From 15ca13a9853e6810c8d082122a66165032b78e5a Mon Sep 17 00:00:00 2001
From: Xnoe <xnoe@xnoe.moe>
Date: Sat, 27 Nov 2021 02:31:23 +0000
Subject: [PATCH] Update atapio.cpp to use new rather than dumb_alloc, add
 file_size

---
 src/kernel/atapio.cpp | 17 ++++++++++++++---
 src/kernel/atapio.h   |  3 +++
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/kernel/atapio.cpp b/src/kernel/atapio.cpp
index 0a6c310..e38b4c9 100644
--- a/src/kernel/atapio.cpp
+++ b/src/kernel/atapio.cpp
@@ -21,10 +21,10 @@ uint16_t sectorsPerFAT;
 
 
 void init_atapio() {
-  rootDirEntries = (uint8_t*)dumb_alloc(8192);
-  FAT1 = (uint16_t*)dumb_alloc(512 * 34);
+  rootDirEntries = new uint8_t[8192];
+  FAT1 = (uint16_t*)(new uint8_t[512 * 34]);
 
-  uint32_t boot_sector = (uint32_t)dumb_alloc(4096);
+  uint32_t boot_sector = new uint32_t[1024];
   read_sectors(0, 1, (uint8_t*)boot_sector);
 
   countReserved = *((uint16_t*)(boot_sector + 0x0e));
@@ -125,4 +125,15 @@ void load_file(char* filename, uint8_t* destination) {
     if (location == 0xffff)
       loaded = true;
   }
+}
+
+uint32_t file_size(char* filename) {
+  for (int i=0; i<countRDEs; i++) {
+    bool found = strcmp(rootDirEntries+(i*32), filename, 11);
+    if (found) {
+      uint32_t* correctEntry = (uint32_t*)(rootDirEntries+(i*32));
+      return correctEntry[7];
+    }
+  }
+  return 0;
 }
\ No newline at end of file
diff --git a/src/kernel/atapio.h b/src/kernel/atapio.h
index 0a1e982..6f5b618 100644
--- a/src/kernel/atapio.h
+++ b/src/kernel/atapio.h
@@ -7,6 +7,7 @@
 #include "types.h"
 #include "strings.h"
 #include "allocate.h"
+#include "global.h"
 
 void init_atapio();
 void read_sector(uint32_t address, uint8_t* buffer);
@@ -14,4 +15,6 @@ void read_sectors(uint32_t address, int count, uint8_t* buffer);
 uint16_t file_exists(char* filename);
 void load_file(char* filename, uint8_t* location);
 
+uint32_t file_size(char* filename);
+
 #endif
\ No newline at end of file