]>
Commit | Line | Data |
---|---|---|
e2641e09 | 1 | #include "redis.h" |
2 | ||
3 | /*----------------------------------------------------------------------------- | |
4 | * Config file parsing | |
5 | *----------------------------------------------------------------------------*/ | |
6 | ||
7 | int yesnotoi(char *s) { | |
8 | if (!strcasecmp(s,"yes")) return 1; | |
9 | else if (!strcasecmp(s,"no")) return 0; | |
10 | else return -1; | |
11 | } | |
12 | ||
13 | void 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 | ||
20 | void 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 */ | |
28 | void 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 | ||
215 | loaderr: | |
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 | ||
227 | void 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 | ||
325 | badfmt: /* 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 | ||
332 | void 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 | ||
412 | void 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 | ||
432 | badarity: | |
3ab20376 PN |
433 | addReplyErrorFormat(c,"Wrong number of arguments for CONFIG %s", |
434 | (char*) c->argv[1]->ptr); | |
e2641e09 | 435 | } |