r10909: Give better shutdown messages

[Samba/nascimento.git] / source3 / passdb / pdb_tdb.c

/*
 * Unix SMB/CIFS implementation. 
 * SMB parameters and setup
 * Copyright (C) Andrew Tridgell   1992-1998
 * Copyright (C) Simo Sorce        2000-2003
 * Copyright (C) Gerald Carter     2000
 * Copyright (C) Jeremy Allison    2001
 * Copyright (C) Andrew Bartlett   2002
 * 
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option)
 * any later version.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 * 
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 675
 * Mass Ave, Cambridge, MA 02139, USA.
 */

#include "includes.h"

#if 0 /* when made a module use this */

static int tdbsam_debug_level = DBGC_ALL;
#undef DBGC_CLASS
#define DBGC_CLASS tdbsam_debug_level

#else

#undef DBGC_CLASS
#define DBGC_CLASS DBGC_PASSDB

#endif

#define TDBSAM_VERSION  2       /* Most recent TDBSAM version */
#define TDBSAM_VERSION_STRING   "INFO/version"
#define PASSDB_FILE_NAME        "passdb.tdb"
#define USERPREFIX              "USER_"
#define RIDPREFIX               "RID_"
#define PRIVPREFIX              "PRIV_"
#define tdbsamver_t     int32

struct tdbsam_privates {
        TDB_CONTEXT     *passwd_tdb;

        /* retrive-once info */
        const char *tdbsam_location;
};

struct pwent_list {
        struct pwent_list *prev, *next;
        TDB_DATA key;
};
static struct pwent_list *tdbsam_pwent_list;


/**
 * Convert old TDBSAM to the latest version.
 * @param pdb_tdb A pointer to the opened TDBSAM file which must be converted. 
 *                This file must be opened with read/write access.
 * @param from Current version of the TDBSAM file.
 * @return True if the conversion has been successful, false otherwise. 
 **/

static BOOL tdbsam_convert(TDB_CONTEXT *pdb_tdb, tdbsamver_t from) 
{
        const char * vstring = TDBSAM_VERSION_STRING;
        SAM_ACCOUNT *user = NULL;
        const char *prefix = USERPREFIX;
        TDB_DATA        data, key, old_key;
        uint8           *buf = NULL;
        BOOL            ret;

        if (pdb_tdb == NULL) {
                DEBUG(0,("tdbsam_convert: Bad TDB Context pointer.\n"));
                return False;
        }

        /* handle a Samba upgrade */
        tdb_lock_bystring(pdb_tdb, vstring, 0);
        
        if (!NT_STATUS_IS_OK(pdb_init_sam(&user))) {
                DEBUG(0,("tdbsam_convert: cannot initialized a SAM_ACCOUNT.\n"));
                return False;
        }

        /* Enumerate all records and convert them */
        key = tdb_firstkey(pdb_tdb);

        while (key.dptr) {
        
                /* skip all non-USER entries (eg. RIDs) */
                while ((key.dsize != 0) && (strncmp(key.dptr, prefix, strlen (prefix)))) {
                        old_key = key;
                        /* increment to next in line */
                        key = tdb_nextkey(pdb_tdb, key);
                        SAFE_FREE(old_key.dptr);
                }
        
                if (key.dptr) {
                        
                        /* read from tdbsam */
                        data = tdb_fetch(pdb_tdb, key);
                        if (!data.dptr) {
                                DEBUG(0,("tdbsam_convert: database entry not found: %s.\n",key.dptr));
                                return False;
                        }
        
                        if (!NT_STATUS_IS_OK(pdb_reset_sam(user))) {
                                DEBUG(0,("tdbsam_convert: cannot reset SAM_ACCOUNT.\n"));
                                SAFE_FREE(data.dptr);
                                return False;
                        }
                        
                        /* unpack the buffer from the former format */
                        DEBUG(10,("tdbsam_convert: Try unpacking a record with (key:%s) (version:%d)\n", key.dptr, from));
                        switch (from) {
                                case 0:
                                        ret = init_sam_from_buffer_v0(user, (uint8 *)data.dptr, data.dsize);
                                        break;
                                case 1:
                                        ret = init_sam_from_buffer_v1(user, (uint8 *)data.dptr, data.dsize);
                                        break;
                                case 2:
                                        ret = init_sam_from_buffer_v2(user, (uint8 *)data.dptr, data.dsize);
                                        break;
                                default:
                                        /* unknown tdbsam version */
                                        ret = False;
                        }
                        if (!ret) {
                                DEBUG(0,("tdbsam_convert: Bad SAM_ACCOUNT entry returned from TDB (key:%s) (version:%d)\n", key.dptr, from));
                                SAFE_FREE(data.dptr);
                                return False;
                        }
        
                        /* We're finished with the old data. */
                        SAFE_FREE(data.dptr);

                        /* pack from the buffer into the new format */
                        DEBUG(10,("tdbsam_convert: Try packing a record (key:%s) (version:%d)\n", key.dptr, from));
                        if ((data.dsize=init_buffer_from_sam (&buf, user, False)) == -1) {
                                DEBUG(0,("tdbsam_convert: cannot pack the SAM_ACCOUNT into the new format\n"));
                                SAFE_FREE(data.dptr);
                                return False;
                        }
                        data.dptr = (char *)buf;
                        
                        /* Store the buffer inside the TDBSAM */
                        if (tdb_store(pdb_tdb, key, data, TDB_MODIFY) != TDB_SUCCESS) {
                                DEBUG(0,("tdbsam_convert: cannot store the SAM_ACCOUNT (key:%s) in new format\n",key.dptr));
                                SAFE_FREE(data.dptr);
                                return False;
                        }
                        
                        SAFE_FREE(data.dptr);
                        
                        /* increment to next in line */
                        old_key = key;
                        key = tdb_nextkey(pdb_tdb, key);
                        SAFE_FREE(old_key.dptr);
                }
                
        }

        pdb_free_sam(&user);
        
        /* upgrade finished */
        tdb_store_int32(pdb_tdb, vstring, TDBSAM_VERSION);
        tdb_unlock_bystring(pdb_tdb, vstring);

        return(True);   
}

