1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/lib/ldb/tools/ldbmodify.c

188 lines
4.6 KiB
C
Raw Normal View History

/*
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 %zu\n",
ldb_strerror(ret),
errstr,
ldb_dn_get_linearized(ldif->msg->dn),
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;
}