odyssey/sources/rules.h

#ifndef ODYSSEY_RULES_H
#define ODYSSEY_RULES_H

/*
 * Odyssey.
 *
 * Scalable PostgreSQL connection pooler.
 */

typedef struct od_rule_storage od_rule_storage_t;
typedef struct od_rule_auth od_rule_auth_t;
typedef struct od_rule od_rule_t;
typedef struct od_rules od_rules_t;

typedef enum {
	OD_RULE_TLS_DISABLE,
	OD_RULE_TLS_ALLOW,
	OD_RULE_TLS_REQUIRE,
	OD_RULE_TLS_VERIFY_CA,
	OD_RULE_TLS_VERIFY_FULL
} od_rule_tls_t;

typedef enum {
	OD_RULE_AUTH_UNDEF,
	OD_RULE_AUTH_NONE,
	OD_RULE_AUTH_BLOCK,
	OD_RULE_AUTH_CLEAR_TEXT,
	OD_RULE_AUTH_MD5,
	OD_RULE_AUTH_SCRAM_SHA_256,
	OD_RULE_AUTH_CERT
} od_rule_auth_type_t;

typedef enum {
	OD_RULE_POOL_SESSION,
	OD_RULE_POOL_TRANSACTION
} od_rule_pool_type_t;

typedef enum {
	OD_RULE_STORAGE_REMOTE,
	OD_RULE_STORAGE_LOCAL,
} od_rule_storage_type_t;

struct od_rule_storage {
	char *name;
	char *type;
	od_rule_storage_type_t storage_type;
	char *host;
	int port;
	od_rule_tls_t tls_mode;
	char *tls;
	char *tls_ca_file;
	char *tls_key_file;
	char *tls_cert_file;
	char *tls_protocols;
	int server_max_routing;
	od_list_t link;
};

struct od_rule_auth {
	char *common_name;
	od_list_t link;
};

struct od_rule {
	/* versioning */
	int mark;
	int obsolete;
	int refs;
	/* id */
	char *db_name;
	int db_name_len;
	int db_is_default;
	char *user_name;
	int user_name_len;
	int user_is_default;
	/* auth */
	char *auth;
	od_rule_auth_type_t auth_mode;
	char *auth_query;
	char *auth_query_db;
	char *auth_query_user;
	int auth_common_name_default;
	od_list_t auth_common_names;
	int auth_common_names_count;

	/*  PAM parametrs */
	char *auth_pam_service;
	od_pam_auth_data_t *auth_pam_data;

	/* password */
	char *password;
	int password_len;
	/* storage */
	od_rule_storage_t *storage;
	char *storage_name;
	char *storage_db;
	char *storage_user;
	int storage_user_len;
	char *storage_password;
	int storage_password_len;
	/* pool */
	od_rule_pool_type_t pool;
	char *pool_sz;
	int pool_size;
	int pool_timeout;
	int pool_ttl;
	int pool_discard;
	int pool_cancel;
	int pool_rollback;
	uint64_t pool_client_idle_timeout; // makes sence only for session pooling
	uint64_t pool_idle_in_transaction_timeout; // makes sence only for session pooling
	/* misc */
	int client_fwd_error;
	int reserve_session_server_connection;
	int application_name_add_host;
	int client_max_set;
	int client_max;
	int log_debug;
	int log_query;
	double *quantiles;
	int quantiles_count;
	uint64_t server_lifetime_us;
	od_list_t link;
};

struct od_rules {
	od_list_t storages;
	od_list_t rules;
};

void od_rules_init(od_rules_t *);
void od_rules_free(od_rules_t *);
int od_rules_validate(od_rules_t *, od_config_t *, od_logger_t *);
int od_rules_merge(od_rules_t *, od_rules_t *);
void od_rules_print(od_rules_t *, od_logger_t *);

/* rule */
od_rule_t *od_rules_add(od_rules_t *);
void od_rules_ref(od_rule_t *);
void od_rules_unref(od_rule_t *);
int od_rules_compare(od_rule_t *, od_rule_t *);

od_rule_t *od_rules_forward(od_rules_t *, char *, char *);

od_rule_t *od_rules_match(od_rules_t *, char *, char *, int, int);

/* storage */
od_rule_storage_t *od_rules_storage_add(od_rules_t *);

od_rule_storage_t *od_rules_storage_match(od_rules_t *, char *);

od_rule_storage_t *od_rules_storage_copy(od_rule_storage_t *);

void od_rules_storage_free(od_rule_storage_t *);

/* auth */
od_rule_auth_t *od_rules_auth_add(od_rule_t *);

void od_rules_auth_free(od_rule_auth_t *);

#endif /* ODYSSEY_RULES_H */