sticker-convert 2.13.3.0__py3-none-any.whl → 2.17.0.0__py3-none-any.whl

sticker_convert/{utils/auth/get_viber_auth.py → auth/auth_viber.py}

@@ -1,235 +1,250 @@
-#!/usr/bin/env python3
-import importlib.util
-import os
-import platform
-import shutil
-import subprocess
-import time
-from functools import partial
-from getpass import getpass
-from pathlib import Path
-from typing import Callable, List, Optional, Tuple, cast
-
-from sticker_convert.utils.process import check_admin, find_pid_by_name, get_mem, killall
-
-MSG_NO_BIN = """Viber Desktop not detected.
-Download and install Viber Desktop,
-then login to Viber Desktop and try again."""
-MSG_NO_AUTH = """Viber Desktop installed,
-but viber_auth not found.
-Please login to Viber Desktop and try again."""
-MSG_SIP_ENABLED = """You need to disable SIP:
-1. Restart computer in Recovery mode
-2. Launch Terminal from the Utilities menu
-3. Run the command `csrutil disable`
-4. Restart your computer"""
-MSG_LAUNCH_FAIL = "Failed to launch Viber"
-MSG_PERMISSION_ERROR = "Failed to read Viber process memory"
-
-
-class GetViberAuth:
-    def __init__(self, cb_ask_str: Callable[..., str] = input) -> None:
-        self.cb_ask_str = cb_ask_str
-
-    def relaunch_viber(self, viber_bin_path: str) -> Optional[int]:
-        killed = killall("viber")
-        if killed:
-            time.sleep(5)
-
-        if platform.system() == "Darwin":
-            cmd = ["open", "-n", viber_bin_path]
-        else:
-            cmd = [viber_bin_path]
-        subprocess.Popen(cmd)
-        time.sleep(10)
-
-        return find_pid_by_name("viber")
-
-    def get_auth_by_pme(
-        self, viber_bin_path: str, relaunch: bool = True
-    ) -> Tuple[Optional[str], str]:
-        from PyMemoryEditor import OpenProcess  # type: ignore
-
-        member_id = None
-        m_token = None
-        m_ts = None
-
-        if relaunch:
-            viber_pid = self.relaunch_viber(viber_bin_path)
-        else:
-            viber_pid = find_pid_by_name("viber")
-        if viber_pid is None:
-            return None, MSG_LAUNCH_FAIL
-
-        try:
-            with OpenProcess(pid=int(viber_pid)) as process:
-                for address in process.search_by_value(str, 18, "X-Viber-Auth-Mid: "):  # type: ignore
-                    member_id_addr = cast(int, address) + 18
-                    member_id_bytes = process.read_process_memory(
-                        member_id_addr, bytes, 20
-                    )
-                    member_id_term = member_id_bytes.find(b"\x0d\x0a")
-                    if member_id_term == -1:
-                        continue
-                    member_id = member_id_bytes[:member_id_term].decode(
-                        encoding="ascii"
-                    )
-                    break
-                if member_id is None:
-                    return None, MSG_NO_AUTH
-
-                for address in process.search_by_value(str, 20, "X-Viber-Auth-Token: "):  # type: ignore
-                    m_token_addr = cast(int, address) + 20
-                    m_token = process.read_process_memory(m_token_addr, str, 64)
-                    break
-                if m_token is None:
-                    return None, MSG_NO_AUTH
-
-                for address in process.search_by_value(  # type: ignore
-                    str, 24, "X-Viber-Auth-Timestamp: "
-                ):
-                    m_ts_addr = cast(int, address) + 24
-                    m_ts = process.read_process_memory(m_ts_addr, str, 13)
-                    break
-                if m_ts is None:
-                    return None, MSG_NO_AUTH
-        except PermissionError:
-            return None, MSG_PERMISSION_ERROR
-
-        viber_auth = f"member_id:{member_id};m_token:{m_token};m_ts:{m_ts}"
-        msg = "Got viber_auth successfully:\n"
-        msg += f"{viber_auth=}\n"
-
-        return viber_auth, msg
-
-    def get_auth_by_dump(
-        self, viber_bin_path: str, relaunch: bool = True
-    ) -> Tuple[Optional[str], str]:
-        member_id = None
-        m_token = None
-        m_ts = None
-
-        if platform.system() == "Darwin":
-            csrutil_status = subprocess.run(
-                ["csrutil", "status"], capture_output=True, text=True
-            ).stdout
-
-            if "enabled" in csrutil_status:
-                return None, MSG_SIP_ENABLED
-
-        if relaunch:
-            viber_pid = self.relaunch_viber(viber_bin_path)
-        else:
-            viber_pid = find_pid_by_name("viber")
-        if viber_pid is None:
-            return None, MSG_LAUNCH_FAIL
-
-        if self.cb_ask_str == input:
-            pw_func = getpass
-        else:
-            pw_func = partial(
-                self.cb_ask_str, initialvalue="", cli_show_initialvalue=False
-            )
-        s = get_mem(viber_pid, pw_func)
-
-        if s is None:
-            return None, "Failed to dump memory"
-
-        member_id_addr = s.find(b"X-Viber-Auth-Mid: ")
-        m_token_addr = s.find(b"X-Viber-Auth-Token: ")
-        m_ts_addr = s.find(b"X-Viber-Auth-Timestamp: ")
-
-        if member_id_addr == -1 or m_token_addr == -1 or m_ts_addr == -1:
-            return None, MSG_NO_AUTH
-
-        member_id_addr += 18
-        m_token_addr += 20
-        m_ts_addr += 24
-
-        member_id_bytes = s[member_id_addr : member_id_addr + 20]
-        member_id_term = member_id_bytes.find(b"\x0d\x0a")
-        if member_id_term == -1:
-            return None, MSG_NO_AUTH
-        member_id = member_id_bytes[:member_id_term].decode(encoding="ascii")
-
-        m_token = s[m_token_addr : m_token_addr + 64].decode(encoding="ascii")
-        m_ts = s[m_ts_addr : m_ts_addr + 13].decode(encoding="ascii")
-
-        viber_auth = f"member_id:{member_id};m_token:{m_token};m_ts:{m_ts}"
-        msg = "Got viber_auth successfully:\n"
-        msg += f"{viber_auth=}\n"
-
-        return viber_auth, msg
-
-    def get_viber_desktop(self) -> Optional[str]:
-        if platform.system() == "Windows":
-            viber_bin_path = os.path.expandvars("%localappdata%/Viber/Viber.exe")
-        elif platform.system() == "Darwin":
-            viber_bin_path = "/Applications/Viber.app"
-        else:
-            if os.path.isfile("/opt/viber/Viber"):
-                viber_bin_path = "/opt/viber/Viber"
-            else:
-                viber_which = shutil.which("Viber")
-                if viber_which is None:
-                    viber_which = shutil.which("viber")
-                if viber_which is None:
-                    viber_which = shutil.which("viber.AppImage")
-                if viber_which is None:
-                    viber_bin_path = "viber"
-                else:
-                    viber_bin_path = viber_which
-
-        if Path(viber_bin_path).is_file() or Path(viber_bin_path).is_dir():
-            return viber_bin_path
-
-        return None
-
-    def get_cred(
-        self,
-        viber_bin_path: Optional[str] = None,
-    ) -> Tuple[Optional[str], str]:
-        if not viber_bin_path:
-            viber_bin_path = self.get_viber_desktop()
-
-        if not viber_bin_path:
-            return None, MSG_NO_BIN
-
-        # get_auth_by_dump()
-        # + Fast
-        # - Requires admin
-
-        # get_auth_by_pme()
-        # + No admin (If have permission to read process)
-        # - Slow
-        # - Cannot run on macOS
-
-        # If admin, prefer get_auth_by_dump() over get_auth_by_pme(), vice versa
-        methods: List[Callable[[str, bool], Tuple[Optional[str], str]]] = []
-        relaunch = True
-        viber_auth = None
-        msg = ""
-
-        pme_present = importlib.util.find_spec("PyMemoryEditor") is not None
-        if platform.system() == "Darwin":
-            methods.append(self.get_auth_by_dump)
-        elif platform.system() == "Windows":
-            methods.append(self.get_auth_by_dump)
-            if pme_present:
-                methods.append(self.get_auth_by_pme)
-        else:
-            if not os.path.isfile("/.dockerenv"):
-                methods.append(self.get_auth_by_dump)
-            if pme_present:
-                methods.append(self.get_auth_by_pme)
-            if check_admin() is False:
-                methods.reverse()
-
-        for method in methods:
-            viber_auth, msg = method(viber_bin_path, relaunch)
-            relaunch = False
-            if viber_auth is not None:
-                break
-
-        killall("viber")
-        return viber_auth, msg
+#!/usr/bin/env python3
+import importlib.util
+import os
+import platform
+import shutil
+import subprocess
+import time
+from pathlib import Path
+from typing import Any, Callable, List, Optional, Tuple, cast
+
+from sticker_convert.auth.auth_base import AuthBase
+from sticker_convert.utils.process import check_admin, find_pid_by_name, get_mem, killall
+from sticker_convert.utils.translate import I
+
+
+class AuthViber(AuthBase):
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        self.MSG_NO_BIN = I(
+            "Viber Desktop not detected.\n"
+            "Download and install Viber Desktop,\n"
+            "then login to Viber Desktop and try again."
+        )
+        self.MSG_NO_AUTH = I(
+            "Viber Desktop installed,\n"
+            "but viber_auth not found.\n"
+            "Please login to Viber Desktop and try again."
+        )
+        self.MSG_SIP_ENABLED = I(
+            "You need to disable SIP:\n"
+            "1. Restart computer in Recovery mode\n"
+            "2. Launch Terminal from the Utilities menu\n"
+            "3. Run the command `csrutil disable`\n"
+            "4. Restart your computer"
+        )
+        self.MSG_LAUNCH_FAIL = I("Failed to launch Viber")
+        self.MSG_PERMISSION_ERROR = I("Failed to read Viber process memory")
+
+        super().__init__(*args, **kwargs)
+
+    def relaunch_viber(self, viber_bin_path: str) -> Optional[int]:
+        killed = killall("viber")
+        if killed:
+            time.sleep(5)
+
+        if platform.system() == "Darwin":
+            cmd = ["open", "-n", viber_bin_path]
+        else:
+            cmd = [viber_bin_path]
+        subprocess.Popen(cmd)
+        time.sleep(10)
+
+        return find_pid_by_name("viber")
+
+    def get_auth_by_pme(
+        self, viber_bin_path: str, relaunch: bool = True
+    ) -> Tuple[Optional[str], str]:
+        from PyMemoryEditor import OpenProcess  # type: ignore
+
+        member_id = None
+        m_token = None
+        m_ts = None
+
+        if relaunch:
+            viber_pid = self.relaunch_viber(viber_bin_path)
+        else:
+            viber_pid = find_pid_by_name("viber")
+        if viber_pid is None:
+            return None, self.MSG_LAUNCH_FAIL
+
+        try:
+            with OpenProcess(pid=int(viber_pid)) as process:
+                for address in process.search_by_value(str, 18, "X-Viber-Auth-Mid: "):  # type: ignore
+                    member_id_addr = cast(int, address) + 18
+                    member_id_bytes = process.read_process_memory(
+                        member_id_addr, bytes, 20
+                    )
+                    member_id_term = member_id_bytes.find(b"\x0d\x0a")
+                    if member_id_term == -1:
+                        continue
+                    member_id = member_id_bytes[:member_id_term].decode(
+                        encoding="ascii"
+                    )
+                    break
+                if member_id is None:
+                    return None, self.MSG_NO_AUTH
+
+                for address in process.search_by_value(str, 20, "X-Viber-Auth-Token: "):  # type: ignore
+                    m_token_addr = cast(int, address) + 20
+                    m_token = process.read_process_memory(m_token_addr, str, 64)
+                    break
+                if m_token is None:
+                    return None, self.MSG_NO_AUTH
+
+                for address in process.search_by_value(  # type: ignore
+                    str, 24, "X-Viber-Auth-Timestamp: "
+                ):
+                    m_ts_addr = cast(int, address) + 24
+                    m_ts = process.read_process_memory(m_ts_addr, str, 13)
+                    break
+                if m_ts is None:
+                    return None, self.MSG_NO_AUTH
+        except PermissionError:
+            return None, self.MSG_PERMISSION_ERROR
+
+        viber_auth = f"member_id:{member_id};m_token:{m_token};m_ts:{m_ts}"
+        msg = I("Got viber_auth successfully:\n")
+        msg += f"{viber_auth=}\n"
+
+        return viber_auth, msg
+
+    def get_auth_by_dump(
+        self, viber_bin_path: str, relaunch: bool = True
+    ) -> Tuple[Optional[str], str]:
+        member_id = None
+        m_token = None
+        m_ts = None
+
+        if platform.system() == "Darwin":
+            csrutil_status = subprocess.run(
+                ["csrutil", "status"], capture_output=True, text=True
+            ).stdout
+
+            if "enabled" in csrutil_status:
+                return None, self.MSG_SIP_ENABLED
+
+        if relaunch:
+            viber_pid = self.relaunch_viber(viber_bin_path)
+        else:
+            viber_pid = find_pid_by_name("viber")
+        if viber_pid is None:
+            return None, self.MSG_LAUNCH_FAIL
+
+        def pw_func(msg: str) -> str:
+            return self.cb.put(
+                (
+                    "ask_str",
+                    None,
+                    {
+                        "message": msg,
+                        "password": True,
+                    },
+                )
+            )
+
+        s = get_mem(viber_pid, pw_func)
+
+        if s is None:
+            return None, I("Failed to dump memory")
+
+        member_id_addr = s.find(b"X-Viber-Auth-Mid: ")
+        m_token_addr = s.find(b"X-Viber-Auth-Token: ")
+        m_ts_addr = s.find(b"X-Viber-Auth-Timestamp: ")
+
+        if member_id_addr == -1 or m_token_addr == -1 or m_ts_addr == -1:
+            return None, self.MSG_NO_AUTH
+
+        member_id_addr += 18
+        m_token_addr += 20
+        m_ts_addr += 24
+
+        member_id_bytes = s[member_id_addr : member_id_addr + 20]
+        member_id_term = member_id_bytes.find(b"\x0d\x0a")
+        if member_id_term == -1:
+            return None, self.MSG_NO_AUTH
+        member_id = member_id_bytes[:member_id_term].decode(encoding="ascii")
+
+        m_token = s[m_token_addr : m_token_addr + 64].decode(encoding="ascii")
+        m_ts = s[m_ts_addr : m_ts_addr + 13].decode(encoding="ascii")
+
+        viber_auth = f"member_id:{member_id};m_token:{m_token};m_ts:{m_ts}"
+        msg = I("Got viber_auth successfully:\n")
+        msg += f"{viber_auth=}\n"
+
+        return viber_auth, msg
+
+    def get_viber_desktop(self) -> Optional[str]:
+        if platform.system() == "Windows":
+            viber_bin_path = os.path.expandvars("%localappdata%/Viber/Viber.exe")
+        elif platform.system() == "Darwin":
+            viber_bin_path = "/Applications/Viber.app"
+        else:
+            if os.path.isfile("/opt/viber/Viber"):
+                viber_bin_path = "/opt/viber/Viber"
+            else:
+                viber_which = shutil.which("Viber")
+                if viber_which is None:
+                    viber_which = shutil.which("viber")
+                if viber_which is None:
+                    viber_which = shutil.which("viber.AppImage")
+                if viber_which is None:
+                    viber_bin_path = "viber"
+                else:
+                    viber_bin_path = viber_which
+
+        if Path(viber_bin_path).is_file() or Path(viber_bin_path).is_dir():
+            return viber_bin_path
+
+        return None
+
+    def get_cred(
+        self,
+        viber_bin_path: Optional[str] = None,
+    ) -> Tuple[Optional[str], str]:
+        msg = I("Getting Viber credentials...\n(This may take a minute)")
+        self.cb.put(("msg_dynamic", (msg,), None))
+        if not viber_bin_path:
+            viber_bin_path = self.get_viber_desktop()
+
+        if not viber_bin_path:
+            return None, self.MSG_NO_BIN
+
+        # get_auth_by_dump()
+        # + Fast
+        # - Requires admin
+
+        # get_auth_by_pme()
+        # + No admin (If have permission to read process)
+        # - Slow
+        # - Cannot run on macOS
+
+        # If admin, prefer get_auth_by_dump() over get_auth_by_pme(), vice versa
+        methods: List[Callable[[str, bool], Tuple[Optional[str], str]]] = []
+        relaunch = True
+        viber_auth = None
+        msg = ""
+
+        pme_present = importlib.util.find_spec("PyMemoryEditor") is not None
+        if platform.system() == "Darwin":
+            methods.append(self.get_auth_by_dump)
+        elif platform.system() == "Windows":
+            methods.append(self.get_auth_by_dump)
+            if pme_present:
+                methods.append(self.get_auth_by_pme)
+        else:
+            if not os.path.isfile("/.dockerenv"):
+                methods.append(self.get_auth_by_dump)
+            if pme_present:
+                methods.append(self.get_auth_by_pme)
+            if check_admin() is False:
+                methods.reverse()
+
+        for method in methods:
+            viber_auth, msg = method(viber_bin_path, relaunch)
+            relaunch = False
+            if viber_auth is not None:
+                break
+
+        killall("viber")
+        self.cb.put(("msg_dynamic", (None,), None))
+        return viber_auth, msg