chiark / gitweb /
libudev: test - use epoll
[elogind.git] / libudev / test-libudev.c
1 /*
2  * test-libudev
3  *
4  * Copyright (C) 2008 Kay Sievers <kay.sievers@vrfy.org>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  */
11
12 #include <stdio.h>
13 #include <stdarg.h>
14 #include <stdlib.h>
15 #include <unistd.h>
16 #include <errno.h>
17 #include <string.h>
18 #include <getopt.h>
19 #include <syslog.h>
20 #include <fcntl.h>
21 #include <sys/epoll.h>
22
23 #include "libudev.h"
24
25 #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
26
27 static void log_fn(struct udev *udev,
28                    int priority, const char *file, int line, const char *fn,
29                    const char *format, va_list args)
30 {
31         printf("test-libudev: %s %s:%d ", fn, file, line);
32         vprintf(format, args);
33 }
34
35 static void print_device(struct udev_device *device)
36 {
37         const char *str;
38         dev_t devnum;
39         int count;
40         struct udev_list_entry *list_entry;
41
42         printf("*** device: %p ***\n", device);
43         str = udev_device_get_action(device);
44         if (str != NULL)
45                 printf("action:    '%s'\n", str);
46
47         str = udev_device_get_syspath(device);
48         printf("syspath:   '%s'\n", str);
49
50         str = udev_device_get_sysname(device);
51         printf("sysname:   '%s'\n", str);
52
53         str = udev_device_get_sysnum(device);
54         if (str != NULL)
55                 printf("sysnum:    '%s'\n", str);
56
57         str = udev_device_get_devpath(device);
58         printf("devpath:   '%s'\n", str);
59
60         str = udev_device_get_subsystem(device);
61         if (str != NULL)
62                 printf("subsystem: '%s'\n", str);
63
64         str = udev_device_get_devtype(device);
65         if (str != NULL)
66                 printf("devtype:   '%s'\n", str);
67
68         str = udev_device_get_driver(device);
69         if (str != NULL)
70                 printf("driver:    '%s'\n", str);
71
72         str = udev_device_get_devnode(device);
73         if (str != NULL)
74                 printf("devname:   '%s'\n", str);
75
76         devnum = udev_device_get_devnum(device);
77         if (major(devnum) > 0)
78                 printf("devnum:    %u:%u\n", major(devnum), minor(devnum));
79
80         count = 0;
81         udev_list_entry_foreach(list_entry, udev_device_get_devlinks_list_entry(device)) {
82                 printf("link:      '%s'\n", udev_list_entry_get_name(list_entry));
83                 count++;
84         }
85         if (count > 0)
86                 printf("found %i links\n", count);
87
88         count = 0;
89         udev_list_entry_foreach(list_entry, udev_device_get_properties_list_entry(device)) {
90                 printf("property:  '%s=%s'\n",
91                        udev_list_entry_get_name(list_entry),
92                        udev_list_entry_get_value(list_entry));
93                 count++;
94         }
95         if (count > 0)
96                 printf("found %i properties\n", count);
97
98         str = udev_device_get_property_value(device, "MAJOR");
99         if (str != NULL)
100                 printf("MAJOR: '%s'\n", str);
101
102         str = udev_device_get_sysattr_value(device, "dev");
103         if (str != NULL)
104                 printf("attr{dev}: '%s'\n", str);
105
106         printf("\n");
107 }
108
109 static int test_device(struct udev *udev, const char *syspath)
110 {
111         struct udev_device *device;
112
113         printf("looking at device: %s\n", syspath);
114         device = udev_device_new_from_syspath(udev, syspath);
115         if (device == NULL) {
116                 printf("no device found\n");
117                 return -1;
118         }
119         print_device(device);
120         udev_device_unref(device);
121         return 0;
122 }
123
124 static int test_device_parents(struct udev *udev, const char *syspath)
125 {
126         struct udev_device *device;
127         struct udev_device *device_parent;
128
129         printf("looking at device: %s\n", syspath);
130         device = udev_device_new_from_syspath(udev, syspath);
131         if (device == NULL)
132                 return -1;
133
134         printf("looking at parents\n");
135         device_parent = device;
136         do {
137                 print_device(device_parent);
138                 device_parent = udev_device_get_parent(device_parent);
139         } while (device_parent != NULL);
140
141         printf("looking at parents again\n");
142         device_parent = device;
143         do {
144                 print_device(device_parent);
145                 device_parent = udev_device_get_parent(device_parent);
146         } while (device_parent != NULL);
147         udev_device_unref(device);
148
149         return 0;
150 }
151
152 static int test_device_devnum(struct udev *udev)
153 {
154         dev_t devnum = makedev(1, 3);
155         struct udev_device *device;
156
157         printf("looking up device: %u:%u\n", major(devnum), minor(devnum));
158         device = udev_device_new_from_devnum(udev, 'c', devnum);
159         if (device == NULL)
160                 return -1;
161         print_device(device);
162         udev_device_unref(device);
163         return 0;
164 }
165
166 static int test_device_subsys_name(struct udev *udev)
167 {
168         struct udev_device *device;
169
170         printf("looking up device: 'block':'sda'\n");
171         device = udev_device_new_from_subsystem_sysname(udev, "block", "sda");
172         if (device == NULL)
173                 return -1;
174         print_device(device);
175         udev_device_unref(device);
176
177         printf("looking up device: 'subsystem':'pci'\n");
178         device = udev_device_new_from_subsystem_sysname(udev, "subsystem", "pci");
179         if (device == NULL)
180                 return -1;
181         print_device(device);
182         udev_device_unref(device);
183
184         printf("looking up device: 'drivers':'scsi:sd'\n");
185         device = udev_device_new_from_subsystem_sysname(udev, "drivers", "scsi:sd");
186         if (device == NULL)
187                 return -1;
188         print_device(device);
189         udev_device_unref(device);
190
191         printf("looking up device: 'module':'printk'\n");
192         device = udev_device_new_from_subsystem_sysname(udev, "module", "printk");
193         if (device == NULL)
194                 return -1;
195         print_device(device);
196         udev_device_unref(device);
197         return 0;
198 }
199
200 static int test_enumerate_print_list(struct udev_enumerate *enumerate)
201 {
202         struct udev_list_entry *list_entry;
203         int count = 0;
204
205         udev_list_entry_foreach(list_entry, udev_enumerate_get_list_entry(enumerate)) {
206                 struct udev_device *device;
207
208                 device = udev_device_new_from_syspath(udev_enumerate_get_udev(enumerate),
209                                                       udev_list_entry_get_name(list_entry));
210                 if (device != NULL) {
211                         printf("device: '%s' (%s)\n",
212                                udev_device_get_syspath(device),
213                                udev_device_get_subsystem(device));
214                         udev_device_unref(device);
215                         count++;
216                 }
217         }
218         printf("found %i devices\n\n", count);
219         return count;
220 }
221
222 static int test_monitor(struct udev *udev)
223 {
224         struct udev_monitor *udev_monitor;
225         int fd_ep;
226         int fd_udev = -1;
227         struct epoll_event ep_udev, ep_stdin;
228
229         fd_ep = epoll_create1(EPOLL_CLOEXEC);
230         if (fd_ep < 0) {
231                 printf("error creating epoll fd: %m\n");
232                 goto out;
233         }
234
235         udev_monitor = udev_monitor_new_from_netlink(udev, "udev");
236         if (udev_monitor == NULL) {
237                 printf("no socket\n");
238                 goto out;
239         }
240         fd_udev = udev_monitor_get_fd(udev_monitor);
241
242         if (udev_monitor_filter_add_match_subsystem_devtype(udev_monitor, "block", NULL) < 0 ||
243             udev_monitor_filter_add_match_subsystem_devtype(udev_monitor, "tty", NULL) < 0 ||
244             udev_monitor_filter_add_match_subsystem_devtype(udev_monitor, "usb", "usb_device") < 0) {
245                 printf("filter failed\n");
246                 goto out;
247         }
248
249         if (udev_monitor_enable_receiving(udev_monitor) < 0) {
250                 printf("bind failed\n");
251                 goto out;
252         }
253
254         memset(&ep_udev, 0, sizeof(struct epoll_event));
255         ep_udev.events = EPOLLIN;
256         ep_udev.data.fd = fd_udev;
257         if (epoll_ctl(fd_ep, EPOLL_CTL_ADD, fd_udev, &ep_udev) < 0) {
258                 printf("fail to add fd to epoll: %m\n");
259                 goto out;
260         }
261
262         memset(&ep_stdin, 0, sizeof(struct epoll_event));
263         ep_stdin.events = EPOLLIN;
264         ep_stdin.data.fd = STDIN_FILENO;
265         if (epoll_ctl(fd_ep, EPOLL_CTL_ADD, STDIN_FILENO, &ep_stdin) < 0) {
266                 printf("fail to add fd to epoll: %m\n");
267                 goto out;
268         }
269
270         for (;;) {
271                 int fdcount;
272                 struct epoll_event ev[4];
273                 struct udev_device *device;
274                 int i;
275
276                 printf("waiting for events from udev, press ENTER to exit\n");
277                 fdcount = epoll_wait(fd_ep, ev, ARRAY_SIZE(ev), -1);
278                 printf("epoll fd count: %i\n", fdcount);
279
280                 for (i = 0; i < fdcount; i++) {
281                         if (ev[i].data.fd == fd_udev && ev[i].events & EPOLLIN) {
282                                 device = udev_monitor_receive_device(udev_monitor);
283                                 if (device == NULL) {
284                                         printf("no device from socket\n");
285                                         continue;
286                                 }
287                                 print_device(device);
288                                 udev_device_unref(device);
289                         }
290
291                         if (ev[i].data.fd == STDIN_FILENO && ev[i].events & EPOLLIN) {
292                                 printf("exiting loop\n");
293                                 goto out;
294                         }
295                 }
296         }
297 out:
298         if (fd_ep >= 0)
299                 close(fd_ep);
300         udev_monitor_unref(udev_monitor);
301         return 0;
302 }
303
304 static int test_queue(struct udev *udev)
305 {
306         struct udev_queue *udev_queue;
307         unsigned long long int seqnum;
308         struct udev_list_entry *list_entry;
309
310         udev_queue = udev_queue_new(udev);
311         if (udev_queue == NULL)
312                 return -1;
313         seqnum = udev_queue_get_kernel_seqnum(udev_queue);
314         printf("seqnum kernel: %llu\n", seqnum);
315         seqnum = udev_queue_get_udev_seqnum(udev_queue);
316         printf("seqnum udev  : %llu\n", seqnum);
317
318         if (udev_queue_get_queue_is_empty(udev_queue))
319                 printf("queue is empty\n");
320         printf("get queue list\n");
321         udev_list_entry_foreach(list_entry, udev_queue_get_queued_list_entry(udev_queue))
322                 printf("queued: '%s' [%s]\n", udev_list_entry_get_name(list_entry), udev_list_entry_get_value(list_entry));
323         printf("\n");
324         printf("get queue list again\n");
325         udev_list_entry_foreach(list_entry, udev_queue_get_queued_list_entry(udev_queue))
326                 printf("queued: '%s' [%s]\n", udev_list_entry_get_name(list_entry), udev_list_entry_get_value(list_entry));
327         printf("\n");
328         printf("get failed list\n");
329         udev_list_entry_foreach(list_entry, udev_queue_get_failed_list_entry(udev_queue))
330                 printf("failed: '%s'\n", udev_list_entry_get_name(list_entry));
331         printf("\n");
332
333         list_entry = udev_queue_get_queued_list_entry(udev_queue);
334         if (list_entry != NULL) {
335                 printf("event [%llu] is queued\n", seqnum);
336                 seqnum = strtoull(udev_list_entry_get_value(list_entry), NULL, 10);
337                 if (udev_queue_get_seqnum_is_finished(udev_queue, seqnum))
338                         printf("event [%llu] is not finished\n", seqnum);
339                 else
340                         printf("event [%llu] is finished\n", seqnum);
341         }
342         printf("\n");
343         udev_queue_unref(udev_queue);
344         return 0;
345 }
346
347 static int test_enumerate(struct udev *udev, const char *subsystem)
348 {
349         struct udev_enumerate *udev_enumerate;
350
351         printf("enumerate '%s'\n", subsystem == NULL ? "<all>" : subsystem);
352         udev_enumerate = udev_enumerate_new(udev);
353         if (udev_enumerate == NULL)
354                 return -1;
355         udev_enumerate_add_match_subsystem(udev_enumerate, subsystem);
356         udev_enumerate_scan_devices(udev_enumerate);
357         test_enumerate_print_list(udev_enumerate);
358         udev_enumerate_unref(udev_enumerate);
359
360         printf("enumerate 'net' + duplicated scan + null + zero\n");
361         udev_enumerate = udev_enumerate_new(udev);
362         if (udev_enumerate == NULL)
363                 return -1;
364         udev_enumerate_add_match_subsystem(udev_enumerate, "net");
365         udev_enumerate_scan_devices(udev_enumerate);
366         udev_enumerate_scan_devices(udev_enumerate);
367         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/zero");
368         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/null");
369         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/zero");
370         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/null");
371         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/zero");
372         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/null");
373         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/null");
374         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/zero");
375         udev_enumerate_add_syspath(udev_enumerate, "/sys/class/mem/zero");
376         udev_enumerate_scan_devices(udev_enumerate);
377         test_enumerate_print_list(udev_enumerate);
378         udev_enumerate_unref(udev_enumerate);
379
380         printf("enumerate 'block'\n");
381         udev_enumerate = udev_enumerate_new(udev);
382         if (udev_enumerate == NULL)
383                 return -1;
384         udev_enumerate_add_match_subsystem(udev_enumerate,"block");
385         udev_enumerate_add_match_is_initialized(udev_enumerate);
386         udev_enumerate_scan_devices(udev_enumerate);
387         test_enumerate_print_list(udev_enumerate);
388         udev_enumerate_unref(udev_enumerate);
389
390         printf("enumerate 'not block'\n");
391         udev_enumerate = udev_enumerate_new(udev);
392         if (udev_enumerate == NULL)
393                 return -1;
394         udev_enumerate_add_nomatch_subsystem(udev_enumerate, "block");
395         udev_enumerate_scan_devices(udev_enumerate);
396         test_enumerate_print_list(udev_enumerate);
397         udev_enumerate_unref(udev_enumerate);
398
399         printf("enumerate 'pci, mem, vc'\n");
400         udev_enumerate = udev_enumerate_new(udev);
401         if (udev_enumerate == NULL)
402                 return -1;
403         udev_enumerate_add_match_subsystem(udev_enumerate, "pci");
404         udev_enumerate_add_match_subsystem(udev_enumerate, "mem");
405         udev_enumerate_add_match_subsystem(udev_enumerate, "vc");
406         udev_enumerate_scan_devices(udev_enumerate);
407         test_enumerate_print_list(udev_enumerate);
408         udev_enumerate_unref(udev_enumerate);
409
410         printf("enumerate 'subsystem'\n");
411         udev_enumerate = udev_enumerate_new(udev);
412         if (udev_enumerate == NULL)
413                 return -1;
414         udev_enumerate_scan_subsystems(udev_enumerate);
415         test_enumerate_print_list(udev_enumerate);
416         udev_enumerate_unref(udev_enumerate);
417
418         printf("enumerate 'property IF_FS_*=filesystem'\n");
419         udev_enumerate = udev_enumerate_new(udev);
420         if (udev_enumerate == NULL)
421                 return -1;
422         udev_enumerate_add_match_property(udev_enumerate, "ID_FS*", "filesystem");
423         udev_enumerate_scan_devices(udev_enumerate);
424         test_enumerate_print_list(udev_enumerate);
425         udev_enumerate_unref(udev_enumerate);
426         return 0;
427 }
428
429 int main(int argc, char *argv[])
430 {
431         struct udev *udev = NULL;
432         static const struct option options[] = {
433                 { "syspath", required_argument, NULL, 'p' },
434                 { "subsystem", required_argument, NULL, 's' },
435                 { "debug", no_argument, NULL, 'd' },
436                 { "help", no_argument, NULL, 'h' },
437                 { "version", no_argument, NULL, 'V' },
438                 {}
439         };
440         const char *syspath = "/devices/virtual/mem/null";
441         const char *subsystem = NULL;
442         char path[1024];
443         const char *str;
444
445         udev = udev_new();
446         printf("context: %p\n", udev);
447         if (udev == NULL) {
448                 printf("no context\n");
449                 return 1;
450         }
451         udev_set_log_fn(udev, log_fn);
452         printf("set log: %p\n", log_fn);
453
454         for (;;) {
455                 int option;
456
457                 option = getopt_long(argc, argv, "+p:s:dhV", options, NULL);
458                 if (option == -1)
459                         break;
460
461                 switch (option) {
462                 case 'p':
463                         syspath = optarg;
464                         break;
465                 case 's':
466                         subsystem = optarg;
467                         break;
468                 case 'd':
469                         if (udev_get_log_priority(udev) < LOG_INFO)
470                                 udev_set_log_priority(udev, LOG_INFO);
471                         break;
472                 case 'h':
473                         printf("--debug --syspath= --subsystem= --help\n");
474                         goto out;
475                 case 'V':
476                         printf("%s\n", VERSION);
477                         goto out;
478                 default:
479                         goto out;
480                 }
481         }
482
483         str = udev_get_sys_path(udev);
484         printf("sys_path: '%s'\n", str);
485         str = udev_get_dev_path(udev);
486         printf("dev_path: '%s'\n", str);
487
488         /* add sys path if needed */
489         if (strncmp(syspath, udev_get_sys_path(udev), strlen(udev_get_sys_path(udev))) != 0) {
490                 snprintf(path, sizeof(path), "%s%s", udev_get_sys_path(udev), syspath);
491                 syspath = path;
492         }
493
494         test_device(udev, syspath);
495         test_device_devnum(udev);
496         test_device_subsys_name(udev);
497         test_device_parents(udev, syspath);
498
499         test_enumerate(udev, subsystem);
500
501         test_queue(udev);
502
503         test_monitor(udev);
504 out:
505         udev_unref(udev);
506         return 0;
507 }