Added unit tests for isInFileJail function from #1066

fortra · Apr 26, 2021 · fb2a6fb · fb2a6fb
1 parent d187315
commit fb2a6fb
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/tests/SMB_RPC/test_smbserver.py b/tests/SMB_RPC/test_smbserver.py
@@ -19,15 +19,26 @@
 
 from six import StringIO, BytesIO, b
 
-from impacket.smbserver import SimpleSMBServer
+from impacket.smbserver import isInFileJail, SimpleSMBServer
 from impacket.smbconnection import SMBConnection, SessionError, compute_lmhash, compute_nthash
 
 
 class SMBServerUnitTests(unittest.TestCase):
     """Unit tests for the SMBServer
     """
 
-    pass
+    def test_isInFileJail(self):
+        """Test validation of common prefix path.
+        """
+        jail_path = "/tmp/jail_path"
+        self.assertTrue(isInFileJail(jail_path, "filename"))
+        self.assertTrue(isInFileJail(jail_path, "./filename"))
+        self.assertTrue(isInFileJail(jail_path, "../jail_path/filename"))
+
+        self.assertFalse(isInFileJail(jail_path, "/filename"))
+        self.assertFalse(isInFileJail(jail_path, "/tmp/filename"))
+        self.assertFalse(isInFileJail(jail_path, "../filename"))
+        self.assertFalse(isInFileJail(jail_path, "../../filename"))
 
 
 class SimpleSMBServerFuncTests(unittest.TestCase):