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
  61 #include <stdio.h>
  62 #include <stdlib.h>
  63 #include <errno.h>
  64 #include <string.h>
  65 #include <inttypes.h>
  66 #include <time.h>
  67 #include <inttypes.h>
  68
  69 /* functions supported:
  70  *   create a library   (no extra operands required)
  71  *   add a module from a library (requires filename and name to give mod.)
  72  *   replace a module in a library (requires given name and filename)
  73  *   delete a module from a library (requires given name)
  74  *   extract a module from the library (requires given name and filename)
  75  *   list modules
  76  */
  77
  78 const char *usage =
  79     "usage:\n"
  80     "  rdflib x libname [extra operands]\n\n"
  81     "  where x is one of:\n"
  82     "    c - create library\n"
  83     "    a - add module (operands = filename module-name)\n"
  84     "    x - extract               (module-name filename)\n"
  85     "    r - replace               (module-name filename)\n"
  86     "    d - delete                (module-name)\n" "    t - list\n";
  87
  88 /* Library signature */
  89 const char *rdl_signature = "RDLIB2", *sig_modname = ".sig";
  90
  91 char **_argv;
  92
  93 #define _ENDIANNESS 0           /* 0 for little, 1 for big */
  94
  95 static void int32_ttolocal(int32_t *l)
  96 {
  97 #if _ENDIANNESS
  98     uint8_t t;
  99     uint8_t *p = (uint8_t *)l;
 100
 101     t = p[0];
 102     p[0] = p[3];
 103     p[3] = t;
 104     t = p[1];
 105     p[1] = p[2];
 106     p[2] = p[1];
 107 #else
 108     (void)l;             /* placate optimizers */
 109 #endif
 110 }
 111
 112 char copybytes(FILE * fp, FILE * fp2, int n)
 113 {
 114     int i, t = 0;
 115
 116     for (i = 0; i < n; i++) {
 117         t = fgetc(fp);
 118         if (t == EOF) {
 119             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 120                     _argv[2]);
 121             exit(1);
 122         }
 123         if (fp2)
 124             if (fputc(t, fp2) == EOF) {
 125                 fprintf(stderr, "rdflib: write error\n");
 126                 exit(1);
 127             }
 128     }
 129     return (char)t;             /* return last char read */
 130 }
 131
 132 int32_t copyint32_t(FILE * fp, FILE * fp2)
 133 {
 134     int32_t l;
 135     int i, t;
 136     uint8_t *p = (uint8_t *)&l;
 137
 138     for (i = 0; i < 4; i++) {   /* skip magic no */
 139         t = fgetc(fp);
 140         if (t == EOF) {
 141             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 142                     _argv[2]);
 143             exit(1);
 144         }
 145         if (fp2)
 146             if (fputc(t, fp2) == EOF) {
 147                 fprintf(stderr, "rdflib: write error\n");
 148                 exit(1);
 149             }
 150         *p++ = t;
 151     }
 152     int32_ttolocal(&l);
 153     return l;
 154 }
 155
 156 int main(int argc, char **argv)
 157 {
 158     FILE *fp, *fp2 = NULL, *fptmp;
 159     char *p, buf[256], c;
 160     int i;
 161     int32_t l;
 162     time_t t;
 163     char rdbuf[10];
 164
 165     _argv = argv;
 166
 167     if (argc < 3 || !strncmp(argv[1], "-h", 2)
 168         || !strncmp(argv[1], "--h", 3)) {
 169         fputs(usage, stdout);
 170         exit(1);
 171     }
 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         fwrite(sig_modname, 1, strlen(sig_modname) + 1, fp);
 182         fwrite(rdl_signature, 1, strlen(rdl_signature), fp);
 183         l = sizeof(t = time(NULL));
 184         fwrite(&l, sizeof(l), 1, fp);
 185         fwrite(&t, 1, l, fp);
 186         fclose(fp);
 187         break;
 188
 189     case 'a':                  /* add module */
 190         if (argc < 5) {
 191             fprintf(stderr, "rdflib: required parameter missing\n");
 192             exit(1);
 193         }
 194         fp = fopen(argv[2], "ab");
 195         if (!fp) {
 196             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 197             perror("rdflib");
 198             exit(1);
 199         }
 200
 201         fp2 = fopen(argv[3], "rb");
 202         if (!fp2) {
 203             fprintf(stderr, "rdflib: could not open '%s'\n", argv[3]);
 204             perror("rdflib");
 205             exit(1);
 206         }
 207
 208         p = argv[4];
 209         do {
 210             if (fputc(*p, fp) == EOF) {
 211                 fprintf(stderr, "rdflib: write error\n");
 212                 exit(1);
 213             }
 214         } while (*p++);
 215
 216         while (!feof(fp2)) {
 217             i = fgetc(fp2);
 218             if (i == EOF) {
 219                 break;
 220             }
 221
 222             if (fputc(i, fp) == EOF) {
 223                 fprintf(stderr, "rdflib: write error\n");
 224                 exit(1);
 225             }
 226         }
 227         fclose(fp2);
 228         fclose(fp);
 229         break;
 230
 231     case 'x':
 232         if (argc < 5) {
 233             fprintf(stderr, "rdflib: required parameter missing\n");
 234             exit(1);
 235         }
 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     case 'd':                  /* delete module */
 321         if (argc < 4) {
 322             fprintf(stderr, "rdflib: required parameter missing\n");
 323             exit(1);
 324         }
 325
 326         fp = fopen(argv[2], "rb");
 327         if (!fp) {
 328             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 329             perror("rdflib");
 330             exit(1);
 331         }
 332
 333         if (argv[1][0] == 'r') {
 334             fp2 = fopen(argv[4], "rb");
 335             if (!fp2) {
 336                 fprintf(stderr, "rdflib: could not open '%s'\n", argv[4]);
 337                 perror("rdflib");
 338                 exit(1);
 339             }
 340         }
 341
 342         fptmp = tmpfile();
 343         if (!fptmp) {
 344             fprintf(stderr, "rdflib: could not open temporary file\n");
 345             perror("rdflib");
 346             exit(1);
 347         }
 348
 349         /* copy library into temporary file */
 350         fseek(fp, 0, SEEK_END); /* get file length */
 351         l = ftell(fp);
 352         fseek(fp, 0, SEEK_SET);
 353         copybytes(fp, fptmp, l);
 354         rewind(fptmp);
 355         freopen(argv[2], "wb", fp);
 356
 357         while (!feof(fptmp)) {
 358             /* read name */
 359             p = buf;
 360             while ((*(p++) = (char)fgetc(fptmp)))
 361                 if (feof(fptmp))
 362                     break;
 363
 364             if (feof(fptmp))
 365                 break;
 366
 367             /* check against desired name */
 368             if (!strcmp(buf, argv[3])) {
 369                 fread(p = rdbuf, 1, sizeof(rdbuf), fptmp);
 370                 l = *(int32_t *)(p + 6);
 371                 fseek(fptmp, l, SEEK_CUR);
 372                 break;
 373             } else {
 374                 fwrite(buf, 1, strlen(buf) + 1, fp);    /* module name */
 375                 if ((c = copybytes(fptmp, fp, 6)) >= '2') {
 376                     l = copyint32_t(fptmp, fp);    /* version 2 or above */
 377                     copybytes(fptmp, fp, l);    /* entire object */
 378                 }
 379             }
 380         }
 381
 382         if (argv[1][0] == 'r') {
 383             /* copy new module into library */
 384             p = argv[3];
 385             do {
 386                 if (fputc(*p, fp) == EOF) {
 387                     fprintf(stderr, "rdflib: write error\n");
 388                     exit(1);
 389                 }
 390             } while (*p++);
 391
 392             while (!feof(fp2)) {
 393                 i = fgetc(fp2);
 394                 if (i == EOF) {
 395                     break;
 396                 }
 397                 if (fputc(i, fp) == EOF) {
 398                     fprintf(stderr, "rdflib: write error\n");
 399                     exit(1);
 400                 }
 401             }
 402             fclose(fp2);
 403         }
 404
 405         /* copy rest of library if any */
 406         while (!feof(fptmp)) {
 407             i = fgetc(fptmp);
 408             if (i == EOF) {
 409                 break;
 410             }
 411
 412             if (fputc(i, fp) == EOF) {
 413                 fprintf(stderr, "rdflib: write error\n");
 414                 exit(1);
 415             }
 416         }
 417
 418         fclose(fp);
 419         fclose(fptmp);
 420         break;
 421
 422     default:
 423         fprintf(stderr, "rdflib: command '%c' not recognized\n",
 424                 argv[1][0]);
 425         exit(1);
 426     }
 427     return 0;
 428 }