[redis.git] / src / config.c

#include "redis.h"

/*-----------------------------------------------------------------------------
 * Config file parsing
 *----------------------------------------------------------------------------*/

int yesnotoi(char *s) {
    if (!strcasecmp(s,"yes")) return 1;
    else if (!strcasecmp(s,"no")) return 0;
    else return -1;
}

void appendServerSaveParams(time_t seconds, int changes) {
    server.saveparams = zrealloc(server.saveparams,sizeof(struct saveparam)*(server.saveparamslen+1));
    server.saveparams[server.saveparamslen].seconds = seconds;
    server.saveparams[server.saveparamslen].changes = changes;
    server.saveparamslen++;
}

void resetServerSaveParams() {
    zfree(server.saveparams);
    server.saveparams = NULL;
    server.saveparamslen = 0;
}

/* I agree, this is a very rudimental way to load a configuration...
   will improve later if the config gets more complex */
void loadServerConfig(char *filename) {
    FILE *fp;
    char buf[REDIS_CONFIGLINE_MAX+1], *err = NULL;
    int linenum = 0;
    sds line = NULL;

    if (filename[0] == '-' && filename[1] == '\0')
        fp = stdin;
    else {
        if ((fp = fopen(filename,"r")) == NULL) {
            redisLog(REDIS_WARNING, "Fatal error, can't open config file '%s'", filename);
            exit(1);
        }
    }

    while(fgets(buf,REDIS_CONFIGLINE_MAX+1,fp) != NULL) {
        sds *argv;
        int argc, j;

        linenum++;
        line = sdsnew(buf);
        line = sdstrim(line," \t\r\n");

        /* Skip comments and blank lines*/
        if (line[0] == '#' || line[0] == '\0') {
            sdsfree(line);
            continue;
        }

        /* Split into arguments */
        argv = sdssplitlen(line,sdslen(line)," ",1,&argc);
        sdstolower(argv[0]);

        /* Execute config directives */
        if (!strcasecmp(argv[0],"timeout") && argc == 2) {
            server.maxidletime = atoi(argv[1]);
            if (server.maxidletime < 0) {
                err = "Invalid timeout value"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"port") && argc == 2) {
            server.port = atoi(argv[1]);
            if (server.port < 1 || server.port > 65535) {
                err = "Invalid port"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"bind") && argc == 2) {
            server.bindaddr = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"save") && argc == 3) {
            int seconds = atoi(argv[1]);
            int changes = atoi(argv[2]);
            if (seconds < 1 || changes < 0) {
                err = "Invalid save parameters"; goto loaderr;
            }
            appendServerSaveParams(seconds,changes);
        } else if (!strcasecmp(argv[0],"dir") && argc == 2) {
            if (chdir(argv[1]) == -1) {
                redisLog(REDIS_WARNING,"Can't chdir to '%s': %s",
                    argv[1], strerror(errno));
                exit(1);
            }
        } else if (!strcasecmp(argv[0],"loglevel") && argc == 2) {
            if (!strcasecmp(argv[1],"debug")) server.verbosity = REDIS_DEBUG;
            else if (!strcasecmp(argv[1],"verbose")) server.verbosity = REDIS_VERBOSE;
            else if (!strcasecmp(argv[1],"notice")) server.verbosity = REDIS_NOTICE;
            else if (!strcasecmp(argv[1],"warning")) server.verbosity = REDIS_WARNING;
            else {
                err = "Invalid log level. Must be one of debug, notice, warning";
                goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"logfile") && argc == 2) {
            FILE *logfp;

            server.logfile = zstrdup(argv[1]);
            if (!strcasecmp(server.logfile,"stdout")) {
                zfree(server.logfile);
                server.logfile = NULL;
            }
            if (server.logfile) {
                /* Test if we are able to open the file. The server will not
                 * be able to abort just for this problem later... */
                logfp = fopen(server.logfile,"a");
                if (logfp == NULL) {
                    err = sdscatprintf(sdsempty(),
                        "Can't open the log file: %s", strerror(errno));
                    goto loaderr;
                }
                fclose(logfp);
            }
        } else if (!strcasecmp(argv[0],"databases") && argc == 2) {
            server.dbnum = atoi(argv[1]);
            if (server.dbnum < 1) {
                err = "Invalid number of databases"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"include") && argc == 2) {
            loadServerConfig(argv[1]);
        } else if (!strcasecmp(argv[0],"maxclients") && argc == 2) {
            server.maxclients = atoi(argv[1]);
        } else if (!strcasecmp(argv[0],"maxmemory") && argc == 2) {
            server.maxmemory = memtoll(argv[1],NULL);
        } else if (!strcasecmp(argv[0],"slaveof") && argc == 3) {
            server.masterhost = sdsnew(argv[1]);
            server.masterport = atoi(argv[2]);
            server.replstate = REDIS_REPL_CONNECT;
        } else if (!strcasecmp(argv[0],"masterauth") && argc == 2) {
        	server.masterauth = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"glueoutputbuf") && argc == 2) {
            if ((server.glueoutputbuf = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"rdbcompression") && argc == 2) {
            if ((server.rdbcompression = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"activerehashing") && argc == 2) {
            if ((server.activerehashing = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"daemonize") && argc == 2) {
            if ((server.daemonize = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"appendonly") && argc == 2) {
            if ((server.appendonly = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"appendfilename") && argc == 2) {
            zfree(server.appendfilename);
            server.appendfilename = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"no-appendfsync-on-rewrite")
                   && argc == 2) {
            if ((server.no_appendfsync_on_rewrite= yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"appendfsync") && argc == 2) {
            if (!strcasecmp(argv[1],"no")) {
                server.appendfsync = APPENDFSYNC_NO;
            } else if (!strcasecmp(argv[1],"always")) {
                server.appendfsync = APPENDFSYNC_ALWAYS;
            } else if (!strcasecmp(argv[1],"everysec")) {
                server.appendfsync = APPENDFSYNC_EVERYSEC;
            } else {
                err = "argument must be 'no', 'always' or 'everysec'";
                goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"requirepass") && argc == 2) {
            server.requirepass = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"pidfile") && argc == 2) {
            zfree(server.pidfile);
            server.pidfile = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"dbfilename") && argc == 2) {
            zfree(server.dbfilename);
            server.dbfilename = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"vm-enabled") && argc == 2) {
            if ((server.vm_enabled = yesnotoi(argv[1])) == -1) {
                err = "argument must be 'yes' or 'no'"; goto loaderr;
            }
        } else if (!strcasecmp(argv[0],"vm-swap-file") && argc == 2) {
            zfree(server.vm_swap_file);
            server.vm_swap_file = zstrdup(argv[1]);
        } else if (!strcasecmp(argv[0],"vm-max-memory") && argc == 2) {
            server.vm_max_memory = memtoll(argv[1],NULL);
        } else if (!strcasecmp(argv[0],"vm-page-size") && argc == 2) {
            server.vm_page_size = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"vm-pages") && argc == 2) {
            server.vm_pages = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"vm-max-threads") && argc == 2) {
            server.vm_max_threads = strtoll(argv[1], NULL, 10);
        } else if (!strcasecmp(argv[0],"hash-max-zipmap-entries") && argc == 2){
            server.hash_max_zipmap_entries = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"hash-max-zipmap-value") && argc == 2){
            server.hash_max_zipmap_value = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"list-max-ziplist-entries") && argc == 2){
            server.list_max_ziplist_entries = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"list-max-ziplist-value") && argc == 2){
            server.list_max_ziplist_value = memtoll(argv[1], NULL);
        } else if (!strcasecmp(argv[0],"set-max-intset-entries") && argc == 2){
            server.set_max_intset_entries = memtoll(argv[1], NULL);
        } else {
            err = "Bad directive or wrong number of arguments"; goto loaderr;
        }
        for (j = 0; j < argc; j++)
            sdsfree(argv[j]);
        zfree(argv);
        sdsfree(line);
    }
    if (fp != stdin) fclose(fp);
    return;

