]> git.saurik.com Git - redis.git/blob - src/redis.h
new preloading implemented, still EXEC not handled correctly, everything to test
[redis.git] / src / redis.h
1 #ifndef __REDIS_H
2 #define __REDIS_H
3
4 #include "fmacros.h"
5 #include "config.h"
6
7 #if defined(__sun)
8 #include "solarisfixes.h"
9 #endif
10
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <time.h>
15 #include <limits.h>
16 #include <unistd.h>
17 #include <errno.h>
18 #include <inttypes.h>
19 #include <pthread.h>
20 #include <syslog.h>
21
22 #include "ae.h" /* Event driven programming library */
23 #include "sds.h" /* Dynamic safe strings */
24 #include "dict.h" /* Hash tables */
25 #include "adlist.h" /* Linked lists */
26 #include "zmalloc.h" /* total memory usage aware version of malloc/free */
27 #include "anet.h" /* Networking the easy way */
28 #include "zipmap.h" /* Compact string -> string data structure */
29 #include "ziplist.h" /* Compact list data structure */
30 #include "intset.h" /* Compact integer set structure */
31 #include "version.h"
32
33 /* Error codes */
34 #define REDIS_OK 0
35 #define REDIS_ERR -1
36
37 /* Static server configuration */
38 #define REDIS_SERVERPORT 6379 /* TCP port */
39 #define REDIS_MAXIDLETIME (60*5) /* default client timeout */
40 #define REDIS_IOBUF_LEN 1024
41 #define REDIS_LOADBUF_LEN 1024
42 #define REDIS_STATIC_ARGS 8
43 #define REDIS_DEFAULT_DBNUM 16
44 #define REDIS_CONFIGLINE_MAX 1024
45 #define REDIS_MAX_SYNC_TIME 60 /* Slave can't take more to sync */
46 #define REDIS_EXPIRELOOKUPS_PER_CRON 10 /* lookup 10 expires per loop */
47 #define REDIS_MAX_WRITE_PER_EVENT (1024*64)
48 #define REDIS_REQUEST_MAX_SIZE (1024*1024*256) /* max bytes in inline command */
49 #define REDIS_SHARED_INTEGERS 10000
50 #define REDIS_REPLY_CHUNK_BYTES (5*1500) /* 5 TCP packets with default MTU */
51 #define REDIS_MAX_LOGMSG_LEN 1024 /* Default maximum length of syslog messages */
52
53 /* Hash table parameters */
54 #define REDIS_HT_MINFILL 10 /* Minimal hash table fill 10% */
55
56 /* Command flags:
57 * REDIS_CMD_DENYOOM:
58 * Commands marked with this flag will return an error when 'maxmemory' is
59 * set and the server is using more than 'maxmemory' bytes of memory.
60 * In short: commands with this flag are denied on low memory conditions.
61 * REDIS_CMD_FORCE_REPLICATION:
62 * Force replication even if dirty is 0. */
63 #define REDIS_CMD_DENYOOM 4
64 #define REDIS_CMD_FORCE_REPLICATION 8
65
66 /* Object types */
67 #define REDIS_STRING 0
68 #define REDIS_LIST 1
69 #define REDIS_SET 2
70 #define REDIS_ZSET 3
71 #define REDIS_HASH 4
72 #define REDIS_VMPOINTER 8
73 /* Object types only used for persistence in .rdb files */
74 #define REDIS_HASH_ZIPMAP 9
75 #define REDIS_LIST_ZIPLIST 10
76 #define REDIS_SET_INTSET 11
77
78 /* Objects encoding. Some kind of objects like Strings and Hashes can be
79 * internally represented in multiple ways. The 'encoding' field of the object
80 * is set to one of this fields for this object. */
81 #define REDIS_ENCODING_RAW 0 /* Raw representation */
82 #define REDIS_ENCODING_INT 1 /* Encoded as integer */
83 #define REDIS_ENCODING_HT 2 /* Encoded as hash table */
84 #define REDIS_ENCODING_ZIPMAP 3 /* Encoded as zipmap */
85 #define REDIS_ENCODING_LINKEDLIST 4 /* Encoded as regular linked list */
86 #define REDIS_ENCODING_ZIPLIST 5 /* Encoded as ziplist */
87 #define REDIS_ENCODING_INTSET 6 /* Encoded as intset */
88
89 /* Object types only used for dumping to disk */
90 #define REDIS_EXPIRETIME 253
91 #define REDIS_SELECTDB 254
92 #define REDIS_EOF 255
93
94 /* Defines related to the dump file format. To store 32 bits lengths for short
95 * keys requires a lot of space, so we check the most significant 2 bits of
96 * the first byte to interpreter the length:
97 *
98 * 00|000000 => if the two MSB are 00 the len is the 6 bits of this byte
99 * 01|000000 00000000 => 01, the len is 14 byes, 6 bits + 8 bits of next byte
100 * 10|000000 [32 bit integer] => if it's 01, a full 32 bit len will follow
101 * 11|000000 this means: specially encoded object will follow. The six bits
102 * number specify the kind of object that follows.
103 * See the REDIS_RDB_ENC_* defines.
104 *
105 * Lenghts up to 63 are stored using a single byte, most DB keys, and may
106 * values, will fit inside. */
107 #define REDIS_RDB_6BITLEN 0
108 #define REDIS_RDB_14BITLEN 1
109 #define REDIS_RDB_32BITLEN 2
110 #define REDIS_RDB_ENCVAL 3
111 #define REDIS_RDB_LENERR UINT_MAX
112
113 /* When a length of a string object stored on disk has the first two bits
114 * set, the remaining two bits specify a special encoding for the object
115 * accordingly to the following defines: */
116 #define REDIS_RDB_ENC_INT8 0 /* 8 bit signed integer */
117 #define REDIS_RDB_ENC_INT16 1 /* 16 bit signed integer */
118 #define REDIS_RDB_ENC_INT32 2 /* 32 bit signed integer */
119 #define REDIS_RDB_ENC_LZF 3 /* string compressed with FASTLZ */
120
121 /* Scheduled IO opeations flags. */
122 #define REDIS_IO_LOAD 1
123 #define REDIS_IO_SAVE 2
124 #define REDIS_IO_LOADINPROG 4
125 #define REDIS_IO_SAVEINPROG 8
126
127 /* Generic IO flags */
128 #define REDIS_IO_ONLYLOADS 1
129 #define REDIS_IO_ASAP 2
130
131 #define REDIS_MAX_COMPLETED_JOBS_PROCESSED 1
132 #define REDIS_THREAD_STACK_SIZE (1024*1024*4)
133
134 /* Client flags */
135 #define REDIS_SLAVE 1 /* This client is a slave server */
136 #define REDIS_MASTER 2 /* This client is a master server */
137 #define REDIS_MONITOR 4 /* This client is a slave monitor, see MONITOR */
138 #define REDIS_MULTI 8 /* This client is in a MULTI context */
139 #define REDIS_BLOCKED 16 /* The client is waiting in a blocking operation */
140 #define REDIS_IO_WAIT 32 /* The client is waiting for Virtual Memory I/O */
141 #define REDIS_DIRTY_CAS 64 /* Watched keys modified. EXEC will fail. */
142 #define REDIS_CLOSE_AFTER_REPLY 128 /* Close after writing entire reply. */
143 #define REDIS_UNBLOCKED 256 /* This client was unblocked and is stored in
144 server.unblocked_clients */
145
146 /* Client request types */
147 #define REDIS_REQ_INLINE 1
148 #define REDIS_REQ_MULTIBULK 2
149
150 /* Slave replication state - slave side */
151 #define REDIS_REPL_NONE 0 /* No active replication */
152 #define REDIS_REPL_CONNECT 1 /* Must connect to master */
153 #define REDIS_REPL_TRANSFER 2 /* Receiving .rdb from master */
154 #define REDIS_REPL_CONNECTED 3 /* Connected to master */
155
156 /* Slave replication state - from the point of view of master
157 * Note that in SEND_BULK and ONLINE state the slave receives new updates
158 * in its output queue. In the WAIT_BGSAVE state instead the server is waiting
159 * to start the next background saving in order to send updates to it. */
160 #define REDIS_REPL_WAIT_BGSAVE_START 3 /* master waits bgsave to start feeding it */
161 #define REDIS_REPL_WAIT_BGSAVE_END 4 /* master waits bgsave to start bulk DB transmission */
162 #define REDIS_REPL_SEND_BULK 5 /* master is sending the bulk DB */
163 #define REDIS_REPL_ONLINE 6 /* bulk DB already transmitted, receive updates */
164
165 /* List related stuff */
166 #define REDIS_HEAD 0
167 #define REDIS_TAIL 1
168
169 /* Sort operations */
170 #define REDIS_SORT_GET 0
171 #define REDIS_SORT_ASC 1
172 #define REDIS_SORT_DESC 2
173 #define REDIS_SORTKEY_MAX 1024
174
175 /* Log levels */
176 #define REDIS_DEBUG 0
177 #define REDIS_VERBOSE 1
178 #define REDIS_NOTICE 2
179 #define REDIS_WARNING 3
180
181 /* Anti-warning macro... */
182 #define REDIS_NOTUSED(V) ((void) V)
183
184 #define ZSKIPLIST_MAXLEVEL 32 /* Should be enough for 2^32 elements */
185 #define ZSKIPLIST_P 0.25 /* Skiplist P = 1/4 */
186
187 /* Append only defines */
188 #define APPENDFSYNC_NO 0
189 #define APPENDFSYNC_ALWAYS 1
190 #define APPENDFSYNC_EVERYSEC 2
191
192 /* Zip structure related defaults */
193 #define REDIS_HASH_MAX_ZIPMAP_ENTRIES 512
194 #define REDIS_HASH_MAX_ZIPMAP_VALUE 64
195 #define REDIS_LIST_MAX_ZIPLIST_ENTRIES 512
196 #define REDIS_LIST_MAX_ZIPLIST_VALUE 64
197 #define REDIS_SET_MAX_INTSET_ENTRIES 512
198
199 /* Sets operations codes */
200 #define REDIS_OP_UNION 0
201 #define REDIS_OP_DIFF 1
202 #define REDIS_OP_INTER 2
203
204 /* Redis maxmemory strategies */
205 #define REDIS_MAXMEMORY_VOLATILE_LRU 0
206 #define REDIS_MAXMEMORY_VOLATILE_TTL 1
207 #define REDIS_MAXMEMORY_VOLATILE_RANDOM 2
208 #define REDIS_MAXMEMORY_ALLKEYS_LRU 3
209 #define REDIS_MAXMEMORY_ALLKEYS_RANDOM 4
210 #define REDIS_MAXMEMORY_NO_EVICTION 5
211
212 /* Diskstore background saving thread states */
213 #define REDIS_BGSAVE_THREAD_UNACTIVE 0
214 #define REDIS_BGSAVE_THREAD_ACTIVE 1
215 #define REDIS_BGSAVE_THREAD_DONE_OK 2
216 #define REDIS_BGSAVE_THREAD_DONE_ERR 3
217
218 /* We can print the stacktrace, so our assert is defined this way: */
219 #define redisAssert(_e) ((_e)?(void)0 : (_redisAssert(#_e,__FILE__,__LINE__),_exit(1)))
220 #define redisPanic(_e) _redisPanic(#_e,__FILE__,__LINE__),_exit(1)
221 void _redisAssert(char *estr, char *file, int line);
222 void _redisPanic(char *msg, char *file, int line);
223
224 /*-----------------------------------------------------------------------------
225 * Data types
226 *----------------------------------------------------------------------------*/
227
228 /* A redis object, that is a type able to hold a string / list / set */
229
230 /* The actual Redis Object */
231 #define REDIS_LRU_CLOCK_MAX ((1<<21)-1) /* Max value of obj->lru */
232 #define REDIS_LRU_CLOCK_RESOLUTION 10 /* LRU clock resolution in seconds */
233 typedef struct redisObject {
234 unsigned type:4;
235 unsigned notused:2; /* Not used */
236 unsigned encoding:4;
237 unsigned lru:22; /* lru time (relative to server.lruclock) */
238 int refcount;
239 void *ptr;
240 /* VM fields are only allocated if VM is active, otherwise the
241 * object allocation function will just allocate
242 * sizeof(redisObjct) minus sizeof(redisObjectVM), so using
243 * Redis without VM active will not have any overhead. */
244 } robj;
245
246 /* The VM pointer structure - identifies an object in the swap file.
247 *
248 * This object is stored in place of the value
249 * object in the main key->value hash table representing a database.
250 * Note that the first fields (type, storage) are the same as the redisObject
251 * structure so that vmPointer strucuters can be accessed even when casted
252 * as redisObject structures.
253 *
254 * This is useful as we don't know if a value object is or not on disk, but we
255 * are always able to read obj->storage to check this. For vmPointer
256 * structures "type" is set to REDIS_VMPOINTER (even if without this field
257 * is still possible to check the kind of object from the value of 'storage').*/
258 typedef struct vmPointer {
259 unsigned type:4;
260 unsigned storage:2; /* REDIS_VM_SWAPPED or REDIS_VM_LOADING */
261 unsigned notused:26;
262 unsigned int vtype; /* type of the object stored in the swap file */
263 off_t page; /* the page at witch the object is stored on disk */
264 off_t usedpages; /* number of pages used on disk */
265 } vmpointer;
266
267 /* Macro used to initalize a Redis object allocated on the stack.
268 * Note that this macro is taken near the structure definition to make sure
269 * we'll update it when the structure is changed, to avoid bugs like
270 * bug #85 introduced exactly in this way. */
271 #define initStaticStringObject(_var,_ptr) do { \
272 _var.refcount = 1; \
273 _var.type = REDIS_STRING; \
274 _var.encoding = REDIS_ENCODING_RAW; \
275 _var.ptr = _ptr; \
276 } while(0);
277
278 typedef struct redisDb {
279 dict *dict; /* The keyspace for this DB */
280 dict *expires; /* Timeout of keys with a timeout set */
281 dict *blocking_keys; /* Keys with clients waiting for data (BLPOP) */
282 dict *io_keys; /* Keys with clients waiting for DS I/O */
283 dict *io_negcache; /* Negative caching for disk store */
284 dict *io_queued; /* Queued IO operations hash table */
285 dict *watched_keys; /* WATCHED keys for MULTI/EXEC CAS */
286 int id;
287 } redisDb;
288
289 /* Client MULTI/EXEC state */
290 typedef struct multiCmd {
291 robj **argv;
292 int argc;
293 struct redisCommand *cmd;
294 } multiCmd;
295
296 typedef struct multiState {
297 multiCmd *commands; /* Array of MULTI commands */
298 int count; /* Total number of MULTI commands */
299 } multiState;
300
301 typedef struct blockingState {
302 robj **keys; /* The key we are waiting to terminate a blocking
303 * operation such as BLPOP. Otherwise NULL. */
304 int count; /* Number of blocking keys */
305 time_t timeout; /* Blocking operation timeout. If UNIX current time
306 * is >= timeout then the operation timed out. */
307 robj *target; /* The key that should receive the element,
308 * for BRPOPLPUSH. */
309 } blockingState;
310
311 /* With multiplexing we need to take per-clinet state.
312 * Clients are taken in a liked list. */
313 typedef struct redisClient {
314 int fd;
315 redisDb *db;
316 int dictid;
317 sds querybuf;
318 int argc;
319 robj **argv;
320 int reqtype;
321 int multibulklen; /* number of multi bulk arguments left to read */
322 long bulklen; /* length of bulk argument in multi bulk request */
323 list *reply;
324 int sentlen;
325 time_t lastinteraction; /* time of the last interaction, used for timeout */
326 int flags; /* REDIS_SLAVE | REDIS_MONITOR | REDIS_MULTI ... */
327 int slaveseldb; /* slave selected db, if this client is a slave */
328 int authenticated; /* when requirepass is non-NULL */
329 int replstate; /* replication state if this is a slave */
330 int repldbfd; /* replication DB file descriptor */
331 long repldboff; /* replication DB file offset */
332 off_t repldbsize; /* replication DB file size */
333 multiState mstate; /* MULTI/EXEC state */
334 blockingState bpop; /* blocking state */
335 list *io_keys; /* Keys this client is waiting to be loaded from the
336 * swap file in order to continue. */
337 list *watched_keys; /* Keys WATCHED for MULTI/EXEC CAS */
338 dict *pubsub_channels; /* channels a client is interested in (SUBSCRIBE) */
339 list *pubsub_patterns; /* patterns a client is interested in (SUBSCRIBE) */
340
341 /* Response buffer */
342 int bufpos;
343 char buf[REDIS_REPLY_CHUNK_BYTES];
344 } redisClient;
345
346 struct saveparam {
347 time_t seconds;
348 int changes;
349 };
350
351 struct sharedObjectsStruct {
352 robj *crlf, *ok, *err, *emptybulk, *czero, *cone, *cnegone, *pong, *space,
353 *colon, *nullbulk, *nullmultibulk, *queued,
354 *emptymultibulk, *wrongtypeerr, *nokeyerr, *syntaxerr, *sameobjecterr,
355 *outofrangeerr, *loadingerr, *plus,
356 *select0, *select1, *select2, *select3, *select4,
357 *select5, *select6, *select7, *select8, *select9,
358 *messagebulk, *pmessagebulk, *subscribebulk, *unsubscribebulk, *mbulk3,
359 *mbulk4, *psubscribebulk, *punsubscribebulk,
360 *integers[REDIS_SHARED_INTEGERS];
361 };
362
363 /* Global server state structure */
364 struct redisServer {
365 /* General */
366 pthread_t mainthread;
367 redisDb *db;
368 dict *commands; /* Command table hahs table */
369 aeEventLoop *el;
370 /* Networking */
371 int port;
372 char *bindaddr;
373 char *unixsocket;
374 int ipfd;
375 int sofd;
376 list *clients;
377 list *slaves, *monitors;
378 char neterr[ANET_ERR_LEN];
379 /* RDB / AOF loading information */
380 int loading;
381 off_t loading_total_bytes;
382 off_t loading_loaded_bytes;
383 time_t loading_start_time;
384 /* Fast pointers to often looked up command */
385 struct redisCommand *delCommand, *multiCommand;
386 int cronloops; /* number of times the cron function run */
387 time_t lastsave; /* Unix time of last save succeeede */
388 /* Fields used only for stats */
389 time_t stat_starttime; /* server start time */
390 long long stat_numcommands; /* number of processed commands */
391 long long stat_numconnections; /* number of connections received */
392 long long stat_expiredkeys; /* number of expired keys */
393 long long stat_evictedkeys; /* number of evicted keys (maxmemory) */
394 long long stat_keyspace_hits; /* number of successful lookups of keys */
395 long long stat_keyspace_misses; /* number of failed lookups of keys */
396 /* Configuration */
397 int verbosity;
398 int maxidletime;
399 int dbnum;
400 int daemonize;
401 int appendonly;
402 int appendfsync;
403 int no_appendfsync_on_rewrite;
404 int shutdown_asap;
405 int activerehashing;
406 char *requirepass;
407 /* Persistence */
408 long long dirty; /* changes to DB from the last save */
409 long long dirty_before_bgsave; /* used to restore dirty on failed BGSAVE */
410 time_t lastfsync;
411 int appendfd;
412 int appendseldb;
413 char *pidfile;
414 pid_t bgsavechildpid;
415 pid_t bgrewritechildpid;
416 int bgsavethread_state;
417 pthread_mutex_t bgsavethread_mutex;
418 pthread_t bgsavethread;
419 sds bgrewritebuf; /* buffer taken by parent during oppend only rewrite */
420 sds aofbuf; /* AOF buffer, written before entering the event loop */
421 struct saveparam *saveparams;
422 int saveparamslen;
423 char *dbfilename;
424 int rdbcompression;
425 char *appendfilename;
426 /* Logging */
427 char *logfile;
428 int syslog_enabled;
429 char *syslog_ident;
430 int syslog_facility;
431 /* Replication related */
432 int isslave;
433 /* Slave specific fields */
434 char *masterauth;
435 char *masterhost;
436 int masterport;
437 redisClient *master; /* client that is master for this slave */
438 int replstate; /* replication status if the instance is a slave */
439 off_t repl_transfer_left; /* bytes left reading .rdb */
440 int repl_transfer_s; /* slave -> master SYNC socket */
441 int repl_transfer_fd; /* slave -> master SYNC temp file descriptor */
442 char *repl_transfer_tmpfile; /* slave-> master SYNC temp file name */
443 time_t repl_transfer_lastio; /* unix time of the latest read, for timeout */
444 int repl_serve_stale_data; /* Serve stale data when link is down? */
445 /* Limits */
446 unsigned int maxclients;
447 unsigned long long maxmemory;
448 int maxmemory_policy;
449 int maxmemory_samples;
450 /* Blocked clients */
451 unsigned int bpop_blocked_clients;
452 unsigned int cache_blocked_clients;
453 list *unblocked_clients; /* list of clients to unblock before next loop */
454 list *cache_io_queue; /* IO operations queue */
455 int cache_flush_delay; /* seconds to wait before flushing keys */
456 /* Sort parameters - qsort_r() is only available under BSD so we
457 * have to take this state global, in order to pass it to sortCompare() */
458 int sort_desc;
459 int sort_alpha;
460 int sort_bypattern;
461 /* Virtual memory configuration */
462 int ds_enabled; /* backend disk in redis.conf */
463 char *ds_path; /* location of the disk store on disk */
464 unsigned long long cache_max_memory;
465 /* Zip structure config */
466 size_t hash_max_zipmap_entries;
467 size_t hash_max_zipmap_value;
468 size_t list_max_ziplist_entries;
469 size_t list_max_ziplist_value;
470 size_t set_max_intset_entries;
471 time_t unixtime; /* Unix time sampled every second. */
472 /* Virtual memory I/O threads stuff */
473 /* An I/O thread process an element taken from the io_jobs queue and
474 * put the result of the operation in the io_done list. While the
475 * job is being processed, it's put on io_processing queue. */
476 list *io_newjobs; /* List of VM I/O jobs yet to be processed */
477 list *io_processing; /* List of VM I/O jobs being processed */
478 list *io_processed; /* List of VM I/O jobs already processed */
479 list *io_ready_clients; /* Clients ready to be unblocked. All keys loaded */
480 pthread_mutex_t io_mutex; /* lock to access io_jobs/io_done/io_thread_job */
481 pthread_cond_t io_condvar; /* I/O threads conditional variable */
482 pthread_attr_t io_threads_attr; /* attributes for threads creation */
483 int io_active_threads; /* Number of running I/O threads */
484 int vm_max_threads; /* Max number of I/O threads running at the same time */
485 /* Our main thread is blocked on the event loop, locking for sockets ready
486 * to be read or written, so when a threaded I/O operation is ready to be
487 * processed by the main thread, the I/O thread will use a unix pipe to
488 * awake the main thread. The followings are the two pipe FDs. */
489 int io_ready_pipe_read;
490 int io_ready_pipe_write;
491 /* Virtual memory stats */
492 unsigned long long vm_stats_used_pages;
493 unsigned long long vm_stats_swapped_objects;
494 unsigned long long vm_stats_swapouts;
495 unsigned long long vm_stats_swapins;
496 /* Pubsub */
497 dict *pubsub_channels; /* Map channels to list of subscribed clients */
498 list *pubsub_patterns; /* A list of pubsub_patterns */
499 /* Misc */
500 unsigned lruclock:22; /* clock incrementing every minute, for LRU */
501 unsigned lruclock_padding:10;
502 };
503
504 typedef struct pubsubPattern {
505 redisClient *client;
506 robj *pattern;
507 } pubsubPattern;
508
509 typedef void redisCommandProc(redisClient *c);
510 typedef int *redisGetKeysProc(struct redisCommand *cmd, robj **argv, int argc, int *numkeys, int flags);
511 struct redisCommand {
512 char *name;
513 redisCommandProc *proc;
514 int arity;
515 int flags;
516 /* Use a function to determine keys arguments in a command line.
517 * Used both for diskstore preloading and Redis Cluster. */
518 redisGetKeysProc *getkeys_proc;
519 /* What keys should be loaded in background when calling this command? */
520 int firstkey; /* The first argument that's a key (0 = no keys) */
521 int lastkey; /* THe last argument that's a key */
522 int keystep; /* The step between first and last key */
523 long long microseconds, calls;
524 };
525
526 struct redisFunctionSym {
527 char *name;
528 unsigned long pointer;
529 };
530
531 typedef struct _redisSortObject {
532 robj *obj;
533 union {
534 double score;
535 robj *cmpobj;
536 } u;
537 } redisSortObject;
538
539 typedef struct _redisSortOperation {
540 int type;
541 robj *pattern;
542 } redisSortOperation;
543
544 /* ZSETs use a specialized version of Skiplists */
545 typedef struct zskiplistNode {
546 robj *obj;
547 double score;
548 struct zskiplistNode *backward;
549 struct zskiplistLevel {
550 struct zskiplistNode *forward;
551 unsigned int span;
552 } level[];
553 } zskiplistNode;
554
555 typedef struct zskiplist {
556 struct zskiplistNode *header, *tail;
557 unsigned long length;
558 int level;
559 } zskiplist;
560
561 typedef struct zset {
562 dict *dict;
563 zskiplist *zsl;
564 } zset;
565
566 /* DIsk store threaded I/O request message */
567 #define REDIS_IOJOB_LOAD 0
568 #define REDIS_IOJOB_SAVE 1
569
570 typedef struct iojob {
571 int type; /* Request type, REDIS_IOJOB_* */
572 redisDb *db;/* Redis database */
573 robj *key; /* This I/O request is about this key */
574 robj *val; /* the value to swap for REDIS_IOJOB_SAVE, otherwise this
575 * field is populated by the I/O thread for REDIS_IOJOB_LOAD. */
576 time_t expire; /* Expire time for this key on REDIS_IOJOB_LOAD */
577 } iojob;
578
579 /* IO operations scheduled -- check dscache.c for more info */
580 typedef struct ioop {
581 int type;
582 redisDb *db;
583 robj *key;
584 time_t ctime; /* This is the creation time of the entry. */
585 } ioop;
586
587 /* Structure to hold list iteration abstraction. */
588 typedef struct {
589 robj *subject;
590 unsigned char encoding;
591 unsigned char direction; /* Iteration direction */
592 unsigned char *zi;
593 listNode *ln;
594 } listTypeIterator;
595
596 /* Structure for an entry while iterating over a list. */
597 typedef struct {
598 listTypeIterator *li;
599 unsigned char *zi; /* Entry in ziplist */
600 listNode *ln; /* Entry in linked list */
601 } listTypeEntry;
602
603 /* Structure to hold set iteration abstraction. */
604 typedef struct {
605 robj *subject;
606 int encoding;
607 int ii; /* intset iterator */
608 dictIterator *di;
609 } setTypeIterator;
610
611 /* Structure to hold hash iteration abstration. Note that iteration over
612 * hashes involves both fields and values. Because it is possible that
613 * not both are required, store pointers in the iterator to avoid
614 * unnecessary memory allocation for fields/values. */
615 typedef struct {
616 int encoding;
617 unsigned char *zi;
618 unsigned char *zk, *zv;
619 unsigned int zklen, zvlen;
620
621 dictIterator *di;
622 dictEntry *de;
623 } hashTypeIterator;
624
625 #define REDIS_HASH_KEY 1
626 #define REDIS_HASH_VALUE 2
627
628 /*-----------------------------------------------------------------------------
629 * Extern declarations
630 *----------------------------------------------------------------------------*/
631
632 extern struct redisServer server;
633 extern struct sharedObjectsStruct shared;
634 extern dictType setDictType;
635 extern dictType zsetDictType;
636 extern double R_Zero, R_PosInf, R_NegInf, R_Nan;
637 dictType hashDictType;
638
639 /*-----------------------------------------------------------------------------
640 * Functions prototypes
641 *----------------------------------------------------------------------------*/
642
643 /* Utils */
644 long long ustime(void);
645
646 /* networking.c -- Networking and Client related operations */
647 redisClient *createClient(int fd);
648 void closeTimedoutClients(void);
649 void freeClient(redisClient *c);
650 void resetClient(redisClient *c);
651 void sendReplyToClient(aeEventLoop *el, int fd, void *privdata, int mask);
652 void addReply(redisClient *c, robj *obj);
653 void *addDeferredMultiBulkLength(redisClient *c);
654 void setDeferredMultiBulkLength(redisClient *c, void *node, long length);
655 void addReplySds(redisClient *c, sds s);
656 void processInputBuffer(redisClient *c);
657 void acceptTcpHandler(aeEventLoop *el, int fd, void *privdata, int mask);
658 void acceptUnixHandler(aeEventLoop *el, int fd, void *privdata, int mask);
659 void readQueryFromClient(aeEventLoop *el, int fd, void *privdata, int mask);
660 void addReplyBulk(redisClient *c, robj *obj);
661 void addReplyBulkCString(redisClient *c, char *s);
662 void addReplyBulkCBuffer(redisClient *c, void *p, size_t len);
663 void addReplyBulkLongLong(redisClient *c, long long ll);
664 void acceptHandler(aeEventLoop *el, int fd, void *privdata, int mask);
665 void addReply(redisClient *c, robj *obj);
666 void addReplySds(redisClient *c, sds s);
667 void addReplyError(redisClient *c, char *err);
668 void addReplyStatus(redisClient *c, char *status);
669 void addReplyDouble(redisClient *c, double d);
670 void addReplyLongLong(redisClient *c, long long ll);
671 void addReplyMultiBulkLen(redisClient *c, long length);
672 void *dupClientReplyValue(void *o);
673 void getClientsMaxBuffers(unsigned long *longest_output_list,
674 unsigned long *biggest_input_buffer);
675
676 #ifdef __GNUC__
677 void addReplyErrorFormat(redisClient *c, const char *fmt, ...)
678 __attribute__((format(printf, 2, 3)));
679 void addReplyStatusFormat(redisClient *c, const char *fmt, ...)
680 __attribute__((format(printf, 2, 3)));
681 #else
682 void addReplyErrorFormat(redisClient *c, const char *fmt, ...);
683 void addReplyStatusFormat(redisClient *c, const char *fmt, ...);
684 #endif
685
686 /* List data type */
687 void listTypeTryConversion(robj *subject, robj *value);
688 void listTypePush(robj *subject, robj *value, int where);
689 robj *listTypePop(robj *subject, int where);
690 unsigned long listTypeLength(robj *subject);
691 listTypeIterator *listTypeInitIterator(robj *subject, int index, unsigned char direction);
692 void listTypeReleaseIterator(listTypeIterator *li);
693 int listTypeNext(listTypeIterator *li, listTypeEntry *entry);
694 robj *listTypeGet(listTypeEntry *entry);
695 void listTypeInsert(listTypeEntry *entry, robj *value, int where);
696 int listTypeEqual(listTypeEntry *entry, robj *o);
697 void listTypeDelete(listTypeEntry *entry);
698 void listTypeConvert(robj *subject, int enc);
699 void unblockClientWaitingData(redisClient *c);
700 int handleClientsWaitingListPush(redisClient *c, robj *key, robj *ele);
701 void popGenericCommand(redisClient *c, int where);
702
703 /* MULTI/EXEC/WATCH... */
704 void unwatchAllKeys(redisClient *c);
705 void initClientMultiState(redisClient *c);
706 void freeClientMultiState(redisClient *c);
707 void queueMultiCommand(redisClient *c, struct redisCommand *cmd);
708 void touchWatchedKey(redisDb *db, robj *key);
709 void touchWatchedKeysOnFlush(int dbid);
710
711 /* Redis object implementation */
712 void decrRefCount(void *o);
713 void incrRefCount(robj *o);
714 void freeStringObject(robj *o);
715 void freeListObject(robj *o);
716 void freeSetObject(robj *o);
717 void freeZsetObject(robj *o);
718 void freeHashObject(robj *o);
719 robj *createObject(int type, void *ptr);
720 robj *createStringObject(char *ptr, size_t len);
721 robj *dupStringObject(robj *o);
722 robj *tryObjectEncoding(robj *o);
723 robj *getDecodedObject(robj *o);
724 size_t stringObjectLen(robj *o);
725 robj *createStringObjectFromLongLong(long long value);
726 robj *createListObject(void);
727 robj *createZiplistObject(void);
728 robj *createSetObject(void);
729 robj *createIntsetObject(void);
730 robj *createHashObject(void);
731 robj *createZsetObject(void);
732 int getLongFromObjectOrReply(redisClient *c, robj *o, long *target, const char *msg);
733 int checkType(redisClient *c, robj *o, int type);
734 int getLongLongFromObjectOrReply(redisClient *c, robj *o, long long *target, const char *msg);
735 int getDoubleFromObjectOrReply(redisClient *c, robj *o, double *target, const char *msg);
736 int getLongLongFromObject(robj *o, long long *target);
737 char *strEncoding(int encoding);
738 int compareStringObjects(robj *a, robj *b);
739 int equalStringObjects(robj *a, robj *b);
740 unsigned long estimateObjectIdleTime(robj *o);
741
742 /* Synchronous I/O with timeout */
743 int syncWrite(int fd, char *ptr, ssize_t size, int timeout);
744 int syncRead(int fd, char *ptr, ssize_t size, int timeout);
745 int syncReadLine(int fd, char *ptr, ssize_t size, int timeout);
746 int fwriteBulkString(FILE *fp, char *s, unsigned long len);
747 int fwriteBulkDouble(FILE *fp, double d);
748 int fwriteBulkLongLong(FILE *fp, long long l);
749 int fwriteBulkObject(FILE *fp, robj *obj);
750
751 /* Replication */
752 void replicationFeedSlaves(list *slaves, int dictid, robj **argv, int argc);
753 void replicationFeedMonitors(list *monitors, int dictid, robj **argv, int argc);
754 int syncWithMaster(void);
755 void updateSlavesWaitingBgsave(int bgsaveerr);
756 void replicationCron(void);
757
758 /* Generic persistence functions */
759 void startLoading(FILE *fp);
760 void loadingProgress(off_t pos);
761 void stopLoading(void);
762
763 /* RDB persistence */
764 int rdbLoad(char *filename);
765 int rdbSaveBackground(char *filename);
766 void rdbRemoveTempFile(pid_t childpid);
767 int rdbSave(char *filename);
768 int rdbSaveObject(FILE *fp, robj *o);
769 off_t rdbSavedObjectLen(robj *o);
770 off_t rdbSavedObjectPages(robj *o);
771 robj *rdbLoadObject(int type, FILE *fp);
772 void backgroundSaveDoneHandler(int exitcode, int bysignal);
773 int rdbSaveKeyValuePair(FILE *fp, robj *key, robj *val, time_t expireitme, time_t now);
774 int rdbLoadType(FILE *fp);
775 time_t rdbLoadTime(FILE *fp);
776 robj *rdbLoadStringObject(FILE *fp);
777 int rdbSaveType(FILE *fp, unsigned char type);
778 int rdbSaveLen(FILE *fp, uint32_t len);
779
780 /* AOF persistence */
781 void flushAppendOnlyFile(void);
782 void feedAppendOnlyFile(struct redisCommand *cmd, int dictid, robj **argv, int argc);
783 void aofRemoveTempFile(pid_t childpid);
784 int rewriteAppendOnlyFileBackground(void);
785 int loadAppendOnlyFile(char *filename);
786 void stopAppendOnly(void);
787 int startAppendOnly(void);
788 void backgroundRewriteDoneHandler(int exitcode, int bysignal);
789
790 /* Sorted sets data type */
791 zskiplist *zslCreate(void);
792 void zslFree(zskiplist *zsl);
793 zskiplistNode *zslInsert(zskiplist *zsl, double score, robj *obj);
794
795 /* Core functions */
796 void freeMemoryIfNeeded(void);
797 int processCommand(redisClient *c);
798 void setupSignalHandlers(void);
799 struct redisCommand *lookupCommand(sds name);
800 struct redisCommand *lookupCommandByCString(char *s);
801 void call(redisClient *c, struct redisCommand *cmd);
802 int prepareForShutdown();
803 void redisLog(int level, const char *fmt, ...);
804 void usage();
805 void updateDictResizePolicy(void);
806 int htNeedsResize(dict *dict);
807 void oom(const char *msg);
808 void populateCommandTable(void);
809 void resetCommandTableStats(void);
810
811 /* Disk store */
812 int dsOpen(void);
813 int dsClose(void);
814 int dsSet(redisDb *db, robj *key, robj *val, time_t expire);
815 robj *dsGet(redisDb *db, robj *key, time_t *expire);
816 int dsDel(redisDb *db, robj *key);
817 int dsExists(redisDb *db, robj *key);
818 void dsFlushDb(int dbid);
819 int dsRdbSaveBackground(char *filename);
820 int dsRdbSave(char *filename);
821
822 /* Disk Store Cache */
823 void dsInit(void);
824 void vmThreadedIOCompletedJob(aeEventLoop *el, int fd, void *privdata, int mask);
825 void lockThreadedIO(void);
826 void unlockThreadedIO(void);
827 void freeIOJob(iojob *j);
828 void queueIOJob(iojob *j);
829 void waitEmptyIOJobsQueue(void);
830 void processAllPendingIOJobs(void);
831 int blockClientOnSwappedKeys(redisClient *c, struct redisCommand *cmd);
832 int dontWaitForSwappedKey(redisClient *c, robj *key);
833 void handleClientsBlockedOnSwappedKey(redisDb *db, robj *key);
834 int cacheFreeOneEntry(void);
835 void cacheScheduleIOAddFlag(redisDb *db, robj *key, long flag);
836 void cacheScheduleIODelFlag(redisDb *db, robj *key, long flag);
837 int cacheScheduleIOGetFlags(redisDb *db, robj *key);
838 void cacheScheduleIO(redisDb *db, robj *key, int type);
839 void cacheCron(void);
840 int cacheKeyMayExist(redisDb *db, robj *key);
841 void cacheSetKeyMayExist(redisDb *db, robj *key);
842 void cacheSetKeyDoesNotExist(redisDb *db, robj *key);
843 void cacheForcePointInTime(void);
844
845 /* Set data type */
846 robj *setTypeCreate(robj *value);
847 int setTypeAdd(robj *subject, robj *value);
848 int setTypeRemove(robj *subject, robj *value);
849 int setTypeIsMember(robj *subject, robj *value);
850 setTypeIterator *setTypeInitIterator(robj *subject);
851 void setTypeReleaseIterator(setTypeIterator *si);
852 int setTypeNext(setTypeIterator *si, robj **objele, int64_t *llele);
853 robj *setTypeNextObject(setTypeIterator *si);
854 int setTypeRandomElement(robj *setobj, robj **objele, int64_t *llele);
855 unsigned long setTypeSize(robj *subject);
856 void setTypeConvert(robj *subject, int enc);
857
858 /* Hash data type */
859 void convertToRealHash(robj *o);
860 void hashTypeTryConversion(robj *subject, robj **argv, int start, int end);
861 void hashTypeTryObjectEncoding(robj *subject, robj **o1, robj **o2);
862 int hashTypeGet(robj *o, robj *key, robj **objval, unsigned char **v, unsigned int *vlen);
863 robj *hashTypeGetObject(robj *o, robj *key);
864 int hashTypeExists(robj *o, robj *key);
865 int hashTypeSet(robj *o, robj *key, robj *value);
866 int hashTypeDelete(robj *o, robj *key);
867 unsigned long hashTypeLength(robj *o);
868 hashTypeIterator *hashTypeInitIterator(robj *subject);
869 void hashTypeReleaseIterator(hashTypeIterator *hi);
870 int hashTypeNext(hashTypeIterator *hi);
871 int hashTypeCurrent(hashTypeIterator *hi, int what, robj **objval, unsigned char **v, unsigned int *vlen);
872 robj *hashTypeCurrentObject(hashTypeIterator *hi, int what);
873 robj *hashTypeLookupWriteOrCreate(redisClient *c, robj *key);
874
875 /* Pub / Sub */
876 int pubsubUnsubscribeAllChannels(redisClient *c, int notify);
877 int pubsubUnsubscribeAllPatterns(redisClient *c, int notify);
878 void freePubsubPattern(void *p);
879 int listMatchPubsubPattern(void *a, void *b);
880
881 /* Utility functions */
882 int stringmatchlen(const char *pattern, int patternLen,
883 const char *string, int stringLen, int nocase);
884 int stringmatch(const char *pattern, const char *string, int nocase);
885 long long memtoll(const char *p, int *err);
886 int ll2string(char *s, size_t len, long long value);
887 int isStringRepresentableAsLong(sds s, long *longval);
888 int isStringRepresentableAsLongLong(sds s, long long *longval);
889 int isObjectRepresentableAsLongLong(robj *o, long long *llongval);
890
891 /* Configuration */
892 void loadServerConfig(char *filename);
893 void appendServerSaveParams(time_t seconds, int changes);
894 void resetServerSaveParams();
895
896 /* db.c -- Keyspace access API */
897 int removeExpire(redisDb *db, robj *key);
898 void propagateExpire(redisDb *db, robj *key);
899 int expireIfNeeded(redisDb *db, robj *key);
900 time_t getExpire(redisDb *db, robj *key);
901 void setExpire(redisDb *db, robj *key, time_t when);
902 robj *lookupKey(redisDb *db, robj *key);
903 robj *lookupKeyRead(redisDb *db, robj *key);
904 robj *lookupKeyWrite(redisDb *db, robj *key);
905 robj *lookupKeyReadOrReply(redisClient *c, robj *key, robj *reply);
906 robj *lookupKeyWriteOrReply(redisClient *c, robj *key, robj *reply);
907 int dbAdd(redisDb *db, robj *key, robj *val);
908 int dbReplace(redisDb *db, robj *key, robj *val);
909 int dbExists(redisDb *db, robj *key);
910 robj *dbRandomKey(redisDb *db);
911 int dbDelete(redisDb *db, robj *key);
912 long long emptyDb();
913 int selectDb(redisClient *c, int id);
914 void signalModifiedKey(redisDb *db, robj *key);
915 void signalFlushedDb(int dbid);
916
917 /* API to get key arguments from commands */
918 #define REDIS_GETKEYS_ALL 0
919 #define REDIS_GETKEYS_PRELOAD 1
920 int *getKeysFromCommand(struct redisCommand *cmd, robj **argv, int argc, int *numkeys, int flags);
921 void getKeysFreeResult(int *result);
922 int *noPreloadGetKeys(struct redisCommand *cmd,robj **argv, int argc, int *numkeys, int flags);
923 int *renameGetKeys(struct redisCommand *cmd,robj **argv, int argc, int *numkeys, int flags);
924 int *zunionInterGetKeys(struct redisCommand *cmd,robj **argv, int argc, int *numkeys, int flags);
925
926 /* Git SHA1 */
927 char *redisGitSHA1(void);
928 char *redisGitDirty(void);
929
930 /* Commands prototypes */
931 void authCommand(redisClient *c);
932 void pingCommand(redisClient *c);
933 void echoCommand(redisClient *c);
934 void setCommand(redisClient *c);
935 void setnxCommand(redisClient *c);
936 void setexCommand(redisClient *c);
937 void getCommand(redisClient *c);
938 void delCommand(redisClient *c);
939 void existsCommand(redisClient *c);
940 void setbitCommand(redisClient *c);
941 void getbitCommand(redisClient *c);
942 void setrangeCommand(redisClient *c);
943 void getrangeCommand(redisClient *c);
944 void incrCommand(redisClient *c);
945 void decrCommand(redisClient *c);
946 void incrbyCommand(redisClient *c);
947 void decrbyCommand(redisClient *c);
948 void selectCommand(redisClient *c);
949 void randomkeyCommand(redisClient *c);
950 void keysCommand(redisClient *c);
951 void dbsizeCommand(redisClient *c);
952 void lastsaveCommand(redisClient *c);
953 void saveCommand(redisClient *c);
954 void bgsaveCommand(redisClient *c);
955 void bgrewriteaofCommand(redisClient *c);
956 void shutdownCommand(redisClient *c);
957 void moveCommand(redisClient *c);
958 void renameCommand(redisClient *c);
959 void renamenxCommand(redisClient *c);
960 void lpushCommand(redisClient *c);
961 void rpushCommand(redisClient *c);
962 void lpushxCommand(redisClient *c);
963 void rpushxCommand(redisClient *c);
964 void linsertCommand(redisClient *c);
965 void lpopCommand(redisClient *c);
966 void rpopCommand(redisClient *c);
967 void llenCommand(redisClient *c);
968 void lindexCommand(redisClient *c);
969 void lrangeCommand(redisClient *c);
970 void ltrimCommand(redisClient *c);
971 void typeCommand(redisClient *c);
972 void lsetCommand(redisClient *c);
973 void saddCommand(redisClient *c);
974 void sremCommand(redisClient *c);
975 void smoveCommand(redisClient *c);
976 void sismemberCommand(redisClient *c);
977 void scardCommand(redisClient *c);
978 void spopCommand(redisClient *c);
979 void srandmemberCommand(redisClient *c);
980 void sinterCommand(redisClient *c);
981 void sinterstoreCommand(redisClient *c);
982 void sunionCommand(redisClient *c);
983 void sunionstoreCommand(redisClient *c);
984 void sdiffCommand(redisClient *c);
985 void sdiffstoreCommand(redisClient *c);
986 void syncCommand(redisClient *c);
987 void flushdbCommand(redisClient *c);
988 void flushallCommand(redisClient *c);
989 void sortCommand(redisClient *c);
990 void lremCommand(redisClient *c);
991 void rpoplpushCommand(redisClient *c);
992 void infoCommand(redisClient *c);
993 void mgetCommand(redisClient *c);
994 void monitorCommand(redisClient *c);
995 void expireCommand(redisClient *c);
996 void expireatCommand(redisClient *c);
997 void getsetCommand(redisClient *c);
998 void ttlCommand(redisClient *c);
999 void persistCommand(redisClient *c);
1000 void slaveofCommand(redisClient *c);
1001 void debugCommand(redisClient *c);
1002 void msetCommand(redisClient *c);
1003 void msetnxCommand(redisClient *c);
1004 void zaddCommand(redisClient *c);
1005 void zincrbyCommand(redisClient *c);
1006 void zrangeCommand(redisClient *c);
1007 void zrangebyscoreCommand(redisClient *c);
1008 void zrevrangebyscoreCommand(redisClient *c);
1009 void zcountCommand(redisClient *c);
1010 void zrevrangeCommand(redisClient *c);
1011 void zcardCommand(redisClient *c);
1012 void zremCommand(redisClient *c);
1013 void zscoreCommand(redisClient *c);
1014 void zremrangebyscoreCommand(redisClient *c);
1015 void multiCommand(redisClient *c);
1016 void execCommand(redisClient *c);
1017 void discardCommand(redisClient *c);
1018 void blpopCommand(redisClient *c);
1019 void brpopCommand(redisClient *c);
1020 void brpoplpushCommand(redisClient *c);
1021 void appendCommand(redisClient *c);
1022 void strlenCommand(redisClient *c);
1023 void zrankCommand(redisClient *c);
1024 void zrevrankCommand(redisClient *c);
1025 void hsetCommand(redisClient *c);
1026 void hsetnxCommand(redisClient *c);
1027 void hgetCommand(redisClient *c);
1028 void hmsetCommand(redisClient *c);
1029 void hmgetCommand(redisClient *c);
1030 void hdelCommand(redisClient *c);
1031 void hlenCommand(redisClient *c);
1032 void zremrangebyrankCommand(redisClient *c);
1033 void zunionstoreCommand(redisClient *c);
1034 void zinterstoreCommand(redisClient *c);
1035 void hkeysCommand(redisClient *c);
1036 void hvalsCommand(redisClient *c);
1037 void hgetallCommand(redisClient *c);
1038 void hexistsCommand(redisClient *c);
1039 void configCommand(redisClient *c);
1040 void hincrbyCommand(redisClient *c);
1041 void subscribeCommand(redisClient *c);
1042 void unsubscribeCommand(redisClient *c);
1043 void psubscribeCommand(redisClient *c);
1044 void punsubscribeCommand(redisClient *c);
1045 void publishCommand(redisClient *c);
1046 void watchCommand(redisClient *c);
1047 void unwatchCommand(redisClient *c);
1048
1049 #if defined(__GNUC__)
1050 void *calloc(size_t count, size_t size) __attribute__ ((deprecated));
1051 void free(void *ptr) __attribute__ ((deprecated));
1052 void *malloc(size_t size) __attribute__ ((deprecated));
1053 void *realloc(void *ptr, size_t size) __attribute__ ((deprecated));
1054 #endif
1055
1056 #endif