chiark / gitweb /
[PATCH] better fix for NAME="foo-%c{N}" gets a truncated name
[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         char *pos;
199         int len;
200
201         strfieldcpy(filename, udev_root);
202         strfieldcat(filename, dev->name);
203
204         switch (dev->type) {
205         case 'b':
206                 dev->mode |= S_IFBLK;
207                 break;
208         case 'c':
209         case 'u':
210                 dev->mode |= S_IFCHR;
211                 break;
212         case 'p':
213                 dev->mode |= S_IFIFO;
214                 break;
215         default:
216                 dbg("unknown node type %c\n", dev->type);
217                 return -EINVAL;
218         }
219
220         /* create parent directories if needed */
221         if (strrchr(dev->name, '/'))
222                 create_path(filename);
223
224         if (dev->owner[0] != '\0') {
225                 char *endptr;
226                 unsigned long id = strtoul(dev->owner, &endptr, 10);
227                 if (endptr[0] == '\0')
228                         uid = (uid_t) id;
229                 else {
230                         struct passwd *pw;
231                         if (strncmp(dev->owner, LOCAL_USER, sizeof(LOCAL_USER)) == 0)
232                                 set_to_local_user(dev->owner);
233
234                         pw = getpwnam(dev->owner);
235                         if (pw == NULL)
236                                 dbg("specified user unknown '%s'", dev->owner);
237                         else
238                                 uid = pw->pw_uid;
239                 }
240         }
241
242         if (dev->group[0] != '\0') {
243                 char *endptr;
244                 unsigned long id = strtoul(dev->group, &endptr, 10);
245                 if (endptr[0] == '\0')
246                         gid = (gid_t) id;
247                 else {
248                         struct group *gr = getgrnam(dev->group);
249                         if (gr == NULL)
250                                 dbg("specified group unknown '%s'", dev->group);
251                         else
252                                 gid = gr->gr_gid;
253                 }
254         }
255
256         if (!fake) {
257                 unlink_entry(filename);
258                 info("creating device node '%s'", filename);
259                 make_node(filename, dev->major, dev->minor, dev->mode, uid, gid);
260         } else {
261                 info("creating device node '%s', major = '%d', minor = '%d', "
262                      "mode = '%#o', uid = '%d', gid = '%d'", filename,
263                      dev->major, dev->minor, (mode_t)dev->mode, uid, gid);
264         }
265
266         /* create partitions if requested */
267         if (dev->partitions > 0) {
268                 info("creating device partition nodes '%s[1-%i]'", filename, dev->partitions);
269                 if (!fake) {
270                         for (i = 1; i <= dev->partitions; i++) {
271                                 strfieldcpy(partitionname, filename);
272                                 strintcat(partitionname, i);
273                                 unlink_entry(partitionname);
274                                 make_node(partitionname, dev->major,
275                                           dev->minor + i, dev->mode, uid, gid);
276                         }
277                 }
278         }
279
280         if (!fake)
281                 selinux_add_node(filename);
282
283         /* create symlink if requested */
284         foreach_strpart(dev->symlink, " ", pos, len) {
285                 strnfieldcpy(linkname, pos, len+1);
286                 strfieldcpy(filename, udev_root);
287                 strfieldcat(filename, linkname);
288                 dbg("symlink '%s' to node '%s' requested", filename, dev->name);
289                 if (!fake)
290                         if (strrchr(linkname, '/'))
291                                 create_path(filename);
292
293                 /* optimize relative link */
294                 linktarget[0] = '\0';
295                 i = 0;
296                 tail = 0;
297                 while ((dev->name[i] == linkname[i]) && dev->name[i]) {
298                         if (dev->name[i] == '/')
299                                 tail = i+1;
300                         i++;
301                 }
302                 while (linkname[i] != '\0') {
303                         if (linkname[i] == '/')
304                                 strfieldcat(linktarget, "../");
305                         i++;
306                 }
307
308                 strfieldcat(linktarget, &dev->name[tail]);
309
310                 if (!fake)
311                         unlink_entry(filename);
312
313                 dbg("symlink(%s, %s)", linktarget, filename);
314                 if (!fake) {
315                         retval = symlink(linktarget, filename);
316                         if (retval != 0)
317                                 dbg("symlink(%s, %s) failed with error '%s'",
318                                     linktarget, filename, strerror(errno));
319                 }
320         }
321
322         return retval;
323 }
324
325 static struct sysfs_class_device *get_class_dev(char *device_name)
326 {
327         char dev_path[SYSFS_PATH_MAX];
328         struct sysfs_class_device *class_dev = NULL;
329
330         strfieldcpy(dev_path, sysfs_path);
331         strfieldcat(dev_path, device_name);
332         dbg("looking at '%s'", dev_path);
333
334         /* open up the sysfs class device for this thing... */
335         class_dev = sysfs_open_class_device_path(dev_path);
336         if (class_dev == NULL) {
337                 dbg ("sysfs_open_class_device_path failed");
338                 goto exit;
339         }
340         dbg("class_dev->name='%s'", class_dev->name);
341
342 exit:
343         return class_dev;
344 }
345
346 /* wait for the "dev" file to show up in the directory in sysfs.
347  * If it doesn't happen in about 10 seconds, give up.
348  */
349 #define SECONDS_TO_WAIT_FOR_DEV         10
350 static int sleep_for_dev(char *path)
351 {
352         char filename[SYSFS_PATH_MAX + 6];
353         int loop = SECONDS_TO_WAIT_FOR_DEV;
354         int retval;
355
356         strfieldcpy(filename, sysfs_path);
357         strfieldcat(filename, path);
358         strfieldcat(filename, "/dev");
359
360         while (loop--) {
361                 struct stat buf;
362
363                 dbg("looking for '%s'", filename);
364                 retval = stat(filename, &buf);
365                 if (retval == 0)
366                         goto exit;
367
368                 /* sleep to give the kernel a chance to create the dev file */
369                 sleep(1);
370         }
371         retval = -ENODEV;
372 exit:
373         return retval;
374 }
375
376 int udev_add_device(char *path, char *subsystem, int fake)
377 {
378         struct sysfs_class_device *class_dev = NULL;
379         struct udevice dev;
380         int retval = -EINVAL;
381
382         memset(&dev, 0x00, sizeof(dev));
383
384         /* for now, the block layer is the only place where block devices are */
385         if (strcmp(subsystem, "block") == 0)
386                 dev.type = 'b';
387         else
388                 dev.type = 'c';
389
390         retval = sleep_for_dev(path);
391         if (retval != 0)
392                 goto exit;
393
394         class_dev = get_class_dev(path);
395         if (class_dev == NULL)
396                 goto exit;
397
398         retval = get_major_minor(class_dev, &dev);
399         if (retval != 0) {
400                 dbg("get_major_minor failed");
401                 goto exit;
402         }
403
404         retval = namedev_name_device(class_dev, &dev);
405         if (retval != 0)
406                 goto exit;
407
408         if (!fake) {
409                 retval = udevdb_add_dev(path, &dev);
410                 if (retval != 0)
411                         dbg("udevdb_add_dev failed, but we are going to try "
412                             "to create the node anyway. But remove might not "
413                             "work properly for this device.");
414
415         }
416         dbg("name='%s'", dev.name);
417         retval = create_node(&dev, fake);
418
419         if ((retval == 0) && (!fake))
420                 sysbus_send_create(&dev, path);
421
422 exit:
423         if (class_dev)
424                 sysfs_close_class_device(class_dev);
425
426         return retval;
427 }