Expose default devices
[media-ctl.git] / src / mediactl.c
index 8f7d0a9..4da334a 100644 (file)
@@ -1,20 +1,22 @@
 /*
- * Media controller test application
+ * Media controller interface library
  *
- * Copyright (C) 2010 Ideas on board SPRL <laurent.pinchart@ideasonboard.com>
+ * Copyright (C) 2010-2011 Ideas on board SPRL
+ *
+ * Contact: Laurent Pinchart <laurent.pinchart@ideasonboard.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; either version 2 of the License, or
+ * it under the terms of the GNU Lesser General Public License as published
+ * by the Free Software Foundation; either version 2.1 of the License, or
  * (at your option) any later version.
  *
  * 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.
+ * GNU Lesser 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.,
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
  */
 
 #include "config.h"
 #include <sys/stat.h>
 #include <sys/types.h>
 
-#include <unistd.h>
+#include <ctype.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <fcntl.h>
-#include <errno.h>
+#include <unistd.h>
 
-#include <linux/videodev2.h>
 #include <linux/media.h>
+#include <linux/videodev2.h>
 
 #include "mediactl.h"
+#include "mediactl-priv.h"
 #include "tools.h"
 
+/* -----------------------------------------------------------------------------
+ * Graph access
+ */
+
 struct media_pad *media_entity_remote_source(struct media_pad *pad)
 {
        unsigned int i;
@@ -61,10 +70,17 @@ struct media_entity *media_get_entity_by_name(struct media_device *media,
 {
        unsigned int i;
 
+       /* A match is impossible if the entity name is longer than the maximum
+        * size we can get from the kernel.
+        */
+       if (length >= FIELD_SIZEOF(struct media_entity_desc, name))
+               return NULL;
+
        for (i = 0; i < media->entities_count; ++i) {
                struct media_entity *entity = &media->entities[i];
 
-               if (strncmp(entity->info.name, name, length) == 0)
+               if (strncmp(entity->info.name, name, length) == 0 &&
+                   entity->info.name[length] == '\0')
                        return entity;
        }
 
@@ -74,18 +90,127 @@ struct media_entity *media_get_entity_by_name(struct media_device *media,
 struct media_entity *media_get_entity_by_id(struct media_device *media,
                                            __u32 id)
 {
+       bool next = id & MEDIA_ENT_ID_FLAG_NEXT;
        unsigned int i;
 
+       id &= ~MEDIA_ENT_ID_FLAG_NEXT;
+
        for (i = 0; i < media->entities_count; ++i) {
                struct media_entity *entity = &media->entities[i];
 
-               if (entity->info.id == id)
+               if ((entity->info.id == id && !next) ||
+                   (entity->info.id > id && next))
                        return entity;
        }
 
        return NULL;
 }
 
+unsigned int media_get_entities_count(struct media_device *media)
+{
+       return media->entities_count;
+}
+
+struct media_entity *media_get_entity(struct media_device *media, unsigned int index)
+{
+       if (index >= media->entities_count)
+               return NULL;
+
+       return &media->entities[index];
+}
+
+const struct media_pad *media_entity_get_pad(struct media_entity *entity, unsigned int index)
+{
+       if (index >= entity->info.pads)
+               return NULL;
+
+       return &entity->pads[index];
+}
+
+unsigned int media_entity_get_links_count(struct media_entity *entity)
+{
+       return entity->num_links;
+}
+
+const struct media_link *media_entity_get_link(struct media_entity *entity, unsigned int index)
+{
+       if (index >= entity->num_links)
+               return NULL;
+
+       return &entity->links[index];
+}
+
+const char *media_entity_get_devname(struct media_entity *entity)
+{
+       return entity->devname[0] ? entity->devname : NULL;
+}
+
+struct media_entity *media_get_default_entity(struct media_device *media,
+                                             unsigned int type)
+{
+       switch (type) {
+       case MEDIA_ENT_T_DEVNODE_V4L:
+               return media->def.v4l;
+       case MEDIA_ENT_T_DEVNODE_FB:
+               return media->def.fb;
+       case MEDIA_ENT_T_DEVNODE_ALSA:
+               return media->def.alsa;
+       case MEDIA_ENT_T_DEVNODE_DVB:
+               return media->def.dvb;
+       }
+}
+
+const struct media_device_info *media_get_info(struct media_device *media)
+{
+       return &media->info;
+}
+
+const char *media_get_devnode(struct media_device *media)
+{
+       return media->devnode;
+}
+
+const struct media_entity_desc *media_entity_get_info(struct media_entity *entity)
+{
+       return &entity->info;
+}
+
+/* -----------------------------------------------------------------------------
+ * Open/close
+ */
+
+static int media_device_open(struct media_device *media)
+{
+       int ret;
+
+       if (media->fd != -1)
+               return 0;
+
+       media_dbg(media, "Opening media device %s\n", media->devnode);
+
+       media->fd = open(media->devnode, O_RDWR);
+       if (media->fd < 0) {
+               ret = -errno;
+               media_dbg(media, "%s: Can't open media device %s\n",
+                         __func__, media->devnode);
+               return ret;
+       }
+
+       return 0;
+}
+
+static void media_device_close(struct media_device *media)
+{
+       if (media->fd != -1) {
+               close(media->fd);
+               media->fd = -1;
+       }
+}
+
+/* -----------------------------------------------------------------------------
+ * Link setup
+ */
+
 int media_setup_link(struct media_device *media,
                     struct media_pad *source,
                     struct media_pad *sink,
@@ -96,6 +221,10 @@ int media_setup_link(struct media_device *media,
        unsigned int i;
        int ret;
 
+       ret = media_device_open(media);
+       if (ret < 0)
+               goto done;
+
        for (i = 0; i < source->entity->num_links; i++) {
                link = &source->entity->links[i];
 
@@ -107,8 +236,9 @@ int media_setup_link(struct media_device *media,
        }
 
        if (i == source->entity->num_links) {
-               printf("%s: Link not found\n", __func__);
-               return -EINVAL;
+               media_dbg(media, "%s: Link not found\n", __func__);
+               ret = -ENOENT;
+               goto done;
        }
 
        /* source pad */
@@ -124,15 +254,21 @@ int media_setup_link(struct media_device *media,
        ulink.flags = flags | (link->flags & MEDIA_LNK_FL_IMMUTABLE);
 
        ret = ioctl(media->fd, MEDIA_IOC_SETUP_LINK, &ulink);
-       if (ret < 0) {
-               printf("%s: Unable to setup link (%s)\n", __func__,
-                       strerror(errno));
-               return ret;
+       if (ret == -1) {
+               ret = -errno;
+               media_dbg(media, "%s: Unable to setup link (%s)\n",
+                         __func__, strerror(errno));
+               goto done;
        }
 
        link->flags = ulink.flags;
        link->twin->flags = ulink.flags;
-       return 0;
+
+       ret = 0;
+
+done:
+       media_device_close(media);
+       return ret;
 }
 
 int media_reset_links(struct media_device *media)
@@ -160,6 +296,10 @@ int media_reset_links(struct media_device *media)
        return 0;
 }
 
+/* -----------------------------------------------------------------------------
+ * Entities, pads and links enumeration
+ */
+
 static struct media_link *media_entity_add_link(struct media_entity *entity)
 {
        if (entity->num_links >= entity->max_links) {
@@ -192,15 +332,17 @@ static int media_enum_links(struct media_device *media)
                unsigned int i;
 
                links.entity = entity->info.id;
-               links.pads = malloc(entity->info.pads * sizeof(struct media_pad_desc));
-               links.links = malloc(entity->info.links * sizeof(struct media_link_desc));
+               links.pads = calloc(entity->info.pads, sizeof(struct media_pad_desc));
+               links.links = calloc(entity->info.links, sizeof(struct media_link_desc));
 
                if (ioctl(media->fd, MEDIA_IOC_ENUM_LINKS, &links) < 0) {
-                       printf("%s: Unable to enumerate pads and links (%s).\n",
-                               __func__, strerror(errno));
+                       ret = -errno;
+                       media_dbg(media,
+                                 "%s: Unable to enumerate pads and links (%s).\n",
+                                 __func__, strerror(errno));
                        free(links.pads);
                        free(links.links);
-                       return -errno;
+                       return ret;
                }
 
                for (i = 0; i < entity->info.pads; ++i) {
@@ -220,9 +362,12 @@ static int media_enum_links(struct media_device *media)
                        sink = media_get_entity_by_id(media, link->sink.entity);
 
                        if (source == NULL || sink == NULL) {
-                               printf("WARNING entity %u link %u from %u/%u to %u/%u is invalid!\n",
-                                       id, i, link->source.entity, link->source.index,
-                                       link->sink.entity, link->sink.index);
+                               media_dbg(media,
+                                         "WARNING entity %u link %u from %u/%u to %u/%u is invalid!\n",
+                                         id, i, link->source.entity,
+                                         link->source.index,
+                                         link->sink.entity,
+                                         link->sink.index);
                                ret = -EINVAL;
                        } else {
                                fwdlink = media_entity_add_link(source);
@@ -266,7 +411,7 @@ static inline void media_udev_close(struct udev *udev)
 }
 
 static int media_get_devname_udev(struct udev *udev,
-               struct media_entity *entity, int verbose)
+               struct media_entity *entity)
 {
        struct udev_device *device;
        dev_t devnum;
@@ -277,8 +422,8 @@ static int media_get_devname_udev(struct udev *udev,
                return -EINVAL;
 
        devnum = makedev(entity->info.v4l.major, entity->info.v4l.minor);
-       if (verbose)
-               printf("looking up device: %u:%u\n", major(devnum), minor(devnum));
+       media_dbg(entity->media, "looking up device: %u:%u\n",
+                 major(devnum), minor(devnum));
        device = udev_device_new_from_devnum(udev, 'c', devnum);
        if (device) {
                p = udev_device_get_devnode(device);
@@ -303,7 +448,7 @@ static inline int media_udev_open(struct udev **udev) { return 0; }
 static inline void media_udev_close(struct udev *udev) { }
 
 static inline int media_get_devname_udev(struct udev *udev,
-               struct media_entity *entity, int verbose)
+               struct media_entity *entity)
 {
        return -ENOTSUP;
 }
@@ -346,7 +491,7 @@ static int media_get_devname_sysfs(struct media_entity *entity)
        return 0;
 }
 
-static int media_enum_entities(struct media_device *media, int verbose)
+static int media_enum_entities(struct media_device *media)
 {
        struct media_entity *entity;
        struct udev *udev;
@@ -356,7 +501,7 @@ static int media_enum_entities(struct media_device *media, int verbose)
 
        ret = media_udev_open(&udev);
        if (ret < 0)
-               printf("%s: Can't get udev context\n", __func__);
+               media_dbg(media, "Can't get udev context\n");
 
        for (id = 0, ret = 0; ; id = entity->info.id) {
                size = (media->entities_count + 1) * sizeof(*media->entities);
@@ -366,6 +511,7 @@ static int media_enum_entities(struct media_device *media, int verbose)
                memset(entity, 0, sizeof(*entity));
                entity->fd = -1;
                entity->info.id = id | MEDIA_ENT_ID_FLAG_NEXT;
+               entity->media = media;
 
                ret = ioctl(media->fd, MEDIA_IOC_ENUM_ENTITIES, &entity->info);
                if (ret < 0) {
@@ -388,13 +534,30 @@ static int media_enum_entities(struct media_device *media, int verbose)
 
                media->entities_count++;
 
+               if (entity->info.flags & MEDIA_ENT_FL_DEFAULT) {
+                       switch (entity->info.type) {
+                       case MEDIA_ENT_T_DEVNODE_V4L:
+                               media->def.v4l = entity;
+                               break;
+                       case MEDIA_ENT_T_DEVNODE_FB:
+                               media->def.fb = entity;
+                               break;
+                       case MEDIA_ENT_T_DEVNODE_ALSA:
+                               media->def.alsa = entity;
+                               break;
+                       case MEDIA_ENT_T_DEVNODE_DVB:
+                               media->def.dvb = entity;
+                               break;
+                       }
+               }
+
                /* Find the corresponding device name. */
                if (media_entity_type(entity) != MEDIA_ENT_T_DEVNODE &&
                    media_entity_type(entity) != MEDIA_ENT_T_V4L2_SUBDEV)
                        continue;
 
                /* Try to get the device name via udev */
-               if (!media_get_devname_udev(udev, entity, verbose))
+               if (!media_get_devname_udev(udev, entity))
                        continue;
 
                /* Fall back to get the device name via sysfs */
@@ -405,60 +568,111 @@ static int media_enum_entities(struct media_device *media, int verbose)
        return ret;
 }
 
-struct media_device *media_open(const char *name, int verbose)
+int media_device_enumerate(struct media_device *media)
 {
-       struct media_device *media;
        int ret;
 
-       media = calloc(1, sizeof(*media));
-       if (media == NULL) {
-               printf("%s: unable to allocate memory\n", __func__);
-               return NULL;
-       }
+       if (media->entities)
+               return 0;
 
-       if (verbose)
-               printf("Opening media device %s\n", name);
-       media->fd = open(name, O_RDWR);
-       if (media->fd < 0) {
-               media_close(media);
-               printf("%s: Can't open media device %s\n", __func__, name);
-               return NULL;
-       }
+       ret = media_device_open(media);
+       if (ret < 0)
+               return ret;
 
-       if (verbose)
-               printf("Enumerating entities\n");
+       ret = ioctl(media->fd, MEDIA_IOC_DEVICE_INFO, &media->info);
+       if (ret < 0) {
+               ret = -errno;
+               media_dbg(media, "%s: Unable to retrieve media device "
+                         "information for device %s (%s)\n", __func__,
+                         media->devnode, strerror(errno));
+               goto done;
+       }
 
-       ret = media_enum_entities(media, verbose);
+       media_dbg(media, "Enumerating entities\n");
 
+       ret = media_enum_entities(media);
        if (ret < 0) {
-               printf("%s: Unable to enumerate entities for device %s (%s)\n",
-                       __func__, name, strerror(-ret));
-               media_close(media);
-               return NULL;
+               media_dbg(media,
+                         "%s: Unable to enumerate entities for device %s (%s)\n",
+                         __func__, media->devnode, strerror(-ret));
+               goto done;
        }
 
-       if (verbose) {
-               printf("Found %u entities\n", media->entities_count);
-               printf("Enumerating pads and links\n");
-       }
+       media_dbg(media, "Found %u entities\n", media->entities_count);
+       media_dbg(media, "Enumerating pads and links\n");
 
        ret = media_enum_links(media);
        if (ret < 0) {
-               printf("%s: Unable to enumerate pads and linksfor device %s\n",
-                       __func__, name);
-               media_close(media);
+               media_dbg(media,
+                         "%s: Unable to enumerate pads and linksfor device %s\n",
+                         __func__, media->devnode);
+               goto done;
+       }
+
+       ret = 0;
+
+done:
+       media_device_close(media);
+       return ret;
+}
+
+/* -----------------------------------------------------------------------------
+ * Create/destroy
+ */
+
+static void media_debug_default(void *ptr, ...)
+{
+}
+
+void media_debug_set_handler(struct media_device *media,
+                            void (*debug_handler)(void *, ...),
+                            void *debug_priv)
+{
+       if (debug_handler) {
+               media->debug_handler = debug_handler;
+               media->debug_priv = debug_priv;
+       } else {
+               media->debug_handler = media_debug_default;
+               media->debug_priv = NULL;
+       }
+}
+
+struct media_device *media_device_new(const char *devnode)
+{
+       struct media_device *media;
+       int ret;
+
+       media = calloc(1, sizeof(*media));
+       if (media == NULL)
+               return NULL;
+
+       media->fd = -1;
+       media->refcount = 1;
+
+       media_debug_set_handler(media, NULL, NULL);
+
+       media->devnode = strdup(devnode);
+       if (media->devnode == NULL) {
+               media_device_unref(media);
                return NULL;
        }
 
        return media;
 }
 
-void media_close(struct media_device *media)
+struct media_device *media_device_ref(struct media_device *media)
+{
+       media->refcount++;
+       return media;
+}
+
+void media_device_unref(struct media_device *media)
 {
        unsigned int i;
 
-       if (media->fd != -1)
-               close(media->fd);
+       media->refcount--;
+       if (media->refcount > 0)
+               return;
 
        for (i = 0; i < media->entities_count; ++i) {
                struct media_entity *entity = &media->entities[i];
@@ -470,5 +684,196 @@ void media_close(struct media_device *media)
        }
 
        free(media->entities);
+       free(media->devnode);
        free(media);
 }
+
+/* -----------------------------------------------------------------------------
+ * Parsing
+ */
+
+struct media_pad *media_parse_pad(struct media_device *media,
+                                 const char *p, char **endp)
+{
+       unsigned int entity_id, pad;
+       struct media_entity *entity;
+       char *end;
+
+       /* endp can be NULL. To avoid spreading NULL checks across the function,
+        * set endp to &end in that case.
+        */
+       if (endp == NULL)
+               endp = &end;
+
+       for (; isspace(*p); ++p);
+
+       if (*p == '"' || *p == '\'') {
+               for (end = (char *)p + 1; *end && *end != '"' && *end != '\''; ++end);
+               if (*end != '"' && *end != '\'') {
+                       media_dbg(media, "missing matching '\"'\n");
+                       *endp = end;
+                       return NULL;
+               }
+
+               entity = media_get_entity_by_name(media, p + 1, end - p - 1);
+               if (entity == NULL) {
+                       media_dbg(media, "no such entity \"%.*s\"\n", end - p - 1, p + 1);
+                       *endp = (char *)p + 1;
+                       return NULL;
+               }
+
+               ++end;
+       } else {
+               entity_id = strtoul(p, &end, 10);
+               entity = media_get_entity_by_id(media, entity_id);
+               if (entity == NULL) {
+                       media_dbg(media, "no such entity %d\n", entity_id);
+                       *endp = (char *)p;
+                       return NULL;
+               }
+       }
+       for (; isspace(*end); ++end);
+
+       if (*end != ':') {
+               media_dbg(media, "Expected ':'\n", *end);
+               *endp = end;
+               return NULL;
+       }
+
+       for (p = end + 1; isspace(*p); ++p);
+
+       pad = strtoul(p, &end, 10);
+
+       if (pad >= entity->info.pads) {
+               media_dbg(media, "No pad '%d' on entity \"%s\". Maximum pad number is %d\n",
+                               pad, entity->info.name, entity->info.pads - 1);
+               *endp = (char *)p;
+               return NULL;
+       }
+
+       for (p = end; isspace(*p); ++p);
+       *endp = (char *)p;
+
+       return &entity->pads[pad];
+}
+
+struct media_link *media_parse_link(struct media_device *media,
+                                   const char *p, char **endp)
+{
+       struct media_link *link;
+       struct media_pad *source;
+       struct media_pad *sink;
+       unsigned int i;
+       char *end;
+
+       source = media_parse_pad(media, p, &end);
+       if (source == NULL) {
+               *endp = end;
+               return NULL;
+       }
+
+       if (end[0] != '-' || end[1] != '>') {
+               *endp = end;
+               media_dbg(media, "Expected '->'\n");
+               return NULL;
+       }
+
+       p = end + 2;
+
+       sink = media_parse_pad(media, p, &end);
+       if (sink == NULL) {
+               *endp = end;
+               return NULL;
+       }
+
+       *endp = end;
+
+       for (i = 0; i < source->entity->num_links; i++) {
+               link = &source->entity->links[i];
+
+               if (link->source == source && link->sink == sink)
+                       return link;
+       }
+
+       media_dbg(media, "No link between \"%s\":%d and \"%s\":%d\n",
+                       source->entity->info.name, source->index,
+                       sink->entity->info.name, sink->index);
+       return NULL;
+}
+
+int media_parse_setup_link(struct media_device *media,
+                          const char *p, char **endp)
+{
+       struct media_link *link;
+       __u32 flags;
+       char *end;
+
+       link = media_parse_link(media, p, &end);
+       if (link == NULL) {
+               media_dbg(media,
+                         "%s: Unable to parse link\n", __func__);
+               *endp = end;
+               return -EINVAL;
+       }
+
+       p = end;
+       if (*p++ != '[') {
+               media_dbg(media, "Unable to parse link flags: expected '['.\n");
+               *endp = (char *)p - 1;
+               return -EINVAL;
+       }
+
+       flags = strtoul(p, &end, 10);
+       for (p = end; isspace(*p); p++);
+       if (*p++ != ']') {
+               media_dbg(media, "Unable to parse link flags: expected ']'.\n");
+               *endp = (char *)p - 1;
+               return -EINVAL;
+       }
+
+       for (; isspace(*p); p++);
+       *endp = (char *)p;
+
+       media_dbg(media,
+                 "Setting up link %u:%u -> %u:%u [%u]\n",
+                 link->source->entity->info.id, link->source->index,
+                 link->sink->entity->info.id, link->sink->index,
+                 flags);
+
+       return media_setup_link(media, link->source, link->sink, flags);
+}
+
+void media_print_streampos(struct media_device *media, const char *p,
+                          const char *end)
+{
+       int pos;
+
+       pos = end - p + 1;
+
+       if (pos < 0)
+               pos = 0;
+       if (pos > strlen(p))
+               pos = strlen(p);
+
+       media_dbg(media, "\n");
+       media_dbg(media, " %s\n", p);
+       media_dbg(media, " %*s\n", pos, "^");
+}
+
+int media_parse_setup_links(struct media_device *media, const char *p)
+{
+       char *end;
+       int ret;
+
+       do {
+               ret = media_parse_setup_link(media, p, &end);
+               if (ret < 0) {
+                       media_print_streampos(media, p, end);
+                       return ret;
+               }
+
+               p = end + 1;
+       } while (*end == ',');
+
+       return *end ? -EINVAL : 0;
+}