From 138119a6a8372862c52b52c404b2172b2d79ae98 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=85=AC=E6=98=8E?= <83812544+Ed1s0nZ@users.noreply.github.com> Date: Mon, 24 Nov 2025 00:37:59 +0800 Subject: [PATCH] Add files via upload --- tools/http-framework-test.yaml | 173 ++++++++++++++++++++++++++++++++- 1 file changed, 170 insertions(+), 3 deletions(-) diff --git a/tools/http-framework-test.yaml b/tools/http-framework-test.yaml index d3a38d1c..07fac1b6 100644 --- a/tools/http-framework-test.yaml +++ b/tools/http-framework-test.yaml @@ -168,6 +168,124 @@ args: except (TypeError, ValueError): return None + def encode_form_data(data: str): + """ + 对application/x-www-form-urlencoded格式的数据进行URL编码 + 解析key=value格式，对值部分进行URL编码，保持键名不变 + 支持多个键值对（用&分隔） + 正确处理值中包含=的情况（使用split('=', 1)只分割第一个=） + 智能处理值中包含&的情况（通过检查&后是否跟着key=模式来判断） + + 注意：假设输入是未编码的原始数据，如果已经是编码的，可能会重复编码 + """ + if not data: + return data + + # 智能分割：找到真正的键值对分隔符& + # &是分隔符的条件：&后面跟着key=模式（即非空白字符后跟=） + def find_key_value_pairs(text): + """找到所有的key=value对，正确处理值中的&和=""" + pairs = [] + i = 0 + text_len = len(text) + + while i < text_len: + # 跳过空白 + while i < text_len and text[i] in ' \t\n\r': + i += 1 + if i >= text_len: + break + + # 查找键名（到第一个=为止） + key_start = i + while i < text_len and text[i] != '=': + i += 1 + + if i >= text_len: + # 没有=，可能是单个值 + remaining = text[key_start:].strip() + if remaining: + pairs.append((None, remaining)) + break + + # 提取键名 + key = text[key_start:i].strip() + i += 1 # 跳过= + + if not key: + continue + + # 查找值的结束位置 + # 值可能包含&，需要判断&是否是分隔符 + value_start = i + value_end = text_len + + # 从值开始位置查找& + j = value_start + while j < text_len: + if text[j] == '&': + # 检查&后面是否跟着key=模式 + k = j + 1 + # 跳过空白 + while k < text_len and text[k] in ' \t\n\r': + k += 1 + # 查找下一个=或& + m = k + while m < text_len and text[m] not in '=&': + m += 1 + # 如果找到=，说明&后面是新的键值对 + if m < text_len and text[m] == '=': + value_end = j + i = j + 1 # 从&后开始下一轮 + break + j += 1 + + # 提取值 + value = text[value_start:value_end] + pairs.append((key, value)) + + if value_end < text_len: + i = value_end + 1 + else: + break + + return pairs + + # 解析所有键值对 + pairs = find_key_value_pairs(data) + parts = [] + + for key, value in pairs: + if key is None: + # 没有键，只有值 + parts.append(urllib.parse.quote_plus(value, safe='')) + else: + # 对值进行URL编码 + encoded_value = urllib.parse.quote_plus(value, safe='') + parts.append(f"{key}={encoded_value}") + + return '&'.join(parts) + + def should_encode_data(headers: list, data: str): + """ + 判断是否需要对POST数据进行URL编码 + 如果Content-Type是application/x-www-form-urlencoded，则需要编码 + """ + if not data: + return False + + content_type = None + for header in headers: + if ':' in header: + h_key, h_value = header.split(':', 1) + if h_key.strip().lower() == 'content-type': + content_type = h_value.strip().lower() + break + + if content_type and 'application/x-www-form-urlencoded' in content_type: + return True + return False + parser = argparse.ArgumentParser(description="Enhanced HTTP testing helper") parser.add_argument("--url", required=True) parser.add_argument("--method", default="GET") @@ -189,6 +307,7 @@ args: parser.add_argument("--show-command", dest="show_command", action="store_true") parser.add_argument("--show-summary", dest="show_summary", action="store_true") parser.add_argument("--response-encoding", dest="response_encoding", default="") + parser.add_argument("--debug", dest="debug", action="store_true") args = parser.parse_args() repeat = max(1, args.repeat) @@ -220,10 +339,52 @@ args: curl_cmd.append("-k") if args.proxy: curl_cmd.extend(["-x", args.proxy]) - for header in parse_headers(args.headers): + + # 解析headers以便检查Content-Type + parsed_headers = parse_headers(args.headers) + for header in parsed_headers: curl_cmd.extend(["-H", header]) - if args.data: - curl_cmd.extend(["--data", args.data]) + + # 处理POST数据：如果是表单数据，需要URL编码 + prepared_data = args.data + if args.debug and args.data: + print("\n===== Debug: POST Data Processing =====") + print(f"Original data: {repr(args.data)}") + print(f"Data length: {len(args.data)} bytes") + # 显示原始数据中的键值对 + if '=' in args.data: + parts = args.data.split('&') + print(f"Detected {len(parts)} key-value pair(s):") + for i, part in enumerate(parts, 1): + if '=' in part: + k, v = part.split('=', 1) + print(f" [{i}] {k} = {repr(v[:50])}{'...' if len(v) > 50 else ''} (length: {len(v)})") + else: + print(f" [{i}] (no key): {repr(part[:50])}{'...' if len(part) > 50 else ''}") + if should_encode_data(parsed_headers, args.data): + print("Content-Type detected: application/x-www-form-urlencoded") + print("Encoding will be applied...") + else: + print("No encoding needed (not form-urlencoded)") + + if args.data and should_encode_data(parsed_headers, args.data): + prepared_data = encode_form_data(args.data) + if args.debug: + print(f"\nEncoded data: {repr(prepared_data)}") + print(f"Encoded length: {len(prepared_data)} bytes") + # 显示编码后的键值对 + if '&' in prepared_data: + encoded_parts = prepared_data.split('&') + print(f"Encoded into {len(encoded_parts)} key-value pair(s):") + for i, part in enumerate(encoded_parts, 1): + if '=' in part: + k, v = part.split('=', 1) + print(f" [{i}] {k} = {repr(v[:80])}{'...' if len(v) > 80 else ''} (length: {len(v)})") + else: + print(f" [{i}] (no key): {repr(part[:80])}{'...' if len(part) > 80 else ''}") + + if prepared_data: + curl_cmd.extend(["--data", prepared_data]) metrics_template = METRIC_MARKER + ":" + "|".join([ "%{time_namelookup}", "%{time_connect}", @@ -425,6 +586,12 @@ parameters: required: false default: true flag: "--show-summary" + - name: "debug" + type: "bool" + description: "调试模式：打印POST数据的原始值和编码后的值，方便排查编码问题" + required: false + default: false + flag: "--debug" - name: "response_encoding" type: "string" description: "强制响应解码使用的编码（默认自动检测，可用于指定GBK等场景）"