[adns] / src / internal.h

/**/

#ifndef ADNS_INTERNAL_H_INCLUDED
#define ADNS_INTERNAL_H_INCLUDED

#define PRINTFFORMAT(a,b) __attribute__((format(printf,a,b)))
typedef unsigned char byte;

#include <stdarg.h>
#include <assert.h>
#include <unistd.h>

#include <sys/time.h>

#include "adns.h"

/* Configuration and constants */

#define MAXSERVERS 5
#define MAXUDPRETRIES /*15*/5
#define UDPRETRYMS 2000
#define TCPMS 30000
#define LOCALRESOURCEMS 20
#define MAXUDPDGRAM 512
#define NSPORT 53
#define MAXDNAME 255

/* Shared data structures */

typedef union {
  adns_status status;
  char *cp;
  adns_rrtype type;
  int i;
  struct in_addr ia;
  unsigned long ul;
} rr_align;

typedef struct {
  int used, avail;
  byte *buf;
} vbuf;

typedef union {
  void *ext;
  int dmaddr_index;
} qcontext;

struct adns__query {
  /* FIXME: make sure this is all init'd properly */
  enum { query_udp, query_tcpwait, query_tcpsent, query_child, query_done } state;
  adns_query back, next, parent;
  struct { adns_query head, tail; } children;
  struct { adns_query back, next; } siblings;
  adns_rrtype type;
  vbuf answer;
  int id, flags, udpretries;
  int udpnextserver;
  unsigned long udpsent, tcpfailed; /* bitmap indexed by server */
  struct timeval timeout;
  byte *querymsg;
  int querylen;
  qcontext context;
  char owner[1];
  /* After the owner name and nul comes the query message, pointed to by querymsg */

  /* Possible states:
   *
   *  state   Queue   child  id   answer    nextudpserver  sentudp     failedtcp
   *
   *  udp     NONE    null   >=0  null      0              zero        zero
   *  udp     timew   null   >=0  null      any            nonzero     zero
   *  udp     NONE    null   >=0  null      any            nonzero     zero
   *
   *  tcpwait timew   null   >=0  null      irrelevant     zero        any
   *  tcpsent timew   null   >=0  null      irrelevant     zero        any
   *
   *  child   childw  set    >=0  partial   irrelevant     irrelevant  irrelevant
   *  done    output  null   -1   set/null  irrelevant     irrelevant  irrelevant
   *
   *			      +------------------------+
   *             START -----> |      udp/NONE          |
   *			      +------------------------+
   *                         /                       |\  \
   *        too big for UDP /             UDP timeout  \  \ send via UDP
   *        do this ASAP!  /              more retries  \  \   do this ASAP!
   *                     |_                  desired     \  _|
   *		  +---------------+     	    	+-----------+
   *              | tcpwait/timew | ____                | udp/timew |
   *              +---------------+     \	    	+-----------+
   *                    |  ^             |                 | |
   *     TCP conn'd;    |  | TCP died    |                 | |
   *     send via TCP   |  | more        |     UDP timeout | |
   *     do this ASAP!  |  | servers     |      no more    | |
   *                    v  | to try      |      retries    | |
   *              +---------------+      |      desired    | |
   *              | tcpsent/timew | ____ |                 | |
   *    	  +---------------+     \|                 | |
   *                  \   \ TCP died     | TCP             | |
   *                   \   \ no more     | timeout         / |
   *                    \   \ servers    |                /  |
   *                     \   \ to try    |               /   |
   *                  got \   \          v             |_    / got
   *                 reply \   _| +------------------+      / reply
   *   	       	       	    \  	  | done/output FAIL |     /
   *                         \    +------------------+    /
   *                          \                          /
   *                           _|                      |_
   *                             (..... got reply ....)
   *                              /                   \
   *        need child query/ies /                     \ no child query
   *                            /                       \
   *                          |_                         _|
   *		    +--------------+		       +----------------+
   *                | child/childw | ----------------> | done/output OK |
   *                +--------------+  children done    +----------------+
   */
};

struct adns__state {
  adns_initflags iflags;
  FILE *diagfile;
  struct { adns_query head, tail; } timew, childw, output;
  int nextid, udpsocket, tcpsocket;
  vbuf rqbuf, tcpsend, tcprecv;
  int nservers, tcpserver;
  enum adns__tcpstate { server_disconnected, server_connecting, server_ok } tcpstate;
  struct timeval tcptimeout;
  struct server {
    struct in_addr addr;
  } servers[MAXSERVERS];
};

