This fixes issue #327, is a very complex fix (unfortunately), details:

[redis.git] / src / zipmap.c

diff --git a/src/zipmap.c b/src/zipmap.c
index be780a828ca849652c2c5d2ea6e8d88f0761d111..65ed29c89ade473dc4fcb123de4780781cd2168c 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);
         }
     }


@@ -144,7 +147,7 @@ static unsigned char *zipmapLookupRaw(unsigned char *zm, unsigned char *key, uns
         /* Match or skip the key */
         l = zipmapDecodeLength(p);
         llen = zipmapEncodeLength(NULL,l);
         /* Match or skip the key */
         l = zipmapDecodeLength(p);
         llen = zipmapEncodeLength(NULL,l);

-        if (k == NULL && l == klen && !memcmp(p+llen,key,l)) {
+        if (key != NULL && k == NULL && l == klen && !memcmp(p+llen,key,l)) {

             /* Only return when the user doesn't care
              * for the total length of the zipmap. */
             if (totlen != NULL) {
             /* Only return when the user doesn't care
              * for the total length of the zipmap. */
             if (totlen != NULL) {


@@ -295,7 +298,7 @@ unsigned char *zipmapDel(unsigned char *zm, unsigned char *key, unsigned int kle
     return zm;
 }
 
     return zm;
 }
 

-/* Call it before to iterate trought elements via zipmapNext() */
+/* Call before iterating through elements via zipmapNext() */

 unsigned char *zipmapRewind(unsigned char *zm) {
     return zm+1;
 }
 unsigned char *zipmapRewind(unsigned char *zm) {
     return zm+1;
 }


@@ -360,6 +363,16 @@ unsigned int zipmapLen(unsigned char *zm) {
     return len;
 }
 
     return len;
 }
 

+/* Return the raw size in bytes of a zipmap, so that we can serialize
+ * the zipmap on disk (or everywhere is needed) just writing the returned
+ * amount of bytes of the C array starting at the zipmap pointer. */
+size_t zipmapBlobLen(unsigned char *zm) {
+    unsigned int totlen;
+    zipmapLookupRaw(zm,NULL,0,&totlen);
+    return totlen;
+}
+
+#ifdef ZIPMAP_TEST_MAIN

 void zipmapRepr(unsigned char *p) {
     unsigned int l;
 
 void zipmapRepr(unsigned char *p) {
     unsigned int l;
 


@@ -393,7 +406,6 @@ void zipmapRepr(unsigned char *p) {
     printf("\n");
 }
 
     printf("\n");
 }
 

-#ifdef ZIPMAP_TEST_MAIN

 int main(void) {
     unsigned char *zm;
 
 int main(void) {
     unsigned char *zm;
 


@@ -440,7 +452,7 @@ int main(void) {
                 vlen, vlen, value);
         }
     }
                 vlen, vlen, value);
         }
     }

-    printf("\nIterate trought elements:\n");
+    printf("\nIterate through elements:\n");

     {
         unsigned char *i = zipmapRewind(zm);
         unsigned char *key, *value;
     {
         unsigned char *i = zipmapRewind(zm);
         unsigned char *key, *value;