summaryrefslogtreecommitdiff
path: root/lib/ldb/tools
diff options
context:
space:
mode:
authorAndrew Bartlett <abartlet@samba.org>2012-04-04 15:17:32 +1000
committerAndrew Bartlett <abartlet@samba.org>2012-04-10 11:14:43 +0200
commite17d12c23b2f56d8d29f7ee43148be85d28154c6 (patch)
tree40f233b97d294c70536f1e597de4b130ecd09807 /lib/ldb/tools
parent501d6d3dd4e3045e069bb2f7a52cf842dd1dfa67 (diff)
downloadsamba-e17d12c23b2f56d8d29f7ee43148be85d28154c6.tar.gz
samba-e17d12c23b2f56d8d29f7ee43148be85d28154c6.tar.bz2
samba-e17d12c23b2f56d8d29f7ee43148be85d28154c6.zip
ldb-tools: Place the whole of an ldif file in a transaction
This ensures that when operating ldbadd and ldbmodify against local ldb files, either an ldif file succeeds or fails as a whole. Also tests to verify that this is working correctly, and an ABI bump due to the extra (private, but exported to ldb* tools) symbol and behaviour change. Andrew Bartlett Autobuild-User: Andrew Bartlett <abartlet@samba.org> Autobuild-Date: Tue Apr 10 11:14:43 CEST 2012 on sn-devel-104
Diffstat (limited to 'lib/ldb/tools')
-rw-r--r--lib/ldb/tools/ldbadd.c45
-rw-r--r--lib/ldb/tools/ldbmodify.c45
2 files changed, 74 insertions, 16 deletions
diff --git a/lib/ldb/tools/ldbadd.c b/lib/ldb/tools/ldbadd.c
index 47fd261841..d0d5ec0ec4 100644
--- a/lib/ldb/tools/ldbadd.c
+++ b/lib/ldb/tools/ldbadd.c
@@ -34,8 +34,8 @@
#include "ldb.h"
#include "tools/cmdline.h"
#include "ldbutil.h"
+#include "include/ldb_private.h"
-static unsigned int failures;
static struct ldb_cmdline *options;
static void usage(struct ldb_context *ldb)
@@ -55,13 +55,23 @@ 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));
return 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(ldb, f))) {
+ while ((ldif = ldb_ldif_read_file_state(ldb, &state))) {
if (ldif->changetype != LDB_CHANGETYPE_ADD &&
ldif->changetype != LDB_CHANGETYPE_NONE) {
fprintf(stderr, "Only CHANGETYPE_ADD records allowed\n");
@@ -73,7 +83,6 @@ static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
fprintf(stderr,
"ERR: Message canonicalize failed - %s\n",
ldb_strerror(ret));
- failures++;
fun_ret = ret;
ldb_ldif_read_free(ldb, ldif);
continue;
@@ -81,10 +90,10 @@ static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
ret = ldb_add_ctrl(ldb, ldif->msg,req_ctrls);
if (ret != LDB_SUCCESS) {
- fprintf(stderr, "ERR: %s : \"%s\" on DN %s\n",
+ fprintf(stderr, "ERR: %s : \"%s\" on DN %s at block before line %llu\n",
ldb_strerror(ret), ldb_errstring(ldb),
- ldb_dn_get_linearized(ldif->msg->dn));
- failures++;
+ ldb_dn_get_linearized(ldif->msg->dn),
+ (unsigned long long)state.line_no);
fun_ret = ret;
} else {
(*count)++;
@@ -93,6 +102,24 @@ static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
}
}
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;
@@ -148,7 +175,11 @@ int main(int argc, const char **argv)
talloc_free(mem_ctx);
- printf("Added %u records with %u failures\n", count, failures);
+ if (ret) {
+ printf("Add failed after processing %u records\n", count);
+ } else {
+ printf("Added %u records successfully\n", count);
+ }
return ret;
}
diff --git a/lib/ldb/tools/ldbmodify.c b/lib/ldb/tools/ldbmodify.c
index 2ca7b62b2c..3a9a2657ce 100644
--- a/lib/ldb/tools/ldbmodify.c
+++ b/lib/ldb/tools/ldbmodify.c
@@ -34,8 +34,8 @@
#include "ldb.h"
#include "tools/cmdline.h"
#include "ldbutil.h"
+#include "include/ldb_private.h"
-static unsigned int failures;
static struct ldb_cmdline *options;
static void usage(struct ldb_context *ldb)
@@ -54,13 +54,23 @@ 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);
}
- while ((ldif = ldb_ldif_read_file(ldb, f))) {
+ 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;
@@ -96,10 +106,10 @@ static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
if (errstr == NULL) {
errstr = ldb_errstring(ldb);
}
- fprintf(stderr, "ERR: (%s) \"%s\" on DN %s\n",
+ 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));
- failures++;
+ errstr, ldb_dn_get_linearized(ldif->msg->dn),
+ (unsigned long long)state.line_no);
fun_ret = ret;
} else {
(*count)++;
@@ -108,13 +118,26 @@ static int process_file(struct ldb_context *ldb, FILE *f, unsigned int *count)
}
}
ldb_ldif_read_free(ldb, ldif);
+ if (ret) {
+ break;
+ }
}
-
- if (!feof(f)) {
+
+ 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;
}
@@ -150,7 +173,11 @@ int main(int argc, const char **argv)
talloc_free(mem_ctx);
- printf("Modified %u records with %u failures\n", count, failures);
-
+ if (ret) {
+ printf("Modify failed after processing %u records\n", count);
+ } else {
+ printf("Modified %u records successfully\n", count);
+ }
+
return ret;
}