nob_read_entire_file

Author: MarekKnapek

nob.h

@@ -2679,18 +2679,68 @@ NOBDEF bool nob_rename(const char *old_path, const char *new_path)
 
 NOBDEF bool nob_read_entire_file(const char *path, Nob_String_Builder *sb)
 {
+#ifdef _WIN32
+    bool result;
+    HANDLE file;
+    DWORD chunk_size;
+    size_t mark;
+    wchar_t *wide_path;
+    DWORD err;
+    size_t new_count;
+    BOOL b;
+    DWORD read;
+
+    result = true;
+    file = INVALID_HANDLE_VALUE;
+    chunk_size = 64 * 1024;
+    mark = nob_temp_save();
+    wide_path = nob__unicode_utf8_to_unicode_utf16_temp(path);
+    file = CreateFileW(wide_path, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+    err = GetLastError();
+    nob_temp_rewind(mark);
+    if (file == INVALID_HANDLE_VALUE)
+    {
+        nob_log(NOB_ERROR, "Could not open file %s for reading: %s\n", path, nob_win32_error_message(err));
+        nob_return_defer(false);
+    }
+    for (;;)
+    {
+        new_count = sb->count + chunk_size;
+        if (new_count > sb->capacity)
+        {
+            sb->items = NOB_DECLTYPE_CAST(sb->items)NOB_REALLOC(sb->items, 2 * new_count);
+            NOB_ASSERT(sb->items != NULL && "Buy more RAM lool!!");
+            sb->capacity = new_count;
+        }
+        b = ReadFile(file, sb->items + sb->count, chunk_size, &read, NULL);
+        if (b == FALSE)
+        {
+            err = GetLastError();
+            nob_log(NOB_ERROR, "Could not read from file %s: %s\n", path, nob_win32_error_message(err));
+            nob_return_defer(false);
+        }
+        sb->count += read;
+        if (read != chunk_size)
+        {
+            break;
+        }
+    }
+defer:
+    if (file != INVALID_HANDLE_VALUE)
+    {
+        b = CloseHandle(file);
+        NOB_ASSERT(b != FALSE);
+    }
+    return result;
+#else
     bool result = true;
 
     FILE *f = fopen(path, "rb");
     size_t new_count = 0;
     long long m = 0;
     if (f == NULL)                 nob_return_defer(false);
     if (fseek(f, 0, SEEK_END) < 0) nob_return_defer(false);
-#ifndef _WIN32
     m = ftell(f);
-#else
-    m = _telli64(_fileno(f));
-#endif
     if (m < 0)                     nob_return_defer(false);
     if (fseek(f, 0, SEEK_SET) < 0) nob_return_defer(false);
 
@@ -2712,6 +2762,7 @@ NOBDEF bool nob_read_entire_file(const char *path, Nob_String_Builder *sb)
     if (!result) nob_log(NOB_ERROR, "Could not read file %s: %s", path, strerror(errno));
     if (f) fclose(f);
     return result;
+#endif // _WIN32
 }
 
 NOBDEF int nob_sb_appendf(Nob_String_Builder *sb, const char *fmt, ...)

tests/unicode.c

@@ -68,6 +68,7 @@ static void test_unicode_utf8_file_operations(void)
 	Nob_Fd fd;
 	Nob_File_Type ft;
 	bool b;
+	Nob_String_Builder sb;
 
 	nob_log(NOB_INFO, "%s", "Testing file operations...");
 	n = NOB_ARRAY_LEN(k_strings);
@@ -92,6 +93,15 @@ static void test_unicode_utf8_file_operations(void)
 		b = nob_write_entire_file(k_strings[i], "test", 4);
 		test(b);
 
+		sb.items = NULL;
+		sb.capacity = 0;
+		sb.count = 0;
+		b = nob_read_entire_file(k_strings[i], &sb);
+		test(b);
+		test(sb.count == 4);
+		test(memcmp(sb.items, "test", 4) == 0);
+		NOB_FREE(sb.items);
+
 		b = nob_delete_file(k_strings[i]);
 		test(b);
 	}