 8d9b62daf3
			
		
	
	8d9b62daf3
	
	
	
		
			
			This commit implements Phase 2 of the CHORUS Task Execution Engine development plan, providing a comprehensive execution environment abstraction layer with Docker container sandboxing support. ## New Features ### Core Sandbox Interface - Comprehensive ExecutionSandbox interface with isolated task execution - Support for command execution, file I/O, environment management - Resource usage monitoring and sandbox lifecycle management - Standardized error handling with SandboxError types and categories ### Docker Container Sandbox Implementation - Full Docker API integration with secure container creation - Transparent repository mounting with configurable read/write access - Advanced security policies with capability dropping and privilege controls - Comprehensive resource limits (CPU, memory, disk, processes, file handles) - Support for tmpfs mounts, masked paths, and read-only bind mounts - Container lifecycle management with proper cleanup and health monitoring ### Security & Resource Management - Configurable security policies with SELinux, AppArmor, and Seccomp support - Fine-grained capability management with secure defaults - Network isolation options with configurable DNS and proxy settings - Resource monitoring with real-time CPU, memory, and network usage tracking - Comprehensive ulimits configuration for process and file handle limits ### Repository Integration - Seamless repository mounting from local paths to container workspaces - Git configuration support with user credentials and global settings - File inclusion/exclusion patterns for selective repository access - Configurable permissions and ownership for mounted repositories ### Testing Infrastructure - Comprehensive test suite with 60+ test cases covering all functionality - Docker integration tests with Alpine Linux containers (skipped in short mode) - Mock sandbox implementation for unit testing without Docker dependencies - Security policy validation tests with read-only filesystem enforcement - Resource usage monitoring and cleanup verification tests ## Technical Details ### Dependencies Added - github.com/docker/docker v28.4.0+incompatible - Docker API client - github.com/docker/go-connections v0.6.0 - Docker connection utilities - github.com/docker/go-units v0.5.0 - Docker units and formatting - Associated Docker API dependencies for complete container management ### Architecture - Interface-driven design enabling multiple sandbox implementations - Comprehensive configuration structures for all sandbox aspects - Resource usage tracking with detailed metrics collection - Error handling with retryable error classification - Proper cleanup and resource management throughout sandbox lifecycle ### Compatibility - Maintains backward compatibility with existing CHORUS architecture - Designed for future integration with Phase 3 Core Task Execution Engine - Extensible design supporting additional sandbox implementations (VM, process) This Phase 2 implementation provides the foundation for secure, isolated task execution that will be integrated with the AI model providers from Phase 1 in the upcoming Phase 3 development. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
		
			
				
	
	
		
			861 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			861 lines
		
	
	
		
			26 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 impl
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"reflect"
 | |
| 	"sync"
 | |
| 
 | |
| 	"google.golang.org/protobuf/encoding/protowire"
 | |
| 	"google.golang.org/protobuf/internal/errors"
 | |
| 	"google.golang.org/protobuf/proto"
 | |
| 	"google.golang.org/protobuf/reflect/protoreflect"
 | |
| 	"google.golang.org/protobuf/reflect/protoregistry"
 | |
| 	"google.golang.org/protobuf/runtime/protoiface"
 | |
| )
 | |
| 
 | |
| type errInvalidUTF8 struct{}
 | |
| 
 | |
| func (errInvalidUTF8) Error() string     { return "string field contains invalid UTF-8" }
 | |
| func (errInvalidUTF8) InvalidUTF8() bool { return true }
 | |
| func (errInvalidUTF8) Unwrap() error     { return errors.Error }
 | |
| 
 | |
| // initOneofFieldCoders initializes the fast-path functions for the fields in a oneof.
 | |
| //
 | |
| // For size, marshal, and isInit operations, functions are set only on the first field
 | |
| // in the oneof. The functions are called when the oneof is non-nil, and will dispatch
 | |
| // to the appropriate field-specific function as necessary.
 | |
| //
 | |
| // The unmarshal function is set on each field individually as usual.
 | |
