Removing redundant code

Author: Michael Skelton

VHostScan.py

@@ -9,10 +9,11 @@
 def print_banner():
     print("+-+-+-+-+-+-+-+-+-+  v. 0.1")
     print("|V|H|o|s|t|S|c|a|n|  Developed by @codingo_ & @__timk")
-    print("+-+-+-+-+-+-+-+-+-+  https://github.com/codingo/VHostScan")
+    print("+-+-+-+-+-+-+-+-+-+  https://github.com/codingo/VHostScan\n")
 
 
 def main():
+    print_banner()
     parser = ArgumentParser()
     parser.add_argument("-t",   dest="target_hosts", required=True, help="Set a target range of addresses to target. Ex 10.11.1.1-255" )
     parser.add_argument("-w",   dest="wordlist", required=False, help="Set the wordlist to use for generated commands. Ex /usr/share/wordlist.txt", default="./wordlists/virtual-host-scanning.txt")
@@ -23,20 +24,7 @@ def main():
     parser.add_argument('--unique-depth', dest='unique_depth', type=int, help='Show likely matches of page content that is found x times (default 1).', default=1)
     arguments = parser.parse_args()
 
-    if len(sys.argv) == 1:
-        print_banner()
-        parser.error("No arguments given.")
-        parser.print_usage
-        sys.exit()
-
-    if arguments.output_directory.endswith('/' or '\\'):
-        arguments.output_directory = arguments.output_directory[:-1]
-    if arguments.target_hosts.endswith('/' or '\\'):
-        arguments.target_hosts = arguments.target_hosts[:-1]
-
-    print_banner()
-
-    scanner = virtual_host_scanner(arguments.target_hosts, arguments.output_directory, arguments.port, arguments.unique_depth, arguments.ignore_http_codes, arguments.ignore_content_length, arguments.wordlist)
+    scanner = virtual_host_scanner(arguments.target_hosts, arguments.port, arguments.unique_depth, arguments.ignore_http_codes, arguments.ignore_content_length, arguments.wordlist)
     scanner.scan()
     print(scanner.likely_matches())
 

lib/core/virtual_host_scanner.py

@@ -17,10 +17,9 @@ class virtual_host_scanner(object):
         output: folder to write output file to
     """
 
-    def __init__(self, target, output, port=80, unique_depth=1, ignore_http_codes='404', ignore_content_length=0, 
+    def __init__(self, target, port=80, unique_depth=1, ignore_http_codes='404', ignore_content_length=0, 
                  wordlist="./wordlists/virtual-host-scanning.txt"):
         self.target = target
-        self.output = output + '/' + target + '_virtualhosts.txt'
         self.port = port
         self.ignore_http_codes = list(map(int, ignore_http_codes.replace(' ', '').split(',')))
         self.ignore_content_length = ignore_content_length
@@ -98,6 +97,4 @@ def likely_matches(self):
         d=df.groupby("val_col").filter(lambda x: len(x) <= self.unique_depth)
         matches=((d["key_col"].values).tolist())
 
-        return matches
-
-    def page_difference_ratios
+        return matches