Adding execute test for AsusWRT Lan RCE module (threat9#526)

* Adding execute test for AsusWRT Lan RCE exploit * Trigger Travis
devkw · Oct 2, 2018 · dd9b4f2 · dd9b4f2
1 parent 91be37a
commit dd9b4f2
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/exploits/routers/asus/test_asuswrt_lan_rce.py b/tests/exploits/routers/asus/test_asuswrt_lan_rce.py
@@ -1,3 +1,4 @@
+import re
 from unittest import mock
 from routersploit.modules.exploits.routers.asus.asuswrt_lan_rce import Exploit
 
@@ -22,3 +23,14 @@ def test_check_success(mocked_shell, target):
 
     assert exploit.check() is None
     assert exploit.run() is None
+
+
+def test_execute(udp_target):
+    infosvr_mock = udp_target.get_command_mock(re.compile(b"^\x0c\x15\x33\x00.{508}$"))  # 512 bytes in total
+    infosvr_mock.return_value = b"\x0c\x163\x00q6b3\x181\xbfJ\xd5\x08)\x00e1f419fdd8adf235c0948c8f3095e34878369301\n\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00Asus4021432\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00255.255.255.0\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00RT-AC68U\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x003.0.0.4.380\x00\x00\x00\x00\x00\x00\x181\xbfJ\xd5\x08\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x82\x80Q\x00\x01\x02\x1f\x92\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x01\xbb\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
+
+    exploit = Exploit()
+    exploit.target = udp_target.host
+    exploit.infosvr_port = udp_target.port
+
+    assert exploit.execute("echo e1f419fdd8adf235c0948c8f3095e34878369301") == "e1f419fdd8adf235c0948c8f3095e34878369301\n"