from socket import *

compile_ip = re.compile(  
    '^(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|[1-9])\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)$')

# -*- coding: utf-8 -*-
from socket import *
import re
from pip._vendor.distlib.compat import raw_input

def portscaner(host,port):         
  try:
    s = socket(AF_INET, SOCK_STREAM) # 實例化socket
    s.connect((host,port))    #對對應(yīng)主機的對應(yīng)端口發(fā)起連接
    print('主機%s的%s端口處于打開狀態(tài)'%(host,port))
    s.close()   #關(guān)閉連接
    Export_Data(port)   #調(diào)用Export_Date
  except Exception as e:
    pass   #利用異常處理，對無法建立連接的端口pass掉


def ChecK_Host(host): #利用正側(cè)表達(dá)式匹配host，判斷是否符合IP地址的格式  并且輸入不為空 
  compile_ip = re.compile(  
    '^(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|[1-9])\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)\.(1\d{2}|2[0-4]\d|25[0-5]|[1-9]\d|\d)$')
  if compile_ip.match(host) and len(host)!=0:  #host符合IP地址的格式且不為空值返回True，否則返回False
    return True
  else:
    return False
    

def Export_Data(port):   #導(dǎo)出數(shù)據(jù)
  port = str(port)     #將port轉(zhuǎn)換為字符串類型
  date = open('Active_Port.txt', 'a') #在文件尾部寫入數(shù)據(jù)，Active_Port.txt,若不存在就新建文件
  date.write(port+'\n')      #將port并行寫入文件
  date.close()    #不要忘了關(guān)閉文件



def main(host):   #主模塊
  if ChecK_Host(host):   #通過ChecK_Host()函數(shù)return 的布爾值進(jìn)行判斷輸入的IP地址是否有誤
    for port in range(1,65536):  #這里可以改成1024
      portscaner(host, port)
    print('\n')
    print('已將結(jié)果導(dǎo)出到 Active_Port.txt')
  else:
    print('IP地址無錯誤，請重新輸入?。?！')



if __name__ == '__main__':
  print('<' * 12 + '單線程端口掃描器' + '>' * 12)
  host = input('請輸入有效的目標(biāo)IPv4地址：')
  main(host)

raw_input('Press Enter to exit...')    #當(dāng)腳本執(zhí)行完后，等待鍵入Enter回車才退出程序（可有可無，主要是為了封裝成exe可執(zhí)行文件后，執(zhí)行完程序dos界面不自動退出）