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 <string.h>
  66 #include <time.h>
  67
  68 /* functions supported:
  69  *   create a library   (no extra operands required)
  70  *   add a module from a library (requires filename and name to give mod.)
  71  *   replace a module in a library (requires given name and filename)
  72  *   delete a module from a library (requires given name)
  73  *   extract a module from the library (requires given name and filename)
  74  *   list modules
  75  */
  76
  77 const char *usage =
  78     "usage:\n"
  79     "  rdflib x libname [extra operands]\n\n"
  80     "  where x is one of:\n"
  81     "    c - create library\n"
  82     "    a - add module (operands = filename module-name)\n"
  83     "    x - extract               (module-name filename)\n"
  84     "    r - replace               (module-name filename)\n"
  85     "    d - delete                (module-name)\n" "    t - list\n";
  86
  87 /* Library signature */
  88 const char *rdl_signature = "RDLIB2", *sig_modname = ".sig";
  89
  90 char **_argv;
  91
  92 #define _ENDIANNESS 0           /* 0 for little, 1 for big */
  93
  94 static void int32_ttolocal(int32_t *l)
  95 {
  96 #if _ENDIANNESS
  97     uint8_t t;
  98     uint8_t *p = (uint8_t *)l;
  99
 100     t = p[0];
 101     p[0] = p[3];
 102     p[3] = t;
 103     t = p[1];
 104     p[1] = p[2];
 105     p[2] = p[1];
 106 #else
 107     (void)l;             /* placate optimizers */
 108 #endif
 109 }
 110
 111 static char copybytes(FILE * fp, FILE * fp2, int n)
 112 {
 113     int i, t = 0;
 114
 115     for (i = 0; i < n; i++) {
 116         t = fgetc(fp);
 117         if (t == EOF) {
 118             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 119                     _argv[2]);
 120             exit(1);
 121         }
 122         if (fp2)
 123             if (fputc(t, fp2) == EOF) {
 124                 fprintf(stderr, "rdflib: write error\n");
 125                 exit(1);
 126             }
 127     }
 128     return (char)t;             /* return last char read */
 129 }
 130
 131 static int32_t copyint32_t(FILE * fp, FILE * fp2)
 132 {
 133     int32_t l;
 134     int i, t;
 135     uint8_t *p = (uint8_t *)&l;
 136
 137     for (i = 0; i < 4; i++) {   /* skip magic no */
 138         t = fgetc(fp);
 139         if (t == EOF) {
 140             fprintf(stderr, "rdflib: premature end of file in '%s'\n",
 141                     _argv[2]);
 142             exit(1);
 143         }
 144         if (fp2)
 145             if (fputc(t, fp2) == EOF) {
 146                 fprintf(stderr, "rdflib: write error\n");
 147                 exit(1);
 148             }
 149         *p++ = t;
 150     }
 151     int32_ttolocal(&l);
 152     return l;
 153 }
 154
 155 int main(int argc, char **argv)
 156 {
 157     FILE *fp, *fp2 = NULL, *fptmp;
 158     char *p, buf[256], c;
 159     int i;
 160     int32_t l;
 161     time_t t;
 162     char rdbuf[10];
 163
 164     _argv = argv;
 165
 166     if (argc < 3 || !strncmp(argv[1], "-h", 2)
 167         || !strncmp(argv[1], "--h", 3)) {
 168         fputs(usage, stdout);
 169         exit(1);
 170     }
 171
 172     rdoff_init();
 173
 174     switch (argv[1][0]) {
 175     case 'c':                  /* create library */
 176         fp = fopen(argv[2], "wb");
 177         if (!fp) {
 178             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 179             perror("rdflib");
 180             exit(1);
 181         }
 182         nasm_write(sig_modname, strlen(sig_modname) + 1, fp);
 183         nasm_write(rdl_signature, strlen(rdl_signature), fp);
 184         t = time(NULL);
 185         fwriteint32_t(t, 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         break;
 237     case 't':
 238         fp = fopen(argv[2], "rb");
 239         if (!fp) {
 240             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 241             perror("rdflib");
 242             exit(1);
 243         }
 244
 245         fp2 = NULL;
 246         while (!feof(fp)) {
 247             /* read name */
 248             p = buf;
 249             while ((*(p++) = (char)fgetc(fp)))
 250                 if (feof(fp))
 251                     break;
 252
 253             if (feof(fp))
 254                 break;
 255
 256             fp2 = NULL;
 257             if (argv[1][0] == 'x') {
 258                 /* check against desired name */
 259                 if (!strcmp(buf, argv[3])) {
 260                     fp2 = fopen(argv[4], "wb");
 261                     if (!fp2) {
 262                         fprintf(stderr, "rdflib: could not open '%s'\n",
 263                                 argv[4]);
 264                         perror("rdflib");
 265                         exit(1);
 266                     }
 267                 }
 268             } else
 269                 printf("%-40s ", buf);
 270
 271             /* step over the RDOFF file, extracting type information for
 272              * the listing, and copying it if fp2 != NULL */
 273
 274             if (buf[0] == '.') {
 275
 276                 if (argv[1][0] == 't')
 277                     for (i = 0; i < 6; i++)
 278                         printf("%c", copybytes(fp, fp2, 1));
 279                 else
 280                     copybytes(fp, fp2, 6);
 281
 282                 l = copyint32_t(fp, fp2);
 283
 284                 if (argv[1][0] == 't')
 285                     printf("   %"PRId32" bytes content\n", l);
 286
 287                 copybytes(fp, fp2, l);
 288             } else if ((c = copybytes(fp, fp2, 6)) >= '2') {    /* version 2 or above */
 289                 l = copyint32_t(fp, fp2);
 290
 291                 if (argv[1][0] == 't')
 292                     printf("RDOFF%c   %"PRId32" bytes content\n", c, l);
 293                 copybytes(fp, fp2, l);  /* entire object */
 294             } else {
 295                 if (argv[1][0] == 't')
 296                     printf("RDOFF1\n");
 297                 /*
 298                  * version 1 object, so we don't have an object content
 299                  * length field.
 300                  */
 301                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* header */
 302                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* text */
 303                 copybytes(fp, fp2, copyint32_t(fp, fp2));  /* data */
 304             }
 305
 306             if (fp2)
 307                 break;
 308         }
 309         fclose(fp);
 310         if (fp2)
 311             fclose(fp2);
 312         else if (argv[1][0] == 'x') {
 313             fprintf(stderr, "rdflib: module '%s' not found in '%s'\n",
 314                     argv[3], argv[2]);
 315             exit(1);
 316         }
 317         break;
 318
 319     case 'r':                  /* replace module */
 320         argc--;
 321         /* fall through */
 322
 323     case 'd':                  /* delete module */
 324         if (argc < 4) {
 325             fprintf(stderr, "rdflib: required parameter missing\n");
 326             exit(1);
 327         }
 328
 329         fp = fopen(argv[2], "rb");
 330         if (!fp) {
 331             fprintf(stderr, "rdflib: could not open '%s'\n", argv[2]);
 332             perror("rdflib");
 333             exit(1);
 334         }
 335
 336         if (argv[1][0] == 'r') {
 337             fp2 = fopen(argv[4], "rb");
 338             if (!fp2) {
 339                 fprintf(stderr, "rdflib: could not open '%s'\n", argv[4]);
 340                 perror("rdflib");
 341                 exit(1);
 342             }
 343         }
 344
 345         fptmp = tmpfile();
 346         if (!fptmp) {
 347             fprintf(stderr, "rdflib: could not open temporary file\n");
 348             perror("rdflib");
 349             exit(1);
 350         }
 351
 352         /* copy library into temporary file */
 353         fseek(fp, 0, SEEK_END); /* get file length */
 354         l = ftell(fp);
 355         fseek(fp, 0, SEEK_SET);
 356         copybytes(fp, fptmp, l);
 357         rewind(fptmp);
 358         if (freopen(argv[2], "wb", fp) == NULL) {
 359             fprintf(stderr, "rdflib: could not reopen '%s'\n", argv[2]);
 360             perror("rdflib");
 361             exit(1);
 362         }
 363
 364         while (!feof(fptmp)) {
 365             /* read name */
 366             p = buf;
 367             while ((*(p++) = (char)fgetc(fptmp)))
 368                 if (feof(fptmp))
 369                     break;
 370
 371             if (feof(fptmp))
 372                 break;
 373
 374             /* check against desired name */
 375             if (!strcmp(buf, argv[3])) {
 376                 if (fread(p = rdbuf, 1, sizeof(rdbuf), fptmp) < 10) {
 377                     nasm_fatal("short read on input");
 378                 }
 379                 l = *(int32_t *)(p + 6);
 380                 fseek(fptmp, l, SEEK_CUR);
 381                 break;
 382             } else {
 383                 nasm_write(buf, strlen(buf) + 1, fp);    /* module name */
 384                 if ((c = copybytes(fptmp, fp, 6)) >= '2') {
 385                     l = copyint32_t(fptmp, fp);    /* version 2 or above */
 386                     copybytes(fptmp, fp, l);    /* entire object */
 387                 }
 388             }
 389         }
 390
 391         if (argv[1][0] == 'r') {
 392             /* copy new module into library */
 393             p = argv[3];
 394             do {
 395                 if (fputc(*p, fp) == EOF) {
 396                     fprintf(stderr, "rdflib: write error\n");
 397                     exit(1);
 398                 }
 399             } while (*p++);
 400
 401             while (!feof(fp2)) {
 402                 i = fgetc(fp2);
 403                 if (i == EOF) {
 404                     break;
 405                 }
 406                 if (fputc(i, fp) == EOF) {
 407                     fprintf(stderr, "rdflib: write error\n");
 408                     exit(1);
 409                 }
 410             }
 411             fclose(fp2);
 412         }
 413
 414         /* copy rest of library if any */
 415         while (!feof(fptmp)) {
 416             i = fgetc(fptmp);
 417             if (i == EOF) {
 418                 break;
 419             }
 420
 421             if (fputc(i, fp) == EOF) {
 422                 fprintf(stderr, "rdflib: write error\n");
 423                 exit(1);
 424             }
 425         }
 426
 427         fclose(fp);
 428         fclose(fptmp);
 429         break;
 430
 431     default:
 432         fprintf(stderr, "rdflib: command '%c' not recognized\n",
 433                 argv[1][0]);
 434         exit(1);
 435     }
 436     return 0;
 437 }