]> git.saurik.com Git - redis.git/blob - src/redis.c
Scripting: add helper functions redis.error_reply() and redis.status_reply().
[redis.git] / src / redis.c
1 /*
2 * Copyright (c) 2009-2010, 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
30 #include "redis.h"
31 #include "slowlog.h"
32 #include "bio.h"
33
34 #include <time.h>
35 #include <signal.h>
36 #include <sys/wait.h>
37 #include <errno.h>
38 #include <assert.h>
39 #include <ctype.h>
40 #include <stdarg.h>
41 #include <arpa/inet.h>
42 #include <sys/stat.h>
43 #include <fcntl.h>
44 #include <sys/time.h>
45 #include <sys/resource.h>
46 #include <sys/uio.h>
47 #include <limits.h>
48 #include <float.h>
49 #include <math.h>
50 #include <sys/resource.h>
51 #include <sys/utsname.h>
52
53 /* Our shared "common" objects */
54
55 struct sharedObjectsStruct shared;
56
57 /* Global vars that are actually used as constants. The following double
58 * values are used for double on-disk serialization, and are initialized
59 * at runtime to avoid strange compiler optimizations. */
60
61 double R_Zero, R_PosInf, R_NegInf, R_Nan;
62
63 /*================================= Globals ================================= */
64
65 /* Global vars */
66 struct redisServer server; /* server global state */
67 struct redisCommand *commandTable;
68
69 /* Our command table.
70 *
71 * Every entry is composed of the following fields:
72 *
73 * name: a string representing the command name.
74 * function: pointer to the C function implementing the command.
75 * arity: number of arguments, it is possible to use -N to say >= N
76 * sflags: command flags as string. See below for a table of flags.
77 * flags: flags as bitmask. Computed by Redis using the 'sflags' field.
78 * get_keys_proc: an optional function to get key arguments from a command.
79 * This is only used when the following three fields are not
80 * enough to specify what arguments are keys.
81 * first_key_index: first argument that is a key
82 * last_key_index: last argument that is a key
83 * key_step: step to get all the keys from first to last argument. For instance
84 * in MSET the step is two since arguments are key,val,key,val,...
85 * microseconds: microseconds of total execution time for this command.
86 * calls: total number of calls of this command.
87 *
88 * The flags, microseconds and calls fields are computed by Redis and should
89 * always be set to zero.
90 *
91 * Command flags are expressed using strings where every character represents
92 * a flag. Later the populateCommandTable() function will take care of
93 * populating the real 'flags' field using this characters.
94 *
95 * This is the meaning of the flags:
96 *
97 * w: write command (may modify the key space).
98 * r: read command (will never modify the key space).
99 * m: may increase memory usage once called. Don't allow if out of memory.
100 * a: admin command, like SAVE or SHUTDOWN.
101 * p: Pub/Sub related command.
102 * f: force replication of this command, regarless of server.dirty.
103 * s: command not allowed in scripts.
104 * R: random command. Command is not deterministic, that is, the same command
105 * with the same arguments, with the same key space, may have different
106 * results. For instance SPOP and RANDOMKEY are two random commands.
107 * S: Sort command output array if called from script, so that the output
108 * is deterministic.
109 * l: Allow command while loading the database.
110 * t: Allow command while a slave has stale data but is not allowed to
111 * server this data. Normally no command is accepted in this condition
112 * but just a few.
113 */
114 struct redisCommand redisCommandTable[] = {
115 {"get",getCommand,2,"r",0,NULL,1,1,1,0,0},
116 {"set",setCommand,3,"wm",0,noPreloadGetKeys,1,1,1,0,0},
117 {"setnx",setnxCommand,3,"wm",0,noPreloadGetKeys,1,1,1,0,0},
118 {"setex",setexCommand,4,"wm",0,noPreloadGetKeys,1,1,1,0,0},
119 {"psetex",psetexCommand,4,"wm",0,noPreloadGetKeys,1,1,1,0,0},
120 {"append",appendCommand,3,"wm",0,NULL,1,1,1,0,0},
121 {"strlen",strlenCommand,2,"r",0,NULL,1,1,1,0,0},
122 {"del",delCommand,-2,"w",0,noPreloadGetKeys,1,-1,1,0,0},
123 {"exists",existsCommand,2,"r",0,NULL,1,1,1,0,0},
124 {"setbit",setbitCommand,4,"wm",0,NULL,1,1,1,0,0},
125 {"getbit",getbitCommand,3,"r",0,NULL,1,1,1,0,0},
126 {"setrange",setrangeCommand,4,"wm",0,NULL,1,1,1,0,0},
127 {"getrange",getrangeCommand,4,"r",0,NULL,1,1,1,0,0},
128 {"substr",getrangeCommand,4,"r",0,NULL,1,1,1,0,0},
129 {"incr",incrCommand,2,"wm",0,NULL,1,1,1,0,0},
130 {"decr",decrCommand,2,"wm",0,NULL,1,1,1,0,0},
131 {"mget",mgetCommand,-2,"r",0,NULL,1,-1,1,0,0},
132 {"rpush",rpushCommand,-3,"wm",0,NULL,1,1,1,0,0},
133 {"lpush",lpushCommand,-3,"wm",0,NULL,1,1,1,0,0},
134 {"rpushx",rpushxCommand,3,"wm",0,NULL,1,1,1,0,0},
135 {"lpushx",lpushxCommand,3,"wm",0,NULL,1,1,1,0,0},
136 {"linsert",linsertCommand,5,"wm",0,NULL,1,1,1,0,0},
137 {"rpop",rpopCommand,2,"w",0,NULL,1,1,1,0,0},
138 {"lpop",lpopCommand,2,"w",0,NULL,1,1,1,0,0},
139 {"brpop",brpopCommand,-3,"ws",0,NULL,1,1,1,0,0},
140 {"brpoplpush",brpoplpushCommand,4,"wms",0,NULL,1,2,1,0,0},
141 {"blpop",blpopCommand,-3,"ws",0,NULL,1,-2,1,0,0},
142 {"llen",llenCommand,2,"r",0,NULL,1,1,1,0,0},
143 {"lindex",lindexCommand,3,"r",0,NULL,1,1,1,0,0},
144 {"lset",lsetCommand,4,"wm",0,NULL,1,1,1,0,0},
145 {"lrange",lrangeCommand,4,"r",0,NULL,1,1,1,0,0},
146 {"ltrim",ltrimCommand,4,"w",0,NULL,1,1,1,0,0},
147 {"lrem",lremCommand,4,"w",0,NULL,1,1,1,0,0},
148 {"rpoplpush",rpoplpushCommand,3,"wm",0,NULL,1,2,1,0,0},
149 {"sadd",saddCommand,-3,"wm",0,NULL,1,1,1,0,0},
150 {"srem",sremCommand,-3,"w",0,NULL,1,1,1,0,0},
151 {"smove",smoveCommand,4,"w",0,NULL,1,2,1,0,0},
152 {"sismember",sismemberCommand,3,"r",0,NULL,1,1,1,0,0},
153 {"scard",scardCommand,2,"r",0,NULL,1,1,1,0,0},
154 {"spop",spopCommand,2,"wRs",0,NULL,1,1,1,0,0},
155 {"srandmember",srandmemberCommand,-2,"rR",0,NULL,1,1,1,0,0},
156 {"sinter",sinterCommand,-2,"rS",0,NULL,1,-1,1,0,0},
157 {"sinterstore",sinterstoreCommand,-3,"wm",0,NULL,1,-1,1,0,0},
158 {"sunion",sunionCommand,-2,"rS",0,NULL,1,-1,1,0,0},
159 {"sunionstore",sunionstoreCommand,-3,"wm",0,NULL,1,-1,1,0,0},
160 {"sdiff",sdiffCommand,-2,"rS",0,NULL,1,-1,1,0,0},
161 {"sdiffstore",sdiffstoreCommand,-3,"wm",0,NULL,1,-1,1,0,0},
162 {"smembers",sinterCommand,2,"rS",0,NULL,1,1,1,0,0},
163 {"zadd",zaddCommand,-4,"wm",0,NULL,1,1,1,0,0},
164 {"zincrby",zincrbyCommand,4,"wm",0,NULL,1,1,1,0,0},
165 {"zrem",zremCommand,-3,"w",0,NULL,1,1,1,0,0},
166 {"zremrangebyscore",zremrangebyscoreCommand,4,"w",0,NULL,1,1,1,0,0},
167 {"zremrangebyrank",zremrangebyrankCommand,4,"w",0,NULL,1,1,1,0,0},
168 {"zunionstore",zunionstoreCommand,-4,"wm",0,zunionInterGetKeys,0,0,0,0,0},
169 {"zinterstore",zinterstoreCommand,-4,"wm",0,zunionInterGetKeys,0,0,0,0,0},
170 {"zrange",zrangeCommand,-4,"r",0,NULL,1,1,1,0,0},
171 {"zrangebyscore",zrangebyscoreCommand,-4,"r",0,NULL,1,1,1,0,0},
172 {"zrevrangebyscore",zrevrangebyscoreCommand,-4,"r",0,NULL,1,1,1,0,0},
173 {"zcount",zcountCommand,4,"r",0,NULL,1,1,1,0,0},
174 {"zrevrange",zrevrangeCommand,-4,"r",0,NULL,1,1,1,0,0},
175 {"zcard",zcardCommand,2,"r",0,NULL,1,1,1,0,0},
176 {"zscore",zscoreCommand,3,"r",0,NULL,1,1,1,0,0},
177 {"zrank",zrankCommand,3,"r",0,NULL,1,1,1,0,0},
178 {"zrevrank",zrevrankCommand,3,"r",0,NULL,1,1,1,0,0},
179 {"hset",hsetCommand,4,"wm",0,NULL,1,1,1,0,0},
180 {"hsetnx",hsetnxCommand,4,"wm",0,NULL,1,1,1,0,0},
181 {"hget",hgetCommand,3,"r",0,NULL,1,1,1,0,0},
182 {"hmset",hmsetCommand,-4,"wm",0,NULL,1,1,1,0,0},
183 {"hmget",hmgetCommand,-3,"r",0,NULL,1,1,1,0,0},
184 {"hincrby",hincrbyCommand,4,"wm",0,NULL,1,1,1,0,0},
185 {"hincrbyfloat",hincrbyfloatCommand,4,"wm",0,NULL,1,1,1,0,0},
186 {"hdel",hdelCommand,-3,"w",0,NULL,1,1,1,0,0},
187 {"hlen",hlenCommand,2,"r",0,NULL,1,1,1,0,0},
188 {"hkeys",hkeysCommand,2,"rS",0,NULL,1,1,1,0,0},
189 {"hvals",hvalsCommand,2,"rS",0,NULL,1,1,1,0,0},
190 {"hgetall",hgetallCommand,2,"r",0,NULL,1,1,1,0,0},
191 {"hexists",hexistsCommand,3,"r",0,NULL,1,1,1,0,0},
192 {"incrby",incrbyCommand,3,"wm",0,NULL,1,1,1,0,0},
193 {"decrby",decrbyCommand,3,"wm",0,NULL,1,1,1,0,0},
194 {"incrbyfloat",incrbyfloatCommand,3,"wm",0,NULL,1,1,1,0,0},
195 {"getset",getsetCommand,3,"wm",0,NULL,1,1,1,0,0},
196 {"mset",msetCommand,-3,"wm",0,NULL,1,-1,2,0,0},
197 {"msetnx",msetnxCommand,-3,"wm",0,NULL,1,-1,2,0,0},
198 {"randomkey",randomkeyCommand,1,"rR",0,NULL,0,0,0,0,0},
199 {"select",selectCommand,2,"r",0,NULL,0,0,0,0,0},
200 {"move",moveCommand,3,"w",0,NULL,1,1,1,0,0},
201 {"rename",renameCommand,3,"w",0,renameGetKeys,1,2,1,0,0},
202 {"renamenx",renamenxCommand,3,"w",0,renameGetKeys,1,2,1,0,0},
203 {"expire",expireCommand,3,"w",0,NULL,1,1,1,0,0},
204 {"expireat",expireatCommand,3,"w",0,NULL,1,1,1,0,0},
205 {"pexpire",pexpireCommand,3,"w",0,NULL,1,1,1,0,0},
206 {"pexpireat",pexpireatCommand,3,"w",0,NULL,1,1,1,0,0},
207 {"keys",keysCommand,2,"rS",0,NULL,0,0,0,0,0},
208 {"dbsize",dbsizeCommand,1,"r",0,NULL,0,0,0,0,0},
209 {"auth",authCommand,2,"rs",0,NULL,0,0,0,0,0},
210 {"ping",pingCommand,1,"r",0,NULL,0,0,0,0,0},
211 {"echo",echoCommand,2,"r",0,NULL,0,0,0,0,0},
212 {"save",saveCommand,1,"ars",0,NULL,0,0,0,0,0},
213 {"bgsave",bgsaveCommand,1,"ar",0,NULL,0,0,0,0,0},
214 {"bgrewriteaof",bgrewriteaofCommand,1,"ar",0,NULL,0,0,0,0,0},
215 {"shutdown",shutdownCommand,-1,"ar",0,NULL,0,0,0,0,0},
216 {"lastsave",lastsaveCommand,1,"r",0,NULL,0,0,0,0,0},
217 {"type",typeCommand,2,"r",0,NULL,1,1,1,0,0},
218 {"multi",multiCommand,1,"rs",0,NULL,0,0,0,0,0},
219 {"exec",execCommand,1,"s",0,NULL,0,0,0,0,0},
220 {"discard",discardCommand,1,"rs",0,NULL,0,0,0,0,0},
221 {"sync",syncCommand,1,"ars",0,NULL,0,0,0,0,0},
222 {"replconf",replconfCommand,-1,"ars",0,NULL,0,0,0,0,0},
223 {"flushdb",flushdbCommand,1,"w",0,NULL,0,0,0,0,0},
224 {"flushall",flushallCommand,1,"w",0,NULL,0,0,0,0,0},
225 {"sort",sortCommand,-2,"wm",0,NULL,1,1,1,0,0},
226 {"info",infoCommand,-1,"rlt",0,NULL,0,0,0,0,0},
227 {"monitor",monitorCommand,1,"ars",0,NULL,0,0,0,0,0},
228 {"ttl",ttlCommand,2,"r",0,NULL,1,1,1,0,0},
229 {"pttl",pttlCommand,2,"r",0,NULL,1,1,1,0,0},
230 {"persist",persistCommand,2,"w",0,NULL,1,1,1,0,0},
231 {"slaveof",slaveofCommand,3,"ast",0,NULL,0,0,0,0,0},
232 {"debug",debugCommand,-2,"as",0,NULL,0,0,0,0,0},
233 {"config",configCommand,-2,"ar",0,NULL,0,0,0,0,0},
234 {"subscribe",subscribeCommand,-2,"rpslt",0,NULL,0,0,0,0,0},
235 {"unsubscribe",unsubscribeCommand,-1,"rpslt",0,NULL,0,0,0,0,0},
236 {"psubscribe",psubscribeCommand,-2,"rpslt",0,NULL,0,0,0,0,0},
237 {"punsubscribe",punsubscribeCommand,-1,"rpslt",0,NULL,0,0,0,0,0},
238 {"publish",publishCommand,3,"pflt",0,NULL,0,0,0,0,0},
239 {"watch",watchCommand,-2,"rs",0,noPreloadGetKeys,1,-1,1,0,0},
240 {"unwatch",unwatchCommand,1,"rs",0,NULL,0,0,0,0,0},
241 {"cluster",clusterCommand,-2,"ar",0,NULL,0,0,0,0,0},
242 {"restore",restoreCommand,4,"awm",0,NULL,1,1,1,0,0},
243 {"migrate",migrateCommand,6,"aw",0,NULL,0,0,0,0,0},
244 {"asking",askingCommand,1,"r",0,NULL,0,0,0,0,0},
245 {"dump",dumpCommand,2,"ar",0,NULL,1,1,1,0,0},
246 {"object",objectCommand,-2,"r",0,NULL,2,2,2,0,0},
247 {"client",clientCommand,-2,"ar",0,NULL,0,0,0,0,0},
248 {"eval",evalCommand,-3,"s",0,zunionInterGetKeys,0,0,0,0,0},
249 {"evalsha",evalShaCommand,-3,"s",0,zunionInterGetKeys,0,0,0,0,0},
250 {"slowlog",slowlogCommand,-2,"r",0,NULL,0,0,0,0,0},
251 {"script",scriptCommand,-2,"ras",0,NULL,0,0,0,0,0},
252 {"time",timeCommand,1,"rR",0,NULL,0,0,0,0,0},
253 {"bitop",bitopCommand,-4,"wm",0,NULL,2,-1,1,0,0},
254 {"bitcount",bitcountCommand,-2,"r",0,NULL,1,1,1,0,0}
255 };
256
257 /*============================ Utility functions ============================ */
258
259 /* Low level logging. To use only for very big messages, otherwise
260 * redisLog() is to prefer. */
261 void redisLogRaw(int level, const char *msg) {
262 const int syslogLevelMap[] = { LOG_DEBUG, LOG_INFO, LOG_NOTICE, LOG_WARNING };
263 const char *c = ".-*#";
264 FILE *fp;
265 char buf[64];
266 int rawmode = (level & REDIS_LOG_RAW);
267
268 level &= 0xff; /* clear flags */
269 if (level < server.verbosity) return;
270
271 fp = (server.logfile == NULL) ? stdout : fopen(server.logfile,"a");
272 if (!fp) return;
273
274 if (rawmode) {
275 fprintf(fp,"%s",msg);
276 } else {
277 int off;
278 struct timeval tv;
279
280 gettimeofday(&tv,NULL);
281 off = strftime(buf,sizeof(buf),"%d %b %H:%M:%S.",localtime(&tv.tv_sec));
282 snprintf(buf+off,sizeof(buf)-off,"%03d",(int)tv.tv_usec/1000);
283 fprintf(fp,"[%d] %s %c %s\n",(int)getpid(),buf,c[level],msg);
284 }
285 fflush(fp);
286
287 if (server.logfile) fclose(fp);
288
289 if (server.syslog_enabled) syslog(syslogLevelMap[level], "%s", msg);
290 }
291
292 /* Like redisLogRaw() but with printf-alike support. This is the funciton that
293 * is used across the code. The raw version is only used in order to dump
294 * the INFO output on crash. */
295 void redisLog(int level, const char *fmt, ...) {
296 va_list ap;
297 char msg[REDIS_MAX_LOGMSG_LEN];
298
299 if ((level&0xff) < server.verbosity) return;
300
301 va_start(ap, fmt);
302 vsnprintf(msg, sizeof(msg), fmt, ap);
303 va_end(ap);
304
305 redisLogRaw(level,msg);
306 }
307
308 /* Log a fixed message without printf-alike capabilities, in a way that is
309 * safe to call from a signal handler.
310 *
311 * We actually use this only for signals that are not fatal from the point
312 * of view of Redis. Signals that are going to kill the server anyway and
313 * where we need printf-alike features are served by redisLog(). */
314 void redisLogFromHandler(int level, const char *msg) {
315 int fd;
316 char buf[64];
317
318 if ((level&0xff) < server.verbosity ||
319 (server.logfile == NULL && server.daemonize)) return;
320 fd = server.logfile ?
321 open(server.logfile, O_APPEND|O_CREAT|O_WRONLY, 0644) :
322 STDOUT_FILENO;
323 if (fd == -1) return;
324 ll2string(buf,sizeof(buf),getpid());
325 if (write(fd,"[",1) == -1) goto err;
326 if (write(fd,buf,strlen(buf)) == -1) goto err;
327 if (write(fd," | signal handler] (",20) == -1) goto err;
328 ll2string(buf,sizeof(buf),time(NULL));
329 if (write(fd,buf,strlen(buf)) == -1) goto err;
330 if (write(fd,") ",2) == -1) goto err;
331 if (write(fd,msg,strlen(msg)) == -1) goto err;
332 if (write(fd,"\n",1) == -1) goto err;
333 err:
334 if (server.logfile) close(fd);
335 }
336
337 /* Return the UNIX time in microseconds */
338 long long ustime(void) {
339 struct timeval tv;
340 long long ust;
341
342 gettimeofday(&tv, NULL);
343 ust = ((long long)tv.tv_sec)*1000000;
344 ust += tv.tv_usec;
345 return ust;
346 }
347
348 /* Return the UNIX time in milliseconds */
349 long long mstime(void) {
350 return ustime()/1000;
351 }
352
353 /* After an RDB dump or AOF rewrite we exit from children using _exit() instead of
354 * exit(), because the latter may interact with the same file objects used by
355 * the parent process. However if we are testing the coverage normal exit() is
356 * used in order to obtain the right coverage information. */
357 void exitFromChild(int retcode) {
358 #ifdef COVERAGE_TEST
359 exit(retcode);
360 #else
361 _exit(retcode);
362 #endif
363 }
364
365 /*====================== Hash table type implementation ==================== */
366
367 /* This is an hash table type that uses the SDS dynamic strings libary as
368 * keys and radis objects as values (objects can hold SDS strings,
369 * lists, sets). */
370
371 void dictVanillaFree(void *privdata, void *val)
372 {
373 DICT_NOTUSED(privdata);
374 zfree(val);
375 }
376
377 void dictListDestructor(void *privdata, void *val)
378 {
379 DICT_NOTUSED(privdata);
380 listRelease((list*)val);
381 }
382
383 int dictSdsKeyCompare(void *privdata, const void *key1,
384 const void *key2)
385 {
386 int l1,l2;
387 DICT_NOTUSED(privdata);
388
389 l1 = sdslen((sds)key1);
390 l2 = sdslen((sds)key2);
391 if (l1 != l2) return 0;
392 return memcmp(key1, key2, l1) == 0;
393 }
394
395 /* A case insensitive version used for the command lookup table. */
396 int dictSdsKeyCaseCompare(void *privdata, const void *key1,
397 const void *key2)
398 {
399 DICT_NOTUSED(privdata);
400
401 return strcasecmp(key1, key2) == 0;
402 }
403
404 void dictRedisObjectDestructor(void *privdata, void *val)
405 {
406 DICT_NOTUSED(privdata);
407
408 if (val == NULL) return; /* Values of swapped out keys as set to NULL */
409 decrRefCount(val);
410 }
411
412 void dictSdsDestructor(void *privdata, void *val)
413 {
414 DICT_NOTUSED(privdata);
415
416 sdsfree(val);
417 }
418
419 int dictObjKeyCompare(void *privdata, const void *key1,
420 const void *key2)
421 {
422 const robj *o1 = key1, *o2 = key2;
423 return dictSdsKeyCompare(privdata,o1->ptr,o2->ptr);
424 }
425
426 unsigned int dictObjHash(const void *key) {
427 const robj *o = key;
428 return dictGenHashFunction(o->ptr, sdslen((sds)o->ptr));
429 }
430
431 unsigned int dictSdsHash(const void *key) {
432 return dictGenHashFunction((unsigned char*)key, sdslen((char*)key));
433 }
434
435 unsigned int dictSdsCaseHash(const void *key) {
436 return dictGenCaseHashFunction((unsigned char*)key, sdslen((char*)key));
437 }
438
439 int dictEncObjKeyCompare(void *privdata, const void *key1,
440 const void *key2)
441 {
442 robj *o1 = (robj*) key1, *o2 = (robj*) key2;
443 int cmp;
444
445 if (o1->encoding == REDIS_ENCODING_INT &&
446 o2->encoding == REDIS_ENCODING_INT)
447 return o1->ptr == o2->ptr;
448
449 o1 = getDecodedObject(o1);
450 o2 = getDecodedObject(o2);
451 cmp = dictSdsKeyCompare(privdata,o1->ptr,o2->ptr);
452 decrRefCount(o1);
453 decrRefCount(o2);
454 return cmp;
455 }
456
457 unsigned int dictEncObjHash(const void *key) {
458 robj *o = (robj*) key;
459
460 if (o->encoding == REDIS_ENCODING_RAW) {
461 return dictGenHashFunction(o->ptr, sdslen((sds)o->ptr));
462 } else {
463 if (o->encoding == REDIS_ENCODING_INT) {
464 char buf[32];
465 int len;
466
467 len = ll2string(buf,32,(long)o->ptr);
468 return dictGenHashFunction((unsigned char*)buf, len);
469 } else {
470 unsigned int hash;
471
472 o = getDecodedObject(o);
473 hash = dictGenHashFunction(o->ptr, sdslen((sds)o->ptr));
474 decrRefCount(o);
475 return hash;
476 }
477 }
478 }
479
480 /* Sets type hash table */
481 dictType setDictType = {
482 dictEncObjHash, /* hash function */
483 NULL, /* key dup */
484 NULL, /* val dup */
485 dictEncObjKeyCompare, /* key compare */
486 dictRedisObjectDestructor, /* key destructor */
487 NULL /* val destructor */
488 };
489
490 /* Sorted sets hash (note: a skiplist is used in addition to the hash table) */
491 dictType zsetDictType = {
492 dictEncObjHash, /* hash function */
493 NULL, /* key dup */
494 NULL, /* val dup */
495 dictEncObjKeyCompare, /* key compare */
496 dictRedisObjectDestructor, /* key destructor */
497 NULL /* val destructor */
498 };
499
500 /* Db->dict, keys are sds strings, vals are Redis objects. */
501 dictType dbDictType = {
502 dictSdsHash, /* hash function */
503 NULL, /* key dup */
504 NULL, /* val dup */
505 dictSdsKeyCompare, /* key compare */
506 dictSdsDestructor, /* key destructor */
507 dictRedisObjectDestructor /* val destructor */
508 };
509
510 /* Db->expires */
511 dictType keyptrDictType = {
512 dictSdsHash, /* hash function */
513 NULL, /* key dup */
514 NULL, /* val dup */
515 dictSdsKeyCompare, /* key compare */
516 NULL, /* key destructor */
517 NULL /* val destructor */
518 };
519
520 /* Command table. sds string -> command struct pointer. */
521 dictType commandTableDictType = {
522 dictSdsCaseHash, /* hash function */
523 NULL, /* key dup */
524 NULL, /* val dup */
525 dictSdsKeyCaseCompare, /* key compare */
526 dictSdsDestructor, /* key destructor */
527 NULL /* val destructor */
528 };
529
530 /* Hash type hash table (note that small hashes are represented with zimpaps) */
531 dictType hashDictType = {
532 dictEncObjHash, /* hash function */
533 NULL, /* key dup */
534 NULL, /* val dup */
535 dictEncObjKeyCompare, /* key compare */
536 dictRedisObjectDestructor, /* key destructor */
537 dictRedisObjectDestructor /* val destructor */
538 };
539
540 /* Keylist hash table type has unencoded redis objects as keys and
541 * lists as values. It's used for blocking operations (BLPOP) and to
542 * map swapped keys to a list of clients waiting for this keys to be loaded. */
543 dictType keylistDictType = {
544 dictObjHash, /* hash function */
545 NULL, /* key dup */
546 NULL, /* val dup */
547 dictObjKeyCompare, /* key compare */
548 dictRedisObjectDestructor, /* key destructor */
549 dictListDestructor /* val destructor */
550 };
551
552 /* Cluster nodes hash table, mapping nodes addresses 1.2.3.4:6379 to
553 * clusterNode structures. */
554 dictType clusterNodesDictType = {
555 dictSdsHash, /* hash function */
556 NULL, /* key dup */
557 NULL, /* val dup */
558 dictSdsKeyCompare, /* key compare */
559 dictSdsDestructor, /* key destructor */
560 NULL /* val destructor */
561 };
562
563 int htNeedsResize(dict *dict) {
564 long long size, used;
565
566 size = dictSlots(dict);
567 used = dictSize(dict);
568 return (size && used && size > DICT_HT_INITIAL_SIZE &&
569 (used*100/size < REDIS_HT_MINFILL));
570 }
571
572 /* If the percentage of used slots in the HT reaches REDIS_HT_MINFILL
573 * we resize the hash table to save memory */
574 void tryResizeHashTables(void) {
575 int j;
576
577 for (j = 0; j < server.dbnum; j++) {
578 if (htNeedsResize(server.db[j].dict))
579 dictResize(server.db[j].dict);
580 if (htNeedsResize(server.db[j].expires))
581 dictResize(server.db[j].expires);
582 }
583 }
584
585 /* Our hash table implementation performs rehashing incrementally while
586 * we write/read from the hash table. Still if the server is idle, the hash
587 * table will use two tables for a long time. So we try to use 1 millisecond
588 * of CPU time at every serverCron() loop in order to rehash some key. */
589 void incrementallyRehash(void) {
590 int j;
591
592 for (j = 0; j < server.dbnum; j++) {
593 /* Keys dictionary */
594 if (dictIsRehashing(server.db[j].dict)) {
595 dictRehashMilliseconds(server.db[j].dict,1);
596 break; /* already used our millisecond for this loop... */
597 }
598 /* Expires */
599 if (dictIsRehashing(server.db[j].expires)) {
600 dictRehashMilliseconds(server.db[j].expires,1);
601 break; /* already used our millisecond for this loop... */
602 }
603 }
604 }
605
606 /* This function is called once a background process of some kind terminates,
607 * as we want to avoid resizing the hash tables when there is a child in order
608 * to play well with copy-on-write (otherwise when a resize happens lots of
609 * memory pages are copied). The goal of this function is to update the ability
610 * for dict.c to resize the hash tables accordingly to the fact we have o not
611 * running childs. */
612 void updateDictResizePolicy(void) {
613 if (server.rdb_child_pid == -1 && server.aof_child_pid == -1)
614 dictEnableResize();
615 else
616 dictDisableResize();
617 }
618
619 /* ======================= Cron: called every 100 ms ======================== */
620
621 /* Try to expire a few timed out keys. The algorithm used is adaptive and
622 * will use few CPU cycles if there are few expiring keys, otherwise
623 * it will get more aggressive to avoid that too much memory is used by
624 * keys that can be removed from the keyspace. */
625 void activeExpireCycle(void) {
626 int j, iteration = 0;
627 long long start = ustime(), timelimit;
628
629 /* We can use at max REDIS_EXPIRELOOKUPS_TIME_PERC percentage of CPU time
630 * per iteration. Since this function gets called with a frequency of
631 * REDIS_HZ times per second, the following is the max amount of
632 * microseconds we can spend in this function. */
633 timelimit = 1000000*REDIS_EXPIRELOOKUPS_TIME_PERC/REDIS_HZ/100;
634 if (timelimit <= 0) timelimit = 1;
635
636 for (j = 0; j < server.dbnum; j++) {
637 int expired;
638 redisDb *db = server.db+j;
639
640 /* Continue to expire if at the end of the cycle more than 25%
641 * of the keys were expired. */
642 do {
643 unsigned long num = dictSize(db->expires);
644 unsigned long slots = dictSlots(db->expires);
645 long long now = mstime();
646
647 /* When there are less than 1% filled slots getting random
648 * keys is expensive, so stop here waiting for better times...
649 * The dictionary will be resized asap. */
650 if (num && slots > DICT_HT_INITIAL_SIZE &&
651 (num*100/slots < 1)) break;
652
653 /* The main collection cycle. Sample random keys among keys
654 * with an expire set, checking for expired ones. */
655 expired = 0;
656 if (num > REDIS_EXPIRELOOKUPS_PER_CRON)
657 num = REDIS_EXPIRELOOKUPS_PER_CRON;
658 while (num--) {
659 dictEntry *de;
660 long long t;
661
662 if ((de = dictGetRandomKey(db->expires)) == NULL) break;
663 t = dictGetSignedIntegerVal(de);
664 if (now > t) {
665 sds key = dictGetKey(de);
666 robj *keyobj = createStringObject(key,sdslen(key));
667
668 propagateExpire(db,keyobj);
669 dbDelete(db,keyobj);
670 decrRefCount(keyobj);
671 expired++;
672 server.stat_expiredkeys++;
673 }
674 }
675 /* We can't block forever here even if there are many keys to
676 * expire. So after a given amount of milliseconds return to the
677 * caller waiting for the other active expire cycle. */
678 iteration++;
679 if ((iteration & 0xf) == 0 && /* check once every 16 cycles. */
680 (ustime()-start) > timelimit) return;
681 } while (expired > REDIS_EXPIRELOOKUPS_PER_CRON/4);
682 }
683 }
684
685 void updateLRUClock(void) {
686 server.lruclock = (server.unixtime/REDIS_LRU_CLOCK_RESOLUTION) &
687 REDIS_LRU_CLOCK_MAX;
688 }
689
690
691 /* Add a sample to the operations per second array of samples. */
692 void trackOperationsPerSecond(void) {
693 long long t = mstime() - server.ops_sec_last_sample_time;
694 long long ops = server.stat_numcommands - server.ops_sec_last_sample_ops;
695 long long ops_sec;
696
697 ops_sec = t > 0 ? (ops*1000/t) : 0;
698
699 server.ops_sec_samples[server.ops_sec_idx] = ops_sec;
700 server.ops_sec_idx = (server.ops_sec_idx+1) % REDIS_OPS_SEC_SAMPLES;
701 server.ops_sec_last_sample_time = mstime();
702 server.ops_sec_last_sample_ops = server.stat_numcommands;
703 }
704
705 /* Return the mean of all the samples. */
706 long long getOperationsPerSecond(void) {
707 int j;
708 long long sum = 0;
709
710 for (j = 0; j < REDIS_OPS_SEC_SAMPLES; j++)
711 sum += server.ops_sec_samples[j];
712 return sum / REDIS_OPS_SEC_SAMPLES;
713 }
714
715 /* Check for timeouts. Returns non-zero if the client was terminated */
716 int clientsCronHandleTimeout(redisClient *c) {
717 time_t now = server.unixtime;
718
719 if (server.maxidletime &&
720 !(c->flags & REDIS_SLAVE) && /* no timeout for slaves */
721 !(c->flags & REDIS_MASTER) && /* no timeout for masters */
722 !(c->flags & REDIS_BLOCKED) && /* no timeout for BLPOP */
723 dictSize(c->pubsub_channels) == 0 && /* no timeout for pubsub */
724 listLength(c->pubsub_patterns) == 0 &&
725 (now - c->lastinteraction > server.maxidletime))
726 {
727 redisLog(REDIS_VERBOSE,"Closing idle client");
728 freeClient(c);
729 return 1;
730 } else if (c->flags & REDIS_BLOCKED) {
731 if (c->bpop.timeout != 0 && c->bpop.timeout < now) {
732 addReply(c,shared.nullmultibulk);
733 unblockClientWaitingData(c);
734 }
735 }
736 return 0;
737 }
738
739 /* The client query buffer is an sds.c string that can end with a lot of
740 * free space not used, this function reclaims space if needed.
741 *
742 * The funciton always returns 0 as it never terminates the client. */
743 int clientsCronResizeQueryBuffer(redisClient *c) {
744 size_t querybuf_size = sdsAllocSize(c->querybuf);
745 time_t idletime = server.unixtime - c->lastinteraction;
746
747 /* There are two conditions to resize the query buffer:
748 * 1) Query buffer is > BIG_ARG and too big for latest peak.
749 * 2) Client is inactive and the buffer is bigger than 1k. */
750 if (((querybuf_size > REDIS_MBULK_BIG_ARG) &&
751 (querybuf_size/(c->querybuf_peak+1)) > 2) ||
752 (querybuf_size > 1024 && idletime > 2))
753 {
754 /* Only resize the query buffer if it is actually wasting space. */
755 if (sdsavail(c->querybuf) > 1024) {
756 c->querybuf = sdsRemoveFreeSpace(c->querybuf);
757 }
758 }
759 /* Reset the peak again to capture the peak memory usage in the next
760 * cycle. */
761 c->querybuf_peak = 0;
762 return 0;
763 }
764
765 void clientsCron(void) {
766 /* Make sure to process at least 1/(REDIS_HZ*10) of clients per call.
767 * Since this function is called REDIS_HZ times per second we are sure that
768 * in the worst case we process all the clients in 10 seconds.
769 * In normal conditions (a reasonable number of clients) we process
770 * all the clients in a shorter time. */
771 int numclients = listLength(server.clients);
772 int iterations = numclients/(REDIS_HZ*10);
773
774 if (iterations < 50)
775 iterations = (numclients < 50) ? numclients : 50;
776 while(listLength(server.clients) && iterations--) {
777 redisClient *c;
778 listNode *head;
779
780 /* Rotate the list, take the current head, process.
781 * This way if the client must be removed from the list it's the
782 * first element and we don't incur into O(N) computation. */
783 listRotate(server.clients);
784 head = listFirst(server.clients);
785 c = listNodeValue(head);
786 /* The following functions do different service checks on the client.
787 * The protocol is that they return non-zero if the client was
788 * terminated. */
789 if (clientsCronHandleTimeout(c)) continue;
790 if (clientsCronResizeQueryBuffer(c)) continue;
791 }
792 }
793
794 /* This is our timer interrupt, called REDIS_HZ times per second.
795 * Here is where we do a number of things that need to be done asynchronously.
796 * For instance:
797 *
798 * - Active expired keys collection (it is also performed in a lazy way on
799 * lookup).
800 * - Software watchdong.
801 * - Update some statistic.
802 * - Incremental rehashing of the DBs hash tables.
803 * - Triggering BGSAVE / AOF rewrite, and handling of terminated children.
804 * - Clients timeout of differnet kinds.
805 * - Replication reconnection.
806 * - Many more...
807 *
808 * Everything directly called here will be called REDIS_HZ times per second,
809 * so in order to throttle execution of things we want to do less frequently
810 * a macro is used: run_with_period(milliseconds) { .... }
811 */
812
813 int serverCron(struct aeEventLoop *eventLoop, long long id, void *clientData) {
814 int j;
815 REDIS_NOTUSED(eventLoop);
816 REDIS_NOTUSED(id);
817 REDIS_NOTUSED(clientData);
818
819 /* Software watchdog: deliver the SIGALRM that will reach the signal
820 * handler if we don't return here fast enough. */
821 if (server.watchdog_period) watchdogScheduleSignal(server.watchdog_period);
822
823 /* We take a cached value of the unix time in the global state because
824 * with virtual memory and aging there is to store the current time
825 * in objects at every object access, and accuracy is not needed.
826 * To access a global var is faster than calling time(NULL) */
827 server.unixtime = time(NULL);
828
829 run_with_period(100) trackOperationsPerSecond();
830
831 /* We have just 22 bits per object for LRU information.
832 * So we use an (eventually wrapping) LRU clock with 10 seconds resolution.
833 * 2^22 bits with 10 seconds resoluton is more or less 1.5 years.
834 *
835 * Note that even if this will wrap after 1.5 years it's not a problem,
836 * everything will still work but just some object will appear younger
837 * to Redis. But for this to happen a given object should never be touched
838 * for 1.5 years.
839 *
840 * Note that you can change the resolution altering the
841 * REDIS_LRU_CLOCK_RESOLUTION define.
842 */
843 updateLRUClock();
844
845 /* Record the max memory used since the server was started. */
846 if (zmalloc_used_memory() > server.stat_peak_memory)
847 server.stat_peak_memory = zmalloc_used_memory();
848
849 /* We received a SIGTERM, shutting down here in a safe way, as it is
850 * not ok doing so inside the signal handler. */
851 if (server.shutdown_asap) {
852 if (prepareForShutdown(0) == REDIS_OK) exit(0);
853 redisLog(REDIS_WARNING,"SIGTERM received but errors trying to shut down the server, check the logs for more information");
854 }
855
856 /* Show some info about non-empty databases */
857 run_with_period(5000) {
858 for (j = 0; j < server.dbnum; j++) {
859 long long size, used, vkeys;
860
861 size = dictSlots(server.db[j].dict);
862 used = dictSize(server.db[j].dict);
863 vkeys = dictSize(server.db[j].expires);
864 if (used || vkeys) {
865 redisLog(REDIS_VERBOSE,"DB %d: %lld keys (%lld volatile) in %lld slots HT.",j,used,vkeys,size);
866 /* dictPrintStats(server.dict); */
867 }
868 }
869 }
870
871 /* We don't want to resize the hash tables while a bacground saving
872 * is in progress: the saving child is created using fork() that is
873 * implemented with a copy-on-write semantic in most modern systems, so
874 * if we resize the HT while there is the saving child at work actually
875 * a lot of memory movements in the parent will cause a lot of pages
876 * copied. */
877 if (server.rdb_child_pid == -1 && server.aof_child_pid == -1) {
878 tryResizeHashTables();
879 if (server.activerehashing) incrementallyRehash();
880 }
881
882 /* Show information about connected clients */
883 if (!server.sentinel_mode) {
884 run_with_period(5000) {
885 redisLog(REDIS_VERBOSE,
886 "%d clients connected (%d slaves), %zu bytes in use",
887 listLength(server.clients)-listLength(server.slaves),
888 listLength(server.slaves),
889 zmalloc_used_memory());
890 }
891 }
892
893 /* We need to do a few operations on clients asynchronously. */
894 clientsCron();
895
896 /* Start a scheduled AOF rewrite if this was requested by the user while
897 * a BGSAVE was in progress. */
898 if (server.rdb_child_pid == -1 && server.aof_child_pid == -1 &&
899 server.aof_rewrite_scheduled)
900 {
901 rewriteAppendOnlyFileBackground();
902 }
903
904 /* Check if a background saving or AOF rewrite in progress terminated. */
905 if (server.rdb_child_pid != -1 || server.aof_child_pid != -1) {
906 int statloc;
907 pid_t pid;
908
909 if ((pid = wait3(&statloc,WNOHANG,NULL)) != 0) {
910 int exitcode = WEXITSTATUS(statloc);
911 int bysignal = 0;
912
913 if (WIFSIGNALED(statloc)) bysignal = WTERMSIG(statloc);
914
915 if (pid == server.rdb_child_pid) {
916 backgroundSaveDoneHandler(exitcode,bysignal);
917 } else {
918 backgroundRewriteDoneHandler(exitcode,bysignal);
919 }
920 updateDictResizePolicy();
921 }
922 } else {
923 /* If there is not a background saving/rewrite in progress check if
924 * we have to save/rewrite now */
925 for (j = 0; j < server.saveparamslen; j++) {
926 struct saveparam *sp = server.saveparams+j;
927
928 if (server.dirty >= sp->changes &&
929 server.unixtime-server.lastsave > sp->seconds) {
930 redisLog(REDIS_NOTICE,"%d changes in %d seconds. Saving...",
931 sp->changes, sp->seconds);
932 rdbSaveBackground(server.rdb_filename);
933 break;
934 }
935 }
936
937 /* Trigger an AOF rewrite if needed */
938 if (server.rdb_child_pid == -1 &&
939 server.aof_child_pid == -1 &&
940 server.aof_rewrite_perc &&
941 server.aof_current_size > server.aof_rewrite_min_size)
942 {
943 long long base = server.aof_rewrite_base_size ?
944 server.aof_rewrite_base_size : 1;
945 long long growth = (server.aof_current_size*100/base) - 100;
946 if (growth >= server.aof_rewrite_perc) {
947 redisLog(REDIS_NOTICE,"Starting automatic rewriting of AOF on %lld%% growth",growth);
948 rewriteAppendOnlyFileBackground();
949 }
950 }
951 }
952
953
954 /* If we postponed an AOF buffer flush, let's try to do it every time the
955 * cron function is called. */
956 if (server.aof_flush_postponed_start) flushAppendOnlyFile(0);
957
958 /* Expire a few keys per cycle, only if this is a master.
959 * On slaves we wait for DEL operations synthesized by the master
960 * in order to guarantee a strict consistency. */
961 if (server.masterhost == NULL) activeExpireCycle();
962
963 /* Close clients that need to be closed asynchronous */
964 freeClientsInAsyncFreeQueue();
965
966 /* Replication cron function -- used to reconnect to master and
967 * to detect transfer failures. */
968 run_with_period(1000) replicationCron();
969
970 /* Run other sub-systems specific cron jobs */
971 run_with_period(1000) {
972 if (server.cluster_enabled) clusterCron();
973 }
974
975 /* Run the sentinel timer if we are in sentinel mode. */
976 run_with_period(100) {
977 if (server.sentinel_mode) sentinelTimer();
978 }
979
980 server.cronloops++;
981 return 1000/REDIS_HZ;
982 }
983
984 /* This function gets called every time Redis is entering the
985 * main loop of the event driven library, that is, before to sleep
986 * for ready file descriptors. */
987 void beforeSleep(struct aeEventLoop *eventLoop) {
988 REDIS_NOTUSED(eventLoop);
989 listNode *ln;
990 redisClient *c;
991
992 /* Try to process pending commands for clients that were just unblocked. */
993 while (listLength(server.unblocked_clients)) {
994 ln = listFirst(server.unblocked_clients);
995 redisAssert(ln != NULL);
996 c = ln->value;
997 listDelNode(server.unblocked_clients,ln);
998 c->flags &= ~REDIS_UNBLOCKED;
999
1000 /* Process remaining data in the input buffer. */
1001 if (c->querybuf && sdslen(c->querybuf) > 0) {
1002 server.current_client = c;
1003 processInputBuffer(c);
1004 server.current_client = NULL;
1005 }
1006 }
1007
1008 /* Write the AOF buffer on disk */
1009 flushAppendOnlyFile(0);
1010 }
1011
1012 /* =========================== Server initialization ======================== */
1013
1014 void createSharedObjects(void) {
1015 int j;
1016
1017 shared.crlf = createObject(REDIS_STRING,sdsnew("\r\n"));
1018 shared.ok = createObject(REDIS_STRING,sdsnew("+OK\r\n"));
1019 shared.err = createObject(REDIS_STRING,sdsnew("-ERR\r\n"));
1020 shared.emptybulk = createObject(REDIS_STRING,sdsnew("$0\r\n\r\n"));
1021 shared.czero = createObject(REDIS_STRING,sdsnew(":0\r\n"));
1022 shared.cone = createObject(REDIS_STRING,sdsnew(":1\r\n"));
1023 shared.cnegone = createObject(REDIS_STRING,sdsnew(":-1\r\n"));
1024 shared.nullbulk = createObject(REDIS_STRING,sdsnew("$-1\r\n"));
1025 shared.nullmultibulk = createObject(REDIS_STRING,sdsnew("*-1\r\n"));
1026 shared.emptymultibulk = createObject(REDIS_STRING,sdsnew("*0\r\n"));
1027 shared.pong = createObject(REDIS_STRING,sdsnew("+PONG\r\n"));
1028 shared.queued = createObject(REDIS_STRING,sdsnew("+QUEUED\r\n"));
1029 shared.wrongtypeerr = createObject(REDIS_STRING,sdsnew(
1030 "-ERR Operation against a key holding the wrong kind of value\r\n"));
1031 shared.nokeyerr = createObject(REDIS_STRING,sdsnew(
1032 "-ERR no such key\r\n"));
1033 shared.syntaxerr = createObject(REDIS_STRING,sdsnew(
1034 "-ERR syntax error\r\n"));
1035 shared.sameobjecterr = createObject(REDIS_STRING,sdsnew(
1036 "-ERR source and destination objects are the same\r\n"));
1037 shared.outofrangeerr = createObject(REDIS_STRING,sdsnew(
1038 "-ERR index out of range\r\n"));
1039 shared.noscripterr = createObject(REDIS_STRING,sdsnew(
1040 "-NOSCRIPT No matching script. Please use EVAL.\r\n"));
1041 shared.loadingerr = createObject(REDIS_STRING,sdsnew(
1042 "-LOADING Redis is loading the dataset in memory\r\n"));
1043 shared.slowscripterr = createObject(REDIS_STRING,sdsnew(
1044 "-BUSY Redis is busy running a script. You can only call SCRIPT KILL or SHUTDOWN NOSAVE.\r\n"));
1045 shared.masterdownerr = createObject(REDIS_STRING,sdsnew(
1046 "-MASTERDOWN Link with MASTER is down and slave-serve-stale-data is set to 'no'.\r\n"));
1047 shared.bgsaveerr = createObject(REDIS_STRING,sdsnew(
1048 "-MISCONF Redis is configured to save RDB snapshots, but is currently not able to persist on disk. Commands that may modify the data set are disabled. Please check Redis logs for details about the error.\r\n"));
1049 shared.roslaveerr = createObject(REDIS_STRING,sdsnew(
1050 "-READONLY You can't write against a read only slave.\r\n"));
1051 shared.oomerr = createObject(REDIS_STRING,sdsnew(
1052 "-OOM command not allowed when used memory > 'maxmemory'.\r\n"));
1053 shared.space = createObject(REDIS_STRING,sdsnew(" "));
1054 shared.colon = createObject(REDIS_STRING,sdsnew(":"));
1055 shared.plus = createObject(REDIS_STRING,sdsnew("+"));
1056
1057 for (j = 0; j < REDIS_SHARED_SELECT_CMDS; j++) {
1058 shared.select[j] = createObject(REDIS_STRING,
1059 sdscatprintf(sdsempty(),"select %d\r\n", j));
1060 }
1061 shared.messagebulk = createStringObject("$7\r\nmessage\r\n",13);
1062 shared.pmessagebulk = createStringObject("$8\r\npmessage\r\n",14);
1063 shared.subscribebulk = createStringObject("$9\r\nsubscribe\r\n",15);
1064 shared.unsubscribebulk = createStringObject("$11\r\nunsubscribe\r\n",18);
1065 shared.psubscribebulk = createStringObject("$10\r\npsubscribe\r\n",17);
1066 shared.punsubscribebulk = createStringObject("$12\r\npunsubscribe\r\n",19);
1067 shared.del = createStringObject("DEL",3);
1068 shared.rpop = createStringObject("RPOP",4);
1069 shared.lpop = createStringObject("LPOP",4);
1070 shared.lpush = createStringObject("LPUSH",5);
1071 for (j = 0; j < REDIS_SHARED_INTEGERS; j++) {
1072 shared.integers[j] = createObject(REDIS_STRING,(void*)(long)j);
1073 shared.integers[j]->encoding = REDIS_ENCODING_INT;
1074 }
1075 for (j = 0; j < REDIS_SHARED_BULKHDR_LEN; j++) {
1076 shared.mbulkhdr[j] = createObject(REDIS_STRING,
1077 sdscatprintf(sdsempty(),"*%d\r\n",j));
1078 shared.bulkhdr[j] = createObject(REDIS_STRING,
1079 sdscatprintf(sdsempty(),"$%d\r\n",j));
1080 }
1081 }
1082
1083 void initServerConfig() {
1084 getRandomHexChars(server.runid,REDIS_RUN_ID_SIZE);
1085 server.runid[REDIS_RUN_ID_SIZE] = '\0';
1086 server.arch_bits = (sizeof(long) == 8) ? 64 : 32;
1087 server.port = REDIS_SERVERPORT;
1088 server.bindaddr = NULL;
1089 server.unixsocket = NULL;
1090 server.unixsocketperm = 0;
1091 server.ipfd = -1;
1092 server.sofd = -1;
1093 server.dbnum = REDIS_DEFAULT_DBNUM;
1094 server.verbosity = REDIS_NOTICE;
1095 server.maxidletime = REDIS_MAXIDLETIME;
1096 server.client_max_querybuf_len = REDIS_MAX_QUERYBUF_LEN;
1097 server.saveparams = NULL;
1098 server.loading = 0;
1099 server.logfile = NULL; /* NULL = log on standard output */
1100 server.syslog_enabled = 0;
1101 server.syslog_ident = zstrdup("redis");
1102 server.syslog_facility = LOG_LOCAL0;
1103 server.daemonize = 0;
1104 server.aof_state = REDIS_AOF_OFF;
1105 server.aof_fsync = AOF_FSYNC_EVERYSEC;
1106 server.aof_no_fsync_on_rewrite = 0;
1107 server.aof_rewrite_perc = REDIS_AOF_REWRITE_PERC;
1108 server.aof_rewrite_min_size = REDIS_AOF_REWRITE_MIN_SIZE;
1109 server.aof_rewrite_base_size = 0;
1110 server.aof_rewrite_scheduled = 0;
1111 server.aof_last_fsync = time(NULL);
1112 server.aof_rewrite_time_last = -1;
1113 server.aof_rewrite_time_start = -1;
1114 server.aof_lastbgrewrite_status = REDIS_OK;
1115 server.aof_delayed_fsync = 0;
1116 server.aof_fd = -1;
1117 server.aof_selected_db = -1; /* Make sure the first time will not match */
1118 server.aof_flush_postponed_start = 0;
1119 server.pidfile = zstrdup("/var/run/redis.pid");
1120 server.rdb_filename = zstrdup("dump.rdb");
1121 server.aof_filename = zstrdup("appendonly.aof");
1122 server.requirepass = NULL;
1123 server.rdb_compression = 1;
1124 server.rdb_checksum = 1;
1125 server.activerehashing = 1;
1126 server.maxclients = REDIS_MAX_CLIENTS;
1127 server.bpop_blocked_clients = 0;
1128 server.maxmemory = 0;
1129 server.maxmemory_policy = REDIS_MAXMEMORY_VOLATILE_LRU;
1130 server.maxmemory_samples = 3;
1131 server.hash_max_ziplist_entries = REDIS_HASH_MAX_ZIPLIST_ENTRIES;
1132 server.hash_max_ziplist_value = REDIS_HASH_MAX_ZIPLIST_VALUE;
1133 server.list_max_ziplist_entries = REDIS_LIST_MAX_ZIPLIST_ENTRIES;
1134 server.list_max_ziplist_value = REDIS_LIST_MAX_ZIPLIST_VALUE;
1135 server.set_max_intset_entries = REDIS_SET_MAX_INTSET_ENTRIES;
1136 server.zset_max_ziplist_entries = REDIS_ZSET_MAX_ZIPLIST_ENTRIES;
1137 server.zset_max_ziplist_value = REDIS_ZSET_MAX_ZIPLIST_VALUE;
1138 server.shutdown_asap = 0;
1139 server.repl_ping_slave_period = REDIS_REPL_PING_SLAVE_PERIOD;
1140 server.repl_timeout = REDIS_REPL_TIMEOUT;
1141 server.cluster_enabled = 0;
1142 server.cluster.configfile = zstrdup("nodes.conf");
1143 server.lua_caller = NULL;
1144 server.lua_time_limit = REDIS_LUA_TIME_LIMIT;
1145 server.lua_client = NULL;
1146 server.lua_timedout = 0;
1147
1148 updateLRUClock();
1149 resetServerSaveParams();
1150
1151 appendServerSaveParams(60*60,1); /* save after 1 hour and 1 change */
1152 appendServerSaveParams(300,100); /* save after 5 minutes and 100 changes */
1153 appendServerSaveParams(60,10000); /* save after 1 minute and 10000 changes */
1154 /* Replication related */
1155 server.masterauth = NULL;
1156 server.masterhost = NULL;
1157 server.masterport = 6379;
1158 server.master = NULL;
1159 server.repl_state = REDIS_REPL_NONE;
1160 server.repl_syncio_timeout = REDIS_REPL_SYNCIO_TIMEOUT;
1161 server.repl_serve_stale_data = 1;
1162 server.repl_slave_ro = 1;
1163 server.repl_down_since = time(NULL);
1164 server.slave_priority = REDIS_DEFAULT_SLAVE_PRIORITY;
1165
1166 /* Client output buffer limits */
1167 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_NORMAL].hard_limit_bytes = 0;
1168 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_NORMAL].soft_limit_bytes = 0;
1169 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_NORMAL].soft_limit_seconds = 0;
1170 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_SLAVE].hard_limit_bytes = 1024*1024*256;
1171 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_SLAVE].soft_limit_bytes = 1024*1024*64;
1172 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_SLAVE].soft_limit_seconds = 60;
1173 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_PUBSUB].hard_limit_bytes = 1024*1024*32;
1174 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_PUBSUB].soft_limit_bytes = 1024*1024*8;
1175 server.client_obuf_limits[REDIS_CLIENT_LIMIT_CLASS_PUBSUB].soft_limit_seconds = 60;
1176
1177 /* Double constants initialization */
1178 R_Zero = 0.0;
1179 R_PosInf = 1.0/R_Zero;
1180 R_NegInf = -1.0/R_Zero;
1181 R_Nan = R_Zero/R_Zero;
1182
1183 /* Command table -- we intiialize it here as it is part of the
1184 * initial configuration, since command names may be changed via
1185 * redis.conf using the rename-command directive. */
1186 server.commands = dictCreate(&commandTableDictType,NULL);
1187 populateCommandTable();
1188 server.delCommand = lookupCommandByCString("del");
1189 server.multiCommand = lookupCommandByCString("multi");
1190 server.lpushCommand = lookupCommandByCString("lpush");
1191 server.lpopCommand = lookupCommandByCString("lpop");
1192 server.rpopCommand = lookupCommandByCString("rpop");
1193
1194 /* Slow log */
1195 server.slowlog_log_slower_than = REDIS_SLOWLOG_LOG_SLOWER_THAN;
1196 server.slowlog_max_len = REDIS_SLOWLOG_MAX_LEN;
1197
1198 /* Debugging */
1199 server.assert_failed = "<no assertion failed>";
1200 server.assert_file = "<no file>";
1201 server.assert_line = 0;
1202 server.bug_report_start = 0;
1203 server.watchdog_period = 0;
1204 }
1205
1206 /* This function will try to raise the max number of open files accordingly to
1207 * the configured max number of clients. It will also account for 32 additional
1208 * file descriptors as we need a few more for persistence, listening
1209 * sockets, log files and so forth.
1210 *
1211 * If it will not be possible to set the limit accordingly to the configured
1212 * max number of clients, the function will do the reverse setting
1213 * server.maxclients to the value that we can actually handle. */
1214 void adjustOpenFilesLimit(void) {
1215 rlim_t maxfiles = server.maxclients+32;
1216 struct rlimit limit;
1217
1218 if (getrlimit(RLIMIT_NOFILE,&limit) == -1) {
1219 redisLog(REDIS_WARNING,"Unable to obtain the current NOFILE limit (%s), assuming 1024 and setting the max clients configuration accordingly.",
1220 strerror(errno));
1221 server.maxclients = 1024-32;
1222 } else {
1223 rlim_t oldlimit = limit.rlim_cur;
1224
1225 /* Set the max number of files if the current limit is not enough
1226 * for our needs. */
1227 if (oldlimit < maxfiles) {
1228 rlim_t f;
1229
1230 f = maxfiles;
1231 while(f > oldlimit) {
1232 limit.rlim_cur = f;
1233 limit.rlim_max = f;
1234 if (setrlimit(RLIMIT_NOFILE,&limit) != -1) break;
1235 f -= 128;
1236 }
1237 if (f < oldlimit) f = oldlimit;
1238 if (f != maxfiles) {
1239 server.maxclients = f-32;
1240 redisLog(REDIS_WARNING,"Unable to set the max number of files limit to %d (%s), setting the max clients configuration to %d.",
1241 (int) maxfiles, strerror(errno), (int) server.maxclients);
1242 } else {
1243 redisLog(REDIS_NOTICE,"Max number of open files set to %d",
1244 (int) maxfiles);
1245 }
1246 }
1247 }
1248 }
1249
1250 void initServer() {
1251 int j;
1252
1253 signal(SIGHUP, SIG_IGN);
1254 signal(SIGPIPE, SIG_IGN);
1255 setupSignalHandlers();
1256
1257 if (server.syslog_enabled) {
1258 openlog(server.syslog_ident, LOG_PID | LOG_NDELAY | LOG_NOWAIT,
1259 server.syslog_facility);
1260 }
1261
1262 server.current_client = NULL;
1263 server.clients = listCreate();
1264 server.clients_to_close = listCreate();
1265 server.slaves = listCreate();
1266 server.monitors = listCreate();
1267 server.unblocked_clients = listCreate();
1268 server.ready_keys = listCreate();
1269
1270 createSharedObjects();
1271 adjustOpenFilesLimit();
1272 server.el = aeCreateEventLoop(server.maxclients+1024);
1273 server.db = zmalloc(sizeof(redisDb)*server.dbnum);
1274
1275 if (server.port != 0) {
1276 server.ipfd = anetTcpServer(server.neterr,server.port,server.bindaddr);
1277 if (server.ipfd == ANET_ERR) {
1278 redisLog(REDIS_WARNING, "Opening port %d: %s",
1279 server.port, server.neterr);
1280 exit(1);
1281 }
1282 }
1283 if (server.unixsocket != NULL) {
1284 unlink(server.unixsocket); /* don't care if this fails */
1285 server.sofd = anetUnixServer(server.neterr,server.unixsocket,server.unixsocketperm);
1286 if (server.sofd == ANET_ERR) {
1287 redisLog(REDIS_WARNING, "Opening socket: %s", server.neterr);
1288 exit(1);
1289 }
1290 }
1291 if (server.ipfd < 0 && server.sofd < 0) {
1292 redisLog(REDIS_WARNING, "Configured to not listen anywhere, exiting.");
1293 exit(1);
1294 }
1295 for (j = 0; j < server.dbnum; j++) {
1296 server.db[j].dict = dictCreate(&dbDictType,NULL);
1297 server.db[j].expires = dictCreate(&keyptrDictType,NULL);
1298 server.db[j].blocking_keys = dictCreate(&keylistDictType,NULL);
1299 server.db[j].ready_keys = dictCreate(&setDictType,NULL);
1300 server.db[j].watched_keys = dictCreate(&keylistDictType,NULL);
1301 server.db[j].id = j;
1302 }
1303 server.pubsub_channels = dictCreate(&keylistDictType,NULL);
1304 server.pubsub_patterns = listCreate();
1305 listSetFreeMethod(server.pubsub_patterns,freePubsubPattern);
1306 listSetMatchMethod(server.pubsub_patterns,listMatchPubsubPattern);
1307 server.cronloops = 0;
1308 server.rdb_child_pid = -1;
1309 server.aof_child_pid = -1;
1310 aofRewriteBufferReset();
1311 server.aof_buf = sdsempty();
1312 server.lastsave = time(NULL);
1313 server.rdb_save_time_last = -1;
1314 server.rdb_save_time_start = -1;
1315 server.dirty = 0;
1316 server.stat_numcommands = 0;
1317 server.stat_numconnections = 0;
1318 server.stat_expiredkeys = 0;
1319 server.stat_evictedkeys = 0;
1320 server.stat_starttime = time(NULL);
1321 server.stat_keyspace_misses = 0;
1322 server.stat_keyspace_hits = 0;
1323 server.stat_peak_memory = 0;
1324 server.stat_fork_time = 0;
1325 server.stat_rejected_conn = 0;
1326 memset(server.ops_sec_samples,0,sizeof(server.ops_sec_samples));
1327 server.ops_sec_idx = 0;
1328 server.ops_sec_last_sample_time = mstime();
1329 server.ops_sec_last_sample_ops = 0;
1330 server.unixtime = time(NULL);
1331 server.lastbgsave_status = REDIS_OK;
1332 server.stop_writes_on_bgsave_err = 1;
1333 aeCreateTimeEvent(server.el, 1, serverCron, NULL, NULL);
1334 if (server.ipfd > 0 && aeCreateFileEvent(server.el,server.ipfd,AE_READABLE,
1335 acceptTcpHandler,NULL) == AE_ERR) redisPanic("Unrecoverable error creating server.ipfd file event.");
1336 if (server.sofd > 0 && aeCreateFileEvent(server.el,server.sofd,AE_READABLE,
1337 acceptUnixHandler,NULL) == AE_ERR) redisPanic("Unrecoverable error creating server.sofd file event.");
1338
1339 if (server.aof_state == REDIS_AOF_ON) {
1340 server.aof_fd = open(server.aof_filename,
1341 O_WRONLY|O_APPEND|O_CREAT,0644);
1342 if (server.aof_fd == -1) {
1343 redisLog(REDIS_WARNING, "Can't open the append-only file: %s",
1344 strerror(errno));
1345 exit(1);
1346 }
1347 }
1348
1349 /* 32 bit instances are limited to 4GB of address space, so if there is
1350 * no explicit limit in the user provided configuration we set a limit
1351 * at 3.5GB using maxmemory with 'noeviction' policy'. This saves
1352 * useless crashes of the Redis instance. */
1353 if (server.arch_bits == 32 && server.maxmemory == 0) {
1354 redisLog(REDIS_WARNING,"Warning: 32 bit instance detected but no memory limit set. Setting 3.5 GB maxmemory limit with 'noeviction' policy now.");
1355 server.maxmemory = 3584LL*(1024*1024); /* 3584 MB = 3.5 GB */
1356 server.maxmemory_policy = REDIS_MAXMEMORY_NO_EVICTION;
1357 }
1358
1359 if (server.cluster_enabled) clusterInit();
1360 scriptingInit();
1361 slowlogInit();
1362 bioInit();
1363 }
1364
1365 /* Populates the Redis Command Table starting from the hard coded list
1366 * we have on top of redis.c file. */
1367 void populateCommandTable(void) {
1368 int j;
1369 int numcommands = sizeof(redisCommandTable)/sizeof(struct redisCommand);
1370
1371 for (j = 0; j < numcommands; j++) {
1372 struct redisCommand *c = redisCommandTable+j;
1373 char *f = c->sflags;
1374 int retval;
1375
1376 while(*f != '\0') {
1377 switch(*f) {
1378 case 'w': c->flags |= REDIS_CMD_WRITE; break;
1379 case 'r': c->flags |= REDIS_CMD_READONLY; break;
1380 case 'm': c->flags |= REDIS_CMD_DENYOOM; break;
1381 case 'a': c->flags |= REDIS_CMD_ADMIN; break;
1382 case 'p': c->flags |= REDIS_CMD_PUBSUB; break;
1383 case 'f': c->flags |= REDIS_CMD_FORCE_REPLICATION; break;
1384 case 's': c->flags |= REDIS_CMD_NOSCRIPT; break;
1385 case 'R': c->flags |= REDIS_CMD_RANDOM; break;
1386 case 'S': c->flags |= REDIS_CMD_SORT_FOR_SCRIPT; break;
1387 case 'l': c->flags |= REDIS_CMD_LOADING; break;
1388 case 't': c->flags |= REDIS_CMD_STALE; break;
1389 default: redisPanic("Unsupported command flag"); break;
1390 }
1391 f++;
1392 }
1393
1394 retval = dictAdd(server.commands, sdsnew(c->name), c);
1395 assert(retval == DICT_OK);
1396 }
1397 }
1398
1399 void resetCommandTableStats(void) {
1400 int numcommands = sizeof(redisCommandTable)/sizeof(struct redisCommand);
1401 int j;
1402
1403 for (j = 0; j < numcommands; j++) {
1404 struct redisCommand *c = redisCommandTable+j;
1405
1406 c->microseconds = 0;
1407 c->calls = 0;
1408 }
1409 }
1410
1411 /* ========================== Redis OP Array API ============================ */
1412
1413 void redisOpArrayInit(redisOpArray *oa) {
1414 oa->ops = NULL;
1415 oa->numops = 0;
1416 }
1417
1418 int redisOpArrayAppend(redisOpArray *oa, struct redisCommand *cmd, int dbid,
1419 robj **argv, int argc, int target)
1420 {
1421 redisOp *op;
1422
1423 oa->ops = zrealloc(oa->ops,sizeof(redisOp)*(oa->numops+1));
1424 op = oa->ops+oa->numops;
1425 op->cmd = cmd;
1426 op->dbid = dbid;
1427 op->argv = argv;
1428 op->argc = argc;
1429 op->target = target;
1430 oa->numops++;
1431 return oa->numops;
1432 }
1433
1434 void redisOpArrayFree(redisOpArray *oa) {
1435 while(oa->numops) {
1436 int j;
1437 redisOp *op;
1438
1439 oa->numops--;
1440 op = oa->ops+oa->numops;
1441 for (j = 0; j < op->argc; j++)
1442 decrRefCount(op->argv[j]);
1443 zfree(op->argv);
1444 }
1445 zfree(oa->ops);
1446 }
1447
1448 /* ====================== Commands lookup and execution ===================== */
1449
1450 struct redisCommand *lookupCommand(sds name) {
1451 return dictFetchValue(server.commands, name);
1452 }
1453
1454 struct redisCommand *lookupCommandByCString(char *s) {
1455 struct redisCommand *cmd;
1456 sds name = sdsnew(s);
1457
1458 cmd = dictFetchValue(server.commands, name);
1459 sdsfree(name);
1460 return cmd;
1461 }
1462
1463 /* Propagate the specified command (in the context of the specified database id)
1464 * to AOF, Slaves and Monitors.
1465 *
1466 * flags are an xor between:
1467 * + REDIS_PROPAGATE_NONE (no propagation of command at all)
1468 * + REDIS_PROPAGATE_AOF (propagate into the AOF file if is enabled)
1469 * + REDIS_PROPAGATE_REPL (propagate into the replication link)
1470 */
1471 void propagate(struct redisCommand *cmd, int dbid, robj **argv, int argc,
1472 int flags)
1473 {
1474 if (server.aof_state != REDIS_AOF_OFF && flags & REDIS_PROPAGATE_AOF)
1475 feedAppendOnlyFile(cmd,dbid,argv,argc);
1476 if (flags & REDIS_PROPAGATE_REPL && listLength(server.slaves))
1477 replicationFeedSlaves(server.slaves,dbid,argv,argc);
1478 }
1479
1480 /* Used inside commands to schedule the propagation of additional commands
1481 * after the current command is propagated to AOF / Replication. */
1482 void alsoPropagate(struct redisCommand *cmd, int dbid, robj **argv, int argc,
1483 int target)
1484 {
1485 redisOpArrayAppend(&server.also_propagate,cmd,dbid,argv,argc,target);
1486 }
1487
1488 /* Call() is the core of Redis execution of a command */
1489 void call(redisClient *c, int flags) {
1490 long long dirty, start = ustime(), duration;
1491
1492 /* Sent the command to clients in MONITOR mode, only if the commands are
1493 * not geneated from reading an AOF. */
1494 if (listLength(server.monitors) && !server.loading)
1495 replicationFeedMonitors(c,server.monitors,c->db->id,c->argv,c->argc);
1496
1497 /* Call the command. */
1498 redisOpArrayInit(&server.also_propagate);
1499 dirty = server.dirty;
1500 c->cmd->proc(c);
1501 dirty = server.dirty-dirty;
1502 duration = ustime()-start;
1503
1504 /* When EVAL is called loading the AOF we don't want commands called
1505 * from Lua to go into the slowlog or to populate statistics. */
1506 if (server.loading && c->flags & REDIS_LUA_CLIENT)
1507 flags &= ~(REDIS_CALL_SLOWLOG | REDIS_CALL_STATS);
1508
1509 /* Log the command into the Slow log if needed, and populate the
1510 * per-command statistics that we show in INFO commandstats. */
1511 if (flags & REDIS_CALL_SLOWLOG)
1512 slowlogPushEntryIfNeeded(c->argv,c->argc,duration);
1513 if (flags & REDIS_CALL_STATS) {
1514 c->cmd->microseconds += duration;
1515 c->cmd->calls++;
1516 }
1517
1518 /* Propagate the command into the AOF and replication link */
1519 if (flags & REDIS_CALL_PROPAGATE) {
1520 int flags = REDIS_PROPAGATE_NONE;
1521
1522 if (c->cmd->flags & REDIS_CMD_FORCE_REPLICATION)
1523 flags |= REDIS_PROPAGATE_REPL;
1524 if (dirty)
1525 flags |= (REDIS_PROPAGATE_REPL | REDIS_PROPAGATE_AOF);
1526 if (flags != REDIS_PROPAGATE_NONE)
1527 propagate(c->cmd,c->db->id,c->argv,c->argc,flags);
1528 }
1529 /* Commands such as LPUSH or BRPOPLPUSH may propagate an additional
1530 * PUSH command. */
1531 if (server.also_propagate.numops) {
1532 int j;
1533 redisOp *rop;
1534
1535 for (j = 0; j < server.also_propagate.numops; j++) {
1536 rop = &server.also_propagate.ops[j];
1537 propagate(rop->cmd, rop->dbid, rop->argv, rop->argc, rop->target);
1538 }
1539 redisOpArrayFree(&server.also_propagate);
1540 }
1541 server.stat_numcommands++;
1542 }
1543
1544 /* If this function gets called we already read a whole
1545 * command, argments are in the client argv/argc fields.
1546 * processCommand() execute the command or prepare the
1547 * server for a bulk read from the client.
1548 *
1549 * If 1 is returned the client is still alive and valid and
1550 * and other operations can be performed by the caller. Otherwise
1551 * if 0 is returned the client was destroied (i.e. after QUIT). */
1552 int processCommand(redisClient *c) {
1553 /* The QUIT command is handled separately. Normal command procs will
1554 * go through checking for replication and QUIT will cause trouble
1555 * when FORCE_REPLICATION is enabled and would be implemented in
1556 * a regular command proc. */
1557 if (!strcasecmp(c->argv[0]->ptr,"quit")) {
1558 addReply(c,shared.ok);
1559 c->flags |= REDIS_CLOSE_AFTER_REPLY;
1560 return REDIS_ERR;
1561 }
1562
1563 /* Now lookup the command and check ASAP about trivial error conditions
1564 * such as wrong arity, bad command name and so forth. */
1565 c->cmd = c->lastcmd = lookupCommand(c->argv[0]->ptr);
1566 if (!c->cmd) {
1567 addReplyErrorFormat(c,"unknown command '%s'",
1568 (char*)c->argv[0]->ptr);
1569 return REDIS_OK;
1570 } else if ((c->cmd->arity > 0 && c->cmd->arity != c->argc) ||
1571 (c->argc < -c->cmd->arity)) {
1572 addReplyErrorFormat(c,"wrong number of arguments for '%s' command",
1573 c->cmd->name);
1574 return REDIS_OK;
1575 }
1576
1577 /* Check if the user is authenticated */
1578 if (server.requirepass && !c->authenticated && c->cmd->proc != authCommand)
1579 {
1580 addReplyError(c,"operation not permitted");
1581 return REDIS_OK;
1582 }
1583
1584 /* If cluster is enabled, redirect here */
1585 if (server.cluster_enabled &&
1586 !(c->cmd->getkeys_proc == NULL && c->cmd->firstkey == 0)) {
1587 int hashslot;
1588
1589 if (server.cluster.state != REDIS_CLUSTER_OK) {
1590 addReplyError(c,"The cluster is down. Check with CLUSTER INFO for more information");
1591 return REDIS_OK;
1592 } else {
1593 int ask;
1594 clusterNode *n = getNodeByQuery(c,c->cmd,c->argv,c->argc,&hashslot,&ask);
1595 if (n == NULL) {
1596 addReplyError(c,"Multi keys request invalid in cluster");
1597 return REDIS_OK;
1598 } else if (n != server.cluster.myself) {
1599 addReplySds(c,sdscatprintf(sdsempty(),
1600 "-%s %d %s:%d\r\n", ask ? "ASK" : "MOVED",
1601 hashslot,n->ip,n->port));
1602 return REDIS_OK;
1603 }
1604 }
1605 }
1606
1607 /* Handle the maxmemory directive.
1608 *
1609 * First we try to free some memory if possible (if there are volatile
1610 * keys in the dataset). If there are not the only thing we can do
1611 * is returning an error. */
1612 if (server.maxmemory) {
1613 int retval = freeMemoryIfNeeded();
1614 if ((c->cmd->flags & REDIS_CMD_DENYOOM) && retval == REDIS_ERR) {
1615 addReply(c, shared.oomerr);
1616 return REDIS_OK;
1617 }
1618 }
1619
1620 /* Don't accept write commands if there are problems persisting on disk. */
1621 if (server.stop_writes_on_bgsave_err &&
1622 server.saveparamslen > 0
1623 && server.lastbgsave_status == REDIS_ERR &&
1624 c->cmd->flags & REDIS_CMD_WRITE)
1625 {
1626 addReply(c, shared.bgsaveerr);
1627 return REDIS_OK;
1628 }
1629
1630 /* Don't accept write commands if this is a read only slave. But
1631 * accept write commands if this is our master. */
1632 if (server.masterhost && server.repl_slave_ro &&
1633 !(c->flags & REDIS_MASTER) &&
1634 c->cmd->flags & REDIS_CMD_WRITE)
1635 {
1636 addReply(c, shared.roslaveerr);
1637 return REDIS_OK;
1638 }
1639
1640 /* Only allow SUBSCRIBE and UNSUBSCRIBE in the context of Pub/Sub */
1641 if ((dictSize(c->pubsub_channels) > 0 || listLength(c->pubsub_patterns) > 0)
1642 &&
1643 c->cmd->proc != subscribeCommand &&
1644 c->cmd->proc != unsubscribeCommand &&
1645 c->cmd->proc != psubscribeCommand &&
1646 c->cmd->proc != punsubscribeCommand) {
1647 addReplyError(c,"only (P)SUBSCRIBE / (P)UNSUBSCRIBE / QUIT allowed in this context");
1648 return REDIS_OK;
1649 }
1650
1651 /* Only allow INFO and SLAVEOF when slave-serve-stale-data is no and
1652 * we are a slave with a broken link with master. */
1653 if (server.masterhost && server.repl_state != REDIS_REPL_CONNECTED &&
1654 server.repl_serve_stale_data == 0 &&
1655 !(c->cmd->flags & REDIS_CMD_STALE))
1656 {
1657 addReply(c, shared.masterdownerr);
1658 return REDIS_OK;
1659 }
1660
1661 /* Loading DB? Return an error if the command has not the
1662 * REDIS_CMD_LOADING flag. */
1663 if (server.loading && !(c->cmd->flags & REDIS_CMD_LOADING)) {
1664 addReply(c, shared.loadingerr);
1665 return REDIS_OK;
1666 }
1667
1668 /* Lua script too slow? Only allow commands with REDIS_CMD_STALE flag. */
1669 if (server.lua_timedout &&
1670 !(c->cmd->proc == shutdownCommand &&
1671 c->argc == 2 &&
1672 tolower(((char*)c->argv[1]->ptr)[0]) == 'n') &&
1673 !(c->cmd->proc == scriptCommand &&
1674 c->argc == 2 &&
1675 tolower(((char*)c->argv[1]->ptr)[0]) == 'k'))
1676 {
1677 addReply(c, shared.slowscripterr);
1678 return REDIS_OK;
1679 }
1680
1681 /* Exec the command */
1682 if (c->flags & REDIS_MULTI &&
1683 c->cmd->proc != execCommand && c->cmd->proc != discardCommand &&
1684 c->cmd->proc != multiCommand && c->cmd->proc != watchCommand)
1685 {
1686 queueMultiCommand(c);
1687 addReply(c,shared.queued);
1688 } else {
1689 call(c,REDIS_CALL_FULL);
1690 if (listLength(server.ready_keys))
1691 handleClientsBlockedOnLists();
1692 }
1693 return REDIS_OK;
1694 }
1695
1696 /*================================== Shutdown =============================== */
1697
1698 int prepareForShutdown(int flags) {
1699 int save = flags & REDIS_SHUTDOWN_SAVE;
1700 int nosave = flags & REDIS_SHUTDOWN_NOSAVE;
1701
1702 redisLog(REDIS_WARNING,"User requested shutdown...");
1703 /* Kill the saving child if there is a background saving in progress.
1704 We want to avoid race conditions, for instance our saving child may
1705 overwrite the synchronous saving did by SHUTDOWN. */
1706 if (server.rdb_child_pid != -1) {
1707 redisLog(REDIS_WARNING,"There is a child saving an .rdb. Killing it!");
1708 kill(server.rdb_child_pid,SIGKILL);
1709 rdbRemoveTempFile(server.rdb_child_pid);
1710 }
1711 if (server.aof_state != REDIS_AOF_OFF) {
1712 /* Kill the AOF saving child as the AOF we already have may be longer
1713 * but contains the full dataset anyway. */
1714 if (server.aof_child_pid != -1) {
1715 redisLog(REDIS_WARNING,
1716 "There is a child rewriting the AOF. Killing it!");
1717 kill(server.aof_child_pid,SIGKILL);
1718 }
1719 /* Append only file: fsync() the AOF and exit */
1720 redisLog(REDIS_NOTICE,"Calling fsync() on the AOF file.");
1721 aof_fsync(server.aof_fd);
1722 }
1723 if ((server.saveparamslen > 0 && !nosave) || save) {
1724 redisLog(REDIS_NOTICE,"Saving the final RDB snapshot before exiting.");
1725 /* Snapshotting. Perform a SYNC SAVE and exit */
1726 if (rdbSave(server.rdb_filename) != REDIS_OK) {
1727 /* Ooops.. error saving! The best we can do is to continue
1728 * operating. Note that if there was a background saving process,
1729 * in the next cron() Redis will be notified that the background
1730 * saving aborted, handling special stuff like slaves pending for
1731 * synchronization... */
1732 redisLog(REDIS_WARNING,"Error trying to save the DB, can't exit.");
1733 return REDIS_ERR;
1734 }
1735 }
1736 if (server.daemonize) {
1737 redisLog(REDIS_NOTICE,"Removing the pid file.");
1738 unlink(server.pidfile);
1739 }
1740 /* Close the listening sockets. Apparently this allows faster restarts. */
1741 if (server.ipfd != -1) close(server.ipfd);
1742 if (server.sofd != -1) close(server.sofd);
1743 if (server.unixsocket) {
1744 redisLog(REDIS_NOTICE,"Removing the unix socket file.");
1745 unlink(server.unixsocket); /* don't care if this fails */
1746 }
1747
1748 redisLog(REDIS_WARNING,"Redis is now ready to exit, bye bye...");
1749 return REDIS_OK;
1750 }
1751
1752 /*================================== Commands =============================== */
1753
1754 /* Return zero if strings are the same, non-zero if they are not.
1755 * The comparison is performed in a way that prevents an attacker to obtain
1756 * information about the nature of the strings just monitoring the execution
1757 * time of the function.
1758 *
1759 * Note that limiting the comparison length to strings up to 512 bytes we
1760 * can avoid leaking any information about the password length and any
1761 * possible branch misprediction related leak.
1762 */
1763 int time_independent_strcmp(char *a, char *b) {
1764 char bufa[REDIS_AUTHPASS_MAX_LEN], bufb[REDIS_AUTHPASS_MAX_LEN];
1765 /* The above two strlen perform len(a) + len(b) operations where either
1766 * a or b are fixed (our password) length, and the difference is only
1767 * relative to the length of the user provided string, so no information
1768 * leak is possible in the following two lines of code. */
1769 int alen = strlen(a);
1770 int blen = strlen(b);
1771 int j;
1772 int diff = 0;
1773
1774 /* We can't compare strings longer than our static buffers.
1775 * Note that this will never pass the first test in practical circumstances
1776 * so there is no info leak. */
1777 if (alen > sizeof(bufa) || blen > sizeof(bufb)) return 1;
1778
1779 memset(bufa,0,sizeof(bufa)); /* Constant time. */
1780 memset(bufb,0,sizeof(bufb)); /* Constant time. */
1781 /* Again the time of the following two copies is proportional to
1782 * len(a) + len(b) so no info is leaked. */
1783 memcpy(bufa,a,alen);
1784 memcpy(bufb,b,blen);
1785
1786 /* Always compare all the chars in the two buffers without
1787 * conditional expressions. */
1788 for (j = 0; j < sizeof(bufa); j++) {
1789 diff |= (bufa[j] ^ bufb[j]);
1790 }
1791 /* Length must be equal as well. */
1792 diff |= alen ^ blen;
1793 return diff; /* If zero strings are the same. */
1794 }
1795
1796 void authCommand(redisClient *c) {
1797 if (!server.requirepass) {
1798 addReplyError(c,"Client sent AUTH, but no password is set");
1799 } else if (!time_independent_strcmp(c->argv[1]->ptr, server.requirepass)) {
1800 c->authenticated = 1;
1801 addReply(c,shared.ok);
1802 } else {
1803 c->authenticated = 0;
1804 addReplyError(c,"invalid password");
1805 }
1806 }
1807
1808 void pingCommand(redisClient *c) {
1809 addReply(c,shared.pong);
1810 }
1811
1812 void echoCommand(redisClient *c) {
1813 addReplyBulk(c,c->argv[1]);
1814 }
1815
1816 void timeCommand(redisClient *c) {
1817 struct timeval tv;
1818
1819 /* gettimeofday() can only fail if &tv is a bad addresss so we
1820 * don't check for errors. */
1821 gettimeofday(&tv,NULL);
1822 addReplyMultiBulkLen(c,2);
1823 addReplyBulkLongLong(c,tv.tv_sec);
1824 addReplyBulkLongLong(c,tv.tv_usec);
1825 }
1826
1827 /* Convert an amount of bytes into a human readable string in the form
1828 * of 100B, 2G, 100M, 4K, and so forth. */
1829 void bytesToHuman(char *s, unsigned long long n) {
1830 double d;
1831
1832 if (n < 1024) {
1833 /* Bytes */
1834 sprintf(s,"%lluB",n);
1835 return;
1836 } else if (n < (1024*1024)) {
1837 d = (double)n/(1024);
1838 sprintf(s,"%.2fK",d);
1839 } else if (n < (1024LL*1024*1024)) {
1840 d = (double)n/(1024*1024);
1841 sprintf(s,"%.2fM",d);
1842 } else if (n < (1024LL*1024*1024*1024)) {
1843 d = (double)n/(1024LL*1024*1024);
1844 sprintf(s,"%.2fG",d);
1845 }
1846 }
1847
1848 /* Create the string returned by the INFO command. This is decoupled
1849 * by the INFO command itself as we need to report the same information
1850 * on memory corruption problems. */
1851 sds genRedisInfoString(char *section) {
1852 sds info = sdsempty();
1853 time_t uptime = server.unixtime-server.stat_starttime;
1854 int j, numcommands;
1855 struct rusage self_ru, c_ru;
1856 unsigned long lol, bib;
1857 int allsections = 0, defsections = 0;
1858 int sections = 0;
1859
1860 if (section) {
1861 allsections = strcasecmp(section,"all") == 0;
1862 defsections = strcasecmp(section,"default") == 0;
1863 }
1864
1865 getrusage(RUSAGE_SELF, &self_ru);
1866 getrusage(RUSAGE_CHILDREN, &c_ru);
1867 getClientsMaxBuffers(&lol,&bib);
1868
1869 /* Server */
1870 if (allsections || defsections || !strcasecmp(section,"server")) {
1871 struct utsname name;
1872 char *mode;
1873
1874 if (server.cluster_enabled) mode = "cluster";
1875 else if (server.sentinel_mode) mode = "sentinel";
1876 else mode = "standalone";
1877
1878 if (sections++) info = sdscat(info,"\r\n");
1879 uname(&name);
1880 info = sdscatprintf(info,
1881 "# Server\r\n"
1882 "redis_version:%s\r\n"
1883 "redis_git_sha1:%s\r\n"
1884 "redis_git_dirty:%d\r\n"
1885 "redis_mode:%s\r\n"
1886 "os:%s %s %s\r\n"
1887 "arch_bits:%d\r\n"
1888 "multiplexing_api:%s\r\n"
1889 "gcc_version:%d.%d.%d\r\n"
1890 "process_id:%ld\r\n"
1891 "run_id:%s\r\n"
1892 "tcp_port:%d\r\n"
1893 "uptime_in_seconds:%ld\r\n"
1894 "uptime_in_days:%ld\r\n"
1895 "lru_clock:%ld\r\n",
1896 REDIS_VERSION,
1897 redisGitSHA1(),
1898 strtol(redisGitDirty(),NULL,10) > 0,
1899 mode,
1900 name.sysname, name.release, name.machine,
1901 server.arch_bits,
1902 aeGetApiName(),
1903 #ifdef __GNUC__
1904 __GNUC__,__GNUC_MINOR__,__GNUC_PATCHLEVEL__,
1905 #else
1906 0,0,0,
1907 #endif
1908 (long) getpid(),
1909 server.runid,
1910 server.port,
1911 uptime,
1912 uptime/(3600*24),
1913 (unsigned long) server.lruclock);
1914 }
1915
1916 /* Clients */
1917 if (allsections || defsections || !strcasecmp(section,"clients")) {
1918 if (sections++) info = sdscat(info,"\r\n");
1919 info = sdscatprintf(info,
1920 "# Clients\r\n"
1921 "connected_clients:%lu\r\n"
1922 "client_longest_output_list:%lu\r\n"
1923 "client_biggest_input_buf:%lu\r\n"
1924 "blocked_clients:%d\r\n",
1925 listLength(server.clients)-listLength(server.slaves),
1926 lol, bib,
1927 server.bpop_blocked_clients);
1928 }
1929
1930 /* Memory */
1931 if (allsections || defsections || !strcasecmp(section,"memory")) {
1932 char hmem[64];
1933 char peak_hmem[64];
1934
1935 bytesToHuman(hmem,zmalloc_used_memory());
1936 bytesToHuman(peak_hmem,server.stat_peak_memory);
1937 if (sections++) info = sdscat(info,"\r\n");
1938 info = sdscatprintf(info,
1939 "# Memory\r\n"
1940 "used_memory:%zu\r\n"
1941 "used_memory_human:%s\r\n"
1942 "used_memory_rss:%zu\r\n"
1943 "used_memory_peak:%zu\r\n"
1944 "used_memory_peak_human:%s\r\n"
1945 "used_memory_lua:%lld\r\n"
1946 "mem_fragmentation_ratio:%.2f\r\n"
1947 "mem_allocator:%s\r\n",
1948 zmalloc_used_memory(),
1949 hmem,
1950 zmalloc_get_rss(),
1951 server.stat_peak_memory,
1952 peak_hmem,
1953 ((long long)lua_gc(server.lua,LUA_GCCOUNT,0))*1024LL,
1954 zmalloc_get_fragmentation_ratio(),
1955 ZMALLOC_LIB
1956 );
1957 }
1958
1959 /* Persistence */
1960 if (allsections || defsections || !strcasecmp(section,"persistence")) {
1961 if (sections++) info = sdscat(info,"\r\n");
1962 info = sdscatprintf(info,
1963 "# Persistence\r\n"
1964 "loading:%d\r\n"
1965 "rdb_changes_since_last_save:%lld\r\n"
1966 "rdb_bgsave_in_progress:%d\r\n"
1967 "rdb_last_save_time:%ld\r\n"
1968 "rdb_last_bgsave_status:%s\r\n"
1969 "rdb_last_bgsave_time_sec:%ld\r\n"
1970 "rdb_current_bgsave_time_sec:%ld\r\n"
1971 "aof_enabled:%d\r\n"
1972 "aof_rewrite_in_progress:%d\r\n"
1973 "aof_rewrite_scheduled:%d\r\n"
1974 "aof_last_rewrite_time_sec:%ld\r\n"
1975 "aof_current_rewrite_time_sec:%ld\r\n"
1976 "aof_last_bgrewrite_status:%s\r\n",
1977 server.loading,
1978 server.dirty,
1979 server.rdb_child_pid != -1,
1980 server.lastsave,
1981 (server.lastbgsave_status == REDIS_OK) ? "ok" : "err",
1982 server.rdb_save_time_last,
1983 (server.rdb_child_pid == -1) ?
1984 -1 : time(NULL)-server.rdb_save_time_start,
1985 server.aof_state != REDIS_AOF_OFF,
1986 server.aof_child_pid != -1,
1987 server.aof_rewrite_scheduled,
1988 server.aof_rewrite_time_last,
1989 (server.aof_child_pid == -1) ?
1990 -1 : time(NULL)-server.aof_rewrite_time_start,
1991 (server.aof_lastbgrewrite_status == REDIS_OK) ? "ok" : "err");
1992
1993 if (server.aof_state != REDIS_AOF_OFF) {
1994 info = sdscatprintf(info,
1995 "aof_current_size:%lld\r\n"
1996 "aof_base_size:%lld\r\n"
1997 "aof_pending_rewrite:%d\r\n"
1998 "aof_buffer_length:%zu\r\n"
1999 "aof_rewrite_buffer_length:%lu\r\n"
2000 "aof_pending_bio_fsync:%llu\r\n"
2001 "aof_delayed_fsync:%lu\r\n",
2002 (long long) server.aof_current_size,
2003 (long long) server.aof_rewrite_base_size,
2004 server.aof_rewrite_scheduled,
2005 sdslen(server.aof_buf),
2006 aofRewriteBufferSize(),
2007 bioPendingJobsOfType(REDIS_BIO_AOF_FSYNC),
2008 server.aof_delayed_fsync);
2009 }
2010
2011 if (server.loading) {
2012 double perc;
2013 time_t eta, elapsed;
2014 off_t remaining_bytes = server.loading_total_bytes-
2015 server.loading_loaded_bytes;
2016
2017 perc = ((double)server.loading_loaded_bytes /
2018 server.loading_total_bytes) * 100;
2019
2020 elapsed = server.unixtime-server.loading_start_time;
2021 if (elapsed == 0) {
2022 eta = 1; /* A fake 1 second figure if we don't have
2023 enough info */
2024 } else {
2025 eta = (elapsed*remaining_bytes)/server.loading_loaded_bytes;
2026 }
2027
2028 info = sdscatprintf(info,
2029 "loading_start_time:%ld\r\n"
2030 "loading_total_bytes:%llu\r\n"
2031 "loading_loaded_bytes:%llu\r\n"
2032 "loading_loaded_perc:%.2f\r\n"
2033 "loading_eta_seconds:%ld\r\n"
2034 ,(unsigned long) server.loading_start_time,
2035 (unsigned long long) server.loading_total_bytes,
2036 (unsigned long long) server.loading_loaded_bytes,
2037 perc,
2038 eta
2039 );
2040 }
2041 }
2042
2043 /* Stats */
2044 if (allsections || defsections || !strcasecmp(section,"stats")) {
2045 if (sections++) info = sdscat(info,"\r\n");
2046 info = sdscatprintf(info,
2047 "# Stats\r\n"
2048 "total_connections_received:%lld\r\n"
2049 "total_commands_processed:%lld\r\n"
2050 "instantaneous_ops_per_sec:%lld\r\n"
2051 "rejected_connections:%lld\r\n"
2052 "expired_keys:%lld\r\n"
2053 "evicted_keys:%lld\r\n"
2054 "keyspace_hits:%lld\r\n"
2055 "keyspace_misses:%lld\r\n"
2056 "pubsub_channels:%ld\r\n"
2057 "pubsub_patterns:%lu\r\n"
2058 "latest_fork_usec:%lld\r\n",
2059 server.stat_numconnections,
2060 server.stat_numcommands,
2061 getOperationsPerSecond(),
2062 server.stat_rejected_conn,
2063 server.stat_expiredkeys,
2064 server.stat_evictedkeys,
2065 server.stat_keyspace_hits,
2066 server.stat_keyspace_misses,
2067 dictSize(server.pubsub_channels),
2068 listLength(server.pubsub_patterns),
2069 server.stat_fork_time);
2070 }
2071
2072 /* Replication */
2073 if (allsections || defsections || !strcasecmp(section,"replication")) {
2074 if (sections++) info = sdscat(info,"\r\n");
2075 info = sdscatprintf(info,
2076 "# Replication\r\n"
2077 "role:%s\r\n",
2078 server.masterhost == NULL ? "master" : "slave");
2079 if (server.masterhost) {
2080 info = sdscatprintf(info,
2081 "master_host:%s\r\n"
2082 "master_port:%d\r\n"
2083 "master_link_status:%s\r\n"
2084 "master_last_io_seconds_ago:%d\r\n"
2085 "master_sync_in_progress:%d\r\n"
2086 ,server.masterhost,
2087 server.masterport,
2088 (server.repl_state == REDIS_REPL_CONNECTED) ?
2089 "up" : "down",
2090 server.master ?
2091 ((int)(server.unixtime-server.master->lastinteraction)) : -1,
2092 server.repl_state == REDIS_REPL_TRANSFER
2093 );
2094
2095 if (server.repl_state == REDIS_REPL_TRANSFER) {
2096 info = sdscatprintf(info,
2097 "master_sync_left_bytes:%lld\r\n"
2098 "master_sync_last_io_seconds_ago:%d\r\n"
2099 , (long long)
2100 (server.repl_transfer_size - server.repl_transfer_read),
2101 (int)(server.unixtime-server.repl_transfer_lastio)
2102 );
2103 }
2104
2105 if (server.repl_state != REDIS_REPL_CONNECTED) {
2106 info = sdscatprintf(info,
2107 "master_link_down_since_seconds:%ld\r\n",
2108 (long)server.unixtime-server.repl_down_since);
2109 }
2110 info = sdscatprintf(info,
2111 "slave_priority:%d\r\n", server.slave_priority);
2112 }
2113 info = sdscatprintf(info,
2114 "connected_slaves:%lu\r\n",
2115 listLength(server.slaves));
2116 if (listLength(server.slaves)) {
2117 int slaveid = 0;
2118 listNode *ln;
2119 listIter li;
2120
2121 listRewind(server.slaves,&li);
2122 while((ln = listNext(&li))) {
2123 redisClient *slave = listNodeValue(ln);
2124 char *state = NULL;
2125 char ip[32];
2126 int port;
2127
2128 if (anetPeerToString(slave->fd,ip,&port) == -1) continue;
2129 switch(slave->replstate) {
2130 case REDIS_REPL_WAIT_BGSAVE_START:
2131 case REDIS_REPL_WAIT_BGSAVE_END:
2132 state = "wait_bgsave";
2133 break;
2134 case REDIS_REPL_SEND_BULK:
2135 state = "send_bulk";
2136 break;
2137 case REDIS_REPL_ONLINE:
2138 state = "online";
2139 break;
2140 }
2141 if (state == NULL) continue;
2142 info = sdscatprintf(info,"slave%d:%s,%d,%s\r\n",
2143 slaveid,ip,slave->slave_listening_port,state);
2144 slaveid++;
2145 }
2146 }
2147 }
2148
2149 /* CPU */
2150 if (allsections || defsections || !strcasecmp(section,"cpu")) {
2151 if (sections++) info = sdscat(info,"\r\n");
2152 info = sdscatprintf(info,
2153 "# CPU\r\n"
2154 "used_cpu_sys:%.2f\r\n"
2155 "used_cpu_user:%.2f\r\n"
2156 "used_cpu_sys_children:%.2f\r\n"
2157 "used_cpu_user_children:%.2f\r\n",
2158 (float)self_ru.ru_stime.tv_sec+(float)self_ru.ru_stime.tv_usec/1000000,
2159 (float)self_ru.ru_utime.tv_sec+(float)self_ru.ru_utime.tv_usec/1000000,
2160 (float)c_ru.ru_stime.tv_sec+(float)c_ru.ru_stime.tv_usec/1000000,
2161 (float)c_ru.ru_utime.tv_sec+(float)c_ru.ru_utime.tv_usec/1000000);
2162 }
2163
2164 /* cmdtime */
2165 if (allsections || !strcasecmp(section,"commandstats")) {
2166 if (sections++) info = sdscat(info,"\r\n");
2167 info = sdscatprintf(info, "# Commandstats\r\n");
2168 numcommands = sizeof(redisCommandTable)/sizeof(struct redisCommand);
2169 for (j = 0; j < numcommands; j++) {
2170 struct redisCommand *c = redisCommandTable+j;
2171
2172 if (!c->calls) continue;
2173 info = sdscatprintf(info,
2174 "cmdstat_%s:calls=%lld,usec=%lld,usec_per_call=%.2f\r\n",
2175 c->name, c->calls, c->microseconds,
2176 (c->calls == 0) ? 0 : ((float)c->microseconds/c->calls));
2177 }
2178 }
2179
2180 /* Cluster */
2181 if (allsections || defsections || !strcasecmp(section,"cluster")) {
2182 if (sections++) info = sdscat(info,"\r\n");
2183 info = sdscatprintf(info,
2184 "# Cluster\r\n"
2185 "cluster_enabled:%d\r\n",
2186 server.cluster_enabled);
2187 }
2188
2189 /* Key space */
2190 if (allsections || defsections || !strcasecmp(section,"keyspace")) {
2191 if (sections++) info = sdscat(info,"\r\n");
2192 info = sdscatprintf(info, "# Keyspace\r\n");
2193 for (j = 0; j < server.dbnum; j++) {
2194 long long keys, vkeys;
2195
2196 keys = dictSize(server.db[j].dict);
2197 vkeys = dictSize(server.db[j].expires);
2198 if (keys || vkeys) {
2199 info = sdscatprintf(info, "db%d:keys=%lld,expires=%lld\r\n",
2200 j, keys, vkeys);
2201 }
2202 }
2203 }
2204 return info;
2205 }
2206
2207 void infoCommand(redisClient *c) {
2208 char *section = c->argc == 2 ? c->argv[1]->ptr : "default";
2209
2210 if (c->argc > 2) {
2211 addReply(c,shared.syntaxerr);
2212 return;
2213 }
2214 sds info = genRedisInfoString(section);
2215 addReplySds(c,sdscatprintf(sdsempty(),"$%lu\r\n",
2216 (unsigned long)sdslen(info)));
2217 addReplySds(c,info);
2218 addReply(c,shared.crlf);
2219 }
2220
2221 void monitorCommand(redisClient *c) {
2222 /* ignore MONITOR if aleady slave or in monitor mode */
2223 if (c->flags & REDIS_SLAVE) return;
2224
2225 c->flags |= (REDIS_SLAVE|REDIS_MONITOR);
2226 c->slaveseldb = 0;
2227 listAddNodeTail(server.monitors,c);
2228 addReply(c,shared.ok);
2229 }
2230
2231 /* ============================ Maxmemory directive ======================== */
2232
2233 /* This function gets called when 'maxmemory' is set on the config file to limit
2234 * the max memory used by the server, before processing a command.
2235 *
2236 * The goal of the function is to free enough memory to keep Redis under the
2237 * configured memory limit.
2238 *
2239 * The function starts calculating how many bytes should be freed to keep
2240 * Redis under the limit, and enters a loop selecting the best keys to
2241 * evict accordingly to the configured policy.
2242 *
2243 * If all the bytes needed to return back under the limit were freed the
2244 * function returns REDIS_OK, otherwise REDIS_ERR is returned, and the caller
2245 * should block the execution of commands that will result in more memory
2246 * used by the server.
2247 */
2248 int freeMemoryIfNeeded(void) {
2249 size_t mem_used, mem_tofree, mem_freed;
2250 int slaves = listLength(server.slaves);
2251
2252 /* Remove the size of slaves output buffers and AOF buffer from the
2253 * count of used memory. */
2254 mem_used = zmalloc_used_memory();
2255 if (slaves) {
2256 listIter li;
2257 listNode *ln;
2258
2259 listRewind(server.slaves,&li);
2260 while((ln = listNext(&li))) {
2261 redisClient *slave = listNodeValue(ln);
2262 unsigned long obuf_bytes = getClientOutputBufferMemoryUsage(slave);
2263 if (obuf_bytes > mem_used)
2264 mem_used = 0;
2265 else
2266 mem_used -= obuf_bytes;
2267 }
2268 }
2269 if (server.aof_state != REDIS_AOF_OFF) {
2270 mem_used -= sdslen(server.aof_buf);
2271 mem_used -= aofRewriteBufferSize();
2272 }
2273
2274 /* Check if we are over the memory limit. */
2275 if (mem_used <= server.maxmemory) return REDIS_OK;
2276
2277 if (server.maxmemory_policy == REDIS_MAXMEMORY_NO_EVICTION)
2278 return REDIS_ERR; /* We need to free memory, but policy forbids. */
2279
2280 /* Compute how much memory we need to free. */
2281 mem_tofree = mem_used - server.maxmemory;
2282 mem_freed = 0;
2283 while (mem_freed < mem_tofree) {
2284 int j, k, keys_freed = 0;
2285
2286 for (j = 0; j < server.dbnum; j++) {
2287 long bestval = 0; /* just to prevent warning */
2288 sds bestkey = NULL;
2289 struct dictEntry *de;
2290 redisDb *db = server.db+j;
2291 dict *dict;
2292
2293 if (server.maxmemory_policy == REDIS_MAXMEMORY_ALLKEYS_LRU ||
2294 server.maxmemory_policy == REDIS_MAXMEMORY_ALLKEYS_RANDOM)
2295 {
2296 dict = server.db[j].dict;
2297 } else {
2298 dict = server.db[j].expires;
2299 }
2300 if (dictSize(dict) == 0) continue;
2301
2302 /* volatile-random and allkeys-random policy */
2303 if (server.maxmemory_policy == REDIS_MAXMEMORY_ALLKEYS_RANDOM ||
2304 server.maxmemory_policy == REDIS_MAXMEMORY_VOLATILE_RANDOM)
2305 {
2306 de = dictGetRandomKey(dict);
2307 bestkey = dictGetKey(de);
2308 }
2309
2310 /* volatile-lru and allkeys-lru policy */
2311 else if (server.maxmemory_policy == REDIS_MAXMEMORY_ALLKEYS_LRU ||
2312 server.maxmemory_policy == REDIS_MAXMEMORY_VOLATILE_LRU)
2313 {
2314 for (k = 0; k < server.maxmemory_samples; k++) {
2315 sds thiskey;
2316 long thisval;
2317 robj *o;
2318
2319 de = dictGetRandomKey(dict);
2320 thiskey = dictGetKey(de);
2321 /* When policy is volatile-lru we need an additonal lookup
2322 * to locate the real key, as dict is set to db->expires. */
2323 if (server.maxmemory_policy == REDIS_MAXMEMORY_VOLATILE_LRU)
2324 de = dictFind(db->dict, thiskey);
2325 o = dictGetVal(de);
2326 thisval = estimateObjectIdleTime(o);
2327
2328 /* Higher idle time is better candidate for deletion */
2329 if (bestkey == NULL || thisval > bestval) {
2330 bestkey = thiskey;
2331 bestval = thisval;
2332 }
2333 }
2334 }
2335
2336 /* volatile-ttl */
2337 else if (server.maxmemory_policy == REDIS_MAXMEMORY_VOLATILE_TTL) {
2338 for (k = 0; k < server.maxmemory_samples; k++) {
2339 sds thiskey;
2340 long thisval;
2341
2342 de = dictGetRandomKey(dict);
2343 thiskey = dictGetKey(de);
2344 thisval = (long) dictGetVal(de);
2345
2346 /* Expire sooner (minor expire unix timestamp) is better
2347 * candidate for deletion */
2348 if (bestkey == NULL || thisval < bestval) {
2349 bestkey = thiskey;
2350 bestval = thisval;
2351 }
2352 }
2353 }
2354
2355 /* Finally remove the selected key. */
2356 if (bestkey) {
2357 long long delta;
2358
2359 robj *keyobj = createStringObject(bestkey,sdslen(bestkey));
2360 propagateExpire(db,keyobj);
2361 /* We compute the amount of memory freed by dbDelete() alone.
2362 * It is possible that actually the memory needed to propagate
2363 * the DEL in AOF and replication link is greater than the one
2364 * we are freeing removing the key, but we can't account for
2365 * that otherwise we would never exit the loop.
2366 *
2367 * AOF and Output buffer memory will be freed eventually so
2368 * we only care about memory used by the key space. */
2369 delta = (long long) zmalloc_used_memory();
2370 dbDelete(db,keyobj);
2371 delta -= (long long) zmalloc_used_memory();
2372 mem_freed += delta;
2373 server.stat_evictedkeys++;
2374 decrRefCount(keyobj);
2375 keys_freed++;
2376
2377 /* When the memory to free starts to be big enough, we may
2378 * start spending so much time here that is impossible to
2379 * deliver data to the slaves fast enough, so we force the
2380 * transmission here inside the loop. */
2381 if (slaves) flushSlavesOutputBuffers();
2382 }
2383 }
2384 if (!keys_freed) return REDIS_ERR; /* nothing to free... */
2385 }
2386 return REDIS_OK;
2387 }
2388
2389 /* =================================== Main! ================================ */
2390
2391 #ifdef __linux__
2392 int linuxOvercommitMemoryValue(void) {
2393 FILE *fp = fopen("/proc/sys/vm/overcommit_memory","r");
2394 char buf[64];
2395
2396 if (!fp) return -1;
2397 if (fgets(buf,64,fp) == NULL) {
2398 fclose(fp);
2399 return -1;
2400 }
2401 fclose(fp);
2402
2403 return atoi(buf);
2404 }
2405
2406 void linuxOvercommitMemoryWarning(void) {
2407 if (linuxOvercommitMemoryValue() == 0) {
2408 redisLog(REDIS_WARNING,"WARNING overcommit_memory is set to 0! Background save may fail under low memory condition. To fix this issue add 'vm.overcommit_memory = 1' to /etc/sysctl.conf and then reboot or run the command 'sysctl vm.overcommit_memory=1' for this to take effect.");
2409 }
2410 }
2411 #endif /* __linux__ */
2412
2413 void createPidFile(void) {
2414 /* Try to write the pid file in a best-effort way. */
2415 FILE *fp = fopen(server.pidfile,"w");
2416 if (fp) {
2417 fprintf(fp,"%d\n",(int)getpid());
2418 fclose(fp);
2419 }
2420 }
2421
2422 void daemonize(void) {
2423 int fd;
2424
2425 if (fork() != 0) exit(0); /* parent exits */
2426 setsid(); /* create a new session */
2427
2428 /* Every output goes to /dev/null. If Redis is daemonized but
2429 * the 'logfile' is set to 'stdout' in the configuration file
2430 * it will not log at all. */
2431 if ((fd = open("/dev/null", O_RDWR, 0)) != -1) {
2432 dup2(fd, STDIN_FILENO);
2433 dup2(fd, STDOUT_FILENO);
2434 dup2(fd, STDERR_FILENO);
2435 if (fd > STDERR_FILENO) close(fd);
2436 }
2437 }
2438
2439 void version() {
2440 printf("Redis server v=%s sha=%s:%d malloc=%s bits=%d\n",
2441 REDIS_VERSION,
2442 redisGitSHA1(),
2443 atoi(redisGitDirty()) > 0,
2444 ZMALLOC_LIB,
2445 sizeof(long) == 4 ? 32 : 64);
2446 exit(0);
2447 }
2448
2449 void usage() {
2450 fprintf(stderr,"Usage: ./redis-server [/path/to/redis.conf] [options]\n");
2451 fprintf(stderr," ./redis-server - (read config from stdin)\n");
2452 fprintf(stderr," ./redis-server -v or --version\n");
2453 fprintf(stderr," ./redis-server -h or --help\n");
2454 fprintf(stderr," ./redis-server --test-memory <megabytes>\n\n");
2455 fprintf(stderr,"Examples:\n");
2456 fprintf(stderr," ./redis-server (run the server with default conf)\n");
2457 fprintf(stderr," ./redis-server /etc/redis/6379.conf\n");
2458 fprintf(stderr," ./redis-server --port 7777\n");
2459 fprintf(stderr," ./redis-server --port 7777 --slaveof 127.0.0.1 8888\n");
2460 fprintf(stderr," ./redis-server /etc/myredis.conf --loglevel verbose\n\n");
2461 fprintf(stderr,"Sentinel mode:\n");
2462 fprintf(stderr," ./redis-server /etc/sentinel.conf --sentinel\n");
2463 exit(1);
2464 }
2465
2466 void redisAsciiArt(void) {
2467 #include "asciilogo.h"
2468 char *buf = zmalloc(1024*16);
2469 char *mode = "stand alone";
2470
2471 if (server.cluster_enabled) mode = "cluster";
2472 else if (server.sentinel_mode) mode = "sentinel";
2473
2474 snprintf(buf,1024*16,ascii_logo,
2475 REDIS_VERSION,
2476 redisGitSHA1(),
2477 strtol(redisGitDirty(),NULL,10) > 0,
2478 (sizeof(long) == 8) ? "64" : "32",
2479 mode, server.port,
2480 (long) getpid()
2481 );
2482 redisLogRaw(REDIS_NOTICE|REDIS_LOG_RAW,buf);
2483 zfree(buf);
2484 }
2485
2486 static void sigtermHandler(int sig) {
2487 REDIS_NOTUSED(sig);
2488
2489 redisLogFromHandler(REDIS_WARNING,"Received SIGTERM, scheduling shutdown...");
2490 server.shutdown_asap = 1;
2491 }
2492
2493 void setupSignalHandlers(void) {
2494 struct sigaction act;
2495
2496 /* When the SA_SIGINFO flag is set in sa_flags then sa_sigaction is used.
2497 * Otherwise, sa_handler is used. */
2498 sigemptyset(&act.sa_mask);
2499 act.sa_flags = 0;
2500 act.sa_handler = sigtermHandler;
2501 sigaction(SIGTERM, &act, NULL);
2502
2503 #ifdef HAVE_BACKTRACE
2504 sigemptyset(&act.sa_mask);
2505 act.sa_flags = SA_NODEFER | SA_RESETHAND | SA_SIGINFO;
2506 act.sa_sigaction = sigsegvHandler;
2507 sigaction(SIGSEGV, &act, NULL);
2508 sigaction(SIGBUS, &act, NULL);
2509 sigaction(SIGFPE, &act, NULL);
2510 sigaction(SIGILL, &act, NULL);
2511 #endif
2512 return;
2513 }
2514
2515 void memtest(size_t megabytes, int passes);
2516
2517 /* Returns 1 if there is --sentinel among the arguments or if
2518 * argv[0] is exactly "redis-sentinel". */
2519 int checkForSentinelMode(int argc, char **argv) {
2520 int j;
2521
2522 if (strstr(argv[0],"redis-sentinel") != NULL) return 1;
2523 for (j = 1; j < argc; j++)
2524 if (!strcmp(argv[j],"--sentinel")) return 1;
2525 return 0;
2526 }
2527
2528 /* Function called at startup to load RDB or AOF file in memory. */
2529 void loadDataFromDisk(void) {
2530 long long start = ustime();
2531 if (server.aof_state == REDIS_AOF_ON) {
2532 if (loadAppendOnlyFile(server.aof_filename) == REDIS_OK)
2533 redisLog(REDIS_NOTICE,"DB loaded from append only file: %.3f seconds",(float)(ustime()-start)/1000000);
2534 } else {
2535 if (rdbLoad(server.rdb_filename) == REDIS_OK) {
2536 redisLog(REDIS_NOTICE,"DB loaded from disk: %.3f seconds",
2537 (float)(ustime()-start)/1000000);
2538 } else if (errno != ENOENT) {
2539 redisLog(REDIS_WARNING,"Fatal error loading the DB. Exiting.");
2540 exit(1);
2541 }
2542 }
2543 }
2544
2545 void redisOutOfMemoryHandler(size_t allocation_size) {
2546 redisLog(REDIS_WARNING,"Out Of Memory allocating %zu bytes!",
2547 allocation_size);
2548 redisPanic("OOM");
2549 }
2550
2551 int main(int argc, char **argv) {
2552 struct timeval tv;
2553
2554 /* We need to initialize our libraries, and the server configuration. */
2555 zmalloc_enable_thread_safeness();
2556 zmalloc_set_oom_handler(redisOutOfMemoryHandler);
2557 srand(time(NULL)^getpid());
2558 gettimeofday(&tv,NULL);
2559 dictSetHashFunctionSeed(tv.tv_sec^tv.tv_usec^getpid());
2560 server.sentinel_mode = checkForSentinelMode(argc,argv);
2561 initServerConfig();
2562
2563 /* We need to init sentinel right now as parsing the configuration file
2564 * in sentinel mode will have the effect of populating the sentinel
2565 * data structures with master nodes to monitor. */
2566 if (server.sentinel_mode) {
2567 initSentinelConfig();
2568 initSentinel();
2569 }
2570
2571 if (argc >= 2) {
2572 int j = 1; /* First option to parse in argv[] */
2573 sds options = sdsempty();
2574 char *configfile = NULL;
2575
2576 /* Handle special options --help and --version */
2577 if (strcmp(argv[1], "-v") == 0 ||
2578 strcmp(argv[1], "--version") == 0) version();
2579 if (strcmp(argv[1], "--help") == 0 ||
2580 strcmp(argv[1], "-h") == 0) usage();
2581 if (strcmp(argv[1], "--test-memory") == 0) {
2582 if (argc == 3) {
2583 memtest(atoi(argv[2]),50);
2584 exit(0);
2585 } else {
2586 fprintf(stderr,"Please specify the amount of memory to test in megabytes.\n");
2587 fprintf(stderr,"Example: ./redis-server --test-memory 4096\n\n");
2588 exit(1);
2589 }
2590 }
2591
2592 /* First argument is the config file name? */
2593 if (argv[j][0] != '-' || argv[j][1] != '-')
2594 configfile = argv[j++];
2595 /* All the other options are parsed and conceptually appended to the
2596 * configuration file. For instance --port 6380 will generate the
2597 * string "port 6380\n" to be parsed after the actual file name
2598 * is parsed, if any. */
2599 while(j != argc) {
2600 if (argv[j][0] == '-' && argv[j][1] == '-') {
2601 /* Option name */
2602 if (sdslen(options)) options = sdscat(options,"\n");
2603 options = sdscat(options,argv[j]+2);
2604 options = sdscat(options," ");
2605 } else {
2606 /* Option argument */
2607 options = sdscatrepr(options,argv[j],strlen(argv[j]));
2608 options = sdscat(options," ");
2609 }
2610 j++;
2611 }
2612 resetServerSaveParams();
2613 loadServerConfig(configfile,options);
2614 sdsfree(options);
2615 } else {
2616 redisLog(REDIS_WARNING, "Warning: no config file specified, using the default config. In order to specify a config file use %s /path/to/%s.conf", argv[0], server.sentinel_mode ? "sentinel" : "redis");
2617 }
2618 if (server.daemonize) daemonize();
2619 initServer();
2620 if (server.daemonize) createPidFile();
2621 redisAsciiArt();
2622
2623 if (!server.sentinel_mode) {
2624 /* Things only needed when not runnign in Sentinel mode. */
2625 redisLog(REDIS_WARNING,"Server started, Redis version " REDIS_VERSION);
2626 #ifdef __linux__
2627 linuxOvercommitMemoryWarning();
2628 #endif
2629 loadDataFromDisk();
2630 if (server.ipfd > 0)
2631 redisLog(REDIS_NOTICE,"The server is now ready to accept connections on port %d", server.port);
2632 if (server.sofd > 0)
2633 redisLog(REDIS_NOTICE,"The server is now ready to accept connections at %s", server.unixsocket);
2634 }
2635
2636 aeSetBeforeSleepProc(server.el,beforeSleep);
2637 aeMain(server.el);
2638 aeDeleteEventLoop(server.el);
2639 return 0;
2640 }
2641
2642 /* The End */