mdw@git.distorted.org.uk Git - adns/blob - src/submit.c

   1 /**/
   2
   3 #include <stdlib.h>
   4 #include <unistd.h>
   5 #include <errno.h>
   6
   7 #include <sys/time.h>
   8
   9 #include "internal.h"
  10
  11 static adns_query allocquery(adns_state ads, const char *owner, int ol,
  12                              int id, const typeinfo *typei,
  13                              adns_queryflags flags, const qcontext *ctx) {
  14   /* Query message used is the one assembled in ads->rqbuf */
  15   adns_query qu;
  16   adns_answer *ans;
  17
  18   qu= malloc(sizeof(*qu)+ol+1+ads->rqbuf.used); if (!qu) return 0;
  19   adns__vbuf_init(&qu->ansbuf);
  20   qu->cname= 0;
  21   qu->state= query_udp;
  22   qu->next= qu->back= qu->parent= 0;
  23   LIST_INIT(qu->children);
  24   qu->siblings.next= qu->siblings.back= 0;
  25   qu->typei= typei;
  26   qu->id= id;
  27   qu->flags= flags;
  28   qu->udpretries= 0;
  29   qu->udpnextserver= 0;
  30   qu->udpsent= qu->tcpfailed= 0;
  31   timerclear(&qu->timeout);
  32   memcpy(&qu->context,ctx,sizeof(qu->context));
  33   memcpy(qu->owner,owner,ol); qu->owner[ol]= 0;
  34   qu->querymsg= qu->owner+ol+1;
  35   memcpy(qu->owner+ol+1,ads->rqbuf.buf,ads->rqbuf.used);
  36   qu->querylen= ads->rqbuf.used;
  37
  38   return qu;
  39 }
  40
  41 static int failsubmit(adns_state ads, const qcontext *ctx, adns_query *query_r,
  42                       adns_queryflags flags, int id, adns_status stat) {
  43   adns_query qu;
  44
  45   ads->rqbuf.used= 0;
  46   qu= allocquery(ads,0,0,id,0,flags,ctx); if (!qu) return errno;
  47   adns__query_fail(ads,qu,stat);
  48   *query_r= qu;
  49   return 0;
  50 }
  51
  52 int adns_submit(adns_state ads,
  53                 const char *owner,
  54                 adns_rrtype type,
  55                 adns_queryflags flags,
  56                 void *context,
  57                 adns_query *query_r) {
  58   adns_query qu;
  59   adns_status stat;
  60   int ol, id, r;
  61   qcontext ctx;
  62   struct timeval now;
  63   const typeinfo *typei;
  64
  65   ctx.ext= context;
  66   id= ads->nextid++;
  67
  68   r= gettimeofday(&now,0); if (r) return errno;
  69
  70   typei= adns__findtype(type);
  71   if (!typei) return failsubmit(ads,context,query_r,flags,id,adns_s_notimplemented);
  72
  73   ol= strlen(owner);
  74   if (ol<=1 || ol>DNS_MAXDOMAIN+1)
  75     return failsubmit(ads,context,query_r,flags,id,adns_s_invaliddomain);
  76   if (owner[ol-1]=='.' && owner[ol-2]!='\\') { flags &= ~adns_qf_search; ol--; }
  77
  78   stat= adns__mkquery(ads,owner,ol,id,typei,flags);
  79   if (stat) return failsubmit(ads,context,query_r,flags,id,stat);
  80
  81   qu= allocquery(ads,owner,ol,id,typei,flags,&ctx); if (!qu) return errno;
  82   adns__query_udp(ads,qu,now);
  83   adns__autosys(ads,now);
  84
  85   *query_r= qu;
  86   return 0;
  87 }
  88
  89 int adns_synchronous(adns_state ads,
  90                      const char *owner,
  91                      adns_rrtype type,
  92                      adns_queryflags flags,
  93                      adns_answer **answer_r) {
  94   adns_query qu;
  95   int r;
  96
  97   r= adns_submit(ads,owner,type,flags,0,&qu);
  98   if (r) return r;
  99
 100   do {
 101     r= adns_wait(ads,&qu,answer_r,0);
 102   } while (r==EINTR);
 103   if (r) adns_cancel(ads,qu);
 104   return r;
 105 }
 106
 107 void adns_cancel(adns_state ads, adns_query query) {
 108   abort(); /* fixme */
 109 }