Handle release of parent files
[lttng-modules.git] / ltt-debugfs-abi.c
index fe70f9a9fd3b9c3ea1ce9f2f66138dee892b68df..4717b81e14621f43792cbf5df36d7627c7143de0 100644 (file)
@@ -34,13 +34,13 @@ static struct dentry *lttng_dentry;
 static const struct file_operations lttng_fops;
 static const struct file_operations lttng_session_fops;
 static const struct file_operations lttng_channel_fops;
+static const struct file_operations lttng_event_fops;
 
 /*
  * LTTng DebugFS ABI structures.
  */
 
 struct lttng_channel {
-       int session;                    /* Session file descriptor */
        int overwrite;                  /* 1: overwrite, 0: discard */
        u64 subbuf_size;
        u64 num_subbuf;
@@ -49,7 +49,6 @@ struct lttng_channel {
 };
 
 struct lttng_event {
-       int channel;                    /* Channel file descriptor */
        enum instrum_type itype;
        char name[];
 };
@@ -58,10 +57,10 @@ static
 int lttng_abi_create_session(void)
 {
        struct ltt_session *session;
-       struct file *session_file;
+       struct file *session_filp;
        int session_fd;
 
-       session = ltt_session_create()
+       session = ltt_session_create();
        if (!session)
                return -ENOMEM;
        session_fd = get_unused_fd_flags(O_RDWR);
@@ -69,14 +68,15 @@ int lttng_abi_create_session(void)
                ret = session_fd;
                goto fd_error;
        }
-       session_file = anon_inode_getfile("[lttng_session]",
+       session_filp = anon_inode_getfile("[lttng_session]",
                                          &lttng_session_fops,
                                          session, O_RDWR);
-       if (IS_ERR(session_file)) {
-               ret = PTR_ERR(session_file);
+       if (IS_ERR(session_filp)) {
+               ret = PTR_ERR(session_filp);
                goto file_error;
        }
-       fd_install(session_fd, session_file);
+       session->file = session_filp;
+       fd_install(session_fd, session_filp);
        return session_fd;
 
 file_error:
@@ -110,23 +110,10 @@ long lttng_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
        }
 }
 
-#ifdef CONFIG_COMPAT
-static
-long lttng_compat_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
-{
-       switch (cmd) {
-       case LTTNG_SESSION:
-               return lttng_abi_create_session();
-       default:
-               return -ENOIOCTLCMD;
-       }
-}
-#endif
-
 static const struct file_operations lttng_fops = {
        .unlocked_ioctl = lttng_ioctl,
 #ifdef CONFIG_COMPAT
-       .compat_ioctl = lttng_compat_ioctl,
+       .compat_ioctl = lttng_ioctl,
 #endif
 }
 
@@ -140,17 +127,8 @@ int lttng_abi_create_channel(struct file *session_filp,
        int chan_fd;
        int ret = 0;
 
-       if (copy_from_user(&chan_param, ucham_param, sizeof(chan_param)))
+       if (copy_from_user(&chan_param, uchan_param, sizeof(chan_param)))
                return -EFAULT;
-       chan = ltt_channel_create(session, chan_param->overwrite, NULL,
-                                 chan_param->subbuf_size,
-                                 chan_param->num_subbuf,
-                                 chan_param->switch_timer_interval,
-                                 chan_param->read_timer_interval);
-       if (!chan) {
-               ret = -ENOMEM;
-               goto chan_error;
-       }
        chan_fd = get_unused_fd_flags(O_RDWR);
        if (chan_fd < 0) {
                ret = chan_fd;
@@ -158,22 +136,37 @@ int lttng_abi_create_channel(struct file *session_filp,
        }
        chan_filp = anon_inode_getfile("[lttng_channel]",
                                       &lttng_channel_fops,
-                                      chan, O_RDWR);
+                                      NULL, O_RDWR);
        if (IS_ERR(chan_filp)) {
                ret = PTR_ERR(chan_filp);
                goto file_error;
        }
-
+       /*
+        * We tolerate no failure path after channel creation. It will stay
+        * invariant for the rest of the session.
+        */
+       chan = ltt_channel_create(session, chan_param->overwrite, NULL,
+                                 chan_param->subbuf_size,
+                                 chan_param->num_subbuf,
+                                 chan_param->switch_timer_interval,
+                                 chan_param->read_timer_interval);
+       if (!chan) {
+               ret = -ENOMEM;
+               goto chan_error;
+       }
+       channel->file = chan_filp;
+       chan_filp->private_data = chan;
+       fd_install(chan_fd, chan_filp);
        /* The channel created holds a reference on the session */
        atomic_inc(&session_filp->f_count);
 
        return chan_fd;
 
+chan_error:
+       fput(chan_filp);
 file_error:
        put_unused_fd(chan_fd);
 fd_error:
-       ltt_channel_destroy(chan);
-chan_error:
        return ret;
 }
 
@@ -201,23 +194,26 @@ long lttng_session_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
        }
 }
 
