changed directory tree of all modules and packages, for easing on future implementations and extensions

kube-hunter.py

@@ -19,10 +19,10 @@ logging.basicConfig(level=loglevel, format='%(asctime)s - [%(levelname)s]: %(mes
 
 import log
 # executes all registrations from sub packages
-import modules
-from modules.discovery import HostDiscovery
-from modules.events import handler
-from modules.discovery.hosts import HostScanEvent
+import src.modules
+from src.modules.discovery import HostDiscovery
+from src.core.events import handler
+from src.modules.discovery.hosts import HostScanEvent
 
 def main():
     logging.info("Started")

log/reporter.py

@@ -1,8 +1,8 @@
 import logging
 from prettytable import PrettyTable
-from modules.events import handler
-from modules.events.types import Vulnerability, Information, Service
-from modules.discovery.kubelet import KubeletExposedHandler
+from src.core.events import handler
+from src.core.events.types import Vulnerability, Information, Service
+from src.modules.discovery.kubelet import KubeletExposedHandler
 
 services = list()
 vulnerabilities = list()

modules/__init__.py

@@ -1,4 +0,0 @@
-import discovery
-import hunting
-import events
-import types

src/__init__.py

@@ -0,0 +1,2 @@
+import core
+import modules

src/core/__init__.py

@@ -0,0 +1,2 @@
+from events import *
+from types import *

src/core/types/defaults.py

@@ -1,4 +1,4 @@
-from ..events import handler
+from ...core.events import handler
 
 class Hunter(object):
     def __init__(self):

src/modules/__init__.py

@@ -0,0 +1,2 @@
+import discovery
+import hunting

src/modules/discovery/dashboard.py

@@ -2,9 +2,9 @@ import json
 
 import requests
 
-from ..events import handler
-from ..events.types import Event, Service, OpenPortEvent
-from ..types import Hunter
+from ...core.events import handler
+from ...core.events.types import Event, Service, OpenPortEvent
+from ...core.types import Hunter
 
 class KubeDashboardEvent(Service, Event):
     """Allows multiple arbitrary operations on the cluster from all connections"""

src/modules/discovery/hosts.py

@@ -8,9 +8,9 @@ import requests
 from netaddr import IPNetwork
 from netifaces import AF_INET, ifaddresses, interfaces
 
-from ..events import handler
-from ..events.types import Event, NewHostEvent
-from ..types import Hunter
+from ...core.events import handler
+from ...core.events.types import Event, NewHostEvent
+from ...core.types import Hunter
 
 class HostScanEvent(Event):
     def __init__(self, pod=False, active=False, predefined_hosts=list()):

src/modules/discovery/kubelet.py

@@ -1,13 +1,13 @@
 import json
 import logging
 from enum import Enum
-from ..types import Hunter
+from ...core.types import Hunter
 
 import requests
 import urllib3
 
-from ..events import handler
-from ..events.types import OpenPortEvent, Kubelet, Vulnerability, Event, Service
+from ...core.events import handler
+from ...core.events.types import OpenPortEvent, Kubelet, Vulnerability, Event, Service
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 """ Services """

src/modules/discovery/ports.py

@@ -1,8 +1,8 @@
 from socket import socket
-from ..types import Hunter
+from ...core.types import Hunter
 
-from ..events import handler
-from ..events.types import NewHostEvent, OpenPortEvent
+from ...core.events import handler
+from ...core.events.types import NewHostEvent, OpenPortEvent
 
 
 default_ports = [8001, 10250, 10255, 30000]

src/modules/discovery/proxy.py

@@ -1,11 +1,11 @@
 import logging
 from collections import defaultdict
-from ..types import Hunter
+from ...core.types import Hunter
 
 from requests import get
 
-from ..events import handler
-from ..events.types import Service, Event, OpenPortEvent
+from ...core.events import handler
+from ...core.events.types import Service, Event, OpenPortEvent
 
 class KubeProxyEvent(Event, Service):
     """proxies from a localhost address to the Kubernetes apiserver"""

src/modules/hunting/dashboard.py

@@ -1,9 +1,9 @@
 import logging
-from ..types import Hunter
+from ...core.types import Hunter
 
 import requests
 
-from ..events import handler
+from ...core.events import handler
 from ..discovery.dashboard import KubeDashboardEvent
 
 @handler.subscribe(KubeDashboardEvent)

src/modules/hunting/kubelet.py

@@ -5,10 +5,10 @@ from enum import Enum
 import requests
 import urllib3
 
-from ..events import handler
-from ..events.types import (KubernetesCluster, Kubelet, Vulnerability, Information, Event)
+from ...core.events import handler
+from ...core.events.types import (KubernetesCluster, Kubelet, Vulnerability, Information, Event)
 from ..discovery.kubelet import KubeletExposedHandler, ReadOnlyKubeletEvent, SecureKubeletEvent
-from ..types import Hunter
+from ...core.types import Hunter
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 

src/modules/hunting/proxy.py

@@ -5,9 +5,9 @@ from requests import get
 
 from ..discovery.dashboard import KubeDashboardEvent
 from ..discovery.proxy import KubeProxyEvent
-from ..events import handler
-from ..events.types import Vulnerability, Event, KubernetesCluster
-from ..types import Hunter
+from ...core.events import handler
+from ...core.events.types import Vulnerability, Event, KubernetesCluster
+from ...core.types import Hunter
 
 
 class Service(Enum):