68a8db1efb0cd13e0e3d59362ab9db1a4a8bc496Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews#define BIND_8_COMPAT /* Pull in <arpa/nameser_compat.h> */

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <sys/time.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <sys/types.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <sys/socket.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <stdio.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <stdlib.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <string.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <limits.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <time.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <unistd.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <netdb.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <netinet/in.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <arpa/nameser.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <resolv.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <math.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#include <errno.h>

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

9db88ce113701a6fb600a67aaa65764d800e25b0Mark Andrews#ifdef HAVE_CONFIG_H

9db88ce113701a6fb600a67aaa65764d800e25b0Mark Andrews#include "config.h"

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#ifndef HAVE_GETADDRINFO

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#include "missing/addrinfo.h"

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#endif

9db88ce113701a6fb600a67aaa65764d800e25b0Mark Andrews#endif

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define DEF_MAX_QUERIES_OUTSTANDING 20

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define DEF_QUERY_TIMEOUT 5 /* in seconds */

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#define DEF_SERVER_TO_QUERY "127.0.0.1"

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#define DEF_SERVER_PORT "53"

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define DEF_BUFFER_SIZE 32 /* in k */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉#define DEF_RTTARRAY_SIZE 50000

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉#define DEF_RTTARRAY_UNIT 100 /* in usec */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define COMMENT_CHAR ';'

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define CONFIG_CHAR '#'

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define MAX_PORT 65535

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define MAX_INPUT_LEN 512

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define MAX_DOMAIN_LEN 255

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define MAX_BUFFER_LEN 8192 /* in bytes */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define HARD_TIMEOUT_EXTRA 5 /* in seconds */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define RESPONSE_BLOCKING_WAIT_TIME 0.1 /* in seconds */

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews#define EDNSLEN 11

384fef003441df4067b773688237700513390f9eEvan Hunt#define DNS_HEADERLEN 12

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define FALSE 0

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define TRUE 1

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define WHITESPACE " \t\n"

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonenum directives_enum { V_SERVER, V_PORT, V_MAXQUERIES, V_MAXWAIT };

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define DIRECTIVES { "server", "port", "maxqueries", "maxwait" }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define DIR_VALUES { V_SERVER, V_PORT, V_MAXQUERIES, V_MAXWAIT }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define QTYPE_STRINGS { \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "A", "NS", "MD", "MF", "CNAME", "SOA", "MB", "MG", "MR", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "NULL", "WKS", "PTR", "HINFO", "MINFO", "MX", "TXT", "RP", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "AFSDB", "X25", "ISDN", "RT", "NSAP", "NSAP-PTR", "SIG", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "KEY", "PX", "GPOS", "AAAA", "LOC", "NXT", "EID", "NIMLOC", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "SRV", "ATMA", "NAPTR", "KX", "CERT", "A6", "DNAME", "SINK", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "OPT", "APL", "DS", "SSHFP", "IPSECKEY", "RRSIG", "NSEC", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "DNSKEY", "DHCID", "NSEC3", "NSEC3PARAM", "TLSA", "HIP", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "NINFO", "RKEY", "TALINK", "CDS", "SPF", "UINFO", "UID", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "GID", "UNSPEC", "NID", "L32", "L64", "LP", "TKEY", "TSIG", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "IXFR", "AXFR", "MAILB", "MAILA", "URI", "CAA", "*", "ANY", \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt "TA", "DLV" \

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define QTYPE_CODES { \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 49, 50, 51, 52, 55, 56, 57, 58, 59, 99, 100, 101, 102, 103, \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 104, 105, 106, 107, 249, 250, 251, 252, 253, 254, 255, 255, \

3192f9981331ae89c91a03f1c0f1e3b843fd4f21Evan Hunt 256, 257, 32768, 32769 \

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews#define RCODE_STRINGS { \

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews "NOERROR", "FORMERR", "SERVFAIL", "NXDOMAIN", \

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews "NOTIMP", "REFUSED", "YXDOMAIN", "YXRRSET", \

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews "NXRRSET", "NOTAUTH", "NOTZONE", "rcode11", \

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews "rcode12", "rcode13", "rcode14", "rcode15" \

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews}

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define QUERY_STATUS_MAGIC 0x51535441U /* QSTA */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson#define VALID_QUERY_STATUS(q) ((q) != NULL && \

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson (q)->magic == QUERY_STATUS_MAGIC)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstruct query_status {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned int magic;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int in_use;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned short int id;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson struct timeval sent_timestamp;

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson char *desc;

384fef003441df4067b773688237700513390f9eEvan Hunt int qtype;

384fef003441df4067b773688237700513390f9eEvan Hunt char qname[MAX_DOMAIN_LEN + 1];

384fef003441df4067b773688237700513390f9eEvan Hunt};

