1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <err.h>
#include <sys/ioctl.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include "utils.h"
void
set_nonblocking(int sock)
{
int on = 1;
if (ioctl(sock, FIONBIO, &on) == -1)
err(EXIT_FAILURE, "ioctl");
}
void
set_nodelay(int fd)
{
int nodelay = 1;
if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &nodelay, sizeof(nodelay)) == -1)
err(EXIT_FAILURE, "ioctl");
}
void
write_all(int fd, const void *buf, size_t buf_size)
{
const unsigned char *p = (const unsigned char *) buf;
while (buf_size) {
ssize_t written = write(fd, p, buf_size);
if (written <= 0)
exit(EXIT_FAILURE);
buf_size -= written;
p += written;
}
}
int
parse_value(const char *s, int min, int max, const unit *units)
{
char *end = NULL;
double val;
errno = 0;
val = strtod(s, &end);
if (errno != 0) {
fprintf(stderr, "Invalid value %s\n", s);
exit(EXIT_FAILURE);
}
if (end == s) {
fprintf(stderr, "No digit in %s\n", s);
exit(EXIT_FAILURE);
}
for (; units->suffix; ++units) {
if (strcmp(end, units->suffix) != 0)
continue;
val *= units->scale;
if (val < min || val > max) {
fprintf(stderr, "Value %s out of range [%d-%d]\n", s, min, max);
exit(EXIT_FAILURE);
}
return (int) val;
}
fprintf(stderr, "Wrong format %s\n", s);
exit(EXIT_FAILURE);
}
|
