usr.sbin/crunch/crunchgen/crunchgen.c

   1 /*
   2  * Copyright (c) 1994 University of Maryland
   3  * All Rights Reserved.
   4  *
   5  * Permission to use, copy, modify, distribute, and sell this software and its
   6  * documentation for any purpose is hereby granted without fee, provided that
   7  * the above copyright notice appear in all copies and that both that
   8  * copyright notice and this permission notice appear in supporting
   9  * documentation, and that the name of U.M. not be used in advertising or
  10  * publicity pertaining to distribution of the software without specific,
  11  * written prior permission.  U.M. makes no representations about the
  12  * suitability of this software for any purpose.  It is provided "as is"
  13  * without express or implied warranty.
  14  *
  15  * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
  17  * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  18  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
  19  * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
  20  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  21  *
  22  * Author: James da Silva, Systems Design and Analysis Group
  23  *                         Computer Science Department
  24  *                         University of Maryland at College Park
  25  */
  26 /*
  27  * ========================================================================
  28  * crunchgen.c
  29  *
  30  * Generates a Makefile and main C file for a crunched executable,
  31  * from specs given in a .conf file.
  32  */
  33
  34 #include <sys/cdefs.h>
  35 __FBSDID("$FreeBSD$");
  36
  37 #include <sys/param.h>
  38 #include <sys/stat.h>
  39
  40 #include <ctype.h>
  41 #include <err.h>
  42 #include <paths.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47
  48 #define CRUNCH_VERSION  "0.2"
  49
  50 #define MAXLINELEN      16384
  51 #define MAXFIELDS        2048
  52
  53
  54 /* internal representation of conf file: */
  55
  56 /* simple lists of strings suffice for most parms */
  57
  58 typedef struct strlst {
  59         struct strlst *next;
  60         char *str;
  61 } strlst_t;
  62
  63 /* progs have structure, each field can be set with "special" or calculated */
  64
  65 typedef struct prog {
  66         struct prog *next;      /* link field */
  67         char *name;             /* program name */
  68         char *ident;            /* C identifier for the program name */
  69         char *srcdir;
  70         char *realsrcdir;
  71         char *objdir;
  72         char *objvar;           /* Makefile variable to replace OBJS */
  73         strlst_t *objs, *objpaths;
  74         strlst_t *buildopts;
  75         strlst_t *keeplist;
  76         strlst_t *links;
  77         strlst_t *libs;
  78         strlst_t *libs_so;
  79         int goterror;
  80 } prog_t;
  81
  82
  83 /* global state */
  84
  85 strlst_t *buildopts = NULL;
  86 strlst_t *srcdirs   = NULL;
  87 strlst_t *libs      = NULL;
  88 strlst_t *libs_so   = NULL;
  89 prog_t   *progs     = NULL;
  90
  91 char confname[MAXPATHLEN], infilename[MAXPATHLEN];
  92 char outmkname[MAXPATHLEN], outcfname[MAXPATHLEN], execfname[MAXPATHLEN];
  93 char tempfname[MAXPATHLEN], cachename[MAXPATHLEN], curfilename[MAXPATHLEN];
  94 char outhdrname[MAXPATHLEN] ;   /* user-supplied header for *.mk */
  95 char *objprefix;                /* where are the objects ? */
  96 char *path_make;
  97 int linenum = -1;
  98 int goterror = 0;
  99
 100 int verbose, readcache;         /* options */
 101 int reading_cache;
 102 int makeobj = 0;                /* add 'make obj' rules to the makefile */
 103
 104 int list_mode;
 105
 106 /* general library routines */
 107
 108 void status(const char *str);
 109 void out_of_memory(void);
 110 void add_string(strlst_t **listp, char *str);
 111 int is_dir(const char *pathname);
 112 int is_nonempty_file(const char *pathname);
 113 int subtract_strlst(strlst_t **lista, strlst_t **listb);
 114 int in_list(strlst_t **listp, char *str);
 115
 116 /* helper routines for main() */
 117
 118 void usage(void);
 119 void parse_conf_file(void);
 120 void gen_outputs(void);
 121
 122 extern char *crunched_skel[];
 123
 124
 125 int
 126 main(int argc, char **argv)
 127 {
 128         char *p;
 129         int optc;
 130
 131         verbose = 1;
 132         readcache = 1;
 133         *outmkname = *outcfname = *execfname = '\0';
 134
 135         path_make = getenv("MAKE");
 136         if (path_make == NULL || *path_make == '\0')
 137                 path_make = "make";
 138
 139         p = getenv("MAKEOBJDIRPREFIX");
 140         if (p == NULL || *p == '\0')
 141                 objprefix = "/usr/obj"; /* default */
 142         else
 143                 if ((objprefix = strdup(p)) == NULL)
 144                         out_of_memory();
 145
 146         while((optc = getopt(argc, argv, "lh:m:c:e:p:foq")) != -1) {
 147                 switch(optc) {
 148                 case 'f':
 149                         readcache = 0;
 150                         break;
 151                 case 'o':
 152                         makeobj = 1;
 153                         break;
 154                 case 'q':
 155                         verbose = 0;
 156                         break;
 157
 158                 case 'm':
 159                         strlcpy(outmkname, optarg, sizeof(outmkname));
 160                         break;
 161                 case 'p':
 162                         if ((objprefix = strdup(optarg)) == NULL)
 163                                 out_of_memory();
 164                         break;
 165
 166                 case 'h':
 167                         strlcpy(outhdrname, optarg, sizeof(outhdrname));
 168                         break;
 169                 case 'c':
 170                         strlcpy(outcfname, optarg, sizeof(outcfname));
 171                         break;
 172                 case 'e':
 173                         strlcpy(execfname, optarg, sizeof(execfname));
 174                         break;
 175
 176                 case 'l':
 177                         list_mode++;
 178                         verbose = 0;
 179                         break;
 180
 181                 case '?':
 182                 default:
 183                         usage();
 184                 }
 185         }
 186
 187         argc -= optind;
 188         argv += optind;
 189
 190         if (argc != 1)
 191                 usage();
 192
 193         /*
 194          * generate filenames
 195          */
 196
 197         strlcpy(infilename, argv[0], sizeof(infilename));
 198
 199         /* confname = `basename infilename .conf` */
 200
 201         if ((p=strrchr(infilename, '/')) != NULL)
 202                 strlcpy(confname, p + 1, sizeof(confname));
 203         else
 204                 strlcpy(confname, infilename, sizeof(confname));
 205
 206         if ((p=strrchr(confname, '.')) != NULL && !strcmp(p, ".conf"))
 207                 *p = '\0';
 208
 209         if (!*outmkname)
 210                 snprintf(outmkname, sizeof(outmkname), "%s.mk", confname);
 211         if (!*outcfname)
 212                 snprintf(outcfname, sizeof(outcfname), "%s.c", confname);
 213         if (!*execfname)
 214                 snprintf(execfname, sizeof(execfname), "%s", confname);
 215
 216         snprintf(cachename, sizeof(cachename), "%s.cache", confname);
 217         snprintf(tempfname, sizeof(tempfname), "%s/crunchgen_%sXXXXXX",
 218         getenv("TMPDIR") ? getenv("TMPDIR") : _PATH_TMP, confname);
 219
 220         parse_conf_file();
 221         if (list_mode)
 222                 exit(goterror);
 223
 224         gen_outputs();
 225
 226         exit(goterror);
 227 }
 228
 229
 230 void
 231 usage(void)
 232 {
 233         fprintf(stderr, "%s%s\n\t%s%s\n", "usage: crunchgen [-foq] ",
 234             "[-h <makefile-header-name>] [-m <makefile>]",
 235             "[-p <obj-prefix>] [-c <c-file-name>] [-e <exec-file>] ",
 236             "<conffile>");
 237         exit(1);
 238 }
 239
 240
 241 /*
 242  * ========================================================================
 243  * parse_conf_file subsystem
 244  *
 245  */
 246
 247 /* helper routines for parse_conf_file */
 248
 249 void parse_one_file(char *filename);
 250 void parse_line(char *pline, int *fc, char **fv, int nf);
 251 void add_srcdirs(int argc, char **argv);
 252 void add_progs(int argc, char **argv);
 253 void add_link(int argc, char **argv);
 254 void add_libs(int argc, char **argv);
 255 void add_libs_so(int argc, char **argv);
 256 void add_buildopts(int argc, char **argv);
 257 void add_special(int argc, char **argv);
 258
 259 prog_t *find_prog(char *str);
 260 void add_prog(char *progname);
 261
 262
 263 void
 264 parse_conf_file(void)
 265 {
 266         if (!is_nonempty_file(infilename))
 267                 errx(1, "fatal: input file \"%s\" not found", infilename);
 268
 269         parse_one_file(infilename);
 270         if (readcache && is_nonempty_file(cachename)) {
 271                 reading_cache = 1;
 272                 parse_one_file(cachename);
 273         }
 274 }
 275
 276
 277 void
 278 parse_one_file(char *filename)
 279 {
 280         char *fieldv[MAXFIELDS];
 281         int fieldc;
 282         void (*f)(int c, char **v);
 283         FILE *cf;
 284         char line[MAXLINELEN];
 285
 286         snprintf(line, sizeof(line), "reading %s", filename);
 287         status(line);
 288         strlcpy(curfilename, filename, sizeof(curfilename));
 289
 290         if ((cf = fopen(curfilename, "r")) == NULL) {
 291                 warn("%s", curfilename);
 292                 goterror = 1;
 293                 return;
 294         }
 295
 296         linenum = 0;
 297         while (fgets(line, MAXLINELEN, cf) != NULL) {
 298                 linenum++;
 299                 parse_line(line, &fieldc, fieldv, MAXFIELDS);
 300
 301                 if (fieldc < 1)
 302                         continue;
 303
 304                 if (!strcmp(fieldv[0], "srcdirs"))
 305                         f = add_srcdirs;
 306                 else if(!strcmp(fieldv[0], "progs"))
 307                         f = add_progs;
 308                 else if(!strcmp(fieldv[0], "ln"))
 309                         f = add_link;
 310                 else if(!strcmp(fieldv[0], "libs"))
 311                         f = add_libs;
 312                 else if(!strcmp(fieldv[0], "libs_so"))
 313                         f = add_libs_so;
 314                 else if(!strcmp(fieldv[0], "buildopts"))
 315                         f = add_buildopts;
 316                 else if(!strcmp(fieldv[0], "special"))
 317                         f = add_special;
 318                 else {
 319                         warnx("%s:%d: skipping unknown command `%s'",
 320                             curfilename, linenum, fieldv[0]);
 321                         goterror = 1;
 322                         continue;
 323                 }
 324
 325                 if (fieldc < 2) {
 326                         warnx("%s:%d: %s %s",
 327                             curfilename, linenum, fieldv[0],
 328                             "command needs at least 1 argument, skipping");
 329                         goterror = 1;
 330                         continue;
 331                 }
 332
 333                 f(fieldc, fieldv);
 334         }
 335
 336         if (ferror(cf)) {
 337                 warn("%s", curfilename);
 338                 goterror = 1;
 339         }
 340         fclose(cf);
 341 }
 342
 343
 344 void
 345 parse_line(char *pline, int *fc, char **fv, int nf)
 346 {
 347         char *p;
 348
 349         p = pline;
 350         *fc = 0;
 351
 352         while (1) {
 353                 while (isspace((unsigned char)*p))
 354                         p++;
 355
 356                 if (*p == '\0' || *p == '#')
 357                         break;
 358
 359                 if (*fc < nf)
 360                         fv[(*fc)++] = p;
 361
 362                 while (*p && !isspace((unsigned char)*p) && *p != '#')
 363                         p++;
 364
 365                 if (*p == '\0' || *p == '#')
 366                         break;
 367
 368                 *p++ = '\0';
 369         }
 370
 371         if (*p)
 372                 *p = '\0';              /* needed for '#' case */
 373 }
 374
 375
 376 void
 377 add_srcdirs(int argc, char **argv)
 378 {
 379         int i;
 380
 381         for (i = 1; i < argc; i++) {
 382                 if (is_dir(argv[i]))
 383                         add_string(&srcdirs, argv[i]);
 384                 else {
 385                         warnx("%s:%d: `%s' is not a directory, skipping it",
 386                             curfilename, linenum, argv[i]);
 387                         goterror = 1;
 388                 }
 389         }
 390 }
 391
 392
 393 void
 394 add_progs(int argc, char **argv)
 395 {
 396         int i;
 397
 398         for (i = 1; i < argc; i++)
 399                 add_prog(argv[i]);
 400 }
 401
 402
 403 void
 404 add_prog(char *progname)
 405 {
 406         prog_t *p1, *p2;
 407
 408         /* add to end, but be smart about dups */
 409
 410         for (p1 = NULL, p2 = progs; p2 != NULL; p1 = p2, p2 = p2->next)
 411                 if (!strcmp(p2->name, progname))
 412                         return;
 413
 414         p2 = malloc(sizeof(prog_t));
 415         if(p2) {
 416                 memset(p2, 0, sizeof(prog_t));
 417                 p2->name = strdup(progname);
 418         }
 419         if (!p2 || !p2->name)
 420                 out_of_memory();
 421
 422         p2->next = NULL;
 423         if (p1 == NULL)
 424                 progs = p2;
 425         else
 426                 p1->next = p2;
 427
 428         p2->ident = NULL;
 429         p2->srcdir = NULL;
 430         p2->realsrcdir = NULL;
 431         p2->objdir = NULL;
 432         p2->links = NULL;
 433         p2->libs = NULL;
 434         p2->libs_so = NULL;
 435         p2->objs = NULL;
 436         p2->keeplist = NULL;
 437         p2->buildopts = NULL;
 438         p2->goterror = 0;
 439
 440         if (list_mode)
 441                 printf("%s\n",progname);
 442 }
 443
 444
 445 void
 446 add_link(int argc, char **argv)
 447 {
 448         int i;
 449         prog_t *p = find_prog(argv[1]);
 450
 451         if (p == NULL) {
 452                 warnx("%s:%d: no prog %s previously declared, skipping link",
 453                     curfilename, linenum, argv[1]);
 454                 goterror = 1;
 455                 return;
 456         }
 457
 458         for (i = 2; i < argc; i++) {
 459                 if (list_mode)
 460                         printf("%s\n",argv[i]);
 461
 462                 add_string(&p->links, argv[i]);
 463         }
 464 }
 465
 466
 467 void
 468 add_libs(int argc, char **argv)
 469 {
 470         int i;
 471
 472         for(i = 1; i < argc; i++) {
 473                 add_string(&libs, argv[i]);
 474                 if ( in_list(&libs_so, argv[i]) )
 475                         warnx("%s:%d: "
 476                                 "library `%s' specified as dynamic earlier",
 477                                 curfilename, linenum, argv[i]);
 478         }
 479 }
 480
 481
 482 void
 483 add_libs_so(int argc, char **argv)
 484 {
 485         int i;
 486
 487         for(i = 1; i < argc; i++) {
 488                 add_string(&libs_so, argv[i]);
 489                 if ( in_list(&libs, argv[i]) )
 490                         warnx("%s:%d: "
 491                                 "library `%s' specified as static earlier",
 492                                 curfilename, linenum, argv[i]);
 493         }
 494 }
 495
 496
 497 void
 498 add_buildopts(int argc, char **argv)
 499 {
 500         int i;
 501
 502         for (i = 1; i < argc; i++)
 503                 add_string(&buildopts, argv[i]);
 504 }
 505
 506
 507 void
 508 add_special(int argc, char **argv)
 509 {
 510         int i;
 511         prog_t *p = find_prog(argv[1]);
 512
 513         if (p == NULL) {
 514                 if (reading_cache)
 515                         return;
 516
 517                 warnx("%s:%d: no prog %s previously declared, skipping special",
 518                     curfilename, linenum, argv[1]);
 519                 goterror = 1;
 520                 return;
 521         }
 522
 523         if (!strcmp(argv[2], "ident")) {
 524                 if (argc != 4)
 525                         goto argcount;
 526                 if ((p->ident = strdup(argv[3])) == NULL)
 527                         out_of_memory();
 528         } else if (!strcmp(argv[2], "srcdir")) {
 529                 if (argc != 4)
 530                         goto argcount;
 531                 if ((p->srcdir = strdup(argv[3])) == NULL)
 532                         out_of_memory();
 533         } else if (!strcmp(argv[2], "objdir")) {
 534                 if(argc != 4)
 535                         goto argcount;
 536                 if((p->objdir = strdup(argv[3])) == NULL)
 537                         out_of_memory();
 538         } else if (!strcmp(argv[2], "objs")) {
 539                 p->objs = NULL;
 540                 for (i = 3; i < argc; i++)
 541                         add_string(&p->objs, argv[i]);
 542         } else if (!strcmp(argv[2], "objpaths")) {
 543                 p->objpaths = NULL;
 544                 for (i = 3; i < argc; i++)
 545                         add_string(&p->objpaths, argv[i]);
 546         } else if (!strcmp(argv[2], "keep")) {
 547                 p->keeplist = NULL;
 548                 for(i = 3; i < argc; i++)
 549                         add_string(&p->keeplist, argv[i]);
 550         } else if (!strcmp(argv[2], "objvar")) {
 551                 if(argc != 4)
 552                         goto argcount;
 553                 if ((p->objvar = strdup(argv[3])) == NULL)
 554                         out_of_memory();
 555         } else if (!strcmp(argv[2], "buildopts")) {
 556                 p->buildopts = NULL;
 557                 for (i = 3; i < argc; i++)
 558                         add_string(&p->buildopts, argv[i]);
 559         } else if (!strcmp(argv[2], "lib")) {
 560                 for (i = 3; i < argc; i++)
 561                         add_string(&p->libs, argv[i]);
 562         } else {
 563                 warnx("%s:%d: bad parameter name `%s', skipping line",
 564                     curfilename, linenum, argv[2]);
 565                 goterror = 1;
 566         }
 567         return;
 568
 569  argcount:
 570         warnx("%s:%d: too %s arguments, expected \"special %s %s <string>\"",
 571             curfilename, linenum, argc < 4? "few" : "many", argv[1], argv[2]);
 572         goterror = 1;
 573 }
 574
 575
 576 prog_t *find_prog(char *str)
 577 {
 578         prog_t *p;
 579
 580         for (p = progs; p != NULL; p = p->next)
 581                 if (!strcmp(p->name, str))
 582                         return p;
 583
 584         return NULL;
 585 }
 586
 587
 588 /*
 589  * ========================================================================
 590  * gen_outputs subsystem
 591  *
 592  */
 593
 594 /* helper subroutines */
 595
 596 void remove_error_progs(void);
 597 void fillin_program(prog_t *p);
 598 void gen_specials_cache(void);
 599 void gen_output_makefile(void);
 600 void gen_output_cfile(void);
 601
 602 void fillin_program_objs(prog_t *p, char *path);
 603 void top_makefile_rules(FILE *outmk);
 604 void prog_makefile_rules(FILE *outmk, prog_t *p);
 605 void output_strlst(FILE *outf, strlst_t *lst);
 606 char *genident(char *str);
 607 char *dir_search(char *progname);
 608
 609
 610 void
 611 gen_outputs(void)
 612 {
 613         prog_t *p;
 614
 615         for (p = progs; p != NULL; p = p->next)
 616                 fillin_program(p);
 617
 618         remove_error_progs();
 619         gen_specials_cache();
 620         gen_output_cfile();
 621         gen_output_makefile();
 622         status("");
 623         fprintf(stderr,
 624             "Run \"%s -f %s\" to build crunched binary.\n",
 625             path_make, outmkname);
 626 }
 627
 628 /*
 629  * run the makefile for the program to find which objects are necessary
 630  */
 631 void
 632 fillin_program(prog_t *p)
 633 {
 634         char path[MAXPATHLEN];
 635         char line[MAXLINELEN];
 636         FILE *f;
 637
 638         snprintf(line, MAXLINELEN, "filling in parms for %s", p->name);
 639         status(line);
 640
 641         if (!p->ident)
 642                 p->ident = genident(p->name);
 643
 644         /* look for the source directory if one wasn't specified by a special */
 645         if (!p->srcdir) {
 646                 p->srcdir = dir_search(p->name);
 647         }
 648
 649         /* Determine the actual srcdir (maybe symlinked). */
 650         if (p->srcdir) {
 651                 snprintf(line, MAXLINELEN, "cd %s && echo -n `/bin/pwd`",
 652                     p->srcdir);
 653                 f = popen(line,"r");
 654                 if (!f)
 655                         errx(1, "Can't execute: %s\n", line);
 656
 657                 path[0] = '\0';
 658                 fgets(path, sizeof path, f);
 659                 if (pclose(f))
 660                         errx(1, "Can't execute: %s\n", line);
 661
 662                 if (!*path)
 663                         errx(1, "Can't perform pwd on: %s\n", p->srcdir);
 664
 665                 p->realsrcdir = strdup(path);
 666         }
 667
 668         /* Unless the option to make object files was specified the
 669         * the objects will be built in the source directory unless
 670         * an object directory already exists.
 671         */
 672         if (!makeobj && !p->objdir && p->srcdir) {
 673                 char *auto_obj;
 674
 675                 auto_obj = NULL;
 676                 snprintf(line, sizeof line, "%s/%s", objprefix, p->realsrcdir);
 677                 if (is_dir(line) ||
 678                     ((auto_obj = getenv("MK_AUTO_OBJ")) != NULL &&
 679                     strcmp(auto_obj, "yes") == 0)) {
 680                         if ((p->objdir = strdup(line)) == NULL)
 681                         out_of_memory();
 682                 } else
 683                         p->objdir = p->realsrcdir;
 684         }
 685
 686         /*
 687         * XXX look for a Makefile.{name} in local directory first.
 688         * This lets us override the original Makefile.
 689         */
 690         snprintf(path, sizeof(path), "Makefile.%s", p->name);
 691         if (is_nonempty_file(path)) {
 692                 snprintf(line, MAXLINELEN, "Using %s for %s", path, p->name);
 693                 status(line);
 694         } else
 695                 if (p->srcdir)
 696                         snprintf(path, sizeof(path), "%s/Makefile", p->srcdir);
 697         if (!p->objs && p->srcdir && is_nonempty_file(path))
 698                 fillin_program_objs(p, path);
 699
 700         if (!p->srcdir && !p->objdir && verbose)
 701                 warnx("%s: %s: %s",
 702                     "warning: could not find source directory",
 703                     infilename, p->name);
 704         if (!p->objs && verbose)
 705                 warnx("%s: %s: warning: could not find any .o files",
 706                     infilename, p->name);
 707
 708         if ((!p->srcdir || !p->objdir) && !p->objs)
 709                 p->goterror = 1;
 710 }
 711
 712 void
 713 fillin_program_objs(prog_t *p, char *path)
 714 {
 715         char *obj, *cp;
 716         int fd, rc;
 717         FILE *f;
 718         char *objvar="OBJS";
 719         strlst_t *s;
 720         char line[MAXLINELEN];
 721
 722         /* discover the objs from the srcdir Makefile */
 723
 724         if ((fd = mkstemp(tempfname)) == -1) {
 725                 perror(tempfname);
 726                 exit(1);
 727         }
 728         if ((f = fdopen(fd, "w")) == NULL) {
 729                 warn("%s", tempfname);
 730                 goterror = 1;
 731                 return;
 732         }
 733         if (p->objvar)
 734                 objvar = p->objvar;
 735
 736         /*
 737         * XXX include outhdrname (e.g. to contain Make variables)
 738         */
 739         if (outhdrname[0] != '\0')
 740                 fprintf(f, ".include \"%s\"\n", outhdrname);
 741         fprintf(f, ".include \"%s\"\n", path);
 742         fprintf(f, ".POSIX:\n");
 743         if (buildopts) {
 744                 fprintf(f, "BUILDOPTS+=");
 745                 output_strlst(f, buildopts);
 746         }
 747         fprintf(f, ".if defined(PROG)\n");
 748         fprintf(f, "%s?=${PROG}.o\n", objvar);
 749         fprintf(f, ".endif\n");
 750         fprintf(f, "loop:\n\t@echo 'OBJS= '${%s}\n", objvar);
 751
 752         fprintf(f, "crunchgen_objs:\n"
 753             "\t@cd %s && %s -f %s $(BUILDOPTS) $(%s_OPTS)",
 754             p->srcdir, path_make, tempfname, p->ident);
 755         for (s = p->buildopts; s != NULL; s = s->next)
 756                 fprintf(f, " %s", s->str);
 757         fprintf(f, " loop\n");
 758
 759         fclose(f);
 760
 761         snprintf(line, MAXLINELEN, "cd %s && %s -f %s -B crunchgen_objs",
 762              p->srcdir, path_make, tempfname);
 763         if ((f = popen(line, "r")) == NULL) {
 764                 warn("submake pipe");
 765                 goterror = 1;
 766                 return;
 767         }
 768
 769         while(fgets(line, MAXLINELEN, f)) {
 770                 if (strncmp(line, "OBJS= ", 6)) {
 771                         warnx("make error: %s", line);
 772                         goterror = 1;
 773                         continue;
 774                 }
 775
 776                 cp = line + 6;
 777                 while (isspace((unsigned char)*cp))
 778                         cp++;
 779
 780                 while(*cp) {
 781                         obj = cp;
 782                         while (*cp && !isspace((unsigned char)*cp))
 783                                 cp++;
 784                         if (*cp)
 785                                 *cp++ = '\0';
 786                         add_string(&p->objs, obj);
 787                         while (isspace((unsigned char)*cp))
 788                                 cp++;
 789                 }
 790         }
 791
 792         if ((rc=pclose(f)) != 0) {
 793                 warnx("make error: make returned %d", rc);
 794                 goterror = 1;
 795         }
 796
 797         unlink(tempfname);
 798 }
 799
 800 void
 801 remove_error_progs(void)
 802 {
 803         prog_t *p1, *p2;
 804
 805         p1 = NULL; p2 = progs;
 806         while (p2 != NULL) {
 807                 if (!p2->goterror)
 808                         p1 = p2, p2 = p2->next;
 809                 else {
 810                         /* delete it from linked list */
 811                         warnx("%s: %s: ignoring program because of errors",
 812                             infilename, p2->name);
 813                         if (p1)
 814                                 p1->next = p2->next;
 815                         else
 816                                 progs = p2->next;
 817                         p2 = p2->next;
 818                 }
 819         }
 820 }
 821
 822 void
 823 gen_specials_cache(void)
 824 {
 825         FILE *cachef;
 826         prog_t *p;
 827         char line[MAXLINELEN];
 828
 829         snprintf(line, MAXLINELEN, "generating %s", cachename);
 830         status(line);
 831
 832         if ((cachef = fopen(cachename, "w")) == NULL) {
 833                 warn("%s", cachename);
 834                 goterror = 1;
 835                 return;
 836         }
 837
 838         fprintf(cachef, "# %s - parm cache generated from %s by crunchgen "
 839             " %s\n\n",
 840             cachename, infilename, CRUNCH_VERSION);
 841
 842         for (p = progs; p != NULL; p = p->next) {
 843                 fprintf(cachef, "\n");
 844                 if (p->srcdir)
 845                         fprintf(cachef, "special %s srcdir %s\n",
 846                             p->name, p->srcdir);
 847                 if (p->objdir)
 848                         fprintf(cachef, "special %s objdir %s\n",
 849                             p->name, p->objdir);
 850                 if (p->objs) {
 851                         fprintf(cachef, "special %s objs", p->name);
 852                         output_strlst(cachef, p->objs);
 853                 }
 854                 if (p->objpaths) {
 855                         fprintf(cachef, "special %s objpaths", p->name);
 856                         output_strlst(cachef, p->objpaths);
 857                 }
 858         }
 859         fclose(cachef);
 860 }
 861
 862
 863 void
 864 gen_output_makefile(void)
 865 {
 866         prog_t *p;
 867         FILE *outmk;
 868         char line[MAXLINELEN];
 869
 870         snprintf(line, MAXLINELEN, "generating %s", outmkname);
 871         status(line);
 872
 873         if ((outmk = fopen(outmkname, "w")) == NULL) {
 874                 warn("%s", outmkname);
 875                 goterror = 1;
 876                 return;
 877         }
 878
 879         fprintf(outmk, "# %s - generated from %s by crunchgen %s\n\n",
 880             outmkname, infilename, CRUNCH_VERSION);
 881
 882         if (outhdrname[0] != '\0')
 883                 fprintf(outmk, ".include \"%s\"\n", outhdrname);
 884
 885         top_makefile_rules(outmk);
 886         for (p = progs; p != NULL; p = p->next)
 887                 prog_makefile_rules(outmk, p);
 888
 889         fprintf(outmk, "\n# ========\n");
 890         fclose(outmk);
 891 }
 892
 893
 894 void
 895 gen_output_cfile(void)
 896 {
 897         char **cp;
 898         FILE *outcf;
 899         prog_t *p;
 900         strlst_t *s;
 901         char line[MAXLINELEN];
 902
 903         snprintf(line, MAXLINELEN, "generating %s", outcfname);
 904         status(line);
 905
 906         if((outcf = fopen(outcfname, "w")) == NULL) {
 907                 warn("%s", outcfname);
 908                 goterror = 1;
 909                 return;
 910         }
 911
 912         fprintf(outcf,
 913             "/* %s - generated from %s by crunchgen %s */\n",
 914             outcfname, infilename, CRUNCH_VERSION);
 915
 916         fprintf(outcf, "#define EXECNAME \"%s\"\n", execfname);
 917         for (cp = crunched_skel; *cp != NULL; cp++)
 918                 fprintf(outcf, "%s\n", *cp);
 919
 920         for (p = progs; p != NULL; p = p->next)
 921                 fprintf(outcf, "extern int _crunched_%s_stub();\n", p->ident);
 922
 923         fprintf(outcf, "\nstruct stub entry_points[] = {\n");
 924         for (p = progs; p != NULL; p = p->next) {
 925                 fprintf(outcf, "\t{ \"%s\", _crunched_%s_stub },\n",
 926                     p->name, p->ident);
 927                 for (s = p->links; s != NULL; s = s->next)
 928                         fprintf(outcf, "\t{ \"%s\", _crunched_%s_stub },\n",
 929                             s->str, p->ident);
 930         }
 931
 932         fprintf(outcf, "\t{ EXECNAME, crunched_main },\n");
 933         fprintf(outcf, "\t{ NULL, NULL }\n};\n");
 934         fclose(outcf);
 935 }
 936
 937
 938 char *genident(char *str)
 939 {
 940         char *n, *s, *d;
 941
 942         /*
 943          * generates a Makefile/C identifier from a program name,
 944          * mapping '-' to '_' and ignoring all other non-identifier
 945          * characters.  This leads to programs named "foo.bar" and
 946          * "foobar" to map to the same identifier.
 947          */
 948
 949         if ((n = strdup(str)) == NULL)
 950                 return NULL;
 951         for (d = s = n; *s != '\0'; s++) {
 952                 if (*s == '-')
 953                         *d++ = '_';
 954                 else if (*s == '_' || isalnum((unsigned char)*s))
 955                         *d++ = *s;
 956         }
 957         *d = '\0';
 958         return n;
 959 }
 960
 961
 962 char *dir_search(char *progname)
 963 {
 964         char path[MAXPATHLEN];
 965         strlst_t *dir;
 966         char *srcdir;
 967
 968         for (dir = srcdirs; dir != NULL; dir = dir->next) {
 969                 snprintf(path, MAXPATHLEN, "%s/%s", dir->str, progname);
 970                 if (!is_dir(path))
 971                         continue;
 972
 973                 if ((srcdir = strdup(path)) == NULL)
 974                         out_of_memory();
 975
 976                 return srcdir;
 977         }
 978         return NULL;
 979 }
 980
 981
 982 void
 983 top_makefile_rules(FILE *outmk)
 984 {
 985         prog_t *p;
 986
 987         fprintf(outmk, "LD?= ld\n");
 988         if ( subtract_strlst(&libs, &libs_so) )
 989                 fprintf(outmk, "# NOTE: Some LIBS declarations below overridden by LIBS_SO\n");
 990
 991         fprintf(outmk, "LIBS+=");
 992         output_strlst(outmk, libs);
 993
 994         fprintf(outmk, "LIBS_SO+=");
 995         output_strlst(outmk, libs_so);
 996
 997         if (makeobj) {
 998                 fprintf(outmk, "MAKEOBJDIRPREFIX?=%s\n", objprefix);
 999                 fprintf(outmk, "MAKEENV=env MAKEOBJDIRPREFIX=$(MAKEOBJDIRPREFIX)\n");
1000                 fprintf(outmk, "CRUNCHMAKE=$(MAKEENV) $(MAKE)\n");
1001         } else {
1002                 fprintf(outmk, "CRUNCHMAKE=$(MAKE)\n");
1003         }
1004
1005         if (buildopts) {
1006                 fprintf(outmk, "BUILDOPTS+=");
1007                 output_strlst(outmk, buildopts);
1008         }
1009
1010         fprintf(outmk, "CRUNCHED_OBJS=");
1011         for (p = progs; p != NULL; p = p->next)
1012                 fprintf(outmk, " %s.lo", p->name);
1013         fprintf(outmk, "\n");
1014
1015         fprintf(outmk, "SUBMAKE_TARGETS=");
1016         for (p = progs; p != NULL; p = p->next)
1017                 fprintf(outmk, " %s_make", p->ident);
1018         fprintf(outmk, "\nSUBCLEAN_TARGETS=");
1019         for (p = progs; p != NULL; p = p->next)
1020                 fprintf(outmk, " %s_clean", p->ident);
1021         fprintf(outmk, "\n\n");
1022
1023         fprintf(outmk, "all: objs exe\nobjs: $(SUBMAKE_TARGETS)\n");
1024         fprintf(outmk, "exe: %s\n", execfname);
1025         fprintf(outmk, "%s: %s.o $(CRUNCHED_OBJS) $(SUBMAKE_TARGETS)\n", execfname, execfname);
1026         fprintf(outmk, ".if defined(LIBS_SO) && !empty(LIBS_SO)\n");
1027         fprintf(outmk, "\t$(CC) -o %s %s.o $(CRUNCHED_OBJS) \\\n",
1028             execfname, execfname);
1029         fprintf(outmk, "\t\t-Xlinker -Bstatic $(LIBS) \\\n");
1030         fprintf(outmk, "\t\t-Xlinker -Bdynamic $(LIBS_SO)\n");
1031         fprintf(outmk, ".else\n");
1032         fprintf(outmk, "\t$(CC) -static -o %s %s.o $(CRUNCHED_OBJS) $(LIBS)\n",
1033             execfname, execfname);
1034         fprintf(outmk, ".endif\n");
1035         fprintf(outmk, "realclean: clean subclean\n");
1036         fprintf(outmk, "clean:\n\trm -f %s *.lo *.o *_stub.c\n", execfname);
1037         fprintf(outmk, "subclean: $(SUBCLEAN_TARGETS)\n");
1038 }
1039
1040
1041 void
1042 prog_makefile_rules(FILE *outmk, prog_t *p)
1043 {
1044         strlst_t *lst;
1045
1046         fprintf(outmk, "\n# -------- %s\n\n", p->name);
1047
1048         fprintf(outmk, "%s_OBJDIR=", p->ident);
1049         if (p->objdir)
1050                 fprintf(outmk, "%s", p->objdir);
1051         else
1052                 fprintf(outmk, "$(MAKEOBJDIRPREFIX)/$(%s_REALSRCDIR)\n",
1053                     p->ident);
1054         fprintf(outmk, "\n");
1055
1056         fprintf(outmk, "%s_OBJPATHS=", p->ident);
1057         if (p->objpaths)
1058                 output_strlst(outmk, p->objpaths);
1059         else {
1060                 for (lst = p->objs; lst != NULL; lst = lst->next) {
1061                         fprintf(outmk, " $(%s_OBJDIR)/%s", p->ident, lst->str);
1062                 }
1063                 fprintf(outmk, "\n");
1064         }
1065         fprintf(outmk, "$(%s_OBJPATHS): .NOMETA\n", p->ident);
1066
1067         if (p->srcdir && p->objs) {
1068                 fprintf(outmk, "%s_SRCDIR=%s\n", p->ident, p->srcdir);
1069                 fprintf(outmk, "%s_REALSRCDIR=%s\n", p->ident, p->realsrcdir);
1070
1071                 fprintf(outmk, "%s_OBJS=", p->ident);
1072                 output_strlst(outmk, p->objs);
1073                 if (p->buildopts != NULL) {
1074                         fprintf(outmk, "%s_OPTS+=", p->ident);
1075                         output_strlst(outmk, p->buildopts);
1076                 }
1077 #if 0
1078                 fprintf(outmk, "$(%s_OBJPATHS): %s_make\n\n", p->ident, p->ident);
1079 #endif
1080                 fprintf(outmk, "%s_make:\n", p->ident);
1081                 fprintf(outmk, "\t(cd $(%s_SRCDIR) && ", p->ident);
1082                 if (makeobj)
1083                         fprintf(outmk, "$(CRUNCHMAKE) obj && ");
1084                 fprintf(outmk, "\\\n");
1085                 fprintf(outmk, "\t\t$(CRUNCHMAKE) $(BUILDOPTS) $(%s_OPTS) depend &&",
1086                     p->ident);
1087                 fprintf(outmk, "\\\n");
1088                 fprintf(outmk, "\t\t$(CRUNCHMAKE) $(BUILDOPTS) $(%s_OPTS) "
1089                     "$(%s_OBJS))",
1090                     p->ident, p->ident);
1091                 fprintf(outmk, "\n");
1092                 fprintf(outmk, "%s_clean:\n", p->ident);
1093                 fprintf(outmk, "\t(cd $(%s_SRCDIR) && $(CRUNCHMAKE) $(BUILDOPTS) clean cleandepend)\n\n",
1094                     p->ident);
1095         } else {
1096                 fprintf(outmk, "%s_make:\n", p->ident);
1097                 fprintf(outmk, "\t@echo \"** cannot make objs for %s\"\n\n",
1098                     p->name);
1099         }
1100
1101         if (p->libs) {
1102                 fprintf(outmk, "%s_LIBS=", p->ident);
1103                 output_strlst(outmk, p->libs);
1104         }
1105
1106         fprintf(outmk, "%s_stub.c:\n", p->name);
1107         fprintf(outmk, "\techo \""
1108             "extern int main(int argc, char **argv, char **envp); "
1109             "int _crunched_%s_stub(int argc, char **argv, char **envp)"
1110             "{return main(argc,argv,envp);}\" >%s_stub.c\n",
1111             p->ident, p->name);
1112         fprintf(outmk, "%s.lo: %s_stub.o $(%s_OBJPATHS)",
1113             p->name, p->name, p->ident);
1114         if (p->libs)
1115                 fprintf(outmk, " $(%s_LIBS)", p->ident);
1116
1117         fprintf(outmk, "\n");
1118         fprintf(outmk, "\t$(CC) -nostdlib -Wl,-dc -r -o %s.lo %s_stub.o $(%s_OBJPATHS)",
1119             p->name, p->name, p->ident);
1120         if (p->libs)
1121                 fprintf(outmk, " $(%s_LIBS)", p->ident);
1122         fprintf(outmk, "\n");
1123         fprintf(outmk, "\tcrunchide -k _crunched_%s_stub ", p->ident);
1124         for (lst = p->keeplist; lst != NULL; lst = lst->next)
1125                 fprintf(outmk, "-k _%s ", lst->str);
1126         fprintf(outmk, "%s.lo\n", p->name);
1127 }
1128
1129 void
1130 output_strlst(FILE *outf, strlst_t *lst)
1131 {
1132         for (; lst != NULL; lst = lst->next)
1133                 if ( strlen(lst->str) )
1134                         fprintf(outf, " %s", lst->str);
1135         fprintf(outf, "\n");
1136 }
1137
1138
1139 /*
1140  * ========================================================================
1141  * general library routines
1142  *
1143  */
1144
1145 void
1146 status(const char *str)
1147 {
1148         static int lastlen = 0;
1149         int len, spaces;
1150
1151         if (!verbose)
1152                 return;
1153
1154         len = strlen(str);
1155         spaces = lastlen - len;
1156         if (spaces < 1)
1157                 spaces = 1;
1158
1159         fprintf(stderr, " [%s]%*.*s\r", str, spaces, spaces, " ");
1160         fflush(stderr);
1161         lastlen = len;
1162 }
1163
1164
1165 void
1166 out_of_memory(void)
1167 {
1168         err(1, "%s: %d: out of memory, stopping", infilename, linenum);
1169 }
1170
1171
1172 void
1173 add_string(strlst_t **listp, char *str)
1174 {
1175         strlst_t *p1, *p2;
1176
1177         /* add to end, but be smart about dups */
1178
1179         for (p1 = NULL, p2 = *listp; p2 != NULL; p1 = p2, p2 = p2->next)
1180                 if (!strcmp(p2->str, str))
1181                         return;
1182
1183         p2 = malloc(sizeof(strlst_t));
1184         if (p2) {
1185                 p2->next = NULL;
1186                 p2->str = strdup(str);
1187         }
1188         if (!p2 || !p2->str)
1189                 out_of_memory();
1190
1191         if (p1 == NULL)
1192                 *listp = p2;
1193         else
1194                 p1->next = p2;
1195 }
1196
1197 int
1198 subtract_strlst(strlst_t **lista, strlst_t **listb)
1199 {
1200         int subtract_count = 0;
1201         strlst_t *p1;
1202         for (p1 = *listb; p1 != NULL; p1 = p1->next)
1203                 if ( in_list(lista, p1->str) ) {
1204                         warnx("Will compile library `%s' dynamically", p1->str);
1205                         strcat(p1->str, "");
1206                         subtract_count++;
1207                 }
1208         return subtract_count;
1209 }
1210
1211 int
1212 in_list(strlst_t **listp, char *str)
1213 {
1214         strlst_t *p1;
1215         for (p1 = *listp; p1 != NULL; p1 = p1->next)
1216                 if (!strcmp(p1->str, str))
1217                         return 1;
1218         return 0;
1219 }
1220
1221 int
1222 is_dir(const char *pathname)
1223 {
1224         struct stat buf;
1225
1226         if (stat(pathname, &buf) == -1)
1227                 return 0;
1228
1229         return S_ISDIR(buf.st_mode);
1230 }
1231
1232 int
1233 is_nonempty_file(const char *pathname)
1234 {
1235         struct stat buf;
1236
1237         if (stat(pathname, &buf) == -1)
1238                 return 0;
1239
1240         return S_ISREG(buf.st_mode) && buf.st_size > 0;
1241 }