加个保护

antlabs · May 18, 2024 · 122fb7e · 122fb7e
1 parent 1702051
commit 122fb7e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/deflate/compression_encode.go b/deflate/compression_encode.go
@@ -129,6 +129,9 @@ func (e *CompressContextTakeover) Compress(payload *[]byte, bit uint8) (encodePa
 		bytespool.PutBytes(encodeBuf)
 	}
 
+	if e != nil && payload != nil {
+		e.dict.Write(*payload)
+	}
 	outBuf := out.Bytes()
 	return &outBuf, nil
 }