From 720d57b9d87b12640f578010ed4093cf24dde5d3 Mon Sep 17 00:00:00 2001
From: Tom Tucker <tom@opengridcomputing.com>
Date: Sun, 23 Nov 2008 09:58:08 -0600
Subject: [PATCH] --- yaml --- r: 119661 b: refs/heads/master c:
 2da2c21d7508d34bc6d600df665d84871b65d2b9 h: refs/heads/master i:   119659:
 0949c8cdcbc6872d8205929d8718a96db468e68f v: v3

---
 [refs]                     | 2 +-
 trunk/net/sunrpc/svcsock.c | 9 +++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/[refs] b/[refs]
index a416863bfe4e..48d5ccf671c4 100644
--- a/[refs]
+++ b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: 2c5e76158fcea6e3b9536a74efa7b5e2e846d374
+refs/heads/master: 2da2c21d7508d34bc6d600df665d84871b65d2b9
diff --git a/trunk/net/sunrpc/svcsock.c b/trunk/net/sunrpc/svcsock.c
index 95293f549e9c..a1951dcc5776 100644
--- a/trunk/net/sunrpc/svcsock.c
+++ b/trunk/net/sunrpc/svcsock.c
@@ -1183,7 +1183,11 @@ int svc_addsock(struct svc_serv *serv,
 	else if (so->state > SS_UNCONNECTED)
 		err = -EISCONN;
 	else {
-		svsk = svc_setup_socket(serv, so, &err, SVC_SOCK_DEFAULTS);
+		if (!try_module_get(THIS_MODULE))
+			err = -ENOENT;
+		else
+			svsk = svc_setup_socket(serv, so, &err,
+						SVC_SOCK_DEFAULTS);
 		if (svsk) {
 			struct sockaddr_storage addr;
 			struct sockaddr *sin = (struct sockaddr *)&addr;
@@ -1196,7 +1200,8 @@ int svc_addsock(struct svc_serv *serv,
 			spin_unlock_bh(&serv->sv_lock);
 			svc_xprt_received(&svsk->sk_xprt);
 			err = 0;
-		}
+		} else
+			module_put(THIS_MODULE);
 	}
 	if (err) {
 		sockfd_put(so);