From 1bdf29dc5b24b982dbe0d7fdeaa9f8cb633a1cc3 Mon Sep 17 00:00:00 2001 From: whokilleddb Date: Mon, 21 Mar 2022 22:19:03 +0530 Subject: [PATCH] Sanitize IP zookeeper/enable_and_wait.py (#24007) * Sanitize IP zookeeper/enable_and_wait.py * Fixed Identation * Fixed Gaierror --- tools/zookeeper/enable_and_wait.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tools/zookeeper/enable_and_wait.py b/tools/zookeeper/enable_and_wait.py index 398ef412b..16b662f59 100755 --- a/tools/zookeeper/enable_and_wait.py +++ b/tools/zookeeper/enable_and_wait.py @@ -3,10 +3,15 @@ import os import sys import time +from socket import gethostbyname, gaierror from tools.zookeeper import Zookeeper def is_online(ip): - return (os.system(f"ping -c 1 {ip} > /dev/null") == 0) + try: + addr = gethostbyname(ip) + return (os.system(f"ping -c 1 {addr} > /dev/null") == 0) + except gaierror: + return False if __name__ == "__main__": z = Zookeeper()