设计该程序的思路:
1、服务器端处于监听状态,等待客户端的连接。同时服务器定义了黑名单('Calculator.exe','QQ.exe','cmd.exe'),不同主机之间进程的名称可能会有所不同;
2、客户端主动连接服务器,创建连接成功后,服务器便将黑名单发送给客户端,并在服务器端打印客户端的IP和端口号,提示发送成功;
3、客户端接收到服务器端发送的黑名单后,将其打印出来,并使系统每隔1秒检验是否运行了黑名单中的进程。如果运行了,则结束该进程。
服务器端
import psutil #psutil能够轻松实现获取系统运行的进程和系统利用率(包括CPU、内存、磁盘、网络等)信息
import socket
from threading import Thread
s=socket.socket() #创建一个socket
s.bind(('0.0.0.0',3333)) #绑定端口;0.0.0.0表示任意ip
s.listen(5) #'5'表示等待连接的最大数量
blacklist=['Calculator.exe','QQ.exe','cmd.exe']
msg=','.join(blacklist) #将列表blacklist转换成字符串
#把黑名单发送给连接它的客户端
def tcplink(sock,addr):
sock.send(msg.encode())
print(addr,'OK!')
sock.close() #发送黑名单完毕后,关闭TCP连接
ts=[]
while 1:
con,addr=s.accept() #接受一个客户端的连接
#创建多线程
t=Thread(target=tcplink,args=(con,addr))
ts.append(t)
t.start()
for t in ts:
t.join()
客户端
import socket
import psutil
import time
s=socket.socket() #创建一个socket
#与服务器端建立连接,每隔5秒尝试一次,直至连接成功
while 1:
try:
s.connect(('127.0.0.1',3333))
break
except:
time.sleep(5)
data=s.recv(1024).decode() #接收从服务器发送过来的黑名单,并保存在变量data中
blacklist=data.split(',') #将字符串data转换为列表blacklist
print(blacklist) #打印blacklist
s.close() #关闭资源
#每隔1秒检验系统是否运行了blacklist中的进程。如果运行了,则结束该进程;没有运行则pass
while 1:
pids=psutil.pids() #以列表的形式获取所有系统运行的进程ID
for i in pids:
try:
if psutil.Process(i).name() in blacklist:
psutil.Process(i).terminate() #结束进程
except:
pass
time.sleep(1)
运行结果如下:
服务器端:
('127.0.0.1', 16533) OK!
客户端:
['Calculator.exe','QQ.exe','cmd.exe']
编写好程序后,分别运行服务器端和客户端的程序,只要客户端打开了服务器端给它发送的黑名单中的程序('Calculator.exe','QQ.exe','cmd.exe'),系统很快便会结束该进程。