From 89f948ae4fe32cdceddfabc8e236785bc7c00ff4 Mon Sep 17 00:00:00 2001
From: ubuntu <ubuntu@ubuntu.com>
Date: Fri, 23 Nov 2018 19:48:14 +0800
Subject: [PATCH] add tupd
---
clients/client-linux.py | 16 ++++++++++++++++
clients/client-psutil.py | 19 +++++++++++++++++++
2 files changed, 35 insertions(+)
diff --git a/clients/client-linux.py b/clients/client-linux.py
index c28435a..6654c55 100755
--- a/clients/client-linux.py
+++ b/clients/client-linux.py
@@ -127,6 +127,21 @@ def liuliang():
NET_OUT += int(netinfo[0][9])
return NET_IN, NET_OUT
+def tupd():
+ '''
+ tcp, udp, process, thread count: for view ddcc attack , then send warning
+ :return:
+ '''
+ s = subprocess.check_output("ss -t|wc -l", shell=True)
+ t = int(s[:-1])-1
+ s = subprocess.check_output("ss -u|wc -l", shell=True)
+ u = int(s[:-1])-1
+ s = subprocess.check_output("ps -ef|wc -l", shell=True)
+ p = int(s[:-1])-2
+ s = subprocess.check_output("ps -xH|wc -l", shell=True)
+ d = int(s[:-1])-2
+ return t,u,p,d
+
def ip_status():
object_check = ['www.10010.com', 'www.189.cn', 'www.10086.cn']
ip_check = 0
@@ -302,6 +317,7 @@ if __name__ == '__main__':
array['ping_10010'] = lostRate.get('10010') * 100
array['ping_189'] = lostRate.get('189') * 100
array['ping_10086'] = lostRate.get('10086') * 100
+ array['tcp'], array['udp'], array['process'], array['thread'] = tupd()
s.send("update " + json.dumps(array) + "\n")
except KeyboardInterrupt:
diff --git a/clients/client-psutil.py b/clients/client-psutil.py
index 77dee21..ea3891a 100755
--- a/clients/client-psutil.py
+++ b/clients/client-psutil.py
@@ -95,6 +95,24 @@ def liuliang():
NET_OUT += v[0]
return NET_IN, NET_OUT
+def tupd():
+ '''
+ tcp, udp, process, thread count: for view ddcc attack , then send warning
+ :return:
+ '''
+ if 'linux' in sys.platform:
+ t = int(os.popen('ss -t|wc -l').read()[:-1])-1
+ u = int(os.popen('ss -u|wc -l').read()[:-1])-1
+ p = int(os.popen('ps -ef|wc -l').read()[:-1])-2
+ d = int(os.popen('ps -xH|wc -l').read()[:-1])-2
+ else:
+ t = int(os.popen('netstat -an|find "TCP" /c').read()[:-1])-1
+ u = int(os.popen('netstat -an|find "UDP" /c').read()[:-1])-1
+ p = len(psutil.pids())
+ # cpu?
+ d = sum([psutil.Process(k).num_threads() for k in [x for x in psutil.pids()]])
+ return t,u,p,d
+
def ip_status():
object_check = ['www.10010.com', 'www.189.cn', 'www.10086.cn']
ip_check = 0
@@ -271,6 +289,7 @@ if __name__ == '__main__':
array['ping_10010'] = lostRate.get('10010') * 100
array['ping_189'] = lostRate.get('189') * 100
array['ping_10086'] = lostRate.get('10086') * 100
+ array['tcp'], array['udp'], array['process'], array['thread'] = tupd()
s.send("update " + json.dumps(array) + "\n")
except KeyboardInterrupt: