X-Git-Url: https://git.exim.org/exim.git/blobdiff_plain/d7d7b7b91dd75cec636fc144da7e27eed860f971..1d28cc061677bd07d9bed48dd84bd5c590247043:/src/src/lookups/sqlite.c diff --git a/src/src/lookups/sqlite.c b/src/src/lookups/sqlite.c index aacf4ef48..6c7af4225 100644 --- a/src/src/lookups/sqlite.c +++ b/src/src/lookups/sqlite.c @@ -1,19 +1,15 @@ -/* $Cambridge: exim/src/src/lookups/sqlite.c,v 1.3 2006/02/07 11:19:01 ph10 Exp $ */ - /************************************************* * Exim - an Internet mail transport agent * *************************************************/ -/* Copyright (c) University of Cambridge 1995 - 2006 */ +/* Copyright (c) The Exim Maintainers 2020 - 2022 */ +/* Copyright (c) University of Cambridge 1995 - 2018 */ /* See the file NOTICE for conditions of use and distribution. */ +/* SPDX-License-Identifier: GPL-2.0-or-later */ #include "../exim.h" #include "lf_functions.h" -#include "sqlite.h" -#ifndef LOOKUP_SQLITE -static void dummy(int x) { dummy(x-1); } -#else #include @@ -23,20 +19,29 @@ static void dummy(int x) { dummy(x-1); } /* See local README for interface description. */ -void * -sqlite_open(uschar *filename, uschar **errmsg) +static void * +sqlite_open(const uschar * filename, uschar ** errmsg) { sqlite3 *db = NULL; int ret; -ret = sqlite3_open((char *)filename, &db); -if (ret != 0) +if (!filename || !*filename) + { + DEBUG(D_lookup) debug_printf_indent("Using sqlite_dbfile: %s\n", sqlite_dbfile); + filename = sqlite_dbfile; + } +if (!filename || *filename != '/') + *errmsg = US"absolute file name expected for \"sqlite\" lookup"; +else if ((ret = sqlite3_open(CCS filename, &db)) != 0) { *errmsg = (void *)sqlite3_errmsg(db); - debug_printf("Error opening database: %s\n", *errmsg); + sqlite3_close(db); + db = NULL; + DEBUG(D_lookup) debug_printf_indent("Error opening database: %s\n", *errmsg); } -sqlite3_busy_timeout(db, 1000 * sqlite_lock_timeout); +if (db) + sqlite3_busy_timeout(db, 1000 * sqlite_lock_timeout); return db; } @@ -47,61 +52,53 @@ return db; /* See local README for interface description. */ -struct strbuf { - uschar *string; - int size; - int len; -}; - -static int sqlite_callback(void *arg, int argc, char **argv, char **azColName) +static int +sqlite_callback(void *arg, int argc, char **argv, char **azColName) { -struct strbuf *res = arg; -int i; +gstring * res = *(gstring **)arg; /* For second and subsequent results, insert \n */ -if (res->string != NULL) - res->string = string_cat(res->string, &res->size, &res->len, US"\n", 1); +if (res) + res = string_catn(res, US"\n", 1); if (argc > 1) { /* For multiple fields, include the field name too */ - for (i = 0; i < argc; i++) + for (int i = 0; i < argc; i++) { - uschar *value = US((argv[i] != NULL)? argv[i]:""); - res->string = lf_quote(US azColName[i], value, Ustrlen(value), res->string, - &res->size, &res->len); + uschar * value = US(argv[i] ? argv[i] : ""); + res = lf_quote(US azColName[i], value, Ustrlen(value), res); } } else - { - res->string = string_append(res->string, &res->size, &res->len, 1, - (argv[0] != NULL)? argv[0]:""); - } + res = string_cat(res, argv[0] ? US argv[0] : US ""); -res->string[res->len] = 0; +/* always return a non-null gstring, even for a zero-length string result */ +*(gstring **)arg = res ? res : string_get(1); return 0; } -int -sqlite_find(void *handle, uschar *filename, uschar *query, int length, - uschar **result, uschar **errmsg, BOOL *do_cache) +static int +sqlite_find(void * handle, const uschar * filename, const uschar * query, + int length, uschar ** result, uschar ** errmsg, uint * do_cache, + const uschar * opts) { int ret; -struct strbuf res = { NULL, 0, 0 }; +gstring * res = NULL; -ret = sqlite3_exec(handle, (char *)query, sqlite_callback, &res, (char **)errmsg); +ret = sqlite3_exec(handle, CS query, sqlite_callback, &res, CSS errmsg); if (ret != SQLITE_OK) { - debug_printf("sqlite3_exec failed: %s\n", *errmsg); + debug_printf_indent("sqlite3_exec failed: %s\n", *errmsg); return FAIL; } -if (res.string == NULL) *do_cache = FALSE; +if (!res) *do_cache = 0; /* on fail, wipe cache */ -*result = res.string; +*result = string_from_gstring(res); return OK; } @@ -113,7 +110,7 @@ return OK; /* See local README for interface description. */ -void sqlite_close(void *handle) +static void sqlite_close(void *handle) { sqlite3_close(handle); } @@ -130,26 +127,25 @@ for sqlite is the single quote, and it is quoted by doubling. Arguments: s the string to be quoted opt additional option text or NULL if none + idx lookup type index Returns: the processed string or NULL for a bad option */ -uschar * -sqlite_quote(uschar *s, uschar *opt) +static uschar * +sqlite_quote(uschar * s, uschar * opt, unsigned idx) { -register int c; -int count = 0; -uschar *t = s; -uschar *quoted; +int c, count = 0; +uschar * t = s, * quoted; -if (opt != NULL) return NULL; /* No options recognized */ +if (opt) return NULL; /* No options recognized */ -while ((c = *t++) != 0) if (c == '\'') count++; +while ((c = *t++)) if (c == '\'') count++; +count += t - s; -if (count == 0) return s; -t = quoted = store_get(Ustrlen(s) + count + 1); +t = quoted = store_get_quoted(count + 1, s, idx); -while ((c = *s++) != 0) +while ((c = *s++)) { if (c == '\'') *t++ = '\''; *t++ = c; @@ -159,6 +155,47 @@ while ((c = *s++) != 0) return quoted; } -#endif /* LOOKUP_SQLITE */ + + +/************************************************* +* Version reporting entry point * +*************************************************/ + +/* See local README for interface description. */ + +#include "../version.h" + +gstring * +sqlite_version_report(gstring * g) +{ +g = string_fmt_append(g, + "Library version: SQLite: Compile: %s\n" + " Runtime: %s\n", + SQLITE_VERSION, sqlite3_libversion()); +#ifdef DYNLOOKUP +g = string_fmt_append(g, + " Exim version %s\n", EXIM_VERSION_STR); +#endif +return g; +} + +static lookup_info _lookup_info = { + .name = US"sqlite", /* lookup name */ + .type = lookup_absfilequery, /* query-style lookup, starts with file name */ + .open = sqlite_open, /* open function */ + .check = NULL, /* no check function */ + .find = sqlite_find, /* find function */ + .close = sqlite_close, /* close function */ + .tidy = NULL, /* no tidy function */ + .quote = sqlite_quote, /* quoting function */ + .version_report = sqlite_version_report /* version reporting */ +}; + +#ifdef DYNLOOKUP +#define sqlite_lookup_module_info _lookup_module_info +#endif + +static lookup_info *_lookup_list[] = { &_lookup_info }; +lookup_module_info sqlite_lookup_module_info = { LOOKUP_MODULE_INFO_MAGIC, _lookup_list, 1 }; /* End of lookups/sqlite.c */