From 2168180ffb288115ec265684860bc957b55e75a0 Mon Sep 17 00:00:00 2001 From: Daniel Sagi Date: Tue, 11 Jun 2019 11:29:39 +0300 Subject: [PATCH] fixed issue with get_random_pod method, .next attribute on generator is deprecated in python3 --- src/modules/hunting/kubelet.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/modules/hunting/kubelet.py b/src/modules/hunting/kubelet.py index a43add1..dafc575 100644 --- a/src/modules/hunting/kubelet.py +++ b/src/modules/hunting/kubelet.py @@ -299,12 +299,14 @@ class SecureKubeletPortHunter(Hunter): if not pod_data: pod_data = next((pod_data for pod_data in pods_data if is_kubesystem_pod(pod_data)), None) - container_data = (container_data for container_data in pod_data["spec"]["containers"]).next() - return { - "name": pod_data["metadata"]["name"], - "container": container_data["name"], - "namespace": pod_data["metadata"]["namespace"] - } + if pod_data: + container_data = next((container_data for container_data in pod_data["spec"]["containers"]), None) + if container_data: + return { + "name": pod_data["metadata"]["name"], + "container": container_data["name"], + "namespace": pod_data["metadata"]["namespace"] + } @handler.subscribe(ExposedRunHandler) class ProveRunHandler(ActiveHunter):