Esempio n. 1
0
    def testCommandTimeout(self, MockExpect):
        '''
        If after invoking a cli command the expected prompt is not found in the response a timeout event is triggered. 
        '''

        responses = [
            'pyco@localhost password: '******'''Linux cencenighe 2.6.32-30-generic #59-Ubuntu SMP Tue Mar 1 21:30:46 UTC 2011 x86_64 GNU/Linux\r
        Ubuntu 10.04.2 LTS\r
        \r
        Welcome to Ubuntu!\r
         * Documentation:  https://help.ubuntu.com/\r
        \r
        Last login: Thu Feb 24 09:05:39 2011 from localhost\r\n$ ''',
            'to be cleared by clear_buffer', '$ ',
            'to be cleared by clear_buffer', 'no prompt'
        ]
        log.info("testLoginTimeout ...")
        h = device('telnet://*****:*****@h')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        h.login()

        self.failUnlessRaises(ConnectionTimedOut, h.send, 'id')
    def testPromptDiscoveryDisabled(self, MockExpect):
        '''
        Disable the prompt discovery. 
        '''

        responses = [
            'pyco@localhost password: '******'''Linux cencenighe 2.6.32-30-generic #59-Ubuntu SMP Tue Mar 1 21:30:46 UTC 2011 x86_64 GNU/Linux\r
        Ubuntu 10.04.2 LTS\r
        \r
        Welcome to Ubuntu!\r
         * Documentation:  https://help.ubuntu.com/\r
        \r
        Last login: Thu Feb 24 09:05:39 2011 from localhost\r\n$ some output''',
            'i am pyco'
        ]

        log.info("testLoginTimeout ...")
        h = device('telnet://*****:*****@h')
        h.discoverPrompt = False

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        h.login()

        h.send('id')
Esempio n. 3
0
    def testTC1(self, MockExpect):
        log.info("testTC1 ...")
        h = device('ssh://*****:*****@host')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        self.failUnlessRaises(PermissionDenied, h.login)
    def testTC1(self, MockExpect):
        log.info("testTC1 ...")
        h = device('ssh://*****:*****@host')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        self.failUnlessRaises(PermissionDenied, h.login)
Esempio n. 5
0
    def testFakeOk(self):
        log.debug("testFakeOk ...")
        h = device("telnet://%s:%s@%s:%d" % ("username", "secret", "localhost", TELNET_PORT))

        h.maxWait = 2
        out = h("id")
        print("--> %s" % out)
        self.assertRegex(out, "uid=[0-9]+\(pyco\).*")
Esempio n. 6
0
    def testNativeTelnet(self):
        log.info("testNative ...")
        user = environ["USER"]
        log.info("USER: [%s]" % user)
        h = device("telnet://%s:%s@%s" % (user, user, "localhost"))
        h.maxWait = 10

        self.assertRaises(PermissionDenied, h, "id")
Esempio n. 7
0
    def testSimpleCommand(self):
        log.debug("starting testSimpleCommand ...")
        target = device('telnet://%s:%s@%s:%d' %
                        ('kenobi', 'secret', 'localhost', TELNET_PORT))

        target.maxWait = 2
        out = target('id')
        target.close()
        self.assertRegex(out, 'uid=[0-9]+\(pyco\).*')
Esempio n. 8
0
    def _testLoginTimeout(self, MockExpect):
        responses = ['unexpected response (expected password prompt)', 'to be cleared by clear_buffer'] 

        log.info("testLoginTimeout ...")
        h = device('telnet://*****:*****@h')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        self.failUnlessRaises(LoginFailed, h.login)
Esempio n. 9
0
    def _testLoginTimeout(self, MockExpect):
        responses = [
            'unexpected response (expected password prompt)',
            'to be cleared by clear_buffer'
        ]

        log.info("testLoginTimeout ...")
        h = device('telnet://*****:*****@h')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        self.failUnlessRaises(LoginFailed, h.login)
Esempio n. 10
0
    def testTC1(self, MockExpect):
        log.info("testTC1 ...")
        h = device('telnet://*****:*****@h')
        h.maxWait = 20

        print(h.patternMap['GROUND'])

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        out = h.login()
    
        h.send('id')
