1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
|
/*
* PgBouncer - Lightweight connection pooler for PostgreSQL.
*
* Copyright (c) 2007-2009 Marko Kreen, Skype Technologies OÜ
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
extern struct StatList user_list;
extern struct AATree user_tree;
extern struct StatList pool_list;
extern struct StatList peer_pool_list;
extern struct StatList database_list;
extern struct StatList peer_list;
extern struct StatList autodatabase_idle_list;
extern struct StatList login_client_list;
extern struct Slab *client_cache;
extern struct Slab *server_cache;
extern struct Slab *db_cache;
extern struct Slab *peer_cache;
extern struct Slab *peer_pool_cache;
extern struct Slab *pool_cache;
extern struct Slab *user_cache;
extern struct Slab *credentials_cache;
extern struct Slab *iobuf_cache;
extern struct Slab *outstanding_request_cache;
extern struct Slab *var_list_cache;
extern struct Slab *server_prepared_statement_cache;
extern PgPreparedStatement *prepared_statements;
extern unsigned long long int last_pgsocket_id;
PgDatabase *find_peer(int peer_id);
PgDatabase *find_database(const char *name);
PgDatabase *find_or_register_database(PgSocket *connection, const char *name);
PgGlobalUser *find_global_user(const char *name);
PgCredentials *find_global_credentials(const char *name);
PgPool *get_pool(PgDatabase *db, PgCredentials *user_credentials);
PgPool *get_peer_pool(PgDatabase *);
PgSocket *compare_connections_by_time(PgSocket *lhs, PgSocket *rhs);
bool evict_connection(PgDatabase *db) _MUSTCHECK;
bool evict_pool_connection(PgPool *pool) _MUSTCHECK;
bool evict_user_connection(PgCredentials *user_credentials) _MUSTCHECK;
bool find_server(PgSocket *client) _MUSTCHECK;
bool life_over(PgSocket *server);
bool release_server(PgSocket *server) /* _MUSTCHECK */;
bool finish_client_login(PgSocket *client) _MUSTCHECK;
bool check_fast_fail(PgSocket *client) _MUSTCHECK;
PgSocket *accept_client(int sock, bool is_unix) _MUSTCHECK;
void disconnect_server(PgSocket *server, bool notify, const char *reason, ...) _PRINTF(3, 4);
void disconnect_client(PgSocket *client, bool notify, const char *reason, ...) _PRINTF(3, 4);
void disconnect_client_sqlstate(PgSocket *client, bool notify, const char *sqlstate, const char *reason);
PgDatabase * add_peer(const char *name, int peer_id) _MUSTCHECK;
PgDatabase * add_database(const char *name) _MUSTCHECK;
PgDatabase *register_auto_database(const char *name);
PgCredentials * add_dynamic_credentials(PgDatabase *db, const char *name, const char *passwd) _MUSTCHECK;
PgCredentials * force_user_credentials(PgDatabase *db, const char *username, const char *passwd) _MUSTCHECK;
bool add_outstanding_request(PgSocket *client, char type, ResponseAction action) _MUSTCHECK;
bool pop_outstanding_request(PgSocket *client, const char types[], bool *skip);
bool clear_outstanding_requests_until(PgSocket *server, const char types[]) _MUSTCHECK;
bool queue_fake_response(PgSocket *client, char request_type) _MUSTCHECK;
PgGlobalUser * update_global_user_passwd(PgGlobalUser *user, const char *passwd) _MUSTCHECK;
PgGlobalUser * find_or_add_new_global_user(const char *name, const char *passwd) _MUSTCHECK;
PgCredentials * find_or_add_new_global_credentials(const char *name, const char *passwd) _MUSTCHECK;
PgCredentials * add_pam_credentials(const char *name, const char *passwd) _MUSTCHECK;
void accept_cancel_request(PgSocket *req);
void forward_cancel_request(PgSocket *server);
void launch_new_connection(PgPool *pool, bool evict_if_needed);
bool use_client_socket(int fd, PgAddr *addr, const char *dbname, const char *username, uint64_t ckey, int oldfd, int linkfd,
const char *client_end, const char *std_string, const char *datestyle, const char *timezone,
const char *password,
const char *scram_client_key, int scram_client_key_len,
const char *scram_server_key, int scram_server_key_len) _MUSTCHECK;
bool use_server_socket(int fd, PgAddr *addr, const char *dbname, const char *username, uint64_t ckey, int oldfd, int linkfd,
const char *client_end, const char *std_string, const char *datestyle, const char *timezone,
const char *password,
const char *scram_client_key, int scram_client_key_len,
const char *scram_server_key, int scram_server_key_len) _MUSTCHECK;
void activate_client(PgSocket *client);
void change_client_state(PgSocket *client, SocketState newstate);
void change_server_state(PgSocket *server, SocketState newstate);
int get_active_client_count(void);
int get_active_server_count(void);
void tag_pool_dirty(PgPool *pool);
void tag_database_dirty(PgDatabase *db);
void tag_autodb_dirty(void);
void tag_host_addr_dirty(const char *host, const struct sockaddr *sa);
void for_each_server(PgPool *pool, void (*func)(PgSocket *sk));
void reuse_just_freed_objects(void);
void init_objects(void);
void init_caches(void);
void objects_cleanup(void);
|