# coding:utf-8
import optparse
from socket import *


def scan(tgtHost, tgtPort):
    try:
        sd = socket(AF_INET, SOCK_STREAM)
        sd.connect((tgtHost, tgtPort))

        # sd.send("wgPython\r\n")
        print('111')
        result = sd.recv(1024)
        print('[-] ip=%s ,port=%d  /tcp open' % (tgtHost, tgtPort))
        print('[+] ' + str(result))
        sd.close()
    except:
        print('[-] ip=%s ,port=%d  /tcp closed' % (tgtHost, tgtPort))


def portScan(tgtHost, tgtPorts):
    try:
        tgtip = gethostbyname(tgtHost)
    except:
        print("[-]Cannot relove '%s' Unknow host :" % tgtHost)
        return
    try:
        tgtName = gethostbyname(tgtip)
        print('\n [+] Scan Result for: ' + tgtName[0])
    except:
        print('\n [+] Scan Result for: ' + tgtip)
    setdefaulttimeout(2)
    # for tgtPort in tgtPorts:
    print('Scanning port' + str(tgtPorts))
    scan(tgtHost, tgtPorts)


def main():
    usage = "myhelp [-p <port>] [-f [filename] [-H <IP>] ]"
    parser = optparse.OptionParser(usage)
    parser.add_option("-H", dest='tgthost', type="string", help="specify target host")
    parser.add_option('-p', dest="tgtPort", type="int", help="specify target port")
    (options, args) = parser.parse_args()
    tgthost = options.tgthost
    tgtports = options.tgtPort
    if (tgthost == None) | (tgtports == None):
        print(parser.usage, '111')
        exit(0)
    portScan(tgthost, tgtports)


if __name__ == '__main__':
    main()