1 files changed, 29 insertions, 26 deletions

diff --git a/usbmuxd/main.c b/usbmuxd/main.c
index 525e095..182178f 100644
--- a/usbmuxd/main.c
+++ b/usbmuxd/main.c
@@ -373,29 +373,6 @@ int main(int argc, char *argv[])
 		goto terminate;
 	}
 
-	usbmuxd_log(LL_INFO, "Creating socket");
-	listenfd = create_socket();
-	if(listenfd < 0)
-		return 1;
-
-	client_init();
-	device_init();
-	usbmuxd_log(LL_INFO, "Initializing USB");
-	if((res = usb_init()) < 0)
-		return 2;
-	usbmuxd_log(LL_INFO, "%d device%s detected", res, (res==1)?"":"s");
-	
-	usbmuxd_log(LL_NOTICE, "Initialization complete");
-
-	if (!foreground) {
-		if (daemonize() < 0) {
-			fprintf(stderr, "usbmuxd: FATAL: Could not daemonize!\n");
-			usbmuxd_log(LL_ERROR, "FATAL: Could not daemonize!");
-			log_disable_syslog();
-			exit(EXIT_FAILURE);
-		}
-	}
-
 	// now open the lockfile and place the lock
 	lfd = fopen(lockfile, "w");
 	if (lfd) {
@@ -403,13 +380,21 @@ int main(int argc, char *argv[])
 		lock.l_whence = SEEK_SET;
 		lock.l_start = 0;
 		lock.l_len = 0;
-		if (fcntl(fileno(lfd), F_SETLK, &lock) == -1) {
+		if ((res = fcntl(fileno(lfd), F_SETLK, &lock)) < 0) {
 			usbmuxd_log(LL_FATAL, "Lockfile locking failed!");
-			log_disable_syslog();
-			exit(EXIT_FAILURE);
+			goto terminate;
 		}
+	} else {
+		usbmuxd_log(LL_FATAL, "Lockfile open failed!");
+		res = -1;
+		goto terminate;
 	}
 
+	usbmuxd_log(LL_INFO, "Creating socket");
+	res = listenfd = create_socket();
+	if(listenfd < 0)
+		goto terminate;
+
 	// drop elevated privileges
 	if (drop_privileges && (getuid() == 0 || geteuid() == 0)) {
 		struct passwd *pw = getpwnam(drop_user);
@@ -446,6 +431,24 @@ int main(int argc, char *argv[])
 		usbmuxd_log(LL_NOTICE, "Successfully dropped privileges to '%s'", drop_user);
 	}
 
+	client_init();
+	device_init();
+	usbmuxd_log(LL_INFO, "Initializing USB");
+	if((res = usb_init()) < 0)
+		goto terminate;
+
+	usbmuxd_log(LL_INFO, "%d device%s detected", res, (res==1)?"":"s");
+
+	usbmuxd_log(LL_NOTICE, "Initialization complete");
+
+	if (!foreground) {
+		if ((res = daemonize()) < 0) {
+			fprintf(stderr, "usbmuxd: FATAL: Could not daemonize!\n");
+			usbmuxd_log(LL_FATAL, "Could not daemonize!");
+			goto terminate;
+		}
+	}
+
 	res = main_loop(listenfd);
 	if(res < 0)
 		usbmuxd_log(LL_FATAL, "main_loop failed");