From d600f0d5b3c601026054189a6a65b681be50fa47 Mon Sep 17 00:00:00 2001
From: Ali Mohammad Pur <ali.mpfard@gmail.com>
Date: Sun, 12 Sep 2021 00:15:47 +0430
Subject: [PATCH] Kernel: Specialize Traits<(Nonnull)OwnPtr<T>> for KString

To make it behave like a string, since KString is always stored as a
(Nonnull)OwnPtr in the kernel.
---
 Kernel/KString.h | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/Kernel/KString.h b/Kernel/KString.h
index 0a06e5478b5..1a2af2f318d 100644
--- a/Kernel/KString.h
+++ b/Kernel/KString.h
@@ -63,4 +63,33 @@ struct Formatter<OwnPtr<Kernel::KString>> : Formatter<StringView> {
     }
 };
 
+template<>
+struct Traits<NonnullOwnPtr<Kernel::KString>> : public GenericTraits<NonnullOwnPtr<Kernel::KString>> {
+    using PeekType = Kernel::KString*;
+    using ConstPeekType = Kernel::KString const*;
+    static unsigned hash(NonnullOwnPtr<Kernel::KString> const& p) { return string_hash(p->characters(), p->length()); }
+    static bool equals(NonnullOwnPtr<Kernel::KString> const& a, NonnullOwnPtr<Kernel::KString> const& b) { return a->view() == b->view(); }
+};
+
+template<>
+struct Traits<OwnPtr<Kernel::KString>> : public GenericTraits<OwnPtr<Kernel::KString>> {
+    using PeekType = Kernel::KString*;
+    using ConstPeekType = Kernel::KString const*;
+    static unsigned hash(OwnPtr<Kernel::KString> const& p)
+    {
+        if (!p)
+            return ptr_hash(nullptr);
+        return string_hash(p->characters(), p->length());
+    }
+    static bool equals(OwnPtr<Kernel::KString> const& a, OwnPtr<Kernel::KString> const& b)
+    {
+        if (!a || !b)
+            return a.ptr() == b.ptr();
+        if (a == b)
+            return true;
+
+        return a->view() == b->view();
+    }
+};
+
 }