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