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
|
// Copyright 2016 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.
//go:build (linux && amd64) || (freebsd && amd64)
// +build linux,amd64 freebsd,amd64
package main
// Test that we can collect a lot of colliding profiling signals from
// an external C thread. This used to fail when built with the race
// detector, because a call of the predeclared function copy was
// turned into a call to runtime.slicecopy, which is not marked nosplit.
/*
#include <signal.h>
#include <stdint.h>
#include <pthread.h>
#include <sched.h>
struct cgoTracebackArg {
uintptr_t context;
uintptr_t sigContext;
uintptr_t* buf;
uintptr_t max;
};
static int raceprofCount;
// We want a bunch of different profile stacks that collide in the
// hash table maintained in runtime/cpuprof.go. This code knows the
// size of the hash table (1 << 10) and knows that the hash function
// is simply multiplicative.
void raceprofTraceback(void* parg) {
struct cgoTracebackArg* arg = (struct cgoTracebackArg*)(parg);
raceprofCount++;
arg->buf[0] = raceprofCount * (1 << 10);
arg->buf[1] = 0;
}
static void* raceprofThread(void* p) {
int i;
for (i = 0; i < 100; i++) {
pthread_kill(pthread_self(), SIGPROF);
sched_yield();
}
return 0;
}
void runRaceprofThread() {
pthread_t tid;
pthread_create(&tid, 0, raceprofThread, 0);
pthread_join(tid, 0);
}
*/
import "C"
import (
"bytes"
"fmt"
"runtime"
"runtime/pprof"
"unsafe"
)
func init() {
register("CgoRaceprof", CgoRaceprof)
}
func CgoRaceprof() {
runtime.SetCgoTraceback(0, unsafe.Pointer(C.raceprofTraceback), nil, nil)
var buf bytes.Buffer
pprof.StartCPUProfile(&buf)
C.runRaceprofThread()
fmt.Println("OK")
}
|
