1 /* Name resolution using adns */
7 #error secnet requires ADNS version 1.0 or above
10 #include <arpa/inet.h>
16 struct resolver_if ops;
25 resolve_answer_fn *answer;
29 static resolve_request_fn resolve_request;
30 static bool_t resolve_request(void *sst, cstring_t name,
31 int port, struct comm_if *comm,
32 resolve_answer_fn *cb, void *cst)
37 const int maxlitlen=50;
39 ssize_t l=strlen(name);
40 if (name[0]=='[' && l<maxlitlen && l>2 && name[l-1]==']') {
41 char trimmed[maxlitlen+1];
42 memcpy(trimmed,name+1,l-2);
48 ca.ia.sin.sin_family=AF_INET;
49 ca.ia.sin.sin_port=htons(port);
50 if (inet_aton(trimmed,&ca.ia.sin.sin_addr))
57 q=safe_malloc(sizeof *q,"resolve_request");
63 rv=adns_submit(st->ast, name, adns_r_addr, 0, q, &q->query);
66 "resolver: failed to submit lookup for %s: %s",name,
75 static int resolver_beforepoll(void *sst, struct pollfd *fds, int *nfds_io,
79 return adns_beforepoll(st->ast, fds, nfds_io, timeout_io, tv_now);
82 static void resolver_afterpoll(void *sst, struct pollfd *fds, int nfds)
91 adns_afterpoll(st->ast, fds, nfds, tv_now);
95 rv=adns_check(st->ast, &aq, &ans, &qp);
98 if (ans->status!=adns_s_ok) {
99 q->answer(q->cst,NULL,0); /* Failure */
104 int ca_len=MIN(ans->nrrs,MAX_PEER_ADDRS);
105 struct comm_addr ca_buf[ca_len];
107 for (rslot=0, wslot=0;
108 rslot<ans->nrrs && wslot<ca_len;
110 adns_rr_addr *ra=&ans->rrs.addr[rslot];
111 struct comm_addr *ca=&ca_buf[wslot];
113 /* copy fields individually so we leave holes zeroed: */
114 switch (ra->addr.sa.sa_family) {
116 assert(ra->len == sizeof(ca->ia.sin));
117 ca->ia.sin.sin_family=ra->addr.inet.sin_family;
118 ca->ia.sin.sin_addr= ra->addr.inet.sin_addr;
119 ca->ia.sin.sin_port= htons(q->port);
124 assert(ra->len == sizeof(ca->ia.sin6));
125 ca->ia.sin6.sin6_family=ra->addr.inet6.sin6_family;
126 ca->ia.sin6.sin6_addr= ra->addr.inet6.sin6_addr;
127 ca->ia.sin6.sin6_port= htons(q->port);
130 #endif /*CONFIG_IPV6*/
135 q->answer(q->cst,ca_buf,wslot);
139 } else if (rv==EAGAIN || rv==ESRCH) {
142 fatal("resolver_afterpoll: adns_check() returned %d",rv);
149 /* Initialise adns, using parameters supplied */
150 static list_t *adnsresolver_apply(closure_t *self, struct cloc loc,
151 dict_t *context, list_t *args)
158 st=safe_malloc(sizeof(*st),"adnsresolver_apply");
159 st->cl.description="adns";
160 st->cl.type=CL_RESOLVER;
162 st->cl.interface=&st->ops;
165 st->ops.request=resolve_request;
168 if (!i || i->type!=t_dict) {
169 cfgfatal(st->loc,"adns","first argument must be a dictionary\n");
172 conf=dict_read_string(d,"config",False,"adns",loc);
175 if (adns_init_strcfg(&st->ast, 0, 0, conf)) {
176 fatal_perror("Failed to initialise ADNS");
179 if (adns_init(&st->ast, 0, 0)) {
180 fatal_perror("Failed to initialise ADNS");
184 register_for_poll(st, resolver_beforepoll, resolver_afterpoll,
185 ADNS_POLLFDS_RECOMMENDED+5,"resolver");
187 return new_closure(&st->cl);
190 void resolver_module(dict_t *dict)
192 add_closure(dict,"adns",adnsresolver_apply);