加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

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

发布时间:2020-12-16 23:55:11 所属栏目:Python 来源:网络整理
导读:server =sys.argv[1span style="color: #000000;"] s =span style="color: #000000;"socket.socket() span style="color: #0000ff;"while 1span style="color: #000000;": span style="color: #0000ff;"try span style="color: #000000;": s.connect((server

server=sys.argv[1<span style="color: #000000;">]
s
=<span style="color: #000000;">socket.socket()
<span style="color: #0000ff;">while
1<span style="color: #000000;">:
<span style="color: #0000ff;">try
<span style="color: #000000;">:
s.connect((server,
1234)) <span style="color: #008000;">#
<span style="color: #008000;">建立tcp连接

<span style="color: #0000ff;">break

<span style="color: #0000ff;">except<span style="color: #000000;">:
time.sleep(5<span style="color: #000000;">)

data=s.recv(1024).decode() <span style="color: #008000;">#<span style="color: #008000;">接收服务器发送的黑名单
blacklist=data.split(<span style="color: #800000;">"<span style="color: #800000;">,<span style="color: #800000;">") <span style="color: #008000;">#<span style="color: #008000;">str-->list
<span style="color: #0000ff;">print<span style="color: #000000;">(blacklist)
s.close() <span style="color: #008000;">#<span style="color: #008000;">接受到黑名单,关闭资源

<span style="color: #0000ff;">while 1<span style="color: #000000;">:
pids=psutil.pids() <span style="color: #008000;">#<span style="color: #008000;">pids所有进程的进程id (list)
<span style="color: #0000ff;">for i <span style="color: #0000ff;">in<span style="color: #000000;"> pids:
<span style="color: #0000ff;">try<span style="color: #000000;">:
<span style="color: #0000ff;">if psutil.Process(i).name() <span style="color: #0000ff;">in blacklist: <span style="color: #008000;">#<span style="color: #008000;">psutil.Process().name() 进程ID对应的程序名
psutil.Process(i).terminate() <span style="color: #008000;">#<span style="color: #008000;">terminate 关闭程序
<span style="color: #0000ff;">except<span style="color: #000000;">:
<span style="color: #0000ff;">pass<span style="color: #000000;">
time.sleep(1)

服务端:

threading =,12345=[,=<span style="color: #0000ff;">def tcplink(sock,addr): <span style="color: #008000;">#<span style="color: #008000;">把黑名单发送给客户端
<span style="color: #000000;"> sock.send(msg.encode())
<span style="color: #0000ff;">print
(addr,<span style="color: #800000;">'
<span style="color: #800000;">OK!
<span style="color: #800000;">'
<span style="color: #000000;">)
sock.close()
<span style="color: #0000ff;">while
1<span style="color: #000000;">:
con,addr
=<span style="color: #000000;">s.accept()
ts
=<span style="color: #000000;">[]
t
=Thread(target=tcplink,args=(con,addr)) <span style="color: #008000;">#
<span style="color: #008000;">创建线程来处理tcp连接

<span style="color: #000000;"> ts.append(t)
t.start()
<span style="color: #0000ff;">for
i <span style="color: #0000ff;">in
<span style="color: #000000;"> ts:
i.join()

测试:

运行QQ和计算器,发现刚运行就被关闭了。

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读