/* From setup.c: */

void adns__vdiag(adns_state ads, const char *pfx, adns_initflags prevent,
		 int serv, const char *fmt, va_list al);
void adns__debug(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);
void adns__warn(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);
void adns__diag(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);

int adns__vbuf_ensure(vbuf *vb, int want);
int adns__vbuf_append(vbuf *vb, const byte *data, int len);
/* 1=>success, 0=>realloc failed */
void adns__vbuf_appendq(vbuf *vb, const byte *data, int len);
void adns__vbuf_init(vbuf *vb);

int adns__setnonblock(adns_state ads, int fd); /* => errno value */

/* From submit.c: */

void adns__query_nomem(adns_state ads, adns_query qu);
void adns__query_fail(adns_state ads, adns_query qu, adns_status stat);

/* From query.c: */

void adns__query_udp(adns_state ads, adns_query qu, struct timeval now);
void adns__query_tcp(adns_state ads, adns_query qu, struct timeval now);
adns_status adns__mkquery(adns_state ads, const char *owner, int ol, int id,
			  adns_rrtype type, adns_queryflags flags);

/* From reply.c: */

void adns__procdgram(adns_state ads, const byte *dgram, int len, int serv);

/* From event.c: */

void adns__tcp_broken(adns_state ads, const char *what, const char *why);
void adns__tcp_tryconnect(adns_state ads, struct timeval now);
void adns__autosys(adns_state ads, struct timeval now);

/* Useful static inline functions: */

static inline void timevaladd(struct timeval *tv_io, long ms) {
  struct timeval tmp;
  assert(ms>=0);
  tmp= *tv_io;
  tmp.tv_usec += (ms%1000)*1000000;
  tmp.tv_sec += ms/1000;
  if (tmp.tv_usec >= 1000000) { tmp.tv_sec++; tmp.tv_usec -= 1000; }
  *tv_io= tmp;
}

static inline int ctype_whitespace(int c) { return c==' ' || c=='\n' || c=='\t'; }
static inline int ctype_digit(int c) { return c>='0' && c<='9'; }

/* Useful macros */

#define LIST_INIT(list) ((list).head= (list).tail= 0)

#define LIST_UNLINK_PART(list,node,part) \
  do { \
    if ((node)->back) (node)->back->part next= (node)->part next; \
      else                        (list).head= (node)->part next; \
    if ((node)->next) (node)->next->part back= (node)->part back; \
      else                        (list).tail= (node)->part back; \
  } while(0)

#define LIST_LINK_TAIL_PART(list,node,part) \
  do { \
    (node)->part back= 0; \
    (node)->part next= (list).tail; \
    if ((list).tail) (list).tail->part back= (node); else (list).part head= (node); \
    (list).tail= (node); \
  } while(0)

#define LIST_UNLINK(list,node) LIST_UNLINK_PART(list,node,)
#define LIST_LINK_TAIL(list,node) LIST_LINK_TAIL_PART(list,node,)

#endif
Commit	Line	Data
dfdbb32c	1	/**/
	2
	3	#ifndef ADNS_INTERNAL_H_INCLUDED
	4	#define ADNS_INTERNAL_H_INCLUDED
	5
d05cc330	6	#define PRINTFFORMAT(a,b) __attribute__((format(printf,a,b)))
	7	typedef unsigned char byte;
	8
	9	#include <stdarg.h>
	10	#include <assert.h>
	11	#include <unistd.h>
	12
dfdbb32c	13	#include <sys/time.h>
	14
	15	#include "adns.h"
	16
	17	/* Configuration and constants */
	18
	19	#define MAXSERVERS 5
de8b18da	20	#define MAXUDPRETRIES /15/5
dfdbb32c	21	#define UDPRETRYMS 2000
	22	#define TCPMS 30000
	23	#define LOCALRESOURCEMS 20
d05cc330	24	#define MAXUDPDGRAM 512
dfdbb32c	25	#define NSPORT 53
d05cc330	26	#define MAXDNAME 255
dfdbb32c	27
	28	/* Shared data structures */
	29
