def test_command_echo(self): sessions = {} users = {'test': BaitUser('test', 'test')} authenticator = Authenticator(users) Session.authenticator = authenticator cap = honeypot_ssh.SSH(sessions, {'enabled': 'True', 'port': 0, 'max_attempts': 3}, users, self.work_dir, self.key) srv = StreamServer(('0.0.0.0', 0), cap.handle_session) srv.start() bee_info = { 'timing': 'regular', 'username': '******', 'password': '******', 'port': srv.server_port, 'server': '127.0.0.1' } beesessions = {} BaitSession.client_id = 'f51171df-c8f6-4af4-86c0-f4e163cf69e8' current_bee = client_ssh.ssh(beesessions, bee_info) current_bee.connect_login() resp = current_bee.echo('just testing!') self.assertTrue('just testing!' in resp)
def test_logout(self): """Tests if the SSH bee can Logout from the SSH capability""" sessions = {} users = {'test': BaitUser('test', 'test')} authenticator = Authenticator(users) Session.authenticator = authenticator cap = honeypot_ssh.SSH(sessions, {'enabled': 'True', 'port': 0, 'max_attempts': 3}, users, self.work_dir, self.key) srv = StreamServer(('0.0.0.0', 0), cap.handle_session) srv.start() bee_info = { 'timing': 'regular', 'username': '******', 'password': '******', 'port': srv.server_port, 'server': '127.0.0.1' } beesessions = {} BaitSession.client_id = 'f51171df-c8f6-4af4-86c0-f4e163cf69e8' current_bee = client_ssh.ssh(beesessions, bee_info) current_bee.connect_login() current_bee.logout() srv.stop()
def test_validate_senses(self): sessions = {} users = {'test': BaitUser('test', 'test')} authenticator = Authenticator(users) Session.authenticator = authenticator cap = honeypot_ssh.SSH(sessions, {'enabled': 'True', 'port': 0, 'max_attempts': 3}, users, self.work_dir, self.key) srv = StreamServer(('0.0.0.0', 0), cap.handle_session) srv.start() bee_info = { 'timing': 'regular', 'username': '******', 'password': '******', 'port': srv.server_port, 'server': '127.0.0.1' } beesessions = {} BaitSession.client_id = 'f51171df-c8f6-4af4-86c0-f4e163cf69e8' current_bee = client_ssh.ssh(beesessions, bee_info) for s in current_bee.senses: sense = getattr(current_bee, s) self.assertTrue(callable(sense))