Fixed bug in AOF rewrite not working because of integer overflow

[redis.git] / src / zipmap.c

diff --git a/src/zipmap.c b/src/zipmap.c
index 693db7b9300278e14f42177a1ed09e76af94729c..9f0fc71815819ea17605ebd904ee4e6ed8157ac0 100644 (file)
--- a/src/zipmap.c
+++ b/src/zipmap.c
@@ -80,6 +80,7 @@
 #include <string.h>
 #include <assert.h>
 #include "zmalloc.h"
 #include <string.h>
 #include <assert.h>
 #include "zmalloc.h"

+#include "endian.h"

 
 #define ZIPMAP_BIGLEN 254
 #define ZIPMAP_END 255
 
 #define ZIPMAP_BIGLEN 254
 #define ZIPMAP_END 255


@@ -108,6 +109,7 @@ static unsigned int zipmapDecodeLength(unsigned char *p) {
 
     if (len < ZIPMAP_BIGLEN) return len;
     memcpy(&len,p+1,sizeof(unsigned int));
 
     if (len < ZIPMAP_BIGLEN) return len;
     memcpy(&len,p+1,sizeof(unsigned int));

+    memrev32ifbe(&len);

     return len;
 }
 
     return len;
 }
 


@@ -123,6 +125,7 @@ static unsigned int zipmapEncodeLength(unsigned char *p, unsigned int len) {
         } else {
             p[0] = ZIPMAP_BIGLEN;
             memcpy(p+1,&len,sizeof(len));
         } else {
             p[0] = ZIPMAP_BIGLEN;
             memcpy(p+1,&len,sizeof(len));

+            memrev32ifbe(p+1);

             return 1+sizeof(len);
         }
     }
             return 1+sizeof(len);
         }
     }