Kernel: UsermodePointer bool cast

This allows UsermodePointer to be checked in if statements without an explicit cast/access
Weiww-Xu · Oct 29, 2021 · 2493e20 · 2493e20
1 parent d01e737
commit 2493e20
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Kernel/include/UserPointer.h b/Kernel/include/UserPointer.h
@@ -36,6 +36,8 @@ template <typename T> class UserPointer {
 
     ALWAYS_INLINE T* Pointer() { return m_ptr; }
 
+    ALWAYS_INLINE operator bool() const { return m_ptr != nullptr; }
+
 private:
     T* m_ptr;
 };
@@ -44,6 +46,9 @@ template <typename T> class UserBuffer {
 public:
     UserBuffer(uintptr_t ptr) : m_ptr(reinterpret_cast<T*>(ptr)) {}
 
+    ALWAYS_INLINE int GetValue(unsigned index, T& kernelValue) const { return UserMemcpy(&kernelValue, &m_ptr[index], sizeof(T)); }
+    ALWAYS_INLINE int StoreValue(unsigned index, const T& kernelValue) { return UserMemcpy(&m_ptr[index], &kernelValue, sizeof(T)); }
+
     ALWAYS_INLINE int Read(T* data, size_t offset, size_t count) const {
         if (!IsUsermodePointer(m_ptr, offset, count)) { // Don't allow kernel memory access
             return 0;