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