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