chiark / gitweb /
[PATCH] cleanup mult field string handling
[elogind.git] / udev-add.c
1 /*
2  * udev-add.c
3  *
4  * Userspace devfs
5  *
6  * Copyright (C) 2003 Greg Kroah-Hartman <greg@kroah.com>
7  *
8  *
9  *      This program is free software; you can redistribute it and/or modify it
10  *      under the terms of the GNU General Public License as published by the
11  *      Free Software Foundation version 2 of the License.
12  * 
13  *      This program is distributed in the hope that it will be useful, but
14  *      WITHOUT ANY WARRANTY; without even the implied warranty of
15  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  *      General Public License for more details.
17  * 
18  *      You should have received a copy of the GNU General Public License along
19  *      with this program; if not, write to the Free Software Foundation, Inc.,
20  *      675 Mass Ave, Cambridge, MA 02139, USA.
21  *
22  */
23
24 #include <stdlib.h>
25 #include <string.h>
26 #include <stdio.h>
27 #include <fcntl.h>
28 #include <unistd.h>
29 #include <errno.h>
30 #include <sys/stat.h>
31 #include <sys/types.h>
32 #include <grp.h>
33 #ifndef __KLIBC__
34 #include <pwd.h>
35 #include <utmp.h>
36 #endif
37
38 #include "libsysfs/sysfs/libsysfs.h"
39 #include "udev.h"
40 #include "udev_version.h"
41 #include "udev_dbus.h"
42 #include "udev_selinux.h"
43 #include "logging.h"
44 #include "namedev.h"
45 #include "udevdb.h"
46 #include "klibc_fixups.h"
47
48 #define LOCAL_USER "$local"
49
50 /* 
51  * Right now the major/minor of a device is stored in a file called
52  * "dev" in sysfs.
53  * The number is stored as:
54  *      MM:mm
55  *              MM is the major
56  *              mm is the minor
57  *              The value is in decimal.
58  */
59 static int get_major_minor(struct sysfs_class_device *class_dev, struct udevice *udev)
60 {
61         int retval = -ENODEV;
62         struct sysfs_attribute *attr = NULL;
63
64         attr = sysfs_get_classdev_attr(class_dev, "dev");
65         if (attr == NULL)
66                 goto exit;
67         dbg("dev='%s'", attr->value);
68
69         if (sscanf(attr->value, "%u:%u", &udev->major, &udev->minor) != 2)
70                 goto exit;
71         dbg("found major=%d, minor=%d", udev->major, udev->minor);
72
73         retval = 0;
74 exit:
75         return retval;
76 }
77
78 static int create_path(char *file)
79 {
80         char p[NAME_SIZE];
81         char *pos;
82         int retval;
83         struct stat stats;
84         
85         strfieldcpy(p, file);
86         pos = strchr(p+1, '/');
87         while (1) {
88                 pos = strchr(pos+1, '/');
89                 if (pos == NULL)
90                         break;
91                 *pos = 0x00;
92                 if (stat(p, &stats)) {
93                         retval = mkdir(p, 0755);
94                         if (retval != 0) {
95                                 dbg("mkdir(%s) failed with error '%s'",
96                                     p, strerror(errno));
97                                 return retval;
98                         }
99                         dbg("created '%s'", p);
100                 }
101                 *pos = '/';
102         }
103         return 0;
104 }
105
106 static int make_node(char *filename, int major, int minor, unsigned int mode, uid_t uid, gid_t gid)
107 {
108         int retval;
109
110         retval = mknod(filename, mode, makedev(major, minor));
111         if (retval != 0) {
112                 dbg("mknod(%s, %#o, %u, %u) failed with error '%s'",
113                     filename, mode, major, minor, strerror(errno));
114                 return retval;
115         }
116
117         dbg("chmod(%s, %#o)", filename, mode);
118         retval = chmod(filename, mode);
119         if (retval != 0) {
120                 dbg("chmod(%s, %#o) failed with error '%s'",
121                     filename, mode, strerror(errno));
122                 return retval;
123         }
124
125         if (uid != 0 || gid != 0) {
126                 dbg("chown(%s, %u, %u)", filename, uid, gid);
127                 retval = chown(filename, uid, gid);
128                 if (retval != 0) {
129                         dbg("chown(%s, %u, %u) failed with error '%s'",
130                             filename, uid, gid, strerror(errno));
131                         return retval;
132                 }
133         }
134
135         return 0;
136 }
137
138 /* get the local logged in user */
139 static void set_to_local_user(char *user)
140 {
141         struct utmp *u;
142         time_t recent = 0;
143
144         strnfieldcpy(user, default_owner_str, OWNER_SIZE);
145         setutent();
146         while (1) {
147                 u = getutent();
148                 if (u == NULL)
149                         break;
150
151                 /* is this a user login ? */
152                 if (u->ut_type != USER_PROCESS)
153                         continue;
154
155                 /* is this a local login ? */
156                 if (strcmp(u->ut_host, ""))
157                         continue;
158
159                 if (u->ut_time > recent) {
160                         recent = u->ut_time;
161                         strnfieldcpy(user, u->ut_user, OWNER_SIZE);
162                         dbg("local user is '%s'", user);
163                         break;
164                 }
165         }
166         endutent();
167 }
168
169 /* Used to unlink existing files to ensure that our new file/symlink is created */
170 static int unlink_entry(char *filename)
171 {
172         struct stat stats;
173         int retval = 0;
174         
175         if (lstat(filename, &stats) == 0) {
176                 if ((stats.st_mode & S_IFMT) != S_IFDIR) {
177                         retval = unlink(filename);
178                         if (retval) {
179                                 dbg("unlink(%s) failed with error '%s'",
180                                     filename, strerror(errno));
181                         }
182                 }
183         }
184         return retval;
185 }
186
187 static int create_node(struct udevice *dev, int fake)
188 {
189         char filename[NAME_SIZE];
190         char linkname[NAME_SIZE];
191         char linktarget[NAME_SIZE];
192         char partitionname[NAME_SIZE];
193         int retval = 0;
194         uid_t uid = 0;
195         gid_t gid = 0;
196         int i;
197         int tail;
198         int pos, len;
199
200         strfieldcpy(filename, udev_root);
201         strfieldcat(filename, dev->name);
202
203         switch (dev->type) {
204         case 'b':
205                 dev->mode |= S_IFBLK;
206                 break;
207         case 'c':
208         case 'u':
209                 dev->mode |= S_IFCHR;
210                 break;
211         case 'p':
212                 dev->mode |= S_IFIFO;
213                 break;
214         default:
215                 dbg("unknown node type %c\n", dev->type);
216                 return -EINVAL;
217         }
218
219         /* create parent directories if needed */
220         if (strrchr(dev->name, '/'))
221                 create_path(filename);
222
223         if (dev->owner[0] != '\0') {
224                 char *endptr;
225                 unsigned long id = strtoul(dev->owner, &endptr, 10);
226                 if (endptr[0] == '\0')
227                         uid = (uid_t) id;
228                 else {
229                         if (strncmp(dev->owner, LOCAL_USER, sizeof(LOCAL_USER)) == 0)
230                                 set_to_local_user(dev->owner);
231
232                         struct passwd *pw = getpwnam(dev->owner);
233                         if (pw == NULL)
234                                 dbg("specified user unknown '%s'", dev->owner);
235                         else
236                                 uid = pw->pw_uid;
237                 }
238         }
239
240         if (dev->group[0] != '\0') {
241                 char *endptr;
242                 unsigned long id = strtoul(dev->group, &endptr, 10);
243                 if (endptr[0] == '\0')
244                         gid = (gid_t) id;
245                 else {
246                         struct group *gr = getgrnam(dev->group);
247                         if (gr == NULL)
248                                 dbg("specified group unknown '%s'", dev->group);
249                         else
250                                 gid = gr->gr_gid;
251                 }
252         }
253
254         if (!fake) {
255                 unlink_entry(filename);
256                 info("creating device node '%s'", filename);
257                 make_node(filename, dev->major, dev->minor, dev->mode, uid, gid);
258         } else {
259                 info("creating device node '%s', major = '%d', minor = '%d', "
260                      "mode = '%#o', uid = '%d', gid = '%d'", filename,
261                      dev->major, dev->minor, (mode_t)dev->mode, uid, gid);
262         }
263
264         /* create partitions if requested */
265         if (dev->partitions > 0) {
266                 info("creating device partition nodes '%s[1-%i]'", filename, dev->partitions);
267                 if (!fake) {
268                         for (i = 1; i <= dev->partitions; i++) {
269                                 strfieldcpy(partitionname, filename);
270                                 strintcat(partitionname, i);
271                                 unlink_entry(partitionname);
272                                 make_node(partitionname, dev->major,
273                                           dev->minor + i, dev->mode, uid, gid);
274                         }
275                 }
276         }
277
278         if (!fake)
279                 selinux_add_node(filename);
280
281         /* create symlink if requested */
282         foreach_strpart(dev->symlink, " ", pos, len) {
283                 strnfieldcpy(linkname, dev->symlink + pos, len+1);
284                 strfieldcpy(filename, udev_root);
285                 strfieldcat(filename, linkname);
286                 dbg("symlink '%s' to node '%s' requested", filename, dev->name);
287                 if (!fake)
288                         if (strrchr(linkname, '/'))
289                                 create_path(filename);
290
291                 /* optimize relative link */
292                 linktarget[0] = '\0';
293                 i = 0;
294                 tail = 0;
295                 while ((dev->name[i] == linkname[i]) && dev->name[i]) {
296                         if (dev->name[i] == '/')
297                                 tail = i+1;
298                         i++;
299                 }
300                 while (linkname[i] != '\0') {
301                         if (linkname[i] == '/')
302                                 strfieldcat(linktarget, "../");
303                         i++;
304                 }
305
306                 strfieldcat(linktarget, &dev->name[tail]);
307
308                 if (!fake)
309                         unlink_entry(filename);
310
311                 dbg("symlink(%s, %s)", linktarget, filename);
312                 if (!fake) {
313                         retval = symlink(linktarget, filename);
314                         if (retval != 0)
315                                 dbg("symlink(%s, %s) failed with error '%s'",
316                                     linktarget, filename, strerror(errno));
317                 }
318         }
319
320         return retval;
321 }
322
323 static struct sysfs_class_device *get_class_dev(char *device_name)
324 {
325         char dev_path[SYSFS_PATH_MAX];
326         struct sysfs_class_device *class_dev = NULL;
327
328         strfieldcpy(dev_path, sysfs_path);
329         strfieldcat(dev_path, device_name);
330         dbg("looking at '%s'", dev_path);
331
332         /* open up the sysfs class device for this thing... */
333         class_dev = sysfs_open_class_device_path(dev_path);
334         if (class_dev == NULL) {
335                 dbg ("sysfs_open_class_device_path failed");
336                 goto exit;
337         }
338         dbg("class_dev->name='%s'", class_dev->name);
339
340 exit:
341         return class_dev;
342 }
343
344 /* wait for the "dev" file to show up in the directory in sysfs.
345  * If it doesn't happen in about 10 seconds, give up.
346  */
347 #define SECONDS_TO_WAIT_FOR_DEV         10
348 static int sleep_for_dev(char *path)
349 {
350         char filename[SYSFS_PATH_MAX + 6];
351         int loop = SECONDS_TO_WAIT_FOR_DEV;
352         int retval;
353
354         strfieldcpy(filename, sysfs_path);
355         strfieldcat(filename, path);
356         strfieldcat(filename, "/dev");
357
358         while (loop--) {
359                 struct stat buf;
360
361                 dbg("looking for '%s'", filename);
362                 retval = stat(filename, &buf);
363                 if (retval == 0)
364                         goto exit;
365
366                 /* sleep to give the kernel a chance to create the dev file */
367                 sleep(1);
368         }
369         retval = -ENODEV;
370 exit:
371         return retval;
372 }
373
374 int udev_add_device(char *path, char *subsystem, int fake)
375 {
376         struct sysfs_class_device *class_dev = NULL;
377         struct udevice dev;
378         int retval = -EINVAL;
379
380         memset(&dev, 0x00, sizeof(dev));
381
382         /* for now, the block layer is the only place where block devices are */
383         if (strcmp(subsystem, "block") == 0)
384                 dev.type = 'b';
385         else
386                 dev.type = 'c';
387
388         retval = sleep_for_dev(path);
389         if (retval != 0)
390                 goto exit;
391
392         class_dev = get_class_dev(path);
393         if (class_dev == NULL)
394                 goto exit;
395
396         retval = get_major_minor(class_dev, &dev);
397         if (retval != 0) {
398                 dbg("get_major_minor failed");
399                 goto exit;
400         }
401
402         retval = namedev_name_device(class_dev, &dev);
403         if (retval != 0)
404                 goto exit;
405
406         if (!fake) {
407                 retval = udevdb_add_dev(path, &dev);
408                 if (retval != 0)
409                         dbg("udevdb_add_dev failed, but we are going to try "
410                             "to create the node anyway. But remove might not "
411                             "work properly for this device.");
412
413         }
414         dbg("name='%s'", dev.name);
415         retval = create_node(&dev, fake);
416
417         if ((retval == 0) && (!fake))
418                 sysbus_send_create(&dev, path);
419
420 exit:
421         if (class_dev)
422                 sysfs_close_class_device(class_dev);
423
424         return retval;
425 }