Merge pull request #53 from fa0311/develop-v4

Develop v4

Author: fa0311

.vscode/launch.json

@@ -0,0 +1,17 @@
+{
+  // IntelliSense を使用して利用可能な属性を学べます。
+  // 既存の属性の説明をホバーして表示します。
+  // 詳細情報は次を確認してください: https://go.microsoft.com/fwlink/?linkid=830387
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "DMMGamePlayerFastLauncher",
+      "type": "python",
+      "request": "launch",
+      "program": "DMMGamePlayerFastLauncher.py",
+      "console": "integratedTerminal",
+      "justMyCode": true,
+      "args": ["umamusume", "--non-bypass-uac"]
+    }
+  ]
+}

DMMGamePlayerFastLauncher.py

@@ -13,6 +13,8 @@
 from urllib.parse import urlparse
 import win32security
 import sys
+import base64
+from Crypto.Cipher import AES
 
 
 class DgpSession:
@@ -65,20 +67,28 @@ def gen_rand_address(self):
         return address[:-1]
 
     def write(self):
+        aes_key = self.get_aes_key()
         for cookie_row in self.db.cursor().execute("select * from cookies"):
-            name = cookie_row[3]
-            value = self.session.cookies.get(name)
-            if value != None:
-                self.db.execute(
-                    f"update cookies set value = '{value}' where name = '{name}'"
-                )
+            value = self.session.cookies.get(cookie_row[3])
+            v10, nonce, _, _ = self.split_encrypted_data(cookie_row[5])
+            cipher = AES.new(aes_key, AES.MODE_GCM, nonce)
+            decrypt_data, mac = cipher.encrypt_and_digest(value.encode())
+            data = self.join_encrypted_data(v10, nonce, decrypt_data, mac)
+            self.db.execute(
+                "update cookies set encrypted_value = ? where name = ?",
+                (memoryview(data), cookie_row[3]),
+            )
         self.db.commit()
 
     def read(self):
+        aes_key = self.get_aes_key()
         for cookie_row in self.db.cursor().execute("select * from cookies"):
