systemd/src/test-udev.c

/*
 * Copyright (C) 2003-2004 Greg Kroah-Hartman <greg@kroah.com>
 * Copyright (C) 2004-2008 Kay Sievers <kay.sievers@vrfy.org>
 *
 * 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, either version 2 of the License, or
 * (at your option) any later version.
 *
 * 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, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <ctype.h>
#include <errno.h>
#include <unistd.h>
#include <syslog.h>
#include <grp.h>
#include <sys/signalfd.h>

#include "udev.h"

void udev_main_log(struct udev *udev, int priority,
		   const char *file, int line, const char *fn,
		   const char *format, va_list args) {}

int main(int argc, char *argv[])
{
	struct udev *udev;
	struct udev_event *event = NULL;
	struct udev_device *dev = NULL;
	struct udev_rules *rules = NULL;
	char syspath[UTIL_PATH_SIZE];
	const char *devpath;
	const char *action;
	sigset_t mask, sigmask_orig;
	int err = -EINVAL;

	udev = udev_new();
	if (udev == NULL)
		exit(1);
	info(udev, "version %s\n", VERSION);
	udev_selinux_init(udev);

	sigprocmask(SIG_SETMASK, NULL, &sigmask_orig);

	action = argv[1];
	if (action == NULL) {
		err(udev, "action missing\n");
		goto out;
	}

	devpath = argv[2];
	if (devpath == NULL) {
		err(udev, "devpath missing\n");
		goto out;
	}

	rules = udev_rules_new(udev, 1);

	util_strscpyl(syspath, sizeof(syspath), udev_get_sys_path(udev), devpath, NULL);
	dev = udev_device_new_from_syspath(udev, syspath);
	if (dev == NULL) {
		info(udev, "unknown device '%s'\n", devpath);
		goto out;
	}

	udev_device_set_action(dev, action);
	event = udev_event_new(dev);

	sigfillset(&mask);
	sigprocmask(SIG_SETMASK, &mask, &sigmask_orig);
	event->fd_signal = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
	if (event->fd_signal < 0) {
		fprintf(stderr, "error creating signalfd\n");
		goto out;
	}

	/* do what devtmpfs usually provides us */
	if (udev_device_get_devnode(dev) != NULL) {
		mode_t mode;

		if (strcmp(udev_device_get_subsystem(dev), "block") == 0)
			mode |= S_IFBLK;
		else
			mode |= S_IFCHR;

		if (strcmp(action, "remove") != 0) {
			util_create_path(udev, udev_device_get_devnode(dev));
			mknod(udev_device_get_devnode(dev), mode, udev_device_get_devnum(dev));
		} else {
			unlink(udev_device_get_devnode(dev));
			util_delete_path(udev, udev_device_get_devnode(dev));
		}
	}

	err = udev_event_execute_rules(event, rules, &sigmask_orig);
	if (err == 0)
		udev_event_execute_run(event, NULL);
out:
	if (event != NULL && event->fd_signal >= 0)
		close(event->fd_signal);
	udev_event_unref(event);
	udev_device_unref(dev);
	udev_rules_unref(rules);
	udev_selinux_exit(udev);
	udev_unref(udev);
	if (err != 0)
		return 1;
	return 0;
}