[glacier] Add a vector class

lib/glacier/container/vector.h

@@ -0,0 +1,110 @@
+#pragma once
+
+#include <glacier/memory/move.h>
+#include <stdint.h>
+
+namespace glcr {
+
+template <typename T>
+class Vector {
+ public:
+  Vector() : data_(nullptr), size_(0), capacity_(0) {}
+
+  Vector(const Vector&) = delete;
+  // TODO: Implement Move
+  Vector(Vector&&) = delete;
+
+  ~Vector() { delete[] data_; }
+
+  // FIXME: Handle downsizing.
+  void Resize(uint64_t capacity);
+
+  // Setters.
+  void PushBack(const T& item);
+  void PushBack(T&& item);
+  template <typename... Args>
+  void EmplaceBack(Args... args);
+
+  // Accessors.
+  T& operator[](uint64_t index);
+  const T& operator[](uint64_t index) const;
+  T& at(uint64_t index);
+  const T& at(uint64_t index) const;
+
+  uint64_t size() const { return size_; }
+  uint64_t capacity() const { return capacity_; }
+
+ private:
+  T* data_;
+  uint64_t size_;
+  uint64_t capacity_;
+
+  void Expand();
+};
+
+template <typename T>
+void Vector<T>::Resize(uint64_t capacity) {
+  T* new_data = reinterpret_cast<T*>(new uint8_t[capacity * sizeof(T)]);
+  for (uint64_t i = 0; i < size_; i++) {
+    new_data[i] = glcr::Move(data_[i]);
+  }
+  delete[] data_;
+  data_ = new_data;
+  capacity_ = capacity;
+}
+
+template <typename T>
+void Vector<T>::PushBack(const T& item) {
+  if (size_ >= capacity_) {
+    Expand();
+  }
+
+  data_[size_++] = item;
+}
+
+template <typename T>
+void Vector<T>::PushBack(T&& item) {
+  if (size_ >= capacity_) {
+    Expand();
+  }
+
+  data_[size_++] = glcr::Move(item);
+}
+
+template <typename T>
+template <typename... Args>
+void Vector<T>::EmplaceBack(Args... args) {
+  if (size_ >= capacity_) {
+    Expand();
+  }
+
+  data_[size_++] = T(args...);
+}
+
+template <typename T>
+T& Vector<T>::operator[](uint64_t index) {
+  return data_[index];
+}
+
+template <typename T>
+const T& Vector<T>::operator[](uint64_t index) const {
+  return data_[index];
+}
+
+template <typename T>
+T& Vector<T>::at(uint64_t index) {
+  return data_[index];
+}
+
+template <typename T>
+const T& Vector<T>::at(uint64_t index) const {
+  return data_[index];
+}
+
+template <typename T>
+void Vector<T>::Expand() {
+  uint64_t new_capacity = capacity_ == 0 ? 1 : capacity_ * 2;
+  Resize(new_capacity);
+}
+
+}  // namespace glcr

zion/memory/kernel_heap.cpp

@@ -68,3 +68,4 @@ void* operator new(uint64_t size) { return GetKernelHeap().Allocate(size); }
 void* operator new[](uint64_t size) { return GetKernelHeap().Allocate(size); }
 
 void operator delete(void*, uint64_t) {}
+void operator delete[](void*) {}

zion/object/process.cpp

@@ -39,25 +39,19 @@ glcr::RefPtr<Thread> Process::CreateThread() {
 
 glcr::RefPtr<Thread> Process::GetThread(uint64_t tid) {
   MutexHolder lock(mutex_);
-  auto iter = threads_.begin();
-  while (iter != threads_.end()) {
-    if ((*iter)->tid() == tid) {
-      return *iter;
-    }
-    ++iter;
+  if (tid >= threads_.size()) {
+    panic("Bad thread access %u on process %u with %u threads.", tid, id_,
+          threads_.size());
   }
-  panic("Bad thread access.");
-  return nullptr;
+  return threads_[tid];
 }
 
 void Process::CheckState() {
   MutexHolder lock(mutex_);
-  auto iter = threads_.begin();
-  while (iter != threads_.end()) {
-    if ((*iter)->GetState() != Thread::FINISHED) {
+  for (uint64_t i = 0; i < threads_.size(); i++) {
+    if (threads_[i]->GetState() != Thread::FINISHED) {
       return;
     }
-    ++iter;
   }
   state_ = FINISHED;
 }

zion/object/process.h

@@ -1,6 +1,6 @@
 #pragma once
 
-#include <glacier/container/linked_list.h>
+#include <glacier/container/vector.h>
 #include <glacier/memory/ref_ptr.h>
 #include <stdint.h>
 
@@ -64,8 +64,7 @@ class Process : public KernelObject {
   State state_;
 
   uint64_t next_thread_id_ = 0;
-  uint64_t next_cap_id_ = 0x100;
 
-  glcr::LinkedList<glcr::RefPtr<Thread>> threads_;
+  glcr::Vector<glcr::RefPtr<Thread>> threads_;
   CapabilityTable caps_;
 };

zion/scheduler/process_manager.cpp

@@ -14,23 +14,15 @@ void ProcessManager::InsertProcess(const glcr::RefPtr<Process>& proc) {
 }
 
 Process& ProcessManager::FromId(uint64_t pid) {
-  auto iter = proc_list_.begin();
-  while (iter != proc_list_.end()) {
-    if ((*iter)->id() == pid) {
-      return **iter;
-    }
-    ++iter;
+  if (pid >= proc_list_.size()) {
+    panic("Bad proc access %u, have %u processes", pid, proc_list_.size());
   }
-
-  panic("Searching for invalid process id");
-  return *((Process*)0);
+  return *proc_list_[pid];
 }
 
 void ProcessManager::DumpProcessStates() {
   dbgln("Process States: %u", proc_list_.size());
-  auto iter = proc_list_.begin();
-  while (iter != proc_list_.end()) {
-    dbgln("%u: %u", (*iter)->id(), (*iter)->GetState());
-    ++iter;
+  for (uint64_t i = 0; i < proc_list_.size(); i++) {
+    dbgln("%u: %u", proc_list_[i]->id(), proc_list_[i]->GetState());
   }
 }

zion/scheduler/process_manager.h

@@ -1,6 +1,6 @@
 #pragma once
 
-#include <glacier/container/linked_list.h>
+#include <glacier/container/vector.h>
 #include <glacier/memory/ref_ptr.h>
 
 #include "object/process.h"
@@ -18,7 +18,7 @@ class ProcessManager {
 
  private:
   // TODO: This should be a hashmap.
-  glcr::LinkedList<glcr::RefPtr<Process>> proc_list_;
+  glcr::Vector<glcr::RefPtr<Process>> proc_list_;
 };
 
 extern ProcessManager* gProcMan;