python 端口扫描主机并将结果保存到CouchDB文档。

Posted 2021-05-09

from datetime import datetime
from collections import OrderedDict
import couchdbkit as cdb
import argparse
import nmap
import sys

db = None

p = argparse.ArgumentParser()

args = OrderedDict([
	("host", {}),
	("--db_uri", {"default": "http://127.0.0.1:5984"})
])
for key, params in args.items():
	p.add_argument(key, **params)

class HostScan(cdb.Document):
	datetime = cdb.DateTimeProperty()
	scan = cdb.DictProperty()

if __name__ == "__main__":
	args = p.parse_args()

	try:
		print "Connecting to DB.."
		s = cdb.Server(uri=args.db_uri)

		print "Getting or creating 'snoopback' database.."
		db = s.get_or_create_db("snoopback")

		nm = nmap.PortScanner()
		print "Scanning %s.." % args.host
		nm.scan(args.host, arguments="-sV")

		print "Saving results to db.."
		HostScan.set_db(db)
		scan = HostScan(
			datetime=datetime.now(),
			scan=nm[args.host]
		)
		scan.save()

	except Exception, e:
		print "Exception: %s" % e
		sys.exit(1)