nmap_scan.py

from __future__ import print_function

import optparse  # Import the module

import nmap  # Import the module


# Script Name		: nmap_scan.py
# Author				: Craig Richards
# Created				: 24th May 2013
# Last Modified		:
# Version				: 1.0
# Modifications		:
# Description			: This scans my scripts directory and gives a count of the different types of scripts, you need nmap installed to run this


def nmapScan(tgtHost, tgtPort):  # Create the function, this fucntion does the scanning
    nmScan = nmap.PortScanner()
    nmScan.scan(tgtHost, tgtPort)
    state = nmScan[tgtHost]["tcp"][int(tgtPort)]["state"]
    print("[*] " + tgtHost + " tcp/" + tgtPort + " " + state)


def main():  # Main Program
    parser = optparse.OptionParser(
        "usage%prog " + "-H <host> -p <port>"
    )  # Display options/help if required
    parser.add_option("-H", dest="tgtHost", type="string", help="specify host")
    parser.add_option("-p", dest="tgtPort", type="string", help="port")
    (options, args) = parser.parse_args()
    tgtHost = options.tgtHost
    tgtPorts = str(options.tgtPort).split(",")

    if (tgtHost == None) | (tgtPorts[0] == None):
        print(parser.usage)
        exit(0)

    for tgtPort in tgtPorts:  # Scan the hosts with the ports etc
        nmapScan(tgtHost, tgtPort)


if __name__ == "__main__":
    main()