Fix rdb.c dictionary iterator release in 2 more places.

b85aae78 · antirez · cd87b3c7 · b85aae78
隐藏空白更改
内联并排

Showing with 8 addition and 2 deletion

src/rdb.c src/rdb.c +8 -2

未找到文件。
--- a/src/rdb.c
+++ b/src/rdb.c
@@ -772,7 +772,10 @@ ssize_t rdbSaveObject(rio *rdb, robj *o) {
            dictIterator *di = dictGetIterator(set);
            dictEntry *de;

-            if ((n = rdbSaveLen(rdb,dictSize(set))) == -1) return -1;
+            if ((n = rdbSaveLen(rdb,dictSize(set))) == -1) {
+                dictReleaseIterator(di);
+                return -1;
+            }
            nwritten += n;

            while((de = dictNext(di)) != NULL) {
@@ -842,7 +845,10 @@ ssize_t rdbSaveObject(rio *rdb, robj *o) {
            dictIterator *di = dictGetIterator(o->ptr);
            dictEntry *de;

-            if ((n = rdbSaveLen(rdb,dictSize((dict*)o->ptr))) == -1) return -1;
+            if ((n = rdbSaveLen(rdb,dictSize((dict*)o->ptr))) == -1) {
+                dictReleaseIterator(di);
+                return -1;
+            }
            nwritten += n;

            while((de = dictNext(di)) != NULL) {