ziplist.c 27.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/* Memory layout of a ziplist, containing "foo", "bar", "quux":
 * <zlbytes><zllen><len>"foo"<len>"bar"<len>"quux"
 *
 * <zlbytes> is an unsigned integer to hold the number of bytes that
 * the ziplist occupies. This is stored to not have to traverse the ziplist
 * to know the new length when pushing.
 *
 * <zllen> is the number of items in the ziplist. When this value is
 * greater than 254, we need to traverse the entire list to know
 * how many items it holds.
 *
 * <len> is the number of bytes occupied by a single entry. When this
 * number is greater than 253, the length will occupy 5 bytes, where
 * the extra bytes contain an unsigned integer to hold the length.
 */

#include <stdio.h>
18
#include <stdlib.h>
19 20
#include <string.h>
#include <assert.h>
21
#include <limits.h>
22 23 24 25
#include "zmalloc.h"
#include "sds.h"
#include "ziplist.h"

26 27 28 29 30
/* Important note: the ZIP_END value is used to depict the end of the
 * ziplist structure. When a pointer contains an entry, the first couple
 * of bytes contain the encoded length of the previous entry. This length
 * is encoded as ZIP_ENC_RAW length, so the first two bits will contain 00
 * and the byte will therefore never have a value of 255. */
31
#define ZIP_END 255
32
#define ZIP_BIGLEN 254
33 34 35 36 37 38 39 40 41 42 43 44 45 46

/* Entry encoding */
#define ZIP_ENC_RAW     0
#define ZIP_ENC_SHORT   1
#define ZIP_ENC_INT     2
#define ZIP_ENC_LLONG   3
#define ZIP_ENCODING(p) ((p)[0] >> 6)

/* Length encoding for raw entries */
#define ZIP_LEN_INLINE  0
#define ZIP_LEN_UINT16  1
#define ZIP_LEN_UINT32  2

/* Utility macros */
47
#define ZIPLIST_BYTES(zl) (*((unsigned int*)(zl)))
48
#define ZIPLIST_TAIL_OFFSET(zl) (*((unsigned int*)((zl)+sizeof(unsigned int))))
49 50
#define ZIPLIST_LENGTH(zl) (*((zl)+2*sizeof(unsigned int)))
#define ZIPLIST_HEADER_SIZE (2*sizeof(unsigned int)+1)
51 52 53
#define ZIPLIST_ENTRY_HEAD(zl) ((zl)+ZIPLIST_HEADER_SIZE)
#define ZIPLIST_ENTRY_TAIL(zl) ((zl)+ZIPLIST_TAIL_OFFSET(zl))
#define ZIPLIST_ENTRY_END(zl) ((zl)+ZIPLIST_BYTES(zl)-1)
54
#define ZIPLIST_INCR_LENGTH(zl,incr) { \
55
    if (ZIPLIST_LENGTH(zl) < ZIP_BIGLEN) ZIPLIST_LENGTH(zl)+=incr; }
56

57 58 59 60 61
typedef struct zlentry {
    unsigned int prevrawlensize, prevrawlen;
    unsigned int lensize, len;
    unsigned int headersize;
    unsigned char encoding;
62
    unsigned char *p;
63 64
} zlentry;

65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
/* Return bytes needed to store integer encoded by 'encoding' */
static unsigned int zipEncodingSize(char encoding) {
    if (encoding == ZIP_ENC_SHORT) {
        return sizeof(short int);
    } else if (encoding == ZIP_ENC_INT) {
        return sizeof(int);
    } else if (encoding == ZIP_ENC_LLONG) {
        return sizeof(long long);
    }
    assert(NULL);
}

/* Decode the encoded length pointed by 'p'. If a pointer to 'lensize' is
 * provided, it is set to the number of bytes required to encode the length. */
static unsigned int zipDecodeLength(unsigned char *p, unsigned int *lensize) {
    unsigned char encoding = ZIP_ENCODING(p), lenenc;
    unsigned int len;

    if (encoding == ZIP_ENC_RAW) {
        lenenc = (p[0] >> 4) & 0x3;
        if (lenenc == ZIP_LEN_INLINE) {
            len = p[0] & 0xf;
            if (lensize) *lensize = 1;
        } else if (lenenc == ZIP_LEN_UINT16) {
            len = p[1] | (p[2] << 8);
            if (lensize) *lensize = 3;
        } else {
            len = p[1] | (p[2] << 8) | (p[3] << 16) | (p[4] << 24);
            if (lensize) *lensize = 5;
        }
    } else {
        len = zipEncodingSize(encoding);
        if (lensize) *lensize = 1;
    }
    return len;
}

