Change interaction when accepting new RPC client connections

Currently the virNetServerServicePtr is responsible for
creating the virNetServerClientPtr instance when accepting
a new connection. Change this so that the virNetServerServicePtr
merely gives virNetServerPtr a virNetSocketPtr instance. The
virNetServerPtr can then create the virNetServerClientPtr
as it desires

Signed-off-by: Daniel P. Berrange <berrange@redhat.com>

diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms
index 649f2c308cf891d9636fff6fca91b4420e7fd61f..0d7c9757b764aec9c103ec6f1eb236d47309bd2f 100644 (file)
--- a/src/libvirt_private.syms
+++ b/src/libvirt_private.syms
@@ -1492,8 +1492,6 @@ virNetServerKeepAliveRequired;
 virNetServerNew;
 virNetServerQuit;
 virNetServerRun;
-virNetServerServiceNewTCP;
-virNetServerServiceNewUNIX;
 virNetServerSetTLSContext;
 virNetServerUpdateServices;
 
@@ -1558,7 +1556,9 @@ virNetServerProgramUnknownError;
 # virnetserverservice.h
 virNetServerServiceClose;
 virNetServerServiceGetAuth;
+virNetServerServiceGetMaxRequests;
 virNetServerServiceGetPort;
+virNetServerServiceGetTLSContext;
 virNetServerServiceIsReadonly;
 virNetServerServiceNewTCP;
 virNetServerServiceNewUNIX;

diff --git a/src/rpc/virnetserver.c b/src/rpc/virnetserver.c
index 15abb5698f69980441daffc2c8a20a4ad167e227..c3eb2e5d9c61a5fbfb3affc842db82f9afe5e8b9 100644 (file)
--- a/src/rpc/virnetserver.c
+++ b/src/rpc/virnetserver.c
@@ -261,11 +261,12 @@ cleanup:
 }
 
 
-static int virNetServerDispatchNewClient(virNetServerServicePtr svc ATTRIBUTE_UNUSED,
-                                         virNetServerClientPtr client,
+static int virNetServerDispatchNewClient(virNetServerServicePtr svc,
+                                         virNetSocketPtr clientsock,
                                          void *opaque)
 {
     virNetServerPtr srv = opaque;
+    virNetServerClientPtr client = NULL;
 
     virNetServerLock(srv);
 
@@ -276,6 +277,13 @@ static int virNetServerDispatchNewClient(virNetServerServicePtr svc ATTRIBUTE_UN
         goto error;
     }
 
+    if (!(client = virNetServerClientNew(clientsock,
+                                         virNetServerServiceGetAuth(svc),
+                                         virNetServerServiceIsReadonly(svc),
+                                         virNetServerServiceGetMaxRequests(svc),
+                                         virNetServerServiceGetTLSContext(svc))))
+        goto error;
+
     if (virNetServerClientInit(client) < 0)
         goto error;
 
@@ -301,6 +309,8 @@ static int virNetServerDispatchNewClient(virNetServerServicePtr svc ATTRIBUTE_UN
     return 0;
 
 error:
+    virNetServerClientClose(client);
+    virObjectUnref(client);
     virNetServerUnlock(srv);
     return -1;
 }

diff --git a/src/rpc/virnetserverclient.c b/src/rpc/virnetserverclient.c
index d135b0fb31c14147e010f01a15015d2132824b8e..9f033c8841d9ba0359ff383c47ea40492aa37c15 100644 (file)
--- a/src/rpc/virnetserverclient.c
+++ b/src/rpc/virnetserverclient.c
@@ -357,7 +357,7 @@ virNetServerClientPtr virNetServerClientNew(virNetSocketPtr sock,
         return NULL;
     }
 
-    client->sock = sock;
+    client->sock = virObjectRef(sock);
     client->auth = auth;
     client->readonly = readonly;
     client->tlsCtxt = virObjectRef(tls);
@@ -385,8 +385,6 @@ virNetServerClientPtr virNetServerClientNew(virNetSocketPtr sock,
     return client;
 
 error:
-    /* XXX ref counting is better than this */
-    client->sock = NULL; /* Caller owns 'sock' upon failure */
     virObjectUnref(client);
     return NULL;
 }

diff --git a/src/rpc/virnetserverservice.c b/src/rpc/virnetserverservice.c
index 000b5b6b93d1f55299db8e1dcf530e3fbec74c6f..eda5ef91c10ff6d45791ba0ae3047456957b3124 100644 (file)
--- a/src/rpc/virnetserverservice.c
+++ b/src/rpc/virnetserverservice.c
@@ -69,40 +69,21 @@ static void virNetServerServiceAccept(virNetSocketPtr sock,
                                       void *opaque)
 {
     virNetServerServicePtr svc = opaque;
-    virNetServerClientPtr client = NULL;
     virNetSocketPtr clientsock = NULL;
 
     if (virNetSocketAccept(sock, &clientsock) < 0)
-        goto error;
+        goto cleanup;
 
     if (!clientsock) /* Connection already went away */
         goto cleanup;
 
-    if (!(client = virNetServerClientNew(clientsock,
-                                         svc->auth,
-                                         svc->readonly,
-                                         svc->nrequests_client_max,
-                                         svc->tls)))
-        goto error;
-
     if (!svc->dispatchFunc)
-        goto error;
-
-    if (svc->dispatchFunc(svc, client, svc->dispatchOpaque) < 0)
-        virNetServerClientClose(client);
+        goto cleanup;
 
-    virObjectUnref(client);
+    svc->dispatchFunc(svc, clientsock, svc->dispatchOpaque);
 
 cleanup:
-    return;
-
-error:
-    if (client) {
-        virNetServerClientClose(client);
-        virObjectUnref(client);
-    } else {
-        virObjectUnref(clientsock);
-    }
+    virObjectUnref(clientsock);
 }
 
 
@@ -240,6 +221,17 @@ bool virNetServerServiceIsReadonly(virNetServerServicePtr svc)
 }
 
 
+size_t virNetServerServiceGetMaxRequests(virNetServerServicePtr svc)
+{
+    return svc->nrequests_client_max;
+}
+
+virNetTLSContextPtr virNetServerServiceGetTLSContext(virNetServerServicePtr svc)
+{
+    return svc->tls;
+}
+
+
 void virNetServerServiceSetDispatcher(virNetServerServicePtr svc,
                                       virNetServerServiceDispatchFunc func,
                                       void *opaque)

diff --git a/src/rpc/virnetserverservice.h b/src/rpc/virnetserverservice.h
index 98fd396349b42b0f5e96d4b73ee9817f61d81d99..cb18e2d8041468e2ca5bf691049095161bc2f6c8 100644 (file)
--- a/src/rpc/virnetserverservice.h
+++ b/src/rpc/virnetserverservice.h
@@ -34,7 +34,7 @@ enum {
 };
 
 typedef int (*virNetServerServiceDispatchFunc)(virNetServerServicePtr svc,
-                                               virNetServerClientPtr client,
+                                               virNetSocketPtr sock,
                                                void *opaque);
 
 virNetServerServicePtr virNetServerServiceNewTCP(const char *nodename,
@@ -55,6 +55,8 @@ int virNetServerServiceGetPort(virNetServerServicePtr svc);
 
 int virNetServerServiceGetAuth(virNetServerServicePtr svc);
 bool virNetServerServiceIsReadonly(virNetServerServicePtr svc);
+size_t virNetServerServiceGetMaxRequests(virNetServerServicePtr svc);
+virNetTLSContextPtr virNetServerServiceGetTLSContext(virNetServerServicePtr svc);
 
 void virNetServerServiceSetDispatcher(virNetServerServicePtr svc,
                                       virNetServerServiceDispatchFunc func,