source: vendor/current/lib/ldb/tools/ldbmodify.c

/* 
   ldb database library

   Copyright (C) Andrew Tridgell  2004

     ** NOTE! The following LGPL license applies to the ldb
     ** library. This does NOT imply that all of Samba is released
     ** under the LGPL
   
   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 3 of the License, or (at your option) any later version.

   This library 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
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, see <http://www.gnu.org/licenses/>.
*/

/*
 *  Name: ldb
 *
 *  Component: ldbmodify
 *
 *  Description: utility to modify records - modelled on ldapmodify
 *
 *  Author: Andrew Tridgell
 */

#include "replace.h"
#include "ldb.h"
#include "tools/cmdline.h"
#include "ldbutil.h"
#include "include/ldb_private.h"

static struct ldb_cmdline *options;

static void usage(struct ldb_context *ldb)
{
        printf("Usage: ldbmodify <options> <ldif...>\n");
        printf("Modifies a ldb based upon ldif change records\n\n");
        ldb_cmdline_help(ldb, "ldbmodify", stdout);
        exit(LDB_ERR_OPERATIONS_ERROR);
}

/*
  process modifies for one file
*/
static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
{
        struct ldb_ldif *ldif;
        int fun_ret = LDB_SUCCESS, ret;
        struct ldb_control **req_ctrls = ldb_parse_control_strings(ldb, ldb, (const char **)options->controls);
        struct ldif_read_file_state state = {
                .f = f
        };

        if (options->controls != NULL &&  req_ctrls== NULL) {
                printf("parsing controls failed: %s\n", ldb_errstring(ldb));
                exit(LDB_ERR_OPERATIONS_ERROR);
        }

        fun_ret = ldb_transaction_start(ldb);
        if (fun_ret != LDB_SUCCESS) {
                fprintf(stderr, "ERR: (%s) on transaction start\n",
                        ldb_errstring(ldb));
                return fun_ret;
        }

        while ((ldif = ldb_ldif_read_file_state(ldb, &state))) {
                struct ldb_dn *olddn;
                bool deleteoldrdn = false;
                struct ldb_dn *newdn;
                const char *errstr = NULL;

                switch (ldif->changetype) {
                case LDB_CHANGETYPE_NONE:
                case LDB_CHANGETYPE_ADD:
                        ret = ldb_add_ctrl(ldb, ldif->msg,req_ctrls);
                        break;
                case LDB_CHANGETYPE_DELETE:
                        ret = ldb_delete_ctrl(ldb, ldif->msg->dn,req_ctrls);
                        break;
                case LDB_CHANGETYPE_MODIFY:
                        ret = ldb_modify_ctrl(ldb, ldif->msg,req_ctrls);
                        break;
                case LDB_CHANGETYPE_MODRDN:
                        ret = ldb_ldif_parse_modrdn(ldb, ldif, ldif, &olddn,
                                                    NULL, &deleteoldrdn,
                                                    NULL, &newdn);
                        if (ret == LDB_SUCCESS) {
                                if (deleteoldrdn) {
                                        ret = ldb_rename(ldb, olddn, newdn);
                                } else {
                                        errstr = "modrdn: deleteoldrdn=0 "
                                                 "not supported.";
                                        ret = LDB_ERR_CONSTRAINT_VIOLATION;
                                }
                        }
                        break;
                }
                if (ret != LDB_SUCCESS) {
                        if (errstr == NULL) {
                                errstr = ldb_errstring(ldb);
                        }
                        fprintf(stderr, "ERR: (%s) \"%s\" on DN %s at block before line %llu\n",
                                ldb_strerror(ret),
                                errstr, ldb_dn_get_linearized(ldif->msg->dn), 
                                (unsigned long long)state.line_no);
                        fun_ret = ret;
                } else {
                        (*count)++;
                        if (options->verbose) {
                                printf("Modified %s\n", ldb_dn_get_linearized(ldif->msg->dn));
                        }
                }
                ldb_ldif_read_free(ldb, ldif);
                if (ret) {
                        break;
                }
        }
        
        if (fun_ret == LDB_SUCCESS && !feof(f)) {
                fprintf(stderr, "Failed to parse ldif\n");
                fun_ret = LDB_ERR_OPERATIONS_ERROR;
        }

        if (fun_ret == LDB_SUCCESS) {
                fun_ret = ldb_transaction_commit(ldb);
                if (fun_ret != LDB_SUCCESS) {
                        fprintf(stderr, "ERR: (%s) on transaction commit\n",
                                ldb_errstring(ldb));
                }
        } else {
                ldb_transaction_cancel(ldb);
        }

        return fun_ret;
}

int main(int argc, const char **argv)
{
        struct ldb_context *ldb;
        unsigned int i, count = 0;
        int ret = LDB_SUCCESS;
        TALLOC_CTX *mem_ctx = talloc_new(NULL);

        ldb = ldb_init(mem_ctx, NULL);
        if (ldb == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }

        options = ldb_cmdline_process(ldb, argc, argv, usage);

        if (options->argc == 0) {
                ret = process_file(ldb, stdin, &count);
        } else {
                for (i=0;i<options->argc;i++) {
                        const char *fname = options->argv[i];
                        FILE *f;
                        f = fopen(fname, "r");
                        if (!f) {
                                perror(fname);
                                return LDB_ERR_OPERATIONS_ERROR;
                        }
                        ret = process_file(ldb, f, &count);
                        fclose(f);
                }
        }

        talloc_free(mem_ctx);

        if (ret) {
                printf("Modify failed after processing %u records\n", count);
        } else {
                printf("Modified %u records successfully\n", count);
        }
                
        return ret;
}