]> git.saurik.com Git - redis.git/blame - redis.c
Integer encoding implemented in dump file. Doc updated
[redis.git] / redis.c
CommitLineData
ed9b544e 1/*
2 * Copyright (c) 2006-2009, Salvatore Sanfilippo <antirez at gmail dot com>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * * Redistributions of source code must retain the above copyright notice,
9 * this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * * Neither the name of Redis nor the names of its contributors may be used
14 * to endorse or promote products derived from this software without
15 * specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 * POSSIBILITY OF SUCH DAMAGE.
28 */
29
23d4709d 30#define REDIS_VERSION "0.09"
31
32#include "fmacros.h"
ed9b544e 33
34#include <stdio.h>
35#include <stdlib.h>
36#include <string.h>
37#include <time.h>
38#include <unistd.h>
39#include <signal.h>
40#include <sys/wait.h>
41#include <errno.h>
42#include <assert.h>
43#include <ctype.h>
44#include <stdarg.h>
45#include <inttypes.h>
46#include <arpa/inet.h>
47#include <sys/stat.h>
48#include <fcntl.h>
49#include <sys/time.h>
50#include <sys/resource.h>
f78fd11b 51#include <limits.h>
ed9b544e 52
53#include "ae.h" /* Event driven programming library */
54#include "sds.h" /* Dynamic safe strings */
55#include "anet.h" /* Networking the easy way */
56#include "dict.h" /* Hash tables */
57#include "adlist.h" /* Linked lists */
58#include "zmalloc.h" /* total memory usage aware version of malloc/free */
59
60/* Error codes */
61#define REDIS_OK 0
62#define REDIS_ERR -1
63
64/* Static server configuration */
65#define REDIS_SERVERPORT 6379 /* TCP port */
66#define REDIS_MAXIDLETIME (60*5) /* default client timeout */
67#define REDIS_QUERYBUF_LEN 1024
68#define REDIS_LOADBUF_LEN 1024
69#define REDIS_MAX_ARGS 16
70#define REDIS_DEFAULT_DBNUM 16
71#define REDIS_CONFIGLINE_MAX 1024
72#define REDIS_OBJFREELIST_MAX 1000000 /* Max number of objects to cache */
73#define REDIS_MAX_SYNC_TIME 60 /* Slave can't take more to sync */
74
75/* Hash table parameters */
76#define REDIS_HT_MINFILL 10 /* Minimal hash table fill 10% */
77#define REDIS_HT_MINSLOTS 16384 /* Never resize the HT under this */
78
79/* Command flags */
80#define REDIS_CMD_BULK 1
81#define REDIS_CMD_INLINE 2
82
83/* Object types */
84#define REDIS_STRING 0
85#define REDIS_LIST 1
86#define REDIS_SET 2
87#define REDIS_HASH 3
f78fd11b 88
89/* Object types only used for dumping to disk */
ed9b544e 90#define REDIS_SELECTDB 254
91#define REDIS_EOF 255
92
f78fd11b 93/* Defines related to the dump file format. To store 32 bits lengths for short
94 * keys requires a lot of space, so we check the most significant 2 bits of
95 * the first byte to interpreter the length:
96 *
97 * 00|000000 => if the two MSB are 00 the len is the 6 bits of this byte
98 * 01|000000 00000000 => 01, the len is 14 byes, 6 bits + 8 bits of next byte
99 * 10|000000 [32 bit integer] => if it's 01, a full 32 bit len will follow
a4d1ba9a 100 * 11|000000 this means: specially encoded object will follow. The six bits
101 * number specify the kind of object that follows.
102 * See the REDIS_RDB_ENC_* defines.
f78fd11b 103 *
10c43610 104 * Lenghts up to 63 are stored using a single byte, most DB keys, and may
105 * values, will fit inside. */
f78fd11b 106#define REDIS_RDB_6BITLEN 0
107#define REDIS_RDB_14BITLEN 1
108#define REDIS_RDB_32BITLEN 2
17be1a4a 109#define REDIS_RDB_ENCVAL 3
f78fd11b 110#define REDIS_RDB_LENERR UINT_MAX
111
a4d1ba9a 112/* When a length of a string object stored on disk has the first two bits
113 * set, the remaining two bits specify a special encoding for the object
114 * accordingly to the following defines: */
115#define REDIS_RDB_ENC_INT8 0 /* 8 bit signed integer */
116#define REDIS_RDB_ENC_INT16 1 /* 16 bit signed integer */
117#define REDIS_RDB_ENC_INT32 2 /* 32 bit signed integer */
118#define REDIS_RDB_ENC_FLZ 3 /* string compressed with FASTLZ */
119
ed9b544e 120/* Client flags */
121#define REDIS_CLOSE 1 /* This client connection should be closed ASAP */
122#define REDIS_SLAVE 2 /* This client is a slave server */
123#define REDIS_MASTER 4 /* This client is a master server */
87eca727 124#define REDIS_MONITOR 8 /* This client is a slave monitor, see MONITOR */
ed9b544e 125
126/* Server replication state */
127#define REDIS_REPL_NONE 0 /* No active replication */
128#define REDIS_REPL_CONNECT 1 /* Must connect to master */
129#define REDIS_REPL_CONNECTED 2 /* Connected to master */
130
131/* List related stuff */
132#define REDIS_HEAD 0
133#define REDIS_TAIL 1
134
135/* Sort operations */
136#define REDIS_SORT_GET 0
137#define REDIS_SORT_DEL 1
138#define REDIS_SORT_INCR 2
139#define REDIS_SORT_DECR 3
140#define REDIS_SORT_ASC 4
141#define REDIS_SORT_DESC 5
142#define REDIS_SORTKEY_MAX 1024
143
144/* Log levels */
145#define REDIS_DEBUG 0
146#define REDIS_NOTICE 1
147#define REDIS_WARNING 2
148
149/* Anti-warning macro... */
150#define REDIS_NOTUSED(V) ((void) V)
151
152/*================================= Data types ============================== */
153
154/* A redis object, that is a type able to hold a string / list / set */
155typedef struct redisObject {
156 int type;
157 void *ptr;
158 int refcount;
159} robj;
160
161/* With multiplexing we need to take per-clinet state.
162 * Clients are taken in a liked list. */
163typedef struct redisClient {
164 int fd;
165 dict *dict;
166 int dictid;
167 sds querybuf;
168 robj *argv[REDIS_MAX_ARGS];
169 int argc;
170 int bulklen; /* bulk read len. -1 if not in bulk read mode */
171 list *reply;
172 int sentlen;
173 time_t lastinteraction; /* time of the last interaction, used for timeout */
87eca727 174 int flags; /* REDIS_CLOSE | REDIS_SLAVE | REDIS_MONITOR */
ed9b544e 175 int slaveseldb; /* slave selected db, if this client is a slave */
abcb223e 176 int authenticated; /* when requirepass is non-NULL */
ed9b544e 177} redisClient;
178
179struct saveparam {
180 time_t seconds;
181 int changes;
182};
183
184/* Global server state structure */
185struct redisServer {
186 int port;
187 int fd;
188 dict **dict;
10c43610 189 dict *sharingpool;
190 unsigned int sharingpoolsize;
ed9b544e 191 long long dirty; /* changes to DB from the last save */
192 list *clients;
87eca727 193 list *slaves, *monitors;
ed9b544e 194 char neterr[ANET_ERR_LEN];
195 aeEventLoop *el;
196 int cronloops; /* number of times the cron function run */
197 list *objfreelist; /* A list of freed objects to avoid malloc() */
198 time_t lastsave; /* Unix time of last save succeeede */
199 int usedmemory; /* Used memory in megabytes */
200 /* Fields used only for stats */
201 time_t stat_starttime; /* server start time */
202 long long stat_numcommands; /* number of processed commands */
203 long long stat_numconnections; /* number of connections received */
204 /* Configuration */
205 int verbosity;
206 int glueoutputbuf;
207 int maxidletime;
208 int dbnum;
209 int daemonize;
ed329fcf 210 char *pidfile;
ed9b544e 211 int bgsaveinprogress;
212 struct saveparam *saveparams;
213 int saveparamslen;
214 char *logfile;
215 char *bindaddr;
216 char *dbfilename;
abcb223e 217 char *requirepass;
10c43610 218 int shareobjects;
ed9b544e 219 /* Replication related */
220 int isslave;
221 char *masterhost;
222 int masterport;
223 redisClient *master;
224 int replstate;
225 /* Sort parameters - qsort_r() is only available under BSD so we
226 * have to take this state global, in order to pass it to sortCompare() */
227 int sort_desc;
228 int sort_alpha;
229 int sort_bypattern;
230};
231
232typedef void redisCommandProc(redisClient *c);
233struct redisCommand {
234 char *name;
235 redisCommandProc *proc;
236 int arity;
237 int flags;
238};
239
240typedef struct _redisSortObject {
241 robj *obj;
242 union {
243 double score;
244 robj *cmpobj;
245 } u;
246} redisSortObject;
247
248typedef struct _redisSortOperation {
249 int type;
250 robj *pattern;
251} redisSortOperation;
252
253struct sharedObjectsStruct {
c937aa89 254 robj *crlf, *ok, *err, *emptybulk, *czero, *cone, *pong, *space,
7b45bfb2 255 *colon, *nullbulk, *nullmultibulk,
c937aa89 256 *emptymultibulk, *wrongtypeerr, *nokeyerr, *syntaxerr, *sameobjecterr,
257 *outofrangeerr, *plus,
ed9b544e 258 *select0, *select1, *select2, *select3, *select4,
259 *select5, *select6, *select7, *select8, *select9;
260} shared;
261
262/*================================ Prototypes =============================== */
263
264static void freeStringObject(robj *o);
265static void freeListObject(robj *o);
266static void freeSetObject(robj *o);
267static void decrRefCount(void *o);
268static robj *createObject(int type, void *ptr);
269static void freeClient(redisClient *c);
f78fd11b 270static int rdbLoad(char *filename);
ed9b544e 271static void addReply(redisClient *c, robj *obj);
272static void addReplySds(redisClient *c, sds s);
273static void incrRefCount(robj *o);
f78fd11b 274static int rdbSaveBackground(char *filename);
ed9b544e 275static robj *createStringObject(char *ptr, size_t len);
87eca727 276static void replicationFeedSlaves(list *slaves, struct redisCommand *cmd, int dictid, robj **argv, int argc);
ed9b544e 277static int syncWithMaster(void);
10c43610 278static robj *tryObjectSharing(robj *o);
ed9b544e 279
abcb223e 280static void authCommand(redisClient *c);
ed9b544e 281static void pingCommand(redisClient *c);
282static void echoCommand(redisClient *c);
283static void setCommand(redisClient *c);
284static void setnxCommand(redisClient *c);
285static void getCommand(redisClient *c);
286static void delCommand(redisClient *c);
287static void existsCommand(redisClient *c);
288static void incrCommand(redisClient *c);
289static void decrCommand(redisClient *c);
290static void incrbyCommand(redisClient *c);
291static void decrbyCommand(redisClient *c);
292static void selectCommand(redisClient *c);
293static void randomkeyCommand(redisClient *c);
294static void keysCommand(redisClient *c);
295static void dbsizeCommand(redisClient *c);
296static void lastsaveCommand(redisClient *c);
297static void saveCommand(redisClient *c);
298static void bgsaveCommand(redisClient *c);
299static void shutdownCommand(redisClient *c);
300static void moveCommand(redisClient *c);
301static void renameCommand(redisClient *c);
302static void renamenxCommand(redisClient *c);
303static void lpushCommand(redisClient *c);
304static void rpushCommand(redisClient *c);
305static void lpopCommand(redisClient *c);
306static void rpopCommand(redisClient *c);
307static void llenCommand(redisClient *c);
308static void lindexCommand(redisClient *c);
309static void lrangeCommand(redisClient *c);
310static void ltrimCommand(redisClient *c);
311static void typeCommand(redisClient *c);
312static void lsetCommand(redisClient *c);
313static void saddCommand(redisClient *c);
314static void sremCommand(redisClient *c);
315static void sismemberCommand(redisClient *c);
316static void scardCommand(redisClient *c);
317static void sinterCommand(redisClient *c);
318static void sinterstoreCommand(redisClient *c);
319static void syncCommand(redisClient *c);
320static void flushdbCommand(redisClient *c);
321static void flushallCommand(redisClient *c);
322static void sortCommand(redisClient *c);
323static void lremCommand(redisClient *c);
324static void infoCommand(redisClient *c);
70003d28 325static void mgetCommand(redisClient *c);
87eca727 326static void monitorCommand(redisClient *c);
ed9b544e 327
328/*================================= Globals ================================= */
329
330/* Global vars */
331static struct redisServer server; /* server global state */
332static struct redisCommand cmdTable[] = {
333 {"get",getCommand,2,REDIS_CMD_INLINE},
334 {"set",setCommand,3,REDIS_CMD_BULK},
335 {"setnx",setnxCommand,3,REDIS_CMD_BULK},
336 {"del",delCommand,2,REDIS_CMD_INLINE},
337 {"exists",existsCommand,2,REDIS_CMD_INLINE},
338 {"incr",incrCommand,2,REDIS_CMD_INLINE},
339 {"decr",decrCommand,2,REDIS_CMD_INLINE},
70003d28 340 {"mget",mgetCommand,-2,REDIS_CMD_INLINE},
ed9b544e 341 {"rpush",rpushCommand,3,REDIS_CMD_BULK},
342 {"lpush",lpushCommand,3,REDIS_CMD_BULK},
343 {"rpop",rpopCommand,2,REDIS_CMD_INLINE},
344 {"lpop",lpopCommand,2,REDIS_CMD_INLINE},
345 {"llen",llenCommand,2,REDIS_CMD_INLINE},
346 {"lindex",lindexCommand,3,REDIS_CMD_INLINE},
347 {"lset",lsetCommand,4,REDIS_CMD_BULK},
348 {"lrange",lrangeCommand,4,REDIS_CMD_INLINE},
349 {"ltrim",ltrimCommand,4,REDIS_CMD_INLINE},
350 {"lrem",lremCommand,4,REDIS_CMD_BULK},
351 {"sadd",saddCommand,3,REDIS_CMD_BULK},
352 {"srem",sremCommand,3,REDIS_CMD_BULK},
353 {"sismember",sismemberCommand,3,REDIS_CMD_BULK},
354 {"scard",scardCommand,2,REDIS_CMD_INLINE},
355 {"sinter",sinterCommand,-2,REDIS_CMD_INLINE},
356 {"sinterstore",sinterstoreCommand,-3,REDIS_CMD_INLINE},
357 {"smembers",sinterCommand,2,REDIS_CMD_INLINE},
358 {"incrby",incrbyCommand,3,REDIS_CMD_INLINE},
359 {"decrby",decrbyCommand,3,REDIS_CMD_INLINE},
360 {"randomkey",randomkeyCommand,1,REDIS_CMD_INLINE},
361 {"select",selectCommand,2,REDIS_CMD_INLINE},
362 {"move",moveCommand,3,REDIS_CMD_INLINE},
363 {"rename",renameCommand,3,REDIS_CMD_INLINE},
364 {"renamenx",renamenxCommand,3,REDIS_CMD_INLINE},
365 {"keys",keysCommand,2,REDIS_CMD_INLINE},
366 {"dbsize",dbsizeCommand,1,REDIS_CMD_INLINE},
abcb223e 367 {"auth",authCommand,2,REDIS_CMD_INLINE},
ed9b544e 368 {"ping",pingCommand,1,REDIS_CMD_INLINE},
369 {"echo",echoCommand,2,REDIS_CMD_BULK},
370 {"save",saveCommand,1,REDIS_CMD_INLINE},
371 {"bgsave",bgsaveCommand,1,REDIS_CMD_INLINE},
372 {"shutdown",shutdownCommand,1,REDIS_CMD_INLINE},
373 {"lastsave",lastsaveCommand,1,REDIS_CMD_INLINE},
374 {"type",typeCommand,2,REDIS_CMD_INLINE},
375 {"sync",syncCommand,1,REDIS_CMD_INLINE},
376 {"flushdb",flushdbCommand,1,REDIS_CMD_INLINE},
377 {"flushall",flushallCommand,1,REDIS_CMD_INLINE},
378 {"sort",sortCommand,-2,REDIS_CMD_INLINE},
379 {"info",infoCommand,1,REDIS_CMD_INLINE},
87eca727 380 {"monitor",monitorCommand,1,REDIS_CMD_INLINE},
ed9b544e 381 {NULL,NULL,0,0}
382};
383
384/*============================ Utility functions ============================ */
385
386/* Glob-style pattern matching. */
387int stringmatchlen(const char *pattern, int patternLen,
388 const char *string, int stringLen, int nocase)
389{
390 while(patternLen) {
391 switch(pattern[0]) {
392 case '*':
393 while (pattern[1] == '*') {
394 pattern++;
395 patternLen--;
396 }
397 if (patternLen == 1)
398 return 1; /* match */
399 while(stringLen) {
400 if (stringmatchlen(pattern+1, patternLen-1,
401 string, stringLen, nocase))
402 return 1; /* match */
403 string++;
404 stringLen--;
405 }
406 return 0; /* no match */
407 break;
408 case '?':
409 if (stringLen == 0)
410 return 0; /* no match */
411 string++;
412 stringLen--;
413 break;
414 case '[':
415 {
416 int not, match;
417
418 pattern++;
419 patternLen--;
420 not = pattern[0] == '^';
421 if (not) {
422 pattern++;
423 patternLen--;
424 }
425 match = 0;
426 while(1) {
427 if (pattern[0] == '\\') {
428 pattern++;
429 patternLen--;
430 if (pattern[0] == string[0])
431 match = 1;
432 } else if (pattern[0] == ']') {
433 break;
434 } else if (patternLen == 0) {
435 pattern--;
436 patternLen++;
437 break;
438 } else if (pattern[1] == '-' && patternLen >= 3) {
439 int start = pattern[0];
440 int end = pattern[2];
441 int c = string[0];
442 if (start > end) {
443 int t = start;
444 start = end;
445 end = t;
446 }
447 if (nocase) {
448 start = tolower(start);
449 end = tolower(end);
450 c = tolower(c);
451 }
452 pattern += 2;
453 patternLen -= 2;
454 if (c >= start && c <= end)
455 match = 1;
456 } else {
457 if (!nocase) {
458 if (pattern[0] == string[0])
459 match = 1;
460 } else {
461 if (tolower((int)pattern[0]) == tolower((int)string[0]))
462 match = 1;
463 }
464 }
465 pattern++;
466 patternLen--;
467 }
468 if (not)
469 match = !match;
470 if (!match)
471 return 0; /* no match */
472 string++;
473 stringLen--;
474 break;
475 }
476 case '\\':
477 if (patternLen >= 2) {
478 pattern++;
479 patternLen--;
480 }
481 /* fall through */
482 default:
483 if (!nocase) {
484 if (pattern[0] != string[0])
485 return 0; /* no match */
486 } else {
487 if (tolower((int)pattern[0]) != tolower((int)string[0]))
488 return 0; /* no match */
489 }
490 string++;
491 stringLen--;
492 break;
493 }
494 pattern++;
495 patternLen--;
496 if (stringLen == 0) {
497 while(*pattern == '*') {
498 pattern++;
499 patternLen--;
500 }
501 break;
502 }
503 }
504 if (patternLen == 0 && stringLen == 0)
505 return 1;
506 return 0;
507}
508
509void redisLog(int level, const char *fmt, ...)
510{
511 va_list ap;
512 FILE *fp;
513
514 fp = (server.logfile == NULL) ? stdout : fopen(server.logfile,"a");
515 if (!fp) return;
516
517 va_start(ap, fmt);
518 if (level >= server.verbosity) {
519 char *c = ".-*";
520 fprintf(fp,"%c ",c[level]);
521 vfprintf(fp, fmt, ap);
522 fprintf(fp,"\n");
523 fflush(fp);
524 }
525 va_end(ap);
526
527 if (server.logfile) fclose(fp);
528}
529
530/*====================== Hash table type implementation ==================== */
531
532/* This is an hash table type that uses the SDS dynamic strings libary as
533 * keys and radis objects as values (objects can hold SDS strings,
534 * lists, sets). */
535
536static int sdsDictKeyCompare(void *privdata, const void *key1,
537 const void *key2)
538{
539 int l1,l2;
540 DICT_NOTUSED(privdata);
541
542 l1 = sdslen((sds)key1);
543 l2 = sdslen((sds)key2);
544 if (l1 != l2) return 0;
545 return memcmp(key1, key2, l1) == 0;
546}
547
548static void dictRedisObjectDestructor(void *privdata, void *val)
549{
550 DICT_NOTUSED(privdata);
551
552 decrRefCount(val);
553}
554
555static int dictSdsKeyCompare(void *privdata, const void *key1,
556 const void *key2)
557{
558 const robj *o1 = key1, *o2 = key2;
559 return sdsDictKeyCompare(privdata,o1->ptr,o2->ptr);
560}
561
562static unsigned int dictSdsHash(const void *key) {
563 const robj *o = key;
564 return dictGenHashFunction(o->ptr, sdslen((sds)o->ptr));
565}
566
567static dictType setDictType = {
568 dictSdsHash, /* hash function */
569 NULL, /* key dup */
570 NULL, /* val dup */
571 dictSdsKeyCompare, /* key compare */
572 dictRedisObjectDestructor, /* key destructor */
573 NULL /* val destructor */
574};
575
576static dictType hashDictType = {
577 dictSdsHash, /* hash function */
578 NULL, /* key dup */
579 NULL, /* val dup */
580 dictSdsKeyCompare, /* key compare */
581 dictRedisObjectDestructor, /* key destructor */
582 dictRedisObjectDestructor /* val destructor */
583};
584
585/* ========================= Random utility functions ======================= */
586
587/* Redis generally does not try to recover from out of memory conditions
588 * when allocating objects or strings, it is not clear if it will be possible
589 * to report this condition to the client since the networking layer itself
590 * is based on heap allocation for send buffers, so we simply abort.
591 * At least the code will be simpler to read... */
592static void oom(const char *msg) {
593 fprintf(stderr, "%s: Out of memory\n",msg);
594 fflush(stderr);
595 sleep(1);
596 abort();
597}
598
599/* ====================== Redis server networking stuff ===================== */
600void closeTimedoutClients(void) {
601 redisClient *c;
602 listIter *li;
603 listNode *ln;
604 time_t now = time(NULL);
605
606 li = listGetIterator(server.clients,AL_START_HEAD);
607 if (!li) return;
608 while ((ln = listNextElement(li)) != NULL) {
609 c = listNodeValue(ln);
610 if (!(c->flags & REDIS_SLAVE) && /* no timeout for slaves */
611 (now - c->lastinteraction > server.maxidletime)) {
612 redisLog(REDIS_DEBUG,"Closing idle client");
613 freeClient(c);
614 }
615 }
616 listReleaseIterator(li);
617}
618
619int serverCron(struct aeEventLoop *eventLoop, long long id, void *clientData) {
620 int j, size, used, loops = server.cronloops++;
621 REDIS_NOTUSED(eventLoop);
622 REDIS_NOTUSED(id);
623 REDIS_NOTUSED(clientData);
624
625 /* Update the global state with the amount of used memory */
626 server.usedmemory = zmalloc_used_memory();
627
628 /* If the percentage of used slots in the HT reaches REDIS_HT_MINFILL
629 * we resize the hash table to save memory */
630 for (j = 0; j < server.dbnum; j++) {
631 size = dictGetHashTableSize(server.dict[j]);
632 used = dictGetHashTableUsed(server.dict[j]);
633 if (!(loops % 5) && used > 0) {
634 redisLog(REDIS_DEBUG,"DB %d: %d keys in %d slots HT.",j,used,size);
a4d1ba9a 635 /* dictPrintStats(server.dict); */
ed9b544e 636 }
637 if (size && used && size > REDIS_HT_MINSLOTS &&
638 (used*100/size < REDIS_HT_MINFILL)) {
639 redisLog(REDIS_NOTICE,"The hash table %d is too sparse, resize it...",j);
640 dictResize(server.dict[j]);
641 redisLog(REDIS_NOTICE,"Hash table %d resized.",j);
642 }
643 }
644
645 /* Show information about connected clients */
646 if (!(loops % 5)) {
647 redisLog(REDIS_DEBUG,"%d clients connected (%d slaves), %d bytes in use",
648 listLength(server.clients)-listLength(server.slaves),
649 listLength(server.slaves),
10c43610 650 server.usedmemory,
651 dictGetHashTableUsed(server.sharingpool));
ed9b544e 652 }
653
654 /* Close connections of timedout clients */
655 if (!(loops % 10))
656 closeTimedoutClients();
657
658 /* Check if a background saving in progress terminated */
659 if (server.bgsaveinprogress) {
660 int statloc;
661 if (wait4(-1,&statloc,WNOHANG,NULL)) {
662 int exitcode = WEXITSTATUS(statloc);
663 if (exitcode == 0) {
664 redisLog(REDIS_NOTICE,
665 "Background saving terminated with success");
666 server.dirty = 0;
667 server.lastsave = time(NULL);
668 } else {
669 redisLog(REDIS_WARNING,
670 "Background saving error");
671 }
672 server.bgsaveinprogress = 0;
673 }
674 } else {
675 /* If there is not a background saving in progress check if
676 * we have to save now */
677 time_t now = time(NULL);
678 for (j = 0; j < server.saveparamslen; j++) {
679 struct saveparam *sp = server.saveparams+j;
680
681 if (server.dirty >= sp->changes &&
682 now-server.lastsave > sp->seconds) {
683 redisLog(REDIS_NOTICE,"%d changes in %d seconds. Saving...",
684 sp->changes, sp->seconds);
f78fd11b 685 rdbSaveBackground(server.dbfilename);
ed9b544e 686 break;
687 }
688 }
689 }
690 /* Check if we should connect to a MASTER */
691 if (server.replstate == REDIS_REPL_CONNECT) {
692 redisLog(REDIS_NOTICE,"Connecting to MASTER...");
693 if (syncWithMaster() == REDIS_OK) {
694 redisLog(REDIS_NOTICE,"MASTER <-> SLAVE sync succeeded");
695 }
696 }
697 return 1000;
698}
699
700static void createSharedObjects(void) {
701 shared.crlf = createObject(REDIS_STRING,sdsnew("\r\n"));
702 shared.ok = createObject(REDIS_STRING,sdsnew("+OK\r\n"));
703 shared.err = createObject(REDIS_STRING,sdsnew("-ERR\r\n"));
c937aa89 704 shared.emptybulk = createObject(REDIS_STRING,sdsnew("$0\r\n\r\n"));
705 shared.czero = createObject(REDIS_STRING,sdsnew(":0\r\n"));
706 shared.cone = createObject(REDIS_STRING,sdsnew(":1\r\n"));
707 shared.nullbulk = createObject(REDIS_STRING,sdsnew("$-1\r\n"));
708 shared.nullmultibulk = createObject(REDIS_STRING,sdsnew("*-1\r\n"));
709 shared.emptymultibulk = createObject(REDIS_STRING,sdsnew("*0\r\n"));
ed9b544e 710 /* no such key */
ed9b544e 711 shared.pong = createObject(REDIS_STRING,sdsnew("+PONG\r\n"));
712 shared.wrongtypeerr = createObject(REDIS_STRING,sdsnew(
713 "-ERR Operation against a key holding the wrong kind of value\r\n"));
ed9b544e 714 shared.nokeyerr = createObject(REDIS_STRING,sdsnew(
715 "-ERR no such key\r\n"));
ed9b544e 716 shared.syntaxerr = createObject(REDIS_STRING,sdsnew(
717 "-ERR syntax error\r\n"));
c937aa89 718 shared.sameobjecterr = createObject(REDIS_STRING,sdsnew(
719 "-ERR source and destination objects are the same\r\n"));
720 shared.outofrangeerr = createObject(REDIS_STRING,sdsnew(
721 "-ERR index out of range\r\n"));
ed9b544e 722 shared.space = createObject(REDIS_STRING,sdsnew(" "));
c937aa89 723 shared.colon = createObject(REDIS_STRING,sdsnew(":"));
724 shared.plus = createObject(REDIS_STRING,sdsnew("+"));
ed9b544e 725 shared.select0 = createStringObject("select 0\r\n",10);
726 shared.select1 = createStringObject("select 1\r\n",10);
727 shared.select2 = createStringObject("select 2\r\n",10);
728 shared.select3 = createStringObject("select 3\r\n",10);
729 shared.select4 = createStringObject("select 4\r\n",10);
730 shared.select5 = createStringObject("select 5\r\n",10);
731 shared.select6 = createStringObject("select 6\r\n",10);
732 shared.select7 = createStringObject("select 7\r\n",10);
733 shared.select8 = createStringObject("select 8\r\n",10);
734 shared.select9 = createStringObject("select 9\r\n",10);
735}
736
737static void appendServerSaveParams(time_t seconds, int changes) {
738 server.saveparams = zrealloc(server.saveparams,sizeof(struct saveparam)*(server.saveparamslen+1));
739 if (server.saveparams == NULL) oom("appendServerSaveParams");
740 server.saveparams[server.saveparamslen].seconds = seconds;
741 server.saveparams[server.saveparamslen].changes = changes;
742 server.saveparamslen++;
743}
744
745static void ResetServerSaveParams() {
746 zfree(server.saveparams);
747 server.saveparams = NULL;
748 server.saveparamslen = 0;
749}
750
751static void initServerConfig() {
752 server.dbnum = REDIS_DEFAULT_DBNUM;
753 server.port = REDIS_SERVERPORT;
754 server.verbosity = REDIS_DEBUG;
755 server.maxidletime = REDIS_MAXIDLETIME;
756 server.saveparams = NULL;
757 server.logfile = NULL; /* NULL = log on standard output */
758 server.bindaddr = NULL;
759 server.glueoutputbuf = 1;
760 server.daemonize = 0;
ed329fcf 761 server.pidfile = "/var/run/redis.pid";
ed9b544e 762 server.dbfilename = "dump.rdb";
abcb223e 763 server.requirepass = NULL;
10c43610 764 server.shareobjects = 0;
ed9b544e 765 ResetServerSaveParams();
766
767 appendServerSaveParams(60*60,1); /* save after 1 hour and 1 change */
768 appendServerSaveParams(300,100); /* save after 5 minutes and 100 changes */
769 appendServerSaveParams(60,10000); /* save after 1 minute and 10000 changes */
770 /* Replication related */
771 server.isslave = 0;
772 server.masterhost = NULL;
773 server.masterport = 6379;
774 server.master = NULL;
775 server.replstate = REDIS_REPL_NONE;
776}
777
778static void initServer() {
779 int j;
780
781 signal(SIGHUP, SIG_IGN);
782 signal(SIGPIPE, SIG_IGN);
783
784 server.clients = listCreate();
785 server.slaves = listCreate();
87eca727 786 server.monitors = listCreate();
ed9b544e 787 server.objfreelist = listCreate();
788 createSharedObjects();
789 server.el = aeCreateEventLoop();
790 server.dict = zmalloc(sizeof(dict*)*server.dbnum);
10c43610 791 server.sharingpool = dictCreate(&setDictType,NULL);
792 server.sharingpoolsize = 1024;
87eca727 793 if (!server.dict || !server.clients || !server.slaves || !server.monitors || !server.el || !server.objfreelist)
ed9b544e 794 oom("server initialization"); /* Fatal OOM */
795 server.fd = anetTcpServer(server.neterr, server.port, server.bindaddr);
796 if (server.fd == -1) {
797 redisLog(REDIS_WARNING, "Opening TCP port: %s", server.neterr);
798 exit(1);
799 }
10c43610 800 for (j = 0; j < server.dbnum; j++)
ed9b544e 801 server.dict[j] = dictCreate(&hashDictType,NULL);
ed9b544e 802 server.cronloops = 0;
803 server.bgsaveinprogress = 0;
804 server.lastsave = time(NULL);
805 server.dirty = 0;
806 server.usedmemory = 0;
807 server.stat_numcommands = 0;
808 server.stat_numconnections = 0;
809 server.stat_starttime = time(NULL);
810 aeCreateTimeEvent(server.el, 1000, serverCron, NULL, NULL);
811}
812
813/* Empty the whole database */
814static void emptyDb() {
815 int j;
816
817 for (j = 0; j < server.dbnum; j++)
818 dictEmpty(server.dict[j]);
819}
820
821/* I agree, this is a very rudimental way to load a configuration...
822 will improve later if the config gets more complex */
823static void loadServerConfig(char *filename) {
824 FILE *fp = fopen(filename,"r");
825 char buf[REDIS_CONFIGLINE_MAX+1], *err = NULL;
826 int linenum = 0;
827 sds line = NULL;
828
829 if (!fp) {
830 redisLog(REDIS_WARNING,"Fatal error, can't open config file");
831 exit(1);
832 }
833 while(fgets(buf,REDIS_CONFIGLINE_MAX+1,fp) != NULL) {
834 sds *argv;
835 int argc, j;
836
837 linenum++;
838 line = sdsnew(buf);
839 line = sdstrim(line," \t\r\n");
840
841 /* Skip comments and blank lines*/
842 if (line[0] == '#' || line[0] == '\0') {
843 sdsfree(line);
844 continue;
845 }
846
847 /* Split into arguments */
848 argv = sdssplitlen(line,sdslen(line)," ",1,&argc);
849 sdstolower(argv[0]);
850
851 /* Execute config directives */
852 if (!strcmp(argv[0],"timeout") && argc == 2) {
853 server.maxidletime = atoi(argv[1]);
854 if (server.maxidletime < 1) {
855 err = "Invalid timeout value"; goto loaderr;
856 }
857 } else if (!strcmp(argv[0],"port") && argc == 2) {
858 server.port = atoi(argv[1]);
859 if (server.port < 1 || server.port > 65535) {
860 err = "Invalid port"; goto loaderr;
861 }
862 } else if (!strcmp(argv[0],"bind") && argc == 2) {
863 server.bindaddr = zstrdup(argv[1]);
864 } else if (!strcmp(argv[0],"save") && argc == 3) {
865 int seconds = atoi(argv[1]);
866 int changes = atoi(argv[2]);
867 if (seconds < 1 || changes < 0) {
868 err = "Invalid save parameters"; goto loaderr;
869 }
870 appendServerSaveParams(seconds,changes);
871 } else if (!strcmp(argv[0],"dir") && argc == 2) {
872 if (chdir(argv[1]) == -1) {
873 redisLog(REDIS_WARNING,"Can't chdir to '%s': %s",
874 argv[1], strerror(errno));
875 exit(1);
876 }
877 } else if (!strcmp(argv[0],"loglevel") && argc == 2) {
878 if (!strcmp(argv[1],"debug")) server.verbosity = REDIS_DEBUG;
879 else if (!strcmp(argv[1],"notice")) server.verbosity = REDIS_NOTICE;
880 else if (!strcmp(argv[1],"warning")) server.verbosity = REDIS_WARNING;
881 else {
882 err = "Invalid log level. Must be one of debug, notice, warning";
883 goto loaderr;
884 }
885 } else if (!strcmp(argv[0],"logfile") && argc == 2) {
886 FILE *fp;
887
888 server.logfile = zstrdup(argv[1]);
889 if (!strcmp(server.logfile,"stdout")) {
890 zfree(server.logfile);
891 server.logfile = NULL;
892 }
893 if (server.logfile) {
894 /* Test if we are able to open the file. The server will not
895 * be able to abort just for this problem later... */
896 fp = fopen(server.logfile,"a");
897 if (fp == NULL) {
898 err = sdscatprintf(sdsempty(),
899 "Can't open the log file: %s", strerror(errno));
900 goto loaderr;
901 }
902 fclose(fp);
903 }
904 } else if (!strcmp(argv[0],"databases") && argc == 2) {
905 server.dbnum = atoi(argv[1]);
906 if (server.dbnum < 1) {
907 err = "Invalid number of databases"; goto loaderr;
908 }
909 } else if (!strcmp(argv[0],"slaveof") && argc == 3) {
910 server.masterhost = sdsnew(argv[1]);
911 server.masterport = atoi(argv[2]);
912 server.replstate = REDIS_REPL_CONNECT;
913 } else if (!strcmp(argv[0],"glueoutputbuf") && argc == 2) {
914 sdstolower(argv[1]);
915 if (!strcmp(argv[1],"yes")) server.glueoutputbuf = 1;
916 else if (!strcmp(argv[1],"no")) server.glueoutputbuf = 0;
917 else {
918 err = "argument must be 'yes' or 'no'"; goto loaderr;
919 }
10c43610 920 } else if (!strcmp(argv[0],"shareobjects") && argc == 2) {
921 sdstolower(argv[1]);
922 if (!strcmp(argv[1],"yes")) server.shareobjects = 1;
923 else if (!strcmp(argv[1],"no")) server.shareobjects = 0;
924 else {
925 err = "argument must be 'yes' or 'no'"; goto loaderr;
926 }
ed9b544e 927 } else if (!strcmp(argv[0],"daemonize") && argc == 2) {
928 sdstolower(argv[1]);
929 if (!strcmp(argv[1],"yes")) server.daemonize = 1;
930 else if (!strcmp(argv[1],"no")) server.daemonize = 0;
931 else {
932 err = "argument must be 'yes' or 'no'"; goto loaderr;
933 }
abcb223e
BH
934 } else if (!strcmp(argv[0],"requirepass") && argc == 2) {
935 server.requirepass = zstrdup(argv[1]);
ed329fcf
LH
936 } else if (!strcmp(argv[0],"pidfile") && argc == 2) {
937 server.pidfile = zstrdup(argv[1]);
ed9b544e 938 } else {
939 err = "Bad directive or wrong number of arguments"; goto loaderr;
940 }
941 for (j = 0; j < argc; j++)
942 sdsfree(argv[j]);
943 zfree(argv);
944 sdsfree(line);
945 }
946 fclose(fp);
947 return;
948
949loaderr:
950 fprintf(stderr, "\n*** FATAL CONFIG FILE ERROR ***\n");
951 fprintf(stderr, "Reading the configuration file, at line %d\n", linenum);
952 fprintf(stderr, ">>> '%s'\n", line);
953 fprintf(stderr, "%s\n", err);
954 exit(1);
955}
956
957static void freeClientArgv(redisClient *c) {
958 int j;
959
960 for (j = 0; j < c->argc; j++)
961 decrRefCount(c->argv[j]);
962 c->argc = 0;
963}
964
965static void freeClient(redisClient *c) {
966 listNode *ln;
967
968 aeDeleteFileEvent(server.el,c->fd,AE_READABLE);
969 aeDeleteFileEvent(server.el,c->fd,AE_WRITABLE);
970 sdsfree(c->querybuf);
971 listRelease(c->reply);
972 freeClientArgv(c);
973 close(c->fd);
974 ln = listSearchKey(server.clients,c);
975 assert(ln != NULL);
976 listDelNode(server.clients,ln);
977 if (c->flags & REDIS_SLAVE) {
87eca727 978 list *l = (c->flags & REDIS_MONITOR) ? server.monitors : server.slaves;
979 ln = listSearchKey(l,c);
ed9b544e 980 assert(ln != NULL);
87eca727 981 listDelNode(l,ln);
ed9b544e 982 }
983 if (c->flags & REDIS_MASTER) {
984 server.master = NULL;
985 server.replstate = REDIS_REPL_CONNECT;
986 }
987 zfree(c);
988}
989
990static void glueReplyBuffersIfNeeded(redisClient *c) {
991 int totlen = 0;
992 listNode *ln = c->reply->head, *next;
993 robj *o;
994
995 while(ln) {
996 o = ln->value;
997 totlen += sdslen(o->ptr);
998 ln = ln->next;
999 /* This optimization makes more sense if we don't have to copy
1000 * too much data */
1001 if (totlen > 1024) return;
1002 }
1003 if (totlen > 0) {
1004 char buf[1024];
1005 int copylen = 0;
1006
1007 ln = c->reply->head;
1008 while(ln) {
1009 next = ln->next;
1010 o = ln->value;
1011 memcpy(buf+copylen,o->ptr,sdslen(o->ptr));
1012 copylen += sdslen(o->ptr);
1013 listDelNode(c->reply,ln);
1014 ln = next;
1015 }
1016 /* Now the output buffer is empty, add the new single element */
1017 addReplySds(c,sdsnewlen(buf,totlen));
1018 }
1019}
1020
1021static void sendReplyToClient(aeEventLoop *el, int fd, void *privdata, int mask) {
1022 redisClient *c = privdata;
1023 int nwritten = 0, totwritten = 0, objlen;
1024 robj *o;
1025 REDIS_NOTUSED(el);
1026 REDIS_NOTUSED(mask);
1027
1028 if (server.glueoutputbuf && listLength(c->reply) > 1)
1029 glueReplyBuffersIfNeeded(c);
1030 while(listLength(c->reply)) {
1031 o = listNodeValue(listFirst(c->reply));
1032 objlen = sdslen(o->ptr);
1033
1034 if (objlen == 0) {
1035 listDelNode(c->reply,listFirst(c->reply));
1036 continue;
1037 }
1038
1039 if (c->flags & REDIS_MASTER) {
1040 nwritten = objlen - c->sentlen;
1041 } else {
a4d1ba9a 1042 nwritten = write(fd, ((char*)o->ptr)+c->sentlen, objlen - c->sentlen);
ed9b544e 1043 if (nwritten <= 0) break;
1044 }
1045 c->sentlen += nwritten;
1046 totwritten += nwritten;
1047 /* If we fully sent the object on head go to the next one */
1048 if (c->sentlen == objlen) {
1049 listDelNode(c->reply,listFirst(c->reply));
1050 c->sentlen = 0;
1051 }
1052 }
1053 if (nwritten == -1) {
1054 if (errno == EAGAIN) {
1055 nwritten = 0;
1056 } else {
1057 redisLog(REDIS_DEBUG,
1058 "Error writing to client: %s", strerror(errno));
1059 freeClient(c);
1060 return;
1061 }
1062 }
1063 if (totwritten > 0) c->lastinteraction = time(NULL);
1064 if (listLength(c->reply) == 0) {
1065 c->sentlen = 0;
1066 aeDeleteFileEvent(server.el,c->fd,AE_WRITABLE);
1067 }
1068}
1069
1070static struct redisCommand *lookupCommand(char *name) {
1071 int j = 0;
1072 while(cmdTable[j].name != NULL) {
1073 if (!strcmp(name,cmdTable[j].name)) return &cmdTable[j];
1074 j++;
1075 }
1076 return NULL;
1077}
1078
1079/* resetClient prepare the client to process the next command */
1080static void resetClient(redisClient *c) {
1081 freeClientArgv(c);
1082 c->bulklen = -1;
1083}
1084
1085/* If this function gets called we already read a whole
1086 * command, argments are in the client argv/argc fields.
1087 * processCommand() execute the command or prepare the
1088 * server for a bulk read from the client.
1089 *
1090 * If 1 is returned the client is still alive and valid and
1091 * and other operations can be performed by the caller. Otherwise
1092 * if 0 is returned the client was destroied (i.e. after QUIT). */
1093static int processCommand(redisClient *c) {
1094 struct redisCommand *cmd;
1095 long long dirty;
1096
1097 sdstolower(c->argv[0]->ptr);
1098 /* The QUIT command is handled as a special case. Normal command
1099 * procs are unable to close the client connection safely */
1100 if (!strcmp(c->argv[0]->ptr,"quit")) {
1101 freeClient(c);
1102 return 0;
1103 }
1104 cmd = lookupCommand(c->argv[0]->ptr);
1105 if (!cmd) {
1106 addReplySds(c,sdsnew("-ERR unknown command\r\n"));
1107 resetClient(c);
1108 return 1;
1109 } else if ((cmd->arity > 0 && cmd->arity != c->argc) ||
1110 (c->argc < -cmd->arity)) {
1111 addReplySds(c,sdsnew("-ERR wrong number of arguments\r\n"));
1112 resetClient(c);
1113 return 1;
1114 } else if (cmd->flags & REDIS_CMD_BULK && c->bulklen == -1) {
1115 int bulklen = atoi(c->argv[c->argc-1]->ptr);
1116
1117 decrRefCount(c->argv[c->argc-1]);
1118 if (bulklen < 0 || bulklen > 1024*1024*1024) {
1119 c->argc--;
1120 addReplySds(c,sdsnew("-ERR invalid bulk write count\r\n"));
1121 resetClient(c);
1122 return 1;
1123 }
1124 c->argc--;
1125 c->bulklen = bulklen+2; /* add two bytes for CR+LF */
1126 /* It is possible that the bulk read is already in the
1127 * buffer. Check this condition and handle it accordingly */
1128 if ((signed)sdslen(c->querybuf) >= c->bulklen) {
1129 c->argv[c->argc] = createStringObject(c->querybuf,c->bulklen-2);
1130 c->argc++;
1131 c->querybuf = sdsrange(c->querybuf,c->bulklen,-1);
1132 } else {
1133 return 1;
1134 }
1135 }
10c43610 1136 /* Let's try to share objects on the command arguments vector */
1137 if (server.shareobjects) {
1138 int j;
1139 for(j = 1; j < c->argc; j++)
1140 c->argv[j] = tryObjectSharing(c->argv[j]);
1141 }
e63943a4 1142 /* Check if the user is authenticated */
1143 if (server.requirepass && !c->authenticated && cmd->proc != authCommand) {
1144 addReplySds(c,sdsnew("-ERR operation not permitted\r\n"));
1145 resetClient(c);
1146 return 1;
1147 }
1148
ed9b544e 1149 /* Exec the command */
1150 dirty = server.dirty;
1151 cmd->proc(c);
1152 if (server.dirty-dirty != 0 && listLength(server.slaves))
87eca727 1153 replicationFeedSlaves(server.slaves,cmd,c->dictid,c->argv,c->argc);
1154 if (listLength(server.monitors))
1155 replicationFeedSlaves(server.monitors,cmd,c->dictid,c->argv,c->argc);
ed9b544e 1156 server.stat_numcommands++;
1157
1158 /* Prepare the client for the next command */
1159 if (c->flags & REDIS_CLOSE) {
1160 freeClient(c);
1161 return 0;
1162 }
1163 resetClient(c);
1164 return 1;
1165}
1166
87eca727 1167static void replicationFeedSlaves(list *slaves, struct redisCommand *cmd, int dictid, robj **argv, int argc) {
1168 listNode *ln = slaves->head;
ed9b544e 1169 robj *outv[REDIS_MAX_ARGS*4]; /* enough room for args, spaces, newlines */
1170 int outc = 0, j;
1171
1172 for (j = 0; j < argc; j++) {
1173 if (j != 0) outv[outc++] = shared.space;
1174 if ((cmd->flags & REDIS_CMD_BULK) && j == argc-1) {
1175 robj *lenobj;
1176
1177 lenobj = createObject(REDIS_STRING,
1178 sdscatprintf(sdsempty(),"%d\r\n",sdslen(argv[j]->ptr)));
1179 lenobj->refcount = 0;
1180 outv[outc++] = lenobj;
1181 }
1182 outv[outc++] = argv[j];
1183 }
1184 outv[outc++] = shared.crlf;
1185
1186 while(ln) {
1187 redisClient *slave = ln->value;
1188 if (slave->slaveseldb != dictid) {
1189 robj *selectcmd;
1190
1191 switch(dictid) {
1192 case 0: selectcmd = shared.select0; break;
1193 case 1: selectcmd = shared.select1; break;
1194 case 2: selectcmd = shared.select2; break;
1195 case 3: selectcmd = shared.select3; break;
1196 case 4: selectcmd = shared.select4; break;
1197 case 5: selectcmd = shared.select5; break;
1198 case 6: selectcmd = shared.select6; break;
1199 case 7: selectcmd = shared.select7; break;
1200 case 8: selectcmd = shared.select8; break;
1201 case 9: selectcmd = shared.select9; break;
1202 default:
1203 selectcmd = createObject(REDIS_STRING,
1204 sdscatprintf(sdsempty(),"select %d\r\n",dictid));
1205 selectcmd->refcount = 0;
1206 break;
1207 }
1208 addReply(slave,selectcmd);
1209 slave->slaveseldb = dictid;
1210 }
1211 for (j = 0; j < outc; j++) addReply(slave,outv[j]);
1212 ln = ln->next;
1213 }
1214}
1215
1216static void readQueryFromClient(aeEventLoop *el, int fd, void *privdata, int mask) {
1217 redisClient *c = (redisClient*) privdata;
1218 char buf[REDIS_QUERYBUF_LEN];
1219 int nread;
1220 REDIS_NOTUSED(el);
1221 REDIS_NOTUSED(mask);
1222
1223 nread = read(fd, buf, REDIS_QUERYBUF_LEN);
1224 if (nread == -1) {
1225 if (errno == EAGAIN) {
1226 nread = 0;
1227 } else {
1228 redisLog(REDIS_DEBUG, "Reading from client: %s",strerror(errno));
1229 freeClient(c);
1230 return;
1231 }
1232 } else if (nread == 0) {
1233 redisLog(REDIS_DEBUG, "Client closed connection");
1234 freeClient(c);
1235 return;
1236 }
1237 if (nread) {
1238 c->querybuf = sdscatlen(c->querybuf, buf, nread);
1239 c->lastinteraction = time(NULL);
1240 } else {
1241 return;
1242 }
1243
1244again:
1245 if (c->bulklen == -1) {
1246 /* Read the first line of the query */
1247 char *p = strchr(c->querybuf,'\n');
1248 size_t querylen;
1249 if (p) {
1250 sds query, *argv;
1251 int argc, j;
1252
1253 query = c->querybuf;
1254 c->querybuf = sdsempty();
1255 querylen = 1+(p-(query));
1256 if (sdslen(query) > querylen) {
1257 /* leave data after the first line of the query in the buffer */
1258 c->querybuf = sdscatlen(c->querybuf,query+querylen,sdslen(query)-querylen);
1259 }
1260 *p = '\0'; /* remove "\n" */
1261 if (*(p-1) == '\r') *(p-1) = '\0'; /* and "\r" if any */
1262 sdsupdatelen(query);
1263
1264 /* Now we can split the query in arguments */
1265 if (sdslen(query) == 0) {
1266 /* Ignore empty query */
1267 sdsfree(query);
1268 return;
1269 }
1270 argv = sdssplitlen(query,sdslen(query)," ",1,&argc);
1271 sdsfree(query);
1272 if (argv == NULL) oom("sdssplitlen");
1273 for (j = 0; j < argc && j < REDIS_MAX_ARGS; j++) {
1274 if (sdslen(argv[j])) {
1275 c->argv[c->argc] = createObject(REDIS_STRING,argv[j]);
1276 c->argc++;
1277 } else {
1278 sdsfree(argv[j]);
1279 }
1280 }
1281 zfree(argv);
1282 /* Execute the command. If the client is still valid
1283 * after processCommand() return and there is something
1284 * on the query buffer try to process the next command. */
1285 if (processCommand(c) && sdslen(c->querybuf)) goto again;
1286 return;
1287 } else if (sdslen(c->querybuf) >= 1024) {
1288 redisLog(REDIS_DEBUG, "Client protocol error");
1289 freeClient(c);
1290 return;
1291 }
1292 } else {
1293 /* Bulk read handling. Note that if we are at this point
1294 the client already sent a command terminated with a newline,
1295 we are reading the bulk data that is actually the last
1296 argument of the command. */
1297 int qbl = sdslen(c->querybuf);
1298
1299 if (c->bulklen <= qbl) {
1300 /* Copy everything but the final CRLF as final argument */
1301 c->argv[c->argc] = createStringObject(c->querybuf,c->bulklen-2);
1302 c->argc++;
1303 c->querybuf = sdsrange(c->querybuf,c->bulklen,-1);
1304 processCommand(c);
1305 return;
1306 }
1307 }
1308}
1309
1310static int selectDb(redisClient *c, int id) {
1311 if (id < 0 || id >= server.dbnum)
1312 return REDIS_ERR;
1313 c->dict = server.dict[id];
1314 c->dictid = id;
1315 return REDIS_OK;
1316}
1317
1318static redisClient *createClient(int fd) {
1319 redisClient *c = zmalloc(sizeof(*c));
1320
1321 anetNonBlock(NULL,fd);
1322 anetTcpNoDelay(NULL,fd);
1323 if (!c) return NULL;
1324 selectDb(c,0);
1325 c->fd = fd;
1326 c->querybuf = sdsempty();
1327 c->argc = 0;
1328 c->bulklen = -1;
1329 c->sentlen = 0;
1330 c->flags = 0;
1331 c->lastinteraction = time(NULL);
abcb223e 1332 c->authenticated = 0;
ed9b544e 1333 if ((c->reply = listCreate()) == NULL) oom("listCreate");
1334 listSetFreeMethod(c->reply,decrRefCount);
1335 if (aeCreateFileEvent(server.el, c->fd, AE_READABLE,
1336 readQueryFromClient, c, NULL) == AE_ERR) {
1337 freeClient(c);
1338 return NULL;
1339 }
1340 if (!listAddNodeTail(server.clients,c)) oom("listAddNodeTail");
1341 return c;
1342}
1343
1344static void addReply(redisClient *c, robj *obj) {
1345 if (listLength(c->reply) == 0 &&
1346 aeCreateFileEvent(server.el, c->fd, AE_WRITABLE,
1347 sendReplyToClient, c, NULL) == AE_ERR) return;
1348 if (!listAddNodeTail(c->reply,obj)) oom("listAddNodeTail");
1349 incrRefCount(obj);
1350}
1351
1352static void addReplySds(redisClient *c, sds s) {
1353 robj *o = createObject(REDIS_STRING,s);
1354 addReply(c,o);
1355 decrRefCount(o);
1356}
1357
1358static void acceptHandler(aeEventLoop *el, int fd, void *privdata, int mask) {
1359 int cport, cfd;
1360 char cip[128];
1361 REDIS_NOTUSED(el);
1362 REDIS_NOTUSED(mask);
1363 REDIS_NOTUSED(privdata);
1364
1365 cfd = anetAccept(server.neterr, fd, cip, &cport);
1366 if (cfd == AE_ERR) {
1367 redisLog(REDIS_DEBUG,"Accepting client connection: %s", server.neterr);
1368 return;
1369 }
1370 redisLog(REDIS_DEBUG,"Accepted %s:%d", cip, cport);
1371 if (createClient(cfd) == NULL) {
1372 redisLog(REDIS_WARNING,"Error allocating resoures for the client");
1373 close(cfd); /* May be already closed, just ingore errors */
1374 return;
1375 }
1376 server.stat_numconnections++;
1377}
1378
1379/* ======================= Redis objects implementation ===================== */
1380
1381static robj *createObject(int type, void *ptr) {
1382 robj *o;
1383
1384 if (listLength(server.objfreelist)) {
1385 listNode *head = listFirst(server.objfreelist);
1386 o = listNodeValue(head);
1387 listDelNode(server.objfreelist,head);
1388 } else {
1389 o = zmalloc(sizeof(*o));
1390 }
1391 if (!o) oom("createObject");
1392 o->type = type;
1393 o->ptr = ptr;
1394 o->refcount = 1;
1395 return o;
1396}
1397
1398static robj *createStringObject(char *ptr, size_t len) {
1399 return createObject(REDIS_STRING,sdsnewlen(ptr,len));
1400}
1401
1402static robj *createListObject(void) {
1403 list *l = listCreate();
1404
1405 if (!l) oom("listCreate");
1406 listSetFreeMethod(l,decrRefCount);
1407 return createObject(REDIS_LIST,l);
1408}
1409
1410static robj *createSetObject(void) {
1411 dict *d = dictCreate(&setDictType,NULL);
1412 if (!d) oom("dictCreate");
1413 return createObject(REDIS_SET,d);
1414}
1415
1416#if 0
1417static robj *createHashObject(void) {
1418 dict *d = dictCreate(&hashDictType,NULL);
1419 if (!d) oom("dictCreate");
1420 return createObject(REDIS_SET,d);
1421}
1422#endif
1423
1424static void freeStringObject(robj *o) {
1425 sdsfree(o->ptr);
1426}
1427
1428static void freeListObject(robj *o) {
1429 listRelease((list*) o->ptr);
1430}
1431
1432static void freeSetObject(robj *o) {
1433 dictRelease((dict*) o->ptr);
1434}
1435
1436static void freeHashObject(robj *o) {
1437 dictRelease((dict*) o->ptr);
1438}
1439
1440static void incrRefCount(robj *o) {
1441 o->refcount++;
1442}
1443
1444static void decrRefCount(void *obj) {
1445 robj *o = obj;
1446 if (--(o->refcount) == 0) {
1447 switch(o->type) {
1448 case REDIS_STRING: freeStringObject(o); break;
1449 case REDIS_LIST: freeListObject(o); break;
1450 case REDIS_SET: freeSetObject(o); break;
1451 case REDIS_HASH: freeHashObject(o); break;
1452 default: assert(0 != 0); break;
1453 }
1454 if (listLength(server.objfreelist) > REDIS_OBJFREELIST_MAX ||
1455 !listAddNodeHead(server.objfreelist,o))
1456 zfree(o);
1457 }
1458}
1459
10c43610 1460/* Try to share an object against the shared objects pool */
1461static robj *tryObjectSharing(robj *o) {
1462 struct dictEntry *de;
1463 unsigned long c;
1464
1465 if (server.shareobjects == 0) return o;
1466
1467 assert(o->type == REDIS_STRING);
1468 de = dictFind(server.sharingpool,o);
1469 if (de) {
1470 robj *shared = dictGetEntryKey(de);
1471
1472 c = ((unsigned long) dictGetEntryVal(de))+1;
1473 dictGetEntryVal(de) = (void*) c;
1474 incrRefCount(shared);
1475 decrRefCount(o);
1476 return shared;
1477 } else {
1478 /* Here we are using a stream algorihtm: Every time an object is
1479 * shared we increment its count, everytime there is a miss we
1480 * recrement the counter of a random object. If this object reaches
1481 * zero we remove the object and put the current object instead. */
1482 if (dictGetHashTableUsed(server.sharingpool) >=
1483 server.sharingpoolsize) {
1484 de = dictGetRandomKey(server.sharingpool);
1485 assert(de != NULL);
1486 c = ((unsigned long) dictGetEntryVal(de))-1;
1487 dictGetEntryVal(de) = (void*) c;
1488 if (c == 0) {
1489 dictDelete(server.sharingpool,de->key);
1490 }
1491 } else {
1492 c = 0; /* If the pool is empty we want to add this object */
1493 }
1494 if (c == 0) {
1495 int retval;
1496
1497 retval = dictAdd(server.sharingpool,o,(void*)1);
1498 assert(retval == DICT_OK);
1499 incrRefCount(o);
1500 }
1501 return o;
1502 }
1503}
1504
ed9b544e 1505/*============================ DB saving/loading ============================ */
1506
f78fd11b 1507static int rdbSaveType(FILE *fp, unsigned char type) {
1508 if (fwrite(&type,1,1,fp) == 0) return -1;
1509 return 0;
1510}
1511
e3566d4b 1512/* check rdbLoadLen() comments for more info */
f78fd11b 1513static int rdbSaveLen(FILE *fp, uint32_t len) {
1514 unsigned char buf[2];
1515
1516 if (len < (1<<6)) {
1517 /* Save a 6 bit len */
10c43610 1518 buf[0] = (len&0xFF)|(REDIS_RDB_6BITLEN<<6);
f78fd11b 1519 if (fwrite(buf,1,1,fp) == 0) return -1;
1520 } else if (len < (1<<14)) {
1521 /* Save a 14 bit len */
10c43610 1522 buf[0] = ((len>>8)&0xFF)|(REDIS_RDB_14BITLEN<<6);
f78fd11b 1523 buf[1] = len&0xFF;
17be1a4a 1524 if (fwrite(buf,2,1,fp) == 0) return -1;
f78fd11b 1525 } else {
1526 /* Save a 32 bit len */
10c43610 1527 buf[0] = (REDIS_RDB_32BITLEN<<6);
f78fd11b 1528 if (fwrite(buf,1,1,fp) == 0) return -1;
1529 len = htonl(len);
1530 if (fwrite(&len,4,1,fp) == 0) return -1;
1531 }
1532 return 0;
1533}
1534
e3566d4b 1535/* String objects in the form "2391" "-100" without any space and with a
1536 * range of values that can fit in an 8, 16 or 32 bit signed value can be
1537 * encoded as integers to save space */
1538int rdbTryIntegerEncoding(sds s, unsigned char *enc) {
1539 long long value;
1540 char *endptr, buf[32];
1541
1542 /* Check if it's possible to encode this value as a number */
1543 value = strtoll(s, &endptr, 10);
1544 if (endptr[0] != '\0') return 0;
1545 snprintf(buf,32,"%lld",value);
1546
1547 /* If the number converted back into a string is not identical
1548 * then it's not possible to encode the string as integer */
1549 if (strlen(buf) != sdslen(s) || memcmp(buf,s,sdslen(s))) return 0;
1550
1551 /* Finally check if it fits in our ranges */
1552 if (value >= -(1<<7) && value <= (1<<7)-1) {
1553 enc[0] = (REDIS_RDB_ENCVAL<<6)|REDIS_RDB_ENC_INT8;
1554 enc[1] = value&0xFF;
1555 return 2;
1556 } else if (value >= -(1<<15) && value <= (1<<15)-1) {
1557 enc[0] = (REDIS_RDB_ENCVAL<<6)|REDIS_RDB_ENC_INT16;
1558 enc[1] = value&0xFF;
1559 enc[2] = (value>>8)&0xFF;
1560 return 3;
1561 } else if (value >= -((long long)1<<31) && value <= ((long long)1<<31)-1) {
1562 enc[0] = (REDIS_RDB_ENCVAL<<6)|REDIS_RDB_ENC_INT32;
1563 enc[1] = value&0xFF;
1564 enc[2] = (value>>8)&0xFF;
1565 enc[3] = (value>>16)&0xFF;
1566 enc[4] = (value>>24)&0xFF;
1567 return 5;
1568 } else {
1569 return 0;
1570 }
1571}
1572
1573/* Save a string objet as [len][data] on disk. If the object is a string
1574 * representation of an integer value we try to safe it in a special form */
10c43610 1575static int rdbSaveStringObject(FILE *fp, robj *obj) {
1576 size_t len = sdslen(obj->ptr);
e3566d4b 1577 int enclen;
10c43610 1578
e3566d4b 1579 if (len <= 11) {
1580 unsigned char buf[5];
1581 if ((enclen = rdbTryIntegerEncoding(obj->ptr,buf)) > 0) {
1582 if (fwrite(buf,enclen,1,fp) == 0) return -1;
1583 return 0;
1584 }
1585 }
10c43610 1586 if (rdbSaveLen(fp,len) == -1) return -1;
1587 if (len && fwrite(obj->ptr,len,1,fp) == 0) return -1;
1588 return 0;
1589}
1590
ed9b544e 1591/* Save the DB on disk. Return REDIS_ERR on error, REDIS_OK on success */
f78fd11b 1592static int rdbSave(char *filename) {
ed9b544e 1593 dictIterator *di = NULL;
1594 dictEntry *de;
ed9b544e 1595 FILE *fp;
1596 char tmpfile[256];
1597 int j;
1598
1599 snprintf(tmpfile,256,"temp-%d.%ld.rdb",(int)time(NULL),(long int)random());
1600 fp = fopen(tmpfile,"w");
1601 if (!fp) {
1602 redisLog(REDIS_WARNING, "Failed saving the DB: %s", strerror(errno));
1603 return REDIS_ERR;
1604 }
f78fd11b 1605 if (fwrite("REDIS0001",9,1,fp) == 0) goto werr;
ed9b544e 1606 for (j = 0; j < server.dbnum; j++) {
1607 dict *d = server.dict[j];
1608 if (dictGetHashTableUsed(d) == 0) continue;
1609 di = dictGetIterator(d);
1610 if (!di) {
1611 fclose(fp);
1612 return REDIS_ERR;
1613 }
1614
1615 /* Write the SELECT DB opcode */
f78fd11b 1616 if (rdbSaveType(fp,REDIS_SELECTDB) == -1) goto werr;
1617 if (rdbSaveLen(fp,j) == -1) goto werr;
ed9b544e 1618
1619 /* Iterate this DB writing every entry */
1620 while((de = dictNext(di)) != NULL) {
1621 robj *key = dictGetEntryKey(de);
1622 robj *o = dictGetEntryVal(de);
1623
f78fd11b 1624 if (rdbSaveType(fp,o->type) == -1) goto werr;
10c43610 1625 if (rdbSaveStringObject(fp,key) == -1) goto werr;
f78fd11b 1626 if (o->type == REDIS_STRING) {
ed9b544e 1627 /* Save a string value */
10c43610 1628 if (rdbSaveStringObject(fp,o) == -1) goto werr;
f78fd11b 1629 } else if (o->type == REDIS_LIST) {
ed9b544e 1630 /* Save a list value */
1631 list *list = o->ptr;
1632 listNode *ln = list->head;
1633
f78fd11b 1634 if (rdbSaveLen(fp,listLength(list)) == -1) goto werr;
ed9b544e 1635 while(ln) {
1636 robj *eleobj = listNodeValue(ln);
f78fd11b 1637
10c43610 1638 if (rdbSaveStringObject(fp,eleobj) == -1) goto werr;
ed9b544e 1639 ln = ln->next;
1640 }
f78fd11b 1641 } else if (o->type == REDIS_SET) {
ed9b544e 1642 /* Save a set value */
1643 dict *set = o->ptr;
1644 dictIterator *di = dictGetIterator(set);
1645 dictEntry *de;
1646
1647 if (!set) oom("dictGetIteraotr");
f78fd11b 1648 if (rdbSaveLen(fp,dictGetHashTableUsed(set)) == -1) goto werr;
ed9b544e 1649 while((de = dictNext(di)) != NULL) {
10c43610 1650 robj *eleobj = dictGetEntryKey(de);
ed9b544e 1651
10c43610 1652 if (rdbSaveStringObject(fp,eleobj) == -1) goto werr;
ed9b544e 1653 }
1654 dictReleaseIterator(di);
1655 } else {
1656 assert(0 != 0);
1657 }
1658 }
1659 dictReleaseIterator(di);
1660 }
1661 /* EOF opcode */
f78fd11b 1662 if (rdbSaveType(fp,REDIS_EOF) == -1) goto werr;
1663
1664 /* Make sure data will not remain on the OS's output buffers */
ed9b544e 1665 fflush(fp);
1666 fsync(fileno(fp));
1667 fclose(fp);
1668
1669 /* Use RENAME to make sure the DB file is changed atomically only
1670 * if the generate DB file is ok. */
1671 if (rename(tmpfile,filename) == -1) {
1672 redisLog(REDIS_WARNING,"Error moving temp DB file on the final destionation: %s", strerror(errno));
1673 unlink(tmpfile);
1674 return REDIS_ERR;
1675 }
1676 redisLog(REDIS_NOTICE,"DB saved on disk");
1677 server.dirty = 0;
1678 server.lastsave = time(NULL);
1679 return REDIS_OK;
1680
1681werr:
1682 fclose(fp);
1683 unlink(tmpfile);
1684 redisLog(REDIS_WARNING,"Write error saving DB on disk: %s", strerror(errno));
1685 if (di) dictReleaseIterator(di);
1686 return REDIS_ERR;
1687}
1688
f78fd11b 1689static int rdbSaveBackground(char *filename) {
ed9b544e 1690 pid_t childpid;
1691
1692 if (server.bgsaveinprogress) return REDIS_ERR;
1693 if ((childpid = fork()) == 0) {
1694 /* Child */
1695 close(server.fd);
f78fd11b 1696 if (rdbSave(filename) == REDIS_OK) {
ed9b544e 1697 exit(0);
1698 } else {
1699 exit(1);
1700 }
1701 } else {
1702 /* Parent */
1703 redisLog(REDIS_NOTICE,"Background saving started by pid %d",childpid);
1704 server.bgsaveinprogress = 1;
1705 return REDIS_OK;
1706 }
1707 return REDIS_OK; /* unreached */
1708}
1709
f78fd11b 1710static int rdbLoadType(FILE *fp) {
1711 unsigned char type;
7b45bfb2 1712 if (fread(&type,1,1,fp) == 0) return -1;
1713 return type;
1714}
1715
e3566d4b 1716/* Load an encoded length from the DB, see the REDIS_RDB_* defines on the top
1717 * of this file for a description of how this are stored on disk.
1718 *
1719 * isencoded is set to 1 if the readed length is not actually a length but
1720 * an "encoding type", check the above comments for more info */
1721static uint32_t rdbLoadLen(FILE *fp, int rdbver, int *isencoded) {
f78fd11b 1722 unsigned char buf[2];
1723 uint32_t len;
1724
e3566d4b 1725 if (isencoded) *isencoded = 0;
f78fd11b 1726 if (rdbver == 0) {
1727 if (fread(&len,4,1,fp) == 0) return REDIS_RDB_LENERR;
1728 return ntohl(len);
1729 } else {
17be1a4a 1730 int type;
1731
f78fd11b 1732 if (fread(buf,1,1,fp) == 0) return REDIS_RDB_LENERR;
17be1a4a 1733 type = (buf[0]&0xC0)>>6;
1734 if (type == REDIS_RDB_6BITLEN) {
f78fd11b 1735 /* Read a 6 bit len */
e3566d4b 1736 return buf[0]&0x3F;
1737 } else if (type == REDIS_RDB_ENCVAL) {
1738 /* Read a 6 bit len encoding type */
1739 if (isencoded) *isencoded = 1;
1740 return buf[0]&0x3F;
17be1a4a 1741 } else if (type == REDIS_RDB_14BITLEN) {
f78fd11b 1742 /* Read a 14 bit len */
1743 if (fread(buf+1,1,1,fp) == 0) return REDIS_RDB_LENERR;
1744 return ((buf[0]&0x3F)<<8)|buf[1];
1745 } else {
1746 /* Read a 32 bit len */
1747 if (fread(&len,4,1,fp) == 0) return REDIS_RDB_LENERR;
1748 return ntohl(len);
1749 }
1750 }
f78fd11b 1751}
1752
e3566d4b 1753static robj *rdbLoadIntegerObject(FILE *fp, int enctype) {
1754 unsigned char enc[4];
1755 long long val;
1756
1757 if (enctype == REDIS_RDB_ENC_INT8) {
1758 if (fread(enc,1,1,fp) == 0) return NULL;
1759 val = (signed char)enc[0];
1760 } else if (enctype == REDIS_RDB_ENC_INT16) {
1761 uint16_t v;
1762 if (fread(enc,2,1,fp) == 0) return NULL;
1763 v = enc[0]|(enc[1]<<8);
1764 val = (int16_t)v;
1765 } else if (enctype == REDIS_RDB_ENC_INT32) {
1766 uint32_t v;
1767 if (fread(enc,4,1,fp) == 0) return NULL;
1768 v = enc[0]|(enc[1]<<8)|(enc[2]<<16)|(enc[3]<<24);
1769 val = (int32_t)v;
1770 } else {
1771 val = 0; /* anti-warning */
1772 assert(0!=0);
1773 }
1774 return createObject(REDIS_STRING,sdscatprintf(sdsempty(),"%lld",val));
1775}
1776
1777static robj *rdbLoadStringObject(FILE*fp, int rdbver) {
1778 int isencoded;
1779 uint32_t len;
f78fd11b 1780 sds val;
1781
e3566d4b 1782 len = rdbLoadLen(fp,rdbver,&isencoded);
1783 if (isencoded) {
1784 switch(len) {
1785 case REDIS_RDB_ENC_INT8:
1786 case REDIS_RDB_ENC_INT16:
1787 case REDIS_RDB_ENC_INT32:
1788 return rdbLoadIntegerObject(fp,len);
1789 default:
1790 assert(0!=0);
1791 }
1792 }
1793
f78fd11b 1794 if (len == REDIS_RDB_LENERR) return NULL;
1795 val = sdsnewlen(NULL,len);
1796 if (len && fread(val,len,1,fp) == 0) {
1797 sdsfree(val);
1798 return NULL;
1799 }
10c43610 1800 return tryObjectSharing(createObject(REDIS_STRING,val));
f78fd11b 1801}
1802
1803static int rdbLoad(char *filename) {
ed9b544e 1804 FILE *fp;
f78fd11b 1805 robj *keyobj = NULL;
1806 uint32_t dbid;
7b45bfb2 1807 int type;
ed9b544e 1808 int retval;
1809 dict *d = server.dict[0];
f78fd11b 1810 char buf[1024];
1811 int rdbver;
ed9b544e 1812 fp = fopen(filename,"r");
1813 if (!fp) return REDIS_ERR;
1814 if (fread(buf,9,1,fp) == 0) goto eoferr;
f78fd11b 1815 buf[9] = '\0';
1816 if (memcmp(buf,"REDIS",5) != 0) {
ed9b544e 1817 fclose(fp);
1818 redisLog(REDIS_WARNING,"Wrong signature trying to load DB from file");
1819 return REDIS_ERR;
1820 }
f78fd11b 1821 rdbver = atoi(buf+5);
1822 if (rdbver > 1) {
1823 fclose(fp);
1824 redisLog(REDIS_WARNING,"Can't handle RDB format version %d",rdbver);
1825 return REDIS_ERR;
1826 }
ed9b544e 1827 while(1) {
1828 robj *o;
1829
1830 /* Read type. */
f78fd11b 1831 if ((type = rdbLoadType(fp)) == -1) goto eoferr;
ed9b544e 1832 if (type == REDIS_EOF) break;
1833 /* Handle SELECT DB opcode as a special case */
1834 if (type == REDIS_SELECTDB) {
e3566d4b 1835 if ((dbid = rdbLoadLen(fp,rdbver,NULL)) == REDIS_RDB_LENERR)
1836 goto eoferr;
ed9b544e 1837 if (dbid >= (unsigned)server.dbnum) {
f78fd11b 1838 redisLog(REDIS_WARNING,"FATAL: Data file was created with a Redis server configured to handle more than %d databases. Exiting\n", server.dbnum);
ed9b544e 1839 exit(1);
1840 }
1841 d = server.dict[dbid];
1842 continue;
1843 }
1844 /* Read key */
f78fd11b 1845 if ((keyobj = rdbLoadStringObject(fp,rdbver)) == NULL) goto eoferr;
ed9b544e 1846
1847 if (type == REDIS_STRING) {
1848 /* Read string value */
f78fd11b 1849 if ((o = rdbLoadStringObject(fp,rdbver)) == NULL) goto eoferr;
ed9b544e 1850 } else if (type == REDIS_LIST || type == REDIS_SET) {
1851 /* Read list/set value */
1852 uint32_t listlen;
f78fd11b 1853
e3566d4b 1854 if ((listlen = rdbLoadLen(fp,rdbver,NULL)) == REDIS_RDB_LENERR)
f78fd11b 1855 goto eoferr;
ed9b544e 1856 o = (type == REDIS_LIST) ? createListObject() : createSetObject();
1857 /* Load every single element of the list/set */
1858 while(listlen--) {
1859 robj *ele;
1860
f78fd11b 1861 if ((ele = rdbLoadStringObject(fp,rdbver)) == NULL) goto eoferr;
ed9b544e 1862 if (type == REDIS_LIST) {
1863 if (!listAddNodeTail((list*)o->ptr,ele))
1864 oom("listAddNodeTail");
1865 } else {
1866 if (dictAdd((dict*)o->ptr,ele,NULL) == DICT_ERR)
1867 oom("dictAdd");
1868 }
ed9b544e 1869 }
1870 } else {
1871 assert(0 != 0);
1872 }
1873 /* Add the new object in the hash table */
f78fd11b 1874 retval = dictAdd(d,keyobj,o);
ed9b544e 1875 if (retval == DICT_ERR) {
f78fd11b 1876 redisLog(REDIS_WARNING,"Loading DB, duplicated key (%s) found! Unrecoverable error, exiting now.", keyobj->ptr);
ed9b544e 1877 exit(1);
1878 }
f78fd11b 1879 keyobj = o = NULL;
ed9b544e 1880 }
1881 fclose(fp);
1882 return REDIS_OK;
1883
1884eoferr: /* unexpected end of file is handled here with a fatal exit */
e3566d4b 1885 if (keyobj) decrRefCount(keyobj);
1886 redisLog(REDIS_WARNING,"Short read or OOM loading DB. Unrecoverable error, exiting now.");
ed9b544e 1887 exit(1);
1888 return REDIS_ERR; /* Just to avoid warning */
1889}
1890
1891/*================================== Commands =============================== */
1892
abcb223e
BH
1893static void authCommand(redisClient *c) {
1894 if (!strcmp(c->argv[1]->ptr, server.requirepass)) {
1895 c->authenticated = 1;
1896 addReply(c,shared.ok);
1897 } else {
1898 c->authenticated = 0;
1899 addReply(c,shared.err);
1900 }
1901}
1902
ed9b544e 1903static void pingCommand(redisClient *c) {
1904 addReply(c,shared.pong);
1905}
1906
1907static void echoCommand(redisClient *c) {
c937aa89 1908 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",
ed9b544e 1909 (int)sdslen(c->argv[1]->ptr)));
1910 addReply(c,c->argv[1]);
1911 addReply(c,shared.crlf);
1912}
1913
1914/*=================================== Strings =============================== */
1915
1916static void setGenericCommand(redisClient *c, int nx) {
1917 int retval;
1918
1919 retval = dictAdd(c->dict,c->argv[1],c->argv[2]);
1920 if (retval == DICT_ERR) {
1921 if (!nx) {
1922 dictReplace(c->dict,c->argv[1],c->argv[2]);
1923 incrRefCount(c->argv[2]);
1924 } else {
c937aa89 1925 addReply(c,shared.czero);
ed9b544e 1926 return;
1927 }
1928 } else {
1929 incrRefCount(c->argv[1]);
1930 incrRefCount(c->argv[2]);
1931 }
1932 server.dirty++;
c937aa89 1933 addReply(c, nx ? shared.cone : shared.ok);
ed9b544e 1934}
1935
1936static void setCommand(redisClient *c) {
a4d1ba9a 1937 setGenericCommand(c,0);
ed9b544e 1938}
1939
1940static void setnxCommand(redisClient *c) {
a4d1ba9a 1941 setGenericCommand(c,1);
ed9b544e 1942}
1943
1944static void getCommand(redisClient *c) {
1945 dictEntry *de;
1946
1947 de = dictFind(c->dict,c->argv[1]);
1948 if (de == NULL) {
c937aa89 1949 addReply(c,shared.nullbulk);
ed9b544e 1950 } else {
1951 robj *o = dictGetEntryVal(de);
1952
1953 if (o->type != REDIS_STRING) {
c937aa89 1954 addReply(c,shared.wrongtypeerr);
ed9b544e 1955 } else {
c937aa89 1956 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",(int)sdslen(o->ptr)));
ed9b544e 1957 addReply(c,o);
1958 addReply(c,shared.crlf);
1959 }
1960 }
1961}
1962
70003d28 1963static void mgetCommand(redisClient *c) {
1964 dictEntry *de;
1965 int j;
1966
c937aa89 1967 addReplySds(c,sdscatprintf(sdsempty(),"*%d\r\n",c->argc-1));
70003d28 1968 for (j = 1; j < c->argc; j++) {
1969 de = dictFind(c->dict,c->argv[j]);
1970 if (de == NULL) {
c937aa89 1971 addReply(c,shared.nullbulk);
70003d28 1972 } else {
1973 robj *o = dictGetEntryVal(de);
1974
1975 if (o->type != REDIS_STRING) {
c937aa89 1976 addReply(c,shared.nullbulk);
70003d28 1977 } else {
c937aa89 1978 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",(int)sdslen(o->ptr)));
70003d28 1979 addReply(c,o);
1980 addReply(c,shared.crlf);
1981 }
1982 }
1983 }
1984}
1985
ed9b544e 1986static void incrDecrCommand(redisClient *c, int incr) {
1987 dictEntry *de;
1988 long long value;
1989 int retval;
1990 robj *o;
1991
1992 de = dictFind(c->dict,c->argv[1]);
1993 if (de == NULL) {
1994 value = 0;
1995 } else {
1996 robj *o = dictGetEntryVal(de);
1997
1998 if (o->type != REDIS_STRING) {
1999 value = 0;
2000 } else {
2001 char *eptr;
2002
2003 value = strtoll(o->ptr, &eptr, 10);
2004 }
2005 }
2006
2007 value += incr;
2008 o = createObject(REDIS_STRING,sdscatprintf(sdsempty(),"%lld",value));
2009 retval = dictAdd(c->dict,c->argv[1],o);
2010 if (retval == DICT_ERR) {
2011 dictReplace(c->dict,c->argv[1],o);
2012 } else {
2013 incrRefCount(c->argv[1]);
2014 }
2015 server.dirty++;
c937aa89 2016 addReply(c,shared.colon);
ed9b544e 2017 addReply(c,o);
2018 addReply(c,shared.crlf);
2019}
2020
2021static void incrCommand(redisClient *c) {
a4d1ba9a 2022 incrDecrCommand(c,1);
ed9b544e 2023}
2024
2025static void decrCommand(redisClient *c) {
a4d1ba9a 2026 incrDecrCommand(c,-1);
ed9b544e 2027}
2028
2029static void incrbyCommand(redisClient *c) {
2030 int incr = atoi(c->argv[2]->ptr);
a4d1ba9a 2031 incrDecrCommand(c,incr);
ed9b544e 2032}
2033
2034static void decrbyCommand(redisClient *c) {
2035 int incr = atoi(c->argv[2]->ptr);
a4d1ba9a 2036 incrDecrCommand(c,-incr);
ed9b544e 2037}
2038
2039/* ========================= Type agnostic commands ========================= */
2040
2041static void delCommand(redisClient *c) {
2042 if (dictDelete(c->dict,c->argv[1]) == DICT_OK) {
2043 server.dirty++;
c937aa89 2044 addReply(c,shared.cone);
ed9b544e 2045 } else {
c937aa89 2046 addReply(c,shared.czero);
ed9b544e 2047 }
2048}
2049
2050static void existsCommand(redisClient *c) {
2051 dictEntry *de;
2052
2053 de = dictFind(c->dict,c->argv[1]);
2054 if (de == NULL)
c937aa89 2055 addReply(c,shared.czero);
ed9b544e 2056 else
c937aa89 2057 addReply(c,shared.cone);
ed9b544e 2058}
2059
2060static void selectCommand(redisClient *c) {
2061 int id = atoi(c->argv[1]->ptr);
2062
2063 if (selectDb(c,id) == REDIS_ERR) {
2064 addReplySds(c,"-ERR invalid DB index\r\n");
2065 } else {
2066 addReply(c,shared.ok);
2067 }
2068}
2069
2070static void randomkeyCommand(redisClient *c) {
2071 dictEntry *de;
2072
2073 de = dictGetRandomKey(c->dict);
2074 if (de == NULL) {
2075 addReply(c,shared.crlf);
2076 } else {
c937aa89 2077 addReply(c,shared.plus);
ed9b544e 2078 addReply(c,dictGetEntryKey(de));
2079 addReply(c,shared.crlf);
2080 }
2081}
2082
2083static void keysCommand(redisClient *c) {
2084 dictIterator *di;
2085 dictEntry *de;
2086 sds pattern = c->argv[1]->ptr;
2087 int plen = sdslen(pattern);
2088 int numkeys = 0, keyslen = 0;
2089 robj *lenobj = createObject(REDIS_STRING,NULL);
2090
2091 di = dictGetIterator(c->dict);
2092 if (!di) oom("dictGetIterator");
2093 addReply(c,lenobj);
2094 decrRefCount(lenobj);
2095 while((de = dictNext(di)) != NULL) {
2096 robj *keyobj = dictGetEntryKey(de);
2097 sds key = keyobj->ptr;
2098 if ((pattern[0] == '*' && pattern[1] == '\0') ||
2099 stringmatchlen(pattern,plen,key,sdslen(key),0)) {
2100 if (numkeys != 0)
2101 addReply(c,shared.space);
2102 addReply(c,keyobj);
2103 numkeys++;
2104 keyslen += sdslen(key);
2105 }
2106 }
2107 dictReleaseIterator(di);
c937aa89 2108 lenobj->ptr = sdscatprintf(sdsempty(),"$%lu\r\n",keyslen+(numkeys ? (numkeys-1) : 0));
ed9b544e 2109 addReply(c,shared.crlf);
2110}
2111
2112static void dbsizeCommand(redisClient *c) {
2113 addReplySds(c,
c937aa89 2114 sdscatprintf(sdsempty(),":%lu\r\n",dictGetHashTableUsed(c->dict)));
ed9b544e 2115}
2116
2117static void lastsaveCommand(redisClient *c) {
2118 addReplySds(c,
c937aa89 2119 sdscatprintf(sdsempty(),":%lu\r\n",server.lastsave));
ed9b544e 2120}
2121
2122static void typeCommand(redisClient *c) {
2123 dictEntry *de;
2124 char *type;
2125
2126 de = dictFind(c->dict,c->argv[1]);
2127 if (de == NULL) {
c937aa89 2128 type = "+none";
ed9b544e 2129 } else {
2130 robj *o = dictGetEntryVal(de);
2131
2132 switch(o->type) {
c937aa89 2133 case REDIS_STRING: type = "+string"; break;
2134 case REDIS_LIST: type = "+list"; break;
2135 case REDIS_SET: type = "+set"; break;
ed9b544e 2136 default: type = "unknown"; break;
2137 }
2138 }
2139 addReplySds(c,sdsnew(type));
2140 addReply(c,shared.crlf);
2141}
2142
2143static void saveCommand(redisClient *c) {
f78fd11b 2144 if (rdbSave(server.dbfilename) == REDIS_OK) {
ed9b544e 2145 addReply(c,shared.ok);
2146 } else {
2147 addReply(c,shared.err);
2148 }
2149}
2150
2151static void bgsaveCommand(redisClient *c) {
2152 if (server.bgsaveinprogress) {
2153 addReplySds(c,sdsnew("-ERR background save already in progress\r\n"));
2154 return;
2155 }
f78fd11b 2156 if (rdbSaveBackground(server.dbfilename) == REDIS_OK) {
ed9b544e 2157 addReply(c,shared.ok);
2158 } else {
2159 addReply(c,shared.err);
2160 }
2161}
2162
2163static void shutdownCommand(redisClient *c) {
2164 redisLog(REDIS_WARNING,"User requested shutdown, saving DB...");
f78fd11b 2165 if (rdbSave(server.dbfilename) == REDIS_OK) {
ed329fcf
LH
2166 if (server.daemonize) {
2167 unlink(server.pidfile);
2168 }
ed9b544e 2169 redisLog(REDIS_WARNING,"Server exit now, bye bye...");
2170 exit(1);
2171 } else {
2172 redisLog(REDIS_WARNING,"Error trying to save the DB, can't exit");
2173 addReplySds(c,sdsnew("-ERR can't quit, problems saving the DB\r\n"));
2174 }
2175}
2176
2177static void renameGenericCommand(redisClient *c, int nx) {
2178 dictEntry *de;
2179 robj *o;
2180
2181 /* To use the same key as src and dst is probably an error */
2182 if (sdscmp(c->argv[1]->ptr,c->argv[2]->ptr) == 0) {
c937aa89 2183 addReply(c,shared.sameobjecterr);
ed9b544e 2184 return;
2185 }
2186
2187 de = dictFind(c->dict,c->argv[1]);
2188 if (de == NULL) {
c937aa89 2189 addReply(c,shared.nokeyerr);
ed9b544e 2190 return;
2191 }
2192 o = dictGetEntryVal(de);
2193 incrRefCount(o);
2194 if (dictAdd(c->dict,c->argv[2],o) == DICT_ERR) {
2195 if (nx) {
2196 decrRefCount(o);
c937aa89 2197 addReply(c,shared.czero);
ed9b544e 2198 return;
2199 }
2200 dictReplace(c->dict,c->argv[2],o);
2201 } else {
2202 incrRefCount(c->argv[2]);
2203 }
2204 dictDelete(c->dict,c->argv[1]);
2205 server.dirty++;
c937aa89 2206 addReply(c,nx ? shared.cone : shared.ok);
ed9b544e 2207}
2208
2209static void renameCommand(redisClient *c) {
2210 renameGenericCommand(c,0);
2211}
2212
2213static void renamenxCommand(redisClient *c) {
2214 renameGenericCommand(c,1);
2215}
2216
2217static void moveCommand(redisClient *c) {
2218 dictEntry *de;
2219 robj *o, *key;
2220 dict *src, *dst;
2221 int srcid;
2222
2223 /* Obtain source and target DB pointers */
2224 src = c->dict;
2225 srcid = c->dictid;
2226 if (selectDb(c,atoi(c->argv[2]->ptr)) == REDIS_ERR) {
c937aa89 2227 addReply(c,shared.outofrangeerr);
ed9b544e 2228 return;
2229 }
2230 dst = c->dict;
2231 c->dict = src;
2232 c->dictid = srcid;
2233
2234 /* If the user is moving using as target the same
2235 * DB as the source DB it is probably an error. */
2236 if (src == dst) {
c937aa89 2237 addReply(c,shared.sameobjecterr);
ed9b544e 2238 return;
2239 }
2240
2241 /* Check if the element exists and get a reference */
2242 de = dictFind(c->dict,c->argv[1]);
2243 if (!de) {
c937aa89 2244 addReply(c,shared.czero);
ed9b544e 2245 return;
2246 }
2247
2248 /* Try to add the element to the target DB */
2249 key = dictGetEntryKey(de);
2250 o = dictGetEntryVal(de);
2251 if (dictAdd(dst,key,o) == DICT_ERR) {
c937aa89 2252 addReply(c,shared.czero);
ed9b544e 2253 return;
2254 }
2255 incrRefCount(key);
2256 incrRefCount(o);
2257
2258 /* OK! key moved, free the entry in the source DB */
2259 dictDelete(src,c->argv[1]);
2260 server.dirty++;
c937aa89 2261 addReply(c,shared.cone);
ed9b544e 2262}
2263
2264/* =================================== Lists ================================ */
2265static void pushGenericCommand(redisClient *c, int where) {
2266 robj *lobj;
2267 dictEntry *de;
2268 list *list;
2269
2270 de = dictFind(c->dict,c->argv[1]);
2271 if (de == NULL) {
2272 lobj = createListObject();
2273 list = lobj->ptr;
2274 if (where == REDIS_HEAD) {
2275 if (!listAddNodeHead(list,c->argv[2])) oom("listAddNodeHead");
2276 } else {
2277 if (!listAddNodeTail(list,c->argv[2])) oom("listAddNodeTail");
2278 }
2279 dictAdd(c->dict,c->argv[1],lobj);
2280 incrRefCount(c->argv[1]);
2281 incrRefCount(c->argv[2]);
2282 } else {
2283 lobj = dictGetEntryVal(de);
2284 if (lobj->type != REDIS_LIST) {
2285 addReply(c,shared.wrongtypeerr);
2286 return;
2287 }
2288 list = lobj->ptr;
2289 if (where == REDIS_HEAD) {
2290 if (!listAddNodeHead(list,c->argv[2])) oom("listAddNodeHead");
2291 } else {
2292 if (!listAddNodeTail(list,c->argv[2])) oom("listAddNodeTail");
2293 }
2294 incrRefCount(c->argv[2]);
2295 }
2296 server.dirty++;
2297 addReply(c,shared.ok);
2298}
2299
2300static void lpushCommand(redisClient *c) {
2301 pushGenericCommand(c,REDIS_HEAD);
2302}
2303
2304static void rpushCommand(redisClient *c) {
2305 pushGenericCommand(c,REDIS_TAIL);
2306}
2307
2308static void llenCommand(redisClient *c) {
2309 dictEntry *de;
2310 list *l;
2311
2312 de = dictFind(c->dict,c->argv[1]);
2313 if (de == NULL) {
c937aa89 2314 addReply(c,shared.czero);
ed9b544e 2315 return;
2316 } else {
2317 robj *o = dictGetEntryVal(de);
2318 if (o->type != REDIS_LIST) {
c937aa89 2319 addReply(c,shared.wrongtypeerr);
ed9b544e 2320 } else {
2321 l = o->ptr;
c937aa89 2322 addReplySds(c,sdscatprintf(sdsempty(),":%d\r\n",listLength(l)));
ed9b544e 2323 }
2324 }
2325}
2326
2327static void lindexCommand(redisClient *c) {
2328 dictEntry *de;
2329 int index = atoi(c->argv[2]->ptr);
2330
2331 de = dictFind(c->dict,c->argv[1]);
2332 if (de == NULL) {
c937aa89 2333 addReply(c,shared.nullbulk);
ed9b544e 2334 } else {
2335 robj *o = dictGetEntryVal(de);
2336
2337 if (o->type != REDIS_LIST) {
c937aa89 2338 addReply(c,shared.wrongtypeerr);
ed9b544e 2339 } else {
2340 list *list = o->ptr;
2341 listNode *ln;
2342
2343 ln = listIndex(list, index);
2344 if (ln == NULL) {
c937aa89 2345 addReply(c,shared.nullbulk);
ed9b544e 2346 } else {
2347 robj *ele = listNodeValue(ln);
c937aa89 2348 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",(int)sdslen(ele->ptr)));
ed9b544e 2349 addReply(c,ele);
2350 addReply(c,shared.crlf);
2351 }
2352 }
2353 }
2354}
2355
2356static void lsetCommand(redisClient *c) {
2357 dictEntry *de;
2358 int index = atoi(c->argv[2]->ptr);
2359
2360 de = dictFind(c->dict,c->argv[1]);
2361 if (de == NULL) {
2362 addReply(c,shared.nokeyerr);
2363 } else {
2364 robj *o = dictGetEntryVal(de);
2365
2366 if (o->type != REDIS_LIST) {
2367 addReply(c,shared.wrongtypeerr);
2368 } else {
2369 list *list = o->ptr;
2370 listNode *ln;
2371
2372 ln = listIndex(list, index);
2373 if (ln == NULL) {
c937aa89 2374 addReply(c,shared.outofrangeerr);
ed9b544e 2375 } else {
2376 robj *ele = listNodeValue(ln);
2377
2378 decrRefCount(ele);
2379 listNodeValue(ln) = c->argv[3];
2380 incrRefCount(c->argv[3]);
2381 addReply(c,shared.ok);
2382 server.dirty++;
2383 }
2384 }
2385 }
2386}
2387
2388static void popGenericCommand(redisClient *c, int where) {
2389 dictEntry *de;
2390
2391 de = dictFind(c->dict,c->argv[1]);
2392 if (de == NULL) {
c937aa89 2393 addReply(c,shared.nullbulk);
ed9b544e 2394 } else {
2395 robj *o = dictGetEntryVal(de);
2396
2397 if (o->type != REDIS_LIST) {
c937aa89 2398 addReply(c,shared.wrongtypeerr);
ed9b544e 2399 } else {
2400 list *list = o->ptr;
2401 listNode *ln;
2402
2403 if (where == REDIS_HEAD)
2404 ln = listFirst(list);
2405 else
2406 ln = listLast(list);
2407
2408 if (ln == NULL) {
c937aa89 2409 addReply(c,shared.nullbulk);
ed9b544e 2410 } else {
2411 robj *ele = listNodeValue(ln);
c937aa89 2412 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",(int)sdslen(ele->ptr)));
ed9b544e 2413 addReply(c,ele);
2414 addReply(c,shared.crlf);
2415 listDelNode(list,ln);
2416 server.dirty++;
2417 }
2418 }
2419 }
2420}
2421
2422static void lpopCommand(redisClient *c) {
2423 popGenericCommand(c,REDIS_HEAD);
2424}
2425
2426static void rpopCommand(redisClient *c) {
2427 popGenericCommand(c,REDIS_TAIL);
2428}
2429
2430static void lrangeCommand(redisClient *c) {
2431 dictEntry *de;
2432 int start = atoi(c->argv[2]->ptr);
2433 int end = atoi(c->argv[3]->ptr);
2434
2435 de = dictFind(c->dict,c->argv[1]);
2436 if (de == NULL) {
c937aa89 2437 addReply(c,shared.nullmultibulk);
ed9b544e 2438 } else {
2439 robj *o = dictGetEntryVal(de);
2440
2441 if (o->type != REDIS_LIST) {
c937aa89 2442 addReply(c,shared.wrongtypeerr);
ed9b544e 2443 } else {
2444 list *list = o->ptr;
2445 listNode *ln;
2446 int llen = listLength(list);
2447 int rangelen, j;
2448 robj *ele;
2449
2450 /* convert negative indexes */
2451 if (start < 0) start = llen+start;
2452 if (end < 0) end = llen+end;
2453 if (start < 0) start = 0;
2454 if (end < 0) end = 0;
2455
2456 /* indexes sanity checks */
2457 if (start > end || start >= llen) {
2458 /* Out of range start or start > end result in empty list */
c937aa89 2459 addReply(c,shared.emptymultibulk);
ed9b544e 2460 return;
2461 }
2462 if (end >= llen) end = llen-1;
2463 rangelen = (end-start)+1;
2464
2465 /* Return the result in form of a multi-bulk reply */
2466 ln = listIndex(list, start);
c937aa89 2467 addReplySds(c,sdscatprintf(sdsempty(),"*%d\r\n",rangelen));
ed9b544e 2468 for (j = 0; j < rangelen; j++) {
2469 ele = listNodeValue(ln);
c937aa89 2470 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",(int)sdslen(ele->ptr)));
ed9b544e 2471 addReply(c,ele);
2472 addReply(c,shared.crlf);
2473 ln = ln->next;
2474 }
2475 }
2476 }
2477}
2478
2479static void ltrimCommand(redisClient *c) {
2480 dictEntry *de;
2481 int start = atoi(c->argv[2]->ptr);
2482 int end = atoi(c->argv[3]->ptr);
2483
2484 de = dictFind(c->dict,c->argv[1]);
2485 if (de == NULL) {
2486 addReply(c,shared.nokeyerr);
2487 } else {
2488 robj *o = dictGetEntryVal(de);
2489
2490 if (o->type != REDIS_LIST) {
2491 addReply(c,shared.wrongtypeerr);
2492 } else {
2493 list *list = o->ptr;
2494 listNode *ln;
2495 int llen = listLength(list);
2496 int j, ltrim, rtrim;
2497
2498 /* convert negative indexes */
2499 if (start < 0) start = llen+start;
2500 if (end < 0) end = llen+end;
2501 if (start < 0) start = 0;
2502 if (end < 0) end = 0;
2503
2504 /* indexes sanity checks */
2505 if (start > end || start >= llen) {
2506 /* Out of range start or start > end result in empty list */
2507 ltrim = llen;
2508 rtrim = 0;
2509 } else {
2510 if (end >= llen) end = llen-1;
2511 ltrim = start;
2512 rtrim = llen-end-1;
2513 }
2514
2515 /* Remove list elements to perform the trim */
2516 for (j = 0; j < ltrim; j++) {
2517 ln = listFirst(list);
2518 listDelNode(list,ln);
2519 }
2520 for (j = 0; j < rtrim; j++) {
2521 ln = listLast(list);
2522 listDelNode(list,ln);
2523 }
2524 addReply(c,shared.ok);
2525 server.dirty++;
2526 }
2527 }
2528}
2529
2530static void lremCommand(redisClient *c) {
2531 dictEntry *de;
2532
2533 de = dictFind(c->dict,c->argv[1]);
2534 if (de == NULL) {
7b45bfb2 2535 addReply(c,shared.nokeyerr);
ed9b544e 2536 } else {
2537 robj *o = dictGetEntryVal(de);
2538
2539 if (o->type != REDIS_LIST) {
c937aa89 2540 addReply(c,shared.wrongtypeerr);
ed9b544e 2541 } else {
2542 list *list = o->ptr;
2543 listNode *ln, *next;
2544 int toremove = atoi(c->argv[2]->ptr);
2545 int removed = 0;
2546 int fromtail = 0;
2547
2548 if (toremove < 0) {
2549 toremove = -toremove;
2550 fromtail = 1;
2551 }
2552 ln = fromtail ? list->tail : list->head;
2553 while (ln) {
ed9b544e 2554 robj *ele = listNodeValue(ln);
a4d1ba9a 2555
2556 next = fromtail ? ln->prev : ln->next;
ed9b544e 2557 if (sdscmp(ele->ptr,c->argv[3]->ptr) == 0) {
2558 listDelNode(list,ln);
2559 server.dirty++;
2560 removed++;
2561 if (toremove && removed == toremove) break;
2562 }
2563 ln = next;
2564 }
c937aa89 2565 addReplySds(c,sdscatprintf(sdsempty(),":%d\r\n",removed));
ed9b544e 2566 }
2567 }
2568}
2569
2570/* ==================================== Sets ================================ */
2571
2572static void saddCommand(redisClient *c) {
2573 dictEntry *de;
2574 robj *set;
2575
2576 de = dictFind(c->dict,c->argv[1]);
2577 if (de == NULL) {
2578 set = createSetObject();
2579 dictAdd(c->dict,c->argv[1],set);
2580 incrRefCount(c->argv[1]);
2581 } else {
2582 set = dictGetEntryVal(de);
2583 if (set->type != REDIS_SET) {
c937aa89 2584 addReply(c,shared.wrongtypeerr);
ed9b544e 2585 return;
2586 }
2587 }
2588 if (dictAdd(set->ptr,c->argv[2],NULL) == DICT_OK) {
2589 incrRefCount(c->argv[2]);
2590 server.dirty++;
c937aa89 2591 addReply(c,shared.cone);
ed9b544e 2592 } else {
c937aa89 2593 addReply(c,shared.czero);
ed9b544e 2594 }
2595}
2596
2597static void sremCommand(redisClient *c) {
2598 dictEntry *de;
2599
2600 de = dictFind(c->dict,c->argv[1]);
2601 if (de == NULL) {
c937aa89 2602 addReply(c,shared.czero);
ed9b544e 2603 } else {
2604 robj *set;
2605
2606 set = dictGetEntryVal(de);
2607 if (set->type != REDIS_SET) {
c937aa89 2608 addReply(c,shared.wrongtypeerr);
ed9b544e 2609 return;
2610 }
2611 if (dictDelete(set->ptr,c->argv[2]) == DICT_OK) {
2612 server.dirty++;
c937aa89 2613 addReply(c,shared.cone);
ed9b544e 2614 } else {
c937aa89 2615 addReply(c,shared.czero);
ed9b544e 2616 }
2617 }
2618}
2619
2620static void sismemberCommand(redisClient *c) {
2621 dictEntry *de;
2622
2623 de = dictFind(c->dict,c->argv[1]);
2624 if (de == NULL) {
c937aa89 2625 addReply(c,shared.czero);
ed9b544e 2626 } else {
2627 robj *set;
2628
2629 set = dictGetEntryVal(de);
2630 if (set->type != REDIS_SET) {
c937aa89 2631 addReply(c,shared.wrongtypeerr);
ed9b544e 2632 return;
2633 }
2634 if (dictFind(set->ptr,c->argv[2]))
c937aa89 2635 addReply(c,shared.cone);
ed9b544e 2636 else
c937aa89 2637 addReply(c,shared.czero);
ed9b544e 2638 }
2639}
2640
2641static void scardCommand(redisClient *c) {
2642 dictEntry *de;
2643 dict *s;
2644
2645 de = dictFind(c->dict,c->argv[1]);
2646 if (de == NULL) {
c937aa89 2647 addReply(c,shared.czero);
ed9b544e 2648 return;
2649 } else {
2650 robj *o = dictGetEntryVal(de);
2651 if (o->type != REDIS_SET) {
c937aa89 2652 addReply(c,shared.wrongtypeerr);
ed9b544e 2653 } else {
2654 s = o->ptr;
c937aa89 2655 addReplySds(c,sdscatprintf(sdsempty(),":%d\r\n",
ed9b544e 2656 dictGetHashTableUsed(s)));
2657 }
2658 }
2659}
2660
2661static int qsortCompareSetsByCardinality(const void *s1, const void *s2) {
2662 dict **d1 = (void*) s1, **d2 = (void*) s2;
2663
2664 return dictGetHashTableUsed(*d1)-dictGetHashTableUsed(*d2);
2665}
2666
2667static void sinterGenericCommand(redisClient *c, robj **setskeys, int setsnum, robj *dstkey) {
2668 dict **dv = zmalloc(sizeof(dict*)*setsnum);
2669 dictIterator *di;
2670 dictEntry *de;
2671 robj *lenobj = NULL, *dstset = NULL;
2672 int j, cardinality = 0;
2673
2674 if (!dv) oom("sinterCommand");
2675 for (j = 0; j < setsnum; j++) {
2676 robj *setobj;
2677 dictEntry *de;
2678
2679 de = dictFind(c->dict,setskeys[j]);
2680 if (!de) {
2681 zfree(dv);
c937aa89 2682 addReply(c,shared.nokeyerr);
ed9b544e 2683 return;
2684 }
2685 setobj = dictGetEntryVal(de);
2686 if (setobj->type != REDIS_SET) {
2687 zfree(dv);
c937aa89 2688 addReply(c,shared.wrongtypeerr);
ed9b544e 2689 return;
2690 }
2691 dv[j] = setobj->ptr;
2692 }
2693 /* Sort sets from the smallest to largest, this will improve our
2694 * algorithm's performace */
2695 qsort(dv,setsnum,sizeof(dict*),qsortCompareSetsByCardinality);
2696
2697 /* The first thing we should output is the total number of elements...
2698 * since this is a multi-bulk write, but at this stage we don't know
2699 * the intersection set size, so we use a trick, append an empty object
2700 * to the output list and save the pointer to later modify it with the
2701 * right length */
2702 if (!dstkey) {
2703 lenobj = createObject(REDIS_STRING,NULL);
2704 addReply(c,lenobj);
2705 decrRefCount(lenobj);
2706 } else {
2707 /* If we have a target key where to store the resulting set
2708 * create this key with an empty set inside */
2709 dstset = createSetObject();
2710 dictDelete(c->dict,dstkey);
2711 dictAdd(c->dict,dstkey,dstset);
2712 incrRefCount(dstkey);
2713 }
2714
2715 /* Iterate all the elements of the first (smallest) set, and test
2716 * the element against all the other sets, if at least one set does
2717 * not include the element it is discarded */
2718 di = dictGetIterator(dv[0]);
2719 if (!di) oom("dictGetIterator");
2720
2721 while((de = dictNext(di)) != NULL) {
2722 robj *ele;
2723
2724 for (j = 1; j < setsnum; j++)
2725 if (dictFind(dv[j],dictGetEntryKey(de)) == NULL) break;
2726 if (j != setsnum)
2727 continue; /* at least one set does not contain the member */
2728 ele = dictGetEntryKey(de);
2729 if (!dstkey) {
c937aa89 2730 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",sdslen(ele->ptr)));
ed9b544e 2731 addReply(c,ele);
2732 addReply(c,shared.crlf);
2733 cardinality++;
2734 } else {
2735 dictAdd(dstset->ptr,ele,NULL);
2736 incrRefCount(ele);
2737 }
2738 }
2739 dictReleaseIterator(di);
2740
2741 if (!dstkey)
c937aa89 2742 lenobj->ptr = sdscatprintf(sdsempty(),"*%d\r\n",cardinality);
ed9b544e 2743 else
2744 addReply(c,shared.ok);
2745 zfree(dv);
2746}
2747
2748static void sinterCommand(redisClient *c) {
2749 sinterGenericCommand(c,c->argv+1,c->argc-1,NULL);
2750}
2751
2752static void sinterstoreCommand(redisClient *c) {
2753 sinterGenericCommand(c,c->argv+2,c->argc-2,c->argv[1]);
2754}
2755
2756static void flushdbCommand(redisClient *c) {
2757 dictEmpty(c->dict);
2758 addReply(c,shared.ok);
f78fd11b 2759 rdbSave(server.dbfilename);
ed9b544e 2760}
2761
2762static void flushallCommand(redisClient *c) {
2763 emptyDb();
2764 addReply(c,shared.ok);
f78fd11b 2765 rdbSave(server.dbfilename);
ed9b544e 2766}
2767
2768redisSortOperation *createSortOperation(int type, robj *pattern) {
2769 redisSortOperation *so = zmalloc(sizeof(*so));
2770 if (!so) oom("createSortOperation");
2771 so->type = type;
2772 so->pattern = pattern;
2773 return so;
2774}
2775
2776/* Return the value associated to the key with a name obtained
2777 * substituting the first occurence of '*' in 'pattern' with 'subst' */
2778robj *lookupKeyByPattern(dict *dict, robj *pattern, robj *subst) {
2779 char *p;
2780 sds spat, ssub;
2781 robj keyobj;
2782 int prefixlen, sublen, postfixlen;
2783 dictEntry *de;
2784 /* Expoit the internal sds representation to create a sds string allocated on the stack in order to make this function faster */
2785 struct {
2786 long len;
2787 long free;
2788 char buf[REDIS_SORTKEY_MAX+1];
2789 } keyname;
2790
2791
2792 spat = pattern->ptr;
2793 ssub = subst->ptr;
2794 if (sdslen(spat)+sdslen(ssub)-1 > REDIS_SORTKEY_MAX) return NULL;
2795 p = strchr(spat,'*');
2796 if (!p) return NULL;
2797
2798 prefixlen = p-spat;
2799 sublen = sdslen(ssub);
2800 postfixlen = sdslen(spat)-(prefixlen+1);
2801 memcpy(keyname.buf,spat,prefixlen);
2802 memcpy(keyname.buf+prefixlen,ssub,sublen);
2803 memcpy(keyname.buf+prefixlen+sublen,p+1,postfixlen);
2804 keyname.buf[prefixlen+sublen+postfixlen] = '\0';
2805 keyname.len = prefixlen+sublen+postfixlen;
2806
2807 keyobj.refcount = 1;
2808 keyobj.type = REDIS_STRING;
2809 keyobj.ptr = ((char*)&keyname)+(sizeof(long)*2);
2810
2811 de = dictFind(dict,&keyobj);
a4d1ba9a 2812 /* printf("lookup '%s' => %p\n", keyname.buf,de); */
ed9b544e 2813 if (!de) return NULL;
2814 return dictGetEntryVal(de);
2815}
2816
2817/* sortCompare() is used by qsort in sortCommand(). Given that qsort_r with
2818 * the additional parameter is not standard but a BSD-specific we have to
2819 * pass sorting parameters via the global 'server' structure */
2820static int sortCompare(const void *s1, const void *s2) {
2821 const redisSortObject *so1 = s1, *so2 = s2;
2822 int cmp;
2823
2824 if (!server.sort_alpha) {
2825 /* Numeric sorting. Here it's trivial as we precomputed scores */
2826 if (so1->u.score > so2->u.score) {
2827 cmp = 1;
2828 } else if (so1->u.score < so2->u.score) {
2829 cmp = -1;
2830 } else {
2831 cmp = 0;
2832 }
2833 } else {
2834 /* Alphanumeric sorting */
2835 if (server.sort_bypattern) {
2836 if (!so1->u.cmpobj || !so2->u.cmpobj) {
2837 /* At least one compare object is NULL */
2838 if (so1->u.cmpobj == so2->u.cmpobj)
2839 cmp = 0;
2840 else if (so1->u.cmpobj == NULL)
2841 cmp = -1;
2842 else
2843 cmp = 1;
2844 } else {
2845 /* We have both the objects, use strcoll */
2846 cmp = strcoll(so1->u.cmpobj->ptr,so2->u.cmpobj->ptr);
2847 }
2848 } else {
2849 /* Compare elements directly */
2850 cmp = strcoll(so1->obj->ptr,so2->obj->ptr);
2851 }
2852 }
2853 return server.sort_desc ? -cmp : cmp;
2854}
2855
2856/* The SORT command is the most complex command in Redis. Warning: this code
2857 * is optimized for speed and a bit less for readability */
2858static void sortCommand(redisClient *c) {
2859 dictEntry *de;
2860 list *operations;
2861 int outputlen = 0;
2862 int desc = 0, alpha = 0;
2863 int limit_start = 0, limit_count = -1, start, end;
2864 int j, dontsort = 0, vectorlen;
2865 int getop = 0; /* GET operation counter */
2866 robj *sortval, *sortby = NULL;
2867 redisSortObject *vector; /* Resulting vector to sort */
2868
2869 /* Lookup the key to sort. It must be of the right types */
2870 de = dictFind(c->dict,c->argv[1]);
2871 if (de == NULL) {
c937aa89 2872 addReply(c,shared.nokeyerr);
ed9b544e 2873 return;
2874 }
2875 sortval = dictGetEntryVal(de);
2876 if (sortval->type != REDIS_SET && sortval->type != REDIS_LIST) {
c937aa89 2877 addReply(c,shared.wrongtypeerr);
ed9b544e 2878 return;
2879 }
2880
2881 /* Create a list of operations to perform for every sorted element.
2882 * Operations can be GET/DEL/INCR/DECR */
2883 operations = listCreate();
092dac2a 2884 listSetFreeMethod(operations,zfree);
ed9b544e 2885 j = 2;
2886
2887 /* Now we need to protect sortval incrementing its count, in the future
2888 * SORT may have options able to overwrite/delete keys during the sorting
2889 * and the sorted key itself may get destroied */
2890 incrRefCount(sortval);
2891
2892 /* The SORT command has an SQL-alike syntax, parse it */
2893 while(j < c->argc) {
2894 int leftargs = c->argc-j-1;
2895 if (!strcasecmp(c->argv[j]->ptr,"asc")) {
2896 desc = 0;
2897 } else if (!strcasecmp(c->argv[j]->ptr,"desc")) {
2898 desc = 1;
2899 } else if (!strcasecmp(c->argv[j]->ptr,"alpha")) {
2900 alpha = 1;
2901 } else if (!strcasecmp(c->argv[j]->ptr,"limit") && leftargs >= 2) {
2902 limit_start = atoi(c->argv[j+1]->ptr);
2903 limit_count = atoi(c->argv[j+2]->ptr);
2904 j+=2;
2905 } else if (!strcasecmp(c->argv[j]->ptr,"by") && leftargs >= 1) {
2906 sortby = c->argv[j+1];
2907 /* If the BY pattern does not contain '*', i.e. it is constant,
2908 * we don't need to sort nor to lookup the weight keys. */
2909 if (strchr(c->argv[j+1]->ptr,'*') == NULL) dontsort = 1;
2910 j++;
2911 } else if (!strcasecmp(c->argv[j]->ptr,"get") && leftargs >= 1) {
2912 listAddNodeTail(operations,createSortOperation(
2913 REDIS_SORT_GET,c->argv[j+1]));
2914 getop++;
2915 j++;
2916 } else if (!strcasecmp(c->argv[j]->ptr,"del") && leftargs >= 1) {
2917 listAddNodeTail(operations,createSortOperation(
2918 REDIS_SORT_DEL,c->argv[j+1]));
2919 j++;
2920 } else if (!strcasecmp(c->argv[j]->ptr,"incr") && leftargs >= 1) {
2921 listAddNodeTail(operations,createSortOperation(
2922 REDIS_SORT_INCR,c->argv[j+1]));
2923 j++;
2924 } else if (!strcasecmp(c->argv[j]->ptr,"get") && leftargs >= 1) {
2925 listAddNodeTail(operations,createSortOperation(
2926 REDIS_SORT_DECR,c->argv[j+1]));
2927 j++;
2928 } else {
2929 decrRefCount(sortval);
2930 listRelease(operations);
c937aa89 2931 addReply(c,shared.syntaxerr);
ed9b544e 2932 return;
2933 }
2934 j++;
2935 }
2936
2937 /* Load the sorting vector with all the objects to sort */
2938 vectorlen = (sortval->type == REDIS_LIST) ?
2939 listLength((list*)sortval->ptr) :
2940 dictGetHashTableUsed((dict*)sortval->ptr);
2941 vector = zmalloc(sizeof(redisSortObject)*vectorlen);
2942 if (!vector) oom("allocating objects vector for SORT");
2943 j = 0;
2944 if (sortval->type == REDIS_LIST) {
2945 list *list = sortval->ptr;
2946 listNode *ln = list->head;
2947 while(ln) {
2948 robj *ele = ln->value;
2949 vector[j].obj = ele;
2950 vector[j].u.score = 0;
2951 vector[j].u.cmpobj = NULL;
2952 ln = ln->next;
2953 j++;
2954 }
2955 } else {
2956 dict *set = sortval->ptr;
2957 dictIterator *di;
2958 dictEntry *setele;
2959
2960 di = dictGetIterator(set);
2961 if (!di) oom("dictGetIterator");
2962 while((setele = dictNext(di)) != NULL) {
2963 vector[j].obj = dictGetEntryKey(setele);
2964 vector[j].u.score = 0;
2965 vector[j].u.cmpobj = NULL;
2966 j++;
2967 }
2968 dictReleaseIterator(di);
2969 }
2970 assert(j == vectorlen);
2971
2972 /* Now it's time to load the right scores in the sorting vector */
2973 if (dontsort == 0) {
2974 for (j = 0; j < vectorlen; j++) {
2975 if (sortby) {
2976 robj *byval;
2977
2978 byval = lookupKeyByPattern(c->dict,sortby,vector[j].obj);
2979 if (!byval || byval->type != REDIS_STRING) continue;
2980 if (alpha) {
2981 vector[j].u.cmpobj = byval;
2982 incrRefCount(byval);
2983 } else {
2984 vector[j].u.score = strtod(byval->ptr,NULL);
2985 }
2986 } else {
2987 if (!alpha) vector[j].u.score = strtod(vector[j].obj->ptr,NULL);
2988 }
2989 }
2990 }
2991
2992 /* We are ready to sort the vector... perform a bit of sanity check
2993 * on the LIMIT option too. We'll use a partial version of quicksort. */
2994 start = (limit_start < 0) ? 0 : limit_start;
2995 end = (limit_count < 0) ? vectorlen-1 : start+limit_count-1;
2996 if (start >= vectorlen) {
2997 start = vectorlen-1;
2998 end = vectorlen-2;
2999 }
3000 if (end >= vectorlen) end = vectorlen-1;
3001
3002 if (dontsort == 0) {
3003 server.sort_desc = desc;
3004 server.sort_alpha = alpha;
3005 server.sort_bypattern = sortby ? 1 : 0;
3006 qsort(vector,vectorlen,sizeof(redisSortObject),sortCompare);
3007 }
3008
3009 /* Send command output to the output buffer, performing the specified
3010 * GET/DEL/INCR/DECR operations if any. */
3011 outputlen = getop ? getop*(end-start+1) : end-start+1;
c937aa89 3012 addReplySds(c,sdscatprintf(sdsempty(),"*%d\r\n",outputlen));
ed9b544e 3013 for (j = start; j <= end; j++) {
3014 listNode *ln = operations->head;
3015 if (!getop) {
c937aa89 3016 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",
ed9b544e 3017 sdslen(vector[j].obj->ptr)));
3018 addReply(c,vector[j].obj);
3019 addReply(c,shared.crlf);
3020 }
3021 while(ln) {
3022 redisSortOperation *sop = ln->value;
3023 robj *val = lookupKeyByPattern(c->dict,sop->pattern,
3024 vector[j].obj);
3025
3026 if (sop->type == REDIS_SORT_GET) {
3027 if (!val || val->type != REDIS_STRING) {
9eb00f21 3028 addReply(c,shared.nullbulk);
ed9b544e 3029 } else {
c937aa89 3030 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",
ed9b544e 3031 sdslen(val->ptr)));
3032 addReply(c,val);
3033 addReply(c,shared.crlf);
3034 }
3035 } else if (sop->type == REDIS_SORT_DEL) {
3036 /* TODO */
3037 }
3038 ln = ln->next;
3039 }
3040 }
3041
3042 /* Cleanup */
3043 decrRefCount(sortval);
3044 listRelease(operations);
3045 for (j = 0; j < vectorlen; j++) {
3046 if (sortby && alpha && vector[j].u.cmpobj)
3047 decrRefCount(vector[j].u.cmpobj);
3048 }
3049 zfree(vector);
3050}
3051
3052static void infoCommand(redisClient *c) {
3053 sds info;
3054 time_t uptime = time(NULL)-server.stat_starttime;
3055
3056 info = sdscatprintf(sdsempty(),
3057 "redis_version:%s\r\n"
3058 "connected_clients:%d\r\n"
3059 "connected_slaves:%d\r\n"
3060 "used_memory:%d\r\n"
3061 "changes_since_last_save:%lld\r\n"
3062 "last_save_time:%d\r\n"
3063 "total_connections_received:%lld\r\n"
3064 "total_commands_processed:%lld\r\n"
3065 "uptime_in_seconds:%d\r\n"
3066 "uptime_in_days:%d\r\n"
3067 ,REDIS_VERSION,
3068 listLength(server.clients)-listLength(server.slaves),
3069 listLength(server.slaves),
3070 server.usedmemory,
3071 server.dirty,
3072 server.lastsave,
3073 server.stat_numconnections,
3074 server.stat_numcommands,
3075 uptime,
3076 uptime/(3600*24)
3077 );
c937aa89 3078 addReplySds(c,sdscatprintf(sdsempty(),"$%d\r\n",sdslen(info)));
ed9b544e 3079 addReplySds(c,info);
70003d28 3080 addReply(c,shared.crlf);
ed9b544e 3081}
3082
3083/* =============================== Replication ============================= */
3084
3085/* Send the whole output buffer syncronously to the slave. This a general operation in theory, but it is actually useful only for replication. */
3086static int flushClientOutput(redisClient *c) {
3087 int retval;
3088 time_t start = time(NULL);
3089
3090 while(listLength(c->reply)) {
3091 if (time(NULL)-start > 5) return REDIS_ERR; /* 5 seconds timeout */
3092 retval = aeWait(c->fd,AE_WRITABLE,1000);
3093 if (retval == -1) {
3094 return REDIS_ERR;
3095 } else if (retval & AE_WRITABLE) {
3096 sendReplyToClient(NULL, c->fd, c, AE_WRITABLE);
3097 }
3098 }
3099 return REDIS_OK;
3100}
3101
a4d1ba9a 3102static int syncWrite(int fd, char *ptr, ssize_t size, int timeout) {
ed9b544e 3103 ssize_t nwritten, ret = size;
3104 time_t start = time(NULL);
3105
3106 timeout++;
3107 while(size) {
3108 if (aeWait(fd,AE_WRITABLE,1000) & AE_WRITABLE) {
3109 nwritten = write(fd,ptr,size);
3110 if (nwritten == -1) return -1;
3111 ptr += nwritten;
3112 size -= nwritten;
3113 }
3114 if ((time(NULL)-start) > timeout) {
3115 errno = ETIMEDOUT;
3116 return -1;
3117 }
3118 }
3119 return ret;
3120}
3121
a4d1ba9a 3122static int syncRead(int fd, char *ptr, ssize_t size, int timeout) {
ed9b544e 3123 ssize_t nread, totread = 0;
3124 time_t start = time(NULL);
3125
3126 timeout++;
3127 while(size) {
3128 if (aeWait(fd,AE_READABLE,1000) & AE_READABLE) {
3129 nread = read(fd,ptr,size);
3130 if (nread == -1) return -1;
3131 ptr += nread;
3132 size -= nread;
3133 totread += nread;
3134 }
3135 if ((time(NULL)-start) > timeout) {
3136 errno = ETIMEDOUT;
3137 return -1;
3138 }
3139 }
3140 return totread;
3141}
3142
3143static int syncReadLine(int fd, char *ptr, ssize_t size, int timeout) {
3144 ssize_t nread = 0;
3145
3146 size--;
3147 while(size) {
3148 char c;
3149
3150 if (syncRead(fd,&c,1,timeout) == -1) return -1;
3151 if (c == '\n') {
3152 *ptr = '\0';
3153 if (nread && *(ptr-1) == '\r') *(ptr-1) = '\0';
3154 return nread;
3155 } else {
3156 *ptr++ = c;
3157 *ptr = '\0';
3158 nread++;
3159 }
3160 }
3161 return nread;
3162}
3163
3164static void syncCommand(redisClient *c) {
3165 struct stat sb;
3166 int fd = -1, len;
3167 time_t start = time(NULL);
3168 char sizebuf[32];
3169
cf3f0c01 3170 /* ignore SYNC if aleady slave or in monitor mode */
3171 if (c->flags & REDIS_SLAVE) return;
3172
ed9b544e 3173 redisLog(REDIS_NOTICE,"Slave ask for syncronization");
f78fd11b 3174 if (flushClientOutput(c) == REDIS_ERR ||
3175 rdbSave(server.dbfilename) != REDIS_OK)
ed9b544e 3176 goto closeconn;
3177
3178 fd = open(server.dbfilename, O_RDONLY);
3179 if (fd == -1 || fstat(fd,&sb) == -1) goto closeconn;
3180 len = sb.st_size;
3181
c937aa89 3182 snprintf(sizebuf,32,"$%d\r\n",len);
ed9b544e 3183 if (syncWrite(c->fd,sizebuf,strlen(sizebuf),5) == -1) goto closeconn;
3184 while(len) {
3185 char buf[1024];
3186 int nread;
3187
3188 if (time(NULL)-start > REDIS_MAX_SYNC_TIME) goto closeconn;
3189 nread = read(fd,buf,1024);
3190 if (nread == -1) goto closeconn;
3191 len -= nread;
3192 if (syncWrite(c->fd,buf,nread,5) == -1) goto closeconn;
3193 }
3194 if (syncWrite(c->fd,"\r\n",2,5) == -1) goto closeconn;
3195 close(fd);
3196 c->flags |= REDIS_SLAVE;
3197 c->slaveseldb = 0;
3198 if (!listAddNodeTail(server.slaves,c)) oom("listAddNodeTail");
3199 redisLog(REDIS_NOTICE,"Syncronization with slave succeeded");
3200 return;
3201
3202closeconn:
3203 if (fd != -1) close(fd);
3204 c->flags |= REDIS_CLOSE;
3205 redisLog(REDIS_WARNING,"Syncronization with slave failed");
3206 return;
3207}
3208
3209static int syncWithMaster(void) {
3210 char buf[1024], tmpfile[256];
3211 int dumpsize;
3212 int fd = anetTcpConnect(NULL,server.masterhost,server.masterport);
3213 int dfd;
3214
3215 if (fd == -1) {
3216 redisLog(REDIS_WARNING,"Unable to connect to MASTER: %s",
3217 strerror(errno));
3218 return REDIS_ERR;
3219 }
3220 /* Issue the SYNC command */
3221 if (syncWrite(fd,"SYNC \r\n",7,5) == -1) {
3222 close(fd);
3223 redisLog(REDIS_WARNING,"I/O error writing to MASTER: %s",
3224 strerror(errno));
3225 return REDIS_ERR;
3226 }
3227 /* Read the bulk write count */
3228 if (syncReadLine(fd,buf,1024,5) == -1) {
3229 close(fd);
3230 redisLog(REDIS_WARNING,"I/O error reading bulk count from MASTER: %s",
3231 strerror(errno));
3232 return REDIS_ERR;
3233 }
c937aa89 3234 dumpsize = atoi(buf+1);
ed9b544e 3235 redisLog(REDIS_NOTICE,"Receiving %d bytes data dump from MASTER",dumpsize);
3236 /* Read the bulk write data on a temp file */
3237 snprintf(tmpfile,256,"temp-%d.%ld.rdb",(int)time(NULL),(long int)random());
3238 dfd = open(tmpfile,O_CREAT|O_WRONLY,0644);
3239 if (dfd == -1) {
3240 close(fd);
3241 redisLog(REDIS_WARNING,"Opening the temp file needed for MASTER <-> SLAVE synchronization: %s",strerror(errno));
3242 return REDIS_ERR;
3243 }
3244 while(dumpsize) {
3245 int nread, nwritten;
3246
3247 nread = read(fd,buf,(dumpsize < 1024)?dumpsize:1024);
3248 if (nread == -1) {
3249 redisLog(REDIS_WARNING,"I/O error trying to sync with MASTER: %s",
3250 strerror(errno));
3251 close(fd);
3252 close(dfd);
3253 return REDIS_ERR;
3254 }
3255 nwritten = write(dfd,buf,nread);
3256 if (nwritten == -1) {
3257 redisLog(REDIS_WARNING,"Write error writing to the DB dump file needed for MASTER <-> SLAVE synchrnonization: %s", strerror(errno));
3258 close(fd);
3259 close(dfd);
3260 return REDIS_ERR;
3261 }
3262 dumpsize -= nread;
3263 }
3264 close(dfd);
3265 if (rename(tmpfile,server.dbfilename) == -1) {
3266 redisLog(REDIS_WARNING,"Failed trying to rename the temp DB into dump.rdb in MASTER <-> SLAVE synchronization: %s", strerror(errno));
3267 unlink(tmpfile);
3268 close(fd);
3269 return REDIS_ERR;
3270 }
3271 emptyDb();
f78fd11b 3272 if (rdbLoad(server.dbfilename) != REDIS_OK) {
ed9b544e 3273 redisLog(REDIS_WARNING,"Failed trying to load the MASTER synchronization DB from disk");
3274 close(fd);
3275 return REDIS_ERR;
3276 }
3277 server.master = createClient(fd);
3278 server.master->flags |= REDIS_MASTER;
3279 server.replstate = REDIS_REPL_CONNECTED;
3280 return REDIS_OK;
3281}
3282
87eca727 3283static void monitorCommand(redisClient *c) {
cf3f0c01 3284 /* ignore MONITOR if aleady slave or in monitor mode */
3285 if (c->flags & REDIS_SLAVE) return;
3286
87eca727 3287 c->flags |= (REDIS_SLAVE|REDIS_MONITOR);
3288 c->slaveseldb = 0;
3289 if (!listAddNodeTail(server.monitors,c)) oom("listAddNodeTail");
3290 addReply(c,shared.ok);
3291}
3292
ed9b544e 3293/* =================================== Main! ================================ */
3294
3295static void daemonize(void) {
3296 int fd;
3297 FILE *fp;
3298
3299 if (fork() != 0) exit(0); /* parent exits */
3300 setsid(); /* create a new session */
3301
3302 /* Every output goes to /dev/null. If Redis is daemonized but
3303 * the 'logfile' is set to 'stdout' in the configuration file
3304 * it will not log at all. */
3305 if ((fd = open("/dev/null", O_RDWR, 0)) != -1) {
3306 dup2(fd, STDIN_FILENO);
3307 dup2(fd, STDOUT_FILENO);
3308 dup2(fd, STDERR_FILENO);
3309 if (fd > STDERR_FILENO) close(fd);
3310 }
3311 /* Try to write the pid file */
ed329fcf 3312 fp = fopen(server.pidfile,"w");
ed9b544e 3313 if (fp) {
3314 fprintf(fp,"%d\n",getpid());
3315 fclose(fp);
3316 }
3317}
3318
3319int main(int argc, char **argv) {
3320 initServerConfig();
3321 if (argc == 2) {
3322 ResetServerSaveParams();
3323 loadServerConfig(argv[1]);
3324 } else if (argc > 2) {
3325 fprintf(stderr,"Usage: ./redis-server [/path/to/redis.conf]\n");
3326 exit(1);
3327 }
3328 initServer();
3329 if (server.daemonize) daemonize();
3330 redisLog(REDIS_NOTICE,"Server started, Redis version " REDIS_VERSION);
f78fd11b 3331 if (rdbLoad(server.dbfilename) == REDIS_OK)
ed9b544e 3332 redisLog(REDIS_NOTICE,"DB loaded from disk");
3333 if (aeCreateFileEvent(server.el, server.fd, AE_READABLE,
3334 acceptHandler, NULL, NULL) == AE_ERR) oom("creating file event");
46713f83 3335 redisLog(REDIS_NOTICE,"The server is now ready to accept connections on port %d", server.port);
ed9b544e 3336 aeMain(server.el);
3337 aeDeleteEventLoop(server.el);
3338 return 0;
3339}