mirror of https://github.com/yandex/odyssey.git
435 lines
9.1 KiB
C
435 lines
9.1 KiB
C
|
|
/*
|
|
* odissey.
|
|
*
|
|
* PostgreSQL connection pooler and request router.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include "od_macro.h"
|
|
#include "od_list.h"
|
|
#include "od_log.h"
|
|
#include "od_scheme.h"
|
|
#include "od_lex.h"
|
|
#include "od_config.h"
|
|
|
|
#define od_keyword(name, token) { name, sizeof(name) - 1, token }
|
|
|
|
static odkeyword_t od_config_keywords[] =
|
|
{
|
|
/* main */
|
|
od_keyword("odissey", OD_LODISSEY),
|
|
od_keyword("yes", OD_LYES),
|
|
od_keyword("no", OD_LNO),
|
|
od_keyword("on", OD_LON),
|
|
od_keyword("off", OD_LOFF),
|
|
od_keyword("daemonize", OD_LDAEMONIZE),
|
|
od_keyword("log_file", OD_LLOG_FILE),
|
|
od_keyword("pid_file", OD_LPID_FILE),
|
|
od_keyword("pooling", OD_LPOOLING),
|
|
/* listen */
|
|
od_keyword("listen", OD_LLISTEN),
|
|
od_keyword("host", OD_LHOST),
|
|
od_keyword("port", OD_LPORT),
|
|
od_keyword("workers", OD_LWORKERS),
|
|
od_keyword("client_max", OD_LCLIENT_MAX),
|
|
/* server */
|
|
od_keyword("server", OD_LSERVER),
|
|
/* routing */
|
|
od_keyword("routing", OD_LROUTING),
|
|
od_keyword("route", OD_LROUTE),
|
|
od_keyword("mode", OD_LMODE),
|
|
od_keyword("user", OD_LUSER),
|
|
od_keyword("password", OD_LPASSWORD),
|
|
od_keyword("pool_min", OD_LPOOL_MIN),
|
|
od_keyword("pool_max", OD_LPOOL_MAX),
|
|
{ NULL, 0, 0 }
|
|
};
|
|
|
|
void
|
|
od_configinit(odconfig_t *config,
|
|
odlog_t *log,
|
|
odscheme_t *scheme)
|
|
{
|
|
od_lexinit(&config->lex);
|
|
config->log = log;
|
|
config->scheme = scheme;
|
|
}
|
|
|
|
int
|
|
od_configopen(odconfig_t *config, char *file)
|
|
{
|
|
/* read file */
|
|
struct stat st;
|
|
int rc = lstat(file, &st);
|
|
if (rc == -1) {
|
|
od_error(config->log, "failed to open config file '%s'",
|
|
file);
|
|
return -1;
|
|
}
|
|
char *config_buf = malloc(st.st_size);
|
|
if (config_buf == NULL) {
|
|
od_error(config->log, "memory allocation error");
|
|
return -1;
|
|
}
|
|
FILE *f = fopen(file, "r");
|
|
if (f == NULL) {
|
|
free(config_buf);
|
|
od_error(config->log, "failed to open config file '%s'",
|
|
file);
|
|
return -1;
|
|
}
|
|
rc = fread(config_buf, st.st_size, 1, f);
|
|
fclose(f);
|
|
if (rc != 1) {
|
|
free(config_buf);
|
|
od_error(config->log, "failed to open config file '%s'",
|
|
file);
|
|
return -1;
|
|
}
|
|
od_lexopen(&config->lex, od_config_keywords, config_buf,
|
|
st.st_size);
|
|
config->scheme->config_file = file;
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
od_configclose(odconfig_t *config)
|
|
{
|
|
od_lexfree(&config->lex);
|
|
}
|
|
|
|
static void
|
|
od_configerror(odconfig_t *config, odtoken_t *tk, char *fmt, ...)
|
|
{
|
|
char msg[256];
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vsnprintf(msg, sizeof(msg), fmt, args);
|
|
va_end(args);
|
|
int line = config->lex.line;
|
|
if (tk)
|
|
line = tk->line;
|
|
od_error(config->log, "%s:%d %s", config->scheme->config_file,
|
|
line, msg);
|
|
}
|
|
|
|
static int
|
|
od_confignext(odconfig_t *config, int id, odtoken_t **tk)
|
|
{
|
|
odtoken_t *tkp = NULL;
|
|
int token = od_lexpop(&config->lex, &tkp);
|
|
if (token == OD_LERROR) {
|
|
od_configerror(config, NULL, "%s", config->lex.error);
|
|
return -1;
|
|
}
|
|
if (tk) {
|
|
*tk = tkp;
|
|
}
|
|
if (token != id) {
|
|
if (id < 0xff && ispunct(id)) {
|
|
od_configerror(config, tkp, "expected '%c'", id);
|
|
return -1;
|
|
}
|
|
od_configerror(config, tkp, "expected '%s'",
|
|
od_lexname_of(&config->lex, id));
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
od_confignext_yes_no(odconfig_t *config, odtoken_t **tk)
|
|
{
|
|
int rc;
|
|
rc = od_lexpop(&config->lex, tk);
|
|
if (rc == OD_LYES)
|
|
return 1;
|
|
if (rc == OD_LNO)
|
|
return 0;
|
|
od_configerror(config, *tk, "expected yes/no");
|
|
return -1;
|
|
}
|
|
|
|
static int
|
|
od_configparse_listen(odconfig_t *config)
|
|
{
|
|
if (od_confignext(config, '{', NULL) == -1)
|
|
return -1;
|
|
odtoken_t *tk;
|
|
int rc;
|
|
int eof = 0;
|
|
while (! eof)
|
|
{
|
|
rc = od_lexpop(&config->lex, &tk);
|
|
switch (rc) {
|
|
/* host */
|
|
case OD_LHOST:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
config->scheme->host = tk->v.string;
|
|
continue;
|
|
/* port */
|
|
case OD_LPORT:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
config->scheme->port = tk->v.num;
|
|
continue;
|
|
/* client_max */
|
|
case OD_LCLIENT_MAX:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
config->scheme->client_max = tk->v.num;
|
|
continue;
|
|
/* workers */
|
|
case OD_LWORKERS:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
config->scheme->workers = tk->v.num;
|
|
continue;
|
|
case OD_LEOF:
|
|
od_configerror(config, tk, "unexpected end of config file");
|
|
return -1;
|
|
case '}':
|
|
eof = 1;
|
|
continue;
|
|
default:
|
|
od_configerror(config, tk, "unknown option");
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
od_configparse_server(odconfig_t *config)
|
|
{
|
|
odscheme_server_t *server =
|
|
od_schemeserver_add(config->scheme);
|
|
if (server == NULL)
|
|
return -1;
|
|
odtoken_t *tk;
|
|
int rc;
|
|
/* name */
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
server->name = tk->v.string;
|
|
if (od_confignext(config, '{', NULL) == -1)
|
|
return -1;
|
|
int eof = 0;
|
|
while (! eof)
|
|
{
|
|
rc = od_lexpop(&config->lex, &tk);
|
|
switch (rc) {
|
|
/* host */
|
|
case OD_LHOST:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
server->host = tk->v.string;
|
|
continue;
|
|
/* port */
|
|
case OD_LPORT:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
server->port = tk->v.num;
|
|
continue;
|
|
case OD_LEOF:
|
|
od_configerror(config, tk, "unexpected end of config file");
|
|
return -1;
|
|
case '}':
|
|
eof = 1;
|
|
continue;
|
|
default:
|
|
od_configerror(config, tk, "unknown option");
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
od_configparse_route(odconfig_t *config, odtoken_t *name)
|
|
{
|
|
odscheme_route_t *route =
|
|
od_schemeroute_add(config->scheme);
|
|
if (route == NULL)
|
|
return -1;
|
|
route->database = name->v.string;
|
|
if (od_confignext(config, '{', NULL) == -1)
|
|
return -1;
|
|
odtoken_t *tk;
|
|
int rc;
|
|
int eof = 0;
|
|
while (! eof)
|
|
{
|
|
rc = od_lexpop(&config->lex, &tk);
|
|
switch (rc) {
|
|
/* route */
|
|
case OD_LROUTE:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
route->route = tk->v.string;
|
|
continue;
|
|
/* client_max */
|
|
case OD_LCLIENT_MAX:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
route->client_max = tk->v.num;
|
|
continue;
|
|
/* pool_min */
|
|
case OD_LPOOL_MIN:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
route->pool_min = tk->v.num;
|
|
continue;
|
|
/* pool_max */
|
|
case OD_LPOOL_MAX:
|
|
if (od_confignext(config, OD_LNUMBER, &tk) == -1)
|
|
return -1;
|
|
route->pool_max = tk->v.num;
|
|
continue;
|
|
/* user */
|
|
case OD_LUSER:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
route->user = tk->v.string;
|
|
continue;
|
|
/* password */
|
|
case OD_LPASSWORD:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
route->password = tk->v.string;
|
|
continue;
|
|
case OD_LEOF:
|
|
od_configerror(config, tk, "unexpected end of config file");
|
|
return -1;
|
|
case '}':
|
|
eof = 1;
|
|
continue;
|
|
default:
|
|
od_configerror(config, tk, "unknown option");
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
od_configparse_routing(odconfig_t *config)
|
|
{
|
|
if (od_confignext(config, '{', NULL) == -1)
|
|
return -1;
|
|
odtoken_t *tk;
|
|
int rc;
|
|
int eof = 0;
|
|
while (! eof)
|
|
{
|
|
rc = od_lexpop(&config->lex, &tk);
|
|
switch (rc) {
|
|
/* mode */
|
|
case OD_LMODE:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
config->scheme->routing = tk->v.string;
|
|
continue;
|
|
/* route (database name) */
|
|
case OD_LSTRING:
|
|
rc = od_configparse_route(config, tk);
|
|
if (rc == -1)
|
|
return -1;
|
|
break;
|
|
case OD_LEOF:
|
|
od_configerror(config, tk, "unexpected end of config file");
|
|
return -1;
|
|
case '}':
|
|
eof = 1;
|
|
continue;
|
|
default:
|
|
od_configerror(config, tk, "unknown option");
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
od_configparse(odconfig_t *config)
|
|
{
|
|
odtoken_t *tk;
|
|
if (od_confignext(config, OD_LODISSEY, NULL) == -1)
|
|
return -1;
|
|
if (od_confignext(config, '{', NULL) == -1)
|
|
return -1;
|
|
int rc;
|
|
int eof = 0;
|
|
while (! eof)
|
|
{
|
|
rc = od_lexpop(&config->lex, &tk);
|
|
switch (rc) {
|
|
/* daemonize */
|
|
case OD_LDAEMONIZE:
|
|
rc = od_confignext_yes_no(config, &tk);
|
|
if (rc == -1)
|
|
return -1;
|
|
config->scheme->daemonize = rc;
|
|
continue;
|
|
/* log_file */
|
|
case OD_LLOG_FILE:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
config->scheme->log_file = tk->v.string;
|
|
continue;
|
|
/* pid_file */
|
|
case OD_LPID_FILE:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
config->scheme->pid_file = tk->v.string;
|
|
continue;
|
|
/* pooling */
|
|
case OD_LPOOLING:
|
|
if (od_confignext(config, OD_LSTRING, &tk) == -1)
|
|
return -1;
|
|
config->scheme->pooling = tk->v.string;
|
|
continue;
|
|
/* listen */
|
|
case OD_LLISTEN:
|
|
rc = od_configparse_listen(config);
|
|
if (rc == -1)
|
|
return -1;
|
|
continue;
|
|
/* server */
|
|
case OD_LSERVER:
|
|
rc = od_configparse_server(config);
|
|
if (rc == -1)
|
|
return -1;
|
|
continue;
|
|
/* routing */
|
|
case OD_LROUTING:
|
|
rc = od_configparse_routing(config);
|
|
if (rc == -1)
|
|
return -1;
|
|
continue;
|
|
case OD_LEOF:
|
|
od_configerror(config, tk, "unexpected end of config file");
|
|
return -1;
|
|
case '}':
|
|
eof = 1;
|
|
continue;
|
|
default:
|
|
od_configerror(config, tk, "unknown option");
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|