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