1 /* Copyright (C) 2009 Pierre-Marc Fournier, Philippe Proulx-Barrette
3 * This library is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU Lesser General Public
5 * License as published by the Free Software Foundation; either
6 * version 2.1 of the License, or (at your option) any later version.
8 * This library is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * Lesser General Public License for more details.
13 * You should have received a copy of the GNU Lesser General Public
14 * License along with this library; if not, write to the Free Software
15 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
28 #include "ust/ustcmd.h"
31 pid_t
*ustcmd_get_online_pids(void)
33 struct dirent
*dirent
;
35 unsigned int ret_size
= 1 * sizeof(pid_t
), i
= 0;
37 dir
= opendir(SOCK_DIR
);
42 pid_t
*ret
= (pid_t
*) malloc(ret_size
);
44 while ((dirent
= readdir(dir
))) {
45 if (!strcmp(dirent
->d_name
, ".") ||
46 !strcmp(dirent
->d_name
, "..")) {
51 if (dirent
->d_type
!= DT_DIR
&&
52 !!strcmp(dirent
->d_name
, "ustd")) {
54 sscanf(dirent
->d_name
, "%u", (unsigned int *) &ret
[i
]);
55 if (pid_is_online(ret
[i
])) {
56 ret_size
+= sizeof(pid_t
);
57 ret
= (pid_t
*) realloc(ret
, ret_size
);
63 ret
[i
] = 0; /* Array end */
76 * Sets marker state (USTCMD_MS_ON or USTCMD_MS_OFF).
78 * @param mn Marker name
79 * @param state Marker's new state
80 * @param pid Traced process ID
81 * @return 0 if successful, or errors {USTCMD_ERR_GEN, USTCMD_ERR_ARG}
83 int ustcmd_set_marker_state(const char *mn
, int state
, pid_t pid
)
85 char *cmd_str
[] = {"disable_marker", "enable_marker"};
90 return USTCMD_ERR_ARG
;
93 if (asprintf(&cmd
, "%s %s", cmd_str
[state
], mn
) < 0) {
94 ERR("ustcmd_set_marker_state : asprintf failed (%s %s)",
96 return USTCMD_ERR_GEN
;
99 result
= ustcmd_send_cmd(cmd
, pid
, NULL
);
102 return USTCMD_ERR_GEN
;
110 * Set subbuffer size.
112 * @param channel_size Channel name and size
113 * @param pid Traced process ID
114 * @return 0 if successful, or error
116 int ustcmd_set_subbuf_size(const char *channel_size
, pid_t pid
)
121 if (asprintf(&cmd
, "%s %s", "set_subbuf_size", channel_size
) < 0) {
122 ERR("ustcmd_set_subbuf_size : asprintf failed (set_subbuf_size %s)",
127 result
= ustcmd_send_cmd(cmd
, pid
, NULL
);
140 * @param channel_num Channel name and num
141 * @param pid Traced process ID
142 * @return 0 if successful, or error
144 int ustcmd_set_subbuf_num(const char *channel_size
, pid_t pid
)
149 if (asprintf(&cmd
, "%s %s", "set_subbuf_num", channel_size
) < 0) {
150 ERR("ustcmd_set_subbuf_num : asprintf failed (set_subbuf_num %s",
155 result
= ustcmd_send_cmd(cmd
, pid
, NULL
);
166 * Get subbuffer size.
168 * @param channel Channel name
169 * @param pid Traced process ID
170 * @return subbuf size if successful, or error
172 int ustcmd_get_subbuf_size(const char *channel
, pid_t pid
)
177 /* format: channel_cpu */
178 if (asprintf(&cmd
, "%s %s_0", "get_subbuf_size", channel
) < 0) {
179 ERR("ustcmd_get_subbuf_size : asprintf failed (get_subbuf_size, %s_0",
184 result
= ustcmd_send_cmd(cmd
, pid
, &reply
);
190 result
= atoi(reply
);
199 * @param channel Channel name
200 * @param pid Traced process ID
201 * @return subbuf cnf if successful, or error
203 int ustcmd_get_subbuf_num(const char *channel
, pid_t pid
)
208 /* format: channel_cpu */
209 if (asprintf(&cmd
, "%s %s_0", "get_n_subbufs", channel
) < 0) {
210 ERR("ustcmd_get_subbuf_num : asprintf failed (get_n_subbufs, %s_0",
215 result
= ustcmd_send_cmd(cmd
, pid
, &reply
);
221 result
= atoi(reply
);
228 * Destroys an UST trace according to a PID.
230 * @param pid Traced process ID
231 * @return 0 if successful, or error USTCMD_ERR_GEN
233 int ustcmd_destroy_trace(pid_t pid
)
237 result
= ustcmd_send_cmd("trace_destroy", pid
, NULL
);
239 return USTCMD_ERR_GEN
;
246 * Starts an UST trace (and setups it) according to a PID.
248 * @param pid Traced process ID
249 * @return 0 if successful, or error USTCMD_ERR_GEN
251 int ustcmd_setup_and_start(pid_t pid
)
255 result
= ustcmd_send_cmd("start", pid
, NULL
);
257 return USTCMD_ERR_GEN
;
264 * Creates an UST trace according to a PID.
266 * @param pid Traced process ID
267 * @return 0 if successful, or error USTCMD_ERR_GEN
269 int ustcmd_create_trace(pid_t pid
)
273 result
= ustcmd_send_cmd("trace_create", pid
, NULL
);
275 return USTCMD_ERR_GEN
;
282 * Starts an UST trace according to a PID.
284 * @param pid Traced process ID
285 * @return 0 if successful, or error USTCMD_ERR_GEN
287 int ustcmd_start_trace(pid_t pid
)
291 result
= ustcmd_send_cmd("trace_start", pid
, NULL
);
293 return USTCMD_ERR_GEN
;
300 * Alloc an UST trace according to a PID.
302 * @param pid Traced process ID
303 * @return 0 if successful, or error USTCMD_ERR_GEN
305 int ustcmd_alloc_trace(pid_t pid
)
309 result
= ustcmd_send_cmd("trace_alloc", pid
, NULL
);
311 return USTCMD_ERR_GEN
;
318 * Stops an UST trace according to a PID.
320 * @param pid Traced process ID
321 * @return 0 if successful, or error USTCMD_ERR_GEN
323 int ustcmd_stop_trace(pid_t pid
)
327 result
= ustcmd_send_cmd("trace_stop", pid
, NULL
);
329 return USTCMD_ERR_GEN
;
336 * Counts newlines ('\n') in a string.
338 * @param str String to search in
339 * @return Total newlines count
341 unsigned int ustcmd_count_nl(const char *str
)
343 unsigned int i
= 0, tot
= 0;
345 while (str
[i
] != '\0') {
346 if (str
[i
] == '\n') {
356 * Frees a CMSF array.
358 * @param cmsf CMSF array to free
359 * @return 0 if successful, or error USTCMD_ERR_ARG
361 int ustcmd_free_cmsf(struct marker_status
*cmsf
)
364 return USTCMD_ERR_ARG
;
368 while (cmsf
[i
].channel
!= NULL
) {
369 free(cmsf
[i
].channel
);
370 free(cmsf
[i
].marker
);
380 * Gets channel/marker/state/format string for a given PID.
382 * @param cmsf Pointer to CMSF array to be filled (callee allocates, caller
383 * frees with `ustcmd_free_cmsf')
384 * @param pid Targeted PID
385 * @return 0 if successful, or -1 on error
387 int ustcmd_get_cmsf(struct marker_status
**cmsf
, const pid_t pid
)
389 char *big_str
= NULL
;
391 struct marker_status
*tmp_cmsf
= NULL
;
392 unsigned int i
= 0, cmsf_ind
= 0;
397 result
= ustcmd_send_cmd("list_markers", pid
, &big_str
);
403 ERR("error while getting markers list");
407 tmp_cmsf
= (struct marker_status
*) malloc(sizeof(struct marker_status
) *
408 (ustcmd_count_nl(big_str
) + 1));
409 if (tmp_cmsf
== NULL
) {
413 /* Parse received reply string (format: "[chan]/[mark] [st] [fs]"): */
414 while (big_str
[i
] != '\0') {
417 sscanf(big_str
+ i
, "marker: %a[^/]/%a[^ ] %c %a[^\n]",
418 &tmp_cmsf
[cmsf_ind
].channel
,
419 &tmp_cmsf
[cmsf_ind
].marker
,
421 &tmp_cmsf
[cmsf_ind
].fs
);
422 tmp_cmsf
[cmsf_ind
].state
= (state
== USTCMD_MS_CHR_ON
?
423 USTCMD_MS_ON
: USTCMD_MS_OFF
); /* Marker state */
425 while (big_str
[i
] != '\n') {
426 ++i
; /* Go to next '\n' */
428 ++i
; /* Skip current pointed '\n' */
431 tmp_cmsf
[cmsf_ind
].channel
= NULL
;
432 tmp_cmsf
[cmsf_ind
].marker
= NULL
;
433 tmp_cmsf
[cmsf_ind
].fs
= NULL
;
444 * @param sock_path Socket path
445 * @param pid Traced process ID
446 * @return 0 if successful, or error
448 int ustcmd_set_sock_path(const char *sock_path
, pid_t pid
)
453 if (asprintf(&cmd
, "%s %s", "set_sock_path", sock_path
) < 0) {
454 ERR("ustcmd_set_sock_path : asprintf failed (set_sock_path, %s",
459 result
= ustcmd_send_cmd(cmd
, pid
, NULL
);
462 return USTCMD_ERR_GEN
;
472 * @param sock_path Pointer to where the socket path will be returned
473 * @param pid Traced process ID
474 * @return 0 if successful, or error
476 int ustcmd_get_sock_path(char **sock_path
, pid_t pid
)
481 if (asprintf(&cmd
, "%s", "get_sock_path") < 0) {
482 ERR("ustcmd_get_sock_path : asprintf failed");
483 return USTCMD_ERR_GEN
;
486 result
= ustcmd_send_cmd(cmd
, pid
, &reply
);
489 return USTCMD_ERR_GEN
;
497 int ustcmd_force_switch(pid_t pid
)
501 result
= ustcmd_send_cmd("force_switch", pid
, NULL
);
503 return USTCMD_ERR_GEN
;
510 * Sends a given command to a traceable process
512 * @param cmd Null-terminated command to send
513 * @param pid Targeted PID
514 * @param reply Pointer to string to be filled with a reply string (must
515 * be NULL if no reply is needed for the given command).
516 * @return -1 if not successful, 0 on EOT, 1 on success
519 int ustcmd_send_cmd(const char *cmd
, const pid_t pid
, char **reply
)
521 struct ustcomm_connection conn
;
524 if (ustcomm_connect_app(pid
, &conn
)) {
525 ERR("could not connect to PID %u", (unsigned int) pid
);
529 retval
= ustcomm_send_request(&conn
, cmd
, reply
);
531 ustcomm_close_app(&conn
);
This page took 0.04166 seconds and 4 git commands to generate.