tools/bin2res.c

   1 /************************************************
   2  *
   3  * Converting binary resources from/to *.rc files
   4  *
   5  * Copyright 1999 Juergen Schmied
   6  * Copyright 2003 Dimitrie O. Paun
   7  *
   8  * This library is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This library is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this library; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  */
  22
  23 #include "config.h"
  24 #include "wine/port.h"
  25
  26 #include <signal.h>
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <ctype.h>
  30 #include <string.h>
  31 #include <sys/stat.h>
  32 #include <limits.h>
  33 #ifdef HAVE_SYS_PARAM_H
  34 # include <sys/param.h>
  35 #endif
  36
  37 static const char *clean_file;
  38
  39 static const char* help =
  40         "Usage: bin2res [OPTIONS] <rsrc.rc>\n"
  41         "  -a archive binaries into the <rsrc.rc> file\n"
  42         "  -x extract binaries from the <rsrc.rc> file\n"
  43         "  -i <filename> archive the named file into the <rsrc.rc> file\n"
  44         "  -o <filename> extract the named file from the <rsrc.rc> file\n"
  45         "  -f force processing of older resources\n"
  46         "  -v causes the command to be verbous during processing\n"
  47         "  -h print this help screen and exit\n"
  48         "\n"
  49         "This tool allows the insertion/extractions of embedded binary\n"
  50         "resources to/from .rc files, for storage within the cvs tree.\n"
  51         "This is accomplished by placing a magic marker in a comment\n"
  52         "just above the resource. The marker consists of the BINRES\n"
  53         "string followed by the file name. For example, to insert a\n"
  54         "brand new binary resource in a .rc file, place the marker\n"
  55         "above empty brackets:\n"
  56         "    /* BINRES idb_std_small.bmp */\n"
  57         "    IDB_STD_SMALL BITMAP idb_std_small.bmp\n"
  58         "    /* {\n"
  59         "    } */\n"
  60         "To merge the binary resources into the .rc file, run:\n"
  61         "   bin2res -a myrsrc.rc\n"
  62         "Only resources that are newer than the .rc are processed.\n"
  63         "To extract the binary resources from the .rc file, run:\n"
  64         "  bin2res -x myrsrc.rc\n"
  65         "Binary files newer than the .rc file are not overwritten.\n"
  66         "\n"
  67         "To force processing of all resources, use the -f flag.\n"
  68         "To process a particular file, use the -i/-o options.\n";
  69
  70 static void usage(void)
  71 {
  72     printf(help);
  73     exit(1);
  74 }
  75
  76 static void cleanup_files(void)
  77 {
  78     if (clean_file) unlink( clean_file );
  79 }
  80
  81 static void exit_on_signal( int sig )
  82 {
  83     exit(1);  /* this will call the atexit functions */
  84 }
  85
  86 static int insert_hexdump (FILE* outfile, FILE* infile)
  87 {
  88     int i, c;
  89
  90     fprintf (outfile, "{\n '");
  91     for (i = 0; (c = fgetc(infile)) != EOF; i++)
  92     {
  93         if (i && (i % 16) == 0) fprintf (outfile, "'\n '");
  94         if (i % 16)  fprintf (outfile, " ");
  95         fprintf(outfile, "%02X", c);
  96     }
  97     fprintf (outfile, "'\n}");
  98
  99     return 1;
 100 }
 101
 102 static int hex2bin(char c)
 103 {
 104     if (!isxdigit(c)) return -1024;
 105     if (isdigit(c)) return c - '0';
 106     return toupper(c) - 'A' + 10;
 107 }
 108
 109 static int extract_hexdump (FILE* outfile, FILE* infile)
 110 {
 111     int byte, c;
 112
 113     while ( (c = fgetc(infile)) != EOF && c != '}')
 114     {
 115         if (isspace(c) || c == '\'') continue;
 116         byte = 16 * hex2bin(c);
 117         c = fgetc(infile);
 118         if (c == EOF) return 0;
 119         byte += hex2bin(c);
 120         if (byte < 0) return 0;
 121         fputc(byte, outfile);
 122     }
 123     return 1;
 124 }
 125
 126 static const char* parse_marker(const char *line, time_t* last_updated)
 127 {
 128     static char res_file_name[PATH_MAX], *rpos, *wpos;
 129     struct stat st;
 130
 131     if (!(rpos = strstr(line, "BINRES"))) return 0;
 132     for (rpos += 6; *rpos && isspace(*rpos); rpos++) /**/;
 133     for (wpos = res_file_name; *rpos && !isspace(*rpos); ) *wpos++ = *rpos++;
 134     *wpos = 0;
 135
 136     *last_updated = (stat(res_file_name, &st) < 0) ? 0 : st.st_mtime;
 137
 138     return res_file_name;
 139 }
 140
 141 static int process_resources(const char* input_file_name, const char* specific_file_name,
 142                       int inserting, int force_processing, int verbose)
 143 {
 144     char buffer[2048], tmp_file_name[PATH_MAX];
 145     const char *res_file_name;
 146     time_t rc_last_update, res_last_update;
 147     FILE *fin, *fres, *ftmp = 0;
 148     struct stat st;
 149     int fd, c;
 150
 151     if (!(fin = fopen(input_file_name, "r"))) return 0;
 152     if (stat(input_file_name, &st) < 0) return 0;
 153     rc_last_update = st.st_mtime;
 154
 155     if (inserting)
 156     {
 157         strcpy(tmp_file_name, input_file_name);
 158         strcat(tmp_file_name, "-XXXXXX.temp");
 159         if ((fd = mkstemps(tmp_file_name, 5)) == -1)
 160         {
 161             strcpy(tmp_file_name, "/tmp/bin2res-XXXXXX.temp");
 162             if ((fd = mkstemps(tmp_file_name, 5)) == -1) return 0;
 163         }
 164         clean_file = tmp_file_name;
 165         if (!(ftmp = fdopen(fd, "w"))) return 0;
 166     }
 167
 168     for (c = EOF; fgets(buffer, sizeof(buffer), fin); c = EOF)
 169     {
 170         if (inserting) fprintf(ftmp, "%s", buffer);
 171         if (!(res_file_name = parse_marker(buffer, &res_last_update))) continue;
 172         if ( (specific_file_name && strcmp(specific_file_name, res_file_name)) ||
 173              (!force_processing && ((rc_last_update < res_last_update) == !inserting)) )
 174         {
 175             if (verbose) printf("skipping '%s'\n", res_file_name);
 176             continue;
 177         }
 178
 179         if (verbose) printf("processing '%s'\n", res_file_name);
 180         while ( (c = fgetc(fin)) != EOF && c != '{')
 181             if (inserting) fputc(c, ftmp);
 182         if (c == EOF) break;
 183
 184         if (inserting)
 185         {
 186             if (!(fres = fopen(res_file_name, "rb"))) break;
 187             if (!insert_hexdump(ftmp, fres)) break;
 188             while ( (c = fgetc(fin)) != EOF && c != '}') /**/;
 189             fclose(fres);
 190         }
 191         else
 192         {
 193             clean_file = res_file_name;
 194             if (!(fres = fopen(res_file_name, "wb"))) break;
 195             if (!extract_hexdump(fres, fin)) break;
 196             fclose(fres);
 197             clean_file = NULL;
 198         }
 199     }
 200
 201     fclose(fin);
 202
 203     if (inserting)
 204     {
 205         fclose(ftmp);
 206         if (c == EOF)
 207         {
 208             if (rename(tmp_file_name, input_file_name) < 0)
 209             {
 210                 /* try unlinking first, Windows rename is brain-damaged */
 211                 if (unlink(input_file_name) < 0 || rename(tmp_file_name, input_file_name) < 0)
 212                     return 0;
 213             }
 214             clean_file = NULL;
 215         }
 216     }
 217
 218     return c == EOF;
 219 }
 220
 221 int main(int argc, char **argv)
 222 {
 223     int convert_dir = 0, optc;
 224     int force_overwrite = 0, verbose = 0;
 225     const char* input_file_name = 0;
 226     const char* specific_file_name = 0;
 227
 228     atexit( cleanup_files );
 229     signal( SIGTERM, exit_on_signal );
 230     signal( SIGINT, exit_on_signal );
 231 #ifdef SIGHUP
 232     signal( SIGHUP, exit_on_signal );
 233 #endif
 234
 235     while((optc = getopt(argc, argv, "axi:o:fhv")) != EOF)
 236     {
 237         switch(optc)
 238         {
 239         case 'a':
 240         case 'x':
 241             if (convert_dir) usage();
 242             convert_dir = optc;
 243         break;
 244         case 'i':
 245         case 'o':
 246             if (specific_file_name) usage();
 247             specific_file_name = optarg;
 248             optc = ((optc == 'i') ? 'a' : 'x');
 249             if (convert_dir && convert_dir != optc) usage();
 250             convert_dir = optc;
 251         break;
 252         case 'f':
 253             force_overwrite = 1;
 254         break;
 255         case 'v':
 256             verbose = 1;
 257         break;
 258         case 'h':
 259             printf(help);
 260             exit(0);
 261         break;
 262         default:
 263             usage();
 264         }
 265     }
 266
 267     if (optind + 1 != argc) usage();
 268     input_file_name = argv[optind];
 269
 270     if (!convert_dir) usage();
 271
 272     if (!process_resources(input_file_name, specific_file_name,
 273                            convert_dir == 'a', force_overwrite, verbose))
 274     {
 275         perror("Processing failed");
 276         exit(1);
 277     }
 278
 279     return 0;
 280 }