Let the daemon terminate only when no more device is attached

1 files changed, 30 insertions, 15 deletions

diff --git a/src/main.c b/src/main.c
index b4b5283..3617e14 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1006,18 +1006,6 @@ static int daemonize()
 	return 0;
 }
 
-/**
- * signal handler function for cleaning up properly
- */
-static void clean_exit(int sig)
-{
-	if (sig == SIGINT) {
-		if (verbose >= 1)
-			fprintf(stderr, "CTRL+C pressed\n");
-	}
-	quit_flag = 1;
-}
-
 static void usage()
 {
 	printf("usage: usbmuxd [options]\n");
@@ -1093,6 +1081,33 @@ static int devices_attached()
 }
 
 /**
+ * signal handler function for cleaning up properly
+ */
+static void handle_signal(int sig)
+{
+	if (sig == SIGTERM) {
+		quit_flag = 1;
+	} else {
+		if (sig == SIGINT) {
+			if (verbose >= 1)
+				fprintf(stderr, "CTRL+C pressed\n");
+		}
+
+		if (verbose >= 1)
+			fprintf(stderr, "Checking if we can terminate (no more devices attached)...\n");
+
+		if (devices_attached() > 0) {
+			// we can't quit, there are still devices attached.
+			if (verbose >= 1)
+				fprintf(stderr, "Refusing to terminate, there are still devices attached. Kill me with signal 15 (TERM) to force quit.\n");
+		} else {
+			// it's safe to quit
+			quit_flag = 1;
+		}
+	}
+}
+
+/**
  * main function. Initializes all stuff and then loops waiting in accept.
  */
 int main(int argc, char **argv)
@@ -1121,9 +1136,9 @@ int main(int argc, char **argv)
 		logmsg(LOG_NOTICE, "starting");
 
 	// signal(SIGHUP, reload_conf); // none yet
-	signal(SIGINT, clean_exit);
-	signal(SIGQUIT, clean_exit);
-	signal(SIGTERM, clean_exit);
+	signal(SIGINT, handle_signal);
+	signal(SIGQUIT, handle_signal);
+	signal(SIGTERM, handle_signal);
 	signal(SIGPIPE, SIG_IGN);
 
 	// check for other running instance