internal_memscan: add extra check

maxthyme · Feb 4, 2017 · a8f4833 · a8f4833
1 parent f65a981
commit a8f4833
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/library/LuaApi.cpp b/library/LuaApi.cpp
@@ -2557,6 +2557,9 @@ static int internal_memscan(lua_State *L)
     for (int i = 0; i <= hcount; i++)
     {
         uint8_t *p = haystack + i*hstep;
+        if (p + nsize > haystack + (hcount * hstep)) {
+            break;
+        }
         if (memcmp(p, needle, nsize) == 0) {
             lua_pushinteger(L, i);
             lua_pushinteger(L, (lua_Integer)p);