mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-02-14 09:59:55 +00:00
0f1739262f
Fix linting issues with flake8 and black. Add pre-commit congifuration, update documnetation for it. Apply linting check in Travis CI.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import logging
|
|
import subprocess
|
|
|
|
from kube_hunter.core.types import Discovery
|
|
from kube_hunter.core.events import handler
|
|
from kube_hunter.core.events.types import HuntStarted, Event
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class KubectlClientEvent(Event):
|
|
"""The API server is in charge of all operations on the cluster."""
|
|
|
|
def __init__(self, version):
|
|
self.version = version
|
|
|
|
def location(self):
|
|
return "local machine"
|
|
|
|
|
|
# Will be triggered on start of every hunt
|
|
@handler.subscribe(HuntStarted)
|
|
class KubectlClientDiscovery(Discovery):
|
|
"""Kubectl Client Discovery
|
|
Checks for the existence of a local kubectl client
|
|
"""
|
|
|
|
def __init__(self, event):
|
|
self.event = event
|
|
|
|
def get_kubectl_binary_version(self):
|
|
version = None
|
|
try:
|
|
# kubectl version --client does not make any connection to the cluster/internet whatsoever.
|
|
version_info = subprocess.check_output("kubectl version --client", stderr=subprocess.STDOUT)
|
|
if b"GitVersion" in version_info:
|
|
# extracting version from kubectl output
|
|
version_info = version_info.decode()
|
|
start = version_info.find("GitVersion")
|
|
version = version_info[start + len("GitVersion':\"") : version_info.find('",', start)]
|
|
except Exception:
|
|
logger.debug("Could not find kubectl client")
|
|
return version
|
|
|
|
def execute(self):
|
|
logger.debug("Attempting to discover a local kubectl client")
|
|
version = self.get_kubectl_binary_version()
|
|
if version:
|
|
self.publish_event(KubectlClientEvent(version=version))
|