#include <errno.h>
#include <stdarg.h>
#include <stdio_ext.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#if HAVE_GETRANDOM
# include <sys/random.h>
#endif
#ifdef HAVE_ERROR_H
# include <error.h>
#else
void error(int status, int errnum, const char *format, ...)
{
va_list ap;
fprintf(stderr, "%s: ", program_invocation_short_name);
va_start(ap, format);
vfprintf(stderr, format, ap);
va_end(ap);
if (errnum)
fprintf(stderr, ": %s\n", strerror(errnum));
else
fprintf(stderr, "\n");
if (status)
exit(status);
}
#endif
int close_stream(FILE *stream)
{
const int flush_status = fflush(stream);
#ifdef HAVE___FPENDING
const int some_pending = (__fpending(stream) != 0);
#endif
const int prev_fail = (ferror(stream) != 0);
const int fclose_fail = (fclose(stream) != 0);
if (flush_status ||
prev_fail || (fclose_fail && (
#ifdef HAVE___FPENDING
some_pending ||
#endif
errno != EBADF))) {
if (!fclose_fail && !(errno == EPIPE))
errno = 0;
return EOF;
}
return 0;
}
void close_stdout(void)
{
if (close_stream(stdout) != 0 && !(errno == EPIPE)) {
if (errno)
error(0, errno, "write error");
else
error(0, 0, "write error");
_exit(EXIT_FAILURE);
}
if (close_stream(stderr) != 0)
_exit(EXIT_FAILURE);
}
long strtol_or_err(char const *const str, char const *const errmesg,
const long min, const long max)
{
long num;
char *end = NULL;
errno = 0;
if (str == NULL || *str == '\0')
goto err;
num = strtol(str, &end, 10);
if (errno || str == end || (end && *end))
goto err;
if (num < min || max < num)
error(EXIT_FAILURE, 0, "%s: '%s': out of range: %lu <= value <= %lu",
errmesg, str, min, max);
return num;
err:
error(EXIT_FAILURE, errno, "%s: '%s'", errmesg, str);
abort();
}
static unsigned int iputil_srand_fallback(void)
{
struct timespec ts;
clock_gettime(CLOCK_REALTIME, &ts);
return ((getpid() << 16) ^ getuid() ^ ts.tv_sec ^ ts.tv_nsec);
}
void iputils_srand(void)
{
unsigned int i;
#if HAVE_GETRANDOM
ssize_t ret;
do {
errno = 0;
ret = getrandom(&i, sizeof(i), GRND_NONBLOCK);
switch (errno) {
case 0:
break;
case EINTR:
continue;
default:
i = iputil_srand_fallback();
goto done;
}
} while (ret != sizeof(i));
done:
#else
i = iputil_srand_fallback();
#endif
srand(i);
/* Consume up to 31 random numbers */
i = rand() & 0x1F;
while (0 < i) {
rand();
i--;
}
}
void timespecsub(struct timespec *a, struct timespec *b, struct timespec *res)
{
res->tv_sec = a->tv_sec - b->tv_sec;
res->tv_nsec = a->tv_nsec - b->tv_nsec;
if (res->tv_nsec < 0) {
res->tv_sec--;
res->tv_nsec += 1000000000L;
}
}