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
|
/* $Id: pgsql_drv.h,v 1.16 2011/06/28 00:13:48 sbajic Exp $ */
/*
DSPAM
COPYRIGHT (C) 2002-2011 DSPAM PROJECT
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _PGSQL_DRV_H
# define _PGSQL_DRV_H
#ifdef HAVE_CONFIG_H
#include <auto-config.h>
#endif
#include <libpq-fe.h>
struct _pgsql_drv_storage
{
PGconn *dbh; /* database connection */
int pg_major_ver; /* database major version */
int pg_minor_ver; /* database minor version */
int pg_micro_ver; /* database micro version */
int pg_token_type; /* type of token */
struct _ds_spam_totals control_totals; /* totals at storage init */
struct _ds_spam_totals merged_totals; /* totals for merged group */
/* control token data; used to measure deltas from getall to setall
* enabling us to code a sql query based on increments/decrements
* instead of query-coded data */
unsigned long long control_token; /* control token crc */
long control_sh; /* control token spam hits at getall */
long control_ih; /* control token innocent hits at getall */
PGresult *iter_user; /* get_nextuser iteration result */
PGresult *iter_token; /* get_nexttoken iteration result */
PGresult *iter_sig; /* get_nextsignature iteration result */
char u_getnextuser[MAX_FILENAME_LENGTH];
struct passwd p_getpwuid;
struct passwd p_getpwnam;
int dbh_attached;
};
/* Driver-specific functions */
int _pgsql_drv_get_spamtotals (DSPAM_CTX * CTX);
int _pgsql_drv_set_spamtotals (DSPAM_CTX * CTX);
void _pgsql_drv_query_error (const char *error, const char *query);
int _pgsql_drv_token_type (struct _pgsql_drv_storage *s, PGresult *result, int column);
char *_pgsql_drv_token_write (int type, unsigned long long token, char *buffer, size_t bufsz);
unsigned long long _pgsql_drv_token_read(int type, char *str);
int _pgsql_drv_get_dbversion (struct _pgsql_drv_storage *s, unsigned int range);
PGconn *_pgsql_drv_connect (DSPAM_CTX *CTX);
struct passwd *_pgsql_drv_getpwnam (DSPAM_CTX * CTX, const char *name);
struct passwd *_pgsql_drv_getpwuid (DSPAM_CTX * CTX, uid_t uid);
DSPAM_CTX *_pgsql_drv_init_tools( const char *home, config_t config,
void *dbh, int mode);
#ifdef VIRTUAL_USERS
struct passwd *_pgsql_drv_setpwnam (DSPAM_CTX * CTX, const char *name);
#endif
#ifdef EXT_LOOKUP
int verified_user;
#endif
#ifdef PREFERENCES_EXTENSION
int _pgsql_drv_set_attributes(DSPAM_CTX *CTX, config_t config);
#endif
#endif /* _PGSQL_DRV_H */
|