From 36be1040970d1225689caf129a9ca67970334830 Mon Sep 17 00:00:00 2001
From: Hallvard Furuseth <hallvard@openldap.org>
Date: Wed, 20 May 2015 03:17:47 +0200
Subject: [PATCH] lmdb-test: mtest*.c: Fix MDB_NOOVERWRITE, plug cursor leak.

Change-Id: Idc2840fafc0d8cb4aaaa2e23b899a9ce7fb2b52b
---
 mtest0.c | 6 ++++--
 mtest2.c | 4 ++--
 mtest6.c | 9 ++++++---
 3 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/mtest0.c b/mtest0.c
index f27132d1..4fe9a15a 100644
--- a/mtest0.c
+++ b/mtest0.c
@@ -54,12 +54,13 @@ int main(int argc,char * argv[])
 
 	key.mv_size = sizeof(int);
 	key.mv_data = sval;
-	data.mv_size = sizeof(sval);
-	data.mv_data = sval;
 
 	printf("Adding %d values\n", count);
 	for (i=0;i<count;i++) {
 		sprintf(sval, "%03x %d foo bar", values[i], values[i]);
+		/* Set <data> in each iteration, since MDB_NOOVERWRITE may modify it */
+		data.mv_size = sizeof(sval);
+		data.mv_data = sval;
 		if (RES(MDB_KEYEXIST, mdb_put(txn, dbi, &key, &data, MDB_NOOVERWRITE))) {
 			j++;
 			data.mv_size = sizeof(sval);
@@ -130,6 +131,7 @@ int main(int argc,char * argv[])
 			(int) key.mv_size,  (char *) key.mv_data,
 			(int) data.mv_size, (char *) data.mv_data);
 
+	mdb_cursor_close(cursor);
 	mdb_txn_abort(txn);
 
 	printf("Deleting with cursor\n");
diff --git a/mtest2.c b/mtest2.c
index d04e08ae..e1ab1ca8 100644
--- a/mtest2.c
+++ b/mtest2.c
@@ -59,12 +59,12 @@ int main(int argc,char * argv[])
 
 	key.mv_size = sizeof(int);
 	key.mv_data = sval;
-	data.mv_size = sizeof(sval);
-	data.mv_data = sval;
 
 	printf("Adding %d values\n", count);
 	for (i=0;i<count;i++) {
 		sprintf(sval, "%03x %d foo bar", values[i], values[i]);
+		data.mv_size = sizeof(sval);
+		data.mv_data = sval;
 		if (RES(MDB_KEYEXIST, mdb_put(txn, dbi, &key, &data, MDB_NOOVERWRITE)))
 			j++;
 	}
diff --git a/mtest6.c b/mtest6.c
index 76b0a9c1..a1255bca 100644
--- a/mtest6.c
+++ b/mtest6.c
@@ -31,7 +31,7 @@ int main(int argc,char * argv[])
 	int i = 0, rc;
 	MDB_env *env;
 	MDB_dbi dbi;
-	MDB_val key, data;
+	MDB_val key, data, sdata;
 	MDB_txn *txn;
 	MDB_stat mst;
 	MDB_cursor *cursor;
@@ -55,25 +55,28 @@ int main(int argc,char * argv[])
 	sval = calloc(1, mst.ms_psize / 4);
 	key.mv_size = sizeof(long);
 	key.mv_data = &kval;
-	data.mv_size = mst.ms_psize / 4 - 30;
-	data.mv_data = sval;
+	sdata.mv_size = mst.ms_psize / 4 - 30;
+	sdata.mv_data = sval;
 
 	printf("Adding 12 values, should yield 3 splits\n");
 	for (i=0;i<12;i++) {
 		kval = i*5;
 		sprintf(sval, "%08lx", kval);
+		data = sdata;
 		(void)RES(MDB_KEYEXIST, mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE));
 	}
 	printf("Adding 12 more values, should yield 3 splits\n");
 	for (i=0;i<12;i++) {
 		kval = i*5+4;
 		sprintf(sval, "%08lx", kval);
+		data = sdata;
 		(void)RES(MDB_KEYEXIST, mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE));
 	}
 	printf("Adding 12 more values, should yield 3 splits\n");
 	for (i=0;i<12;i++) {
 		kval = i*5+1;
 		sprintf(sval, "%08lx", kval);
+		data = sdata;
 		(void)RES(MDB_KEYEXIST, mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE));
 	}
 	E(mdb_cursor_get(cursor, &key, &data, MDB_FIRST));