#!/usr/bin/env python
#coding:utf-8
# Author: Zeroh
# Modified: Cobranail

import re
import sys
import Queue
import threading
import optparse
import requests
from IPy import IP

printLock = threading.Semaphore(1)  #lock Screen print
TimeOut = (0.5,1)  #request timeout, for slow connection

#User-Agent
header = {'User-Agent' : 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/44.0.2403.125 Safari/537.36','Connection':'close'}

class scan():

  def __init__(self,cidr, threads_num, ports):
    self.threads_num = threads_num
    self.ports=[]
    ports_set=[]
    if ',' in ports:
      ports_set=ports.split(',')
    else:
      ports_set=[ports]
    #print ports_set
    for ps in ports_set:
      if '-' in ps:
        pp = ps.split('-')
        self.ports = self.ports+range(int(pp[0]),int(pp[1])+1)
      else:
        self.ports.append(ps)
    #print self.ports
    self.cidr = IP(cidr)
	#build ip queue
    self.IPs = Queue.Queue()
    for ip in self.cidr:
      for port in self.ports:
        ipp = str(ip)+':'+str(port)
        #print ipp
        self.IPs.put(ipp)

  def request(self):
    with threading.Lock():
      while self.IPs.qsize() > 0:
        ip = self.IPs.get()
        #print str(ip)
        try:
          r = requests.Session().get('http://'+str(ip),headers=header,timeout=TimeOut)
          status = r.status_code
          title = re.search(r'<title>(.*)</title>', r.text) #get the title
          if title:
            title = title.group(1).strip().strip("\r").strip("\n")[:30]
          else:
            title = "None"
          banner = ''
          try:
            banner += r.headers['Server'][:20] #get the server banner
          except:pass
          printLock.acquire()
          print "|%-24s|%-6s|%-20s|%-30s|" % (ip,status,banner,title)
          print "+------------------------+------+--------------------+------------------------------+"

          #Save log
          with open("./log/"+self.cidr.strNormal(3)+".log",'a') as f:
            f.write(ip+"\n")

        except Exception,e:
          printLock.acquire()
        finally:
          printLock.release()

  #Multi thread
  def run(self):
    for i in range(self.threads_num):
      t = threading.Thread(target=self.request)
      t.start()

if __name__ == "__main__":
  parser = optparse.OptionParser("Usage: %prog [options] target")
  parser.add_option("-t", "--thread", dest = "threads_num",
    default = 10, type = "int",
    help = "[optional]number of  theads,default=10")
  parser.add_option("-p", "--ports", dest = "ports",
    default = '80', type = "string",
    help = "[optional]number of  theads,default=10")
  (options, args) = parser.parse_args()
  if len(args) < 1:
    parser.print_help()
    sys.exit(0)

  print "+------------------------+------+--------------------+------------------------------+"
  print "|          IP            |Status|       Server       |            Title             |"
  print "+------------------------+------+--------------------+------------------------------+"

  s = scan(cidr=args[0],threads_num=options.threads_num, ports=options.ports)
  s.run()