git-svn-id: http://svn.ipfire.org/svn/ipfire/IPFire/source@16 ea5c0bd1-69bd-2848...

[people/teissler/ipfire-2.x.git] / src / misc-progs / restartdhcp.c

/* SmoothWall helper program - restartdhcp\r
 *\r
 * This program is distributed under the terms of the GNU General Public\r
 * Licence.  See the file COPYING for details.\r
 *\r
 * (c) Lawrence Manning, 2001\r
 * Simple program intended to be installed setuid(0) that can be used for\r
 * restarting DHCPd.\r
 * \r
 * $Id: restartdhcp.c,v 1.5.2.1 2004/11/03 13:50:26 alanh Exp $\r
 * \r
 */\r
\r
#include "libsmooth.h"\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <unistd.h>\r
#include <fcntl.h>\r
#include <string.h>\r
#include <sys/types.h>\r
#include <sys/stat.h>\r
#include <signal.h>\r
#include "setuid.h"\r
\r
int main(void)\r
{\r
	int fd = -1;\r
	int fdblue = -1;\r
	char buffer[STRING_SIZE];\r
	char blue_dev[STRING_SIZE] = "", green_dev[STRING_SIZE] = "";\r
	int pid;\r
	struct keyvalue *kv = NULL;\r
	\r
	if (!(initsetuid()))\r
		exit(1);\r
		\r
	memset(buffer, 0, STRING_SIZE);\r
\r
	/* Init the keyvalue structure */\r
	kv=initkeyvalues();\r
\r
	/* Read in the current values */\r
	if (!readkeyvalues(kv, CONFIG_ROOT "/ethernet/settings"))\r
	{\r
		fprintf(stderr, "Cannot read ethernet settings\n");\r
		exit(1);\r
	}\r
\r
	if (!findkey(kv, "GREEN_DEV", green_dev))\r
	{\r
		fprintf(stderr, "Cannot read GREEN_DEV\n");\r
		exit(1);\r
	}\r
\r
	if (!VALID_DEVICE(green_dev))\r
	{\r
		fprintf(stderr, "Bad GREEN_DEV: %s\n", green_dev);\r
		exit(1);\r
	}\r
\r
	/* Get the BLUE interface details */\r
	findkey(kv, "BLUE_DEV", blue_dev);\r
\r
	freekeyvalues(kv);\r
\r
	if ((fdblue = open(CONFIG_ROOT "/dhcp/enable_blue", O_RDONLY)) != -1)\r
	{\r
		close(fdblue);\r
		if (!VALID_DEVICE(blue_dev))\r
		{\r
			fprintf(stderr, "Bad BLUE_DEV: %s\n", blue_dev);\r
			exit(1);\r
		}\r
	}\r
\r
	if ((fd = open("/var/run/dhcpd.pid", O_RDONLY)) != -1)\r
	{\r
		if (read(fd, buffer, STRING_SIZE - 1) == -1)\r
			fprintf(stderr, "Couldn't read from pid file\n");\r
		else\r
		{\r
			pid = atoi(buffer);\r
			if (pid <= 1)\r
				fprintf(stderr, "Bad pid value\n");\r
			else\r
			{\r
				if (kill(pid, SIGTERM) == -1)\r
					fprintf(stderr, "Unable to send SIGTERM\n");\r
				else\r
					unlink("/var/run/dhcpd.pid");\r
			}\r
		}\r
		safe_system("/bin/killall -KILL dhcpd");\r
		close(fd);\r
	}\r
\r
	safe_system("/sbin/iptables -F DHCPBLUEINPUT");\r
\r
	buffer[0] = '\0';\r
\r
	if ((fd = open(CONFIG_ROOT "/dhcp/enable_green", O_RDONLY)) != -1)\r
	{\r
		close(fd);\r
		if ((fdblue = open(CONFIG_ROOT "/dhcp/enable_blue", O_RDONLY)) != -1)\r
		{\r
			close(fdblue);\r
			\r
			snprintf(buffer, STRING_SIZE-1, "/sbin/iptables -A DHCPBLUEINPUT -p tcp --source-port 68 --destination-port 67 -i %s -j ACCEPT > /dev/null 2>&1", blue_dev);\r
			safe_system(buffer);\r
			snprintf(buffer, STRING_SIZE-1, "/sbin/iptables -A DHCPBLUEINPUT -p udp --source-port 68 --destination-port 67 -i %s -j ACCEPT > /dev/null 2>&1", blue_dev);\r
			safe_system(buffer);\r
			snprintf(buffer, STRING_SIZE-1, "/usr/sbin/dhcpd -q %s %s", green_dev, blue_dev);\r
		} else {\r
			snprintf(buffer, STRING_SIZE-1, "/usr/sbin/dhcpd -q %s", green_dev);\r
		}\r
		safe_system(buffer);\r
	} else {\r
		if ((fdblue = open(CONFIG_ROOT "/dhcp/enable_blue", O_RDONLY)) != -1)\r
		{\r
			close(fdblue);\r
\r
			snprintf(buffer, STRING_SIZE-1, "/sbin/iptables -A DHCPBLUEINPUT -p tcp --source-port 68 --destination-port 67 -i %s -j ACCEPT > /dev/null 2>&1", blue_dev);\r
			safe_system(buffer);\r
			snprintf(buffer, STRING_SIZE-1, "/sbin/iptables -A DHCPBLUEINPUT -p udp --source-port 68 --destination-port 67 -i %s -j ACCEPT > /dev/null 2>&1", blue_dev);\r
			safe_system(buffer);\r
			snprintf(buffer, STRING_SIZE-1,  "/usr/sbin/dhcpd -q %s", blue_dev);\r
			safe_system(buffer);\r
		}\r
	}\r
\r
	if (buffer[0] != '\0')\r
	{\r
		/* Silly dhcpd creates pids with mode 640 */\r
		sleep (1);\r
		if ((fd = open("/var/run/dhcpd.pid", 0)) == -1)\r
		{\r
			fprintf(stderr, "No pid file\n");\r
			return 1;\r
		}\r
		fchmod(fd, 00644);\r
		close(fd);\r
	}\r
\r
	return 0;\r
}\r