63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
import t, c
|
||
from stdint import *
|
||
import stdio
|
||
import stdlib
|
||
import memhub
|
||
import hashlib
|
||
|
||
|
||
POOL_SIZE: t.CDefine = 16777480
|
||
|
||
|
||
@t.CExport
|
||
def main() -> int:
|
||
arena: bytes = stdlib.malloc(POOL_SIZE)
|
||
if arena is None:
|
||
return 1
|
||
mb: memhub.MemBuddy | t.CPtr = memhub.MemBuddy(arena, POOL_SIZE)
|
||
if mb is None:
|
||
return 1
|
||
hashlib._mbuddy = (memhub.MemManager | t.CPtr)(mb)
|
||
|
||
# 测试 "abc" 的 SHA1
|
||
# 期望: a9993e364706816aba3e25717850c26c9cd0d89d
|
||
s: str = "abc"
|
||
stdio.printf("input: %s\n", s)
|
||
|
||
# 计算 len(s)
|
||
n: t.CSizeT = len(s)
|
||
stdio.printf("len=%d\n", n)
|
||
|
||
# 构造 sha1 对象
|
||
ctx: hashlib.sha1 | t.CPtr = hashlib.sha1()
|
||
if ctx is None:
|
||
stdio.printf("ctx is None\n")
|
||
return 1
|
||
|
||
ctx.update(s)
|
||
|
||
digest: bytes = mb.alloc(hashlib.SHA1_DIGEST_LEN)
|
||
if digest is None:
|
||
stdio.printf("digest alloc failed\n")
|
||
return 1
|
||
|
||
ctx.final(digest)
|
||
|
||
# 打印 digest（十六进制）
|
||
stdio.printf("SHA1: ")
|
||
for i in range(hashlib.SHA1_DIGEST_LEN):
|
||
b: t.CUInt8T = digest[i]
|
||
hi: t.CInt = (b >> 4) & 0xF
|
||
lo: t.CInt = b & 0xF
|
||
if hi < 10:
|
||
stdio.printf("%c", '0' + hi)
|
||
else:
|
||
stdio.printf("%c", 'a' + (hi - 10))
|
||
if lo < 10:
|
||
stdio.printf("%c", '0' + lo)
|
||
else:
|
||
stdio.printf("%c", 'a' + (lo - 10))
|
||
stdio.printf("\n")
|
||
stdio.printf("expect: a9993e364706816aba3e25717850c26c9cd0d89d\n")
|
||
return 0
|