7c3adbe3389c532b28bc6d5490e92b103466ddb3
[media-ctl.git] / 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 }