Fix: liblttng-ust-fork Makefile flags mismatch
[lttng-ust.git] / liblttng-ust-ctl / ustctl.c
index 32fa2ebc33e92e054a7591435fa9aadffef47009..d443a2978db76d67d96fcb21b0e0353e60c2b0c2 100644 (file)
@@ -2,21 +2,21 @@
  * Copyright (C) 2011 - Julien Desfossez <julien.desfossez@polymtl.ca>
  *                      Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; only version 2
- * of the License.
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; version 2 of the License only.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
+#define _GNU_SOURCE
 #include <string.h>
 #include <lttng/ust-ctl.h>
 #include <lttng/ust-abi.h>
@@ -51,9 +51,8 @@ int ustctl_release_handle(int sock, int handle)
                lum.handle = handle;
                lum.cmd = LTTNG_UST_RELEASE;
                ret = ustcomm_send_app_cmd(sock, &lum, &lur);
-               if (ret < 0) {
+               if (ret)
                        return ret;
-               }
        }
        return 0;
 }
@@ -65,15 +64,20 @@ int ustctl_release_object(int sock, struct lttng_ust_object_data *data)
 {
        int ret;
 
+       if (!data)
+               return -EINVAL;
+
        if (data->shm_fd >= 0) {
                ret = close(data->shm_fd);
                if (ret < 0) {
+                       ret = -errno;
                        return ret;
                }
        }
        if (data->wait_fd >= 0) {
                ret = close(data->wait_fd);
                if (ret < 0) {
+                       ret = -errno;
                        return ret;
                }
        }
@@ -96,14 +100,7 @@ int ustctl_register_done(int sock)
        ret = ustcomm_send_app_cmd(sock, &lum, &lur);
        if (ret)
                return ret;
-       if (lur.ret_code != USTCOMM_OK) {
-               DBG("Return code: %s", ustcomm_get_readable_code(lur.ret_code));
-               goto error;
-       }
        return 0;
-
-error:
-       return -1;
 }
 
 /*
@@ -137,6 +134,9 @@ int ustctl_open_metadata(int sock, int session_handle,
        struct lttng_ust_object_data *metadata_data;
        int ret, err = 0;
 
+       if (!chops || !_metadata_data)
+               return -EINVAL;
+
        metadata_data = malloc(sizeof(*metadata_data));
        if (!metadata_data)
                return -ENOMEM;
@@ -156,17 +156,13 @@ int ustctl_open_metadata(int sock, int session_handle,
                free(metadata_data);
                return ret;
        }
-       if (lur.ret_code != USTCOMM_OK) {
-               free(metadata_data);
-               return lur.ret_code;
-       }
        metadata_data->handle = lur.ret_val;
        DBG("received metadata handle %u", metadata_data->handle);
        metadata_data->memory_map_size = lur.u.channel.memory_map_size;
        /* get shm fd */
        ret = ustcomm_recv_fd(sock);
        if (ret < 0)
-               err = 1;
+               err = ret;
        else
                metadata_data->shm_fd = ret;
        /*
@@ -176,7 +172,7 @@ int ustctl_open_metadata(int sock, int session_handle,
        /* get wait fd */
        ret = ustcomm_recv_fd(sock);
        if (ret < 0)
-               err = 1;
+               err = ret;
        else
                metadata_data->wait_fd = ret;
        if (err)
