1
1
mirror of https://github.com/systemd/systemd-stable.git synced 2024-10-31 16:21:11 +03:00
systemd-stable/udevstart.c
Andrey Borzenkov f5f0c34f6d substitute format chars in RUN after rule matching
Apply substitutions before running a program, not while the rule
is parsed. It allows to use environment variables set during rule
processing as command arguments.
2006-01-24 20:10:48 +01:00

377 lines
8.8 KiB
C

/*
* udevstart.c
*
* Copyright (C) 2004 Greg Kroah-Hartman <greg@kroah.com>
* Copyright (C) 2004 Kay Sievers <kay@vrfy.org>
*
* Quick and dirty way to populate a /dev with udev if your system
* does not have access to a shell. Based originally on a patch
* from:
* Harald Hoyer <harald@redhat.com>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation version 2 of the License.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#include <stdlib.h>
#include <stddef.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <ctype.h>
#include <dirent.h>
#include <signal.h>
#include <syslog.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <sys/types.h>
#include "udev.h"
#include "udev_rules.h"
static const char *udev_run_str;
static const char *udev_log_str;
static struct udev_rules rules;
#ifdef USE_LOG
void log_message(int priority, const char *format, ...)
{
va_list args;
if (priority > udev_log_priority)
return;
va_start(args, format);
vsyslog(priority, format, args);
va_end(args);
}
#endif
struct device {
struct list_head node;
char path[PATH_SIZE];
};
/* sort files in lexical order */
static int device_list_insert(const char *path, struct list_head *device_list)
{
struct device *loop_device;
struct device *new_device;
const char *devpath = &path[strlen(sysfs_path)];
dbg("insert: '%s'\n", devpath);
list_for_each_entry(loop_device, device_list, node) {
if (strcmp(loop_device->path, devpath) > 0) {
break;
}
}
new_device = malloc(sizeof(struct device));
if (new_device == NULL) {
dbg("error malloc");
return -ENOMEM;
}
strlcpy(new_device->path, devpath, sizeof(new_device->path));
list_add_tail(&new_device->node, &loop_device->node);
dbg("add '%s'" , new_device->path);
return 0;
}
/* list of devices that we should run last due to any one of a number of reasons */
static char *last_list[] = {
"/block/dm", /* on here because dm wants to have the block devices around before it */
NULL,
};
/* list of devices that we should run first due to any one of a number of reasons */
static char *first_list[] = {
"/class/mem",
"/class/tty",
NULL,
};
static int add_device(const char *devpath)
{
struct sysfs_device *dev;
struct udevice *udev;
/* clear and set environment for next event */
clearenv();
setenv("ACTION", "add", 1);
setenv("UDEV_START", "1", 1);
if (udev_log_str)
setenv("UDEV_LOG", udev_log_str, 1);
if (udev_run_str)
setenv("UDEV_RUN", udev_run_str, 1);
dev = sysfs_device_get(devpath);
if (dev == NULL)
return -1;
udev = udev_device_init();
if (udev == NULL)
return -1;
/* override built-in sysfs device */
udev->dev = dev;
strcpy(udev->action, "add");
udev->devt = udev_device_get_devt(udev);
if (strcmp(udev->dev->subsystem, "net") != 0) {
udev->devt = udev_device_get_devt(udev);
if (major(udev->devt) == 0)
return -1;
}
dbg("add '%s'", udev->dev->devpath);
setenv("DEVPATH", udev->dev->devpath, 1);
setenv("SUBSYSTEM", udev->dev->subsystem, 1);
udev_rules_get_name(&rules, udev);
if (udev->ignore_device) {
dbg("device event will be ignored");
goto exit;
}
if (udev->name[0] == '\0') {
dbg("device node creation supressed");
goto run;
}
udev_add_device(udev);
run:
if (udev_run && !list_empty(&udev->run_list)) {
struct name_entry *name_loop;
dbg("executing run list");
list_for_each_entry(name_loop, &udev->run_list, node) {
if (strncmp(name_loop->name, "socket:", strlen("socket:")) == 0)
pass_env_to_socket(&name_loop->name[strlen("socket:")], udev->dev->devpath, "add");
else {
char program[PATH_SIZE];
strlcpy(program, name_loop->name, sizeof(program));
apply_format(udev, program, sizeof(program));
run_program(program, udev->dev->subsystem, NULL, 0, NULL, (udev_log_priority >= LOG_INFO));
}
}
}
exit:
udev_device_cleanup(udev);
return 0;
}
static void exec_list(struct list_head *device_list)
{
struct device *loop_device;
struct device *tmp_device;
int i;
/* handle the "first" type devices first */
list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
for (i = 0; first_list[i] != NULL; i++) {
if (strncmp(loop_device->path, first_list[i], strlen(first_list[i])) == 0) {
add_device(loop_device->path);
list_del(&loop_device->node);
free(loop_device);
break;
}
}
}
/* handle the devices we are allowed to, excluding the "last" type devices */
list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
int found = 0;
for (i = 0; last_list[i] != NULL; i++) {
if (strncmp(loop_device->path, last_list[i], strlen(last_list[i])) == 0) {
found = 1;
break;
}
}
if (found)
continue;
add_device(loop_device->path);
list_del(&loop_device->node);
free(loop_device);
}
/* handle the rest of the devices left over, if any */
list_for_each_entry_safe(loop_device, tmp_device, device_list, node) {
add_device(loop_device->path);
list_del(&loop_device->node);
free(loop_device);
}
}
static int has_devt(const char *path)
{
char filename[PATH_SIZE];
struct stat statbuf;
snprintf(filename, sizeof(filename), "%s/dev", path);
filename[sizeof(filename)-1] = '\0';
if (stat(filename, &statbuf) == 0)
return 1;
return 0;
}
static void udev_scan_block(struct list_head *device_list)
{
char base[PATH_SIZE];
DIR *dir;
struct dirent *dent;
snprintf(base, sizeof(base), "%s/block", sysfs_path);
base[sizeof(base)-1] = '\0';
dir = opendir(base);
if (dir != NULL) {
for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
char dirname[PATH_SIZE];
DIR *dir2;
struct dirent *dent2;
if (dent->d_name[0] == '.')
continue;
snprintf(dirname, sizeof(dirname), "%s/%s", base, dent->d_name);
dirname[sizeof(dirname)-1] = '\0';
if (has_devt(dirname))
device_list_insert(dirname, device_list);
else
continue;
/* look for partitions */
dir2 = opendir(dirname);
if (dir2 != NULL) {
for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
char dirname2[PATH_SIZE];
if (dent2->d_name[0] == '.')
continue;
snprintf(dirname2, sizeof(dirname2), "%s/%s", dirname, dent2->d_name);
dirname2[sizeof(dirname2)-1] = '\0';
if (has_devt(dirname2))
device_list_insert(dirname2, device_list);
}
closedir(dir2);
}
}
closedir(dir);
}
}
static void udev_scan_class(struct list_head *device_list)
{
char base[PATH_SIZE];
DIR *dir;
struct dirent *dent;
snprintf(base, sizeof(base), "%s/class", sysfs_path);
base[sizeof(base)-1] = '\0';
dir = opendir(base);
if (dir != NULL) {
for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
char dirname[PATH_SIZE];
DIR *dir2;
struct dirent *dent2;
if (dent->d_name[0] == '.')
continue;
snprintf(dirname, sizeof(dirname), "%s/%s", base, dent->d_name);
dirname[sizeof(dirname)-1] = '\0';
dir2 = opendir(dirname);
if (dir2 != NULL) {
for (dent2 = readdir(dir2); dent2 != NULL; dent2 = readdir(dir2)) {
char dirname2[PATH_SIZE];
if (dent2->d_name[0] == '.')
continue;
snprintf(dirname2, sizeof(dirname2), "%s/%s", dirname, dent2->d_name);
dirname2[sizeof(dirname2)-1] = '\0';
if (has_devt(dirname2) || strcmp(dent->d_name, "net") == 0)
device_list_insert(dirname2, device_list);
}
closedir(dir2);
}
}
closedir(dir);
}
}
static void asmlinkage sig_handler(int signum)
{
switch (signum) {
case SIGALRM:
exit(1);
case SIGINT:
case SIGTERM:
exit(20 + signum);
}
}
int main(int argc, char *argv[], char *envp[])
{
LIST_HEAD(device_list);
struct sigaction act;
logging_init("udevstart");
udev_config_init();
dbg("version %s", UDEV_VERSION);
udev_run_str = getenv("UDEV_RUN");
udev_log_str = getenv("UDEV_LOG");
/* disable all logging if not explicitely requested */
if (udev_log_str == NULL)
udev_log_priority = 0;
/* set signal handlers */
memset(&act, 0x00, sizeof(act));
act.sa_handler = (void (*) (int))sig_handler;
sigemptyset (&act.sa_mask);
act.sa_flags = 0;
sigaction(SIGALRM, &act, NULL);
sigaction(SIGINT, &act, NULL);
sigaction(SIGTERM, &act, NULL);
/* trigger timeout to prevent hanging processes */
alarm(UDEV_ALARM_TIMEOUT);
sysfs_init();
udev_rules_init(&rules, 1);
udev_scan_class(&device_list);
udev_scan_block(&device_list);
exec_list(&device_list);
udev_rules_cleanup(&rules);
sysfs_cleanup();
logging_close();
return 0;
}