Merge remote-tracking branch 'upstream/master'

[cc65] / libsrc / cbm / readdir.c

diff --git a/libsrc/cbm/readdir.c b/libsrc/cbm/readdir.c
index 0a20bdd06660504a296d40bcd2a79cd95964efd0..d6947316bf0039466b7c0ac7b810c787572b509b 100644 (file)
--- a/libsrc/cbm/readdir.c
+++ b/libsrc/cbm/readdir.c
@@ -18,8 +18,8 @@ struct dirent* __fastcall__ readdir (register DIR* dir)
     register unsigned char* b;
     register unsigned char i;
     register unsigned char count;
-    unsigned char s;
-    unsigned char j;
+    static unsigned char s;
+    static unsigned char j;
     unsigned char buffer[0x40];
     static struct dirent entry;
 
@@ -32,7 +32,6 @@ struct dirent* __fastcall__ readdir (register DIR* dir)
         /* errno already set */
         goto exitpoint;
     }
-    dir->off += 2;
 
     /* Read the number of blocks */
     if (!_dirread (dir, &entry.d_blocks, sizeof (entry.d_blocks))) {
@@ -40,27 +39,27 @@ struct dirent* __fastcall__ readdir (register DIR* dir)
     }
 
     /* Read the next file entry into the buffer */
-    for (count = 0, b = buffer; count < sizeof (buffer); ++count, ++b) {
+    for (count = 0, b = buffer; count < sizeof (buffer); ++b) {
         if (!_dirread1 (dir, b)) {
             goto exitpoint;
         }
+        ++count;
         if (*b == '\0') {
             break;
         }
     }
 
+    /* Bump the directory offset and include the bytes for line-link and size */
+    dir->off += count + 4;
+
     /* End of directory is reached if the buffer contains "blocks free". It is
-     * sufficient here to check for the leading 'b'. To avoid problems if we're
-     * called again, read until end of directory.
+     * sufficient here to check for the leading 'b'. buffer will contain at
+     * least one byte if we come here.
      */
-    if (count > 0 && buffer[0] == 'b') {
-        while (_dirread1 (dir, buffer)) ;
-        return 0;
+    if (buffer[0] == 'b') {
+        goto exitpoint;
     }
 
-    /* Bump the directory offset */
-    dir->off += count;
-
     /* Parse the buffer for the filename and file type */
     i = 0;
     j = 0;
@@ -79,9 +78,17 @@ struct dirent* __fastcall__ readdir (register DIR* dir)
             case 1:
                 /* Within file name */
                 if (*b == '"') {
+                    /* End of file name found. */
                     entry.d_name[j] = '\0';
                     entry.d_namlen = j;
-                    s = 2;
+                    if (entry.d_off > 2) {
+                        /* Proceed with file type */
+                        s = 2;
+                    } else {
+                        /* This is a disk header, so we're done */
+                        entry.d_type = _CBM_T_HEADER;
+                        return &entry;
+                    }
                 } else if (j < sizeof (entry.d_name) - 1) {
                     entry.d_name[j] = *b;
                     ++j;
@@ -106,8 +113,8 @@ struct dirent* __fastcall__ readdir (register DIR* dir)
                 /* Distinguish DEL or DIR file type entries */
                 switch (*b) {
                     case 'e':                                   break;
-                    case 'i': entry.d_type = CBM_T_DIR;         break;
-                    default:  entry.d_type = CBM_T_OTHER;       break;
+                    case 'i': entry.d_type = _CBM_T_DIR;        break;
+                    default:  entry.d_type = _CBM_T_OTHER;      break;
                 }
                 return &entry;
         }