OSSIM项目研究记录（七）

2021SC@SDUSC

---

前言

本次源码分析取自agent目录下的command.py，主要实现的功能是代理信息的处理、插件状态的查询等。

一、引入模块

from __init__ import __version__
from uuid import UUID
from bson import BSON
from bson.codec_options import CodecOptions
from bson.binary import STANDARD

1.从_init_.py文件中引入_version_.这里我查看文件得到：

__version__ = "VERSION"

2.UUID是Universally Unique Identifier的缩写，它是在一定的范围内（从特定的名字空间到全球）唯一的机器生成的标识符。UUID具有以下涵义：

• 经由一定的算法机器生成
• 非人工指定，非人工识别
• 在特定的范围内重复的可能性极小

3.bson
BSON是一种类json的一种二进制形式的存储格式，简称Binary JSON，它和JSON一样，支持内嵌的文档对象和数组对象，但是BSON有JSON没有的一些数据类型，如Date和BinData类型。

二、类

定义一个抽象父类，函数体都用pass占位。

class Command(object):
    """Command Parent abstract class"""

    def __init__(self):
        pass

    def to_string(self):
        """Abstract method, must be overwritten"""
        pass

    def to_bson(self):
        """Abstract method, must be overwritten"""
        pass

    def is_idm_event(self):
        return False

类：插件未知状态。定义三个函数
主要作用是返回bson信息并输出"插件进程未知“，且返回插件id

class PluginUnknownState(Command):
    def __init__(self, plugin_id):
        super(PluginUnknownState, self).__init__()
        self.plugin_id = plugin_id

    def to_bson(self):
        data = {'plugin-process-unknown': {'plugin_id': int(self.plugin_id)}}
        return BSON.encode(data)

    def to_string(self):
        return "plugin-process-unknown plugin_id=\"%s\"\n" % self.plugin_id

类PluginStartState、PluginStopState、PluginEnableState、PluginDisableState作用类似。

类：附件插件，返回附件插件的信息和状态

class AppendPlugin(Command):
    MSG = 'session-append-plugin id="{0}" plugin_id="{1}" enabled="{2}" state="{3}"\n'
    PLUGIN_STARTED = 1
    PLUGIN_STOPPED = 2

    def __init__(self, plugin_id, sequence_id, state, enabled):
        super(AppendPlugin, self).__init__()
        self.plugin_id = plugin_id
        self.sequence_id = sequence_id
        self.state = state
        self.enabled = enabled

    def to_bson(self):
        state = int(AppendPlugin.PLUGIN_STARTED) if self.state == "start" else int(AppendPlugin.PLUGIN_STOPPED)
        data = {
            'session-append-plugin': {'plugin_id': self.plugin_id,
                                      'id': self.sequence_id,
                                      'enabled': bool(True),
                                      'state': state}}
        return BSON.encode(data)

    def to_string(self):
        return AppendPlugin.MSG.format(self.sequence_id,
                                       self.plugin_id,
                                       self.enabled,
                                       self.state)