DeepSeek/deepseek-harness
1
0
Fork 0
mirror of https://github.com/deepseek-ai/deepseek-harness synced 2026-08-15 21:04:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(jsonl): preserve private decoder checksum validation

Browse Source
This commit is contained in:
imccyu
2026-08-05 23:05:00 +08:00
parent 28df863bdd
commit 02c61d4481
1 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
packages/session-persistence/session-persistence-jsonl/src/zstd-private-decoder.ts
View File

@@ -24,6 +24,7 @@ interface NodeZstdPrivateHandle {
type NodeZstdPrivateWriteState = Uint32Array & { 0: number; 1: number }
interface NodeZstdPrivateState {
[key: symbol]: unknown
_handle: NodeZstdPrivateHandle | null
_writeState: NodeZstdPrivateWriteState
_defaultFlushFlag: number
@@ -32,17 +33,23 @@ interface NodeZstdPrivateState {
type NodeZstdPrivateStream = ReturnType<typeof createZstdDecompress> & NodeZstdPrivateState
/** Return the stream with its observed private Node contract, or reject that optimization. */
function privateZstdStream(stream: ReturnType<typeof createZstdDecompress>): NodeZstdPrivateStream | undefined {
function privateZstdStream(
stream: ReturnType<typeof createZstdDecompress>,
): { stream: NodeZstdPrivateStream; errorKey: symbol } | undefined {
const candidate = stream as unknown as Partial<NodeZstdPrivateState>
const handle = candidate._handle
const errorKey = Reflect.ownKeys(stream).find((key): key is symbol => (
typeof key === 'symbol' && key.description === 'kError'
))
if (
typeof handle !== 'object' || handle === null
|| typeof (handle as { writeSync?: unknown }).writeSync !== 'function'
|| !(candidate._writeState instanceof Uint32Array)
|| candidate._writeState.length < 2
|| typeof candidate._defaultFlushFlag !== 'number'
|| errorKey === undefined
) return undefined
return stream as NodeZstdPrivateStream
return { stream: stream as NodeZstdPrivateStream, errorKey }
}
/**
@@ -57,7 +64,10 @@ export class NodePrivateZstdFrameDecoder implements ZstdFrameDecoder {
private started = false
private closed = false
private constructor(private readonly stream: NodeZstdPrivateStream) {
private constructor(
private readonly stream: NodeZstdPrivateStream,
private readonly errorKey: symbol,
) {
this.stream.on('error', (error: Error) => {
this.decoderError ??= error
})
@@ -70,8 +80,10 @@ export class NodePrivateZstdFrameDecoder implements ZstdFrameDecoder {
*/
static create(): NodePrivateZstdFrameDecoder | undefined {
const stream = createZstdDecompress({ chunkSize: DECODE_CHUNK_SIZE })
const privateStream = privateZstdStream(stream)
if (privateStream !== undefined) return new NodePrivateZstdFrameDecoder(privateStream)
const privateAccess = privateZstdStream(stream)
if (privateAccess !== undefined) {
return new NodePrivateZstdFrameDecoder(privateAccess.stream, privateAccess.errorKey)
}
stream.close()
return undefined
}
@@ -116,6 +128,11 @@ export class NodePrivateZstdFrameDecoder implements ZstdFrameDecoder {
DECODE_CHUNK_SIZE,
)
if (this.decoderError !== undefined) throw this.decoderError
const internalError = this.stream[this.errorKey]
if (internalError !== undefined && internalError !== null) {
if (internalError instanceof Error) throw internalError
throw new Error('Zstandard decoder exposed a non-Error internal failure')
}
const outputAfter = this.stream._writeState[0]
const inputAfter = this.stream._writeState[1]