-#ifdef CONFIG_COMPAT
+/*
+ * Called when the last file reference is dropped.
+ *
+ * Big fat note: channels and events are invariant for the whole session after
+ * their creation. So this session destruction also destroys all channel and
+ * event structures specific to this session (they are not destroyed when their
+ * individual file is released).
+ */
 static
-long lttng_session_compat_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
+int lttng_session_release(struct inode *inode, struct file *file)
 {
-       switch (cmd) {
-       case LTTNG_CHANNEL:
-               return lttng_abi_create_channel(filp, (struct lttng_channel __user *)arg);
-       default:
-               return -ENOIOCTLCMD;
-       }
+       struct ltt_session *session = file->private_data;
+       return ltt_session_destroy(session);
 }
-#endif
 
 static const struct file_operations lttng_session_fops = {
+       .release = lttng_session_release,
        .unlocked_ioctl = lttng_session_ioctl,
 #ifdef CONFIG_COMPAT
-       .compat_ioctl = lttng_session_compat_ioctl,
+       .compat_ioctl = lttng_session_ioctl,
 #endif
 }
 
@@ -238,13 +234,13 @@ int lttng_abi_open_stream(struct file *channel_filp)
                goto fd_error;
        }
        stream_filp = anon_inode_getfile("[lttng_stream]",
-                                        &lttng_stream_fops,
+                                        &lib_ring_buffer_file_operations,
                                         buf, O_RDWR);
        if (IS_ERR(stream_filp)) {
                ret = PTR_ERR(stream_filp);
                goto file_error;
        }
-
+       fd_install(stream_fd, stream_filp);
        /* The stream holds a reference on the channel */
        atomic_inc(&channel_filp->f_count);
        return stream_fd;
@@ -256,6 +252,63 @@ fd_error:
        return ret;
 }
 