Esempio n. 11
0
    def testNoPassword(self, MockExpect):
        from pyco.device import cliIsConnected
        log.info("testNoPassword ...")
        #h = device('telnet://%s:%s@%s' % (hop1['username'], hop1['password'], hop1['name']))
        h = device('telnet://*****:*****@h')
        
        h.remove_event('username_event', 'GROUND')
        h.add_event_action('timeout', action=cliIsConnected, endState='USER_PROMPT')
        
        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = self.responses

        out = h.login()
    def testNoPromptAfterResponse(self, MockExpect):
        from pyco.device import cliIsConnected
        log.info("testNoPromptAfterResponse ...")
        cisco = device('telnet://*****:*****@h')
        
        from pyco.actions import sendUsername
        cisco.add_event_action('username-event', pattern='Username: '******'GROUND')
        
        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = self.responses

        out = cisco.login()
    
        self.failUnlessRaises(ConnectionTimedOut, cisco.send, 'show version')
Esempio n. 13
0
    def testNoPassword(self, MockExpect):
        from pyco.device import cliIsConnected
        log.info("testNoPassword ...")
        #h = device('telnet://%s:%s@%s' % (hop1['username'], hop1['password'], hop1['name']))
        h = device('telnet://*****:*****@h')

        h.remove_event('username_event', 'GROUND')
        h.add_event_action('timeout',
                           action=cliIsConnected,
                           endState='USER_PROMPT')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = self.responses

        out = h.login()
Esempio n. 14
0
 def testMulti(self):
     log.debug("testMulti ...")
     setface("ciscoios")
     
     h = device('telnet://%s:%s@%s:%d/ciscoios' % 
                ('obi-wan-kenobi', 'secret', 'localhost',TELNET_PORT))
     h.maxWait = 2
     
     #p = Client()[:]
     
     hosts = [h] * 10
     
     with Pool(5) as p:
         results = p.map(command, hosts)
         self.assertEqual(len(results), 10, "expected 10 command outputs")
         self.assertNotIn(None, results, "some commands failed unexpectely")
Esempio n. 15
0
    def testMulti(self):
        log.debug("testMulti ...")
        setface("ciscoios")

        h = device('telnet://%s:%s@%s:%d/ciscoios' %
                   ('obi-wan-kenobi', 'secret', 'localhost', TELNET_PORT))
        h.maxWait = 2

        #p = Client()[:]

        hosts = [h] * 10

        with Pool(5) as p:
            results = p.map(command, hosts)
            self.assertEqual(len(results), 10, "expected 10 command outputs")
            self.assertNotIn(None, results, "some commands failed unexpectely")
    def testNoPromptAfterResponse(self, MockExpect):
        from pyco.device import cliIsConnected
        log.info("testNoPromptAfterResponse ...")
        cisco = device('telnet://*****:*****@h')

        from pyco.actions import sendUsername
        cisco.add_event_action('username-event',
                               pattern='Username: '******'GROUND')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = self.responses

        out = cisco.login()

        self.failUnlessRaises(ConnectionTimedOut, cisco.send, 'show version')
Esempio n. 17
0
    def testCommandTimeout(self, MockExpect):
        '''
        If after invoking a cli command the expected prompt is not found in the response a timeout event is triggered. 
        '''
        
        responses = ['pyco@localhost password: '******'''Linux cencenighe 2.6.32-30-generic #59-Ubuntu SMP Tue Mar 1 21:30:46 UTC 2011 x86_64 GNU/Linux\r
        Ubuntu 10.04.2 LTS\r
        \r
        Welcome to Ubuntu!\r
         * Documentation:  https://help.ubuntu.com/\r
        \r
        Last login: Thu Feb 24 09:05:39 2011 from localhost\r\n$ ''',
        'to be cleared by clear_buffer', '$ ','to be cleared by clear_buffer', 'no prompt']
        log.info("testLoginTimeout ...")
        h = device('telnet://*****:*****@h')

        MockExpect.side_effect = simulator.side_effect
        simulator.side_effect.responses = responses

        h.login()
         
        self.failUnlessRaises(ConnectionTimedOut, h.send, 'id')
Esempio n. 18
0
            if target.has_event_handlers(target.currentEvent):
                log.debug("[%s] got [%s] event; invoking handlers: [%s]" % (target.name, target.currentEvent.name, target.get_event_handlers(target.currentEvent)))
                for eh in target.get_event_handlers(target.currentEvent):
                    eh(target)
           
            stateChanged = target.process(target.currentEvent)
            response += self.pipe.before
            if isinstance(self.pipe.after, str) and not target.currentEvent.isPromptMatch():
                response += self.pipe.after

        return response