d05cc330	30	typedef union {
dfdbb32c	31	adns_status status;
	32	char *cp;
	33	adns_rrtype type;
d05cc330	34	int i;
dfdbb32c	35	struct in_addr ia;
dfdbb32c	36	unsigned long ul;
d05cc330	37	} rr_align;
	38
	39	typedef struct {
	40	int used, avail;
	41	byte *buf;
	42	} vbuf;
	43
	44	typedef union {
	45	void *ext;
	46	int dmaddr_index;
	47	} qcontext;
dfdbb32c	48
	49	struct adns__query {
	50	/* FIXME: make sure this is all init'd properly */
96e79df5	51	enum { query_udp, query_tcpwait, query_tcpsent, query_child, query_done } state;
d05cc330	52	adns_query back, next, parent;
dfdbb32c	53	struct { adns_query head, tail; } children;
	54	struct { adns_query back, next; } siblings;
	55	adns_rrtype type;
d05cc330	56	vbuf answer;
96e79df5	57	int id, flags, udpretries;
d05cc330	58	int udpnextserver;
d05cc330	59	unsigned long udpsent, tcpfailed; /* bitmap indexed by server */
dfdbb32c	60	struct timeval timeout;
d05cc330	61	byte *querymsg;
dfdbb32c	62	int querylen;
d05cc330	63	qcontext context;
dfdbb32c	64	char owner[1];
d05cc330	65	/* After the owner name and nul comes the query message, pointed to by querymsg */
96e79df5	66
dfdbb32c	67	/* Possible states:
96e79df5	68	*
	69	* state Queue child id answer nextudpserver sentudp failedtcp
	70	*
	71	* udp NONE null >=0 null 0 zero zero
	72	* udp timew null >=0 null any nonzero zero
	73	* udp NONE null >=0 null any nonzero zero
	74	*
	75	* tcpwait timew null >=0 null irrelevant zero any
	76	* tcpsent timew null >=0 null irrelevant zero any
	77	*
	78	* child childw set >=0 partial irrelevant irrelevant irrelevant
	79	* done output null -1 set/null irrelevant irrelevant irrelevant
	80	*
	81	* +------------------------+
	82	* START -----> \| udp/NONE \|
	83	* +------------------------+
	84	* / \|\ \
	85	* too big for UDP / UDP timeout \ \ send via UDP
	86	* do this ASAP! / more retries \ \ do this ASAP!
	87	* \|_ desired \ _\|
	88	* +---------------+ +-----------+
	89	* \| tcpwait/timew \| ____ \| udp/timew \|
	90	* +---------------+ \ +-----------+
	91	* \| ^ \| \| \|
	92	* TCP conn'd; \| \| TCP died \| \| \|
	93	* send via TCP \| \| more \| UDP timeout \| \|
	94	* do this ASAP! \| \| servers \| no more \| \|
	95	* v \| to try \| retries \| \|
	96	* +---------------+ \| desired \| \|
	97	* \| tcpsent/timew \| ____ \| \| \|
	98	* +---------------+ \\| \| \|
	99	* \ \ TCP died \| TCP \| \|
	100	* \ \ no more \| timeout / \|
	101	* \ \ servers \| / \|
	102	* \ \ to try \| / \|
	103	* got \ \ v \|_ / got
	104	* reply \ _\| +------------------+ / reply
	105	* \ \| done/output FAIL \| /
	106	* \ +------------------+ /
	107	* \ /
	108	* _\| \|_
	109	* (..... got reply ....)
	110	* / \
	111	* need child query/ies / \ no child query
	112	* / \
	113	* \|_ _\|
	114	* +--------------+ +----------------+
	115	* \| child/childw \| ----------------> \| done/output OK \|
	116	* +--------------+ children done +----------------+
dfdbb32c	117	*/
	118	};
	119
dfdbb32c	120	struct adns__state {
dfdbb32c	121	adns_initflags iflags;
dfdbb32c	122	FILE *diagfile;
96e79df5	123	struct { adns_query head, tail; } timew, childw, output;
d05cc330	124	int nextid, udpsocket, tcpsocket;
d05cc330	125	vbuf rqbuf, tcpsend, tcprecv;
dfdbb32c	126	int nservers, tcpserver;
71324651	127	enum adns__tcpstate { server_disconnected, server_connecting, server_ok } tcpstate;
dfdbb32c	128	struct timeval tcptimeout;
	129	struct server {
	130	struct in_addr addr;
	131	} servers[MAXSERVERS];
	132	};
	133
	134	/* From setup.c: */
	135
