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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#ifndef __CLI_P_H
#define __CLI_P_H
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include "triton.h"
struct cli_client_t
{
uint8_t *cmdline;
int (*send)(struct cli_client_t *, const void *buf, int size);
int (*sendv)(struct cli_client_t *, const char *fmt, va_list ap);
void (*disconnect)(struct cli_client_t *);
};
int cli_process_cmd(struct cli_client_t *cln);
/* Format peer address (IPv4, IPv6 or IPv4-mapped IPv6) for logging.
* buf must be at least INET6_ADDRSTRLEN bytes long. */
static inline const char *cli_addr_str(const struct sockaddr_storage *addr,
char *buf, size_t size)
{
const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *)addr;
const struct sockaddr_in *sin = (const struct sockaddr_in *)addr;
buf[0] = '\0';
if (addr->ss_family == AF_INET6) {
if (IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
inet_ntop(AF_INET, &sin6->sin6_addr.s6_addr32[3], buf, size);
else
inet_ntop(AF_INET6, &sin6->sin6_addr, buf, size);
} else
inet_ntop(AF_INET, &sin->sin_addr, buf, size);
return buf;
}
/* Parse "host:port", "[host]:port" or ":port" listener specification.
* str is modified in place, *host points into str afterwards (NULL for
* empty host). For unbracketed hosts the last ':' separates the port,
* so bare IPv6 addresses like "::1:2001" are accepted too.
* Returns 0 on success, -1 on invalid format. */
static inline int cli_parse_hostport(char *str, const char **host, int *port)
{
char *d;
if (*str == '[') {
++str;
d = strchr(str, ']');
if (!d || d[1] != ':')
return -1;
*d++ = '\0';
} else {
d = strrchr(str, ':');
if (!d)
return -1;
}
*d = '\0';
*port = atoi(d + 1);
if (*port <= 0)
return -1;
*host = *str ? str : NULL;
return 0;
}
/* Fill sockaddr for binding a CLI listener. host may be an IPv4 or IPv6
* address; NULL host means any IPv4 address (use "::" for IPv6 wildcard).
* Returns 0 on success, -1 if host is not a valid address. */
static inline int cli_bind_addr(const char *host, int port,
struct sockaddr_storage *addr, socklen_t *len)
{
struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)addr;
struct sockaddr_in *sin = (struct sockaddr_in *)addr;
memset(addr, 0, sizeof(*addr));
if (host && inet_pton(AF_INET6, host, &sin6->sin6_addr) > 0) {
sin6->sin6_family = AF_INET6;
sin6->sin6_port = htons(port);
*len = sizeof(*sin6);
} else {
sin->sin_family = AF_INET;
sin->sin_port = htons(port);
if (!host)
sin->sin_addr.s_addr = htonl(INADDR_ANY);
else if (inet_pton(AF_INET, host, &sin->sin_addr) <= 0)
return -1;
*len = sizeof(*sin);
}
return 0;
}
extern char *conf_cli_passwd;
extern char *conf_cli_prompt;
#endif
|
