80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
import t, c
|
||
from stdint import *
|
||
import stdio
|
||
import stdlib
|
||
import socket
|
||
import memhub
|
||
import requests
|
||
|
||
|
||
# 全局 MBuddy 指针
|
||
_mbuddy: memhub.MemBuddy | t.CPtr
|
||
|
||
|
||
@t.CExport
|
||
def main() -> int:
|
||
stdio.printf("WebTest: HTTP client library test\n")
|
||
|
||
# 初始化内存管理器（16MB arena）
|
||
arena: bytes = stdlib.malloc(16 * 1024 * 1024)
|
||
if arena is None:
|
||
stdio.printf("FATAL: malloc arena failed\n")
|
||
return 1
|
||
|
||
_mbuddy = memhub.MemBuddy(arena, 16 * 1024 * 1024)
|
||
requests._mbuddy = _mbuddy
|
||
|
||
# 初始化 Socket 库
|
||
result: INT = socket.SocketInit()
|
||
stdio.printf("SocketInit: %d\n", result)
|
||
|
||
# 测试 Socket OOP API
|
||
sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
if sock.fd != socket.INVALID_SOCKET:
|
||
stdio.printf("Socket: OK (fd=%d)\n", sock.fd)
|
||
sock.close()
|
||
stdio.printf("Socket.close: OK\n")
|
||
else:
|
||
stdio.printf("Socket: FAIL\n")
|
||
|
||
# 测试 DNS 解析 — 通过 Socket.connect 间接测试（内部调用 _MakeAddr）
|
||
stdio.printf("\n--- DNS test ---\n")
|
||
dns_sock = socket.Socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
if dns_sock.fd != socket.INVALID_SOCKET:
|
||
dns_sock.settimeout(5000)
|
||
result: INT = dns_sock.connect("example.com", 80)
|
||
if result == 0:
|
||
stdio.printf("DNS: example.com resolved OK\n")
|
||
else:
|
||
stdio.printf("DNS: example.com resolved but connect failed (err in _MakeAddr debug above)\n")
|
||
dns_sock.close()
|
||
|
||
# 测试 HTTP GET — 使用 DNS 解析 example.com
|
||
stdio.printf("\n--- HTTP GET test (example.com) ---\n")
|
||
resp = requests.get("http://example.com/")
|
||
if resp is not None:
|
||
stdio.printf("requests.get: OK\n")
|
||
code: INT = resp.status_code
|
||
length: INT = resp.body_len
|
||
stdio.printf(" status_code: %d\n", code)
|
||
stdio.printf(" body_len: %d\n", length)
|
||
if resp.text is not None and length > 0:
|
||
print_len: INT = 300
|
||
if length < print_len:
|
||
print_len = length
|
||
stdio.printf(" text (first %d chars):\n", print_len)
|
||
txt: str = resp.text
|
||
i: INT = 0
|
||
while i < print_len and txt[i] != 0:
|
||
stdio.printf("%c", txt[i])
|
||
i += 1
|
||
stdio.printf("\n")
|
||
requests.release(resp)
|
||
else:
|
||
stdio.printf("requests.get: FAIL (network unavailable)\n")
|
||
|
||
# 清理
|
||
socket.SocketCleanup()
|
||
stdio.printf("\nSocketCleanup: OK\n")
|
||
stdio.printf("WebTest: ALL PASSED\n")
|
||
return 0 |