/* Encode the length 'l' writing it in 'p'. If p is NULL it just returns
 * the amount of bytes required to encode such a length. */
static unsigned int zipEncodeLength(unsigned char *p, char encoding, unsigned int rawlen) {
    unsigned char len = 1, lenenc, buf[5];
    if (encoding == ZIP_ENC_RAW) {
        if (rawlen <= 0xf) {
            if (!p) return len;
            lenenc = ZIP_LEN_INLINE;
            buf[0] = rawlen;
        } else if (rawlen <= 0xffff) {
            len += 2;
            if (!p) return len;
            lenenc = ZIP_LEN_UINT16;
            buf[1] = (rawlen     ) & 0xff;
            buf[2] = (rawlen >> 8) & 0xff;
        } else {
            len += 4;
            if (!p) return len;
            lenenc = ZIP_LEN_UINT32;
            buf[1] = (rawlen      ) & 0xff;
            buf[2] = (rawlen >>  8) & 0xff;
            buf[3] = (rawlen >> 16) & 0xff;
            buf[4] = (rawlen >> 24) & 0xff;
        }
        buf[0] = (lenenc << 4) | (buf[0] & 0xf);
    }
    if (!p) return len;

    /* Apparently we need to store the length in 'p' */
    buf[0] = (encoding << 6) | (buf[0] & 0x3f);
    memcpy(p,buf,len);
    return len;
}

136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164
/* Decode the length of the previous element stored at "p". */
static unsigned int zipPrevDecodeLength(unsigned char *p, unsigned int *lensize) {
    unsigned int len = *p;
    if (len < ZIP_BIGLEN) {
        if (lensize) *lensize = 1;
    } else {
        if (lensize) *lensize = 1+sizeof(len);
        memcpy(&len,p+1,sizeof(len));
    }
    return len;
}

/* Encode the length of the previous entry and write it to "p". Return the
 * number of bytes needed to encode this length if "p" is NULL. */
static unsigned int zipPrevEncodeLength(unsigned char *p, unsigned int len) {
    if (p == NULL) {
        return (len < ZIP_BIGLEN) ? 1 : sizeof(len)+1;
    } else {
        if (len < ZIP_BIGLEN) {
            p[0] = len;
            return 1;
        } else {
            p[0] = ZIP_BIGLEN;
            memcpy(p+1,&len,sizeof(len));
            return 1+sizeof(len);
        }
    }
}

165 166 167 168
/* Return the difference in number of bytes needed to store the new length
 * "len" on the entry pointed to by "p". */
static int zipPrevLenByteDiff(unsigned char *p, unsigned int len) {
    unsigned int prevlensize;
169 170
    zipPrevDecodeLength(p,&prevlensize);
    return zipPrevEncodeLength(NULL,len)-prevlensize;
171 172
}

173 174 175
/* Check if string pointed to by 'entry' can be encoded as an integer.
 * Stores the integer value in 'v' and its encoding in 'encoding'.
 * Warning: this function requires a NULL-terminated string! */
P
Pieter Noordhuis 已提交
176
static int zipTryEncoding(char *entry, long long *v, char *encoding) {
177 178 179 180
    long long value;
    char *eptr;

    if (entry[0] == '-' || (entry[0] >= '0' && entry[0] <= '9')) {
P
Pieter Noordhuis 已提交
181
        value = strtoll(entry,&eptr,10);
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234
        if (eptr[0] != '\0') return 0;
        if (value >= SHRT_MIN && value <= SHRT_MAX) {
            *encoding = ZIP_ENC_SHORT;
        } else if (value >= INT_MIN && value <= INT_MAX) {
            *encoding = ZIP_ENC_INT;
        } else {
            *encoding = ZIP_ENC_LLONG;
        }
        *v = value;
        return 1;
    }
    return 0;
}

