chiark / gitweb /
[PATCH] - unlink bugfix
[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                         strfieldcpy(user, u->ut_user);
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[255];
190         char linktarget[255];
191         char partitionname[255];
192         char *linkname;
193         char *symlinks;
194         int retval = 0;
195         uid_t uid = 0;
196         gid_t gid = 0;
197         int i;
198         int tail;
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         if (dev->symlink[0] != '\0') {
283                 symlinks = dev->symlink;
284                 while (1) {
285                         linkname = strsep(&symlinks, " ");
286                         if (linkname == NULL || linkname[0] == '\0')
287                                 break;
288
289                         strfieldcpy(filename, udev_root);
290                         strfieldcat(filename, linkname);
291                         dbg("symlink '%s' to node '%s' requested", filename, dev->name);
292                         if (!fake)
293                                 if (strrchr(linkname, '/'))
294                                         create_path(filename);
295
296                         /* optimize relative link */
297                         linktarget[0] = '\0';
298                         i = 0;
299                         tail = 0;
300                         while ((dev->name[i] == linkname[i]) && dev->name[i]) {
301                                 if (dev->name[i] == '/')
302                                         tail = i+1;
303                                 i++;
304                         }
305                         while (linkname[i] != '\0') {
306                                 if (linkname[i] == '/')
307                                         strfieldcat(linktarget, "../");
308                                 i++;
309                         }
310
311                         strfieldcat(linktarget, &dev->name[tail]);
312
313                         if (!fake)
314                                 unlink_entry(filename);
315
316                         dbg("symlink(%s, %s)", linktarget, filename);
317                         if (!fake) {
318                                 retval = symlink(linktarget, filename);
319                                 if (retval != 0)
320                                         dbg("symlink(%s, %s) failed with error '%s'",
321                                             linktarget, filename, strerror(errno));
322                         }
323                 }
324         }
325
326         return retval;
327 }
328
329 static struct sysfs_class_device *get_class_dev(char *device_name)
330 {
331         char dev_path[SYSFS_PATH_MAX];
332         struct sysfs_class_device *class_dev = NULL;
333
334         strfieldcpy(dev_path, sysfs_path);
335         strfieldcat(dev_path, device_name);
336         dbg("looking at '%s'", dev_path);
337
338         /* open up the sysfs class device for this thing... */
339         class_dev = sysfs_open_class_device_path(dev_path);
340         if (class_dev == NULL) {
341                 dbg ("sysfs_open_class_device_path failed");
342                 goto exit;
343         }
344         dbg("class_dev->name='%s'", class_dev->name);
345
346 exit:
347         return class_dev;
348 }
349
350 /* wait for the "dev" file to show up in the directory in sysfs.
351  * If it doesn't happen in about 10 seconds, give up.
352  */
353 #define SECONDS_TO_WAIT_FOR_DEV         10
354 static int sleep_for_dev(char *path)
355 {
356         char filename[SYSFS_PATH_MAX + 6];
357         int loop = SECONDS_TO_WAIT_FOR_DEV;
358         int retval;
359
360         strfieldcpy(filename, sysfs_path);
361         strfieldcat(filename, path);
362         strfieldcat(filename, "/dev");
363
364         while (loop--) {
365                 struct stat buf;
366
367                 dbg("looking for '%s'", filename);
368                 retval = stat(filename, &buf);
369                 if (retval == 0)
370                         goto exit;
371
372                 /* sleep to give the kernel a chance to create the dev file */
373                 sleep(1);
374         }
375         retval = -ENODEV;
376 exit:
377         return retval;
378 }
379
380 int udev_add_device(char *path, char *subsystem, int fake)
381 {
382         struct sysfs_class_device *class_dev = NULL;
383         struct udevice dev;
384         int retval = -EINVAL;
385
386         memset(&dev, 0x00, sizeof(dev));
387
388         /* for now, the block layer is the only place where block devices are */
389         if (strcmp(subsystem, "block") == 0)
390                 dev.type = 'b';
391         else
392                 dev.type = 'c';
393
394         retval = sleep_for_dev(path);
395         if (retval != 0)
396                 goto exit;
397
398         class_dev = get_class_dev(path);
399         if (class_dev == NULL)
400                 goto exit;
401
402         retval = get_major_minor(class_dev, &dev);
403         if (retval != 0) {
404                 dbg("get_major_minor failed");
405                 goto exit;
406         }
407
408         retval = namedev_name_device(class_dev, &dev);
409         if (retval != 0)
410                 goto exit;
411
412         if (!fake) {
413                 retval = udevdb_add_dev(path, &dev);
414                 if (retval != 0)
415                         dbg("udevdb_add_dev failed, but we are going to try "
416                             "to create the node anyway. But remove might not "
417                             "work properly for this device.");
418
419         }
420         dbg("name='%s'", dev.name);
421         retval = create_node(&dev, fake);
422
423         if ((retval == 0) && (!fake))
424                 sysbus_send_create(&dev, path);
425
426 exit:
427         if (class_dev)
428                 sysfs_close_class_device(class_dev);
429
430         return retval;
431 }