loaderr:
    fprintf(stderr, "\n*** FATAL CONFIG FILE ERROR ***\n");
    fprintf(stderr, "Reading the configuration file, at line %d\n", linenum);
    fprintf(stderr, ">>> '%s'\n", line);
    fprintf(stderr, "%s\n", err);
    exit(1);
}

/*-----------------------------------------------------------------------------
 * CONFIG command for remote configuration
 *----------------------------------------------------------------------------*/

void configSetCommand(redisClient *c) {
    robj *o = getDecodedObject(c->argv[3]);
    long long ll;

    if (!strcasecmp(c->argv[2]->ptr,"dbfilename")) {
        zfree(server.dbfilename);
        server.dbfilename = zstrdup(o->ptr);
    } else if (!strcasecmp(c->argv[2]->ptr,"requirepass")) {
        zfree(server.requirepass);
        server.requirepass = zstrdup(o->ptr);
    } else if (!strcasecmp(c->argv[2]->ptr,"masterauth")) {
        zfree(server.masterauth);
        server.masterauth = zstrdup(o->ptr);
    } else if (!strcasecmp(c->argv[2]->ptr,"maxmemory")) {
        if (getLongLongFromObject(o,&ll) == REDIS_ERR ||
            ll < 0) goto badfmt;
        server.maxmemory = ll;
    } else if (!strcasecmp(c->argv[2]->ptr,"timeout")) {
        if (getLongLongFromObject(o,&ll) == REDIS_ERR ||
            ll < 0 || ll > LONG_MAX) goto badfmt;
        server.maxidletime = ll;
    } else if (!strcasecmp(c->argv[2]->ptr,"appendfsync")) {
        if (!strcasecmp(o->ptr,"no")) {
            server.appendfsync = APPENDFSYNC_NO;
        } else if (!strcasecmp(o->ptr,"everysec")) {
            server.appendfsync = APPENDFSYNC_EVERYSEC;
        } else if (!strcasecmp(o->ptr,"always")) {
            server.appendfsync = APPENDFSYNC_ALWAYS;
        } else {
            goto badfmt;
        }
    } else if (!strcasecmp(c->argv[2]->ptr,"no-appendfsync-on-rewrite")) {
        int yn = yesnotoi(o->ptr);

        if (yn == -1) goto badfmt;
        server.no_appendfsync_on_rewrite = yn;
    } else if (!strcasecmp(c->argv[2]->ptr,"appendonly")) {
        int old = server.appendonly;
        int new = yesnotoi(o->ptr);

        if (new == -1) goto badfmt;
        if (old != new) {
            if (new == 0) {
                stopAppendOnly();
            } else {
                if (startAppendOnly() == REDIS_ERR) {
                    addReplySds(c,sdscatprintf(sdsempty(),
                        "-ERR Unable to turn on AOF. Check server logs.\r\n"));
                    decrRefCount(o);
                    return;
                }
            }
        }
    } else if (!strcasecmp(c->argv[2]->ptr,"save")) {
        int vlen, j;
        sds *v = sdssplitlen(o->ptr,sdslen(o->ptr)," ",1,&vlen);

        /* Perform sanity check before setting the new config:
         * - Even number of args
         * - Seconds >= 1, changes >= 0 */
        if (vlen & 1) {
            sdsfreesplitres(v,vlen);
            goto badfmt;
        }
        for (j = 0; j < vlen; j++) {
            char *eptr;
            long val;

            val = strtoll(v[j], &eptr, 10);
            if (eptr[0] != '\0' ||
                ((j & 1) == 0 && val < 1) ||
                ((j & 1) == 1 && val < 0)) {
                sdsfreesplitres(v,vlen);
                goto badfmt;
            }
        }
        /* Finally set the new config */
        resetServerSaveParams();
        for (j = 0; j < vlen; j += 2) {
            time_t seconds;
            int changes;

            seconds = strtoll(v[j],NULL,10);
            changes = strtoll(v[j+1],NULL,10);
            appendServerSaveParams(seconds, changes);
        }
        sdsfreesplitres(v,vlen);
    } else {
        addReplySds(c,sdscatprintf(sdsempty(),
            "-ERR not supported CONFIG parameter %s\r\n",
            (char*)c->argv[2]->ptr));
        decrRefCount(o);
        return;
    }
    decrRefCount(o);
    addReply(c,shared.ok);
    return;

badfmt: /* Bad format errors */
    addReplySds(c,sdscatprintf(sdsempty(),
        "-ERR invalid argument '%s' for CONFIG SET '%s'\r\n",
            (char*)o->ptr,
            (char*)c->argv[2]->ptr));
    decrRefCount(o);
}

