/* $NetBSD: mkdep.c,v 1.15 2003/01/19 00:42:45 elric Exp $ */ /*- * Copyright (c) 1999 The NetBSD Foundation, Inc. * All rights reserved. * * This code is derived from software contributed to The NetBSD Foundation * by Matthias Scheler. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. All advertising materials mentioning features or use of this software * must display the following acknowledgement: * This product includes software developed by the NetBSD * Foundation, Inc. and its contributors. * 4. Neither the name of The NetBSD Foundation nor the names of its * contributors may be used to endorse or promote products derived * from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. */ #include #if defined(__COPYRIGHT) && !defined(lint) __COPYRIGHT("@(#) Copyright (c) 1999 The NetBSD Foundation, Inc.\n\ All rights reserved.\n"); #endif /* not lint */ #if defined(__RCSID) && !defined(lint) __RCSID("$NetBSD: mkdep.c,v 1.15 2003/01/19 00:42:45 elric Exp $"); #endif /* not lint */ #if HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include #include #include #include #include #include #include #include #include "findcc.h" #define DEFAULT_PATH _PATH_DEFPATH #define DEFAULT_FILENAME ".depend" int tmpfd; char tmpfilename[MAXPATHLEN]; static void finish __P((int)); static void usage __P((void)); int main __P((int, char **)); static void usage() { (void)fprintf(stderr, "usage: %s [-a] [-p] [-f file] flags file ...\n", getprogname()); exit(EXIT_FAILURE); } void finish(signo) int signo; { if (tmpfd != -1) { (void)close(tmpfd); (void)unlink(tmpfilename); } exit(EXIT_FAILURE); } int main(argc, argv) int argc; char **argv; { /* LINTED local definition of index */ int aflag, pflag, index, status; pid_t cpid, pid; char *filename, *CC, *pathname, **args; const char *tmpdir; /* LINTED local definition of tmpfile */ FILE *tmpfile, *dependfile; char buffer[32768]; setlocale(LC_ALL, ""); setprogname(argv[0]); aflag = 0; pflag = 0; filename = DEFAULT_FILENAME; /* XXX should use getopt(). */ for (index = 1; index < argc; index++) { if (strcmp(argv[index], "-a") == 0) aflag = 1; else if (strcmp(argv[index], "-f") == 0) { if (++index < argc) filename = argv[index]; } else if (strcmp(argv[index], "-p") == 0) pflag = 1; else break; } argc -= index; argv += index; if (argc == 0) usage(); if ((CC = getenv("CC")) == NULL) CC = DEFAULT_CC; if ((pathname = findcc(CC)) == NULL) if (!setenv("PATH", DEFAULT_PATH, 1)) pathname = findcc(CC); if (pathname == NULL) { (void)fprintf(stderr, "%s: %s: not found\n", getprogname(), CC); exit(EXIT_FAILURE); } if ((args = malloc((argc + 3) * sizeof(char *))) == NULL) { perror(getprogname()); exit(EXIT_FAILURE); } args[0] = CC; args[1] = "-M"; (void)memcpy(&args[2], argv, (argc + 1) * sizeof(char *)); if ((tmpdir = getenv("TMPDIR")) == NULL) tmpdir = _PATH_TMP; (void)snprintf(tmpfilename, sizeof (tmpfilename), "%s/%s", tmpdir, "mkdepXXXXXX"); /* set signal handler */ tmpfd = -1; (void)signal(SIGINT, finish); (void)signal(SIGHUP, finish); (void)signal(SIGQUIT, finish); (void)signal(SIGTERM, finish); if ((tmpfd = mkstemp (tmpfilename)) < 0) { warn("unable to create temporary file %s", tmpfilename); exit(EXIT_FAILURE); } switch (cpid = vfork()) { case 0: (void)dup2(tmpfd, STDOUT_FILENO); (void)close(tmpfd); (void)execv(pathname, args); _exit(EXIT_FAILURE); /* NOTREACHED */ case -1: (void)fprintf(stderr, "%s: unable to fork.\n", getprogname()); (void)close(tmpfd); (void)unlink(tmpfilename); exit(EXIT_FAILURE); } while (((pid = wait(&status)) != cpid) && (pid >= 0)) continue; if (status) { (void)fprintf(stderr, "%s: compile failed.\n", getprogname()); (void)close(tmpfd); (void)unlink(tmpfilename); exit(EXIT_FAILURE); } (void)lseek(tmpfd, (off_t)0, SEEK_SET); if ((tmpfile = fdopen(tmpfd, "r")) == NULL) { (void)fprintf(stderr, "%s: unable to read temporary file %s\n", getprogname(), tmpfilename); (void)close(tmpfd); (void)unlink(tmpfilename); exit(EXIT_FAILURE); } if ((dependfile = fopen(filename, aflag ? "a" : "w")) == NULL) { (void)fprintf(stderr, "%s: unable to %s to file %s\n", getprogname(), aflag ? "append" : "write", filename); (void)fclose(tmpfile); (void)unlink(tmpfilename); exit(EXIT_FAILURE); } while (fgets(buffer, sizeof(buffer), tmpfile) != NULL) { char *ptr; if (pflag && ((ptr = strstr(buffer, ".o")) != NULL)) { char *colon; colon = ptr + 2; while (isspace(*colon)) colon++; if (*colon == ':') (void)strcpy(ptr, colon); } ptr = buffer; while (*ptr) { if (isspace(*ptr++)) if ((ptr[0] == '.') && (ptr[1] == '/')) (void)strcpy(ptr, ptr + 2); } (void)fputs(buffer, dependfile); } (void)fclose(dependfile); (void)fclose(tmpfile); (void)unlink(tmpfilename); exit(EXIT_SUCCESS); }