/* Store integer 'value' at 'p', encoded as 'encoding' */
static void zipSaveInteger(unsigned char *p, long long value, char encoding) {
    short int s;
    int i;
    long long l;
    if (encoding == ZIP_ENC_SHORT) {
        s = value;
        memcpy(p,&s,sizeof(s));
    } else if (encoding == ZIP_ENC_INT) {
        i = value;
        memcpy(p,&i,sizeof(i));
    } else if (encoding == ZIP_ENC_LLONG) {
        l = value;
        memcpy(p,&l,sizeof(l));
    } else {
        assert(NULL);
    }
}

/* Read integer encoded as 'encoding' from 'p' */
static long long zipLoadInteger(unsigned char *p, char encoding) {
    short int s;
    int i;
    long long l, ret;
    if (encoding == ZIP_ENC_SHORT) {
        memcpy(&s,p,sizeof(s));
        ret = s;
    } else if (encoding == ZIP_ENC_INT) {
        memcpy(&i,p,sizeof(i));
        ret = i;
    } else if (encoding == ZIP_ENC_LLONG) {
        memcpy(&l,p,sizeof(l));
        ret = l;
    } else {
        assert(NULL);
    }
    return ret;
}

235 236 237
/* Return a struct with all information about an entry. */
static zlentry zipEntry(unsigned char *p) {
    zlentry e;
238
    e.prevrawlen = zipPrevDecodeLength(p,&e.prevrawlensize);
239 240 241
    e.len = zipDecodeLength(p+e.prevrawlensize,&e.lensize);
    e.headersize = e.prevrawlensize+e.lensize;
    e.encoding = ZIP_ENCODING(p+e.prevrawlensize);
242
    e.p = p;
243 244 245
    return e;
}

246
/* Return the total number of bytes used by the entry at "p". */
247
static unsigned int zipRawEntryLength(unsigned char *p) {
248 249
    zlentry e = zipEntry(p);
    return e.headersize + e.len;
250 251
}

252 253 254 255 256
/* Create a new empty ziplist. */
unsigned char *ziplistNew(void) {
    unsigned int bytes = ZIPLIST_HEADER_SIZE+1;
    unsigned char *zl = zmalloc(bytes);
    ZIPLIST_BYTES(zl) = bytes;
257
    ZIPLIST_TAIL_OFFSET(zl) = ZIPLIST_HEADER_SIZE;
258 259 260 261 262
    ZIPLIST_LENGTH(zl) = 0;
    zl[bytes-1] = ZIP_END;
    return zl;
}

263
/* Resize the ziplist. */
264
static unsigned char *ziplistResize(unsigned char *zl, unsigned int len) {
265
    zl = zrealloc(zl,len);
266 267 268 269 270
    ZIPLIST_BYTES(zl) = len;
    zl[len-1] = ZIP_END;
    return zl;
}

271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
/* Delete "num" entries, starting at "p". Returns pointer to the ziplist. */
static unsigned char *__ziplistDelete(unsigned char *zl, unsigned char *p, int num) {
    unsigned int i, totlen, deleted = 0;
    int nextdiff = 0;
    zlentry first = zipEntry(p);
    for (i = 0; p[0] != ZIP_END && i < num; i++) {
        p += zipRawEntryLength(p);
        deleted++;
    }

    totlen = p-first.p;
    if (totlen > 0) {
        if (p[0] != ZIP_END) {
            /* Tricky: storing the prevlen in this entry might reduce or
             * increase the number of bytes needed, compared to the current
             * prevlen. Note that we can always store this length because
             * it was previously stored by an entry that is being deleted. */
            nextdiff = zipPrevLenByteDiff(p,first.prevrawlen);
289
            zipPrevEncodeLength(p-nextdiff,first.prevrawlen);
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305

            /* Update offset for tail */
            ZIPLIST_TAIL_OFFSET(zl) -= totlen+nextdiff;

            /* Move tail to the front of the ziplist */
            memmove(first.p,p-nextdiff,ZIPLIST_BYTES(zl)-(p-zl)-1+nextdiff);
        } else {
            /* The entire tail was deleted. No need to move memory. */
            ZIPLIST_TAIL_OFFSET(zl) = (first.p-zl)-first.prevrawlen;
        }

        /* Resize and update length */
        zl = ziplistResize(zl, ZIPLIST_BYTES(zl)-totlen+nextdiff);
        ZIPLIST_INCR_LENGTH(zl,-deleted);
    }
    return zl;
306 307
}

