hoelee/curl-impersonate
1
0
Fork 0
mirror of https://github.com/lwthiker/curl-impersonate.git synced 2025-08-09 05:09:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Run the LD_PRELOAD injection tests on Linux only

Browse Source 
Injecting libcurl-impersonate with LD_PRELOAD is supported on
Linux only. On Mac there is DYLD_INSERT_LIBRARIES but it
reuqires more work to be fully functional.
This commit is contained in:
lwthiker
2022-04-23 10:43:57 +03:00
parent 90012e78de
commit d0c5777420
1 changed files with 41 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
tests/test_impersonate.py
View File

@@ -1,6 +1,7 @@
import os import os
import io import io
import re import re
import sys
import logging import logging
import subprocess import subprocess
import tempfile import tempfile
@@ -145,7 +146,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "chrome98" "CURL_IMPERSONATE": "chrome98"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"chrome_98.0.4758.102_win10" "chrome_98.0.4758.102_win10"
), ),
( (
@@ -153,7 +154,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "chrome99" "CURL_IMPERSONATE": "chrome99"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"chrome_99.0.4844.51_win10" "chrome_99.0.4844.51_win10"
), ),
( (
@@ -161,7 +162,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "chrome99_android" "CURL_IMPERSONATE": "chrome99_android"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"chrome_99.0.4844.73_android12-pixel6" "chrome_99.0.4844.73_android12-pixel6"
), ),
( (
@@ -169,7 +170,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "edge98" "CURL_IMPERSONATE": "edge98"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"edge_98.0.1108.62_win10" "edge_98.0.1108.62_win10"
), ),
( (
@@ -177,7 +178,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "edge99" "CURL_IMPERSONATE": "edge99"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"edge_99.0.1150.30_win10" "edge_99.0.1150.30_win10"
), ),
( (
@@ -185,7 +186,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "safari15_3" "CURL_IMPERSONATE": "safari15_3"
}, },
"libcurl-impersonate-chrome.so", "libcurl-impersonate-chrome",
"safari_15.3_macos11.6.4" "safari_15.3_macos11.6.4"
), ),
( (
@@ -193,7 +194,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "ff91esr" "CURL_IMPERSONATE": "ff91esr"
}, },
"libcurl-impersonate-ff.so", "libcurl-impersonate-ff",
"firefox_91.6.0esr_win10" "firefox_91.6.0esr_win10"
), ),
( (
@@ -201,7 +202,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "ff95" "CURL_IMPERSONATE": "ff95"
}, },
"libcurl-impersonate-ff.so", "libcurl-impersonate-ff",
"firefox_95.0.2_win10" "firefox_95.0.2_win10"
), ),
( (
@@ -209,7 +210,7 @@ class TestImpersonation:
{ {
"CURL_IMPERSONATE": "ff98" "CURL_IMPERSONATE": "ff98"
}, },
"libcurl-impersonate-ff.so", "libcurl-impersonate-ff",
"firefox_98.0_win10" "firefox_98.0_win10"
) )
] ]
@@ -255,6 +256,12 @@ class TestImpersonation:
p.terminate() p.terminate()
p.wait(timeout=10) p.wait(timeout=10)
def _set_ld_preload(self, env_vars, lib):
if sys.platform.startswith("linux"):
env_vars["LD_PRELOAD"] = lib + ".so"
elif sys.platform.startswith("darwin"):
env_vars["DYLD_INSERT_LIBRARIES"] = lib + ".dylib"
def _run_curl(self, curl_binary, env_vars, extra_args, url, def _run_curl(self, curl_binary, env_vars, extra_args, url,
output="/dev/null"): output="/dev/null"):
env = os.environ.copy() env = os.environ.copy()
@@ -368,9 +375,15 @@ class TestImpersonation:
pytestconfig.getoption("install_dir"), "bin", curl_binary pytestconfig.getoption("install_dir"), "bin", curl_binary
) )
if ld_preload: if ld_preload:
env_vars["LD_PRELOAD"] = os.path.join( # Injecting libcurl-impersonate with LD_PRELOAD is supported on
# Linux only. On Mac there is DYLD_INSERT_LIBRARIES but it
# reuqires more work to be functional.
if not sys.platform.startswith("linux"):
pytest.skip()
self._set_ld_preload(env_vars, os.path.join(
pytestconfig.getoption("install_dir"), "lib", ld_preload pytestconfig.getoption("install_dir"), "lib", ld_preload
) ))
ret = self._run_curl(curl_binary, ret = self._run_curl(curl_binary,
env_vars=env_vars, env_vars=env_vars,
@@ -425,9 +438,16 @@ class TestImpersonation:
pytestconfig.getoption("install_dir"), "bin", curl_binary pytestconfig.getoption("install_dir"), "bin", curl_binary
) )
if ld_preload: if ld_preload:
env_vars["LD_PRELOAD"] = os.path.join( # Injecting libcurl-impersonate with LD_PRELOAD is supported on
# Linux only. On Mac there is DYLD_INSERT_LIBRARIES but it
# reuqires more work to be functional.
if not sys.platform.startswith("linux"):
pytest.skip()
self._set_ld_preload(env_vars, os.path.join(
pytestconfig.getoption("install_dir"), "lib", ld_preload pytestconfig.getoption("install_dir"), "lib", ld_preload
) ))
ret = self._run_curl(curl_binary, ret = self._run_curl(curl_binary,
env_vars=env_vars, env_vars=env_vars,
extra_args=["-k"], extra_args=["-k"],
@@ -481,9 +501,15 @@ class TestImpersonation:
pytestconfig.getoption("install_dir"), "bin", curl_binary pytestconfig.getoption("install_dir"), "bin", curl_binary
) )
if ld_preload: if ld_preload:
env_vars["LD_PRELOAD"] = os.path.join( # Injecting libcurl-impersonate with LD_PRELOAD is supported on
# Linux only. On Mac there is DYLD_INSERT_LIBRARIES but it
# reuqires more work to be functional.
if not sys.platform.startswith("linux"):
pytest.skip()
self._set_ld_preload(env_vars, os.path.join(
pytestconfig.getoption("install_dir"), "lib", ld_preload pytestconfig.getoption("install_dir"), "lib", ld_preload
) ))
output = tempfile.mkstemp()[1] output = tempfile.mkstemp()[1]
ret = self._run_curl(curl_binary, ret = self._run_curl(curl_binary,