lib/isc/portset.c

1 /*	$NetBSD: portset.c,v 1.1 2024/02/18 20:57:50 christos Exp $	*/
22 #include <isc/portset.h>
30  * Internal representation of portset.  It's an array of 32-bit integers, each
40 portset_isset(isc_portset_t *portset, in_port_t port) {
41 	return ((portset->buf[port >> 5] & ((uint32_t)1 << (port & 31))) != 0);
45 portset_add(isc_portset_t *portset, in_port_t port) {
46 	if (!portset_isset(portset, port)) {
47 		portset->nports++;
48 		portset->buf[port >> 5] |= ((uint32_t)1 << (port & 31));
53 portset_remove(isc_portset_t *portset, in_port_t port) {
54 	if (portset_isset(portset, port)) {
55 		portset->nports--;
56 		portset->buf[port >> 5] &= ~((uint32_t)1 << (port & 31));
62 	isc_portset_t *portset;
66 	portset = isc_mem_get(mctx, sizeof(*portset));
69 	memset(portset, 0, sizeof(*portset));
70 	*portsetp = portset;
77 	isc_portset_t *portset;
80 	portset = *portsetp;
82 	isc_mem_put(mctx, portset, sizeof(*portset));
86 isc_portset_isset(isc_portset_t *portset, in_port_t port) {
87 	REQUIRE(portset != NULL);
89 	return (portset_isset(portset, port));
93 isc_portset_nports(isc_portset_t *portset) {
94 	REQUIRE(portset != NULL);
96 	return (portset->nports);
100 isc_portset_add(isc_portset_t *portset, in_port_t port) {
101 	REQUIRE(portset != NULL);
103 	portset_add(portset, port);
107 isc_portset_remove(isc_portset_t *portset, in_port_t port) {
108 	portset_remove(portset, port);
112 isc_portset_addrange(isc_portset_t *portset, in_port_t port_lo,
116 	REQUIRE(portset != NULL);
121 		portset_add(portset, p);
126 isc_portset_removerange(isc_portset_t *portset, in_port_t port_lo,
130 	REQUIRE(portset != NULL);
135 		portset_remove(portset, p);