chiark / gitweb /
don't include stropts.h, some libc's don't like it
[elogind.git] / udev_rules_parse.c
1 /*
2  * udev_rules_parse.c
3  *
4  * Copyright (C) 2003,2004 Greg Kroah-Hartman <greg@kroah.com>
5  * Copyright (C) 2003-2005 Kay Sievers <kay.sievers@vrfy.org>
6  *
7  *      This program is free software; you can redistribute it and/or modify it
8  *      under the terms of the GNU General Public License as published by the
9  *      Free Software Foundation version 2 of the License.
10  * 
11  *      This program is distributed in the hope that it will be useful, but
12  *      WITHOUT ANY WARRANTY; without even the implied warranty of
13  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  *      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  *      675 Mass Ave, Cambridge, MA 02139, USA.
19  *
20  */
21
22 #include <stddef.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <stdio.h>
26 #include <ctype.h>
27 #include <unistd.h>
28 #include <sys/stat.h>
29 #include <errno.h>
30
31 #include "udev.h"
32 #include "udev_rules.h"
33
34
35 void udev_rules_iter_init(struct udev_rules *rules)
36 {
37         dbg("bufsize=%zi", rules->bufsize);
38         rules->current = 0;
39 }
40
41 struct udev_rule *udev_rules_iter_next(struct udev_rules *rules)
42 {
43         static struct udev_rule *rule;
44
45         if (!rules)
46                 return NULL;
47
48         dbg("current=%zi", rules->current);
49         if (rules->current >= rules->bufsize) {
50                 dbg("no more rules");
51                 return NULL;
52         }
53
54         /* get next rule */
55         rule = (struct udev_rule *) (rules->buf + rules->current);
56         rules->current += sizeof(struct udev_rule) + rule->bufsize;
57
58         return rule;
59 }
60
61 struct udev_rule *udev_rules_iter_label(struct udev_rules *rules, const char *label)
62 {
63         static struct udev_rule *rule;
64
65 next:
66         dbg("current=%zi", rules->current);
67         if (rules->current >= rules->bufsize) {
68                 dbg("no more rules");
69                 return NULL;
70         }
71         rule = (struct udev_rule *) (rules->buf + rules->current);
72
73         if (strcmp(&rule->buf[rule->label.val_off], label) != 0) {
74                 dbg("moving forward, looking for label '%s'", label);
75                 rules->current += sizeof(struct udev_rule) + rule->bufsize;
76                 goto next;
77         }
78
79         dbg("found label '%s'", label);
80         return rule;
81 }
82
83 static int get_key(char **line, char **key, enum key_operation *operation, char **value)
84 {
85         char *linepos;
86         char *temp;
87
88         linepos = *line;
89         if (linepos == NULL && linepos[0] == '\0')
90                 return -1;
91
92         /* skip whitespace */
93         while (isspace(linepos[0]) || linepos[0] == ',')
94                 linepos++;
95
96         /* get the key */
97         if (linepos[0] == '\0')
98                 return -1;
99         *key = linepos;
100
101         while (1) {
102                 linepos++;
103                 if (linepos[0] == '\0')
104                         return -1;
105                 if (isspace(linepos[0]))
106                         break;
107                 if (linepos[0] == '=')
108                         break;
109                 if (linepos[0] == '+')
110                         break;
111                 if (linepos[0] == '!')
112                         break;
113                 if (linepos[0] == ':')
114                         break;
115         }
116
117         /* remember end of key */
118         temp = linepos;
119
120         /* skip whitespace after key */
121         while (isspace(linepos[0]))
122                 linepos++;
123         if (linepos[0] == '\0')
124                 return -1;
125
126         /* get operation type */
127         if (linepos[0] == '=' && linepos[1] == '=') {
128                 *operation = KEY_OP_MATCH;
129                 linepos += 2;
130                 dbg("operator=match");
131         } else if (linepos[0] == '!' && linepos[1] == '=') {
132                 *operation = KEY_OP_NOMATCH;
133                 linepos += 2;
134                 dbg("operator=nomatch");
135         } else if (linepos[0] == '+' && linepos[1] == '=') {
136                 *operation = KEY_OP_ADD;
137                 linepos += 2;
138                 dbg("operator=add");
139         } else if (linepos[0] == '=') {
140                 *operation = KEY_OP_ASSIGN;
141                 linepos++;
142                 dbg("operator=assign");
143         } else if (linepos[0] == ':' && linepos[1] == '=') {
144                 *operation = KEY_OP_ASSIGN_FINAL;
145                 linepos += 2;
146                 dbg("operator=assign_final");
147         } else
148                 return -1;
149
150         /* terminate key */
151         temp[0] = '\0';
152         dbg("key='%s'", *key);
153
154         /* skip whitespace after operator */
155         while (isspace(linepos[0]))
156                 linepos++;
157         if (linepos[0] == '\0')
158                 return -1;
159
160         /* get the value*/
161         if (linepos[0] == '"')
162                 linepos++;
163         else
164                 return -1;
165         *value = linepos;
166
167         temp = strchr(linepos, '"');
168         if (!temp)
169                 return -1;
170         temp[0] = '\0';
171         temp++;
172         dbg("value='%s'", *value);
173
174         /* move line to next key */
175         *line = temp;
176
177         return 0;
178 }
179
180 /* extract possible KEY{attr} */
181 static char *get_key_attribute(char *str)
182 {
183         char *pos;
184         char *attr;
185
186         attr = strchr(str, '{');
187         if (attr != NULL) {
188                 attr++;
189                 pos = strchr(attr, '}');
190                 if (pos == NULL) {
191                         err("missing closing brace for format");
192                         return NULL;
193                 }
194                 pos[0] = '\0';
195                 dbg("attribute='%s'", attr);
196                 return attr;
197         }
198
199         return NULL;
200 }
201
202 static int add_rule_key(struct udev_rule *rule, struct key *key,
203                         enum key_operation operation, const char *value)
204 {
205         size_t val_len = strnlen(value, PATH_SIZE);
206
207         key->operation = operation;
208
209         key->val_off = rule->bufsize;
210         strlcpy(rule->buf + rule->bufsize, value, val_len+1);
211         rule->bufsize += val_len+1;
212
213         return 0;
214 }
215
216 static int add_rule_key_pair(struct udev_rule *rule, struct key_pairs *pairs,
217                              enum key_operation operation, const char *key, const char *value)
218 {
219         size_t key_len = strnlen(key, PATH_SIZE);
220
221         if (pairs->count >= PAIRS_MAX) {
222                 err("skip, too many keys in a single rule");
223                 return -1;
224         }
225
226         add_rule_key(rule, &pairs->keys[pairs->count].key, operation, value);
227
228         /* add the key-name of the pair */
229         pairs->keys[pairs->count].key_name_off = rule->bufsize;
230         strlcpy(rule->buf + rule->bufsize, key, key_len+1);
231         rule->bufsize += key_len+1;
232
233         pairs->count++;
234
235         return 0;
236 }
237
238 static int add_to_rules(struct udev_rules *rules, char *line, const char *filename, unsigned int lineno)
239 {
240         struct udev_rule *rule;
241         size_t rule_size;
242         int valid;
243         char *linepos;
244         char *attr;
245         size_t padding;
246         int retval;
247
248         /* get all the keys */
249         rule = calloc(1, sizeof (struct udev_rule) + LINE_SIZE);
250         if (!rule) {
251                 err("malloc failed");
252                 return -1;
253         }
254         linepos = line;
255         valid = 0;
256
257         while (1) {
258                 char *key;
259                 char *value;
260                 enum key_operation operation = KEY_OP_UNSET;
261
262                 retval = get_key(&linepos, &key, &operation, &value);
263                 if (retval)
264                         break;
265
266                 if (strcasecmp(key, "LABEL") == 0) {
267                         add_rule_key(rule, &rule->label, operation, value);
268                         valid = 1;
269                         continue;
270                 }
271
272                 if (strcasecmp(key, "GOTO") == 0) {
273                         add_rule_key(rule, &rule->goto_label, operation, value);
274                         valid = 1;
275                         continue;
276                 }
277
278                 if (strcasecmp(key, "KERNEL") == 0) {
279                         if (operation != KEY_OP_MATCH &&
280                             operation != KEY_OP_NOMATCH) {
281                                 err("invalid KERNEL operation");
282                                 goto invalid;
283                         }
284                         add_rule_key(rule, &rule->kernel_name, operation, value);
285                         valid = 1;
286                         continue;
287                 }
288
289                 if (strcasecmp(key, "SUBSYSTEM") == 0) {
290                         if (operation != KEY_OP_MATCH &&
291                             operation != KEY_OP_NOMATCH) {
292                                 err("invalid SUBSYSTEM operation");
293                                 goto invalid;
294                         }
295                         add_rule_key(rule, &rule->subsystem, operation, value);
296                         valid = 1;
297                         continue;
298                 }
299
300                 if (strcasecmp(key, "ACTION") == 0) {
301                         if (operation != KEY_OP_MATCH &&
302                             operation != KEY_OP_NOMATCH) {
303                                 err("invalid ACTION operation");
304                                 goto invalid;
305                         }
306                         add_rule_key(rule, &rule->action, operation, value);
307                         valid = 1;
308                         continue;
309                 }
310
311                 if (strcasecmp(key, "DEVPATH") == 0) {
312                         if (operation != KEY_OP_MATCH &&
313                             operation != KEY_OP_NOMATCH) {
314                                 err("invalid DEVPATH operation");
315                                 goto invalid;
316                         }
317                         add_rule_key(rule, &rule->devpath, operation, value);
318                         valid = 1;
319                         continue;
320                 }
321
322                 if (strcasecmp(key, "BUS") == 0) {
323                         if (operation != KEY_OP_MATCH &&
324                             operation != KEY_OP_NOMATCH) {
325                                 err("invalid BUS operation");
326                                 goto invalid;
327                         }
328                         add_rule_key(rule, &rule->bus, operation, value);
329                         valid = 1;
330                         continue;
331                 }
332
333                 if (strcasecmp(key, "ID") == 0) {
334                         if (operation != KEY_OP_MATCH &&
335                             operation != KEY_OP_NOMATCH) {
336                                 err("invalid ID operation");
337                                 goto invalid;
338                         }
339                         add_rule_key(rule, &rule->id, operation, value);
340                         valid = 1;
341                         continue;
342                 }
343
344                 if (strncasecmp(key, "SYSFS", sizeof("SYSFS")-1) == 0) {
345                         attr = get_key_attribute(key + sizeof("SYSFS")-1);
346                         if (attr == NULL) {
347                                 err("error parsing SYSFS attribute in '%s'", line);
348                                 continue;
349                         }
350                         add_rule_key_pair(rule, &rule->sysfs, operation, attr, value);
351                         valid = 1;
352                         continue;
353                 }
354
355                 if (strcasecmp(key, "WAIT_FOR_SYSFS") == 0) {
356                         add_rule_key(rule, &rule->wait_for_sysfs, operation, value);
357                         valid = 1;
358                         continue;
359                 }
360
361                 if (strncasecmp(key, "ENV", sizeof("ENV")-1) == 0) {
362                         attr = get_key_attribute(key + sizeof("ENV")-1);
363                         if (attr == NULL) {
364                                 err("error parsing ENV attribute");
365                                 continue;
366                         }
367                         add_rule_key_pair(rule, &rule->env, operation, attr, value);
368                         valid = 1;
369                         continue;
370                 }
371
372                 if (strcasecmp(key, "MODALIAS") == 0) {
373                         add_rule_key(rule, &rule->modalias, operation, value);
374                         valid = 1;
375                         continue;
376                 }
377
378                 if (strncasecmp(key, "IMPORT", sizeof("IMPORT")-1) == 0) {
379                         attr = get_key_attribute(key + sizeof("IMPORT")-1);
380                         if (attr && strstr(attr, "program")) {
381                                 dbg("IMPORT will be executed");
382                                 rule->import_type  = IMPORT_PROGRAM;
383                         } else if (attr && strstr(attr, "file")) {
384                                 dbg("IMPORT will be included as file");
385                                 rule->import_type  = IMPORT_FILE;
386                         } else if (attr && strstr(attr, "parent")) {
387                                 dbg("IMPORT will include the parent values");
388                                 rule->import_type = IMPORT_PARENT;
389                         } else {
390                                 /* figure it out if it is executable */
391                                 char file[PATH_SIZE];
392                                 char *pos;
393                                 struct stat stats;
394
395                                 strlcpy(file, value, sizeof(file));
396                                 pos = strchr(file, ' ');
397                                 if (pos)
398                                         pos[0] = '\0';
399
400                                 /* allow programs in /lib/udev called without the path */
401                                 if (strchr(file, '/') == NULL) {
402                                         strlcpy(file, "/lib/udev/", sizeof(file));
403                                         strlcat(file, value, sizeof(file));
404                                         pos = strchr(file, ' ');
405                                         if (pos)
406                                                 pos[0] = '\0';
407                                 }
408
409                                 dbg("IMPORT auto mode for '%s'", file);
410                                 if (!lstat(file, &stats) && (stats.st_mode & S_IXUSR)) {
411                                         dbg("IMPORT is executable, will be executed (autotype)");
412                                         rule->import_type  = IMPORT_PROGRAM;
413                                 } else {
414                                         dbg("IMPORT is not executable, will be included as file (autotype)");
415                                         rule->import_type  = IMPORT_FILE;
416                                 }
417                         }
418                         add_rule_key(rule, &rule->import, operation, value);
419                         valid = 1;
420                         continue;
421                 }
422
423                 if (strcasecmp(key, "DRIVER") == 0) {
424                         add_rule_key(rule, &rule->driver, operation, value);
425                         valid = 1;
426                         continue;
427                 }
428
429                 if (strcasecmp(key, "RESULT") == 0) {
430                         add_rule_key(rule, &rule->result, operation, value);
431                         valid = 1;
432                         continue;
433                 }
434
435                 if (strcasecmp(key, "PROGRAM") == 0) {
436                         add_rule_key(rule, &rule->program, operation, value);
437                         valid = 1;
438                         continue;
439                 }
440
441                 if (strncasecmp(key, "NAME", sizeof("NAME")-1) == 0) {
442                         attr = get_key_attribute(key + sizeof("NAME")-1);
443                         if (attr != NULL) {
444                                 if (strstr(attr, "all_partitions") != NULL) {
445                                         dbg("creation of partition nodes requested");
446                                         rule->partitions = DEFAULT_PARTITIONS_COUNT;
447                                 }
448                                 if (strstr(attr, "ignore_remove") != NULL) {
449                                         dbg("remove event should be ignored");
450                                         rule->ignore_remove = 1;
451                                 }
452                         }
453                         if (value[0] == '\0')
454                                 dbg("name empty, node creation supressed");
455                         add_rule_key(rule, &rule->name, operation, value);
456                         continue;
457                 }
458
459                 if (strcasecmp(key, "SYMLINK") == 0) {
460                         add_rule_key(rule, &rule->symlink, operation, value);
461                         valid = 1;
462                         continue;
463                 }
464
465                 if (strcasecmp(key, "OWNER") == 0) {
466                         valid = 1;
467                         if (rules->resolve_names && (!strchr(value, '$') && !strchr(value, '%'))) {
468                                 char *endptr;
469                                 strtoul(value, &endptr, 10);
470                                 if (endptr[0] != '\0') {
471                                         char owner[32];
472                                         uid_t uid = lookup_user(value);
473                                         dbg("replacing username='%s' by id=%i", value, uid);
474                                         sprintf(owner, "%u", (unsigned int) uid);
475                                         add_rule_key(rule, &rule->owner, operation, owner);
476                                         continue;
477                                 }
478                         }
479
480                         add_rule_key(rule, &rule->owner, operation, value);
481                         continue;
482                 }
483
484                 if (strcasecmp(key, "GROUP") == 0) {
485                         valid = 1;
486                         if (rules->resolve_names && (!strchr(value, '$') && !strchr(value, '%'))) {
487                                 char *endptr;
488                                 strtoul(value, &endptr, 10);
489                                 if (endptr[0] != '\0') {
490                                         char group[32];
491                                         gid_t gid = lookup_group(value);
492                                         dbg("replacing groupname='%s' by id=%i", value, gid);
493                                         sprintf(group, "%u", (unsigned int) gid);
494                                         add_rule_key(rule, &rule->group, operation, group);
495                                         continue;
496                                 }
497                         }
498
499                         add_rule_key(rule, &rule->group, operation, value);
500                         continue;
501                 }
502
503                 if (strcasecmp(key, "MODE") == 0) {
504                         rule->mode = strtol(value, NULL, 8);
505                         rule->mode_operation = operation;
506                         valid = 1;
507                         continue;
508                 }
509
510                 if (strcasecmp(key, "RUN") == 0) {
511                         add_rule_key(rule, &rule->run, operation, value);
512                         valid = 1;
513                         continue;
514                 }
515
516                 if (strcasecmp(key, "OPTIONS") == 0) {
517                         if (strstr(value, "last_rule") != NULL) {
518                                 dbg("last rule to be applied");
519                                 rule->last_rule = 1;
520                         }
521                         if (strstr(value, "ignore_device") != NULL) {
522                                 dbg("device should be ignored");
523                                 rule->ignore_device = 1;
524                         }
525                         if (strstr(value, "ignore_remove") != NULL) {
526                                 dbg("remove event should be ignored");
527                                 rule->ignore_remove = 1;
528                         }
529                         if (strstr(value, "all_partitions") != NULL) {
530                                 dbg("creation of partition nodes requested");
531                                 rule->partitions = DEFAULT_PARTITIONS_COUNT;
532                         }
533                         valid = 1;
534                         continue;
535                 }
536
537                 err("unknown key '%s'", key);
538         }
539
540         /* skip line if not any valid key was found */
541         if (!valid)
542                 goto invalid;
543
544         /* grow buffer and add rule */
545         rule_size = sizeof(struct udev_rule) + rule->bufsize;
546         padding = (sizeof(size_t) - rule_size % sizeof(size_t)) % sizeof(size_t);
547         dbg("add %zi padding bytes", padding);
548         rule_size += padding;
549         rule->bufsize += padding;
550
551         rules->buf = realloc(rules->buf, rules->bufsize + rule_size);
552         if (!rules->buf) {
553                 err("realloc failed");
554                 goto exit;
555         }
556         dbg("adding rule to offset %zi", rules->bufsize);
557         memcpy(rules->buf + rules->bufsize, rule, rule_size);
558         rules->bufsize += rule_size;
559 exit:
560         free(rule);
561         return 0;
562
563 invalid:
564         free(rule);
565         err("invalid rule '%s:%u'", filename, lineno);
566         return -1;
567 }
568
569 static int parse_file(struct udev_rules *rules, const char *filename)
570 {
571         char line[LINE_SIZE];
572         char *bufline;
573         unsigned int lineno;
574         char *buf;
575         size_t bufsize;
576         size_t cur;
577         size_t count;
578         int retval = 0;
579
580         if (file_map(filename, &buf, &bufsize) != 0) {
581                 err("can't open '%s' as rules file: %s", filename, strerror(errno));
582                 return -1;
583         }
584         dbg("reading '%s' as rules file", filename);
585
586         /* loop through the whole file */
587         cur = 0;
588         lineno = 0;
589         while (cur < bufsize) {
590                 unsigned int i, j;
591
592                 count = buf_get_line(buf, bufsize, cur);
593                 bufline = &buf[cur];
594                 cur += count+1;
595                 lineno++;
596
597                 if (count >= sizeof(line)) {
598                         err("line too long, rule skipped '%s:%u'", filename, lineno);
599                         continue;
600                 }
601
602                 /* eat the whitespace */
603                 while ((count > 0) && isspace(bufline[0])) {
604                         bufline++;
605                         count--;
606                 }
607                 if (count == 0)
608                         continue;
609
610                 /* see if this is a comment */
611                 if (bufline[0] == COMMENT_CHARACTER)
612                         continue;
613
614                 /* skip backslash and newline from multi line rules */
615                 for (i = j = 0; i < count; i++) {
616                         if (bufline[i] == '\\' && bufline[i+1] == '\n')
617                                 continue;
618
619                         line[j++] = bufline[i];
620                 }
621                 line[j] = '\0';
622
623                 dbg("read '%s'", line);
624                 add_to_rules(rules, line, filename, lineno);
625         }
626
627         file_unmap(buf, bufsize);
628         return retval;
629 }
630
631 int udev_rules_init(struct udev_rules *rules, int resolve_names)
632 {
633         struct stat stats;
634         int retval;
635
636         memset(rules, 0x00, sizeof(struct udev_rules));
637         rules->resolve_names = resolve_names;
638
639         /* parse rules file or all matching files in directory */
640         if (stat(udev_rules_filename, &stats) != 0)
641                 return -1;
642
643         if ((stats.st_mode & S_IFMT) != S_IFDIR) {
644                 dbg("parse single rules file '%s'", udev_rules_filename);
645                 retval = parse_file(rules, udev_rules_filename);
646         } else {
647                 struct name_entry *name_loop, *name_tmp;
648                 LIST_HEAD(name_list);
649
650                 dbg("parse rules directory '%s'", udev_rules_filename);
651                 retval = add_matching_files(&name_list, udev_rules_filename, RULEFILE_SUFFIX);
652
653                 list_for_each_entry_safe(name_loop, name_tmp, &name_list, node) {
654                         if (stat(name_loop->name, &stats) == 0) {
655                                 if (stats.st_size)
656                                         parse_file(rules, name_loop->name);
657                                 else
658                                         dbg("empty rules file '%s'", name_loop->name);
659                         } else
660                                 err("could not read '%s': %s", name_loop->name, strerror(errno));
661                         list_del(&name_loop->node);
662                         free(name_loop);
663                 }
664         }
665
666         return retval;
667 }
668
669 void udev_rules_cleanup(struct udev_rules *rules)
670 {
671         if (rules->buf) {
672                 free(rules->buf);
673                 rules->buf = NULL;
674         }
675 }