308
/* Insert item at "p". */
P
Pieter Noordhuis 已提交
309
static unsigned char *__ziplistInsert(unsigned char *zl, unsigned char *p, char *s, unsigned int slen) {
310 311 312
    unsigned int curlen = ZIPLIST_BYTES(zl), reqlen, prevlen = 0;
    unsigned int offset, nextdiff = 0;
    unsigned char *tail;
313 314
    char encoding = ZIP_ENC_RAW;
    long long value;
315
    zlentry entry;
316

317 318 319 320
    /* Find out prevlen for the entry that is inserted. */
    if (p[0] != ZIP_END) {
        entry = zipEntry(p);
        prevlen = entry.prevrawlen;
321
    } else {
322
        tail = ZIPLIST_ENTRY_TAIL(zl);
323 324 325
        if (tail[0] != ZIP_END) {
            prevlen = zipRawEntryLength(tail);
        }
326 327
    }

328
    /* See if the entry can be encoded */
329
    if (zipTryEncoding(s,&value,&encoding)) {
330 331
        reqlen = zipEncodingSize(encoding);
    } else {
332
        reqlen = slen;
333
    }
334 335 336

    /* We need space for both the length of the previous entry and
     * the length of the payload. */
337
    reqlen += zipPrevEncodeLength(NULL,prevlen);
338 339 340 341 342
    reqlen += zipEncodeLength(NULL,encoding,slen);

    /* When the insert position is not equal to the tail, we need to
     * make sure that the next entry can hold this entry's length in
     * its prevlen field. */
P
Pieter Noordhuis 已提交
343
    nextdiff = (p[0] != ZIP_END) ? zipPrevLenByteDiff(p,reqlen) : 0;
344 345 346 347 348 349 350 351 352 353 354

    /* Store offset because a realloc may change the address of zl. */
    offset = p-zl;
    zl = ziplistResize(zl,curlen+reqlen+nextdiff);
    p = zl+offset;

    /* Apply memory move when necessary and update tail offset. */
    if (p[0] != ZIP_END) {
        /* Subtract one because of the ZIP_END bytes */
        memmove(p+reqlen,p-nextdiff,curlen-offset-1+nextdiff);
        /* Encode this entry's raw length in the next entry. */
355
        zipPrevEncodeLength(p+reqlen,reqlen);
356 357
        /* Update offset for tail */
        ZIPLIST_TAIL_OFFSET(zl) += reqlen+nextdiff;
358
    } else {
359 360
        /* This element will be the new tail. */
        ZIPLIST_TAIL_OFFSET(zl) = p-zl;
361 362
    }

363
    /* Write the entry */
364
    p += zipPrevEncodeLength(p,prevlen);
365
    p += zipEncodeLength(p,encoding,slen);
366 367 368
    if (encoding != ZIP_ENC_RAW) {
        zipSaveInteger(p,value,encoding);
    } else {
369
        memcpy(p,s,slen);
370
    }
371
    ZIPLIST_INCR_LENGTH(zl,1);
372 373 374
    return zl;
}

P
Pieter Noordhuis 已提交
375
unsigned char *ziplistPush(unsigned char *zl, char *s, unsigned int slen, int where) {
376
    unsigned char *p;
377
    p = (where == ZIPLIST_HEAD) ? ZIPLIST_ENTRY_HEAD(zl) : ZIPLIST_ENTRY_END(zl);
378 379 380
    return __ziplistInsert(zl,p,s,slen);
}

381
unsigned char *ziplistPop(unsigned char *zl, sds *target, int where) {
382
    zlentry entry;
383
    unsigned char *p;
384 385
    long long value;
    if (target) *target = NULL;
386 387

    /* Get pointer to element to remove */
388
    p = (where == ZIPLIST_HEAD) ? ZIPLIST_ENTRY_HEAD(zl) : ZIPLIST_ENTRY_TAIL(zl);
389
    if (*p == ZIP_END) return zl;
390

391
    entry = zipEntry(p);
392
    if (target) {
393 394
        if (entry.encoding == ZIP_ENC_RAW) {
            *target = sdsnewlen(p+entry.headersize,entry.len);
395
        } else {
396
            value = zipLoadInteger(p+entry.headersize,entry.encoding);
397 398 399
            *target = sdscatprintf(sdsempty(), "%lld", value);
        }
    }
400

401
    zl = __ziplistDelete(zl,p,1);
402 403 404
    return zl;
}

