X-Git-Url: http://shamusworld.gotdns.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=init-ng.c;h=7dba91aeb3aa375fcbad2410753ba6928d9733c1;hb=HEAD;hp=3b969be0e62235aa03ee59df65cc1f0e173d1d9d;hpb=9f23f04c3792ce4c64fec9f1b8116e0586f50ce0;p=init-ng

diff --git a/init-ng.c b/init-ng.c
index 3b969be..7dba91a 100644
--- a/init-ng.c
+++ b/init-ng.c
@@ -10,21 +10,20 @@
 // (C) 2017 Underground Software
 //
 
-#define _GNU_SOURCE		// Without this, functions in unistd.h go missing
 #include <unistd.h>
 #include <errno.h>
 #include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <linux/reboot.h>
+#include <sys/reboot.h>
 #include <sys/syscall.h>
-#include <sys/socket.h>
+#include <sys/stat.h>
 #include <sys/types.h>
 #include <sys/un.h>
 #include <sys/wait.h>
 
-static const char socketPath[] = "/run/initctl";
+static const char fifoPath[] = "/dev/initctl";
 static int shuttingDown = 0;
 static int verbose = 0;		// N.B.: There is no mechanism to set this ATM
 static struct sigaction action;
@@ -39,7 +38,7 @@ static char * args[256];	// 256 arguments ought to be enough for anybody ;-)
 // Find a process in the monitored process list (MPL) by PID. Returns the index
 // of the entry in the MPL.
 //
