sticker-convert 2.9.0__tar.gz → 2.9.1__tar.gz

{sticker_convert-2.9.0/src/sticker_convert.egg-info → sticker_convert-2.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sticker-convert
-Version: 2.9.0
+Version: 2.9.1
 Summary: Convert (animated) stickers to/from WhatsApp, Telegram, Signal, Line, Kakao, Viber, iMessage. Written in Python.
 Author-email: laggykiller <chaudominic2@gmail.com>
 Maintainer-email: laggykiller <chaudominic2@gmail.com>
@@ -376,11 +376,11 @@ Requires-Dist: mergedeep~=1.3.4
 Requires-Dist: numpy>=1.22.4
 Requires-Dist: Pillow~=10.3.0
 Requires-Dist: pyoxipng~=9.0.0
-Requires-Dist: python-telegram-bot~=21.2
+Requires-Dist: python-telegram-bot~=21.3
 Requires-Dist: psutil~=5.9.8
 Requires-Dist: PyMemoryEditor~=1.5.22
 Requires-Dist: requests~=2.32.3
-Requires-Dist: rlottie_python~=1.3.5
+Requires-Dist: rlottie_python~=1.3.6
 Requires-Dist: signalstickers-client-fork-laggykiller~=3.3.0.post2
 Requires-Dist: sqlcipher3-wheels~=0.5.2.post1
 Requires-Dist: tqdm~=4.66.4

{sticker_convert-2.9.0 → sticker_convert-2.9.1}/requirements.txt

@@ -10,11 +10,11 @@ mergedeep~=1.3.4
 numpy>=1.22.4
 Pillow~=10.3.0
 pyoxipng~=9.0.0
-python-telegram-bot~=21.2
+python-telegram-bot~=21.3
 psutil~=5.9.8
 PyMemoryEditor~=1.5.22
 requests~=2.32.3
-rlottie_python~=1.3.5
+rlottie_python~=1.3.6
 signalstickers-client-fork-laggykiller~=3.3.0.post2
 sqlcipher3-wheels~=0.5.2.post1
 tqdm~=4.66.4

{sticker_convert-2.9.0 → sticker_convert-2.9.1}/src/sticker_convert/gui_components/windows/viber_get_auth_window.py

@@ -51,11 +51,17 @@ class ViberGetAuthWindow(BaseWindow):
                 justify="left",
                 anchor="w",
             )
+        else:
+            self.explanation_lbl2 = Label(
+                self.frame_info,
+                text="You may be asked for admin password.",
+                justify="left",
+                anchor="w",
+            )
 
         self.explanation_lbl0.grid(column=0, row=0, sticky="w", padx=3, pady=3)
         self.explanation_lbl1.grid(column=0, row=1, sticky="w", padx=3, pady=3)
