Fixes user being able to attempt to run attack without host and issue #6 \(Can't specify port\)

This commit is contained in:
GettingGodlyInThisGame 2016-11-12 12:45:26 +13:00
parent a7851b5570
commit d23a1f9427
1 changed files with 7 additions and 2 deletions

View File

@ -10,7 +10,12 @@ parser.set_defaults(verbose=False)
parser.set_defaults(randuseragent=False) parser.set_defaults(randuseragent=False)
args = parser.parse_args() args = parser.parse_args()
if len(sys.argv)==1: if len(sys.argv)<=1:
parser.print_help()
sys.exit(1)
if not args.host:
print("Host required!")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
@ -54,7 +59,7 @@ user_agents = [
def init_socket(ip): def init_socket(ip):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(4) s.settimeout(4)
s.connect((ip,80)) s.connect((ip,args.port))
s.send("GET /?{} HTTP/1.1\r\n".format(random.randint(0, 2000)).encode("utf-8")) s.send("GET /?{} HTTP/1.1\r\n".format(random.randint(0, 2000)).encode("utf-8"))
if args.randuseragent: if args.randuseragent: