blob: 7bb79708268e0110ae8da7b213dd3d1dbe97ac3e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#!/usr/bin/python
def find_potential_super(path="/etc/tinc/retiolum/hosts"):
import os
import re
needle_addr = re.compile("Address\s*=\s*(.*)")
needle_port = re.compile("Port\s*=\s*(.*)")
for f in os.listdir(path):
with open(path+"/"+f) as of:
addrs = []
port = "655"
for line in of.readlines():
addr_found = needle_addr.match(line)
if addr_found:
addrs.append(addr_found.group(1))
port_found = needle_port.match(line)
if port_found:
port = port_found.group(1)
if addrs : yield (f ,[(addr ,int(port)) for addr in addrs])
def try_connect(addr):
try:
from socket import socket,AF_INET,SOCK_STREAM
s = socket(AF_INET,SOCK_STREAM)
s.settimeout(2)
s.connect(addr)
s.settimeout(None)
s.close()
return addr
except Exception as e:
pass
#return ()
def check_one_super(ha):
host,addrs = ha
valid_addrs = []
for addr in addrs:
ret = try_connect(addr)
if ret: valid_addrs.append(ret)
if valid_addrs: return (host,valid_addrs)
def check_all_the_super(path="/etc/tinc/retiolum/hosts"):
from multiprocessing import Pool
p = Pool(20)
return filter(None,p.map(check_one_super,find_potential_super(path)))
if __name__ == "__main__":
"""
usage
"""
for host,addrs in check_all_the_super():
print(host,addrs)
|
