Bug 7284 - Properly lock ztdynamic modules under kernel 2.6

git-svn-id: http://svn.digium.com/svn/zaptel/trunk@1155 5390a7c7-147a-4af0-8ec9-7488f05a26cb

1 files changed, 5 insertions, 0 deletions

diff --git a/ztd-loc.c b/ztd-loc.c
index 100bd9f..0458293 100644
--- a/ztd-loc.c
+++ b/ztd-loc.c
@@ -162,6 +162,8 @@ static int digit2int(char d)
 		printk("TDMoL: Removed interface for %s, key %d id %d\n", z->span->name, z->key, z->id);
 #ifndef LINUX26
 		MOD_DEC_USE_COUNT;
+#else
+		module_put(THIS_MODULE);
 #endif
 		kfree(z);
 	}
@@ -229,6 +231,9 @@ static int digit2int(char d)
 		spin_unlock_irqrestore(&zlock, flags);
 #ifndef LINUX26
 		MOD_INC_USE_COUNT;
+#else
+		if(!try_module_get(THIS_MODULE))
+			printk("TDMoL: Unable to increment module use count\n");
 #endif
 
 		printk("TDMoL: Added new interface for %s, key %d id %d\n", span->name, z->key, z->id);