-        if self.explanation_lbl2 is not None:
-            self.explanation_lbl2.grid(column=0, row=2, sticky="w", padx=3, pady=3)
+        self.explanation_lbl2.grid(column=0, row=2, sticky="w", padx=3, pady=3)
 
         # Start button frame
         self.launch_btn = Button(

sticker_convert-2.9.1/src/sticker_convert/resources/memdump_windows.ps1

@@ -0,0 +1,8 @@
+param($processId)
+$dumpFilePath = "$Env:TEMP\memdump.bin.$processId"
+
+Powershell -c rundll32.exe C:\Windows\System32\comsvcs.dll, MiniDump $processId $dumpFilePath full
+
+icacls $dumpFilePath /remove "Everyone"
+icacls $dumpFilePath /grant "Everyone:(r)"
+icacls $dumpFilePath /grant "Everyone:(w)"

sticker_convert-2.9.1/src/sticker_convert/utils/auth/get_viber_auth.py

@@ -0,0 +1,441 @@
+#!/usr/bin/env python3
+import importlib.util
+import os
+import platform
+import shutil
+import signal
+import subprocess
+import time
+from getpass import getpass
+from pathlib import Path
+from typing import Callable, List, Optional, Tuple, cast
+
+from sticker_convert.definitions import ROOT_DIR
+
+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_NO_PGREP = "pgrep command or psutil python package is necessary"
+MSG_LAUNCH_FAIL = "Failed to launch Viber"
+MSG_PERMISSION_ERROR = "Failed to read Viber process memory"
+
+
+def check_admin_windows() -> bool:
+    username = os.getenv("username")
+    if username is None:
+        return False
+
+    s = subprocess.run(
+        ["net", "user", username],
+        capture_output=True,
+        text=True,
+    ).stdout
+
+    return True if "*Administrators" in s else False
+
+
+def check_admin_linux() -> bool:
+    s = subprocess.run(
+        ["sudo", "-l"],
+        capture_output=True,
+        text=True,
+    ).stdout
+
+    return True if "may run the following commands" in s else False
+
+
+def killall(name: str) -> bool:
+    result = False
+
+    while True:
+        pid = find_pid_by_name(name)
+        if pid is not None:
+            os.kill(pid, signal.SIGTERM)
+            result = True
+        else:
+            break
+
+    return result
+
+
+def find_pid_by_name(name: str) -> Optional[int]:
+    if platform.system() == "Windows":
+        s = subprocess.run(
+            ["powershell", "-c", "Get-Process", f"*{name}*"],
+            capture_output=True,
+            text=True,
+        ).stdout
+
+        for line in s.split("\n"):
+            if name in line.lower():
+                info = name.split()
+                pid = info[5]
+                if pid.isnumeric():
+                    return int(pid)
+
+        return None
+    else:
+        if platform.system() == "Darwin":
+            pattern = "Viber"
+        else:
+            pattern = ".*[Vv]iber.*"
+        pid = (
+            subprocess.run(["pgrep", pattern], capture_output=True, text=True)
+            .stdout.split("\n")[0]
+            .strip()
+        )
+        if pid == "" or pid.isnumeric() is False:
+            return None
+        else:
+            return int(pid)
+
+
+if importlib.util.find_spec("psutil"):
+    import psutil
+
+    def killall(name: str) -> bool:
+        result = False
+
+        for proc in psutil.process_iter():  # type: ignore
+            if name in proc.name().lower():
+                proc.kill()
+                result = True
+
+        return result
+
+    def find_pid_by_name(name: str) -> Optional[int]:
+        for proc in psutil.process_iter():  # type: ignore
+            if name in proc.name().lower():
+                return proc.pid
+
+        return None
+
+
+class GetViberAuth:
+    def __init__(self, cb_ask_str: Callable[..., str] = input):
+        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_mem_windows(self, viber_pid: int) -> Tuple[Optional[bytes], str]:
+        from pathlib import WindowsPath
+
+        memdump_ps_path = str(WindowsPath(ROOT_DIR / "resources/memdump_windows.ps1"))
+        arglist = (
+            f'-NoProfile -ExecutionPolicy Bypass -File "{memdump_ps_path}" {viber_pid}'
+        )
+        dump_fpath = os.path.expandvars(f"%temp%/memdump.bin.{viber_pid}")
+
+        cmd = [
+            "powershell",
+            "-NoProfile",
+            "-ExecutionPolicy",
+            "Bypass",
+            "-Command",
+            f"Start-Process -Verb RunAs powershell -ArgumentList '{arglist}'",
+        ]
+
+        subprocess.run(cmd, capture_output=True, text=True)
+
+        while True:
+            try:
+                with open(dump_fpath, "rb") as f:
+                    s = f.read()
+                if len(s) != 0:
+                    break
+                time.sleep(1)
+            except (FileNotFoundError, PermissionError):
+                pass
+
+        while True:
+            try:
+                os.remove(dump_fpath)
+                break
+            except PermissionError:
+                pass
+
+        return s, ""
+
+    def get_mem_linux(self, viber_pid: int) -> Tuple[Optional[bytes], str]:
+        memdump_sh_path = (ROOT_DIR / "resources/memdump_linux.sh").as_posix()
+
+        s = subprocess.run(
+            [
+                memdump_sh_path,
+                str(viber_pid),
+            ],
+            capture_output=True,
+        ).stdout
+
+        if len(s) > 1000:
+            pass
+        elif shutil.which("pkexec") and os.getenv("DISPLAY"):
+            s = subprocess.run(
+                [
+                    "pkexec",
+                    memdump_sh_path,
+                    str(viber_pid),
+                ],
+                capture_output=True,
+            ).stdout
+        else:
+            prompt = "Enter sudo password: "
+            if self.cb_ask_str != input:
+                sudo_password = self.cb_ask_str(
+                    prompt, initialvalue="", cli_show_initialvalue=False
+                )
+            else:
+                sudo_password = getpass(prompt)
+            sudo_password_pipe = subprocess.Popen(
+                ("echo", sudo_password), stdout=subprocess.PIPE
+            )
+            s = subprocess.run(
+                [
+                    "sudo",
+                    "-S",
+                    memdump_sh_path,
+                    str(viber_pid),
+                ],
+                capture_output=True,
+                stdin=sudo_password_pipe.stdout,
+            ).stdout
+
+        return s, ""
+
+    def get_mem_darwin(self, viber_pid: int) -> Tuple[Optional[bytes], str]:
+        subprocess.run(
+            [
+                "lldb",
+                "--attach-pid",
+                str(viber_pid),
+                "-o",
+                "process save-core /tmp/viber.dmp",
+                "-o",
+                "quit",
+            ],
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+        )
+
+        with open("/tmp/viber.dmp", "rb") as f:
+            s = f.read()
+
+        os.remove("/tmp/viber.dmp")
+
+        return s, ""
+
+    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
+        elif (
+            platform.system() != "Windows"
+            and shutil.which("pgrep") is None
+            and importlib.find_loader("psutil") is None
+        ):
+            return None, MSG_NO_PGREP
+
+        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 platform.system() == "Windows":
+            s, msg = self.get_mem_windows(viber_pid)
+        elif platform.system() == "Darwin":
+            s, msg = self.get_mem_darwin(viber_pid)
+        else:
+            s, msg = self.get_mem_linux(viber_pid)
+
+        if s is None:
+            return None, msg
+
+        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)
+            if check_admin_windows() is False:
+                methods.reverse()
+        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_linux() 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

