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