Return to portset.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / ntp / lib / isc

Annotation of embedaddon/ntp/lib/isc/portset.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2008  Internet Systems Consortium, Inc. ("ISC")
        !             3:  *
        !             4:  * Permission to use, copy, modify, and/or distribute this software for any
        !             5:  * purpose with or without fee is hereby granted, provided that the above
        !             6:  * copyright notice and this permission notice appear in all copies.
        !             7:  *
        !             8:  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
        !             9:  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
        !            10:  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
        !            11:  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
        !            12:  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
        !            13:  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
        !            14:  * PERFORMANCE OF THIS SOFTWARE.
        !            15:  */
        !            16: 
        !            17: /* $Id: portset.c,v 1.4 2008/06/24 23:24:35 marka Exp $ */
        !            18: 
        !            19: /*! \file */
        !            20: 
        !            21: #include <config.h>
        !            22: 
        !            23: #include <isc/mem.h>
        !            24: #include <isc/portset.h>
        !            25: #include <isc/string.h>
        !            26: #include <isc/types.h>
        !            27: #include <isc/util.h>
        !            28: 
        !            29: #define ISC_PORTSET_BUFSIZE (65536 / (sizeof(isc_uint32_t) * 8))
        !            30: 
        !            31: /*%
        !            32:  * Internal representation of portset.  It's an array of 32-bit integers, each
        !            33:  * bit corresponding to a single port in the ascending order.  For example,
        !            34:  * the second most significant bit of buf[0] corresponds to port 1.
        !            35:  */
        !            36: struct isc_portset {
        !            37:        unsigned int nports;    /*%< number of ports in the set */
        !            38:        isc_uint32_t buf[ISC_PORTSET_BUFSIZE];
        !            39: };
        !            40: 
        !            41: static inline isc_boolean_t
        !            42: portset_isset(isc_portset_t *portset, in_port_t port) {
        !            43:        return (ISC_TF((portset->buf[port >> 5] & (1 << (port & 31))) != 0));
        !            44: }
        !            45: 
        !            46: static inline void
        !            47: portset_add(isc_portset_t *portset, in_port_t port) {
        !            48:        if (!portset_isset(portset, port)) {
        !            49:                portset->nports++;
        !            50:                portset->buf[port >> 5] |= (1 << (port & 31));
        !            51:        }
        !            52: }
        !            53: 
        !            54: static inline void
        !            55: portset_remove(isc_portset_t *portset, in_port_t port) {
        !            56:        if (portset_isset(portset, port)) {
        !            57:                portset->nports--;
        !            58:                portset->buf[port >> 5] &= ~(1 << (port & 31));
        !            59:        }
        !            60: }
        !            61: 
        !            62: isc_result_t
        !            63: isc_portset_create(isc_mem_t *mctx, isc_portset_t **portsetp) {
        !            64:        isc_portset_t *portset;
        !            65: 
        !            66:        REQUIRE(portsetp != NULL && *portsetp == NULL);
        !            67: 
        !            68:        portset = isc_mem_get(mctx, sizeof(*portset));
        !            69:        if (portset == NULL)
        !            70:                return (ISC_R_NOMEMORY);
        !            71: 
        !            72:        /* Make the set 'empty' by default */
        !            73:        memset(portset, 0, sizeof(*portset));
        !            74:        *portsetp = portset;
        !            75: 
        !            76:        return (ISC_R_SUCCESS);
        !            77: }
        !            78: 
        !            79: void
        !            80: isc_portset_destroy(isc_mem_t *mctx, isc_portset_t **portsetp) {
        !            81:        isc_portset_t *portset;
        !            82: 
        !            83:        REQUIRE(portsetp != NULL);
        !            84:        portset = *portsetp;
        !            85: 
        !            86:        isc_mem_put(mctx, portset, sizeof(*portset));
        !            87: }
        !            88: 
        !            89: isc_boolean_t
        !            90: isc_portset_isset(isc_portset_t *portset, in_port_t port) {
        !            91:        REQUIRE(portset != NULL);
        !            92: 
        !            93:        return (portset_isset(portset, port));
        !            94: }
        !            95: 
        !            96: unsigned int
        !            97: isc_portset_nports(isc_portset_t *portset) {
        !            98:        REQUIRE(portset != NULL);
        !            99: 
        !           100:        return (portset->nports);
        !           101: }
        !           102: 
        !           103: void
        !           104: isc_portset_add(isc_portset_t *portset, in_port_t port) {
        !           105:        REQUIRE(portset != NULL);
        !           106: 
        !           107:        portset_add(portset, port);
        !           108: }
        !           109: 
        !           110: void
        !           111: isc_portset_remove(isc_portset_t *portset, in_port_t port) {
        !           112:        portset_remove(portset, port);
        !           113: }
        !           114: 
        !           115: void
        !           116: isc_portset_addrange(isc_portset_t *portset, in_port_t port_lo,
        !           117:                     in_port_t port_hi)
        !           118: {
        !           119:        in_port_t p;
        !           120: 
        !           121:        REQUIRE(portset != NULL);
        !           122:        REQUIRE(port_lo <= port_hi);
        !           123: 
        !           124:        p = port_lo;
        !           125:        do {
        !           126:                portset_add(portset, p);
        !           127:        } while (p++ < port_hi);
        !           128: }
        !           129: 
        !           130: void
        !           131: isc_portset_removerange(isc_portset_t *portset, in_port_t port_lo,
        !           132:                        in_port_t port_hi)
        !           133: {
        !           134:        in_port_t p;
        !           135: 
        !           136:        REQUIRE(portset != NULL);
        !           137:        REQUIRE(port_lo <= port_hi);
        !           138: 
        !           139:        p = port_lo;
        !           140:        do {
        !           141:                portset_remove(portset, p);
        !           142:        } while (p++ < port_hi);
        !           143: }