From 62a03bbf3235e733c09d6c7ef4f20ae4d43b81bf Mon Sep 17 00:00:00 2001
From: Werner Almesberger <werner@almesberger.net>
Date: Wed, 11 May 2011 05:21:45 -0300
Subject: [PATCH] tools/usbwait/usbwait.c: new option -r to require removal
 before appearance

- usbwait.c (need_removal, wait_for_usb): if need_removal is set, wait
  for the device to be removed before trying to find it
- usbwait.c (usage, main): new option -r to set need_removal
---
 tools/usbwait/usbwait.c | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/tools/usbwait/usbwait.c b/tools/usbwait/usbwait.c
index 344da71..f5fa739 100644
--- a/tools/usbwait/usbwait.c
+++ b/tools/usbwait/usbwait.c
@@ -24,19 +24,27 @@
 
 static useconds_t interval_us = DEFAULT_POLL_S*1000000;
 static unsigned long timeout = 0;
+static int need_removal = 0;
 
 
 static void wait_for_usb(void)
 {
 	struct timeval to, now;
+	usb_dev_handle *dev;
 
 	gettimeofday(&to, NULL);
 	to.tv_sec += timeout;
 
 	while (1) {
 		usb_rescan();
-		if (open_usb(0, 0))
-			return;
+		dev = open_usb(0, 0);
+		if (dev) {
+			if (!need_removal)
+				return;
+			usb_close(dev);
+		} else {
+			need_removal = 0;
+		}
 		if (timeout) {
 			gettimeofday(&now, NULL);
 			if (now.tv_sec > to.tv_sec)
@@ -59,9 +67,10 @@ static void wait_for_usb(void)
 static void usage(const char *name)
 {
 	fprintf(stderr,
-"usage: %s [-i poll_s] [-p path] [-t timeout_s] [vendor]:[product]\n\n"
+"usage: %s [-i poll_s] [-p path] [-r] [-t timeout_s] [vendor]:[product]\n\n"
 "  -i poll_s     poll interval in seconds (default: %g s)\n"
 "  -p path       USB device path\n"
+"  -r            wait for device removal first\n"
 "  -t timeout_s  timeout in seconds (default: infinite)\n"
   , name, DEFAULT_POLL_S);
 	exit(1);
@@ -73,7 +82,7 @@ int main(int argc, char **argv)
 	char *end;
 	int c;
 
-	while ((c = getopt(argc, argv, "i:p:t:")) != EOF)
+	while ((c = getopt(argc, argv, "i:p:rt:")) != EOF)
 		switch (c) {
 		case 'i':
 			interval_us = strtod(optarg, &end)*1000000;
@@ -83,6 +92,9 @@ int main(int argc, char **argv)
 		case 'p':
 			restrict_usb_path(optarg);
 			break;
+		case 'r':
+			need_removal = 1;
+			break;
 		case 't':
 			timeout = strtoul(optarg, &end, 0);
 			if (*end || !timeout)