Add argsparse for flooder function

Signed-off-by: Fernando Fernández <ffernandezmancera@gmail.com>

Author: Fernando Fernández

pentesting-multitool.py

@@ -9,48 +9,62 @@ def main():
                                                  ' It will be developed using Python3 adding some external libraries as DNSPython, pythonwhois or scapy.')
 
     # Argument for DNS function
-
-    parser.add_argument('-DNS', metavar='--DNS', type=str, help='Using it you can get information about a DNS record of a specific domain.\n'
+    parser.add_argument('-d', metavar='--DNS', type=str, help='Using it you can get information about a DNS record of a specific domain.\n'
                         'Usage: <domain> -r <record> options: <-f --file>')
 
     # Argument for whois function
-
-    parser.add_argument('-whois', metavar='--whois', type=str, help='Using it you can get information about a domain. '
+    parser.add_argument('-w', metavar='--whois', type=str, help='Using it you can get information about a domain. '
                                                                    'Usage: <domain> options: <-f --file>.')
 
     # Argument for shodan search function
-
-    parser.add_argument('-shodan', metavar='--shodan', type=str, help='Using it you can get information about the devices and services connected in a network.'
-                                                                        'Usage: <search> (separating search arguments by \"-\") options: <-f --file, -full >')
+    parser.add_argument('-s', metavar='--shodan', type=str, help='Using it you can get information about the devices and services connected in a network.'
+                                                                        'Usage: <search> (separating search arguments by \"-\") options: <-f --file, -u >')
 
     # Argument for banner grabber function
-
-    parser.add_argument('-bgrabber', metavar='--bgrabber', type=str, help='Using it you can get information about some services in the specified ports and IP.'
+    parser.add_argument('-b', metavar='--bgrabber', type=str, help='Using it you can get information about some services in the specified ports and IP.'
                                                                           'Usage: <ip> -p <ports>(separated by \":\") options: <-file --file>.')
 
+    # Argument for flooder function
+    parser.add_argument('-o', metavar='--flooder', type=int, help='Using it you can do a DOS Attack (Ping of death) or generate the PCAP file for do it.'
+                                                                        'Usage: <number of packets (generator mode) or sends(flooder mode)> -f <filename> options: <-g --generator>.')
+
+    #Argument for fuzzer function
+    parser.add_argument('-z', metavar='--fuzzer', type=str, help='Using it you can do a fuzzing test and write a PCAP with results. Please read documentation for know about the PCAP format.'
+                                                                      'Usage:<ip> -ng <number of generations> -n <number of packets> -l <layer (UDP, TCP, ICMP)> -f <PCAP file name>')
+
     # Optionals
+    parser.add_argument('-l', metavar='--layer', required='-z' in sys.argv or '--fuzzer' in sys.argv, help='Layer that you want to use.')
 
-    parser.add_argument('-p', metavar='--ports', required='-p' in sys.argv or '--ports' in sys.argv, type=str, help='Specify ports.')
+    parser.add_argument('-n', metavar='--number', required='-z' in sys.argv or '--fuzzer' in sys.argv, type=int, help='Number of packets.')
 
-    parser.add_argument('-full', action='store_true', help='Specify if you want the complete information or just the IP on shodan search.')
+    parser.add_argument('-ng', metavar='--ngenerations', required='-z' in sys.argv or '--fuzzer' in sys.argv, type=int, help='Number of generations.')
 
-    parser.add_argument('-r', metavar='--record', required='-DNS' in sys.argv or '--DNS' in sys.argv, type=str, help='Record of a specific domain.')
+    parser.add_argument('-g', action='store_true', help='Specify if you want to generate the file.')
 
-    parser.add_argument('-f', metavar='--file',type=str, help='Filename where you want to save the result.')
+    parser.add_argument('-p', metavar='--ports', required='-b' in sys.argv or '--bgrabber' in sys.argv, type=str, help='Specify ports.')
 
-    #Defining args
+    parser.add_argument('-u', action='store_true', help='Specify if you want the complete information or just the IP on shodan search.')
+
+    parser.add_argument('-r', metavar='--record', required='-d' in sys.argv or '--DNS' in sys.argv, type=str, help='Record of a specific domain.')
 
+    parser.add_argument('-f', metavar='--file', required='-o' in sys.argv or '--flooder' in sys.argv, type=str, help='Filename where you want to save the result.')
+
+    #Defining args
     args = parser.parse_args()
 
     #Calls
-    if args.DNS is not None:
-        dns_utility.dns_scan(args.DNS, args.r, args.f)
-    elif args.whois is not None:
-        whois_utility.whois_scan(args.whois, args.f)
-    elif args.shodan is not None:
-        shodan_utility.shodan_search(args.shodan, args.f, args.full)
-    elif args.bgrabber is not None:
-        bannergrab_utility.banner_grabbing(args.bgrabber, args.p, args.f)
+    if args.d is not None:
+        dns_utility.dns_scan(args.d, args.r, args.f)
+    elif args.w is not None:
+        whois_utility.whois_scan(args.w, args.f)
+    elif args.s is not None:
+        shodan_utility.shodan_search(args.s, args.f, args.u)
+    elif args.b is not None:
+        bannergrab_utility.banner_grabbing(args.b, args.p, args.f)
+    elif args.o is not None:
+        flooder_utility.flooder_attack(args.o, args.g, args.f)
+    elif args.z is not None:
+        fuzzing_utility.fuzzing_attack(args.z, args.ng, args.n, args.l, args.f)
 
 if __name__ == '__main__':
     main()

utility_scripts/flooder_utility.py

@@ -1,28 +1,11 @@
 #! /usr/bin/env python3
 
 from scapy.all import IP, ICMP, wrpcap, rdpcap, sendpfast
-import sys
 
 class flooder_attack(object):
 
-    def check(self):
-        if len(sys.argv) < 5:
-            print('Usage: python3 pentesting-multitool flooder <mode> <number of packets (generator mode) or sends(flooder mode) <filename>\n'
-                  '<mode> : You can use it for generate a pcap file with an specified with \"generator\" or use it for do a flooding attack with \"flood\" \n')
-            exit()
+    def generator(self, n, filename):
 
-        if sys.argv[2] == "generator":
-            self.generator()
-        elif sys.argv[2] == "flood":
-            self.flooder()
-        else:
-            print("Invalid argument.  (pentesting-multitool flooder <?>)")
-            exit()
-
-    def generator(self):
-
-        n = int(sys.argv[3])
-        filename = sys.argv[4]
         time = 0.00114108 * n + 0.157758
         minutes = time/60
 
@@ -33,10 +16,7 @@ def generator(self):
 
         print('%s packets generated.' % (n))
 
-    def flooder(self):
-
-        n = int(sys.argv[3])
-        filename = sys.argv[4]
+    def flooder(self, n, filename):
 
         print('Reading pcap file.')
         pkgs = rdpcap(filename)
@@ -46,6 +26,10 @@ def flooder(self):
             sendpfast(pkgs)
             print('Done, part %s of %s' % ((i + 1), n))
 
-    def __init__(self):
+    def __init__(self, number, generator, filename):
         print('Initializing')
-        self.check()
+        if generator is True:
+            self.generator(number, filename)
+        else:
+            self.flooder(number, filename)
+        exit()