diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms index 649f2c308cf891d9636fff6fca91b4420e7fd61f..0d7c9757b764aec9c103ec6f1eb236d47309bd2f 100644 --- 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 --- 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 --- 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 --- 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 --- 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,