/**
 * Open the TDB passwd database, check version and convert it if needed.
 * @param name filename of the tdbsam file.
 * @param open_flags file access mode.
 * @return a TDB_CONTEXT handle on the tdbsam file.
 **/

static TDB_CONTEXT * tdbsam_tdbopen (const char *name, int open_flags)
{
        TDB_CONTEXT     *pdb_tdb;
        tdbsamver_t     version;
        
        /* Try to open tdb passwd */
        if (!(pdb_tdb = tdb_open_log(name, 0, TDB_DEFAULT, 
                                     open_flags, 0600))) {
                DEBUG(0, ("Unable to open/create TDB passwd\n"));
                return NULL;
        }

        /* Check the version */
        version = (tdbsamver_t) tdb_fetch_int32(pdb_tdb, 
                                                TDBSAM_VERSION_STRING);
        if (version == -1)
                version = 0;    /* Version not found, assume version 0 */
        
        /* Compare the version */
        if (version > TDBSAM_VERSION) {
                /* Version more recent than the latest known */ 
                DEBUG(0, ("TDBSAM version unknown: %d\n", version));
                tdb_close(pdb_tdb);
                pdb_tdb = NULL;
        } 
        else if (version < TDBSAM_VERSION) {
                /* Older version, must be converted */
                DEBUG(1, ("TDBSAM version too old (%d), trying to convert it.\n", version));
                
                /* Reopen the pdb file with read-write access if needed */
                if (!(open_flags & O_RDWR)) {
                        DEBUG(10, ("tdbsam_tdbopen: TDB file opened with read only access, reopen it with read-write access.\n"));
                        tdb_close(pdb_tdb);
                        pdb_tdb = tdb_open_log(name, 0, TDB_DEFAULT, (open_flags & 07777770) | O_RDWR, 0600);
                }
                
                /* Convert */
                if (!tdbsam_convert(pdb_tdb, version)){
                        DEBUG(0, ("tdbsam_tdbopen: Error when trying to convert tdbsam: %s\n",name));
                        tdb_close(pdb_tdb);
                        pdb_tdb = NULL;
                } else {
                        DEBUG(1, ("TDBSAM converted successfully.\n"));
                }

                /* Reopen the pdb file as it must be */
                if (!(open_flags & O_RDWR)) {
                        tdb_close(pdb_tdb);
                        pdb_tdb = tdb_open_log(name, 0, TDB_DEFAULT, open_flags, 0600);
                }
        }
        
        return pdb_tdb;
}

/*****************************************************************************
 Utility functions to close the tdb sam database
 ****************************************************************************/