#    def processResponseWithTimeout(self, target, checkPoint):
#        patterns = [pexpect.TIMEOUT]
#        return self.patternMatch(target, checkPoint, patterns, target.maxWait, exactMatch=True)
        
    def processResponse(self, target, checkPoint):
        '''
        '''
        return self.patternMatch(target, checkPoint, [TIMEOUT], target.maxWait)
        
        
 


# The source point of all paths        
SOURCE_HOST = device('__source_host__')

# the source is connected for definition 
SOURCE_HOST.is_connected = lambda : True
        
Esempio n. 19
0
    def testWrongPassword(self):
        log.debug("testWrongPassword ...")
        h = device("telnet://%s:%s@%s:%d" % ("username", "wrong_pwd", "localhost", TELNET_PORT))

        self.assertRaises(PermissionDenied, h, "id")
Esempio n. 20
0
 def testWrongPassword(self):
     log.debug("testWrongPassword ...")
     target = device('telnet://%s:%s@%s:%d' %
                     ('kenobi', 'wrong_pwd', 'localhost', TELNET_PORT))
     target.close()
     self.assertRaises(PermissionDenied, target, 'id')
Esempio n. 21
0
def doUname(hostname, user, pwd, protocol):
    print("doUname")
    h = device("%s://%s:%s@%s:7777" % (protocol, user, pwd, hostname))
    out = h('uname -a')
Esempio n. 22
0
 def testWrongUsername(self):
     log.debug("testWrongUsername ...")
     target = device('telnet://%s:%s@%s:%d' %
                     ('fake_username', 'secret', 'localhost', TELNET_PORT))
     target.close()
     self.assertRaises(PermissionDenied, target, 'id')
Esempio n. 23
0
 def testConnectionRefused(self):
     log.debug("testConnectionRefused ...")
     target = device('telnet://%s:%s@%s:%d' %
                     ('fake_username', 'secret', 'localhost', 9999))
     target.close()
     self.assertRaises(ConnectionRefused, target, 'id')
Esempio n. 24
0
def doUname(hostname,user,pwd,protocol):
    print("doUname")
    h=device("%s://%s:%s@%s:7777" % (protocol,user,pwd,hostname))
    out=h('uname -a')
Esempio n. 25
0
def command(host):
    try:
        print("----> [%s]" % host.driver)
        if (host.driver.name == "ciscoios"):
            cioscoios_show_ip_local_pool(host)
        elif(host.driver.name == 'juniper'):
            print("Thommy lets happen!")
        else:
            print("unknown device type %s" % host.driver)
    except Exception:
        #print("%s: interrogazione fallita, vedere file di log per i dettagli" % host.name)
        #traceback.print_exc(file=sys.stdout)
        logging.exception("interrogazione fallita")
    finally:
        pass

if __name__ == '__main__':

    indb = Database()
    
    with Pool(4) as pool:
    
        for n in indb.getall():
            print("telnetting %s" % n.name)
            host = device('telnet://%s:%s@%s:%d/ciscoios' % 
                   (n.username, n.password, 'localhost', TELNET_PORT))
            
            pool.apply(command, (host,))
    
Esempio n. 26
0
            if target.has_event_handlers(target.currentEvent):
                log.debug("[%s] got [%s] event; invoking handlers: [%s]" %
                          (target.name, target.currentEvent.name,
                           target.get_event_handlers(target.currentEvent)))
                for eh in target.get_event_handlers(target.currentEvent):
                    eh(target)

            stateChanged = target.process(target.currentEvent)
            response += self.pipe.before
            if isinstance(self.pipe.after,
                          str) and not target.currentEvent.isPromptMatch():
                response += self.pipe.after

        return response

#    def processResponseWithTimeout(self, target, checkPoint):
#        patterns = [pexpect.TIMEOUT]
#        return self.patternMatch(target, checkPoint, patterns, target.maxWait, exactMatch=True)

    def processResponse(self, target, checkPoint):
        '''
        '''
        return self.patternMatch(target, checkPoint, [TIMEOUT], target.maxWait)


# The source point of all paths
SOURCE_HOST = device('__source_host__')

# the source is connected for definition
SOURCE_HOST.is_connected = lambda: True