code/default/x_tunnel/local/cloudflare_front/test.py - master - githubmirror/XX-Net

code/default/x_tunnel/local/cloudflare_front/test.py

76 строк2 KB

Micheal X

5.5.13 refactor codes, fix minor bugs, add log collection function for report issues.

19 авг 2023, 09:26

19 авг 2023, 09:26aea538f

#!/usr/bin/env python2
# coding:utf-8

import os
import sys
import time

import utils

current_path = os.path.dirname(os.path.abspath(__file__))
root_path = os.path.abspath(os.path.join(current_path, os.pardir, os.pardir, os.pardir))
python_path = root_path

sys.path.append(root_path)

noarch_lib = os.path.abspath(os.path.join(python_path, 'lib', 'noarch'))
sys.path.append(noarch_lib)

if sys.platform == "win32":
    win32_lib = os.path.abspath(os.path.join(python_path, 'lib', 'win32'))
    sys.path.append(win32_lib)
elif sys.platform.startswith("linux"):
    linux_lib = os.path.abspath(os.path.join(python_path, 'lib', 'linux'))
    sys.path.append(linux_lib)
elif sys.platform == "darwin":
    darwin_lib = os.path.abspath(os.path.join(python_path, 'lib', 'darwin'))
    sys.path.append(darwin_lib)
    extra_lib = "/System/Library/Frameworks/Python.framework/Versions/2.7/Extras/lib/python"
    sys.path.append(extra_lib)

import x_tunnel.local.cloudflare_front as front
import env_info

from xlog import getLogger
xlog = getLogger("cloudflare_front")
xlog.set_buffer(2000)

data_path = env_info.data_path
module_data_path = os.path.join(data_path, 'x_tunnel')


def get_dns():
    start_time = time.time()
    # content, status, response = front.request("GET", "scan1.xx-net.org", "/", timeout=10)
    content, status, response = front.front.request("GET", "dns.xx-net.org", path="/query?domain=www.google.com")

    time_cost = time.time() - start_time
    xlog.info("GET cost:%f", time_cost)
    xlog.info("status:%d content:%s", status, content)
    front.front.stop()


def post_data():
    start_time = time.time()
    data = utils.generate_random_lowercase(900 * 1024)
    content, status, response = front.front.request("POST", "v3.xx-net.org", path="/upload_logs?session_id=tyjiugru",
                                                    data=data,
                                                    headers={"Content-Length": str(len(data))})

    time_cost = time.time() - start_time
    xlog.info("POST cost:%f", time_cost)
    xlog.info("status:%d content:%s", status, content)
    front.front.stop()


if __name__ == '__main__':
    import traceback

    try:
        # post_data()
        get_dns()
    except Exception:
        traceback.print_exc(file=sys.stdout)
    except KeyboardInterrupt:
        front.stop()
        sys.exit()