]> git.saurik.com Git - redis.git/blame - src/ziplist.c
Add ziplist test for deleting next to last entries
[redis.git] / src / ziplist.c
CommitLineData
c4705381
PN
1/* The ziplist is a specially encoded dually linked list that is designed
2 * to be very memory efficient. It stores both strings and integer values,
3 * where integers are encoded as actual integers instead of a series of
4 * characters. It allows push and pop operations on either side of the list
5 * in O(1) time. However, because every operation requires a reallocation of
6 * the memory used by the ziplist, the actual complexity is related to the
7 * amount of memory used by the ziplist.
11ac6ff6 8 *
c4705381 9 * ----------------------------------------------------------------------------
11ac6ff6 10 *
c4705381
PN
11 * ZIPLIST OVERALL LAYOUT:
12 * The general layout of the ziplist is as follows:
13 * <zlbytes><zltail><zllen><entry><entry><zlend>
11ac6ff6 14 *
c4705381
PN
15 * <zlbytes> is an unsigned integer to hold the number of bytes that the
16 * ziplist occupies. This value needs to be stored to be able to resize the
17 * entire structure without the need to traverse it first.
18 *
19 * <zltail> is the offset to the last entry in the list. This allows a pop
20 * operation on the far side of the list without the need for full traversal.
21 *
22 * <zllen> is the number of entries.When this value is larger than 2**16-2,
23 * we need to traverse the entire list to know how many items it holds.
24 *
25 * <zlend> is a single byte special value, equal to 255, which indicates the
26 * end of the list.
27 *
28 * ZIPLIST ENTRIES:
29 * Every entry in the ziplist is prefixed by a header that contains two pieces
30 * of information. First, the length of the previous entry is stored to be
31 * able to traverse the list from back to front. Second, the encoding with an
32 * optional string length of the entry itself is stored.
33 *
34 * The length of the previous entry is encoded in the following way:
35 * If this length is smaller than 254 bytes, it will only consume a single
36 * byte that takes the length as value. When the length is greater than or
37 * equal to 254, it will consume 5 bytes. The first byte is set to 254 to
38 * indicate a larger value is following. The remaining 4 bytes take the
39 * length of the previous entry as value.
40 *
41 * The other header field of the entry itself depends on the contents of the
42 * entry. When the entry is a string, the first 2 bits of this header will hold
43 * the type of encoding used to store the length of the string, followed by the
44 * actual length of the string. When the entry is an integer the first 2 bits
45 * are both set to 1. The following 2 bits are used to specify what kind of
46 * integer will be stored after this header. An overview of the different
47 * types and encodings is as follows:
48 *
49 * |00pppppp| - 1 byte
50 * String value with length less than or equal to 63 bytes (6 bits).
51 * |01pppppp|qqqqqqqq| - 2 bytes
52 * String value with length less than or equal to 16383 bytes (14 bits).
53 * |10______|qqqqqqqq|rrrrrrrr|ssssssss|tttttttt| - 5 bytes
54 * String value with length greater than or equal to 16384 bytes.
dcd4efe9 55 * |11000000| - 1 byte
c4705381 56 * Integer encoded as int16_t (2 bytes).
dcd4efe9 57 * |11010000| - 1 byte
c4705381 58 * Integer encoded as int32_t (4 bytes).
dcd4efe9 59 * |11100000| - 1 byte
c4705381 60 * Integer encoded as int64_t (8 bytes).
dcd4efe9 61 * |11110000| - 1 byte
dd51571c 62 * Integer encoded as 24 bit signed (3 bytes).
dcd4efe9 63 * |11111110| - 1 byte
64 * Integer encoded as 8 bit signed (1 byte).
65 * |1111xxxx| - (with xxxx between 0000 and 1101) immediate 4 bit integer.
66 * Unsigned integer from 0 to 12. The encoded value is actually from
67 * 1 to 13 because 0000 and 1111 can not be used, so 1 should be
68 * subtracted from the encoded 4 bit value to obtain the right value.
69 * |11111111| - End of ziplist.
dd51571c 70 *
71 * All the integers are represented in little endian byte order.
11ac6ff6
PN
72 */
73
74#include <stdio.h>
29b14d5f 75#include <stdlib.h>
11ac6ff6 76#include <string.h>
e1f93d4b 77#include <stdint.h>
11ac6ff6 78#include <assert.h>
29b14d5f 79#include <limits.h>
11ac6ff6 80#include "zmalloc.h"
edf23aff 81#include "util.h"
11ac6ff6 82#include "ziplist.h"
7a3e3720 83#include "endianconv.h"
11ac6ff6 84
37fff074 85#define ZIP_END 255
aa549962 86#define ZIP_BIGLEN 254
37fff074 87
c4705381 88/* Different encoding/length possibilities */
dcd4efe9 89#define ZIP_STR_MASK 0xc0
90#define ZIP_INT_MASK 0x30
c4705381
PN
91#define ZIP_STR_06B (0 << 6)
92#define ZIP_STR_14B (1 << 6)
93#define ZIP_STR_32B (2 << 6)
94#define ZIP_INT_16B (0xc0 | 0<<4)
95#define ZIP_INT_32B (0xc0 | 1<<4)
96#define ZIP_INT_64B (0xc0 | 2<<4)
ad91404a 97#define ZIP_INT_24B (0xc0 | 3<<4)
dcd4efe9 98#define ZIP_INT_8B 0xfe
99/* 4 bit integer immediate encoding */
100#define ZIP_INT_IMM_MASK 0x0f
101#define ZIP_INT_IMM_MIN 0xf1 /* 11110001 */
102#define ZIP_INT_IMM_MAX 0xfd /* 11111101 */
103#define ZIP_INT_IMM_VAL(v) (v & ZIP_INT_IMM_MASK)
ad91404a
GT
104
105#define INT24_MAX 0x7fffff
106#define INT24_MIN (-INT24_MAX - 1)
37fff074 107
fe458402
PN
108/* Macro to determine type */
109#define ZIP_IS_STR(enc) (((enc) & ZIP_STR_MASK) < ZIP_STR_MASK)
37fff074
PN
110
111/* Utility macros */
e1f93d4b
PN
112#define ZIPLIST_BYTES(zl) (*((uint32_t*)(zl)))
113#define ZIPLIST_TAIL_OFFSET(zl) (*((uint32_t*)((zl)+sizeof(uint32_t))))
114#define ZIPLIST_LENGTH(zl) (*((uint16_t*)((zl)+sizeof(uint32_t)*2)))
115#define ZIPLIST_HEADER_SIZE (sizeof(uint32_t)*2+sizeof(uint16_t))
116#define ZIPLIST_ENTRY_HEAD(zl) ((zl)+ZIPLIST_HEADER_SIZE)
8e0ef249 117#define ZIPLIST_ENTRY_TAIL(zl) ((zl)+intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl)))
118#define ZIPLIST_ENTRY_END(zl) ((zl)+intrev32ifbe(ZIPLIST_BYTES(zl))-1)
e1f93d4b
PN
119
120/* We know a positive increment can only be 1 because entries can only be
121 * pushed one at a time. */
f6eb1747 122#define ZIPLIST_INCR_LENGTH(zl,incr) { \
56538477 123 if (ZIPLIST_LENGTH(zl) < UINT16_MAX) \
3fa19b7d 124 ZIPLIST_LENGTH(zl) = intrev16ifbe(intrev16ifbe(ZIPLIST_LENGTH(zl))+incr); \
56538477 125}
11ac6ff6 126
a5456b2c
PN
127typedef struct zlentry {
128 unsigned int prevrawlensize, prevrawlen;
129 unsigned int lensize, len;
130 unsigned int headersize;
131 unsigned char encoding;
0c0d0564 132 unsigned char *p;
a5456b2c
PN
133} zlentry;
134
dcd4efe9 135/* Extract the encoding from the byte pointed by 'ptr' and set it into
136 * 'encoding'. */
137#define ZIP_ENTRY_ENCODING(ptr, encoding) do { \
138 (encoding) = (ptr[0]); \
139 if ((encoding) < ZIP_STR_MASK) (encoding) &= ZIP_STR_MASK; \
fe458402 140} while(0)
c4705381 141
37fff074 142/* Return bytes needed to store integer encoded by 'encoding' */
c4705381
PN
143static unsigned int zipIntSize(unsigned char encoding) {
144 switch(encoding) {
dcd4efe9 145 case ZIP_INT_8B: return 1;
146 case ZIP_INT_16B: return 2;
147 case ZIP_INT_24B: return 3;
148 case ZIP_INT_32B: return 4;
149 case ZIP_INT_64B: return 8;
150 default: return 0; /* 4 bit immediate */
37fff074
PN
151 }
152 assert(NULL);
8ce39260 153 return 0;
37fff074
PN
154}
155
37fff074
PN
156/* Encode the length 'l' writing it in 'p'. If p is NULL it just returns
157 * the amount of bytes required to encode such a length. */
c4705381
PN
158static unsigned int zipEncodeLength(unsigned char *p, unsigned char encoding, unsigned int rawlen) {
159 unsigned char len = 1, buf[5];
160
161 if (ZIP_IS_STR(encoding)) {
162 /* Although encoding is given it may not be set for strings,
163 * so we determine it here using the raw length. */
164 if (rawlen <= 0x3f) {
37fff074 165 if (!p) return len;
c4705381
PN
166 buf[0] = ZIP_STR_06B | rawlen;
167 } else if (rawlen <= 0x3fff) {
168 len += 1;
37fff074 169 if (!p) return len;
c4705381
PN
170 buf[0] = ZIP_STR_14B | ((rawlen >> 8) & 0x3f);
171 buf[1] = rawlen & 0xff;
37fff074
PN
172 } else {
173 len += 4;
174 if (!p) return len;
c4705381
PN
175 buf[0] = ZIP_STR_32B;
176 buf[1] = (rawlen >> 24) & 0xff;
177 buf[2] = (rawlen >> 16) & 0xff;
178 buf[3] = (rawlen >> 8) & 0xff;
179 buf[4] = rawlen & 0xff;
37fff074 180 }
c4705381
PN
181 } else {
182 /* Implies integer encoding, so length is always 1. */
183 if (!p) return len;
184 buf[0] = encoding;
37fff074 185 }
37fff074 186
c4705381 187 /* Store this length at p */
37fff074
PN
188 memcpy(p,buf,len);
189 return len;
190}
191
fe458402
PN
192/* Decode the length encoded in 'ptr'. The 'encoding' variable will hold the
193 * entries encoding, the 'lensize' variable will hold the number of bytes
194 * required to encode the entries length, and the 'len' variable will hold the
195 * entries length. */
196#define ZIP_DECODE_LENGTH(ptr, encoding, lensize, len) do { \
197 ZIP_ENTRY_ENCODING((ptr), (encoding)); \
198 if ((encoding) < ZIP_STR_MASK) { \
199 if ((encoding) == ZIP_STR_06B) { \
200 (lensize) = 1; \
201 (len) = (ptr)[0] & 0x3f; \
202 } else if ((encoding) == ZIP_STR_14B) { \
203 (lensize) = 2; \
204 (len) = (((ptr)[0] & 0x3f) << 8) | (ptr)[1]; \
205 } else if (encoding == ZIP_STR_32B) { \
206 (lensize) = 5; \
207 (len) = ((ptr)[1] << 24) | \
208 ((ptr)[2] << 16) | \
209 ((ptr)[3] << 8) | \
210 ((ptr)[4]); \
211 } else { \
212 assert(NULL); \
213 } \
214 } else { \
215 (lensize) = 1; \
216 (len) = zipIntSize(encoding); \
217 } \
218} while(0);
7b1f85c0
PN
219
220/* Encode the length of the previous entry and write it to "p". Return the
221 * number of bytes needed to encode this length if "p" is NULL. */
222static unsigned int zipPrevEncodeLength(unsigned char *p, unsigned int len) {
223 if (p == NULL) {
224 return (len < ZIP_BIGLEN) ? 1 : sizeof(len)+1;
225 } else {
226 if (len < ZIP_BIGLEN) {
227 p[0] = len;
228 return 1;
229 } else {
230 p[0] = ZIP_BIGLEN;
231 memcpy(p+1,&len,sizeof(len));
f2204374 232 memrev32ifbe(p+1);
7b1f85c0
PN
233 return 1+sizeof(len);
234 }
235 }
236}
237
169d2ef1
PN
238/* Encode the length of the previous entry and write it to "p". This only
239 * uses the larger encoding (required in __ziplistCascadeUpdate). */
240static void zipPrevEncodeLengthForceLarge(unsigned char *p, unsigned int len) {
241 if (p == NULL) return;
242 p[0] = ZIP_BIGLEN;
243 memcpy(p+1,&len,sizeof(len));
f2204374 244 memrev32ifbe(p+1);
169d2ef1
PN
245}
246
fe458402
PN
247/* Decode the number of bytes required to store the length of the previous
248 * element, from the perspective of the entry pointed to by 'ptr'. */
249#define ZIP_DECODE_PREVLENSIZE(ptr, prevlensize) do { \
250 if ((ptr)[0] < ZIP_BIGLEN) { \
251 (prevlensize) = 1; \
252 } else { \
253 (prevlensize) = 5; \
254 } \
255} while(0);
256
257/* Decode the length of the previous element, from the perspective of the entry
258 * pointed to by 'ptr'. */
259#define ZIP_DECODE_PREVLEN(ptr, prevlensize, prevlen) do { \
260 ZIP_DECODE_PREVLENSIZE(ptr, prevlensize); \
261 if ((prevlensize) == 1) { \
262 (prevlen) = (ptr)[0]; \
263 } else if ((prevlensize) == 5) { \
264 assert(sizeof((prevlensize)) == 4); \
265 memcpy(&(prevlen), ((char*)(ptr)) + 1, 4); \
b64281cc 266 memrev32ifbe(&prevlen); \
fe458402
PN
267 } \
268} while(0);
269
270/* Return the difference in number of bytes needed to store the length of the
271 * previous element 'len', in the entry pointed to by 'p'. */
dcb9cf4e
PN
272static int zipPrevLenByteDiff(unsigned char *p, unsigned int len) {
273 unsigned int prevlensize;
fe458402
PN
274 ZIP_DECODE_PREVLENSIZE(p, prevlensize);
275 return zipPrevEncodeLength(NULL, len) - prevlensize;
276}
277
278/* Return the total number of bytes used by the entry pointed to by 'p'. */
279static unsigned int zipRawEntryLength(unsigned char *p) {
280 unsigned int prevlensize, encoding, lensize, len;
281 ZIP_DECODE_PREVLENSIZE(p, prevlensize);
282 ZIP_DECODE_LENGTH(p + prevlensize, encoding, lensize, len);
283 return prevlensize + lensize + len;
dcb9cf4e
PN
284}
285
37fff074 286/* Check if string pointed to by 'entry' can be encoded as an integer.
61712508 287 * Stores the integer value in 'v' and its encoding in 'encoding'. */
288static int zipTryEncoding(unsigned char *entry, unsigned int entrylen, long long *v, unsigned char *encoding) {
37fff074 289 long long value;
37fff074 290
61712508 291 if (entrylen >= 32 || entrylen == 0) return 0;
edf23aff 292 if (string2ll((char*)entry,entrylen,&value)) {
61712508 293 /* Great, the string can be encoded. Check what's the smallest
294 * of our encoding types that can hold this value. */
dcd4efe9 295 if (value >= 0 && value <= 12) {
296 *encoding = ZIP_INT_IMM_MIN+value;
297 } else if (value >= INT8_MIN && value <= INT8_MAX) {
298 *encoding = ZIP_INT_8B;
299 } else if (value >= INT16_MIN && value <= INT16_MAX) {
c4705381 300 *encoding = ZIP_INT_16B;
ad91404a
GT
301 } else if (value >= INT24_MIN && value <= INT24_MAX) {
302 *encoding = ZIP_INT_24B;
e1f93d4b 303 } else if (value >= INT32_MIN && value <= INT32_MAX) {
c4705381 304 *encoding = ZIP_INT_32B;
37fff074 305 } else {
c4705381 306 *encoding = ZIP_INT_64B;
37fff074
PN
307 }
308 *v = value;
309 return 1;
310 }
311 return 0;
312}
313
314/* Store integer 'value' at 'p', encoded as 'encoding' */
e1f93d4b
PN
315static void zipSaveInteger(unsigned char *p, int64_t value, unsigned char encoding) {
316 int16_t i16;
317 int32_t i32;
318 int64_t i64;
dcd4efe9 319 if (encoding == ZIP_INT_8B) {
82675c86 320 ((int8_t*)p)[0] = (int8_t)value;
dcd4efe9 321 } else if (encoding == ZIP_INT_16B) {
e1f93d4b
PN
322 i16 = value;
323 memcpy(p,&i16,sizeof(i16));
f2204374 324 memrev16ifbe(p);
ad91404a
GT
325 } else if (encoding == ZIP_INT_24B) {
326 i32 = value<<8;
327 memrev32ifbe(&i32);
82675c86 328 memcpy(p,((uint8_t*)&i32)+1,sizeof(i32)-sizeof(uint8_t));
c4705381 329 } else if (encoding == ZIP_INT_32B) {
e1f93d4b
PN
330 i32 = value;
331 memcpy(p,&i32,sizeof(i32));
f2204374 332 memrev32ifbe(p);
c4705381 333 } else if (encoding == ZIP_INT_64B) {
e1f93d4b
PN
334 i64 = value;
335 memcpy(p,&i64,sizeof(i64));
f2204374 336 memrev64ifbe(p);
dcd4efe9 337 } else if (encoding >= ZIP_INT_IMM_MIN && encoding <= ZIP_INT_IMM_MAX) {
338 /* Nothing to do, the value is stored in the encoding itself. */
37fff074
PN
339 } else {
340 assert(NULL);
341 }
342}
343
344/* Read integer encoded as 'encoding' from 'p' */
e1f93d4b
PN
345static int64_t zipLoadInteger(unsigned char *p, unsigned char encoding) {
346 int16_t i16;
347 int32_t i32;
8ce39260 348 int64_t i64, ret = 0;
dcd4efe9 349 if (encoding == ZIP_INT_8B) {
82675c86 350 ret = ((int8_t*)p)[0];
dcd4efe9 351 } else if (encoding == ZIP_INT_16B) {
e1f93d4b 352 memcpy(&i16,p,sizeof(i16));
f2204374 353 memrev16ifbe(&i16);
e1f93d4b 354 ret = i16;
c4705381 355 } else if (encoding == ZIP_INT_32B) {
e1f93d4b 356 memcpy(&i32,p,sizeof(i32));
66d1b021 357 memrev32ifbe(&i32);
e1f93d4b 358 ret = i32;
ad91404a
GT
359 } else if (encoding == ZIP_INT_24B) {
360 i32 = 0;
82675c86 361 memcpy(((uint8_t*)&i32)+1,p,sizeof(i32)-sizeof(uint8_t));
ad91404a
GT
362 memrev32ifbe(&i32);
363 ret = i32>>8;
c4705381 364 } else if (encoding == ZIP_INT_64B) {
e1f93d4b 365 memcpy(&i64,p,sizeof(i64));
3fa19b7d 366 memrev64ifbe(&i64);
e1f93d4b 367 ret = i64;
dcd4efe9 368 } else if (encoding >= ZIP_INT_IMM_MIN && encoding <= ZIP_INT_IMM_MAX) {
369 ret = (encoding & ZIP_INT_IMM_MASK)-1;
37fff074
PN
370 } else {
371 assert(NULL);
372 }
373 return ret;
374}
375
a5456b2c
PN
376/* Return a struct with all information about an entry. */
377static zlentry zipEntry(unsigned char *p) {
378 zlentry e;
fe458402
PN
379
380 ZIP_DECODE_PREVLEN(p, e.prevrawlensize, e.prevrawlen);
381 ZIP_DECODE_LENGTH(p + e.prevrawlensize, e.encoding, e.lensize, e.len);
382 e.headersize = e.prevrawlensize + e.lensize;
0c0d0564 383 e.p = p;
a5456b2c
PN
384 return e;
385}
386
11ac6ff6
PN
387/* Create a new empty ziplist. */
388unsigned char *ziplistNew(void) {
389 unsigned int bytes = ZIPLIST_HEADER_SIZE+1;
390 unsigned char *zl = zmalloc(bytes);
56538477 391 ZIPLIST_BYTES(zl) = intrev32ifbe(bytes);
392 ZIPLIST_TAIL_OFFSET(zl) = intrev32ifbe(ZIPLIST_HEADER_SIZE);
11ac6ff6
PN
393 ZIPLIST_LENGTH(zl) = 0;
394 zl[bytes-1] = ZIP_END;
395 return zl;
396}
397
37fff074 398/* Resize the ziplist. */
11ac6ff6 399static unsigned char *ziplistResize(unsigned char *zl, unsigned int len) {
37fff074 400 zl = zrealloc(zl,len);
56538477 401 ZIPLIST_BYTES(zl) = intrev32ifbe(len);
11ac6ff6
PN
402 zl[len-1] = ZIP_END;
403 return zl;
404}
405
169d2ef1
PN
406/* When an entry is inserted, we need to set the prevlen field of the next
407 * entry to equal the length of the inserted entry. It can occur that this
408 * length cannot be encoded in 1 byte and the next entry needs to be grow
409 * a bit larger to hold the 5-byte encoded prevlen. This can be done for free,
410 * because this only happens when an entry is already being inserted (which
411 * causes a realloc and memmove). However, encoding the prevlen may require
412 * that this entry is grown as well. This effect may cascade throughout
413 * the ziplist when there are consecutive entries with a size close to
414 * ZIP_BIGLEN, so we need to check that the prevlen can be encoded in every
415 * consecutive entry.
416 *
417 * Note that this effect can also happen in reverse, where the bytes required
418 * to encode the prevlen field can shrink. This effect is deliberately ignored,
419 * because it can cause a "flapping" effect where a chain prevlen fields is
420 * first grown and then shrunk again after consecutive inserts. Rather, the
421 * field is allowed to stay larger than necessary, because a large prevlen
422 * field implies the ziplist is holding large entries anyway.
423 *
424 * The pointer "p" points to the first entry that does NOT need to be
425 * updated, i.e. consecutive fields MAY need an update. */
426static unsigned char *__ziplistCascadeUpdate(unsigned char *zl, unsigned char *p) {
56538477 427 size_t curlen = intrev32ifbe(ZIPLIST_BYTES(zl)), rawlen, rawlensize;
69298a05 428 size_t offset, noffset, extra;
169d2ef1
PN
429 unsigned char *np;
430 zlentry cur, next;
431
432 while (p[0] != ZIP_END) {
433 cur = zipEntry(p);
434 rawlen = cur.headersize + cur.len;
435 rawlensize = zipPrevEncodeLength(NULL,rawlen);
436
437 /* Abort if there is no next entry. */
438 if (p[rawlen] == ZIP_END) break;
439 next = zipEntry(p+rawlen);
440
441 /* Abort when "prevlen" has not changed. */
442 if (next.prevrawlen == rawlen) break;
443
444 if (next.prevrawlensize < rawlensize) {
445 /* The "prevlen" field of "next" needs more bytes to hold
446 * the raw length of "cur". */
447 offset = p-zl;
448 extra = rawlensize-next.prevrawlensize;
449 zl = ziplistResize(zl,curlen+extra);
169d2ef1
PN
450 p = zl+offset;
451
b7d3bf51 452 /* Current pointer and offset for next element. */
169d2ef1
PN
453 np = p+rawlen;
454 noffset = np-zl;
b7d3bf51
PN
455
456 /* Update tail offset when next element is not the tail element. */
3fa19b7d 457 if ((zl+intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))) != np) {
458 ZIPLIST_TAIL_OFFSET(zl) =
459 intrev32ifbe(intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))+extra);
460 }
b7d3bf51
PN
461
462 /* Move the tail to the back. */
169d2ef1
PN
463 memmove(np+rawlensize,
464 np+next.prevrawlensize,
465 curlen-noffset-next.prevrawlensize-1);
466 zipPrevEncodeLength(np,rawlen);
467
468 /* Advance the cursor */
469 p += rawlen;
306c6a02 470 curlen += extra;
169d2ef1
PN
471 } else {
472 if (next.prevrawlensize > rawlensize) {
473 /* This would result in shrinking, which we want to avoid.
474 * So, set "rawlen" in the available bytes. */
475 zipPrevEncodeLengthForceLarge(p+rawlen,rawlen);
476 } else {
477 zipPrevEncodeLength(p+rawlen,rawlen);
478 }
479
480 /* Stop here, as the raw length of "next" has not changed. */
481 break;
482 }
483 }
484 return zl;
485}
486
0c0d0564 487/* Delete "num" entries, starting at "p". Returns pointer to the ziplist. */
b6eb9703 488static unsigned char *__ziplistDelete(unsigned char *zl, unsigned char *p, unsigned int num) {
0c0d0564 489 unsigned int i, totlen, deleted = 0;
69298a05
PN
490 size_t offset;
491 int nextdiff = 0;
169d2ef1
PN
492 zlentry first, tail;
493
494 first = zipEntry(p);
0c0d0564
PN
495 for (i = 0; p[0] != ZIP_END && i < num; i++) {
496 p += zipRawEntryLength(p);
497 deleted++;
498 }
499
500 totlen = p-first.p;
501 if (totlen > 0) {
502 if (p[0] != ZIP_END) {
503 /* Tricky: storing the prevlen in this entry might reduce or
504 * increase the number of bytes needed, compared to the current
505 * prevlen. Note that we can always store this length because
506 * it was previously stored by an entry that is being deleted. */
507 nextdiff = zipPrevLenByteDiff(p,first.prevrawlen);
7b1f85c0 508 zipPrevEncodeLength(p-nextdiff,first.prevrawlen);
0c0d0564
PN
509
510 /* Update offset for tail */
3fa19b7d 511 ZIPLIST_TAIL_OFFSET(zl) =
512 intrev32ifbe(intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))-totlen);
169d2ef1
PN
513
514 /* When the tail contains more than one entry, we need to take
515 * "nextdiff" in account as well. Otherwise, a change in the
516 * size of prevlen doesn't have an effect on the *tail* offset. */
517 tail = zipEntry(p);
3fa19b7d 518 if (p[tail.headersize+tail.len] != ZIP_END) {
cab1105c 519 ZIPLIST_TAIL_OFFSET(zl) =
3fa19b7d 520 intrev32ifbe(intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))+nextdiff);
521 }
0c0d0564
PN
522
523 /* Move tail to the front of the ziplist */
56538477 524 memmove(first.p,p-nextdiff,
525 intrev32ifbe(ZIPLIST_BYTES(zl))-(p-zl)-1+nextdiff);
0c0d0564
PN
526 } else {
527 /* The entire tail was deleted. No need to move memory. */
56538477 528 ZIPLIST_TAIL_OFFSET(zl) =
529 intrev32ifbe((first.p-zl)-first.prevrawlen);
0c0d0564
PN
530 }
531
532 /* Resize and update length */
169d2ef1 533 offset = first.p-zl;
56538477 534 zl = ziplistResize(zl, intrev32ifbe(ZIPLIST_BYTES(zl))-totlen+nextdiff);
0c0d0564 535 ZIPLIST_INCR_LENGTH(zl,-deleted);
169d2ef1
PN
536 p = zl+offset;
537
538 /* When nextdiff != 0, the raw length of the next entry has changed, so
539 * we need to cascade the update throughout the ziplist */
540 if (nextdiff != 0)
541 zl = __ziplistCascadeUpdate(zl,p);
0c0d0564
PN
542 }
543 return zl;
11ac6ff6
PN
544}
545
6435c767 546/* Insert item at "p". */
b6eb9703 547static unsigned char *__ziplistInsert(unsigned char *zl, unsigned char *p, unsigned char *s, unsigned int slen) {
56538477 548 size_t curlen = intrev32ifbe(ZIPLIST_BYTES(zl)), reqlen, prevlen = 0;
69298a05
PN
549 size_t offset;
550 int nextdiff = 0;
c4705381 551 unsigned char encoding = 0;
f013f400 552 long long value = 123456789; /* initialized to avoid warning. Using a value
553 that is easy to see if for some reason
554 we use it uninitialized. */
169d2ef1 555 zlentry entry, tail;
11ac6ff6 556
6435c767
PN
557 /* Find out prevlen for the entry that is inserted. */
558 if (p[0] != ZIP_END) {
559 entry = zipEntry(p);
560 prevlen = entry.prevrawlen;
dcb9cf4e 561 } else {
169d2ef1
PN
562 unsigned char *ptail = ZIPLIST_ENTRY_TAIL(zl);
563 if (ptail[0] != ZIP_END) {
564 prevlen = zipRawEntryLength(ptail);
6435c767 565 }
dcb9cf4e
PN
566 }
567
29b14d5f 568 /* See if the entry can be encoded */
61712508 569 if (zipTryEncoding(s,slen,&value,&encoding)) {
c4705381
PN
570 /* 'encoding' is set to the appropriate integer encoding */
571 reqlen = zipIntSize(encoding);
29b14d5f 572 } else {
c4705381
PN
573 /* 'encoding' is untouched, however zipEncodeLength will use the
574 * string length to figure out how to encode it. */
6435c767 575 reqlen = slen;
29b14d5f 576 }
dcb9cf4e
PN
577 /* We need space for both the length of the previous entry and
578 * the length of the payload. */
7b1f85c0 579 reqlen += zipPrevEncodeLength(NULL,prevlen);
6435c767
PN
580 reqlen += zipEncodeLength(NULL,encoding,slen);
581
582 /* When the insert position is not equal to the tail, we need to
583 * make sure that the next entry can hold this entry's length in
584 * its prevlen field. */
177a0a0b 585 nextdiff = (p[0] != ZIP_END) ? zipPrevLenByteDiff(p,reqlen) : 0;
6435c767
PN
586
587 /* Store offset because a realloc may change the address of zl. */
588 offset = p-zl;
589 zl = ziplistResize(zl,curlen+reqlen+nextdiff);
590 p = zl+offset;
591
592 /* Apply memory move when necessary and update tail offset. */
593 if (p[0] != ZIP_END) {
594 /* Subtract one because of the ZIP_END bytes */
595 memmove(p+reqlen,p-nextdiff,curlen-offset-1+nextdiff);
169d2ef1 596
6435c767 597 /* Encode this entry's raw length in the next entry. */
7b1f85c0 598 zipPrevEncodeLength(p+reqlen,reqlen);
169d2ef1 599
6435c767 600 /* Update offset for tail */
3fa19b7d 601 ZIPLIST_TAIL_OFFSET(zl) =
602 intrev32ifbe(intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))+reqlen);
169d2ef1
PN
603
604 /* When the tail contains more than one entry, we need to take
605 * "nextdiff" in account as well. Otherwise, a change in the
606 * size of prevlen doesn't have an effect on the *tail* offset. */
607 tail = zipEntry(p+reqlen);
3fa19b7d 608 if (p[reqlen+tail.headersize+tail.len] != ZIP_END) {
609 ZIPLIST_TAIL_OFFSET(zl) =
610 intrev32ifbe(intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl))+nextdiff);
611 }
11ac6ff6 612 } else {
6435c767 613 /* This element will be the new tail. */
56538477 614 ZIPLIST_TAIL_OFFSET(zl) = intrev32ifbe(p-zl);
dcb9cf4e
PN
615 }
616
169d2ef1
PN
617 /* When nextdiff != 0, the raw length of the next entry has changed, so
618 * we need to cascade the update throughout the ziplist */
619 if (nextdiff != 0) {
620 offset = p-zl;
621 zl = __ziplistCascadeUpdate(zl,p+reqlen);
622 p = zl+offset;
623 }
624
11ac6ff6 625 /* Write the entry */
7b1f85c0 626 p += zipPrevEncodeLength(p,prevlen);
6435c767 627 p += zipEncodeLength(p,encoding,slen);
c4705381 628 if (ZIP_IS_STR(encoding)) {
6435c767 629 memcpy(p,s,slen);
c4705381
PN
630 } else {
631 zipSaveInteger(p,value,encoding);
29b14d5f 632 }
f6eb1747 633 ZIPLIST_INCR_LENGTH(zl,1);
11ac6ff6
PN
634 return zl;
635}
636
b6eb9703 637unsigned char *ziplistPush(unsigned char *zl, unsigned char *s, unsigned int slen, int where) {
6435c767 638 unsigned char *p;
1ce81fa5 639 p = (where == ZIPLIST_HEAD) ? ZIPLIST_ENTRY_HEAD(zl) : ZIPLIST_ENTRY_END(zl);
6435c767
PN
640 return __ziplistInsert(zl,p,s,slen);
641}
642
c03206fd
PN
643/* Returns an offset to use for iterating with ziplistNext. When the given
644 * index is negative, the list is traversed back to front. When the list
645 * doesn't contain an element at the provided index, NULL is returned. */
646unsigned char *ziplistIndex(unsigned char *zl, int index) {
647 unsigned char *p;
648 zlentry entry;
649 if (index < 0) {
650 index = (-index)-1;
651 p = ZIPLIST_ENTRY_TAIL(zl);
652 if (p[0] != ZIP_END) {
653 entry = zipEntry(p);
654 while (entry.prevrawlen > 0 && index--) {
655 p -= entry.prevrawlen;
656 entry = zipEntry(p);
657 }
658 }
659 } else {
660 p = ZIPLIST_ENTRY_HEAD(zl);
661 while (p[0] != ZIP_END && index--) {
662 p += zipRawEntryLength(p);
663 }
08253bf4 664 }
177a0a0b 665 return (p[0] == ZIP_END || index > 0) ? NULL : p;
08253bf4
PN
666}
667
d51ebef5 668/* Return pointer to next entry in ziplist.
669 *
670 * zl is the pointer to the ziplist
671 * p is the pointer to the current element
672 *
673 * The element after 'p' is returned, otherwise NULL if we are at the end. */
8632fb30
PN
674unsigned char *ziplistNext(unsigned char *zl, unsigned char *p) {
675 ((void) zl);
d71b9865
PN
676
677 /* "p" could be equal to ZIP_END, caused by ziplistDelete,
678 * and we should return NULL. Otherwise, we should return NULL
679 * when the *next* element is ZIP_END (there is no next entry). */
680 if (p[0] == ZIP_END) {
681 return NULL;
d71b9865 682 }
fe458402
PN
683
684 p += zipRawEntryLength(p);
685 if (p[0] == ZIP_END) {
686 return NULL;
687 }
688
689 return p;
75d8978e
PN
690}
691
033fb554 692/* Return pointer to previous entry in ziplist. */
8632fb30
PN
693unsigned char *ziplistPrev(unsigned char *zl, unsigned char *p) {
694 zlentry entry;
695
696 /* Iterating backwards from ZIP_END should return the tail. When "p" is
697 * equal to the first element of the list, we're already at the head,
698 * and should return NULL. */
699 if (p[0] == ZIP_END) {
700 p = ZIPLIST_ENTRY_TAIL(zl);
701 return (p[0] == ZIP_END) ? NULL : p;
702 } else if (p == ZIPLIST_ENTRY_HEAD(zl)) {
703 return NULL;
704 } else {
705 entry = zipEntry(p);
169d2ef1 706 assert(entry.prevrawlen > 0);
8632fb30
PN
707 return p-entry.prevrawlen;
708 }
033fb554
PN
709}
710
75d8978e
PN
711/* Get entry pointer to by 'p' and store in either 'e' or 'v' depending
712 * on the encoding of the entry. 'e' is always set to NULL to be able
713 * to find out whether the string pointer or the integer value was set.
714 * Return 0 if 'p' points to the end of the zipmap, 1 otherwise. */
b6eb9703 715unsigned int ziplistGet(unsigned char *p, unsigned char **sstr, unsigned int *slen, long long *sval) {
a5456b2c 716 zlentry entry;
c03206fd 717 if (p == NULL || p[0] == ZIP_END) return 0;
03e52931 718 if (sstr) *sstr = NULL;
dcb9cf4e 719
a5456b2c 720 entry = zipEntry(p);
c4705381 721 if (ZIP_IS_STR(entry.encoding)) {
03e52931
PN
722 if (sstr) {
723 *slen = entry.len;
b6eb9703 724 *sstr = p+entry.headersize;
75d8978e
PN
725 }
726 } else {
03e52931
PN
727 if (sval) {
728 *sval = zipLoadInteger(p+entry.headersize,entry.encoding);
75d8978e 729 }
08253bf4 730 }
75d8978e 731 return 1;
08253bf4
PN
732}
733
033fb554 734/* Insert an entry at "p". */
b6eb9703 735unsigned char *ziplistInsert(unsigned char *zl, unsigned char *p, unsigned char *s, unsigned int slen) {
033fb554 736 return __ziplistInsert(zl,p,s,slen);
779deb60
PN
737}
738
0f10458c
PN
739/* Delete a single entry from the ziplist, pointed to by *p.
740 * Also update *p in place, to be able to iterate over the
741 * ziplist, while deleting entries. */
6a8e35ad 742unsigned char *ziplistDelete(unsigned char *zl, unsigned char **p) {
69298a05 743 size_t offset = *p-zl;
0c0d0564 744 zl = __ziplistDelete(zl,*p,1);
0f10458c 745
0c0d0564 746 /* Store pointer to current element in p, because ziplistDelete will
0f3dfa87
PN
747 * do a realloc which might result in a different "zl"-pointer.
748 * When the delete direction is back to front, we might delete the last
749 * entry and end up with "p" pointing to ZIP_END, so check this. */
6a8e35ad 750 *p = zl+offset;
0f10458c
PN
751 return zl;
752}
753
033fb554
PN
754/* Delete a range of entries from the ziplist. */
755unsigned char *ziplistDeleteRange(unsigned char *zl, unsigned int index, unsigned int num) {
756 unsigned char *p = ziplistIndex(zl,index);
757 return (p == NULL) ? zl : __ziplistDelete(zl,p,num);
758}
759
c09c2c3b 760/* Compare entry pointer to by 'p' with 'entry'. Return 1 if equal. */
b6eb9703 761unsigned int ziplistCompare(unsigned char *p, unsigned char *sstr, unsigned int slen) {
a5456b2c 762 zlentry entry;
b6eb9703
PN
763 unsigned char sencoding;
764 long long zval, sval;
177a0a0b 765 if (p[0] == ZIP_END) return 0;
c09c2c3b 766
a5456b2c 767 entry = zipEntry(p);
c4705381 768 if (ZIP_IS_STR(entry.encoding)) {
c09c2c3b 769 /* Raw compare */
a5456b2c 770 if (entry.len == slen) {
03e52931 771 return memcmp(p+entry.headersize,sstr,slen) == 0;
c09c2c3b
PN
772 } else {
773 return 0;
774 }
c4aace90 775 } else {
0ef88927
PN
776 /* Try to compare encoded values. Don't compare encoding because
777 * different implementations may encoded integers differently. */
61712508 778 if (zipTryEncoding(sstr,slen,&sval,&sencoding)) {
0ef88927
PN
779 zval = zipLoadInteger(p+entry.headersize,entry.encoding);
780 return zval == sval;
c4aace90 781 }
c09c2c3b 782 }
c4aace90 783 return 0;
c09c2c3b
PN
784}
785
fe458402
PN
786/* Find pointer to the entry equal to the specified entry. Skip 'skip' entries
787 * between every comparison. Returns NULL when the field could not be found. */
788unsigned char *ziplistFind(unsigned char *p, unsigned char *vstr, unsigned int vlen, unsigned int skip) {
789 int skipcnt = 0;
790 unsigned char vencoding = 0;
791 long long vll = 0;
792
793 while (p[0] != ZIP_END) {
794 unsigned int prevlensize, encoding, lensize, len;
795 unsigned char *q;
796
797 ZIP_DECODE_PREVLENSIZE(p, prevlensize);
798 ZIP_DECODE_LENGTH(p + prevlensize, encoding, lensize, len);
799 q = p + prevlensize + lensize;
800
801 if (skipcnt == 0) {
802 /* Compare current entry with specified entry */
803 if (ZIP_IS_STR(encoding)) {
804 if (len == vlen && memcmp(q, vstr, vlen) == 0) {
805 return p;
806 }
807 } else {
8361d6c4 808 /* Find out if the searched field can be encoded. Note that
809 * we do it only the first time, once done vencoding is set
810 * to non-zero and vll is set to the integer value. */
fe458402 811 if (vencoding == 0) {
fe458402 812 if (!zipTryEncoding(vstr, vlen, &vll, &vencoding)) {
8361d6c4 813 /* If the entry can't be encoded we set it to
814 * UCHAR_MAX so that we don't retry again the next
815 * time. */
fe458402
PN
816 vencoding = UCHAR_MAX;
817 }
fe458402
PN
818 /* Must be non-zero by now */
819 assert(vencoding);
820 }
821
8361d6c4 822 /* Compare current entry with specified entry, do it only
823 * if vencoding != UCHAR_MAX because if there is no encoding
824 * possible for the field it can't be a valid integer. */
825 if (vencoding != UCHAR_MAX) {
fe458402
PN
826 long long ll = zipLoadInteger(q, encoding);
827 if (ll == vll) {
828 return p;
829 }
830 }
831 }
832
833 /* Reset skip count */
834 skipcnt = skip;
835 } else {
836 /* Skip entry */
837 skipcnt--;
838 }
839
840 /* Move to next entry */
841 p = q + len;
842 }
843
844 return NULL;
845}
846
6205b463
PN
847/* Return length of ziplist. */
848unsigned int ziplistLen(unsigned char *zl) {
849 unsigned int len = 0;
56538477 850 if (intrev16ifbe(ZIPLIST_LENGTH(zl)) < UINT16_MAX) {
851 len = intrev16ifbe(ZIPLIST_LENGTH(zl));
6205b463
PN
852 } else {
853 unsigned char *p = zl+ZIPLIST_HEADER_SIZE;
854 while (*p != ZIP_END) {
855 p += zipRawEntryLength(p);
856 len++;
857 }
858
859 /* Re-store length if small enough */
56538477 860 if (len < UINT16_MAX) ZIPLIST_LENGTH(zl) = intrev16ifbe(len);
6205b463
PN
861 }
862 return len;
863}
864
d4fb9f41 865/* Return ziplist blob size in bytes. */
866size_t ziplistBlobLen(unsigned char *zl) {
56538477 867 return intrev32ifbe(ZIPLIST_BYTES(zl));
4812cf28
PN
868}
869
11ac6ff6 870void ziplistRepr(unsigned char *zl) {
c8d9e7f4 871 unsigned char *p;
169d2ef1 872 int index = 0;
c8d9e7f4 873 zlentry entry;
11ac6ff6 874
169d2ef1
PN
875 printf(
876 "{total bytes %d} "
877 "{length %u}\n"
878 "{tail offset %u}\n",
56538477 879 intrev32ifbe(ZIPLIST_BYTES(zl)),
880 intrev16ifbe(ZIPLIST_LENGTH(zl)),
881 intrev32ifbe(ZIPLIST_TAIL_OFFSET(zl)));
1ce81fa5 882 p = ZIPLIST_ENTRY_HEAD(zl);
11ac6ff6 883 while(*p != ZIP_END) {
c8d9e7f4 884 entry = zipEntry(p);
169d2ef1
PN
885 printf(
886 "{"
887 "addr 0x%08lx, "
888 "index %2d, "
889 "offset %5ld, "
890 "rl: %5u, "
891 "hs %2u, "
892 "pl: %5u, "
893 "pls: %2u, "
894 "payload %5u"
895 "} ",
10c12171 896 (long unsigned)p,
169d2ef1 897 index,
10c12171 898 (unsigned long) (p-zl),
169d2ef1
PN
899 entry.headersize+entry.len,
900 entry.headersize,
901 entry.prevrawlen,
902 entry.prevrawlensize,
903 entry.len);
c8d9e7f4 904 p += entry.headersize;
c4705381 905 if (ZIP_IS_STR(entry.encoding)) {
169d2ef1 906 if (entry.len > 40) {
10c12171 907 if (fwrite(p,40,1,stdout) == 0) perror("fwrite");
169d2ef1
PN
908 printf("...");
909 } else {
10c12171 910 if (entry.len &&
911 fwrite(p,entry.len,1,stdout) == 0) perror("fwrite");
169d2ef1 912 }
29b14d5f 913 } else {
3688d7f3 914 printf("%lld", (long long) zipLoadInteger(p,entry.encoding));
29b14d5f 915 }
11ac6ff6 916 printf("\n");
c8d9e7f4 917 p += entry.len;
169d2ef1 918 index++;
11ac6ff6
PN
919 }
920 printf("{end}\n\n");
921}
922
923#ifdef ZIPLIST_TEST_MAIN
ffc15852 924#include <sys/time.h>
306c6a02
PN
925#include "adlist.h"
926#include "sds.h"
927
928#define debug(f, ...) { if (DEBUG) printf(f, __VA_ARGS__); }
11ac6ff6 929
08253bf4
PN
930unsigned char *createList() {
931 unsigned char *zl = ziplistNew();
b84186ff
PN
932 zl = ziplistPush(zl, (unsigned char*)"foo", 3, ZIPLIST_TAIL);
933 zl = ziplistPush(zl, (unsigned char*)"quux", 4, ZIPLIST_TAIL);
934 zl = ziplistPush(zl, (unsigned char*)"hello", 5, ZIPLIST_HEAD);
935 zl = ziplistPush(zl, (unsigned char*)"1024", 4, ZIPLIST_TAIL);
08253bf4
PN
936 return zl;
937}
938
29b14d5f
PN
939unsigned char *createIntList() {
940 unsigned char *zl = ziplistNew();
941 char buf[32];
942
943 sprintf(buf, "100");
b84186ff 944 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_TAIL);
29b14d5f 945 sprintf(buf, "128000");
b84186ff 946 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_TAIL);
29b14d5f 947 sprintf(buf, "-100");
b84186ff 948 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_HEAD);
29b14d5f 949 sprintf(buf, "4294967296");
b84186ff 950 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_HEAD);
29b14d5f 951 sprintf(buf, "non integer");
b84186ff 952 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_TAIL);
29b14d5f 953 sprintf(buf, "much much longer non integer");
b84186ff 954 zl = ziplistPush(zl, (unsigned char*)buf, strlen(buf), ZIPLIST_TAIL);
29b14d5f
PN
955 return zl;
956}
957
ffc15852
PN
958long long usec(void) {
959 struct timeval tv;
960 gettimeofday(&tv,NULL);
961 return (((long long)tv.tv_sec)*1000000)+tv.tv_usec;
962}
963
964void stress(int pos, int num, int maxsize, int dnum) {
965 int i,j,k;
966 unsigned char *zl;
967 char posstr[2][5] = { "HEAD", "TAIL" };
968 long long start;
969 for (i = 0; i < maxsize; i+=dnum) {
970 zl = ziplistNew();
971 for (j = 0; j < i; j++) {
972 zl = ziplistPush(zl,(unsigned char*)"quux",4,ZIPLIST_TAIL);
973 }
974
975 /* Do num times a push+pop from pos */
976 start = usec();
977 for (k = 0; k < num; k++) {
978 zl = ziplistPush(zl,(unsigned char*)"quux",4,pos);
979 zl = ziplistDeleteRange(zl,0,1);
980 }
981 printf("List size: %8d, bytes: %8d, %dx push+pop (%s): %6lld usec\n",
56538477 982 i,intrev32ifbe(ZIPLIST_BYTES(zl)),num,posstr[pos],usec()-start);
ffc15852
PN
983 zfree(zl);
984 }
985}
986
306974f5
PN
987void pop(unsigned char *zl, int where) {
988 unsigned char *p, *vstr;
989 unsigned int vlen;
990 long long vlong;
991
992 p = ziplistIndex(zl,where == ZIPLIST_HEAD ? 0 : -1);
993 if (ziplistGet(p,&vstr,&vlen,&vlong)) {
994 if (where == ZIPLIST_HEAD)
995 printf("Pop head: ");
996 else
997 printf("Pop tail: ");
998
999 if (vstr)
10c12171 1000 if (vlen && fwrite(vstr,vlen,1,stdout) == 0) perror("fwrite");
306974f5
PN
1001 else
1002 printf("%lld", vlong);
1003
1004 printf("\n");
1005 ziplistDeleteRange(zl,-1,1);
1006 } else {
1007 printf("ERROR: Could not pop\n");
1008 exit(1);
1009 }
1010}
1011
b7d3bf51 1012int randstring(char *target, unsigned int min, unsigned int max) {
306c6a02
PN
1013 int p, len = min+rand()%(max-min+1);
1014 int minval, maxval;
1015 switch(rand() % 3) {
1016 case 0:
1017 minval = 0;
1018 maxval = 255;
1019 break;
1020 case 1:
1021 minval = 48;
1022 maxval = 122;
1023 break;
1024 case 2:
1025 minval = 48;
1026 maxval = 52;
1027 break;
1028 default:
1029 assert(NULL);
1030 }
1031
1032 while(p < len)
1033 target[p++] = minval+rand()%(maxval-minval+1);
b7d3bf51 1034 return len;
306c6a02
PN
1035}
1036
89bf6f58
PN
1037void verify(unsigned char *zl, zlentry *e) {
1038 int i;
1039 int len = ziplistLen(zl);
1040 zlentry _e;
1041
1042 for (i = 0; i < len; i++) {
1043 memset(&e[i], 0, sizeof(zlentry));
1044 e[i] = zipEntry(ziplistIndex(zl, i));
1045
1046 memset(&_e, 0, sizeof(zlentry));
1047 _e = zipEntry(ziplistIndex(zl, -len+i));
1048
1049 assert(memcmp(&e[i], &_e, sizeof(zlentry)) == 0);
1050 }
1051}
1052
08253bf4 1053int main(int argc, char **argv) {
a24ba809 1054 unsigned char *zl, *p;
b84186ff 1055 unsigned char *entry;
335d16bc 1056 unsigned int elen;
75d8978e 1057 long long value;
08253bf4 1058
84403fe7
PN
1059 /* If an argument is given, use it as the random seed. */
1060 if (argc == 2)
1061 srand(atoi(argv[1]));
1062
29b14d5f
PN
1063 zl = createIntList();
1064 ziplistRepr(zl);
1065
08253bf4 1066 zl = createList();
11ac6ff6
PN
1067 ziplistRepr(zl);
1068
306974f5 1069 pop(zl,ZIPLIST_TAIL);
11ac6ff6
PN
1070 ziplistRepr(zl);
1071
306974f5 1072 pop(zl,ZIPLIST_HEAD);
11ac6ff6
PN
1073 ziplistRepr(zl);
1074
306974f5 1075 pop(zl,ZIPLIST_TAIL);
dcb9cf4e
PN
1076 ziplistRepr(zl);
1077
306974f5 1078 pop(zl,ZIPLIST_TAIL);
dcb9cf4e
PN
1079 ziplistRepr(zl);
1080
c03206fd
PN
1081 printf("Get element at index 3:\n");
1082 {
1083 zl = createList();
1084 p = ziplistIndex(zl, 3);
1085 if (!ziplistGet(p, &entry, &elen, &value)) {
1086 printf("ERROR: Could not access index 3\n");
1087 return 1;
1088 }
1089 if (entry) {
10c12171 1090 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
c03206fd
PN
1091 printf("\n");
1092 } else {
1093 printf("%lld\n", value);
1094 }
1095 printf("\n");
1096 }
1097
1098 printf("Get element at index 4 (out of range):\n");
1099 {
1100 zl = createList();
1101 p = ziplistIndex(zl, 4);
1102 if (p == NULL) {
1103 printf("No entry\n");
1104 } else {
1105 printf("ERROR: Out of range index should return NULL, returned offset: %ld\n", p-zl);
1106 return 1;
1107 }
1108 printf("\n");
1109 }
1110
1111 printf("Get element at index -1 (last element):\n");
1112 {
1113 zl = createList();
1114 p = ziplistIndex(zl, -1);
1115 if (!ziplistGet(p, &entry, &elen, &value)) {
1116 printf("ERROR: Could not access index -1\n");
1117 return 1;
1118 }
1119 if (entry) {
10c12171 1120 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
c03206fd
PN
1121 printf("\n");
1122 } else {
1123 printf("%lld\n", value);
1124 }
1125 printf("\n");
1126 }
1127
1128 printf("Get element at index -4 (first element):\n");
1129 {
1130 zl = createList();
1131 p = ziplistIndex(zl, -4);
1132 if (!ziplistGet(p, &entry, &elen, &value)) {
1133 printf("ERROR: Could not access index -4\n");
1134 return 1;
1135 }
1136 if (entry) {
10c12171 1137 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
c03206fd
PN
1138 printf("\n");
1139 } else {
1140 printf("%lld\n", value);
1141 }
1142 printf("\n");
1143 }
1144
1145 printf("Get element at index -5 (reverse out of range):\n");
1146 {
1147 zl = createList();
1148 p = ziplistIndex(zl, -5);
1149 if (p == NULL) {
1150 printf("No entry\n");
1151 } else {
1152 printf("ERROR: Out of range index should return NULL, returned offset: %ld\n", p-zl);
1153 return 1;
1154 }
1155 printf("\n");
1156 }
1157
08253bf4
PN
1158 printf("Iterate list from 0 to end:\n");
1159 {
1160 zl = createList();
1161 p = ziplistIndex(zl, 0);
75d8978e 1162 while (ziplistGet(p, &entry, &elen, &value)) {
335d16bc 1163 printf("Entry: ");
75d8978e 1164 if (entry) {
10c12171 1165 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
75d8978e
PN
1166 } else {
1167 printf("%lld", value);
1168 }
8632fb30 1169 p = ziplistNext(zl,p);
75d8978e 1170 printf("\n");
08253bf4
PN
1171 }
1172 printf("\n");
1173 }
1174
1175 printf("Iterate list from 1 to end:\n");
1176 {
1177 zl = createList();
1178 p = ziplistIndex(zl, 1);
75d8978e 1179 while (ziplistGet(p, &entry, &elen, &value)) {
335d16bc 1180 printf("Entry: ");
75d8978e 1181 if (entry) {
10c12171 1182 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
75d8978e
PN
1183 } else {
1184 printf("%lld", value);
1185 }
8632fb30 1186 p = ziplistNext(zl,p);
75d8978e 1187 printf("\n");
08253bf4
PN
1188 }
1189 printf("\n");
1190 }
1191
1192 printf("Iterate list from 2 to end:\n");
1193 {
1194 zl = createList();
1195 p = ziplistIndex(zl, 2);
75d8978e 1196 while (ziplistGet(p, &entry, &elen, &value)) {
335d16bc 1197 printf("Entry: ");
75d8978e 1198 if (entry) {
10c12171 1199 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
75d8978e
PN
1200 } else {
1201 printf("%lld", value);
1202 }
8632fb30 1203 p = ziplistNext(zl,p);
75d8978e 1204 printf("\n");
08253bf4
PN
1205 }
1206 printf("\n");
1207 }
1208
1209 printf("Iterate starting out of range:\n");
1210 {
1211 zl = createList();
75d8978e
PN
1212 p = ziplistIndex(zl, 4);
1213 if (!ziplistGet(p, &entry, &elen, &value)) {
08253bf4
PN
1214 printf("No entry\n");
1215 } else {
1216 printf("ERROR\n");
1217 }
779deb60
PN
1218 printf("\n");
1219 }
1220
0f3dfa87
PN
1221 printf("Iterate from back to front:\n");
1222 {
1223 zl = createList();
1224 p = ziplistIndex(zl, -1);
1225 while (ziplistGet(p, &entry, &elen, &value)) {
1226 printf("Entry: ");
1227 if (entry) {
10c12171 1228 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
0f3dfa87
PN
1229 } else {
1230 printf("%lld", value);
1231 }
8632fb30 1232 p = ziplistPrev(zl,p);
0f3dfa87
PN
1233 printf("\n");
1234 }
1235 printf("\n");
1236 }
1237
1238 printf("Iterate from back to front, deleting all items:\n");
1239 {
1240 zl = createList();
1241 p = ziplistIndex(zl, -1);
1242 while (ziplistGet(p, &entry, &elen, &value)) {
1243 printf("Entry: ");
1244 if (entry) {
10c12171 1245 if (elen && fwrite(entry,elen,1,stdout) == 0) perror("fwrite");
0f3dfa87
PN
1246 } else {
1247 printf("%lld", value);
1248 }
8632fb30
PN
1249 zl = ziplistDelete(zl,&p);
1250 p = ziplistPrev(zl,p);
0f3dfa87
PN
1251 printf("\n");
1252 }
1253 printf("\n");
1254 }
1255
779deb60
PN
1256 printf("Delete inclusive range 0,0:\n");
1257 {
1258 zl = createList();
ba5b4bde 1259 zl = ziplistDeleteRange(zl, 0, 1);
779deb60
PN
1260 ziplistRepr(zl);
1261 }
1262
1263 printf("Delete inclusive range 0,1:\n");
1264 {
1265 zl = createList();
ba5b4bde 1266 zl = ziplistDeleteRange(zl, 0, 2);
779deb60
PN
1267 ziplistRepr(zl);
1268 }
1269
1270 printf("Delete inclusive range 1,2:\n");
1271 {
1272 zl = createList();
ba5b4bde 1273 zl = ziplistDeleteRange(zl, 1, 2);
779deb60
PN
1274 ziplistRepr(zl);
1275 }
1276
1277 printf("Delete with start index out of range:\n");
1278 {
1279 zl = createList();
ba5b4bde 1280 zl = ziplistDeleteRange(zl, 5, 1);
779deb60
PN
1281 ziplistRepr(zl);
1282 }
1283
1284 printf("Delete with num overflow:\n");
1285 {
1286 zl = createList();
ba5b4bde 1287 zl = ziplistDeleteRange(zl, 1, 5);
779deb60 1288 ziplistRepr(zl);
08253bf4
PN
1289 }
1290
0f10458c
PN
1291 printf("Delete foo while iterating:\n");
1292 {
1293 zl = createList();
b84186ff
PN
1294 p = ziplistIndex(zl,0);
1295 while (ziplistGet(p,&entry,&elen,&value)) {
1296 if (entry && strncmp("foo",(char*)entry,elen) == 0) {
0f10458c 1297 printf("Delete foo\n");
b84186ff 1298 zl = ziplistDelete(zl,&p);
0f10458c
PN
1299 } else {
1300 printf("Entry: ");
75d8978e 1301 if (entry) {
10c12171 1302 if (elen && fwrite(entry,elen,1,stdout) == 0)
1303 perror("fwrite");
75d8978e 1304 } else {
b84186ff 1305 printf("%lld",value);
75d8978e 1306 }
b84186ff 1307 p = ziplistNext(zl,p);
75d8978e 1308 printf("\n");
0f10458c
PN
1309 }
1310 }
1311 printf("\n");
1312 ziplistRepr(zl);
c09c2c3b
PN
1313 }
1314
b0d605c1
PN
1315 printf("Regression test for >255 byte strings:\n");
1316 {
1317 char v1[257],v2[257];
1318 memset(v1,'x',256);
1319 memset(v2,'y',256);
1320 zl = ziplistNew();
1321 zl = ziplistPush(zl,(unsigned char*)v1,strlen(v1),ZIPLIST_TAIL);
1322 zl = ziplistPush(zl,(unsigned char*)v2,strlen(v2),ZIPLIST_TAIL);
1323
1324 /* Pop values again and compare their value. */
1325 p = ziplistIndex(zl,0);
1326 assert(ziplistGet(p,&entry,&elen,&value));
306c6a02 1327 assert(strncmp(v1,(char*)entry,elen) == 0);
b0d605c1
PN
1328 p = ziplistIndex(zl,1);
1329 assert(ziplistGet(p,&entry,&elen,&value));
306c6a02 1330 assert(strncmp(v2,(char*)entry,elen) == 0);
b0d605c1
PN
1331 printf("SUCCESS\n\n");
1332 }
1333
89bf6f58
PN
1334 printf("Regression test deleting next to last entries:\n");
1335 {
1336 char v[3][257];
1337 zlentry e[3];
1338 int i;
1339
1340 for (i = 0; i < (sizeof(v)/sizeof(v[0])); i++) {
1341 memset(v[i], 'a' + i, sizeof(v[0]));
1342 }
1343
1344 v[0][256] = '\0';
1345 v[1][ 1] = '\0';
1346 v[2][256] = '\0';
1347
1348 zl = ziplistNew();
1349 for (i = 0; i < (sizeof(v)/sizeof(v[0])); i++) {
1350 zl = ziplistPush(zl, (unsigned char *) v[i], strlen(v[i]), ZIPLIST_TAIL);
1351 }
1352
1353 verify(zl, e);
1354
1355 assert(e[0].prevrawlensize == 1);
1356 assert(e[1].prevrawlensize == 5);
1357 assert(e[2].prevrawlensize == 1);
1358
1359 /* Deleting entry 1 will increase `prevrawlensize` for entry 2 */
1360 unsigned char *p = e[1].p;
1361 zl = ziplistDelete(zl, &p);
1362
1363 verify(zl, e);
1364
1365 assert(e[0].prevrawlensize == 1);
1366 assert(e[1].prevrawlensize == 5);
1367
1368 printf("SUCCESS\n\n");
1369 }
1370
dbaa41c6
PN
1371 printf("Create long list and check indices:\n");
1372 {
1373 zl = ziplistNew();
1374 char buf[32];
1375 int i,len;
1376 for (i = 0; i < 1000; i++) {
1377 len = sprintf(buf,"%d",i);
b84186ff 1378 zl = ziplistPush(zl,(unsigned char*)buf,len,ZIPLIST_TAIL);
dbaa41c6
PN
1379 }
1380 for (i = 0; i < 1000; i++) {
1381 p = ziplistIndex(zl,i);
1382 assert(ziplistGet(p,NULL,NULL,&value));
1383 assert(i == value);
1384
1385 p = ziplistIndex(zl,-i-1);
1386 assert(ziplistGet(p,NULL,NULL,&value));
1387 assert(999-i == value);
1388 }
1389 printf("SUCCESS\n\n");
1390 }
1391
c09c2c3b
PN
1392 printf("Compare strings with ziplist entries:\n");
1393 {
1394 zl = createList();
b84186ff
PN
1395 p = ziplistIndex(zl,0);
1396 if (!ziplistCompare(p,(unsigned char*)"hello",5)) {
dcb9cf4e 1397 printf("ERROR: not \"hello\"\n");
a24ba809 1398 return 1;
c09c2c3b 1399 }
b84186ff 1400 if (ziplistCompare(p,(unsigned char*)"hella",5)) {
dcb9cf4e 1401 printf("ERROR: \"hella\"\n");
a24ba809 1402 return 1;
c09c2c3b
PN
1403 }
1404
b84186ff
PN
1405 p = ziplistIndex(zl,3);
1406 if (!ziplistCompare(p,(unsigned char*)"1024",4)) {
dcb9cf4e 1407 printf("ERROR: not \"1024\"\n");
a24ba809 1408 return 1;
c09c2c3b 1409 }
b84186ff 1410 if (ziplistCompare(p,(unsigned char*)"1025",4)) {
dcb9cf4e 1411 printf("ERROR: \"1025\"\n");
a24ba809 1412 return 1;
c09c2c3b 1413 }
169d2ef1
PN
1414 printf("SUCCESS\n\n");
1415 }
1416
1417 printf("Stress with random payloads of different encoding:\n");
1418 {
306c6a02 1419 int i,j,len,where;
169d2ef1 1420 unsigned char *p;
306c6a02 1421 char buf[1024];
b7d3bf51 1422 int buflen;
306c6a02
PN
1423 list *ref;
1424 listNode *refnode;
1425
1426 /* Hold temp vars from ziplist */
1427 unsigned char *sstr;
1428 unsigned int slen;
1429 long long sval;
1430
306c6a02
PN
1431 for (i = 0; i < 20000; i++) {
1432 zl = ziplistNew();
1433 ref = listCreate();
1434 listSetFreeMethod(ref,sdsfree);
1435 len = rand() % 256;
1436
1437 /* Create lists */
1438 for (j = 0; j < len; j++) {
1439 where = (rand() & 1) ? ZIPLIST_HEAD : ZIPLIST_TAIL;
b7d3bf51
PN
1440 if (rand() % 2) {
1441 buflen = randstring(buf,1,sizeof(buf)-1);
1442 } else {
1443 switch(rand() % 3) {
1444 case 0:
1445 buflen = sprintf(buf,"%lld",(0LL + rand()) >> 20);
1446 break;
1447 case 1:
1448 buflen = sprintf(buf,"%lld",(0LL + rand()));
1449 break;
1450 case 2:
1451 buflen = sprintf(buf,"%lld",(0LL + rand()) << 20);
1452 break;
1453 default:
1454 assert(NULL);
1455 }
306c6a02
PN
1456 }
1457
1458 /* Add to ziplist */
b7d3bf51 1459 zl = ziplistPush(zl, (unsigned char*)buf, buflen, where);
169d2ef1 1460
306c6a02
PN
1461 /* Add to reference list */
1462 if (where == ZIPLIST_HEAD) {
b7d3bf51 1463 listAddNodeHead(ref,sdsnewlen(buf, buflen));
306c6a02 1464 } else if (where == ZIPLIST_TAIL) {
b7d3bf51 1465 listAddNodeTail(ref,sdsnewlen(buf, buflen));
306c6a02
PN
1466 } else {
1467 assert(NULL);
1468 }
169d2ef1
PN
1469 }
1470
306c6a02
PN
1471 assert(listLength(ref) == ziplistLen(zl));
1472 for (j = 0; j < len; j++) {
1473 /* Naive way to get elements, but similar to the stresser
1474 * executed from the Tcl test suite. */
1475 p = ziplistIndex(zl,j);
1476 refnode = listIndex(ref,j);
1477
1478 assert(ziplistGet(p,&sstr,&slen,&sval));
1479 if (sstr == NULL) {
b7d3bf51 1480 buflen = sprintf(buf,"%lld",sval);
306c6a02 1481 } else {
b7d3bf51
PN
1482 buflen = slen;
1483 memcpy(buf,sstr,buflen);
1484 buf[buflen] = '\0';
306c6a02 1485 }
b7d3bf51 1486 assert(memcmp(buf,listNodeValue(refnode),buflen) == 0);
306c6a02
PN
1487 }
1488 zfree(zl);
1489 listRelease(ref);
169d2ef1
PN
1490 }
1491 printf("SUCCESS\n\n");
0f10458c
PN
1492 }
1493
ffc15852
PN
1494 printf("Stress with variable ziplist size:\n");
1495 {
1496 stress(ZIPLIST_HEAD,100000,16384,256);
1497 stress(ZIPLIST_TAIL,100000,16384,256);
1498 }
1499
11ac6ff6
PN
1500 return 0;
1501}
ffc15852 1502
11ac6ff6 1503#endif