| func (mi *MessageInfo) initOneofFieldCoders(od protoreflect.OneofDescriptor, si structInfo) {
 | |
| 	fs := si.oneofsByName[od.Name()]
 | |
| 	ft := fs.Type
 | |
| 	oneofFields := make(map[reflect.Type]*coderFieldInfo)
 | |
| 	needIsInit := false
 | |
| 	fields := od.Fields()
 | |
| 	for i, lim := 0, fields.Len(); i < lim; i++ {
 | |
| 		fd := od.Fields().Get(i)
 | |
| 		num := fd.Number()
 | |
| 		// Make a copy of the original coderFieldInfo for use in unmarshaling.
 | |
| 		//
 | |
| 		// oneofFields[oneofType].funcs.marshal is the field-specific marshal function.
 | |
| 		//
 | |
| 		// mi.coderFields[num].marshal is set on only the first field in the oneof,
 | |
| 		// and dispatches to the field-specific marshaler in oneofFields.
 | |
| 		cf := *mi.coderFields[num]
 | |
| 		ot := si.oneofWrappersByNumber[num]
 | |
| 		cf.ft = ot.Field(0).Type
 | |
| 		cf.mi, cf.funcs = fieldCoder(fd, cf.ft)
 | |
| 		oneofFields[ot] = &cf
 | |
| 		if cf.funcs.isInit != nil {
 | |
| 			needIsInit = true
 | |
| 		}
 | |
| 		mi.coderFields[num].funcs.unmarshal = func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 			var vw reflect.Value         // pointer to wrapper type
 | |
| 			vi := p.AsValueOf(ft).Elem() // oneof field value of interface kind
 | |
| 			if !vi.IsNil() && !vi.Elem().IsNil() && vi.Elem().Elem().Type() == ot {
 | |
| 				vw = vi.Elem()
 | |
| 			} else {
 | |
| 				vw = reflect.New(ot)
 | |
| 			}
 | |
| 			out, err := cf.funcs.unmarshal(b, pointerOfValue(vw).Apply(zeroOffset), wtyp, &cf, opts)
 | |
| 			if err != nil {
 | |
| 				return out, err
 | |
| 			}
 | |
| 			vi.Set(vw)
 | |
| 			return out, nil
 | |
| 		}
 | |
| 	}
 | |
| 	getInfo := func(p pointer) (pointer, *coderFieldInfo) {
 | |
| 		v := p.AsValueOf(ft).Elem()
 | |
| 		if v.IsNil() {
 | |
| 			return pointer{}, nil
 | |
| 		}
 | |
| 		v = v.Elem() // interface -> *struct
 | |
| 		if v.IsNil() {
 | |
| 			return pointer{}, nil
 | |
| 		}
 | |
| 		return pointerOfValue(v).Apply(zeroOffset), oneofFields[v.Elem().Type()]
 | |
| 	}
 | |
| 	first := mi.coderFields[od.Fields().Get(0).Number()]
 | |
| 	first.funcs.size = func(p pointer, _ *coderFieldInfo, opts marshalOptions) int {
 | |
| 		p, info := getInfo(p)
 | |
| 		if info == nil || info.funcs.size == nil {
 | |
| 			return 0
 | |
| 		}
 | |
| 		return info.funcs.size(p, info, opts)
 | |
| 	}
 | |
| 	first.funcs.marshal = func(b []byte, p pointer, _ *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 		p, info := getInfo(p)
 | |
| 		if info == nil || info.funcs.marshal == nil {
 | |
| 			return b, nil
 | |
| 		}
 | |
| 		return info.funcs.marshal(b, p, info, opts)
 | |
| 	}
 | |
| 	first.funcs.merge = func(dst, src pointer, _ *coderFieldInfo, opts mergeOptions) {
 | |
| 		srcp, srcinfo := getInfo(src)
 | |
| 		if srcinfo == nil || srcinfo.funcs.merge == nil {
 | |
| 			return
 | |
| 		}
 | |
| 		dstp, dstinfo := getInfo(dst)
 | |
| 		if dstinfo != srcinfo {
 | |
| 			dst.AsValueOf(ft).Elem().Set(reflect.New(src.AsValueOf(ft).Elem().Elem().Elem().Type()))
 | |
| 			dstp = pointerOfValue(dst.AsValueOf(ft).Elem().Elem()).Apply(zeroOffset)
 | |
| 		}
 | |
| 		srcinfo.funcs.merge(dstp, srcp, srcinfo, opts)
 | |
| 	}
 | |