void configGetCommand(redisClient *c) {
    robj *o = getDecodedObject(c->argv[2]);
    void *replylen = addDeferredMultiBulkLength(c);
    char *pattern = o->ptr;
    int matches = 0;

    if (stringmatch(pattern,"dbfilename",0)) {
        addReplyBulkCString(c,"dbfilename");
        addReplyBulkCString(c,server.dbfilename);
        matches++;
    }
    if (stringmatch(pattern,"requirepass",0)) {
        addReplyBulkCString(c,"requirepass");
        addReplyBulkCString(c,server.requirepass);
        matches++;
    }
    if (stringmatch(pattern,"masterauth",0)) {
        addReplyBulkCString(c,"masterauth");
        addReplyBulkCString(c,server.masterauth);
        matches++;
    }
    if (stringmatch(pattern,"maxmemory",0)) {
        char buf[128];

        ll2string(buf,128,server.maxmemory);
        addReplyBulkCString(c,"maxmemory");
        addReplyBulkCString(c,buf);
        matches++;
    }
    if (stringmatch(pattern,"timeout",0)) {
        char buf[128];

        ll2string(buf,128,server.maxidletime);
        addReplyBulkCString(c,"timeout");
        addReplyBulkCString(c,buf);
        matches++;
    }
    if (stringmatch(pattern,"appendonly",0)) {
        addReplyBulkCString(c,"appendonly");
        addReplyBulkCString(c,server.appendonly ? "yes" : "no");
        matches++;
    }
    if (stringmatch(pattern,"no-appendfsync-on-rewrite",0)) {
        addReplyBulkCString(c,"no-appendfsync-on-rewrite");
        addReplyBulkCString(c,server.no_appendfsync_on_rewrite ? "yes" : "no");
        matches++;
    }
    if (stringmatch(pattern,"appendfsync",0)) {
        char *policy;

        switch(server.appendfsync) {
        case APPENDFSYNC_NO: policy = "no"; break;
        case APPENDFSYNC_EVERYSEC: policy = "everysec"; break;
        case APPENDFSYNC_ALWAYS: policy = "always"; break;
        default: policy = "unknown"; break; /* too harmless to panic */
        }
        addReplyBulkCString(c,"appendfsync");
        addReplyBulkCString(c,policy);
        matches++;
    }
    if (stringmatch(pattern,"save",0)) {
        sds buf = sdsempty();
        int j;

        for (j = 0; j < server.saveparamslen; j++) {
            buf = sdscatprintf(buf,"%ld %d",
                    server.saveparams[j].seconds,
                    server.saveparams[j].changes);
            if (j != server.saveparamslen-1)
                buf = sdscatlen(buf," ",1);
        }
        addReplyBulkCString(c,"save");
        addReplyBulkCString(c,buf);
        sdsfree(buf);
        matches++;
    }
    decrRefCount(o);
    setDeferredMultiBulkLength(c,replylen,matches*2);
}