405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425
/* Returns an offset to use for iterating with ziplistNext. When the given
 * index is negative, the list is traversed back to front. When the list
 * doesn't contain an element at the provided index, NULL is returned. */
unsigned char *ziplistIndex(unsigned char *zl, int index) {
    unsigned char *p;
    zlentry entry;
    if (index < 0) {
        index = (-index)-1;
        p = ZIPLIST_ENTRY_TAIL(zl);
        if (p[0] != ZIP_END) {
            entry = zipEntry(p);
            while (entry.prevrawlen > 0 && index--) {
                p -= entry.prevrawlen;
                entry = zipEntry(p);
            }
        }
    } else {
        p = ZIPLIST_ENTRY_HEAD(zl);
        while (p[0] != ZIP_END && index--) {
            p += zipRawEntryLength(p);
        }
426
    }
P
Pieter Noordhuis 已提交
427
    return (p[0] == ZIP_END || index > 0) ? NULL : p;
428 429
}

430
/* Return pointer to next entry in ziplist. */
431 432
unsigned char *ziplistNext(unsigned char *zl, unsigned char *p) {
    ((void) zl);
P
Pieter Noordhuis 已提交
433
    return (p[0] == ZIP_END) ? NULL : p+zipRawEntryLength(p);
434 435
}

436
/* Return pointer to previous entry in ziplist. */
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451
unsigned char *ziplistPrev(unsigned char *zl, unsigned char *p) {
    zlentry entry;

    /* Iterating backwards from ZIP_END should return the tail. When "p" is
     * equal to the first element of the list, we're already at the head,
     * and should return NULL. */
    if (p[0] == ZIP_END) {
        p = ZIPLIST_ENTRY_TAIL(zl);
        return (p[0] == ZIP_END) ? NULL : p;
    } else if (p == ZIPLIST_ENTRY_HEAD(zl)) {
        return NULL;
    } else {
        entry = zipEntry(p);
        return p-entry.prevrawlen;
    }
452 453
}

454 455 456 457
/* Get entry pointer to by 'p' and store in either 'e' or 'v' depending
 * on the encoding of the entry. 'e' is always set to NULL to be able
 * to find out whether the string pointer or the integer value was set.
 * Return 0 if 'p' points to the end of the zipmap, 1 otherwise. */
P
Pieter Noordhuis 已提交
458
unsigned int ziplistGet(unsigned char *p, char **sstr, unsigned int *slen, long long *sval) {
459
    zlentry entry;
460
    if (p == NULL || p[0] == ZIP_END) return 0;
461
    if (sstr) *sstr = NULL;
462

463 464
    entry = zipEntry(p);
    if (entry.encoding == ZIP_ENC_RAW) {
465 466
        if (sstr) {
            *slen = entry.len;
P
Pieter Noordhuis 已提交
467
            *sstr = (char*)p+entry.headersize;
468 469
        }
    } else {
470 471
        if (sval) {
            *sval = zipLoadInteger(p+entry.headersize,entry.encoding);
472
        }
473
    }
474
    return 1;
475 476
}

477 478 479
/* Insert an entry at "p". */
unsigned char *ziplistInsert(unsigned char *zl, unsigned char *p, char *s, unsigned int slen) {
    return __ziplistInsert(zl,p,s,slen);
480 481
}

482 483 484
/* Delete a single entry from the ziplist, pointed to by *p.
 * Also update *p in place, to be able to iterate over the
 * ziplist, while deleting entries. */
485
unsigned char *ziplistDelete(unsigned char *zl, unsigned char **p) {
486 487
    unsigned int offset = *p-zl;
    zl = __ziplistDelete(zl,*p,1);
488

489
    /* Store pointer to current element in p, because ziplistDelete will
490 491 492
     * do a realloc which might result in a different "zl"-pointer.
     * When the delete direction is back to front, we might delete the last
     * entry and end up with "p" pointing to ZIP_END, so check this. */
493
    *p = zl+offset;
494 495 496
    return zl;
}

497 498 499 500 501 502
/* Delete a range of entries from the ziplist. */
unsigned char *ziplistDeleteRange(unsigned char *zl, unsigned int index, unsigned int num) {
    unsigned char *p = ziplistIndex(zl,index);
    return (p == NULL) ? zl : __ziplistDelete(zl,p,num);
}

