rdoff/rdflib.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 1996-2009 The NASM Authors - All Rights Reserved
   4  *   See the file AUTHORS included with the NASM distribution for
   5  *   the specific copyright holders.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following
   9  *   conditions are met:
  10  *
  11  *   * Redistributions of source code must retain the above copyright
  12  *     notice, this list of conditions and the following disclaimer.
  13  *   * Redistributions in binary form must reproduce the above
  14  *     copyright notice, this list of conditions and the following
  15  *     disclaimer in the documentation and/or other materials provided
  16  *     with the distribution.
  17  *
  18  *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  19  *     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  20  *     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  21  *     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22  *     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  23  *     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  *     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25  *     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  26  *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  *     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28  *     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29  *     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  30  *     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  *
  32  * ----------------------------------------------------------------------- */
  33
  34 /* rdflib - manipulate RDOFF library files (.rdl) */
  35
  36 /*
  37  * an rdoff library is simply a sequence of RDOFF object files, each
  38  * preceded by the name of the module, an ASCII string of up to 255
  39  * characters, terminated by a zero.
  40  *
  41  * When a library is being created, special signature block is placed
  42  * in the beginning of the file. It is a string 'RDLIB' followed by a
  43  * version number, then int32_t content size and a int32_t time stamp.
  44  * The module name of the signature block is '.sig'.
  45  *
  46  *
  47  * There may be an optional directory placed on the end of the file.
  48  * The format of the directory will be 'RDLDD' followed by a version
  49  * number, followed by the length of the directory, and then the
  50  * directory, the format of which has not yet been designed.
  51  * The module name of the directory must be '.dir'.
  52  *
  53  * All module names beginning with '.' are reserved for possible future
  54  * extensions. The linker ignores all such modules, assuming they have
  55  * the format of a six uint8_t type & version identifier followed by int32_t
  56  * content size, followed by data.
  57  */
  58
  59 #include "compiler.h"
  60 #include "rdfutils.h"
  61
  62 #include <stdio.h>
  63 #include <stdlib.h>
  64 #include <errno.h>
  65 #include <time.h>
  66
  67 /* functions supported:
  68  *   create a library   (no extra operands required)
  69  *   add a module from a library (requires filename and name to give mod.)
  70  *   replace a module in a library (requires given name and filename)
  71  *   delete a module from a library (requires given name)
  72  *   extract a module from the library (requires given name and filename)
  73  *   list modules
  74  */
  75
  76 const char *usage =
  77     "usage:\n"
  78     "  rdflib x libname [extra operands]\n\n"
  79     "  where x is one of:\n"
  80     "    c - create library\n"
  81     "    a - add module (operands = filename module-name)\n"
  82     "    x - extract               (module-name filename)\n"
  83     "    r - replace               (module-name filename)\n"
  84     "    d - delete                (module-name)\n" "    t - list\n";
  85
  86 /* Library signature */
  87 const char *rdl_signature = "RDLIB2", *sig_modname = ".sig";
  88
  89 char **_argv;
  90
  91 #define _ENDIANNESS 0           /* 0 for little, 1 for big */
  92
  93 static void int32_ttolocal(int32_t *l)
  94 {
  95 #if _ENDIANNESS
  96     uint8_t t;
  97     uint8_t *p = (uint8_t *)l;
  98
  99     t = p[0];
 100     p[0] = p[3];
 101     p[3] = t;
 102     t = p[1];
 103     p[1] = p[2];
 104     p[2] = p[1];
 105 #else
 106     (void)l;             /* placate optimizers */
 107 #endif
 108 }
 109
 110 static char copybytes(FILE * fp, FILE * fp2, int n)
 111 {
 112     int i, t = 0;
 113
 114     for (i = 0; i < n; i++) {
 115         t = fgetc(fp);
 116         if (t == EOF) {
 117             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 118                     _argv[2]);
 119             exit(1);
 120         }
 121         if (fp2)
 122             if (fputc(t, fp2) == EOF) {
 123                 fprintf(stderr, "rdflib: write error\n");
 124                 exit(1);
 125             }
 126     }
 127     return (char)t;             /* return last char read */
 128 }
 129
 130 static int32_t copyint32_t(FILE * fp, FILE * fp2)
 131 {
 132     int32_t l;
 133     int i, t;
 134     uint8_t *p = (uint8_t *)&l;
 135
 136     for (i = 0; i < 4; i++) {   /* skip magic no */
 137         t = fgetc(fp);
 138         if (t == EOF) {
 139             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 140                     _argv[2]);
 141             exit(1);
 142         }
 143         if (fp2)
 144             if (fputc(t, fp2) == EOF) {
 145                 fprintf(stderr, "rdflib: write error\n");
 146                 exit(1);
 147             }
 148         *p++ = t;
 149     }
 150     int32_ttolocal(&l);
 151     return l;
 152 }
 153
 154 int main(int argc, char **argv)
 155 {
 156     FILE *fp, *fp2 = NULL, *fptmp;
 157     char *p, buf[256], c;
 158     int i;
 159     int32_t l;
 160     time_t t;
 161     char rdbuf[10];
 162
 163     _argv = argv;
 164
 165     if (argc < 3 || !strncmp(argv[1], "-h", 2)
 166         || !strncmp(argv[1], "--h", 3)) {
 167         fputs(usage, stdout);
 168         exit(1);
 169     }
 170
 171     rdoff_init();
 172
 173     switch (argv[1][0]) {
 174     case 'c':                  /* create library */
 175         fp = fopen(argv[2], "wb");
 176         if (!fp) {
 177             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 178             perror("rdflib");
 179             exit(1);
 180         }
 181         nasm_write(sig_modname, strlen(sig_modname) + 1, fp);
 182         nasm_write(rdl_signature, strlen(rdl_signature), fp);
 183         t = time(NULL);
 184         fwriteint32_t(t, fp);
 185         fclose(fp);
 186         break;
 187
 188     case 'a':                  /* add module */
 189         if (argc < 5) {
 190             fprintf(stderr, "rdflib: required parameter missing\n");
 191             exit(1);
 192         }
 193         fp = fopen(argv[2], "ab");
 194         if (!fp) {
 195             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 196             perror("rdflib");
 197             exit(1);
 198         }
 199
 200         fp2 = fopen(argv[3], "rb");
 201         if (!fp2) {
 202             fprintf(stderr, "rdflib: could not open '%s'\n", argv[3]);
 203             perror("rdflib");
 204             exit(1);
 205         }
 206
 207         p = argv[4];
 208         do {
 209             if (fputc(*p, fp) == EOF) {
 210                 fprintf(stderr, "rdflib: write error\n");
 211                 exit(1);
 212             }
 213         } while (*p++);
 214
 215         while (!feof(fp2)) {
 216             i = fgetc(fp2);
 217             if (i == EOF) {
 218                 break;
 219             }
 220
 221             if (fputc(i, fp) == EOF) {
 222                 fprintf(stderr, "rdflib: write error\n");
 223                 exit(1);
 224             }
 225         }
 226         fclose(fp2);
 227         fclose(fp);
 228         break;
 229
 230     case 'x':
 231         if (argc < 5) {
 232             fprintf(stderr, "rdflib: required parameter missing\n");
 233             exit(1);
 234         }
 235         break;
 236     case 't':
 237         fp = fopen(argv[2], "rb");
 238         if (!fp) {
 239             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 240             perror("rdflib");
 241             exit(1);
 242         }
 243
 244         fp2 = NULL;
 245         while (!feof(fp)) {
 246             /* read name */
 247             p = buf;
 248             while ((*(p++) = (char)fgetc(fp)))
 249                 if (feof(fp))
 250                     break;
 251
 252             if (feof(fp))
 253                 break;
 254
 255             fp2 = NULL;
 256             if (argv[1][0] == 'x') {
 257                 /* check against desired name */
 258                 if (!strcmp(buf, argv[3])) {
 259                     fp2 = fopen(argv[4], "wb");
 260                     if (!fp2) {
 261                         fprintf(stderr, "rdflib: could not open '%s'\n",
 262                                 argv[4]);
 263                         perror("rdflib");
 264                         exit(1);
 265                     }
 266                 }
 267             } else
 268                 printf("%-40s ", buf);
 269
 270             /* step over the RDOFF file, extracting type information for
 271              * the listing, and copying it if fp2 != NULL */
 272
 273             if (buf[0] == '.') {
 274
 275                 if (argv[1][0] == 't')
 276                     for (i = 0; i < 6; i++)
 277                         printf("%c", copybytes(fp, fp2, 1));
 278                 else
 279                     copybytes(fp, fp2, 6);
 280
 281                 l = copyint32_t(fp, fp2);
 282
 283                 if (argv[1][0] == 't')
 284                     printf("   %"PRId32" bytes content\n", l);
 285
 286                 copybytes(fp, fp2, l);
 287             } else if ((c = copybytes(fp, fp2, 6)) >= '2') {    /* version 2 or above */
 288                 l = copyint32_t(fp, fp2);
 289
 290                 if (argv[1][0] == 't')
 291                     printf("RDOFF%c   %"PRId32" bytes content\n", c, l);
 292                 copybytes(fp, fp2, l);  /* entire object */
 293             } else {
 294                 if (argv[1][0] == 't')
 295                     printf("RDOFF1\n");
 296                 /*
 297                  * version 1 object, so we don't have an object content
 298                  * length field.
 299                  */
 300                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* header */
 301                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* text */
 302                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* data */
 303             }
 304
 305             if (fp2)
 306                 break;
 307         }
 308         fclose(fp);
 309         if (fp2)
 310             fclose(fp2);
 311         else if (argv[1][0] == 'x') {
 312             fprintf(stderr, "rdflib: module '%s' not found in '%s'\n",
 313                     argv[3], argv[2]);
 314             exit(1);
 315         }
 316         break;
 317
 318     case 'r':                  /* replace module */
 319         argc--;
 320         /* fall through */
 321
 322     case 'd':                  /* delete module */
 323         if (argc < 4) {
 324             fprintf(stderr, "rdflib: required parameter missing\n");
 325             exit(1);
 326         }
 327
 328         fp = fopen(argv[2], "rb");
 329         if (!fp) {
 330             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 331             perror("rdflib");
 332             exit(1);
 333         }
 334
 335         if (argv[1][0] == 'r') {
 336             fp2 = fopen(argv[4], "rb");
 337             if (!fp2) {
 338                 fprintf(stderr, "rdflib: could not open '%s'\n", argv[4]);
 339                 perror("rdflib");
 340                 exit(1);
 341             }
 342         }
 343
 344         fptmp = tmpfile();
 345         if (!fptmp) {
 346             fprintf(stderr, "rdflib: could not open temporary file\n");
 347             perror("rdflib");
 348             exit(1);
 349         }
 350
 351         /* copy library into temporary file */
 352         fseek(fp, 0, SEEK_END); /* get file length */
 353         l = ftell(fp);
 354         fseek(fp, 0, SEEK_SET);
 355         copybytes(fp, fptmp, l);
 356         rewind(fptmp);
 357         if (freopen(argv[2], "wb", fp) == NULL) {
 358             fprintf(stderr, "rdflib: could not reopen '%s'\n", argv[2]);
 359             perror("rdflib");
 360             exit(1);
 361         }
 362
 363         while (!feof(fptmp)) {
 364             /* read name */
 365             p = buf;
 366             while ((*(p++) = (char)fgetc(fptmp)))
 367                 if (feof(fptmp))
 368                     break;
 369
 370             if (feof(fptmp))
 371                 break;
 372
 373             /* check against desired name */
 374             if (!strcmp(buf, argv[3])) {
 375                 if (fread(p = rdbuf, 1, sizeof(rdbuf), fptmp) < 10) {
 376                     nasm_fatal("short read on input");
 377                 }
 378                 l = *(int32_t *)(p + 6);
 379                 fseek(fptmp, l, SEEK_CUR);
 380                 break;
 381             } else {
 382                 nasm_write(buf, strlen(buf) + 1, fp);    /* module name */
 383                 if ((c = copybytes(fptmp, fp, 6)) >= '2') {
 384                     l = copyint32_t(fptmp, fp);    /* version 2 or above */
 385                     copybytes(fptmp, fp, l);    /* entire object */
 386                 }
 387             }
 388         }
 389
 390         if (argv[1][0] == 'r') {
 391             /* copy new module into library */
 392             p = argv[3];
 393             do {
 394                 if (fputc(*p, fp) == EOF) {
 395                     fprintf(stderr, "rdflib: write error\n");
 396                     exit(1);
 397                 }
 398             } while (*p++);
 399
 400             while (!feof(fp2)) {
 401                 i = fgetc(fp2);
 402                 if (i == EOF) {
 403                     break;
 404                 }
 405                 if (fputc(i, fp) == EOF) {
 406                     fprintf(stderr, "rdflib: write error\n");
 407                     exit(1);
 408                 }
 409             }
 410             fclose(fp2);
 411         }
 412
 413         /* copy rest of library if any */
 414         while (!feof(fptmp)) {
 415             i = fgetc(fptmp);
 416             if (i == EOF) {
 417                 break;
 418             }
 419
 420             if (fputc(i, fp) == EOF) {
 421                 fprintf(stderr, "rdflib: write error\n");
 422                 exit(1);
 423             }
 424         }
 425
 426         fclose(fp);
 427         fclose(fptmp);
 428         break;
 429
 430     default:
 431         fprintf(stderr, "rdflib: command '%c' not recognized\n",
 432                 argv[1][0]);
 433         exit(1);
 434     }
 435     return 0;
 436 }