crunchgen.c revision 56995
11722Sjkh/*
21722Sjkh * Copyright (c) 1994 University of Maryland
31722Sjkh * All Rights Reserved.
41722Sjkh *
51722Sjkh * Permission to use, copy, modify, distribute, and sell this software and its
61722Sjkh * documentation for any purpose is hereby granted without fee, provided that
71722Sjkh * the above copyright notice appear in all copies and that both that
81722Sjkh * copyright notice and this permission notice appear in supporting
91722Sjkh * documentation, and that the name of U.M. not be used in advertising or
101722Sjkh * publicity pertaining to distribution of the software without specific,
111722Sjkh * written prior permission.  U.M. makes no representations about the
121722Sjkh * suitability of this software for any purpose.  It is provided "as is"
131722Sjkh * without express or implied warranty.
141722Sjkh *
151722Sjkh * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
161722Sjkh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
171722Sjkh * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
181722Sjkh * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
191722Sjkh * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
201722Sjkh * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
211722Sjkh *
221722Sjkh * Author: James da Silva, Systems Design and Analysis Group
231722Sjkh *			   Computer Science Department
241722Sjkh *			   University of Maryland at College Park
2556995Sluigi *
2656995Sluigi * $FreeBSD: head/usr.sbin/crunch/crunchgen/crunchgen.c 56995 2000-02-05 10:29:19Z luigi $
271722Sjkh */
281722Sjkh/*
291722Sjkh * ========================================================================
301722Sjkh * crunchgen.c
311722Sjkh *
321722Sjkh * Generates a Makefile and main C file for a crunched executable,
338857Srgrimes * from specs given in a .conf file.
341722Sjkh */
3529453Scharnier#include <ctype.h>
3629453Scharnier#include <err.h>
3729453Scharnier#include <stdio.h>
381722Sjkh#include <stdlib.h>
3929453Scharnier#include <string.h>
401722Sjkh#include <unistd.h>
411722Sjkh
421722Sjkh#include <sys/types.h>
431722Sjkh#include <sys/stat.h>
441722Sjkh#include <sys/param.h>
451722Sjkh
461722Sjkh#define CRUNCH_VERSION	"0.2"
471722Sjkh
481722Sjkh#define MAXLINELEN	16384
491722Sjkh#define MAXFIELDS 	 2048
501722Sjkh
511722Sjkh
521722Sjkh/* internal representation of conf file: */
531722Sjkh
541722Sjkh/* simple lists of strings suffice for most parms */
551722Sjkh
561722Sjkhtypedef struct strlst {
571722Sjkh    struct strlst *next;
581722Sjkh    char *str;
591722Sjkh} strlst_t;
601722Sjkh
611722Sjkh/* progs have structure, each field can be set with "special" or calculated */
621722Sjkh
631722Sjkhtypedef struct prog {
6456995Sluigi    struct prog *next;		/* link field */
6556995Sluigi    char *name;			/* program name */
6656995Sluigi    char *ident;		/* C identifier for the program name */
6756995Sluigi    char *srcdir;
6856995Sluigi    char *objdir;
6956995Sluigi    char *objvar;		/* Makefile variable to replace OBJS */
701722Sjkh    strlst_t *objs, *objpaths;
7130120Sjoerg    strlst_t *keeplist;
721722Sjkh    strlst_t *links;
731722Sjkh    int goterror;
741722Sjkh} prog_t;
751722Sjkh
761722Sjkh
771722Sjkh/* global state */
781722Sjkh
791722Sjkhstrlst_t *srcdirs = NULL;
801722Sjkhstrlst_t *libs    = NULL;
811722Sjkhprog_t   *progs   = NULL;
821722Sjkh
831722Sjkhchar line[MAXLINELEN];
841722Sjkh
851722Sjkhchar confname[MAXPATHLEN], infilename[MAXPATHLEN];
861722Sjkhchar outmkname[MAXPATHLEN], outcfname[MAXPATHLEN], execfname[MAXPATHLEN];
871722Sjkhchar tempfname[MAXPATHLEN], cachename[MAXPATHLEN], curfilename[MAXPATHLEN];
8856995Sluigichar outhdrname[MAXPATHLEN] ; /* user-supplied header for *.mk */
891722Sjkhint linenum = -1;
901722Sjkhint goterror = 0;
911722Sjkh
921722Sjkhint verbose, readcache;	/* options */
931722Sjkhint reading_cache;
941722Sjkh
956696Sphkint list_mode;
966696Sphk
971722Sjkh/* general library routines */
981722Sjkh
991722Sjkhvoid status(char *str);
1001722Sjkhvoid out_of_memory(void);
1011722Sjkhvoid add_string(strlst_t **listp, char *str);
1021722Sjkhint is_dir(char *pathname);
1031722Sjkhint is_nonempty_file(char *pathname);
1041722Sjkh
1051722Sjkh/* helper routines for main() */
1061722Sjkh
1078857Srgrimesvoid usage(void);
1081722Sjkhvoid parse_conf_file(void);
1091722Sjkhvoid gen_outputs(void);
1101722Sjkh
1111722Sjkh
1121722Sjkhint main(int argc, char **argv)
1131722Sjkh{
1141722Sjkh    char *p;
1151722Sjkh    int optc;
1161722Sjkh
1171722Sjkh    verbose = 1;
1181722Sjkh    readcache = 1;
1191722Sjkh    *outmkname = *outcfname = *execfname = '\0';
1208857Srgrimes
12156995Sluigi    while((optc = getopt(argc, argv, "lh:m:c:e:fq")) != -1) {
1221722Sjkh	switch(optc) {
1231722Sjkh	case 'f':	readcache = 0; break;
1241722Sjkh	case 'q':	verbose = 0; break;
1251722Sjkh
1261722Sjkh	case 'm':	strcpy(outmkname, optarg); break;
12756995Sluigi	case 'h':	strcpy(outhdrname, optarg); break;
1281722Sjkh	case 'c':	strcpy(outcfname, optarg); break;
1291722Sjkh	case 'e':	strcpy(execfname, optarg); break;
1306696Sphk	case 'l':	list_mode++; verbose = 0; break;
1311722Sjkh
1321722Sjkh	case '?':
1331722Sjkh	default:	usage();
1341722Sjkh	}
1351722Sjkh    }
1361722Sjkh
1371722Sjkh    argc -= optind;
1381722Sjkh    argv += optind;
1391722Sjkh
1401722Sjkh    if(argc != 1) usage();
1411722Sjkh
1428857Srgrimes    /*
1431722Sjkh     * generate filenames
1441722Sjkh     */
1451722Sjkh
1461722Sjkh    strcpy(infilename, argv[0]);
1471722Sjkh
1481722Sjkh    /* confname = `basename infilename .conf` */
1491722Sjkh
1501722Sjkh    if((p=strrchr(infilename, '/')) != NULL) strcpy(confname, p+1);
1511722Sjkh    else strcpy(confname, infilename);
1521722Sjkh    if((p=strrchr(confname, '.')) != NULL && !strcmp(p, ".conf")) *p = '\0';
1531722Sjkh
1541722Sjkh    if(!*outmkname) sprintf(outmkname, "%s.mk", confname);
1551722Sjkh    if(!*outcfname) sprintf(outcfname, "%s.c", confname);
1561722Sjkh    if(!*execfname) sprintf(execfname, "%s", confname);
1571722Sjkh
1581722Sjkh    sprintf(cachename, "%s.cache", confname);
1591722Sjkh    sprintf(tempfname, ".tmp_%sXXXXXX", confname);
1601722Sjkh    if(mktemp(tempfname) == NULL) {
1611722Sjkh	perror(tempfname);
1621722Sjkh	exit(1);
1631722Sjkh    }
1641722Sjkh
1651722Sjkh    parse_conf_file();
1666696Sphk    if (list_mode)
1676696Sphk	exit(goterror);
1686696Sphk
1691722Sjkh    gen_outputs();
1701722Sjkh
1711722Sjkh    exit(goterror);
1721722Sjkh}
1731722Sjkh
1741722Sjkh
1751722Sjkhvoid usage(void)
1761722Sjkh{
17729453Scharnier    fprintf(stderr, "%s\n%s\n",
17829453Scharnier		"usage: crunchgen [-fq] [-m <makefile>] [-c <c file>]",
17929453Scharnier		"                 [-e <exec file>] <conffile>");
1801722Sjkh    exit(1);
1811722Sjkh}
1821722Sjkh
1831722Sjkh
1841722Sjkh/*
1851722Sjkh * ========================================================================
1861722Sjkh * parse_conf_file subsystem
1871722Sjkh *
1881722Sjkh */
1891722Sjkh
1901722Sjkh/* helper routines for parse_conf_file */
1911722Sjkh
1921722Sjkhvoid parse_one_file(char *filename);
1938857Srgrimesvoid parse_line(char *line, int *fc, char **fv, int nf);
1941722Sjkhvoid add_srcdirs(int argc, char **argv);
1951722Sjkhvoid add_progs(int argc, char **argv);
1961722Sjkhvoid add_link(int argc, char **argv);
1971722Sjkhvoid add_libs(int argc, char **argv);
1981722Sjkhvoid add_special(int argc, char **argv);
1991722Sjkh
2001722Sjkhprog_t *find_prog(char *str);
2011722Sjkhvoid add_prog(char *progname);
2021722Sjkh
2031722Sjkh
2041722Sjkhvoid parse_conf_file(void)
2051722Sjkh{
20629453Scharnier    if(!is_nonempty_file(infilename))
20729453Scharnier		errx(1, "fatal: input file \"%s\" not found", infilename);
2081722Sjkh    parse_one_file(infilename);
2091722Sjkh    if(readcache && is_nonempty_file(cachename)) {
2101722Sjkh	reading_cache = 1;
2111722Sjkh	parse_one_file(cachename);
2121722Sjkh    }
2131722Sjkh}
2141722Sjkh
2151722Sjkh
2161722Sjkhvoid parse_one_file(char *filename)
2171722Sjkh{
2181722Sjkh    char *fieldv[MAXFIELDS];
2191722Sjkh    int fieldc;
2201722Sjkh    void (*f)(int c, char **v);
2211722Sjkh    FILE *cf;
2221722Sjkh
2231722Sjkh    sprintf(line, "reading %s", filename);
2241722Sjkh    status(line);
2251722Sjkh    strcpy(curfilename, filename);
2261722Sjkh
2271722Sjkh    if((cf = fopen(curfilename, "r")) == NULL) {
22829453Scharnier	warn("%s", curfilename);
2291722Sjkh	goterror = 1;
2301722Sjkh	return;
2311722Sjkh    }
2321722Sjkh
2331722Sjkh    linenum = 0;
2341722Sjkh    while(fgets(line, MAXLINELEN, cf) != NULL) {
2351722Sjkh	linenum++;
2361722Sjkh	parse_line(line, &fieldc, fieldv, MAXFIELDS);
2371722Sjkh	if(fieldc < 1) continue;
2381722Sjkh	if(!strcmp(fieldv[0], "srcdirs"))	f = add_srcdirs;
2391722Sjkh	else if(!strcmp(fieldv[0], "progs"))    f = add_progs;
2401722Sjkh	else if(!strcmp(fieldv[0], "ln"))	f = add_link;
2411722Sjkh	else if(!strcmp(fieldv[0], "libs"))	f = add_libs;
2421722Sjkh	else if(!strcmp(fieldv[0], "special"))	f = add_special;
2431722Sjkh	else {
24429453Scharnier	    warnx("%s:%d: skipping unknown command `%s'",
2451722Sjkh		    curfilename, linenum, fieldv[0]);
2461722Sjkh	    goterror = 1;
2471722Sjkh	    continue;
2481722Sjkh	}
2491722Sjkh	if(fieldc < 2) {
25029453Scharnier	    warnx("%s:%d: %s command needs at least 1 argument, skipping",
2511722Sjkh		    curfilename, linenum, fieldv[0]);
2521722Sjkh	    goterror = 1;
2531722Sjkh	    continue;
2541722Sjkh	}
2551722Sjkh	f(fieldc, fieldv);
2561722Sjkh    }
2571722Sjkh
2581722Sjkh    if(ferror(cf)) {
25929453Scharnier	warn("%s", curfilename);
2601722Sjkh	goterror = 1;
2611722Sjkh    }
2621722Sjkh    fclose(cf);
2631722Sjkh}
2641722Sjkh
2651722Sjkh
2661722Sjkhvoid parse_line(char *line, int *fc, char **fv, int nf)
2671722Sjkh{
2681722Sjkh    char *p;
2691722Sjkh
2701722Sjkh    p = line;
2711722Sjkh    *fc = 0;
2721722Sjkh    while(1) {
2731722Sjkh	while(isspace(*p)) p++;
2741722Sjkh	if(*p == '\0' || *p == '#') break;
2751722Sjkh
2761722Sjkh	if(*fc < nf) fv[(*fc)++] = p;
2771722Sjkh	while(*p && !isspace(*p) && *p != '#') p++;
2781722Sjkh	if(*p == '\0' || *p == '#') break;
2791722Sjkh	*p++ = '\0';
2801722Sjkh    }
2811722Sjkh    if(*p) *p = '\0';		/* needed for '#' case */
2821722Sjkh}
2831722Sjkh
2841722Sjkh
2851722Sjkhvoid add_srcdirs(int argc, char **argv)
2861722Sjkh{
2871722Sjkh    int i;
2881722Sjkh
2891722Sjkh    for(i=1;i<argc;i++) {
2901722Sjkh	if(is_dir(argv[i]))
2911722Sjkh	    add_string(&srcdirs, argv[i]);
2921722Sjkh	else {
29329453Scharnier	    warnx("%s:%d: `%s' is not a directory, skipping it",
2941722Sjkh		    curfilename, linenum, argv[i]);
2951722Sjkh	    goterror = 1;
2961722Sjkh	}
2971722Sjkh    }
2981722Sjkh}
2991722Sjkh
3001722Sjkh
3011722Sjkhvoid add_progs(int argc, char **argv)
3021722Sjkh{
3031722Sjkh    int i;
3041722Sjkh
3051722Sjkh    for(i=1;i<argc;i++)
3061722Sjkh	add_prog(argv[i]);
3071722Sjkh}
3081722Sjkh
3091722Sjkh
3101722Sjkhvoid add_prog(char *progname)
3111722Sjkh{
3121722Sjkh    prog_t *p1, *p2;
3131722Sjkh
3141722Sjkh    /* add to end, but be smart about dups */
3151722Sjkh
3161722Sjkh    for(p1 = NULL, p2 = progs; p2 != NULL; p1 = p2, p2 = p2->next)
3171722Sjkh	if(!strcmp(p2->name, progname)) return;
3181722Sjkh
3191722Sjkh    p2 = malloc(sizeof(prog_t));
32019822Sjoerg    if(p2) {
32119822Sjoerg	memset(p2, 0, sizeof(prog_t));
32219822Sjoerg	p2->name = strdup(progname);
32319822Sjoerg    }
3248857Srgrimes    if(!p2 || !p2->name)
3251722Sjkh	out_of_memory();
3261722Sjkh
3271722Sjkh    p2->next = NULL;
3281722Sjkh    if(p1 == NULL) progs = p2;
3291722Sjkh    else p1->next = p2;
3301722Sjkh
3311722Sjkh    p2->ident = p2->srcdir = p2->objdir = NULL;
33230120Sjoerg    p2->links = p2->objs = p2->keeplist = NULL;
3331722Sjkh    p2->goterror = 0;
3346696Sphk    if (list_mode)
3356696Sphk        printf("%s\n",progname);
3361722Sjkh}
3371722Sjkh
3381722Sjkh
3391722Sjkhvoid add_link(int argc, char **argv)
3401722Sjkh{
3411722Sjkh    int i;
3421722Sjkh    prog_t *p = find_prog(argv[1]);
3431722Sjkh
3441722Sjkh    if(p == NULL) {
34529453Scharnier	warnx("%s:%d: no prog %s previously declared, skipping link",
3461722Sjkh		curfilename, linenum, argv[1]);
3471722Sjkh	goterror = 1;
3481722Sjkh	return;
3491722Sjkh    }
3506696Sphk    for(i=2;i<argc;i++) {
3516696Sphk	if (list_mode)
3526696Sphk		printf("%s\n",argv[i]);
3531722Sjkh	add_string(&p->links, argv[i]);
3546696Sphk    }
3551722Sjkh}
3561722Sjkh
3571722Sjkh
3581722Sjkhvoid add_libs(int argc, char **argv)
3591722Sjkh{
3601722Sjkh    int i;
3611722Sjkh
3621722Sjkh    for(i=1;i<argc;i++)
3631722Sjkh	add_string(&libs, argv[i]);
3641722Sjkh}
3651722Sjkh
3661722Sjkh
3671722Sjkhvoid add_special(int argc, char **argv)
3681722Sjkh{
3691722Sjkh    int i;
3701722Sjkh    prog_t *p = find_prog(argv[1]);
3711722Sjkh
3721722Sjkh    if(p == NULL) {
3731722Sjkh	if(reading_cache) return;
37429453Scharnier	warnx("%s:%d: no prog %s previously declared, skipping special",
3751722Sjkh		curfilename, linenum, argv[1]);
3761722Sjkh	goterror = 1;
3771722Sjkh	return;
3781722Sjkh    }
3791722Sjkh
3801722Sjkh    if(!strcmp(argv[2], "ident")) {
3811722Sjkh	if(argc != 4) goto argcount;
3821722Sjkh	if((p->ident = strdup(argv[3])) == NULL)
3831722Sjkh	    out_of_memory();
3841722Sjkh    }
3851722Sjkh    else if(!strcmp(argv[2], "srcdir")) {
3861722Sjkh	if(argc != 4) goto argcount;
3871722Sjkh	if((p->srcdir = strdup(argv[3])) == NULL)
3881722Sjkh	    out_of_memory();
3891722Sjkh    }
3901722Sjkh    else if(!strcmp(argv[2], "objdir")) {
3911722Sjkh	if(argc != 4) goto argcount;
3921722Sjkh	if((p->objdir = strdup(argv[3])) == NULL)
3931722Sjkh	    out_of_memory();
3941722Sjkh    }
3951722Sjkh    else if(!strcmp(argv[2], "objs")) {
3961722Sjkh	p->objs = NULL;
3971722Sjkh	for(i=3;i<argc;i++)
3981722Sjkh	    add_string(&p->objs, argv[i]);
3991722Sjkh    }
4001722Sjkh    else if(!strcmp(argv[2], "objpaths")) {
4011722Sjkh	p->objpaths = NULL;
4021722Sjkh	for(i=3;i<argc;i++)
4031722Sjkh	    add_string(&p->objpaths, argv[i]);
4041722Sjkh    }
40530120Sjoerg    else if(!strcmp(argv[2], "keep")) {
40630120Sjoerg	p->keeplist = NULL;
40730120Sjoerg	for(i=3;i<argc;i++)
40830120Sjoerg	    add_string(&p->keeplist, argv[i]);
40930120Sjoerg    }
41056995Sluigi    else if(!strcmp(argv[2], "objvar")) {
41156995Sluigi	if(argc != 4)
41256995Sluigi	    goto argcount;
41356995Sluigi	if((p->objvar = strdup(argv[3])) == NULL)
41456995Sluigi	    out_of_memory();
41556995Sluigi    }
4161722Sjkh    else {
41729453Scharnier	warnx("%s:%d: bad parameter name `%s', skipping line",
4181722Sjkh		curfilename, linenum, argv[2]);
4191722Sjkh	goterror = 1;
4201722Sjkh    }
4211722Sjkh    return;
4221722Sjkh
4231722Sjkh
4241722Sjkh argcount:
42529453Scharnier    warnx("%s:%d: too %s arguments, expected \"special %s %s <string>\"",
4261722Sjkh	    curfilename, linenum, argc < 4? "few" : "many", argv[1], argv[2]);
4271722Sjkh    goterror = 1;
4281722Sjkh}
4291722Sjkh
4301722Sjkh
4311722Sjkhprog_t *find_prog(char *str)
4321722Sjkh{
4331722Sjkh    prog_t *p;
4341722Sjkh
4351722Sjkh    for(p = progs; p != NULL; p = p->next)
4361722Sjkh	if(!strcmp(p->name, str)) return p;
4371722Sjkh
4381722Sjkh    return NULL;
4391722Sjkh}
4401722Sjkh
4411722Sjkh
4421722Sjkh/*
4431722Sjkh * ========================================================================
4441722Sjkh * gen_outputs subsystem
4451722Sjkh *
4461722Sjkh */
4471722Sjkh
4481722Sjkh/* helper subroutines */
4491722Sjkh
4501722Sjkhvoid remove_error_progs(void);
4511722Sjkhvoid fillin_program(prog_t *p);
4521722Sjkhvoid gen_specials_cache(void);
4531722Sjkhvoid gen_output_makefile(void);
4541722Sjkhvoid gen_output_cfile(void);
4551722Sjkh
4561722Sjkhvoid fillin_program_objs(prog_t *p, char *path);
4571722Sjkhvoid top_makefile_rules(FILE *outmk);
4581722Sjkhvoid prog_makefile_rules(FILE *outmk, prog_t *p);
4591722Sjkhvoid output_strlst(FILE *outf, strlst_t *lst);
4601722Sjkhchar *genident(char *str);
4611722Sjkhchar *dir_search(char *progname);
4621722Sjkh
4631722Sjkh
4641722Sjkhvoid gen_outputs(void)
4651722Sjkh{
4661722Sjkh    prog_t *p;
4671722Sjkh
4681722Sjkh    for(p = progs; p != NULL; p = p->next)
4691722Sjkh	fillin_program(p);
4701722Sjkh
4711722Sjkh    remove_error_progs();
4721722Sjkh    gen_specials_cache();
4731722Sjkh    gen_output_cfile();
4741722Sjkh    gen_output_makefile();
4751722Sjkh    status("");
4768857Srgrimes    fprintf(stderr,
4771722Sjkh	    "Run \"make -f %s objs exe\" to build crunched binary.\n",
4781722Sjkh	    outmkname);
4791722Sjkh}
4801722Sjkh
48156995Sluigi/*
48256995Sluigi * run the makefile for the program to find which objects are necessary
48356995Sluigi */
4841722Sjkhvoid fillin_program(prog_t *p)
4851722Sjkh{
4861722Sjkh    char path[MAXPATHLEN];
4871722Sjkh    char *srcparent;
4881722Sjkh    strlst_t *s;
4891722Sjkh
4901722Sjkh    sprintf(line, "filling in parms for %s", p->name);
4911722Sjkh    status(line);
4921722Sjkh
4938857Srgrimes    if(!p->ident)
4941722Sjkh	p->ident = genident(p->name);
4951722Sjkh    if(!p->srcdir) {
4961722Sjkh	srcparent = dir_search(p->name);
4971722Sjkh	if(srcparent)
4981722Sjkh	    sprintf(path, "%s/%s", srcparent, p->name);
4991722Sjkh	if(is_dir(path))
5001722Sjkh	    p->srcdir = strdup(path);
5011722Sjkh    }
5021722Sjkh    if(!p->objdir && p->srcdir) {
50317430Sphk	FILE *f;
50417430Sphk
50532704Ssos	sprintf(path, "cd %s && echo -n /usr/obj`/bin/pwd`", p->srcdir);
50617430Sphk        p->objdir = p->srcdir;
50717430Sphk	f = popen(path,"r");
50817430Sphk	if (f) {
50917430Sphk	    fgets(path,sizeof path, f);
51017430Sphk	    if (!pclose(f)) {
51117430Sphk		if(is_dir(path))
51217430Sphk		    p->objdir = strdup(path);
51317430Sphk	    }
51417430Sphk	}
5151722Sjkh    }
51656995Sluigi/*
51756995Sluigi * XXX look for a Makefile.{name} in local directory first.
51856995Sluigi * This lets us override the original Makefile.
51956995Sluigi */
52056995Sluigi    sprintf(path, "Makefile.%s", p->name);
52156995Sluigi    if (is_nonempty_file(path)) {
52256995Sluigi       sprintf(line, "Using %s for %s", path, p->name);
52356995Sluigi       status(line);
52456995Sluigi    } else
5251722Sjkh    if(p->srcdir) sprintf(path, "%s/Makefile", p->srcdir);
5261722Sjkh    if(!p->objs && p->srcdir && is_nonempty_file(path))
5271722Sjkh	fillin_program_objs(p, path);
5281722Sjkh
5291722Sjkh    if(!p->objpaths && p->objdir && p->objs)
5301722Sjkh	for(s = p->objs; s != NULL; s = s->next) {
5311722Sjkh	    sprintf(line, "%s/%s", p->objdir, s->str);
5321722Sjkh	    add_string(&p->objpaths, line);
5331722Sjkh	}
5341722Sjkh
5351722Sjkh    if(!p->srcdir && verbose)
53629453Scharnier	warnx("%s: %s: warning: could not find source directory",
5371722Sjkh		infilename, p->name);
5381722Sjkh    if(!p->objs && verbose)
53929453Scharnier	warnx("%s: %s: warning: could not find any .o files",
5401722Sjkh		infilename, p->name);
5411722Sjkh
5421722Sjkh    if(!p->objpaths) {
54329453Scharnier	warnx("%s: %s: error: no objpaths specified or calculated",
5441722Sjkh		infilename, p->name);
5451722Sjkh	p->goterror = goterror = 1;
5461722Sjkh    }
5471722Sjkh}
5481722Sjkh
5491722Sjkhvoid fillin_program_objs(prog_t *p, char *path)
5501722Sjkh{
5511722Sjkh    char *obj, *cp;
5521722Sjkh    int rc;
5531722Sjkh    FILE *f;
55456995Sluigi    char *objvar="OBJS";
5551722Sjkh
5561722Sjkh    /* discover the objs from the srcdir Makefile */
5571722Sjkh
5581722Sjkh    if((f = fopen(tempfname, "w")) == NULL) {
55929453Scharnier	warn("%s", tempfname);
5601722Sjkh	goterror = 1;
5611722Sjkh	return;
5621722Sjkh    }
56356995Sluigi    if (p->objvar)
56456995Sluigi	objvar = p->objvar ;
5658857Srgrimes
56656995Sluigi    /*
56756995Sluigi     * XXX include outhdrname (e.g. to contain Make variables)
56856995Sluigi     */
56956995Sluigi    if (outhdrname[0] != '\0')
57056995Sluigi	fprintf(f, ".include \"%s\"\n", outhdrname);
5711722Sjkh    fprintf(f, ".include \"%s\"\n", path);
57256995Sluigi    fprintf(f, ".if defined(PROG) && !defined(%s)\n", objvar);
57356995Sluigi    fprintf(f, "%s=${PROG}.o\n", objvar);
5741722Sjkh    fprintf(f, ".endif\n");
57556995Sluigi    fprintf(f, "crunchgen_objs:\n\t@echo 'OBJS= '${%s}\n", objvar);
5761722Sjkh    fclose(f);
5771722Sjkh
5781722Sjkh    sprintf(line, "make -f %s crunchgen_objs 2>&1", tempfname);
5791722Sjkh    if((f = popen(line, "r")) == NULL) {
58029453Scharnier	warn("submake pipe");
5811722Sjkh	goterror = 1;
5821722Sjkh	return;
5831722Sjkh    }
5841722Sjkh
5851722Sjkh    while(fgets(line, MAXLINELEN, f)) {
5861722Sjkh	if(strncmp(line, "OBJS= ", 6)) {
58729453Scharnier	    warnx("make error: %s", line);
5888857Srgrimes	    goterror = 1;
5891722Sjkh	    continue;
5901722Sjkh	}
5911722Sjkh	cp = line + 6;
5921722Sjkh	while(isspace(*cp)) cp++;
5931722Sjkh	while(*cp) {
5941722Sjkh	    obj = cp;
5951722Sjkh	    while(*cp && !isspace(*cp)) cp++;
5961722Sjkh	    if(*cp) *cp++ = '\0';
5971722Sjkh	    add_string(&p->objs, obj);
5981722Sjkh	    while(isspace(*cp)) cp++;
5991722Sjkh	}
6001722Sjkh    }
6011722Sjkh    if((rc=pclose(f)) != 0) {
60229453Scharnier	warnx("make error: make returned %d", rc);
6031722Sjkh	goterror = 1;
6041722Sjkh    }
6051722Sjkh    unlink(tempfname);
6061722Sjkh}
6071722Sjkh
6081722Sjkhvoid remove_error_progs(void)
6091722Sjkh{
6101722Sjkh    prog_t *p1, *p2;
6111722Sjkh
6128857Srgrimes    p1 = NULL; p2 = progs;
6138857Srgrimes    while(p2 != NULL) {
6141722Sjkh	if(!p2->goterror)
6151722Sjkh	    p1 = p2, p2 = p2->next;
6161722Sjkh	else {
6171722Sjkh	    /* delete it from linked list */
61829453Scharnier	    warnx("%s: %s: ignoring program because of errors",
6191722Sjkh		    infilename, p2->name);
6201722Sjkh	    if(p1) p1->next = p2->next;
6211722Sjkh	    else progs = p2->next;
6221722Sjkh	    p2 = p2->next;
6231722Sjkh	}
6241722Sjkh    }
6251722Sjkh}
6261722Sjkh
6271722Sjkhvoid gen_specials_cache(void)
6281722Sjkh{
6291722Sjkh    FILE *cachef;
6301722Sjkh    prog_t *p;
6311722Sjkh
6321722Sjkh    sprintf(line, "generating %s", cachename);
6331722Sjkh    status(line);
6341722Sjkh
6351722Sjkh    if((cachef = fopen(cachename, "w")) == NULL) {
63629453Scharnier	warn("%s", cachename);
6371722Sjkh	goterror = 1;
6381722Sjkh	return;
6391722Sjkh    }
6401722Sjkh
6411722Sjkh    fprintf(cachef, "# %s - parm cache generated from %s by crunchgen %s\n\n",
6421722Sjkh	    cachename, infilename, CRUNCH_VERSION);
6431722Sjkh
6441722Sjkh    for(p = progs; p != NULL; p = p->next) {
6451722Sjkh	fprintf(cachef, "\n");
6461722Sjkh	if(p->srcdir)
6471722Sjkh	    fprintf(cachef, "special %s srcdir %s\n", p->name, p->srcdir);
6481722Sjkh	if(p->objdir)
6491722Sjkh	    fprintf(cachef, "special %s objdir %s\n", p->name, p->objdir);
6501722Sjkh	if(p->objs) {
6511722Sjkh	    fprintf(cachef, "special %s objs", p->name);
6521722Sjkh	    output_strlst(cachef, p->objs);
6531722Sjkh	}
6541722Sjkh	fprintf(cachef, "special %s objpaths", p->name);
6551722Sjkh	output_strlst(cachef, p->objpaths);
6561722Sjkh    }
6571722Sjkh    fclose(cachef);
6581722Sjkh}
6591722Sjkh
6601722Sjkh
6611722Sjkhvoid gen_output_makefile(void)
6621722Sjkh{
6631722Sjkh    prog_t *p;
6641722Sjkh    FILE *outmk;
6651722Sjkh
6661722Sjkh    sprintf(line, "generating %s", outmkname);
6671722Sjkh    status(line);
6681722Sjkh
6691722Sjkh    if((outmk = fopen(outmkname, "w")) == NULL) {
67029453Scharnier	warn("%s", outmkname);
6711722Sjkh	goterror = 1;
6721722Sjkh	return;
6731722Sjkh    }
6741722Sjkh
6751722Sjkh    fprintf(outmk, "# %s - generated from %s by crunchgen %s\n\n",
6761722Sjkh	    outmkname, infilename, CRUNCH_VERSION);
67756995Sluigi    if (outhdrname[0] != '\0')
67856995Sluigi	fprintf(outmk, ".include \"%s\"\n", outhdrname);
6791722Sjkh
6801722Sjkh    top_makefile_rules(outmk);
6811722Sjkh
6821722Sjkh    for(p = progs; p != NULL; p = p->next)
6838857Srgrimes	prog_makefile_rules(outmk, p);
6841722Sjkh
6851722Sjkh    fprintf(outmk, "\n# ========\n");
6861722Sjkh    fclose(outmk);
6871722Sjkh}
6881722Sjkh
6891722Sjkh
6901722Sjkhvoid gen_output_cfile(void)
6911722Sjkh{
6921722Sjkh    extern char *crunched_skel[];
6931722Sjkh    char **cp;
6941722Sjkh    FILE *outcf;
6951722Sjkh    prog_t *p;
6961722Sjkh    strlst_t *s;
6971722Sjkh
6981722Sjkh    sprintf(line, "generating %s", outcfname);
6991722Sjkh    status(line);
7001722Sjkh
7011722Sjkh    if((outcf = fopen(outcfname, "w")) == NULL) {
70229453Scharnier	warn("%s", outcfname);
7031722Sjkh	goterror = 1;
7041722Sjkh	return;
7051722Sjkh    }
7061722Sjkh
7078857Srgrimes    fprintf(outcf,
7081722Sjkh	  "/* %s - generated from %s by crunchgen %s */\n",
7091722Sjkh	    outcfname, infilename, CRUNCH_VERSION);
7101722Sjkh
7111722Sjkh    fprintf(outcf, "#define EXECNAME \"%s\"\n", execfname);
7121722Sjkh    for(cp = crunched_skel; *cp != NULL; cp++)
7131722Sjkh	fprintf(outcf, "%s\n", *cp);
7141722Sjkh
7151722Sjkh    for(p = progs; p != NULL; p = p->next)
7161722Sjkh	fprintf(outcf, "extern int _crunched_%s_stub();\n", p->ident);
7171722Sjkh
7181722Sjkh    fprintf(outcf, "\nstruct stub entry_points[] = {\n");
7191722Sjkh    for(p = progs; p != NULL; p = p->next) {
7201722Sjkh	fprintf(outcf, "\t{ \"%s\", _crunched_%s_stub },\n",
7211722Sjkh		p->name, p->ident);
7221722Sjkh	for(s = p->links; s != NULL; s = s->next)
7231722Sjkh	    fprintf(outcf, "\t{ \"%s\", _crunched_%s_stub },\n",
7241722Sjkh		    s->str, p->ident);
7251722Sjkh    }
7268857Srgrimes
7271722Sjkh    fprintf(outcf, "\t{ EXECNAME, crunched_main },\n");
7281722Sjkh    fprintf(outcf, "\t{ NULL, NULL }\n};\n");
7291722Sjkh    fclose(outcf);
7301722Sjkh}
7311722Sjkh
7321722Sjkh
7331722Sjkhchar *genident(char *str)
7341722Sjkh{
7351722Sjkh    char *n,*s,*d;
7361722Sjkh
7371722Sjkh    /*
7381722Sjkh     * generates a Makefile/C identifier from a program name, mapping '-' to
7391722Sjkh     * '_' and ignoring all other non-identifier characters.  This leads to
7401722Sjkh     * programs named "foo.bar" and "foobar" to map to the same identifier.
7411722Sjkh     */
7421722Sjkh
7431722Sjkh    if((n = strdup(str)) == NULL)
7441722Sjkh	return NULL;
7451722Sjkh    for(d = s = n; *s != '\0'; s++) {
7461722Sjkh	if(*s == '-') *d++ = '_';
7471722Sjkh	else if(*s == '_' || isalnum(*s)) *d++ = *s;
7481722Sjkh    }
7491722Sjkh    *d = '\0';
7501722Sjkh    return n;
7511722Sjkh}
7521722Sjkh
7531722Sjkh
7541722Sjkhchar *dir_search(char *progname)
7551722Sjkh{
7561722Sjkh    char path[MAXPATHLEN];
7571722Sjkh    strlst_t *dir;
7581722Sjkh
7591722Sjkh    for(dir=srcdirs; dir != NULL; dir=dir->next) {
7601722Sjkh	sprintf(path, "%s/%s", dir->str, progname);
7611722Sjkh	if(is_dir(path)) return dir->str;
7621722Sjkh    }
7631722Sjkh    return NULL;
7641722Sjkh}
7651722Sjkh
7661722Sjkh
7671722Sjkhvoid top_makefile_rules(FILE *outmk)
7681722Sjkh{
7691722Sjkh    prog_t *p;
7701722Sjkh
7711722Sjkh    fprintf(outmk, "LIBS=");
7721722Sjkh    output_strlst(outmk, libs);
7731722Sjkh
7741722Sjkh    fprintf(outmk, "CRUNCHED_OBJS=");
7751722Sjkh    for(p = progs; p != NULL; p = p->next)
7761722Sjkh	fprintf(outmk, " %s.lo", p->name);
7771722Sjkh    fprintf(outmk, "\n");
7781722Sjkh
7791722Sjkh    fprintf(outmk, "SUBMAKE_TARGETS=");
7801722Sjkh    for(p = progs; p != NULL; p = p->next)
7811722Sjkh	fprintf(outmk, " %s_make", p->ident);
78232589Sbrian    fprintf(outmk, "\nSUBCLEAN_TARGETS=");
78332589Sbrian    for(p = progs; p != NULL; p = p->next)
78432589Sbrian	fprintf(outmk, " %s_clean", p->ident);
7851722Sjkh    fprintf(outmk, "\n\n");
7861722Sjkh
7878857Srgrimes    fprintf(outmk, "%s: %s.o $(CRUNCHED_OBJS)\n",
7881722Sjkh	    execfname, execfname);
7891722Sjkh    fprintf(outmk, "\t$(CC) -static -o %s %s.o $(CRUNCHED_OBJS) $(LIBS)\n",
7901722Sjkh	    execfname, execfname);
7911722Sjkh    fprintf(outmk, "\tstrip %s\n", execfname);
7921722Sjkh    fprintf(outmk, "all: objs exe\nobjs: $(SUBMAKE_TARGETS)\n");
7931722Sjkh    fprintf(outmk, "exe: %s\n", execfname);
79432589Sbrian    fprintf(outmk, "realclean: clean subclean\n");
7951722Sjkh    fprintf(outmk, "clean:\n\trm -f %s *.lo *.o *_stub.c\n",
7961722Sjkh	    execfname);
79732589Sbrian    fprintf(outmk, "subclean: $(SUBCLEAN_TARGETS)\n");
7981722Sjkh}
7991722Sjkh
8001722Sjkh
8011722Sjkhvoid prog_makefile_rules(FILE *outmk, prog_t *p)
8021722Sjkh{
80330120Sjoerg    strlst_t *lst;
80430120Sjoerg
8051722Sjkh    fprintf(outmk, "\n# -------- %s\n\n", p->name);
8061722Sjkh
8071722Sjkh    if(p->srcdir && p->objs) {
8081722Sjkh	fprintf(outmk, "%s_SRCDIR=%s\n", p->ident, p->srcdir);
8091722Sjkh	fprintf(outmk, "%s_OBJS=", p->ident);
8101722Sjkh	output_strlst(outmk, p->objs);
8111722Sjkh	fprintf(outmk, "%s_make:\n", p->ident);
81232589Sbrian	fprintf(outmk, "\t(cd $(%s_SRCDIR) && make depend && make $(%s_OBJS))\n",
8131722Sjkh		p->ident, p->ident);
81432589Sbrian	fprintf(outmk, "%s_clean:\n", p->ident);
81532589Sbrian	fprintf(outmk, "\t(cd $(%s_SRCDIR) && make clean)\n\n", p->ident);
8161722Sjkh    }
8171722Sjkh    else
8188857Srgrimes	fprintf(outmk, "%s_make:\n\t@echo \"** cannot make objs for %s\"\n\n",
8191722Sjkh		p->ident, p->name);
8201722Sjkh
8211722Sjkh    fprintf(outmk,   "%s_OBJPATHS=", p->ident);
8221722Sjkh    output_strlst(outmk, p->objpaths);
8231722Sjkh
8241722Sjkh    fprintf(outmk, "%s_stub.c:\n", p->name);
8251722Sjkh    fprintf(outmk, "\techo \""
8261722Sjkh	           "int _crunched_%s_stub(int argc, char **argv, char **envp)"
8271722Sjkh	           "{return main(argc,argv,envp);}\" >%s_stub.c\n",
8281722Sjkh	    p->ident, p->name);
8291722Sjkh    fprintf(outmk, "%s.lo: %s_stub.o $(%s_OBJPATHS)\n",
8301722Sjkh	    p->name, p->name, p->ident);
8318857Srgrimes    fprintf(outmk, "\tld -dc -r -o %s.lo %s_stub.o $(%s_OBJPATHS)\n",
8321722Sjkh	    p->name, p->name, p->ident);
83339171Sjkh    fprintf(outmk, "\tcrunchide -k _crunched_%s_stub ", p->ident);
83430120Sjoerg    for(lst = p->keeplist; lst != NULL; lst = lst->next)
83530120Sjoerg      fprintf(outmk, "-k _%s ", lst->str);
83630120Sjoerg    fprintf(outmk, "%s.lo\n", p->name);
8371722Sjkh}
8381722Sjkh
8391722Sjkhvoid output_strlst(FILE *outf, strlst_t *lst)
8401722Sjkh{
8411722Sjkh    for(; lst != NULL; lst = lst->next)
8421722Sjkh	fprintf(outf, " %s", lst->str);
8431722Sjkh    fprintf(outf, "\n");
8441722Sjkh}
8451722Sjkh
8461722Sjkh
8471722Sjkh/*
8481722Sjkh * ========================================================================
8491722Sjkh * general library routines
8501722Sjkh *
8511722Sjkh */
8521722Sjkh
8531722Sjkhvoid status(char *str)
8541722Sjkh{
8551722Sjkh    static int lastlen = 0;
8561722Sjkh    int len, spaces;
8571722Sjkh
8581722Sjkh    if(!verbose) return;
8591722Sjkh
8601722Sjkh    len = strlen(str);
8611722Sjkh    spaces = lastlen - len;
8621722Sjkh    if(spaces < 1) spaces = 1;
8631722Sjkh
8641722Sjkh    fprintf(stderr, " [%s]%*.*s\r", str, spaces, spaces, " ");
8651722Sjkh    fflush(stderr);
8661722Sjkh    lastlen = len;
8671722Sjkh}
8681722Sjkh
8691722Sjkh
8701722Sjkhvoid out_of_memory(void)
8711722Sjkh{
87229453Scharnier    errx(1, "%s: %d: out of memory, stopping", infilename, linenum);
8731722Sjkh}
8741722Sjkh
8751722Sjkh
8761722Sjkhvoid add_string(strlst_t **listp, char *str)
8771722Sjkh{
8781722Sjkh    strlst_t *p1, *p2;
8791722Sjkh
8801722Sjkh    /* add to end, but be smart about dups */
8811722Sjkh
8821722Sjkh    for(p1 = NULL, p2 = *listp; p2 != NULL; p1 = p2, p2 = p2->next)
8831722Sjkh	if(!strcmp(p2->str, str)) return;
8841722Sjkh
8851722Sjkh    p2 = malloc(sizeof(strlst_t));
88619822Sjoerg    if(p2) {
88719822Sjoerg	memset(p2, 0, sizeof(strlst_t));
88819822Sjoerg	p2->str = strdup(str);
88919822Sjoerg    }
8901722Sjkh    if(!p2 || !p2->str)
8911722Sjkh	out_of_memory();
8921722Sjkh
8931722Sjkh    p2->next = NULL;
8941722Sjkh    if(p1 == NULL) *listp = p2;
8951722Sjkh    else p1->next = p2;
8961722Sjkh}
8971722Sjkh
8981722Sjkh
8991722Sjkhint is_dir(char *pathname)
9001722Sjkh{
9011722Sjkh    struct stat buf;
9021722Sjkh
9031722Sjkh    if(stat(pathname, &buf) == -1)
9041722Sjkh	return 0;
9051722Sjkh    return S_ISDIR(buf.st_mode);
9061722Sjkh}
9071722Sjkh
9081722Sjkhint is_nonempty_file(char *pathname)
9091722Sjkh{
9101722Sjkh    struct stat buf;
9111722Sjkh
9121722Sjkh    if(stat(pathname, &buf) == -1)
9131722Sjkh	return 0;
9141722Sjkh
9151722Sjkh    return S_ISREG(buf.st_mode) && buf.st_size > 0;
9161722Sjkh}
917