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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#include <errno.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>
#include "triton.h"
#include "utils.h"
#include "memdebug.h"
extern int urandom_fd;
void __export u_inet_ntoa(in_addr_t addr, char *str)
{
addr = ntohl(addr);
sprintf(str, "%i.%i.%i.%i", (addr >> 24) & 0xff, (addr >> 16) & 0xff, (addr >> 8) & 0xff, addr & 0xff);
}
int __export u_readlong(long int *dst, const char *src,
long int min, long int max)
{
char *src_stop = NULL;
long int rv;
if (dst == NULL || src == NULL || src[0] == '\0')
return -1;
errno = 0;
rv = strtol(src, &src_stop, 0);
if (errno != 0 || *src_stop != '\0' || rv < min || rv > max) {
return -1;
} else {
*dst = rv;
return 0;
}
}
int __export u_parse_ip4addr(const char *src, struct in_addr *addr,
const char **err_msg)
{
struct addrinfo hint = {
.ai_flags = AI_NUMERICHOST,
.ai_family = AF_INET,
.ai_socktype = 0,
.ai_protocol = 0,
};
struct addrinfo *ainfo;
int err;
err = getaddrinfo(src, NULL, &hint, &ainfo);
if (err) {
*err_msg = gai_strerror(err);
return err;
}
*addr = ((struct sockaddr_in *)ainfo->ai_addr)->sin_addr;
freeaddrinfo(ainfo);
return 0;
}
int __export u_randbuf(void *buf, size_t buf_len, int *err)
{
uint8_t *u8buf = buf;
ssize_t rd_len;
while (buf_len) {
rd_len = read(urandom_fd, u8buf, buf_len);
if (rd_len < 0) {
if (errno == EINTR)
rd_len = 0;
else {
if (err)
*err = errno;
return -1;
}
} else if (rd_len == 0) {
if (err)
*err = 0;
return -1;
}
u8buf += rd_len;
buf_len -= rd_len;
}
return 0;
}
|
