Move ProcExec() into proc.c
[dragonfly.git] / usr.bin / make / proc.c
1 /*-
2  * Copyright (C) 2005 Max Okumoto.
3  *      All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $DragonFly: src/usr.bin/make/proc.c,v 1.1 2005/05/23 18:24:59 okumoto Exp $
27  */
28
29 #include <unistd.h>
30 #include <errno.h>
31 #include <string.h>
32 #include <fcntl.h>
33
34 #include "proc.h"
35 #include "shell.h"
36 #include "util.h"
37
38 /**
39  * Replace the current process.
40  */
41 void
42 ProcExec(const ProcStuff *ps)
43 {
44         if (ps->in != STDIN_FILENO) {
45                 /*
46                  * Redirect the child's stdin to the input fd
47                  * and reset it to the beginning (again).
48                  */
49                 if (dup2(ps->in, STDIN_FILENO) == -1)
50                         Punt("Cannot dup2: %s", strerror(errno));
51                 lseek(STDIN_FILENO, (off_t)0, SEEK_SET);
52         }
53
54         if (ps->out != STDOUT_FILENO) {
55                 /*
56                  * Redirect the child's stdout to the output fd.
57                  */
58                 if (dup2(ps->out, STDOUT_FILENO) == -1)
59                         Punt("Cannot dup2: %s", strerror(errno));
60                 close(ps->out);
61         }
62
63         if (ps->err != STDERR_FILENO) {
64                 /*
65                  * Redirect the child's stderr to the err fd.
66                  */
67                 if (dup2(ps->err, STDERR_FILENO) == -1)
68                         Punt("Cannot dup2: %s", strerror(errno));
69                 close(ps->err);
70         }
71
72         if (ps->merge_errors) {
73                 /*
74                  * Send stderr to parent process too.
75                  */
76                 if (dup2(STDOUT_FILENO, STDERR_FILENO) == -1)
77                         Punt("Cannot dup2: %s", strerror(errno));
78         }
79
80         /*
81          * The file descriptors for stdin, stdout, or stderr might
82          * have been marked close-on-exec.  Clear the flag on all
83          * of them.
84          */
85         fcntl(STDIN_FILENO, F_SETFD,
86             fcntl(STDIN_FILENO, F_GETFD) & (~FD_CLOEXEC));
87         fcntl(STDOUT_FILENO, F_SETFD,
88             fcntl(STDOUT_FILENO, F_GETFD) & (~FD_CLOEXEC));
89         fcntl(STDERR_FILENO, F_SETFD,
90             fcntl(STDERR_FILENO, F_GETFD) & (~FD_CLOEXEC));
91
92         if (ps->pgroup) {
93 #ifdef USE_PGRP
94                 /*
95                  * Become a process group leader, so we can kill it and all
96                  * its descendants in one fell swoop, by killing its process
97                  * family, but not commit suicide.
98                  */
99 #if defined(SYSV)
100                 setsid();
101 #else
102                 setpgid(0, getpid());
103 #endif
104 #endif /* USE_PGRP */
105         }
106
107         if (ps->searchpath) {
108                 execvp(ps->argv[0], ps->argv);
109
110                 write(STDERR_FILENO, ps->argv[0], strlen(ps->argv[0]));
111                 write(STDERR_FILENO, ":", 1);
112                 write(STDERR_FILENO, strerror(errno), strlen(strerror(errno)));
113                 write(STDERR_FILENO, "\n", 1);
114         } else {
115                 execv(commandShell->path, ps->argv);
116
117                 write(STDERR_FILENO,
118                       "Could not execute shell\n",
119                       sizeof("Could not execute shell"));
120         }
121
122         /*
123          * Since we are the child process, exit without flushing buffers.
124          */
125         _exit(1);
126 }
127