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