-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
tagger.go
133 lines (106 loc) · 3.07 KB
/
tagger.go
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2016-present Datadog, Inc.
package testtagger
import (
"fmt"
"os"
"runtime"
"strings"
"unsafe"
common "github.com/DataDog/datadog-agent/rtloader/test/common"
"github.com/DataDog/datadog-agent/rtloader/test/helpers"
)
/*
#include "rtloader_mem.h"
#include "datadog_agent_rtloader.h"
extern char **Tags(char*, int);
static void initTaggerTests(rtloader_t *rtloader) {
set_cgo_free_cb(rtloader, _free);
set_tags_cb(rtloader, Tags);
}
*/
import "C"
var (
rtloader *C.rtloader_t
tmpfile *os.File
)
func setUp() error {
// Initialize memory tracking
helpers.InitMemoryTracker()
rtloader = (*C.rtloader_t)(common.GetRtLoader())
if rtloader == nil {
return fmt.Errorf("make failed")
}
var err error
tmpfile, err = os.CreateTemp("", "testout")
if err != nil {
return err
}
C.initTaggerTests(rtloader)
// Updates sys.path so testing Check can be found
C.add_python_path(rtloader, C.CString("../python"))
if ok := C.init(rtloader); ok != 1 {
return fmt.Errorf("`init` failed: %s", C.GoString(C.get_error(rtloader)))
}
return nil
}
func tearDown() {
os.Remove(tmpfile.Name())
}
func run(call string) (string, error) {
tmpfile.Truncate(0)
code := (*C.char)(helpers.TrackedCString(fmt.Sprintf(`
try:
import tagger
%s
except Exception as e:
with open(r'%s', 'w') as f:
f.write("{}: {}\n".format(type(e).__name__, e))
`, call, tmpfile.Name())))
defer C._free(unsafe.Pointer(code))
runtime.LockOSThread()
state := C.ensure_gil(rtloader)
ret := C.run_simple_string(rtloader, code) == 1
C.release_gil(rtloader, state)
runtime.UnlockOSThread()
if !ret {
return "", fmt.Errorf("`run_simple_string` errored")
}
output, err := os.ReadFile(tmpfile.Name())
return strings.TrimSpace(string(output)), err
}
//revive:disable
//export Tags
func Tags(id *C.char, cardinality C.int) **C.char {
goID := C.GoString(id)
if goID != "base" {
return nil
}
length := 4
cTags := C._malloc(C.size_t(length) * C.size_t(unsafe.Sizeof(uintptr(0))))
// convert the C array to a Go Array so we can index it
indexTag := (*[1<<29 - 1]*C.char)(cTags)[:length:length]
indexTag[length-1] = nil
// dummy value for each cardinality value
switch cardinality {
case C.DATADOG_AGENT_RTLOADER_TAGGER_LOW:
indexTag[0] = (*C.char)(helpers.TrackedCString("a"))
indexTag[1] = (*C.char)(helpers.TrackedCString("b"))
indexTag[2] = (*C.char)(helpers.TrackedCString("c"))
case C.DATADOG_AGENT_RTLOADER_TAGGER_HIGH:
indexTag[0] = (*C.char)(helpers.TrackedCString("A"))
indexTag[1] = (*C.char)(helpers.TrackedCString("B"))
indexTag[2] = (*C.char)(helpers.TrackedCString("C"))
case C.DATADOG_AGENT_RTLOADER_TAGGER_ORCHESTRATOR:
indexTag[0] = (*C.char)(helpers.TrackedCString("1"))
indexTag[1] = (*C.char)(helpers.TrackedCString("2"))
indexTag[2] = (*C.char)(helpers.TrackedCString("3"))
default:
C._free(cTags)
return nil
}
return (**C.char)(cTags)
}
//revive:enable