@@ -187,7 +183,7 @@ int ustctl_open_metadata(int sock, int session_handle,
 error:
        (void) ustctl_release_object(sock, metadata_data);
        free(metadata_data);
-       return -EINVAL;
+       return err;
 }
 
 int ustctl_create_channel(int sock, int session_handle,
@@ -199,6 +195,9 @@ int ustctl_create_channel(int sock, int session_handle,
        struct lttng_ust_object_data *channel_data;
        int ret, err = 0;
 
+       if (!chops || !_channel_data)
+               return -EINVAL;
+
        channel_data = malloc(sizeof(*channel_data));
        if (!channel_data)
                return -ENOMEM;
@@ -218,17 +217,13 @@ int ustctl_create_channel(int sock, int session_handle,
                free(channel_data);
                return ret;
        }
-       if (lur.ret_code != USTCOMM_OK) {
-               free(channel_data);
-               return lur.ret_code;
-       }
        channel_data->handle = lur.ret_val;
        DBG("received channel handle %u", channel_data->handle);
        channel_data->memory_map_size = lur.u.channel.memory_map_size;
        /* get shm fd */
        ret = ustcomm_recv_fd(sock);
        if (ret < 0)
-               err = 1;
+               err = ret;
        else
                channel_data->shm_fd = ret;
        /*
@@ -238,7 +233,7 @@ int ustctl_create_channel(int sock, int session_handle,
        /* get wait fd */
        ret = ustcomm_recv_fd(sock);
        if (ret < 0)
-               err = 1;
+               err = ret;
        else
                channel_data->wait_fd = ret;
        if (err)
@@ -249,13 +244,14 @@ int ustctl_create_channel(int sock, int session_handle,
 error:
        (void) ustctl_release_object(sock, channel_data);
        free(channel_data);
-       return -EINVAL;
+       return err;
 }
 
 /*
- * Return -ENOENT if no more stream is available for creation.
+ * Return -LTTNG_UST_ERR_NOENT if no more stream is available for creation.
  * Return 0 on success.
- * Return negative error value on error.
+ * Return negative error value on system error.
+ * Return positive error value on UST error.
  */
 int ustctl_create_stream(int sock, struct lttng_ust_object_data *channel_data,
                struct lttng_ust_object_data **_stream_data)
@@ -265,6 +261,9 @@ int ustctl_create_stream(int sock, struct lttng_ust_object_data *channel_data,
        struct lttng_ust_object_data *stream_data;
        int ret, fd, err = 0;
 
+       if (!channel_data || !_stream_data)
+               return -EINVAL;
+
        stream_data = malloc(sizeof(*stream_data));
        if (!stream_data)
                return -ENOMEM;
@@ -277,18 +276,13 @@ int ustctl_create_stream(int sock, struct lttng_ust_object_data *channel_data,
                free(stream_data);
                return ret;
        }
-       if (lur.ret_code != USTCOMM_OK) {
-               free(stream_data);
-               return lur.ret_code;
-       }
-
        stream_data->handle = lur.ret_val;
        DBG("received stream handle %u", stream_data->handle);
        stream_data->memory_map_size = lur.u.stream.memory_map_size;
        /* get shm fd */
        fd = ustcomm_recv_fd(sock);
        if (fd < 0)
-               err = 1;
+               err = fd;
        else
                stream_data->shm_fd = fd;
        /*
@@ -298,7 +292,7 @@ int ustctl_create_stream(int sock, struct lttng_ust_object_data *channel_data,
        /* get wait fd */
        fd = ustcomm_recv_fd(sock);
        if (fd < 0)
-               err = 1;
+               err = fd;
        else
                stream_data->wait_fd = fd;
        if (err)
@@ -309,7 +303,7 @@ int ustctl_create_stream(int sock, struct lttng_ust_object_data *channel_data,
 error:
        (void) ustctl_release_object(sock, stream_data);
        free(stream_data);
-       return -EINVAL;
+       return err;
 }
 
 int ustctl_create_event(int sock, struct lttng_ust_event *ev,
@@ -321,6 +315,9 @@ int ustctl_create_event(int sock, struct lttng_ust_event *ev,
        struct lttng_ust_object_data *event_data;
        int ret;
 
+       if (!channel_data || !_event_data)
+               return -EINVAL;
+
        event_data = malloc(sizeof(*event_data));
        if (!event_data)
                return -ENOMEM;
@@ -353,6 +350,9 @@ int ustctl_add_context(int sock, struct lttng_ust_context *ctx,
        struct lttng_ust_object_data *context_data;
        int ret;
 
+       if (!obj_data || !_context_data)
+               return -EINVAL;
+
        context_data = malloc(sizeof(*context_data));
        if (!context_data)
                return -ENOMEM;
@@ -372,6 +372,39 @@ int ustctl_add_context(int sock, struct lttng_ust_context *ctx,
        return ret;
 }
 
+int ustctl_set_filter(int sock, struct lttng_ust_filter_bytecode *bytecode,
+               struct lttng_ust_object_data *obj_data)
+{
+       struct ustcomm_ust_msg lum;
+       struct ustcomm_ust_reply lur;
+       int ret;
+
+       if (!obj_data)
+               return -EINVAL;
+
+       memset(&lum, 0, sizeof(lum));
+       lum.handle = obj_data->handle;
+       lum.cmd = LTTNG_UST_FILTER;
+       lum.u.filter.data_size = bytecode->len;
+       lum.u.filter.reloc_offset = bytecode->reloc_offset;
+       lum.u.filter.seqnum = bytecode->seqnum;
+
+       ret = ustcomm_send_app_msg(sock, &lum);
+       if (ret)
+               return ret;
+       /* send var len bytecode */
+       ret = ustcomm_send_unix_sock(sock, bytecode->data,
+                               bytecode->len);
+       if (ret < 0) {
+               if (ret == -ECONNRESET)
+                       fprintf(stderr, "remote end closed connection\n");
+               return ret;
+       }
+       if (ret != bytecode->len)
+               return -EINVAL;
+       return ustcomm_recv_app_reply(sock, &lur, lum.handle, lum.cmd);
+}
+
 /* Enable event, channel and session ioctl */
 int ustctl_enable(int sock, struct lttng_ust_object_data *object)
 {
@@ -379,6 +412,9 @@ int ustctl_enable(int sock, struct lttng_ust_object_data *object)
        struct ustcomm_ust_reply lur;
        int ret;
 
+       if (!object)
+               return -EINVAL;
+
        memset(&lum, 0, sizeof(lum));
        lum.handle = object->handle;
        lum.cmd = LTTNG_UST_ENABLE;
@@ -396,6 +432,9 @@ int ustctl_disable(int sock, struct lttng_ust_object_data *object)
        struct ustcomm_ust_reply lur;
        int ret;
 
+       if (!object)
+               return -EINVAL;
+
        memset(&lum, 0, sizeof(lum));
        lum.handle = object->handle;
        lum.cmd = LTTNG_UST_DISABLE;
@@ -446,6 +485,9 @@ int ustctl_tracepoint_list_get(int sock, int tp_list_handle,
        struct ustcomm_ust_reply lur;
        int ret;
 
+       if (!iter)
+               return -EINVAL;
+
        memset(&lum, 0, sizeof(lum));
        lum.handle = tp_list_handle;
        lum.cmd = LTTNG_UST_TRACEPOINT_LIST_GET;
@@ -459,12 +501,61 @@ int ustctl_tracepoint_list_get(int sock, int tp_list_handle,
        return 0;
 }
 
+int ustctl_tracepoint_field_list(int sock)
+{
+       struct ustcomm_ust_msg lum;
+       struct ustcomm_ust_reply lur;
+       int ret, tp_field_list_handle;
+
+       memset(&lum, 0, sizeof(lum));
+       lum.handle = LTTNG_UST_ROOT_HANDLE;
+       lum.cmd = LTTNG_UST_TRACEPOINT_FIELD_LIST;
+       ret = ustcomm_send_app_cmd(sock, &lum, &lur);
+       if (ret)
+               return ret;
+       tp_field_list_handle = lur.ret_val;
+       DBG("received tracepoint field list handle %u", tp_field_list_handle);
+       return tp_field_list_handle;
+}
+
+int ustctl_tracepoint_field_list_get(int sock, int tp_field_list_handle,
+               struct lttng_ust_field_iter *iter)
+{
+       struct ustcomm_ust_msg lum;
+       struct ustcomm_ust_reply lur;
+       int ret;
+       ssize_t len;
+
+       if (!iter)
+               return -EINVAL;
+
+       memset(&lum, 0, sizeof(lum));
+       lum.handle = tp_field_list_handle;
+       lum.cmd = LTTNG_UST_TRACEPOINT_FIELD_LIST_GET;
+       ret = ustcomm_send_app_cmd(sock, &lum, &lur);
+       if (ret)
+               return ret;
+       len = ustcomm_recv_unix_sock(sock, iter, sizeof(*iter));
+       if (len != sizeof(*iter)) {
+               return -EINVAL;
+       }
+       DBG("received tracepoint field list entry event_name %s event_loglevel %d field_name %s field_type %d",
+               iter->event_name,
+               iter->loglevel,
+               iter->field_name,
+               iter->type);
+       return 0;
+}
+
 int ustctl_tracer_version(int sock, struct lttng_ust_tracer_version *v)
 {
        struct ustcomm_ust_msg lum;
        struct ustcomm_ust_reply lur;
        int ret;
 
+       if (!v)
+               return -EINVAL;
+
        memset(&lum, 0, sizeof(lum));
        lum.handle = LTTNG_UST_ROOT_HANDLE;
        lum.cmd = LTTNG_UST_TRACER_VERSION;
@@ -494,6 +585,9 @@ int ustctl_wait_quiescent(int sock)
 
 int ustctl_calibrate(int sock, struct lttng_ust_calibrate *calibrate)
 {
+       if (!calibrate)
+               return -EINVAL;
+
        return -ENOSYS;
 }
 
@@ -503,6 +597,9 @@ int ustctl_sock_flush_buffer(int sock, struct lttng_ust_object_data *object)
        struct ustcomm_ust_reply lur;
        int ret;
 
+       if (!object)
+               return -EINVAL;
+
        memset(&lum, 0, sizeof(lum));
        lum.handle = object->handle;
        lum.cmd = LTTNG_UST_FLUSH_BUFFER;
@@ -524,6 +621,9 @@ struct lttng_ust_shm_handle *ustctl_map_channel(struct lttng_ust_object_data *ch
        struct lttng_ust_lib_ring_buffer_config *config;
        int ret;
 
+       if (!chan_data)
+               return NULL;
+
        handle = channel_handle_create(chan_data->shm_fd,
                chan_data->wait_fd,
                chan_data->memory_map_size);
@@ -585,6 +685,7 @@ struct lttng_ust_shm_handle *ustctl_map_channel(struct lttng_ust_object_data *ch
        default:
                ERR("Unknown client type %d", config->client_type);
                channel_destroy(chan, handle, 1);
+               free(handle->shadow_chan);
                return NULL;
        }
        /* Replace the object table pointer. */
@@ -605,6 +706,9 @@ int ustctl_add_stream(struct lttng_ust_shm_handle *handle,
 {
        int ret;
 
+       if (!handle || !stream_data)
+               return -EINVAL;
+
        if (!stream_data->handle)
                return -ENOENT;
        /* map stream */
@@ -629,8 +733,10 @@ void ustctl_unmap_channel(struct lttng_ust_shm_handle *handle)
 {
        struct channel *chan;
 
+       assert(handle);
        chan = shmp(handle, handle->chan);
        channel_destroy(chan, handle, 1);
+       free(handle->shadow_chan);
 }
 
 /*
@@ -639,12 +745,16 @@ void ustctl_unmap_channel(struct lttng_ust_shm_handle *handle)
 struct lttng_ust_lib_ring_buffer *ustctl_open_stream_read(struct lttng_ust_shm_handle *handle,
        int cpu)
 {
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
        int *shm_fd, *wait_fd;
        uint64_t *memory_map_size;
        struct lttng_ust_lib_ring_buffer *buf;
        int ret;
 
+       if (!handle)
+               return NULL;
+
+       chan = handle->shadow_chan;
        buf = channel_get_ring_buffer(&chan->backend.config,
                chan, cpu, handle, &shm_fd, &wait_fd, &memory_map_size);
        if (!buf)
@@ -668,6 +778,7 @@ struct lttng_ust_lib_ring_buffer *ustctl_open_stream_read(struct lttng_ust_shm_h
 void ustctl_close_stream_read(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       assert(handle && buf);
        lib_ring_buffer_release_read(buf, handle, 1);
 }
 
@@ -676,6 +787,8 @@ void ustctl_close_stream_read(struct lttng_ust_shm_handle *handle,
 void *ustctl_get_mmap_base(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       if (!handle || !buf)
+               return NULL;
        return shmp(handle, buf->backend.memory_map);
 }
 
@@ -685,8 +798,12 @@ int ustctl_get_mmap_len(struct lttng_ust_shm_handle *handle,
                unsigned long *len)
 {
        unsigned long mmap_buf_len;
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
+
+       if (!handle || !buf || !len)
+               return -EINVAL;
 
+       chan = handle->shadow_chan;
        if (chan->backend.config.output != RING_BUFFER_MMAP)
                return -EINVAL;
        mmap_buf_len = chan->backend.buf_size;
@@ -703,8 +820,12 @@ int ustctl_get_max_subbuf_size(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf,
                unsigned long *len)
 {
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
 
+       if (!handle || !buf || !len)
+               return -EINVAL;
+
+       chan = handle->shadow_chan;
        *len = chan->backend.subbuf_size;
        return 0;
 }
@@ -718,9 +839,13 @@ int ustctl_get_max_subbuf_size(struct lttng_ust_shm_handle *handle,
 int ustctl_get_mmap_read_offset(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *off)
 {
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
        unsigned long sb_bindex;
 
+       if (!handle || !buf || !off)
+               return -EINVAL;
+
+       chan = handle->shadow_chan;
        if (chan->backend.config.output != RING_BUFFER_MMAP)
                return -EINVAL;
        sb_bindex = subbuffer_id_get_index(&chan->backend.config,
@@ -733,8 +858,12 @@ int ustctl_get_mmap_read_offset(struct lttng_ust_shm_handle *handle,
 int ustctl_get_subbuf_size(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *len)
 {
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
+
+       if (!handle || !buf || !len)
+               return -EINVAL;
 
+       chan = handle->shadow_chan;
        *len = lib_ring_buffer_get_read_data_size(&chan->backend.config, buf,
                handle);
        return 0;
@@ -744,8 +873,12 @@ int ustctl_get_subbuf_size(struct lttng_ust_shm_handle *handle,
 int ustctl_get_padded_subbuf_size(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *len)
 {
-       struct channel *chan = handle->shadow_chan;
+       struct channel *chan;
+
+       if (!handle || !buf || !len)
+               return -EINVAL;
 
+       chan = handle->shadow_chan;
        *len = lib_ring_buffer_get_read_data_size(&chan->backend.config, buf,
                handle);
        *len = PAGE_ALIGN(*len);
@@ -756,6 +889,9 @@ int ustctl_get_padded_subbuf_size(struct lttng_ust_shm_handle *handle,
 int ustctl_get_next_subbuf(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       if (!handle || !buf)
+               return -EINVAL;
+
        return lib_ring_buffer_get_next_subbuf(buf, handle);
 }
 
@@ -764,6 +900,9 @@ int ustctl_get_next_subbuf(struct lttng_ust_shm_handle *handle,
 int ustctl_put_next_subbuf(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       if (!handle || !buf)
+               return -EINVAL;
+
        lib_ring_buffer_put_next_subbuf(buf, handle);
        return 0;
 }
@@ -774,6 +913,9 @@ int ustctl_put_next_subbuf(struct lttng_ust_shm_handle *handle,
 int ustctl_snapshot(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       if (!handle || !buf)
+               return -EINVAL;
+
        return lib_ring_buffer_snapshot(buf, &buf->cons_snapshot,
                        &buf->prod_snapshot, handle);
 }
@@ -782,6 +924,9 @@ int ustctl_snapshot(struct lttng_ust_shm_handle *handle,
 int ustctl_snapshot_get_consumed(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *pos)
 {
+       if (!handle || !buf || !pos)
+               return -EINVAL;
+
        *pos = buf->cons_snapshot;
        return 0;
 }
@@ -790,6 +935,9 @@ int ustctl_snapshot_get_consumed(struct lttng_ust_shm_handle *handle,
 int ustctl_snapshot_get_produced(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *pos)
 {
+       if (!handle || !buf || !pos)
+               return -EINVAL;
+
        *pos = buf->prod_snapshot;
        return 0;
 }
@@ -798,6 +946,9 @@ int ustctl_snapshot_get_produced(struct lttng_ust_shm_handle *handle,
 int ustctl_get_subbuf(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf, unsigned long *pos)
 {
+       if (!handle || !buf || !pos)
+               return -EINVAL;
+
        return lib_ring_buffer_get_subbuf(buf, *pos, handle);
 }
 
@@ -805,6 +956,9 @@ int ustctl_get_subbuf(struct lttng_ust_shm_handle *handle,
 int ustctl_put_subbuf(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf)
 {
+       if (!handle || !buf)
+               return -EINVAL;
+
        lib_ring_buffer_put_subbuf(buf, handle);
        return 0;
 }
@@ -813,6 +967,7 @@ void ustctl_flush_buffer(struct lttng_ust_shm_handle *handle,
                struct lttng_ust_lib_ring_buffer *buf,
                int producer_active)
 {
+       assert(handle && buf);
        lib_ring_buffer_switch_slow(buf,
                producer_active ? SWITCH_ACTIVE : SWITCH_FLUSH,
                handle);
This page took 0.02995 seconds and 4 git commands to generate.