chiark / gitweb /
85828e32d72c22362e6a7ccaf5cc3014d653ee6e
[elogind.git] / src / udev / udev-builtin-usb_id.c
1 /*
2  * USB device properties and persistent device path
3  *
4  * Copyright (c) 2005 SUSE Linux Products GmbH, Germany
5  *   Author: Hannes Reinecke <hare@suse.de>
6  *
7  * Copyright (C) 2005-2011 Kay Sievers <kay.sievers@vrfy.org>
8  *
9  * This program is free software: you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation, either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
21  */
22
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <stdarg.h>
26 #include <unistd.h>
27 #include <string.h>
28 #include <ctype.h>
29 #include <fcntl.h>
30 #include <errno.h>
31
32 #include "udev.h"
33
34 static void set_usb_iftype(char *to, int if_class_num, size_t len)
35 {
36         char *type = "generic";
37
38         switch (if_class_num) {
39         case 1:
40                 type = "audio";
41                 break;
42         case 2: /* CDC-Control */
43                 break;
44         case 3:
45                 type = "hid";
46                 break;
47         case 5: /* Physical */
48                 break;
49         case 6:
50                 type = "media";
51                 break;
52         case 7:
53                 type = "printer";
54                 break;
55         case 8:
56                 type = "storage";
57                 break;
58         case 9:
59                 type = "hub";
60                 break;
61         case 0x0a: /* CDC-Data */
62                 break;
63         case 0x0b: /* Chip/Smart Card */
64                 break;
65         case 0x0d: /* Content Security */
66                 break;
67         case 0x0e:
68                 type = "video";
69                 break;
70         case 0xdc: /* Diagnostic Device */
71                 break;
72         case 0xe0: /* Wireless Controller */
73                 break;
74         case 0xfe: /* Application-specific */
75                 break;
76         case 0xff: /* Vendor-specific */
77                 break;
78         default:
79                 break;
80         }
81         strncpy(to, type, len);
82         to[len-1] = '\0';
83 }
84
85 static int set_usb_mass_storage_ifsubtype(char *to, const char *from, size_t len)
86 {
87         int type_num = 0;
88         char *eptr;
89         char *type = "generic";
90
91         type_num = strtoul(from, &eptr, 0);
92         if (eptr != from) {
93                 switch (type_num) {
94                 case 2:
95                         type = "atapi";
96                         break;
97                 case 3:
98                         type = "tape";
99                         break;
100                 case 4: /* UFI */
101                 case 5: /* SFF-8070i */
102                         type = "floppy";
103                         break;
104                 case 1: /* RBC devices */
105                         type = "rbc";
106                         break;
107                 case 6: /* Transparent SPC-2 devices */
108                         type = "scsi";
109                         break;
110                 default:
111                         break;
112                 }
113         }
114         util_strscpy(to, len, type);
115         return type_num;
116 }
117
118 static void set_scsi_type(char *to, const char *from, size_t len)
119 {
120         int type_num;
121         char *eptr;
122         char *type = "generic";
123
124         type_num = strtoul(from, &eptr, 0);
125         if (eptr != from) {
126                 switch (type_num) {
127                 case 0:
128                 case 0xe:
129                         type = "disk";
130                         break;
131                 case 1:
132                         type = "tape";
133                         break;
134                 case 4:
135                 case 7:
136                 case 0xf:
137                         type = "optical";
138                         break;
139                 case 5:
140                         type = "cd";
141                         break;
142                 default:
143                         break;
144                 }
145         }
146         util_strscpy(to, len, type);
147 }
148
149 #define USB_DT_DEVICE                        0x01
150 #define USB_DT_INTERFACE                0x04
151
152 static int dev_if_packed_info(struct udev_device *dev, char *ifs_str, size_t len)
153 {
154         char *filename = NULL;
155         int fd;
156         ssize_t size;
157         unsigned char buf[18 + 65535];
158         unsigned int pos, strpos;
159         struct usb_interface_descriptor {
160                 u_int8_t        bLength;
161                 u_int8_t        bDescriptorType;
162                 u_int8_t        bInterfaceNumber;
163                 u_int8_t        bAlternateSetting;
164                 u_int8_t        bNumEndpoints;
165                 u_int8_t        bInterfaceClass;
166                 u_int8_t        bInterfaceSubClass;
167                 u_int8_t        bInterfaceProtocol;
168                 u_int8_t        iInterface;
169         } __attribute__((packed));
170         int err = 0;
171
172         if (asprintf(&filename, "%s/descriptors", udev_device_get_syspath(dev)) < 0) {
173                 err = -1;
174                 goto out;
175         }
176         fd = open(filename, O_RDONLY|O_CLOEXEC);
177         if (fd < 0) {
178                 fprintf(stderr, "error opening USB device 'descriptors' file\n");
179                 err = -1;
180                 goto out;
181         }
182         size = read(fd, buf, sizeof(buf));
183         close(fd);
184         if (size < 18 || size == sizeof(buf)) {
185                 err = -1;
186                 goto out;
187         }
188
189         pos = 0;
190         strpos = 0;
191         ifs_str[0] = '\0';
192         while (pos < sizeof(buf) && strpos+7 < len-2) {
193                 struct usb_interface_descriptor *desc;
194                 char if_str[8];
195
196                 desc = (struct usb_interface_descriptor *) &buf[pos];
197                 if (desc->bLength < 3)
198                         break;
199                 pos += desc->bLength;
200
201                 if (desc->bDescriptorType != USB_DT_INTERFACE)
202                         continue;
203
204                 if (snprintf(if_str, 8, ":%02x%02x%02x",
205                              desc->bInterfaceClass,
206                              desc->bInterfaceSubClass,
207                              desc->bInterfaceProtocol) != 7)
208                         continue;
209
210                 if (strstr(ifs_str, if_str) != NULL)
211                         continue;
212
213                 memcpy(&ifs_str[strpos], if_str, 8),
214                 strpos += 7;
215         }
216         if (strpos > 0) {
217                 ifs_str[strpos++] = ':';
218                 ifs_str[strpos++] = '\0';
219         }
220 out:
221         free(filename);
222         return err;
223 }
224
225 /*
226  * A unique USB identification is generated like this:
227  *
228  * 1.) Get the USB device type from InterfaceClass and InterfaceSubClass
229  * 2.) If the device type is 'Mass-Storage/SPC-2' or 'Mass-Storage/RBC'
230  *     use the SCSI vendor and model as USB-Vendor and USB-model.
231  * 3.) Otherwise use the USB manufacturer and product as
232  *     USB-Vendor and USB-model. Any non-printable characters
233  *     in those strings will be skipped; a slash '/' will be converted
234  *     into a full stop '.'.
235  * 4.) If that fails, too, we will use idVendor and idProduct
236  *     as USB-Vendor and USB-model.
237  * 5.) The USB identification is the USB-vendor and USB-model
238  *     string concatenated with an underscore '_'.
239  * 6.) If the device supplies a serial number, this number
240  *     is concatenated with the identification with an underscore '_'.
241  */
242 static int builtin_usb_id(struct udev_device *dev, int argc, char *argv[], bool test)
243 {
244         char vendor_str[64];
245         char vendor_str_enc[256];
246         const char *vendor_id;
247         char model_str[64];
248         char model_str_enc[256];
249         const char *product_id;
250         char serial_str[UTIL_NAME_SIZE];
251         char packed_if_str[UTIL_NAME_SIZE];
252         char revision_str[64];
253         char type_str[64];
254         char instance_str[64];
255         const char *ifnum = NULL;
256         const char *driver = NULL;
257         char serial[256];
258
259         struct udev *udev = udev_device_get_udev(dev);
260         struct udev_device *dev_interface = NULL;
261         struct udev_device *dev_usb = NULL;
262         const char *if_class, *if_subclass;
263         int if_class_num;
264         int protocol = 0;
265         size_t l;
266         char *s;
267
268         vendor_str[0] = '\0';
269         model_str[0] = '\0';
270         serial_str[0] = '\0';
271         packed_if_str[0] = '\0';
272         revision_str[0] = '\0';
273         type_str[0] = '\0';
274         instance_str[0] = '\0';
275
276         dbg(udev, "syspath %s\n", udev_device_get_syspath(dev));
277
278         /* shortcut, if we are called directly for a "usb_device" type */
279         if (udev_device_get_devtype(dev) != NULL && strcmp(udev_device_get_devtype(dev), "usb_device") == 0) {
280                 dev_if_packed_info(dev, packed_if_str, sizeof(packed_if_str));
281                 dev_usb = dev;
282                 goto fallback;
283         }
284
285         /* usb interface directory */
286         dev_interface = udev_device_get_parent_with_subsystem_devtype(dev, "usb", "usb_interface");
287         if (dev_interface == NULL) {
288                 info(udev, "unable to access usb_interface device of '%s'\n",
289                      udev_device_get_syspath(dev));
290                 return EXIT_FAILURE;
291         }
292
293         ifnum = udev_device_get_sysattr_value(dev_interface, "bInterfaceNumber");
294         driver = udev_device_get_sysattr_value(dev_interface, "driver");
295
296         if_class = udev_device_get_sysattr_value(dev_interface, "bInterfaceClass");
297         if (!if_class) {
298                 info(udev, "%s: cannot get bInterfaceClass attribute\n",
299                      udev_device_get_sysname(dev));
300                 return EXIT_FAILURE;
301         }
302
303         if_class_num = strtoul(if_class, NULL, 16);
304         if (if_class_num == 8) {
305                 /* mass storage */
306                 if_subclass = udev_device_get_sysattr_value(dev_interface, "bInterfaceSubClass");
307                 if (if_subclass != NULL)
308                         protocol = set_usb_mass_storage_ifsubtype(type_str, if_subclass, sizeof(type_str)-1);
309         } else {
310                 set_usb_iftype(type_str, if_class_num, sizeof(type_str)-1);
311         }
312
313         info(udev, "%s: if_class %d protocol %d\n",
314              udev_device_get_syspath(dev_interface), if_class_num, protocol);
315
316         /* usb device directory */
317         dev_usb = udev_device_get_parent_with_subsystem_devtype(dev_interface, "usb", "usb_device");
318         if (!dev_usb) {
319                 info(udev, "unable to find parent 'usb' device of '%s'\n",
320                      udev_device_get_syspath(dev));
321                 return EXIT_FAILURE;
322         }
323
324         /* all interfaces of the device in a single string */
325         dev_if_packed_info(dev_usb, packed_if_str, sizeof(packed_if_str));
326
327         /* mass storage : SCSI or ATAPI */
328         if ((protocol == 6 || protocol == 2)) {
329                 struct udev_device *dev_scsi;
330                 const char *scsi_model, *scsi_vendor, *scsi_type, *scsi_rev;
331                 int host, bus, target, lun;
332
333                 /* get scsi device */
334                 dev_scsi = udev_device_get_parent_with_subsystem_devtype(dev, "scsi", "scsi_device");
335                 if (dev_scsi == NULL) {
336                         info(udev, "unable to find parent 'scsi' device of '%s'\n",
337                              udev_device_get_syspath(dev));
338                         goto fallback;
339                 }
340                 if (sscanf(udev_device_get_sysname(dev_scsi), "%d:%d:%d:%d", &host, &bus, &target, &lun) != 4) {
341                         info(udev, "invalid scsi device '%s'\n", udev_device_get_sysname(dev_scsi));
342                         goto fallback;
343                 }
344
345                 /* Generic SPC-2 device */
346                 scsi_vendor = udev_device_get_sysattr_value(dev_scsi, "vendor");
347                 if (!scsi_vendor) {
348                         info(udev, "%s: cannot get SCSI vendor attribute\n",
349                              udev_device_get_sysname(dev_scsi));
350                         goto fallback;
351                 }
352                 udev_util_encode_string(scsi_vendor, vendor_str_enc, sizeof(vendor_str_enc));
353                 util_replace_whitespace(scsi_vendor, vendor_str, sizeof(vendor_str)-1);
354                 util_replace_chars(vendor_str, NULL);
355
356                 scsi_model = udev_device_get_sysattr_value(dev_scsi, "model");
357                 if (!scsi_model) {
358                         info(udev, "%s: cannot get SCSI model attribute\n",
359                              udev_device_get_sysname(dev_scsi));
360                         goto fallback;
361                 }
362                 udev_util_encode_string(scsi_model, model_str_enc, sizeof(model_str_enc));
363                 util_replace_whitespace(scsi_model, model_str, sizeof(model_str)-1);
364                 util_replace_chars(model_str, NULL);
365
366                 scsi_type = udev_device_get_sysattr_value(dev_scsi, "type");
367                 if (!scsi_type) {
368                         info(udev, "%s: cannot get SCSI type attribute\n",
369                              udev_device_get_sysname(dev_scsi));
370                         goto fallback;
371                 }
372                 set_scsi_type(type_str, scsi_type, sizeof(type_str)-1);
373
374                 scsi_rev = udev_device_get_sysattr_value(dev_scsi, "rev");
375                 if (!scsi_rev) {
376                         info(udev, "%s: cannot get SCSI revision attribute\n",
377                              udev_device_get_sysname(dev_scsi));
378                         goto fallback;
379                 }
380                 util_replace_whitespace(scsi_rev, revision_str, sizeof(revision_str)-1);
381                 util_replace_chars(revision_str, NULL);
382
383                 /*
384                  * some broken devices have the same identifiers
385                  * for all luns, export the target:lun number
386                  */
387                 sprintf(instance_str, "%d:%d", target, lun);
388         }
389
390 fallback:
391         vendor_id = udev_device_get_sysattr_value(dev_usb, "idVendor");
392         product_id = udev_device_get_sysattr_value(dev_usb, "idProduct");
393
394         /* fallback to USB vendor & device */
395         if (vendor_str[0] == '\0') {
396                 const char *usb_vendor = NULL;
397
398                 usb_vendor = udev_device_get_sysattr_value(dev_usb, "manufacturer");
399                 if (!usb_vendor)
400                         usb_vendor = vendor_id;
401                 if (!usb_vendor) {
402                         info(udev, "No USB vendor information available\n");
403                         return EXIT_FAILURE;
404                 }
405                 udev_util_encode_string(usb_vendor, vendor_str_enc, sizeof(vendor_str_enc));
406                 util_replace_whitespace(usb_vendor, vendor_str, sizeof(vendor_str)-1);
407                 util_replace_chars(vendor_str, NULL);
408         }
409
410         if (model_str[0] == '\0') {
411                 const char *usb_model = NULL;
412
413                 usb_model = udev_device_get_sysattr_value(dev_usb, "product");
414                 if (!usb_model)
415                         usb_model = product_id;
416                 if (!usb_model) {
417                         dbg(udev, "No USB model information available\n");
418                         return EXIT_FAILURE;
419                 }
420                 udev_util_encode_string(usb_model, model_str_enc, sizeof(model_str_enc));
421                 util_replace_whitespace(usb_model, model_str, sizeof(model_str)-1);
422                 util_replace_chars(model_str, NULL);
423         }
424
425         if (revision_str[0] == '\0') {
426                 const char *usb_rev;
427
428                 usb_rev = udev_device_get_sysattr_value(dev_usb, "bcdDevice");
429                 if (usb_rev) {
430                         util_replace_whitespace(usb_rev, revision_str, sizeof(revision_str)-1);
431                         util_replace_chars(revision_str, NULL);
432                 }
433         }
434
435         if (serial_str[0] == '\0') {
436                 const char *usb_serial;
437
438                 usb_serial = udev_device_get_sysattr_value(dev_usb, "serial");
439                 if (usb_serial) {
440                         util_replace_whitespace(usb_serial, serial_str, sizeof(serial_str)-1);
441                         util_replace_chars(serial_str, NULL);
442                 }
443         }
444
445         s = serial;
446         l = util_strpcpyl(&s, sizeof(serial), vendor_str, "_", model_str, NULL);
447         if (serial_str[0] != '\0')
448                 l = util_strpcpyl(&s, l, "_", serial_str, NULL);
449
450         if (instance_str[0] != '\0')
451                 util_strpcpyl(&s, l, "-", instance_str, NULL);
452
453         udev_builtin_add_property(dev, test, "ID_VENDOR", vendor_str);
454         udev_builtin_add_property(dev, test, "ID_VENDOR_ENC", vendor_str_enc);
455         udev_builtin_add_property(dev, test, "ID_VENDOR_ID", vendor_id);
456         udev_builtin_add_property(dev, test, "ID_MODEL", model_str);
457         udev_builtin_add_property(dev, test, "ID_MODEL_ENC", model_str_enc);
458         udev_builtin_add_property(dev, test, "ID_MODEL_ID", product_id);
459         udev_builtin_add_property(dev, test, "ID_REVISION", revision_str);
460         udev_builtin_add_property(dev, test, "ID_SERIAL", serial);
461         if (serial_str[0] != '\0')
462                 udev_builtin_add_property(dev, test, "ID_SERIAL_SHORT", serial_str);
463         if (type_str[0] != '\0')
464                 udev_builtin_add_property(dev, test, "ID_TYPE", type_str);
465         if (instance_str[0] != '\0')
466                 udev_builtin_add_property(dev, test, "ID_INSTANCE", instance_str);
467         udev_builtin_add_property(dev, test, "ID_BUS", "usb");
468         if (packed_if_str[0] != '\0')
469                 udev_builtin_add_property(dev, test, "ID_USB_INTERFACES", packed_if_str);
470         if (ifnum != NULL)
471                 udev_builtin_add_property(dev, test, "ID_USB_INTERFACE_NUM", ifnum);
472         if (driver != NULL)
473                 udev_builtin_add_property(dev, test, "ID_USB_DRIVER", driver);
474         return EXIT_SUCCESS;
475 }
476
477 const struct udev_builtin udev_builtin_usb_id = {
478         .name = "usb_id",
479         .cmd = builtin_usb_id,
480         .help = "usb device properties",
481         .run_once = true,
482 };