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 */
|