static void tdbsam_tdbclose ( struct tdbsam_privates *state )
{
        if ( !state )
                return;
                
        if ( state->passwd_tdb ) {
                tdb_close( state->passwd_tdb );
                state->passwd_tdb = NULL;
        }
        
        return;
                
}

/****************************************************************************
 creates a list of user keys
****************************************************************************/

static int tdbsam_traverse_setpwent(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
{
        const char *prefix = USERPREFIX;
        int  prefixlen = strlen (prefix);
        struct pwent_list *ptr;
        
        if ( strncmp(key.dptr, prefix, prefixlen) == 0 ) {
                if ( !(ptr=SMB_MALLOC_P(struct pwent_list)) ) {
                        DEBUG(0,("tdbsam_traverse_setpwent: Failed to malloc new entry for list\n"));
                        
                        /* just return 0 and let the traversal continue */
                        return 0;
                }
                ZERO_STRUCTP(ptr);
                
                /* save a copy of the key */
                
                ptr->key.dptr = memdup( key.dptr, key.dsize );
                ptr->key.dsize = key.dsize;
                
                DLIST_ADD( tdbsam_pwent_list, ptr );
        
        }
        
        
        return 0;
}

/***************************************************************
 Open the TDB passwd database for SAM account enumeration.
 Save a list of user keys for iteration.
****************************************************************/

static NTSTATUS tdbsam_setsampwent(struct pdb_methods *my_methods, BOOL update, uint16 acb_mask)
{
        uint32 flags = update ? (O_RDWR|O_CREAT) : O_RDONLY;
        
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        
        if ( !(tdb_state->passwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, flags )) ) 
                return NT_STATUS_UNSUCCESSFUL;

        tdb_traverse( tdb_state->passwd_tdb, tdbsam_traverse_setpwent, NULL );
        
        return NT_STATUS_OK;
}


/***************************************************************
 End enumeration of the TDB passwd list.
****************************************************************/

static void tdbsam_endsampwent(struct pdb_methods *my_methods)
{
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        struct pwent_list *ptr, *ptr_next;
        
        tdbsam_tdbclose( tdb_state );
        
        /* clear out any remaining entries in the list */
        
        for ( ptr=tdbsam_pwent_list; ptr; ptr = ptr_next ) {
                ptr_next = ptr->next;
                DLIST_REMOVE( tdbsam_pwent_list, ptr );
                SAFE_FREE( ptr->key.dptr);
                SAFE_FREE( ptr );
        }
        
        DEBUG(7, ("endtdbpwent: closed sam database.\n"));
}

/*****************************************************************
 Get one SAM_ACCOUNT from the TDB (next in line)
*****************************************************************/

static NTSTATUS tdbsam_getsampwent(struct pdb_methods *my_methods, SAM_ACCOUNT *user)
{
        NTSTATUS                nt_status = NT_STATUS_UNSUCCESSFUL;
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        TDB_DATA                data;
        struct pwent_list       *pkey;

        if ( !user ) {
                DEBUG(0,("tdbsam_getsampwent: SAM_ACCOUNT is NULL.\n"));
                return nt_status;
        }

        if ( !tdbsam_pwent_list ) {
                DEBUG(4,("tdbsam_getsampwent: end of list\n"));
                tdbsam_tdbclose( tdb_state );
                return nt_status;
        }
        
        if ( !tdb_state->passwd_tdb ) {
                if ( !(tdb_state->passwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_RDONLY)) )
                        return nt_status;
        }

        /* pull the next entry */
                
        pkey = tdbsam_pwent_list;
        DLIST_REMOVE( tdbsam_pwent_list, pkey );
        
        data = tdb_fetch(tdb_state->passwd_tdb, pkey->key);

        SAFE_FREE( pkey->key.dptr);
        SAFE_FREE( pkey);
        
        if (!data.dptr) {
                DEBUG(5,("pdb_getsampwent: database entry not found.  Was the user deleted?\n"));
                return nt_status;
        }
  
        if (!init_sam_from_buffer(user, (unsigned char *)data.dptr, data.dsize)) {
                DEBUG(0,("pdb_getsampwent: Bad SAM_ACCOUNT entry returned from TDB!\n"));
        }
        
        SAFE_FREE( data.dptr );
        

        return NT_STATUS_OK;
}

/******************************************************************
 Lookup a name in the SAM TDB
******************************************************************/

static NTSTATUS tdbsam_getsampwnam (struct pdb_methods *my_methods, SAM_ACCOUNT *user, const char *sname)
{
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        TDB_CONTEXT     *pwd_tdb;
        TDB_DATA        data, key;
        fstring         keystr;
        fstring         name;

        if ( !user ) {
                DEBUG(0,("pdb_getsampwnam: SAM_ACCOUNT is NULL.\n"));
                return nt_status;
        }

        /* Data is stored in all lower-case */
        fstrcpy(name, sname);
        strlower_m(name);

        /* set search key */
        slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
        key.dptr = keystr;
        key.dsize = strlen(keystr) + 1;

        /* open the accounts TDB */
        if (!(pwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_RDONLY))) {
        
                if (errno == ENOENT) {
                        /*
                         * TDB file doesn't exist, so try to create new one. This is useful to avoid
                         * confusing error msg when adding user account first time
                         */
                        if (!(pwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_CREAT ))) {
                                DEBUG(0, ("pdb_getsampwnam: TDB passwd (%s) did not exist. File successfully created.\n",
                                          tdb_state->tdbsam_location));
                        } else {
                                DEBUG(0, ("pdb_getsampwnam: TDB passwd (%s) does not exist. Couldn't create new one. Error was: %s\n",
                                          tdb_state->tdbsam_location, strerror(errno)));
                        }
                        
                        /* requested user isn't there anyway */
                        nt_status = NT_STATUS_NO_SUCH_USER;
                        return nt_status;
                }
                DEBUG(0, ("pdb_getsampwnam: Unable to open TDB passwd (%s)!\n", tdb_state->tdbsam_location));
                return nt_status;
        }

        /* get the record */
        data = tdb_fetch(pwd_tdb, key);
        if (!data.dptr) {
                DEBUG(5,("pdb_getsampwnam (TDB): error fetching database.\n"));
                DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
                DEBUGADD(5, (" Key: %s\n", keystr));
                tdb_close(pwd_tdb);
                return nt_status;
        }
  
        /* unpack the buffer */
        if (!init_sam_from_buffer(user, (unsigned char *)data.dptr, data.dsize)) {
                DEBUG(0,("pdb_getsampwent: Bad SAM_ACCOUNT entry returned from TDB!\n"));
                SAFE_FREE(data.dptr);
                tdb_close(pwd_tdb);
                return nt_status;
        }
        SAFE_FREE(data.dptr);

        /* no further use for database, close it now */
        tdb_close(pwd_tdb);
        
        return NT_STATUS_OK;
}

/***************************************************************************
 Search by rid
 **************************************************************************/

static NTSTATUS tdbsam_getsampwrid (struct pdb_methods *my_methods, SAM_ACCOUNT *user, uint32 rid)
{
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        TDB_CONTEXT             *pwd_tdb;
        TDB_DATA                data, key;
        fstring                 keystr;
        fstring                 name;
        
        if (user==NULL) {
                DEBUG(0,("pdb_getsampwrid: SAM_ACCOUNT is NULL.\n"));
                return nt_status;
        }

        /* set search key */
        slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, rid);
        key.dptr = keystr;
        key.dsize = strlen (keystr) + 1;

        /* open the accounts TDB */
        if (!(pwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_RDONLY))) {
                DEBUG(0, ("pdb_getsampwrid: Unable to open TDB rid database!\n"));
                return nt_status;
        }

        /* get the record */
        data = tdb_fetch (pwd_tdb, key);
        if (!data.dptr) {
                DEBUG(5,("pdb_getsampwrid (TDB): error looking up RID %d by key %s.\n", rid, keystr));
                DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
                tdb_close (pwd_tdb);
                return nt_status;
        }


        fstrcpy(name, data.dptr);
        SAFE_FREE(data.dptr);
        
        tdb_close (pwd_tdb);
        
        return tdbsam_getsampwnam (my_methods, user, name);
}

static NTSTATUS tdbsam_getsampwsid(struct pdb_methods *my_methods, SAM_ACCOUNT * user, const DOM_SID *sid)
{
        uint32 rid;
        if (!sid_peek_check_rid(get_global_sam_sid(), sid, &rid))
                return NT_STATUS_UNSUCCESSFUL;
        return tdbsam_getsampwrid(my_methods, user, rid);
}

/***************************************************************************
 Delete a SAM_ACCOUNT
****************************************************************************/

