c0fc891fbd4624381bb1a6e38a3293c514bd0c2d
[exim.git] / src / src / lookups / sqlite.c
1 /*************************************************
2 *     Exim - an Internet mail transport agent    *
3 *************************************************/
4
5 /* Copyright (c) University of Cambridge 1995 - 2009 */
6 /* See the file NOTICE for conditions of use and distribution. */
7
8 #include "../exim.h"
9 #include "lf_functions.h"
10
11 #include <sqlite3.h>
12
13
14 /*************************************************
15 *              Open entry point                  *
16 *************************************************/
17
18 /* See local README for interface description. */
19
20 static void *
21 sqlite_open(uschar *filename, uschar **errmsg)
22 {
23 sqlite3 *db = NULL;
24 int ret;
25
26 ret = sqlite3_open((char *)filename, &db);
27 if (ret != 0)
28   {
29   *errmsg = (void *)sqlite3_errmsg(db);
30   debug_printf("Error opening database: %s\n", *errmsg);
31   }
32
33 sqlite3_busy_timeout(db, 1000 * sqlite_lock_timeout);
34 return db;
35 }
36
37
38 /*************************************************
39 *               Find entry point                 *
40 *************************************************/
41
42 /* See local README for interface description. */
43
44 struct strbuf {
45   uschar *string;
46   int size;
47   int len;
48 };
49
50 static int sqlite_callback(void *arg, int argc, char **argv, char **azColName)
51 {
52 struct strbuf *res = arg;
53 int i;
54
55 /* For second and subsequent results, insert \n */
56
57 if (res->string != NULL)
58   res->string = string_cat(res->string, &res->size, &res->len, US"\n", 1);
59
60 if (argc > 1)
61   {
62   /* For multiple fields, include the field name too */
63   for (i = 0; i < argc; i++)
64     {
65     uschar *value = US((argv[i] != NULL)? argv[i]:"<NULL>");
66     res->string = lf_quote(US azColName[i], value, Ustrlen(value), res->string,
67       &res->size, &res->len);
68     }
69   }
70
71 else
72   {
73   res->string = string_append(res->string, &res->size, &res->len, 1,
74     (argv[0] != NULL)? argv[0]:"<NULL>");
75   }
76
77 res->string[res->len] = 0;
78 return 0;
79 }
80
81
82 static int
83 sqlite_find(void *handle, uschar *filename, uschar *query, int length,
84   uschar **result, uschar **errmsg, BOOL *do_cache)
85 {
86 int ret;
87 struct strbuf res = { NULL, 0, 0 };
88
89 ret = sqlite3_exec(handle, (char *)query, sqlite_callback, &res, (char **)errmsg);
90 if (ret != SQLITE_OK)
91   {
92   debug_printf("sqlite3_exec failed: %s\n", *errmsg);
93   return FAIL;
94   }
95
96 if (res.string == NULL) *do_cache = FALSE;
97
98 *result = res.string;
99 return OK;
100 }
101
102
103
104 /*************************************************
105 *               Close entry point                *
106 *************************************************/
107
108 /* See local README for interface description. */
109
110 static void sqlite_close(void *handle)
111 {
112 sqlite3_close(handle);
113 }
114
115
116
117 /*************************************************
118 *               Quote entry point                *
119 *************************************************/
120
121 /* From what I have found so far, the only character that needs to be quoted
122 for sqlite is the single quote, and it is quoted by doubling.
123
124 Arguments:
125   s          the string to be quoted
126   opt        additional option text or NULL if none
127
128 Returns:     the processed string or NULL for a bad option
129 */
130
131 static uschar *
132 sqlite_quote(uschar *s, uschar *opt)
133 {
134 register int c;
135 int count = 0;
136 uschar *t = s;
137 uschar *quoted;
138
139 if (opt != NULL) return NULL;     /* No options recognized */
140
141 while ((c = *t++) != 0) if (c == '\'') count++;
142
143 if (count == 0) return s;
144 t = quoted = store_get(Ustrlen(s) + count + 1);
145
146 while ((c = *s++) != 0)
147   {
148   if (c == '\'') *t++ = '\'';
149   *t++ = c;
150   }
151
152 *t = 0;
153 return quoted;
154 }
155
156
157
158 /*************************************************
159 *         Version reporting entry point          *
160 *************************************************/
161
162 /* See local README for interface description. */
163
164 #include "../version.h"
165
166 void
167 sqlite_version_report(FILE *f)
168 {
169 fprintf(f, "Library version: SQLite: Compile: %s\n"
170            "                         Runtime: %s\n",
171         SQLITE_VERSION, sqlite3_libversion());
172 #ifdef DYNLOOKUP
173 fprintf(f, "                         Exim version %s\n", EXIM_VERSION_STR);
174 #endif
175 }
176
177 static lookup_info _lookup_info = {
178   US"sqlite",                    /* lookup name */
179   lookup_absfilequery,           /* query-style lookup, starts with file name */
180   sqlite_open,                   /* open function */
181   NULL,                          /* no check function */
182   sqlite_find,                   /* find function */
183   sqlite_close,                  /* close function */
184   NULL,                          /* no tidy function */
185   sqlite_quote,                  /* quoting function */
186   sqlite_version_report          /* version reporting */
187 };
188
189 #ifdef DYNLOOKUP
190 #define sqlite_lookup_module_info _lookup_module_info
191 #endif
192
193 static lookup_info *_lookup_list[] = { &_lookup_info };
194 lookup_module_info sqlite_lookup_module_info = { LOOKUP_MODULE_INFO_MAGIC, _lookup_list, 1 };
195
196 /* End of lookups/sqlite.c */