class ShellUniqCommandTests(unittest.TestCase):
"""Tests for cowrie/commands/uniq.py."""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_uniq_command_001(self) -> None:
self.proto.lineReceived(b"echo test | uniq\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_uniq_command_002(self) -> None:
self.proto.lineReceived(b"echo -e \"test\ntest\ntest\" | uniq\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_uniq_command_003(self) -> None:
self.proto.lineReceived(b"uniq\n")
self.proto.lineReceived(b"test\n")
self.proto.lineReceived(b"test\n")
self.proto.lineReceived(b"test\n")
self.proto.handle_CTRL_D()
self.assertEqual(self.tr.value(), b"test\n\n" + PROMPT)
class ShellFtpGetCommandTests(unittest.TestCase):
"""Tests for cowrie/commands/ftpget.py."""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_help_command(self) -> None:
usage = b"BusyBox v1.20.2 (2016-06-22 15:12:53 EDT) multi-call binary.\n" \
b"\n" \
b"Usage: ftpget [OPTIONS] HOST [LOCAL_FILE] REMOTE_FILE\n" \
b"\n" \
b"Download a file via FTP\n" \
b"\n" \
b" -c Continue previous transfer\n" \
b" -v Verbose\n" \
b" -u USER Username\n" \
b" -p PASS Password\n" \
b" -P NUM Port\n\n"
self.proto.lineReceived(b"ftpget\n")
self.assertEqual(self.tr.value(), usage + PROMPT)
class ShellTeeCommandTests(unittest.TestCase):
"""Tests for cowrie/commands/tee.py."""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_tee_command_001(self) -> None:
self.proto.lineReceived(b"tee /a/b/c/d\n")
self.assertEqual(self.tr.value(),
b"tee: /a/b/c/d: No such file or directory\n")
# tee still waiting input from stdin
self.proto.handle_CTRL_C()
def test_tee_command_002(self) -> None:
self.proto.lineReceived(b"tee /a/b/c/d\n")
self.proto.handle_CTRL_C()
self.assertEqual(
self.tr.value(),
b"tee: /a/b/c/d: No such file or directory\n^C\n" + PROMPT)
def test_tee_command_003(self) -> None:
self.proto.lineReceived(b"tee a\n")
self.proto.lineReceived(b"test\n")
self.proto.handle_CTRL_D()
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_tee_command_004(self) -> None:
self.proto.lineReceived(b"echo test | tee\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_tee_command_005(self) -> None:
self.proto.lineReceived(b"tee\n")
self.proto.lineReceived(b"test\n")
self.proto.handle_CTRL_D()
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
class ShellPipeCommandsTests(unittest.TestCase):
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_shell_pipe_with_cat_tail(self) -> None:
self.proto.lineReceived(b"echo test | tail -n 1\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_shell_pipe_with_cat_head(self) -> None:
self.proto.lineReceived(b"echo test | head -n 1\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
class ShellEchoCommandTests(unittest.TestCase):
"""Tests for cowrie/commands/awk.py."""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_awk_command_001(self) -> None:
self.proto.lineReceived(b"echo \"test test\" | awk \"{ print $0 }\"\n")
self.assertEqual(self.tr.value(), b"test test\n" + PROMPT)
def test_awk_command_002(self) -> None:
self.proto.lineReceived(b"echo \"test\" | awk \"{ print $1 }\"\n")
self.assertEqual(self.tr.value(), b"test\n" + PROMPT)
def test_awk_command_003(self) -> None:
self.proto.lineReceived(b"echo \"test test\" | awk \"{ print $1 $2 }\"\n")
self.assertEqual(self.tr.value(), b"test test\n" + PROMPT)
def test_awk_command_004(self) -> None:
self.proto.lineReceived(b"echo \"test test\" | awk \"{ print $1,$2 }\"\n")
self.assertEqual(self.tr.value(), b"test test\n" + PROMPT)
def test_awk_command_005(self) -> None:
self.proto.lineReceived(b"echo \"test test\" | awk \"{ print $1$2 }\"\n")
self.assertEqual(self.tr.value(), b"testtest\n" + PROMPT)
class ShellBase64CommandTests(unittest.TestCase):
"""Tests for cowrie/commands/base64.py"""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_base64_command_001(self) -> None:
self.proto.lineReceived(b"echo cowrie | base64")
self.assertEqual(self.tr.value(), b"Y293cmllCg==\n" + PROMPT)
def test_base64_command_002(self) -> None:
self.proto.lineReceived(b"echo Y293cmllCg== | base64 -d")
self.assertEqual(self.tr.value(), b"cowrie\n" + PROMPT)
def setUp(self) -> None:
self.proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
self.tr = FakeTransport("", "31337")
self.proto.makeConnection(self.tr)
self.tr.clear()
class ShellFileCommandsTests(unittest.TestCase):
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
cpuinfo = proto.fs.file_contents("/proc/cpuinfo")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_cat_output(self) -> None:
self.proto.lineReceived(b"cat /proc/cpuinfo\n")
self.assertEqual(self.tr.value(), self.cpuinfo + PROMPT)
def test_grep_output(self) -> None:
lines = [
line.strip() for line in self.cpuinfo.splitlines()
if b"cpu" in line
]
lines.append(b"")
self.proto.lineReceived(b"grep cpu /proc/cpuinfo\n")
self.assertEqual(self.tr.value(), b"\n".join(lines) + PROMPT)
def test_tail_output(self) -> None:
lines = [line.strip() for line in self.cpuinfo.splitlines()][-10:]
lines.append(b"")
self.proto.lineReceived(b"tail -n 10 /proc/cpuinfo\n")
self.assertEqual(self.tr.value(), b"\n".join(lines) + PROMPT)
def test_head_output(self) -> None:
lines = [line.strip() for line in self.cpuinfo.splitlines()][:10]
lines.append(b"")
self.proto.lineReceived(b"head -n 10 /proc/cpuinfo\n")
self.assertEqual(self.tr.value(), b"\n".join(lines) + PROMPT)
def test_rm_output(self) -> None:
self.proto.lineReceived(b"rm /usr/bin/gcc\n")
self.assertEqual(self.tr.value(), PROMPT)
def test_rm_error_output(self) -> None: # TODO: quotes?..
self.proto.lineReceived(f"rm {NONEXISTEN_FILE:s}\n".encode())
self.assertEqual(
self.tr.value(),
f"rm: cannot remove `{NONEXISTEN_FILE:s}': No such file or directory\n"
.encode() + PROMPT)
def test_cp_output(self) -> None:
self.proto.lineReceived(b"cp /usr/bin/gcc /tmp\n")
self.assertEqual(self.tr.value(), PROMPT)
def test_cp_error_output(self) -> None: # TODO: quotes?..
self.proto.lineReceived(f"cp {NONEXISTEN_FILE:s} /tmp\n".encode())
self.assertEqual(
self.tr.value(),
f"cp: cannot stat `{NONEXISTEN_FILE:s}': No such file or directory\n"
.encode() + PROMPT)
def test_mv_output(self) -> None:
self.proto.lineReceived(b"mv /usr/bin/awk /tmp\n")
self.assertEqual(self.tr.value(), PROMPT)
def test_mv_error_output(self) -> None: # TODO: quotes?..
self.proto.lineReceived(f"mv {NONEXISTEN_FILE:s} /tmp\n".encode())
self.assertEqual(
self.tr.value(),
f"mv: cannot stat `{NONEXISTEN_FILE:s}': No such file or directory\n"
.encode() + PROMPT)
def test_mkdir_output(self) -> None:
path = "/tmp/hello"
self.proto.lineReceived(f"mkdir {path:s}\n".encode())
self.assertEqual(self.tr.value(), PROMPT)
self.assertTrue(self.proto.fs.exists(path))
self.assertTrue(self.proto.fs.isdir(path))
def test_mkdir_error_output(self) -> None: # TODO: quotes?..
path = "/etc"
self.proto.lineReceived(f"mkdir {path:s}\n".encode())
self.assertEqual(
self.tr.value(),
f"mkdir: cannot create directory `{path:s}': File exists\n".encode(
) + PROMPT)
def test_rmdir_output(self) -> None:
path = "/tmp/bye"
self.proto.lineReceived(f"mkdir {path:s}\n".encode())
self.tr.clear()
self.proto.lineReceived(f"rmdir {path:s}\n".encode())
self.assertEqual(self.tr.value(), PROMPT)
self.assertFalse(self.proto.fs.exists(path))
def test_rmdir_error_output(self) -> None: # TODO: quotes?..
self.proto.lineReceived(f"rmdir {NONEXISTEN_FILE:s}\n".encode())
self.assertEqual(
self.tr.value(),
f"rmdir: failed to remove `{NONEXISTEN_FILE:s}': No such file or directory\n"
.encode() + PROMPT)
def test_pwd_output(self) -> None:
self.proto.lineReceived(b"pwd\n")
self.assertEqual(self.tr.value(),
self.proto.cwd.encode() + b"\n" + PROMPT)
def test_touch_output(self) -> None:
path = "/tmp/test.txt"
self.proto.lineReceived(f"touch {path:s}\n".encode())
self.assertEqual(self.tr.value(), PROMPT)
self.assertTrue(self.proto.fs.exists(path))
class ShellChmodCommandTests(unittest.TestCase):
"""Test for cowrie/commands/chmod.py."""
proto = HoneyPotInteractiveProtocol(FakeAvatar(FakeServer()))
tr = FakeTransport("", "31337")
@classmethod
def setUpClass(cls) -> None:
cls.proto.makeConnection(cls.tr)
@classmethod
def tearDownClass(cls) -> None:
cls.proto.connectionLost("tearDown From Unit Test")
def setUp(self) -> None:
self.tr.clear()
def test_chmod_command_001(self) -> None:
self.proto.lineReceived(b"chmod")
self.assertEqual(
self.tr.value(),
b"chmod: missing operand\n" + TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_002(self) -> None:
self.proto.lineReceived(b"chmod -x")
self.assertEqual(
self.tr.value(),
b"chmod: missing operand\n" + TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_003(self) -> None:
self.proto.lineReceived(b"chmod +x")
self.assertEqual(
self.tr.value(),
b"chmod: missing operand after \xe2\x80\x98+x\xe2\x80\x99\n" +
TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_004(self) -> None:
self.proto.lineReceived(b"chmod -A")
self.assertEqual(
self.tr.value(),
b"chmod: invalid option -- 'A'\n" + TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_005(self) -> None:
self.proto.lineReceived(b"chmod --A")
self.assertEqual(
self.tr.value(), b"chmod: unrecognized option '--A'\n" +
TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_006(self) -> None:
self.proto.lineReceived(b"chmod -x abcd")
self.assertEqual(
self.tr.value(),
b"chmod: cannot access 'abcd': No such file or directory\n" +
PROMPT)
def test_chmod_command_007(self) -> None:
self.proto.lineReceived(b"chmod abcd efgh")
self.assertEqual(
self.tr.value(),
b"chmod: invalid mode: \xe2\x80\x98abcd\xe2\x80\x99\n" +
TRY_CHMOD_HELP_MSG + PROMPT)
def test_chmod_command_008(self) -> None:
self.proto.lineReceived(b"chmod +x .ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_009(self) -> None:
self.proto.lineReceived(b"chmod -R +x .ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_010(self) -> None:
self.proto.lineReceived(b"chmod +x /root/.ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_011(self) -> None:
self.proto.lineReceived(b"chmod +x ~/.ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_012(self) -> None:
self.proto.lineReceived(b"chmod a+x .ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_013(self) -> None:
self.proto.lineReceived(b"chmod ug+x .ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_014(self) -> None:
self.proto.lineReceived(b"chmod 777 .ssh")
self.assertEqual(self.tr.value(), PROMPT)
def test_chmod_command_015(self) -> None:
self.proto.lineReceived(b"chmod 0755 .ssh")
self.assertEqual(self.tr.value(), PROMPT)