Adding suport to custom tcp port

checker.py

@@ -31,13 +31,21 @@
 }
 
 
-if len(sys.argv) != 2:
-	print("{} <ip>".format(sys.argv[0]))
+if len(sys.argv) < 2:
+	print("{} <ip> [port]".format(sys.argv[0]))
 	sys.exit(1)
 
 target = sys.argv[1]
+port = 445
 
-conn = MYSMB(target)
+try:
+	if sys.argv[2] != '':
+		port = int(sys.argv[2])
+except:
+	pass
+
+print('Trying to connect to %s:%d' % (target, port))
+conn = MYSMB(target, port)
 try:
 	conn.login(USERNAME, PASSWORD)
 except smb.SessionError as e:

mysmb.py

@@ -104,7 +104,7 @@ def _setup_login_packet_hook(maxBufferSize):
 
 
 class MYSMB(smb.SMB):
-	def __init__(self, remote_host, use_ntlmv2=True, timeout=8):
+	def __init__(self, remote_host, remote_port=445, use_ntlmv2=True, timeout=8):
 		self.__use_ntlmv2 = use_ntlmv2
 		self._default_tid = 0
 		self._pid = os.getpid() & 0xffff
@@ -115,7 +115,7 @@ def __init__(self, remote_host, use_ntlmv2=True, timeout=8):
 		self._last_tid = 0  # last tid from connect_tree()
 		self._last_fid = 0  # last fid from nt_create_andx()
 		self._smbConn = None
-		smb.SMB.__init__(self, remote_host, remote_host, timeout=timeout)
+		smb.SMB.__init__(self, remote_host, remote_host, sess_port=remote_port, timeout=timeout)
 
 	def set_pid(self, pid):
 		self._pid = pid