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