503
/* Compare entry pointer to by 'p' with 'entry'. Return 1 if equal. */
P
Pieter Noordhuis 已提交
504
unsigned int ziplistCompare(unsigned char *p, char *sstr, unsigned int slen) {
505
    zlentry entry;
P
Pieter Noordhuis 已提交
506
    char sencoding;
507
    long long val, sval;
P
Pieter Noordhuis 已提交
508
    if (p[0] == ZIP_END) return 0;
509

510 511
    entry = zipEntry(p);
    if (entry.encoding == ZIP_ENC_RAW) {
512
        /* Raw compare */
513
        if (entry.len == slen) {
514
            return memcmp(p+entry.headersize,sstr,slen) == 0;
515 516 517
        } else {
            return 0;
        }
518
    } else {
519
        /* Try to compare encoded values */
520
        if (zipTryEncoding(sstr,&sval,&sencoding)) {
521 522 523 524
            if (entry.encoding == sencoding) {
                val = zipLoadInteger(p+entry.headersize,entry.encoding);
                return val == sval;
            }
525
        }
526
    }
527
    return 0;
528 529
}

530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547
/* Return length of ziplist. */
unsigned int ziplistLen(unsigned char *zl) {
    unsigned int len = 0;
    if (ZIPLIST_LENGTH(zl) < ZIP_BIGLEN) {
        len = ZIPLIST_LENGTH(zl);
    } else {
        unsigned char *p = zl+ZIPLIST_HEADER_SIZE;
        while (*p != ZIP_END) {
            p += zipRawEntryLength(p);
            len++;
        }

        /* Re-store length if small enough */
        if (len < ZIP_BIGLEN) ZIPLIST_LENGTH(zl) = len;
    }
    return len;
}

548 549 550 551 552
/* Return size in bytes of ziplist. */
unsigned int ziplistSize(unsigned char *zl) {
    return ZIPLIST_BYTES(zl);
}

553
void ziplistRepr(unsigned char *zl) {
554 555
    unsigned char *p;
    zlentry entry;
556

557
    printf("{total bytes %d} {length %u}\n",ZIPLIST_BYTES(zl), ZIPLIST_LENGTH(zl));
558
    p = ZIPLIST_ENTRY_HEAD(zl);
559
    while(*p != ZIP_END) {
560 561 562 563 564
        entry = zipEntry(p);
        printf("{offset %ld, header %u, payload %u} ",p-zl,entry.headersize,entry.len);
        p += entry.headersize;
        if (entry.encoding == ZIP_ENC_RAW) {
            fwrite(p,entry.len,1,stdout);
565
        } else {
566
            printf("%lld", zipLoadInteger(p,entry.encoding));
567
        }
568
        printf("\n");
569
        p += entry.len;
570 571 572 573 574 575
    }
    printf("{end}\n\n");
}

#ifdef ZIPLIST_TEST_MAIN

576 577
unsigned char *createList() {
    unsigned char *zl = ziplistNew();
P
Pieter Noordhuis 已提交
578 579 580 581
    zl = ziplistPush(zl, "foo", 3, ZIPLIST_TAIL);
    zl = ziplistPush(zl, "quux", 4, ZIPLIST_TAIL);
    zl = ziplistPush(zl, "hello", 5, ZIPLIST_HEAD);
    zl = ziplistPush(zl, "1024", 4, ZIPLIST_TAIL);
582 583 584
    return zl;
}

585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
unsigned char *createIntList() {
    unsigned char *zl = ziplistNew();
    char buf[32];

    sprintf(buf, "100");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_TAIL);
    sprintf(buf, "128000");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_TAIL);
    sprintf(buf, "-100");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_HEAD);
    sprintf(buf, "4294967296");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_HEAD);
    sprintf(buf, "non integer");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_TAIL);
    sprintf(buf, "much much longer non integer");
    zl = ziplistPush(zl, buf, strlen(buf), ZIPLIST_TAIL);
    return zl;
}

