summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-mangler/v2/pointer.go
blob: d732b2f67d4b8b0c072b7fdb6287568bae48c777 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package mangler

import (
	"reflect"
	"unsafe"

	"codeberg.org/gruf/go-xunsafe"
)

// derefPointerType returns a Mangler capable of dereferencing
// and formatting the given pointer type currently in TypeIter{}.
// note this will fetch a sub-Mangler for resulting value type.
func derefPointerType(t xunsafe.TypeIter) Mangler {
	var derefs int
	var indirects int64
	rtype := t.Type
	flags := t.Flag

	// Iteratively dereference pointer types.
	for rtype.Kind() == reflect.Pointer {

		// Only if this is actual indirect memory do we
		// perform a derefence, otherwise we just skip over
		// and increase the dereference indicator, i.e. '1'.
		if flags&xunsafe.Reflect_flagIndir != 0 {
			indirects |= 1 << derefs
		}
		derefs++

		// Get next elem type.
		rtype = rtype.Elem()

		// Get next set of dereferenced element type flags.
		flags = xunsafe.ReflectPointerElemFlags(flags, rtype)
	}

	// Ensure this is a reasonable number of derefs.
	if derefs > 4*int(unsafe.Sizeof(indirects)) {
		return nil
	}

	// Wrap value as TypeIter.
	vt := t.Child(rtype, flags)

	// Get value mangler.
	fn := loadOrGet(vt)
	if fn == nil {
		return nil
	}

	return func(buf []byte, ptr unsafe.Pointer) []byte {
		for i := 0; i < derefs; i++ {
			switch {
			case indirects&1<<i == 0:
				// No dereference needed.
				buf = append(buf, '1')

			case ptr == nil:
				// Nil value, return here.
				buf = append(buf, '0')
				return buf

			default:
				// Further deref ptr.
				buf = append(buf, '1')
				ptr = *(*unsafe.Pointer)(ptr)
			}
		}

		if ptr == nil {
			// Final nil val check.
			buf = append(buf, '0')
			return buf
		}

		// Mangle fully deref'd.
		buf = append(buf, '1')
		buf = fn(buf, ptr)
		return buf
	}
}