blob: 22c30357b267325add97dd24946f899b6ca3b174
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Utility routines. |
4 | * |
5 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> |
6 | * |
7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. |
8 | */ |
9 | #include "libbb.h" |
10 | #if ENABLE_FEATURE_SYSLOG |
11 | # include <syslog.h> |
12 | #endif |
13 | |
14 | #if ENABLE_FEATURE_SYSLOG |
15 | smallint syslog_level = LOG_ERR; |
16 | #endif |
17 | smallint logmode = LOGMODE_STDIO; |
18 | const char *msg_eol = "\n"; |
19 | |
20 | void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr) |
21 | { |
22 | char *msg, *msg1; |
23 | char stack_msg[80]; |
24 | int applet_len, strerr_len, msgeol_len, used; |
25 | |
26 | if (!logmode) |
27 | return; |
28 | |
29 | if (!s) /* nomsg[_and_die] uses NULL fmt */ |
30 | s = ""; /* some libc don't like printf(NULL) */ |
31 | |
32 | applet_len = strlen(applet_name) + 2; /* "applet: " */ |
33 | strerr_len = strerr ? strlen(strerr) : 0; |
34 | msgeol_len = strlen(msg_eol); |
35 | |
36 | /* This costs ~90 bytes of code, but avoids costly |
37 | * malloc()[in vasprintf]+realloc()+memmove()+free() in 99% of cases. |
38 | * ~40% speedup. |
39 | */ |
40 | if ((int)sizeof(stack_msg) - applet_len > 0) { |
41 | va_list p2; |
42 | |
43 | /* It is not portable to use va_list twice, need to va_copy it */ |
44 | va_copy(p2, p); |
45 | used = vsnprintf(stack_msg + applet_len, (int)sizeof(stack_msg) - applet_len, s, p2); |
46 | va_end(p2); |
47 | msg = stack_msg; |
48 | used += applet_len; |
49 | if (used < (int)sizeof(stack_msg) - 3 - msgeol_len - strerr_len) |
50 | goto add_pfx_and_sfx; |
51 | } |
52 | |
53 | used = vasprintf(&msg, s, p); |
54 | if (used < 0) |
55 | return; |
56 | |
57 | /* This is ugly and costs +60 bytes compared to multiple |
58 | * fprintf's, but is guaranteed to do a single write. |
59 | * This is needed for e.g. httpd logging, when multiple |
60 | * children can produce log messages simultaneously. */ |
61 | |
62 | /* can't use xrealloc: it calls error_msg on failure, |
63 | * that may result in a recursion */ |
64 | /* +3 is for ": " before strerr and for terminating NUL */ |
65 | msg1 = realloc(msg, applet_len + used + strerr_len + msgeol_len + 3); |
66 | if (!msg1) { |
67 | msg[used++] = '\n'; /* overwrites NUL */ |
68 | applet_len = 0; |
69 | } else { |
70 | msg = msg1; |
71 | /* TODO: maybe use writev instead of memmoving? Need full_writev? */ |
72 | memmove(msg + applet_len, msg, used); |
73 | used += applet_len; |
74 | add_pfx_and_sfx: |
75 | strcpy(msg, applet_name); |
76 | msg[applet_len - 2] = ':'; |
77 | msg[applet_len - 1] = ' '; |
78 | if (strerr) { |
79 | if (s[0]) { /* not perror_nomsg? */ |
80 | msg[used++] = ':'; |
81 | msg[used++] = ' '; |
82 | } |
83 | strcpy(&msg[used], strerr); |
84 | used += strerr_len; |
85 | } |
86 | strcpy(&msg[used], msg_eol); |
87 | used += msgeol_len; |
88 | } |
89 | |
90 | if (logmode & LOGMODE_STDIO) { |
91 | fflush_all(); |
92 | full_write(STDERR_FILENO, msg, used); |
93 | } |
94 | #if ENABLE_FEATURE_SYSLOG |
95 | if (logmode & LOGMODE_SYSLOG) { |
96 | syslog(syslog_level, "%s", msg + applet_len); |
97 | } |
98 | #endif |
99 | if (msg != stack_msg) |
100 | free(msg); |
101 | } |
102 | |
103 | #ifdef VERSION_WITH_WRITEV |
104 | /* Code size is approximately the same, but currently it's the only user |
105 | * of writev in entire bbox. __libc_writev in uclibc is ~50 bytes. */ |
106 | void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr) |
107 | { |
108 | int strerr_len, msgeol_len; |
109 | struct iovec iov[3]; |
110 | |
111 | #define used (iov[2].iov_len) |
112 | #define msgv (iov[2].iov_base) |
113 | #define msgc ((char*)(iov[2].iov_base)) |
114 | #define msgptr (&(iov[2].iov_base)) |
115 | |
116 | if (!logmode) |
117 | return; |
118 | |
119 | if (!s) /* nomsg[_and_die] uses NULL fmt */ |
120 | s = ""; /* some libc don't like printf(NULL) */ |
121 | |
122 | /* Prevent "derefing type-punned ptr will break aliasing rules" */ |
123 | used = vasprintf((char**)(void*)msgptr, s, p); |
124 | if (used < 0) |
125 | return; |
126 | |
127 | /* This is ugly and costs +60 bytes compared to multiple |
128 | * fprintf's, but is guaranteed to do a single write. |
129 | * This is needed for e.g. httpd logging, when multiple |
130 | * children can produce log messages simultaneously. */ |
131 | |
132 | strerr_len = strerr ? strlen(strerr) : 0; |
133 | msgeol_len = strlen(msg_eol); |
134 | /* +3 is for ": " before strerr and for terminating NUL */ |
135 | msgv = xrealloc(msgv, used + strerr_len + msgeol_len + 3); |
136 | if (strerr) { |
137 | msgc[used++] = ':'; |
138 | msgc[used++] = ' '; |
139 | strcpy(msgc + used, strerr); |
140 | used += strerr_len; |
141 | } |
142 | strcpy(msgc + used, msg_eol); |
143 | used += msgeol_len; |
144 | |
145 | if (logmode & LOGMODE_STDIO) { |
146 | iov[0].iov_base = (char*)applet_name; |
147 | iov[0].iov_len = strlen(applet_name); |
148 | iov[1].iov_base = (char*)": "; |
149 | iov[1].iov_len = 2; |
150 | /*iov[2].iov_base = msgc;*/ |
151 | /*iov[2].iov_len = used;*/ |
152 | fflush_all(); |
153 | writev(STDERR_FILENO, iov, 3); |
154 | } |
155 | # if ENABLE_FEATURE_SYSLOG |
156 | if (logmode & LOGMODE_SYSLOG) { |
157 | syslog(LOG_ERR, "%s", msgc); |
158 | } |
159 | # endif |
160 | free(msgc); |
161 | } |
162 | #endif |
163 | |
164 | |
165 | void FAST_FUNC bb_error_msg_and_die(const char *s, ...) |
166 | { |
167 | va_list p; |
168 | |
169 | va_start(p, s); |
170 | bb_verror_msg(s, p, NULL); |
171 | va_end(p); |
172 | xfunc_die(); |
173 | } |
174 | |
175 | void FAST_FUNC bb_error_msg(const char *s, ...) |
176 | { |
177 | va_list p; |
178 | |
179 | va_start(p, s); |
180 | bb_verror_msg(s, p, NULL); |
181 | va_end(p); |
182 | } |
183 |