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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
| # coding=utf-8
import socket
import requests
import os
import random
import commands, time
# 下载网页
def download(url):
res = requests.get(url)
return res.text
# 域名转IP
def host_to_ip(host):
'''
Returns the IP address of a given hostname
CLI Example:
.. code-block:: bash
salt '*' network.host_to_ip example.com
'''
try:
family, socktype, proto, canonname, sockaddr = socket.getaddrinfo(
host, 0, socket.AF_UNSPEC, socket.SOCK_STREAM)[0]
if family == socket.AF_INET:
ip, port = sockaddr
elif family == socket.AF_INET6:
ip, port, flow_info, scope_id = sockaddr
except Exception:
ip = None
return ip
def service_status(service_name):
cmd = "ps -ef |grep \"%s\"|grep -v grep |awk '{print $2}'" % service_name
result = os.popen(cmd).read().strip()
try:
service_pid = result.split()[0]
if service_pid:
print "\033[32;1m%s monitor service is running...\033[0m" % service_name
return "Running"
except IndexError:
print "\033[31;1m%s service is not running....\033[0m" % service_name
return "Dead"
def stop_service(ss_config='all', local_port=0, customize=""):
if customize:
script = customize
elif ss_config == 'all':
script = "ss-local"
else:
server, port, pwd, method = ss_config
script = "ss-local -s {0} -p {1} -k {2} -m {3} -l {4}".format(server, port, pwd, method, local_port)
service_name = script
cmd = "ps -ef| grep \"%s\"|grep -v grep |awk '{print $2}'|xargs kill -9" % service_name
if service_status(service_name) == 'Running':
cmd_result = os.system(cmd)
if cmd_result == 0:
print '..............\n'
time.sleep(1)
print '\033[31;1m%s stopped! \033[0m' % service_name
else:
print '\033[31;1mCannot stop %s service successfully,please manually kill the pid!\033[0m' % service_name
else:
print 'Service is not running...,nothing to kill! '
def start_service(ss_config, local_port):
server, port, pwd, method = ss_config
script = "ss-local -s {0} -p {1} -k {2} -m {3} -l {4}".format(server, port, pwd, method, local_port)
print "Checking service status....."
if service_status(script) == 'Running':
print "\033[33;1mNetflow service is already running!\033[0m\n"
else:
print "Starting service...."
cmd = 'nohup %s >> phantom.log 2>&1 &' % script
print cmd
if os.system(cmd) == 0:
time.sleep(5)
print '\033[32;1mNetflow service started successfully!\n\033[0m'
else:
print '\033[32;1mNetflow service started failed!\n\033[0m'
def restart_service(service_name, ss_config, local_port):
server, port, pwd, method = ss_config
script = "ss-local -s {0} -p {1} -k {2} -m {3} -l {4}".format(server, port, pwd, method, local_port)
print "Checking service status....."
if service_status(service_name) == 'Running':
print "\033[33;1mNetflow service is already running!\033[0m\n"
else:
print "Starting service...."
cmd = 'nohup %s >> phantom.log 2>&1 &' % script
print cmd
if os.system(cmd) == 0:
time.sleep(5)
print '\033[32;1mNetflow service started successfully!\n\033[0m'
else:
print '\033[32;1mNetflow service started failed!\n\033[0m'
def check_ss_live(ss_config):
server, port, pwd, method = ss_config
local_port = 8800 + random.randrange(10, 99)
start_service(ss_config, local_port)
cmd = "curl --socks5 127.0.0.1:{} http://ip111cn.appspot.com".format(local_port)
start = time.time()
result = commands.getstatusoutput(cmd)
stop = time.time()
print(stop - start)
stop_service(ss_config, local_port)
if server in result[1]:
print ss_config, "OK!"
return stop - start, ss_config
else:
return False, ss_config
|
