fix decoding problem

google · Oct 12, 2024 · 605f0ff · 605f0ff
1 parent 6a11024
commit 605f0ff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/launcher/internal/logging/logging.go b/launcher/internal/logging/logging.go
@@ -215,13 +215,13 @@ func (l *logger) Error(msg string, args ...any) {
 // Write implements the Writer interface for workload logging redirection.
 func (l *logger) Write(p []byte) (int, error) {
 	// Base64 decode input.
-	var decodedMessage []byte
-	_, err := base64.RawStdEncoding.Decode(decodedMessage, p)
+	decodedMessage := make([]byte, base64.StdEncoding.DecodedLen(len(p)))
+	n, err := base64.StdEncoding.Decode(decodedMessage, p)
 	if err != nil {
 		return 0, fmt.Errorf("error decoding message for cloud logging: %v", err)
 	}
 
-	pl := payload{payloadMessageKey: decodedMessage}
+	pl := payload{payloadMessageKey: decodedMessage[:n]}
 
 	if len(l.instanceName) > 0 {
 		pl[payloadInstanceNameKey] = l.instanceName