X-Git-Url: http://git.lttng.org/?p=ust.git;a=blobdiff_plain;f=ustd%2Fustd.c;h=c0f4c596eca0ce9a12165a1d090a3c655db0fcdf;hp=046ac0c0835a551bcc15273b596226d1db757518;hb=7032c7d350bfa093db533713df8a14df37360bdc;hpb=9c9048271f2e13c3f4b9d19220d72d421f63bf6a diff --git a/ustd/ustd.c b/ustd/ustd.c index 046ac0c..c0f4c59 100644 --- a/ustd/ustd.c +++ b/ustd/ustd.c @@ -1,4 +1,5 @@ /* Copyright (C) 2009 Pierre-Marc Fournier + * 2010 Alexis Halle * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public @@ -22,7 +23,6 @@ #include #include #include -#include #include #include @@ -32,165 +32,39 @@ #include #include -#include "ustd.h" +#include "ust/ustd.h" #include "usterr.h" -#include "ustcomm.h" - -/* return value: 0 = subbuffer is finished, it won't produce data anymore - * 1 = got subbuffer successfully - * <0 = error - */ - -#define GET_SUBBUF_OK 1 -#define GET_SUBBUF_DONE 0 -#define GET_SUBBUF_DIED 2 - -#define PUT_SUBBUF_OK 1 -#define PUT_SUBBUF_DIED 0 -#define PUT_SUBBUF_PUSHED 2 -#define PUT_SUBBUF_DONE 3 char *sock_path=NULL; char *trace_path=NULL; int daemon_mode = 0; char *pidfile = NULL; -/* Number of active buffers and the mutex to protect it. */ -int active_buffers = 0; -pthread_mutex_t active_buffers_mutex = PTHREAD_MUTEX_INITIALIZER; -/* Whether a request to end the program was received. */ -volatile sig_atomic_t terminate_req = 0; +struct libustd_instance *instance; -int get_subbuffer(struct buffer_info *buf) -{ - char *send_msg=NULL; - char *received_msg=NULL; - char *rep_code=NULL; - int retval; - int result; - - asprintf(&send_msg, "get_subbuffer %s", buf->name); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - if((result == -1 && (errno == ECONNRESET || errno == EPIPE)) || result == 0) { - DBG("app died while being traced"); - retval = GET_SUBBUF_DIED; - goto end; - } - else if(result < 0) { - ERR("get_subbuffer: ustcomm_send_request failed"); - retval = -1; - goto end; - } - - result = sscanf(received_msg, "%as %ld", &rep_code, &buf->consumed_old); - if(result != 2 && result != 1) { - ERR("unable to parse response to get_subbuffer"); - retval = -1; - free(received_msg); - goto end_rep; - } - - if(!strcmp(rep_code, "OK")) { - DBG("got subbuffer %s", buf->name); - retval = GET_SUBBUF_OK; - } - else if(nth_token_is(received_msg, "END", 0) == 1) { - retval = GET_SUBBUF_DONE; - goto end_rep; - } - else if(!strcmp(received_msg, "NOTFOUND")) { - DBG("For buffer %s, the trace was not found. This likely means it was destroyed by the user.", buf->name); - retval = GET_SUBBUF_DONE; - goto end_rep; - } - else { - DBG("error getting subbuffer %s", buf->name); - retval = -1; - } - - /* FIMXE: free correctly the stuff */ -end_rep: - if(rep_code) - free(rep_code); -end: - if(send_msg) - free(send_msg); - if(received_msg) - free(received_msg); - - return retval; -} +struct buffer_info_local { + /* output file */ + int file_fd; + /* the offset we must truncate to, to unput the last subbuffer */ + off_t previous_offset; +}; -int put_subbuffer(struct buffer_info *buf) +static int write_pidfile(const char *file_name, pid_t pid) { - char *send_msg=NULL; - char *received_msg=NULL; - char *rep_code=NULL; - int retval; - int result; - - asprintf(&send_msg, "put_subbuffer %s %ld", buf->name, buf->consumed_old); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - if(result < 0 && (errno == ECONNRESET || errno == EPIPE)) { - retval = PUT_SUBBUF_DIED; - goto end; - } - else if(result < 0) { - ERR("put_subbuffer: send_message failed"); - retval = -1; - goto end; - } - else if(result == 0) { - /* Program seems finished. However this might not be - * the last subbuffer that has to be collected. - */ - retval = PUT_SUBBUF_DIED; - goto end; - } - - result = sscanf(received_msg, "%as", &rep_code); - if(result != 1) { - ERR("unable to parse response to put_subbuffer"); - retval = -1; - goto end_rep; - } + FILE *pidfp; - if(!strcmp(rep_code, "OK")) { - DBG("subbuffer put %s", buf->name); - retval = PUT_SUBBUF_OK; - } - else if(!strcmp(received_msg, "NOTFOUND")) { - DBG("For buffer %s, the trace was not found. This likely means it was destroyed by the user.", buf->name); - /* However, maybe this was not the last subbuffer. So - * we return the program died. - */ - retval = PUT_SUBBUF_DIED; - goto end_rep; - } - else { - DBG("put_subbuffer: received error, we were pushed"); - retval = PUT_SUBBUF_PUSHED; - goto end_rep; + pidfp = fopen(file_name, "w"); + if(!pidfp) { + PERROR("fopen (%s)", file_name); + WARN("killing child process"); + return -1; } -end_rep: - if(rep_code) - free(rep_code); - -end: - if(send_msg) - free(send_msg); - if(received_msg) - free(received_msg); + fprintf(pidfp, "%d\n", pid); - return retval; -} + fclose(pidfp); -void decrement_active_buffers(void *arg) -{ - pthread_mutex_lock(&active_buffers_mutex); - active_buffers--; - pthread_mutex_unlock(&active_buffers_mutex); + return 0; } int create_dir_if_needed(char *dir) @@ -207,152 +81,100 @@ int create_dir_if_needed(char *dir) return 0; } -int is_directory(const char *dir) +int unwrite_last_subbuffer(struct buffer_info *buf) { int result; - struct stat st; + struct buffer_info_local *buf_local = buf->user_data; - result = stat(dir, &st); + result = ftruncate(buf_local->file_fd, buf_local->previous_offset); if(result == -1) { - PERROR("stat"); - return 0; + PERROR("ftruncate"); + return -1; } - if(!S_ISDIR(st.st_mode)) { - return 0; + result = lseek(buf_local->file_fd, buf_local->previous_offset, SEEK_SET); + if(result == (int)(off_t)-1) { + PERROR("lseek"); + return -1; } - return 1; + return 0; } -struct buffer_info *connect_buffer(pid_t pid, const char *bufname) +int write_current_subbuffer(struct buffer_info *buf) { - struct buffer_info *buf; - char *send_msg; - char *received_msg; int result; - char *tmp; - int fd; - struct shmid_ds shmds; - - buf = (struct buffer_info *) malloc(sizeof(struct buffer_info)); - if(buf == NULL) { - ERR("add_buffer: insufficient memory"); - return NULL; - } + struct buffer_info_local *buf_local = buf->user_data; - buf->name = bufname; - buf->pid = pid; + void *subbuf_mem = buf->mem + (buf->consumed_old & (buf->n_subbufs * buf->subbuf_size-1)); - /* connect to app */ - result = ustcomm_connect_app(buf->pid, &buf->conn); - if(result) { - WARN("unable to connect to process, it probably died before we were able to connect"); - return NULL; - } + size_t cur_sb_size = subbuffer_data_size(subbuf_mem); - /* get pidunique */ - asprintf(&send_msg, "get_pidunique"); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - free(send_msg); - if(result == -1) { - ERR("problem in ustcomm_send_request(get_pidunique)"); - return NULL; - } - if(result == 0) { - goto error; + off_t cur_offset = lseek(buf_local->file_fd, 0, SEEK_CUR); + if(cur_offset == (off_t)-1) { + PERROR("lseek"); + return -1; } - result = sscanf(received_msg, "%lld", &buf->pidunique); - if(result != 1) { - ERR("unable to parse response to get_pidunique"); - return NULL; - } - free(received_msg); - DBG("got pidunique %lld", buf->pidunique); + buf_local->previous_offset = cur_offset; + DBG("previous_offset: %ld", cur_offset); - /* get shmid */ - asprintf(&send_msg, "get_shmid %s", buf->name); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - free(send_msg); + result = patient_write(buf_local->file_fd, subbuf_mem, cur_sb_size); if(result == -1) { - ERR("problem in ustcomm_send_request(get_shmid)"); - return NULL; - } - if(result == 0) { - goto error; + PERROR("write"); + return -1; } - result = sscanf(received_msg, "%d %d", &buf->shmid, &buf->bufstruct_shmid); - if(result != 2) { - ERR("unable to parse response to get_shmid (\"%s\")", received_msg); - return NULL; - } - free(received_msg); - DBG("got shmids %d %d", buf->shmid, buf->bufstruct_shmid); + return 0; +} - /* get n_subbufs */ - asprintf(&send_msg, "get_n_subbufs %s", buf->name); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - free(send_msg); - if(result == -1) { - ERR("problem in ustcomm_send_request(g_n_subbufs)"); - return NULL; - } - if(result == 0) { - goto error; - } +int on_read_subbuffer(struct libustd_callbacks *data, struct buffer_info *buf) +{ + return write_current_subbuffer(buf); +} - result = sscanf(received_msg, "%d", &buf->n_subbufs); - if(result != 1) { - ERR("unable to parse response to get_n_subbufs"); - return NULL; - } - free(received_msg); - DBG("got n_subbufs %d", buf->n_subbufs); +int on_read_partial_subbuffer(struct libustd_callbacks *data, struct buffer_info *buf, + long subbuf_index, unsigned long valid_length) +{ + struct buffer_info_local *buf_local = buf->user_data; + char *tmp; + int result; + unsigned long pad_size; - /* get subbuf size */ - asprintf(&send_msg, "get_subbuf_size %s", buf->name); - result = ustcomm_send_request(&buf->conn, send_msg, &received_msg); - free(send_msg); + result = patient_write(buf_local->file_fd, buf->mem + subbuf_index * buf->subbuf_size, valid_length); if(result == -1) { - ERR("problem in ustcomm_send_request(get_subbuf_size)"); - return NULL; - } - if(result == 0) { - goto error; + ERR("Error writing to buffer file"); + return; } - result = sscanf(received_msg, "%d", &buf->subbuf_size); - if(result != 1) { - ERR("unable to parse response to get_subbuf_size"); - return NULL; + /* pad with empty bytes */ + pad_size = PAGE_ALIGN(valid_length)-valid_length; + if(pad_size) { + tmp = zmalloc(pad_size); + result = patient_write(buf_local->file_fd, tmp, pad_size); + if(result == -1) { + ERR("Error writing to buffer file"); + return; + } + free(tmp); } - free(received_msg); - DBG("got subbuf_size %d", buf->subbuf_size); - /* attach memory */ - buf->mem = shmat(buf->shmid, NULL, 0); - if(buf->mem == (void *) 0) { - PERROR("shmat"); - return NULL; - } - DBG("successfully attached buffer memory"); +} - buf->bufstruct_mem = shmat(buf->bufstruct_shmid, NULL, 0); - if(buf->bufstruct_mem == (void *) 0) { - PERROR("shmat"); - return NULL; - } - DBG("successfully attached buffer bufstruct memory"); +int on_open_buffer(struct libustd_callbacks *data, struct buffer_info *buf) +{ + char *tmp; + int result; + int fd; + struct buffer_info_local *buf_local = + zmalloc(sizeof(struct buffer_info_local)); - /* obtain info on the memory segment */ - result = shmctl(buf->shmid, IPC_STAT, &shmds); - if(result == -1) { - PERROR("shmctl"); - return NULL; + if(!buf_local) { + ERR("could not allocate buffer_info_local struct"); + return 1; } - buf->memlen = shmds.shm_segsz; + + buf->user_data = buf_local; /* open file for output */ if(!trace_path) { @@ -363,7 +185,7 @@ struct buffer_info *connect_buffer(pid_t pid, const char *bufname) result = create_dir_if_needed(USTD_DEFAULT_TRACE_PATH); if(result == -1) { ERR("could not create directory %s", USTD_DEFAULT_TRACE_PATH); - return NULL; + return 1; } trace_path = USTD_DEFAULT_TRACE_PATH; @@ -374,7 +196,7 @@ struct buffer_info *connect_buffer(pid_t pid, const char *bufname) if(result == -1) { ERR("could not create directory %s", tmp); free(tmp); - return NULL; + return 1; } free(tmp); @@ -383,218 +205,67 @@ struct buffer_info *connect_buffer(pid_t pid, const char *bufname) if(result == -1) { PERROR("open"); ERR("failed opening trace file %s", tmp); - return NULL; + return 1; } - buf->file_fd = fd; + buf_local->file_fd = fd; free(tmp); - pthread_mutex_lock(&active_buffers_mutex); - active_buffers++; - pthread_mutex_unlock(&active_buffers_mutex); - - return buf; - -error: - free(buf); - return NULL; -} - -static void destroy_buffer(struct buffer_info *buf) -{ - int result; - - result = ustcomm_close_app(&buf->conn); - if(result == -1) { - WARN("problem calling ustcomm_close_app"); - } - - result = shmdt(buf->mem); - if(result == -1) { - PERROR("shmdt"); - } - - result = shmdt(buf->bufstruct_mem); - if(result == -1) { - PERROR("shmdt"); - } - - result = close(buf->file_fd); - if(result == -1) { - PERROR("close"); - } - - free(buf); -} - -int unwrite_last_subbuffer(struct buffer_info *buf) -{ - int result; - - result = ftruncate(buf->file_fd, buf->previous_offset); - if(result == -1) { - PERROR("ftruncate"); - return -1; - } - - result = lseek(buf->file_fd, buf->previous_offset, SEEK_SET); - if(result == (int)(off_t)-1) { - PERROR("lseek"); - return -1; - } - return 0; } -int write_current_subbuffer(struct buffer_info *buf) +int on_close_buffer(struct libustd_callbacks *data, struct buffer_info *buf) { - int result; - - void *subbuf_mem = buf->mem + (buf->consumed_old & (buf->n_subbufs * buf->subbuf_size-1)); - - size_t cur_sb_size = subbuffer_data_size(subbuf_mem); - - off_t cur_offset = lseek(buf->file_fd, 0, SEEK_CUR); - if(cur_offset == (off_t)-1) { - PERROR("lseek"); - return -1; - } - - buf->previous_offset = cur_offset; - DBG("previous_offset: %ld", cur_offset); - - result = patient_write(buf->file_fd, subbuf_mem, cur_sb_size); + struct buffer_info_local *buf_local = buf->user_data; + int result = close(buf_local->file_fd); + free(buf_local); if(result == -1) { - PERROR("write"); - /* FIXME: maybe drop this trace */ - return -1; + PERROR("close"); } - return 0; } -int consumer_loop(struct buffer_info *buf) +int on_put_error(struct libustd_callbacks *data, struct buffer_info *buf) { - int result; - - pthread_cleanup_push(decrement_active_buffers, NULL); - - for(;;) { - /* get the subbuffer */ - result = get_subbuffer(buf); - if(result == -1) { - ERR("error getting subbuffer"); - continue; - } - else if(result == GET_SUBBUF_DONE) { - /* this is done */ - break; - } - else if(result == GET_SUBBUF_DIED) { - finish_consuming_dead_subbuffer(buf); - break; - } - - /* write data to file */ - write_current_subbuffer(buf); - /* FIXME: handle return value? */ - - /* put the subbuffer */ - result = put_subbuffer(buf); - if(result == -1) { - ERR("unknown error putting subbuffer (channel=%s)", buf->name); - break; - } - else if(result == PUT_SUBBUF_PUSHED) { - ERR("Buffer overflow (channel=%s), reader pushed. This channel will not be usable passed this point.", buf->name); - break; - } - else if(result == PUT_SUBBUF_DIED) { - DBG("application died while putting subbuffer"); - /* Skip the first subbuffer. We are not sure it is trustable - * because the put_subbuffer() did not complete. - */ - unwrite_last_subbuffer(buf); - finish_consuming_dead_subbuffer(buf); - break; - } - else if(result == PUT_SUBBUF_DONE) { - /* Done with this subbuffer */ - /* FIXME: add a case where this branch is used? Upon - * normal trace termination, at put_subbuf time, a - * special last-subbuffer code could be returned by - * the listener. - */ - break; - } - else if(result == PUT_SUBBUF_OK) { - } - } - - DBG("thread for buffer %s is stopping", buf->name); - - /* FIXME: destroy, unalloc... */ - - pthread_cleanup_pop(1); - - return 0; + unwrite_last_subbuffer(buf); } -struct consumer_thread_args { - pid_t pid; - const char *bufname; -}; - -void *consumer_thread(void *arg) +struct libustd_callbacks *new_callbacks() { - struct buffer_info *buf = (struct buffer_info *) arg; - struct consumer_thread_args *args = (struct consumer_thread_args *) arg; + struct libustd_callbacks *callbacks = + zmalloc(sizeof(struct libustd_callbacks)); - DBG("GOT ARGS: pid %d bufname %s", args->pid, args->bufname); - - buf = connect_buffer(args->pid, args->bufname); - if(buf == NULL) { - ERR("failed to connect to buffer"); - goto end; - } + if(!callbacks) + return NULL; - consumer_loop(buf); + callbacks->on_open_buffer = on_open_buffer; + callbacks->on_close_buffer = on_close_buffer; + callbacks->on_read_subbuffer = on_read_subbuffer; + callbacks->on_read_partial_subbuffer = on_read_partial_subbuffer; + callbacks->on_put_error = on_put_error; + callbacks->on_new_thread = NULL; + callbacks->on_close_thread = NULL; + callbacks->on_trace_end = NULL; - free(args->bufname); - destroy_buffer(buf); + return callbacks; - end: - /* bufname is free'd in free_buffer() */ - free(args); - return NULL; } -int start_consuming_buffer(pid_t pid, const char *bufname) +int is_directory(const char *dir) { - pthread_t thr; - struct consumer_thread_args *args; int result; + struct stat st; - DBG("beginning of start_consuming_buffer: args: pid %d bufname %s", pid, bufname); - - args = (struct consumer_thread_args *) malloc(sizeof(struct consumer_thread_args)); - - args->pid = pid; - args->bufname = strdup(bufname); - DBG("beginning2 of start_consuming_buffer: args: pid %d bufname %s", args->pid, args->bufname); - - result = pthread_create(&thr, NULL, consumer_thread, args); + result = stat(dir, &st); if(result == -1) { - ERR("pthread_create failed"); - return -1; + PERROR("stat"); + return 0; } - result = pthread_detach(thr); - if(result == -1) { - ERR("pthread_detach failed"); - return -1; + + if(!S_ISDIR(st.st_mode)) { + return 0; } - DBG("end of start_consuming_buffer: args: pid %d bufname %s", args->pid, args->bufname); - return 0; + return 1; } void usage(void) @@ -666,34 +337,21 @@ int parse_args(int argc, char **argv) void sigterm_handler(int sig) { - terminate_req = 1; -} - -static int write_pidfile(const char *file_name, pid_t pid) -{ - FILE *pidfp; - - pidfp = fopen(file_name, "w"); - if(!pidfp) { - PERROR("fopen (%s)", pidfile); - WARN("killing child process"); - return -1; - } - - fprintf(pidfp, "%d\n", pid); - - fclose(pidfp); - - return 0; + libustd_stop_instance(instance, 0); } int start_ustd(int fd) { - struct ustcomm_ustd ustd; int result; sigset_t sigset; struct sigaction sa; + struct libustd_callbacks *callbacks = new_callbacks(); + if(!callbacks) { + PERROR("new_callbacks"); + return 1; + } + result = sigemptyset(&sigset); if(result == -1) { PERROR("sigemptyset"); @@ -701,7 +359,7 @@ int start_ustd(int fd) } sa.sa_handler = sigterm_handler; sa.sa_mask = sigset; - sa.sa_flags = SA_RESTART; + sa.sa_flags = 0; result = sigaction(SIGTERM, &sa, NULL); if(result == -1) { PERROR("sigaction"); @@ -713,9 +371,15 @@ int start_ustd(int fd) return 1; } - result = ustcomm_init_ustd(&ustd, sock_path); - if(result == -1) { - ERR("failed to initialize socket"); + instance = libustd_new_instance(callbacks, sock_path); + if(!instance) { + ERR("failed to create libustd instance"); + return 1; + } + + result = libustd_init_instance(instance); + if(result) { + ERR("failed to initialize libustd instance"); return 1; } @@ -763,57 +427,9 @@ int start_ustd(int fd) } } - /* app loop */ - for(;;) { - char *recvbuf; - - /* check for requests on our public socket */ - result = ustcomm_ustd_recv_message(&ustd, &recvbuf, NULL, 100); - if(result == -1) { - ERR("error in ustcomm_ustd_recv_message"); - goto loop_end; - } - if(result > 0) { - if(!strncmp(recvbuf, "collect", 7)) { - pid_t pid; - char *bufname; - int result; - - result = sscanf(recvbuf, "%*s %d %50as", &pid, &bufname); - if(result != 2) { - ERR("parsing error: %s", recvbuf); - goto free_bufname; - } - - result = start_consuming_buffer(pid, bufname); - if(result < 0) { - ERR("error in add_buffer"); - goto free_bufname; - } - - free_bufname: - free(bufname); - } - else { - WARN("unknown command: %s", recvbuf); - } - - free(recvbuf); - } - - loop_end: - - if(terminate_req) { - pthread_mutex_lock(&active_buffers_mutex); - if(active_buffers == 0) { - pthread_mutex_unlock(&active_buffers_mutex); - break; - } - pthread_mutex_unlock(&active_buffers_mutex); - } - } + libustd_start_instance(instance); - ustcomm_fini_ustd(&ustd); + free(callbacks); return 0; }