af4c0eac6a5c35acb93d7973251386b13437328d
[exim.git] / src / src / lookups / sqlite.c
1 /* $Cambridge: exim/src/src/lookups/sqlite.c,v 1.1 2005/08/01 13:20:28 ph10 Exp $ */
2
3 /*************************************************
4 *     Exim - an Internet mail transport agent    *
5 *************************************************/
6
7 /* Copyright (c) University of Cambridge 1995 - 2005 */
8 /* See the file NOTICE for conditions of use and distribution. */
9
10 #include "../exim.h"
11 #include "lf_functions.h"
12 #include "sqlite.h"
13
14 #ifndef LOOKUP_SQLITE
15 static void dummy(int x) { dummy(x-1); }
16 #else
17 #include <sqlite3.h>
18
19
20 /*************************************************
21 *              Open entry point                  *
22 *************************************************/
23
24 /* See local README for interface description. */
25
26 void *
27 sqlite_open(uschar *filename, uschar **errmsg)
28 {
29 sqlite3 *db = NULL;
30 int ret;
31
32 ret = sqlite3_open((char *)filename, &db);
33 if (ret != 0)
34   {
35   *errmsg = (void *)sqlite3_errmsg(db);
36   debug_printf("Error opening database: %s\n", *errmsg);
37   }
38
39 return db;
40 }
41
42
43 /*************************************************
44 *               Find entry point                 *
45 *************************************************/
46
47 /* See local README for interface description. */
48
49 struct strbuf {
50   uschar *string;
51   int size;
52   int len;
53 };
54
55 static int sqlite_callback(void *arg, int argc, char **argv, char **azColName)
56 {
57 struct strbuf *res = arg;
58 int i;
59
60 /* For second and subsequent results, insert \n */
61
62 if (res->string != NULL)
63   res->string = string_cat(res->string, &res->size, &res->len, US"\n", 1);
64
65 if (argc > 1)
66   {
67   /* For multiple fields, include the field name too */
68   for (i = 0; i < argc; i++)
69     {
70     uschar *value = US((argv[i] != NULL)? argv[i]:"<NULL>");
71     res->string = lf_quote(US azColName[i], value, Ustrlen(value), res->string,
72       &res->size, &res->len);
73     }
74   }
75
76 else
77   {
78   res->string = string_append(res->string, &res->size, &res->len, 1,
79     (argv[0] != NULL)? argv[0]:"<NULL>");
80   }
81
82 res->string[res->len] = 0;
83 return 0;
84 }
85
86
87 int
88 sqlite_find(void *handle, uschar *filename, uschar *query, int length,
89   uschar **result, uschar **errmsg, BOOL *do_cache)
90 {
91 int ret;
92 struct strbuf res = { NULL, 0, 0 };
93
94 ret = sqlite3_exec(handle, (char *)query, sqlite_callback, &res, (char **)errmsg);
95 if (ret != SQLITE_OK)
96   {
97   debug_printf("sqlite3_exec failed: %s\n", *errmsg);
98   return FAIL;
99   }
100
101 if (res.string == NULL) *do_cache = FALSE;
102
103 *result = res.string;
104 return OK;
105 }
106
107
108
109 /*************************************************
110 *               Close entry point                *
111 *************************************************/
112
113 /* See local README for interface description. */
114
115 void sqlite_close(void *handle)
116 {
117 sqlite3_close(handle);
118 }
119
120
121
122 /*************************************************
123 *               Quote entry point                *
124 *************************************************/
125
126 /* From what I have found so far, the only character that needs to be quoted
127 for sqlite is the single quote, and it is quoted by doubling.
128
129 Arguments:
130   s          the string to be quoted
131   opt        additional option text or NULL if none
132
133 Returns:     the processed string or NULL for a bad option
134 */
135
136 uschar *
137 sqlite_quote(uschar *s, uschar *opt)
138 {
139 register int c;
140 int count = 0;
141 uschar *t = s;
142 uschar *quoted;
143
144 if (opt != NULL) return NULL;     /* No options recognized */
145
146 while ((c = *t++) != 0) if (c == '\'') count++;
147
148 if (count == 0) return s;
149 t = quoted = store_get(Ustrlen(s) + count + 1);
150
151 while ((c = *s++) != 0)
152   {
153   if (c == '\'') *t++ = '\'';
154   *t++ = c;
155   }
156
157 *t = 0;
158 return quoted;
159 }
160
161 #endif /* LOOKUP_SQLITE */
162
163 /* End of lookups/sqlite.c */