4 Copyright (C) Derrell Lipman 2005
5 Copyright (C) Simo Sorce 2005-2009
7 ** NOTE! The following LGPL license applies to the ldb
8 ** library. This does NOT imply that all of Samba is released
11 This library is free software; you can redistribute it and/or
12 modify it under the terms of the GNU Lesser General Public
13 License as published by the Free Software Foundation; either
14 version 3 of the License, or (at your option) any later version.
16 This library is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19 Lesser General Public License for more details.
21 You should have received a copy of the GNU Lesser General Public
22 License along with this library; if not, see <http://www.gnu.org/licenses/>.
28 * Component: ldb sqlite3 backend
30 * Description: core files for SQLITE3 backend
32 * Author: Derrell Lipman (based on Andrew Tridgell's LDAP backend)
35 #include "ldb_module.h"
39 struct lsqlite3_private
{
46 struct ldb_module
*module
;
47 struct ldb_request
*req
;
50 long long current_eid
;
51 const char * const * attrs
;
52 struct ldb_reply
*ares
;
55 struct tevent_timer
*timeout_event
;
59 * Macros used throughout
64 # define TRUE (! FALSE)
67 #define RESULT_ATTR_TABLE "temp_result_attrs"
70 /* for testing, define to nothing, (create non-temporary table) */
71 #define TEMPTAB "TEMPORARY"
76 sqlite3_stmt
* stmtGetEID
= NULL
;
78 static char *lsqlite3_tprintf(TALLOC_CTX
*mem_ctx
, const char *fmt
, ...)
84 str
= sqlite3_vmprintf(fmt
, ap
);
87 if (str
== NULL
) return NULL
;
89 ret
= talloc_strdup(mem_ctx
, str
);
99 static char base160tab
[161] = {
100 48 ,49 ,50 ,51 ,52 ,53 ,54 ,55 ,56 ,57 , /* 0-9 */
101 58 ,59 ,65 ,66 ,67 ,68 ,69 ,70 ,71 ,72 , /* : ; A-H */
102 73 ,74 ,75 ,76 ,77 ,78 ,79 ,80 ,81 ,82 , /* I-R */
103 83 ,84 ,85 ,86 ,87 ,88 ,89 ,90 ,97 ,98 , /* S-Z , a-b */
104 99 ,100,101,102,103,104,105,106,107,108, /* c-l */
105 109,110,111,112,113,114,115,116,117,118, /* m-v */
106 119,120,121,122,160,161,162,163,164,165, /* w-z, latin1 */
107 166,167,168,169,170,171,172,173,174,175, /* latin1 */
108 176,177,178,179,180,181,182,183,184,185, /* latin1 */
109 186,187,188,189,190,191,192,193,194,195, /* latin1 */
110 196,197,198,199,200,201,202,203,204,205, /* latin1 */
111 206,207,208,209,210,211,212,213,214,215, /* latin1 */
112 216,217,218,219,220,221,222,223,224,225, /* latin1 */
113 226,227,228,229,230,231,232,233,234,235, /* latin1 */
114 236,237,238,239,240,241,242,243,244,245, /* latin1 */
115 246,247,248,249,250,251,252,253,254,255, /* latin1 */
123 * Convert an unsigned long integer into a base160 representation of the
128 * value to be converted
131 * character array, 5 bytes long, into which the base160 representation
132 * will be placed. The result will be a four-digit representation of the
133 * number (with leading zeros prepended as necessary), and null
140 base160_sql(sqlite3_context
* hContext
,
142 sqlite3_value
** argv
)
148 val
= sqlite3_value_int64(argv
[0]);
150 for (i
= 3; i
>= 0; i
--) {
152 result
[i
] = base160tab
[val
% 160];
158 sqlite3_result_text(hContext
, result
, -1, SQLITE_TRANSIENT
);
165 * This function enhances sqlite by adding a "base160_next()" function which is
166 * accessible via queries.
168 * Retrieve the next-greater number in the base160 sequence for the terminal
169 * tree node (the last four digits). Only one tree level (four digits) is
173 * A character string: either an empty string (in which case no operation is
174 * performed), or a string of base160 digits with a length of a multiple of
178 * Upon return, the trailing four digits (one tree level) will have been
182 base160next_sql(sqlite3_context
* hContext
,
184 sqlite3_value
** argv
)
189 char * pBase160
= strdup((const char *)sqlite3_value_text(argv
[0]));
190 char * pStart
= pBase160
;
193 * We need a minimum of four digits, and we will always get a multiple
196 if (pBase160
!= NULL
&&
197 (len
= strlen(pBase160
)) >= 4 &&
200 if (pBase160
== NULL
) {
202 sqlite3_result_null(hContext
);
206 pBase160
+= strlen(pBase160
) - 1;
208 /* We only carry through four digits: one level in the tree */
209 for (i
= 0; i
< 4; i
++) {
211 /* What base160 value does this digit have? */
212 pTab
= strchr(base160tab
, *pBase160
);
214 /* Is there a carry? */
215 if (pTab
< base160tab
+ sizeof(base160tab
) - 1) {
218 * Nope. Just increment this value and we're
226 * There's a carry. This value gets
227 * base160tab[0], we decrement the buffer
228 * pointer to get the next higher-order digit,
229 * and continue in the loop.
231 *pBase160
-- = base160tab
[0];
235 sqlite3_result_text(hContext
,
240 sqlite3_result_value(hContext
, argv
[0]);
241 if (pBase160
!= NULL
) {
247 static char *parsetree_to_sql(struct ldb_module
*module
,
249 const struct ldb_parse_tree
*t
)
251 struct ldb_context
*ldb
;
252 const struct ldb_schema_attribute
*a
;
253 struct ldb_val value
, subval
;
254 char *wild_card_string
;
260 ldb
= ldb_module_get_ctx(module
);
262 switch(t
->operation
) {
265 tmp
= parsetree_to_sql(module
, mem_ctx
, t
->u
.list
.elements
[0]);
266 if (tmp
== NULL
) return NULL
;
268 for (i
= 1; i
< t
->u
.list
.num_elements
; i
++) {
270 child
= parsetree_to_sql(module
, mem_ctx
, t
->u
.list
.elements
[i
]);
271 if (child
== NULL
) return NULL
;
273 tmp
= talloc_asprintf_append(tmp
, " INTERSECT %s ", child
);
274 if (tmp
== NULL
) return NULL
;
277 ret
= talloc_asprintf(mem_ctx
, "SELECT * FROM ( %s )\n", tmp
);
283 tmp
= parsetree_to_sql(module
, mem_ctx
, t
->u
.list
.elements
[0]);
284 if (tmp
== NULL
) return NULL
;
286 for (i
= 1; i
< t
->u
.list
.num_elements
; i
++) {
288 child
= parsetree_to_sql(module
, mem_ctx
, t
->u
.list
.elements
[i
]);
289 if (child
== NULL
) return NULL
;
291 tmp
= talloc_asprintf_append(tmp
, " UNION %s ", child
);
292 if (tmp
== NULL
) return NULL
;
295 return talloc_asprintf(mem_ctx
, "SELECT * FROM ( %s ) ", tmp
);
299 child
= parsetree_to_sql(module
, mem_ctx
, t
->u
.isnot
.child
);
300 if (child
== NULL
) return NULL
;
302 return talloc_asprintf(mem_ctx
,
303 "SELECT eid FROM ldb_entry "
304 "WHERE eid NOT IN ( %s ) ", child
);
306 case LDB_OP_EQUALITY
:
308 * For simple searches, we want to retrieve the list of EIDs that
309 * match the criteria.
311 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.equality
.attr
);
312 if (attr
== NULL
) return NULL
;
313 a
= ldb_schema_attribute_by_name(ldb
, attr
);
315 /* Get a canonicalised copy of the data */
316 a
->syntax
->canonicalise_fn(ldb
, mem_ctx
, &(t
->u
.equality
.value
), &value
);
317 if (value
.data
== NULL
) {
321 if (strcasecmp(t
->u
.equality
.attr
, "dn") == 0) {
322 /* DN query is a special ldb case */
323 const char *cdn
= ldb_dn_get_casefold(
324 ldb_dn_new(mem_ctx
, ldb
,
325 (const char *)value
.data
));
327 return lsqlite3_tprintf(mem_ctx
,
328 "SELECT eid FROM ldb_entry "
329 "WHERE norm_dn = '%q'", cdn
);
332 /* A normal query. */
333 return lsqlite3_tprintf(mem_ctx
,
334 "SELECT eid FROM ldb_attribute_values "
335 "WHERE norm_attr_name = '%q' "
336 "AND norm_attr_value = '%q'",
342 case LDB_OP_SUBSTRING
:
344 wild_card_string
= talloc_strdup(mem_ctx
,
345 (t
->u
.substring
.start_with_wildcard
)?"*":"");
346 if (wild_card_string
== NULL
) return NULL
;
348 for (i
= 0; t
->u
.substring
.chunks
[i
]; i
++) {
349 wild_card_string
= talloc_asprintf_append(wild_card_string
, "%s*",
350 t
->u
.substring
.chunks
[i
]->data
);
351 if (wild_card_string
== NULL
) return NULL
;
354 if ( ! t
->u
.substring
.end_with_wildcard
) {
355 /* remove last wildcard */
356 wild_card_string
[strlen(wild_card_string
) - 1] = '\0';
359 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.substring
.attr
);
360 if (attr
== NULL
) return NULL
;
361 a
= ldb_schema_attribute_by_name(ldb
, attr
);
363 subval
.data
= (void *)wild_card_string
;
364 subval
.length
= strlen(wild_card_string
) + 1;
366 /* Get a canonicalised copy of the data */
367 a
->syntax
->canonicalise_fn(ldb
, mem_ctx
, &(subval
), &value
);
368 if (value
.data
== NULL
) {
372 return lsqlite3_tprintf(mem_ctx
,
373 "SELECT eid FROM ldb_attribute_values "
374 "WHERE norm_attr_name = '%q' "
375 "AND norm_attr_value GLOB '%q'",
380 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.equality
.attr
);
381 if (attr
== NULL
) return NULL
;
382 a
= ldb_schema_attribute_by_name(ldb
, attr
);
384 /* Get a canonicalised copy of the data */
385 a
->syntax
->canonicalise_fn(ldb
, mem_ctx
, &(t
->u
.equality
.value
), &value
);
386 if (value
.data
== NULL
) {
390 return lsqlite3_tprintf(mem_ctx
,
391 "SELECT eid FROM ldb_attribute_values "
392 "WHERE norm_attr_name = '%q' "
393 "AND ldap_compare(norm_attr_value, '>=', '%q', '%q') ",
399 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.equality
.attr
);
400 if (attr
== NULL
) return NULL
;
401 a
= ldb_schema_attribute_by_name(ldb
, attr
);
403 /* Get a canonicalised copy of the data */
404 a
->syntax
->canonicalise_fn(ldb
, mem_ctx
, &(t
->u
.equality
.value
), &value
);
405 if (value
.data
== NULL
) {
409 return lsqlite3_tprintf(mem_ctx
,
410 "SELECT eid FROM ldb_attribute_values "
411 "WHERE norm_attr_name = '%q' "
412 "AND ldap_compare(norm_attr_value, '<=', '%q', '%q') ",
418 if (strcasecmp(t
->u
.present
.attr
, "dn") == 0) {
419 return talloc_strdup(mem_ctx
, "SELECT eid FROM ldb_entry");
422 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.present
.attr
);
423 if (attr
== NULL
) return NULL
;
425 return lsqlite3_tprintf(mem_ctx
,
426 "SELECT eid FROM ldb_attribute_values "
427 "WHERE norm_attr_name = '%q' ",
431 attr
= ldb_attr_casefold(mem_ctx
, t
->u
.equality
.attr
);
432 if (attr
== NULL
) return NULL
;
433 a
= ldb_schema_attribute_by_name(ldb
, attr
);
435 /* Get a canonicalised copy of the data */
436 a
->syntax
->canonicalise_fn(ldb
, mem_ctx
, &(t
->u
.equality
.value
), &value
);
437 if (value
.data
== NULL
) {
441 return lsqlite3_tprintf(mem_ctx
,
442 "SELECT eid FROM ldb_attribute_values "
443 "WHERE norm_attr_name = '%q' "
444 "AND ldap_compare(norm_attr_value, '~%', 'q', '%q') ",
449 case LDB_OP_EXTENDED
:
450 #warning "work out how to handle bitops"
457 /* should never occur */
465 * This function is used for the common case of queries that return a single
468 * NOTE: If more than one value is returned by the query, all but the first
469 * one will be ignored.
472 query_int(const struct lsqlite3_private
* lsqlite3
,
480 sqlite3_stmt
* pStmt
;
483 /* Begin access to variable argument list */
484 va_start(args
, pSql
);
486 /* Format the query */
487 if ((p
= sqlite3_vmprintf(pSql
, args
)) == NULL
) {
493 * Prepare and execute the SQL statement. Loop allows retrying on
494 * certain errors, e.g. SQLITE_SCHEMA occurs if the schema changes,
495 * requiring retrying the operation.
497 for (bLoop
= TRUE
; bLoop
; ) {
499 /* Compile the SQL statement into sqlite virtual machine */
500 if ((ret
= sqlite3_prepare(lsqlite3
->sqlite
,
504 NULL
)) == SQLITE_SCHEMA
) {
505 if (stmtGetEID
!= NULL
) {
506 sqlite3_finalize(stmtGetEID
);
510 } else if (ret
!= SQLITE_OK
) {
514 /* One row expected */
515 if ((ret
= sqlite3_step(pStmt
)) == SQLITE_SCHEMA
) {
516 if (stmtGetEID
!= NULL
) {
517 sqlite3_finalize(stmtGetEID
);
520 (void) sqlite3_finalize(pStmt
);
522 } else if (ret
!= SQLITE_ROW
) {
523 (void) sqlite3_finalize(pStmt
);
527 /* Get the value to be returned */
528 *pRet
= sqlite3_column_int64(pStmt
, 0);
530 /* Free the virtual machine */
531 if ((ret
= sqlite3_finalize(pStmt
)) == SQLITE_SCHEMA
) {
532 if (stmtGetEID
!= NULL
) {
533 sqlite3_finalize(stmtGetEID
);
537 } else if (ret
!= SQLITE_OK
) {
538 (void) sqlite3_finalize(pStmt
);
543 * Normal condition is only one time through loop. Loop is
544 * rerun in error conditions, via "continue", above.
549 /* All done with variable argument list */
553 /* Free the memory we allocated for our query string */
560 * This is a bad hack to support ldap style comparisons within sqlite.
561 * val is the attribute in the row currently under test
562 * func is the desired test "<=" ">=" "~" ":"
563 * cmp is the value to compare against (eg: "test")
564 * attr is the attribute name the value of which we want to test
567 static void lsqlite3_compare(sqlite3_context
*ctx
, int argc
,
568 sqlite3_value
**argv
)
570 struct ldb_context
*ldb
= (struct ldb_context
*)sqlite3_user_data(ctx
);
571 const char *val
= (const char *)sqlite3_value_text(argv
[0]);
572 const char *func
= (const char *)sqlite3_value_text(argv
[1]);
573 const char *cmp
= (const char *)sqlite3_value_text(argv
[2]);
574 const char *attr
= (const char *)sqlite3_value_text(argv
[3]);
575 const struct ldb_schema_attribute
*a
;
583 a
= ldb_schema_attribute_by_name(ldb
, attr
);
584 valX
.data
= (uint8_t *)cmp
;
585 valX
.length
= strlen(cmp
);
586 valY
.data
= (uint8_t *)val
;
587 valY
.length
= strlen(val
);
588 ret
= a
->syntax
->comparison_fn(ldb
, ldb
, &valY
, &valX
);
590 sqlite3_result_int(ctx
, 1);
592 sqlite3_result_int(ctx
, 0);
597 a
= ldb_schema_attribute_by_name(ldb
, attr
);
598 valX
.data
= (uint8_t *)cmp
;
599 valX
.length
= strlen(cmp
);
600 valY
.data
= (uint8_t *)val
;
601 valY
.length
= strlen(val
);
602 ret
= a
->syntax
->comparison_fn(ldb
, ldb
, &valY
, &valX
);
604 sqlite3_result_int(ctx
, 1);
606 sqlite3_result_int(ctx
, 0);
612 sqlite3_result_int(ctx
, 0);
618 sqlite3_result_int(ctx
, 0);
625 sqlite3_result_error(ctx
, "Value must start with a special operation char (<>~:)!", -1);
630 /* rename a record */
631 static int lsqlite3_safe_rollback(sqlite3
*sqlite
)
637 ret
= sqlite3_exec(sqlite
, "ROLLBACK;", NULL
, NULL
, &errmsg
);
638 if (ret
!= SQLITE_OK
) {
640 printf("lsqlite3_safe_rollback: Error: %s\n", errmsg
);
649 /* return an eid as result */
650 static int lsqlite3_eid_callback(void *result
, int col_num
, char **cols
, char **names
)
652 long long *eid
= (long long *)result
;
654 if (col_num
!= 1) return SQLITE_ABORT
;
655 if (strcasecmp(names
[0], "eid") != 0) return SQLITE_ABORT
;
657 *eid
= atoll(cols
[0]);
662 * add a single set of ldap message values to a ldb_message
664 static int lsqlite3_search_callback(void *result
, int col_num
, char **cols
, char **names
)
666 struct ldb_context
*ldb
;
667 struct lsql_context
*ac
;
668 struct ldb_message
*msg
;
673 ac
= talloc_get_type(result
, struct lsql_context
);
674 ldb
= ldb_module_get_ctx(ac
->module
);
676 /* eid, dn, attr_name, attr_value */
677 if (col_num
!= 4) return SQLITE_ABORT
;
679 eid
= atoll(cols
[0]);
682 msg
= ac
->ares
->message
;
685 if (eid
!= ac
->current_eid
) { /* here begin a new entry */
687 /* call the async callback for the last entry
688 * except the first time */
689 if (ac
->current_eid
!= 0) {
690 msg
= ldb_msg_canonicalize(ldb
, msg
);
691 if (!msg
) return SQLITE_ABORT
;
693 ret
= ldb_module_send_entry(ac
->req
, msg
, NULL
);
694 if (ret
!= LDB_SUCCESS
) {
695 ac
->callback_failed
= true;
701 ac
->ares
= talloc_zero(ac
, struct ldb_reply
);
702 if (!ac
->ares
) return SQLITE_ABORT
;
704 msg
= ldb_msg_new(ac
->ares
);
705 if (!msg
) return SQLITE_ABORT
;
707 ac
->ares
->type
= LDB_REPLY_ENTRY
;
708 ac
->current_eid
= eid
;
711 if (msg
->dn
== NULL
) {
712 msg
->dn
= ldb_dn_new(msg
, ldb
, cols
[1]);
719 for (i
= 0; ac
->attrs
[i
]; i
++) {
720 if (strcasecmp(cols
[2], ac
->attrs
[i
]) == 0) {
725 if (!found
) goto done
;
728 if (ldb_msg_add_string(msg
, cols
[2], cols
[3]) != 0) {
733 ac
->ares
->message
= msg
;
740 * lsqlite3_get_eid_ndn()
742 * These functions are used for the very common case of retrieving an EID value
743 * given a (normalized) DN.
746 static long long lsqlite3_get_eid_ndn(sqlite3
*sqlite
, void *mem_ctx
, const char *norm_dn
)
754 query
= lsqlite3_tprintf(mem_ctx
, "SELECT eid "
756 "WHERE norm_dn = '%q';", norm_dn
);
757 if (query
== NULL
) return -1;
759 ret
= sqlite3_exec(sqlite
, query
, lsqlite3_eid_callback
, &eid
, &errmsg
);
760 if (ret
!= SQLITE_OK
) {
762 printf("lsqlite3_get_eid: Fatal Error: %s\n", errmsg
);
771 static long long lsqlite3_get_eid(struct lsqlite3_private
*lsqlite3
,
774 TALLOC_CTX
*local_ctx
;
778 /* ignore ltdb specials */
779 if (ldb_dn_is_special(dn
)) {
783 /* create a local ctx */
784 local_ctx
= talloc_named(lsqlite3
, 0, "lsqlite3_get_eid local context");
785 if (local_ctx
== NULL
) {
789 cdn
= ldb_dn_alloc_casefold(local_ctx
, dn
);
792 eid
= lsqlite3_get_eid_ndn(lsqlite3
->sqlite
, local_ctx
, cdn
);
795 talloc_free(local_ctx
);
800 * Interface functions referenced by lsqlite3_ops
803 /* search for matching records, by tree */
804 int lsql_search(struct lsql_context
*ctx
)
806 struct ldb_module
*module
= ctx
->module
;
807 struct ldb_request
*req
= ctx
->req
;
808 struct lsqlite3_private
*lsqlite3
;
809 struct ldb_context
*ldb
;
816 ldb
= ldb_module_get_ctx(module
);
817 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
818 struct lsqlite3_private
);
820 if ((( ! ldb_dn_is_valid(req
->op
.search
.base
)) ||
821 ldb_dn_is_null(req
->op
.search
.base
)) &&
822 (req
->op
.search
.scope
== LDB_SCOPE_BASE
||
823 req
->op
.search
.scope
== LDB_SCOPE_ONELEVEL
)) {
824 return LDB_ERR_OPERATIONS_ERROR
;
827 if (req
->op
.search
.base
) {
828 norm_basedn
= ldb_dn_alloc_casefold(ctx
, req
->op
.search
.base
);
829 if (norm_basedn
== NULL
) {
830 return LDB_ERR_OPERATIONS_ERROR
;
832 } else norm_basedn
= talloc_strdup(ctx
, "");
834 /* Convert filter into a series of SQL conditions (constraints) */
835 sqlfilter
= parsetree_to_sql(module
, ctx
, req
->op
.search
.tree
);
837 switch(req
->op
.search
.scope
) {
838 case LDB_SCOPE_DEFAULT
:
839 case LDB_SCOPE_SUBTREE
:
840 if (*norm_basedn
!= '\0') {
841 query
= lsqlite3_tprintf(ctx
,
842 "SELECT entry.eid,\n"
846 " FROM ldb_entry AS entry\n"
848 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
849 " ON av.eid = entry.eid\n"
851 " WHERE entry.eid IN\n"
852 " (SELECT DISTINCT ldb_entry.eid\n"
854 " WHERE (ldb_entry.norm_dn GLOB('*,%q')\n"
855 " OR ldb_entry.norm_dn = '%q')\n"
856 " AND ldb_entry.eid IN\n"
860 " ORDER BY entry.eid ASC;",
865 query
= lsqlite3_tprintf(ctx
,
866 "SELECT entry.eid,\n"
870 " FROM ldb_entry AS entry\n"
872 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
873 " ON av.eid = entry.eid\n"
875 " WHERE entry.eid IN\n"
876 " (SELECT DISTINCT ldb_entry.eid\n"
878 " WHERE ldb_entry.eid IN\n"
882 " ORDER BY entry.eid ASC;",
889 query
= lsqlite3_tprintf(ctx
,
890 "SELECT entry.eid,\n"
894 " FROM ldb_entry AS entry\n"
896 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
897 " ON av.eid = entry.eid\n"
899 " WHERE entry.eid IN\n"
900 " (SELECT DISTINCT ldb_entry.eid\n"
902 " WHERE ldb_entry.norm_dn = '%q'\n"
903 " AND ldb_entry.eid IN\n"
907 " ORDER BY entry.eid ASC;",
912 case LDB_SCOPE_ONELEVEL
:
913 query
= lsqlite3_tprintf(ctx
,
914 "SELECT entry.eid,\n"
918 " FROM ldb_entry AS entry\n"
920 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
921 " ON av.eid = entry.eid\n"
923 " WHERE entry.eid IN\n"
924 " (SELECT DISTINCT ldb_entry.eid\n"
926 " WHERE norm_dn GLOB('*,%q')\n"
927 " AND NOT norm_dn GLOB('*,*,%q')\n"
928 " AND ldb_entry.eid IN\n(%s)\n"
931 " ORDER BY entry.eid ASC;",
939 return LDB_ERR_OPERATIONS_ERROR
;
943 printf ("%s\n", query);
946 ctx
->current_eid
= 0;
947 ctx
->attrs
= req
->op
.search
.attrs
;
950 ldb_request_set_state(req
, LDB_ASYNC_PENDING
);
952 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, lsqlite3_search_callback
, ctx
, &errmsg
);
953 if (ret
!= SQLITE_OK
) {
955 ldb_set_errstring(ldb
, errmsg
);
958 return LDB_ERR_OPERATIONS_ERROR
;
961 /* complete the last message if any */
963 ctx
->ares
->message
= ldb_msg_canonicalize(ldb
, ctx
->ares
->message
);
964 if (ctx
->ares
->message
== NULL
) {
965 return LDB_ERR_OPERATIONS_ERROR
;
968 ret
= ldb_module_send_entry(req
, ctx
->ares
->message
, NULL
);
969 if (ret
!= LDB_SUCCESS
) {
979 static int lsql_add(struct lsql_context
*ctx
)
981 struct ldb_module
*module
= ctx
->module
;
982 struct ldb_request
*req
= ctx
->req
;
983 struct lsqlite3_private
*lsqlite3
;
984 struct ldb_context
*ldb
;
985 struct ldb_message
*msg
= req
->op
.add
.message
;
993 ldb
= ldb_module_get_ctx(module
);
994 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
995 struct lsqlite3_private
);
997 /* See if this is an ltdb special */
998 if (ldb_dn_is_special(msg
->dn
)) {
1001 c = ldb_dn_new(local_ctx, ldb, "@INDEXLIST");
1002 if (ldb_dn_compare(ldb, msg->dn, c) == 0) {
1003 #warning "should we handle indexes somehow ?"
1004 ret = LDB_ERR_UNWILLING_TO_PERFORM;
1008 /* Others return an error */
1009 return LDB_ERR_UNWILLING_TO_PERFORM
;
1012 /* create linearized and normalized dns */
1013 dn
= ldb_dn_alloc_linearized(ctx
, msg
->dn
);
1014 ndn
= ldb_dn_alloc_casefold(ctx
, msg
->dn
);
1015 if (dn
== NULL
|| ndn
== NULL
) {
1016 return LDB_ERR_OPERATIONS_ERROR
;
1019 query
= lsqlite3_tprintf(ctx
,
1021 "INSERT OR ABORT INTO ldb_entry "
1022 "('dn', 'norm_dn') "
1023 "VALUES ('%q', '%q');",
1025 if (query
== NULL
) {
1026 return LDB_ERR_OPERATIONS_ERROR
;
1029 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1030 if (ret
!= SQLITE_OK
) {
1032 ldb_set_errstring(ldb
, errmsg
);
1035 return LDB_ERR_OPERATIONS_ERROR
;
1038 eid
= lsqlite3_get_eid_ndn(lsqlite3
->sqlite
, ctx
, ndn
);
1040 return LDB_ERR_OPERATIONS_ERROR
;
1043 for (i
= 0; i
< msg
->num_elements
; i
++) {
1044 const struct ldb_message_element
*el
= &msg
->elements
[i
];
1045 const struct ldb_schema_attribute
*a
;
1049 /* Get a case-folded copy of the attribute name */
1050 attr
= ldb_attr_casefold(ctx
, el
->name
);
1052 return LDB_ERR_OPERATIONS_ERROR
;
1055 a
= ldb_schema_attribute_by_name(ldb
, el
->name
);
1057 if (el
->num_value
== 0) {
1058 ldb_asprintf_errstring(ldb
, "attribute %s on %s specified, but with 0 values (illegal)",
1059 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1060 return LDB_ERR_CONSTRAINT_VIOLATION
;
1062 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1063 if (el
->num_values
> 1) {
1064 ldb_asprintf_errstring(ldb
, "SINGLE-VALUED attribute %s on %s specified more than once",
1065 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1066 return LDB_ERR_CONSTRAINT_VIOLATION
;
1070 /* For each value of the specified attribute name... */
1071 for (j
= 0; j
< el
->num_values
; j
++) {
1072 struct ldb_val value
;
1075 /* Get a canonicalised copy of the data */
1076 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1077 if (value
.data
== NULL
) {
1078 return LDB_ERR_OPERATIONS_ERROR
;
1081 insert
= lsqlite3_tprintf(ctx
,
1082 "INSERT OR ROLLBACK INTO ldb_attribute_values "
1083 "('eid', 'attr_name', 'norm_attr_name',"
1084 " 'attr_value', 'norm_attr_value') "
1085 "VALUES ('%lld', '%q', '%q', '%q', '%q');",
1086 eid
, el
->name
, attr
,
1087 el
->values
[j
].data
, value
.data
);
1088 if (insert
== NULL
) {
1089 return LDB_ERR_OPERATIONS_ERROR
;
1092 ret
= sqlite3_exec(lsqlite3
->sqlite
, insert
, NULL
, NULL
, &errmsg
);
1093 if (ret
!= SQLITE_OK
) {
1095 ldb_set_errstring(ldb
, errmsg
);
1098 return LDB_ERR_OPERATIONS_ERROR
;
1106 /* modify a record */
1107 static int lsql_modify(struct lsql_context
*ctx
)
1109 struct ldb_module
*module
= ctx
->module
;
1110 struct ldb_request
*req
= ctx
->req
;
1111 struct lsqlite3_private
*lsqlite3
;
1112 struct ldb_context
*ldb
;
1113 struct ldb_message
*msg
= req
->op
.mod
.message
;
1119 ldb
= ldb_module_get_ctx(module
);
1120 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1121 struct lsqlite3_private
);
1123 /* See if this is an ltdb special */
1124 if (ldb_dn_is_special(msg
->dn
)) {
1125 /* Others return an error */
1126 return LDB_ERR_UNWILLING_TO_PERFORM
;
1129 eid
= lsqlite3_get_eid(lsqlite3
, msg
->dn
);
1131 return LDB_ERR_OPERATIONS_ERROR
;
1134 for (i
= 0; i
< msg
->num_elements
; i
++) {
1135 const struct ldb_message_element
*el
= &msg
->elements
[i
];
1136 const struct ldb_schema_attribute
*a
;
1137 int flags
= el
->flags
& LDB_FLAG_MOD_MASK
;
1142 if (ldb_attr_cmp(el
->name
, "distinguishedName") == 0) {
1143 ldb_asprintf_errstring(ldb
, "it is not permitted to perform a modify on 'distinguishedName' (use rename instead): %s",
1144 ldb_dn_get_linearized(msg
->dn
));
1145 return LDB_ERR_CONSTRAINT_VIOLATION
;
1148 /* Get a case-folded copy of the attribute name */
1149 attr
= ldb_attr_casefold(ctx
, el
->name
);
1151 return LDB_ERR_OPERATIONS_ERROR
;
1154 a
= ldb_schema_attribute_by_name(ldb
, el
->name
);
1158 case LDB_FLAG_MOD_REPLACE
:
1160 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1161 if (el
->num_values
> 1) {
1162 ldb_asprintf_errstring(ldb
, "SINGLE-VALUE attribute %s on %s specified more than once",
1163 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1164 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1168 for (j
=0; j
<el
->num_values
; j
++) {
1169 if (ldb_msg_find_val(el
, &el
->values
[j
]) != &el
->values
[j
]) {
1170 ldb_asprintf_errstring(ldb
, "%s: value #%d provided more than once", el
->name
, j
);
1171 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1175 /* remove all attributes before adding the replacements */
1176 mod
= lsqlite3_tprintf(ctx
,
1177 "DELETE FROM ldb_attribute_values "
1178 "WHERE eid = '%lld' "
1179 "AND norm_attr_name = '%q';",
1182 return LDB_ERR_OPERATIONS_ERROR
;
1185 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1186 if (ret
!= SQLITE_OK
) {
1188 ldb_set_errstring(ldb
, errmsg
);
1191 return LDB_ERR_OPERATIONS_ERROR
;
1194 /* MISSING break is INTENTIONAL */
1196 case LDB_FLAG_MOD_ADD
:
1198 if (el
->num_values
== 0) {
1199 ldb_asprintf_errstring(ldb
, "attribute %s on %s specified, but with 0 values (illigal)",
1200 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1201 return LDB_ERR_CONSTRAINT_VIOLATION
;
1204 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1205 if (el
->num_values
> 1) {
1206 ldb_asprintf_errstring(ldb
, "SINGLE-VALUE attribute %s on %s specified more than once",
1207 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1208 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1212 #warning "We should throw an error if no value is provided!"
1213 /* For each value of the specified attribute name... */
1214 for (j
= 0; j
< el
->num_values
; j
++) {
1215 struct ldb_val value
;
1217 /* Get a canonicalised copy of the data */
1218 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1219 if (value
.data
== NULL
) {
1220 return LDB_ERR_OPERATIONS_ERROR
;
1223 mod
= lsqlite3_tprintf(ctx
,
1224 "INSERT OR ROLLBACK INTO ldb_attribute_values "
1225 "('eid', 'attr_name', 'norm_attr_name',"
1226 " 'attr_value', 'norm_attr_value') "
1227 "VALUES ('%lld', '%q', '%q', '%q', '%q');",
1228 eid
, el
->name
, attr
,
1229 el
->values
[j
].data
, value
.data
);
1232 return LDB_ERR_OPERATIONS_ERROR
;
1235 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1236 if (ret
!= SQLITE_OK
) {
1238 ldb_set_errstring(ldb
, errmsg
);
1241 return LDB_ERR_OPERATIONS_ERROR
;
1247 case LDB_FLAG_MOD_DELETE
:
1248 #warning "We should throw an error if the attribute we are trying to delete does not exist!"
1249 if (el
->num_values
== 0) {
1250 mod
= lsqlite3_tprintf(ctx
,
1251 "DELETE FROM ldb_attribute_values "
1252 "WHERE eid = '%lld' "
1253 "AND norm_attr_name = '%q';",
1256 return LDB_ERR_OPERATIONS_ERROR
;
1259 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1260 if (ret
!= SQLITE_OK
) {
1262 ldb_set_errstring(ldb
, errmsg
);
1265 return LDB_ERR_OPERATIONS_ERROR
;
1269 /* For each value of the specified attribute name... */
1270 for (j
= 0; j
< el
->num_values
; j
++) {
1271 struct ldb_val value
;
1273 /* Get a canonicalised copy of the data */
1274 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1275 if (value
.data
== NULL
) {
1276 return LDB_ERR_OPERATIONS_ERROR
;
1279 mod
= lsqlite3_tprintf(ctx
,
1280 "DELETE FROM ldb_attribute_values "
1281 "WHERE eid = '%lld' "
1282 "AND norm_attr_name = '%q' "
1283 "AND norm_attr_value = '%q';",
1284 eid
, attr
, value
.data
);
1287 return LDB_ERR_OPERATIONS_ERROR
;
1290 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1291 if (ret
!= SQLITE_OK
) {
1293 ldb_set_errstring(ldb
, errmsg
);
1296 return LDB_ERR_OPERATIONS_ERROR
;
1307 /* delete a record */
1308 static int lsql_delete(struct lsql_context
*ctx
)
1310 struct ldb_module
*module
= ctx
->module
;
1311 struct ldb_request
*req
= ctx
->req
;
1312 struct lsqlite3_private
*lsqlite3
;
1313 struct ldb_context
*ldb
;
1319 ldb
= ldb_module_get_ctx(module
);
1320 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1321 struct lsqlite3_private
);
1323 eid
= lsqlite3_get_eid(lsqlite3
, req
->op
.del
.dn
);
1325 return LDB_ERR_OPERATIONS_ERROR
;
1328 query
= lsqlite3_tprintf(ctx
,
1330 "DELETE FROM ldb_entry WHERE eid = %lld; "
1331 /* Delete attributes */
1332 "DELETE FROM ldb_attribute_values WHERE eid = %lld; ",
1334 if (query
== NULL
) {
1335 return LDB_ERR_OPERATIONS_ERROR
;
1338 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1339 if (ret
!= SQLITE_OK
) {
1341 ldb_set_errstring(ldb
, errmsg
);
1344 return LDB_ERR_OPERATIONS_ERROR
;
1350 /* rename a record */
1351 static int lsql_rename(struct lsql_context
*ctx
)
1353 struct ldb_module
*module
= ctx
->module
;
1354 struct ldb_request
*req
= ctx
->req
;
1355 struct lsqlite3_private
*lsqlite3
;
1356 struct ldb_context
*ldb
;
1357 char *new_dn
, *new_cdn
, *old_cdn
;
1362 ldb
= ldb_module_get_ctx(module
);
1363 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1364 struct lsqlite3_private
);
1366 /* create linearized and normalized dns */
1367 old_cdn
= ldb_dn_alloc_casefold(ctx
, req
->op
.rename
.olddn
);
1368 new_cdn
= ldb_dn_alloc_casefold(ctx
, req
->op
.rename
.newdn
);
1369 new_dn
= ldb_dn_alloc_linearized(ctx
, req
->op
.rename
.newdn
);
1370 if (old_cdn
== NULL
|| new_cdn
== NULL
|| new_dn
== NULL
) {
1371 return LDB_ERR_OPERATIONS_ERROR
;
1374 /* build the SQL query */
1375 query
= lsqlite3_tprintf(ctx
,
1376 "UPDATE ldb_entry SET dn = '%q', norm_dn = '%q' "
1377 "WHERE norm_dn = '%q';",
1378 new_dn
, new_cdn
, old_cdn
);
1379 if (query
== NULL
) {
1380 return LDB_ERR_OPERATIONS_ERROR
;
1384 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1385 if (ret
!= SQLITE_OK
) {
1387 ldb_set_errstring(ldb
, errmsg
);
1390 return LDB_ERR_OPERATIONS_ERROR
;
1396 static int lsql_start_trans(struct ldb_module
* module
)
1400 struct lsqlite3_private
*lsqlite3
;
1402 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1403 struct lsqlite3_private
);
1405 if (lsqlite3
->trans_count
== 0) {
1406 ret
= sqlite3_exec(lsqlite3
->sqlite
, "BEGIN IMMEDIATE;", NULL
, NULL
, &errmsg
);
1407 if (ret
!= SQLITE_OK
) {
1409 printf("lsqlite3_start_trans: error: %s\n", errmsg
);
1416 lsqlite3
->trans_count
++;
1421 static int lsql_end_trans(struct ldb_module
*module
)
1425 struct lsqlite3_private
*lsqlite3
;
1427 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1428 struct lsqlite3_private
);
1430 if (lsqlite3
->trans_count
> 0) {
1431 lsqlite3
->trans_count
--;
1434 if (lsqlite3
->trans_count
== 0) {
1435 ret
= sqlite3_exec(lsqlite3
->sqlite
, "COMMIT;", NULL
, NULL
, &errmsg
);
1436 if (ret
!= SQLITE_OK
) {
1438 printf("lsqlite3_end_trans: error: %s\n", errmsg
);
1448 static int lsql_del_trans(struct ldb_module
*module
)
1450 struct lsqlite3_private
*lsqlite3
;
1452 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1453 struct lsqlite3_private
);
1455 if (lsqlite3
->trans_count
> 0) {
1456 lsqlite3
->trans_count
--;
1459 if (lsqlite3
->trans_count
== 0) {
1460 return lsqlite3_safe_rollback(lsqlite3
->sqlite
);
1466 static int destructor(struct lsqlite3_private
*lsqlite3
)
1468 if (lsqlite3
->sqlite
) {
1469 sqlite3_close(lsqlite3
->sqlite
);
1474 static void lsql_request_done(struct lsql_context
*ctx
, int error
)
1476 struct ldb_context
*ldb
;
1477 struct ldb_request
*req
;
1478 struct ldb_reply
*ares
;
1480 ldb
= ldb_module_get_ctx(ctx
->module
);
1483 /* if we already returned an error just return */
1484 if (ldb_request_get_status(req
) != LDB_SUCCESS
) {
1488 ares
= talloc_zero(req
, struct ldb_reply
);
1491 req
->callback(req
, NULL
);
1494 ares
->type
= LDB_REPLY_DONE
;
1495 ares
->error
= error
;
1497 req
->callback(req
, ares
);
1500 static void lsql_timeout(struct tevent_context
*ev
,
1501 struct tevent_timer
*te
,
1505 struct lsql_context
*ctx
;
1506 ctx
= talloc_get_type(private_data
, struct lsql_context
);
1508 lsql_request_done(ctx
, LDB_ERR_TIME_LIMIT_EXCEEDED
);
1511 static void lsql_callback(struct tevent_context
*ev
,
1512 struct tevent_timer
*te
,
1516 struct lsql_context
*ctx
;
1519 ctx
= talloc_get_type(private_data
, struct lsql_context
);
1521 switch (ctx
->req
->operation
) {
1523 ret
= lsql_search(ctx
);
1526 ret
= lsql_add(ctx
);
1529 ret
= lsql_modify(ctx
);
1532 ret
= lsql_delete(ctx
);
1535 ret
= lsql_rename(ctx
);
1539 ret = lsql_extended(ctx);
1543 /* no other op supported */
1544 ret
= LDB_ERR_UNWILLING_TO_PERFORM
;
1547 if (!ctx
->callback_failed
) {
1548 /* Once we are done, we do not need timeout events */
1549 talloc_free(ctx
->timeout_event
);
1550 lsql_request_done(ctx
, ret
);
1554 static int lsql_handle_request(struct ldb_module
*module
, struct ldb_request
*req
)
1556 struct ldb_context
*ldb
;
1557 struct tevent_context
*ev
;
1558 struct lsql_context
*ac
;
1559 struct tevent_timer
*te
;
1562 if (check_critical_controls(req
->controls
)) {
1563 return LDB_ERR_UNSUPPORTED_CRITICAL_EXTENSION
;
1566 if (req
->starttime
== 0 || req
->timeout
== 0) {
1567 ldb_set_errstring(ldb
, "Invalid timeout settings");
1568 return LDB_ERR_TIME_LIMIT_EXCEEDED
;
1571 ldb
= ldb_module_get_ctx(module
);
1572 ev
= ldb_get_event_context(ldb
);
1574 ac
= talloc_zero(req
, struct lsql_context
);
1576 ldb_set_errstring(ldb
, "Out of Memory");
1577 return LDB_ERR_OPERATIONS_ERROR
;
1580 ac
->module
= module
;
1585 te
= tevent_add_timer(ev
, ac
, tv
, lsql_callback
, ac
);
1587 return LDB_ERR_OPERATIONS_ERROR
;
1590 tv
.tv_sec
= req
->starttime
+ req
->timeout
;
1591 ac
->timeout_event
= tevent_add_timer(ev
, ac
, tv
, lsql_timeout
, ac
);
1592 if (NULL
== ac
->timeout_event
) {
1593 return LDB_ERR_OPERATIONS_ERROR
;
1600 * Table of operations for the sqlite3 backend
1602 static const struct ldb_module_ops lsqlite3_ops
= {
1604 .search
= lsql_handle_request
,
1605 .add
= lsql_handle_request
,
1606 .modify
= lsql_handle_request
,
1607 .del
= lsql_handle_request
,
1608 .rename
= lsql_handle_request
,
1609 .extended
= lsql_handle_request
,
1610 .start_transaction
= lsql_start_trans
,
1611 .end_transaction
= lsql_end_trans
,
1612 .del_transaction
= lsql_del_trans
,
1619 static int initialize(struct lsqlite3_private
*lsqlite3
,
1620 struct ldb_context
*ldb
, const char *url
, int flags
)
1622 TALLOC_CTX
*local_ctx
;
1629 /* create a local ctx */
1630 local_ctx
= talloc_named(lsqlite3
, 0, "lsqlite3_rename local context");
1631 if (local_ctx
== NULL
) {
1635 schema
= lsqlite3_tprintf(local_ctx
,
1638 "CREATE TABLE ldb_info AS "
1639 " SELECT 'LDB' AS database_type,"
1640 " '1.0' AS version;"
1643 * The entry table holds the information about an entry.
1644 * This table is used to obtain the EID of the entry and to
1645 * support scope=one and scope=base. The parent and child
1646 * table is included in the entry table since all the other
1647 * attributes are dependent on EID.
1649 "CREATE TABLE ldb_entry "
1651 " eid INTEGER PRIMARY KEY AUTOINCREMENT,"
1652 " dn TEXT UNIQUE NOT NULL,"
1653 " norm_dn TEXT UNIQUE NOT NULL"
1657 "CREATE TABLE ldb_object_classes"
1659 " class_name TEXT PRIMARY KEY,"
1660 " parent_class_name TEXT,"
1661 " tree_key TEXT UNIQUE,"
1662 " max_child_num INTEGER DEFAULT 0"
1666 * We keep a full listing of attribute/value pairs here
1668 "CREATE TABLE ldb_attribute_values"
1670 " eid INTEGER REFERENCES ldb_entry,"
1672 " norm_attr_name TEXT,"
1674 " norm_attr_value TEXT "
1681 "CREATE INDEX ldb_attribute_values_eid_idx "
1682 " ON ldb_attribute_values (eid);"
1684 "CREATE INDEX ldb_attribute_values_name_value_idx "
1685 " ON ldb_attribute_values (attr_name, norm_attr_value);"
1693 "CREATE TRIGGER ldb_object_classes_insert_tr"
1695 " ON ldb_object_classes"
1698 " UPDATE ldb_object_classes"
1699 " SET tree_key = COALESCE(tree_key, "
1701 " SELECT tree_key || "
1702 " (SELECT base160(max_child_num + 1)"
1703 " FROM ldb_object_classes"
1704 " WHERE class_name = "
1705 " new.parent_class_name)"
1706 " FROM ldb_object_classes "
1707 " WHERE class_name = new.parent_class_name "
1709 " UPDATE ldb_object_classes "
1710 " SET max_child_num = max_child_num + 1"
1711 " WHERE class_name = new.parent_class_name;"
1715 * Table initialization
1718 "INSERT INTO ldb_object_classes "
1719 " (class_name, tree_key) "
1721 " ('TOP', '0001');");
1723 /* Skip protocol indicator of url */
1724 if (strncmp(url
, "sqlite3://", 10) != 0) {
1725 return SQLITE_MISUSE
;
1728 /* Update pointer to just after the protocol indicator */
1731 /* Try to open the (possibly empty/non-existent) database */
1732 if ((ret
= sqlite3_open(url
, &lsqlite3
->sqlite
)) != SQLITE_OK
) {
1736 /* In case this is a new database, enable auto_vacuum */
1737 ret
= sqlite3_exec(lsqlite3
->sqlite
, "PRAGMA auto_vacuum = 1;", NULL
, NULL
, &errmsg
);
1738 if (ret
!= SQLITE_OK
) {
1740 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1746 if (flags
& LDB_FLG_NOSYNC
) {
1748 ret
= sqlite3_exec(lsqlite3
->sqlite
, "PRAGMA synchronous = OFF;", NULL
, NULL
, &errmsg
);
1749 if (ret
!= SQLITE_OK
) {
1751 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1760 /* Establish a busy timeout of 30 seconds */
1761 if ((ret
= sqlite3_busy_timeout(lsqlite3
->sqlite
,
1762 30000)) != SQLITE_OK
) {
1766 /* Create a function, callable from sql, to increment a tree_key */
1768 sqlite3_create_function(lsqlite3
->sqlite
,/* handle */
1769 "base160_next", /* function name */
1770 1, /* number of args */
1771 SQLITE_ANY
, /* preferred text type */
1772 NULL
, /* user data */
1773 base160next_sql
, /* called func */
1774 NULL
, /* step func */
1775 NULL
/* final func */
1780 /* Create a function, callable from sql, to convert int to base160 */
1782 sqlite3_create_function(lsqlite3
->sqlite
,/* handle */
1783 "base160", /* function name */
1784 1, /* number of args */
1785 SQLITE_ANY
, /* preferred text type */
1786 NULL
, /* user data */
1787 base160_sql
, /* called func */
1788 NULL
, /* step func */
1789 NULL
/* final func */
1794 /* Create a function, callable from sql, to perform various comparisons */
1796 sqlite3_create_function(lsqlite3
->sqlite
, /* handle */
1797 "ldap_compare", /* function name */
1798 4, /* number of args */
1799 SQLITE_ANY
, /* preferred text type */
1800 ldb
, /* user data */
1801 lsqlite3_compare
, /* called func */
1802 NULL
, /* step func */
1803 NULL
/* final func */
1808 /* Begin a transaction */
1809 ret
= sqlite3_exec(lsqlite3
->sqlite
, "BEGIN EXCLUSIVE;", NULL
, NULL
, &errmsg
);
1810 if (ret
!= SQLITE_OK
) {
1812 printf("lsqlite3: initialization error: %s\n", errmsg
);
1819 /* Determine if this is a new database. No tables means it is. */
1820 if (query_int(lsqlite3
,
1823 " FROM sqlite_master\n"
1824 " WHERE type = 'table';") != 0) {
1828 if (queryInt
== 0) {
1830 * Create the database schema
1832 ret
= sqlite3_exec(lsqlite3
->sqlite
, schema
, NULL
, NULL
, &errmsg
);
1833 if (ret
!= SQLITE_OK
) {
1835 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1842 * Ensure that the database we opened is one of ours
1844 if (query_int(lsqlite3
,
1847 " (SELECT COUNT(*) = 2"
1848 " FROM sqlite_master "
1849 " WHERE type = 'table' "
1853 " 'ldb_object_classes' "
1859 " WHERE database_type = 'LDB' "
1860 " AND version = '1.0'"
1864 /* It's not one that we created. See ya! */
1869 /* Commit the transaction */
1870 ret
= sqlite3_exec(lsqlite3
->sqlite
, "COMMIT;", NULL
, NULL
, &errmsg
);
1871 if (ret
!= SQLITE_OK
) {
1873 printf("lsqlite3: iniialization error: %s\n", errmsg
);
1882 if (rollback
) lsqlite3_safe_rollback(lsqlite3
->sqlite
);
1883 sqlite3_close(lsqlite3
->sqlite
);
1888 * connect to the database
1890 static int lsqlite3_connect(struct ldb_context
*ldb
,
1893 const char *options
[],
1894 struct ldb_module
**_module
)
1896 struct ldb_module
*module
;
1897 struct lsqlite3_private
*lsqlite3
;
1901 module
= ldb_module_new(ldb
, ldb
, "ldb_sqlite3 backend", &lsqlite3_ops
);
1902 if (!module
) return LDB_ERR_OPERATIONS_ERROR
;
1904 lsqlite3
= talloc(module
, struct lsqlite3_private
);
1909 lsqlite3
->sqlite
= NULL
;
1910 lsqlite3
->options
= NULL
;
1911 lsqlite3
->trans_count
= 0;
1913 ret
= initialize(lsqlite3
, ldb
, url
, flags
);
1914 if (ret
!= SQLITE_OK
) {
1918 talloc_set_destructor(lsqlite3
, destructor
);
1920 ldb_module_set_private(module
, lsqlite3
);
1924 * take a copy of the options array, so we don't have to rely
1925 * on the caller keeping it around (it might be dynamic)
1927 for (i
=0;options
[i
];i
++) ;
1929 lsqlite3
->options
= talloc_array(lsqlite3
, char *, i
+1);
1930 if (!lsqlite3
->options
) {
1934 for (i
=0;options
[i
];i
++) {
1936 lsqlite3
->options
[i
+1] = NULL
;
1937 lsqlite3
->options
[i
] =
1938 talloc_strdup(lsqlite3
->options
, options
[i
]);
1939 if (!lsqlite3
->options
[i
]) {
1949 if (lsqlite3
&& lsqlite3
->sqlite
!= NULL
) {
1950 (void) sqlite3_close(lsqlite3
->sqlite
);
1952 talloc_free(lsqlite3
);
1953 return LDB_ERR_OPERATIONS_ERROR
;
1956 const struct ldb_backend_ops ldb_sqlite3_backend_ops
= {
1958 .connect_fn
= lsqlite3_connect