首页 > > 网络编程 > 其它 >

python简单的监控脚本-利用socket、psutil阻止远…

2019-03-10 11:52:32来源：博客园阅读 ()

python简单的监控脚本-利用socket、psutil阻止远程主机运行特定程序

psutil是一个跨平台的库（http://code.google.com/p/psutil/），能够轻松的实现获取系统运行的进程和系统利用率（CPU、内存、磁盘、网络等）信息。它主要应用于系统监控，分析和限制系统资源及进程的管理。

psutil是第三方库需要安装。pip install psutil

利用socket，psutil实现阻止远程主机运行特定的程序,本例的黑名但是直接写死的,也可以加载一个黑名单文件,方便以后扩展

客户端:

#控制进程,客户端
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()