int
pfr_buf_load(struct pfr_buffer *b, char *file, int nonetwork)
{
FILE *fp;
char buf[BUF_SIZE];
int rv;
int ev = 0;
if (file == NULL)
return (0);
if (!strcmp(file, "-"))
fp = stdin;
else {
fp = pfctl_fopen(file, "r");
if (fp == NULL)
return (-1);
}
while ((rv = pfr_next_token(buf, fp)) == 1)
if ((ev = append_addr(b, buf, nonetwork)) == -1) {
rv = -1;
break;
}
if (ev == 1) /* expected further append_addr call */
rv = -1;
if (fp != stdin)
fclose(fp);
return (rv);
}
int
pfr_buf_load(struct pfr_buffer *b, char *file, int nonetwork,
int (*append_addr)(struct pfr_buffer *, char *, int))
{
FILE *fp;
char buf[BUF_SIZE];
int rv;
if (file == NULL)
return (0);
if (!strcmp(file, "-"))
fp = stdin;
else {
fp = pfctl_fopen(file, "r");
if (fp == NULL)
return (-1);
}
while ((rv = pfr_next_token(buf, fp)) == 1)
if (append_addr(b, buf, nonetwork)) {
rv = -1;
break;
}
if (fp != stdin)
fclose(fp);
return (rv);
}
int
main(int argc, char *argv[])
{
int c;
char *interface = NULL;
struct sockaddr_in *t_addr;
struct sockaddr_in6 *t_addr6;
struct sockaddr *tmp_addrs = NULL;
/* Parse the arguments. */
while ((c = getopt(argc, argv, ":H:L:P:S:a:h:p:c:d:lm:sx:X:o:t:M:r:w:Di:TB:C:O:")) >= 0 ) {
switch (c) {
case 'H':
local_host = optarg;
break;
case 'L':
role = MIXED;
listeners = atoi(optarg);
if (listeners > MAX_POLL_SKS) {
usage(argv[0]);
exit(1);
}
break;
case 'P':
local_port = atoi(optarg);
break;
case 'S':
role = MIXED;
tosend = atoi(optarg);
if (tosend > MAX_POLL_SKS) {
usage(argv[0]);
exit(1);
}
break;
case 'a':
assoc_pattern = atoi(optarg);
if (assoc_pattern < ASSOC_PATTERN_SEQUENTIAL
|| assoc_pattern > ASSOC_PATTERN_RANDOM ) {
usage(argv[0]);
exit(1);
}
break;
case 'h':
remote_host = optarg;
break;
case 'D':
drain = 1;
do_exit = 0;
break;
case 'p':
remote_port = atoi(optarg);
break;
case 's':
if (role != NOT_DEFINED) {
printf("%s: only -s or -l\n", argv[0]);
usage(argv[0]);
exit(1);
}
role = CLIENT;
break;
case 'l':
if (role != NOT_DEFINED) {
printf("%s: only -s or -l\n", argv[0]);
usage(argv[0]);
exit(1);
}
role = SERVER;
break;
case 'd':
debug_level = atoi(optarg);
if (debug_level < DEBUG_NONE
|| debug_level > DEBUG_MAX) {
usage(argv[0]);
exit(1);
}
break;
case 'x':
repeat = atoi(optarg);
if (!repeat) {
xflag = 1;
repeat = BIG_REPEAT;
}
break;
case 'X':
msg_cnt = atoi(optarg);
if ((msg_cnt <= 0) || (msg_cnt > MSG_CNT)) {
usage(argv[0]);
exit(1);
}
break;
case 'c':
test_case = atoi(optarg);
if (test_case > NCASES) {
usage(argv[0]);
exit(1);
}
if (test_case < 0) {
size_arg = -test_case;
}
break;
case 'o':
order_pattern = atoi(optarg);
if (order_pattern < ORDER_PATTERN_UNORDERED
|| order_pattern > ORDER_PATTERN_RANDOM ) {
usage(argv[0]);
exit(1);
}
break;
case 'O':
timetolive = atoi(optarg);
if (timetolive < 0) {
usage(argv[0]);
exit(1);
}
break;
case 't':
stream_pattern = atoi(optarg);
if (stream_pattern < STREAM_PATTERN_SEQUENTIAL
|| stream_pattern > STREAM_PATTERN_RANDOM ) {
usage(argv[0]);
exit(1);
}
break;
case 'M':
max_stream = atoi(optarg);
if (max_stream < 0
|| max_stream >= (1<<16)) {
usage(argv[0]);
exit(1);
}
break;
case 'r':
seed = atoi(optarg);
break;
case 'm':
max_msgsize = atoi(optarg);
#if 0
if ((max_msgsize < DEFAULT_MIN_WINDOW) ||
(max_msgsize > 65515)) {
usage(argv[0]);
exit(1);
}
#endif
break;
case 'i':
interface = optarg;
if_index = if_nametoindex(interface);
if (!if_index) {
printf("Interface %s unknown\n", interface);
exit(1);
}
break;
case 'T':
socket_type = SOCK_STREAM;
break;
case 'B':
tmp_addrs = append_addr(optarg, bindx_add_addrs,
&bindx_add_count);
if (NULL == tmp_addrs) {
fprintf(stderr, "No memory to add ");
fprintf(stderr, "%s\n", optarg);
exit(1);
}
bindx_add_addrs = tmp_addrs;
break;
case 'C':
tmp_addrs = append_addr(optarg, connectx_addrs,
&connectx_count);
if (NULL == tmp_addrs) {
fprintf(stderr, "No memory to add ");
fprintf(stderr, "%s\n", optarg);
exit(1);
}
connectx_addrs = tmp_addrs;
break;
case '?':
default:
usage(argv[0]);
exit(0);
}
} /* while() */
if (NOT_DEFINED == role) {
usage(argv[0]);
exit(1);
}
if (SERVER == role && NULL == local_host && remote_host != NULL) {
fprintf (stderr, "%s: Server needs local address, "
"not remote address\n", argv[0]);
usage(argv[0]);
exit(1);
}
if (CLIENT == role && NULL == remote_host && connectx_count == 0) {
fprintf (stderr, "%s: Client needs at least remote address "
"& port\n", argv[0]);
usage(argv[0]);
exit(1);
}
if (MIXED == role) {
if (listeners && NULL == local_host) {
fprintf (stderr, "%s: Servers need local address\n",
argv[0]);
usage(argv[0]);
exit(1);
}
if (tosend && NULL == remote_host) {
fprintf (stderr, "%s: Clients need remote address ",
argv[0]);
fprintf (stderr, "& port\n");
usage(argv[0]);
exit(1);
}
}
if (optind < argc) {
fprintf(stderr, "%s: non-option arguments are illegal: ",
argv[0]);
while (optind < argc)
fprintf(stderr, "%s ", argv[optind++]);
fprintf (stderr, "\n");
usage(argv[0]);
exit(1);
}
if (remote_host != NULL && connectx_count != 0) {
fprintf(stderr, "%s: You can not provide both -h and -C options.\n",
argv[0]);
usage(argv[0]);
exit(1);
}
if (remote_host != NULL && remote_port != 0) {
struct addrinfo *res;
int error;
char *host_s, *serv_s;
if ((host_s = malloc(NI_MAXHOST)) == NULL) {
fprintf(stderr, "\n*** host_s malloc failed!!! ***\n");
exit(1);
}
if ((serv_s = malloc(NI_MAXSERV)) == NULL) {
fprintf(stderr, "\n*** serv_s malloc failed!!! ***\n");
exit(1);
}
error = getaddrinfo(remote_host, 0, NULL, &res);
if (error) {
printf("%s.\n", gai_strerror(error));
usage(argv[0]);
exit(1);
}
switch (res->ai_family) {
case AF_INET:
t_addr = (struct sockaddr_in *)&s_rem;
t_addr->sin_family = AF_INET;
t_addr->sin_port = htons(remote_port);
inet_pton(AF_INET, remote_host,
&t_addr->sin_addr);
r_len = sizeof (struct sockaddr_in);
#ifdef __FreeBSD__
t_addr->sin_len = r_len;
#endif
break;
case AF_INET6:
t_addr6 = (struct sockaddr_in6 *)&s_rem;
if (interface)
t_addr6->sin6_scope_id =
if_nametoindex(interface);
t_addr6->sin6_family = AF_INET6;
t_addr6->sin6_port = htons(remote_port);
inet_pton(AF_INET6, remote_host,
&t_addr6->sin6_addr);
r_len = sizeof (struct sockaddr_in6);
#ifdef __FreeBSD__
t_addr6->sin6_len = r_len;
#endif
break;
}
getnameinfo((struct sockaddr *)&s_rem, r_len, host_s,
NI_MAXHOST, serv_s, NI_MAXSERV, NI_NUMERICHOST);
DEBUG_PRINT(DEBUG_MAX, "remote:addr=%s, port=%s, family=%d\n",
host_s, serv_s, res->ai_family);
}
if (connectx_count != 0) {
switch (connectx_addrs->sa_family) {
case AF_INET:
t_addr = (struct sockaddr_in *)&s_rem;
r_len = sizeof(struct sockaddr_in);
memcpy(t_addr, connectx_addrs, r_len);
t_addr->sin_port = htons(remote_port);
break;
case AF_INET6:
t_addr6 = (struct sockaddr_in6 *)&s_rem;
r_len = sizeof(struct sockaddr_in6);
memcpy(t_addr6, connectx_addrs, r_len);
t_addr6->sin6_port = htons(remote_port);
break;
}
}
if (local_host != NULL) {
struct addrinfo *res;
int error;
char *host_s, *serv_s;
struct sockaddr_in *t_addr;
struct sockaddr_in6 *t_addr6;
if ((host_s = malloc(NI_MAXHOST)) == NULL) {
fprintf(stderr, "\n*** host_s malloc failed!!! ***\n");
exit(1);
}
if ((serv_s = malloc(NI_MAXSERV)) == NULL) {
fprintf(stderr, "\n*** serv_s malloc failed!!! ***\n");
exit(1);
}
if (strcmp(local_host, "0") == 0)
local_host = "0.0.0.0";
error = getaddrinfo(local_host, 0, NULL, &res);
if (error) {
printf("%s.\n", gai_strerror(error));
usage(argv[0]);
exit(1);
}
switch (res->ai_family) {
case AF_INET:
t_addr = (struct sockaddr_in *)&s_loc;
t_addr->sin_family = AF_INET;
t_addr->sin_port = htons(local_port);
inet_pton(AF_INET, local_host,
&t_addr->sin_addr);
l_len = sizeof (struct sockaddr_in);
#ifdef __FreeBSD__
t_addr->sin_len = l_len;
#endif
break;
case AF_INET6:
t_addr6 = (struct sockaddr_in6 *)&s_loc;
if (interface)
t_addr6->sin6_scope_id =
if_nametoindex(interface);
t_addr6->sin6_family = AF_INET6;
t_addr6->sin6_port = htons(local_port);
inet_pton(AF_INET6, local_host,
&t_addr6->sin6_addr);
l_len = sizeof (struct sockaddr_in6);
#ifdef __FreeBSD__
t_addr6->sin6_len = l_len;
#endif
break;
}
error = getnameinfo((struct sockaddr *)&s_loc, l_len, host_s,
NI_MAXHOST, serv_s, NI_MAXSERV, NI_NUMERICHOST);
if (error)
printf("%s..\n", gai_strerror(error));
DEBUG_PRINT(DEBUG_MAX, "local:addr=%s, port=%s, family=%d\n",
host_s, serv_s, res->ai_family);
}
/* A half-hearted attempt to seed rand() */
if (seed == 0 ) {
seed = time(0);
DEBUG_PRINT(DEBUG_NONE, "seed = %d\n", seed);
}
srand(seed);
/* Let the testing begin. */
start_test(role);
return 0;
} /* main() */
si_item_t *
si_ipnode_byname(si_mod_t *si, const char *name, int family, int flags, const char *interface, uint32_t *err)
{
int i, status, want;
uint32_t if4, if6;
struct in_addr addr4;
struct in6_addr addr6;
si_item_t *item4, *item6;
build_hostent_t *out;
struct hostent *h;
uint32_t unused;
memset(&addr4, 0, sizeof(struct in_addr));
memset(&addr6, 0, sizeof(struct in6_addr));
if (err != NULL) *err = 0;
if (family == AF_INET)
{
status = inet_aton(name, &addr4);
if (status == 1)
{
/* create a host entry */
item4 = make_hostent(si, name, addr4);
if (item4 == NULL)
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
return item4;
}
}
else if (family == AF_INET6)
{
status = inet_pton(family, name, &addr6);
if (status == 1)
{
/* create a host entry */
item6 = make_hostent6(si, name, addr6);
if (item6 == NULL)
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
return item6;
}
status = inet_aton(name, &addr4);
if (status == 1)
{
if (!(flags & (AI_V4MAPPED | AI_V4MAPPED_CFG)))
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND;
return NULL;
}
addr6.__u6_addr.__u6_addr32[0] = 0x00000000;
addr6.__u6_addr.__u6_addr32[1] = 0x00000000;
addr6.__u6_addr.__u6_addr32[2] = htonl(0x0000ffff);
memmove(&(addr6.__u6_addr.__u6_addr32[3]), &(addr4.s_addr), IPV4_ADDR_LEN);
/* create a host entry */
item6 = make_hostent6(si, name, addr6);
if (item6 == NULL)
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
return item6;
}
}
else
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
/*
* IF AI_ADDRCONFIG is set, we need to know what interface flavors we really have.
*/
if4 = 0;
if6 = 0;
if (flags & AI_ADDRCONFIG)
{
if (si_inet_config(&if4, &if6) < 0)
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
/* Bail out if there are no interfaces */
if ((if4 == 0) && (if6 == 0))
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
}
/*
* Figure out what we want.
* If user asked for AF_INET, we only want V4 addresses.
*/
want = WANT_A4_ONLY;
if (family == AF_INET)
{
if ((flags & AI_ADDRCONFIG) && (if4 == 0))
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
}
else if (family == AF_INET6)
{
/* family == AF_INET6 */
want = WANT_A6_ONLY;
if (flags & (AI_V4MAPPED | AI_V4MAPPED_CFG))
{
if (flags & AI_ALL)
{
want = WANT_A6_PLUS_MAPPED_A4;
}
else
{
want = WANT_A6_OR_MAPPED_A4_IF_NO_A6;
}
}
else
{
if ((flags & AI_ADDRCONFIG) && (if6 == 0))
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
}
}
else
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
item6 = NULL;
item4 = NULL;
/* fetch IPv6 data if required */
if ((want == WANT_A6_ONLY) || (want == WANT_A6_OR_MAPPED_A4_IF_NO_A6) || (want == WANT_A6_PLUS_MAPPED_A4))
{
item6 = si_host_byname(si, name, AF_INET6, interface, (uint32_t *)err);
}
/* fetch IPv4 data if required */
if ((want == WANT_A4_ONLY) || (want == WANT_A6_PLUS_MAPPED_A4) || ((want == WANT_A6_OR_MAPPED_A4_IF_NO_A6) && (item6 == NULL)))
{
item4 = si_host_byname(si, name, AF_INET, interface, (uint32_t *)err);
}
if (want == WANT_A4_ONLY)
{
si_item_release(item6);
if ((item4 == NULL) && (err != NULL)) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND;
return item4;
}
if (want == WANT_A6_ONLY)
{
si_item_release(item4);
if ((item6 == NULL) && (err != NULL)) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND;
return item6;
}
if ((item6 == NULL) && (item4 == NULL))
{
if (err != NULL) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND;
return NULL;
}
/* output item will have IPv6 + mapped IPv4 addresses */
out = (build_hostent_t *)calloc(1, sizeof(build_hostent_t));
if (out == NULL)
{
si_item_release(item4);
si_item_release(item6);
if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY;
return NULL;
}
if (item4 != NULL)
{
h = (struct hostent *)((uintptr_t)item4 + sizeof(si_item_t));
out->host.h_name = lower_case(h->h_name);
if (h->h_aliases != NULL)
{
for (i = 0; h->h_aliases[i] != NULL; i++) merge_alias(h->h_aliases[i], out);
}
for (i = 0; h->h_addr_list[i] != 0; i++)
{
addr6.__u6_addr.__u6_addr32[0] = 0x00000000;
addr6.__u6_addr.__u6_addr32[1] = 0x00000000;
addr6.__u6_addr.__u6_addr32[2] = htonl(0x0000ffff);
memmove(&(addr6.__u6_addr.__u6_addr32[3]), h->h_addr_list[i], IPV4_ADDR_LEN);
append_addr((const char *)&addr6, IPV6_ADDR_LEN, out);
}
}
if (item6 != NULL)
{
h = (struct hostent *)((uintptr_t)item6 + sizeof(si_item_t));
if (out->host.h_name == NULL) out->host.h_name = lower_case(h->h_name);
if (h->h_aliases != NULL)
{
for (i = 0; h->h_aliases[i] != NULL; i++) merge_alias(h->h_aliases[i], out);
}
for (i = 0; h->h_addr_list[i] != 0; i++) append_addr(h->h_addr_list[i], IPV6_ADDR_LEN, out);
}
si_item_release(item4);
si_item_release(item6);
unused = 0;
item6 = (si_item_t *)LI_ils_create("L4444s*44c", (unsigned long)si, CATEGORY_HOST_IPV6, 1, unused, unused, out->host.h_name, out->host.h_aliases, AF_INET6, IPV6_ADDR_LEN, out->host.h_addr_list);
free_build_hostent(out);
return item6;
}
