Changes malloc to zmalloc

[ust.git] / libustd / libustd.c

diff --git a/libustd/libustd.c b/libustd/libustd.c
index 3c934536c6785e6d2519b37cc11886dc2351cc8e..49351318df20ee4ce69f84bfbd630646fee9ea76 100644 (file)
--- a/libustd/libustd.c
+++ b/libustd/libustd.c
@@ -29,7 +29,8 @@
 #include <errno.h>
 #include <assert.h>
 
-#include "libustd.h"
+#include <ust/ustd.h>
+#include "lowlevel.h"
 #include "usterr.h"
 #include "ustcomm.h"
 
@@ -58,7 +59,7 @@ int get_subbuffer(struct buffer_info *buf)
        int result;
 
        asprintf(&send_msg, "get_subbuffer %s", buf->name);
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       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;
@@ -118,7 +119,7 @@ int put_subbuffer(struct buffer_info *buf)
        int result;
 
        asprintf(&send_msg, "put_subbuffer %s %ld", buf->name, buf->consumed_old);
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       result = ustcomm_send_request(buf->conn, send_msg, &received_msg);
        if(result < 0 && (errno == ECONNRESET || errno == EPIPE)) {
                retval = PUT_SUBBUF_DIED;
                goto end;
@@ -190,17 +191,24 @@ struct buffer_info *connect_buffer(struct libustd_instance *instance, pid_t pid,
        int result;
        struct shmid_ds shmds;
 
-       buf = (struct buffer_info *) malloc(sizeof(struct buffer_info));
+       buf = (struct buffer_info *) zmalloc(sizeof(struct buffer_info));
        if(buf == NULL) {
                ERR("add_buffer: insufficient memory");
                return NULL;
        }
 
+       buf->conn = malloc(sizeof(struct ustcomm_connection));
+       if(buf->conn == NULL) {
+               ERR("add_buffer: insufficient memory");
+               free(buf);
+               return NULL;
+       }
+
        buf->name = bufname;
        buf->pid = pid;
 
        /* connect to app */
-       result = ustcomm_connect_app(buf->pid, &buf->conn);
+       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;
@@ -208,7 +216,7 @@ struct buffer_info *connect_buffer(struct libustd_instance *instance, pid_t pid,
 
        /* get pidunique */
        asprintf(&send_msg, "get_pidunique");
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       result = ustcomm_send_request(buf->conn, send_msg, &received_msg);
        free(send_msg);
        if(result == -1) {
                ERR("problem in ustcomm_send_request(get_pidunique)");
@@ -228,7 +236,7 @@ struct buffer_info *connect_buffer(struct libustd_instance *instance, pid_t pid,
 
        /* get shmid */
        asprintf(&send_msg, "get_shmid %s", buf->name);
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       result = ustcomm_send_request(buf->conn, send_msg, &received_msg);
        free(send_msg);
        if(result == -1) {
                ERR("problem in ustcomm_send_request(get_shmid)");
@@ -248,7 +256,7 @@ struct buffer_info *connect_buffer(struct libustd_instance *instance, pid_t pid,
 
        /* get n_subbufs */
        asprintf(&send_msg, "get_n_subbufs %s", buf->name);
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       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)");
@@ -268,7 +276,7 @@ struct buffer_info *connect_buffer(struct libustd_instance *instance, pid_t pid,
 
        /* get subbuf size */
        asprintf(&send_msg, "get_subbuf_size %s", buf->name);
-       result = ustcomm_send_request(&buf->conn, send_msg, &received_msg);
+       result = ustcomm_send_request(buf->conn, send_msg, &received_msg);
        free(send_msg);
        if(result == -1) {
                ERR("problem in ustcomm_send_request(get_subbuf_size)");
@@ -328,7 +336,7 @@ static void destroy_buffer(struct libustd_callbacks *callbacks,
 {
        int result;
 
-       result = ustcomm_close_app(&buf->conn);
+       result = ustcomm_close_app(buf->conn);
        if(result == -1) {
                WARN("problem calling ustcomm_close_app");
        }
@@ -346,6 +354,7 @@ static void destroy_buffer(struct libustd_callbacks *callbacks,
        if(callbacks->on_close_buffer)
                callbacks->on_close_buffer(callbacks, buf);
 
+       free(buf->conn);
        free(buf);
 }
 
@@ -486,7 +495,7 @@ int start_consuming_buffer(
 
        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 = (struct consumer_thread_args *) zmalloc(sizeof(struct consumer_thread_args));
 
        args->pid = pid;
        args->bufname = strdup(bufname);
@@ -523,7 +532,7 @@ int libustd_start_instance(struct libustd_instance *instance)
                char *recvbuf;
 
                /* check for requests on our public socket */
-               result = ustcomm_ustd_recv_message(&instance->comm, &recvbuf, NULL, timeout);
+               result = ustcomm_ustd_recv_message(instance->comm, &recvbuf, NULL, timeout);
                if(result == -1 && errno == EINTR) {
                        /* Caught signal */
                }
@@ -586,10 +595,11 @@ int libustd_start_instance(struct libustd_instance *instance)
 void libustd_delete_instance(struct libustd_instance *instance)
 {
        if(instance->is_init)
-               ustcomm_fini_ustd(&instance->comm);
+               ustcomm_fini_ustd(instance->comm);
 
        pthread_mutex_destroy(&instance->mutex);
        free(instance->sock_path);
+       free(instance->comm);
        free(instance);
 }
 
@@ -638,10 +648,16 @@ struct libustd_instance *libustd_new_instance(
        struct libustd_callbacks *callbacks, char *sock_path)
 {
        struct libustd_instance *instance =
-               malloc(sizeof(struct libustd_instance));
+               zmalloc(sizeof(struct libustd_instance));
        if(!instance)
                return NULL;
 
+       instance->comm = malloc(sizeof(struct ustcomm_ustd));
+       if(!instance->comm) {
+               free(instance);
+               return NULL;
+       }
+
        instance->callbacks = callbacks;
        instance->quit_program = 0;
        instance->is_init = 0;
@@ -659,7 +675,7 @@ struct libustd_instance *libustd_new_instance(
 int libustd_init_instance(struct libustd_instance *instance)
 {
        int result;
-       result = ustcomm_init_ustd(&instance->comm, instance->sock_path);
+       result = ustcomm_init_ustd(instance->comm, instance->sock_path);
        if(result == -1) {
                ERR("failed to initialize socket");
                return 1;