1 files changed, 57 insertions, 9 deletions

diff --git a/main.c b/main.c
index c2467a2..6773d0e 100644
--- a/main.c
+++ b/main.c
@@ -42,13 +42,13 @@ int create_socket(void) {
 	int listenfd;
 
 	if(unlink(socket_path) == -1 && errno != ENOENT) {
-		usbmuxd_log(LOG_FATAL, "unlink(%s) failed: %s", socket_path, strerror(errno));
+		usbmuxd_log(LL_FATAL, "unlink(%s) failed: %s", socket_path, strerror(errno));
 		return -1;
 	}
 
 	listenfd = socket(AF_UNIX, SOCK_STREAM, 0);
 	if (listenfd == -1) {
-		usbmuxd_log(LOG_FATAL, "socket() failed: %s", strerror(errno));
+		usbmuxd_log(LL_FATAL, "socket() failed: %s", strerror(errno));
 		return -1;
 	}
 
@@ -56,37 +56,85 @@ int create_socket(void) {
 	bind_addr.sun_family = AF_UNIX;
 	strcpy(bind_addr.sun_path, socket_path);
 	if (bind(listenfd, (struct sockaddr*)&bind_addr, sizeof(bind_addr)) != 0) {
-		usbmuxd_log(LOG_FATAL, "bind() failed: %s", strerror(errno));
+		usbmuxd_log(LL_FATAL, "bind() failed: %s", strerror(errno));
 		return -1;
 	}
 
 	// Start listening
 	if (listen(listenfd, 5) != 0) {
-		usbmuxd_log(LOG_FATAL, "listen() failed: %s", strerror(errno));
+		usbmuxd_log(LL_FATAL, "listen() failed: %s", strerror(errno));
 		return -1;
 	}
 
 	return listenfd;
 }
 
+int main_loop(int listenfd)
+{
+	int to, cnt, i;
+	struct fdlist pollfds;
+	
+	while(1) {
+		usbmuxd_log(LL_SPEW, "main_loop iteration");
+		to = usb_get_timeout();
+		usbmuxd_log(LL_SPEW, "USB timeout is %d ms", to);
+		
+		fdlist_create(&pollfds);
+		usb_get_fds(&pollfds);
+		usbmuxd_log(LL_SPEW, "fd count is %d", pollfds.count);
+		
+		cnt = poll(pollfds.fds, pollfds.count, to);
+		usbmuxd_log(LL_SPEW, "poll() returned %d", cnt);
+		
+		if(cnt == 0) {
+			if(usb_process() < 0) {
+				usbmuxd_log(LL_FATAL, "usb_process() failed");
+				return -1;
+			}
+		} else {
+			for(i=0; i<pollfds.count; i++) {
+				if(pollfds.fds[i].revents) {
+					if(pollfds.owners[i] == FD_USB) {
+						if(usb_process() < 0) {
+							usbmuxd_log(LL_FATAL, "usb_process() failed");
+							return -1;
+						}
+					}
+				}
+			}
+		}
+		fdlist_free(&pollfds);
+	}
+}
+
 int main(int argc, char *argv[])
 {
 	int listenfd;
 	int res;
 
-	usbmuxd_log(LOG_NOTICE, "usbmux v0.1 starting up");
+	usbmuxd_log(LL_NOTICE, "usbmux v0.1 starting up");
 
-	usbmuxd_log(LOG_INFO, "Creating socket");
+	usbmuxd_log(LL_INFO, "Creating socket");
 	listenfd = create_socket();
 	if(listenfd < 0)
 		return 1;
 
-	usbmuxd_log(LOG_INFO, "Initializing USB");
+	usbmuxd_log(LL_INFO, "Initializing USB");
 	if((res = usb_init()) < 0)
 		return 2;
-	usbmuxd_log(LOG_INFO, "%d device%s detected", res, (res==1)?"":"s");
+	usbmuxd_log(LL_INFO, "%d device%s detected", res, (res==1)?"":"s");
 	
-	usbmuxd_log(LOG_NOTICE, "initialization complete");
+	usbmuxd_log(LL_NOTICE, "Initialization complete");
+
+	res = main_loop(listenfd);
+	if(res < 0)
+		usbmuxd_log(LL_FATAL, "main_loop failed");
+
+	usbmuxd_log(LL_NOTICE, "usbmux shutting down");
+	usb_shutdown();
+	usbmuxd_log(LL_NOTICE, "Shutdown complete");
 	
+	if(res < 0)
+		return -res;
 	return 0;
 }