diff options
| author | bunnei <bunneidev@gmail.com> | 2021-05-16 09:54:15 +0200 |
|---|---|---|
| committer | bunnei <bunneidev@gmail.com> | 2021-05-21 06:41:51 +0200 |
| commit | 342170fcd364b584c381f53b405c89ae3ca5e3e8 (patch) | |
| tree | 8ea42686a67477539d627adc4332245e2682715a /src/common/tree.h | |
| parent | hle: kernel: Implement CloneCurrentObject and improve session management. (diff) | |
| download | yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar.gz yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar.bz2 yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar.lz yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar.xz yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.tar.zst yuzu-342170fcd364b584c381f53b405c89ae3ca5e3e8.zip | |
Diffstat (limited to 'src/common/tree.h')
| -rw-r--r-- | src/common/tree.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/common/tree.h b/src/common/tree.h index 9d2d0df4e..18faa4a48 100644 --- a/src/common/tree.h +++ b/src/common/tree.h @@ -43,6 +43,8 @@ * The maximum height of a red-black tree is 2lg (n+1). */ +#include "common/assert.h" + namespace Common { template <typename T> class RBHead { @@ -325,6 +327,10 @@ void RB_REMOVE_COLOR(RBHead<Node>* head, Node* parent, Node* elm) { while ((elm == nullptr || RB_IS_BLACK(elm)) && elm != head->Root() && parent != nullptr) { if (RB_LEFT(parent) == elm) { tmp = RB_RIGHT(parent); + if (!tmp) { + ASSERT_MSG(false, "tmp is invalid!"); + break; + } if (RB_IS_RED(tmp)) { RB_SET_BLACKRED(tmp, parent); RB_ROTATE_LEFT(head, parent, tmp); @@ -366,6 +372,11 @@ void RB_REMOVE_COLOR(RBHead<Node>* head, Node* parent, Node* elm) { tmp = RB_LEFT(parent); } + if (!tmp) { + ASSERT_MSG(false, "tmp is invalid!"); + break; + } + if ((RB_LEFT(tmp) == nullptr || RB_IS_BLACK(RB_LEFT(tmp))) && (RB_RIGHT(tmp) == nullptr || RB_IS_BLACK(RB_RIGHT(tmp)))) { RB_SET_COLOR(tmp, EntryColor::Red); |