* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
+#define _GNU_SOURCE
#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <fcntl.h>
#include <string.h>
#include <dirent.h>
-#
#include "ustcomm.h"
#include "ustcmd.h"
-#define _GNU_SOURCE
-
-pid_t* ustcmd_get_online_pids(void) {
- struct dirent* dirent;
- DIR* dir;
- char* ping_res;
+pid_t *ustcmd_get_online_pids(void)
+{
+ struct dirent *dirent;
+ DIR *dir;
+ unsigned int ret_size = 1 * sizeof(pid_t), i = 0;
dir = opendir(SOCK_DIR);
if (!dir) {
return NULL;
}
- unsigned int ret_size = 1 * sizeof(pid_t), i = 0;
- pid_t* ret = (pid_t*) malloc(ret_size);
+ pid_t *ret = (pid_t *) malloc(ret_size);
- while (dirent = readdir(dir)) {
+ while ((dirent = readdir(dir))) {
if (!strcmp(dirent->d_name, ".") ||
!strcmp(dirent->d_name, "..")) {
if (dirent->d_type != DT_DIR &&
!!strcmp(dirent->d_name, "ustd")) {
- sscanf(dirent->d_name, "%u", (unsigned int*) &ret[i]);
+ sscanf(dirent->d_name, "%u", (unsigned int *) &ret[i]);
if (pid_is_online(ret[i])) {
ret_size += sizeof(pid_t);
- ret = (pid_t*) realloc(ret, ret_size);
+ ret = (pid_t *) realloc(ret, ret_size);
++i;
}
}
}
- ret[i] = 0; // Array end
+ ret[i] = 0; /* Array end */
- if (ret[0] == 0) { // No PID at all..
+ if (ret[0] == 0) {
+ /* No PID at all */
free(ret);
return NULL;
}
* @param pid Traced process ID
* @return 0 if successful, or errors {USTCMD_ERR_GEN, USTCMD_ERR_ARG}
*/
-int ustcmd_set_marker_state(const char* mn, int state, pid_t pid) {
+int ustcmd_set_marker_state(const char *mn, int state, pid_t pid)
+{
+ char *cmd_str [] = {"disable_marker", "enable_marker"};
+ char *cmd;
+ int result;
+
if (mn == NULL) {
return USTCMD_ERR_ARG;
}
- char* cmd_str [] = {"disable_marker", "enable_marker"};
- char* cmd;
asprintf(&cmd, "%s %s", cmd_str[state], mn);
- int tres;
- if (tres = ustcmd_shoot(cmd, pid, NULL)) {
+ result = ustcmd_send_cmd(cmd, pid, NULL);
+ if (result) {
free(cmd);
return USTCMD_ERR_GEN;
}
* @param pid Traced process ID
* @return 0 if successful, or error USTCMD_ERR_GEN
*/
-int ustcmd_destroy_trace(pid_t pid) {
- int tres;
+int ustcmd_destroy_trace(pid_t pid)
+{
+ int result;
- if (tres = ustcmd_shoot("destroy", pid, NULL)) {
+ result = ustcmd_send_cmd("trace_destroy", pid, NULL);
+ if (result) {
return USTCMD_ERR_GEN;
}
* @param pid Traced process ID
* @return 0 if successful, or error USTCMD_ERR_GEN
*/
-int ustcmd_setup_and_start(pid_t pid) {
- int tres;
+int ustcmd_setup_and_start(pid_t pid)
+{
+ int result;
+
+ result = ustcmd_send_cmd("start", pid, NULL);
+ if (result) {
+ return USTCMD_ERR_GEN;
+ }
+
+ return 0;
+}
+
+/**
+ * Creates an UST trace according to a PID.
+ *
+ * @param pid Traced process ID
+ * @return 0 if successful, or error USTCMD_ERR_GEN
+ */
+int ustcmd_create_trace(pid_t pid)
+{
+ int result;
- if (tres = ustcmd_shoot("start", pid, NULL)) {
+ result = ustcmd_send_cmd("trace_create", pid, NULL);
+ if (result) {
return USTCMD_ERR_GEN;
}
* @param pid Traced process ID
* @return 0 if successful, or error USTCMD_ERR_GEN
*/
-int ustcmd_start_trace(pid_t pid) {
- int tres;
+int ustcmd_start_trace(pid_t pid)
+{
+ int result;
- if (tres = ustcmd_shoot("trace_start", pid, NULL)) {
+ result = ustcmd_send_cmd("trace_start", pid, NULL);
+ if (result) {
return USTCMD_ERR_GEN;
}
* @param pid Traced process ID
* @return 0 if successful, or error USTCMD_ERR_GEN
*/
-int ustcmd_stop_trace(pid_t pid) {
- int tres;
+int ustcmd_stop_trace(pid_t pid)
+{
+ int result;
- if (tres = ustcmd_shoot("trace_stop", pid, NULL)) {
+ result = ustcmd_send_cmd("trace_stop", pid, NULL);
+ if (result) {
return USTCMD_ERR_GEN;
}
* @param str String to search in
* @return Total newlines count
*/
-unsigned int ustcmd_count_nl(const char* str) {
+unsigned int ustcmd_count_nl(const char *str)
+{
unsigned int i = 0, tot = 0;
while (str[i] != '\0') {
* @param cmsf CMSF array to free
* @return 0 if successful, or error USTCMD_ERR_ARG
*/
-int ustcmd_free_cmsf(struct USTcmd_cmsf* cmsf) {
+int ustcmd_free_cmsf(struct marker_status *cmsf)
+{
if (cmsf == NULL) {
return USTCMD_ERR_ARG;
}
* @param pid Targeted PID
* @return 0 if successful, or errors {USTCMD_ERR_ARG, USTCMD_ERR_GEN}
*/
-int ustcmd_get_cmsf(struct USTcmd_cmsf** cmsf, const pid_t pid) {
+int ustcmd_get_cmsf(struct marker_status **cmsf, const pid_t pid)
+{
+ char *big_str = NULL;
+ int result;
+ struct marker_status *tmp_cmsf = NULL;
+ unsigned int i = 0, cmsf_ind = 0;
+
if (cmsf == NULL) {
return USTCMD_ERR_ARG;
}
- char* big_str = NULL;
- int tres;
-
- if (tres = ustcmd_shoot("list_markers", pid, &big_str)) {
+ result = ustcmd_send_cmd("list_markers", pid, &big_str);
+ if (result) {
return USTCMD_ERR_GEN;
}
return USTCMD_ERR_GEN;
}
- struct USTcmd_cmsf* tmp_cmsf = NULL;
- tmp_cmsf = (struct USTcmd_cmsf*) malloc(sizeof(struct USTcmd_cmsf) *
+ tmp_cmsf = (struct marker_status *) malloc(sizeof(struct marker_status) *
(ustcmd_count_nl(big_str) + 1));
if (tmp_cmsf == NULL) {
return USTCMD_ERR_GEN;
}
- // Parse received reply string (format: "[chan]/[mark] [st] [fs]"):
- unsigned int i = 0, cur_st, cmsf_ind = 0;
+ /* Parse received reply string (format: "[chan]/[mark] [st] [fs]"): */
while (big_str[i] != '\0') {
- /* RAW METHOD (REPLACED BY SSCANF):
- cur_st = i; // Set current start at beginning of current line
- while (big_str[i] != '/') {
- ++i; // Go to next '/'
- }
- tmp_cmsf[cmsf_ind].channel =
- strndup(big_str + cur_st, i - cur_st);
-
- ++i; // Go after '/'
- cur_st = i; // Set current start at beginning of marker name
- while (big_str[i] != ' ') {
- ++i; // Go to next ' '
- }
- tmp_cmsf[cmsf_ind].marker =
- strndup(big_str + cur_st, i - cur_st);
-
- ++i; // Go after ' '
- tmp_cmsf[cmsf_ind].state = (big_str[i] == USTCMD_MS_CHR_ON ?
- USTCMD_MS_ON : USTCMD_MS_OFF); // Marker state
-
- i += 2; // Go to format string (after state and another ' ')
- cur_st = i; // Set current start at beginning of format string
- while (big_str[i] != '\n') {
- ++i; // Go to next '\n'
- }
- tmp_cmsf[cmsf_ind].fs =
- strndup(big_str + cur_st, i - cur_st);
- */
-
char state;
- sscanf(big_str + i, "%a[^/]/%a[^ ] %c %a[^\n]",
+
+ sscanf(big_str + i, "marker: %a[^/]/%a[^ ] %c %a[^\n]",
&tmp_cmsf[cmsf_ind].channel,
&tmp_cmsf[cmsf_ind].marker,
&state,
&tmp_cmsf[cmsf_ind].fs);
tmp_cmsf[cmsf_ind].state = (state == USTCMD_MS_CHR_ON ?
- USTCMD_MS_ON : USTCMD_MS_OFF); // Marker state
+ USTCMD_MS_ON : USTCMD_MS_OFF); /* Marker state */
while (big_str[i] != '\n') {
- ++i; // Go to next '\n'
+ ++i; /* Go to next '\n' */
}
- ++i; // Skip current pointed '\n'
+ ++i; /* Skip current pointed '\n' */
++cmsf_ind;
}
tmp_cmsf[cmsf_ind].channel = NULL;
* be NULL if no reply is needed for the given command).
* @return 0 if successful, or errors {USTCMD_ERR_ARG, USTCMD_ERR_CONN}
*/
-int ustcmd_shoot(const char* cmd, const pid_t pid, char** reply) {
+
+int ustcmd_send_cmd(const char *cmd, const pid_t pid, char **reply)
+{
+ struct ustcomm_connection conn;
+
if (cmd == NULL) {
return USTCMD_ERR_ARG;
}
- struct ustcomm_connection conn;
if (ustcomm_connect_app(pid, &conn)) {
- fprintf(stderr, "ustcmd_shoot: could not connect to PID %u\n",
+ fprintf(stderr, "ustcmd_send_cmd: could not connect to PID %u\n",
(unsigned int) pid);
return USTCMD_ERR_CONN;
}
return 0;
}
-