python简单的监控脚本-利用sock

#控制进程,客户端
import sys
import psutil
import socket
import time

server=sys.argv[1]
s=socket.socket()
while 1:
    try:
        s.connect((server,1234)) #建立tcp连接
        break
    except:
        time.sleep(5)

data=s.recv(1024).decode() #接收服务器发送的黑名单
blacklist=data.split(",")  #str-->list
print(blacklist)  
s.close()       #接受到黑名单，关闭资源

while 1:
    pids=psutil.pids()    #pids所有进程的进程id  (list)
    for i in pids:
        try:
            if psutil.Process(i).name() in blacklist:   #psutil.Process().name() 进程ID对应的程序名
                psutil.Process(i).terminate()   #terminate 关闭程序
        except:
            pass
    time.sleep(1)        

##控制进程，服务端
import sys
import psutil
import socket
import time
from threading import Thread
s=socket.socket()
s.bind(("0.0.0.0",1234))
s.listen(5)
blacklist=['Calculator.exe','QQ.exe']
msg=','.join(blacklist)

def tcplink(sock,addr): #把黑名单发送给客户端
    sock.send(msg.encode())
    print(addr,'OK!')
    sock.close()
while 1:
    con,addr=s.accept()
    ts=[]
    t=Thread(target=tcplink,args=(con,addr))   #创建线程来处理tcp连接
    ts.append(t)
    t.start()
    for i in ts:
        i.join()