Line data    Source code

       1             : /*
       2             :  * ldb connection and module initialisation
       3             :  *
       4             :  *  Copyright (C) Andrew Bartlett <abartlet@samba.org> 2018
       5             :  *
       6             :  * This program is free software; you can redistribute it and/or modify
       7             :  * it under the terms of the GNU General Public License as published by
       8             :  * the Free Software Foundation; either version 3 of the License, or
       9             :  * (at your option) any later version.
      10             :  *
      11             :  * This program is distributed in the hope that it will be useful,
      12             :  * but WITHOUT ANY WARRANTY; without even the implied warranty of
      13             :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      14             :  * GNU General Public License for more details.
      15             :  *
      16             :  * You should have received a copy of the GNU General Public License
      17             :  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
      18             :  *
      19             :  */
      20             : #include "ldb_private.h"
      21             : #include "../ldb_tdb/ldb_tdb.h"
      22             : #ifdef HAVE_LMDB
      23             : #include "../ldb_mdb/ldb_mdb.h"
      24             : #endif /* HAVE_LMDB */
      25             : 
      26             : /*
      27             :   connect to the database
      28             : */
      29          62 : static int lldb_connect(struct ldb_context *ldb,
      30             :                         const char *url,
      31             :                         unsigned int flags,
      32             :                         const char *options[],
      33             :                         struct ldb_module **module)
      34             : {
      35           5 :         const char *path;
      36           5 :         int ret;
      37             : 
      38             :         /*
      39             :          * Check and remove the url prefix
      40             :          */
      41          62 :         if (strchr(url, ':')) {
      42          62 :                 if (strncmp(url, "ldb://", 6) != 0) {
      43           0 :                         ldb_debug(ldb, LDB_DEBUG_ERROR,
      44             :                                   "Invalid ldb URL '%s'", url);
      45           0 :                         return LDB_ERR_OPERATIONS_ERROR;
      46             :                 }
      47          62 :                 path = url+6;
      48             :         } else {
      49           0 :                 path = url;
      50             :         }
      51             : 
      52             :         /*
      53             :          * Don't create the database if it's not there
      54             :          */
      55          62 :         flags |= LDB_FLG_DONT_CREATE_DB;
      56             : #ifdef HAVE_LMDB
      57             :         /*
      58             :          * Try opening the database as an lmdb
      59             :          */
      60          62 :         ret = lmdb_connect(ldb, path, flags, options, module);
      61          62 :         if (ret == LDB_SUCCESS) {
      62           3 :                 return ret;
      63             :         }
      64          59 :         if (ret != LDB_ERR_UNAVAILABLE) {
      65           0 :                 return ret;
      66             :         }
      67             : 
      68             :         /*
      69             :          * Not mdb so try as tdb
      70             :          */
      71             : #endif /* HAVE_LMDB */
      72          59 :         ret = ltdb_connect(ldb, path, flags, options, module);
      73          59 :         return ret;
      74             : }
      75             : 
      76        6198 : int ldb_ldb_init(const char *version)
      77             : {
      78        6198 :         LDB_MODULE_CHECK_VERSION(version);
      79        6198 :         return ldb_register_backend("ldb", lldb_connect, false);
      80             : }