[Linux] linux审计脚本 -- 录屏用户操作 (记录用户屏幕输出内容)

原创

大大刺猬

发布于 2024-07-16 14:07:06

860

发布于 2024-07-16 14:07:06

文章被收录于专栏：大大刺猬大大刺猬

导读

之前我记得有人问过linux上, 怎么记录用户的所有操作,包括输出的结果. 那时候我还不知道. 今天看python文档pty的时候看到个官方例子就是使用一个伪终端来模拟终端. 这不就是我们要的方法么.

官方链接: https://docs.python.org/3/library/pty.html

操作过程

官方的例子, 实际上就可以直接使用的. 但我们稍微改一改, 完善一下.

首先创建一个目录, 专门用来记录终端的输出信息的.

mkdir -p /data_audit
chmod o+rw /data_audit # 得给其它账号写的权限

然后在/data_audit文件下创建一个脚本(模拟终端的).

cat << EOF > /data_audit/.pty.py
#!/usr/bin/env python
import argparse
import os
import pty
import sys
import time

parser = argparse.ArgumentParser()
parser.add_argument('-a', dest='append', action='store_true')
parser.add_argument('-p', dest='use_python', action='store_true')
parser.add_argument('filename', nargs='?', default='.ptylogin.log')
options = parser.parse_args()

shell = sys.executable if options.use_python else os.environ.get('SHELL', 'sh')
filename = options.filename
mode = 'ab' if options.append else 'wb'

with open(filename, mode) as script:
    def read(fd):
        data = os.read(fd, 1024)
        script.write(data)
        return data

    print('WARNING! pty login on file:'+filename)
    script.write(('login %s\n' % time.asctime()).encode())

    pty.spawn(shell, read)

    script.write(('logout %s\n' % time.asctime()).encode())

EOF

接着就是配置/etc/profile来实现每次时直接调用该脚本. 添加如下内容即可

if [ "`whoami`" != "root" ];then # 一般root账号是不允许登录的, 我们这里就排除root账号了
    mkdir -p /data_audit/`whoami` # 每个用户一个目录
    filename=/data_audit/`whoami`/`tty|sed 's/\///g'` 
    touch ${filename} # 创建相关日志文件
    python /data_audit/.pty.py -a ${filename}
    exit # 等用户退出登录时,我们也要退出
fi

到这里我们就配置好了.