384fef003441df4067b773688237700513390f9eEvan Hunt

384fef003441df4067b773688237700513390f9eEvan Huntstruct query_mininfo { /* minimum info for timeout queries */

384fef003441df4067b773688237700513390f9eEvan Hunt int qtype; /* use -1 if N/A */

384fef003441df4067b773688237700513390f9eEvan Hunt struct timeval sent_timestamp;

384fef003441df4067b773688237700513390f9eEvan Hunt char qname[MAX_DOMAIN_LEN + 1];

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson};

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

66f51d8678c8fcc7256354b599317f67455120fdMark Andrewsint is_uint(char *test_int, unsigned int *result);

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int max_queries_outstanding; /* init 0 */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int query_timeout = DEF_QUERY_TIMEOUT;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint ignore_config_changes = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int socket_bufsize = DEF_BUFFER_SIZE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉int family = AF_UNSPEC;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint use_stdin = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonchar *datafile_name; /* init NULL */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonchar *server_to_query; /* init NULL */

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉char *server_port; /* init NULL */

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉struct addrinfo *server_ai; /* init NULL */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint run_only_once = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint use_timelimit = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int run_timelimit; /* init 0 */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int print_interval; /* init 0 */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int target_qps; /* init 0 */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint serverset = FALSE, portset = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint queriesset = FALSE, timeoutset = FALSE;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrewsint edns = FALSE, dnssec = FALSE;

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrewsint countrcodes = FALSE;

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrewsint rcodecounts[16] = {0};

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafssonint verbose = FALSE;

66f51d8678c8fcc7256354b599317f67455120fdMark Andrewsint recurse = 1;

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonint setup_phase = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas GustafssonFILE *datafile_ptr; /* init NULL */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int runs_through_file; /* init 0 */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int num_queries_sent; /* init 0 */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int num_queries_sent_interval;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int num_queries_outstanding; /* init 0 */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int num_queries_timed_out; /* init 0 */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int num_queries_possiblydelayed; /* init 0 */

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int num_queries_timed_out_interval;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int num_queries_possiblydelayed_interval;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstruct timeval time_of_program_start;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstruct timeval time_of_first_query;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double time_of_first_query_sec;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉struct timeval time_of_first_query_interval;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstruct timeval time_of_end_of_run;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉struct timeval time_of_stop_sending;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉struct timeval time_of_queryset_start;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double query_interval;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉struct timeval time_of_next_queryset;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_max = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_max_interval = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_min = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_min_interval = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_total;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉double rtt_total_interval;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉int rttarray_size = DEF_RTTARRAY_SIZE;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉int rttarray_unit = DEF_RTTARRAY_UNIT;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int *rttarray = NULL;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int *rttarray_interval = NULL;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int rtt_overflows;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉unsigned int rtt_overflows_interval;

384fef003441df4067b773688237700513390f9eEvan Huntunsigned int rtt_counted;

384fef003441df4067b773688237700513390f9eEvan Huntunsigned int rtt_counted_interval;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉char *rtt_histogram_file = NULL;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstruct query_status *status; /* init NULL */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int query_status_allocated; /* init 0 */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉int query_socket = -1;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉int socket4 = -1, socket6 = -1;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrewsstatic char *rcode_strings[] = RCODE_STRINGS;

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews

384fef003441df4067b773688237700513390f9eEvan Huntstatic struct query_mininfo *timeout_queries;

