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