python实现linux下的简单进程监

2023-01-31 01:01:01 python 简单 进程

最近看writeup看的有点头疼,深深感受到了自己的无知。确实还需要学习很多东西、一点一点的积累!加油!

python确实很强大哦~,要想学好Python就得自己多动手堆代码!
无聊写了一个进程监控的脚本,就当做是练习。其实最终实现的功能也很简单。
记录如下:
0x00:
首先这种写法要学会哦,虽然不知道具体有什么用吧,但是这代码写的,看着就很有水平!哈!

def main():
    while 1:
        psutil_process()

if __name__=="__main__":
    main()

name ==”main“的作用是为了对py的运行方式进行检测,如果模块是被导入,name的值为模块的名字,如果是直接执行name的值==”main
网上找了一个解释,贴在这里,希望有懂得人来讲讲具体区别。
0x01:
进程监控、刚开始我的想法是通过两次执行执行ps -e命令,比较不同。于是就上网
找到了subprocess模块。具体的用法还是自行百度,网上资料很多的。
主要是用subprocess.Popen()这个方法通过shell创建一个子进程,不过这个方法不太可行,主要的问题出在两者的比较上。。大家有兴趣可以去做一下,主要是为了练习python
0x02:
继续百度于是找到了psutil这个模块
代码如下:

import subprocess
import sys
import psutil
import time
import datetime
log_file='monitor_log.txt'
def print_process_infORMation(pid):
    p=psutil.Process(pid)
    message='[!][Add]pid:'+str(p.pid)+'  '+'name:'+p.name()+'  '+'username:'+p.username()+'  '+'status:'+p.status()+'  '+'start_time:'+datetime.datetime.fromtimestamp(p.create_time()).strftime("%H:%M:%S")
    print message
    fp=open(log_file,'a')
    try:
        message+='\n'
        fp.writelines(message)
    finally:
        fp.close()

def psutil_process():
    pids_1=psutil.pids()
    time.sleep(2)
    pids_2=psutil.pids()
    for pid in pids_2:
        if pid in pids_1:
            #print 'ok!'
            pass
        else:
            print_process_information(pid)
def main():
    while 1:
        psutil_process()

if __name__=="__main__":
    main()

0x03:
最后可以在改进一下

message='[!][Add]pid:'+str(p.pid)+'  '+'name:'+p.name()+'  '+'username:'+p.username()+'  '+'status:'+p.status()+'  '+'start_time:'+datetime.datetime.fromtimestamp(p.create_time()).strftime("%H:%M:%S")

这里可以用格式化输出。。做个元组什么的
0x04:
通过这个小小的进程监控,我学习了subprocess和psutil,了解了管道的单方向性。Perfect!

相关文章