helos1/memory/memory.hpp
2021-10-10 14:11:45 +08:00

61 lines
1.3 KiB
C++

#pragma once
#include <cstddef>
#include <limits>
#include <new>
#include "memory.h"
namespace helos {
class bad_alloc: public std::exception {
public:
bad_alloc() noexcept {}
bad_alloc(const bad_alloc &other) noexcept {}
bad_alloc &operator=(const bad_alloc &other) noexcept { return *this; }
virtual const char *what() const noexcept override { return "helos::bad_alloc"; }
};
// kAllocator is a class wrapper for kMalloc/Free satisfying the named requirement Allocator.
template<typename Type>
class kAllocator {
public:
typedef Type value_type;
kAllocator() = default;
template<typename Other>
constexpr kAllocator(const kAllocator<Other> &) {}
Type *allocate(std::size_t n) {
return kMalloc(n * sizeof(Type));
}
void deallocate(Type *p, std::size_t n) {
kFree(p);
}
};
template<class T, class U>
bool operator==(const kAllocator<T> &, const kAllocator<U> &) {
return true;
}
template<class T, class U>
bool operator!=(const kAllocator<T> &, const kAllocator<U> &) {
return false;
}
} // namespace helos
// globally overload the new and delete operators
// so keep this header at the top of every source file
//
// operators new and delete only call kMalloc/kFree, so C++ code
// must stay after paging setup
void *operator new(std::size_t size);
void operator delete(void *ptr) noexcept;