+static
+int lttng_abi_create_event(struct file *channel_filp,
+                          struct lttng_event __user *uevent_param)
+{
+       struct ltt_channel *channel = channel_filp->private_data;
+       struct ltt_event *event;
+       char *event_name;
+       struct lttng_event event_param;
+       int event_fd, ret;
+
+       if (copy_from_user(&event_param, uevent_param, sizeof(event_param)))
+               return -EFAULT;
+       event_name = kmalloc(PATH_MAX, GFP_KERNEL);
+       if (!event_name)
+               return -ENOMEM;
+       if (strncpy_from_user(event_name, &uevent_param->name, PATH_MAX)) {
+               ret = -EFAULT;
+               goto name_error;
+       }
+       event_name[PATH_MAX - 1] = '\0';
+       event_fd = get_unused_fd_flags(O_RDWR);
+       if (event_fd < 0) {
+               ret = event_fd;
+               goto fd_error;
+       }
+       event_filp = anon_inode_getfile("[lttng_event]",
+                                       &lttng_event_fops, /* TODO: filter */
+                                       NULL, O_RDWR);
+       if (IS_ERR(event_filp)) {
+               ret = PTR_ERR(event_filp);
+               goto file_error;
+       }
+       /*
+        * We tolerate no failure path after event creation. It will stay
+        * invariant for the rest of the session.
+        */
+       event = ltt_event_create(channel, event_param->itype, event_name, NULL);
+       if (!event) {
+               goto event_error;
+               ret = -EEXIST;
+       }
+       event_filp->private_data = event;
+       fd_install(event_fd, event_filp);
+       /* The event holds a reference on the channel */
+       atomic_inc(&channel_filp->f_count);
+       kfree(event_name);
+       return event_fd;
+
+event_error:
+       fput(event_filp);
+file_error:
+       put_unused_fd(event_fd);
+fd_error:
+name_error:
+       kfree(event_name);
+       return ret;
+}
 
 /**
  *     lttng_channel_ioctl - lttng syscall through ioctl
@@ -268,10 +321,6 @@ fd_error:
  *      LTTNG_STREAM
  *              Returns an event stream file descriptor or failure.
  *              (typically, one event stream records events from one CPU)
- *      LTTNG_STREAM_NOTIFIER
- *              Returns a file descriptor that can be used to monitor
- *              addition/removal of streams to/from a channel. (e.g. notifier
- *              called on CPU hotplug).
  *     LTTNG_EVENT
  *             Returns an event file descriptor or failure.
  *
@@ -284,8 +333,6 @@ long lttng_channel_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
        switch (cmd) {
        case LTTNG_STREAM:
                return lttng_abi_open_stream(filp);
-       case LTTNG_STREAM_NOTIFIER:
-               return lttng_abi_open_stream_notifier(filp);
        case LTTNG_EVENT:
                return lttng_abi_create_event(filp, (struct lttng_event __user *)arg);
        default:
@@ -293,30 +340,60 @@ long lttng_channel_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
        }
 }
 
-#ifdef CONFIG_COMPAT
-static
-long lttng_channel_compat_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
+/**
+ *     lttng_channel_poll - lttng stream addition/removal monitoring
+ *
+ *     @filp: the file
+ *     @wait: poll table
+ */
+unsigned int lttng_channel_poll(struct file *filp, poll_table *wait)
 {
-       switch (cmd) {
-       case LTTNG_STREAM:
-               return lttng_abi_get_stream(filp);
-       case LTTNG_STREAM_NOTIFIER:
-               return lttng_abi_get_stream_notifier(filp);
-       case LTTNG_EVENT:
-               return lttng_abi_create_event(filp, (struct lttng_event __user *)arg);
-       default:
-               return -ENOIOCTLCMD;
+       struct ltt_channel *channel = filp->private_data;
+       unsigned int mask = 0;
+
+       if (filp->f_mode & FMODE_READ) {
+               poll_wait_set_exclusive(wait);
+               poll_wait(filp, &channel->notify_wait, wait);
+
+               /* TODO: identify when the channel is being finalized. */
+               if (finalized)
+                       return POLLHUP;
+               else
+                       return POLLIN | POLLRDNORM;
        }
+       return mask;
+
+}
+
+static
+int lttng_channel_release(struct inode *inode, struct file *file)
+{
+       struct ltt_channel *channel = file->private_data;
+       fput(channel->session->file);
+       return 0;
 }
-#endif
 
 static const struct file_operations lttng_channel_fops = {
+       .release = lttng_channel_release,
+       .poll = lttng_channel_poll,
        .unlocked_ioctl = lttng_channel_ioctl,
 #ifdef CONFIG_COMPAT
-       .compat_ioctl = lttng_channel_compat_ioctl,
+       .compat_ioctl = lttng_channel_ioctl,
 #endif
 }
 
+static
+int lttng_event_release(struct inode *inode, struct file *file)
+{
+       struct ltt_event *event = file->private_data;
+       fput(event->chan->file);
+       return 0;
+}
+
+static const struct file_operations lttng_event_fops = {
+       .release = lttng_event_release,
+}
+
 static int __init ltt_debugfs_abi_init(void)
 {
        int ret = 0;
This page took 0.027132 seconds and 4 git commands to generate.