60cfce122f
This makes dig work, and shows that we can actually transmit as well as receive. Both UDP and TCP connections work, with the restriction that a full message must be contained in each TCP write, not split across two seperate tcp 'packets'.
41 lines
1.2 KiB
C
41 lines
1.2 KiB
C
#include <isc/mutex.h>
|
|
#include <isc/socket.h>
|
|
#include <isc/task.h>
|
|
|
|
#define UDP_INPUT_BUFFER_SIZE (64 * 1024) /* 64k */
|
|
|
|
typedef struct __udp_listener udp_listener_t;
|
|
typedef struct __udp_cctx udp_cctx_t;
|
|
|
|
struct __udp_cctx {
|
|
unsigned char *buf; /* input buffer */
|
|
isc_mem_t *mctx; /* memory context used to allocate */
|
|
udp_listener_t *parent; /* controlling listener */
|
|
u_int slot; /* slot # in tasks[] (and ctxs[]) array */
|
|
u_int count; /* XXX debug */
|
|
};
|
|
|
|
struct __udp_listener {
|
|
isc_socket_t *sock; /* the socket */
|
|
u_int nwstart; /* workers to start */
|
|
u_int nwkeep; /* workers to keep */
|
|
u_int nwmax; /* workers max */
|
|
isc_mem_t *mctx;
|
|
dns_result_t (*dispatch)(isc_mem_t *, isc_region_t *, unsigned int);
|
|
isc_mutex_t lock;
|
|
|
|
/* locked */
|
|
isc_task_t **tasks; /* list of tasks */
|
|
u_int nwactive; /* workers active */
|
|
udp_cctx_t **ctxs; /* list of contexts */
|
|
};
|
|
|
|
udp_listener_t *udp_listener_allocate(isc_mem_t *mctx, u_int nwmax);
|
|
|
|
isc_result_t udp_listener_start(udp_listener_t *l,
|
|
isc_socket_t *sock, isc_taskmgr_t *tmgr,
|
|
u_int nwstart, u_int nwkeep, u_int nwtimeout,
|
|
dns_result_t (*dispatch)(isc_mem_t *,
|
|
isc_region_t *,
|
|
unsigned int));
|