From 7f3d89b2647489b69ca2c6c91a48b49fd0a519aa Mon Sep 17 00:00:00 2001
From: blueloveTH <blueloveTH@foxmail.com>
Date: Sun, 13 Aug 2023 02:21:28 +0800
Subject: [PATCH] more fix...

---
 include/pocketpy/namedict.h | 2 +-
 src/vm.cpp                  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/pocketpy/namedict.h b/include/pocketpy/namedict.h
index 51b65842..7871fc0f 100644
--- a/include/pocketpy/namedict.h
+++ b/include/pocketpy/namedict.h
@@ -32,7 +32,7 @@ struct NameDictImpl {
 #define HASH_PROBE(key, ok, i)          \
 ok = false;                             \
 i = _hash(key, _mask, _hash_seed);      \
-while(true) {       \
+for(int _j=0; _j<_capacity; _j++) {       \
     if(!_items[i].first.empty()){       \
         if(_items[i].first == (key)) { ok = true; break; }  \
     }else{                                                  \
diff --git a/src/vm.cpp b/src/vm.cpp
index 9537515c..51bd578b 100644
--- a/src/vm.cpp
+++ b/src/vm.cpp
@@ -1036,7 +1036,7 @@ void Dict::_probe(PyObject *key, bool &ok, int &i) const{
     i64 hash = vm->py_hash(key);
     i = hash & _mask;
     // std::cout << CAST(Str, vm->py_repr(key)) << " " << hash << " " << i << std::endl;
-    while(true) {
+    for(int j=0; j<_capacity; j++) {
         if(_items[i].first != nullptr){
             if(vm->py_equals(_items[i].first, key)) { ok = true; break; }
         }else{