simonxiao86 · November 22, 2023 01:47 · Nov 3, 2023 · Nov 3, 2023 · Nov 3, 2023
diff --git a/locked_cookie_test.py b/locked_cookie_test.py
@@ -21,15 +21,15 @@
 
 rstrtmgr = windll.LoadLibrary("Rstrtmgr")
 
-session_handle = DWORD(0)
-session_flags = DWORD(0)
-session_key = (WCHAR * 256)()
-
 @WINFUNCTYPE(None, UINT)
 def callback(percent_complete: UINT) -> None:
     print(f"Unlocking file status: {percent_complete}% done")
 
 def unlock_cookies():
+    session_handle = DWORD(0)
+    session_flags = DWORD(0)
+    session_key = (WCHAR * 256)()
+
     result = DWORD(rstrtmgr.RmStartSession(byref(session_handle), session_flags, session_key)).value
 
     if result != ERROR_SUCCESS:

diff --git a/locked_cookie_test.py b/locked_cookie_test.py
@@ -25,6 +25,10 @@
 session_flags = DWORD(0)
 session_key = (WCHAR * 256)()
 
+@WINFUNCTYPE(None, UINT)
+def callback(percent_complete: UINT) -> None:
+    print(f"Unlocking file status: {percent_complete}% done")
+
 def unlock_cookies():
     result = DWORD(rstrtmgr.RmStartSession(byref(session_handle), session_flags, session_key)).value
 
@@ -46,14 +50,13 @@ def unlock_cookies():
         if result not in (ERROR_SUCCESS, ERROR_MORE_DATA):
             raise RuntimeError(f"RmGetList returned non-successful result: {result}")
 
-        @WINFUNCTYPE(None, UINT)
-        def callback(percent_complete: UINT) -> None:
-            print(f"Unlocking file status: {percent_complete}% done")
-
-        result = DWORD(rstrtmgr.RmShutdown(session_handle, RmForceShutdown, callback)).value
+        if proc_info_needed.value:
+            result = DWORD(rstrtmgr.RmShutdown(session_handle, RmForceShutdown, callback)).value
 
-        if result != ERROR_SUCCESS:
-            raise RuntimeError(f"RmShutdown returned non-successful result: {result}")
+            if result != ERROR_SUCCESS:
+                raise RuntimeError(f"RmShutdown returned non-successful result: {result}")
+        else:
+            print("File is not locked")
     finally:
         result = DWORD(rstrtmgr.RmEndSession(session_handle)).value
 
@@ -64,7 +67,7 @@ def callback(percent_complete: UINT) -> None:
 # Use backoff here since there is a race condition between unlocking the file and reading it.
 # Technically we're killing a process within chrome that holds the lock. Chrome can/will restart it,
 # .. so we have to fetch cookies before it re-locks the file. Generally on my system we get them
-# .... though one time we didn't.
+# .... though one time we didn't. I think it has to do with opening a new tab.. idk. Maybe not necessary?
 @backoff.on_exception(backoff.constant, PermissionError, max_tries=5)
 def fetch_cookies():
     unlock_cookies()

diff --git a/locked_cookie_test.py b/locked_cookie_test.py
@@ -0,0 +1,74 @@
+'''
+Proof of concept way to get cookies from chrome on Windows.. even when they're locked.
+Does not require admin rights.
+
+Includes a pure-python version of release_file_lock from:
+https://github.com/thewh1teagle/rookie/blob/02995bbbb692f775e12368e7fb2b728775c88ddd/rookie-rs/src/winapi.rs#L63
+
+(C) - MIT License 2023 - Charles Machalow
+'''
+import os
+from ctypes import windll, byref, create_unicode_buffer, pointer, WINFUNCTYPE
+from ctypes.wintypes import DWORD, WCHAR, UINT
+import browser_cookie3 # pip install browser-cookie3
+import backoff # pip install backoff
+
+ERROR_SUCCESS = 0
+ERROR_MORE_DATA  = 234
+RmForceShutdown = 1
+
+cookies_path = os.path.expandvars(r"%LOCALAPPDATA%\Google\Chrome\User Data\Default\Network\Cookies")
+
+rstrtmgr = windll.LoadLibrary("Rstrtmgr")
+
+session_handle = DWORD(0)
+session_flags = DWORD(0)
+session_key = (WCHAR * 256)()
+
+def unlock_cookies():
+    result = DWORD(rstrtmgr.RmStartSession(byref(session_handle), session_flags, session_key)).value
+
+    if result != ERROR_SUCCESS:
+        raise RuntimeError(f"RmStartSession returned non-zero result: {result}")
+
+    try:
+        result = DWORD(rstrtmgr.RmRegisterResources(session_handle, 1, byref(pointer(create_unicode_buffer(cookies_path))), 0, None, 0, None)).value
+
+        if result != ERROR_SUCCESS:
+            raise RuntimeError(f"RmRegisterResources returned non-zero result: {result}")
+
+        proc_info_needed = DWORD(0)
+        proc_info = DWORD(0)
+        reboot_reasons = DWORD(0)
+
+        result = DWORD(rstrtmgr.RmGetList(session_handle, byref(proc_info_needed), byref(proc_info), None, byref(reboot_reasons))).value
+
+        if result not in (ERROR_SUCCESS, ERROR_MORE_DATA):
+            raise RuntimeError(f"RmGetList returned non-successful result: {result}")
+
+        @WINFUNCTYPE(None, UINT)
+        def callback(percent_complete: UINT) -> None:
+            print(f"Unlocking file status: {percent_complete}% done")
+
+        result = DWORD(rstrtmgr.RmShutdown(session_handle, RmForceShutdown, callback)).value
+
+        if result != ERROR_SUCCESS:
+            raise RuntimeError(f"RmShutdown returned non-successful result: {result}")
+    finally:
+        result = DWORD(rstrtmgr.RmEndSession(session_handle)).value
+
+        if result != ERROR_SUCCESS:
+            raise RuntimeError(f"RmEndSession returned non-successful result: {result}")
+
+
+# Use backoff here since there is a race condition between unlocking the file and reading it.
+# Technically we're killing a process within chrome that holds the lock. Chrome can/will restart it,
+# .. so we have to fetch cookies before it re-locks the file. Generally on my system we get them
+# .... though one time we didn't.
+@backoff.on_exception(backoff.constant, PermissionError, max_tries=5)
+def fetch_cookies():
+    unlock_cookies()
+    return browser_cookie3.chrome(cookies_path)
+
+cookies = fetch_cookies()
+print(len(cookies))