From efcf9ac3de59e6651bd7e732f778506d9b1d3d80 Mon Sep 17 00:00:00 2001
From: hklbgd <hklbgd_assange@oosm.org>
Date: Sat, 1 Jun 2019 21:11:43 +0200
Subject: add getPortInfo function

---
 tmap | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'tmap')

diff --git a/tmap b/tmap
index 1d61f97..fac70d3 100755
--- a/tmap
+++ b/tmap
@@ -9,6 +9,28 @@ import threading
 
 VERSION='1.2.1'
 
+def getPortInfo(port, portfile):
+    """
+           Get info about service commonly used on this port
+    """
+    ## Read every line until EOF
+    while (line = f.readline()) != '':
+        ## Don't read lines that are commented out
+        if line[0] == '#':
+            pass
+
+        ## Split line using tab as delimiter and read the port number
+        split_line = line.split('\t')
+        portnum = split_line[1].split('/')[0]
+
+        ## If portnum equals port return the name of service
+        if int(portnum) != port:
+            pass
+        else:
+            return split_line[0]
+    return "portinfo_error"
+
+
 def getBanner(host, port, wait, notor):
     """
             Receive first 80 bytes from port, return string of received data
-- 
cgit v1.2.3