]> git.saurik.com Git - redis.git/blame - src/config.c
Prevent clients from making too large multibulk requests
[redis.git] / src / config.c
CommitLineData
e2641e09 1#include "redis.h"
2
3/*-----------------------------------------------------------------------------
4 * Config file parsing
5 *----------------------------------------------------------------------------*/
6
7int yesnotoi(char *s) {
8 if (!strcasecmp(s,"yes")) return 1;
9 else if (!strcasecmp(s,"no")) return 0;
10 else return -1;
11}
12
13void appendServerSaveParams(time_t seconds, int changes) {
14 server.saveparams = zrealloc(server.saveparams,sizeof(struct saveparam)*(server.saveparamslen+1));
15 server.saveparams[server.saveparamslen].seconds = seconds;
16 server.saveparams[server.saveparamslen].changes = changes;
17 server.saveparamslen++;
18}
19
20void resetServerSaveParams() {
21 zfree(server.saveparams);
22 server.saveparams = NULL;
23 server.saveparamslen = 0;
24}
25
26/* I agree, this is a very rudimental way to load a configuration...
27 will improve later if the config gets more complex */
28void loadServerConfig(char *filename) {
29 FILE *fp;
30 char buf[REDIS_CONFIGLINE_MAX+1], *err = NULL;
31 int linenum = 0;
32 sds line = NULL;
33
34 if (filename[0] == '-' && filename[1] == '\0')
35 fp = stdin;
36 else {
37 if ((fp = fopen(filename,"r")) == NULL) {
38 redisLog(REDIS_WARNING, "Fatal error, can't open config file '%s'", filename);
39 exit(1);
40 }
41 }
42
43 while(fgets(buf,REDIS_CONFIGLINE_MAX+1,fp) != NULL) {
44 sds *argv;
45 int argc, j;
46
47 linenum++;
48 line = sdsnew(buf);
49 line = sdstrim(line," \t\r\n");
50
51 /* Skip comments and blank lines*/
52 if (line[0] == '#' || line[0] == '\0') {
53 sdsfree(line);
54 continue;
55 }
56
57 /* Split into arguments */
58 argv = sdssplitlen(line,sdslen(line)," ",1,&argc);
59 sdstolower(argv[0]);
60
61 /* Execute config directives */
62 if (!strcasecmp(argv[0],"timeout") && argc == 2) {
63 server.maxidletime = atoi(argv[1]);
64 if (server.maxidletime < 0) {
65 err = "Invalid timeout value"; goto loaderr;
66 }
67 } else if (!strcasecmp(argv[0],"port") && argc == 2) {
68 server.port = atoi(argv[1]);
69 if (server.port < 1 || server.port > 65535) {
70 err = "Invalid port"; goto loaderr;
71 }
72 } else if (!strcasecmp(argv[0],"bind") && argc == 2) {
73 server.bindaddr = zstrdup(argv[1]);
74 } else if (!strcasecmp(argv[0],"save") && argc == 3) {
75 int seconds = atoi(argv[1]);
76 int changes = atoi(argv[2]);
77 if (seconds < 1 || changes < 0) {
78 err = "Invalid save parameters"; goto loaderr;
79 }
80 appendServerSaveParams(seconds,changes);
81 } else if (!strcasecmp(argv[0],"dir") && argc == 2) {
82 if (chdir(argv[1]) == -1) {
83 redisLog(REDIS_WARNING,"Can't chdir to '%s': %s",
84 argv[1], strerror(errno));
85 exit(1);
86 }
87 } else if (!strcasecmp(argv[0],"loglevel") && argc == 2) {
88 if (!strcasecmp(argv[1],"debug")) server.verbosity = REDIS_DEBUG;
89 else if (!strcasecmp(argv[1],"verbose")) server.verbosity = REDIS_VERBOSE;
90 else if (!strcasecmp(argv[1],"notice")) server.verbosity = REDIS_NOTICE;
91 else if (!strcasecmp(argv[1],"warning")) server.verbosity = REDIS_WARNING;
92 else {
93 err = "Invalid log level. Must be one of debug, notice, warning";
94 goto loaderr;
95 }
96 } else if (!strcasecmp(argv[0],"logfile") && argc == 2) {
97 FILE *logfp;
98
99 server.logfile = zstrdup(argv[1]);
100 if (!strcasecmp(server.logfile,"stdout")) {
101 zfree(server.logfile);
102 server.logfile = NULL;
103 }
104 if (server.logfile) {
105 /* Test if we are able to open the file. The server will not
106 * be able to abort just for this problem later... */
107 logfp = fopen(server.logfile,"a");
108 if (logfp == NULL) {
109 err = sdscatprintf(sdsempty(),
110 "Can't open the log file: %s", strerror(errno));
111 goto loaderr;
112 }
113 fclose(logfp);
114 }
115 } else if (!strcasecmp(argv[0],"databases") && argc == 2) {
116 server.dbnum = atoi(argv[1]);
117 if (server.dbnum < 1) {
118 err = "Invalid number of databases"; goto loaderr;
119 }
120 } else if (!strcasecmp(argv[0],"include") && argc == 2) {
121 loadServerConfig(argv[1]);
122 } else if (!strcasecmp(argv[0],"maxclients") && argc == 2) {
123 server.maxclients = atoi(argv[1]);
124 } else if (!strcasecmp(argv[0],"maxmemory") && argc == 2) {
125 server.maxmemory = memtoll(argv[1],NULL);
126 } else if (!strcasecmp(argv[0],"slaveof") && argc == 3) {
127 server.masterhost = sdsnew(argv[1]);
128 server.masterport = atoi(argv[2]);
129 server.replstate = REDIS_REPL_CONNECT;
130 } else if (!strcasecmp(argv[0],"masterauth") && argc == 2) {
131 server.masterauth = zstrdup(argv[1]);
132 } else if (!strcasecmp(argv[0],"glueoutputbuf") && argc == 2) {
133 if ((server.glueoutputbuf = yesnotoi(argv[1])) == -1) {
134 err = "argument must be 'yes' or 'no'"; goto loaderr;
135 }
136 } else if (!strcasecmp(argv[0],"rdbcompression") && argc == 2) {
137 if ((server.rdbcompression = yesnotoi(argv[1])) == -1) {
138 err = "argument must be 'yes' or 'no'"; goto loaderr;
139 }
140 } else if (!strcasecmp(argv[0],"activerehashing") && argc == 2) {
141 if ((server.activerehashing = yesnotoi(argv[1])) == -1) {
142 err = "argument must be 'yes' or 'no'"; goto loaderr;
143 }
144 } else if (!strcasecmp(argv[0],"daemonize") && argc == 2) {
145 if ((server.daemonize = yesnotoi(argv[1])) == -1) {
146 err = "argument must be 'yes' or 'no'"; goto loaderr;
147 }
148 } else if (!strcasecmp(argv[0],"appendonly") && argc == 2) {
149 if ((server.appendonly = yesnotoi(argv[1])) == -1) {
150 err = "argument must be 'yes' or 'no'"; goto loaderr;
151 }
152 } else if (!strcasecmp(argv[0],"appendfilename") && argc == 2) {
153 zfree(server.appendfilename);
154 server.appendfilename = zstrdup(argv[1]);
155 } else if (!strcasecmp(argv[0],"no-appendfsync-on-rewrite")
156 && argc == 2) {
157 if ((server.no_appendfsync_on_rewrite= yesnotoi(argv[1])) == -1) {
158 err = "argument must be 'yes' or 'no'"; goto loaderr;
159 }
160 } else if (!strcasecmp(argv[0],"appendfsync") && argc == 2) {
161 if (!strcasecmp(argv[1],"no")) {
162 server.appendfsync = APPENDFSYNC_NO;
163 } else if (!strcasecmp(argv[1],"always")) {
164 server.appendfsync = APPENDFSYNC_ALWAYS;
165 } else if (!strcasecmp(argv[1],"everysec")) {
166 server.appendfsync = APPENDFSYNC_EVERYSEC;
167 } else {
168 err = "argument must be 'no', 'always' or 'everysec'";
169 goto loaderr;
170 }
171 } else if (!strcasecmp(argv[0],"requirepass") && argc == 2) {
172 server.requirepass = zstrdup(argv[1]);
173 } else if (!strcasecmp(argv[0],"pidfile") && argc == 2) {
174 zfree(server.pidfile);
175 server.pidfile = zstrdup(argv[1]);
176 } else if (!strcasecmp(argv[0],"dbfilename") && argc == 2) {
177 zfree(server.dbfilename);
178 server.dbfilename = zstrdup(argv[1]);
179 } else if (!strcasecmp(argv[0],"vm-enabled") && argc == 2) {
180 if ((server.vm_enabled = yesnotoi(argv[1])) == -1) {
181 err = "argument must be 'yes' or 'no'"; goto loaderr;
182 }
183 } else if (!strcasecmp(argv[0],"vm-swap-file") && argc == 2) {
184 zfree(server.vm_swap_file);
185 server.vm_swap_file = zstrdup(argv[1]);
186 } else if (!strcasecmp(argv[0],"vm-max-memory") && argc == 2) {
187 server.vm_max_memory = memtoll(argv[1],NULL);
188 } else if (!strcasecmp(argv[0],"vm-page-size") && argc == 2) {
189 server.vm_page_size = memtoll(argv[1], NULL);
190 } else if (!strcasecmp(argv[0],"vm-pages") && argc == 2) {
191 server.vm_pages = memtoll(argv[1], NULL);
192 } else if (!strcasecmp(argv[0],"vm-max-threads") && argc == 2) {
193 server.vm_max_threads = strtoll(argv[1], NULL, 10);
194 } else if (!strcasecmp(argv[0],"hash-max-zipmap-entries") && argc == 2){
195 server.hash_max_zipmap_entries = memtoll(argv[1], NULL);
196 } else if (!strcasecmp(argv[0],"hash-max-zipmap-value") && argc == 2){
197 server.hash_max_zipmap_value = memtoll(argv[1], NULL);
198 } else if (!strcasecmp(argv[0],"list-max-ziplist-entries") && argc == 2){
199 server.list_max_ziplist_entries = memtoll(argv[1], NULL);
200 } else if (!strcasecmp(argv[0],"list-max-ziplist-value") && argc == 2){
201 server.list_max_ziplist_value = memtoll(argv[1], NULL);
96ffb2fe
PN
202 } else if (!strcasecmp(argv[0],"set-max-intset-entries") && argc == 2){
203 server.set_max_intset_entries = memtoll(argv[1], NULL);
e2641e09 204 } else {
205 err = "Bad directive or wrong number of arguments"; goto loaderr;
206 }
207 for (j = 0; j < argc; j++)
208 sdsfree(argv[j]);
209 zfree(argv);
210 sdsfree(line);
211 }
212 if (fp != stdin) fclose(fp);
213 return;
214
215loaderr:
216 fprintf(stderr, "\n*** FATAL CONFIG FILE ERROR ***\n");
217 fprintf(stderr, "Reading the configuration file, at line %d\n", linenum);
218 fprintf(stderr, ">>> '%s'\n", line);
219 fprintf(stderr, "%s\n", err);
220 exit(1);
221}
222
223/*-----------------------------------------------------------------------------
224 * CONFIG command for remote configuration
225 *----------------------------------------------------------------------------*/
226
227void configSetCommand(redisClient *c) {
228 robj *o = getDecodedObject(c->argv[3]);
229 long long ll;
230
231 if (!strcasecmp(c->argv[2]->ptr,"dbfilename")) {
232 zfree(server.dbfilename);
233 server.dbfilename = zstrdup(o->ptr);
234 } else if (!strcasecmp(c->argv[2]->ptr,"requirepass")) {
235 zfree(server.requirepass);
236 server.requirepass = zstrdup(o->ptr);
237 } else if (!strcasecmp(c->argv[2]->ptr,"masterauth")) {
238 zfree(server.masterauth);
239 server.masterauth = zstrdup(o->ptr);
240 } else if (!strcasecmp(c->argv[2]->ptr,"maxmemory")) {
241 if (getLongLongFromObject(o,&ll) == REDIS_ERR ||
242 ll < 0) goto badfmt;
243 server.maxmemory = ll;
b4f2e412 244 if (server.maxmemory) freeMemoryIfNeeded();
e2641e09 245 } else if (!strcasecmp(c->argv[2]->ptr,"timeout")) {
246 if (getLongLongFromObject(o,&ll) == REDIS_ERR ||
247 ll < 0 || ll > LONG_MAX) goto badfmt;
248 server.maxidletime = ll;
249 } else if (!strcasecmp(c->argv[2]->ptr,"appendfsync")) {
250 if (!strcasecmp(o->ptr,"no")) {
251 server.appendfsync = APPENDFSYNC_NO;
252 } else if (!strcasecmp(o->ptr,"everysec")) {
253 server.appendfsync = APPENDFSYNC_EVERYSEC;
254 } else if (!strcasecmp(o->ptr,"always")) {
255 server.appendfsync = APPENDFSYNC_ALWAYS;
256 } else {
257 goto badfmt;
258 }
259 } else if (!strcasecmp(c->argv[2]->ptr,"no-appendfsync-on-rewrite")) {
260 int yn = yesnotoi(o->ptr);
261
262 if (yn == -1) goto badfmt;
263 server.no_appendfsync_on_rewrite = yn;
264 } else if (!strcasecmp(c->argv[2]->ptr,"appendonly")) {
265 int old = server.appendonly;
266 int new = yesnotoi(o->ptr);
267
268 if (new == -1) goto badfmt;
269 if (old != new) {
270 if (new == 0) {
271 stopAppendOnly();
272 } else {
273 if (startAppendOnly() == REDIS_ERR) {
3ab20376
PN
274 addReplyError(c,
275 "Unable to turn on AOF. Check server logs.");
e2641e09 276 decrRefCount(o);
277 return;
278 }
279 }
280 }
281 } else if (!strcasecmp(c->argv[2]->ptr,"save")) {
282 int vlen, j;
283 sds *v = sdssplitlen(o->ptr,sdslen(o->ptr)," ",1,&vlen);
284
285 /* Perform sanity check before setting the new config:
286 * - Even number of args
287 * - Seconds >= 1, changes >= 0 */
288 if (vlen & 1) {
289 sdsfreesplitres(v,vlen);
290 goto badfmt;
291 }
292 for (j = 0; j < vlen; j++) {
293 char *eptr;
294 long val;
295
296 val = strtoll(v[j], &eptr, 10);
297 if (eptr[0] != '\0' ||
298 ((j & 1) == 0 && val < 1) ||
299 ((j & 1) == 1 && val < 0)) {
300 sdsfreesplitres(v,vlen);
301 goto badfmt;
302 }
303 }
304 /* Finally set the new config */
305 resetServerSaveParams();
306 for (j = 0; j < vlen; j += 2) {
307 time_t seconds;
308 int changes;
309
310 seconds = strtoll(v[j],NULL,10);
311 changes = strtoll(v[j+1],NULL,10);
312 appendServerSaveParams(seconds, changes);
313 }
314 sdsfreesplitres(v,vlen);
315 } else {
3ab20376
PN
316 addReplyErrorFormat(c,"Unsupported CONFIG parameter: %s",
317 (char*)c->argv[2]->ptr);
e2641e09 318 decrRefCount(o);
319 return;
320 }
321 decrRefCount(o);
322 addReply(c,shared.ok);
323 return;
324
325badfmt: /* Bad format errors */
3ab20376 326 addReplyErrorFormat(c,"Invalid argument '%s' for CONFIG SET '%s'",
e2641e09 327 (char*)o->ptr,
3ab20376 328 (char*)c->argv[2]->ptr);
e2641e09 329 decrRefCount(o);
330}
331
332void configGetCommand(redisClient *c) {
333 robj *o = getDecodedObject(c->argv[2]);
b301c1fc 334 void *replylen = addDeferredMultiBulkLength(c);
e2641e09 335 char *pattern = o->ptr;
336 int matches = 0;
337
e2641e09 338 if (stringmatch(pattern,"dbfilename",0)) {
339 addReplyBulkCString(c,"dbfilename");
340 addReplyBulkCString(c,server.dbfilename);
341 matches++;
342 }
343 if (stringmatch(pattern,"requirepass",0)) {
344 addReplyBulkCString(c,"requirepass");
345 addReplyBulkCString(c,server.requirepass);
346 matches++;
347 }
348 if (stringmatch(pattern,"masterauth",0)) {
349 addReplyBulkCString(c,"masterauth");
350 addReplyBulkCString(c,server.masterauth);
351 matches++;
352 }
353 if (stringmatch(pattern,"maxmemory",0)) {
354 char buf[128];
355
356 ll2string(buf,128,server.maxmemory);
357 addReplyBulkCString(c,"maxmemory");
358 addReplyBulkCString(c,buf);
359 matches++;
360 }
361 if (stringmatch(pattern,"timeout",0)) {
362 char buf[128];
363
364 ll2string(buf,128,server.maxidletime);
365 addReplyBulkCString(c,"timeout");
366 addReplyBulkCString(c,buf);
367 matches++;
368 }
369 if (stringmatch(pattern,"appendonly",0)) {
370 addReplyBulkCString(c,"appendonly");
371 addReplyBulkCString(c,server.appendonly ? "yes" : "no");
372 matches++;
373 }
374 if (stringmatch(pattern,"no-appendfsync-on-rewrite",0)) {
375 addReplyBulkCString(c,"no-appendfsync-on-rewrite");
376 addReplyBulkCString(c,server.no_appendfsync_on_rewrite ? "yes" : "no");
377 matches++;
378 }
379 if (stringmatch(pattern,"appendfsync",0)) {
380 char *policy;
381
382 switch(server.appendfsync) {
383 case APPENDFSYNC_NO: policy = "no"; break;
384 case APPENDFSYNC_EVERYSEC: policy = "everysec"; break;
385 case APPENDFSYNC_ALWAYS: policy = "always"; break;
386 default: policy = "unknown"; break; /* too harmless to panic */
387 }
388 addReplyBulkCString(c,"appendfsync");
389 addReplyBulkCString(c,policy);
390 matches++;
391 }
392 if (stringmatch(pattern,"save",0)) {
393 sds buf = sdsempty();
394 int j;
395
396 for (j = 0; j < server.saveparamslen; j++) {
397 buf = sdscatprintf(buf,"%ld %d",
398 server.saveparams[j].seconds,
399 server.saveparams[j].changes);
400 if (j != server.saveparamslen-1)
401 buf = sdscatlen(buf," ",1);
402 }
403 addReplyBulkCString(c,"save");
404 addReplyBulkCString(c,buf);
405 sdsfree(buf);
406 matches++;
407 }
408 decrRefCount(o);
b301c1fc 409 setDeferredMultiBulkLength(c,replylen,matches*2);
e2641e09 410}
411
412void configCommand(redisClient *c) {
413 if (!strcasecmp(c->argv[1]->ptr,"set")) {
414 if (c->argc != 4) goto badarity;
415 configSetCommand(c);
416 } else if (!strcasecmp(c->argv[1]->ptr,"get")) {
417 if (c->argc != 3) goto badarity;
418 configGetCommand(c);
419 } else if (!strcasecmp(c->argv[1]->ptr,"resetstat")) {
420 if (c->argc != 2) goto badarity;
421 server.stat_numcommands = 0;
422 server.stat_numconnections = 0;
423 server.stat_expiredkeys = 0;
424 server.stat_starttime = time(NULL);
425 addReply(c,shared.ok);
426 } else {
3ab20376
PN
427 addReplyError(c,
428 "CONFIG subcommand must be one of GET, SET, RESETSTAT");
e2641e09 429 }
430 return;
431
432badarity:
3ab20376
PN
433 addReplyErrorFormat(c,"Wrong number of arguments for CONFIG %s",
434 (char*) c->argv[1]->ptr);
e2641e09 435}