384fef003441df4067b773688237700513390f9eEvan Hunt

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonstatic unsigned short

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonget_uint16(unsigned char *buf) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned short ret;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ret = buf[0] * 256 + buf[1];

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (ret);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonshow_startup_info(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson printf("\n"

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson"\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonshow_usage(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr,

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson"\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson DEF_SERVER_TO_QUERY, DEF_SERVER_PORT,

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson DEF_MAX_QUERIES_OUTSTANDING, DEF_QUERY_TIMEOUT,

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 DEF_BUFFER_SIZE, DEF_RTTARRAY_SIZE, DEF_RTTARRAY_UNIT);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonset_datafile(char *new_file) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson char *dfname_tmp;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((new_file == NULL) || (new_file[0] == '\0')) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error: null datafile name\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((dfname_tmp = malloc(strlen(new_file) + 1)) == NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error allocating memory for datafile name: "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "%s\n", new_file);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson free(datafile_name);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson datafile_name = dfname_tmp;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson strcpy(datafile_name, new_file);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson use_stdin = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonset_input_stdin(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson use_stdin = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson free(datafile_name);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson datafile_name = NULL;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonset_server(char *new_name) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson static struct hostent *server_he;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson /* If no change in server name, don't do anything... */

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((server_to_query != NULL) && (new_name != NULL))

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (strcmp(new_name, server_to_query) == 0)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((new_name == NULL) || (new_name[0] == '\0')) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error: null server name\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson free(server_to_query);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson server_to_query = NULL;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((server_to_query = malloc(strlen(new_name) + 1)) == NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error allocating memory for server name: "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "%s\n", new_name);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson strcpy(server_to_query, new_name);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉set_server_port(char *new_port) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 unsigned int uint_val;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((is_uint(new_port, &uint_val)) != TRUE)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (uint_val && uint_val > MAX_PORT)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (server_port != NULL && new_port != NULL &&

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 strcmp(server_port, new_port) == 0)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (0);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 free(server_port);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 server_port = NULL;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((server_port = malloc(strlen(new_port) + 1)) == NULL) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Error allocating memory for server port: "

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "%s\n", new_port);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 strcpy(server_port, new_port);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉set_server_sa(void) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 struct addrinfo hints, *res;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 static struct protoent *proto;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 int error;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (proto == NULL && (proto = getprotobyname("udp")) == NULL) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Error: getprotobyname call failed");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 memset(&hints, 0, sizeof(hints));

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_family = family;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_socktype = SOCK_DGRAM;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_protocol = proto->p_proto;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((error = getaddrinfo(server_to_query, server_port,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 &hints, &res)) != 0) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Error: getaddrinfo(%s, %s) failed\n",

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 server_to_query, server_port);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 /* replace the server's addrinfo */

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (server_ai != NULL)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 freeaddrinfo(server_ai);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 server_ai = res;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (0);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉}

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonis_digit(char d) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (d < '0' || d > '9')

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonis_uint(char *test_int, unsigned int *result) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned long int value;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson char *end;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (test_int == NULL)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_digit(test_int[0]) == FALSE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson value = strtoul(test_int, &end, 10);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((errno == ERANGE) || (*end != '\0') || (value > UINT_MAX))

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson *result = (unsigned int)value;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonset_max_queries(unsigned int new_max) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson static unsigned int size_qs = sizeof(struct query_status);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson struct query_status *temp_stat;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned int count;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (new_max > query_status_allocated) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson temp_stat = realloc(status, new_max * size_qs);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (temp_stat == NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error resizing query_status\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson status = temp_stat;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson /*

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson count = query_status_allocated;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson for (; count < new_max; count++) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson status[count].in_use = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson status[count].magic = QUERY_STATUS_MAGIC;

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson status[count].desc = NULL;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson query_status_allocated = new_max;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson max_queries_outstanding = new_max;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonparse_args(int argc, char **argv) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int c;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned int uint_arg_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 while ((c = getopt(argc, argv,

7d8bfa8f8ea4c222b9403fe97dba0523d768f667Mark Andrews "f:q:t:i:nd:s:p:1l:b:eDcvr:RT:u:H:h")) != -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson switch (c) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 case 'f':

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (strcmp(optarg, "inet") == 0)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 family = AF_INET;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#ifdef AF_INET6

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 else if (strcmp(optarg, "inet6") == 0)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 family = AF_INET6;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#endif

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 else if (strcmp(optarg, "any") == 0)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 family = AF_UNSPEC;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 else {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Invalid address family: %s\n",

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 optarg);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'q':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_uint(optarg, &uint_arg_val) == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson set_max_queries(uint_arg_val);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson queriesset = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Option requires a positive "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "integer value: -%c %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson c, optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 't':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_uint(optarg, &uint_arg_val) == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson query_timeout = uint_arg_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson timeoutset = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Option requires a positive "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "integer value: -%c %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson c, optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'n':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ignore_config_changes = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'd':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (set_datafile(optarg) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error setting datafile "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "name: %s\n", optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 's':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (set_server(optarg) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error setting server "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "name: %s\n", optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson serverset = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'p':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_uint(optarg, &uint_arg_val) == TRUE &&

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson uint_arg_val < MAX_PORT)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 set_server_port(optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson portset = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Option requires a positive "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "integer between 0 and %d: -%c %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson MAX_PORT - 1, c, optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case '1':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson run_only_once = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'l':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_uint(optarg, &uint_arg_val) == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson use_timelimit = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson run_timelimit = uint_arg_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Option requires a positive "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "integer: -%c %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson c, optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case 'b':

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (is_uint(optarg, &uint_arg_val) == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson socket_bufsize = uint_arg_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Option requires a positive "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "integer: -%c %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson c, optarg);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews case 'e':

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews edns = TRUE;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews break;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews case 'D':

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews dnssec = TRUE;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews edns = TRUE;

8ac1acc30d0f405222ffa7b2b93131d9d4e18599Mark Andrews break;

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews case 'c':

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews countrcodes = TRUE;

e2cf0e8ff9ff37121518af5b34b9e4de7abbb47cMark Andrews break;

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson case 'v':

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson verbose = 1;

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 case 'i':

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (is_uint(optarg, &uint_arg_val) == TRUE)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 print_interval = uint_arg_val;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 else {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr, "Invalid interval: %s\n",

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 optarg);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 break;

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews case 'R':

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews recurse = 0;

66f51d8678c8fcc7256354b599317f67455120fdMark Andrews break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 case 'r':

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (is_uint(optarg, &uint_arg_val) == TRUE)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray_size = uint_arg_val;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 else {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr, "Invalid RTT array size: %s\n",

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 optarg);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 case 'u':

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (is_uint(optarg, &uint_arg_val) == TRUE)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray_unit = uint_arg_val;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 else {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr, "Invalid RTT unit: %s\n",

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 optarg);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 case 'H':

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rtt_histogram_file = optarg;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 break;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 case 'T':

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (is_uint(optarg, &uint_arg_val) == TRUE)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 target_qps = uint_arg_val;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 else {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr, "Invalid target qps: %s\n",

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 optarg);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 break;

bd3946db298667e769804b2e978df2d6dcd85e29Mark Andrews case 'h':

bd3946db298667e769804b2e978df2d6dcd85e29Mark Andrews return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson default:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid option: -%c\n", optopt);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (run_only_once == FALSE && use_timelimit == FALSE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson run_only_once = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonopen_datafile(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (use_stdin == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson datafile_ptr = stdin;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((datafile_ptr = fopen(datafile_name, "r")) == NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error: unable to open datafile: %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson datafile_name);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonclose_datafile(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((use_stdin == FALSE) && (datafile_ptr != NULL)) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (fclose(datafile_ptr) != 0) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error: unable to close datafile\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonopen_socket(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int sock;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int ret;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int bufsize;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 struct addrinfo hints, *res;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 memset(&hints, 0, sizeof(hints));

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_family = server_ai->ai_family;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_socktype = server_ai->ai_socktype;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_protocol = server_ai->ai_protocol;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 hints.ai_flags = AI_PASSIVE;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((ret = getaddrinfo(NULL, "0", &hints, &res)) != 0) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Error: getaddrinfo for bind socket failed: %s\n",

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 gai_strerror(ret));

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((sock = socket(res->ai_family, SOCK_DGRAM,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 res->ai_protocol)) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error: socket call failed");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 goto fail;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#if defined(AF_INET6) && defined(IPV6_V6ONLY)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (res->ai_family == AF_INET6) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 int on = 1;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 &on, sizeof(on)) == -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Warning: setsockopt(IPV6_V6ONLY) failed\n");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#endif

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (bind(sock, res->ai_addr, res->ai_addrlen) == -1)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Error: bind call failed");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 freeaddrinfo(res);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson bufsize = 1024 * socket_bufsize;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ret = setsockopt(sock, SOL_SOCKET, SO_RCVBUF,

1123872a47604644f6d74d7d8929472201db96f3Andreas Gustafsson (char *) &bufsize, sizeof(bufsize));

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (ret < 0)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Warning: setsockbuf(SO_RCVBUF) failed\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ret = setsockopt(sock, SOL_SOCKET, SO_SNDBUF,

1123872a47604644f6d74d7d8929472201db96f3Andreas Gustafsson (char *) &bufsize, sizeof(bufsize));

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (ret < 0)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Warning: setsockbuf(SO_SNDBUF) failed\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (sock);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fail:

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (res)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 freeaddrinfo(res);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonclose_socket(void) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (socket4 != -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (close(socket4) != 0) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Error: unable to close IPv4 socket\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (socket6 != -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (close(socket6) != 0) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Error: unable to close IPv6 socket\n");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 query_socket = -1;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉change_socket(void) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 int s, *sockp;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 switch (server_ai->ai_family) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 case AF_INET:

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 sockp = &socket4;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 break;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#ifdef AF_INET6

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 case AF_INET6:

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 sockp = &socket6;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 break;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉#endif

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 default:

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "unexpected address family: %d\n",

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 server_ai->ai_family);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 exit(1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (*sockp == -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((s = open_socket()) == -1)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 *sockp = s;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (*sockp);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉}

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉reset_rttarray(int size) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (rttarray != NULL)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 free(rttarray);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (rttarray_interval != NULL)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 free(rttarray_interval);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray = NULL;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray_interval = NULL;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rtt_max = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rtt_min = -1;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (size > 0) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray = malloc(size * sizeof(rttarray[0]));

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (rttarray == NULL) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr,

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 "Error: allocating memory for RTT array\n");

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 memset(rttarray, 0, size * sizeof(rttarray[0]));

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 rttarray_interval = malloc(size *

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 sizeof(rttarray_interval[0]));

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (rttarray_interval == NULL) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 fprintf(stderr,

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 "Error: allocating memory for RTT array\n");

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 memset(rttarray_interval, 0,

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 size * sizeof(rttarray_interval[0]));

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (0);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉}

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉set_query_interval(unsigned int qps) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (qps == 0)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (0);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 query_interval = (1.0 / (double)qps);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (0);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉}

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonsetup(int argc, char **argv) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson set_input_stdin();

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (set_max_queries(DEF_MAX_QUERIES_OUTSTANDING) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "%s: Unable to set default max outstanding "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "queries\n", argv[0]);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (set_server(DEF_SERVER_TO_QUERY) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "%s: Error setting default server name\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson argv[0]);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (set_server_port(DEF_SERVER_PORT) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "%s: Error setting default server port\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson argv[0]);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (parse_args(argc, argv) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson show_usage();

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (open_datafile() == -1)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (set_server_sa() == -1)

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (-1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((query_socket = change_socket()) == -1)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (reset_rttarray(rttarray_size) == -1)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (set_query_interval(target_qps) == -1)

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 return (-1);

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonset_timenow(struct timeval *tv) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (gettimeofday(tv, NULL) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Error in gettimeofday(). Using inaccurate "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "time() instead\n");

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson tv->tv_sec = time(NULL);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson tv->tv_usec = 0;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉addtv(struct timeval *tv1, struct timeval *tv2, struct timeval *tv_result) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 tv_result->tv_sec = tv1->tv_sec + tv2->tv_sec;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 tv_result->tv_usec = tv1->tv_usec + tv2->tv_usec;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 if (tv_result->tv_usec > 1000000) {

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 tv_result->tv_sec++;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 tv_result->tv_usec -= 1000000;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 }

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉}

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssondifftv(struct timeval tv1, struct timeval tv2) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson long diff_sec, diff_usec;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson double diff;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson diff_sec = tv1.tv_sec - tv2.tv_sec;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson diff_usec = tv1.tv_usec - tv2.tv_usec;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson diff = (double)diff_sec + ((double)diff_usec / 1000000.0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return (diff);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssontimelimit_reached(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson struct timeval time_now;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson set_timenow(&time_now);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (use_timelimit == FALSE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (setup_phase == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (difftv(time_now, time_of_program_start)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson < (double)(run_timelimit + HARD_TIMEOUT_EXTRA))

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (difftv(time_now, time_of_first_query)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson < (double)run_timelimit)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonkeep_sending(int *reached_end_input) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson static int stop = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (stop == TRUE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((*reached_end_input == FALSE) && (timelimit_reached() == FALSE))

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else if ((*reached_end_input == TRUE) && (run_only_once == FALSE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson && (timelimit_reached() == FALSE)) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson rewind(datafile_ptr);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson *reached_end_input = FALSE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson runs_through_file++;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (TRUE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (*reached_end_input == TRUE)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson runs_through_file++;

b89095ba15e9163f2bacc46239d8f69d1a2b9ad3Tatuya JINMEI 神明達哉 set_timenow(&time_of_stop_sending);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson stop = TRUE;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (FALSE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonunsigned int

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonqueries_outstanding(void) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (num_queries_outstanding);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonnext_input_line(char *line, int n) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson char *result;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson do {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson result = fgets(line, n, datafile_ptr);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } while ((result != NULL) &&

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ((line[0] == COMMENT_CHAR) || (line[0] == '\n')));

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (result == NULL)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (0);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (strlen(line));

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonidentify_directive(char *dir) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson static char *directives[] = DIRECTIVES;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson static int dir_values[] = DIR_VALUES;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned int index, num_directives;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson num_directives = sizeof(directives) / sizeof(directives[0]);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (num_directives > (sizeof(dir_values) / sizeof(int)))

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson num_directives = sizeof(dir_values) / sizeof(int);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson for (index = 0; index < num_directives; index++) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (strcmp(dir, directives[index]) == 0)

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (dir_values[index]);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return (-1);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafssonupdate_config(char *config_change_desc) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson char *directive, *config_value, *trailing_garbage;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson char conf_copy[MAX_INPUT_LEN + 1];

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson unsigned int uint_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int directive_number;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson int check;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 int old_af;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (ignore_config_changes == TRUE) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Ignoring configuration change: %s",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson config_change_desc);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson strcpy(conf_copy, config_change_desc);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson ++config_change_desc;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson if (*config_change_desc == '\0') {

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson fprintf(stderr, "Invalid config: No directive present: %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson conf_copy);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (index(WHITESPACE, *config_change_desc) != NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Space before directive or "

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson "no directive present: %s\n", conf_copy);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson directive = strtok(config_change_desc, WHITESPACE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson config_value = strtok(NULL, WHITESPACE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson trailing_garbage = strtok(NULL, WHITESPACE);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((directive_number = identify_directive(directive)) == -1) {

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson fprintf(stderr, "Invalid config: Bad directive: %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson conf_copy);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (config_value == NULL) {

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson fprintf(stderr, "Invalid config: No value present: %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson conf_copy);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (trailing_garbage != NULL) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Config warning: "

81e805888bcd4bd2a6d5c986fade6ffdf7d88a54Andreas Gustafsson "trailing garbage: %s\n", conf_copy);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson switch(directive_number) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case V_SERVER:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (serverset && (setup_phase == TRUE)) {

6d7939a85025621aa11e9cfe933629113bd0963aMark Andrews fprintf(stderr, "Config change overridden by command "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "line: %s\n", directive);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (set_server(config_value) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Set server error: unable to change "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "the server name to '%s'\n", config_value);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 old_af = server_ai->ai_family;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (set_server_sa() == -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Set server error: unable to resolve "

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "a new server '%s'\n",

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 config_value);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (old_af != server_ai->ai_family) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if ((query_socket = change_socket()) == -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 /* XXX: this is fatal */

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr, "Set server error: "

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "unable to open a new socket "

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "for '%s'\n", config_value);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 exit(1);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case V_PORT:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (portset && (setup_phase == TRUE)) {

6d7939a85025621aa11e9cfe933629113bd0963aMark Andrews fprintf(stderr, "Config change overridden by command "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "line: %s\n", directive);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson check = is_uint(config_value, &uint_val);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((check == TRUE) && (uint_val > 0)) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (set_server_port(config_value) == -1) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Bad value for"

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson " %s: %s\n", directive, config_value);

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 } else {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 if (set_server_sa() == -1) {

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 fprintf(stderr,

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 "Failed to set a new port\n");

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 return;

f620c5e527746a2ec3d90a11d21abd8a114746dfTatuya JINMEI 神明達哉 }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Bad value for "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "%s: %s\n", directive, config_value);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case V_MAXQUERIES:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (queriesset && (setup_phase == TRUE)) {

6d7939a85025621aa11e9cfe933629113bd0963aMark Andrews fprintf(stderr, "Config change overridden by command "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "line: %s\n", directive);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson check = is_uint(config_value, &uint_val);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((check == TRUE) && (uint_val > 0)) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson set_max_queries(uint_val);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Bad value for "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "%s: %s\n", directive, config_value);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson case V_MAXWAIT:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if (timeoutset && (setup_phase == TRUE)) {

6d7939a85025621aa11e9cfe933629113bd0963aMark Andrews fprintf(stderr, "Config change overridden by command "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "line: %s\n", directive);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson return;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson check = is_uint(config_value, &uint_val);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson if ((check == TRUE) && (uint_val > 0)) {

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson query_timeout = uint_val;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson } else

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Bad value for "

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson "%s: %s\n", directive, config_value);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson default:

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson fprintf(stderr, "Invalid config: Bad directive: %s\n",

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson directive);

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson break;

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson }

4848fe4ad2c0ba6e2e69e4a2617727f8556d79a0Andreas Gustafsson}