604
int main(int argc, char **argv) {
P
Pieter Noordhuis 已提交
605 606
    unsigned char *zl, *p;
    char *entry;
607
    unsigned int elen;
608
    long long value;
609 610
    sds s;

611 612 613
    zl = createIntList();
    ziplistRepr(zl);

614
    zl = createList();
615 616 617 618 619 620 621 622 623 624
    ziplistRepr(zl);

    zl = ziplistPop(zl, &s, ZIPLIST_TAIL);
    printf("Pop tail: %s (length %ld)\n", s, sdslen(s));
    ziplistRepr(zl);

    zl = ziplistPop(zl, &s, ZIPLIST_HEAD);
    printf("Pop head: %s (length %ld)\n", s, sdslen(s));
    ziplistRepr(zl);

625 626 627 628 629 630 631 632
    zl = ziplistPop(zl, &s, ZIPLIST_TAIL);
    printf("Pop tail: %s (length %ld)\n", s, sdslen(s));
    ziplistRepr(zl);

    zl = ziplistPop(zl, &s, ZIPLIST_TAIL);
    printf("Pop tail: %s (length %ld)\n", s, sdslen(s));
    ziplistRepr(zl);

633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
    printf("Get element at index 3:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 3);
        if (!ziplistGet(p, &entry, &elen, &value)) {
            printf("ERROR: Could not access index 3\n");
            return 1;
        }
        if (entry) {
            fwrite(entry,elen,1,stdout);
            printf("\n");
        } else {
            printf("%lld\n", value);
        }
        printf("\n");
    }

    printf("Get element at index 4 (out of range):\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 4);
        if (p == NULL) {
            printf("No entry\n");
        } else {
            printf("ERROR: Out of range index should return NULL, returned offset: %ld\n", p-zl);
            return 1;
        }
        printf("\n");
    }

    printf("Get element at index -1 (last element):\n");
    {
        zl = createList();
        p = ziplistIndex(zl, -1);
        if (!ziplistGet(p, &entry, &elen, &value)) {
            printf("ERROR: Could not access index -1\n");
            return 1;
        }
        if (entry) {
            fwrite(entry,elen,1,stdout);
            printf("\n");
        } else {
            printf("%lld\n", value);
        }
        printf("\n");
    }

    printf("Get element at index -4 (first element):\n");
    {
        zl = createList();
        p = ziplistIndex(zl, -4);
        if (!ziplistGet(p, &entry, &elen, &value)) {
            printf("ERROR: Could not access index -4\n");
            return 1;
        }
        if (entry) {
            fwrite(entry,elen,1,stdout);
            printf("\n");
        } else {
            printf("%lld\n", value);
        }
        printf("\n");
    }

    printf("Get element at index -5 (reverse out of range):\n");
    {
        zl = createList();
        p = ziplistIndex(zl, -5);
        if (p == NULL) {
            printf("No entry\n");
        } else {
            printf("ERROR: Out of range index should return NULL, returned offset: %ld\n", p-zl);
            return 1;
        }
        printf("\n");
    }

710 711 712 713
    printf("Iterate list from 0 to end:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 0);
714
        while (ziplistGet(p, &entry, &elen, &value)) {
715
            printf("Entry: ");
716 717 718 719 720
            if (entry) {
                fwrite(entry,elen,1,stdout);
            } else {
                printf("%lld", value);
            }
721
            p = ziplistNext(zl,p);
722
            printf("\n");
723 724 725 726 727 728 729 730
        }
        printf("\n");
    }

    printf("Iterate list from 1 to end:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 1);
731
        while (ziplistGet(p, &entry, &elen, &value)) {
732
            printf("Entry: ");
733 734 735 736 737
            if (entry) {
                fwrite(entry,elen,1,stdout);
            } else {
                printf("%lld", value);
            }
738
            p = ziplistNext(zl,p);
739
            printf("\n");
740 741 742 743 744 745 746 747
        }
        printf("\n");
    }

    printf("Iterate list from 2 to end:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 2);
748
        while (ziplistGet(p, &entry, &elen, &value)) {
749
            printf("Entry: ");
750 751 752 753 754
            if (entry) {
                fwrite(entry,elen,1,stdout);
            } else {
                printf("%lld", value);
            }
755
            p = ziplistNext(zl,p);
756
            printf("\n");
757 758 759 760 761 762 763
        }
        printf("\n");
    }

    printf("Iterate starting out of range:\n");
    {
        zl = createList();
764 765
        p = ziplistIndex(zl, 4);
        if (!ziplistGet(p, &entry, &elen, &value)) {
766 767 768 769
            printf("No entry\n");
        } else {
            printf("ERROR\n");
        }
770 771 772
        printf("\n");
    }

773 774 775 776 777 778 779 780 781 782 783
    printf("Iterate from back to front:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, -1);
        while (ziplistGet(p, &entry, &elen, &value)) {
            printf("Entry: ");
            if (entry) {
                fwrite(entry,elen,1,stdout);
            } else {
                printf("%lld", value);
            }
784
            p = ziplistPrev(zl,p);
785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800
            printf("\n");
        }
        printf("\n");
    }

    printf("Iterate from back to front, deleting all items:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, -1);
        while (ziplistGet(p, &entry, &elen, &value)) {
            printf("Entry: ");
            if (entry) {
                fwrite(entry,elen,1,stdout);
            } else {
                printf("%lld", value);
            }
801 802
            zl = ziplistDelete(zl,&p);
            p = ziplistPrev(zl,p);
803 804 805 806 807
            printf("\n");
        }
        printf("\n");
    }

808 809 810
    printf("Delete inclusive range 0,0:\n");
    {
        zl = createList();
811
        zl = ziplistDeleteRange(zl, 0, 1);
812 813 814 815 816 817
        ziplistRepr(zl);
    }

    printf("Delete inclusive range 0,1:\n");
    {
        zl = createList();
818
        zl = ziplistDeleteRange(zl, 0, 2);
819 820 821 822 823 824
        ziplistRepr(zl);
    }

    printf("Delete inclusive range 1,2:\n");
    {
        zl = createList();
825
        zl = ziplistDeleteRange(zl, 1, 2);
826 827 828 829 830 831
        ziplistRepr(zl);
    }

    printf("Delete with start index out of range:\n");
    {
        zl = createList();
832
        zl = ziplistDeleteRange(zl, 5, 1);
833 834 835 836 837 838
        ziplistRepr(zl);
    }

    printf("Delete with num overflow:\n");
    {
        zl = createList();
839
        zl = ziplistDeleteRange(zl, 1, 5);
840
        ziplistRepr(zl);
841 842
    }

843 844 845 846
    printf("Delete foo while iterating:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 0);
847 848
        while (ziplistGet(p, &entry, &elen, &value)) {
            if (entry && strncmp("foo", entry, elen) == 0) {
849
                printf("Delete foo\n");
850
                zl = ziplistDelete(zl, &p, ZIPLIST_TAIL);
851 852
            } else {
                printf("Entry: ");
853 854 855 856 857 858 859
                if (entry) {
                    fwrite(entry,elen,1,stdout);
                } else {
                    printf("%lld", value);
                }
                p = ziplistNext(p);
                printf("\n");
860 861 862 863
            }
        }
        printf("\n");
        ziplistRepr(zl);
864 865
    }

866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886
    printf("Create long list and check indices:\n");
    {
        zl = ziplistNew();
        char buf[32];
        int i,len;
        for (i = 0; i < 1000; i++) {
            len = sprintf(buf,"%d",i);
            zl = ziplistPush(zl,buf,len,ZIPLIST_TAIL);
        }
        for (i = 0; i < 1000; i++) {
            p = ziplistIndex(zl,i);
            assert(ziplistGet(p,NULL,NULL,&value));
            assert(i == value);

            p = ziplistIndex(zl,-i-1);
            assert(ziplistGet(p,NULL,NULL,&value));
            assert(999-i == value);
        }
        printf("SUCCESS\n\n");
    }

887 888 889 890 891
    printf("Compare strings with ziplist entries:\n");
    {
        zl = createList();
        p = ziplistIndex(zl, 0);
        if (!ziplistCompare(p,"hello",5)) {
892
            printf("ERROR: not \"hello\"\n");
P
Pieter Noordhuis 已提交
893
            return 1;
894 895
        }
        if (ziplistCompare(p,"hella",5)) {
896
            printf("ERROR: \"hella\"\n");
P
Pieter Noordhuis 已提交
897
            return 1;
898 899 900 901
        }

        p = ziplistIndex(zl, 3);
        if (!ziplistCompare(p,"1024",4)) {
902
            printf("ERROR: not \"1024\"\n");
P
Pieter Noordhuis 已提交
903
            return 1;
904 905
        }
        if (ziplistCompare(p,"1025",4)) {
906
            printf("ERROR: \"1025\"\n");
P
Pieter Noordhuis 已提交
907
            return 1;
908 909
        }
        printf("SUCCESS\n");
910 911
    }

912 913 914
    return 0;
}
#endif