+            _, nonce, data, mac = self.split_encrypted_data(cookie_row[5])
+            cipher = AES.new(aes_key, AES.MODE_GCM, nonce)
+            value = cipher.decrypt_and_verify(data, mac).decode()
             cookie_data = {
                 "name": cookie_row[3],
-                "value": cookie_row[4],
+                "value": value,
                 "domain": cookie_row[1],
                 "path": cookie_row[6],
                 "secure": cookie_row[8],
@@ -149,6 +159,28 @@ def open(self):
         self.session = requests.session()
         self.cookies = self.session.cookies
 
+    def get_aes_key(self):
+        with open(self.DGP5_PATH + "\\Local State", "r") as f:
+            local_state = json.load(f)
+        encrypted_key = base64.b64decode(
+            local_state["os_crypt"]["encrypted_key"].encode()
+        )[5:]
+        key = win32crypt.CryptUnprotectData(encrypted_key, None, None, None, 0)[1]
+        return key
+
+    def split_encrypted_data(self, encrypted_data: bytes) -> bytes:
+        return (
+            encrypted_data[0:3],
+            encrypted_data[3:15],
+            encrypted_data[15:-16],
+            encrypted_data[-16:],
+        )
+
+    def join_encrypted_data(
+        self, v10: bytes, nonce: bytes, data: bytes, mac: bytes
+    ) -> bytes:
+        return v10 + nonce + data + mac
+
     def close(self):
         self.db.close()
 
@@ -327,6 +359,7 @@ def run_bypass_uac():
 argpar.add_argument("--https-proxy-uri", default=None)
 argpar.add_argument("--non-request-admin", action="store_true")
 argpar.add_argument("--non-bypass-uac", action="store_true")
+argpar.add_argument("--force-bypass-uac", action="store_true")
 argpar.add_argument("--schtasks-path", default="schtasks.exe")
 
 try:
@@ -414,7 +447,12 @@ def run_bypass_uac():
     dmm_args = response["data"]["execute_args"].split(" ")
     if arg.game_args is not None:
         dmm_args = dmm_args + arg.game_args.split(" ")
-    print(game_path)
+    if (
+        arg.force_bypass_uac
+        and not arg.non_bypass_uac
+        and response["data"]["is_administrator"]
+    ):
+        run_bypass_uac()
     start_time = time.time()
     process = process_manager.run([game_path] + dmm_args)
     for line in process.stdout:

docs/README-advance.md

@@ -29,12 +29,16 @@
 | --https-proxy-uri   |            | None         | String | None |            |
 | --non-request-admin |            | False        | Bool               | deprecated |
 | --non-bypass-uac    |            | False        | Bool               |            |
+| --force-bypass-uac  |            | False        | Bool               |            |
 | --schtasks-path     |            | schtasks.exe | String             |            |
 
 ```mermaid
     graph TD;
-        ゲームのプロセスを開始 --権限不備で起動しない--> non-bypass-uac;
-        ゲームのプロセスを開始 --Start/Error--> 続行;
+        ゲームが権限を要求 --> force-bypass-uac;
+        force-bypass-uac --true--> non-bypass-uac
+        force-bypass-uac --false--> 権限与えずゲームを開始
+        権限与えずゲームを開始 --起動しない--> non-bypass-uac;
+        権限与えずゲームを開始 --起動--> 続行;
         non-bypass-uac --false--> 最初のリクエスト;
         最初のリクエスト --no-->ゲームに権限与える
         最初のリクエスト --yes--> non-request-admin
@@ -43,15 +47,16 @@
         non-request-admin --false--> UAC;
         UAC --allow--> ゲームに権限与える
         UAC --disabled--> 続行
-        skip-exception --true--> 続行
+        skip-exception --true--> ゲームに権限与えにず続行
         skip-exception --false--> エラー
 ```
 
 **最初のリクエスト** - 最初のリクエストは、管理者権限を要求します  
 これは、タスクスケジューラーに自動的に権限を昇格させるプログラムを登録するために必要です
 
 特にこだわりが無ければ product_id 以外の引数は不要です  
-ゲームや環境によっては`game-path` `https-proxy-uri` `schtasks-path` などが必要です
+ゲームや環境によっては`game-path` `https-proxy-uri` `schtasks-path` などが必要です  
+更に高速化を求めるのであれば `--force-bypass-uac` などが必要ですが不安定になる可能性があります  
 
 ### game-path
 
@@ -112,7 +117,9 @@ Socks5
 
 ### non-bypass-uac
 
-この引数を使用すると UAC の自動昇格を行わなくなります
+この引数を使用すると UAC の自動昇格を行わなくなります  
+環境によってUACの自動昇格が行われなかったり起動速度を優先したい人はこれを付けてください  
+(開発者向けヒント; ビルド前の開発環境ではこの引数は必須です)
 
 指定していない場合はタスクスケジューラを使った権限の自動昇格を行います  
 タスクの詳細はこのコマンドで確認できます  
@@ -123,6 +130,15 @@ Socks5
 `Get-ScheduledTask | where TaskPath -eq "\Microsoft\Windows\DMMGamePlayerFastLauncher" | Unregister-ScheduledTask -Confirm:$false`
 または `.\tools\refresh.ps1`
 
+### force-bypass-uac
+
+ゲームが管理者権限を求めた際に必ずUACの自動昇格を行います  
+管理者権限でないと起動できないゲーム(non-bypass-uacを付けた際に必ずUACを要求するゲーム)はこれを付けると早く起動できます  
+ウマ娘やプリコネRなどの稀に要求するゲームは無い方が早いかもしれません  
+
+例:  
+`%AppData%\DMMGamePlayerFastLauncher\DMMGamePlayerFastLauncher.exe umamusume --force-bypass-uac`
+
 ### schtasks-path
 
 `schtasks.exe`のパスを指定します

requirements.txt

@@ -2,7 +2,7 @@
 ; SEE THE DOCUMENTATION FOR DETAILS ON CREATING INNO SETUP SCRIPT FILES!
 
 #define MyAppName "DMMGamePlayerFastLauncher"
-#define MyAppVersion "4.6.0"
+#define MyAppVersion "4.7.0"
 #define MyAppPublisher "yuki"
 #define MyAppURL "https://github.com/fa0311/DMMGamePlayerFastLauncher"
 #define MyAppExeName "DMMGamePlayerFastLauncher.exe"