3
0

libbb.h 98 KB

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