contrib/bind-9.3/lib/bind/irs/gen_sv.c

   1 /*
   2  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
   3  * Copyright (c) 1996,1999 by Internet Software Consortium.
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  15  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 #if !defined(LINT) && !defined(CODECENTER)
  19 static const char rcsid[] = "$Id: gen_sv.c,v 1.1.206.1 2004/03/09 08:33:35 marka Exp $";
  20 #endif
  21
  22 /* Imports */
  23
  24 #include "port_before.h"
  25
  26 #include <sys/types.h>
  27 #include <netinet/in.h>
  28 #include <arpa/nameser.h>
  29 #include <resolv.h>
  30
  31 #include <errno.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34
  35 #include <isc/memcluster.h>
  36 #include <irs.h>
  37
  38 #include "port_after.h"
  39
  40 #include "irs_p.h"
  41 #include "gen_p.h"
  42
  43 /* Types */
  44
  45 struct pvt {
  46         struct irs_rule *       rules;
  47         struct irs_rule *       rule;
  48         struct __res_state *    res;
  49         void                    (*free_res)(void *);
  50 };
  51
  52 /* Forward */
  53
  54 static void                     sv_close(struct irs_sv*);
  55 static struct servent *         sv_next(struct irs_sv *);
  56 static struct servent *         sv_byname(struct irs_sv *, const char *,
  57                                           const char *);
  58 static struct servent *         sv_byport(struct irs_sv *, int, const char *);
  59 static void                     sv_rewind(struct irs_sv *);
  60 static void                     sv_minimize(struct irs_sv *);
  61 static struct __res_state *     sv_res_get(struct irs_sv *);
  62 static void                     sv_res_set(struct irs_sv *,
  63                                               struct __res_state *,
  64                                               void (*)(void *));
  65
  66 /* Public */
  67
  68 struct irs_sv *
  69 irs_gen_sv(struct irs_acc *this) {
  70         struct gen_p *accpvt = (struct gen_p *)this->private;
  71         struct irs_sv *sv;
  72         struct pvt *pvt;
  73
  74         if (!(sv = memget(sizeof *sv))) {
  75                 errno = ENOMEM;
  76                 return (NULL);
  77         }
  78         memset(sv, 0x5e, sizeof *sv);
  79         if (!(pvt = memget(sizeof *pvt))) {
  80                 memput(sv, sizeof *sv);
  81                 errno = ENOMEM;
  82                 return (NULL);
  83         }
  84         memset(pvt, 0, sizeof *pvt);
  85         pvt->rules = accpvt->map_rules[irs_sv];
  86         pvt->rule = pvt->rules;
  87         sv->private = pvt;
  88         sv->close = sv_close;
  89         sv->next = sv_next;
  90         sv->byname = sv_byname;
  91         sv->byport = sv_byport;
  92         sv->rewind = sv_rewind;
  93         sv->minimize = sv_minimize;
  94         sv->res_get = sv_res_get;
  95         sv->res_set = sv_res_set;
  96         return (sv);
  97 }
  98
  99 /* Methods */
 100
 101 static void
 102 sv_close(struct irs_sv *this) {
 103         struct pvt *pvt = (struct pvt *)this->private;
 104
 105         memput(pvt, sizeof *pvt);
 106         memput(this, sizeof *this);
 107 }
 108
 109 static struct servent *
 110 sv_next(struct irs_sv *this) {
 111         struct pvt *pvt = (struct pvt *)this->private;
 112         struct servent *rval;
 113         struct irs_sv *sv;
 114
 115         while (pvt->rule) {
 116                 sv = pvt->rule->inst->sv;
 117                 rval = (*sv->next)(sv);
 118                 if (rval)
 119                         return (rval);
 120                 if (!(pvt->rule->flags & IRS_CONTINUE))
 121                         break;
 122                 pvt->rule = pvt->rule->next;
 123                 if (pvt->rule) {
 124                         sv = pvt->rule->inst->sv;
 125                         (*sv->rewind)(sv);
 126                 }
 127         }
 128         return (NULL);
 129 }
 130
 131 static struct servent *
 132 sv_byname(struct irs_sv *this, const char *name, const char *proto) {
 133         struct pvt *pvt = (struct pvt *)this->private;
 134         struct irs_rule *rule;
 135         struct servent *rval;
 136         struct irs_sv *sv;
 137
 138         rval = NULL;
 139         for (rule = pvt->rules; rule; rule = rule->next) {
 140                 sv = rule->inst->sv;
 141                 rval = (*sv->byname)(sv, name, proto);
 142                 if (rval || !(rule->flags & IRS_CONTINUE))
 143                         break;
 144         }
 145         return (rval);
 146 }
 147
 148 static struct servent *
 149 sv_byport(struct irs_sv *this, int port, const char *proto) {
 150         struct pvt *pvt = (struct pvt *)this->private;
 151         struct irs_rule *rule;
 152         struct servent *rval;
 153         struct irs_sv *sv;
 154
 155         rval = NULL;
 156         for (rule = pvt->rules; rule; rule = rule->next) {
 157                 sv = rule->inst->sv;
 158                 rval = (*sv->byport)(sv, port, proto);
 159                 if (rval || !(rule->flags & IRS_CONTINUE))
 160                         break;
 161         }
 162         return (rval);
 163 }
 164
 165 static void
 166 sv_rewind(struct irs_sv *this) {
 167         struct pvt *pvt = (struct pvt *)this->private;
 168         struct irs_sv *sv;
 169
 170         pvt->rule = pvt->rules;
 171         if (pvt->rule) {
 172                 sv = pvt->rule->inst->sv;
 173                 (*sv->rewind)(sv);
 174         }
 175 }
 176
 177 static void
 178 sv_minimize(struct irs_sv *this) {
 179         struct pvt *pvt = (struct pvt *)this->private;
 180         struct irs_rule *rule;
 181
 182         for (rule = pvt->rules; rule != NULL; rule = rule->next) {
 183                 struct irs_sv *sv = rule->inst->sv;
 184
 185                 (*sv->minimize)(sv);
 186         }
 187 }
 188
 189 static struct __res_state *
 190 sv_res_get(struct irs_sv *this) {
 191         struct pvt *pvt = (struct pvt *)this->private;
 192
 193         if (!pvt->res) {
 194                 struct __res_state *res;
 195                 res = (struct __res_state *)malloc(sizeof *res);
 196                 if (!res) {
 197                         errno = ENOMEM;
 198                         return (NULL);
 199                 }
 200                 memset(res, 0, sizeof *res);
 201                 sv_res_set(this, res, free);
 202         }
 203
 204         return (pvt->res);
 205 }
 206
 207 static void
 208 sv_res_set(struct irs_sv *this, struct __res_state *res,
 209                 void (*free_res)(void *)) {
 210         struct pvt *pvt = (struct pvt *)this->private;
 211         struct irs_rule *rule;
 212
 213         if (pvt->res && pvt->free_res) {
 214                 res_nclose(pvt->res);
 215                 (*pvt->free_res)(pvt->res);
 216         }
 217
 218         pvt->res = res;
 219         pvt->free_res = free_res;
 220
 221         for (rule = pvt->rules; rule != NULL; rule = rule->next) {
 222                 struct irs_sv *sv = rule->inst->sv;
 223
 224                 if (sv->res_set)
 225                         (*sv->res_set)(sv, pvt->res, NULL);
 226         }
 227 }