static NTSTATUS tdbsam_delete_sam_account(struct pdb_methods *my_methods, SAM_ACCOUNT *sam_pass)
{
        NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        TDB_CONTEXT     *pwd_tdb;
        TDB_DATA        key;
        fstring         keystr;
        uint32          rid;
        fstring         name;
        
        fstrcpy(name, pdb_get_username(sam_pass));
        strlower_m(name);
        
        /* open the TDB */
        if (!(pwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_RDWR))) {
                DEBUG(0, ("Unable to open TDB passwd!"));
                return nt_status;
        }
  
        /* set the search key */
        slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
        key.dptr = keystr;
        key.dsize = strlen (keystr) + 1;
        
        rid = pdb_get_user_rid(sam_pass);

        /* it's outaa here!  8^) */
        if (tdb_delete(pwd_tdb, key) != TDB_SUCCESS) {
                DEBUG(5, ("Error deleting entry from tdb passwd database!\n"));
                DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
                tdb_close(pwd_tdb); 
                return nt_status;
        }       

        /* delete also the RID key */

        /* set the search key */
        slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, rid);
        key.dptr = keystr;
        key.dsize = strlen (keystr) + 1;

        /* it's outaa here!  8^) */
        if (tdb_delete(pwd_tdb, key) != TDB_SUCCESS) {
                DEBUG(5, ("Error deleting entry from tdb rid database!\n"));
                DEBUGADD(5, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
                tdb_close(pwd_tdb); 
                return nt_status;
        }
        
        tdb_close(pwd_tdb);
        
        return NT_STATUS_OK;
}

/***************************************************************************
 Update the TDB SAM
****************************************************************************/

static BOOL tdb_update_sam(struct pdb_methods *my_methods, SAM_ACCOUNT* newpwd, int flag)
{
        struct tdbsam_privates *tdb_state = (struct tdbsam_privates *)my_methods->private_data;
        TDB_CONTEXT     *pwd_tdb = NULL;
        TDB_DATA        key, data;
        uint8           *buf = NULL;
        fstring         keystr;
        fstring         name;
        BOOL            ret = True;
        uint32          user_rid;

        /* invalidate the existing TDB iterator if it is open */
        
        if (tdb_state->passwd_tdb) {
                tdb_close(tdb_state->passwd_tdb);
                tdb_state->passwd_tdb = NULL;
        }

        /* open the account TDB passwd*/
        
        pwd_tdb = tdbsam_tdbopen(tdb_state->tdbsam_location, O_RDWR | O_CREAT);
        
        if (!pwd_tdb) {
                DEBUG(0, ("tdb_update_sam: Unable to open TDB passwd (%s)!\n", 
                        tdb_state->tdbsam_location));
                return False;
        }

        if (!pdb_get_group_rid(newpwd)) {
                DEBUG (0,("tdb_update_sam: Failing to store a SAM_ACCOUNT for [%s] without a primary group RID\n",
                        pdb_get_username(newpwd)));
                ret = False;
                goto done;
        }

        if ( !(user_rid = pdb_get_user_rid(newpwd)) ) {
                DEBUG(0,("tdb_update_sam: SAM_ACCOUNT (%s) with no RID!\n", pdb_get_username(newpwd)));
                ret = False;
                goto done;
        }

        /* copy the SAM_ACCOUNT struct into a BYTE buffer for storage */
        if ((data.dsize=init_buffer_from_sam (&buf, newpwd, False)) == -1) {
                DEBUG(0,("tdb_update_sam: ERROR - Unable to copy SAM_ACCOUNT info BYTE buffer!\n"));
                ret = False;
                goto done;
        }
        data.dptr = (char *)buf;

        fstrcpy(name, pdb_get_username(newpwd));
        strlower_m(name);
        
        DEBUG(5, ("Storing %saccount %s with RID %d\n", flag == TDB_INSERT ? "(new) " : "", name, user_rid));

        /* setup the USER index key */
        slprintf(keystr, sizeof(keystr)-1, "%s%s", USERPREFIX, name);
        key.dptr = keystr;
        key.dsize = strlen(keystr) + 1;

        /* add the account */
        if (tdb_store(pwd_tdb, key, data, flag) != TDB_SUCCESS) {
                DEBUG(0, ("Unable to modify passwd TDB!"));
                DEBUGADD(0, (" Error: %s", tdb_errorstr(pwd_tdb)));
                DEBUGADD(0, (" occured while storing the main record (%s)\n", keystr));
                ret = False;
                goto done;
        }
        
        /* setup RID data */
        data.dsize = strlen(name) + 1;
        data.dptr = name;

        /* setup the RID index key */
        slprintf(keystr, sizeof(keystr)-1, "%s%.8x", RIDPREFIX, user_rid);
        key.dptr = keystr;
        key.dsize = strlen (keystr) + 1;
        
        /* add the reference */
        if (tdb_store(pwd_tdb, key, data, flag) != TDB_SUCCESS) {
                DEBUG(0, ("Unable to modify TDB passwd !"));
                DEBUGADD(0, (" Error: %s\n", tdb_errorstr(pwd_tdb)));
                DEBUGADD(0, (" occured while storing the RID index (%s)\n", keystr));
                ret = False;
                goto done;
        }

done:   
        /* cleanup */
        tdb_close (pwd_tdb);
        SAFE_FREE(buf);
        
        return (ret);   
}