void configCommand(redisClient *c) {
    if (!strcasecmp(c->argv[1]->ptr,"set")) {
        if (c->argc != 4) goto badarity;
        configSetCommand(c);
    } else if (!strcasecmp(c->argv[1]->ptr,"get")) {
        if (c->argc != 3) goto badarity;
        configGetCommand(c);
    } else if (!strcasecmp(c->argv[1]->ptr,"resetstat")) {
        if (c->argc != 2) goto badarity;
        server.stat_numcommands = 0;
        server.stat_numconnections = 0;
        server.stat_expiredkeys = 0;
        server.stat_starttime = time(NULL);
        addReply(c,shared.ok);
    } else {
        addReplySds(c,sdscatprintf(sdsempty(),
            "-ERR CONFIG subcommand must be one of GET, SET, RESETSTAT\r\n"));
    }
    return;

badarity:
    addReplySds(c,sdscatprintf(sdsempty(),
        "-ERR Wrong number of arguments for CONFIG %s\r\n",
        (char*) c->argv[1]->ptr));
}
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){
96ffb2fe PN	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;
	244	} else if (!strcasecmp(c->argv[2]->ptr,"timeout")) {
	245	if (getLongLongFromObject(o,&ll) == REDIS_ERR \|\|
	246	ll < 0 \|\| ll > LONG_MAX) goto badfmt;
	247	server.maxidletime = ll;
	248	} else if (!strcasecmp(c->argv[2]->ptr,"appendfsync")) {
	249	if (!strcasecmp(o->ptr,"no")) {
	250	server.appendfsync = APPENDFSYNC_NO;
	251	} else if (!strcasecmp(o->ptr,"everysec")) {
	252	server.appendfsync = APPENDFSYNC_EVERYSEC;
	253	} else if (!strcasecmp(o->ptr,"always")) {
	254	server.appendfsync = APPENDFSYNC_ALWAYS;
	255	} else {
	256	goto badfmt;
	257	}
	258	} else if (!strcasecmp(c->argv[2]->ptr,"no-appendfsync-on-rewrite")) {
	259	int yn = yesnotoi(o->ptr);
	260
	261	if (yn == -1) goto badfmt;
	262	server.no_appendfsync_on_rewrite = yn;
	263	} else if (!strcasecmp(c->argv[2]->ptr,"appendonly")) {
	264	int old = server.appendonly;
	265	int new = yesnotoi(o->ptr);
	266
	267	if (new == -1) goto badfmt;
268	if (old != new) {
269	if (new == 0) {
270	stopAppendOnly();
271	} else {
272	if (startAppendOnly() == REDIS_ERR) {
273	addReplySds(c,sdscatprintf(sdsempty(),
274	"-ERR Unable to turn on AOF. Check server logs.\r\n"));
275	decrRefCount(o);
276	return;
277	}
278	}
279	}
280	} else if (!strcasecmp(c->argv[2]->ptr,"save")) {
281	int vlen, j;
282	sds *v = sdssplitlen(o->ptr,sdslen(o->ptr)," ",1,&vlen);
283
284	/* Perform sanity check before setting the new config:
285	* - Even number of args
286	* - Seconds >= 1, changes >= 0 */
287	if (vlen & 1) {
288	sdsfreesplitres(v,vlen);
289	goto badfmt;
290	}
291	for (j = 0; j < vlen; j++) {
292	char *eptr;
293	long val;
294
295	val = strtoll(v[j], &eptr, 10);
296	if (eptr[0] != '\0' \|\|
297	((j & 1) == 0 && val < 1) \|\|
298	((j & 1) == 1 && val < 0)) {
299	sdsfreesplitres(v,vlen);
300	goto badfmt;
301	}
302	}
303	/* Finally set the new config */
304	resetServerSaveParams();
305	for (j = 0; j < vlen; j += 2) {
306	time_t seconds;
307	int changes;
308
309	seconds = strtoll(v[j],NULL,10);
310	changes = strtoll(v[j+1],NULL,10);
311	appendServerSaveParams(seconds, changes);
312	}
313	sdsfreesplitres(v,vlen);
314	} else {
315	addReplySds(c,sdscatprintf(sdsempty(),
316	"-ERR not supported CONFIG parameter %s\r\n",
317	(char*)c->argv[2]->ptr));
318	decrRefCount(o);
319	return;
320	}
321	decrRefCount(o);
322	addReply(c,shared.ok);
323	return;
324
325	badfmt: /* Bad format errors */
326	addReplySds(c,sdscatprintf(sdsempty(),
327	"-ERR invalid argument '%s' for CONFIG SET '%s'\r\n",
328	(char*)o->ptr,
329	(char*)c->argv[2]->ptr));
330	decrRefCount(o);
331	}
332
333	void configGetCommand(redisClient *c) {
334	robj *o = getDecodedObject(c->argv[2]);
b301c1fc	335	void *replylen = addDeferredMultiBulkLength(c);
e2641e09	336	char *pattern = o->ptr;
	337	int matches = 0;
	338
e2641e09	339	if (stringmatch(pattern,"dbfilename",0)) {
	340	addReplyBulkCString(c,"dbfilename");
	341	addReplyBulkCString(c,server.dbfilename);
	342	matches++;
	343	}
	344	if (stringmatch(pattern,"requirepass",0)) {
	345	addReplyBulkCString(c,"requirepass");
	346	addReplyBulkCString(c,server.requirepass);
	347	matches++;
	348	}
	349	if (stringmatch(pattern,"masterauth",0)) {
	350	addReplyBulkCString(c,"masterauth");
	351	addReplyBulkCString(c,server.masterauth);
	352	matches++;
	353	}
	354	if (stringmatch(pattern,"maxmemory",0)) {
	355	char buf[128];
	356
	357	ll2string(buf,128,server.maxmemory);
	358	addReplyBulkCString(c,"maxmemory");
	359	addReplyBulkCString(c,buf);
	360	matches++;
	361	}
	362	if (stringmatch(pattern,"timeout",0)) {
	363	char buf[128];
	364
	365	ll2string(buf,128,server.maxidletime);
	366	addReplyBulkCString(c,"timeout");
	367	addReplyBulkCString(c,buf);
	368	matches++;
	369	}
	370	if (stringmatch(pattern,"appendonly",0)) {
	371	addReplyBulkCString(c,"appendonly");
	372	addReplyBulkCString(c,server.appendonly ? "yes" : "no");
	373	matches++;
	374	}
	375	if (stringmatch(pattern,"no-appendfsync-on-rewrite",0)) {
	376	addReplyBulkCString(c,"no-appendfsync-on-rewrite");
	377	addReplyBulkCString(c,server.no_appendfsync_on_rewrite ? "yes" : "no");
	378	matches++;
	379	}
	380	if (stringmatch(pattern,"appendfsync",0)) {
	381	char *policy;
	382
	383	switch(server.appendfsync) {
	384	case APPENDFSYNC_NO: policy = "no"; break;
	385	case APPENDFSYNC_EVERYSEC: policy = "everysec"; break;
	386	case APPENDFSYNC_ALWAYS: policy = "always"; break;
	387	default: policy = "unknown"; break; /* too harmless to panic */
	388	}
	389	addReplyBulkCString(c,"appendfsync");
	390	addReplyBulkCString(c,policy);
	391	matches++;
	392	}
	393	if (stringmatch(pattern,"save",0)) {
	394	sds buf = sdsempty();
	395	int j;
	396
	397	for (j = 0; j < server.saveparamslen; j++) {
	398	buf = sdscatprintf(buf,"%ld %d",
	399	server.saveparams[j].seconds,
	400	server.saveparams[j].changes);
	401	if (j != server.saveparamslen-1)
	402	buf = sdscatlen(buf," ",1);
403	}
404	addReplyBulkCString(c,"save");
405	addReplyBulkCString(c,buf);
406	sdsfree(buf);
407	matches++;
408	}
409	decrRefCount(o);
b301c1fc	410	setDeferredMultiBulkLength(c,replylen,matches*2);
e2641e09	411	}
	412
	413	void configCommand(redisClient *c) {
	414	if (!strcasecmp(c->argv[1]->ptr,"set")) {
	415	if (c->argc != 4) goto badarity;
	416	configSetCommand(c);
	417	} else if (!strcasecmp(c->argv[1]->ptr,"get")) {
	418	if (c->argc != 3) goto badarity;
	419	configGetCommand(c);
	420	} else if (!strcasecmp(c->argv[1]->ptr,"resetstat")) {
	421	if (c->argc != 2) goto badarity;
	422	server.stat_numcommands = 0;
	423	server.stat_numconnections = 0;
	424	server.stat_expiredkeys = 0;
	425	server.stat_starttime = time(NULL);
	426	addReply(c,shared.ok);
	427	} else {
	428	addReplySds(c,sdscatprintf(sdsempty(),
	429	"-ERR CONFIG subcommand must be one of GET, SET, RESETSTAT\r\n"));
	430	}
	431	return;
	432
	433	badarity:
	434	addReplySds(c,sdscatprintf(sdsempty(),
	435	"-ERR Wrong number of arguments for CONFIG %s\r\n",
	436	(char*) c->argv[1]->ptr));
	437	}