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