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
;
672 ac
= talloc_get_type(result
, struct lsql_context
);
673 ldb
= ldb_module_get_ctx(ac
->module
);
675 /* eid, dn, attr_name, attr_value */
676 if (col_num
!= 4) return SQLITE_ABORT
;
678 eid
= atoll(cols
[0]);
681 msg
= ac
->ares
->message
;
684 if (eid
!= ac
->current_eid
) { /* here begin a new entry */
686 /* call the async callback for the last entry
687 * except the first time */
688 if (ac
->current_eid
!= 0) {
689 msg
= ldb_msg_canonicalize(ldb
, msg
);
690 if (!msg
) return SQLITE_ABORT
;
692 ret
= ldb_module_send_entry(ac
->req
, msg
, NULL
);
693 if (ret
!= LDB_SUCCESS
) {
694 ac
->callback_failed
= true;
700 ac
->ares
= talloc_zero(ac
, struct ldb_reply
);
701 if (!ac
->ares
) return SQLITE_ABORT
;
703 msg
= ldb_msg_new(ac
->ares
);
704 if (!msg
) return SQLITE_ABORT
;
706 ac
->ares
->type
= LDB_REPLY_ENTRY
;
707 ac
->current_eid
= eid
;
710 if (msg
->dn
== NULL
) {
711 msg
->dn
= ldb_dn_new(msg
, ldb
, cols
[1]);
718 for (i
= 0; ac
->attrs
[i
]; i
++) {
719 if (strcasecmp(cols
[2], ac
->attrs
[i
]) == 0) {
724 if (!found
) goto done
;
727 if (ldb_msg_add_string(msg
, cols
[2], cols
[3]) != 0) {
732 ac
->ares
->message
= msg
;
739 * lsqlite3_get_eid_ndn()
741 * These functions are used for the very common case of retrieving an EID value
742 * given a (normalized) DN.
745 static long long lsqlite3_get_eid_ndn(sqlite3
*sqlite
, void *mem_ctx
, const char *norm_dn
)
753 query
= lsqlite3_tprintf(mem_ctx
, "SELECT eid "
755 "WHERE norm_dn = '%q';", norm_dn
);
756 if (query
== NULL
) return -1;
758 ret
= sqlite3_exec(sqlite
, query
, lsqlite3_eid_callback
, &eid
, &errmsg
);
759 if (ret
!= SQLITE_OK
) {
761 printf("lsqlite3_get_eid: Fatal Error: %s\n", errmsg
);
770 static long long lsqlite3_get_eid(struct lsqlite3_private
*lsqlite3
,
773 TALLOC_CTX
*local_ctx
;
777 /* ignore ltdb specials */
778 if (ldb_dn_is_special(dn
)) {
782 /* create a local ctx */
783 local_ctx
= talloc_named(lsqlite3
, 0, "lsqlite3_get_eid local context");
784 if (local_ctx
== NULL
) {
788 cdn
= ldb_dn_alloc_casefold(local_ctx
, dn
);
791 eid
= lsqlite3_get_eid_ndn(lsqlite3
->sqlite
, local_ctx
, cdn
);
794 talloc_free(local_ctx
);
799 * Interface functions referenced by lsqlite3_ops
802 /* search for matching records, by tree */
803 int lsql_search(struct lsql_context
*ctx
)
805 struct ldb_module
*module
= ctx
->module
;
806 struct ldb_request
*req
= ctx
->req
;
807 struct lsqlite3_private
*lsqlite3
;
808 struct ldb_context
*ldb
;
815 ldb
= ldb_module_get_ctx(module
);
816 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
817 struct lsqlite3_private
);
819 if ((( ! ldb_dn_is_valid(req
->op
.search
.base
)) ||
820 ldb_dn_is_null(req
->op
.search
.base
)) &&
821 (req
->op
.search
.scope
== LDB_SCOPE_BASE
||
822 req
->op
.search
.scope
== LDB_SCOPE_ONELEVEL
)) {
823 return LDB_ERR_OPERATIONS_ERROR
;
826 if (req
->op
.search
.base
) {
827 norm_basedn
= ldb_dn_alloc_casefold(ctx
, req
->op
.search
.base
);
828 if (norm_basedn
== NULL
) {
829 return LDB_ERR_OPERATIONS_ERROR
;
831 } else norm_basedn
= talloc_strdup(ctx
, "");
833 /* Convert filter into a series of SQL conditions (constraints) */
834 sqlfilter
= parsetree_to_sql(module
, ctx
, req
->op
.search
.tree
);
836 switch(req
->op
.search
.scope
) {
837 case LDB_SCOPE_DEFAULT
:
838 case LDB_SCOPE_SUBTREE
:
839 if (*norm_basedn
!= '\0') {
840 query
= lsqlite3_tprintf(ctx
,
841 "SELECT entry.eid,\n"
845 " FROM ldb_entry AS entry\n"
847 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
848 " ON av.eid = entry.eid\n"
850 " WHERE entry.eid IN\n"
851 " (SELECT DISTINCT ldb_entry.eid\n"
853 " WHERE (ldb_entry.norm_dn GLOB('*,%q')\n"
854 " OR ldb_entry.norm_dn = '%q')\n"
855 " AND ldb_entry.eid IN\n"
859 " ORDER BY entry.eid ASC;",
864 query
= lsqlite3_tprintf(ctx
,
865 "SELECT entry.eid,\n"
869 " FROM ldb_entry AS entry\n"
871 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
872 " ON av.eid = entry.eid\n"
874 " WHERE entry.eid IN\n"
875 " (SELECT DISTINCT ldb_entry.eid\n"
877 " WHERE ldb_entry.eid IN\n"
881 " ORDER BY entry.eid ASC;",
888 query
= lsqlite3_tprintf(ctx
,
889 "SELECT entry.eid,\n"
893 " FROM ldb_entry AS entry\n"
895 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
896 " ON av.eid = entry.eid\n"
898 " WHERE entry.eid IN\n"
899 " (SELECT DISTINCT ldb_entry.eid\n"
901 " WHERE ldb_entry.norm_dn = '%q'\n"
902 " AND ldb_entry.eid IN\n"
906 " ORDER BY entry.eid ASC;",
911 case LDB_SCOPE_ONELEVEL
:
912 query
= lsqlite3_tprintf(ctx
,
913 "SELECT entry.eid,\n"
917 " FROM ldb_entry AS entry\n"
919 " LEFT OUTER JOIN ldb_attribute_values AS av\n"
920 " ON av.eid = entry.eid\n"
922 " WHERE entry.eid IN\n"
923 " (SELECT DISTINCT ldb_entry.eid\n"
925 " WHERE norm_dn GLOB('*,%q')\n"
926 " AND NOT norm_dn GLOB('*,*,%q')\n"
927 " AND ldb_entry.eid IN\n(%s)\n"
930 " ORDER BY entry.eid ASC;",
938 return LDB_ERR_OPERATIONS_ERROR
;
942 printf ("%s\n", query);
945 ctx
->current_eid
= 0;
946 ctx
->attrs
= req
->op
.search
.attrs
;
949 ldb_request_set_state(req
, LDB_ASYNC_PENDING
);
951 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, lsqlite3_search_callback
, ctx
, &errmsg
);
952 if (ret
!= SQLITE_OK
) {
954 ldb_set_errstring(ldb
, errmsg
);
957 return LDB_ERR_OPERATIONS_ERROR
;
960 /* complete the last message if any */
962 ctx
->ares
->message
= ldb_msg_canonicalize(ldb
, ctx
->ares
->message
);
963 if (ctx
->ares
->message
== NULL
) {
964 return LDB_ERR_OPERATIONS_ERROR
;
967 ret
= ldb_module_send_entry(req
, ctx
->ares
->message
, NULL
);
968 if (ret
!= LDB_SUCCESS
) {
978 static int lsql_add(struct lsql_context
*ctx
)
980 struct ldb_module
*module
= ctx
->module
;
981 struct ldb_request
*req
= ctx
->req
;
982 struct lsqlite3_private
*lsqlite3
;
983 struct ldb_context
*ldb
;
984 struct ldb_message
*msg
= req
->op
.add
.message
;
992 ldb
= ldb_module_get_ctx(module
);
993 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
994 struct lsqlite3_private
);
996 /* See if this is an ltdb special */
997 if (ldb_dn_is_special(msg
->dn
)) {
1000 c = ldb_dn_new(local_ctx, ldb, "@INDEXLIST");
1001 if (ldb_dn_compare(ldb, msg->dn, c) == 0) {
1002 #warning "should we handle indexes somehow ?"
1003 ret = LDB_ERR_UNWILLING_TO_PERFORM;
1007 /* Others return an error */
1008 return LDB_ERR_UNWILLING_TO_PERFORM
;
1011 /* create linearized and normalized dns */
1012 dn
= ldb_dn_alloc_linearized(ctx
, msg
->dn
);
1013 ndn
= ldb_dn_alloc_casefold(ctx
, msg
->dn
);
1014 if (dn
== NULL
|| ndn
== NULL
) {
1015 return LDB_ERR_OPERATIONS_ERROR
;
1018 query
= lsqlite3_tprintf(ctx
,
1020 "INSERT OR ABORT INTO ldb_entry "
1021 "('dn', 'norm_dn') "
1022 "VALUES ('%q', '%q');",
1024 if (query
== NULL
) {
1025 return LDB_ERR_OPERATIONS_ERROR
;
1028 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1029 if (ret
!= SQLITE_OK
) {
1031 ldb_set_errstring(ldb
, errmsg
);
1034 return LDB_ERR_OPERATIONS_ERROR
;
1037 eid
= lsqlite3_get_eid_ndn(lsqlite3
->sqlite
, ctx
, ndn
);
1039 return LDB_ERR_OPERATIONS_ERROR
;
1042 for (i
= 0; i
< msg
->num_elements
; i
++) {
1043 const struct ldb_message_element
*el
= &msg
->elements
[i
];
1044 const struct ldb_schema_attribute
*a
;
1048 /* Get a case-folded copy of the attribute name */
1049 attr
= ldb_attr_casefold(ctx
, el
->name
);
1051 return LDB_ERR_OPERATIONS_ERROR
;
1054 a
= ldb_schema_attribute_by_name(ldb
, el
->name
);
1056 if (el
->num_value
== 0) {
1057 ldb_asprintf_errstring(ldb
, "attribute %s on %s specified, but with 0 values (illegal)",
1058 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1059 return LDB_ERR_CONSTRAINT_VIOLATION
;
1061 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1062 if (el
->num_values
> 1) {
1063 ldb_asprintf_errstring(ldb
, "SINGLE-VALUED attribute %s on %s specified more than once",
1064 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1065 return LDB_ERR_CONSTRAINT_VIOLATION
;
1069 /* For each value of the specified attribute name... */
1070 for (j
= 0; j
< el
->num_values
; j
++) {
1071 struct ldb_val value
;
1074 /* Get a canonicalised copy of the data */
1075 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1076 if (value
.data
== NULL
) {
1077 return LDB_ERR_OPERATIONS_ERROR
;
1080 insert
= lsqlite3_tprintf(ctx
,
1081 "INSERT OR ROLLBACK INTO ldb_attribute_values "
1082 "('eid', 'attr_name', 'norm_attr_name',"
1083 " 'attr_value', 'norm_attr_value') "
1084 "VALUES ('%lld', '%q', '%q', '%q', '%q');",
1085 eid
, el
->name
, attr
,
1086 el
->values
[j
].data
, value
.data
);
1087 if (insert
== NULL
) {
1088 return LDB_ERR_OPERATIONS_ERROR
;
1091 ret
= sqlite3_exec(lsqlite3
->sqlite
, insert
, NULL
, NULL
, &errmsg
);
1092 if (ret
!= SQLITE_OK
) {
1094 ldb_set_errstring(ldb
, errmsg
);
1097 return LDB_ERR_OPERATIONS_ERROR
;
1105 /* modify a record */
1106 static int lsql_modify(struct lsql_context
*ctx
)
1108 struct ldb_module
*module
= ctx
->module
;
1109 struct ldb_request
*req
= ctx
->req
;
1110 struct lsqlite3_private
*lsqlite3
;
1111 struct ldb_context
*ldb
;
1112 struct ldb_message
*msg
= req
->op
.mod
.message
;
1118 ldb
= ldb_module_get_ctx(module
);
1119 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1120 struct lsqlite3_private
);
1122 /* See if this is an ltdb special */
1123 if (ldb_dn_is_special(msg
->dn
)) {
1124 /* Others return an error */
1125 return LDB_ERR_UNWILLING_TO_PERFORM
;
1128 eid
= lsqlite3_get_eid(lsqlite3
, msg
->dn
);
1130 return LDB_ERR_OPERATIONS_ERROR
;
1133 for (i
= 0; i
< msg
->num_elements
; i
++) {
1134 const struct ldb_message_element
*el
= &msg
->elements
[i
];
1135 const struct ldb_schema_attribute
*a
;
1136 int flags
= el
->flags
& LDB_FLAG_MOD_MASK
;
1141 if (ldb_attr_cmp(el
->name
, "distinguishedName") == 0) {
1142 ldb_asprintf_errstring(ldb
, "it is not permitted to perform a modify on 'distinguishedName' (use rename instead): %s",
1143 ldb_dn_get_linearized(msg
->dn
));
1144 return LDB_ERR_CONSTRAINT_VIOLATION
;
1147 /* Get a case-folded copy of the attribute name */
1148 attr
= ldb_attr_casefold(ctx
, el
->name
);
1150 return LDB_ERR_OPERATIONS_ERROR
;
1153 a
= ldb_schema_attribute_by_name(ldb
, el
->name
);
1157 case LDB_FLAG_MOD_REPLACE
:
1159 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1160 if (el
->num_values
> 1) {
1161 ldb_asprintf_errstring(ldb
, "SINGLE-VALUE attribute %s on %s specified more than once",
1162 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1163 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1167 for (j
=0; j
<el
->num_values
; j
++) {
1168 if (ldb_msg_find_val(el
, &el
->values
[j
]) != &el
->values
[j
]) {
1169 ldb_asprintf_errstring(ldb
, "%s: value #%d provided more than once", el
->name
, j
);
1170 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1174 /* remove all attributes before adding the replacements */
1175 mod
= lsqlite3_tprintf(ctx
,
1176 "DELETE FROM ldb_attribute_values "
1177 "WHERE eid = '%lld' "
1178 "AND norm_attr_name = '%q';",
1181 return LDB_ERR_OPERATIONS_ERROR
;
1184 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1185 if (ret
!= SQLITE_OK
) {
1187 ldb_set_errstring(ldb
, errmsg
);
1190 return LDB_ERR_OPERATIONS_ERROR
;
1193 /* MISSING break is INTENTIONAL */
1195 case LDB_FLAG_MOD_ADD
:
1197 if (el
->num_values
== 0) {
1198 ldb_asprintf_errstring(ldb
, "attribute %s on %s specified, but with 0 values (illigal)",
1199 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1200 return LDB_ERR_CONSTRAINT_VIOLATION
;
1203 if (a
&& a
->flags
& LDB_ATTR_FLAG_SINGLE_VALUE
) {
1204 if (el
->num_values
> 1) {
1205 ldb_asprintf_errstring(ldb
, "SINGLE-VALUE attribute %s on %s specified more than once",
1206 el
->name
, ldb_dn_get_linearized(msg
->dn
));
1207 return LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS
;
1211 #warning "We should throw an error if no value is provided!"
1212 /* For each value of the specified attribute name... */
1213 for (j
= 0; j
< el
->num_values
; j
++) {
1214 struct ldb_val value
;
1216 /* Get a canonicalised copy of the data */
1217 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1218 if (value
.data
== NULL
) {
1219 return LDB_ERR_OPERATIONS_ERROR
;
1222 mod
= lsqlite3_tprintf(ctx
,
1223 "INSERT OR ROLLBACK INTO ldb_attribute_values "
1224 "('eid', 'attr_name', 'norm_attr_name',"
1225 " 'attr_value', 'norm_attr_value') "
1226 "VALUES ('%lld', '%q', '%q', '%q', '%q');",
1227 eid
, el
->name
, attr
,
1228 el
->values
[j
].data
, value
.data
);
1231 return LDB_ERR_OPERATIONS_ERROR
;
1234 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1235 if (ret
!= SQLITE_OK
) {
1237 ldb_set_errstring(ldb
, errmsg
);
1240 return LDB_ERR_OPERATIONS_ERROR
;
1246 case LDB_FLAG_MOD_DELETE
:
1247 #warning "We should throw an error if the attribute we are trying to delete does not exist!"
1248 if (el
->num_values
== 0) {
1249 mod
= lsqlite3_tprintf(ctx
,
1250 "DELETE FROM ldb_attribute_values "
1251 "WHERE eid = '%lld' "
1252 "AND norm_attr_name = '%q';",
1255 return LDB_ERR_OPERATIONS_ERROR
;
1258 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1259 if (ret
!= SQLITE_OK
) {
1261 ldb_set_errstring(ldb
, errmsg
);
1264 return LDB_ERR_OPERATIONS_ERROR
;
1268 /* For each value of the specified attribute name... */
1269 for (j
= 0; j
< el
->num_values
; j
++) {
1270 struct ldb_val value
;
1272 /* Get a canonicalised copy of the data */
1273 a
->syntax
->canonicalise_fn(ldb
, ctx
, &(el
->values
[j
]), &value
);
1274 if (value
.data
== NULL
) {
1275 return LDB_ERR_OPERATIONS_ERROR
;
1278 mod
= lsqlite3_tprintf(ctx
,
1279 "DELETE FROM ldb_attribute_values "
1280 "WHERE eid = '%lld' "
1281 "AND norm_attr_name = '%q' "
1282 "AND norm_attr_value = '%q';",
1283 eid
, attr
, value
.data
);
1286 return LDB_ERR_OPERATIONS_ERROR
;
1289 ret
= sqlite3_exec(lsqlite3
->sqlite
, mod
, NULL
, NULL
, &errmsg
);
1290 if (ret
!= SQLITE_OK
) {
1292 ldb_set_errstring(ldb
, errmsg
);
1295 return LDB_ERR_OPERATIONS_ERROR
;
1306 /* delete a record */
1307 static int lsql_delete(struct lsql_context
*ctx
)
1309 struct ldb_module
*module
= ctx
->module
;
1310 struct ldb_request
*req
= ctx
->req
;
1311 struct lsqlite3_private
*lsqlite3
;
1312 struct ldb_context
*ldb
;
1318 ldb
= ldb_module_get_ctx(module
);
1319 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1320 struct lsqlite3_private
);
1322 eid
= lsqlite3_get_eid(lsqlite3
, req
->op
.del
.dn
);
1324 return LDB_ERR_OPERATIONS_ERROR
;
1327 query
= lsqlite3_tprintf(ctx
,
1329 "DELETE FROM ldb_entry WHERE eid = %lld; "
1330 /* Delete attributes */
1331 "DELETE FROM ldb_attribute_values WHERE eid = %lld; ",
1333 if (query
== NULL
) {
1334 return LDB_ERR_OPERATIONS_ERROR
;
1337 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1338 if (ret
!= SQLITE_OK
) {
1340 ldb_set_errstring(ldb
, errmsg
);
1343 return LDB_ERR_OPERATIONS_ERROR
;
1349 /* rename a record */
1350 static int lsql_rename(struct lsql_context
*ctx
)
1352 struct ldb_module
*module
= ctx
->module
;
1353 struct ldb_request
*req
= ctx
->req
;
1354 struct lsqlite3_private
*lsqlite3
;
1355 struct ldb_context
*ldb
;
1356 char *new_dn
, *new_cdn
, *old_cdn
;
1361 ldb
= ldb_module_get_ctx(module
);
1362 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1363 struct lsqlite3_private
);
1365 /* create linearized and normalized dns */
1366 old_cdn
= ldb_dn_alloc_casefold(ctx
, req
->op
.rename
.olddn
);
1367 new_cdn
= ldb_dn_alloc_casefold(ctx
, req
->op
.rename
.newdn
);
1368 new_dn
= ldb_dn_alloc_linearized(ctx
, req
->op
.rename
.newdn
);
1369 if (old_cdn
== NULL
|| new_cdn
== NULL
|| new_dn
== NULL
) {
1370 return LDB_ERR_OPERATIONS_ERROR
;
1373 /* build the SQL query */
1374 query
= lsqlite3_tprintf(ctx
,
1375 "UPDATE ldb_entry SET dn = '%q', norm_dn = '%q' "
1376 "WHERE norm_dn = '%q';",
1377 new_dn
, new_cdn
, old_cdn
);
1378 if (query
== NULL
) {
1379 return LDB_ERR_OPERATIONS_ERROR
;
1383 ret
= sqlite3_exec(lsqlite3
->sqlite
, query
, NULL
, NULL
, &errmsg
);
1384 if (ret
!= SQLITE_OK
) {
1386 ldb_set_errstring(ldb
, errmsg
);
1389 return LDB_ERR_OPERATIONS_ERROR
;
1395 static int lsql_start_trans(struct ldb_module
* module
)
1399 struct lsqlite3_private
*lsqlite3
;
1401 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1402 struct lsqlite3_private
);
1404 if (lsqlite3
->trans_count
== 0) {
1405 ret
= sqlite3_exec(lsqlite3
->sqlite
, "BEGIN IMMEDIATE;", NULL
, NULL
, &errmsg
);
1406 if (ret
!= SQLITE_OK
) {
1408 printf("lsqlite3_start_trans: error: %s\n", errmsg
);
1415 lsqlite3
->trans_count
++;
1420 static int lsql_end_trans(struct ldb_module
*module
)
1424 struct lsqlite3_private
*lsqlite3
;
1426 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1427 struct lsqlite3_private
);
1429 if (lsqlite3
->trans_count
> 0) {
1430 lsqlite3
->trans_count
--;
1433 if (lsqlite3
->trans_count
== 0) {
1434 ret
= sqlite3_exec(lsqlite3
->sqlite
, "COMMIT;", NULL
, NULL
, &errmsg
);
1435 if (ret
!= SQLITE_OK
) {
1437 printf("lsqlite3_end_trans: error: %s\n", errmsg
);
1447 static int lsql_del_trans(struct ldb_module
*module
)
1449 struct lsqlite3_private
*lsqlite3
;
1451 lsqlite3
= talloc_get_type(ldb_module_get_private(module
),
1452 struct lsqlite3_private
);
1454 if (lsqlite3
->trans_count
> 0) {
1455 lsqlite3
->trans_count
--;
1458 if (lsqlite3
->trans_count
== 0) {
1459 return lsqlite3_safe_rollback(lsqlite3
->sqlite
);
1465 static int destructor(struct lsqlite3_private
*lsqlite3
)
1467 if (lsqlite3
->sqlite
) {
1468 sqlite3_close(lsqlite3
->sqlite
);
1473 static void lsql_request_done(struct lsql_context
*ctx
, int error
)
1475 struct ldb_context
*ldb
;
1476 struct ldb_request
*req
;
1477 struct ldb_reply
*ares
;
1479 ldb
= ldb_module_get_ctx(ctx
->module
);
1482 /* if we already returned an error just return */
1483 if (ldb_request_get_status(req
) != LDB_SUCCESS
) {
1487 ares
= talloc_zero(req
, struct ldb_reply
);
1490 req
->callback(req
, NULL
);
1493 ares
->type
= LDB_REPLY_DONE
;
1494 ares
->error
= error
;
1496 req
->callback(req
, ares
);
1499 static void lsql_timeout(struct tevent_context
*ev
,
1500 struct tevent_timer
*te
,
1504 struct lsql_context
*ctx
;
1505 ctx
= talloc_get_type(private_data
, struct lsql_context
);
1507 lsql_request_done(ctx
, LDB_ERR_TIME_LIMIT_EXCEEDED
);
1510 static void lsql_callback(struct tevent_context
*ev
,
1511 struct tevent_timer
*te
,
1515 struct lsql_context
*ctx
;
1518 ctx
= talloc_get_type(private_data
, struct lsql_context
);
1520 switch (ctx
->req
->operation
) {
1522 ret
= lsql_search(ctx
);
1525 ret
= lsql_add(ctx
);
1528 ret
= lsql_modify(ctx
);
1531 ret
= lsql_delete(ctx
);
1534 ret
= lsql_rename(ctx
);
1538 ret = lsql_extended(ctx);
1542 /* no other op supported */
1543 ret
= LDB_ERR_UNWILLING_TO_PERFORM
;
1546 if (!ctx
->callback_failed
) {
1547 /* Once we are done, we do not need timeout events */
1548 talloc_free(ctx
->timeout_event
);
1549 lsql_request_done(ctx
, ret
);
1553 static int lsql_handle_request(struct ldb_module
*module
, struct ldb_request
*req
)
1555 struct ldb_context
*ldb
;
1556 struct tevent_context
*ev
;
1557 struct lsql_context
*ac
;
1558 struct tevent_timer
*te
;
1561 if (check_critical_controls(req
->controls
)) {
1562 return LDB_ERR_UNSUPPORTED_CRITICAL_EXTENSION
;
1565 if (req
->starttime
== 0 || req
->timeout
== 0) {
1566 ldb_set_errstring(ldb
, "Invalid timeout settings");
1567 return LDB_ERR_TIME_LIMIT_EXCEEDED
;
1570 ldb
= ldb_module_get_ctx(module
);
1571 ev
= ldb_get_event_context(ldb
);
1573 ac
= talloc_zero(req
, struct lsql_context
);
1575 ldb_set_errstring(ldb
, "Out of Memory");
1576 return LDB_ERR_OPERATIONS_ERROR
;
1579 ac
->module
= module
;
1584 te
= tevent_add_timer(ev
, ac
, tv
, lsql_callback
, ac
);
1586 return LDB_ERR_OPERATIONS_ERROR
;
1589 tv
.tv_sec
= req
->starttime
+ req
->timeout
;
1590 ac
->timeout_event
= tevent_add_timer(ev
, ac
, tv
, lsql_timeout
, ac
);
1591 if (NULL
== ac
->timeout_event
) {
1592 return LDB_ERR_OPERATIONS_ERROR
;
1599 * Table of operations for the sqlite3 backend
1601 static const struct ldb_module_ops lsqlite3_ops
= {
1603 .search
= lsql_handle_request
,
1604 .add
= lsql_handle_request
,
1605 .modify
= lsql_handle_request
,
1606 .del
= lsql_handle_request
,
1607 .rename
= lsql_handle_request
,
1608 .extended
= lsql_handle_request
,
1609 .start_transaction
= lsql_start_trans
,
1610 .end_transaction
= lsql_end_trans
,
1611 .del_transaction
= lsql_del_trans
,
1618 static int initialize(struct lsqlite3_private
*lsqlite3
,
1619 struct ldb_context
*ldb
, const char *url
, int flags
)
1621 TALLOC_CTX
*local_ctx
;
1628 /* create a local ctx */
1629 local_ctx
= talloc_named(lsqlite3
, 0, "lsqlite3_rename local context");
1630 if (local_ctx
== NULL
) {
1634 schema
= lsqlite3_tprintf(local_ctx
,
1637 "CREATE TABLE ldb_info AS "
1638 " SELECT 'LDB' AS database_type,"
1639 " '1.0' AS version;"
1642 * The entry table holds the information about an entry.
1643 * This table is used to obtain the EID of the entry and to
1644 * support scope=one and scope=base. The parent and child
1645 * table is included in the entry table since all the other
1646 * attributes are dependent on EID.
1648 "CREATE TABLE ldb_entry "
1650 " eid INTEGER PRIMARY KEY AUTOINCREMENT,"
1651 " dn TEXT UNIQUE NOT NULL,"
1652 " norm_dn TEXT UNIQUE NOT NULL"
1656 "CREATE TABLE ldb_object_classes"
1658 " class_name TEXT PRIMARY KEY,"
1659 " parent_class_name TEXT,"
1660 " tree_key TEXT UNIQUE,"
1661 " max_child_num INTEGER DEFAULT 0"
1665 * We keep a full listing of attribute/value pairs here
1667 "CREATE TABLE ldb_attribute_values"
1669 " eid INTEGER REFERENCES ldb_entry,"
1671 " norm_attr_name TEXT,"
1673 " norm_attr_value TEXT "
1680 "CREATE INDEX ldb_attribute_values_eid_idx "
1681 " ON ldb_attribute_values (eid);"
1683 "CREATE INDEX ldb_attribute_values_name_value_idx "
1684 " ON ldb_attribute_values (attr_name, norm_attr_value);"
1692 "CREATE TRIGGER ldb_object_classes_insert_tr"
1694 " ON ldb_object_classes"
1697 " UPDATE ldb_object_classes"
1698 " SET tree_key = COALESCE(tree_key, "
1700 " SELECT tree_key || "
1701 " (SELECT base160(max_child_num + 1)"
1702 " FROM ldb_object_classes"
1703 " WHERE class_name = "
1704 " new.parent_class_name)"
1705 " FROM ldb_object_classes "
1706 " WHERE class_name = new.parent_class_name "
1708 " UPDATE ldb_object_classes "
1709 " SET max_child_num = max_child_num + 1"
1710 " WHERE class_name = new.parent_class_name;"
1714 * Table initialization
1717 "INSERT INTO ldb_object_classes "
1718 " (class_name, tree_key) "
1720 " ('TOP', '0001');");
1722 /* Skip protocol indicator of url */
1723 if (strncmp(url
, "sqlite3://", 10) != 0) {
1724 return SQLITE_MISUSE
;
1727 /* Update pointer to just after the protocol indicator */
1730 /* Try to open the (possibly empty/non-existent) database */
1731 if ((ret
= sqlite3_open(url
, &lsqlite3
->sqlite
)) != SQLITE_OK
) {
1735 /* In case this is a new database, enable auto_vacuum */
1736 ret
= sqlite3_exec(lsqlite3
->sqlite
, "PRAGMA auto_vacuum = 1;", NULL
, NULL
, &errmsg
);
1737 if (ret
!= SQLITE_OK
) {
1739 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1745 if (flags
& LDB_FLG_NOSYNC
) {
1747 ret
= sqlite3_exec(lsqlite3
->sqlite
, "PRAGMA synchronous = OFF;", NULL
, NULL
, &errmsg
);
1748 if (ret
!= SQLITE_OK
) {
1750 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1759 /* Establish a busy timeout of 30 seconds */
1760 if ((ret
= sqlite3_busy_timeout(lsqlite3
->sqlite
,
1761 30000)) != SQLITE_OK
) {
1765 /* Create a function, callable from sql, to increment a tree_key */
1767 sqlite3_create_function(lsqlite3
->sqlite
,/* handle */
1768 "base160_next", /* function name */
1769 1, /* number of args */
1770 SQLITE_ANY
, /* preferred text type */
1771 NULL
, /* user data */
1772 base160next_sql
, /* called func */
1773 NULL
, /* step func */
1774 NULL
/* final func */
1779 /* Create a function, callable from sql, to convert int to base160 */
1781 sqlite3_create_function(lsqlite3
->sqlite
,/* handle */
1782 "base160", /* function name */
1783 1, /* number of args */
1784 SQLITE_ANY
, /* preferred text type */
1785 NULL
, /* user data */
1786 base160_sql
, /* called func */
1787 NULL
, /* step func */
1788 NULL
/* final func */
1793 /* Create a function, callable from sql, to perform various comparisons */
1795 sqlite3_create_function(lsqlite3
->sqlite
, /* handle */
1796 "ldap_compare", /* function name */
1797 4, /* number of args */
1798 SQLITE_ANY
, /* preferred text type */
1799 ldb
, /* user data */
1800 lsqlite3_compare
, /* called func */
1801 NULL
, /* step func */
1802 NULL
/* final func */
1807 /* Begin a transaction */
1808 ret
= sqlite3_exec(lsqlite3
->sqlite
, "BEGIN EXCLUSIVE;", NULL
, NULL
, &errmsg
);
1809 if (ret
!= SQLITE_OK
) {
1811 printf("lsqlite3: initialization error: %s\n", errmsg
);
1818 /* Determine if this is a new database. No tables means it is. */
1819 if (query_int(lsqlite3
,
1822 " FROM sqlite_master\n"
1823 " WHERE type = 'table';") != 0) {
1827 if (queryInt
== 0) {
1829 * Create the database schema
1831 ret
= sqlite3_exec(lsqlite3
->sqlite
, schema
, NULL
, NULL
, &errmsg
);
1832 if (ret
!= SQLITE_OK
) {
1834 printf("lsqlite3 initializaion error: %s\n", errmsg
);
1841 * Ensure that the database we opened is one of ours
1843 if (query_int(lsqlite3
,
1846 " (SELECT COUNT(*) = 2"
1847 " FROM sqlite_master "
1848 " WHERE type = 'table' "
1852 " 'ldb_object_classes' "
1858 " WHERE database_type = 'LDB' "
1859 " AND version = '1.0'"
1863 /* It's not one that we created. See ya! */
1868 /* Commit the transaction */
1869 ret
= sqlite3_exec(lsqlite3
->sqlite
, "COMMIT;", NULL
, NULL
, &errmsg
);
1870 if (ret
!= SQLITE_OK
) {
1872 printf("lsqlite3: iniialization error: %s\n", errmsg
);
1881 if (rollback
) lsqlite3_safe_rollback(lsqlite3
->sqlite
);
1882 sqlite3_close(lsqlite3
->sqlite
);
1887 * connect to the database
1889 static int lsqlite3_connect(struct ldb_context
*ldb
,
1892 const char *options
[],
1893 struct ldb_module
**_module
)
1895 struct ldb_module
*module
;
1896 struct lsqlite3_private
*lsqlite3
;
1899 module
= ldb_module_new(ldb
, ldb
, "ldb_sqlite3 backend", &lsqlite3_ops
);
1900 if (!module
) return LDB_ERR_OPERATIONS_ERROR
;
1902 lsqlite3
= talloc(module
, struct lsqlite3_private
);
1907 lsqlite3
->sqlite
= NULL
;
1908 lsqlite3
->options
= NULL
;
1909 lsqlite3
->trans_count
= 0;
1911 ret
= initialize(lsqlite3
, ldb
, url
, flags
);
1912 if (ret
!= SQLITE_OK
) {
1916 talloc_set_destructor(lsqlite3
, destructor
);
1918 ldb_module_set_private(module
, lsqlite3
);
1922 * take a copy of the options array, so we don't have to rely
1923 * on the caller keeping it around (it might be dynamic)
1925 for (i
=0;options
[i
];i
++) ;
1927 lsqlite3
->options
= talloc_array(lsqlite3
, char *, i
+1);
1928 if (!lsqlite3
->options
) {
1932 for (i
=0;options
[i
];i
++) {
1934 lsqlite3
->options
[i
+1] = NULL
;
1935 lsqlite3
->options
[i
] =
1936 talloc_strdup(lsqlite3
->options
, options
[i
]);
1937 if (!lsqlite3
->options
[i
]) {
1947 if (lsqlite3
&& lsqlite3
->sqlite
!= NULL
) {
1948 (void) sqlite3_close(lsqlite3
->sqlite
);
1950 talloc_free(lsqlite3
);
1951 return LDB_ERR_OPERATIONS_ERROR
;
1954 const struct ldb_backend_ops ldb_sqlite3_backend_ops
= {
1956 .connect_fn
= lsqlite3_connect