Пример #1
0
    def __init__(self, config):
        """
        XMPP agent class.
        """
        log.info("Starting agent...")
        self.config = config

        log.info("Setting up certificate")
        self.public_key = None
        try:
            if Crypto.version_info[:2] >= (2, 2):
                _public_key = self._read_pub_key()
                if _public_key:
                    key = PublicKey.RSA.importKey(_public_key)
                    self.public_key = key.publickey()
        except:
            pass

        if not self.public_key:
            log.warn('PyCrypto not available or version is < 2.2: Please upgrade: http://www.pycrypto.org/')

        log.info("Loading commands...")
        self.command_runner = CommandRunner(config['Plugins'])

        log.debug("Loading XMPP...")
        observers = [
            ('/iq', self.__onIq),
        ]
        Client.__init__(self,
                        self.config['XMPP'],
                        observers,
                        resource='ecm_agent-%d' % AGENT_VERSION_PROTOCOL
        )
Пример #2
0
    def _processCommand(self, message):
        log.debug('Process Command')

        if self.public_key:
            if not self._verify_message(message):
                log.critical('[RSA CHECK: Failed] Command from %s has bad signature (Ignored)' % message.from_)
                result = (_E_UNVERIFIED_COMMAND, '', 'Bad signature', 0)
                self._onCallFinished(result, message)
                return
        else:
            # No public key, RSA functions are not available on this system
            log.warn('[RSA CHECK: No available] WARNING: Running unverified Command from %s' % message.from_)

        flush_callback = self._Flush
        message.command_replaced = message.command.replace('.', '_')
        d = self.command_runner.run_command(message.command_replaced, message.command_args, flush_callback, message)

        if d:
            d.addCallbacks(self._onCallFinished, self._onCallFailed,
                           callbackKeywords={'message': message},
                           errbackKeywords={'message': message},
            )
            return d

        else:
            log.info("Command Ignored: Unknown command: %s" % message.command)
            result = (_E_RUNNING_COMMAND, '', "Unknown command: %s" % message.command, 0)
            self._onCallFinished(result, message)

        return
Пример #3
0
 def _onPossibleErrorIq(self, elem):
     sender = elem['from']
     for el in elem.elements():
         if el.name == 'error' and el['code'] == '404':
             log.warn('Received a 404 code from the server, setting the target user as offline')
             if sender in self._online_contacts:
                 self._online_contacts.remove(sender)
             else:
                 log.debug('Received a 404 from %s which not (anymore?) in the online contacts list.')
Пример #4
0
    def __onIq(self, msg):
        """
        A new IQ message has been received and we should process it.
        """
        log.debug('__onIq')

        log.debug("q Message received: \n%s" % msg.toXml())
        log.debug("Message type: %s" % msg['type'])

        if self.keepalive.running:
            log.debug("Stop keepalived")
            self.keepalive.stop()
        
        log.debug("Starting keepalived")
        self.keepalive.start(KEEPALIVED_TIMEOUT, now=False)

        self.timestamp = time()

        message = IqMessage(msg)
        recv_command = message.command.replace('.', '_')

        if recv_command in self.running_commands:
            if self.timestamp > self.running_commands[recv_command]:
                del self.running_commands[recv_command]
                self.num_running_commands -= 1
                log.debug("Deleted %s from running_commands dict as should have been completed" % (recv_command))

        if recv_command not in self.running_commands:
            log.debug('recieved new command: %s with message: %s' % (message.command, message))

            if hasattr(message, 'command') and hasattr(message, 'from_'):
                log.debug('online contacts: %s' % self._online_contacts)

                if message.from_ not in self._online_contacts:
                    log.warn('IQ sender not in roster (%s), dropping message' % message.from_)
                else:
                    self.running_commands[recv_command] = self.timestamp + int(message.command_args['timeout'])
                    self.num_running_commands += 1
                    log.debug("Running commands: names: %s numbers: %i" % (self.running_commands, self.num_running_commands))
                    self._processCommand(message)

        else:
            log.debug("already running given command %s" %recv_command)
            result = (_E_RUNNING_COMMAND, '', 'another command is running', 0)
            self._send(result, message)

        del msg
        del message
Пример #5
0
    def __onIq(self, msg):
        """
        A new IQ message has been received and we should process it.
        """
        log.debug('__onIq')
        mem_clean('__onIq [start]')
        self.running_commands += 1

        log.debug("q Message received: \n%s" % msg.toXml())
        log.debug("Message type: %s" % msg['type'])

        if msg['type'] == 'set':
            message = IqMessage(msg)

            if hasattr(message, 'command') and hasattr(message, 'from_'):
                log.debug('online contacts: %s' % self._online_contacts)

                if message.from_ not in self._online_contacts:
                    log.warn('IQ sender not in roster (%s), dropping message' % message.from_)
                else:
                    self._processCommand(message)
            else:
                log.warn('Unknown ecm_message received: Full XML:\n%s' % (msg.toXml()))
                
            del message
            
        else:
            log.warn('Unknown IQ type received: Full XML:\n%s' % (msg.toXml()))

        del msg
        mem_clean('__onIq [end]')
Пример #6
0
    def __onIq(self, msg):
        """
        A new IQ message has been received and we should process it.
        """
        log.debug('__onIq')
        message_type = msg['type']

        log.debug("q Message received: \n%s" % msg.toXml())
        log.debug("Message type: %s" % message_type)

        if message_type == 'set':
            #Parse and check message format
            message = IqMessage(msg)

            if hasattr(message, 'command') and hasattr(message, 'from_'):
                log.debug('online contacts: %s' % self._online_contacts)

                if message.from_ not in self._online_contacts:
                    log.warn('IQ sender not in roster (%s), dropping message'
                           % message.from_)
                else:
                    self._processCommand(message)
            else:
                log.warn('Unknown ecm_message received: "%s" Full XML:\n%s'
                       % (message_type, msg.toXml()))
        else:
            log.warn('Unknown IQ type received: "%s" Full XML:\n%s'
                   % (message_type, msg.toXml()))
Пример #7
0
 def _onErrorRunning(self, failure):
     log.warn('Command failed to execute: %s' % failure)
     return 255, '', ''