fix: Use base64 to encode not utf-8 bytes

See also: #45654

Signed-off-by: yangxuan <[email protected]>

Author: XuanYang-cn

internal/flushcommon/syncmgr/pack_writer_v2.go

@@ -18,6 +18,7 @@ package syncmgr
 
 import (
 	"context"
+	"encoding/base64"
 	"math"
 
 	"github.com/apache/arrow/go/v17/arrow/array"
@@ -167,7 +168,7 @@ func (bw *BulkPackWriterV2) writeInserts(ctx context.Context, pack *SyncPack) (m
 				pluginContext := indexcgopb.StoragePluginContext{
 					EncryptionZoneId: ez.EzID,
 					CollectionId:     ez.CollectionID,
-					EncryptionKey:    string(unsafe),
+					EncryptionKey:    base64.StdEncoding.EncodeToString(unsafe),
 				}
 				pluginContextPtr = &pluginContext
 			}

internal/querynodev2/segments/collection.go

@@ -17,6 +17,7 @@
 package segments
 
 import (
+	"encoding/base64"
 	"fmt"
 	"sync"
 
@@ -398,7 +399,7 @@ func putOrUpdateStorageContext(properties []*commonpb.KeyValuePair, collectionID
 		ez := hookutil.GetEzByCollProperties(properties, collectionID)
 		if ez != nil {
 			key := hookutil.GetCipher().GetUnsafeKey(ez.EzID, ez.CollectionID)
-			err := segcore.PutOrRefPluginContext(ez, string(key))
+			err := segcore.PutOrRefPluginContext(ez, base64.StdEncoding.EncodeToString(key))
 			if err != nil {
 				log.Error("failed to put or update plugin context", zap.Int64("collectionID", collectionID), zap.Error(err))
 			}

internal/storage/rw.go

@@ -18,6 +18,7 @@ package storage
 
 import (
 	"context"
+	"encoding/base64"
 	"fmt"
 	sio "io"
 	"sort"
@@ -253,7 +254,7 @@ func NewBinlogRecordReader(ctx context.Context, binlogs []*datapb.FieldBinlog, s
 				pluginContext = &indexcgopb.StoragePluginContext{
 					EncryptionZoneId: ez.EzID,
 					CollectionId:     ez.CollectionID,
-					EncryptionKey:    string(unsafe),
+					EncryptionKey:    base64.StdEncoding.EncodeToString(unsafe),
 				}
 			}
 		}
@@ -326,7 +327,7 @@ func NewBinlogRecordWriter(ctx context.Context, collectionID, partitionID, segme
 				pluginContext = &indexcgopb.StoragePluginContext{
 					EncryptionZoneId: ez.EzID,
 					CollectionId:     ez.CollectionID,
-					EncryptionKey:    string(unsafe),
+					EncryptionKey:    base64.StdEncoding.EncodeToString(unsafe),
 				}
 			}
 		}

internal/util/hookutil/cipher.go

@@ -19,6 +19,7 @@ package hookutil
 import (
 	"bytes"
 	"context"
+	"encoding/base64"
 	"fmt"
 	"plugin"
 	"strconv"
@@ -132,7 +133,7 @@ func GetStoragePluginContext(properties []*commonpb.KeyValuePair, collectionID i
 			},
 			{
 				Key:   CipherConfigUnsafeEZK,
-				Value: string(key),
+				Value: base64.StdEncoding.EncodeToString(key),
 			},
 		}
 		return pluginContext

internal/util/hookutil/cipher_test.go

@@ -18,6 +18,7 @@ package hookutil
 
 import (
 	"context"
+	"encoding/base64"
 	"fmt"
 	"sync"
 	"testing"
@@ -243,7 +244,7 @@ func (s *CipherSuite) TestGetStoragePluginContext() {
 	s.Equal(CipherConfigCreateEZ, result[0].Key)
 	s.Equal("1", result[0].Value)
 	s.Equal(CipherConfigUnsafeEZK, result[1].Key)
-	s.Equal("unsafe key", result[1].Value)
+	s.Equal(base64.StdEncoding.EncodeToString([]byte("unsafe key")), result[1].Value)
 
 	result = GetStoragePluginContext([]*commonpb.KeyValuePair{}, 2)
 	s.Nil(result)