do a basic all-NUL check for the identify data before accepting it is valid

1 files changed, 14 insertions, 0 deletions

diff --git a/atasmart.c b/atasmart.c
index 435de05..ffc5fab 100644
--- a/atasmart.c
+++ b/atasmart.c
@@ -527,10 +527,12 @@ static int disk_identify_device(SkDisk *d) {
         uint16_t cmd[6];
         int ret;
         size_t len = 512;
+        const uint8_t *p;
 
         if (d->type == SK_DISK_TYPE_BLOB)
                 return 0;
 
+        memset(d->identify, 0, len);
         memset(cmd, 0, sizeof(cmd));
 
         cmd[1] = htons(1);
@@ -543,6 +545,18 @@ static int disk_identify_device(SkDisk *d) {
                 return -1;
         }
 
+        /* Check if IDENTIFY data is all NULs */
+        for (p = d->identify; p < (const uint8_t*) d->identify+len; p++)
+                if (*p) {
+                        p = NULL;
+                        break;
+                }
+
+        if (p) {
+                errno = EIO;
+                return -1;
+        }
+
         d->identify_valid = TRUE;
 
         return 0;