csgo_tools/sda_code.py at master · ShaBaoFa/csgo_tools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import os.path
import sys

from time import time
from Crypto.Hash import HMAC, SHA1
import requests
from struct import pack, unpack
import json
from base64 import b64decode

# from apscheduler.schedulers.background import BackgroundScheduler

two_factor_time_query_url = "https://api.steampowered.com/ITwoFactorService/QueryTime/v0001"

def resource_path(relative_path):
    """ 获取资源文件的绝对路径 """
    base_path = getattr(sys, '_MEIPASS', os.path.dirname(os.path.abspath(__file__)))
    return os.path.join(base_path, relative_path)

def get_app_path():
    if hasattr(sys, 'frozen'):
        # If the application is frozen (using PyInstaller)
        return os.path.dirname(sys.executable)
    else:
        # If the application is not frozen
        return os.path.dirname(os.path.abspath(__file__))

def get_external_file_path(filename):
    app_path = get_app_path()
    if hasattr(sys, 'frozen') and sys.platform.startswith('darwin'):
        # 打包后的运行环境
        external_path = os.path.abspath(os.path.join(app_path, '..', '..', '..'))
    else:
        # 开发调试环境
        external_path = app_path
    return os.path.join(external_path, filename)

# 得到时间补偿量，否则不能产生正确的code
def get_time_offset():
    twofactor_querytime_response = requests.post(url=two_factor_time_query_url, params={'http_timeout': 10}).json()
    ts = int(time())
    return int(twofactor_querytime_response.get('response', {}).get('server_time', ts)) - ts


def hmac_sha1(secret, data):
    return HMAC.new(secret, data, SHA1).digest()


# 根据校准后的时间生成令牌
def generate_twofactor_code_for_time(shared_secret, aligned_time):
    hmac = hmac_sha1(b64decode(shared_secret),
                     pack('>Q', int(aligned_time) // 30))  # this will NOT stop working in 2038

    start = ord(hmac[19:20]) & 0xF
    codeint = unpack('>I', hmac[start:start + 4])[0] & 0x7fffffff

    charset = '23456789BCDFGHJKMNPQRTVWXY'
    code = ''

    for _ in range(5):
        codeint, i = divmod(codeint, len(charset))
        code += charset[i]
    return code


# 模拟生成验证码
def generator_code(steam_id, user_name):
    # 在控制器中打印出steam_id和user_name
    if sys.platform.startswith('darwin'):
        steam_id_path = os.path.join(get_external_file_path('maFiles'), f'{steam_id}.maFile')
        user_name_path = os.path.join(get_external_file_path('maFiles'), f'{user_name}.maFile')
    elif sys.platform.startswith('win'):
        steam_id_path = os.path.join(get_external_file_path('maFiles'), f'{steam_id}.maFile')
        user_name_path = os.path.join(get_external_file_path('maFiles'), f'{user_name}.maFile')
    # 从文件中读入shared_secrets
    if steam_id and os.path.exists(steam_id_path):
        with open(steam_id_path) as fn:
            js = fn.read()
            dic = json.loads(js)
            shared_secret = dic.get('shared_secret')
    elif os.path.exists(user_name_path):
        with open(user_name_path) as fn:
            js = fn.read()
            dic = json.loads(js)
            shared_secret = dic.get('shared_secret')
    if shared_secret:
        aligned_time = int(time() + get_time_offset())  # 补偿后的时间
        two_factor_code = generate_twofactor_code_for_time(shared_secret, aligned_time)
        if len(two_factor_code) == 5:
            return True, two_factor_code
        else:
            return False
    else:
        return False