Exemple #1
0
 def __init__(self, ix, task_queue, pid):
     self.serverName = "%s_%d" % (config.get_mac_address(), pid)
     self.conn = pymongo.Connection(config.MONGO_CONN)
     self.pageurls = self.conn.pongo.pageurls
     self.task_queue = task_queue
     self.ix = ix
     self.p = re.compile(u"(http://blog.csdn.net/[^\\/]+/article/details/\\d+)")
Exemple #2
0
 def __init__(self, ix, task_queue, pid):
     self.serverName = "%s_%d" % (config.get_mac_address(), pid)
     self.conn = pymongo.Connection(config.MONGO_CONN)
     self.pageurls = self.conn.pongo.pageurls
     self.task_queue = task_queue
     self.ix = ix
     self.p = re.compile(
         u'(http://blog.csdn.net/[^\\/]+/article/details/\\d+)')
Exemple #3
0
    def deal_pkg(self, header, body):
        res = None
        if header.cmd == pkg.SET_CHANNEL_CTRL_REQ:
            res = pkg.Header(cmd=pkg.SET_CHANNEL_CTRL_RSP)
            for k in ('mode', 'enable', 'mu_factor'):
                Config.set(header.channel, k, getattr(body, k))

            wsets = [(w.begin, w.noise, w.end) for w in body.data]
            Config.set(header.channel, 'win_settings', wsets)
        elif header.cmd == pkg.GET_BASE_INFO_REQ:
            res = pkg.Header(cmd=pkg.GET_BASE_INFO_RSP)
            body = pkg.BaseInfo(
                mac=get_mac_address(self.ifname),
                hwid='Emultr',
                hwcode='EM',
                hw_version=chr(2) + chr(0),
                sw_version=chr(2) + chr(0),
                sw_revision=0,
                proto_version=2,
                channel_num=2,
                machine_id=0,
                ip_num=1,
                slot_id=0
            )
            res = pkg.combine(res, body)
        elif header.cmd == pkg.SET_SWITCH_CTRL_REQ:
            res = pkg.Header(cmd=pkg.SET_SWITCH_CTRL_RSP)
            if body.led == 1:       # light on
                if self.test:
                    algorithm_test(self.ip + ":" + str(header.channel))
                led = 'LED RED'
            elif body.led == 0xff:  # blink
                led = 'LED BLINK'
            elif body.led == 0:     # light off
                led = 'LED BLACK'
            logger.critical((self.ip, led))
        elif header.cmd == pkg.GET_CHANNEL_CTRL_REQ:
            res = pkg.Header(cmd=pkg.GET_CHANNEL_CTRL_RSP)
            ch_ctl = pkg.ChannelCtrl()
            for k in ('enable', 'mu_factor', 'mode'):
                setattr(ch_ctl, k, Config.get(header.channel, k))
            win_sts = Config.get(header.channel, 'win_settings')
            ch_ctl.win_size = len(win_sts)
            for ws in win_sts:
                wp = pkg.ChannelCtrl.WinSetting(
                    begin=ws[0],
                    end=ws[1],
                    noise=ws[2]
                )
                ch_ctl.data += wp.pack()
            res.data = ch_ctl.pack()
        else:
            logger.info(repr((header, body)))
        if res:
            res.seq = header.seq
            return res.pack()
Exemple #4
0
HEARTBEAT_LIVENESS = config.HEARTBEAT_LIVENESS  # 3..5 is reasonable
HEARTBEAT_INTERVAL = config.HEARTBEAT_INTERVAL  # Seconds

INTERVAL_INIT = config.INTERVAL_INIT
INTERVAL_MAX = config.INTERVAL_MAX

#  Paranoid Pirate Protocol constants
PPP_READY = config.PPP_READY  # Signals worker is ready
PPP_HEARTBEAT = config.PPP_HEARTBEAT  # Signals worker heartbeat

INDEX_PATH = "indexdir"
HOST = 'localhost'
WORKER_HOST = "tcp://localhost:5556"
SUBSCRIBER_HOST = "tcp://localhost:5557"

mac_address = config.get_mac_address()


def worker_socket(context, poller, pid):
    """Helper function that returns a new configured socket
       connected to the Paranoid Pirate queue"""
    worker = context.socket(zmq.DEALER)  # DEALER
    identity = "work-%s-%d:%04X-%04X" % (mac_address, pid, randint(
        0, 0x10000), randint(0, 0x10000))
    worker.setsockopt(zmq.IDENTITY, identity)
    poller.register(worker, zmq.POLLIN)
    worker.connect(WORKER_HOST)
    worker.send(PPP_READY)
    return worker