[ ADD ] Added port range specification and single scan feature.

Author: encoderpie

scan.py

@@ -11,23 +11,35 @@
 d.set_background_title('Port Scanner With pythondialog - Encoderpie')
 d.add_persistent_args(['--no-nl-expand'])
 
-# Port scan def
-def portScan():
-    startPort = 1
-    endPort = 65535
+# Help menu def
+def helpMenu():
+    d.msgbox('''
+The scanning process starts when you enter the \
+IP address or hostname and press 'OK'.
+''', width=0, height=0, title='Help')
+
+# Default ports
+defaultStartPort = 1
+defaultEndPort = 65534
+
+# Port scan
+def portScan(targetInput, startPort, endPort):
     try:
-        target = socket.gethostbyname(user_input)
+        target = socket.gethostbyname(targetInput)
+        scanningPortsText = f'Scanning port(s): {startPort} - {endPort}'
+        if (startPort == endPort):
+            scanningPortsText = f'Scanning port: {startPort}'
         try:
-            gauge_text = '''
-Scanning Target: {} \
-
-Scanning started at: {}
-Open ports:
-            '''.format(target, str(datetime.now()))
-            d.gauge_start(text=gauge_text, height=0, width=0, percent=0, title='Ports Scanning...')
+            gauge_text = f'''
+Scanning target: {target} 
+Scanning started at: {str(datetime.now())}
+{scanningPortsText}
+Open port(s):
+            '''
+            d.gauge_start(text=gauge_text, height=0, width=0, percent=0, title='Port(s) Scanning...')
             openPorts = []
             startScanTime = timer()
-            for port in range(startPort, endPort):
+            for port in range(startPort, endPort + 1):
                 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                 socket.setdefaulttimeout(1)
                 result = s.connect_ex((target,port))
@@ -40,12 +52,16 @@ def portScan():
             d.gauge_update(100)
             d.gauge_stop()
             openPortsText = ''
+            if (len(openPorts) == 0):
+                openPortsText = f'{openPortsText}\n   The open port was not found.'
             for openport in openPorts:
                 openPortsText = f'{openPortsText}\n   {str(openport)} is open port'
-            openPortsText = f'{openPortsText}\nPort scan process took {str(int(startfinishDiffTime))} seconds'
-            msg = f'Open ports: {openPortsText}'
+            startEndTimeDiffText = int(startfinishDiffTime)
+            if (startEndTimeDiffText > 0):
+                openPortsText = f'{openPortsText}\nPort scan process took {str(startEndTimeDiffText)} seconds'
+            msg = f'Port scanning completed!\nOpen port(s): {openPortsText}'
         except KeyboardInterrupt:
-            msg = 'Exiting Program.'
+            msg = 'Goodbye.'
         except socket.gaierror:
             msg = 'Hostname Could Not Be Resolved.'
         except socket.error:
@@ -55,40 +71,56 @@ def portScan():
     d.msgbox(f'{msg}\nExiting!', width=0, height=0, title='Result')  
     sys.exit(0)
 
-# Help menu def
-def helpMenu():
-    d.msgbox(''' \
+# PORT input menu 
+def portInputScreen(ip):
+    userPortFormat = d.menu(f'Target IP/Hostname: {str(ip)}\nSelect the port scanning process.',
+                            choices=[
+                                ('1.', f'Scan all ports between {defaultStartPort} and {defaultEndPort}'), 
+                                ('2.', 'Scan a single port'), 
+                                ('3.', 'Scan all ports in the specified range')],
+                            height=None, width=None, menu_height=None, title='Port scanning process')[1]
+    if (userPortFormat == '1.'):
+        portScan(ip, defaultStartPort, defaultEndPort)
+    elif (userPortFormat == '2.'):
+        code, userPortSingle = d.inputbox('Input port', init='80', width=0, height=0, title='Port', help_button=True)
+        if code == d.OK:
+            portScan(ip, int(userPortSingle), int(userPortSingle))
+    elif (userPortFormat == '3.'):
+        code, userPortRange = d.inputbox('Input port range', init='80 443', width=0, height=0, title='Port range', help_button=True)
+        if code == d.OK:
+            portRangeParsed = userPortRange.split(' ')
+            portScan(ip, int(portRangeParsed[0]), int(portRangeParsed[1]) - 1)
+    if code == d.CANCEL:
+        msg = 'You pressed the Cancel button in the previous menu.'
+    elif code == d.ESC:
+        msg = 'You pressed the Escape key in the previous menu.'
+    elif code == d.HELP:
+        helpMenu()
+    d.msgbox(f'{msg}\nExiting!', width=0, height=0, title='Exiting')
+    sys.exit(0)
 
-Welcome to Port Scanner With pythondialog, \
-Press the 'OK' button at the bottom to start scanning the IP, \
-press the 'help' button at the bottom to get help.
-''', width=0, height=0, title='Help')
+# IP input menu
+def ipInputScreen():
+    code, target = d.inputbox('Input target IP or Host address', init='192.168.', width=0, height=0, title='IP or Host name', help_button=True)
+    if code == d.OK:
+        portInputScreen(target)
+    elif code == d.CANCEL:
+        msg = 'You pressed the Cancel button in the previous menu.'
+    elif code == d.ESC:
+        msg = 'You pressed the Escape key in the previous menu.'
+    elif code == d.HELP:
+        helpMenu()
+    d.msgbox(f'{msg}\nExiting!', width=0, height=0, title='Exiting')
+    sys.exit(0)
 
 # Welcome screen
-welcomeScreen = d.msgbox(''' \
-
+welcomeScreen = d.msgbox('''
 Welcome to Port Scanner With pythondialog, \
 Press the 'OK' button at the bottom to start scanning the IP, \
 press the 'help' button at the bottom to get help.
 ''', width=0, height=0, title='Port Scanner With pythondialog', help_button=True)
-
-if welcomeScreen == d.HELP:
+if welcomeScreen == d.OK:
+    ipInputScreen()
+elif welcomeScreen == d.HELP:
     helpMenu()
-
-# IP, PORT input
-code, user_input = d.inputbox('Input target IP or Host address', init='192.168.', width=0, height=0, title='IP or Host name', help_button=True)
-
-if code == d.OK:
-    portScan()
-elif code == d.CANCEL:
-    msg = 'You chose the Cancel button in the previous dialog.'
-elif code == d.ESC:
-    msg = 'You pressed the Escape key in the previous dialog.'
-elif code == d.HELP:
-    helpMenu()
-else:
-    msg = 'Unexpected exit code from d.inputbox(). Please report a bug.'
-
-d.msgbox(f'{msg}\nExiting!', width=0, height=0, title='Exiting')  
-
 sys.exit(0)