-int FindMonitoredPID(pid_t pid)
+static int FindMonitoredPID(pid_t pid)
 {
 	for(int i=0; i<freeProcessSlot; i++)
 	{
@@ -54,7 +53,7 @@ int FindMonitoredPID(pid_t pid)
 //
 // Remove the given process from the MPL by index
 //
-void RemoveEntry(int entry)
+static void RemoveEntry(int entry)
 {
 	free(processCmd[entry]);
 	free(processName[entry]);
@@ -68,7 +67,7 @@ void RemoveEntry(int entry)
 //
 // Add a process to the MPL
 //
-void AddEntry(pid_t pid, char * name, char * command)
+static void AddEntry(pid_t pid, const char * name, const char * command)
 {
 	int i;
 
@@ -91,7 +90,7 @@ void AddEntry(pid_t pid, char * name, char * command)
 // Take a given space delimited string and turn it into an array of strings
 // suitable for execv().
 //
-void MakeArgumentList(char * s)
+static void MakeArgumentList(const char * s)
 {
 	char buf[4096];
 	char * tokSave;
@@ -120,7 +119,7 @@ void MakeArgumentList(char * s)
 // monitor: Tells Launch whether or not this process should be monitored
 //          and respawned if it goes away (0 = no, 1 = yes)
 //
-pid_t Launch(char * name, char * command, int monitor)
+static pid_t Launch(const char * name, const char * command, int monitor)
 {
 	if (verbose)
 		printf("Starting %s (%smonitored)...\n", name, (monitor ? "" : "un"));
@@ -158,7 +157,7 @@ pid_t Launch(char * name, char * command, int monitor)
 // At some point, we'll read inittab instead of recording just our own internal
 // processes...
 //
-void HandleSIGCHLD(int signum, siginfo_t * info, void * ptr)
+static void HandleSIGCHLD(int signum)
 {
 	while (1)
 	{
@@ -196,9 +195,9 @@ void HandleSIGCHLD(int signum, siginfo_t * info, void * ptr)
 //
 // Start up the system, using the awesome power of OpenRC
 //
-void Init(void)
+static void Init(void)
 {
-	printf("*** init-ng v1.0.0 starting...\n");
+	printf("*** init-ng v1.0.1 starting...\n");
 
 	pid_t pid = Launch("openrc", "/sbin/rc sysinit", 0);
 	waitpid(pid, NULL, 0);
@@ -211,7 +210,7 @@ void Init(void)
 }
 
 
-void ShutDown(int cmd)
+static void ShutDown(int cmd)
 {
 	shuttingDown = 1;
 	pid_t pid = Launch("openrc", "/sbin/rc reboot", 0);
@@ -221,41 +220,26 @@ void ShutDown(int cmd)
 	// to work without any problem as is.
 	while (waitpid(pid, NULL, 0) != pid);
 
-	// Could just call sync() as well...
-	syscall(SYS_sync);
-	syscall(SYS_reboot, LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2C, cmd);
+	sync();
+	reboot(cmd);
 }
 
 
 //
-// Connect to PID 1's socket to send a command
+// Connect to PID 1's FIFO to send a command
 //
-void DoCmd(char * cmd)
+static void DoCmd(const char * cmd)
 {
-	struct sockaddr_un remote;
-	int s = socket(AF_UNIX, SOCK_STREAM, 0);
+	FILE * file = fopen(fifoPath, "w");
 
-	if (s == -1)
+	if (file == NULL)
 	{
-		// Handle error
-		printf("Could not get socket in DoCmd()...\n");
+		perror("DoCmd: fopen");
 		return;
 	}
 
-	remote.sun_family = AF_UNIX;
-	strcpy(remote.sun_path, socketPath);
-	int len = strlen(remote.sun_path) + sizeof(remote.sun_family);
-	int s2 = connect(s, (struct sockaddr *)&remote, len);
-
-	if (s2 == -1)
-	{
-		// Handle error
-		printf("Could not connect in DoCmd()...\n");
-		return;
-	}
-
-	send(s, cmd, strlen(cmd), 0);
-	close(s);
+	size_t ignored = fwrite(cmd, 1, strlen(cmd), file);
+	fclose(file);
 }
 
 
@@ -278,7 +262,7 @@ int main(int argc, char * argv[])
 			}
 			else if (strcmp(argv[1], "test") == 0)
 			{
-				ShutDown(LINUX_REBOOT_CMD_POWER_OFF);
+				ShutDown(RB_POWER_OFF);
 			}
 			else
 				printf("Invalid command\n");
@@ -304,81 +288,58 @@ int main(int argc, char * argv[])
 
 	// Install SIGCHLD signal handler to do process monitoring
 	memset(&action, 0, sizeof(action));
-	action.sa_sigaction = HandleSIGCHLD;
+	action.sa_handler = HandleSIGCHLD;
 	action.sa_flags = SA_SIGINFO;
 	sigaction(SIGCHLD, &action, NULL);
 
-	// Create socket to listen on for commands...
-	struct sockaddr_un local, remote;
-	unsigned int s = socket(AF_UNIX, SOCK_STREAM, 0);
-
-	if (s == -1)
+	// Create a named pipe to listen on for commands
+	if (mkfifo(fifoPath, 0600) == -1)
 	{
-		perror("socket");
-		printf("init-ng: socket() failed!\n");
-	}
-
-	local.sun_family = AF_UNIX;
-	strcpy(local.sun_path, socketPath);
-	unlink(local.sun_path);
-	int len = strlen(local.sun_path) + sizeof(local.sun_family);
-
-	if (bind(s, (struct sockaddr *)&local, len) == -1)
-	{
-		perror("bind");
-		printf("init-ng: bind() failed!\n");
-	}
-
-	// Queue size of 20 ought to be enough for anybody ;-)
-	if (listen(s, 20) == -1)
-	{
-		perror("listen");
-		printf("init-ng: listen() failed!\n");
+		// This is bad... If this happens, we have no comms
+		perror("mkfifo");
 	}
 
 	// Main loop for PID 1. All commands are received and acted upon from here.
 	while (1)
 	{
-		int t = sizeof(remote);
-		unsigned int s2 = accept(s, (struct sockaddr *)&remote, &t);
+		// This will block until a command is sent down the pipe...
+		FILE * file = fopen(fifoPath, "r");
 
-		if (s2 == -1)
+		if (file == NULL)
 		{
 			// Check to see if the signal handler blew things up on us
 			if (errno == EINTR)
 				continue;
 
-			// An unanticipated error occurred, handle it
-			perror("accept");
-			printf("init-ng: accept() failed!\n");
+			perror("fopen");
+			continue;
 		}
 
-		// Get the message from the socket
-		int n = recv(s2, buf, 2048, 0);
-		buf[n] = 0;
+		// Grab whatever was shoved into the FIFO
+		size_t read = fread(buf, 1, 2047, file);
+		buf[read] = 0;
+		fclose(file);
 
 		if (verbose)
-			printf("PID1: Received \"%s\" from socket...\n", buf);
+			printf("PID1: Received \"%s\" from FIFO...\n", buf);
 
 		// Now do something with it...
 		if (strcmp(buf, "poweroff") == 0)
 		{
-			ShutDown(LINUX_REBOOT_CMD_POWER_OFF);
+			ShutDown(RB_POWER_OFF);
 		}
 		else if (strcmp(buf, "restart") == 0)
 		{
-			ShutDown(LINUX_REBOOT_CMD_RESTART);
+			ShutDown(RB_AUTOBOOT);
 		}
 		else if (strcmp(buf, "halt") == 0)
 		{
-			ShutDown(LINUX_REBOOT_CMD_HALT);
+			ShutDown(RB_HALT_SYSTEM);
 		}
 		else if (strcmp(buf, "status") == 0)
 		{
 			printf("[Status placeholder]\n");
 		}
-
-		close(s2);
 	}
 
 	// We should never get here...