{sticker_convert-2.9.0 → sticker_convert-2.9.1}/src/sticker_convert/version.py

@@ -1,3 +1,3 @@
 #!/usr/bin/env python3
 
-__version__ = "2.9.0"
+__version__ = "2.9.1"

{sticker_convert-2.9.0 → sticker_convert-2.9.1/src/sticker_convert.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sticker-convert
-Version: 2.9.0
+Version: 2.9.1
 Summary: Convert (animated) stickers to/from WhatsApp, Telegram, Signal, Line, Kakao, Viber, iMessage. Written in Python.
 Author-email: laggykiller <chaudominic2@gmail.com>
 Maintainer-email: laggykiller <chaudominic2@gmail.com>
@@ -376,11 +376,11 @@ Requires-Dist: mergedeep~=1.3.4
 Requires-Dist: numpy>=1.22.4
 Requires-Dist: Pillow~=10.3.0
 Requires-Dist: pyoxipng~=9.0.0
-Requires-Dist: python-telegram-bot~=21.2
+Requires-Dist: python-telegram-bot~=21.3
 Requires-Dist: psutil~=5.9.8
 Requires-Dist: PyMemoryEditor~=1.5.22
 Requires-Dist: requests~=2.32.3
-Requires-Dist: rlottie_python~=1.3.5
+Requires-Dist: rlottie_python~=1.3.6
 Requires-Dist: signalstickers-client-fork-laggykiller~=3.3.0.post2
 Requires-Dist: sqlcipher3-wheels~=0.5.2.post1
 Requires-Dist: tqdm~=4.66.4

{sticker_convert-2.9.0 → sticker_convert-2.9.1}/src/sticker_convert.egg-info/SOURCES.txt

@@ -83,7 +83,8 @@ src/sticker_convert/resources/compression.json
 src/sticker_convert/resources/emoji.json
 src/sticker_convert/resources/help.json
 src/sticker_convert/resources/input.json
-src/sticker_convert/resources/memdump.sh
+src/sticker_convert/resources/memdump_linux.sh
+src/sticker_convert/resources/memdump_windows.ps1
 src/sticker_convert/resources/output.json
 src/sticker_convert/uploaders/__init__.py
 src/sticker_convert/uploaders/compress_wastickers.py

{sticker_convert-2.9.0 → sticker_convert-2.9.1}/src/sticker_convert.egg-info/requires.txt

@@ -10,11 +10,11 @@ mergedeep~=1.3.4
 numpy>=1.22.4
 Pillow~=10.3.0
 pyoxipng~=9.0.0
-python-telegram-bot~=21.2
+python-telegram-bot~=21.3
 psutil~=5.9.8
 PyMemoryEditor~=1.5.22
 requests~=2.32.3
-rlottie_python~=1.3.5
+rlottie_python~=1.3.6
 signalstickers-client-fork-laggykiller~=3.3.0.post2
 sqlcipher3-wheels~=0.5.2.post1
 tqdm~=4.66.4