pmap: make 32-bit version work better on 64-bit kernels
[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 /* Return old set in the same set: */
548 int sigprocmask2(int how, sigset_t *set) FAST_FUNC;
549 /* Standard handler which just records signo */
550 extern smallint bb_got_signal;
551 void record_signo(int signo); /* not FAST_FUNC! */
552
553
554 void xsetgid(gid_t gid) FAST_FUNC;
555 void xsetuid(uid_t uid) FAST_FUNC;
556 void xsetegid(gid_t egid) FAST_FUNC;
557 void xseteuid(uid_t euid) FAST_FUNC;
558 void xchdir(const char *path) FAST_FUNC;
559 void xfchdir(int fd) FAST_FUNC;
560 void xchroot(const char *path) FAST_FUNC;
561 void xsetenv(const char *key, const char *value) FAST_FUNC;
562 void bb_unsetenv(const char *key) FAST_FUNC;
563 void bb_unsetenv_and_free(char *key) FAST_FUNC;
564 void xunlink(const char *pathname) FAST_FUNC;
565 void xstat(const char *pathname, struct stat *buf) FAST_FUNC;
566 void xfstat(int fd, struct stat *buf, const char *errmsg) FAST_FUNC;
567 int open3_or_warn(const char *pathname, int flags, int mode) FAST_FUNC;
568 int open_or_warn(const char *pathname, int flags) FAST_FUNC;
569 int xopen3(const char *pathname, int flags, int mode) FAST_FUNC;
570 int xopen(const char *pathname, int flags) FAST_FUNC;
571 int xopen_nonblocking(const char *pathname) FAST_FUNC;
572 int xopen_as_uid_gid(const char *pathname, int flags, uid_t u, gid_t g) FAST_FUNC;
573 int open_or_warn_stdin(const char *pathname) FAST_FUNC;
574 int xopen_stdin(const char *pathname) FAST_FUNC;
575 void xrename(const char *oldpath, const char *newpath) FAST_FUNC;
576 int rename_or_warn(const char *oldpath, const char *newpath) FAST_FUNC;
577 off_t xlseek(int fd, off_t offset, int whence) FAST_FUNC;
578 int xmkstemp(char *template) FAST_FUNC;
579 off_t fdlength(int fd) FAST_FUNC;
580
581 uoff_t FAST_FUNC get_volume_size_in_bytes(int fd,
582                 const char *override,
583                 unsigned override_units,
584                 int extend);
585
586 void xpipe(int filedes[2]) FAST_FUNC;
587 /* In this form code with pipes is much more readable */
588 struct fd_pair { int rd; int wr; };
589 #define piped_pair(pair)  pipe(&((pair).rd))
590 #define xpiped_pair(pair) xpipe(&((pair).rd))
591
592 /* Useful for having small structure members/global variables */
593 typedef int8_t socktype_t;
594 typedef int8_t family_t;
595 struct BUG_too_small {
596         char BUG_socktype_t_too_small[(0
597                         | SOCK_STREAM
598                         | SOCK_DGRAM
599                         | SOCK_RDM
600                         | SOCK_SEQPACKET
601                         | SOCK_RAW
602                         ) <= 127 ? 1 : -1];
603         char BUG_family_t_too_small[(0
604                         | AF_UNSPEC
605                         | AF_INET
606                         | AF_INET6
607                         | AF_UNIX
608 #ifdef AF_PACKET
609                         | AF_PACKET
610 #endif
611 #ifdef AF_NETLINK
612                         | AF_NETLINK
613 #endif
614                         /* | AF_DECnet */
615                         /* | AF_IPX */
616                         ) <= 127 ? 1 : -1];
617 };
618
619
620 void parse_datestr(const char *date_str, struct tm *ptm) FAST_FUNC;
621 time_t validate_tm_time(const char *date_str, struct tm *ptm) FAST_FUNC;
622 char *strftime_HHMMSS(char *buf, unsigned len, time_t *tp) FAST_FUNC;
623 char *strftime_YYYYMMDDHHMMSS(char *buf, unsigned len, time_t *tp) FAST_FUNC;
624
625 int xsocket(int domain, int type, int protocol) FAST_FUNC;
626 void xbind(int sockfd, struct sockaddr *my_addr, socklen_t addrlen) FAST_FUNC;
627 void xlisten(int s, int backlog) FAST_FUNC;
628 void xconnect(int s, const struct sockaddr *s_addr, socklen_t addrlen) FAST_FUNC;
629 ssize_t xsendto(int s, const void *buf, size_t len, const struct sockaddr *to,
630                                 socklen_t tolen) FAST_FUNC;
631
632 int setsockopt_int(int fd, int level, int optname, int optval) FAST_FUNC;
633 int setsockopt_1(int fd, int level, int optname) FAST_FUNC;
634 int setsockopt_SOL_SOCKET_int(int fd, int optname, int optval) FAST_FUNC;
635 int setsockopt_SOL_SOCKET_1(int fd, int optname) FAST_FUNC;
636 /* SO_REUSEADDR allows a server to rebind to an address that is already
637  * "in use" by old connections to e.g. previous server instance which is
638  * killed or crashed. Without it bind will fail until all such connections
639  * time out. Linux does not allow multiple live binds on same ip:port
640  * regardless of SO_REUSEADDR (unlike some other flavors of Unix).
641  * Turn it on before you call bind(). */
642 void setsockopt_reuseaddr(int fd) FAST_FUNC; /* On Linux this never fails. */
643 int setsockopt_keepalive(int fd) FAST_FUNC;
644 int setsockopt_broadcast(int fd) FAST_FUNC;
645 int setsockopt_bindtodevice(int fd, const char *iface) FAST_FUNC;
646 int bb_getsockname(int sockfd, void *addr, socklen_t addrlen) FAST_FUNC;
647 /* NB: returns port in host byte order */
648 unsigned bb_lookup_port(const char *port, const char *protocol, unsigned default_port) FAST_FUNC;
649 #if ENABLE_FEATURE_ETC_SERVICES
650 # define bb_lookup_std_port(portstr, protocol, portnum) bb_lookup_port(portstr, protocol, portnum)
651 #else
652 # define bb_lookup_std_port(portstr, protocol, portnum) (portnum)
653 #endif
654 typedef struct len_and_sockaddr {
655         socklen_t len;
656         union {
657                 struct sockaddr sa;
658                 struct sockaddr_in sin;
659 #if ENABLE_FEATURE_IPV6
660                 struct sockaddr_in6 sin6;
661 #endif
662         } u;
663 } len_and_sockaddr;
664 enum {
665         LSA_LEN_SIZE = offsetof(len_and_sockaddr, u),
666         LSA_SIZEOF_SA = sizeof(
667                 union {
668                         struct sockaddr sa;
669                         struct sockaddr_in sin;
670 #if ENABLE_FEATURE_IPV6
671                         struct sockaddr_in6 sin6;
672 #endif
673                 }
674         )
675 };
676 /* Create stream socket, and allocate suitable lsa.
677  * (lsa of correct size and lsa->sa.sa_family (AF_INET/AF_INET6))
678  * af == AF_UNSPEC will result in trying to create IPv6 socket,
679  * and if kernel doesn't support it, fall back to IPv4.
680  * This is useful if you plan to bind to resulting local lsa.
681  */
682 int xsocket_type(len_and_sockaddr **lsap, int af, int sock_type) FAST_FUNC;
683 int xsocket_stream(len_and_sockaddr **lsap) FAST_FUNC;
684 /* Create server socket bound to bindaddr:port. bindaddr can be NULL,
685  * numeric IP ("N.N.N.N") or numeric IPv6 address,
686  * and can have ":PORT" suffix (for IPv6 use "[X:X:...:X]:PORT").
687  * Only if there is no suffix, port argument is used */
688 /* NB: these set SO_REUSEADDR before bind */
689 int create_and_bind_stream_or_die(const char *bindaddr, int port) FAST_FUNC;
690 int create_and_bind_dgram_or_die(const char *bindaddr, int port) FAST_FUNC;
691 /* Create client TCP socket connected to peer:port. Peer cannot be NULL.
692  * Peer can be numeric IP ("N.N.N.N"), numeric IPv6 address or hostname,
693  * and can have ":PORT" suffix (for IPv6 use "[X:X:...:X]:PORT").
694  * If there is no suffix, port argument is used */
695 int create_and_connect_stream_or_die(const char *peer, int port) FAST_FUNC;
696 /* Connect to peer identified by lsa */
697 int xconnect_stream(const len_and_sockaddr *lsa) FAST_FUNC;
698 /* Get local address of bound or accepted socket */
699 len_and_sockaddr *get_sock_lsa(int fd) FAST_FUNC RETURNS_MALLOC;
700 /* Get remote address of connected or accepted socket */
701 len_and_sockaddr *get_peer_lsa(int fd) FAST_FUNC RETURNS_MALLOC;
702 /* Return malloc'ed len_and_sockaddr with socket address of host:port
703  * Currently will return IPv4 or IPv6 sockaddrs only
704  * (depending on host), but in theory nothing prevents e.g.
705  * UNIX socket address being returned, IPX sockaddr etc...
706  * On error does bb_error_msg and returns NULL */
707 len_and_sockaddr* host2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
708 /* Version which dies on error */
709 len_and_sockaddr* xhost2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
710 len_and_sockaddr* xdotted2sockaddr(const char *host, int port) FAST_FUNC RETURNS_MALLOC;
711 /* Same, useful if you want to force family (e.g. IPv6) */
712 #if !ENABLE_FEATURE_IPV6
713 #define host_and_af2sockaddr(host, port, af) host2sockaddr((host), (port))
714 #define xhost_and_af2sockaddr(host, port, af) xhost2sockaddr((host), (port))
715 #else
716 len_and_sockaddr* host_and_af2sockaddr(const char *host, int port, sa_family_t af) FAST_FUNC RETURNS_MALLOC;
717 len_and_sockaddr* xhost_and_af2sockaddr(const char *host, int port, sa_family_t af) FAST_FUNC RETURNS_MALLOC;
718 #endif
719 /* Assign sin[6]_port member if the socket is an AF_INET[6] one,
720  * otherwise no-op. Useful for ftp.
721  * NB: does NOT do htons() internally, just direct assignment. */
722 void set_nport(struct sockaddr *sa, unsigned port) FAST_FUNC;
723 /* Retrieve sin[6]_port or return -1 for non-INET[6] lsa's */
724 int get_nport(const struct sockaddr *sa) FAST_FUNC;
725 /* Reverse DNS. Returns NULL on failure. */
726 char* xmalloc_sockaddr2host(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
727 /* This one doesn't append :PORTNUM */
728 char* xmalloc_sockaddr2host_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
729 /* This one also doesn't fall back to dotted IP (returns NULL) */
730 char* xmalloc_sockaddr2hostonly_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
731 /* inet_[ap]ton on steroids */
732 char* xmalloc_sockaddr2dotted(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
733 char* xmalloc_sockaddr2dotted_noport(const struct sockaddr *sa) FAST_FUNC RETURNS_MALLOC;
734 // "old" (ipv4 only) API
735 // users: traceroute.c hostname.c - use _list_ of all IPs
736 struct hostent *xgethostbyname(const char *name) FAST_FUNC;
737 // Also mount.c and inetd.c are using gethostbyname(),
738 // + inet_common.c has additional IPv4-only stuff
739
740
741 struct tls_aes {
742         uint32_t key[60];
743         unsigned rounds;
744 };
745 #define TLS_MAX_MAC_SIZE 32
746 #define TLS_MAX_KEY_SIZE 32
747 #define TLS_MAX_IV_SIZE   4
748 struct tls_handshake_data; /* opaque */
749 typedef struct tls_state {
750         unsigned flags;
751
752         int ofd;
753         int ifd;
754
755         unsigned min_encrypted_len_on_read;
756         uint16_t cipher_id;
757         unsigned MAC_size;
758         unsigned key_size;
759         unsigned IV_size;
760
761         uint8_t *outbuf;
762         int     outbuf_size;
763
764         int     inbuf_size;
765         int     ofs_to_buffered;
766         int     buffered_size;
767         uint8_t *inbuf;
768
769         struct tls_handshake_data *hsd;
770
771         // RFC 5246
772         // sequence number
773         //   Each connection state contains a sequence number, which is
774         //   maintained separately for read and write states.  The sequence
775         //   number MUST be set to zero whenever a connection state is made the
776         //   active state.  Sequence numbers are of type uint64 and may not
777         //   exceed 2^64-1.
778         /*uint64_t read_seq64_be;*/
779         uint64_t write_seq64_be;
780
781         /*uint8_t *server_write_MAC_key;*/
782         uint8_t *client_write_key;
783         uint8_t *server_write_key;
784         uint8_t *client_write_IV;
785         uint8_t *server_write_IV;
786         uint8_t client_write_MAC_key[TLS_MAX_MAC_SIZE];
787         uint8_t server_write_MAC_k__[TLS_MAX_MAC_SIZE];
788         uint8_t client_write_k__[TLS_MAX_KEY_SIZE];
789         uint8_t server_write_k__[TLS_MAX_KEY_SIZE];
790         uint8_t client_write_I_[TLS_MAX_IV_SIZE];
791         uint8_t server_write_I_[TLS_MAX_IV_SIZE];
792
793         struct tls_aes aes_encrypt;
794         struct tls_aes aes_decrypt;
795         uint8_t H[16]; //used by AES_GCM
796 } tls_state_t;
797
798 static inline tls_state_t *new_tls_state(void)
799 {
800         tls_state_t *tls = xzalloc(sizeof(*tls));
801         return tls;
802 }
803 void tls_handshake(tls_state_t *tls, const char *sni) FAST_FUNC;
804 #define TLSLOOP_EXIT_ON_LOCAL_EOF (1 << 0)
805 void tls_run_copy_loop(tls_state_t *tls, unsigned flags) FAST_FUNC;
806
807
808 void socket_want_pktinfo(int fd) FAST_FUNC;
809 ssize_t send_to_from(int fd, void *buf, size_t len, int flags,
810                 const struct sockaddr *to,
811                 const struct sockaddr *from,
812                 socklen_t tolen) FAST_FUNC;
813 ssize_t recv_from_to(int fd, void *buf, size_t len, int flags,
814                 struct sockaddr *from,
815                 struct sockaddr *to,
816                 socklen_t sa_size) FAST_FUNC;
817
818 uint16_t inet_cksum(uint16_t *addr, int len) FAST_FUNC;
819 int parse_pasv_epsv(char *buf) FAST_FUNC;
820
821 /* 0 if argv[0] is NULL: */
822 unsigned string_array_len(char **argv) FAST_FUNC;
823 void overlapping_strcpy(char *dst, const char *src) FAST_FUNC;
824 char *safe_strncpy(char *dst, const char *src, size_t size) FAST_FUNC;
825 char *strncpy_IFNAMSIZ(char *dst, const char *src) FAST_FUNC;
826 unsigned count_strstr(const char *str, const char *sub) FAST_FUNC;
827 char *xmalloc_substitute_string(const char *src, int count, const char *sub, const char *repl) FAST_FUNC;
828 /* Guaranteed to NOT be a macro (smallest code). Saves nearly 2k on uclibc.
829  * But potentially slow, don't use in one-billion-times loops */
830 int bb_putchar(int ch) FAST_FUNC;
831 /* Note: does not use stdio, writes to fd 2 directly */
832 int bb_putchar_stderr(char ch) FAST_FUNC;
833 char *xasprintf(const char *format, ...) __attribute__ ((format(printf, 1, 2))) FAST_FUNC RETURNS_MALLOC;
834 char *auto_string(char *str) FAST_FUNC;
835 // gcc-4.1.1 still isn't good enough at optimizing it
836 // (+200 bytes compared to macro)
837 //static ALWAYS_INLINE
838 //int LONE_DASH(const char *s) { return s[0] == '-' && !s[1]; }
839 //static ALWAYS_INLINE
840 //int NOT_LONE_DASH(const char *s) { return s[0] != '-' || s[1]; }
841 #define LONE_DASH(s)     ((s)[0] == '-' && !(s)[1])
842 #define NOT_LONE_DASH(s) ((s)[0] != '-' || (s)[1])
843 #define LONE_CHAR(s,c)     ((s)[0] == (c) && !(s)[1])
844 #define NOT_LONE_CHAR(s,c) ((s)[0] != (c) || (s)[1])
845 #define DOT_OR_DOTDOT(s) ((s)[0] == '.' && (!(s)[1] || ((s)[1] == '.' && !(s)[2])))
846
847 typedef struct uni_stat_t {
848         unsigned byte_count;
849         unsigned unicode_count;
850         unsigned unicode_width;
851 } uni_stat_t;
852 /* Returns a string with unprintable chars replaced by '?' or
853  * SUBST_WCHAR. This function is unicode-aware. */
854 const char* FAST_FUNC printable_string(const char *str);
855 const char* FAST_FUNC printable_string2(uni_stat_t *stats, const char *str);
856 /* Prints unprintable char ch as ^C or M-c to file
857  * (M-c is used only if ch is ORed with PRINTABLE_META),
858  * else it is printed as-is (except for ch = 0x9b) */
859 enum { PRINTABLE_META = 0x100 };
860 void fputc_printable(int ch, FILE *file) FAST_FUNC;
861 /* Return a string that is the printable representation of character ch.
862  * Buffer must hold at least four characters. */
863 enum {
864         VISIBLE_ENDLINE   = 1 << 0,
865         VISIBLE_SHOW_TABS = 1 << 1,
866 };
867 void visible(unsigned ch, char *buf, int flags) FAST_FUNC;
868
869 extern ssize_t safe_read(int fd, void *buf, size_t count) FAST_FUNC;
870 extern ssize_t nonblock_immune_read(int fd, void *buf, size_t count) FAST_FUNC;
871 // NB: will return short read on error, not -1,
872 // if some data was read before error occurred
873 extern ssize_t full_read(int fd, void *buf, size_t count) FAST_FUNC;
874 extern void xread(int fd, void *buf, size_t count) FAST_FUNC;
875 extern unsigned char xread_char(int fd) FAST_FUNC;
876 extern ssize_t read_close(int fd, void *buf, size_t maxsz) FAST_FUNC;
877 extern ssize_t open_read_close(const char *filename, void *buf, size_t maxsz) FAST_FUNC;
878 // Reads one line a-la fgets (but doesn't save terminating '\n').
879 // Reads byte-by-byte. Useful when it is important to not read ahead.
880 // Bytes are appended to pfx (which must be malloced, or NULL).
881 extern char *xmalloc_reads(int fd, size_t *maxsz_p) FAST_FUNC;
882 /* Reads block up to *maxsz_p (default: INT_MAX - 4095) */
883 extern void *xmalloc_read(int fd, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
884 /* Returns NULL if file can't be opened (default max size: INT_MAX - 4095) */
885 extern void *xmalloc_open_read_close(const char *filename, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
886 /* Never returns NULL */
887 extern void *xmalloc_xopen_read_close(const char *filename, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
888
889 #if defined(ARG_MAX) && (ARG_MAX >= 60*1024 || !defined(_SC_ARG_MAX))
890 /* Use _constant_ maximum if: defined && (big enough || no variable one exists) */
891 # define bb_arg_max() ((unsigned)ARG_MAX)
892 #elif defined(_SC_ARG_MAX)
893 /* Else use variable one (a bit more expensive) */
894 unsigned bb_arg_max(void) FAST_FUNC;
895 #else
896 /* If all else fails */
897 # define bb_arg_max() ((unsigned)(32 * 1024))
898 #endif
899 unsigned bb_clk_tck(void) FAST_FUNC;
900
901 #define SEAMLESS_COMPRESSION (0 \
902  || ENABLE_FEATURE_SEAMLESS_XZ \
903  || ENABLE_FEATURE_SEAMLESS_LZMA \
904  || ENABLE_FEATURE_SEAMLESS_BZ2 \
905  || ENABLE_FEATURE_SEAMLESS_GZ \
906  || ENABLE_FEATURE_SEAMLESS_Z)
907
908 #if SEAMLESS_COMPRESSION
909 /* Autodetects gzip/bzip2 formats. fd may be in the middle of the file! */
910 int setup_unzip_on_fd(int fd, int fail_if_not_compressed) FAST_FUNC;
911 /* Autodetects .gz etc */
912 extern int open_zipped(const char *fname, int fail_if_not_compressed) FAST_FUNC;
913 extern void *xmalloc_open_zipped_read_close(const char *fname, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
914 #else
915 # define setup_unzip_on_fd(...) (0)
916 # define open_zipped(fname, fail_if_not_compressed)  open((fname), O_RDONLY);
917 # define xmalloc_open_zipped_read_close(fname, maxsz_p) xmalloc_open_read_close((fname), (maxsz_p))
918 #endif
919 /* lzma has no signature, need a little helper. NB: exist only for ENABLE_FEATURE_SEAMLESS_LZMA=y */
920 void setup_lzma_on_fd(int fd) FAST_FUNC;
921
922 extern ssize_t safe_write(int fd, const void *buf, size_t count) FAST_FUNC;
923 // NB: will return short write on error, not -1,
924 // if some data was written before error occurred
925 extern ssize_t full_write(int fd, const void *buf, size_t count) FAST_FUNC;
926 extern void xwrite(int fd, const void *buf, size_t count) FAST_FUNC;
927 extern void xwrite_str(int fd, const char *str) FAST_FUNC;
928 extern ssize_t full_write1_str(const char *str) FAST_FUNC;
929 extern ssize_t full_write2_str(const char *str) FAST_FUNC;
930 extern void xopen_xwrite_close(const char* file, const char *str) FAST_FUNC;
931
932 /* Close fd, but check for failures (some types of write errors) */
933 extern void xclose(int fd) FAST_FUNC;
934
935 /* Reads and prints to stdout till eof, then closes FILE. Exits on error: */
936 extern void xprint_and_close_file(FILE *file) FAST_FUNC;
937
938 /* Reads a line from a text file, up to a newline or NUL byte, inclusive.
939  * Returns malloc'ed char*. If end is NULL '\n' isn't considered
940  * end of line. If end isn't NULL, length of the chunk is stored in it.
941  * Returns NULL if EOF/error.
942  */
943 extern char *bb_get_chunk_from_file(FILE *file, size_t *end) FAST_FUNC;
944 /* Reads up to (and including) TERMINATING_STRING: */
945 extern char *xmalloc_fgets_str(FILE *file, const char *terminating_string) FAST_FUNC RETURNS_MALLOC;
946 /* Same, with limited max size, and returns the length (excluding NUL): */
947 extern char *xmalloc_fgets_str_len(FILE *file, const char *terminating_string, size_t *maxsz_p) FAST_FUNC RETURNS_MALLOC;
948 /* Chops off TERMINATING_STRING from the end: */
949 extern char *xmalloc_fgetline_str(FILE *file, const char *terminating_string) FAST_FUNC RETURNS_MALLOC;
950 /* Reads up to (and including) "\n" or NUL byte: */
951 extern char *xmalloc_fgets(FILE *file) FAST_FUNC RETURNS_MALLOC;
952 /* Chops off '\n' from the end, unlike fgets: */
953 extern char *xmalloc_fgetline(FILE *file) FAST_FUNC RETURNS_MALLOC;
954 /* Same, but doesn't try to conserve space (may have some slack after the end) */
955 /* extern char *xmalloc_fgetline_fast(FILE *file) FAST_FUNC RETURNS_MALLOC; */
956
957 void die_if_ferror(FILE *file, const char *msg) FAST_FUNC;
958 void die_if_ferror_stdout(void) FAST_FUNC;
959 int fflush_all(void) FAST_FUNC;
960 void fflush_stdout_and_exit(int retval) NORETURN FAST_FUNC;
961 int fclose_if_not_stdin(FILE *file) FAST_FUNC;
962 FILE* xfopen(const char *filename, const char *mode) FAST_FUNC;
963 /* Prints warning to stderr and returns NULL on failure: */
964 FILE* fopen_or_warn(const char *filename, const char *mode) FAST_FUNC;
965 /* "Opens" stdin if filename is special, else just opens file: */
966 FILE* xfopen_stdin(const char *filename) FAST_FUNC;
967 FILE* fopen_or_warn_stdin(const char *filename) FAST_FUNC;
968 FILE* fopen_for_read(const char *path) FAST_FUNC;
969 FILE* xfopen_for_read(const char *path) FAST_FUNC;
970 FILE* fopen_for_write(const char *path) FAST_FUNC;
971 FILE* xfopen_for_write(const char *path) FAST_FUNC;
972 FILE* xfdopen_for_read(int fd) FAST_FUNC;
973 FILE* xfdopen_for_write(int fd) FAST_FUNC;
974
975 int bb_pstrcmp(const void *a, const void *b) /* not FAST_FUNC! */;
976 void qsort_string_vector(char **sv, unsigned count) FAST_FUNC;
977
978 /* Wrapper which restarts poll on EINTR or ENOMEM.
979  * On other errors complains [perror("poll")] and returns.
980  * Warning! May take (much) longer than timeout_ms to return!
981  * If this is a problem, use bare poll and open-code EINTR/ENOMEM handling */
982 int safe_poll(struct pollfd *ufds, nfds_t nfds, int timeout_ms) FAST_FUNC;
983
984 char *safe_gethostname(void) FAST_FUNC;
985
986 /* Convert each alpha char in str to lower-case */
987 char* str_tolower(char *str) FAST_FUNC;
988
989 char *utoa(unsigned n) FAST_FUNC;
990 char *itoa(int n) FAST_FUNC;
991 /* Returns a pointer past the formatted number, does NOT null-terminate */
992 char *utoa_to_buf(unsigned n, char *buf, unsigned buflen) FAST_FUNC;
993 char *itoa_to_buf(int n, char *buf, unsigned buflen) FAST_FUNC;
994 /* Intelligent formatters of bignums */
995 char *smart_ulltoa4(unsigned long long ul, char buf[4], const char *scale) FAST_FUNC;
996 char *smart_ulltoa5(unsigned long long ul, char buf[5], const char *scale) FAST_FUNC;
997 /* If block_size == 0, display size without fractional part,
998  * else display (size * block_size) with one decimal digit.
999  * If display_unit == 0, show value no bigger than 1024 with suffix (K,M,G...),
1000  * else divide by display_unit and do not use suffix. */
1001 #define HUMAN_READABLE_MAX_WIDTH      7  /* "1024.0G" */
1002 #define HUMAN_READABLE_MAX_WIDTH_STR "7"
1003 //TODO: provide pointer to buf (avoid statics)?
1004 const char *make_human_readable_str(unsigned long long size,
1005                 unsigned long block_size, unsigned long display_unit) FAST_FUNC;
1006 /* Put a string of hex bytes ("1b2e66fe"...), return advanced pointer */
1007 char *bin2hex(char *dst, const char *src, int count) FAST_FUNC;
1008 /* Reverse */
1009 char* hex2bin(char *dst, const char *src, int count) FAST_FUNC;
1010
1011 /* Generate a UUID */
1012 void generate_uuid(uint8_t *buf) FAST_FUNC;
1013
1014 /* Last element is marked by mult == 0 */
1015 struct suffix_mult {
1016         char suffix[4];
1017         unsigned mult;
1018 };
1019 extern const struct suffix_mult bkm_suffixes[];
1020 #define km_suffixes (bkm_suffixes + 1)
1021 extern const struct suffix_mult cwbkMG_suffixes[];
1022 #define kMG_suffixes (cwbkMG_suffixes + 3)
1023 extern const struct suffix_mult kmg_i_suffixes[];
1024
1025 #include "xatonum.h"
1026 /* Specialized: */
1027
1028 /* Using xatoi() instead of naive atoi() is not always convenient -
1029  * in many places people want *non-negative* values, but store them
1030  * in signed int. Therefore we need this one:
1031  * dies if input is not in [0, INT_MAX] range. Also will reject '-0' etc.
1032  * It should really be named xatoi_nonnegative (since it allows 0),
1033  * but that would be too long.
1034  */
1035 int xatoi_positive(const char *numstr) FAST_FUNC;
1036
1037 /* Useful for reading port numbers */
1038 uint16_t xatou16(const char *numstr) FAST_FUNC;
1039
1040 #if ENABLE_FLOAT_DURATION
1041 typedef double duration_t;
1042 void sleep_for_duration(duration_t duration) FAST_FUNC;
1043 #else
1044 typedef unsigned duration_t;
1045 #define sleep_for_duration(duration) sleep(duration)
1046 #endif
1047 duration_t parse_duration_str(char *str) FAST_FUNC;
1048
1049 /* These parse entries in /etc/passwd and /etc/group.  This is desirable
1050  * for BusyBox since we want to avoid using the glibc NSS stuff, which
1051  * increases target size and is often not needed on embedded systems.  */
1052 long xuname2uid(const char *name) FAST_FUNC;
1053 long xgroup2gid(const char *name) FAST_FUNC;
1054 /* wrapper: allows string to contain numeric uid or gid */
1055 unsigned long get_ug_id(const char *s, long FAST_FUNC (*xname2id)(const char *)) FAST_FUNC;
1056 struct bb_uidgid_t {
1057         uid_t uid;
1058         gid_t gid;
1059 };
1060 /* always sets uid and gid; returns 0 on failure */
1061 int get_uidgid(struct bb_uidgid_t*, const char*) FAST_FUNC;
1062 /* always sets uid and gid; exits on failure */
1063 void xget_uidgid(struct bb_uidgid_t*, const char*) FAST_FUNC;
1064 /* chown-like handling of "user[:[group]" */
1065 void parse_chown_usergroup_or_die(struct bb_uidgid_t *u, char *user_group) FAST_FUNC;
1066 struct passwd* xgetpwnam(const char *name) FAST_FUNC;
1067 struct group* xgetgrnam(const char *name) FAST_FUNC;
1068 struct passwd* xgetpwuid(uid_t uid) FAST_FUNC;
1069 struct group* xgetgrgid(gid_t gid) FAST_FUNC;
1070 char* xuid2uname(uid_t uid) FAST_FUNC;
1071 char* xgid2group(gid_t gid) FAST_FUNC;
1072 char* uid2uname(uid_t uid) FAST_FUNC;
1073 char* gid2group(gid_t gid) FAST_FUNC;
1074 char* uid2uname_utoa(uid_t uid) FAST_FUNC;
1075 char* gid2group_utoa(gid_t gid) FAST_FUNC;
1076 /* versions which cache results (useful for ps, ls etc) */
1077 const char* get_cached_username(uid_t uid) FAST_FUNC;
1078 const char* get_cached_groupname(gid_t gid) FAST_FUNC;
1079 void clear_username_cache(void) FAST_FUNC;
1080 /* internally usernames are saved in fixed-sized char[] buffers */
1081 enum { USERNAME_MAX_SIZE = 32 - sizeof(uid_t) };
1082 #if ENABLE_FEATURE_CHECK_NAMES
1083 void die_if_bad_username(const char* name) FAST_FUNC;
1084 #else
1085 #define die_if_bad_username(name) ((void)(name))
1086 #endif
1087 /*
1088  * Returns (-1) terminated malloced result of getgroups().
1089  * Reallocs group_array (useful for repeated calls).
1090  * ngroups is an initial size of array. It is rounded up to 32 for realloc.
1091  * ngroups is updated on return.
1092  * ngroups can be NULL: bb_getgroups(NULL, NULL) is valid usage.
1093  * Dies on errors (on Linux, only xrealloc can cause this, not internal getgroups call).
1094  */
1095 gid_t *bb_getgroups(int *ngroups, gid_t *group_array) FAST_FUNC;
1096
1097 #if ENABLE_FEATURE_UTMP
1098 void FAST_FUNC write_new_utmp(pid_t pid, int new_type, const char *tty_name, const char *username, const char *hostname);
1099 void FAST_FUNC update_utmp(pid_t pid, int new_type, const char *tty_name, const char *username, const char *hostname);
1100 void FAST_FUNC update_utmp_DEAD_PROCESS(pid_t pid);
1101 #else
1102 # define write_new_utmp(pid, new_type, tty_name, username, hostname) ((void)0)
1103 # define update_utmp(pid, new_type, tty_name, username, hostname) ((void)0)
1104 # define update_utmp_DEAD_PROCESS(pid) ((void)0)
1105 #endif
1106
1107
1108 int file_is_executable(const char *name) FAST_FUNC;
1109 char *find_executable(const char *filename, char **PATHp) FAST_FUNC;
1110 int executable_exists(const char *filename) FAST_FUNC;
1111
1112 /* BB_EXECxx always execs (it's not doing NOFORK/NOEXEC stuff),
1113  * but it may exec busybox and call applet instead of searching PATH.
1114  */
1115 #if ENABLE_FEATURE_PREFER_APPLETS
1116 int BB_EXECVP(const char *file, char *const argv[]) FAST_FUNC;
1117 #define BB_EXECLP(prog,cmd,...) \
1118         do { \
1119                 if (find_applet_by_name(prog) >= 0) \
1120                         execlp(bb_busybox_exec_path, cmd, __VA_ARGS__); \
1121                 execlp(prog, cmd, __VA_ARGS__); \
1122         } while (0)
1123 #else
1124 #define BB_EXECVP(prog,cmd)     execvp(prog,cmd)
1125 #define BB_EXECLP(prog,cmd,...) execlp(prog,cmd,__VA_ARGS__)
1126 #endif
1127 void BB_EXECVP_or_die(char **argv) NORETURN FAST_FUNC;
1128 void exec_prog_or_SHELL(char **argv) NORETURN FAST_FUNC;
1129
1130 /* xvfork() can't be a _function_, return after vfork in child mangles stack
1131  * in the parent. It must be a macro. */
1132 #define xvfork() \
1133 ({ \
1134         pid_t bb__xvfork_pid = vfork(); \
1135         if (bb__xvfork_pid < 0) \
1136                 bb_perror_msg_and_die("vfork"); \
1137         bb__xvfork_pid; \
1138 })
1139 #if BB_MMU
1140 pid_t xfork(void) FAST_FUNC;
1141 #endif
1142 void xvfork_parent_waits_and_exits(void) FAST_FUNC;
1143
1144 /* NOMMU friendy fork+exec: */
1145 pid_t spawn(char **argv) FAST_FUNC;
1146 pid_t xspawn(char **argv) FAST_FUNC;
1147
1148 pid_t safe_waitpid(pid_t pid, int *wstat, int options) FAST_FUNC;
1149 pid_t wait_any_nohang(int *wstat) FAST_FUNC;
1150 /* wait4pid: unlike waitpid, waits ONLY for one process.
1151  * Returns sig + 0x180 if child is killed by signal.
1152  * It's safe to pass negative 'pids' from failed [v]fork -
1153  * wait4pid will return -1 (and will not clobber [v]fork's errno).
1154  * IOW: rc = wait4pid(spawn(argv));
1155  *      if (rc < 0) bb_perror_msg("%s", argv[0]);
1156  *      if (rc > 0) bb_error_msg("exit code: %d", rc & 0xff);
1157  */
1158 int wait4pid(pid_t pid) FAST_FUNC;
1159 int wait_for_exitstatus(pid_t pid) FAST_FUNC;
1160 /************************************************************************/
1161 /* spawn_and_wait/run_nofork_applet/run_applet_no_and_exit need to work */
1162 /* carefully together to reinit some global state while not disturbing  */
1163 /* other. Be careful if you change them. Consult docs/nofork_noexec.txt */
1164 /************************************************************************/
1165 /* Same as wait4pid(spawn(argv)), but with NOFORK/NOEXEC if configured: */
1166 int spawn_and_wait(char **argv) FAST_FUNC;
1167 /* Does NOT check that applet is NOFORK, just blindly runs it */
1168 int run_nofork_applet(int applet_no, char **argv) FAST_FUNC;
1169 void run_noexec_applet_and_exit(int a, const char *name, char **argv) NORETURN FAST_FUNC;
1170 #ifndef BUILD_INDIVIDUAL
1171 int find_applet_by_name(const char *name) FAST_FUNC;
1172 void run_applet_no_and_exit(int a, const char *name, char **argv) NORETURN FAST_FUNC;
1173 #endif
1174 #if defined(__linux__)
1175 void set_task_comm(const char *comm) FAST_FUNC;
1176 #else
1177 # define set_task_comm(name) ((void)0)
1178 #endif
1179
1180 /* Helpers for daemonization.
1181  *
1182  * bb_daemonize(flags) = daemonize, does not compile on NOMMU
1183  *
1184  * bb_daemonize_or_rexec(flags, argv) = daemonizes on MMU (and ignores argv),
1185  *      rexec's itself on NOMMU with argv passed as command line.
1186  * Thus bb_daemonize_or_rexec may cause your <applet>_main() to be re-executed
1187  * from the start. (It will detect it and not reexec again second time).
1188  * You have to audit carefully that you don't do something twice as a result
1189  * (opening files/sockets, parsing config files etc...)!
1190  *
1191  * Both of the above will redirect fd 0,1,2 to /dev/null and drop ctty
1192  * (will do setsid()).
1193  *
1194  * fork_or_rexec(argv) = bare-bones fork on MMU,
1195  *      "vfork + re-exec ourself" on NOMMU. No fd redirection, no setsid().
1196  *      On MMU ignores argv.
1197  *
1198  * Helper for network daemons in foreground mode:
1199  *
1200  * bb_sanitize_stdio() = make sure that fd 0,1,2 are opened by opening them
1201  * to /dev/null if they are not.
1202  */
1203 enum {
1204         DAEMON_CHDIR_ROOT = 1,
1205         DAEMON_DEVNULL_STDIO = 2,
1206         DAEMON_CLOSE_EXTRA_FDS = 4,
1207         DAEMON_ONLY_SANITIZE = 8, /* internal use */
1208         DAEMON_DOUBLE_FORK = 16, /* double fork to avoid controlling tty */
1209 };
1210 #if BB_MMU
1211   enum { re_execed = 0 };
1212 # define fork_or_rexec(argv)                xfork()
1213 # define bb_daemonize_or_rexec(flags, argv) bb_daemonize_or_rexec(flags)
1214 # define bb_daemonize(flags)                bb_daemonize_or_rexec(flags, bogus)
1215 #else
1216   extern bool re_execed;
1217   /* Note: re_exec() and fork_or_rexec() do argv[0][0] |= 0x80 on NOMMU!
1218    * _Parent_ needs to undo it if it doesn't want to have argv[0] mangled.
1219    */
1220   void re_exec(char **argv) NORETURN FAST_FUNC;
1221   pid_t fork_or_rexec(char **argv) FAST_FUNC;
1222   int  BUG_fork_is_unavailable_on_nommu(void) FAST_FUNC;
1223   int  BUG_daemon_is_unavailable_on_nommu(void) FAST_FUNC;
1224   void BUG_bb_daemonize_is_unavailable_on_nommu(void) FAST_FUNC;
1225 # define fork()          BUG_fork_is_unavailable_on_nommu()
1226 # define xfork()         BUG_fork_is_unavailable_on_nommu()
1227 # define daemon(a,b)     BUG_daemon_is_unavailable_on_nommu()
1228 # define bb_daemonize(a) BUG_bb_daemonize_is_unavailable_on_nommu()
1229 #endif
1230 void bb_daemonize_or_rexec(int flags, char **argv) FAST_FUNC;
1231 void bb_sanitize_stdio(void) FAST_FUNC;
1232 #define bb_daemon_helper(arg) bb_daemonize_or_rexec((arg) | DAEMON_ONLY_SANITIZE, NULL)
1233 /* Clear dangerous stuff, set PATH. Return 1 if was run by different user. */
1234 int sanitize_env_if_suid(void) FAST_FUNC;
1235
1236
1237 /* For top, ps. Some argv[i] are replaced by malloced "-opt" strings */
1238 void make_all_argv_opts(char **argv) FAST_FUNC;
1239 char* single_argv(char **argv) FAST_FUNC;
1240 extern const char *const bb_argv_dash[]; /* { "-", NULL } */
1241 extern uint32_t option_mask32;
1242 uint32_t getopt32(char **argv, const char *applet_opts, ...) FAST_FUNC;
1243 # define No_argument "\0"
1244 # define Required_argument "\001"
1245 # define Optional_argument "\002"
1246 #if ENABLE_LONG_OPTS
1247 uint32_t getopt32long(char **argv, const char *optstring, const char *longopts, ...) FAST_FUNC;
1248 #else
1249 #define getopt32long(argv,optstring,longopts,...) \
1250         getopt32(argv,optstring,##__VA_ARGS__)
1251 #endif
1252 /* BSD-derived getopt() functions require that optind be set to 1 in
1253  * order to reset getopt() state.  This used to be generally accepted
1254  * way of resetting getopt().  However, glibc's getopt()
1255  * has additional getopt() state beyond optind (specifically, glibc
1256  * extensions such as '+' and '-' at the start of the string), and requires
1257  * that optind be set to zero to reset its state.  BSD-derived versions
1258  * of getopt() misbehaved if optind is set to 0 in order to reset getopt(),
1259  * and glibc's getopt() used to coredump if optind is set 1 in order
1260  * to reset getopt().
1261  * Then BSD introduced additional variable "optreset" which should be
1262  * set to 1 in order to reset getopt().  Sigh.  Standards, anyone?
1263  *
1264  * By ~2008, OpenBSD 3.4 was changed to survive glibc-like optind = 0
1265  * (to interpret it as if optreset was set).
1266  */
1267 #if 1 /*def __GLIBC__*/
1268 #define GETOPT_RESET() (optind = 0)
1269 #else /* BSD style */
1270 #define GETOPT_RESET() (optind = 1)
1271 #endif
1272
1273
1274 /* Having next pointer as a first member allows easy creation
1275  * of "llist-compatible" structs, and using llist_FOO functions
1276  * on them.
1277  */
1278 typedef struct llist_t {
1279         struct llist_t *link;
1280         char *data;
1281 } llist_t;
1282 void llist_add_to(llist_t **old_head, void *data) FAST_FUNC;
1283 void llist_add_to_end(llist_t **list_head, void *data) FAST_FUNC;
1284 void *llist_pop(llist_t **elm) FAST_FUNC;
1285 void llist_unlink(llist_t **head, llist_t *elm) FAST_FUNC;
1286 void llist_free(llist_t *elm, void (*freeit)(void *data)) FAST_FUNC;
1287 llist_t *llist_rev(llist_t *list) FAST_FUNC;
1288 llist_t *llist_find_str(llist_t *first, const char *str) FAST_FUNC;
1289 /* BTW, surprisingly, changing API to
1290  *   llist_t *llist_add_to(llist_t *old_head, void *data)
1291  * etc does not result in smaller code... */
1292
1293 /* start_stop_daemon and udhcpc are special - they want
1294  * to create pidfiles regardless of FEATURE_PIDFILE */
1295 #if ENABLE_FEATURE_PIDFILE || defined(WANT_PIDFILE)
1296 /* True only if we created pidfile which is *file*, not /dev/null etc */
1297 extern smallint wrote_pidfile;
1298 void write_pidfile(const char *path) FAST_FUNC;
1299 #define remove_pidfile(path) do { if (wrote_pidfile) unlink(path); } while (0)
1300 #else
1301 enum { wrote_pidfile = 0 };
1302 #define write_pidfile(path)  ((void)0)
1303 #define remove_pidfile(path) ((void)0)
1304 #endif
1305
1306 enum {
1307         LOGMODE_NONE = 0,
1308         LOGMODE_STDIO = (1 << 0),
1309         LOGMODE_SYSLOG = (1 << 1) * ENABLE_FEATURE_SYSLOG,
1310         LOGMODE_BOTH = LOGMODE_SYSLOG + LOGMODE_STDIO,
1311 };
1312 extern const char *msg_eol;
1313 extern smallint syslog_level;
1314 extern smallint logmode;
1315 extern uint8_t xfunc_error_retval;
1316 extern void (*die_func)(void);
1317 void xfunc_die(void) NORETURN FAST_FUNC;
1318 void bb_show_usage(void) NORETURN FAST_FUNC;
1319 void bb_error_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1320 void bb_error_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1321 void bb_perror_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1322 void bb_simple_perror_msg(const char *s) FAST_FUNC;
1323 void bb_perror_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1324 void bb_simple_perror_msg_and_die(const char *s) NORETURN FAST_FUNC;
1325 void bb_herror_msg(const char *s, ...) __attribute__ ((format (printf, 1, 2))) FAST_FUNC;
1326 void bb_herror_msg_and_die(const char *s, ...) __attribute__ ((noreturn, format (printf, 1, 2))) FAST_FUNC;
1327 void bb_perror_nomsg_and_die(void) NORETURN FAST_FUNC;
1328 void bb_perror_nomsg(void) FAST_FUNC;
1329 void bb_verror_msg(const char *s, va_list p, const char *strerr) FAST_FUNC;
1330 void bb_die_memory_exhausted(void) NORETURN FAST_FUNC;
1331 void bb_logenv_override(void) FAST_FUNC;
1332
1333 /* We need to export XXX_main from libbusybox
1334  * only if we build "individual" binaries
1335  */
1336 #if ENABLE_FEATURE_INDIVIDUAL
1337 #define MAIN_EXTERNALLY_VISIBLE EXTERNALLY_VISIBLE
1338 #else
1339 #define MAIN_EXTERNALLY_VISIBLE
1340 #endif
1341
1342 /* Embedded script support */
1343 char *get_script_content(unsigned n) FAST_FUNC;
1344 int scripted_main(int argc, char** argv);
1345
1346 /* Applets which are useful from another applets */
1347 int bb_cat(char** argv) FAST_FUNC;
1348 int ash_main(int argc, char** argv)
1349 #if ENABLE_ASH || ENABLE_SH_IS_ASH || ENABLE_BASH_IS_ASH
1350                 MAIN_EXTERNALLY_VISIBLE
1351 #endif
1352 ;
1353 int hush_main(int argc, char** argv)
1354 #if ENABLE_HUSH || ENABLE_SH_IS_HUSH || ENABLE_BASH_IS_HUSH
1355                 MAIN_EXTERNALLY_VISIBLE
1356 #endif
1357 ;
1358 /* If shell needs them, they exist even if not enabled as applets */
1359 int echo_main(int argc, char** argv) IF_ECHO(MAIN_EXTERNALLY_VISIBLE);
1360 int printf_main(int argc, char **argv) IF_PRINTF(MAIN_EXTERNALLY_VISIBLE);
1361 int test_main(int argc, char **argv)
1362 #if ENABLE_TEST || ENABLE_TEST1 || ENABLE_TEST2
1363                 MAIN_EXTERNALLY_VISIBLE
1364 #endif
1365 ;
1366 int kill_main(int argc, char **argv)
1367 #if ENABLE_KILL || ENABLE_KILLALL || ENABLE_KILLALL5
1368                 MAIN_EXTERNALLY_VISIBLE
1369 #endif
1370 ;
1371 /* Similar, but used by chgrp, not shell */
1372 int chown_main(int argc, char **argv) IF_CHOWN(MAIN_EXTERNALLY_VISIBLE);
1373 /* Used by ftpd */
1374 int ls_main(int argc, char **argv) IF_LS(MAIN_EXTERNALLY_VISIBLE);
1375 /* Don't need IF_xxx() guard for these */
1376 int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1377 int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1378
1379 #if ENABLE_ROUTE
1380 void bb_displayroutes(int noresolve, int netstatfmt) FAST_FUNC;
1381 #endif
1382
1383 struct number_state {
1384         unsigned width;
1385         unsigned start;
1386         unsigned inc;
1387         const char *sep;
1388         const char *empty_str;
1389         smallint all, nonempty;
1390 };
1391 int print_numbered_lines(struct number_state *ns, const char *filename) FAST_FUNC;
1392
1393
1394 /* Networking */
1395 /* This structure defines protocol families and their handlers. */
1396 struct aftype {
1397         const char *name;
1398         const char *title;
1399         int af;
1400         int alen;
1401         char*       FAST_FUNC (*print)(unsigned char *);
1402         const char* FAST_FUNC (*sprint)(struct sockaddr *, int numeric);
1403         int         FAST_FUNC (*input)(/*int type,*/ const char *bufp, struct sockaddr *);
1404         void        FAST_FUNC (*herror)(char *text);
1405         int         FAST_FUNC (*rprint)(int options);
1406         int         FAST_FUNC (*rinput)(int typ, int ext, char **argv);
1407         /* may modify src */
1408         int         FAST_FUNC (*getmask)(char *src, struct sockaddr *mask, char *name);
1409 };
1410 /* This structure defines hardware protocols and their handlers. */
1411 struct hwtype {
1412         const char *name;
1413         const char *title;
1414         int type;
1415         int alen;
1416         char* FAST_FUNC (*print)(unsigned char *);
1417         int   FAST_FUNC (*input)(const char *, struct sockaddr *);
1418         int   FAST_FUNC (*activate)(int fd);
1419         int suppress_null_addr;
1420 };
1421 #define IFNAME_SHOW_DOWNED_TOO ((char*)(intptr_t)1)
1422 int display_interfaces(char *ifname) FAST_FUNC;
1423 int in_ether(const char *bufp, struct sockaddr *sap) FAST_FUNC;
1424 #if ENABLE_FEATURE_HWIB
1425 int in_ib(const char *bufp, struct sockaddr *sap) FAST_FUNC;
1426 #else
1427 #define in_ib(a, b) 1 /* fail */
1428 #endif
1429 const struct aftype *get_aftype(const char *name) FAST_FUNC;
1430 const struct hwtype *get_hwtype(const char *name) FAST_FUNC;
1431 const struct hwtype *get_hwntype(int type) FAST_FUNC;
1432
1433
1434 extern int fstype_matches(const char *fstype, const char *comma_list) FAST_FUNC;
1435 #ifdef HAVE_MNTENT_H
1436 extern struct mntent *find_mount_point(const char *name, int subdir_too) FAST_FUNC;
1437 #endif
1438 extern void erase_mtab(const char * name) FAST_FUNC;
1439 extern unsigned int tty_baud_to_value(speed_t speed) FAST_FUNC;
1440 extern speed_t tty_value_to_baud(unsigned int value) FAST_FUNC;
1441 #if ENABLE_DESKTOP
1442 extern void bb_warn_ignoring_args(char *arg) FAST_FUNC;
1443 #else
1444 # define bb_warn_ignoring_args(arg) ((void)0)
1445 #endif
1446
1447 extern int get_linux_version_code(void) FAST_FUNC;
1448
1449 extern char *query_loop(const char *device) FAST_FUNC;
1450 extern int del_loop(const char *device) FAST_FUNC;
1451 /*
1452  * If *devname is not NULL, use that name, otherwise try to find free one,
1453  * malloc and return it in *devname.
1454  * return value is the opened fd to the loop device, or < on error
1455  */
1456 extern int set_loop(char **devname, const char *file, unsigned long long offset, unsigned flags) FAST_FUNC;
1457 /* These constants match linux/loop.h (without BB_ prefix): */
1458 #define BB_LO_FLAGS_READ_ONLY 1
1459 #define BB_LO_FLAGS_AUTOCLEAR 4
1460
1461 /* Returns malloced str */
1462 char *bb_ask_noecho(int fd, int timeout, const char *prompt) FAST_FUNC;
1463 /* Like bb_ask_noecho, but asks on stdin with no timeout.  */
1464 char *bb_ask_noecho_stdin(const char *prompt) FAST_FUNC;
1465
1466 int bb_ask_y_confirmation_FILE(FILE *fp) FAST_FUNC;
1467 int bb_ask_y_confirmation(void) FAST_FUNC;
1468
1469 /* Returns -1 if input is invalid. current_mode is a base for e.g. "u+rw" */
1470 int bb_parse_mode(const char* s, unsigned cur_mode) FAST_FUNC;
1471
1472 /*
1473  * Config file parser
1474  */
1475 enum {
1476         PARSE_COLLAPSE  = 0x00010000, // treat consecutive delimiters as one
1477         PARSE_TRIM      = 0x00020000, // trim leading and trailing delimiters
1478 // TODO: COLLAPSE and TRIM seem to always go in pair
1479         PARSE_GREEDY    = 0x00040000, // last token takes entire remainder of the line
1480         PARSE_MIN_DIE   = 0x00100000, // die if < min tokens found
1481         // keep a copy of current line
1482         PARSE_KEEP_COPY = 0x00200000 * ENABLE_FEATURE_CROND_D,
1483         PARSE_EOL_COMMENTS = 0x00400000, // comments are recognized even if they aren't the first char
1484         PARSE_ALT_COMMENTS = 0x00800000, // delim[0] and delim[1] are two different allowed comment chars
1485         // (so far, delim[0] will only work as comment char for full-line comment)
1486         // (IOW: it works as if PARSE_EOL_COMMENTS is not set. sysctl applet is okay with this)
1487         PARSE_WS_COMMENTS  = 0x01000000, // comments are recognized even if there is whitespace before
1488         // ("line start><space><tab><space>#comment" is also comment, not only "line start>#comment")
1489         // NORMAL is:
1490         // * remove leading and trailing delimiters and collapse
1491         //   multiple delimiters into one
1492         // * warn and continue if less than mintokens delimiters found
1493         // * grab everything into last token
1494         // * comments are recognized even if they aren't the first char
1495         PARSE_NORMAL    = PARSE_COLLAPSE | PARSE_TRIM | PARSE_GREEDY | PARSE_EOL_COMMENTS,
1496 };
1497 typedef struct parser_t {
1498         FILE *fp;
1499         char *data;
1500         char *line, *nline;
1501         size_t line_alloc, nline_alloc;
1502         int lineno;
1503 } parser_t;
1504 parser_t* config_open(const char *filename) FAST_FUNC;
1505 parser_t* config_open2(const char *filename, FILE* FAST_FUNC (*fopen_func)(const char *path)) FAST_FUNC;
1506 /* delims[0] is a comment char (use '\0' to disable), the rest are token delimiters */
1507 int config_read(parser_t *parser, char **tokens, unsigned flags, const char *delims) FAST_FUNC;
1508 #define config_read(parser, tokens, max, min, str, flags) \
1509         config_read(parser, tokens, ((flags) | (((min) & 0xFF) << 8) | ((max) & 0xFF)), str)
1510 void config_close(parser_t *parser) FAST_FUNC;
1511
1512 /* Concatenate path and filename to new allocated buffer.
1513  * Add "/" only as needed (no duplicate "//" are produced).
1514  * If path is NULL, it is assumed to be "/".
1515  * filename should not be NULL. */
1516 char *concat_path_file(const char *path, const char *filename) FAST_FUNC;
1517 /* Returns NULL on . and .. */
1518 char *concat_subpath_file(const char *path, const char *filename) FAST_FUNC;
1519
1520
1521 int bb_make_directory(char *path, long mode, int flags) FAST_FUNC;
1522
1523 int get_signum(const char *name) FAST_FUNC;
1524 const char *get_signame(int number) FAST_FUNC;
1525 void print_signames(void) FAST_FUNC;
1526
1527 char *bb_simplify_path(const char *path) FAST_FUNC;
1528 /* Returns ptr to NUL */
1529 char *bb_simplify_abs_path_inplace(char *path) FAST_FUNC;
1530
1531 #ifndef LOGIN_FAIL_DELAY
1532 #define LOGIN_FAIL_DELAY 3
1533 #endif
1534 extern void bb_do_delay(int seconds) FAST_FUNC;
1535 extern void change_identity(const struct passwd *pw) FAST_FUNC;
1536 extern void run_shell(const char *shell, int loginshell, const char **args) NORETURN FAST_FUNC;
1537
1538 /* Returns $SHELL, getpwuid(getuid())->pw_shell, or DEFAULT_SHELL.
1539  * Note that getpwuid result might need xstrdup'ing
1540  * if there is a possibility of intervening getpwxxx() calls.
1541  */
1542 const char *get_shell_name(void) FAST_FUNC;
1543
1544 #if ENABLE_FEATURE_SETPRIV_CAPABILITIES || ENABLE_RUN_INIT
1545 unsigned cap_name_to_number(const char *cap) FAST_FUNC;
1546 void printf_cap(const char *pfx, unsigned cap_no) FAST_FUNC;
1547 void drop_capability(int cap_ordinal) FAST_FUNC;
1548 /* Structures inside "struct caps" are Linux-specific and libcap-specific: */
1549 #define DEFINE_STRUCT_CAPS \
1550 struct caps { \
1551         struct __user_cap_header_struct header; \
1552         unsigned u32s; \
1553         struct __user_cap_data_struct data[2]; \
1554 }
1555 void getcaps(void *caps) FAST_FUNC;
1556 #endif
1557
1558 #if ENABLE_SELINUX
1559 extern void renew_current_security_context(void) FAST_FUNC;
1560 extern void set_current_security_context(security_context_t sid) FAST_FUNC;
1561 extern context_t set_security_context_component(security_context_t cur_context,
1562                                                 char *user, char *role, char *type, char *range) FAST_FUNC;
1563 extern void setfscreatecon_or_die(security_context_t scontext) FAST_FUNC;
1564 extern void selinux_preserve_fcontext(int fdesc) FAST_FUNC;
1565 #else
1566 #define selinux_preserve_fcontext(fdesc) ((void)0)
1567 #endif
1568 extern void selinux_or_die(void) FAST_FUNC;
1569
1570
1571 /* setup_environment:
1572  * if chdir pw->pw_dir: ok: else if to_tmp == 1: goto /tmp else: goto / or die
1573  * if clear_env = 1: cd(pw->pw_dir), clear environment, then set
1574  *   TERM=(old value)
1575  *   USER=pw->pw_name, LOGNAME=pw->pw_name
1576  *   PATH=bb_default_[root_]path
1577  *   HOME=pw->pw_dir
1578  *   SHELL=shell
1579  * else if change_env = 1:
1580  *   if not root (if pw->pw_uid != 0):
1581  *     USER=pw->pw_name, LOGNAME=pw->pw_name
1582  *   HOME=pw->pw_dir
1583  *   SHELL=shell
1584  * else does nothing
1585  *
1586  * NB: CHANGEENV and CLEARENV use setenv() - this leaks memory!
1587  * If setup_environment() is used is vforked child, this leaks memory _in parent too_!
1588  */
1589 #define SETUP_ENV_CHANGEENV (1 << 0)
1590 #define SETUP_ENV_CLEARENV  (1 << 1)
1591 #define SETUP_ENV_TO_TMP    (1 << 2)
1592 #define SETUP_ENV_NO_CHDIR  (1 << 4)
1593 void setup_environment(const char *shell, int flags, const struct passwd *pw) FAST_FUNC;
1594 void nuke_str(char *str) FAST_FUNC;
1595 #if ENABLE_FEATURE_SECURETTY && !ENABLE_PAM
1596 int is_tty_secure(const char *short_tty) FAST_FUNC;
1597 #else
1598 static ALWAYS_INLINE int is_tty_secure(const char *short_tty UNUSED_PARAM) { return 1; }
1599 #endif
1600 #define CHECKPASS_PW_HAS_EMPTY_PASSWORD 2
1601 int check_password(const struct passwd *pw, const char *plaintext) FAST_FUNC;
1602 int ask_and_check_password_extended(const struct passwd *pw, int timeout, const char *prompt) FAST_FUNC;
1603 int ask_and_check_password(const struct passwd *pw) FAST_FUNC;
1604 /* Returns a malloced string */
1605 #if !ENABLE_USE_BB_CRYPT
1606 #define pw_encrypt(clear, salt, cleanup) pw_encrypt(clear, salt)
1607 #endif
1608 extern char *pw_encrypt(const char *clear, const char *salt, int cleanup) FAST_FUNC;
1609 extern int obscure(const char *old, const char *newval, const struct passwd *pwdp) FAST_FUNC;
1610 /*
1611  * rnd is additional random input. New one is returned.
1612  * Useful if you call crypt_make_salt many times in a row:
1613  * rnd = crypt_make_salt(buf1, 4, 0);
1614  * rnd = crypt_make_salt(buf2, 4, rnd);
1615  * rnd = crypt_make_salt(buf3, 4, rnd);
1616  * (otherwise we risk having same salt generated)
1617  */
1618 extern int crypt_make_salt(char *p, int cnt /*, int rnd*/) FAST_FUNC;
1619 /* "$N$" + sha_salt_16_bytes + NUL */
1620 #define MAX_PW_SALT_LEN (3 + 16 + 1)
1621 extern char* crypt_make_pw_salt(char p[MAX_PW_SALT_LEN], const char *algo) FAST_FUNC;
1622
1623
1624 /* Returns number of lines changed, or -1 on error */
1625 #if !(ENABLE_FEATURE_ADDUSER_TO_GROUP || ENABLE_FEATURE_DEL_USER_FROM_GROUP)
1626 #define update_passwd(filename, username, data, member) \
1627         update_passwd(filename, username, data)
1628 #endif
1629 extern int update_passwd(const char *filename,
1630                 const char *username,
1631                 const char *data,
1632                 const char *member) FAST_FUNC;
1633
1634 int index_in_str_array(const char *const string_array[], const char *key) FAST_FUNC;
1635 int index_in_strings(const char *strings, const char *key) FAST_FUNC;
1636 int index_in_substr_array(const char *const string_array[], const char *key) FAST_FUNC;
1637 int index_in_substrings(const char *strings, const char *key) FAST_FUNC;
1638 const char *nth_string(const char *strings, int n) FAST_FUNC;
1639
1640 extern void print_login_issue(const char *issue_file, const char *tty) FAST_FUNC;
1641 extern void print_login_prompt(void) FAST_FUNC;
1642
1643 char *xmalloc_ttyname(int fd) FAST_FUNC RETURNS_MALLOC;
1644 /* NB: typically you want to pass fd 0, not 1. Think 'applet | grep something' */
1645 int get_terminal_width_height(int fd, unsigned *width, unsigned *height) FAST_FUNC;
1646 int get_terminal_width(int fd) FAST_FUNC;
1647
1648 int tcsetattr_stdin_TCSANOW(const struct termios *tp) FAST_FUNC;
1649 #define TERMIOS_CLEAR_ISIG      (1 << 0)
1650 #define TERMIOS_RAW_CRNL_INPUT  (1 << 1)
1651 #define TERMIOS_RAW_CRNL_OUTPUT (1 << 2)
1652 #define TERMIOS_RAW_CRNL        (TERMIOS_RAW_CRNL_INPUT|TERMIOS_RAW_CRNL_OUTPUT)
1653 #define TERMIOS_RAW_INPUT       (1 << 3)
1654 int get_termios_and_make_raw(int fd, struct termios *newterm, struct termios *oldterm, int flags) FAST_FUNC;
1655 int set_termios_to_raw(int fd, struct termios *oldterm, int flags) FAST_FUNC;
1656
1657 /* NB: "unsigned request" is crucial! "int request" will break some arches! */
1658 int ioctl_or_perror(int fd, unsigned request, void *argp, const char *fmt,...) __attribute__ ((format (printf, 4, 5))) FAST_FUNC;
1659 int ioctl_or_perror_and_die(int fd, unsigned request, void *argp, const char *fmt,...) __attribute__ ((format (printf, 4, 5))) FAST_FUNC;
1660 #if ENABLE_IOCTL_HEX2STR_ERROR
1661 int bb_ioctl_or_warn(int fd, unsigned request, void *argp, const char *ioctl_name) FAST_FUNC;
1662 int bb_xioctl(int fd, unsigned request, void *argp, const char *ioctl_name) FAST_FUNC;
1663 #define ioctl_or_warn(fd,request,argp) bb_ioctl_or_warn(fd,request,argp,#request)
1664 #define xioctl(fd,request,argp)        bb_xioctl(fd,request,argp,#request)
1665 #else
1666 int bb_ioctl_or_warn(int fd, unsigned request, void *argp) FAST_FUNC;
1667 int bb_xioctl(int fd, unsigned request, void *argp) FAST_FUNC;
1668 #define ioctl_or_warn(fd,request,argp) bb_ioctl_or_warn(fd,request,argp)
1669 #define xioctl(fd,request,argp)        bb_xioctl(fd,request,argp)
1670 #endif
1671
1672 char *is_in_ino_dev_hashtable(const struct stat *statbuf) FAST_FUNC;
1673 void add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name) FAST_FUNC;
1674 void reset_ino_dev_hashtable(void) FAST_FUNC;
1675 #ifdef __GLIBC__
1676 /* At least glibc has horrendously large inline for this, so wrap it */
1677 unsigned long long bb_makedev(unsigned major, unsigned minor) FAST_FUNC;
1678 #undef makedev
1679 #define makedev(a,b) bb_makedev(a,b)
1680 #endif
1681
1682
1683 /* "Keycodes" that report an escape sequence.
1684  * We use something which fits into signed char,
1685  * yet doesn't represent any valid Unicode character.
1686  * Also, -1 is reserved for error indication and we don't use it. */
1687 enum {
1688         KEYCODE_UP        =  -2,
1689         KEYCODE_DOWN      =  -3,
1690         KEYCODE_RIGHT     =  -4,
1691         KEYCODE_LEFT      =  -5,
1692         KEYCODE_HOME      =  -6,
1693         KEYCODE_END       =  -7,
1694         KEYCODE_INSERT    =  -8,
1695         KEYCODE_DELETE    =  -9,
1696         KEYCODE_PAGEUP    = -10,
1697         KEYCODE_PAGEDOWN  = -11,
1698         KEYCODE_BACKSPACE = -12, /* Used only if Alt/Ctrl/Shifted */
1699         KEYCODE_D         = -13, /* Used only if Alted */
1700 #if 0
1701         KEYCODE_FUN1      = ,
1702         KEYCODE_FUN2      = ,
1703         KEYCODE_FUN3      = ,
1704         KEYCODE_FUN4      = ,
1705         KEYCODE_FUN5      = ,
1706         KEYCODE_FUN6      = ,
1707         KEYCODE_FUN7      = ,
1708         KEYCODE_FUN8      = ,
1709         KEYCODE_FUN9      = ,
1710         KEYCODE_FUN10     = ,
1711         KEYCODE_FUN11     = ,
1712         KEYCODE_FUN12     = ,
1713 #endif
1714         /* ^^^^^ Be sure that last defined value is small enough.
1715          * Current read_key() code allows going up to -32 (0xfff..fffe0).
1716          * This gives three upper bits in LSB to play with:
1717          * KEYCODE_foo values are 0xfff..fffXX, lowest XX bits are: scavvvvv,
1718          * s=0 if SHIFT, c=0 if CTRL, a=0 if ALT,
1719          * vvvvv bits are the same for same key regardless of "shift bits".
1720          */
1721         //KEYCODE_SHIFT_...   = KEYCODE_...   & ~0x80,
1722         KEYCODE_CTRL_RIGHT    = KEYCODE_RIGHT & ~0x40,
1723         KEYCODE_CTRL_LEFT     = KEYCODE_LEFT  & ~0x40,
1724         KEYCODE_ALT_RIGHT     = KEYCODE_RIGHT & ~0x20,
1725         KEYCODE_ALT_LEFT      = KEYCODE_LEFT  & ~0x20,
1726         KEYCODE_ALT_BACKSPACE = KEYCODE_BACKSPACE & ~0x20,
1727         KEYCODE_ALT_D         = KEYCODE_D     & ~0x20,
1728
1729         KEYCODE_CURSOR_POS = -0x100, /* 0xfff..fff00 */
1730         /* How long is the longest ESC sequence we know?
1731          * We want it big enough to be able to contain
1732          * cursor position sequence "ESC [ 9999 ; 9999 R"
1733          */
1734         KEYCODE_BUFFER_SIZE = 16
1735 };
1736 /* Note: fd may be in blocking or non-blocking mode, both make sense.
1737  * For one, less uses non-blocking mode.
1738  * Only the first read syscall inside read_key may block indefinitely
1739  * (unless fd is in non-blocking mode),
1740  * subsequent reads will time out after a few milliseconds.
1741  * Return of -1 means EOF or error (errno == 0 on EOF).
1742  * buffer[0] is used as a counter of buffered chars and must be 0
1743  * on first call.
1744  * timeout:
1745  * -2: do not poll(-1) for input - read() it, return on EAGAIN at once
1746  * -1: poll(-1) (i.e. block even on NONBLOCKed fd)
1747  * >=0: poll() for TIMEOUT milliseconds, return -1/EAGAIN on timeout
1748  */
1749 int64_t read_key(int fd, char *buffer, int timeout) FAST_FUNC;
1750 void read_key_ungets(char *buffer, const char *str, unsigned len) FAST_FUNC;
1751
1752
1753 #if ENABLE_FEATURE_EDITING
1754 /* It's NOT just ENABLEd or disabled. It's a number: */
1755 # if defined CONFIG_FEATURE_EDITING_HISTORY && CONFIG_FEATURE_EDITING_HISTORY > 0
1756 #  define MAX_HISTORY (CONFIG_FEATURE_EDITING_HISTORY + 0)
1757 unsigned size_from_HISTFILESIZE(const char *hp) FAST_FUNC;
1758 # else
1759 #  define MAX_HISTORY 0
1760 # endif
1761 typedef struct line_input_t {
1762         int flags;
1763         int timeout;
1764         const char *path_lookup;
1765 # if MAX_HISTORY
1766         int cnt_history;
1767         int cur_history;
1768         int max_history; /* must never be <= 0 */
1769 #  if ENABLE_FEATURE_EDITING_SAVEHISTORY
1770         /* meaning of this field depends on FEATURE_EDITING_SAVE_ON_EXIT:
1771          * if !FEATURE_EDITING_SAVE_ON_EXIT: "how many lines are
1772          * in on-disk history"
1773          * if FEATURE_EDITING_SAVE_ON_EXIT: "how many in-memory lines are
1774          * also in on-disk history (and thus need to be skipped on save)"
1775          */
1776         unsigned cnt_history_in_file;
1777         const char *hist_file;
1778 #  endif
1779         char *history[MAX_HISTORY + 1];
1780 # endif
1781 } line_input_t;
1782 enum {
1783         DO_HISTORY       = 1 * (MAX_HISTORY > 0),
1784         TAB_COMPLETION   = 2 * ENABLE_FEATURE_TAB_COMPLETION,
1785         USERNAME_COMPLETION = 4 * ENABLE_FEATURE_USERNAME_COMPLETION,
1786         VI_MODE          = 8 * ENABLE_FEATURE_EDITING_VI,
1787         WITH_PATH_LOOKUP = 0x10,
1788         FOR_SHELL        = DO_HISTORY | TAB_COMPLETION | USERNAME_COMPLETION,
1789 };
1790 line_input_t *new_line_input_t(int flags) FAST_FUNC;
1791 void free_line_input_t(line_input_t *n) FAST_FUNC;
1792 /*
1793  * maxsize must be >= 2.
1794  * Returns:
1795  * -1 on read errors or EOF, or on bare Ctrl-D,
1796  * 0  on ctrl-C (the line entered is still returned in 'command'),
1797  * >0 length of input string, including terminating '\n'
1798  */
1799 int read_line_input(line_input_t *st, const char *prompt, char *command, int maxsize) FAST_FUNC;
1800 void show_history(const line_input_t *st) FAST_FUNC;
1801 # if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
1802 void save_history(line_input_t *st);
1803 # endif
1804 #else
1805 #define MAX_HISTORY 0
1806 int read_line_input(const char* prompt, char* command, int maxsize) FAST_FUNC;
1807 #define read_line_input(state, prompt, command, maxsize) \
1808         read_line_input(prompt, command, maxsize)
1809 #endif
1810
1811
1812 #ifndef COMM_LEN
1813 # ifdef TASK_COMM_LEN
1814 enum { COMM_LEN = TASK_COMM_LEN };
1815 # else
1816 /* synchronize with sizeof(task_struct.comm) in /usr/include/linux/sched.h */
1817 enum { COMM_LEN = 16 };
1818 # endif
1819 #endif
1820
1821 struct smaprec {
1822         unsigned long mapped_rw;
1823         unsigned long mapped_ro;
1824         unsigned long shared_clean;
1825         unsigned long shared_dirty;
1826         unsigned long private_clean;
1827         unsigned long private_dirty;
1828         unsigned long stack;
1829         unsigned long smap_pss, smap_swap;
1830         unsigned long smap_size;
1831         // For mixed 32/64 userspace, 32-bit pmap still needs
1832         // 64-bit field here to correctly show 64-bit processes:
1833         unsigned long long smap_start;
1834         // (strictly speaking, other fields need to be wider too,
1835         // but they are in kbytes, not bytes, and they hold sizes,
1836         // not start addresses, sizes tend to be less than 4 terabytes)
1837         char smap_mode[5];
1838         char *smap_name;
1839 };
1840
1841 #if !ENABLE_PMAP
1842 #define procps_read_smaps(pid, total, cb, data) \
1843         procps_read_smaps(pid, total)
1844 #endif
1845 int FAST_FUNC procps_read_smaps(pid_t pid, struct smaprec *total,
1846                 void (*cb)(struct smaprec *, void *), void *data);
1847
1848 typedef struct procps_status_t {
1849         DIR *dir;
1850         IF_FEATURE_SHOW_THREADS(DIR *task_dir;)
1851         uint8_t shift_pages_to_bytes;
1852         uint8_t shift_pages_to_kb;
1853 /* Fields are set to 0/NULL if failed to determine (or not requested) */
1854         uint16_t argv_len;
1855         char *argv0;
1856         char *exe;
1857         IF_SELINUX(char *context;)
1858         IF_FEATURE_SHOW_THREADS(unsigned main_thread_pid;)
1859         /* Everything below must contain no ptrs to malloc'ed data:
1860          * it is memset(0) for each process in procps_scan() */
1861         unsigned long vsz, rss; /* we round it to kbytes */
1862         unsigned long stime, utime;
1863         unsigned long start_time;
1864         unsigned pid;
1865         unsigned ppid;
1866         unsigned pgid;
1867         unsigned sid;
1868         unsigned uid;
1869         unsigned gid;
1870 #if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
1871         unsigned ruid;
1872         unsigned rgid;
1873         int niceness;
1874 #endif
1875         unsigned tty_major,tty_minor;
1876 #if ENABLE_FEATURE_TOPMEM
1877         struct smaprec smaps;
1878 #endif
1879         char state[4];
1880         /* basename of executable in exec(2), read from /proc/N/stat
1881          * (if executable is symlink or script, it is NOT replaced
1882          * by link target or interpreter name) */
1883         char comm[COMM_LEN];
1884         /* user/group? - use passwd/group parsing functions */
1885 #if ENABLE_FEATURE_TOP_SMP_PROCESS
1886         int last_seen_on_cpu;
1887 #endif
1888 } procps_status_t;
1889 /* flag bits for procps_scan(xx, flags) calls */
1890 enum {
1891         PSSCAN_PID      = 1 << 0,
1892         PSSCAN_PPID     = 1 << 1,
1893         PSSCAN_PGID     = 1 << 2,
1894         PSSCAN_SID      = 1 << 3,
1895         PSSCAN_UIDGID   = 1 << 4,
1896         PSSCAN_COMM     = 1 << 5,
1897         /* PSSCAN_CMD      = 1 << 6, - use read_cmdline instead */
1898         PSSCAN_ARGV0    = 1 << 7,
1899         PSSCAN_EXE      = 1 << 8,
1900         PSSCAN_STATE    = 1 << 9,
1901         PSSCAN_VSZ      = 1 << 10,
1902         PSSCAN_RSS      = 1 << 11,
1903         PSSCAN_STIME    = 1 << 12,
1904         PSSCAN_UTIME    = 1 << 13,
1905         PSSCAN_TTY      = 1 << 14,
1906         PSSCAN_SMAPS    = (1 << 15) * ENABLE_FEATURE_TOPMEM,
1907         /* NB: used by find_pid_by_name(). Any applet using it
1908          * needs to be mentioned here. */
1909         PSSCAN_ARGVN    = (1 << 16) * (ENABLE_KILLALL
1910                                 || ENABLE_PGREP || ENABLE_PKILL
1911                                 || ENABLE_PIDOF
1912                                 || ENABLE_SESTATUS
1913                                 ),
1914         PSSCAN_CONTEXT  = (1 << 17) * ENABLE_SELINUX,
1915         PSSCAN_START_TIME = 1 << 18,
1916         PSSCAN_CPU      = (1 << 19) * ENABLE_FEATURE_TOP_SMP_PROCESS,
1917         PSSCAN_NICE     = (1 << 20) * ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS,
1918         PSSCAN_RUIDGID  = (1 << 21) * ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS,
1919         PSSCAN_TASKS    = (1 << 22) * ENABLE_FEATURE_SHOW_THREADS,
1920 };
1921 //procps_status_t* alloc_procps_scan(void) FAST_FUNC;
1922 void free_procps_scan(procps_status_t* sp) FAST_FUNC;
1923 procps_status_t* procps_scan(procps_status_t* sp, int flags) FAST_FUNC;
1924 /* Format cmdline (up to col chars) into char buf[size] */
1925 /* Puts [comm] if cmdline is empty (-> process is a kernel thread) */
1926 void read_cmdline(char *buf, int size, unsigned pid, const char *comm) FAST_FUNC;
1927 pid_t *find_pid_by_name(const char* procName) FAST_FUNC;
1928 pid_t *pidlist_reverse(pid_t *pidList) FAST_FUNC;
1929 int starts_with_cpu(const char *str) FAST_FUNC;
1930 unsigned get_cpu_count(void) FAST_FUNC;
1931
1932
1933 /* Use strict=1 if you process input from untrusted source:
1934  * it will return NULL on invalid %xx (bad hex chars)
1935  * and str + 1 if decoded char is / or NUL.
1936  * In non-strict mode, it always succeeds (returns str),
1937  * and also it additionally decoded '+' to space.
1938  */
1939 char *percent_decode_in_place(char *str, int strict) FAST_FUNC;
1940
1941
1942 extern const char bb_uuenc_tbl_base64[] ALIGN1;
1943 extern const char bb_uuenc_tbl_std[] ALIGN1;
1944 void bb_uuencode(char *store, const void *s, int length, const char *tbl) FAST_FUNC;
1945 enum {
1946         BASE64_FLAG_UU_STOP = 0x100,
1947         /* Sign-extends to a value which never matches fgetc result: */
1948         BASE64_FLAG_NO_STOP_CHAR = 0x80,
1949 };
1950 const char *decode_base64(char **pp_dst, const char *src) FAST_FUNC;
1951 void read_base64(FILE *src_stream, FILE *dst_stream, int flags) FAST_FUNC;
1952
1953 typedef struct md5_ctx_t {
1954         uint8_t wbuffer[64]; /* always correctly aligned for uint64_t */
1955         void (*process_block)(struct md5_ctx_t*) FAST_FUNC;
1956         uint64_t total64;    /* must be directly before hash[] */
1957         uint32_t hash[8];    /* 4 elements for md5, 5 for sha1, 8 for sha256 */
1958 } md5_ctx_t;
1959 typedef struct md5_ctx_t sha1_ctx_t;
1960 typedef struct md5_ctx_t sha256_ctx_t;
1961 typedef struct sha512_ctx_t {
1962         uint64_t total64[2];  /* must be directly before hash[] */
1963         uint64_t hash[8];
1964         uint8_t wbuffer[128]; /* always correctly aligned for uint64_t */
1965 } sha512_ctx_t;
1966 typedef struct sha3_ctx_t {
1967         uint64_t state[25];
1968         unsigned bytes_queued;
1969         unsigned input_block_bytes;
1970 } sha3_ctx_t;
1971 void md5_begin(md5_ctx_t *ctx) FAST_FUNC;
1972 void md5_hash(md5_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1973 unsigned md5_end(md5_ctx_t *ctx, void *resbuf) FAST_FUNC;
1974 void sha1_begin(sha1_ctx_t *ctx) FAST_FUNC;
1975 #define sha1_hash md5_hash
1976 unsigned sha1_end(sha1_ctx_t *ctx, void *resbuf) FAST_FUNC;
1977 void sha256_begin(sha256_ctx_t *ctx) FAST_FUNC;
1978 #define sha256_hash md5_hash
1979 #define sha256_end  sha1_end
1980 void sha512_begin(sha512_ctx_t *ctx) FAST_FUNC;
1981 void sha512_hash(sha512_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1982 unsigned sha512_end(sha512_ctx_t *ctx, void *resbuf) FAST_FUNC;
1983 void sha3_begin(sha3_ctx_t *ctx) FAST_FUNC;
1984 void sha3_hash(sha3_ctx_t *ctx, const void *buffer, size_t len) FAST_FUNC;
1985 unsigned sha3_end(sha3_ctx_t *ctx, void *resbuf) FAST_FUNC;
1986 /* TLS benefits from knowing that sha1 and sha256 share these. Give them "agnostic" names too */
1987 typedef struct md5_ctx_t md5sha_ctx_t;
1988 #define md5sha_hash md5_hash
1989 #define sha_end sha1_end
1990
1991 extern uint32_t *global_crc32_table;
1992 uint32_t *crc32_filltable(uint32_t *tbl256, int endian) FAST_FUNC;
1993 uint32_t *crc32_new_table_le(void) FAST_FUNC;
1994 uint32_t *global_crc32_new_table_le(void) FAST_FUNC;
1995 uint32_t crc32_block_endian1(uint32_t val, const void *buf, unsigned len, uint32_t *crc_table) FAST_FUNC;
1996 uint32_t crc32_block_endian0(uint32_t val, const void *buf, unsigned len, uint32_t *crc_table) FAST_FUNC;
1997
1998 typedef struct masks_labels_t {
1999         const char *labels;
2000         const int masks[];
2001 } masks_labels_t;
2002 int print_flags_separated(const int *masks, const char *labels,
2003                 int flags, const char *separator) FAST_FUNC;
2004 int print_flags(const masks_labels_t *ml, int flags) FAST_FUNC;
2005
2006 typedef struct bb_progress_t {
2007         unsigned last_size;
2008         unsigned last_update_sec;
2009         unsigned last_change_sec;
2010         unsigned start_sec;
2011         /*unsigned last_eta;*/
2012         const char *curfile;
2013 } bb_progress_t;
2014
2015 #define is_bb_progress_inited(p) ((p)->curfile != NULL)
2016 #define bb_progress_free(p) do { \
2017         if (ENABLE_UNICODE_SUPPORT) free((char*)((p)->curfile)); \
2018         (p)->curfile = NULL; \
2019 } while (0)
2020 void bb_progress_init(bb_progress_t *p, const char *curfile) FAST_FUNC;
2021 int bb_progress_update(bb_progress_t *p,
2022                         uoff_t beg_range,
2023                         uoff_t transferred,
2024                         uoff_t totalsize) FAST_FUNC;
2025
2026 unsigned ubi_devnum_from_devname(const char *str) FAST_FUNC;
2027 int ubi_get_volid_by_name(unsigned ubi_devnum, const char *vol_name) FAST_FUNC;
2028
2029
2030 extern const char *applet_name;
2031
2032 /* Some older linkers don't perform string merging, we used to have common strings
2033  * as global arrays to do it by hand. But:
2034  * (1) newer linkers do it themselves,
2035  * (2) however, they DONT merge string constants with global arrays,
2036  * even if the value is the same (!). Thus global arrays actually
2037  * increased size a bit: for example, "/etc/passwd" string from libc
2038  * wasn't merged with bb_path_passwd_file[] array!
2039  * Therefore now we use #defines.
2040  */
2041 /* "BusyBox vN.N.N (timestamp or extra_version)" */
2042 extern const char bb_banner[] ALIGN1;
2043 extern const char bb_msg_memory_exhausted[] ALIGN1;
2044 extern const char bb_msg_invalid_date[] ALIGN1;
2045 #define bb_msg_read_error "read error"
2046 #define bb_msg_write_error "write error"
2047 extern const char bb_msg_unknown[] ALIGN1;
2048 extern const char bb_msg_can_not_create_raw_socket[] ALIGN1;
2049 extern const char bb_msg_perm_denied_are_you_root[] ALIGN1;
2050 extern const char bb_msg_you_must_be_root[] ALIGN1;
2051 extern const char bb_msg_requires_arg[] ALIGN1;
2052 extern const char bb_msg_invalid_arg_to[] ALIGN1;
2053 extern const char bb_msg_standard_input[] ALIGN1;
2054 extern const char bb_msg_standard_output[] ALIGN1;
2055
2056 /* NB: (bb_hexdigits_upcase[i] | 0x20) -> lowercase hex digit */
2057 extern const char bb_hexdigits_upcase[] ALIGN1;
2058
2059 extern const char bb_path_wtmp_file[] ALIGN1;
2060
2061 /* Busybox mount uses either /proc/mounts or /etc/mtab to
2062  * get the list of currently mounted filesystems */
2063 #define bb_path_mtab_file IF_FEATURE_MTAB_SUPPORT("/etc/mtab")IF_NOT_FEATURE_MTAB_SUPPORT("/proc/mounts")
2064
2065 #define bb_path_passwd_file  _PATH_PASSWD
2066 #define bb_path_group_file   _PATH_GROUP
2067 #define bb_path_shadow_file  _PATH_SHADOW
2068 #define bb_path_gshadow_file _PATH_GSHADOW
2069
2070 #define bb_path_motd_file "/etc/motd"
2071
2072 #define bb_dev_null "/dev/null"
2073 extern const char bb_busybox_exec_path[] ALIGN1;
2074 /* allow default system PATH to be extended via CFLAGS */
2075 #ifndef BB_ADDITIONAL_PATH
2076 #define BB_ADDITIONAL_PATH ""
2077 #endif
2078 #define BB_PATH_ROOT_PATH "PATH=/sbin:/usr/sbin:/bin:/usr/bin" BB_ADDITIONAL_PATH
2079 extern const char bb_PATH_root_path[] ALIGN1; /* BB_PATH_ROOT_PATH */
2080 #define bb_default_root_path (bb_PATH_root_path + sizeof("PATH"))
2081 /* util-linux manpage says /sbin:/bin:/usr/sbin:/usr/bin,
2082  * but I want to save a few bytes here:
2083  */
2084 #define bb_default_path      (bb_PATH_root_path + sizeof("PATH=/sbin:/usr/sbin"))
2085
2086 extern const int const_int_0;
2087 //extern const int const_int_1;
2088
2089 /* This struct is deliberately not defined. */
2090 /* See docs/keep_data_small.txt */
2091 struct globals;
2092 /* '*const' ptr makes gcc optimize code much better.
2093  * Magic prevents ptr_to_globals from going into rodata.
2094  * If you want to assign a value, use SET_PTR_TO_GLOBALS(x) */
2095 extern struct globals *const ptr_to_globals;
2096 /* At least gcc 3.4.6 on mipsel system needs optimization barrier */
2097 #define barrier() __asm__ __volatile__("":::"memory")
2098 #define SET_PTR_TO_GLOBALS(x) do { \
2099         (*(struct globals**)&ptr_to_globals) = (void*)(x); \
2100         barrier(); \
2101 } while (0)
2102 #define FREE_PTR_TO_GLOBALS() do { \
2103         if (ENABLE_FEATURE_CLEAN_UP) { \
2104                 free(ptr_to_globals); \
2105         } \
2106 } while (0)
2107
2108 /* You can change LIBBB_DEFAULT_LOGIN_SHELL, but don't use it,
2109  * use bb_default_login_shell and following defines.
2110  * If you change LIBBB_DEFAULT_LOGIN_SHELL,
2111  * don't forget to change increment constant. */
2112 #define LIBBB_DEFAULT_LOGIN_SHELL  "-/bin/sh"
2113 extern const char bb_default_login_shell[] ALIGN1;
2114 /* "/bin/sh" */
2115 #define DEFAULT_SHELL              (bb_default_login_shell+1)
2116 /* "sh" */
2117 #define DEFAULT_SHELL_SHORT_NAME   (bb_default_login_shell+6)
2118
2119 /* The following devices are the same on all systems.  */
2120 #define CURRENT_TTY "/dev/tty"
2121 #define DEV_CONSOLE "/dev/console"
2122
2123 #if defined(__FreeBSD_kernel__)
2124 # define CURRENT_VC CURRENT_TTY
2125 # define VC_1 "/dev/ttyv0"
2126 # define VC_2 "/dev/ttyv1"
2127 # define VC_3 "/dev/ttyv2"
2128 # define VC_4 "/dev/ttyv3"
2129 # define VC_5 "/dev/ttyv4"
2130 # define VC_FORMAT "/dev/ttyv%d"
2131 #elif defined(__GNU__)
2132 # define CURRENT_VC CURRENT_TTY
2133 # define VC_1 "/dev/tty1"
2134 # define VC_2 "/dev/tty2"
2135 # define VC_3 "/dev/tty3"
2136 # define VC_4 "/dev/tty4"
2137 # define VC_5 "/dev/tty5"
2138 # define VC_FORMAT "/dev/tty%d"
2139 #elif ENABLE_FEATURE_DEVFS
2140 /*Linux, obsolete devfs names */
2141 # define CURRENT_VC "/dev/vc/0"
2142 # define VC_1 "/dev/vc/1"
2143 # define VC_2 "/dev/vc/2"
2144 # define VC_3 "/dev/vc/3"
2145 # define VC_4 "/dev/vc/4"
2146 # define VC_5 "/dev/vc/5"
2147 # define VC_FORMAT "/dev/vc/%d"
2148 # define LOOP_FORMAT "/dev/loop/%u"
2149 # define LOOP_NAMESIZE (sizeof("/dev/loop/") + sizeof(int)*3 + 1)
2150 # define LOOP_NAME "/dev/loop/"
2151 # define FB_0 "/dev/fb/0"
2152 #else
2153 /*Linux, normal names */
2154 # define CURRENT_VC "/dev/tty0"
2155 # define VC_1 "/dev/tty1"
2156 # define VC_2 "/dev/tty2"
2157 # define VC_3 "/dev/tty3"
2158 # define VC_4 "/dev/tty4"
2159 # define VC_5 "/dev/tty5"
2160 # define VC_FORMAT "/dev/tty%d"
2161 # define LOOP_FORMAT "/dev/loop%u"
2162 # define LOOP_NAMESIZE (sizeof("/dev/loop") + sizeof(int)*3 + 1)
2163 # define LOOP_NAME "/dev/loop"
2164 # define FB_0 "/dev/fb0"
2165 #endif
2166
2167 // storage helpers for mk*fs utilities
2168 char BUG_wrong_field_size(void);
2169 #define STORE_LE(field, value) \
2170 do { \
2171         if (sizeof(field) == 4) \
2172                 field = SWAP_LE32((uint32_t)(value)); \
2173         else if (sizeof(field) == 2) \
2174                 field = SWAP_LE16((uint16_t)(value)); \
2175         else if (sizeof(field) == 1) \
2176                 field = (uint8_t)(value); \
2177         else \
2178                 BUG_wrong_field_size(); \
2179 } while (0)
2180
2181 #define FETCH_LE32(field) \
2182         (sizeof(field) == 4 ? SWAP_LE32(field) : BUG_wrong_field_size())
2183
2184
2185 #define ARRAY_SIZE(x) ((unsigned)(sizeof(x) / sizeof((x)[0])))
2186 #define BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
2187
2188
2189 /* We redefine ctype macros. Unicode-correct handling of char types
2190  * can't be done with such byte-oriented operations anyway,
2191  * we don't lose anything.
2192  */
2193 #undef isalnum
2194 #undef isalpha
2195 #undef isascii
2196 #undef isblank
2197 #undef iscntrl
2198 #undef isdigit
2199 #undef isgraph
2200 #undef islower
2201 #undef isprint
2202 #undef ispunct
2203 #undef isspace
2204 #undef isupper
2205 #undef isxdigit
2206 #undef toupper
2207 #undef tolower
2208
2209 /* We save ~500 bytes on isdigit alone.
2210  * BTW, x86 likes (unsigned char) cast more than (unsigned). */
2211
2212 /* These work the same for ASCII and Unicode,
2213  * assuming no one asks "is this a *Unicode* letter?" using isalpha(letter) */
2214 #define isascii(a) ((unsigned char)(a) <= 0x7f)
2215 #define isdigit(a) ((unsigned char)((a) - '0') <= 9)
2216 #define isupper(a) ((unsigned char)((a) - 'A') <= ('Z' - 'A'))
2217 #define islower(a) ((unsigned char)((a) - 'a') <= ('z' - 'a'))
2218 #define isalpha(a) ((unsigned char)(((a)|0x20) - 'a') <= ('z' - 'a'))
2219 #define isblank(a) ({ unsigned char bb__isblank = (a); bb__isblank == ' ' || bb__isblank == '\t'; })
2220 #define iscntrl(a) ({ unsigned char bb__iscntrl = (a); bb__iscntrl < ' ' || bb__iscntrl == 0x7f; })
2221 /* In POSIX/C locale isspace is only these chars: "\t\n\v\f\r" and space.
2222  * "\t\n\v\f\r" happen to have ASCII codes 9,10,11,12,13.
2223  */
2224 #define isspace(a) ({ unsigned char bb__isspace = (a) - 9; bb__isspace == (' ' - 9) || bb__isspace <= (13 - 9); })
2225 // Unsafe wrt NUL: #define ispunct(a) (strchr("!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~", (a)) != NULL)
2226 #define ispunct(a) (strchrnul("!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~", (a))[0])
2227 // Bigger code: #define isalnum(a) ({ unsigned char bb__isalnum = (a) - '0'; bb__isalnum <= 9 || ((bb__isalnum - ('A' - '0')) & 0xdf) <= 25; })
2228 #define isalnum(a) bb_ascii_isalnum(a)
2229 static ALWAYS_INLINE int bb_ascii_isalnum(unsigned char a)
2230 {
2231         unsigned char b = a - '0';
2232         if (b <= 9)
2233                 return (b <= 9);
2234         b = (a|0x20) - 'a';
2235         return b <= 'z' - 'a';
2236 }
2237 #define isxdigit(a) bb_ascii_isxdigit(a)
2238 static ALWAYS_INLINE int bb_ascii_isxdigit(unsigned char a)
2239 {
2240         unsigned char b = a - '0';
2241         if (b <= 9)
2242                 return (b <= 9);
2243         b = (a|0x20) - 'a';
2244         return b <= 'f' - 'a';
2245 }
2246 #define toupper(a) bb_ascii_toupper(a)
2247 static ALWAYS_INLINE unsigned char bb_ascii_toupper(unsigned char a)
2248 {
2249         unsigned char b = a - 'a';
2250         if (b <= ('z' - 'a'))
2251                 a -= 'a' - 'A';
2252         return a;
2253 }
2254 #define tolower(a) bb_ascii_tolower(a)
2255 static ALWAYS_INLINE unsigned char bb_ascii_tolower(unsigned char a)
2256 {
2257         unsigned char b = a - 'A';
2258         if (b <= ('Z' - 'A'))
2259                 a += 'a' - 'A';
2260         return a;
2261 }
2262
2263 /* In ASCII and Unicode, these are likely to be very different.
2264  * Let's prevent ambiguous usage from the start */
2265 #define isgraph(a) isgraph_is_ambiguous_dont_use(a)
2266 #define isprint(a) isprint_is_ambiguous_dont_use(a)
2267 /* NB: must not treat EOF as isgraph or isprint */
2268 #define isgraph_asciionly(a) ((unsigned)((a) - 0x21) <= 0x7e - 0x21)
2269 #define isprint_asciionly(a) ((unsigned)((a) - 0x20) <= 0x7e - 0x20)
2270
2271
2272 /* Simple unit-testing framework */
2273
2274 typedef void (*bbunit_testfunc)(void);
2275
2276 struct bbunit_listelem {
2277         const char* name;
2278         bbunit_testfunc testfunc;
2279 };
2280
2281 void bbunit_registertest(struct bbunit_listelem* test);
2282 void bbunit_settestfailed(void);
2283
2284 #define BBUNIT_DEFINE_TEST(NAME) \
2285         static void bbunit_##NAME##_test(void); \
2286         static struct bbunit_listelem bbunit_##NAME##_elem = { \
2287                 .name = #NAME, \
2288                 .testfunc = bbunit_##NAME##_test, \
2289         }; \
2290         static void INIT_FUNC bbunit_##NAME##_register(void) \
2291         { \
2292                 bbunit_registertest(&bbunit_##NAME##_elem); \
2293         } \
2294         static void bbunit_##NAME##_test(void)
2295
2296 /*
2297  * Both 'goto bbunit_end' and 'break' are here only to get rid
2298  * of compiler warnings.
2299  */
2300 #define BBUNIT_ENDTEST \
2301         do { \
2302                 goto bbunit_end; \
2303         bbunit_end: \
2304                 break; \
2305         } while (0)
2306
2307 #define BBUNIT_PRINTASSERTFAIL \
2308         do { \
2309                 bb_error_msg( \
2310                         "[ERROR] Assertion failed in file %s, line %d", \
2311                         __FILE__, __LINE__); \
2312         } while (0)
2313
2314 #define BBUNIT_ASSERTION_FAILED \
2315         do { \
2316                 bbunit_settestfailed(); \
2317                 goto bbunit_end; \
2318         } while (0)
2319
2320 /*
2321  * Assertions.
2322  * For now we only offer assertions which cause tests to fail
2323  * immediately. In the future 'expects' might be added too -
2324  * similar to those offered by the gtest framework.
2325  */
2326 #define BBUNIT_ASSERT_EQ(EXPECTED, ACTUAL) \
2327         do { \
2328                 if ((EXPECTED) != (ACTUAL)) { \
2329                         BBUNIT_PRINTASSERTFAIL; \
2330                         bb_error_msg("[ERROR] '%s' isn't equal to '%s'", \
2331                                                 #EXPECTED, #ACTUAL); \
2332                         BBUNIT_ASSERTION_FAILED; \
2333                 } \
2334         } while (0)
2335
2336 #define BBUNIT_ASSERT_NOTEQ(EXPECTED, ACTUAL) \
2337         do { \
2338                 if ((EXPECTED) == (ACTUAL)) { \
2339                         BBUNIT_PRINTASSERTFAIL; \
2340                         bb_error_msg("[ERROR] '%s' is equal to '%s'", \
2341                                                 #EXPECTED, #ACTUAL); \
2342                         BBUNIT_ASSERTION_FAILED; \
2343                 } \
2344         } while (0)
2345
2346 #define BBUNIT_ASSERT_NOTNULL(PTR) \
2347         do { \
2348                 if ((PTR) == NULL) { \
2349                         BBUNIT_PRINTASSERTFAIL; \
2350                         bb_error_msg("[ERROR] '%s' is NULL!", #PTR); \
2351                         BBUNIT_ASSERTION_FAILED; \
2352                 } \
2353         } while (0)
2354
2355 #define BBUNIT_ASSERT_NULL(PTR) \
2356         do { \
2357                 if ((PTR) != NULL) { \
2358                         BBUNIT_PRINTASSERTFAIL; \
2359                         bb_error_msg("[ERROR] '%s' is not NULL!", #PTR); \
2360                         BBUNIT_ASSERTION_FAILED; \
2361                 } \
2362         } while (0)
2363
2364 #define BBUNIT_ASSERT_FALSE(STATEMENT) \
2365         do { \
2366                 if ((STATEMENT)) { \
2367                         BBUNIT_PRINTASSERTFAIL; \
2368                         bb_error_msg("[ERROR] Statement '%s' evaluated to true!", \
2369                                                                 #STATEMENT); \
2370                         BBUNIT_ASSERTION_FAILED; \
2371                 } \
2372         } while (0)
2373
2374 #define BBUNIT_ASSERT_TRUE(STATEMENT) \
2375         do { \
2376                 if (!(STATEMENT)) { \
2377                         BBUNIT_PRINTASSERTFAIL; \
2378                         bb_error_msg("[ERROR] Statement '%s' evaluated to false!", \
2379                                         #STATEMENT); \
2380                         BBUNIT_ASSERTION_FAILED; \
2381                 } \
2382         } while (0)
2383
2384 #define BBUNIT_ASSERT_STREQ(STR1, STR2) \
2385         do { \
2386                 if (strcmp(STR1, STR2) != 0) { \
2387                         BBUNIT_PRINTASSERTFAIL; \
2388                         bb_error_msg("[ERROR] Strings '%s' and '%s' " \
2389                                         "are not the same", STR1, STR2); \
2390                         BBUNIT_ASSERTION_FAILED; \
2391                 } \
2392         } while (0)
2393
2394 #define BBUNIT_ASSERT_STRNOTEQ(STR1, STR2) \
2395         do { \
2396                 if (strcmp(STR1, STR2) == 0) { \
2397                         BBUNIT_PRINTASSERTFAIL; \
2398                         bb_error_msg("[ERROR] Strings '%s' and '%s' " \
2399                                         "are the same, but were " \
2400                                         "expected to differ", STR1, STR2); \
2401                         BBUNIT_ASSERTION_FAILED; \
2402                 } \
2403         } while (0)
2404
2405
2406 POP_SAVED_FUNCTION_VISIBILITY
2407
2408 #endif