From 0488138c43e93d335933a5073bc2003b4cfb2338 Mon Sep 17 00:00:00 2001
From: PartialVolume <nick.craig.law@gmail.com>
Date: Wed, 1 Apr 2020 21:29:08 +0100
Subject: [PATCH] Fix --nousb missing a drive

If a USB device generated a particular error
code from smartmontools, the --nousb option
incorrectly did not exclude it from the
enumerated drives.

This has now been corrected.
---
 src/device.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/device.c b/src/device.c
index d9100a1..d9f6e04 100644
--- a/src/device.c
+++ b/src/device.c
@@ -159,7 +159,8 @@ int check_device( nwipe_context_t*** c, PedDevice* dev, int dcount )
         /* retrieve bus and drive serial number, HOWEVER we are only interested in the bus at this time */
         r = nwipe_get_device_bus_type_and_serialno( dev->path, &bus, tmp_serial );
 
-        if( r == 0 || r == 5 )
+        /* See nwipe_get_device_bus_type_and_serialno() function for meaning of these codes */
+        if( r == 0 || ( r >= 3 && r <= 6 ) )
         {
             if( bus == NWIPE_DEVICE_USB )
             {