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

sticker_convert/auth/auth_viber.py

@@ -0,0 +1,250 @@
+#!/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