src/mediactl.c

   1 /*
   2  * Media controller test application
   3  *
   4  * Copyright (C) 2010 Ideas on board SPRL <laurent.pinchart@ideasonboard.com>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; if not, write to the Free Software Foundation, Inc.,
  18  */
  19
  20 #include "config.h"
  21
  22 #include <sys/ioctl.h>
  23 #include <sys/stat.h>
  24 #include <sys/types.h>
  25
  26 #include <unistd.h>
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <fcntl.h>
  31 #include <errno.h>
  32
  33 #include <linux/videodev2.h>
  34 #include <linux/media.h>
  35
  36 #include "mediactl.h"
  37 #include "tools.h"
  38
  39 #ifdef DEBUG
  40 #define dprintf(...) printf(__VA_ARGS__)
  41 #else
  42 #define dprintf(...)
  43 #endif
  44
  45 struct media_pad *media_entity_remote_source(struct media_pad *pad)
  46 {
  47         unsigned int i;
  48
  49         if (!(pad->flags & MEDIA_PAD_FL_SINK))
  50                 return NULL;
  51
  52         for (i = 0; i < pad->entity->num_links; ++i) {
  53                 struct media_link *link = &pad->entity->links[i];
  54
  55                 if (!(link->flags & MEDIA_LNK_FL_ENABLED))
  56                         continue;
  57
  58                 if (link->sink == pad)
  59                         return link->source;
  60         }
  61
  62         return NULL;
  63 }
  64
  65 struct media_entity *media_get_entity_by_name(struct media_device *media,
  66                                               const char *name, size_t length)
  67 {
  68         unsigned int i;
  69
  70         for (i = 0; i < media->entities_count; ++i) {
  71                 struct media_entity *entity = &media->entities[i];
  72
  73                 if (strncmp(entity->info.name, name, length) == 0)
  74                         return entity;
  75         }
  76
  77         return NULL;
  78 }
  79
  80 struct media_entity *media_get_entity_by_id(struct media_device *media,
  81                                             __u32 id)
  82 {
  83         unsigned int i;
  84
  85         for (i = 0; i < media->entities_count; ++i) {
  86                 struct media_entity *entity = &media->entities[i];
  87
  88                 if (entity->info.id == id)
  89                         return entity;
  90         }
  91
  92         return NULL;
  93 }
  94
  95 int media_setup_link(struct media_device *media,
  96                      struct media_pad *source,
  97                      struct media_pad *sink,
  98                      __u32 flags)
  99 {
 100         struct media_link *link;
 101         struct media_link_desc ulink;
 102         unsigned int i;
 103         int ret;
 104
 105         for (i = 0; i < source->entity->num_links; i++) {
 106                 link = &source->entity->links[i];
 107
 108                 if (link->source->entity == source->entity &&
 109                     link->source->index == source->index &&
 110                     link->sink->entity == sink->entity &&
 111                     link->sink->index == sink->index)
 112                         break;
 113         }
 114
 115         if (i == source->entity->num_links) {
 116                 dprintf("%s: Link not found\n", __func__);
 117                 return -ENOENT;
 118         }
 119
 120         /* source pad */
 121         ulink.source.entity = source->entity->info.id;
 122         ulink.source.index = source->index;
 123         ulink.source.flags = MEDIA_PAD_FL_SOURCE;
 124
 125         /* sink pad */
 126         ulink.sink.entity = sink->entity->info.id;
 127         ulink.sink.index = sink->index;
 128         ulink.sink.flags = MEDIA_PAD_FL_SINK;
 129
 130         ulink.flags = flags | (link->flags & MEDIA_LNK_FL_IMMUTABLE);
 131
 132         ret = ioctl(media->fd, MEDIA_IOC_SETUP_LINK, &ulink);
 133         if (ret == -1) {
 134                 dprintf("%s: Unable to setup link (%s)\n", __func__,
 135                         strerror(errno));
 136                 return -errno;
 137         }
 138
 139         link->flags = ulink.flags;
 140         link->twin->flags = ulink.flags;
 141         return 0;
 142 }
 143
 144 int media_reset_links(struct media_device *media)
 145 {
 146         unsigned int i, j;
 147         int ret;
 148
 149         for (i = 0; i < media->entities_count; ++i) {
 150                 struct media_entity *entity = &media->entities[i];
 151
 152                 for (j = 0; j < entity->num_links; j++) {
 153                         struct media_link *link = &entity->links[j];
 154
 155                         if (link->flags & MEDIA_LNK_FL_IMMUTABLE ||
 156                             link->source->entity != entity)
 157                                 continue;
 158
 159                         ret = media_setup_link(media, link->source, link->sink,
 160                                                link->flags & ~MEDIA_LNK_FL_ENABLED);
 161                         if (ret < 0)
 162                                 return ret;
 163                 }
 164         }
 165
 166         return 0;
 167 }
 168
 169 static struct media_link *media_entity_add_link(struct media_entity *entity)
 170 {
 171         if (entity->num_links >= entity->max_links) {
 172                 struct media_link *links = entity->links;
 173                 unsigned int max_links = entity->max_links * 2;
 174                 unsigned int i;
 175
 176                 links = realloc(links, max_links * sizeof *links);
 177                 if (links == NULL)
 178                         return NULL;
 179
 180                 for (i = 0; i < entity->num_links; ++i)
 181                         links[i].twin->twin = &links[i];
 182
 183                 entity->max_links = max_links;
 184                 entity->links = links;
 185         }
 186
 187         return &entity->links[entity->num_links++];
 188 }
 189
 190 static int media_enum_links(struct media_device *media)
 191 {
 192         __u32 id;
 193         int ret = 0;
 194
 195         for (id = 1; id <= media->entities_count; id++) {
 196                 struct media_entity *entity = &media->entities[id - 1];
 197                 struct media_links_enum links;
 198                 unsigned int i;
 199
 200                 links.entity = entity->info.id;
 201                 links.pads = malloc(entity->info.pads * sizeof(struct media_pad_desc));
 202                 links.links = malloc(entity->info.links * sizeof(struct media_link_desc));
 203
 204                 if (ioctl(media->fd, MEDIA_IOC_ENUM_LINKS, &links) < 0) {
 205                         dprintf("%s: Unable to enumerate pads and links (%s).\n",
 206                                 __func__, strerror(errno));
 207                         free(links.pads);
 208                         free(links.links);
 209                         return -errno;
 210                 }
 211
 212                 for (i = 0; i < entity->info.pads; ++i) {
 213                         entity->pads[i].entity = entity;
 214                         entity->pads[i].index = links.pads[i].index;
 215                         entity->pads[i].flags = links.pads[i].flags;
 216                 }
 217
 218                 for (i = 0; i < entity->info.links; ++i) {
 219                         struct media_link_desc *link = &links.links[i];
 220                         struct media_link *fwdlink;
 221                         struct media_link *backlink;
 222                         struct media_entity *source;
 223                         struct media_entity *sink;
 224
 225                         source = media_get_entity_by_id(media, link->source.entity);
 226                         sink = media_get_entity_by_id(media, link->sink.entity);
 227
 228                         if (source == NULL || sink == NULL) {
 229                                 dprintf("WARNING entity %u link %u from %u/%u to %u/%u is invalid!\n",
 230                                         id, i, link->source.entity, link->source.index,
 231                                         link->sink.entity, link->sink.index);
 232                                 ret = -EINVAL;
 233                         } else {
 234                                 fwdlink = media_entity_add_link(source);
 235                                 fwdlink->source = &source->pads[link->source.index];
 236                                 fwdlink->sink = &sink->pads[link->sink.index];
 237                                 fwdlink->flags = link->flags;
 238
 239                                 backlink = media_entity_add_link(sink);
 240                                 backlink->source = &source->pads[link->source.index];
 241                                 backlink->sink = &sink->pads[link->sink.index];
 242                                 backlink->flags = link->flags;
 243
 244                                 fwdlink->twin = backlink;
 245                                 backlink->twin = fwdlink;
 246                         }
 247                 }
 248
 249                 free(links.pads);
 250                 free(links.links);
 251         }
 252
 253         return ret;
 254 }
 255
 256 #ifdef HAVE_LIBUDEV
 257
 258 #include <libudev.h>
 259
 260 static inline int media_udev_open(struct udev **udev)
 261 {
 262         *udev = udev_new();
 263         if (*udev == NULL)
 264                 return -ENOMEM;
 265         return 0;
 266 }
 267
 268 static inline void media_udev_close(struct udev *udev)
 269 {
 270         if (udev != NULL)
 271                 udev_unref(udev);
 272 }
 273
 274 static int media_get_devname_udev(struct udev *udev,
 275                 struct media_entity *entity, int verbose)
 276 {
 277         struct udev_device *device;
 278         dev_t devnum;
 279         const char *p;
 280         int ret = -ENODEV;
 281
 282         if (udev == NULL)
 283                 return -EINVAL;
 284
 285         devnum = makedev(entity->info.v4l.major, entity->info.v4l.minor);
 286         if (verbose)
 287                 printf("looking up device: %u:%u\n", major(devnum), minor(devnum));
 288         device = udev_device_new_from_devnum(udev, 'c', devnum);
 289         if (device) {
 290                 p = udev_device_get_devnode(device);
 291                 if (p) {
 292                         strncpy(entity->devname, p, sizeof(entity->devname));
 293                         entity->devname[sizeof(entity->devname) - 1] = '\0';
 294                 }
 295                 ret = 0;
 296         }
 297
 298         udev_device_unref(device);
 299
 300         return ret;
 301 }
 302
 303 #else   /* HAVE_LIBUDEV */
 304
 305 struct udev;
 306
 307 static inline int media_udev_open(struct udev **udev) { return 0; }
 308
 309 static inline void media_udev_close(struct udev *udev) { }
 310
 311 static inline int media_get_devname_udev(struct udev *udev,
 312                 struct media_entity *entity, int verbose)
 313 {
 314         return -ENOTSUP;
 315 }
 316
 317 #endif  /* HAVE_LIBUDEV */
 318
 319 static int media_get_devname_sysfs(struct media_entity *entity)
 320 {
 321         struct stat devstat;
 322         char devname[32];
 323         char sysname[32];
 324         char target[1024];
 325         char *p;
 326         int ret;
 327
 328         sprintf(sysname, "/sys/dev/char/%u:%u", entity->info.v4l.major,
 329                 entity->info.v4l.minor);
 330         ret = readlink(sysname, target, sizeof(target));
 331         if (ret < 0)
 332                 return -errno;
 333
 334         target[ret] = '\0';
 335         p = strrchr(target, '/');
 336         if (p == NULL)
 337                 return -EINVAL;
 338
 339         sprintf(devname, "/dev/%s", p + 1);
 340         ret = stat(devname, &devstat);
 341         if (ret < 0)
 342                 return -errno;
 343
 344         /* Sanity check: udev might have reordered the device nodes.
 345          * Make sure the major/minor match. We should really use
 346          * libudev.
 347          */
 348         if (major(devstat.st_rdev) == entity->info.v4l.major &&
 349             minor(devstat.st_rdev) == entity->info.v4l.minor)
 350                 strcpy(entity->devname, devname);
 351
 352         return 0;
 353 }
 354
 355 static int media_enum_entities(struct media_device *media, int verbose)
 356 {
 357         struct media_entity *entity;
 358         struct udev *udev;
 359         unsigned int size;
 360         __u32 id;
 361         int ret;
 362
 363         ret = media_udev_open(&udev);
 364         if (ret < 0)
 365                 printf("%s: Can't get udev context\n", __func__);
 366
 367         for (id = 0, ret = 0; ; id = entity->info.id) {
 368                 size = (media->entities_count + 1) * sizeof(*media->entities);
 369                 media->entities = realloc(media->entities, size);
 370
 371                 entity = &media->entities[media->entities_count];
 372                 memset(entity, 0, sizeof(*entity));
 373                 entity->fd = -1;
 374                 entity->info.id = id | MEDIA_ENT_ID_FLAG_NEXT;
 375
 376                 ret = ioctl(media->fd, MEDIA_IOC_ENUM_ENTITIES, &entity->info);
 377                 if (ret < 0) {
 378                         ret = errno != EINVAL ? -errno : 0;
 379                         break;
 380                 }
 381
 382                 /* Number of links (for outbound links) plus number of pads (for
 383                  * inbound links) is a good safe initial estimate of the total
 384                  * number of links.
 385                  */
 386                 entity->max_links = entity->info.pads + entity->info.links;
 387
 388                 entity->pads = malloc(entity->info.pads * sizeof(*entity->pads));
 389                 entity->links = malloc(entity->max_links * sizeof(*entity->links));
 390                 if (entity->pads == NULL || entity->links == NULL) {
 391                         ret = -ENOMEM;
 392                         break;
 393                 }
 394
 395                 media->entities_count++;
 396
 397                 /* Find the corresponding device name. */
 398                 if (media_entity_type(entity) != MEDIA_ENT_T_DEVNODE &&
 399                     media_entity_type(entity) != MEDIA_ENT_T_V4L2_SUBDEV)
 400                         continue;
 401
 402                 /* Try to get the device name via udev */
 403                 if (!media_get_devname_udev(udev, entity, verbose))
 404                         continue;
 405
 406                 /* Fall back to get the device name via sysfs */
 407                 media_get_devname_sysfs(entity);
 408         }
 409
 410         media_udev_close(udev);
 411         return ret;
 412 }
 413
 414 struct media_device *media_open(const char *name, int verbose)
 415 {
 416         struct media_device *media;
 417         int ret;
 418
 419         media = calloc(1, sizeof(*media));
 420         if (media == NULL) {
 421                 dprintf("%s: unable to allocate memory\n", __func__);
 422                 return NULL;
 423         }
 424
 425         if (verbose)
 426                 dprintf("Opening media device %s\n", name);
 427
 428         media->fd = open(name, O_RDWR);
 429         if (media->fd < 0) {
 430                 media_close(media);
 431                 dprintf("%s: Can't open media device %s\n", __func__, name);
 432                 return NULL;
 433         }
 434
 435         if (verbose)
 436                 dprintf("Enumerating entities\n");
 437
 438         ret = media_enum_entities(media, verbose);
 439
 440         if (ret < 0) {
 441                 dprintf("%s: Unable to enumerate entities for device %s (%s)\n",
 442                         __func__, name, strerror(-ret));
 443                 media_close(media);
 444                 return NULL;
 445         }
 446
 447         if (verbose) {
 448                 dprintf("Found %u entities\n", media->entities_count);
 449                 dprintf("Enumerating pads and links\n");
 450         }
 451
 452         ret = media_enum_links(media);
 453         if (ret < 0) {
 454                 dprintf("%s: Unable to enumerate pads and linksfor device %s\n",
 455                         __func__, name);
 456                 media_close(media);
 457                 return NULL;
 458         }
 459
 460         return media;
 461 }
 462
 463 void media_close(struct media_device *media)
 464 {
 465         unsigned int i;
 466
 467         if (media->fd != -1)
 468                 close(media->fd);
 469
 470         for (i = 0; i < media->entities_count; ++i) {
 471                 struct media_entity *entity = &media->entities[i];
 472
 473                 free(entity->pads);
 474                 free(entity->links);
 475                 if (entity->fd != -1)
 476                         close(entity->fd);
 477         }
 478
 479         free(media->entities);
 480         free(media);
 481 }
 482
 483 struct media_pad *media_parse_pad(struct media_device *media,
 484                                   const char *p, char **endp)
 485 {
 486         unsigned int entity_id, pad;
 487         struct media_entity *entity;
 488         char *end;
 489
 490         for (; isspace(*p); ++p);
 491
 492         if (*p == '"') {
 493                 for (end = (char *)p + 1; *end && *end != '"'; ++end);
 494                 if (*end != '"')
 495                         return NULL;
 496
 497                 entity = media_get_entity_by_name(media, p + 1, end - p - 1);
 498                 if (entity == NULL)
 499                         return NULL;
 500
 501                 ++end;
 502         } else {
 503                 entity_id = strtoul(p, &end, 10);
 504                 entity = media_get_entity_by_id(media, entity_id);
 505                 if (entity == NULL)
 506                         return NULL;
 507         }
 508         for (; isspace(*end); ++end);
 509
 510         if (*end != ':')
 511                 return NULL;
 512         for (p = end + 1; isspace(*p); ++p);
 513
 514         pad = strtoul(p, &end, 10);
 515         for (p = end; isspace(*p); ++p);
 516
 517         if (pad >= entity->info.pads)
 518                 return NULL;
 519
 520         for (p = end; isspace(*p); ++p);
 521         if (endp)
 522                 *endp = (char *)p;
 523
 524         return &entity->pads[pad];
 525 }
 526
 527 struct media_link *media_parse_link(struct media_device *media,
 528                                     const char *p, char **endp)
 529 {
 530         struct media_link *link;
 531         struct media_pad *source;
 532         struct media_pad *sink;
 533         unsigned int i;
 534         char *end;
 535
 536         source = media_parse_pad(media, p, &end);
 537         if (source == NULL)
 538                 return NULL;
 539
 540         if (end[0] != '-' || end[1] != '>')
 541                 return NULL;
 542         p = end + 2;
 543
 544         sink = media_parse_pad(media, p, &end);
 545         if (sink == NULL)
 546                 return NULL;
 547
 548         *endp = end;
 549
 550         for (i = 0; i < source->entity->num_links; i++) {
 551                 link = &source->entity->links[i];
 552
 553                 if (link->source == source && link->sink == sink)
 554                         return link;
 555         }
 556
 557         return NULL;
 558 }
 559
 560 int media_parse_setup_link(struct media_device *media,
 561                            const char *p, char **endp)
 562 {
 563         struct media_link *link;
 564         __u32 flags;
 565         char *end;
 566
 567         link = media_parse_link(media, p, &end);
 568         if (link == NULL) {
 569                 dprintf("Unable to parse link\n");
 570                 return -EINVAL;
 571         }
 572
 573         p = end;
 574         if (*p++ != '[') {
 575                 dprintf("Unable to parse link flags\n");
 576                 return -EINVAL;
 577         }
 578
 579         flags = strtoul(p, &end, 10);
 580         for (p = end; isspace(*p); p++);
 581         if (*p++ != ']') {
 582                 dprintf("Unable to parse link flags\n");
 583                 return -EINVAL;
 584         }
 585
 586         for (; isspace(*p); p++);
 587         *endp = (char *)p;
 588
 589         dprintf("Setting up link %u:%u -> %u:%u [%u]\n",
 590                 link->source->entity->info.id, link->source->index,
 591                 link->sink->entity->info.id, link->sink->index,
 592                 flags);
 593
 594         return media_setup_link(media, link->source, link->sink, flags);
 595 }
 596
 597 int media_parse_setup_links(struct media_device *media, const char *p)
 598 {
 599         char *end;
 600         int ret;
 601
 602         do {
 603                 ret = media_parse_setup_link(media, p, &end);
 604                 if (ret < 0)
 605                         return ret;
 606
 607                 p = end + 1;
 608         } while (*end == ',');
 609
 610         return *end ? -EINVAL : 0;
 611 }