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