| 	if needIsInit {
 | |
| 		first.funcs.isInit = func(p pointer, _ *coderFieldInfo) error {
 | |
| 			p, info := getInfo(p)
 | |
| 			if info == nil || info.funcs.isInit == nil {
 | |
| 				return nil
 | |
| 			}
 | |
| 			return info.funcs.isInit(p, info)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func makeWeakMessageFieldCoder(fd protoreflect.FieldDescriptor) pointerCoderFuncs {
 | |
| 	var once sync.Once
 | |
| 	var messageType protoreflect.MessageType
 | |
| 	lazyInit := func() {
 | |
| 		once.Do(func() {
 | |
| 			messageName := fd.Message().FullName()
 | |
| 			messageType, _ = protoregistry.GlobalTypes.FindMessageByName(messageName)
 | |
| 		})
 | |
| 	}
 | |
| 
 | |
| 	return pointerCoderFuncs{
 | |
| 		size: func(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 			m, ok := p.WeakFields().get(f.num)
 | |
| 			if !ok {
 | |
| 				return 0
 | |
| 			}
 | |
| 			lazyInit()
 | |
| 			if messageType == nil {
 | |
| 				panic(fmt.Sprintf("weak message %v is not linked in", fd.Message().FullName()))
 | |
| 			}
 | |
| 			return sizeMessage(m, f.tagsize, opts)
 | |
| 		},
 | |
| 		marshal: func(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 			m, ok := p.WeakFields().get(f.num)
 | |
| 			if !ok {
 | |
| 				return b, nil
 | |
| 			}
 | |
| 			lazyInit()
 | |
| 			if messageType == nil {
 | |
| 				panic(fmt.Sprintf("weak message %v is not linked in", fd.Message().FullName()))
 | |
| 			}
 | |
| 			return appendMessage(b, m, f.wiretag, opts)
 | |
| 		},
 | |
| 		unmarshal: func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 			fs := p.WeakFields()
 | |
| 			m, ok := fs.get(f.num)
 | |
| 			if !ok {
 | |
| 				lazyInit()
 | |
| 				if messageType == nil {
 | |
| 					return unmarshalOutput{}, errUnknown
 | |
| 				}
 | |
| 				m = messageType.New().Interface()
 | |
| 				fs.set(f.num, m)
 | |
| 			}
 | |
| 			return consumeMessage(b, m, wtyp, opts)
 | |
| 		},
 | |
| 		isInit: func(p pointer, f *coderFieldInfo) error {
 | |
| 			m, ok := p.WeakFields().get(f.num)
 | |
| 			if !ok {
 | |
| 				return nil
 | |
| 			}
 | |
| 			return proto.CheckInitialized(m)
 | |
| 		},
 | |
| 		merge: func(dst, src pointer, f *coderFieldInfo, opts mergeOptions) {
 | |
| 			sm, ok := src.WeakFields().get(f.num)
 | |
| 			if !ok {
 | |
| 				return
 | |
| 			}
 | |
| 			dm, ok := dst.WeakFields().get(f.num)
 | |
| 			if !ok {
 | |
| 				lazyInit()
 | |
| 				if messageType == nil {
 | |
| 					panic(fmt.Sprintf("weak message %v is not linked in", fd.Message().FullName()))
 | |
| 				}
 | |
| 				dm = messageType.New().Interface()
 | |
| 				dst.WeakFields().set(f.num, dm)
 | |
| 			}
 | |
| 			opts.Merge(dm, sm)
 | |
| 		},
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func makeMessageFieldCoder(fd protoreflect.FieldDescriptor, ft reflect.Type) pointerCoderFuncs {
 | |
| 	if mi := getMessageInfo(ft); mi != nil {
 | |
| 		funcs := pointerCoderFuncs{
 | |
| 			size:      sizeMessageInfo,
 | |
| 			marshal:   appendMessageInfo,
 | |
| 			unmarshal: consumeMessageInfo,
 | |
| 			merge:     mergeMessage,
 | |
| 		}
 | |
| 		if needsInitCheck(mi.Desc) {
 | |
| 			funcs.isInit = isInitMessageInfo
 | |
| 		}
 | |
| 		return funcs
 | |
| 	} else {
 | |
| 		return pointerCoderFuncs{
 | |
| 			size: func(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return sizeMessage(m, f.tagsize, opts)
 | |
| 			},
 | |
| 			marshal: func(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return appendMessage(b, m, f.wiretag, opts)
 | |
| 			},
 | |
| 			unmarshal: func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 				mp := p.AsValueOf(ft).Elem()
 | |
| 				if mp.IsNil() {
 | |
| 					mp.Set(reflect.New(ft.Elem()))
 | |
| 				}
 | |
| 				return consumeMessage(b, asMessage(mp), wtyp, opts)
 | |
| 			},
 | |
| 			isInit: func(p pointer, f *coderFieldInfo) error {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return proto.CheckInitialized(m)
 | |
| 			},
 | |
| 			merge: mergeMessage,
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func sizeMessageInfo(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 	return protowire.SizeBytes(f.mi.sizePointer(p.Elem(), opts)) + f.tagsize
 | |
| }
 | |
| 
 | |
| func appendMessageInfo(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 	calculatedSize := f.mi.sizePointer(p.Elem(), opts)
 | |
| 	b = protowire.AppendVarint(b, f.wiretag)
 | |
| 	b = protowire.AppendVarint(b, uint64(calculatedSize))
 | |
| 	before := len(b)
 | |
| 	b, err := f.mi.marshalAppendPointer(b, p.Elem(), opts)
 | |
| 	if measuredSize := len(b) - before; calculatedSize != measuredSize && err == nil {
 | |
| 		return nil, errors.MismatchedSizeCalculation(calculatedSize, measuredSize)
 | |
| 	}
 | |
| 	return b, err
 | |
| }
 | |
| 
 | |
| func consumeMessageInfo(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.BytesType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	v, n := protowire.ConsumeBytes(b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	if p.Elem().IsNil() {
 | |
| 		p.SetPointer(pointerOfValue(reflect.New(f.mi.GoReflectType.Elem())))
 | |
| 	}
 | |
| 	o, err := f.mi.unmarshalPointer(v, p.Elem(), 0, opts)
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	out.n = n
 | |
| 	out.initialized = o.initialized
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func isInitMessageInfo(p pointer, f *coderFieldInfo) error {
 | |
| 	return f.mi.checkInitializedPointer(p.Elem())
 | |
| }
 | |
| 
 | |
| func sizeMessage(m proto.Message, tagsize int, opts marshalOptions) int {
 | |
| 	return protowire.SizeBytes(opts.Options().Size(m)) + tagsize
 | |
| }
 | |
| 
 | |
| func appendMessage(b []byte, m proto.Message, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	mopts := opts.Options()
 | |
| 	calculatedSize := mopts.Size(m)
 | |
| 	b = protowire.AppendVarint(b, wiretag)
 | |
| 	b = protowire.AppendVarint(b, uint64(calculatedSize))
 | |
| 	before := len(b)
 | |
| 	b, err := mopts.MarshalAppend(b, m)
 | |
| 	if measuredSize := len(b) - before; calculatedSize != measuredSize && err == nil {
 | |
| 		return nil, errors.MismatchedSizeCalculation(calculatedSize, measuredSize)
 | |
| 	}
 | |
| 	return b, err
 | |
| }
 | |
| 
 | |
| func consumeMessage(b []byte, m proto.Message, wtyp protowire.Type, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.BytesType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	v, n := protowire.ConsumeBytes(b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     v,
 | |
| 		Message: m.ProtoReflect(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func sizeMessageValue(v protoreflect.Value, tagsize int, opts marshalOptions) int {
 | |
| 	m := v.Message().Interface()
 | |
| 	return sizeMessage(m, tagsize, opts)
 | |
| }
 | |
| 
 | |
| func appendMessageValue(b []byte, v protoreflect.Value, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	m := v.Message().Interface()
 | |
| 	return appendMessage(b, m, wiretag, opts)
 | |
| }
 | |
| 
 | |
| func consumeMessageValue(b []byte, v protoreflect.Value, _ protowire.Number, wtyp protowire.Type, opts unmarshalOptions) (protoreflect.Value, unmarshalOutput, error) {
 | |
| 	m := v.Message().Interface()
 | |
| 	out, err := consumeMessage(b, m, wtyp, opts)
 | |
| 	return v, out, err
 | |
| }
 | |
| 
 | |
| func isInitMessageValue(v protoreflect.Value) error {
 | |
| 	m := v.Message().Interface()
 | |
| 	return proto.CheckInitialized(m)
 | |
| }
 | |
| 
 | |
| var coderMessageValue = valueCoderFuncs{
 | |
| 	size:      sizeMessageValue,
 | |
| 	marshal:   appendMessageValue,
 | |
| 	unmarshal: consumeMessageValue,
 | |
| 	isInit:    isInitMessageValue,
 | |
| 	merge:     mergeMessageValue,
 | |
| }
 | |
| 
 | |
| func sizeGroupValue(v protoreflect.Value, tagsize int, opts marshalOptions) int {
 | |
| 	m := v.Message().Interface()
 | |
| 	return sizeGroup(m, tagsize, opts)
 | |
| }
 | |
| 
 | |
| func appendGroupValue(b []byte, v protoreflect.Value, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	m := v.Message().Interface()
 | |
| 	return appendGroup(b, m, wiretag, opts)
 | |
| }
 | |
| 
 | |
| func consumeGroupValue(b []byte, v protoreflect.Value, num protowire.Number, wtyp protowire.Type, opts unmarshalOptions) (protoreflect.Value, unmarshalOutput, error) {
 | |
| 	m := v.Message().Interface()
 | |
| 	out, err := consumeGroup(b, m, num, wtyp, opts)
 | |
| 	return v, out, err
 | |
| }
 | |
| 
 | |
| var coderGroupValue = valueCoderFuncs{
 | |
| 	size:      sizeGroupValue,
 | |
| 	marshal:   appendGroupValue,
 | |
| 	unmarshal: consumeGroupValue,
 | |
| 	isInit:    isInitMessageValue,
 | |
| 	merge:     mergeMessageValue,
 | |
| }
 | |
| 
 | |
| func makeGroupFieldCoder(fd protoreflect.FieldDescriptor, ft reflect.Type) pointerCoderFuncs {
 | |
| 	num := fd.Number()
 | |
| 	if mi := getMessageInfo(ft); mi != nil {
 | |
| 		funcs := pointerCoderFuncs{
 | |
| 			size:      sizeGroupType,
 | |
| 			marshal:   appendGroupType,
 | |
| 			unmarshal: consumeGroupType,
 | |
| 			merge:     mergeMessage,
 | |
| 		}
 | |
| 		if needsInitCheck(mi.Desc) {
 | |
| 			funcs.isInit = isInitMessageInfo
 | |
| 		}
 | |
| 		return funcs
 | |
| 	} else {
 | |
| 		return pointerCoderFuncs{
 | |
| 			size: func(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return sizeGroup(m, f.tagsize, opts)
 | |
| 			},
 | |
| 			marshal: func(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return appendGroup(b, m, f.wiretag, opts)
 | |
| 			},
 | |
| 			unmarshal: func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 				mp := p.AsValueOf(ft).Elem()
 | |
| 				if mp.IsNil() {
 | |
| 					mp.Set(reflect.New(ft.Elem()))
 | |
| 				}
 | |
| 				return consumeGroup(b, asMessage(mp), num, wtyp, opts)
 | |
| 			},
 | |
| 			isInit: func(p pointer, f *coderFieldInfo) error {
 | |
| 				m := asMessage(p.AsValueOf(ft).Elem())
 | |
| 				return proto.CheckInitialized(m)
 | |
| 			},
 | |
| 			merge: mergeMessage,
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func sizeGroupType(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 	return 2*f.tagsize + f.mi.sizePointer(p.Elem(), opts)
 | |
| }
 | |
| 
 | |
| func appendGroupType(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 	b = protowire.AppendVarint(b, f.wiretag) // start group
 | |
| 	b, err := f.mi.marshalAppendPointer(b, p.Elem(), opts)
 | |
| 	b = protowire.AppendVarint(b, f.wiretag+1) // end group
 | |
| 	return b, err
 | |
| }
 | |
| 
 | |
| func consumeGroupType(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.StartGroupType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	if p.Elem().IsNil() {
 | |
| 		p.SetPointer(pointerOfValue(reflect.New(f.mi.GoReflectType.Elem())))
 | |
| 	}
 | |
| 	return f.mi.unmarshalPointer(b, p.Elem(), f.num, opts)
 | |
| }
 | |
| 
 | |
| func sizeGroup(m proto.Message, tagsize int, opts marshalOptions) int {
 | |
| 	return 2*tagsize + opts.Options().Size(m)
 | |
| }
 | |
| 
 | |
| func appendGroup(b []byte, m proto.Message, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	b = protowire.AppendVarint(b, wiretag) // start group
 | |
| 	b, err := opts.Options().MarshalAppend(b, m)
 | |
| 	b = protowire.AppendVarint(b, wiretag+1) // end group
 | |
| 	return b, err
 | |
| }
 | |
| 
 | |
| func consumeGroup(b []byte, m proto.Message, num protowire.Number, wtyp protowire.Type, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.StartGroupType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	b, n := protowire.ConsumeGroup(num, b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     b,
 | |
| 		Message: m.ProtoReflect(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func makeMessageSliceFieldCoder(fd protoreflect.FieldDescriptor, ft reflect.Type) pointerCoderFuncs {
 | |
| 	if mi := getMessageInfo(ft); mi != nil {
 | |
| 		funcs := pointerCoderFuncs{
 | |
| 			size:      sizeMessageSliceInfo,
 | |
| 			marshal:   appendMessageSliceInfo,
 | |
| 			unmarshal: consumeMessageSliceInfo,
 | |
| 			merge:     mergeMessageSlice,
 | |
| 		}
 | |
| 		if needsInitCheck(mi.Desc) {
 | |
| 			funcs.isInit = isInitMessageSliceInfo
 | |
| 		}
 | |
| 		return funcs
 | |
| 	}
 | |
| 	return pointerCoderFuncs{
 | |
| 		size: func(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 			return sizeMessageSlice(p, ft, f.tagsize, opts)
 | |
| 		},
 | |
| 		marshal: func(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 			return appendMessageSlice(b, p, f.wiretag, ft, opts)
 | |
| 		},
 | |
| 		unmarshal: func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 			return consumeMessageSlice(b, p, ft, wtyp, opts)
 | |
| 		},
 | |
| 		isInit: func(p pointer, f *coderFieldInfo) error {
 | |
| 			return isInitMessageSlice(p, ft)
 | |
| 		},
 | |
| 		merge: mergeMessageSlice,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func sizeMessageSliceInfo(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 	s := p.PointerSlice()
 | |
| 	n := 0
 | |
| 	for _, v := range s {
 | |
| 		n += protowire.SizeBytes(f.mi.sizePointer(v, opts)) + f.tagsize
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendMessageSliceInfo(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 	s := p.PointerSlice()
 | |
| 	var err error
 | |
| 	for _, v := range s {
 | |
| 		b = protowire.AppendVarint(b, f.wiretag)
 | |
| 		siz := f.mi.sizePointer(v, opts)
 | |
| 		b = protowire.AppendVarint(b, uint64(siz))
 | |
| 		before := len(b)
 | |
| 		b, err = f.mi.marshalAppendPointer(b, v, opts)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		if measuredSize := len(b) - before; siz != measuredSize {
 | |
| 			return nil, errors.MismatchedSizeCalculation(siz, measuredSize)
 | |
| 		}
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeMessageSliceInfo(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.BytesType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	v, n := protowire.ConsumeBytes(b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	m := reflect.New(f.mi.GoReflectType.Elem()).Interface()
 | |
| 	mp := pointerOfIface(m)
 | |
| 	o, err := f.mi.unmarshalPointer(v, mp, 0, opts)
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	p.AppendPointerSlice(mp)
 | |
| 	out.n = n
 | |
| 	out.initialized = o.initialized
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func isInitMessageSliceInfo(p pointer, f *coderFieldInfo) error {
 | |
| 	s := p.PointerSlice()
 | |
| 	for _, v := range s {
 | |
| 		if err := f.mi.checkInitializedPointer(v); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func sizeMessageSlice(p pointer, goType reflect.Type, tagsize int, opts marshalOptions) int {
 | |
| 	mopts := opts.Options()
 | |
| 	s := p.PointerSlice()
 | |
| 	n := 0
 | |
| 	for _, v := range s {
 | |
| 		m := asMessage(v.AsValueOf(goType.Elem()))
 | |
| 		n += protowire.SizeBytes(mopts.Size(m)) + tagsize
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendMessageSlice(b []byte, p pointer, wiretag uint64, goType reflect.Type, opts marshalOptions) ([]byte, error) {
 | |
| 	mopts := opts.Options()
 | |
| 	s := p.PointerSlice()
 | |
| 	var err error
 | |
| 	for _, v := range s {
 | |
| 		m := asMessage(v.AsValueOf(goType.Elem()))
 | |
| 		b = protowire.AppendVarint(b, wiretag)
 | |
| 		siz := mopts.Size(m)
 | |
| 		b = protowire.AppendVarint(b, uint64(siz))
 | |
| 		before := len(b)
 | |
| 		b, err = mopts.MarshalAppend(b, m)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		if measuredSize := len(b) - before; siz != measuredSize {
 | |
| 			return nil, errors.MismatchedSizeCalculation(siz, measuredSize)
 | |
| 		}
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeMessageSlice(b []byte, p pointer, goType reflect.Type, wtyp protowire.Type, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.BytesType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	v, n := protowire.ConsumeBytes(b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	mp := reflect.New(goType.Elem())
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     v,
 | |
| 		Message: asMessage(mp).ProtoReflect(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	p.AppendPointerSlice(pointerOfValue(mp))
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func isInitMessageSlice(p pointer, goType reflect.Type) error {
 | |
| 	s := p.PointerSlice()
 | |
| 	for _, v := range s {
 | |
| 		m := asMessage(v.AsValueOf(goType.Elem()))
 | |
| 		if err := proto.CheckInitialized(m); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // Slices of messages
 | |
| 
 | |
| func sizeMessageSliceValue(listv protoreflect.Value, tagsize int, opts marshalOptions) int {
 | |
| 	mopts := opts.Options()
 | |
| 	list := listv.List()
 | |
| 	n := 0
 | |
| 	for i, llen := 0, list.Len(); i < llen; i++ {
 | |
| 		m := list.Get(i).Message().Interface()
 | |
| 		n += protowire.SizeBytes(mopts.Size(m)) + tagsize
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendMessageSliceValue(b []byte, listv protoreflect.Value, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	list := listv.List()
 | |
| 	mopts := opts.Options()
 | |
| 	for i, llen := 0, list.Len(); i < llen; i++ {
 | |
| 		m := list.Get(i).Message().Interface()
 | |
| 		b = protowire.AppendVarint(b, wiretag)
 | |
| 		siz := mopts.Size(m)
 | |
| 		b = protowire.AppendVarint(b, uint64(siz))
 | |
| 		before := len(b)
 | |
| 		var err error
 | |
| 		b, err = mopts.MarshalAppend(b, m)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		if measuredSize := len(b) - before; siz != measuredSize {
 | |
| 			return nil, errors.MismatchedSizeCalculation(siz, measuredSize)
 | |
| 		}
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeMessageSliceValue(b []byte, listv protoreflect.Value, _ protowire.Number, wtyp protowire.Type, opts unmarshalOptions) (_ protoreflect.Value, out unmarshalOutput, err error) {
 | |
| 	list := listv.List()
 | |
| 	if wtyp != protowire.BytesType {
 | |
| 		return protoreflect.Value{}, out, errUnknown
 | |
| 	}
 | |
| 	v, n := protowire.ConsumeBytes(b)
 | |
| 	if n < 0 {
 | |
| 		return protoreflect.Value{}, out, errDecode
 | |
| 	}
 | |
| 	m := list.NewElement()
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     v,
 | |
| 		Message: m.Message(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return protoreflect.Value{}, out, err
 | |
| 	}
 | |
| 	list.Append(m)
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return listv, out, nil
 | |
| }
 | |
| 
 | |
| func isInitMessageSliceValue(listv protoreflect.Value) error {
 | |
| 	list := listv.List()
 | |
| 	for i, llen := 0, list.Len(); i < llen; i++ {
 | |
| 		m := list.Get(i).Message().Interface()
 | |
| 		if err := proto.CheckInitialized(m); err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| var coderMessageSliceValue = valueCoderFuncs{
 | |
| 	size:      sizeMessageSliceValue,
 | |
| 	marshal:   appendMessageSliceValue,
 | |
| 	unmarshal: consumeMessageSliceValue,
 | |
| 	isInit:    isInitMessageSliceValue,
 | |
| 	merge:     mergeMessageListValue,
 | |
| }
 | |
| 
 | |
| func sizeGroupSliceValue(listv protoreflect.Value, tagsize int, opts marshalOptions) int {
 | |
| 	mopts := opts.Options()
 | |
| 	list := listv.List()
 | |
| 	n := 0
 | |
| 	for i, llen := 0, list.Len(); i < llen; i++ {
 | |
| 		m := list.Get(i).Message().Interface()
 | |
| 		n += 2*tagsize + mopts.Size(m)
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendGroupSliceValue(b []byte, listv protoreflect.Value, wiretag uint64, opts marshalOptions) ([]byte, error) {
 | |
| 	list := listv.List()
 | |
| 	mopts := opts.Options()
 | |
| 	for i, llen := 0, list.Len(); i < llen; i++ {
 | |
| 		m := list.Get(i).Message().Interface()
 | |
| 		b = protowire.AppendVarint(b, wiretag) // start group
 | |
| 		var err error
 | |
| 		b, err = mopts.MarshalAppend(b, m)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		b = protowire.AppendVarint(b, wiretag+1) // end group
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeGroupSliceValue(b []byte, listv protoreflect.Value, num protowire.Number, wtyp protowire.Type, opts unmarshalOptions) (_ protoreflect.Value, out unmarshalOutput, err error) {
 | |
| 	list := listv.List()
 | |
| 	if wtyp != protowire.StartGroupType {
 | |
| 		return protoreflect.Value{}, out, errUnknown
 | |
| 	}
 | |
| 	b, n := protowire.ConsumeGroup(num, b)
 | |
| 	if n < 0 {
 | |
| 		return protoreflect.Value{}, out, errDecode
 | |
| 	}
 | |
| 	m := list.NewElement()
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     b,
 | |
| 		Message: m.Message(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return protoreflect.Value{}, out, err
 | |
| 	}
 | |
| 	list.Append(m)
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return listv, out, nil
 | |
| }
 | |
| 
 | |
| var coderGroupSliceValue = valueCoderFuncs{
 | |
| 	size:      sizeGroupSliceValue,
 | |
| 	marshal:   appendGroupSliceValue,
 | |
| 	unmarshal: consumeGroupSliceValue,
 | |
| 	isInit:    isInitMessageSliceValue,
 | |
| 	merge:     mergeMessageListValue,
 | |
| }
 | |
| 
 | |
| func makeGroupSliceFieldCoder(fd protoreflect.FieldDescriptor, ft reflect.Type) pointerCoderFuncs {
 | |
| 	num := fd.Number()
 | |
| 	if mi := getMessageInfo(ft); mi != nil {
 | |
| 		funcs := pointerCoderFuncs{
 | |
| 			size:      sizeGroupSliceInfo,
 | |
| 			marshal:   appendGroupSliceInfo,
 | |
| 			unmarshal: consumeGroupSliceInfo,
 | |
| 			merge:     mergeMessageSlice,
 | |
| 		}
 | |
| 		if needsInitCheck(mi.Desc) {
 | |
| 			funcs.isInit = isInitMessageSliceInfo
 | |
| 		}
 | |
| 		return funcs
 | |
| 	}
 | |
| 	return pointerCoderFuncs{
 | |
| 		size: func(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 			return sizeGroupSlice(p, ft, f.tagsize, opts)
 | |
| 		},
 | |
| 		marshal: func(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 			return appendGroupSlice(b, p, f.wiretag, ft, opts)
 | |
| 		},
 | |
| 		unmarshal: func(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 			return consumeGroupSlice(b, p, num, wtyp, ft, opts)
 | |
| 		},
 | |
| 		isInit: func(p pointer, f *coderFieldInfo) error {
 | |
| 			return isInitMessageSlice(p, ft)
 | |
| 		},
 | |
| 		merge: mergeMessageSlice,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func sizeGroupSlice(p pointer, messageType reflect.Type, tagsize int, opts marshalOptions) int {
 | |
| 	mopts := opts.Options()
 | |
| 	s := p.PointerSlice()
 | |
| 	n := 0
 | |
| 	for _, v := range s {
 | |
| 		m := asMessage(v.AsValueOf(messageType.Elem()))
 | |
| 		n += 2*tagsize + mopts.Size(m)
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendGroupSlice(b []byte, p pointer, wiretag uint64, messageType reflect.Type, opts marshalOptions) ([]byte, error) {
 | |
| 	s := p.PointerSlice()
 | |
| 	var err error
 | |
| 	for _, v := range s {
 | |
| 		m := asMessage(v.AsValueOf(messageType.Elem()))
 | |
| 		b = protowire.AppendVarint(b, wiretag) // start group
 | |
| 		b, err = opts.Options().MarshalAppend(b, m)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		b = protowire.AppendVarint(b, wiretag+1) // end group
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeGroupSlice(b []byte, p pointer, num protowire.Number, wtyp protowire.Type, goType reflect.Type, opts unmarshalOptions) (out unmarshalOutput, err error) {
 | |
| 	if wtyp != protowire.StartGroupType {
 | |
| 		return out, errUnknown
 | |
| 	}
 | |
| 	b, n := protowire.ConsumeGroup(num, b)
 | |
| 	if n < 0 {
 | |
| 		return out, errDecode
 | |
| 	}
 | |
| 	mp := reflect.New(goType.Elem())
 | |
| 	o, err := opts.Options().UnmarshalState(protoiface.UnmarshalInput{
 | |
| 		Buf:     b,
 | |
| 		Message: asMessage(mp).ProtoReflect(),
 | |
| 	})
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	p.AppendPointerSlice(pointerOfValue(mp))
 | |
| 	out.n = n
 | |
| 	out.initialized = o.Flags&protoiface.UnmarshalInitialized != 0
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func sizeGroupSliceInfo(p pointer, f *coderFieldInfo, opts marshalOptions) int {
 | |
| 	s := p.PointerSlice()
 | |
| 	n := 0
 | |
| 	for _, v := range s {
 | |
| 		n += 2*f.tagsize + f.mi.sizePointer(v, opts)
 | |
| 	}
 | |
| 	return n
 | |
| }
 | |
| 
 | |
| func appendGroupSliceInfo(b []byte, p pointer, f *coderFieldInfo, opts marshalOptions) ([]byte, error) {
 | |
| 	s := p.PointerSlice()
 | |
| 	var err error
 | |
| 	for _, v := range s {
 | |
| 		b = protowire.AppendVarint(b, f.wiretag) // start group
 | |
| 		b, err = f.mi.marshalAppendPointer(b, v, opts)
 | |
| 		if err != nil {
 | |
| 			return b, err
 | |
| 		}
 | |
| 		b = protowire.AppendVarint(b, f.wiretag+1) // end group
 | |
| 	}
 | |
| 	return b, nil
 | |
| }
 | |
| 
 | |
| func consumeGroupSliceInfo(b []byte, p pointer, wtyp protowire.Type, f *coderFieldInfo, opts unmarshalOptions) (unmarshalOutput, error) {
 | |
| 	if wtyp != protowire.StartGroupType {
 | |
| 		return unmarshalOutput{}, errUnknown
 | |
| 	}
 | |
| 	m := reflect.New(f.mi.GoReflectType.Elem()).Interface()
 | |
| 	mp := pointerOfIface(m)
 | |
| 	out, err := f.mi.unmarshalPointer(b, mp, f.num, opts)
 | |
| 	if err != nil {
 | |
| 		return out, err
 | |
| 	}
 | |
| 	p.AppendPointerSlice(mp)
 | |
| 	return out, nil
 | |
| }
 | |
| 
 | |
| func asMessage(v reflect.Value) protoreflect.ProtoMessage {
 | |
| 	if m, ok := v.Interface().(protoreflect.ProtoMessage); ok {
 | |
| 		return m
 | |
| 	}
 | |
| 	return legacyWrapMessage(v).Interface()
 | |
| }
 |