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