d05cc330	136	void adns__vdiag(adns_state ads, const char *pfx, adns_initflags prevent,
dfdbb32c	137	int serv, const char *fmt, va_list al);
71324651	138	void adns__debug(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);
	139	void adns__warn(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);
	140	void adns__diag(adns_state ads, int serv, const char *fmt, ...) PRINTFFORMAT(3,4);
dfdbb32c	141
d05cc330	142	int adns__vbuf_ensure(vbuf *vb, int want);
d05cc330	143	int adns__vbuf_append(vbuf vb, const byte data, int len);
96e79df5	144	/* 1=>success, 0=>realloc failed */
d05cc330	145	void adns__vbuf_appendq(vbuf vb, const byte data, int len);
	146	void adns__vbuf_init(vbuf *vb);
	147
	148	int adns__setnonblock(adns_state ads, int fd); /* => errno value */
96e79df5	149
dfdbb32c	150	/* From submit.c: */
dfdbb32c	151
d05cc330	152	void adns__query_nomem(adns_state ads, adns_query qu);
dfdbb32c	153	void adns__query_fail(adns_state ads, adns_query qu, adns_status stat);
	154
	155	/* From query.c: */
	156
96e79df5	157	void adns__query_udp(adns_state ads, adns_query qu, struct timeval now);
96e79df5	158	void adns__query_tcp(adns_state ads, adns_query qu, struct timeval now);
48cb0b4b	159	adns_status adns__mkquery(adns_state ads, const char *owner, int ol, int id,
d05cc330	160	adns_rrtype type, adns_queryflags flags);
	161
	162	/* From reply.c: */
	163
	164	void adns__procdgram(adns_state ads, const byte *dgram, int len, int serv);
71324651	165
71324651	166	/* From event.c: */
d05cc330	167
71324651	168	void adns__tcp_broken(adns_state ads, const char what, const char why);
d05cc330	169	void adns__tcp_tryconnect(adns_state ads, struct timeval now);
d05cc330	170	void adns__autosys(adns_state ads, struct timeval now);
dfdbb32c	171
	172	/* Useful static inline functions: */
	173
	174	static inline void timevaladd(struct timeval *tv_io, long ms) {
	175	struct timeval tmp;
	176	assert(ms>=0);
	177	tmp= *tv_io;
de8b18da	178	tmp.tv_usec += (ms%1000)*1000000;
dfdbb32c	179	tmp.tv_sec += ms/1000;
de8b18da	180	if (tmp.tv_usec >= 1000000) { tmp.tv_sec++; tmp.tv_usec -= 1000; }
dfdbb32c	181	*tv_io= tmp;
96e79df5	182	}
dfdbb32c	183
	184	static inline int ctype_whitespace(int c) { return c==' ' \|\| c=='\n' \|\| c=='\t'; }
	185	static inline int ctype_digit(int c) { return c>='0' && c<='9'; }
	186
	187	/* Useful macros */
	188
d05cc330	189	#define LIST_INIT(list) ((list).head= (list).tail= 0)
d05cc330	190
dfdbb32c	191	#define LIST_UNLINK_PART(list,node,part) \
	192	do { \
	193	if ((node)->back) (node)->back->part next= (node)->part next; \
	194	else (list).head= (node)->part next; \
	195	if ((node)->next) (node)->next->part back= (node)->part back; \
	196	else (list).tail= (node)->part back; \
	197	} while(0)
	198
	199	#define LIST_LINK_TAIL_PART(list,node,part) \
	200	do { \
	201	(node)->part back= 0; \
	202	(node)->part next= (list).tail; \
	203	if ((list).tail) (list).tail->part back= (node); else (list).part head= (node); \
	204	(list).tail= (node); \
	205	} while(0)
	206
	207	#define LIST_UNLINK(list,node) LIST_UNLINK_PART(list,node,)
d05cc330	208	#define LIST_LINK_TAIL(list,node) LIST_LINK_TAIL_PART(list,node,)
dfdbb32c	209
dfdbb32c	210	#endif