Replace atomic operations in bin/named/client.c with isc_refcount reference counting

diff --git a/bin/named/client.c b/bin/named/client.c
index 845326abc059e37667a5b1c01a670085ed8d1a9c..29fecadca8c50860c3363bb5f96e68283b70bffc 100644 (file)
--- a/bin/named/client.c
+++ b/bin/named/client.c
@@ -402,12 +402,10 @@ tcpconn_detach(ns_client_t *client) {
 static void
 mark_tcp_active(ns_client_t *client, bool active) {
        if (active && !client->tcpactive) {
-               isc_atomic_xadd(&client->interface->ntcpactive, 1);
+               isc_refcount_increment0(&client->interface->ntcpactive, NULL);
                client->tcpactive = active;
        } else if (!active && client->tcpactive) {
-               uint32_t old =
-                       isc_atomic_xadd(&client->interface->ntcpactive, -1);
-               INSIST(old > 0);
+               isc_refcount_decrement(&client->interface->ntcpactive, NULL);
                client->tcpactive = active;
        }
 }
@@ -554,7 +552,7 @@ exit_check(ns_client_t *client) {
                if (client->mortal && TCP_CLIENT(client) &&
                    client->newstate != NS_CLIENTSTATE_FREED &&
                    !ns_g_clienttest &&
-                   isc_atomic_xadd(&client->interface->ntcpaccepting, 0) == 0)
+                   isc_refcount_current(&client->interface->ntcpaccepting) == 0)
                {
                        /* Nobody else is accepting */
                        client->mortal = false;
@@ -3328,7 +3326,6 @@ client_newconn(isc_task_t *task, isc_event_t *event) {
        isc_result_t result;
        ns_client_t *client = event->ev_arg;
        isc_socket_newconnev_t *nevent = (isc_socket_newconnev_t *)event;
-       uint32_t old;
 
        REQUIRE(event->ev_type == ISC_SOCKEVENT_NEWCONN);
        REQUIRE(NS_CLIENT_VALID(client));
@@ -3348,8 +3345,7 @@ client_newconn(isc_task_t *task, isc_event_t *event) {
        INSIST(client->naccepts == 1);
        client->naccepts--;
 
-       old = isc_atomic_xadd(&client->interface->ntcpaccepting, -1);
-       INSIST(old > 0);
+       isc_refcount_decrement(&client->interface->ntcpaccepting, NULL);
 
        /*
         * We must take ownership of the new socket before the exit
@@ -3480,8 +3476,8 @@ client_accept(ns_client_t *client) {
                 * quota is tcp-clients plus the number of listening
                 * interfaces plus 1.)
                 */
-               exit = (isc_atomic_xadd(&client->interface->ntcpactive, 0) >
-                       (client->tcpactive ? 1 : 0));
+               exit = (isc_refcount_current(&client->interface->ntcpactive) >
+                       (client->tcpactive ? 1U : 0U));
                if (exit) {
                        client->newstate = NS_CLIENTSTATE_INACTIVE;
                        (void)exit_check(client);
@@ -3539,7 +3535,7 @@ client_accept(ns_client_t *client) {
         * listening for connections itself to prevent the interface
         * going dead.
         */
-       isc_atomic_xadd(&client->interface->ntcpaccepting, 1);
+       isc_refcount_increment0(&client->interface->ntcpaccepting, NULL);
 }
 
 static void

diff --git a/bin/named/include/named/interfacemgr.h b/bin/named/include/named/interfacemgr.h
index 3535ef22a8e79893975c5731f6ebdd1c21b17198..6e10f210fdb29637c4bdb24c58d05f8dd80ebcf3 100644 (file)
--- a/bin/named/include/named/interfacemgr.h
+++ b/bin/named/include/named/interfacemgr.h
@@ -45,6 +45,7 @@
 #include <isc/magic.h>
 #include <isc/mem.h>
 #include <isc/socket.h>
+#include <isc/refcount.h>
 
 #include <dns/result.h>
 
@@ -75,11 +76,11 @@ struct ns_interface {
                                                /*%< UDP dispatchers. */
        isc_socket_t *          tcpsocket;      /*%< TCP socket. */
        isc_dscp_t              dscp;           /*%< "listen-on" DSCP value */
-       int32_t                 ntcpaccepting;  /*%< Number of clients
+       isc_refcount_t          ntcpaccepting;  /*%< Number of clients
                                                     ready to accept new
                                                     TCP connections on this
                                                     interface */
-       int32_t                 ntcpactive;     /*%< Number of clients
+       isc_refcount_t          ntcpactive;     /*%< Number of clients
                                                     servicing TCP queries
                                                     (whether accepting or
                                                     connected) */

diff --git a/bin/named/interfacemgr.c b/bin/named/interfacemgr.c
index d9f6df5802c570697bcb438a46d5a44b01f7e390..135533be6be85de698e392134a3b201e4adabb6a 100644 (file)
--- a/bin/named/interfacemgr.c
+++ b/bin/named/interfacemgr.c
@@ -386,8 +386,8 @@ ns_interface_create(ns_interfacemgr_t *mgr, isc_sockaddr_t *addr,
         * connections will be handled in parallel even though there is
         * only one client initially.
         */
-       ifp->ntcpaccepting = 0;
-       ifp->ntcpactive = 0;
+       isc_refcount_init(&ifp->ntcpaccepting, 0);
+       isc_refcount_init(&ifp->ntcpactive, 0);
 
        ifp->nudpdispatch = 0;
 
@@ -618,6 +618,9 @@ ns_interface_destroy(ns_interface_t *ifp) {
 
        ns_interfacemgr_detach(&ifp->mgr);
 
+       isc_refcount_destroy(&ifp->ntcpactive);
+       isc_refcount_destroy(&ifp->ntcpaccepting);
+
        ifp->magic = 0;
        isc_mem_put(mctx, ifp, sizeof(*ifp));
 }