blob: 6060154386b44fdc3e53b30d5575738adbd51651 [file] [log] [blame]
Joe Tsaifa02f4e2018-09-12 16:20:37 -07001// Copyright 2018 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
Joe Tsaibc534a92018-12-18 16:07:48 -08005// +build purego appengine
Joe Tsaifa02f4e2018-09-12 16:20:37 -07006
7package impl
8
9import (
10 "fmt"
11 "reflect"
12)
13
Joe Tsaic0e4bb22019-07-06 13:05:11 -070014const UnsafeEnabled = false
15
Joe Tsaifa02f4e2018-09-12 16:20:37 -070016// offset represents the offset to a struct field, accessible from a pointer.
17// The offset is the field index into a struct.
Joe Tsaic0e4bb22019-07-06 13:05:11 -070018type offset struct {
19 index int
20 export exporter
21}
Joe Tsaifa02f4e2018-09-12 16:20:37 -070022
23// offsetOf returns a field offset for the struct field.
Joe Tsaic0e4bb22019-07-06 13:05:11 -070024func offsetOf(f reflect.StructField, x exporter) offset {
Joe Tsaifa02f4e2018-09-12 16:20:37 -070025 if len(f.Index) != 1 {
26 panic("embedded structs are not supported")
27 }
Joe Tsaic0e4bb22019-07-06 13:05:11 -070028 if f.PkgPath == "" {
29 return offset{index: f.Index[0]} // field is already exported
30 }
31 if x == nil {
32 panic("exporter must be provided for unexported field")
33 }
34 return offset{index: f.Index[0], export: x}
Joe Tsaifa02f4e2018-09-12 16:20:37 -070035}
36
Damien Neilc37adef2019-04-01 13:49:56 -070037// IsValid reports whether the offset is valid.
Joe Tsaic0e4bb22019-07-06 13:05:11 -070038func (f offset) IsValid() bool { return f.index >= 0 }
Damien Neilc37adef2019-04-01 13:49:56 -070039
40// invalidOffset is an invalid field offset.
Joe Tsaic0e4bb22019-07-06 13:05:11 -070041var invalidOffset = offset{index: -1}
Damien Neilc37adef2019-04-01 13:49:56 -070042
43// zeroOffset is a noop when calling pointer.Apply.
Joe Tsaic0e4bb22019-07-06 13:05:11 -070044var zeroOffset = offset{index: 0}
Damien Neilc37adef2019-04-01 13:49:56 -070045
Joe Tsaifa02f4e2018-09-12 16:20:37 -070046// pointer is an abstract representation of a pointer to a struct or field.
47type pointer struct{ v reflect.Value }
48
49// pointerOfValue returns v as a pointer.
50func pointerOfValue(v reflect.Value) pointer {
51 return pointer{v: v}
52}
53
Joe Tsaic6b75612018-09-13 14:24:37 -070054// pointerOfIface returns the pointer portion of an interface.
Joe Tsai6cf80c42018-12-01 04:57:09 -080055func pointerOfIface(v interface{}) pointer {
56 return pointer{v: reflect.ValueOf(v)}
Joe Tsaic6b75612018-09-13 14:24:37 -070057}
58
Joe Tsai6cf80c42018-12-01 04:57:09 -080059// IsNil reports whether the pointer is nil.
60func (p pointer) IsNil() bool {
61 return p.v.IsNil()
62}
63
64// Apply adds an offset to the pointer to derive a new pointer
Joe Tsaifa02f4e2018-09-12 16:20:37 -070065// to a specified field. The current pointer must be pointing at a struct.
Joe Tsai6cf80c42018-12-01 04:57:09 -080066func (p pointer) Apply(f offset) pointer {
Joe Tsaic0e4bb22019-07-06 13:05:11 -070067 if f.export != nil {
68 if v := reflect.ValueOf(f.export(p.v.Interface(), f.index)); v.IsValid() {
69 return pointer{v: v}
70 }
71 }
72 return pointer{v: p.v.Elem().Field(f.index).Addr()}
Joe Tsaifa02f4e2018-09-12 16:20:37 -070073}
74
Joe Tsai6cf80c42018-12-01 04:57:09 -080075// AsValueOf treats p as a pointer to an object of type t and returns the value.
76// It is equivalent to reflect.ValueOf(p.AsIfaceOf(t))
77func (p pointer) AsValueOf(t reflect.Type) reflect.Value {
Damien Neil927aaba2019-05-09 12:18:44 -070078 if got := p.v.Type().Elem(); got != t {
79 panic(fmt.Sprintf("invalid type: got %v, want %v", got, t))
Joe Tsaifa02f4e2018-09-12 16:20:37 -070080 }
81 return p.v
82}
Joe Tsai6cf80c42018-12-01 04:57:09 -080083
84// AsIfaceOf treats p as a pointer to an object of type t and returns the value.
85// It is equivalent to p.AsValueOf(t).Interface()
86func (p pointer) AsIfaceOf(t reflect.Type) interface{} {
87 return p.AsValueOf(t).Interface()
88}
Damien Neilc37adef2019-04-01 13:49:56 -070089
90func (p pointer) Bool() *bool { return p.v.Interface().(*bool) }
91func (p pointer) BoolPtr() **bool { return p.v.Interface().(**bool) }
92func (p pointer) BoolSlice() *[]bool { return p.v.Interface().(*[]bool) }
93func (p pointer) Int32() *int32 { return p.v.Interface().(*int32) }
94func (p pointer) Int32Ptr() **int32 { return p.v.Interface().(**int32) }
95func (p pointer) Int32Slice() *[]int32 { return p.v.Interface().(*[]int32) }
96func (p pointer) Int64() *int64 { return p.v.Interface().(*int64) }
97func (p pointer) Int64Ptr() **int64 { return p.v.Interface().(**int64) }
98func (p pointer) Int64Slice() *[]int64 { return p.v.Interface().(*[]int64) }
99func (p pointer) Uint32() *uint32 { return p.v.Interface().(*uint32) }
100func (p pointer) Uint32Ptr() **uint32 { return p.v.Interface().(**uint32) }
101func (p pointer) Uint32Slice() *[]uint32 { return p.v.Interface().(*[]uint32) }
102func (p pointer) Uint64() *uint64 { return p.v.Interface().(*uint64) }
103func (p pointer) Uint64Ptr() **uint64 { return p.v.Interface().(**uint64) }
104func (p pointer) Uint64Slice() *[]uint64 { return p.v.Interface().(*[]uint64) }
105func (p pointer) Float32() *float32 { return p.v.Interface().(*float32) }
106func (p pointer) Float32Ptr() **float32 { return p.v.Interface().(**float32) }
107func (p pointer) Float32Slice() *[]float32 { return p.v.Interface().(*[]float32) }
108func (p pointer) Float64() *float64 { return p.v.Interface().(*float64) }
109func (p pointer) Float64Ptr() **float64 { return p.v.Interface().(**float64) }
110func (p pointer) Float64Slice() *[]float64 { return p.v.Interface().(*[]float64) }
111func (p pointer) String() *string { return p.v.Interface().(*string) }
112func (p pointer) StringPtr() **string { return p.v.Interface().(**string) }
113func (p pointer) StringSlice() *[]string { return p.v.Interface().(*[]string) }
114func (p pointer) Bytes() *[]byte { return p.v.Interface().(*[]byte) }
115func (p pointer) BytesSlice() *[][]byte { return p.v.Interface().(*[][]byte) }
Joe Tsai89d49632019-06-04 16:20:00 -0700116func (p pointer) Extensions() *map[int32]ExtensionField {
117 return p.v.Interface().(*map[int32]ExtensionField)
Damien Neilc37adef2019-04-01 13:49:56 -0700118}
119
120func (p pointer) Elem() pointer {
121 return pointer{v: p.v.Elem()}
122}
123
124// PointerSlice copies []*T from p as a new []pointer.
125// This behavior differs from the implementation in pointer_unsafe.go.
126func (p pointer) PointerSlice() []pointer {
127 // TODO: reconsider this
128 if p.v.IsNil() {
129 return nil
130 }
131 n := p.v.Elem().Len()
132 s := make([]pointer, n)
133 for i := 0; i < n; i++ {
134 s[i] = pointer{v: p.v.Elem().Index(i)}
135 }
136 return s
137}