From 813c3f9bf82e4b9116b865758bfa32e6645f2e89 Mon Sep 17 00:00:00 2001
From: fslongjin <fslongjin@vip.qq.com>
Date: Mon, 25 Jul 2022 17:01:37 +0800
Subject: [PATCH] bitree: delete node

---
 .vscode/settings.json  |  4 ++-
 kernel/common/bitree.c | 74 ++++++++++++++++++++++++++++++++++++++++--
 kernel/common/bitree.h |  6 ++++
 3 files changed, 80 insertions(+), 4 deletions(-)

diff --git a/.vscode/settings.json b/.vscode/settings.json
index c84e9508..1930ef11 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -110,7 +110,9 @@
         "boot_info.h": "c",
         "pci.h": "c",
         "time.h": "c",
-        "ia64_msi.h": "c"
+        "ia64_msi.h": "c",
+        "errno.h": "c",
+        "bug.h": "c"
     },
     "C_Cpp.errorSquiggles": "Enabled",
     "esbonio.sphinx.confDir": ""
diff --git a/kernel/common/bitree.c b/kernel/common/bitree.c
index 1fe65183..67fca5c9 100644
--- a/kernel/common/bitree.c
+++ b/kernel/common/bitree.c
@@ -105,7 +105,8 @@ int bt_query(struct bt_root_t *root, void *value, uint64_t *ret_addr)
     struct bt_node_t *this_node = root->bt_node;
     struct bt_node_t tmp_node = {0};
     tmp_node.value = value;
-    while (this_node != NULL && !equal(root, &this_node, &tmp_node))
+
+    while (this_node != NULL && !equal(root, this_node, &tmp_node))
     {
         if (smaller(root, &tmp_node, this_node))
             this_node = this_node->left;
@@ -113,6 +114,73 @@ int bt_query(struct bt_root_t *root, void *value, uint64_t *ret_addr)
             this_node = this_node->right;
     }
 
-    *ret_addr = (uint64_t)ret_addr;
-    return 0;
+    if (equal(root, this_node, &tmp_node))
+    {
+        *ret_addr = (uint64_t)this_node;
+        return 0;
+    }
+    else
+    {
+        // 找不到则返回-1，且addr设为0
+        *ret_addr = NULL;
+        return -1;
+    }
+}
+
+static struct bt_node_t *bt_get_minimum(struct bt_node_t *this_node)
+{
+    while (this_node->left != NULL)
+        this_node = this_node->left;
+    return this_node;
+}
+
+/**
+ * @brief 删除结点
+ * 
+ * @param root 树根 
+ * @param value 待删除结点的值
+ * @return int 返回码
+ */
+int bt_delete(struct bt_root_t *root, void *value)
+{
+    uint64_t tmp_addr;
+    int retval;
+
+    // 寻找待删除结点
+    retval = bt_query(root, value, &tmp_addr);
+    if (retval != 0 || tmp_addr == NULL)
+        return retval;
+
+    struct bt_node_t *this_node = (struct bt_node_t *)tmp_addr;
+    struct bt_node_t *to_delete = NULL, *to_delete_son = NULL;
+    if (this_node->left == NULL || this_node->right == NULL)
+        to_delete = this_node;
+    else
+    {
+        to_delete = bt_get_minimum(this_node->right);
+        // 释放要被删除的值，并把下一个结点的值替换上来
+        root->release(this_node->value);
+        this_node->value = to_delete->value;
+    }
+
+    if (to_delete->left != NULL)
+        to_delete_son = to_delete->left;
+    else
+        to_delete_son = to_delete->right;
+
+    if (to_delete_son != NULL)
+        to_delete_son->parent = to_delete->parent;
+
+    if (to_delete->parent == NULL)
+        root->bt_node = to_delete_son;
+    else
+    {
+        if (to_delete->parent->left == to_delete)
+            to_delete->parent->left = to_delete_son;
+        else
+            to_delete->parent->right = to_delete_son;
+    }
+
+    // 释放最终要删除的结点的对象
+    kfree(to_delete);
 }
\ No newline at end of file
diff --git a/kernel/common/bitree.h b/kernel/common/bitree.h
index aa236258..3224635f 100644
--- a/kernel/common/bitree.h
+++ b/kernel/common/bitree.h
@@ -1,4 +1,5 @@
 #pragma once
+#include <common/glib.h>
 
 struct bt_node_t
 {
@@ -13,6 +14,11 @@ struct bt_root_t
 {
     struct bt_node_t *bt_node;
     int (*cmp)(struct bt_node_t *a, struct bt_node_t *b); // 比较函数   a>b 返回1， a==b返回0, a<b返回-1
+    /**
+     * @brief 释放结点的value的函数
+     * @param value 结点的值
+     */
+    int (*release)(void *value);
 };
 
 /**