added --exit-on-no-devices option.

1 files changed, 42 insertions, 1 deletions

diff --git a/main.c b/main.c
index 0af0735..8560aac 100644
--- a/main.c
+++ b/main.c
@@ -53,6 +53,7 @@ static int quit_flag = 0;
 static int fsock = -1;
 static int verbose = DEBUG_LEVEL;
 static int foreground = 0;
+static int exit_on_no_devices = 0;
 
 struct device_use_info {
     uint32_t device_id;
@@ -869,12 +870,13 @@ static void parse_opts(int argc, char **argv)
 	{ "help",       0, NULL, 'h' },
 	{ "foreground", 0, NULL, 'f' },
 	{ "verbose",    0, NULL, 'v' },
+	{ "exit-on-no-devices", 0, NULL, 'e' },
 	{ NULL,         0, NULL, 0}
     };
     int c;
 
     while (1) {
-	c = getopt_long(argc, argv, "hfv", longopts, (int *) 0);
+	c = getopt_long(argc, argv, "hfve", longopts, (int *) 0);
 	if (c == -1) {
 	    break;
 	}
@@ -889,6 +891,9 @@ static void parse_opts(int argc, char **argv)
 	    case 'v':
 		sock_stuff_set_verbose(++verbose);
 		break;
+	    case 'e':
+		exit_on_no_devices = 1;
+		break;
 	    default:
 		usage();
 		exit(2);
@@ -897,6 +902,35 @@ static void parse_opts(int argc, char **argv)
 }
 
 /**
+ * checks for attached devices
+ *
+ * @return number of devices found
+ */
+static int devices_attached()
+{
+    struct usb_bus *bus;
+    struct usb_device *dev;
+    int res = 0;
+
+    usb_init();
+    usb_find_busses();
+    usb_find_devices();
+
+    for (bus = usb_get_busses(); bus; bus = bus->next) {
+	for (dev = bus->devices; dev; dev = dev->next) {
+	    if (dev->descriptor.idVendor == 0x05ac
+		&& dev->descriptor.idProduct >= 0x1290
+		&& dev->descriptor.idProduct <= 0x1293)
+	    {
+		res++;
+	    }
+	}
+    }
+
+    return res;
+}
+
+/**
  * main function. Initializes all stuff and then loops waiting in accept.
  */
 int main(int argc, char **argv)
@@ -944,6 +978,13 @@ int main(int argc, char **argv)
 	}
     }
 
+    if (exit_on_no_devices) {
+	if (devices_attached() <= 0) {
+	    logmsg(LOG_NOTICE, "no devices attached. exiting.");
+	    return 0;
+	}
+    }
+
     fsock = create_unix_socket(USBMUXD_SOCKET_FILE);
     if (fsock < 0) {
 	logmsg(LOG_ERR, "Could not create socket, exiting");