/***************************************************************************
 Modifies an existing SAM_ACCOUNT
****************************************************************************/

static NTSTATUS tdbsam_update_sam_account (struct pdb_methods *my_methods, SAM_ACCOUNT *newpwd)
{
        if (tdb_update_sam(my_methods, newpwd, TDB_MODIFY))
                return NT_STATUS_OK;
        else
                return NT_STATUS_UNSUCCESSFUL;
}

/***************************************************************************
 Adds an existing SAM_ACCOUNT
****************************************************************************/

static NTSTATUS tdbsam_add_sam_account (struct pdb_methods *my_methods, SAM_ACCOUNT *newpwd)
{
        if (tdb_update_sam(my_methods, newpwd, TDB_INSERT))
                return NT_STATUS_OK;
        else
                return NT_STATUS_UNSUCCESSFUL;
}

static void free_private_data(void **vp) 
{
        struct tdbsam_privates **tdb_state = (struct tdbsam_privates **)vp;
        tdbsam_tdbclose(*tdb_state);
        *tdb_state = NULL;

        /* No need to free any further, as it is talloc()ed */
}




/**
 * Init tdbsam backend
 *
 * @param pdb_context initialised passdb context
 * @param pdb_method backend methods structure to be filled with function pointers
 * @param location the backend tdb file location
 *
 * @return nt_status code
 **/

static NTSTATUS pdb_init_tdbsam(PDB_CONTEXT *pdb_context, PDB_METHODS **pdb_method, const char *location)
{
        NTSTATUS nt_status;
        struct tdbsam_privates *tdb_state;

        if (!NT_STATUS_IS_OK(nt_status = make_pdb_methods(pdb_context->mem_ctx, pdb_method))) {
                return nt_status;
        }

        (*pdb_method)->name = "tdbsam";

        (*pdb_method)->setsampwent = tdbsam_setsampwent;
        (*pdb_method)->endsampwent = tdbsam_endsampwent;
        (*pdb_method)->getsampwent = tdbsam_getsampwent;
        (*pdb_method)->getsampwnam = tdbsam_getsampwnam;
        (*pdb_method)->getsampwsid = tdbsam_getsampwsid;
        (*pdb_method)->add_sam_account = tdbsam_add_sam_account;
        (*pdb_method)->update_sam_account = tdbsam_update_sam_account;
        (*pdb_method)->delete_sam_account = tdbsam_delete_sam_account;

        tdb_state = TALLOC_ZERO_P(pdb_context->mem_ctx, struct tdbsam_privates);

        if (!tdb_state) {
                DEBUG(0, ("talloc() failed for tdbsam private_data!\n"));
                return NT_STATUS_NO_MEMORY;
        }

        if (location) {
                tdb_state->tdbsam_location = talloc_strdup(pdb_context->mem_ctx, location);
        } else {
                pstring tdbfile;
                get_private_directory(tdbfile);
                pstrcat(tdbfile, "/");
                pstrcat(tdbfile, PASSDB_FILE_NAME);
                tdb_state->tdbsam_location = talloc_strdup(pdb_context->mem_ctx, tdbfile);
        }

        (*pdb_method)->private_data = tdb_state;

        (*pdb_method)->free_private_data = free_private_data;

        return NT_STATUS_OK;
}

NTSTATUS pdb_tdbsam_init(void)
{
        return smb_register_passdb(PASSDB_INTERFACE_VERSION, "tdbsam", pdb_init_tdbsam);
}