author | Jun Wu <quark@fb.com> |
Tue, 05 Apr 2016 14:48:09 +0100 | |
changeset 28787 | ea86cdcd9b50 |
parent 28165 | c6705c6303dd |
child 28788 | 57a78a64de44 |
permissions | -rw-r--r-- |
28060 | 1 |
/* |
2 |
* Utility functions |
|
3 |
* |
|
4 |
* Copyright (c) 2011 Yuya Nishihara <yuya@tcha.org> |
|
5 |
* |
|
6 |
* This software may be used and distributed according to the terms of the |
|
7 |
* GNU General Public License version 2 or any later version. |
|
8 |
*/ |
|
9 |
||
10 |
#include <signal.h> |
|
11 |
#include <stdarg.h> |
|
12 |
#include <stdio.h> |
|
13 |
#include <stdlib.h> |
|
28084
3fc45956c978
chg: initialize sigaction fields more reliably
Yuya Nishihara <yuya@tcha.org>
parents:
28060
diff
changeset
|
14 |
#include <string.h> |
28060 | 15 |
#include <sys/types.h> |
16 |
#include <sys/wait.h> |
|
17 |
#include <unistd.h> |
|
18 |
||
19 |
#include "util.h" |
|
20 |
||
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
21 |
static int colorenabled = 0; |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
22 |
|
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
23 |
static inline void fsetcolor(FILE *fp, const char *code) |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
24 |
{ |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
25 |
if (!colorenabled) |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
26 |
return; |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
27 |
fprintf(fp, "\033[%sm", code); |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
28 |
} |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
29 |
|
28060 | 30 |
void abortmsg(const char *fmt, ...) |
31 |
{ |
|
32 |
va_list args; |
|
33 |
va_start(args, fmt); |
|
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
34 |
fsetcolor(stderr, "1;31"); |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
35 |
fputs("chg: abort: ", stderr); |
28060 | 36 |
vfprintf(stderr, fmt, args); |
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
37 |
fsetcolor(stderr, ""); |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
38 |
fputc('\n', stderr); |
28060 | 39 |
va_end(args); |
40 |
||
41 |
exit(255); |
|
42 |
} |
|
43 |
||
44 |
static int debugmsgenabled = 0; |
|
45 |
||
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
46 |
void enablecolor(void) |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
47 |
{ |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
48 |
colorenabled = 1; |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
49 |
} |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
50 |
|
28060 | 51 |
void enabledebugmsg(void) |
52 |
{ |
|
53 |
debugmsgenabled = 1; |
|
54 |
} |
|
55 |
||
56 |
void debugmsg(const char *fmt, ...) |
|
57 |
{ |
|
58 |
if (!debugmsgenabled) |
|
59 |
return; |
|
60 |
||
61 |
va_list args; |
|
62 |
va_start(args, fmt); |
|
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
63 |
fsetcolor(stderr, "1;30"); |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
64 |
fputs("chg: debug: ", stderr); |
28060 | 65 |
vfprintf(stderr, fmt, args); |
28787
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
66 |
fsetcolor(stderr, ""); |
ea86cdcd9b50
chg: use color in debug/error messages conditionally
Jun Wu <quark@fb.com>
parents:
28165
diff
changeset
|
67 |
fputc('\n', stderr); |
28060 | 68 |
va_end(args); |
69 |
} |
|
70 |
||
28165
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
71 |
void *mallocx(size_t size) |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
72 |
{ |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
73 |
void *result = malloc(size); |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
74 |
if (!result) |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
75 |
abortmsg("failed to malloc"); |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
76 |
return result; |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
77 |
} |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
78 |
|
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
79 |
void *reallocx(void *ptr, size_t size) |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
80 |
{ |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
81 |
void *result = realloc(ptr, size); |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
82 |
if (!result) |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
83 |
abortmsg("failed to realloc"); |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
84 |
return result; |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
85 |
} |
c6705c6303dd
chg: add utility functions mallocx, reallocx
Jun Wu <quark@fb.com>
parents:
28084
diff
changeset
|
86 |
|
28060 | 87 |
/* |
88 |
* Execute a shell command in mostly the same manner as system(), with the |
|
89 |
* give environment variables, after chdir to the given cwd. Returns a status |
|
90 |
* code compatible with the Python subprocess module. |
|
91 |
*/ |
|
92 |
int runshellcmd(const char *cmd, const char *envp[], const char *cwd) |
|
93 |
{ |
|
94 |
enum { F_SIGINT = 1, F_SIGQUIT = 2, F_SIGMASK = 4, F_WAITPID = 8 }; |
|
95 |
unsigned int doneflags = 0; |
|
96 |
int status = 0; |
|
97 |
struct sigaction newsa, oldsaint, oldsaquit; |
|
98 |
sigset_t oldmask; |
|
99 |
||
100 |
/* block or mask signals just as system() does */ |
|
28084
3fc45956c978
chg: initialize sigaction fields more reliably
Yuya Nishihara <yuya@tcha.org>
parents:
28060
diff
changeset
|
101 |
memset(&newsa, 0, sizeof(newsa)); |
28060 | 102 |
newsa.sa_handler = SIG_IGN; |
103 |
newsa.sa_flags = 0; |
|
104 |
if (sigemptyset(&newsa.sa_mask) < 0) |
|
105 |
goto done; |
|
106 |
if (sigaction(SIGINT, &newsa, &oldsaint) < 0) |
|
107 |
goto done; |
|
108 |
doneflags |= F_SIGINT; |
|
109 |
if (sigaction(SIGQUIT, &newsa, &oldsaquit) < 0) |
|
110 |
goto done; |
|
111 |
doneflags |= F_SIGQUIT; |
|
112 |
||
113 |
if (sigaddset(&newsa.sa_mask, SIGCHLD) < 0) |
|
114 |
goto done; |
|
115 |
if (sigprocmask(SIG_BLOCK, &newsa.sa_mask, &oldmask) < 0) |
|
116 |
goto done; |
|
117 |
doneflags |= F_SIGMASK; |
|
118 |
||
119 |
pid_t pid = fork(); |
|
120 |
if (pid < 0) |
|
121 |
goto done; |
|
122 |
if (pid == 0) { |
|
123 |
sigaction(SIGINT, &oldsaint, NULL); |
|
124 |
sigaction(SIGQUIT, &oldsaquit, NULL); |
|
125 |
sigprocmask(SIG_SETMASK, &oldmask, NULL); |
|
126 |
if (cwd && chdir(cwd) < 0) |
|
127 |
_exit(127); |
|
128 |
const char *argv[] = {"sh", "-c", cmd, NULL}; |
|
129 |
if (envp) { |
|
130 |
execve("/bin/sh", (char **)argv, (char **)envp); |
|
131 |
} else { |
|
132 |
execv("/bin/sh", (char **)argv); |
|
133 |
} |
|
134 |
_exit(127); |
|
135 |
} else { |
|
136 |
if (waitpid(pid, &status, 0) < 0) |
|
137 |
goto done; |
|
138 |
doneflags |= F_WAITPID; |
|
139 |
} |
|
140 |
||
141 |
done: |
|
142 |
if (doneflags & F_SIGINT) |
|
143 |
sigaction(SIGINT, &oldsaint, NULL); |
|
144 |
if (doneflags & F_SIGQUIT) |
|
145 |
sigaction(SIGQUIT, &oldsaquit, NULL); |
|
146 |
if (doneflags & F_SIGMASK) |
|
147 |
sigprocmask(SIG_SETMASK, &oldmask, NULL); |
|
148 |
||
149 |
/* no way to report other errors, use 127 (= shell termination) */ |
|
150 |
if (!(doneflags & F_WAITPID)) |
|
151 |
return 127; |
|
152 |
if (WIFEXITED(status)) |
|
153 |
return WEXITSTATUS(status); |
|
154 |
if (WIFSIGNALED(status)) |
|
155 |
return -WTERMSIG(status); |
|
156 |
return 127; |
|
157 |
} |