File size: 3,854 Bytes
dc0849b |
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 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
from flask import Flask, request, jsonify
import os
import requests
app = Flask(__name__)
# 定义全局字符串变量
str_values = ''
str_uvalues = ''
# 从环境变量中获取密码
password = os.environ.get('PASSWORD', '123456')
# 定义一个函数，接受 Cookie-Values 和要保留的键的列表作为参数
def filter_cookie_values(cookie_values, keep_keys):
new_cookie_values = ""
pairs = [pair.strip() for pair in cookie_values.split(';')]
for pair in pairs:
if '=' in pair:
key, value = pair.split('=', 1)
key = key.strip()
value = value.strip()
if key in keep_keys:
new_cookie_values += f'{key}={value}; '
new_cookie_values = new_cookie_values[:-2]
return new_cookie_values
@app.route('/SET', methods=['POST'])
def set_value():
method = request.method
pwd = request.args.get('pwd')
if not pwd or pwd!= password:
return 'Invalid password', 401
keep_keys = ["_U",
"MUID",
'KievRPSSecAuth',
'cct',
'buid',
'ak_bmsc',
'bm_sv',
'cl_dtrs',
'_RwBf',
'SRCHHPGUSR',
'WLS']
keep_keys_u = ["_U",
"WLS"]
cookie_values = request.headers.get('Cookie-Values')
set_value = filter_cookie_values(cookie_values, keep_keys)
get_u_value = filter_cookie_values(cookie_values, keep_keys_u)
if set_value:
str_values = set_value
if get_u_value and get_u_value not in str_uvalues:
str_uvalues += ';' + get_u_value
return 'Set value successfully'
else:
return 'No Cookie-Values in header', 400
@app.route('/GET', methods=['GET'])
def get_value():
method = request.method
pwd = request.args.get('pwd')
if not pwd or pwd!= password:
return 'Invalid password', 401
result = {'result': {'cookies': str_values}}
return jsonify(result)
@app.route('/CLS', methods=['GET', 'POST'])
def clear_value():
method = request.method
pwd = request.args.get('pwd')
if not pwd or pwd!= password:
return 'Invalid password', 401
replaced_str = str_uvalues.replace(';', '<br>')
str_values = ''
str_uvalues = ''
return 'Clear value successfully' + '\n' + replaced_str
@app.route('/HisU', methods=['GET', 'POST'])
def history_u():
method = request.method
pwd = request.args.get('pwd')
if not pwd or pwd!= password:
return 'Invalid password', 401
replaced_str = str_uvalues.replace(';', '<br>')
return 'Ukey History:' + '\n' + replaced_str
@app.route('/', methods=['GET', 'POST'])
def root():
return 'Please visit /SET /GET or /CLS with?pwd=xxxxxx'
@app.route('/q', methods=['GET'])
def q_route():
url = request.args.get('url')
if not url:
return jsonify({'error': 'URL is required'}), 400
keep_keys = ["_U",
"MUID",
'KievRPSSecAuth',
'cct',
'buid',
'ak_bmsc',
'bm_sv',
'_RwBf',
'SRCHHPGUSR',
'WLS']
response = requests.get(url, headers={
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0',
'Accept-Language': 'zh-CN,zh;q=0.9'
})
set_cookie_array = response.headers.get('set-cookie', [])
if set_cookie_array:
cookies = [cookie.split(';')[0] for cookie in set_cookie_array]
cookies_str = '; '.join(cookies)
else:
cookies_str = ''
set_value = filter_cookie_values(cookies_str, keep_keys)
result = {'result': {'cookies': set_value}}
return jsonify(result)
if __name__ == '__main__':
app.run(port=7860) |