[glacier] Add a vector class

lib/glacier/container/linked_list.h

@@ -2,7 +2,7 @@
 
 #include <stdint.h>
 
-#include "debug/debug.h"
+namespace glcr {
 
 template <typename T>
 class LinkedList {
@@ -32,10 +32,6 @@ class LinkedList {
   }
 
   T PopFront() {
-    if (size_ == 0 || front_ == nullptr) {
-      panic("Popping from empty list");
-    }
-
     size_--;
 
     ListItem* old_front = front_;
@@ -45,33 +41,6 @@ class LinkedList {
     return ret;
   }
 
-  /*
-   * Returns the front item in the list and pushes the passed item to the back.
-   *
-   * Done in one function to avoid a memory alloc/dealloc during scheduling.
-   **/
-  T CycleFront(const T& new_item) {
-    if (size_ == 0 || front_ == nullptr) {
-      panic("Cycling empty list");
-    }
-
-    T ret = front_->item;
-    front_->item = new_item;
-    if (size_ == 1) {
-      return ret;
-    }
-
-    ListItem* old_front = front_;
-    ListItem* iter = front_;
-    front_ = front_->next;
-    while (iter->next != nullptr) {
-      iter = iter->next;
-    }
-    iter->next = old_front;
-    old_front->next = nullptr;
-    return ret;
-  }
-
   T PeekFront() const { return front_->item; }
 
   struct ListItem {
@@ -105,3 +74,5 @@ class LinkedList {
 
   ListItem* front_ = nullptr;
 };
+
+}  // namespace glcr

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/capability/capability_table.cpp

@@ -1,5 +1,7 @@
 #include "capability/capability_table.h"
 
+#include "debug/debug.h"
+
 CapabilityTable::CapabilityTable() {}
 
 uint64_t CapabilityTable::AddExistingCapability(

zion/capability/capability_table.h

@@ -1,9 +1,9 @@
 #pragma once
 
+#include <glacier/container/linked_list.h>
 #include <glacier/memory/ref_ptr.h>
 
 #include "capability/capability.h"
-#include "lib/linked_list.h"
 #include "lib/mutex.h"
 
 class CapabilityTable {
@@ -30,7 +30,7 @@ class CapabilityTable {
     uint64_t id;
     glcr::RefPtr<Capability> cap;
   };
-  LinkedList<CapEntry> capabilities_;
+  glcr::LinkedList<CapEntry> capabilities_;
 };
 
 template <typename T>

zion/lib/message_queue.cpp

@@ -1,5 +1,6 @@
 #include "lib/message_queue.h"
 
+#include "debug/debug.h"
 #include "scheduler/scheduler.h"
 
 z_err_t UnboundedMessageQueue::PushBack(uint64_t num_bytes, const void* bytes,

zion/lib/message_queue.h

@@ -1,13 +1,13 @@
 #pragma once
 
 #include <glacier/container/intrusive_list.h>
+#include <glacier/container/linked_list.h>
 #include <glacier/memory/ref_ptr.h>
 #include <glacier/memory/shared_ptr.h>
 #include <glacier/status/error.h>
 
 #include "capability/capability.h"
 #include "include/ztypes.h"
-#include "lib/linked_list.h"
 #include "lib/mutex.h"
 
 class MessageQueue {
@@ -51,10 +51,10 @@ class UnboundedMessageQueue : public MessageQueue {
     uint64_t num_bytes;
     uint8_t* bytes;
 
-    LinkedList<glcr::RefPtr<Capability>> caps;
+    glcr::LinkedList<glcr::RefPtr<Capability>> caps;
   };
 
-  LinkedList<glcr::SharedPtr<Message>> pending_messages_;
+  glcr::LinkedList<glcr::SharedPtr<Message>> pending_messages_;
 };
 
 class SingleMessageQueue : public MessageQueue {
@@ -79,5 +79,5 @@ class SingleMessageQueue : public MessageQueue {
   bool has_read_ = false;
   uint64_t num_bytes_;
   uint8_t* bytes_;
-  LinkedList<glcr::RefPtr<Capability>> caps_;
+  glcr::LinkedList<glcr::RefPtr<Capability>> caps_;
 };

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/address_space.cpp

@@ -1,5 +1,6 @@
 #include "object/address_space.h"
 
+#include "debug/debug.h"
 #include "memory/kernel_stack_manager.h"
 #include "memory/paging_util.h"
 #include "memory/physical_memory.h"

zion/object/address_space.h

@@ -1,5 +1,6 @@
 #pragma once
 
+#include <glacier/container/linked_list.h>
 #include <glacier/memory/ref_ptr.h>
 #include <stdint.h>
 
@@ -86,7 +87,7 @@ class AddressSpace : public KernelObject {
     uint64_t vaddr;
     glcr::RefPtr<MemoryObject> mem_obj;
   };
-  LinkedList<MemoryMapping> memory_mappings_;
+  glcr::LinkedList<MemoryMapping> memory_mappings_;
 
   MemoryMapping* GetMemoryMappingForAddr(uint64_t vaddr);
 };

zion/object/memory_object.h

@@ -1,6 +1,7 @@
 #pragma once
 
-#include "lib/linked_list.h"
+#include <glacier/container/linked_list.h>
+
 #include "object/kernel_object.h"
 
 class MemoryObject;
@@ -37,7 +38,7 @@ class MemoryObject : public KernelObject {
 
   virtual uint64_t PageNumberToPhysAddr(uint64_t page_num);
 
-  LinkedList<uint64_t> phys_page_list_;
+  glcr::LinkedList<uint64_t> phys_page_list_;
 };
 
 class FixedMemoryObject : public MemoryObject {

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;
+  if (tid >= threads_.size()) {
+    panic("Bad thread access %u on process %u with %u threads.", tid, id_,
+          threads_.size());
   }
-    ++iter;
-  }
-  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,11 +1,11 @@
 #pragma once
 
+#include <glacier/container/vector.h>
 #include <glacier/memory/ref_ptr.h>
 #include <stdint.h>
 
 #include "capability/capability.h"
 #include "capability/capability_table.h"
-#include "lib/linked_list.h"
 #include "lib/mutex.h"
 #include "object/address_space.h"
 #include "object/channel.h"
@@ -64,8 +64,7 @@ class Process : public KernelObject {
   State state_;
 
   uint64_t next_thread_id_ = 0;
-  uint64_t next_cap_id_ = 0x100;
 
-  LinkedList<glcr::RefPtr<Thread>> threads_;
+  glcr::Vector<glcr::RefPtr<Thread>> threads_;
   CapabilityTable caps_;
 };

zion/scheduler/process_manager.cpp

@@ -1,5 +1,7 @@
 #include "scheduler/process_manager.h"
 
+#include "debug/debug.h"
+
 ProcessManager* gProcMan = nullptr;
 
 void ProcessManager::Init() {
@@ -12,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;
+  if (pid >= proc_list_.size()) {
+    panic("Bad proc access %u, have %u processes", pid, proc_list_.size());
   }
-    ++iter;
-  }
-
-  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,8 +1,8 @@
 #pragma once
 
+#include <glacier/container/vector.h>
 #include <glacier/memory/ref_ptr.h>
 
-#include "lib/linked_list.h"
 #include "object/process.h"
 
 class ProcessManager {
@@ -18,7 +18,7 @@ class ProcessManager {
 
  private:
   // TODO: This should be a hashmap.
-  LinkedList<glcr::RefPtr<Process>> proc_list_;
+  glcr::Vector<glcr::RefPtr<Process>> proc_list_;
 };
 
 extern ProcessManager* gProcMan;

zion/scheduler/scheduler.cpp

@@ -2,7 +2,6 @@
 
 #include "common/gdt.h"
 #include "debug/debug.h"
-#include "lib/linked_list.h"
 #include "scheduler/process_manager.h"
 
 namespace {

zion/syscall/process.cpp

@@ -1,6 +1,7 @@
 #include "syscall/process.h"
 
 #include "capability/capability.h"
+#include "debug/debug.h"
 #include "scheduler/process_manager.h"
 #include "scheduler/scheduler.h"
 

zion/syscall/syscall.cpp

@@ -3,6 +3,7 @@
 #include <stdint.h>
 
 #include "common/msr.h"
+#include "debug/debug.h"
 #include "scheduler/scheduler.h"
 #include "syscall/address_space.h"
 #include "syscall/capability.h"

zion/syscall/thread.cpp

@@ -1,6 +1,7 @@
 #include "syscall/thread.h"
 
 #include "capability/capability.h"
+#include "debug/debug.h"
 #include "scheduler/scheduler.h"
 
 glcr::ErrorCode ThreadCreate(ZThreadCreateReq* req) {
@@ -29,6 +30,7 @@ glcr::ErrorCode ThreadExit(ZThreadExitReq*) {
   auto curr_thread = gScheduler->CurrentThread();
   curr_thread->Exit();
   panic("Returned from thread exit");
+  UNREACHABLE
 }
 
 glcr::ErrorCode ThreadWait(ZThreadWaitReq* req) {