mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-01-01 11:58:21 +00:00
8c86fc5e7d
Immediately abort (un)marshal operations when encountering invalid UTF-8 data in proto3 strings. No other proto implementation supports non-UTF-8 data in proto3 strings (and many reject it in proto2 strings as well). Producing invalid output is an interoperability threat (other implementations won't be able to read it). The case where existing string data is found to contain non-UTF8 data is better handled by changing the field to the `bytes` type, which (aside from UTF-8 validation) is wire-compatible with `string`. Remove the errors.NonFatal type, since there are no remaining cases where it is needed. "Non-fatal" errors which produce results and a non-nil error are problematic because they compose poorly; the better approach is to take an option like AllowPartial indicating which conditions to check for. Change-Id: I9d189ec6ffda7b5d96d094aa1b290af2e3f23736 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/183098 Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
// Copyright 2019 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style.
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package proto
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"google.golang.org/protobuf/internal/errors"
|
|
pref "google.golang.org/protobuf/reflect/protoreflect"
|
|
)
|
|
|
|
// IsInitialized returns an error if any required fields in m are not set.
|
|
func IsInitialized(m Message) error {
|
|
if methods := protoMethods(m); methods != nil && methods.IsInitialized != nil {
|
|
// TODO: Do we need a way to disable the fast path here?
|
|
//
|
|
// TODO: Should detailed information about missing
|
|
// fields always be provided by the slow-but-informative
|
|
// reflective implementation?
|
|
return methods.IsInitialized(m)
|
|
}
|
|
return isInitialized(m.ProtoReflect(), nil)
|
|
}
|
|
|
|
// IsInitialized returns an error if any required fields in m are not set.
|
|
func isInitialized(m pref.Message, stack []interface{}) error {
|
|
md := m.Descriptor()
|
|
fds := md.Fields()
|
|
for i, nums := 0, md.RequiredNumbers(); i < nums.Len(); i++ {
|
|
fd := fds.ByNumber(nums.Get(i))
|
|
if !m.Has(fd) {
|
|
stack = append(stack, fd.Name())
|
|
return newRequiredNotSetError(stack)
|
|
}
|
|
}
|
|
var err error
|
|
m.Range(func(fd pref.FieldDescriptor, v pref.Value) bool {
|
|
// Recurse into fields containing message values.
|
|
stack := append(stack, fd.Name())
|
|
switch {
|
|
case fd.IsList():
|
|
if fd.Message() == nil {
|
|
return true
|
|
}
|
|
for i, list := 0, v.List(); i < list.Len() && err == nil; i++ {
|
|
stack := append(stack, "[", i, "].")
|
|
err = isInitialized(list.Get(i).Message(), stack)
|
|
}
|
|
case fd.IsMap():
|
|
if fd.MapValue().Message() == nil {
|
|
return true
|
|
}
|
|
v.Map().Range(func(key pref.MapKey, v pref.Value) bool {
|
|
stack := append(stack, "[", key, "].")
|
|
err = isInitialized(v.Message(), stack)
|
|
return err == nil
|
|
})
|
|
default:
|
|
if fd.Message() == nil {
|
|
return true
|
|
}
|
|
stack := append(stack, ".")
|
|
err = isInitialized(v.Message(), stack)
|
|
}
|
|
return err == nil
|
|
})
|
|
return err
|
|
}
|
|
|
|
func newRequiredNotSetError(stack []interface{}) error {
|
|
var buf bytes.Buffer
|
|
for _, s := range stack {
|
|
fmt.Fprint(&buf, s)
|
|
}
|
|
return errors.RequiredNotSet(buf.String())
|
|
}
|