1 /* vi: set sw=4 ts=4: */
5 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6 * Copyright (C) 2006 Rob Landley
7 * Copyright (C) 2006 Denys Vlasenko
9 * Licensed under GPLv2, see file LICENSE in this source tree.
12 /* We need to have separate xfuncs.c and xfuncs_printf.c because
13 * with current linkers, even with section garbage collection,
14 * if *.o module references any of XXXprintf functions, you pull in
15 * entire printf machinery. Even if you do not use the function
16 * which uses XXXprintf.
18 * xfuncs.c contains functions (not necessarily xfuncs)
19 * which do not pull in printf, directly or indirectly.
20 * xfunc_printf.c contains those which do.
22 * TODO: move xmalloc() and xatonum() here.
27 /* Turn on nonblocking I/O on a fd */
28 int FAST_FUNC
ndelay_on(int fd
)
30 return fcntl(fd
, F_SETFL
, fcntl(fd
, F_GETFL
) | O_NONBLOCK
);
33 int FAST_FUNC
ndelay_off(int fd
)
35 return fcntl(fd
, F_SETFL
, fcntl(fd
, F_GETFL
) & ~O_NONBLOCK
);
38 int FAST_FUNC
close_on_exec_on(int fd
)
40 return fcntl(fd
, F_SETFD
, FD_CLOEXEC
);
43 char* FAST_FUNC
strncpy_IFNAMSIZ(char *dst
, const char *src
)
46 enum { IFNAMSIZ
= 16 };
48 return strncpy(dst
, src
, IFNAMSIZ
);
52 /* Convert unsigned integer to ascii, writing into supplied buffer.
53 * A truncated result contains the first few digits of the result ala strncpy.
54 * Returns a pointer past last generated digit, does _not_ store NUL.
56 void BUG_sizeof(void);
57 char* FAST_FUNC
utoa_to_buf(unsigned n
, char *buf
, unsigned buflen
)
64 // 2^32-1 = 4294967295
66 #if UINT_MAX > 4294967295 /* prevents warning about "const too large" */
69 // 2^64-1 = 18446744073709551615
70 i
= 10000000000000000000;
77 if (res
|| out
|| i
== 1) {
88 /* Convert signed integer to ascii, like utoa_to_buf() */
89 char* FAST_FUNC
itoa_to_buf(int n
, char *buf
, unsigned buflen
)
98 return utoa_to_buf((unsigned)n
, buf
, buflen
);
101 // The following two functions use a static buffer, so calling either one a
102 // second time will overwrite previous results.
104 // The largest 32 bit integer is -2 billion plus NUL, or 1+10+1=12 bytes.
105 // It so happens that sizeof(int) * 3 is enough for 32+ bit ints.
106 // (sizeof(int) * 3 + 2 is correct for any width, even 8-bit)
108 static char local_buf
[sizeof(int) * 3];
110 /* Convert unsigned integer to ascii using a static buffer (returned). */
111 char* FAST_FUNC
utoa(unsigned n
)
113 *(utoa_to_buf(n
, local_buf
, sizeof(local_buf
) - 1)) = '\0';
118 /* Convert signed integer to ascii using a static buffer (returned). */
119 char* FAST_FUNC
itoa(int n
)
121 *(itoa_to_buf(n
, local_buf
, sizeof(local_buf
) - 1)) = '\0';
126 /* Emit a string of hex representation of bytes */
127 char* FAST_FUNC
bin2hex(char *p
, const char *cp
, int count
)
130 unsigned char c
= *cp
++;
131 /* put lowercase hex digits */
132 *p
++ = 0x20 | bb_hexdigits_upcase
[c
>> 4];
133 *p
++ = 0x20 | bb_hexdigits_upcase
[c
& 0xf];
139 /* Convert "[x]x[:][x]x[:][x]x[:][x]x" hex string to binary, no more than COUNT bytes */
140 char* FAST_FUNC
hex2bin(char *dst
, const char *str
, int count
)
143 while (*str
&& count
) {
148 else if ((c
|0x20) >= 'a' && (c
|0x20) <= 'f')
149 val
= (c
|0x20) - ('a' - 10);
156 else if ((c
|0x20) >= 'a' && (c
|0x20) <= 'f')
157 val
|= (c
|0x20) - ('a' - 10);
158 else if (c
== ':' || c
== '\0')
170 errno
= (*str
? ERANGE
: 0);
174 /* Return how long the file at fd is, if there's any way to determine it. */
176 off_t FAST_FUNC
fdlength(int fd
)
178 off_t bottom
= 0, top
= 0, pos
;
181 // If the ioctl works for this, return it.
183 if (ioctl(fd
, BLKGETSIZE
, &size
) >= 0) return size
*512;
185 // FIXME: explain why lseek(SEEK_END) is not used here!
187 // If not, do a binary search for the last location we can read. (Some
188 // block devices don't do BLKGETSIZE right.)
193 pos
= bottom
+ (top
- bottom
) / 2;
195 // If we can read from the current location, it's bigger.
197 if (lseek(fd
, pos
, SEEK_SET
)>=0 && safe_read(fd
, &temp
, 1)==1) {
198 if (bottom
== top
) bottom
= top
= (top
+1) * 2;
201 // If we can't, it's smaller.
210 } while (bottom
+ 1 != top
);
216 int FAST_FUNC
bb_putchar_stderr(char ch
)
218 return write(STDERR_FILENO
, &ch
, 1);
221 ssize_t FAST_FUNC
full_write1_str(const char *str
)
223 return full_write(STDOUT_FILENO
, str
, strlen(str
));
226 ssize_t FAST_FUNC
full_write2_str(const char *str
)
228 return full_write(STDERR_FILENO
, str
, strlen(str
));
231 static int wh_helper(int value
, int def_val
, const char *env_name
, int *err
)
234 char *s
= getenv(env_name
);
237 /* If LINES/COLUMNS are set, pretent that there is
238 * no error getting w/h, this prevents some ugly
239 * cursor tricks by our callers */
243 if (value
<= 1 || value
>= 30000)
248 /* It is perfectly ok to pass in a NULL for either width or for
249 * height, in which case that value will not be set. */
250 int FAST_FUNC
get_terminal_width_height(int fd
, unsigned *width
, unsigned *height
)
257 /* I've seen ioctl returning 0, but row/col is (still?) 0.
258 * We treat that as an error too. */
259 err
= ioctl(fd
, TIOCGWINSZ
, &win
) != 0 || win
.ws_row
== 0;
261 *height
= wh_helper(win
.ws_row
, 24, "LINES", &err
);
263 *width
= wh_helper(win
.ws_col
, 80, "COLUMNS", &err
);
267 int FAST_FUNC
tcsetattr_stdin_TCSANOW(const struct termios
*tp
)
269 return tcsetattr(STDIN_FILENO
, TCSANOW
, tp
);
272 pid_t FAST_FUNC
safe_waitpid(pid_t pid
, int *wstat
, int options
)
277 r
= waitpid(pid
, wstat
, options
);
278 while ((r
== -1) && (errno
== EINTR
));
282 pid_t FAST_FUNC
wait_any_nohang(int *wstat
)
284 return safe_waitpid(-1, wstat
, WNOHANG
);
287 // Wait for the specified child PID to exit, returning child's error return.
288 int FAST_FUNC
wait4pid(pid_t pid
)
293 /*errno = ECHILD; -- wrong. */
294 /* we expect errno to be already set from failed [v]fork/exec */
297 if (safe_waitpid(pid
, &status
, 0) == -1)
299 if (WIFEXITED(status
))
300 return WEXITSTATUS(status
);
301 if (WIFSIGNALED(status
))
302 return WTERMSIG(status
) + 0x180;
306 char * FAST_FUNC
unparse_uuid(const uint8_t *uu
, char *out
)
308 char uuid_string
[32];
310 bin2hex(uuid_string
, (char *)uu
, 16);
311 /* f.e. UUID=dfd9c173-be52-4d27-99a5-c34c6c2ff55f */
312 sprintf(out
, "%.8s-%.4s-%.4s-%.4s-%.12s",
322 static unsigned char fromhex(char c
)
326 return ((c
|0x20) - 'a' + 10);
329 /* Parse & verify UUID string */
330 int FAST_FUNC
parse_uuid(const char *s
, uint8_t *uuid
)
334 if (strlen(s
) != 36 || s
[8] != '-' || s
[13] != '-'
335 || s
[18] != '-' || s
[23] != '-'
339 for (i
= 0; i
< 16; i
++) {
342 if (!isxdigit(s
[0]) || !isxdigit(s
[1]))
344 uuid
[i
] = ((fromhex(s
[0]) << 4) | fromhex(s
[1]));