zookeeper 未授权访问扫描脚本

zookeeper 未授权访问扫描脚本，使用python3
输入：input.txt，每行一个ip
输出：result.txt，ip和端口

# coding=utf-8
import socket

def check(ip, port, timeout):
    try:
        socket.setdefaulttimeout(timeout)
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.connect((ip, int(port)))
        s.send("envi".encode("utf-8"))
        data = s.recv(1024).decode("utf-8")
        s.close()
        if 'Environment' in data:
            print('Zookeeper Unauthorized access')
            return True
        else:
            return False
    except Exception as e:
        print(e)
        return False


def main():
    result = []
    with open('./input.txt','r') as fr:
        ips = fr.readlines()
    for ip in ips:
        ip = ip.strip()
        print(ip)
        for port in [2181, 80, 443, 2180, 2182]:
            if check(ip, port, 2):
                result.append(ip + '\t' + str(port) + '\n')
                break
    with open('./result.txt', 'w') as fw:
        fw.writelines(result)


if __name__ == '__main__':
    main()