libbb.h 99 KB

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