Added the beginnings of the changes needed for the setup.py to work.

Changed around the directory structure to have the files a part of the VHostScan
package so it could be used for pkg_resources. Also changed the DEFAULT_WORDLIST_FILE to
use pkg_resources instead of __file__. Also changed some of the imports so they are using relative
imports instead of the full path.

This appears to work with atleast the -t flag, however, more testing should be done and the tests that
already exist should be able to run when running `python setup.py test`

Author: ccsplit

MANIFEST.in

@@ -0,0 +1 @@
+recursive-include VHostScan *.txt

VHostScan.py renamed to VHostScan/VHostScan.py

@@ -1,22 +1,19 @@
 #!/usr/bin/python
 
-import os
 import sys
 import dns.resolver
 from argparse import ArgumentParser
 from socket import gethostbyaddr
-from lib.core.virtual_host_scanner import *
-from lib.helpers.output_helper import *
-from lib.helpers.file_helper import load_random_user_agents
-from lib.helpers.wordlist_helper import WordList
-from lib.core.__version__ import __version__
-from lib.input import cli_argument_parser
-
-DEFAULT_WORDLIST_FILE = os.path.join(
-    os.path.dirname(os.path.abspath(__file__)),
-    'wordlists',
-    'virtual-host-scanning.txt'
-)
+from pkg_resources import resource_filename
+from .lib.core.virtual_host_scanner import *
+from .lib.helpers.output_helper import *
+from .lib.helpers.file_helper import load_random_user_agents
+from .lib.helpers.wordlist_helper import WordList
+from .lib.core.__version__ import __version__
+from .lib.input import cli_argument_parser
+
+DEFAULT_WORDLIST_FILE = resource_filename(
+    'VHostScan', 'wordlists/virtual-host-scanning.txt')
 
 
 def print_banner():

tests/__init__.py renamed to VHostScan/__init__.py

@@ -5,7 +5,7 @@
 import hashlib
 import pandas as pd
 import time
-from lib.core.discovered_host import *
+from .discovered_host import *
 
 import urllib3
 urllib3.disable_warnings()

lib/__init__.py renamed to VHostScan/lib/__init__.py

@@ -1,5 +1,5 @@
-from lib.core.discovered_host import *
-from lib.helpers.file_helper import *
+from ..core.discovered_host import *
+from .file_helper import *
 import time
 from fuzzywuzzy import fuzz
 import itertools

lib/core/__init__.py renamed to VHostScan/lib/core/__init__.py

@@ -1,13 +1,10 @@
 import sys
-import os
-from lib.helpers.file_helper import get_combined_word_lists
-
-DEFAULT_WORDLIST_FILE = os.path.join(
-    os.path.dirname(os.path.abspath(__file__)),
-    '../..',
-    'wordlists',
-    'virtual-host-scanning.txt'
-)
+from .file_helper import get_combined_word_lists
+from pkg_resources import resource_filename
+
+
+DEFAULT_WORDLIST_FILE = resource_filename(
+    'VHostScan', 'wordlists/virtual-host-scanning.txt')
 
 
 class WordList:

lib/core/__version__.py renamed to VHostScan/lib/core/__version__.py

@@ -0,0 +1,31 @@
+from setuptools import find_packages, setup
+from VHostScan.lib.core.__version__ import __version__
+
+
+def dependencies(file):
+    with open(file) as f:
+        return f.read().splitlines()
+
+
+with open("README.md") as f:
+    setup(
+        name="VHostScan",
+        license="GPLv3",
+        description="A virtual host scanner that performs reverse lookups, "
+                    "can be used with pivot tools, detect catch-all"
+                    "scenarios, aliases and dynamic default pages.",
+        long_description=f.read(),
+        author="codingo",
+        version=__version__,
+        author_email="[email protected]",
+        url="http://github.com/codingo/VHostScan",
+        packages=find_packages(exclude=('tests')),
+        package_data={'VHostScan': ['*.txt']},
+        entry_points={
+            'console_scripts': [
+                'vhostscan = VHostScan.VHostScan:main'
+            ]
+        },
+        install_requires=dependencies('requirements.txt'),
+        tests_require=dependencies('test-requirements.txt'),
+        include_package_data=True)

lib/core/discovered_host.py renamed to VHostScan/lib/core/discovered_host.py

@@ -1,5 +1,3 @@
--r requirements.txt
-
 pytest==3.2.3
 pytest-mock==1.6.3
 pep8==1.7.0