Initial import from FreeBSD RELENG_4:
[dragonfly.git] / contrib / binutils / binutils / resres.c
1 /* resres.c: read_res_file and write_res_file implementation for windres.
2    Copyright 1998, 1999 Free Software Foundation, Inc.
3    Written by Anders Norlander <anorland@hem2.passagen.se>.
4
5    This file is part of GNU Binutils.
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
20    02111-1307, USA.  */
21
22 /* FIXME: This file does not work correctly in a cross configuration.
23    It assumes that it can use fread and fwrite to read and write
24    integers.  It does no swapping.  */
25
26 #include "bfd.h"
27 #include "bucomm.h"
28 #include "libiberty.h"
29 #include "windres.h"
30
31 #include <assert.h>
32 #include <time.h>
33
34 struct res_hdr
35   {
36     unsigned long data_size;
37     unsigned long header_size;
38   };
39
40 static void write_res_directory
41   PARAMS ((const struct res_directory *,
42            const struct res_id *, const struct res_id *,
43            int *, int));
44 static void write_res_resource
45   PARAMS ((const struct res_id *, const struct res_id *,
46            const struct res_resource *, int *));
47 static void write_res_bin
48   PARAMS ((const struct res_resource *, const struct res_id *,
49            const struct res_id *, const struct res_res_info *));
50
51 static void write_res_id PARAMS ((const struct res_id *));
52 static void write_res_info PARAMS ((const struct res_res_info *));
53 static void write_res_data PARAMS ((const void *, size_t, int));
54 static void write_res_header
55   PARAMS ((unsigned long, const struct res_id *, const struct res_id *,
56            const struct res_res_info *));
57
58 static int read_resource_entry PARAMS ((void));
59 static void read_res_data PARAMS ((void *, size_t, int));
60 static void read_res_id PARAMS ((struct res_id *));
61 static unichar *read_unistring PARAMS ((int *));
62 static void skip_null_resource PARAMS ((void));
63
64 static unsigned long get_id_size PARAMS ((const struct res_id *));
65 static void res_align_file PARAMS ((void));
66
67 static void
68   res_add_resource
69   PARAMS ((struct res_resource *, const struct res_id *,
70            const struct res_id *, int, int));
71
72 void
73   res_append_resource
74   PARAMS ((struct res_directory **, struct res_resource *,
75            int, const struct res_id *, int));
76
77 static struct res_directory *resources = NULL;
78
79 static FILE *fres;
80 static const char *filename;
81
82 extern char *program_name;
83
84 /* Read resource file */
85 struct res_directory *
86 read_res_file (fn)
87      const char *fn;
88 {
89   filename = fn;
90   fres = fopen (filename, "rb");
91   if (fres == NULL)
92     fatal ("can't open `%s' for output: %s", filename, strerror (errno));
93
94   skip_null_resource ();
95
96   while (read_resource_entry ())
97     ;
98
99   fclose (fres);
100
101   return resources;
102 }
103
104 /* Write resource file */
105 void
106 write_res_file (fn, resdir)
107      const char *fn;
108      const struct res_directory *resdir;
109 {
110   int language;
111   static const unsigned char sign[] =
112   {0x00, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00,
113    0xFF, 0xFF, 0x00, 0x00, 0xFF, 0xFF, 0x00, 0x00,
114    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
115    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
116   long fpos;
117
118   filename = fn;
119
120   fres = fopen (filename, "wb");
121   if (fres == NULL)
122     fatal ("can't open `%s' for output: %s", filename, strerror (errno));
123
124   /* Write 32 bit resource signature */
125   write_res_data (sign, sizeof (sign), 1);
126
127   /* write resources */
128
129   language = -1;
130   write_res_directory (resdir, (const struct res_id *) NULL,
131                        (const struct res_id *) NULL, &language, 1);
132
133   /* end file on DWORD boundary */
134   fpos = ftell (fres);
135   if (fpos % 4)
136     write_res_data (sign, fpos % 4, 1);
137
138   fclose (fres);
139 }
140
141 /* Read a resource entry, returns 0 when all resources are read */
142 static int
143 read_resource_entry (void)
144 {
145   struct res_id type;
146   struct res_id name;
147   struct res_res_info resinfo;
148   struct res_hdr reshdr;
149   long version;
150   void *buff;
151
152   struct res_resource *r;
153
154   res_align_file ();
155
156   /* Read header */
157   if (fread (&reshdr, sizeof (reshdr), 1, fres) != 1)
158     return 0;
159
160   /* read resource type */
161   read_res_id (&type);
162   /* read resource id */
163   read_res_id (&name);
164
165   res_align_file ();
166
167   /* Read additional resource header */
168   read_res_data (&resinfo.version, sizeof (resinfo.version), 1);
169   read_res_data (&resinfo.memflags, sizeof (resinfo.memflags), 1);
170   read_res_data (&resinfo.language, sizeof (resinfo.language), 1);
171   read_res_data (&version, sizeof (version), 1);
172   read_res_data (&resinfo.characteristics, sizeof (resinfo.characteristics), 1);
173
174   res_align_file ();
175
176   /* Allocate buffer for data */
177   buff = res_alloc (reshdr.data_size);
178   /* Read data */
179   read_res_data (buff, reshdr.data_size, 1);
180   /* Convert binary data to resource */
181   r = bin_to_res (type, buff, reshdr.data_size, 0);
182   r->res_info = resinfo;
183   /* Add resource to resource directory */
184   res_add_resource (r, &type, &name, resinfo.language, 0);
185
186   return 1;
187 }
188
189 /* write resource directory to binary resource file */
190 static void
191 write_res_directory (rd, type, name, language, level)
192      const struct res_directory *rd;
193      const struct res_id *type;
194      const struct res_id *name;
195      int *language;
196      int level;
197 {
198   const struct res_entry *re;
199
200   for (re = rd->entries; re != NULL; re = re->next)
201     {
202       switch (level)
203         {
204         case 1:
205           /* If we're at level 1, the key of this resource is the
206              type.  This normally duplicates the information we have
207              stored with the resource itself, but we need to remember
208              the type if this is a user define resource type.  */
209           type = &re->id;
210           break;
211
212         case 2:
213           /* If we're at level 2, the key of this resource is the name
214              we are going to use in the rc printout. */
215           name = &re->id;
216           break;
217
218         case 3:
219           /* If we're at level 3, then this key represents a language.
220              Use it to update the current language.  */
221           if (!re->id.named
222               && re->id.u.id != (unsigned long) *language
223               && (re->id.u.id & 0xffff) == re->id.u.id)
224             {
225               *language = re->id.u.id;
226             }
227           break;
228
229         default:
230           break;
231         }
232
233       if (re->subdir)
234         write_res_directory (re->u.dir, type, name, language, level + 1);
235       else
236         {
237           if (level == 3)
238             {
239               /* This is the normal case: the three levels are
240                  TYPE/NAME/LANGUAGE.  NAME will have been set at level
241                  2, and represents the name to use.  We probably just
242                  set LANGUAGE, and it will probably match what the
243                  resource itself records if anything.  */
244               write_res_resource (type, name, re->u.res, language);
245             }
246           else
247             {
248               fprintf (stderr, "// Resource at unexpected level %d\n", level);
249               write_res_resource (type, (struct res_id *) NULL, re->u.res,
250                                   language);
251             }
252         }
253     }
254
255 }
256
257 static void
258 write_res_resource (type, name, res, language)
259      const struct res_id *type;
260      const struct res_id *name;
261      const struct res_resource *res;
262      int *language ATTRIBUTE_UNUSED;
263 {
264   int rt;
265
266   switch (res->type)
267     {
268     default:
269       abort ();
270
271     case RES_TYPE_ACCELERATOR:
272       rt = RT_ACCELERATOR;
273       break;
274
275     case RES_TYPE_BITMAP:
276       rt = RT_BITMAP;
277       break;
278
279     case RES_TYPE_CURSOR:
280       rt = RT_CURSOR;
281       break;
282
283     case RES_TYPE_GROUP_CURSOR:
284       rt = RT_GROUP_CURSOR;
285       break;
286
287     case RES_TYPE_DIALOG:
288       rt = RT_DIALOG;
289       break;
290
291     case RES_TYPE_FONT:
292       rt = RT_FONT;
293       break;
294
295     case RES_TYPE_FONTDIR:
296       rt = RT_FONTDIR;
297       break;
298
299     case RES_TYPE_ICON:
300       rt = RT_ICON;
301       break;
302
303     case RES_TYPE_GROUP_ICON:
304       rt = RT_GROUP_ICON;
305       break;
306
307     case RES_TYPE_MENU:
308       rt = RT_MENU;
309       break;
310
311     case RES_TYPE_MESSAGETABLE:
312       rt = RT_MESSAGETABLE;
313       break;
314
315     case RES_TYPE_RCDATA:
316       rt = RT_RCDATA;
317       break;
318
319     case RES_TYPE_STRINGTABLE:
320       rt = RT_STRING;
321       break;
322
323     case RES_TYPE_USERDATA:
324       rt = 0;
325       break;
326
327     case RES_TYPE_VERSIONINFO:
328       rt = RT_VERSION;
329       break;
330     }
331
332   if (rt != 0
333       && type != NULL
334       && (type->named || type->u.id != (unsigned long) rt))
335     {
336       fprintf (stderr, "// Unexpected resource type mismatch: ");
337       res_id_print (stderr, *type, 1);
338       fprintf (stderr, " != %d", rt);
339       abort ();
340     }
341
342   write_res_bin (res, type, name, &res->res_info);
343   return;
344 }
345
346 /* Write a resource in binary resource format */
347 static void
348 write_res_bin (res, type, name, resinfo)
349      const struct res_resource *res;
350      const struct res_id *type;
351      const struct res_id *name;
352      const struct res_res_info *resinfo;
353 {
354   unsigned long datasize = 0;
355   const struct bindata *bin_rep, *data;
356
357   bin_rep = res_to_bin (res, 0);
358   for (data = bin_rep; data != NULL; data = data->next)
359     datasize += data->length;
360
361   write_res_header (datasize, type, name, resinfo);
362
363   for (data = bin_rep; data != NULL; data = data->next)
364     write_res_data (data->data, data->length, 1);
365 }
366
367 /* Get number of bytes needed to store an id in binary format */
368 static unsigned long
369 get_id_size (id)
370      const struct res_id *id;
371 {
372   if (id->named)
373     return sizeof (unichar) * (id->u.n.length + 1);
374   else
375     return sizeof (unichar) * 2;
376 }
377
378 /* Write a resource header */
379 static void
380 write_res_header (datasize, type, name, resinfo)
381      unsigned long datasize;
382      const struct res_id *type;
383      const struct res_id *name;
384      const struct res_res_info *resinfo;
385 {
386   struct res_hdr reshdr;
387   reshdr.data_size = datasize;
388   reshdr.header_size = 24 + get_id_size (type) + get_id_size (name);
389
390   reshdr.header_size = (reshdr.header_size + 3) & ~3;
391
392   res_align_file ();
393   write_res_data (&reshdr, sizeof (reshdr), 1);
394   write_res_id (type);
395   write_res_id (name);
396
397   res_align_file ();
398
399   write_res_info (resinfo);
400   res_align_file ();
401 }
402
403
404 /* Write data to file, abort on failure */
405 static void
406 write_res_data (data, size, count)
407      const void *data;
408      size_t size;
409      int count;
410 {
411   if (fwrite (data, size, count, fres) != (size_t) count)
412     fatal ("%s: could not write to file", filename);
413 }
414
415 /* Read data from file, abort on failure */
416 static void
417 read_res_data (data, size, count)
418      void *data;
419      size_t size;
420      int count;
421 {
422   if (fread (data, size, count, fres) != (size_t) count)
423     fatal ("%s: unexpected end of file", filename);
424 }
425
426 /* Write a resource id */
427 static void
428 write_res_id (id)
429      const struct res_id *id;
430 {
431   if (id->named)
432     {
433       unsigned long len = id->u.n.length;
434       unichar null_term = 0;
435       write_res_data (id->u.n.name, len * sizeof (unichar), 1);
436       write_res_data (&null_term, sizeof (null_term), 1);
437     }
438   else
439     {
440       unsigned short i = 0xFFFF;
441       write_res_data (&i, sizeof (i), 1);
442       i = id->u.id;
443       write_res_data (&i, sizeof (i), 1);
444     }
445 }
446
447 /* Write resource info */
448 static void
449 write_res_info (info)
450      const struct res_res_info *info;
451 {
452   write_res_data (&info->version, sizeof (info->version), 1);
453   write_res_data (&info->memflags, sizeof (info->memflags), 1);
454   write_res_data (&info->language, sizeof (info->language), 1);
455   write_res_data (&info->version, sizeof (info->version), 1);
456   write_res_data (&info->characteristics, sizeof (info->characteristics), 1);
457 }
458
459 /* read a resource identifier */
460 void 
461 read_res_id (id)
462      struct res_id *id;
463 {
464   unsigned short ord;
465   unichar *id_s = NULL;
466   int len;
467
468   read_res_data (&ord, sizeof (ord), 1);
469   if (ord == 0xFFFF)            /* an ordinal id */
470     {
471       read_res_data (&ord, sizeof (ord), 1);
472       id->named = 0;
473       id->u.id = ord;
474     }
475   else
476     /* named id */
477     {
478       if (fseek (fres, -sizeof (ord), SEEK_CUR) != 0)
479         fatal ("%s: %s: could not seek in file", program_name, filename);
480       id_s = read_unistring (&len);
481       id->named = 1;
482       id->u.n.length = len;
483       id->u.n.name = id_s;
484     }
485 }
486
487 /* Read a null terminated UNICODE string */
488 static unichar *
489 read_unistring (len)
490      int *len;
491 {
492   unichar *s;
493   unichar c;
494   unichar *p;
495   int l;
496
497   *len = 0;
498   l = 0;
499
500   /* there are hardly any names longer than 256 characters */
501   p = s = (unichar *) xmalloc (sizeof (unichar) * 256);
502   do
503     {
504       read_res_data (&c, sizeof (c), 1);
505       *p++ = c;
506       if (c != 0)
507         l++;
508     }
509   while (c != 0);
510   *len = l;
511   return s;
512 }
513
514 /* align file on DWORD boundary */
515 static void
516 res_align_file (void)
517 {
518   int pos = ftell (fres);
519   int skip = ((pos + 3) & ~3) - pos;
520   if (fseek (fres, skip, SEEK_CUR) != 0)
521     fatal ("%s: %s: unable to align file", program_name, filename);
522 }
523
524 /* Check if file is a win32 binary resource file, if so
525    skip past the null resource. Returns 0 if successful, -1 on
526    error.
527  */
528 static void
529 skip_null_resource (void)
530 {
531   struct res_hdr reshdr =
532   {0, 0};
533   read_res_data (&reshdr, sizeof (reshdr), 1);
534   if ((reshdr.data_size != 0) || (reshdr.header_size != 0x20))
535     goto skip_err;
536
537   /* Subtract size of HeaderSize and DataSize */
538   if (fseek (fres, reshdr.header_size - 8, SEEK_CUR) != 0)
539     goto skip_err;
540
541   return;
542
543 skip_err:
544   fprintf (stderr, "%s: %s: Not a valid WIN32 resource file\n", program_name,
545            filename);
546   xexit (1);
547 }
548
549 /* Add a resource to resource directory */
550 void
551 res_add_resource (r, type, id, language, dupok)
552      struct res_resource *r;
553      const struct res_id *type;
554      const struct res_id *id;
555      int language;
556      int dupok;
557 {
558   struct res_id a[3];
559
560   a[0] = *type;
561   a[1] = *id;
562   a[2].named = 0;
563   a[2].u.id = language;
564   res_append_resource (&resources, r, 3, a, dupok);
565 }
566
567 /* Append a resource to resource directory.
568    This is just copied from define_resource
569    and modified to add an existing resource.
570  */
571 void
572 res_append_resource (resources, resource, cids, ids, dupok)
573      struct res_directory **resources;
574      struct res_resource *resource;
575      int cids;
576      const struct res_id *ids;
577      int dupok;
578 {
579   struct res_entry *re = NULL;
580   int i;
581
582   assert (cids > 0);
583   for (i = 0; i < cids; i++)
584     {
585       struct res_entry **pp;
586
587       if (*resources == NULL)
588         {
589           static unsigned long timeval;
590
591           /* Use the same timestamp for every resource created in a
592              single run.  */
593           if (timeval == 0)
594             timeval = time (NULL);
595
596           *resources = ((struct res_directory *)
597                         res_alloc (sizeof **resources));
598           (*resources)->characteristics = 0;
599           (*resources)->time = timeval;
600           (*resources)->major = 0;
601           (*resources)->minor = 0;
602           (*resources)->entries = NULL;
603         }
604
605       for (pp = &(*resources)->entries; *pp != NULL; pp = &(*pp)->next)
606         if (res_id_cmp ((*pp)->id, ids[i]) == 0)
607           break;
608
609       if (*pp != NULL)
610         re = *pp;
611       else
612         {
613           re = (struct res_entry *) res_alloc (sizeof *re);
614           re->next = NULL;
615           re->id = ids[i];
616           if ((i + 1) < cids)
617             {
618               re->subdir = 1;
619               re->u.dir = NULL;
620             }
621           else
622             {
623               re->subdir = 0;
624               re->u.res = NULL;
625             }
626
627           *pp = re;
628         }
629
630       if ((i + 1) < cids)
631         {
632           if (!re->subdir)
633             {
634               fprintf (stderr, "%s: ", program_name);
635               res_ids_print (stderr, i, ids);
636               fprintf (stderr, ": expected to be a directory\n");
637               xexit (1);
638             }
639
640           resources = &re->u.dir;
641         }
642     }
643
644   if (re->subdir)
645     {
646       fprintf (stderr, "%s: ", program_name);
647       res_ids_print (stderr, cids, ids);
648       fprintf (stderr, ": expected to be a leaf\n");
649       xexit (1);
650     }
651
652   if (re->u.res != NULL)
653     {
654       if (dupok)
655         return;
656
657       fprintf (stderr, "%s: warning: ", program_name);
658       res_ids_print (stderr, cids, ids);
659       fprintf (stderr, ": duplicate value\n");
660     }
661
662   re->u.res = resource;
663 }