Python init示例

示例#1

文件： av_agent.py 项目： elfland/HackTeam-test-av2

    def __init__(self, vm, redis='localhost', session=None):
        self.vm = vm
        self.host = redis
        self.session = session
        command.init()
        shutil.rmtree('build', ignore_errors=True)
        if os.path.exists(config.basedir_crop):
            shutil.rmtree(config.basedir_crop)

        if 'logging' not in locals():

            from AVCommon.logger import logging
            locals()['logging']=logging

        logging.debug("vm: %s host: %s session: %s" % (self.vm, self.host, session))

        command.context["report"] = self.report

        # TODO
        # load default.yaml as a default SET command.
        f = open("AVAgent/default.yaml")
        y = yaml.load(f)
        for k,v in y.items():
            command.context[k] = v
        logging.debug("command.context size: %s" % len(command.context))

示例#2

文件： av_master.py 项目： elfland/HackTeam-test-av2

 def __init__(self, args):
     self.args = args
     self.vm_names = args.vm.split(',')
     self.procedure = args.procedure.upper()
     self.pool = args.pool
     command.init()

示例#3

文件： test_commands.py 项目： BwRy/test-av2

from AVCommon.commands.client import SET

__author__ = 'fabrizio'

import sys
import os

sys.path.append(os.path.split(os.getcwd())[0])
sys.path.append(os.getcwd())

from AVCommon.logger import logging

from AVCommon import command
from AVCommon import helper

command.init()

def test_START_VM():
    procs = """Process list: 40
pid=0, owner=, cmd=[System Process]
pid=4, owner=NT AUTHORITY\SYSTEM, cmd=System
pid=264, owner=NT AUTHORITY\SYSTEM, cmd=smss.exe
pid=360, owner=NT AUTHORITY\SYSTEM, cmd=csrss.exe
pid=400, owner=NT AUTHORITY\SYSTEM, cmd=wininit.exe
pid=412, owner=NT AUTHORITY\SYSTEM, cmd=csrss.exe
pid=480, owner=NT AUTHORITY\SYSTEM, cmd=services.exe
pid=488, owner=NT AUTHORITY\SYSTEM, cmd=lsass.exe
pid=500, owner=NT AUTHORITY\SYSTEM, cmd=lsm.exe
pid=508, owner=NT AUTHORITY\SYSTEM, cmd=winlogon.exe
pid=624, owner=NT AUTHORITY\SYSTEM, cmd=svchost.exe
pid=696, owner=NT AUTHORITY\NETWORK SERVICE, cmd=svchost.exe

示例#4

文件： test_commands.py 项目： elfland/HackTeam-test-av2

from AVCommon.commands.client import SET

__author__ = 'fabrizio'

import sys
import os

sys.path.append(os.path.split(os.getcwd())[0])
sys.path.append(os.getcwd())

from AVCommon.logger import logging

from AVCommon import command
from AVCommon import helper

command.init()


def test_START_VM():
    procs = """Process list: 40
pid=0, owner=, cmd=[System Process]
pid=4, owner=NT AUTHORITY\SYSTEM, cmd=System
pid=264, owner=NT AUTHORITY\SYSTEM, cmd=smss.exe
pid=360, owner=NT AUTHORITY\SYSTEM, cmd=csrss.exe
pid=400, owner=NT AUTHORITY\SYSTEM, cmd=wininit.exe
pid=412, owner=NT AUTHORITY\SYSTEM, cmd=csrss.exe
pid=480, owner=NT AUTHORITY\SYSTEM, cmd=services.exe
pid=488, owner=NT AUTHORITY\SYSTEM, cmd=lsass.exe
pid=500, owner=NT AUTHORITY\SYSTEM, cmd=lsm.exe
pid=508, owner=NT AUTHORITY\SYSTEM, cmd=winlogon.exe
pid=624, owner=NT AUTHORITY\SYSTEM, cmd=svchost.exe