libbb.h 86 KB

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