Пример #1
0
    def __init__(self, is_rpc = True):
        super(MessageManager, self).__init__()
        self.setDaemon(True)

        if self.DEBUG:
            import logging
            snakemq.init_logging(open("snakemq-%s.log" % getpid(), "w"))
            logger = logging.getLogger("snakemq")
            logger.setLevel(logging.DEBUG)

        self.__pid = getpid()
        self.__rpc_name = "golismero-rpc-%d" % self.__pid
        self.__queue_name = "golismero-queue-%d" % self.__pid
        if is_rpc:
            self.__name = self.__rpc_name
        else:
            self.__name = self.__queue_name
        self.debug("__init__(%r)" % self.name)

        self.__link = snakemq.link.Link()
        self.__packeter = snakemq.packeter.Packeter(self.__link)
        self.__messaging = snakemq.messaging.Messaging(
            self.__name, "", self.__packeter)
        self.__messaging.on_message_recv.add(self.__callback)

        self.__address = None

        self.__queue = Queue()
        ##self.__queue.mutex = RLock()
        ##self.__queue.not_empty = Condition(self.__queue.mutex)
        ##self.__queue.not_full = Condition(self.__queue.mutex)
        ##self.__queue.all_tasks_done = Condition(self.__queue.mutex)

        self.debug("__init__(%r) => completed" % self.name)
Пример #2
0
    def __init__(self, is_rpc=True):
        super(MessageManager, self).__init__()
        self.setDaemon(True)

        if self.DEBUG:
            import logging
            snakemq.init_logging(open("snakemq-%s.log" % getpid(), "w"))
            logger = logging.getLogger("snakemq")
            logger.setLevel(logging.DEBUG)

        self.__pid = getpid()
        self.__rpc_name = "golismero-rpc-%d" % self.__pid
        self.__queue_name = "golismero-queue-%d" % self.__pid
        if is_rpc:
            self.__name = self.__rpc_name
        else:
            self.__name = self.__queue_name
        self.debug("__init__(%r)" % self.name)

        self.__link = snakemq.link.Link()
        self.__packeter = snakemq.packeter.Packeter(self.__link)
        self.__messaging = snakemq.messaging.Messaging(self.__name, "",
                                                       self.__packeter)
        self.__messaging.on_message_recv.add(self.__callback)

        self.__address = None

        self.__queue = Queue()
        ##self.__queue.mutex = RLock()
        ##self.__queue.not_empty = Condition(self.__queue.mutex)
        ##self.__queue.not_full = Condition(self.__queue.mutex)
        ##self.__queue.all_tasks_done = Condition(self.__queue.mutex)

        self.debug("__init__(%r) => completed" % self.name)
Пример #3
0
import argparse
import os
import sys

from ConfigParser import RawConfigParser
from getpass import getpass
from glob import glob
from os import getenv, getpid
from thread import get_ident
from traceback import format_exc

# Hack to disable logging in SnakeMQ.
import snakemq
if path.sep == "\\":
    snakemq.init_logging(open("nul", "w"))
else:
    snakemq.init_logging(open("/dev/null", "w"))

#------------------------------------------------------------------------------
# GoLismero modules

from golismero.api.config import Config
from golismero.api.external import run_external_tool
from golismero.api.logger import Logger
from golismero.api.plugin import CATEGORIES, STAGES
from golismero.common import OrchestratorConfig, AuditConfig, get_profile, \
     get_available_profiles, get_default_plugins_folder
from golismero.main import launcher
from golismero.main.console import get_terminal_size, colorize, Console
from golismero.main.testing import PluginTester
Пример #4
0
#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
sys.path.insert(0, "../")

import logging

import snakemq
import snakemq.link

snakemq.init_logging()
logger = logging.getLogger("snakemq")
logger.setLevel(logging.DEBUG)

sslcfg = snakemq.link.SSLConfig("testpeer.key", "testpeer.crt")

s = snakemq.link.Link()
s.add_listener(("", 4000), ssl_config=sslcfg)

s.loop()
Пример #5
0
sys.path.insert(0, "../")

import logging

import snakemq
import snakemq.link
import snakemq.packeter
import snakemq.messaging
import snakemq.message


def on_recv(conn, ident, message):
    print("received from", conn, ident, message)


snakemq.init_logging()
logger = logging.getLogger("snakemq")
logger.setLevel(logging.DEBUG)

ssl_cfg = snakemq.link.SSLConfig("../tests/unittests/testkey.pem", "../tests/unittests/testcert.pem")
s = snakemq.link.Link()
s.add_connector(("localhost", 4000), ssl_config=ssl_cfg)

pktr = snakemq.packeter.Packeter(s)

m = snakemq.messaging.Messaging("xconnector", "", pktr)
m.on_message_recv.add(on_recv)

s.loop()
Пример #6
0
import argparse
import os
import sys

from ConfigParser import RawConfigParser
from getpass import getpass
from glob import glob
from os import getenv, getpid
from thread import get_ident
from traceback import format_exc

# Hack to disable logging in SnakeMQ.
import snakemq
if path.sep == "\\":
    snakemq.init_logging(open("nul", "w"))
else:
    snakemq.init_logging(open("/dev/null", "w"))


#------------------------------------------------------------------------------
# GoLismero modules

from golismero.api.config import Config
from golismero.api.external import run_external_tool
from golismero.api.logger import Logger
from golismero.api.plugin import CATEGORIES, STAGES
from golismero.common import OrchestratorConfig, AuditConfig, get_profile, \
     get_available_profiles, get_default_plugins_folder
from golismero.main import launcher
from golismero.main.console import get_terminal_size, colorize, Console