Line data Source code
1 : #define _GNU_SOURCE
2 : #include "../../shared/fd_config.h"
3 : #include "../../shared/fd_sys_util.h"
4 : #include "../../../disco/metrics/fd_metrics.h"
5 :
6 : #include <errno.h>
7 : #include <stdio.h>
8 : #include <unistd.h>
9 : #include <sys/wait.h>
10 : #include <sys/random.h>
11 :
12 : static int record_pid;
13 :
14 : static void
15 0 : parent_signal( int sig ) {
16 0 : FD_LOG_NOTICE(( "Received signal %s\n", fd_io_strsignal( sig ) ));
17 0 : if( FD_LIKELY( record_pid ) ) {
18 0 : if( FD_UNLIKELY( -1==kill( record_pid, SIGINT ) ) ) FD_LOG_ERR(( "kill() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
19 0 : }
20 0 : }
21 :
22 : static void
23 0 : install_parent_signals( void ) {
24 0 : struct sigaction sa = {
25 0 : .sa_handler = parent_signal,
26 0 : .sa_flags = 0,
27 0 : };
28 0 : if( FD_UNLIKELY( sigaction( SIGTERM, &sa, NULL ) ) )
29 0 : FD_LOG_ERR(( "sigaction(SIGTERM) failed (%i-%s)", errno, fd_io_strerror( errno ) ));
30 0 : if( FD_UNLIKELY( sigaction( SIGINT, &sa, NULL ) ) )
31 0 : FD_LOG_ERR(( "sigaction(SIGINT) failed (%i-%s)", errno, fd_io_strerror( errno ) ));
32 0 : }
33 :
34 : void
35 : flame_cmd_perm( args_t * args FD_PARAM_UNUSED,
36 : fd_cap_chk_t * chk,
37 0 : config_t const * config FD_PARAM_UNUSED ) {
38 0 : fd_cap_chk_root( chk, "flame", "read system performance counters with `/usr/bin/perf`" );
39 0 : }
40 :
41 : void
42 : flame_cmd_args( int * pargc,
43 : char *** pargv,
44 0 : args_t * args ) {
45 :
46 0 : if( FD_UNLIKELY( !*pargc ) ) FD_LOG_ERR(( "usage: flame [all|tile|tile:idx|agave]" ));
47 0 : strncpy( args->flame.name, **pargv, sizeof( args->flame.name ) - 1 );
48 :
49 0 : (*pargc)--;
50 0 : (*pargv)++;
51 0 : }
52 :
53 : void
54 : flame_cmd_fn( args_t * args,
55 0 : config_t * config ) {
56 0 : install_parent_signals();
57 :
58 0 : fd_topo_join_workspaces( &config->topo, FD_SHMEM_JOIN_MODE_READ_ONLY );
59 0 : fd_topo_fill( &config->topo );
60 :
61 0 : ulong tile_cnt = 0UL;
62 0 : ulong tile_idxs[ 128UL ];
63 :
64 0 : int whole_process = 0;
65 0 : if( FD_UNLIKELY( !strcmp( "all", args->flame.name ) ) ) {
66 0 : FD_TEST( config->topo.tile_cnt<sizeof(tile_idxs)/sizeof(tile_idxs[0]) );
67 0 : for( ulong i=0UL; i<config->topo.tile_cnt; i++ ) {
68 0 : tile_idxs[ tile_cnt ] = i;
69 0 : tile_cnt++;
70 0 : }
71 0 : } else if( FD_UNLIKELY( !strcmp( "agave", args->flame.name ) ) ) {
72 : /* Find the bank tile so we can get the Agave PID */
73 0 : ulong bank_tile_idx = fd_topo_find_tile( &config->topo, "bank", 0UL );
74 0 : if( FD_UNLIKELY( bank_tile_idx==ULONG_MAX ) ) FD_LOG_ERR(( "tile `bank` not found" ));
75 0 : whole_process = 1;
76 0 : tile_idxs[ 0 ] = bank_tile_idx;
77 0 : tile_cnt = 1UL;
78 0 : } else {
79 0 : char * sep = strchr( args->flame.name, ':' );
80 :
81 0 : ulong tile_idx;
82 0 : if( FD_UNLIKELY( !sep ) ) {
83 0 : tile_idx = fd_topo_find_tile( &config->topo, args->flame.name, 0UL );
84 0 : } else {
85 0 : char * endptr;
86 0 : *sep = '\0';
87 0 : ulong kind_id = strtoul( sep+1, &endptr, 10 );
88 0 : if( FD_UNLIKELY( *endptr!='\0' || kind_id==ULONG_MAX ) ) FD_LOG_ERR(( "invalid tile kind id provided `%s`", sep+1 ));
89 0 : tile_idx = fd_topo_find_tile( &config->topo, args->flame.name, kind_id );
90 0 : }
91 :
92 0 : if( FD_UNLIKELY( tile_idx==ULONG_MAX ) ) FD_LOG_ERR(( "tile `%s` not found", args->flame.name ));
93 0 : tile_idxs[ 0 ] = tile_idx;
94 0 : tile_cnt = 1UL;
95 0 : }
96 :
97 0 : char threads[ 4096 ] = {0};
98 0 : ulong len = 0UL;
99 0 : for( ulong i=0UL; i<tile_cnt; i++ ) {
100 0 : if( FD_LIKELY( i!=0UL ) ) {
101 0 : FD_TEST( fd_cstr_printf_check( threads+len, sizeof(threads)-len, NULL, "," ) );
102 0 : len += 1UL;
103 0 : }
104 :
105 0 : ulong tid = fd_metrics_tile( config->topo.tiles[ tile_idxs[ i ] ].metrics )[ FD_METRICS_GAUGE_TILE_TID_OFF ];
106 0 : ulong pid = fd_metrics_tile( config->topo.tiles[ tile_idxs[ i ] ].metrics )[ FD_METRICS_GAUGE_TILE_PID_OFF ];
107 :
108 0 : FD_TEST( pid<=INT_MAX );
109 0 : if( FD_UNLIKELY( -1==kill( (int)pid, 0 ) ) ) {
110 0 : if( FD_UNLIKELY( errno==ESRCH ) ) FD_LOG_ERR(( "tile %s:%lu is not running", config->topo.tiles[ i ].name, config->topo.tiles[ i ].kind_id ));
111 0 : else FD_LOG_ERR(( "kill() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
112 0 : }
113 :
114 0 : ulong arg_len;
115 0 : FD_TEST( fd_cstr_printf_check( threads+len, sizeof(threads)-len, &arg_len, "%lu", fd_ulong_if( whole_process, pid, tid ) ) );
116 0 : len += arg_len;
117 0 : }
118 0 : FD_TEST( len<sizeof(threads) );
119 :
120 0 : FD_LOG_NOTICE(( "/usr/bin/perf script record flamegraph -F 99 -%c %s && /usr/bin/perf script report flamegraph", fd_char_if( whole_process, 'p', 't' ), threads ));
121 :
122 0 : record_pid = fork();
123 0 : if( FD_UNLIKELY( -1==record_pid ) ) FD_LOG_ERR(( "fork() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
124 0 : if( FD_LIKELY( !record_pid ) ) {
125 0 : char * args[ 11 ] = {
126 0 : "/usr/bin/perf",
127 0 : "script",
128 0 : "record",
129 0 : "flamegraph",
130 0 : "-F",
131 0 : "99",
132 0 : whole_process ? "-p" : "-t",
133 0 : threads,
134 0 : NULL,
135 0 : };
136 0 : if( FD_UNLIKELY( -1==execve( "/usr/bin/perf", (char * const *)args, NULL ) ) ) FD_LOG_ERR(( "execve() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
137 0 : }
138 :
139 0 : FD_LOG_NOTICE(( "Perf collection running. Send SIGINT / Crl+C to stop." ));
140 :
141 0 : for(;;) {
142 0 : int wstatus;
143 0 : int exited_pid = waitpid( -1, &wstatus, 0 );
144 0 : if( FD_UNLIKELY( -1==exited_pid ) ) {
145 0 : if( FD_LIKELY( errno==EAGAIN || errno==EINTR ) ) continue;
146 0 : FD_LOG_ERR(( "waitpid() failed (%d-%s)", errno, fd_io_strerror( errno ) ));
147 0 : }
148 :
149 0 : int graceful_exit = !WIFEXITED( wstatus ) && WTERMSIG( wstatus )==SIGINT;
150 0 : if( FD_UNLIKELY( !graceful_exit ) ) {
151 0 : if( FD_UNLIKELY( !WIFEXITED( wstatus ) ) ) FD_LOG_ERR(( "perf record exited unexpectedly with signal %d (%s)", WTERMSIG( wstatus ), fd_io_strsignal( WTERMSIG( wstatus ) ) ));
152 0 : if( FD_UNLIKELY( WEXITSTATUS( wstatus ) ) ) FD_LOG_ERR(( "perf record exited unexpectedly with code %d", WEXITSTATUS( wstatus ) ));
153 0 : }
154 0 : break;
155 0 : }
156 :
157 0 : int report_pid = fork();
158 0 : if( FD_UNLIKELY( -1==report_pid ) ) FD_LOG_ERR(( "fork() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
159 0 : if( FD_LIKELY( !report_pid ) ) {
160 0 : char * args[ 7 ] = {
161 0 : "/usr/bin/perf",
162 0 : "script",
163 0 : "report",
164 0 : "flamegraph",
165 0 : NULL,
166 0 : };
167 0 : if( FD_UNLIKELY( -1==execve( "/usr/bin/perf", (char * const *)args, NULL ) ) ) FD_LOG_ERR(( "execve() failed (%i-%s)", errno, fd_io_strerror( errno ) ));
168 0 : }
169 :
170 0 : for(;;) {
171 0 : int wstatus;
172 0 : int exited_pid = waitpid( -1, &wstatus, 0 );
173 0 : if( FD_UNLIKELY( -1==exited_pid ) ) {
174 0 : if( FD_LIKELY( errno==EAGAIN || errno==EINTR ) ) continue;
175 0 : FD_LOG_ERR(( "waitpid() failed (%d-%s)", errno, fd_io_strerror( errno ) ));
176 0 : }
177 :
178 0 : if( FD_UNLIKELY( !WIFEXITED( wstatus ) ) ) FD_LOG_ERR(( "perf report exited unexpectedly with signal %d (%s)", WTERMSIG( wstatus ), fd_io_strsignal( WTERMSIG( wstatus ) ) ));
179 0 : if( FD_UNLIKELY( WEXITSTATUS( wstatus ) ) ) FD_LOG_ERR(( "perf report exited unexpectedly with code %d", WEXITSTATUS( wstatus ) ));
180 0 : break;
181 0 : }
182 :
183 0 : fd_sys_util_exit_group( 0 );
184 0 : }
|
