host value checking

1 files changed, 12 insertions, 4 deletions

diff --git a/tmap b/tmap
index 541e4e3..2ef3f71 100755
--- a/tmap
+++ b/tmap
@@ -104,11 +104,19 @@ def main():
 	## Scan each host in HOSTS list
 	for h in HOSTS:
 		if len(h.split("/")) == 2:
-			ips = IPv4Network(unicode(h))
-			for i in ips.hosts():
-				portScan(str(i), PORTS, WAIT_TIME, CLEARNET)
-		else:
+			try:
+				ips = IPv4Network(unicode(h))
+				for i in ips.hosts():
+					portScan(str(i), PORTS, WAIT_TIME, CLEARNET)
+			except:
+				parser.print_help()
+				exit()
+
+		elif len(h.split("/")) < 2:
 			portScan(h, PORTS, WAIT_TIME, CLEARNET)
+		else:
+			parser.print_help()
+			exit()
 
 	endTime = time.time()
 	totalTime = endTime - startTime