TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-05 00:34:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

增强KV字节编码安全性

Browse Source
This commit is contained in:
GoEdgeLab
2024-04-24 14:33:15 +08:00
parent f80788ea54
commit 1cf63ff17c
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/utils/kvstore/value_encoder_bytes.go
View File

@@ -10,7 +10,13 @@ func NewBytesValueEncoder[T []byte]() *BytesValueEncoder[T] {
} }
func (this *BytesValueEncoder[T]) Encode(value T) ([]byte, error) { func (this *BytesValueEncoder[T]) Encode(value T) ([]byte, error) {
return value, nil if len(value) == 0 {
return nil, nil
}
var resultValue = make([]byte, len(value))
copy(resultValue, value)
return resultValue, nil
} }
func (this *BytesValueEncoder[T]) EncodeField(value T, fieldName string) ([]byte, error) { func (this *BytesValueEncoder[T]) EncodeField(value T, fieldName string) ([]byte, error) {
@@ -19,6 +25,11 @@ func (this *BytesValueEncoder[T]) EncodeField(value T, fieldName string) ([]byte
} }
func (this *BytesValueEncoder[T]) Decode(valueData []byte) (value T, err error) { func (this *BytesValueEncoder[T]) Decode(valueData []byte) (value T, err error) {
value = valueData if len(valueData) == 0 {
return
}
value = make([]byte, len(valueData))
copy(value, valueData)
return return
} }