/*      $NetBSD: gen_pr.c,v 1.1.1.2 2012/09/09 16:07:56 christos Exp $  */

/*
* Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
* Copyright (c) 1996,1999 by Internet Software Consortium.
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#if !defined(LINT) && !defined(CODECENTER)
static const char rcsid[] = "Id: gen_pr.c,v 1.3 2005/04/27 04:56:24 sra Exp ";
#endif

/* Imports */

#include "port_before.h"

#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/nameser.h>

#include <errno.h>
#include <resolv.h>
#include <stdlib.h>
#include <string.h>

#include <isc/memcluster.h>
#include <irs.h>

#include "port_after.h"

#include "irs_p.h"
#include "gen_p.h"

/* Types */

struct pvt {
       struct irs_rule *       rules;
       struct irs_rule *       rule;
       struct __res_state *    res;
       void                    (*free_res)(void *);
};

/* Forward */

static void                     pr_close(struct irs_pr*);
static struct protoent *        pr_next(struct irs_pr *);
static struct protoent *        pr_byname(struct irs_pr *, const char *);
static struct protoent *        pr_bynumber(struct irs_pr *, int);
static void                     pr_rewind(struct irs_pr *);
static void                     pr_minimize(struct irs_pr *);
static struct __res_state *     pr_res_get(struct irs_pr *);
static void                     pr_res_set(struct irs_pr *,
                                          struct __res_state *,
                                          void (*)(void *));

/* Public */

struct irs_pr *
irs_gen_pr(struct irs_acc *this) {
       struct gen_p *accpvt = (struct gen_p *)this->private;
       struct irs_pr *pr;
       struct pvt *pvt;

       if (!(pr = memget(sizeof *pr))) {
               errno = ENOMEM;
               return (NULL);
       }
       memset(pr, 0x5e, sizeof *pr);
       if (!(pvt = memget(sizeof *pvt))) {
               memput(pr, sizeof *pr);
               errno = ENOMEM;
               return (NULL);
       }
       memset(pvt, 0, sizeof *pvt);
       pvt->rules = accpvt->map_rules[irs_pr];
       pvt->rule = pvt->rules;
       pr->private = pvt;
       pr->close = pr_close;
       pr->next = pr_next;
       pr->byname = pr_byname;
       pr->bynumber = pr_bynumber;
       pr->rewind = pr_rewind;
       pr->minimize = pr_minimize;
       pr->res_get = pr_res_get;
       pr->res_set = pr_res_set;
       return (pr);
}

/* Methods */

static void
pr_close(struct irs_pr *this) {
       struct pvt *pvt = (struct pvt *)this->private;

       memput(pvt, sizeof *pvt);
       memput(this, sizeof *this);
}

static struct protoent *
pr_next(struct irs_pr *this) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct protoent *rval;
       struct irs_pr *pr;

       while (pvt->rule) {
               pr = pvt->rule->inst->pr;
               rval = (*pr->next)(pr);
               if (rval)
                       return (rval);
               if (!(pvt->rules->flags & IRS_CONTINUE))
                       break;
               pvt->rule = pvt->rule->next;
               if (pvt->rule) {
                       pr = pvt->rule->inst->pr;
                       (*pr->rewind)(pr);
               }
       }
       return (NULL);
}

static struct protoent *
pr_byname(struct irs_pr *this, const char *name) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct irs_rule *rule;
       struct protoent *rval;
       struct irs_pr *pr;

       rval = NULL;
       for (rule = pvt->rules; rule; rule = rule->next) {
               pr = rule->inst->pr;
               rval = (*pr->byname)(pr, name);
               if (rval || !(rule->flags & IRS_CONTINUE))
                       break;
       }
       return (rval);
}

static struct protoent *
pr_bynumber(struct irs_pr *this, int proto) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct irs_rule *rule;
       struct protoent *rval;
       struct irs_pr *pr;

       rval = NULL;
       for (rule = pvt->rules; rule; rule = rule->next) {
               pr = rule->inst->pr;
               rval = (*pr->bynumber)(pr, proto);
               if (rval || !(rule->flags & IRS_CONTINUE))
                       break;
       }
       return (rval);
}

static void
pr_rewind(struct irs_pr *this) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct irs_pr *pr;

       pvt->rule = pvt->rules;
       if (pvt->rule) {
               pr = pvt->rule->inst->pr;
               (*pr->rewind)(pr);
       }
}

static void
pr_minimize(struct irs_pr *this) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct irs_rule *rule;

       for (rule = pvt->rules; rule != NULL; rule = rule->next) {
               struct irs_pr *pr = rule->inst->pr;

               (*pr->minimize)(pr);
       }
}

static struct __res_state *
pr_res_get(struct irs_pr *this) {
       struct pvt *pvt = (struct pvt *)this->private;

       if (!pvt->res) {
               struct __res_state *res;
               res = (struct __res_state *)malloc(sizeof *res);
               if (!res) {
                       errno = ENOMEM;
                       return (NULL);
               }
               memset(res, 0, sizeof *res);
               pr_res_set(this, res, free);
       }

       return (pvt->res);
}

static void
pr_res_set(struct irs_pr *this, struct __res_state *res,
               void (*free_res)(void *)) {
       struct pvt *pvt = (struct pvt *)this->private;
       struct irs_rule *rule;

       if (pvt->res && pvt->free_res) {
               res_nclose(pvt->res);
               (*pvt->free_res)(pvt->res);
       }

       pvt->res = res;
       pvt->free_res = free_res;

       for (rule = pvt->rules; rule != NULL; rule = rule->next) {
               struct irs_pr *pr = rule->inst->pr;

               if (pr->res_set)
                       (*pr->res_set)(pr, pvt->res, NULL);
       }
}

/*! \file */

You are viewing proxied material from ftp.icm.edu.pl. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.