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)