ash: recognize embedded scripts in SH_STANDALONE mode
[oweals/busybox.git] / include / libbb.h
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Busybox main internal header file
4  *
5  * Based in part on code from sash, Copyright (c) 1999 by David I. Bell
6  * Permission has been granted to redistribute this code under GPL.
7  *
8  * Licensed under GPLv2, see file LICENSE in this source tree.
9  */
10 #ifndef LIBBB_H
11 #define LIBBB_H 1
12
13 #include "platform.h"
14
15 #include <ctype.h>
16 #include <dirent.h>
17 #include <errno.h>
18 #include <fcntl.h>
19 #include <inttypes.h>
20 #include <netdb.h>
21 #include <setjmp.h>
22 #include <signal.h>
23 #include <paths.h>
24 #if defined __UCLIBC__ /* TODO: and glibc? */
25 /* use inlined versions of these: */
26 # define sigfillset(s)    __sigfillset(s)
27 # define sigemptyset(s)   __sigemptyset(s)
28 # define sigisemptyset(s) __sigisemptyset(s)
29 #endif
30 #include <stdint.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <stdarg.h>
34 #include <stddef.h>
35 #include <string.h>
36 /* There are two incompatible basename's, let's not use them! */
37 /* See the dirname/basename man page for details */
38 #include <libgen.h> /* dirname,basename */
39 #undef basename
40 #define basename dont_use_basename
41 #include <poll.h>
42 #include <sys/ioctl.h>
43 #include <sys/mman.h>
44 #include <sys/resource.h>
45 #include <sys/socket.h>
46 #include <sys/stat.h>
47 #include <sys/time.h>
48 #include <sys/types.h>
49 #if !defined(major) || defined(__GLIBC__)
50 # include <sys/sysmacros.h>
51 #endif
52 #include <sys/wait.h>
53 #include <termios.h>
54 #include <time.h>
55 #include <sys/param.h>
56 #include <pwd.h>
57 #include <grp.h>
58 #if ENABLE_FEATURE_SHADOWPASSWDS
59 # if !ENABLE_USE_BB_SHADOW
60 /* If using busybox's shadow implementation, do not include the shadow.h
61  * header as the toolchain may not provide it at all.
62  */
63 #  include <shadow.h>
64 # endif
65 #endif
66 #if defined(ANDROID) || defined(__ANDROID__)
67 # define endpwent() ((void)0)
68 # define endgrent() ((void)0)
69 #endif
70 #ifdef HAVE_MNTENT_H
71 # include <mntent.h>
72 #endif
73 #ifdef HAVE_SYS_STATFS_H
74 # include <sys/statfs.h>
75 #endif
76 /* Don't do this here:
77  * #include <sys/sysinfo.h>
78  * Some linux/ includes pull in conflicting definition
79  * of struct sysinfo (only in some toolchanins), which breaks build.
80  * Include sys/sysinfo.h only in those files which need it.
81  */
82 #if ENABLE_SELINUX
83 # include <selinux/selinux.h>
84 # include <selinux/context.h>
85 #endif
86 #if ENABLE_FEATURE_UTMP
87 # if defined __UCLIBC__ && ( \
88     (UCLIBC_VERSION >= KERNEL_VERSION(0, 9, 32) \
89      && UCLIBC_VERSION < KERNEL_VERSION(0, 9, 34) \
90      && defined __UCLIBC_HAS_UTMPX__ \
91     ) || ( \
92          UCLIBC_VERSION >= KERNEL_VERSION(0, 9, 34) \
93         ) \
94   )
95 #  include <utmpx.h>
96 # elif defined __UCLIBC__
97 #  include <utmp.h>
98 #  define utmpx utmp
99 #  define setutxent setutent
100 #  define endutxent endutent
101 #  define getutxent getutent
102 #  define getutxid getutid
103 #  define getutxline getutline
104 #  define pututxline pututline
105 #  define utmpxname utmpname
106 #  define updwtmpx updwtmp
107 #  define _PATH_UTMPX _PATH_UTMP
108 # else
109 #  include <utmp.h>
110 #  include <utmpx.h>
111 #  if defined _PATH_UTMP && !defined _PATH_UTMPX
112 #   define _PATH_UTMPX _PATH_UTMP
113 #  endif
114 # endif
115 #endif
116 #if ENABLE_LOCALE_SUPPORT
117 # include <locale.h>
118 #else
119 # define setlocale(x,y) ((void)0)
120 #endif
121 #ifdef DMALLOC
122 # include <dmalloc.h>
123 #endif
124 /* Just in case libc doesn't define some of these... */
125 #ifndef _PATH_PASSWD
126 #define _PATH_PASSWD  "/etc/passwd"
127 #endif
128 #ifndef _PATH_GROUP
129 #define _PATH_GROUP   "/etc/group"
130 #endif
131 #ifndef _PATH_SHADOW
132 #define _PATH_SHADOW  "/etc/shadow"
133 #endif
134 #ifndef _PATH_GSHADOW
135 #define _PATH_GSHADOW "/etc/gshadow"
136 #endif
137 #if defined __FreeBSD__ || defined __OpenBSD__
138 # include <netinet/in.h>
139 # include <arpa/inet.h>
140 #elif defined __APPLE__
141 # include <netinet/in.h>
142 #else
143 # include <arpa/inet.h>
144 //This breaks on bionic:
145 //# if !defined(__socklen_t_defined) && !defined(_SOCKLEN_T_DECLARED)
146 ///* We #define socklen_t *after* includes, otherwise we get
147 // * typedef redefinition errors from system headers
148 // * (in case "is it defined already" detection above failed)
149 // */
150 //#  define socklen_t bb_socklen_t
151 //   typedef unsigned socklen_t;
152 //# endif
153 //if this is still needed, add a fix along the lines of
154 //  ifdef SPECIFIC_BROKEN_LIBC_CHECK / typedef socklen_t / endif
155 //in platform.h instead!
156 #endif
157 #ifndef HAVE_CLEARENV
158 # define clearenv() do { if (environ) environ[0] = NULL; } while (0)
159 #endif
160 #ifndef HAVE_FDATASYNC
161 # define fdatasync fsync
162 #endif
163 #ifndef HAVE_XTABS
164 # define XTABS TAB3
165 #endif
166 /*
167  * Use '%m' to append error string on platforms that support it,
168  * '%s' and strerror() on those that don't.
169  */
170 #ifdef HAVE_PRINTF_PERCENTM
171 # define STRERROR_FMT    "%m"
172 # define STRERROR_ERRNO  /*nothing*/
173 #else
174 # define STRERROR_FMT    "%s"
175 # define STRERROR_ERRNO  ,strerror(errno)
176 #endif
177
178
179 /* Some libc's forget to declare these, do it ourself */
180
181 extern char **environ;
182 /* klogctl is in libc's klog.h, but we cheat and not #include that */
183 int klogctl(int type, char *b, int len);
184 #ifndef PATH_MAX
185 # define PATH_MAX 256
186 #endif
187 #ifndef BUFSIZ
188 # define BUFSIZ 4096
189 #endif
190
191
192 /* Busybox does not use threads, we can speed up stdio. */
193 #ifdef HAVE_UNLOCKED_STDIO
194 # undef  getc
195 # define getc(stream)   getc_unlocked(stream)
196 # undef  getchar
197 # define getchar()      getchar_unlocked()
198 # undef  putc
199 # define putc(c,stream) putc_unlocked(c,stream)
200 # undef  putchar
201 # define putchar(c)     putchar_unlocked(c)
202 # undef  fgetc
203 # define fgetc(stream)  getc_unlocked(stream)
204 # undef  fputc
205 # define fputc(c,stream) putc_unlocked(c,stream)
206 #endif
207 /* Above functions are required by POSIX.1-2008, below ones are extensions */
208 #ifdef HAVE_UNLOCKED_LINE_OPS
209 # undef  fgets
210 # define fgets(s,n,stream) fgets_unlocked(s,n,stream)
211 # undef  fputs
212 # define fputs(s,stream) fputs_unlocked(s,stream)
213 /* musl <= 1.1.15 does not support fflush_unlocked(NULL) */
214 //# undef  fflush
215 //# define fflush(stream) fflush_unlocked(stream)
216 # undef  feof
217 # define feof(stream)   feof_unlocked(stream)
218 # undef  ferror
219 # define ferror(stream) ferror_unlocked(stream)
220 # undef  fileno
221 # define fileno(stream) fileno_unlocked(stream)
222 #endif
223
224
225 /* Make all declarations hidden (-fvisibility flag only affects definitions) */
226 /* (don't include system headers after this until corresponding pop!) */
227 PUSH_AND_SET_FUNCTION_VISIBILITY_TO_HIDDEN
228
229
230 #if ENABLE_USE_BB_PWD_GRP
231 # include "pwd_.h"
232 # include "grp_.h"
233 #endif
234 #if ENABLE_FEATURE_SHADOWPASSWDS
235 # if ENABLE_USE_BB_SHADOW
236 #  include "shadow_.h"
237 # endif
238 #endif
239
240 /* Tested to work correctly with all int types (IIRC :]) */
241 #define MAXINT(T) (T)( \
242         ((T)-1) > 0 \
243         ? (T)-1 \
244         : (T)~((T)1 << (sizeof(T)*8-1)) \
245         )
246
247 #define MININT(T) (T)( \
248         ((T)-1) > 0 \
249         ? (T)0 \
250         : ((T)1 << (sizeof(T)*8-1)) \
251         )
252
253 /* Large file support */
254 /* Note that CONFIG_LFS=y forces bbox to be built with all common ops
255  * (stat, lseek etc) mapped to "largefile" variants by libc.
256  * Practically it means that open() automatically has O_LARGEFILE added
257  * and all filesize/file_offset parameters and struct members are "large"
258  * (in today's world - signed 64bit). For full support of large files,
259  * we need a few helper #defines (below) and careful use of off_t
260  * instead of int/ssize_t. No lseek64(), O_LARGEFILE etc necessary */
261 #if ENABLE_LFS
262 /* CONFIG_LFS is on */
263 # if ULONG_MAX > 0xffffffff
264 /* "long" is long enough on this system */
265 typedef unsigned long uoff_t;
266 #  define XATOOFF(a) xatoul_range((a), 0, LONG_MAX)
267 /* usage: sz = BB_STRTOOFF(s, NULL, 10); if (errno || sz < 0) die(); */
268 #  define BB_STRTOOFF bb_strtoul
269 #  define STRTOOFF strtoul
270 /* usage: printf("size: %"OFF_FMT"d (%"OFF_FMT"x)\n", sz, sz); */
271 #  define OFF_FMT "l"
272 # else
273 /* "long" is too short, need "long long" */
274 typedef unsigned long long uoff_t;
275 #  define XATOOFF(a) xatoull_range((a), 0, LLONG_MAX)
276 #  define BB_STRTOOFF bb_strtoull
277 #  define STRTOOFF strtoull
278 #  define OFF_FMT "ll"
279 # endif
280 #else
281 /* CONFIG_LFS is off */
282 # if UINT_MAX == 0xffffffff
283 /* While sizeof(off_t) == sizeof(int), off_t is typedef'ed to long anyway.
284  * gcc will throw warnings on printf("%d", off_t). Crap... */
285 typedef unsigned long uoff_t;
286 #  define XATOOFF(a) xatoi_positive(a)
287 #  define BB_STRTOOFF bb_strtou
288 #  define STRTOOFF strtol
289 #  define OFF_FMT "l"
290 # else
291 typedef unsigned long uoff_t;
292 #  define XATOOFF(a) xatoul_range((a), 0, LONG_MAX)
293 #  define BB_STRTOOFF bb_strtoul
294 #  define STRTOOFF strtol
295 #  define OFF_FMT "l"
296 # endif
297 #endif
298 /* scary. better ideas? (but do *test* them first!) */
299 #define OFF_T_MAX  ((off_t)~((off_t)1 << (sizeof(off_t)*8-1)))
300 /* Users report bionic to use 32-bit off_t even if LARGEFILE support is requested.
301  * We misdetected that. Don't let it build:
302  */
303 struct BUG_off_t_size_is_misdetected {
304         char BUG_off_t_size_is_misdetected[sizeof(off_t) == sizeof(uoff_t) ? 1 : -1];
305 };
306
307 /* Some useful definitions */
308 #undef FALSE
309 #define FALSE   ((int) 0)
310 #undef TRUE
311 #define TRUE    ((int) 1)
312 #undef SKIP
313 #define SKIP    ((int) 2)
314
315 /* Macros for min/max.  */
316 #ifndef MIN
317 #define MIN(a,b) (((a)<(b))?(a):(b))
318 #endif
319
320 #ifndef MAX
321 #define MAX(a,b) (((a)>(b))?(a):(b))
322 #endif
323
324 /* buffer allocation schemes */
325 #if ENABLE_FEATURE_BUFFERS_GO_ON_STACK
326 #define RESERVE_CONFIG_BUFFER(buffer,len)  char buffer[len]
327 #define RESERVE_CONFIG_UBUFFER(buffer,len) unsigned char buffer[len]
328 #define RELEASE_CONFIG_BUFFER(buffer)      ((void)0)
329 #else
330 #if ENABLE_FEATURE_BUFFERS_GO_IN_BSS
331 #define RESERVE_CONFIG_BUFFER(buffer,len)  static          char buffer[len]
332 #define RESERVE_CONFIG_UBUFFER(buffer,len) static unsigned char buffer[len]
333 #define RELEASE_CONFIG_BUFFER(buffer)      ((void)0)
334 #else
335 #define RESERVE_CONFIG_BUFFER(buffer,len)  char *buffer = xmalloc(len)
336 #define RESERVE_CONFIG_UBUFFER(buffer,len) unsigned char *buffer = xmalloc(len)
337 #define RELEASE_CONFIG_BUFFER(buffer)      free(buffer)
338 #endif
339 #endif
340
341 #if defined(__GLIBC__)
342 /* glibc uses __errno_location() to get a ptr to errno */
343 /* We can just memorize it once - no multithreading in busybox :) */
344 extern int *const bb_errno;
345 #undef errno
346 #define errno (*bb_errno)
347 #endif
348
349 #if !(ULONG_MAX > 0xffffffff)
350 /* Only 32-bit CPUs need this, 64-bit ones use inlined version */
351 uint64_t bb_bswap_64(uint64_t x) FAST_FUNC;
352 #endif
353
354 unsigned long FAST_FUNC isqrt(unsigned long long N);
355
356 unsigned long long monotonic_ns(void) FAST_FUNC;
357 unsigned long long monotonic_us(void) FAST_FUNC;
358 unsigned long long monotonic_ms(void) FAST_FUNC;
359 unsigned monotonic_sec(void) FAST_FUNC;
360
361 extern void chomp(char *s) FAST_FUNC;
362 extern char *trim(char *s) FAST_FUNC;
363 extern char *skip_whitespace(const char *) FAST_FUNC;
364 extern char *skip_non_whitespace(const char *) FAST_FUNC;
365 extern char *skip_dev_pfx(const char *tty_name) FAST_FUNC;
366
367 extern char *strrstr(const char *haystack, const char *needle) FAST_FUNC;
368
369 /* dmalloc will redefine these to it's own implementation. It is safe
370  * to have the prototypes here unconditionally.  */
371 void *malloc_or_warn(size_t size) FAST_FUNC RETURNS_MALLOC;
372 void *xmalloc(size_t size) FAST_FUNC RETURNS_MALLOC;
373 void *xzalloc(size_t size) FAST_FUNC RETURNS_MALLOC;
374 void *xrealloc(void *old, size_t size) FAST_FUNC;
375 /* After v = xrealloc_vector(v, SHIFT, idx) it's ok to use
376  * at least v[idx] and v[idx+1], for all idx values.
377  * SHIFT specifies how many new elements are added (1:2, 2:4, ..., 8:256...)
378  * when all elements are used up. New elements are zeroed out.
379  * xrealloc_vector(v, SHIFT, idx) *MUST* be called with consecutive IDXs -
380  * skipping an index is a bad bug - it may miss a realloc!
381  */
382 #define xrealloc_vector(vector, shift, idx) \
383         xrealloc_vector_helper((vector), (sizeof((vector)[0]) << 8) + (shift), (idx))
384 void* xrealloc_vector_helper(void *vector, unsigned sizeof_and_shift, int idx) FAST_FUNC;
385 char *xstrdup(const char *s) FAST_FUNC RETURNS_MALLOC;
386 char *xstrndup(const char *s, int n) FAST_FUNC RETURNS_MALLOC;
387 void *xmemdup(const void *s, int n) FAST_FUNC RETURNS_MALLOC;
388
389
390 //TODO: supply a pointer to char[11] buffer (avoid statics)?
391 extern const char *bb_mode_string(mode_t mode) FAST_FUNC;
392 extern int is_directory(const char *name, int followLinks) FAST_FUNC;
393 enum {  /* cp.c, mv.c, install.c depend on these values. CAREFUL when changing them! */
394         FILEUTILS_PRESERVE_STATUS = 1 << 0, /* -p */
395         FILEUTILS_DEREFERENCE     = 1 << 1, /* !-d */
396         FILEUTILS_RECUR           = 1 << 2, /* -R */
397         FILEUTILS_FORCE           = 1 << 3, /* -f */
398         FILEUTILS_INTERACTIVE     = 1 << 4, /* -i */
399         FILEUTILS_MAKE_HARDLINK   = 1 << 5, /* -l */
400         FILEUTILS_MAKE_SOFTLINK   = 1 << 6, /* -s */
401         FILEUTILS_DEREF_SOFTLINK  = 1 << 7, /* -L */
402         FILEUTILS_DEREFERENCE_L0  = 1 << 8, /* -H */
403         /* -a = -pdR (mapped in cp.c) */
404         /* -r = -dR  (mapped in cp.c) */
405         /* -P = -d   (mapped in cp.c) */
406         FILEUTILS_VERBOSE         = (1 << 12) * ENABLE_FEATURE_VERBOSE, /* -v */
407         FILEUTILS_UPDATE          = 1 << 13, /* -u */
408         FILEUTILS_NO_TARGET_DIR   = 1 << 14, /* -T */
409 #if ENABLE_SELINUX
410         FILEUTILS_PRESERVE_SECURITY_CONTEXT = 1 << 15, /* -c */
411 #endif
412         FILEUTILS_RMDEST          = 1 << (16 - !ENABLE_SELINUX), /* --remove-destination */
413         /* bit 17 skipped for "cp --parents" */
414         FILEUTILS_REFLINK         = 1 << (18 - !ENABLE_SELINUX), /* cp --reflink=auto */
415         FILEUTILS_REFLINK_ALWAYS  = 1 << (19 - !ENABLE_SELINUX), /* cp --reflink[=always] */
416         /*
417          * Hole. cp may have some bits set here,
418          * they should not affect remove_file()/copy_file()
419          */
420 #if ENABLE_SELINUX
421         FILEUTILS_SET_SECURITY_CONTEXT = 1 << 30,
422 #endif
423         FILEUTILS_IGNORE_CHMOD_ERR = 1 << 31,
424 };
425 #define FILEUTILS_CP_OPTSTR "pdRfilsLHarPvuT" IF_SELINUX("c")
426 extern int remove_file(const char *path, int flags) FAST_FUNC;
427 /* NB: without FILEUTILS_RECUR in flags, it will basically "cat"
428  * the source, not copy (unless "source" is a directory).
429  * This makes "cp /dev/null file" and "install /dev/null file" (!!!)
430  * work coreutils-compatibly. */
431 extern int copy_file(const char *source, const char *dest, int flags) FAST_FUNC;
432
433 enum {
434         ACTION_RECURSE        = (1 << 0),
435         ACTION_FOLLOWLINKS    = (1 << 1),
436         ACTION_FOLLOWLINKS_L0 = (1 << 2),
437         ACTION_DEPTHFIRST     = (1 << 3),
438         /*ACTION_REVERSE      = (1 << 4), - unused */
439         ACTION_QUIET          = (1 << 5),
440         ACTION_DANGLING_OK    = (1 << 6),
441 };
442 typedef uint8_t recurse_flags_t;
443 extern int recursive_action(const char *fileName, unsigned flags,
444         int FAST_FUNC (*fileAction)(const char *fileName, struct stat* statbuf, void* userData, int depth),
445         int FAST_FUNC (*dirAction)(const char *fileName, struct stat* statbuf, void* userData, int depth),
446         void* userData, unsigned depth) FAST_FUNC;
447 extern int device_open(const char *device, int mode) FAST_FUNC;
448 enum { GETPTY_BUFSIZE = 16 }; /* more than enough for "/dev/ttyXXX" */
449 extern int xgetpty(char *line) FAST_FUNC;
450 extern int get_console_fd_or_die(void) FAST_FUNC;
451 extern void console_make_active(int fd, const int vt_num) FAST_FUNC;
452 extern char *find_block_device(const char *path) FAST_FUNC;
453 /* bb_copyfd_XX print read/write errors and return -1 if they occur */
454 extern off_t bb_copyfd_eof(int fd1, int fd2) FAST_FUNC;
455 extern off_t bb_copyfd_size(int fd1, int fd2, off_t size) FAST_FUNC;
456 extern void bb_copyfd_exact_size(int fd1, int fd2, off_t size) FAST_FUNC;
457 /* "short" copy can be detected by return value < size */
458 /* this helper yells "short read!" if param is not -1 */
459 extern void complain_copyfd_and_die(off_t sz) NORETURN FAST_FUNC;
460
461 extern char bb_process_escape_sequence(const char **ptr) FAST_FUNC;
462 char* strcpy_and_process_escape_sequences(char *dst, const char *src) FAST_FUNC;
463 /* xxxx_strip version can modify its parameter:
464  * "/"        -> "/"
465  * "abc"      -> "abc"
466  * "abc/def"  -> "def"
467  * "abc/def/" -> "def" !!
468  */
469 char *bb_get_last_path_component_strip(char *path) FAST_FUNC;
470 /* "abc/def/" -> "" and it never modifies 'path' */
471 char *bb_get_last_path_component_nostrip(const char *path) FAST_FUNC;
472 /* Simpler version: does not special case "/" string */
473 const char *bb_basename(const char *name) FAST_FUNC;
474 /* NB: can violate const-ness (similarly to strchr) */
475 char *last_char_is(const char *s, int c) FAST_FUNC;
476 const char* endofname(const char *name) FAST_FUNC;
477 char *is_prefixed_with(const char *string, const char *key) FAST_FUNC;
478 char *is_suffixed_with(const char *string, const char *key) FAST_FUNC;
479
480 int ndelay_on(int fd) FAST_FUNC;
481 int ndelay_off(int fd) FAST_FUNC;
482 void close_on_exec_on(int fd) FAST_FUNC;
483 void xdup2(int, int) FAST_FUNC;
484 void xmove_fd(int, int) FAST_FUNC;
485
486
487 DIR *xopendir(const char *path) FAST_FUNC;
488 DIR *warn_opendir(const char *path) FAST_FUNC;
489
490 char *xmalloc_realpath(const char *path) FAST_FUNC RETURNS_MALLOC;
491 char *xmalloc_realpath_coreutils(const char *path) FAST_FUNC RETURNS_MALLOC;
492 char *xmalloc_readlink(const char *path) FAST_FUNC RETURNS_MALLOC;
493 char *xmalloc_readlink_or_warn(const char *path) FAST_FUNC RETURNS_MALLOC;
494 /* !RETURNS_MALLOC: it's a realloc-like function */
495 char *xrealloc_getcwd_or_warn(char *cwd) FAST_FUNC;
496
497 char *xmalloc_follow_symlinks(const char *path) FAST_FUNC RETURNS_MALLOC;
498
499
500 enum {
501         /* bb_signals(BB_FATAL_SIGS, handler) catches all signals which
502          * otherwise would kill us, except for those resulting from bugs:
503          * SIGSEGV, SIGILL, SIGFPE.
504          * Other fatal signals not included (TODO?):
505          * SIGBUS   Bus error (bad memory access)
506          * SIGPOLL  Pollable event. Synonym of SIGIO
507          * SIGPROF  Profiling timer expired
508          * SIGSYS   Bad argument to routine
509          * SIGTRAP  Trace/breakpoint trap
510          *
511          * The only known arch with some of these sigs not fitting
512          * into 32 bits is parisc (SIGXCPU=33, SIGXFSZ=34, SIGSTKFLT=36).
513          * Dance around with long long to guard against that...
514          */
515         BB_FATAL_SIGS = (int)(0
516                 + (1LL << SIGHUP)
517                 + (1LL << SIGINT)
518                 + (1LL << SIGTERM)
519                 + (1LL << SIGPIPE)   // Write to pipe with no readers
520                 + (1LL << SIGQUIT)   // Quit from keyboard
521                 + (1LL << SIGABRT)   // Abort signal from abort(3)
522                 + (1LL << SIGALRM)   // Timer signal from alarm(2)
523                 + (1LL << SIGVTALRM) // Virtual alarm clock
524                 + (1LL << SIGXCPU)   // CPU time limit exceeded
525                 + (1LL << SIGXFSZ)   // File size limit exceeded
526                 + (1LL << SIGUSR1)   // Yes kids, these are also fatal!
527                 + (1LL << SIGUSR2)
528                 + 0),
529 };
530 void bb_signals(int sigs, void (*f)(int)) FAST_FUNC;
531 /* Unlike signal() and bb_signals, sets handler with sigaction()
532  * and in a way that while signal handler is run, no other signals
533  * will be blocked; syscalls will not be restarted: */
534 void bb_signals_recursive_norestart(int sigs, void (*f)(int)) FAST_FUNC;
535 /* syscalls like read() will be interrupted with EINTR: */
536 void signal_no_SA_RESTART_empty_mask(int sig, void (*handler)(int)) FAST_FUNC;
537 /* syscalls like read() won't be interrupted (though select/poll will be): */
538 void signal_SA_RESTART_empty_mask(int sig, void (*handler)(int)) FAST_FUNC;
539 void wait_for_any_sig(void) FAST_FUNC;
540 void kill_myself_with_sig(int sig) NORETURN FAST_FUNC;
541 void sig_block(int sig) FAST_FUNC;
542 void sig_unblock(int sig) FAST_FUNC;
543 /* Will do sigaction(signum, act, NULL): */
544 int sigaction_set(int sig, const struct sigaction *act) FAST_FUNC;
545 /* SIG_BLOCK/SIG_UNBLOCK all signals: */
546 int sigprocmask_allsigs(int how) FAST_FUNC;
547 /* Standard handler which just records signo */
548 extern smallint bb_got_signal;
549 void record_signo(int signo); /* not FAST_FUNC! */
550
551
552 void xsetgid(gid_t gid) FAST_FUNC;
553 void xsetuid(uid_t uid) FAST_FUNC;
554 void xsetegid(gid_t egid) FAST_FUNC;
555 void xseteuid(uid_t euid) FAST_FUNC;
556 void xchdir(const char *path) FAST_FUNC;
557 void xfchdir(int fd) FAST_FUNC;
558 void xchroot(const char *path) FAST_FUNC;
559 void xsetenv(const char *key, const char *value) FAST_FUNC;
560 void bb_unsetenv(const char *key) FAST_FUNC;
561 void bb_unsetenv_and_free(char *key) FAST_FUNC;
562 void xunlink(const char *pathname) FAST_FUNC;
563 void xstat(const char *pathname, struct stat *buf) FAST_FUNC;
564 void xfstat(int fd, struct stat *buf, const char *errmsg) FAST_FUNC;
565 int open3_or_warn(const char *pathname, int flags, int mode) FAST_FUNC;
566 int open_or_warn(const char *pathname, int flags) FAST_FUNC;
567 int xopen3(const char *pathname, int flags, int mode) FAST_FUNC;
568 int xopen(const char *pathname, int flags) FAST_FUNC;
569 int xopen_nonblocking(const char *pathname) FAST_FUNC;
570 int xopen_as_uid_gid(const char *pathname, int flags, uid_t u, gid_t g) FAST_FUNC;
571 int open_or_warn_stdin(const char *pathname) FAST_FUNC;
572 int xopen_stdin(const char *pathname) FAST_FUNC;
573 void xrename(const char *oldpath, const char *newpath) FAST_FUNC;
574 int rename_or_warn(const char *oldpath, const char *newpath) FAST_FUNC;
575 off_t xlseek(int fd, off_t offset, int whence) FAST_FUNC;
576 int xmkstemp(char *template) FAST_FUNC;
577 off_t fdlength(int fd) FAST_FUNC;
578
579 uoff_t FAST_FUNC get_volume_size_in_bytes(int fd,
580                 const char *override,
581                 unsigned override_units,
582                 int extend);
583
584 void xpipe(int filedes[2]) FAST_FUNC;
585 /* In this form code with pipes is much more readable */
586 struct fd_pair { int rd; int wr; };
587 #define piped_pair(pair)  pipe(&((pair).rd))
588 #define xpiped_pair(pair) xpipe(&((pair).rd))
589
590 /* Useful for having small structure members/global variables */
591 typedef int8_t socktype_t;
592 typedef int8_t family_t;
593 struct BUG_too_small {
594         char BUG_socktype_t_too_small[(0
595                         | SOCK_STREAM
596                         | SOCK_DGRAM
597                         | SOCK_RDM
598                         | SOCK_SEQPACKET
599                         | SOCK_RAW
600                         ) <= 127 ? 1 : -1];
601         char BUG_family_t_too_small[(0
602                         | AF_UNSPEC
603                         | AF_INET
604                         | AF_INET6
605                         | AF_UNIX
606 #ifdef AF_PACKET
607                         | AF_PACKET
608 #endif
609 #ifdef AF_NETLINK
610                         | AF_NETLINK
611 #endif
612                         /* | AF_DECnet */
613                         /* | AF_IPX */
614                         ) <= 127 ? 1 : -1];
615 };
616
617
618 void parse_datestr(const char *date_str, struct tm *ptm) FAST_FUNC;
619 time_t validate_tm_time(const char *date_str, struct tm *ptm) FAST_FUNC;
620 char *strftime_HHMMSS(char *buf, unsigned len, time_t *tp) FAST_FUNC;
621 char *strftime_YYYYMMDDHHMMSS(char *buf, unsigned len, time_t *tp) FAST_FUNC;
622
623 int xsocket(int domain, int type, int protocol) FAST_FUNC;
624 void xbind(int sockfd, struct sockaddr *my_addr, socklen_t addrlen) FAST_FUNC;
625 void xlisten(int s, int backlog) FAST_FUNC;
626 void xconnect(int s, const struct sockaddr *s_addr, socklen_t addrlen) FAST_FUNC;
627 ssize_t xsendto(int s, const void *buf, size_t len, const struct sockaddr *to,
628                                 socklen_t tolen) FAST_FUNC;
629
630 int setsockopt_int(int fd, int level, int optname, int optval) FAST_FUNC;
631 int setsockopt_1(int fd, int level, int optname) FAST_FUNC;
632 int setsockopt_SOL_SOCKET_int(int fd, int optname, int optval) FAST_FUNC;
633 int setsockopt_SOL_SOCKET_1(int fd, int optname) FAST_FUNC;
634 /* SO_REUSEADDR allows a server to rebind to an address that is already
635  * "in use" by old connections to e.g. previous server instance which is
636  * killed or crashed. Without it bind will fail until all such connections
637  * time out. Linux does not allow multiple live binds on same ip:port
638  * regardless of SO_REUSEADDR (unlike some other flavors of Unix).
639  * Turn it on before you call bind(). */
640 void setsockopt_reuseaddr(int fd) FAST_FUNC; /* On Linux this never fails. */
641 int setsockopt_keepalive(int fd) FAST_FUNC;
642 int setsockopt_broadcast(int fd) FAST_FUNC;
643 int setsockopt_bindtodevice(int fd, const char *iface) FAST_FUNC;
644 int bb_getsockname(int sockfd, void *addr, socklen_t addrlen) FAST_FUNC;
645 /* NB: returns port in host byte order */
646 unsigned bb_lookup_port(const char *port, const char *protocol, unsigned default_port) FAST_FUNC;
647 #if ENABLE_FEATURE_ETC_SERVICES
648 # define bb_lookup_std_port(portstr, protocol, portnum) bb_lookup_port(portstr, protocol, portnum)
649 #else
650 # define bb_lookup_std_port(portstr, protocol, portnum) (portnum)
651 #endif
652 typedef struct len_and_sockaddr {
653         socklen_t len;
654         union {
655                 struct sockaddr sa;
656                 struct sockaddr_in sin;
657 #if ENABLE_FEATURE_IPV6
658                 struct sockaddr_in6 sin6;
659 #endif
660         } u;
661 } len_and_sockaddr;
662 enum {
663         LSA_LEN_SIZE = offsetof(len_and_sockaddr, u),
664         LSA_SIZEOF_SA = sizeof(
665                 union {
666                         struct sockaddr sa;
667                         struct sockaddr_in sin;
668 #if ENABLE_FEATURE_IPV6
669                         struct sockaddr_in6 sin6;
670 #endif
671                 }
672         )
673 };
674 /* Create stream socket, and allocate suitable lsa.
675  * (lsa of correct size and lsa->sa.sa_family (AF_INET/AF_INET6))
676  * af == AF_UNSPEC will result in trying to create IPv6 socket,
677  * and if kernel doesn't support it, fall back to IPv4.
678  * This is useful if you plan to bind to resulting local lsa.
679  */
680 int xsocket_type(len_and_sockaddr **lsap, int af, int sock_type) FAST_FUNC;
681 int xsocket_stream(len_and_sockaddr **lsap) FAST_FUNC;
682 /* Create server socket bound to bindaddr:port. bindaddr can be NULL,
683  * numeric IP ("N.N.N.N") or numeric IPv6 address,
684  * and can have ":PORT" suffix (for IPv6 use "[X:X:...:X]:PORT").
685  * Only if there is no suffix, port argument is used */
686 /* NB: these set SO_REUSEADDR before bind */
687 int create_and_bind_stream_or_die(const char *bindaddr, int port) FAST_FUNC;
688 int create_and_bind_dgram_or_die(const char *bindaddr, int port) FAST_FUNC;
689 /* Create client TCP socket connected to peer:port. Peer cannot be NULL.
690  * Peer can be numeric IP ("N.N.N.N"), numeric IPv6 address or hostname,
691  * and can have ":PORT" suffix (for IPv6 use "[X:X:...:X]:PORT").
692  * If there is no suffix, port argument is used */
693 int create_and_connect_stream_or_die(const char *peer, int port) FAST_FUNC;
694 /* Connect to peer identified by lsa */
695 int xconnect_stream(const len_and_sockaddr *lsa) FAST_FUNC;
696 /* Get local address of bound or accepted socket */
697 len_and_sockaddr *get_sock_lsa(int fd) FAST_FUNC RETURNS_MALLOC;
698 /* Get remote address of connected or accepted socket */
699 len_and_sockaddr *get_peer_lsa(int fd) FAST_FUNC RETURNS_MALLOC;
700 /* Return malloc'ed len_and_sockaddr with socket address of host:port
701  * Currently will return IPv4 or IPv6 sockaddrs only
702  * (depending on host), but in theory nothing prevents e.g.
703  * UNIX socket address being returned, IPX sockaddr etc...
704  * On error does bb_error_msg and returns NULL */
705 len_and_sockaddr* host2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
706 /* Version which dies on error */
707 len_and_sockaddr* xhost2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
708 len_and_sockaddr* xdotted2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
709 /* Same, useful if you want to force family (e.g. IPv6) */
710 #if !ENABLE_FEATURE_IPV6
711 #define host_and_af2sockaddr(host, port, af) host2sockaddr((host), (port))
712 #define xhost_and_af2sockaddr(host, port, af) xhost2sockaddr((host), (port))
713 #else
714 len_and_sockaddr* host_and_af2sockaddr(const char *host, int port, sa_family_t af) FAST_FUNC RETURNS_MALLOC;
715 len_and_sockaddr* xhost_and_af2sockaddr(const char *host, int port, sa_family_t af) FAST_FUNC RETURNS_MALLOC;
716 #endif
717 /* Assign sin[6]_port member if the socket is an AF_INET[6] one,
718  * otherwise no-op. Useful for ftp.
719  * NB: does NOT do htons() internally, just direct assignment. */
720 void set_nport(struct sockaddr *sa, unsigned port) FAST_FUNC;
721 /* Retrieve sin[6]_port or return -1 for non-INET[6] lsa's */
722 int get_nport(const struct sockaddr *sa) FAST_FUNC;
723 /* Reverse DNS. Returns NULL on failure. */
724 char* xmalloc_sockaddr2host(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
725 /* This one doesn't append :PORTNUM */
726 char* xmalloc_sockaddr2host_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
727 /* This one also doesn't fall back to dotted IP (returns NULL) */
728 char* xmalloc_sockaddr2hostonly_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
729 /* inet_[ap]ton on steroids */
730 char* xmalloc_sockaddr2dotted(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
731 char* xmalloc_sockaddr2dotted_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
732 // "old" (ipv4 only) API
733 // users: traceroute.c hostname.c - use _list_ of all IPs
734 struct hostent *xgethostbyname(const char *name) FAST_FUNC;
735 // Also mount.c and inetd.c are using gethostbyname(),
736 // + inet_common.c has additional IPv4-only stuff
737
738
739 #define TLS_MAX_MAC_SIZE 32
740 #define TLS_MAX_KEY_SIZE 32
741 struct tls_handshake_data; /* opaque */
742 typedef struct tls_state {
743         int ofd;
744         int ifd;
745
746         unsigned min_encrypted_len_on_read;
747         uint16_t cipher_id;
748         uint8_t  encrypt_on_write;
749         unsigned MAC_size;
750         unsigned key_size;
751
752         uint8_t *outbuf;
753         int     outbuf_size;
754
755         int     inbuf_size;
756         int     ofs_to_buffered;
757         int     buffered_size;
758         uint8_t *inbuf;
759
760         struct tls_handshake_data *hsd;
761
762         // RFC 5246
763         // sequence number
764         //   Each connection state contains a sequence number, which is
765         //   maintained separately for read and write states.  The sequence
766         //   number MUST be set to zero whenever a connection state is made the
767         //   active state.  Sequence numbers are of type uint64 and may not
768         //   exceed 2^64-1.
769         /*uint64_t read_seq64_be;*/
770         uint64_t write_seq64_be;
771
772         uint8_t *client_write_key;
773         uint8_t *server_write_key;
774         uint8_t client_write_MAC_key[TLS_MAX_MAC_SIZE];
775         uint8_t server_write_MAC_k__[TLS_MAX_MAC_SIZE];
776         uint8_t client_write_k__[TLS_MAX_KEY_SIZE];
777         uint8_t server_write_k__[TLS_MAX_KEY_SIZE];
778 } tls_state_t;
779
780 static inline tls_state_t *new_tls_state(void)
781 {
782         tls_state_t *tls = xzalloc(sizeof(*tls));
783         return tls;
784 }
785 void tls_handshake(tls_state_t *tls, const char *sni) FAST_FUNC;
786 #define TLSLOOP_EXIT_ON_LOCAL_EOF (1 << 0)
787 void tls_run_copy_loop(tls_state_t *tls, unsigned flags) FAST_FUNC;
788
789
790 void socket_want_pktinfo(int fd) FAST_FUNC;
791 ssize_t send_to_from(int fd, void *buf, size_t len, int flags,
792                 const struct sockaddr *to,
793                 const struct sockaddr *from,
794                 socklen_t tolen) FAST_FUNC;
795 ssize_t recv_from_to(int fd, void *buf, size_t len, int flags,
796                 struct sockaddr *from,
797                 struct sockaddr *to,
798                 socklen_t sa_size) FAST_FUNC;
799
800 uint16_t inet_cksum(uint16_t *addr, int len) FAST_FUNC;
801 int parse_pasv_epsv(char *buf) FAST_FUNC;
802
803 /* 0 if argv[0] is NULL: */
804 unsigned string_array_len(char **argv) FAST_FUNC;
805 void overlapping_strcpy(char *dst, const char *src) FAST_FUNC;
806 char *safe_strncpy(char *dst, const char *src, size_t size) FAST_FUNC;
807 char *strncpy_IFNAMSIZ(char *dst, const char *src) FAST_FUNC;
808 unsigned count_strstr(const char *str, const char *sub) FAST_FUNC;
809 char *xmalloc_substitute_string(const char *src, int count, const char *sub, const char *repl) FAST_FUNC;
810 /* Guaranteed to NOT be a macro (smallest code). Saves nearly 2k on uclibc.
811  * But potentially slow, don't use in one-billion-times loops */
812 int bb_putchar(int ch) FAST_FUNC;
813 /* Note: does not use stdio, writes to fd 2 directly */
814 int bb_putchar_stderr(char ch) FAST_FUNC;
815 char *xasprintf(const char *format, ...) __attribute__ ((format(printf, 1, 2))) FAST_FUNC RETURNS_MALLOC;
816 char *auto_string(char *str) FAST_FUNC;
817 // gcc-4.1.1 still isn't good enough at optimizing it
818 // (+200 bytes compared to macro)
819 //static ALWAYS_INLINE
820 //int LONE_DASH(const char *s) { return s[0] == '-' && !s[1]; }
821 //static ALWAYS_INLINE
822 //int NOT_LONE_DASH(const char *s) { return s[0] != '-' || s[1]; }
823 #define LONE_DASH(s)     ((s)[0] == '-' && !(s)[1])
824 #define NOT_LONE_DASH(s) ((s)[0] != '-' || (s)[1])
825 #define LONE_CHAR(s,c)     ((s)[0] == (c) && !(s)[1])
826 #define NOT_LONE_CHAR(s,c) ((s)[0] != (c) || (s)[1])
827 #define DOT_OR_DOTDOT(s) ((s)[0] == '.' && (!(s)[1] || ((s)[1] == '.' && !(s)[2])))
828
829 typedef struct uni_stat_t {
830         unsigned byte_count;
831         unsigned unicode_count;
832         unsigned unicode_width;
833 } uni_stat_t;
834 /* Returns a string with unprintable chars replaced by '?' or
835  * SUBST_WCHAR. This function is unicode-aware. */
836 const char* FAST_FUNC printable_string(const char *str);
837 const char* FAST_FUNC printable_string2(uni_stat_t *stats, const char *str);
838 /* Prints unprintable char ch as ^C or M-c to file
839  * (M-c is used only if ch is ORed with PRINTABLE_META),
840  * else it is printed as-is (except for ch = 0x9b) */
841 enum { PRINTABLE_META = 0x100 };
842 void fputc_printable(int ch, FILE *file) FAST_FUNC;
843 /* Return a string that is the printable representation of character ch.
844  * Buffer must hold at least four characters. */
845 enum {
846         VISIBLE_ENDLINE   = 1 << 0,
847         VISIBLE_SHOW_TABS = 1 << 1,
848 };
849 void visible(unsigned ch, char *buf, int flags) FAST_FUNC;
850
851 extern ssize_t safe_read(int fd, void *buf, size_t count) FAST_FUNC;
852 extern ssize_t nonblock_immune_read(int fd, void *buf, size_t count) FAST_FUNC;
853 // NB: will return short read on error, not -1,
854 // if some data was read before error occurred
855 extern ssize_t full_read(int fd, void *buf, size_t count) FAST_FUNC;
856 extern void xread(int fd, void *buf, size_t count) FAST_FUNC;
857 extern unsigned char xread_char(int fd) FAST_FUNC;
858 extern ssize_t read_close(int fd, void *buf, size_t maxsz) FAST_FUNC;
859 extern ssize_t open_read_close(const char *filename, void *buf, size_t maxsz) FAST_FUNC;
860 // Reads one line a-la fgets (but doesn't save terminating '\n').
861 // Reads byte-by-byte. Useful when it is important to not read ahead.
862 // Bytes are appended to pfx (which must be malloced, or NULL).
863 extern char *xmalloc_reads(int fd, size_t *maxsz_p) FAST_FUNC;
864 /* Reads block up to *maxsz_p (default: INT_MAX - 4095) */
865 extern void *xmalloc_read(int fd, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
866 /* Returns NULL if file can't be opened (default max size: INT_MAX - 4095) */
867 extern void *xmalloc_open_read_close(const char *filename, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
868 /* Never returns NULL */
869 extern void *xmalloc_xopen_read_close(const char *filename, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
870
871 #if defined(ARG_MAX) && (ARG_MAX >= 60*1024 || !defined(_SC_ARG_MAX))
872 /* Use _constant_ maximum if: defined && (big enough || no variable one exists) */
873 # define bb_arg_max() ((unsigned)ARG_MAX)
874 #elif defined(_SC_ARG_MAX)
875 /* Else use variable one (a bit more expensive) */
876 unsigned bb_arg_max(void) FAST_FUNC;
877 #else
878 /* If all else fails */
879 # define bb_arg_max() ((unsigned)(32 * 1024))
880 #endif
881 unsigned bb_clk_tck(void) FAST_FUNC;
882
883 #define SEAMLESS_COMPRESSION (0 \
884  || ENABLE_FEATURE_SEAMLESS_XZ \
885  || ENABLE_FEATURE_SEAMLESS_LZMA \
886  || ENABLE_FEATURE_SEAMLESS_BZ2 \
887  || ENABLE_FEATURE_SEAMLESS_GZ \
888  || ENABLE_FEATURE_SEAMLESS_Z)
889
890 #if SEAMLESS_COMPRESSION
891 /* Autodetects gzip/bzip2 formats. fd may be in the middle of the file! */
892 int setup_unzip_on_fd(int fd, int fail_if_not_compressed) FAST_FUNC;
893 /* Autodetects .gz etc */
894 extern int open_zipped(const char *fname, int fail_if_not_compressed) FAST_FUNC;
895 extern void *xmalloc_open_zipped_read_close(const char *fname, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
896 #else
897 # define setup_unzip_on_fd(...) (0)
898 # define open_zipped(fname, fail_if_not_compressed)  open((fname), O_RDONLY);
899 # define xmalloc_open_zipped_read_close(fname, maxsz_p) xmalloc_open_read_close((fname), (maxsz_p))
900 #endif
901 /* lzma has no signature, need a little helper. NB: exist only for ENABLE_FEATURE_SEAMLESS_LZMA=y */
902 void setup_lzma_on_fd(int fd) FAST_FUNC;
903
904 extern ssize_t safe_write(int fd, const void *buf, size_t count) FAST_FUNC;
905 // NB: will return short write on error, not -1,
906 // if some data was written before error occurred
907 extern ssize_t full_write(int fd, const void *buf, size_t count) FAST_FUNC;
908 extern void xwrite(int fd, const void *buf, size_t count) FAST_FUNC;
909 extern void xwrite_str(int fd, const char *str) FAST_FUNC;
910 extern ssize_t full_write1_str(const char *str) FAST_FUNC;
911 extern ssize_t full_write2_str(const char *str) FAST_FUNC;
912 extern void xopen_xwrite_close(const char* file, const char *str) FAST_FUNC;
913
914 /* Close fd, but check for failures (some types of write errors) */
915 extern void xclose(int fd) FAST_FUNC;
916
917 /* Reads and prints to stdout till eof, then closes FILE. Exits on error: */
918 extern void xprint_and_close_file(FILE *file) FAST_FUNC;
919
920 /* Reads a line from a text file, up to a newline or NUL byte, inclusive.
921  * Returns malloc'ed char*. If end is NULL '\n' isn't considered
922  * end of line. If end isn't NULL, length of the chunk is stored in it.
923  * Returns NULL if EOF/error.
924  */
925 extern char *bb_get_chunk_from_file(FILE *file, size_t *end) FAST_FUNC;
926 /* Reads up to (and including) TERMINATING_STRING: */
927 extern char *xmalloc_fgets_str(FILE *file, const char *terminating_string) FAST_FUNC RETURNS_MALLOC;
928 /* Same, with limited max size, and returns the length (excluding NUL): */
929 extern char *xmalloc_fgets_str_len(FILE *file, const char *terminating_string, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
930 /* Chops off TERMINATING_STRING from the end: */
931 extern char *xmalloc_fgetline_str(FILE *file, const char *terminating_string) FAST_FUNC RETURNS_MALLOC;
932 /* Reads up to (and including) "\n" or NUL byte: */
933 extern char *xmalloc_fgets(FILE *file) FAST_FUNC RETURNS_MALLOC;
934 /* Chops off '\n' from the end, unlike fgets: */
935 extern char *xmalloc_fgetline(FILE *file) FAST_FUNC RETURNS_MALLOC;
936 /* Same, but doesn't try to conserve space (may have some slack after the end) */
937 /* extern char *xmalloc_fgetline_fast(FILE *file) FAST_FUNC RETURNS_MALLOC; */
938
939 void die_if_ferror(FILE *file, const char *msg) FAST_FUNC;
940 void die_if_ferror_stdout(void) FAST_FUNC;
941 int fflush_all(void) FAST_FUNC;
942 void fflush_stdout_and_exit(int retval) NORETURN FAST_FUNC;
943 int fclose_if_not_stdin(FILE *file) FAST_FUNC;
944 FILE* xfopen(const char *filename, const char *mode) FAST_FUNC;
945 /* Prints warning to stderr and returns NULL on failure: */
946 FILE* fopen_or_warn(const char *filename, const char *mode) FAST_FUNC;
947 /* "Opens" stdin if filename is special, else just opens file: */
948 FILE* xfopen_stdin(const char *filename) FAST_FUNC;
949 FILE* fopen_or_warn_stdin(const char *filename) FAST_FUNC;
950 FILE* fopen_for_read(const char *path) FAST_FUNC;
951 FILE* xfopen_for_read(const char *path) FAST_FUNC;
952 FILE* fopen_for_write(const char *path) FAST_FUNC;
953 FILE* xfopen_for_write(const char *path) FAST_FUNC;
954 FILE* xfdopen_for_read(int fd) FAST_FUNC;
955 FILE* xfdopen_for_write(int fd) FAST_FUNC;
956
957 int bb_pstrcmp(const void *a, const void *b) /* not FAST_FUNC! */;
958 void qsort_string_vector(char **sv, unsigned count) FAST_FUNC;
959
960 /* Wrapper which restarts poll on EINTR or ENOMEM.
961  * On other errors complains [perror("poll")] and returns.
962  * Warning! May take (much) longer than timeout_ms to return!
963  * If this is a problem, use bare poll and open-code EINTR/ENOMEM handling */
964 int safe_poll(struct pollfd *ufds, nfds_t nfds, int timeout_ms) FAST_FUNC;
965
966 char *safe_gethostname(void) FAST_FUNC;
967
968 /* Convert each alpha char in str to lower-case */
969 char* str_tolower(char *str) FAST_FUNC;
970
971 char *utoa(unsigned n) FAST_FUNC;
972 char *itoa(int n) FAST_FUNC;
973 /* Returns a pointer past the formatted number, does NOT null-terminate */
974 char *utoa_to_buf(unsigned n, char *buf, unsigned buflen) FAST_FUNC;
975 char *itoa_to_buf(int n, char *buf, unsigned buflen) FAST_FUNC;
976 /* Intelligent formatters of bignums */
977 char *smart_ulltoa4(unsigned long long ul, char buf[4], const char *scale) FAST_FUNC;
978 char *smart_ulltoa5(unsigned long long ul, char buf[5], const char *scale) FAST_FUNC;
979 /* If block_size == 0, display size without fractional part,
980  * else display (size * block_size) with one decimal digit.
981  * If display_unit == 0, show value no bigger than 1024 with suffix (K,M,G...),
982  * else divide by display_unit and do not use suffix. */
983 #define HUMAN_READABLE_MAX_WIDTH      7  /* "1024.0G" */
984 #define HUMAN_READABLE_MAX_WIDTH_STR "7"
985 //TODO: provide pointer to buf (avoid statics)?
986 const char *make_human_readable_str(unsigned long long size,
987                 unsigned long block_size, unsigned long display_unit) FAST_FUNC;
988 /* Put a string of hex bytes ("1b2e66fe"...), return advanced pointer */
989 char *bin2hex(char *dst, const char *src, int count) FAST_FUNC;
990 /* Reverse */
991 char* hex2bin(char *dst, const char *src, int count) FAST_FUNC;
992
993 /* Generate a UUID */
994 void generate_uuid(uint8_t *buf) FAST_FUNC;
995
996 /* Last element is marked by mult == 0 */
997 struct suffix_mult {
998         char suffix[4];
999         unsigned mult;
1000 };
1001 extern const struct suffix_mult bkm_suffixes[];
1002 #define km_suffixes (bkm_suffixes + 1)
1003 extern const struct suffix_mult cwbkMG_suffixes[];
1004 #define kMG_suffixes (cwbkMG_suffixes + 3)
1005 extern const struct suffix_mult kmg_i_suffixes[];
1006
1007 #include "xatonum.h"
1008 /* Specialized: */
1009
1010 /* Using xatoi() instead of naive atoi() is not always convenient -
1011  * in many places people want *non-negative* values, but store them
1012  * in signed int. Therefore we need this one:
1013  * dies if input is not in [0, INT_MAX] range. Also will reject '-0' etc.
1014  * It should really be named xatoi_nonnegative (since it allows 0),
1015  * but that would be too long.
1016  */
1017 int xatoi_positive(const char *numstr) FAST_FUNC;
1018
1019 /* Useful for reading port numbers */
1020 uint16_t xatou16(const char *numstr) FAST_FUNC;
1021
1022 #if ENABLE_FLOAT_DURATION
1023 typedef double duration_t;
1024 void sleep_for_duration(duration_t duration) FAST_FUNC;
1025 #else
1026 typedef unsigned duration_t;
1027 #define sleep_for_duration(duration) sleep(duration)
1028 #endif
1029 duration_t parse_duration_str(char *str) FAST_FUNC;
1030
1031 /* These parse entries in /etc/passwd and /etc/group.  This is desirable
1032  * for BusyBox since we want to avoid using the glibc NSS stuff, which
1033  * increases target size and is often not needed on embedded systems.  */
1034 long xuname2uid(const char *name) FAST_FUNC;
1035 long xgroup2gid(const char *name) FAST_FUNC;
1036 /* wrapper: allows string to contain numeric uid or gid */
1037 unsigned long get_ug_id(const char *s, long FAST_FUNC (*xname2id)(const char *)) FAST_FUNC;
1038 struct bb_uidgid_t {
1039         uid_t uid;
1040         gid_t gid;
1041 };
1042 /* always sets uid and gid; returns 0 on failure */
1043 int get_uidgid(struct bb_uidgid_t*, const char*) FAST_FUNC;
1044 /* always sets uid and gid; exits on failure */
1045 void xget_uidgid(struct bb_uidgid_t*, const char*) FAST_FUNC;
1046 /* chown-like handling of "user[:[group]" */
1047 void parse_chown_usergroup_or_die(struct bb_uidgid_t *u, char *user_group) FAST_FUNC;
1048 struct passwd* xgetpwnam(const char *name) FAST_FUNC;
1049 struct group* xgetgrnam(const char *name) FAST_FUNC;
1050 struct passwd* xgetpwuid(uid_t uid) FAST_FUNC;
1051 struct group* xgetgrgid(gid_t gid) FAST_FUNC;
1052 char* xuid2uname(uid_t uid) FAST_FUNC;
1053 char* xgid2group(gid_t gid) FAST_FUNC;
1054 char* uid2uname(uid_t uid) FAST_FUNC;
1055 char* gid2group(gid_t gid) FAST_FUNC;
1056 char* uid2uname_utoa(uid_t uid) FAST_FUNC;
1057 char* gid2group_utoa(gid_t gid) FAST_FUNC;
1058 /* versions which cache results (useful for ps, ls etc) */
1059 const char* get_cached_username(uid_t uid) FAST_FUNC;
1060 const char* get_cached_groupname(gid_t gid) FAST_FUNC;
1061 void clear_username_cache(void) FAST_FUNC;
1062 /* internally usernames are saved in fixed-sized char[] buffers */
1063 enum { USERNAME_MAX_SIZE = 32 - sizeof(uid_t) };
1064 #if ENABLE_FEATURE_CHECK_NAMES
1065 void die_if_bad_username(const char* name) FAST_FUNC;
1066 #else
1067 #define die_if_bad_username(name) ((void)(name))
1068 #endif
1069 /*
1070  * Returns (-1) terminated malloced result of getgroups().
1071  * Reallocs group_array (useful for repeated calls).
1072  * ngroups is an initial size of array. It is rounded up to 32 for realloc.
1073  * ngroups is updated on return.
1074  * ngroups can be NULL: bb_getgroups(NULL, NULL) is valid usage.
1075  * Dies on errors (on Linux, only xrealloc can cause this, not internal getgroups call).
1076  */
1077 gid_t *bb_getgroups(int *ngroups, gid_t *group_array) FAST_FUNC;
1078
1079 #if ENABLE_FEATURE_UTMP
1080 void FAST_FUNC write_new_utmp(pid_t pid, int new_type, const char *tty_name, const char *username, const char *hostname);
1081 void FAST_FUNC update_utmp(pid_t pid, int new_type, const char *tty_name, const char *username, const char *hostname);
1082 void FAST_FUNC update_utmp_DEAD_PROCESS(pid_t pid);
1083 #else
1084 # define write_new_utmp(pid, new_type, tty_name, username, hostname) ((void)0)
1085 # define update_utmp(pid, new_type, tty_name, username, hostname) ((void)0)
1086 # define update_utmp_DEAD_PROCESS(pid) ((void)0)
1087 #endif
1088
1089
1090 int file_is_executable(const char *name) FAST_FUNC;
1091 char *find_executable(const char *filename, char **PATHp) FAST_FUNC;
1092 int executable_exists(const char *filename) FAST_FUNC;
1093
1094 /* BB_EXECxx always execs (it's not doing NOFORK/NOEXEC stuff),
1095  * but it may exec busybox and call applet instead of searching PATH.
1096  */
1097 #if ENABLE_FEATURE_PREFER_APPLETS
1098 int BB_EXECVP(const char *file, char *const argv[]) FAST_FUNC;
1099 #define BB_EXECLP(prog,cmd,...) \
1100         do { \
1101                 if (find_applet_by_name(prog) >= 0) \
1102                         execlp(bb_busybox_exec_path, cmd, __VA_ARGS__); \
1103                 execlp(prog, cmd, __VA_ARGS__); \
1104         } while (0)
1105 #else
1106 #define BB_EXECVP(prog,cmd)     execvp(prog,cmd)
1107 #define BB_EXECLP(prog,cmd,...) execlp(prog,cmd,__VA_ARGS__)
1108 #endif
1109 void BB_EXECVP_or_die(char **argv) NORETURN FAST_FUNC;
1110 void exec_prog_or_SHELL(char **argv) NORETURN FAST_FUNC;
1111
1112 /* xvfork() can't be a _function_, return after vfork in child mangles stack
1113  * in the parent. It must be a macro. */
1114 #define xvfork() \
1115 ({ \
1116         pid_t bb__xvfork_pid = vfork(); \
1117         if (bb__xvfork_pid < 0) \
1118                 bb_perror_msg_and_die("vfork"); \
1119         bb__xvfork_pid; \
1120 })
1121 #if BB_MMU
1122 pid_t xfork(void) FAST_FUNC;
1123 #endif
1124 void xvfork_parent_waits_and_exits(void) FAST_FUNC;
1125
1126 /* NOMMU friendy fork+exec: */
1127 pid_t spawn(char **argv) FAST_FUNC;
1128 pid_t xspawn(char **argv) FAST_FUNC;
1129
1130 pid_t safe_waitpid(pid_t pid, int *wstat, int options) FAST_FUNC;
1131 pid_t wait_any_nohang(int *wstat) FAST_FUNC;
1132 /* wait4pid: unlike waitpid, waits ONLY for one process.
1133  * Returns sig + 0x180 if child is killed by signal.
1134  * It's safe to pass negative 'pids' from failed [v]fork -
1135  * wait4pid will return -1 (and will not clobber [v]fork's errno).
1136  * IOW: rc = wait4pid(spawn(argv));
1137  *      if (rc < 0) bb_perror_msg("%s", argv[0]);
1138  *      if (rc > 0) bb_error_msg("exit code: %d", rc & 0xff);
1139  */
1140 int wait4pid(pid_t pid) FAST_FUNC;
1141 int wait_for_exitstatus(pid_t pid) FAST_FUNC;
1142 /************************************************************************/
1143 /* spawn_and_wait/run_nofork_applet/run_applet_no_and_exit need to work */
1144 /* carefully together to reinit some global state while not disturbing  */
1145 /* other. Be careful if you change them. Consult docs/nofork_noexec.txt */
1146 /************************************************************************/
1147 /* Same as wait4pid(spawn(argv)), but with NOFORK/NOEXEC if configured: */
1148 int spawn_and_wait(char **argv) FAST_FUNC;
1149 /* Does NOT check that applet is NOFORK, just blindly runs it */
1150 int run_nofork_applet(int applet_no, char **argv) FAST_FUNC;
1151 void run_noexec_applet_and_exit(int a, const char *name, char **argv) NORETURN FAST_FUNC;
1152 #ifndef BUILD_INDIVIDUAL
1153 int find_applet_by_name(const char *name) FAST_FUNC;
1154 void run_applet_no_and_exit(int a, const char *name, char **argv) NORETURN FAST_FUNC;
1155 #endif
1156 #if defined(__linux__)
1157 void set_task_comm(const char *comm) FAST_FUNC;
1158 #else
1159 # define set_task_comm(name) ((void)0)
1160 #endif
1161
1162 /* Helpers for daemonization.
1163  *
1164  * bb_daemonize(flags) = daemonize, does not compile on NOMMU
1165  *
1166  * bb_daemonize_or_rexec(flags, argv) = daemonizes on MMU (and ignores argv),
1167  *      rexec's itself on NOMMU with argv passed as command line.
1168  * Thus bb_daemonize_or_rexec may cause your <applet>_main() to be re-executed
1169  * from the start. (It will detect it and not reexec again second time).
1170  * You have to audit carefully that you don't do something twice as a result
1171  * (opening files/sockets, parsing config files etc...)!
1172  *
1173  * Both of the above will redirect fd 0,1,2 to /dev/null and drop ctty
1174  * (will do setsid()).
1175  *
1176  * fork_or_rexec(argv) = bare-bones fork on MMU,
1177  *      "vfork + re-exec ourself" on NOMMU. No fd redirection, no setsid().
1178  *      On MMU ignores argv.
1179  *
1180  * Helper for network daemons in foreground mode:
1181  *
1182  * bb_sanitize_stdio() = make sure that fd 0,1,2 are opened by opening them
1183  * to /dev/null if they are not.
1184  */
1185 enum {
1186         DAEMON_CHDIR_ROOT = 1,
1187         DAEMON_DEVNULL_STDIO = 2,
1188         DAEMON_CLOSE_EXTRA_FDS = 4,
1189         DAEMON_ONLY_SANITIZE = 8, /* internal use */
1190         DAEMON_DOUBLE_FORK = 16, /* double fork to avoid controlling tty */
1191 };
1192 #if BB_MMU
1193   enum { re_execed = 0 };
1194 # define fork_or_rexec(argv)                xfork()
1195 # define bb_daemonize_or_rexec(flags, argv) bb_daemonize_or_rexec(flags)
1196 # define bb_daemonize(flags)                bb_daemonize_or_rexec(flags, bogus)
1197 #else
1198   extern bool re_execed;
1199   /* Note: re_exec() and fork_or_rexec() do argv[0][0] |= 0x80 on NOMMU!
1200    * _Parent_ needs to undo it if it doesn't want to have argv[0] mangled.
1201    */
1202   void re_exec(char **argv) NORETURN FAST_FUNC;
1203   pid_t fork_or_rexec(char **argv) FAST_FUNC;
1204   int  BUG_fork_is_unavailable_on_nommu(void) FAST_FUNC;
1205   int  BUG_daemon_is_unavailable_on_nommu(void) FAST_FUNC;
1206   void BUG_bb_daemonize_is_unavailable_on_nommu(void) FAST_FUNC;
1207 # define fork()          BUG_fork_is_unavailable_on_nommu()
1208 # define xfork()         BUG_fork_is_unavailable_on_nommu()
1209 # define daemon(a,b)     BUG_daemon_is_unavailable_on_nommu()
1210 # define bb_daemonize(a) BUG_bb_daemonize_is_unavailable_on_nommu()
1211 #endif
1212 void bb_daemonize_or_rexec(int flags, char **argv) FAST_FUNC;
1213 void bb_sanitize_stdio(void) FAST_FUNC;
1214 #define bb_daemon_helper(arg) bb_daemonize_or_rexec((arg) | DAEMON_ONLY_SANITIZE, NULL)
1215 /* Clear dangerous stuff, set PATH. Return 1 if was run by different user. */
1216 int sanitize_env_if_suid(void) FAST_FUNC;
1217
1218
1219 /* For top, ps. Some argv[i] are replaced by malloced "-opt" strings */
1220 void make_all_argv_opts(char **argv) FAST_FUNC;
1221 char* single_argv(char **argv) FAST_FUNC;
1222 extern const char *const bb_argv_dash[]; /* { "-", NULL } */
1223 extern uint32_t option_mask32;
1224 uint32_t getopt32(char **argv, const char *applet_opts, ...) FAST_FUNC;
1225 # define No_argument "\0"
1226 # define Required_argument "\001"
1227 # define Optional_argument "\002"
1228 #if ENABLE_LONG_OPTS
1229 uint32_t getopt32long(char **argv, const char *optstring, const char *longopts, ...) FAST_FUNC;
1230 #else
1231 #define getopt32long(argv,optstring,longopts,...) \
1232         getopt32(argv,optstring,##__VA_ARGS__)
1233 #endif
1234 /* BSD-derived getopt() functions require that optind be set to 1 in
1235  * order to reset getopt() state.  This used to be generally accepted
1236  * way of resetting getopt().  However, glibc's getopt()
1237  * has additional getopt() state beyond optind (specifically, glibc
1238  * extensions such as '+' and '-' at the start of the string), and requires
1239  * that optind be set to zero to reset its state.  BSD-derived versions
1240  * of getopt() misbehaved if optind is set to 0 in order to reset getopt(),
1241  * and glibc's getopt() used to coredump if optind is set 1 in order
1242  * to reset getopt().
1243  * Then BSD introduced additional variable "optreset" which should be
1244  * set to 1 in order to reset getopt().  Sigh.  Standards, anyone?
1245  *
1246  * By ~2008, OpenBSD 3.4 was changed to survive glibc-like optind = 0
1247  * (to interpret it as if optreset was set).
1248  */
1249 #if 1 /*def __GLIBC__*/
1250 #define GETOPT_RESET() (optind = 0)
1251 #else /* BSD style */
1252 #define GETOPT_RESET() (optind = 1)
1253 #endif
1254
1255
1256 /* Having next pointer as a first member allows easy creation
1257  * of "llist-compatible" structs, and using llist_FOO functions
1258  * on them.
1259  */
1260 typedef struct llist_t {
1261         struct llist_t *link;
1262         char *data;
1263 } llist_t;
1264 void llist_add_to(llist_t **old_head, void *data) FAST_FUNC;
1265 void llist_add_to_end(llist_t **list_head, void *data) FAST_FUNC;
1266 void *llist_pop(llist_t **elm) FAST_FUNC;
1267 void llist_unlink(llist_t **head, llist_t *elm) FAST_FUNC;
1268 void llist_free(llist_t *elm, void (*freeit)(void *data)) FAST_FUNC;
1269 llist_t *llist_rev(llist_t *list) FAST_FUNC;
1270 llist_t *llist_find_str(llist_t *first, const char *str) FAST_FUNC;
1271 /* BTW, surprisingly, changing API to
1272  *   llist_t *llist_add_to(llist_t *old_head, void *data)
1273  * etc does not result in smaller code... */
1274
1275 /* start_stop_daemon and udhcpc are special - they want
1276  * to create pidfiles regardless of FEATURE_PIDFILE */
1277 #if ENABLE_FEATURE_PIDFILE || defined(WANT_PIDFILE)
1278 /* True only if we created pidfile which is *file*, not /dev/null etc */
1279 extern smallint wrote_pidfile;
1280 void write_pidfile(const char *path) FAST_FUNC;
1281 #define remove_pidfile(path) do { if (wrote_pidfile) unlink(path); } while (0)
1282 #else
1283 enum { wrote_pidfile = 0 };
1284 #define write_pidfile(path)  ((void)0)
1285 #define remove_pidfile(path) ((void)0)
1286 #endif
1287
1288 enum {
1289         LOGMODE_NONE = 0,
1290         LOGMODE_STDIO = (1 << 0),
1291         LOGMODE_SYSLOG = (1 << 1) * ENABLE_FEATURE_SYSLOG,
1292         LOGMODE_BOTH = LOGMODE_SYSLOG + LOGMODE_STDIO,
1293 };
1294 extern const char *msg_eol;
1295 extern smallint syslog_level;
1296 extern smallint logmode;
1297 extern uint8_t xfunc_error_retval;
1298 extern void (*die_func)(void);
1299 void xfunc_die(void) NORETURN FAST_FUNC;
1300 void bb_show_usage(void) NORETURN FAST_FUNC;
1301 void bb_error_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1302 void bb_error_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1303 void bb_perror_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1304 void bb_simple_perror_msg(const char *s) FAST_FUNC;
1305 void bb_perror_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1306 void bb_simple_perror_msg_and_die(const char *s) NORETURN FAST_FUNC;
1307 void bb_herror_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1308 void bb_herror_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1309 void bb_perror_nomsg_and_die(void) NORETURN FAST_FUNC;
1310 void bb_perror_nomsg(void) FAST_FUNC;
1311 void bb_verror_msg(const char *s, va_list p, const char *strerr) FAST_FUNC;
1312 void bb_die_memory_exhausted(void) NORETURN FAST_FUNC;
1313 void bb_logenv_override(void) FAST_FUNC;
1314
1315 /* We need to export XXX_main from libbusybox
1316  * only if we build "individual" binaries
1317  */
1318 #if ENABLE_FEATURE_INDIVIDUAL
1319 #define MAIN_EXTERNALLY_VISIBLE EXTERNALLY_VISIBLE
1320 #else
1321 #define MAIN_EXTERNALLY_VISIBLE
1322 #endif
1323
1324 /* Embedded script support */
1325 int find_script_by_name(const char *name) FAST_FUNC;
1326 char *get_script_content(unsigned n) FAST_FUNC;
1327
1328 /* Applets which are useful from another applets */
1329 int bb_cat(char** argv) FAST_FUNC;
1330 int ash_main(int argc, char** argv)
1331 #if ENABLE_ASH || ENABLE_SH_IS_ASH || ENABLE_BASH_IS_ASH
1332                 MAIN_EXTERNALLY_VISIBLE
1333 #endif
1334 ;
1335 /* If shell needs them, they exist even if not enabled as applets */
1336 int echo_main(int argc, char** argv) IF_ECHO(MAIN_EXTERNALLY_VISIBLE);
1337 int printf_main(int argc, char **argv) IF_PRINTF(MAIN_EXTERNALLY_VISIBLE);
1338 int test_main(int argc, char **argv)
1339 #if ENABLE_TEST || ENABLE_TEST1 || ENABLE_TEST2
1340                 MAIN_EXTERNALLY_VISIBLE
1341 #endif
1342 ;
1343 int kill_main(int argc, char **argv)
1344 #if ENABLE_KILL || ENABLE_KILLALL || ENABLE_KILLALL5
1345                 MAIN_EXTERNALLY_VISIBLE
1346 #endif
1347 ;
1348 /* Similar, but used by chgrp, not shell */
1349 int chown_main(int argc, char **argv) IF_CHOWN(MAIN_EXTERNALLY_VISIBLE);
1350 /* Used by ftpd */
1351 int ls_main(int argc, char **argv) IF_LS(MAIN_EXTERNALLY_VISIBLE);
1352 /* Don't need IF_xxx() guard for these */
1353 int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1354 int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1355
1356 #if ENABLE_ROUTE
1357 void bb_displayroutes(int noresolve, int netstatfmt) FAST_FUNC;
1358 #endif
1359
1360 struct number_state {
1361         unsigned width;
1362         unsigned start;
1363         unsigned inc;
1364         const char *sep;
1365         const char *empty_str;
1366         smallint all, nonempty;
1367 };
1368 void print_numbered_lines(struct number_state *ns, const char *filename) FAST_FUNC;
1369
1370
1371 /* Networking */
1372 /* This structure defines protocol families and their handlers. */
1373 struct aftype {
1374         const char *name;
1375         const char *title;
1376         int af;
1377         int alen;
1378         char*       FAST_FUNC (*print)(unsigned char *);
1379         const char* FAST_FUNC (*sprint)(struct sockaddr *, int numeric);
1380         int         FAST_FUNC (*input)(/*int type,*/ const char *bufp, struct sockaddr *);
1381         void        FAST_FUNC (*herror)(char *text);
1382         int         FAST_FUNC (*rprint)(int options);
1383         int         FAST_FUNC (*rinput)(int typ, int ext, char **argv);
1384         /* may modify src */
1385         int         FAST_FUNC (*getmask)(char *src, struct sockaddr *mask, char *name);
1386 };
1387 /* This structure defines hardware protocols and their handlers. */
1388 struct hwtype {
1389         const char *name;
1390         const char *title;
1391         int type;
1392         int alen;
1393         char* FAST_FUNC (*print)(unsigned char *);
1394         int   FAST_FUNC (*input)(const char *, struct sockaddr *);
1395         int   FAST_FUNC (*activate)(int fd);
1396         int suppress_null_addr;
1397 };
1398 #define IFNAME_SHOW_DOWNED_TOO ((char*)(intptr_t)1)
1399 int display_interfaces(char *ifname) FAST_FUNC;
1400 int in_ether(const char *bufp, struct sockaddr *sap) FAST_FUNC;
1401 #if ENABLE_FEATURE_HWIB
1402 int in_ib(const char *bufp, struct sockaddr *sap) FAST_FUNC;
1403 #else
1404 #define in_ib(a, b) 1 /* fail */
1405 #endif
1406 const struct aftype *get_aftype(const char *name) FAST_FUNC;
1407 const struct hwtype *get_hwtype(const char *name) FAST_FUNC;
1408 const struct hwtype *get_hwntype(int type) FAST_FUNC;
1409
1410
1411 extern int fstype_matches(const char *fstype, const char *comma_list) FAST_FUNC;
1412 #ifdef HAVE_MNTENT_H
1413 extern struct mntent *find_mount_point(const char *name, int subdir_too) FAST_FUNC;
1414 #endif
1415 extern void erase_mtab(const char * name) FAST_FUNC;
1416 extern unsigned int tty_baud_to_value(speed_t speed) FAST_FUNC;
1417 extern speed_t tty_value_to_baud(unsigned int value) FAST_FUNC;
1418 #if ENABLE_DESKTOP
1419 extern void bb_warn_ignoring_args(char *arg) FAST_FUNC;
1420 #else
1421 # define bb_warn_ignoring_args(arg) ((void)0)
1422 #endif
1423
1424 extern int get_linux_version_code(void) FAST_FUNC;
1425
1426 extern char *query_loop(const char *device) FAST_FUNC;
1427 extern int del_loop(const char *device) FAST_FUNC;
1428 /*
1429  * If *devname is not NULL, use that name, otherwise try to find free one,
1430  * malloc and return it in *devname.
1431  * return value is the opened fd to the loop device, or < on error
1432  */
1433 extern int set_loop(char **devname, const char *file, unsigned long long offset, unsigned flags) FAST_FUNC;
1434 /* These constants match linux/loop.h (without BB_ prefix): */
1435 #define BB_LO_FLAGS_READ_ONLY 1
1436 #define BB_LO_FLAGS_AUTOCLEAR 4
1437
1438 /* Returns malloced str */
1439 char *bb_ask_noecho(int fd, int timeout, const char *prompt) FAST_FUNC;
1440 /* Like bb_ask_noecho, but asks on stdin with no timeout.  */
1441 char *bb_ask_noecho_stdin(const char *prompt) FAST_FUNC;
1442
1443 int bb_ask_y_confirmation_FILE(FILE *fp) FAST_FUNC;
1444 int bb_ask_y_confirmation(void) FAST_FUNC;
1445
1446 /* Returns -1 if input is invalid. current_mode is a base for e.g. "u+rw" */
1447 int bb_parse_mode(const char* s, unsigned cur_mode) FAST_FUNC;
1448
1449 /*
1450  * Config file parser
1451  */
1452 enum {
1453         PARSE_COLLAPSE  = 0x00010000, // treat consecutive delimiters as one
1454         PARSE_TRIM      = 0x00020000, // trim leading and trailing delimiters
1455 // TODO: COLLAPSE and TRIM seem to always go in pair
1456         PARSE_GREEDY    = 0x00040000, // last token takes entire remainder of the line
1457         PARSE_MIN_DIE   = 0x00100000, // die if < min tokens found
1458         // keep a copy of current line
1459         PARSE_KEEP_COPY = 0x00200000 * ENABLE_FEATURE_CROND_D,
1460         PARSE_EOL_COMMENTS = 0x00400000, // comments are recognized even if they aren't the first char
1461         PARSE_ALT_COMMENTS = 0x00800000, // delim[0] and delim[1] are two different allowed comment chars
1462         // (so far, delim[0] will only work as comment char for full-line comment)
1463         // (IOW: it works as if PARSE_EOL_COMMENTS is not set. sysctl applet is okay with this)
1464         PARSE_WS_COMMENTS  = 0x01000000, // comments are recognized even if there is whitespace before
1465         // ("line start><space><tab><space>#comment" is also comment, not only "line start>#comment")
1466         // NORMAL is:
1467         // * remove leading and trailing delimiters and collapse
1468         //   multiple delimiters into one
1469         // * warn and continue if less than mintokens delimiters found
1470         // * grab everything into last token
1471         // * comments are recognized even if they aren't the first char
1472         PARSE_NORMAL    = PARSE_COLLAPSE | PARSE_TRIM | PARSE_GREEDY | PARSE_EOL_COMMENTS,
1473 };
1474 typedef struct parser_t {
1475         FILE *fp;
1476         char *data;
1477         char *line, *nline;
1478         size_t line_alloc, nline_alloc;
1479         int lineno;
1480 } parser_t;
1481 parser_t* config_open(const char *filename) FAST_FUNC;
1482 parser_t* config_open2(const char *filename, FILE* FAST_FUNC (*fopen_func)(const char *path)) FAST_FUNC;
1483 /* delims[0] is a comment char (use '\0' to disable), the rest are token delimiters */
1484 int config_read(parser_t *parser, char **tokens, unsigned flags, const char *delims) FAST_FUNC;
1485 #define config_read(parser, tokens, max, min, str, flags) \
1486         config_read(parser, tokens, ((flags) | (((min) & 0xFF) << 8) | ((max) & 0xFF)), str)
1487 void config_close(parser_t *parser) FAST_FUNC;
1488
1489 /* Concatenate path and filename to new allocated buffer.
1490  * Add "/" only as needed (no duplicate "//" are produced).
1491  * If path is NULL, it is assumed to be "/".
1492  * filename should not be NULL. */
1493 char *concat_path_file(const char *path, const char *filename) FAST_FUNC;
1494 /* Returns NULL on . and .. */
1495 char *concat_subpath_file(const char *path, const char *filename) FAST_FUNC;
1496
1497
1498 int bb_make_directory(char *path, long mode, int flags) FAST_FUNC;
1499
1500 int get_signum(const char *name) FAST_FUNC;
1501 const char *get_signame(int number) FAST_FUNC;
1502 void print_signames(void) FAST_FUNC;
1503
1504 char *bb_simplify_path(const char *path) FAST_FUNC;
1505 /* Returns ptr to NUL */
1506 char *bb_simplify_abs_path_inplace(char *path) FAST_FUNC;
1507
1508 #ifndef LOGIN_FAIL_DELAY
1509 #define LOGIN_FAIL_DELAY 3
1510 #endif
1511 extern void bb_do_delay(int seconds) FAST_FUNC;
1512 extern void change_identity(const struct passwd *pw) FAST_FUNC;
1513 extern void run_shell(const char *shell, int loginshell, const char **args) NORETURN FAST_FUNC;
1514
1515 /* Returns $SHELL, getpwuid(getuid())->pw_shell, or DEFAULT_SHELL.
1516  * Note that getpwuid result might need xstrdup'ing
1517  * if there is a possibility of intervening getpwxxx() calls.
1518  */
1519 const char *get_shell_name(void) FAST_FUNC;
1520
1521 #if ENABLE_FEATURE_SETPRIV_CAPABILITIES || ENABLE_RUN_INIT
1522 unsigned cap_name_to_number(const char *cap) FAST_FUNC;
1523 void printf_cap(const char *pfx, unsigned cap_no) FAST_FUNC;
1524 void drop_capability(int cap_ordinal) FAST_FUNC;
1525 /* Structures inside "struct caps" are Linux-specific and libcap-specific: */
1526 #define DEFINE_STRUCT_CAPS \
1527 struct caps { \
1528         struct __user_cap_header_struct header; \
1529         unsigned u32s; \
1530         struct __user_cap_data_struct data[2]; \
1531 }
1532 void getcaps(void *caps) FAST_FUNC;
1533 #endif
1534
1535 #if ENABLE_SELINUX
1536 extern void renew_current_security_context(void) FAST_FUNC;
1537 extern void set_current_security_context(security_context_t sid) FAST_FUNC;
1538 extern context_t set_security_context_component(security_context_t cur_context,
1539                                                 char *user, char *role, char *type, char *range) FAST_FUNC;
1540 extern void setfscreatecon_or_die(security_context_t scontext) FAST_FUNC;
1541 extern void selinux_preserve_fcontext(int fdesc) FAST_FUNC;
1542 #else
1543 #define selinux_preserve_fcontext(fdesc) ((void)0)
1544 #endif
1545 extern void selinux_or_die(void) FAST_FUNC;
1546
1547
1548 /* setup_environment:
1549  * if chdir pw->pw_dir: ok: else if to_tmp == 1: goto /tmp else: goto / or die
1550  * if clear_env = 1: cd(pw->pw_dir), clear environment, then set
1551  *   TERM=(old value)
1552  *   USER=pw->pw_name, LOGNAME=pw->pw_name
1553  *   PATH=bb_default_[root_]path
1554  *   HOME=pw->pw_dir
1555  *   SHELL=shell
1556  * else if change_env = 1:
1557  *   if not root (if pw->pw_uid != 0):
1558  *     USER=pw->pw_name, LOGNAME=pw->pw_name
1559  *   HOME=pw->pw_dir
1560  *   SHELL=shell
1561  * else does nothing
1562  *
1563  * NB: CHANGEENV and CLEARENV use setenv() - this leaks memory!
1564  * If setup_environment() is used is vforked child, this leaks memory _in parent too_!
1565  */
1566 #define SETUP_ENV_CHANGEENV (1 << 0)
1567 #define SETUP_ENV_CLEARENV  (1 << 1)
1568 #define SETUP_ENV_TO_TMP    (1 << 2)
1569 #define SETUP_ENV_NO_CHDIR  (1 << 4)
1570 void setup_environment(const char *shell, int flags, const struct passwd *pw) FAST_FUNC;
1571 void nuke_str(char *str) FAST_FUNC;
1572 #if ENABLE_FEATURE_SECURETTY && !ENABLE_PAM
1573 int is_tty_secure(const char *short_tty) FAST_FUNC;
1574 #else
1575 static ALWAYS_INLINE int is_tty_secure(const char *short_tty UNUSED_PARAM) { return 1; }
1576 #endif
1577 #define CHECKPASS_PW_HAS_EMPTY_PASSWORD 2
1578 int check_password(const struct passwd *pw, const char *plaintext) FAST_FUNC;
1579 int ask_and_check_password_extended(const struct passwd *pw, int timeout, const char *prompt) FAST_FUNC;
1580 int ask_and_check_password(const struct passwd *pw) FAST_FUNC;
1581 /* Returns a malloced string */
1582 #if !ENABLE_USE_BB_CRYPT
1583 #define pw_encrypt(clear, salt, cleanup) pw_encrypt(clear, salt)
1584 #endif
1585 extern char *pw_encrypt(const char *clear, const char *salt, int cleanup) FAST_FUNC;
1586 extern int obscure(const char *old, const char *newval, const struct passwd *pwdp) FAST_FUNC;
1587 /*
1588  * rnd is additional random input. New one is returned.
1589  * Useful if you call crypt_make_salt many times in a row:
1590  * rnd = crypt_make_salt(buf1, 4, 0);
1591  * rnd = crypt_make_salt(buf2, 4, rnd);
1592  * rnd = crypt_make_salt(buf3, 4, rnd);
1593  * (otherwise we risk having same salt generated)
1594  */
1595 extern int crypt_make_salt(char *p, int cnt /*, int rnd*/) FAST_FUNC;
1596 /* "$N$" + sha_salt_16_bytes + NUL */
1597 #define MAX_PW_SALT_LEN (3 + 16 + 1)
1598 extern char* crypt_make_pw_salt(char p[MAX_PW_SALT_LEN], const char *algo) FAST_FUNC;
1599
1600
1601 /* Returns number of lines changed, or -1 on error */
1602 #if !(ENABLE_FEATURE_ADDUSER_TO_GROUP || ENABLE_FEATURE_DEL_USER_FROM_GROUP)
1603 #define update_passwd(filename, username, data, member) \
1604         update_passwd(filename, username, data)
1605 #endif
1606 extern int update_passwd(const char *filename,
1607                 const char *username,
1608                 const char *data,
1609                 const char *member) FAST_FUNC;
1610
1611 int index_in_str_array(const char *const string_array[], const char *key) FAST_FUNC;
1612 int index_in_strings(const char *strings, const char *key) FAST_FUNC;
1613 int index_in_substr_array(const char *const string_array[], const char *key) FAST_FUNC;
1614 int index_in_substrings(const char *strings, const char *key) FAST_FUNC;
1615 const char *nth_string(const char *strings, int n) FAST_FUNC;
1616
1617 extern void print_login_issue(const char *issue_file, const char *tty) FAST_FUNC;
1618 extern void print_login_prompt(void) FAST_FUNC;
1619
1620 char *xmalloc_ttyname(int fd) FAST_FUNC RETURNS_MALLOC;
1621 /* NB: typically you want to pass fd 0, not 1. Think 'applet | grep something' */
1622 int get_terminal_width_height(int fd, unsigned *width, unsigned *height) FAST_FUNC;
1623 int get_terminal_width(int fd) FAST_FUNC;
1624
1625 int tcsetattr_stdin_TCSANOW(const struct termios *tp) FAST_FUNC;
1626 #define TERMIOS_CLEAR_ISIG      (1 << 0)
1627 #define TERMIOS_RAW_CRNL_INPUT  (1 << 1)
1628 #define TERMIOS_RAW_CRNL_OUTPUT (1 << 2)
1629 #define TERMIOS_RAW_CRNL        (TERMIOS_RAW_CRNL_INPUT|TERMIOS_RAW_CRNL_OUTPUT)
1630 #define TERMIOS_RAW_INPUT       (1 << 3)
1631 int get_termios_and_make_raw(int fd, struct termios *newterm, struct termios *oldterm, int flags) FAST_FUNC;
1632 int set_termios_to_raw(int fd, struct termios *oldterm, int flags) FAST_FUNC;
1633
1634 /* NB: "unsigned request" is crucial! "int request" will break some arches! */
1635 int ioctl_or_perror(int fd, unsigned request, void *argp, const char *fmt,...) __attribute__ ((format (printf, 4, 5))) FAST_FUNC;
1636 int ioctl_or_perror_and_die(int fd, unsigned request, void *argp, const char *fmt,...) __attribute__ ((format (printf, 4, 5))) FAST_FUNC;
1637 #if ENABLE_IOCTL_HEX2STR_ERROR
1638 int bb_ioctl_or_warn(int fd, unsigned request, void *argp, const char *ioctl_name) FAST_FUNC;
1639 int bb_xioctl(int fd, unsigned request, void *argp, const char *ioctl_name) FAST_FUNC;
1640 #define ioctl_or_warn(fd,request,argp) bb_ioctl_or_warn(fd,request,argp,#request)
1641 #define xioctl(fd,request,argp)        bb_xioctl(fd,request,argp,#request)
1642 #else
1643 int bb_ioctl_or_warn(int fd, unsigned request, void *argp) FAST_FUNC;
1644 int bb_xioctl(int fd, unsigned request, void *argp) FAST_FUNC;
1645 #define ioctl_or_warn(fd,request,argp) bb_ioctl_or_warn(fd,request,argp)
1646 #define xioctl(fd,request,argp)        bb_xioctl(fd,request,argp)
1647 #endif
1648
1649 char *is_in_ino_dev_hashtable(const struct stat *statbuf) FAST_FUNC;
1650 void add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name) FAST_FUNC;
1651 void reset_ino_dev_hashtable(void) FAST_FUNC;
1652 #ifdef __GLIBC__
1653 /* At least glibc has horrendously large inline for this, so wrap it */
1654 unsigned long long bb_makedev(unsigned major, unsigned minor) FAST_FUNC;
1655 #undef makedev
1656 #define makedev(a,b) bb_makedev(a,b)
1657 #endif
1658
1659
1660 /* "Keycodes" that report an escape sequence.
1661  * We use something which fits into signed char,
1662  * yet doesn't represent any valid Unicode character.
1663  * Also, -1 is reserved for error indication and we don't use it. */
1664 enum {
1665         KEYCODE_UP        =  -2,
1666         KEYCODE_DOWN      =  -3,
1667         KEYCODE_RIGHT     =  -4,
1668         KEYCODE_LEFT      =  -5,
1669         KEYCODE_HOME      =  -6,
1670         KEYCODE_END       =  -7,
1671         KEYCODE_INSERT    =  -8,
1672         KEYCODE_DELETE    =  -9,
1673         KEYCODE_PAGEUP    = -10,
1674         KEYCODE_PAGEDOWN  = -11,
1675         KEYCODE_BACKSPACE = -12, /* Used only if Alt/Ctrl/Shifted */
1676         KEYCODE_D         = -13, /* Used only if Alted */
1677 #if 0
1678         KEYCODE_FUN1      = ,
1679         KEYCODE_FUN2      = ,
1680         KEYCODE_FUN3      = ,
1681         KEYCODE_FUN4      = ,
1682         KEYCODE_FUN5      = ,
1683         KEYCODE_FUN6      = ,
1684         KEYCODE_FUN7      = ,
1685         KEYCODE_FUN8      = ,
1686         KEYCODE_FUN9      = ,
1687         KEYCODE_FUN10     = ,
1688         KEYCODE_FUN11     = ,
1689         KEYCODE_FUN12     = ,
1690 #endif
1691         /* ^^^^^ Be sure that last defined value is small enough.
1692          * Current read_key() code allows going up to -32 (0xfff..fffe0).
1693          * This gives three upper bits in LSB to play with:
1694          * KEYCODE_foo values are 0xfff..fffXX, lowest XX bits are: scavvvvv,
1695          * s=0 if SHIFT, c=0 if CTRL, a=0 if ALT,
1696          * vvvvv bits are the same for same key regardless of "shift bits".
1697          */
1698         //KEYCODE_SHIFT_...   = KEYCODE_...   & ~0x80,
1699         KEYCODE_CTRL_RIGHT    = KEYCODE_RIGHT & ~0x40,
1700         KEYCODE_CTRL_LEFT     = KEYCODE_LEFT  & ~0x40,
1701         KEYCODE_ALT_RIGHT     = KEYCODE_RIGHT & ~0x20,
1702         KEYCODE_ALT_LEFT      = KEYCODE_LEFT  & ~0x20,
1703         KEYCODE_ALT_BACKSPACE = KEYCODE_BACKSPACE & ~0x20,
1704         KEYCODE_ALT_D         = KEYCODE_D     & ~0x20,
1705
1706         KEYCODE_CURSOR_POS = -0x100, /* 0xfff..fff00 */
1707         /* How long is the longest ESC sequence we know?
1708          * We want it big enough to be able to contain
1709          * cursor position sequence "ESC [ 9999 ; 9999 R"
1710          */
1711         KEYCODE_BUFFER_SIZE = 16
1712 };
1713 /* Note: fd may be in blocking or non-blocking mode, both make sense.
1714  * For one, less uses non-blocking mode.
1715  * Only the first read syscall inside read_key may block indefinitely
1716  * (unless fd is in non-blocking mode),
1717  * subsequent reads will time out after a few milliseconds.
1718  * Return of -1 means EOF or error (errno == 0 on EOF).
1719  * buffer[0] is used as a counter of buffered chars and must be 0
1720  * on first call.
1721  * timeout:
1722  * -2: do not poll(-1) for input - read() it, return on EAGAIN at once
1723  * -1: poll(-1) (i.e. block even on NONBLOCKed fd)
1724  * >=0: poll() for TIMEOUT milliseconds, return -1/EAGAIN on timeout
1725  */
1726 int64_t read_key(int fd, char *buffer, int timeout) FAST_FUNC;
1727 void read_key_ungets(char *buffer, const char *str, unsigned len) FAST_FUNC;
1728
1729
1730 #if ENABLE_FEATURE_EDITING
1731 /* It's NOT just ENABLEd or disabled. It's a number: */
1732 # if defined CONFIG_FEATURE_EDITING_HISTORY && CONFIG_FEATURE_EDITING_HISTORY > 0
1733 #  define MAX_HISTORY (CONFIG_FEATURE_EDITING_HISTORY + 0)
1734 unsigned size_from_HISTFILESIZE(const char *hp) FAST_FUNC;
1735 # else
1736 #  define MAX_HISTORY 0
1737 # endif
1738 typedef struct line_input_t {
1739         int flags;
1740         int timeout;
1741         const char *path_lookup;
1742 # if MAX_HISTORY
1743         int cnt_history;
1744         int cur_history;
1745         int max_history; /* must never be <= 0 */
1746 #  if ENABLE_FEATURE_EDITING_SAVEHISTORY
1747         /* meaning of this field depends on FEATURE_EDITING_SAVE_ON_EXIT:
1748          * if !FEATURE_EDITING_SAVE_ON_EXIT: "how many lines are
1749          * in on-disk history"
1750          * if FEATURE_EDITING_SAVE_ON_EXIT: "how many in-memory lines are
1751          * also in on-disk history (and thus need to be skipped on save)"
1752          */
1753         unsigned cnt_history_in_file;
1754         const char *hist_file;
1755 #  endif
1756         char *history[MAX_HISTORY + 1];
1757 # endif
1758 } line_input_t;
1759 enum {
1760         DO_HISTORY       = 1 * (MAX_HISTORY > 0),
1761         TAB_COMPLETION   = 2 * ENABLE_FEATURE_TAB_COMPLETION,
1762         USERNAME_COMPLETION = 4 * ENABLE_FEATURE_USERNAME_COMPLETION,
1763         VI_MODE          = 8 * ENABLE_FEATURE_EDITING_VI,
1764         WITH_PATH_LOOKUP = 0x10,
1765         FOR_SHELL        = DO_HISTORY | TAB_COMPLETION | USERNAME_COMPLETION,
1766 };
1767 line_input_t *new_line_input_t(int flags) FAST_FUNC;
1768 /* So far static: void free_line_input_t(line_input_t *n) FAST_FUNC; */
1769 /*
1770  * maxsize must be >= 2.
1771  * Returns:
1772  * -1 on read errors or EOF, or on bare Ctrl-D,
1773  * 0  on ctrl-C (the line entered is still returned in 'command'),
1774  * >0 length of input string, including terminating '\n'
1775  */
1776 int read_line_input(line_input_t *st, const char *prompt, char *command, int maxsize) FAST_FUNC;
1777 void show_history(const line_input_t *st) FAST_FUNC;
1778 # if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
1779 void save_history(line_input_t *st);
1780 # endif
1781 #else
1782 #define MAX_HISTORY 0
1783 int read_line_input(const char* prompt, char* command, int maxsize) FAST_FUNC;
1784 #define read_line_input(state, prompt, command, maxsize) \
1785         read_line_input(prompt, command, maxsize)
1786 #endif
1787
1788
1789 #ifndef COMM_LEN
1790 # ifdef TASK_COMM_LEN
1791 enum { COMM_LEN = TASK_COMM_LEN };
1792 # else
1793 /* synchronize with sizeof(task_struct.comm) in /usr/include/linux/sched.h */
1794 enum { COMM_LEN = 16 };
1795 # endif
1796 #endif
1797
1798 struct smaprec {
1799         unsigned long mapped_rw;
1800         unsigned long mapped_ro;
1801         unsigned long shared_clean;
1802         unsigned long shared_dirty;
1803         unsigned long private_clean;
1804         unsigned long private_dirty;
1805         unsigned long stack;
1806         unsigned long smap_pss, smap_swap;
1807         unsigned long smap_size;
1808         unsigned long smap_start;
1809         char smap_mode[5];
1810         char *smap_name;
1811 };
1812
1813 #if !ENABLE_PMAP
1814 #define procps_read_smaps(pid, total, cb, data) \
1815         procps_read_smaps(pid, total)
1816 #endif
1817 int FAST_FUNC procps_read_smaps(pid_t pid, struct smaprec *total,
1818                 void (*cb)(struct smaprec *, void *), void *data);
1819
1820 typedef struct procps_status_t {
1821         DIR *dir;
1822         IF_FEATURE_SHOW_THREADS(DIR *task_dir;)
1823         uint8_t shift_pages_to_bytes;
1824         uint8_t shift_pages_to_kb;
1825 /* Fields are set to 0/NULL if failed to determine (or not requested) */
1826         uint16_t argv_len;
1827         char *argv0;
1828         char *exe;
1829         IF_SELINUX(char *context;)
1830         IF_FEATURE_SHOW_THREADS(unsigned main_thread_pid;)
1831         /* Everything below must contain no ptrs to malloc'ed data:
1832          * it is memset(0) for each process in procps_scan() */
1833         unsigned long vsz, rss; /* we round it to kbytes */
1834         unsigned long stime, utime;
1835         unsigned long start_time;
1836         unsigned pid;
1837         unsigned ppid;
1838         unsigned pgid;
1839         unsigned sid;
1840         unsigned uid;
1841         unsigned gid;
1842 #if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
1843         unsigned ruid;
1844         unsigned rgid;
1845         int niceness;
1846 #endif
1847         unsigned tty_major,tty_minor;
1848 #if ENABLE_FEATURE_TOPMEM
1849         struct smaprec smaps;
1850 #endif
1851         char state[4];
1852         /* basename of executable in exec(2), read from /proc/N/stat
1853          * (if executable is symlink or script, it is NOT replaced
1854          * by link target or interpreter name) */
1855         char comm[COMM_LEN];
1856         /* user/group? - use passwd/group parsing functions */
1857 #if ENABLE_FEATURE_TOP_SMP_PROCESS
1858         int last_seen_on_cpu;
1859 #endif
1860 } procps_status_t;
1861 /* flag bits for procps_scan(xx, flags) calls */
1862 enum {
1863         PSSCAN_PID      = 1 << 0,
1864         PSSCAN_PPID     = 1 << 1,
1865         PSSCAN_PGID     = 1 << 2,
1866         PSSCAN_SID      = 1 << 3,
1867         PSSCAN_UIDGID   = 1 << 4,
1868         PSSCAN_COMM     = 1 << 5,
1869         /* PSSCAN_CMD      = 1 << 6, - use read_cmdline instead */
1870         PSSCAN_ARGV0    = 1 << 7,
1871         PSSCAN_EXE      = 1 << 8,
1872         PSSCAN_STATE    = 1 << 9,
1873         PSSCAN_VSZ      = 1 << 10,
1874         PSSCAN_RSS      = 1 << 11,
1875         PSSCAN_STIME    = 1 << 12,
1876         PSSCAN_UTIME    = 1 << 13,
1877         PSSCAN_TTY      = 1 << 14,
1878         PSSCAN_SMAPS    = (1 << 15) * ENABLE_FEATURE_TOPMEM,
1879         /* NB: used by find_pid_by_name(). Any applet using it
1880          * needs to be mentioned here. */
1881         PSSCAN_ARGVN    = (1 << 16) * (ENABLE_KILLALL
1882                                 || ENABLE_PGREP || ENABLE_PKILL
1883                                 || ENABLE_PIDOF
1884                                 || ENABLE_SESTATUS
1885                                 ),
1886         PSSCAN_CONTEXT  = (1 << 17) * ENABLE_SELINUX,
1887         PSSCAN_START_TIME = 1 << 18,
1888         PSSCAN_CPU      = (1 << 19) * ENABLE_FEATURE_TOP_SMP_PROCESS,
1889         PSSCAN_NICE     = (1 << 20) * ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS,
1890         PSSCAN_RUIDGID  = (1 << 21) * ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS,
1891         PSSCAN_TASKS    = (1 << 22) * ENABLE_FEATURE_SHOW_THREADS,
1892 };
1893 //procps_status_t* alloc_procps_scan(void) FAST_FUNC;
1894 void free_procps_scan(procps_status_t* sp) FAST_FUNC;
1895 procps_status_t* procps_scan(procps_status_t* sp, int flags) FAST_FUNC;
1896 /* Format cmdline (up to col chars) into char buf[size] */
1897 /* Puts [comm] if cmdline is empty (-> process is a kernel thread) */
1898 void read_cmdline(char *buf, int size, unsigned pid, const char *comm) FAST_FUNC;
1899 pid_t *find_pid_by_name(const char* procName) FAST_FUNC;
1900 pid_t *pidlist_reverse(pid_t *pidList) FAST_FUNC;
1901 int starts_with_cpu(const char *str) FAST_FUNC;
1902 unsigned get_cpu_count(void) FAST_FUNC;
1903
1904
1905 /* Use strict=1 if you process input from untrusted source:
1906  * it will return NULL on invalid %xx (bad hex chars)
1907  * and str + 1 if decoded char is / or NUL.
1908  * In non-strict mode, it always succeeds (returns str),
1909  * and also it additionally decoded '+' to space.
1910  */
1911 char *percent_decode_in_place(char *str, int strict) FAST_FUNC;
1912
1913
1914 extern const char bb_uuenc_tbl_base64[] ALIGN1;
1915 extern const char bb_uuenc_tbl_std[] ALIGN1;
1916 void bb_uuencode(char *store, const void *s, int length, const char *tbl) FAST_FUNC;
1917 enum {
1918         BASE64_FLAG_UU_STOP = 0x100,
1919         /* Sign-extends to a value which never matches fgetc result: */
1920         BASE64_FLAG_NO_STOP_CHAR = 0x80,
1921 };
1922 const char *decode_base64(char **pp_dst, const char *src) FAST_FUNC;
1923 void read_base64(FILE *src_stream, FILE *dst_stream, int flags) FAST_FUNC;
1924
1925 typedef struct md5_ctx_t {
1926         uint8_t wbuffer[64]; /* always correctly aligned for uint64_t */
1927         void (*process_block)(struct md5_ctx_t*) FAST_FUNC;
1928         uint64_t total64;    /* must be directly before hash[] */
1929         uint32_t hash[8];    /* 4 elements for md5, 5 for sha1, 8 for sha256 */
1930 } md5_ctx_t;
1931 typedef struct md5_ctx_t sha1_ctx_t;
1932 typedef struct md5_ctx_t sha256_ctx_t;
1933 typedef struct sha512_ctx_t {
1934         uint64_t total64[2];  /* must be directly before hash[] */
1935         uint64_t hash[8];
1936         uint8_t wbuffer[128]; /* always correctly aligned for uint64_t */
1937 } sha512_ctx_t;
1938 typedef struct sha3_ctx_t {
1939         uint64_t state[25];
1940         unsigned bytes_queued;
1941         unsigned input_block_bytes;
1942 } sha3_ctx_t;
1943 void md5_begin(md5_ctx_t *ctx) FAST_FUNC;
1944 void md5_hash(md5_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1945 unsigned md5_end(md5_ctx_t *ctx, void *resbuf) FAST_FUNC;
1946 void sha1_begin(sha1_ctx_t *ctx) FAST_FUNC;
1947 #define sha1_hash md5_hash
1948 unsigned sha1_end(sha1_ctx_t *ctx, void *resbuf) FAST_FUNC;
1949 void sha256_begin(sha256_ctx_t *ctx) FAST_FUNC;
1950 #define sha256_hash md5_hash
1951 #define sha256_end  sha1_end
1952 void sha512_begin(sha512_ctx_t *ctx) FAST_FUNC;
1953 void sha512_hash(sha512_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1954 unsigned sha512_end(sha512_ctx_t *ctx, void *resbuf) FAST_FUNC;
1955 void sha3_begin(sha3_ctx_t *ctx) FAST_FUNC;
1956 void sha3_hash(sha3_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1957 unsigned sha3_end(sha3_ctx_t *ctx, void *resbuf) FAST_FUNC;
1958 /* TLS benefits from knowing that sha1 and sha256 share these. Give them "agnostic" names too */
1959 typedef struct md5_ctx_t md5sha_ctx_t;
1960 #define md5sha_hash md5_hash
1961 #define sha_end sha1_end
1962
1963 extern uint32_t *global_crc32_table;
1964 uint32_t *crc32_filltable(uint32_t *tbl256, int endian) FAST_FUNC;
1965 uint32_t *crc32_new_table_le(void) FAST_FUNC;
1966 uint32_t *global_crc32_new_table_le(void) FAST_FUNC;
1967 uint32_t crc32_block_endian1(uint32_t val, const void *buf, unsigned len, uint32_t *crc_table) FAST_FUNC;
1968 uint32_t crc32_block_endian0(uint32_t val, const void *buf, unsigned len, uint32_t *crc_table) FAST_FUNC;
1969
1970 typedef struct masks_labels_t {
1971         const char *labels;
1972         const int masks[];
1973 } masks_labels_t;
1974 int print_flags_separated(const int *masks, const char *labels,
1975                 int flags, const char *separator) FAST_FUNC;
1976 int print_flags(const masks_labels_t *ml, int flags) FAST_FUNC;
1977
1978 typedef struct bb_progress_t {
1979         unsigned last_size;
1980         unsigned last_update_sec;
1981         unsigned last_change_sec;
1982         unsigned start_sec;
1983         /*unsigned last_eta;*/
1984         const char *curfile;
1985 } bb_progress_t;
1986
1987 #define is_bb_progress_inited(p) ((p)->curfile != NULL)
1988 #define bb_progress_free(p) do { \
1989         if (ENABLE_UNICODE_SUPPORT) free((char*)((p)->curfile)); \
1990         (p)->curfile = NULL; \
1991 } while (0)
1992 void bb_progress_init(bb_progress_t *p, const char *curfile) FAST_FUNC;
1993 void bb_progress_update(bb_progress_t *p,
1994                         uoff_t beg_range,
1995                         uoff_t transferred,
1996                         uoff_t totalsize) FAST_FUNC;
1997
1998 unsigned ubi_devnum_from_devname(const char *str) FAST_FUNC;
1999 int ubi_get_volid_by_name(unsigned ubi_devnum, const char *vol_name) FAST_FUNC;
2000
2001
2002 extern const char *applet_name;
2003
2004 /* Some older linkers don't perform string merging, we used to have common strings
2005  * as global arrays to do it by hand. But:
2006  * (1) newer linkers do it themselves,
2007  * (2) however, they DONT merge string constants with global arrays,
2008  * even if the value is the same (!). Thus global arrays actually
2009  * increased size a bit: for example, "/etc/passwd" string from libc
2010  * wasn't merged with bb_path_passwd_file[] array!
2011  * Therefore now we use #defines.
2012  */
2013 /* "BusyBox vN.N.N (timestamp or extra_version)" */
2014 extern const char bb_banner[] ALIGN1;
2015 extern const char bb_msg_memory_exhausted[] ALIGN1;
2016 extern const char bb_msg_invalid_date[] ALIGN1;
2017 #define bb_msg_read_error "read error"
2018 #define bb_msg_write_error "write error"
2019 extern const char bb_msg_unknown[] ALIGN1;
2020 extern const char bb_msg_can_not_create_raw_socket[] ALIGN1;
2021 extern const char bb_msg_perm_denied_are_you_root[] ALIGN1;
2022 extern const char bb_msg_you_must_be_root[] ALIGN1;
2023 extern const char bb_msg_requires_arg[] ALIGN1;
2024 extern const char bb_msg_invalid_arg_to[] ALIGN1;
2025 extern const char bb_msg_standard_input[] ALIGN1;
2026 extern const char bb_msg_standard_output[] ALIGN1;
2027
2028 /* NB: (bb_hexdigits_upcase[i] | 0x20) -> lowercase hex digit */
2029 extern const char bb_hexdigits_upcase[] ALIGN1;
2030
2031 extern const char bb_path_wtmp_file[] ALIGN1;
2032
2033 /* Busybox mount uses either /proc/mounts or /etc/mtab to
2034  * get the list of currently mounted filesystems */
2035 #define bb_path_mtab_file IF_FEATURE_MTAB_SUPPORT("/etc/mtab")IF_NOT_FEATURE_MTAB_SUPPORT("/proc/mounts")
2036
2037 #define bb_path_passwd_file  _PATH_PASSWD
2038 #define bb_path_group_file   _PATH_GROUP
2039 #define bb_path_shadow_file  _PATH_SHADOW
2040 #define bb_path_gshadow_file _PATH_GSHADOW
2041
2042 #define bb_path_motd_file "/etc/motd"
2043
2044 #define bb_dev_null "/dev/null"
2045 extern const char bb_busybox_exec_path[] ALIGN1;
2046 /* allow default system PATH to be extended via CFLAGS */
2047 #ifndef BB_ADDITIONAL_PATH
2048 #define BB_ADDITIONAL_PATH ""
2049 #endif
2050 #define BB_PATH_ROOT_PATH "PATH=/sbin:/usr/sbin:/bin:/usr/bin" BB_ADDITIONAL_PATH
2051 extern const char bb_PATH_root_path[] ALIGN1; /* BB_PATH_ROOT_PATH */
2052 #define bb_default_root_path (bb_PATH_root_path + sizeof("PATH"))
2053 /* util-linux manpage says /sbin:/bin:/usr/sbin:/usr/bin,
2054  * but I want to save a few bytes here:
2055  */
2056 #define bb_default_path      (bb_PATH_root_path + sizeof("PATH=/sbin:/usr/sbin"))
2057
2058 extern const int const_int_0;
2059 //extern const int const_int_1;
2060
2061 /* This struct is deliberately not defined. */
2062 /* See docs/keep_data_small.txt */
2063 struct globals;
2064 /* '*const' ptr makes gcc optimize code much better.
2065  * Magic prevents ptr_to_globals from going into rodata.
2066  * If you want to assign a value, use SET_PTR_TO_GLOBALS(x) */
2067 extern struct globals *const ptr_to_globals;
2068 /* At least gcc 3.4.6 on mipsel system needs optimization barrier */
2069 #define barrier() __asm__ __volatile__("":::"memory")
2070 #define SET_PTR_TO_GLOBALS(x) do { \
2071         (*(struct globals**)&ptr_to_globals) = (void*)(x); \
2072         barrier(); \
2073 } while (0)
2074 #define FREE_PTR_TO_GLOBALS() do { \
2075         if (ENABLE_FEATURE_CLEAN_UP) { \
2076                 free(ptr_to_globals); \
2077         } \
2078 } while (0)
2079
2080 /* You can change LIBBB_DEFAULT_LOGIN_SHELL, but don't use it,
2081  * use bb_default_login_shell and following defines.
2082  * If you change LIBBB_DEFAULT_LOGIN_SHELL,
2083  * don't forget to change increment constant. */
2084 #define LIBBB_DEFAULT_LOGIN_SHELL  "-/bin/sh"
2085 extern const char bb_default_login_shell[] ALIGN1;
2086 /* "/bin/sh" */
2087 #define DEFAULT_SHELL              (bb_default_login_shell+1)
2088 /* "sh" */
2089 #define DEFAULT_SHELL_SHORT_NAME   (bb_default_login_shell+6)
2090
2091 /* The following devices are the same on all systems.  */
2092 #define CURRENT_TTY "/dev/tty"
2093 #define DEV_CONSOLE "/dev/console"
2094
2095 #if defined(__FreeBSD_kernel__)
2096 # define CURRENT_VC CURRENT_TTY
2097 # define VC_1 "/dev/ttyv0"
2098 # define VC_2 "/dev/ttyv1"
2099 # define VC_3 "/dev/ttyv2"
2100 # define VC_4 "/dev/ttyv3"
2101 # define VC_5 "/dev/ttyv4"
2102 # define VC_FORMAT "/dev/ttyv%d"
2103 #elif defined(__GNU__)
2104 # define CURRENT_VC CURRENT_TTY
2105 # define VC_1 "/dev/tty1"
2106 # define VC_2 "/dev/tty2"
2107 # define VC_3 "/dev/tty3"
2108 # define VC_4 "/dev/tty4"
2109 # define VC_5 "/dev/tty5"
2110 # define VC_FORMAT "/dev/tty%d"
2111 #elif ENABLE_FEATURE_DEVFS
2112 /*Linux, obsolete devfs names */
2113 # define CURRENT_VC "/dev/vc/0"
2114 # define VC_1 "/dev/vc/1"
2115 # define VC_2 "/dev/vc/2"
2116 # define VC_3 "/dev/vc/3"
2117 # define VC_4 "/dev/vc/4"
2118 # define VC_5 "/dev/vc/5"
2119 # define VC_FORMAT "/dev/vc/%d"
2120 # define LOOP_FORMAT "/dev/loop/%u"
2121 # define LOOP_NAMESIZE (sizeof("/dev/loop/") + sizeof(int)*3 + 1)
2122 # define LOOP_NAME "/dev/loop/"
2123 # define FB_0 "/dev/fb/0"
2124 #else
2125 /*Linux, normal names */
2126 # define CURRENT_VC "/dev/tty0"
2127 # define VC_1 "/dev/tty1"
2128 # define VC_2 "/dev/tty2"
2129 # define VC_3 "/dev/tty3"
2130 # define VC_4 "/dev/tty4"
2131 # define VC_5 "/dev/tty5"
2132 # define VC_FORMAT "/dev/tty%d"
2133 # define LOOP_FORMAT "/dev/loop%u"
2134 # define LOOP_NAMESIZE (sizeof("/dev/loop") + sizeof(int)*3 + 1)
2135 # define LOOP_NAME "/dev/loop"
2136 # define FB_0 "/dev/fb0"
2137 #endif
2138
2139 // storage helpers for mk*fs utilities
2140 char BUG_wrong_field_size(void);
2141 #define STORE_LE(field, value) \
2142 do { \
2143         if (sizeof(field) == 4) \
2144                 field = SWAP_LE32((uint32_t)(value)); \
2145         else if (sizeof(field) == 2) \
2146                 field = SWAP_LE16((uint16_t)(value)); \
2147         else if (sizeof(field) == 1) \
2148                 field = (uint8_t)(value); \
2149         else \
2150                 BUG_wrong_field_size(); \
2151 } while (0)
2152
2153 #define FETCH_LE32(field) \
2154         (sizeof(field) == 4 ? SWAP_LE32(field) : BUG_wrong_field_size())
2155
2156
2157 #define ARRAY_SIZE(x) ((unsigned)(sizeof(x) / sizeof((x)[0])))
2158 #define BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
2159
2160
2161 /* We redefine ctype macros. Unicode-correct handling of char types
2162  * can't be done with such byte-oriented operations anyway,
2163  * we don't lose anything.
2164  */
2165 #undef isalnum
2166 #undef isalpha
2167 #undef isascii
2168 #undef isblank
2169 #undef iscntrl
2170 #undef isdigit
2171 #undef isgraph
2172 #undef islower
2173 #undef isprint
2174 #undef ispunct
2175 #undef isspace
2176 #undef isupper
2177 #undef isxdigit
2178 #undef toupper
2179 #undef tolower
2180
2181 /* We save ~500 bytes on isdigit alone.
2182  * BTW, x86 likes (unsigned char) cast more than (unsigned). */
2183
2184 /* These work the same for ASCII and Unicode,
2185  * assuming no one asks "is this a *Unicode* letter?" using isalpha(letter) */
2186 #define isascii(a) ((unsigned char)(a) <= 0x7f)
2187 #define isdigit(a) ((unsigned char)((a) - '0') <= 9)
2188 #define isupper(a) ((unsigned char)((a) - 'A') <= ('Z' - 'A'))
2189 #define islower(a) ((unsigned char)((a) - 'a') <= ('z' - 'a'))
2190 #define isalpha(a) ((unsigned char)(((a)|0x20) - 'a') <= ('z' - 'a'))
2191 #define isblank(a) ({ unsigned char bb__isblank = (a); bb__isblank == ' ' || bb__isblank == '\t'; })
2192 #define iscntrl(a) ({ unsigned char bb__iscntrl = (a); bb__iscntrl < ' ' || bb__iscntrl == 0x7f; })
2193 /* In POSIX/C locale isspace is only these chars: "\t\n\v\f\r" and space.
2194  * "\t\n\v\f\r" happen to have ASCII codes 9,10,11,12,13.
2195  */
2196 #define isspace(a) ({ unsigned char bb__isspace = (a) - 9; bb__isspace == (' ' - 9) || bb__isspace <= (13 - 9); })
2197 // Unsafe wrt NUL: #define ispunct(a) (strchr("!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~", (a)) != NULL)
2198 #define ispunct(a) (strchrnul("!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~", (a))[0])
2199 // Bigger code: #define isalnum(a) ({ unsigned char bb__isalnum = (a) - '0'; bb__isalnum <= 9 || ((bb__isalnum - ('A' - '0')) & 0xdf) <= 25; })
2200 #define isalnum(a) bb_ascii_isalnum(a)
2201 static ALWAYS_INLINE int bb_ascii_isalnum(unsigned char a)
2202 {
2203         unsigned char b = a - '0';
2204         if (b <= 9)
2205                 return (b <= 9);
2206         b = (a|0x20) - 'a';
2207         return b <= 'z' - 'a';
2208 }
2209 #define isxdigit(a) bb_ascii_isxdigit(a)
2210 static ALWAYS_INLINE int bb_ascii_isxdigit(unsigned char a)
2211 {
2212         unsigned char b = a - '0';
2213         if (b <= 9)
2214                 return (b <= 9);
2215         b = (a|0x20) - 'a';
2216         return b <= 'f' - 'a';
2217 }
2218 #define toupper(a) bb_ascii_toupper(a)
2219 static ALWAYS_INLINE unsigned char bb_ascii_toupper(unsigned char a)
2220 {
2221         unsigned char b = a - 'a';
2222         if (b <= ('z' - 'a'))
2223                 a -= 'a' - 'A';
2224         return a;
2225 }
2226 #define tolower(a) bb_ascii_tolower(a)
2227 static ALWAYS_INLINE unsigned char bb_ascii_tolower(unsigned char a)
2228 {
2229         unsigned char b = a - 'A';
2230         if (b <= ('Z' - 'A'))
2231                 a += 'a' - 'A';
2232         return a;
2233 }
2234
2235 /* In ASCII and Unicode, these are likely to be very different.
2236  * Let's prevent ambiguous usage from the start */
2237 #define isgraph(a) isgraph_is_ambiguous_dont_use(a)
2238 #define isprint(a) isprint_is_ambiguous_dont_use(a)
2239 /* NB: must not treat EOF as isgraph or isprint */
2240 #define isgraph_asciionly(a) ((unsigned)((a) - 0x21) <= 0x7e - 0x21)
2241 #define isprint_asciionly(a) ((unsigned)((a) - 0x20) <= 0x7e - 0x20)
2242
2243
2244 /* Simple unit-testing framework */
2245
2246 typedef void (*bbunit_testfunc)(void);
2247
2248 struct bbunit_listelem {
2249         const char* name;
2250         bbunit_testfunc testfunc;
2251 };
2252
2253 void bbunit_registertest(struct bbunit_listelem* test);
2254 void bbunit_settestfailed(void);
2255
2256 #define BBUNIT_DEFINE_TEST(NAME) \
2257         static void bbunit_##NAME##_test(void); \
2258         static struct bbunit_listelem bbunit_##NAME##_elem = { \
2259                 .name = #NAME, \
2260                 .testfunc = bbunit_##NAME##_test, \
2261         }; \
2262         static void INIT_FUNC bbunit_##NAME##_register(void) \
2263         { \
2264                 bbunit_registertest(&bbunit_##NAME##_elem); \
2265         } \
2266         static void bbunit_##NAME##_test(void)
2267
2268 /*
2269  * Both 'goto bbunit_end' and 'break' are here only to get rid
2270  * of compiler warnings.
2271  */
2272 #define BBUNIT_ENDTEST \
2273         do { \
2274                 goto bbunit_end; \
2275         bbunit_end: \
2276                 break; \
2277         } while (0)
2278
2279 #define BBUNIT_PRINTASSERTFAIL \
2280         do { \
2281                 bb_error_msg( \
2282                         "[ERROR] Assertion failed in file %s, line %d", \
2283                         __FILE__, __LINE__); \
2284         } while (0)
2285
2286 #define BBUNIT_ASSERTION_FAILED \
2287         do { \
2288                 bbunit_settestfailed(); \
2289                 goto bbunit_end; \
2290         } while (0)
2291
2292 /*
2293  * Assertions.
2294  * For now we only offer assertions which cause tests to fail
2295  * immediately. In the future 'expects' might be added too -
2296  * similar to those offered by the gtest framework.
2297  */
2298 #define BBUNIT_ASSERT_EQ(EXPECTED, ACTUAL) \
2299         do { \
2300                 if ((EXPECTED) != (ACTUAL)) { \
2301                         BBUNIT_PRINTASSERTFAIL; \
2302                         bb_error_msg("[ERROR] '%s' isn't equal to '%s'", \
2303                                                 #EXPECTED, #ACTUAL); \
2304                         BBUNIT_ASSERTION_FAILED; \
2305                 } \
2306         } while (0)
2307
2308 #define BBUNIT_ASSERT_NOTEQ(EXPECTED, ACTUAL) \
2309         do { \
2310                 if ((EXPECTED) == (ACTUAL)) { \
2311                         BBUNIT_PRINTASSERTFAIL; \
2312                         bb_error_msg("[ERROR] '%s' is equal to '%s'", \
2313                                                 #EXPECTED, #ACTUAL); \
2314                         BBUNIT_ASSERTION_FAILED; \
2315                 } \
2316         } while (0)
2317
2318 #define BBUNIT_ASSERT_NOTNULL(PTR) \
2319         do { \
2320                 if ((PTR) == NULL) { \
2321                         BBUNIT_PRINTASSERTFAIL; \
2322                         bb_error_msg("[ERROR] '%s' is NULL!", #PTR); \
2323                         BBUNIT_ASSERTION_FAILED; \
2324                 } \
2325         } while (0)
2326
2327 #define BBUNIT_ASSERT_NULL(PTR) \
2328         do { \
2329                 if ((PTR) != NULL) { \
2330                         BBUNIT_PRINTASSERTFAIL; \
2331                         bb_error_msg("[ERROR] '%s' is not NULL!", #PTR); \
2332                         BBUNIT_ASSERTION_FAILED; \
2333                 } \
2334         } while (0)
2335
2336 #define BBUNIT_ASSERT_FALSE(STATEMENT) \
2337         do { \
2338                 if ((STATEMENT)) { \
2339                         BBUNIT_PRINTASSERTFAIL; \
2340                         bb_error_msg("[ERROR] Statement '%s' evaluated to true!", \
2341                                                                 #STATEMENT); \
2342                         BBUNIT_ASSERTION_FAILED; \
2343                 } \
2344         } while (0)
2345
2346 #define BBUNIT_ASSERT_TRUE(STATEMENT) \
2347         do { \
2348                 if (!(STATEMENT)) { \
2349                         BBUNIT_PRINTASSERTFAIL; \
2350                         bb_error_msg("[ERROR] Statement '%s' evaluated to false!", \
2351                                         #STATEMENT); \
2352                         BBUNIT_ASSERTION_FAILED; \
2353                 } \
2354         } while (0)
2355
2356 #define BBUNIT_ASSERT_STREQ(STR1, STR2) \
2357         do { \
2358                 if (strcmp(STR1, STR2) != 0) { \
2359                         BBUNIT_PRINTASSERTFAIL; \
2360                         bb_error_msg("[ERROR] Strings '%s' and '%s' " \
2361                                         "are not the same", STR1, STR2); \
2362                         BBUNIT_ASSERTION_FAILED; \
2363                 } \
2364         } while (0)
2365
2366 #define BBUNIT_ASSERT_STRNOTEQ(STR1, STR2) \
2367         do { \
2368                 if (strcmp(STR1, STR2) == 0) { \
2369                         BBUNIT_PRINTASSERTFAIL; \
2370                         bb_error_msg("[ERROR] Strings '%s' and '%s' " \
2371                                         "are the same, but were " \
2372                                         "expected to differ", STR1, STR2); \
2373                         BBUNIT_ASSERTION_FAILED; \
2374                 } \
2375         } while (0)
2376
2377
2378 POP_SAVED_FUNCTION_VISIBILITY
2379
2380 #endif