summaryrefslogtreecommitdiff
path: root/source4/scripting/devel/addlotscontacts
diff options
context:
space:
mode:
authorMatthieu Patou <mat@matws.net>2011-11-11 19:01:54 +0100
committerMatthieu Patou <mat@samba.org>2011-11-11 22:02:53 +0100
commit691fb365f8405d92a269abcef1028aa648d12580 (patch)
tree305354fb840a74727226239a2ba7a3cdc67ab930 /source4/scripting/devel/addlotscontacts
parent7c9b3cdcf79088ff284d2383676d3f26ae42f8fb (diff)
downloadsamba-691fb365f8405d92a269abcef1028aa648d12580.tar.gz
samba-691fb365f8405d92a269abcef1028aa648d12580.tar.bz2
samba-691fb365f8405d92a269abcef1028aa648d12580.zip
s4: Simple test script to create lots of contacts to stress the LDB
Autobuild-User: Matthieu Patou <mat@samba.org> Autobuild-Date: Fri Nov 11 22:02:53 CET 2011 on sn-devel-104
Diffstat (limited to 'source4/scripting/devel/addlotscontacts')
-rw-r--r--source4/scripting/devel/addlotscontacts96
1 files changed, 96 insertions, 0 deletions
diff --git a/source4/scripting/devel/addlotscontacts b/source4/scripting/devel/addlotscontacts
new file mode 100644
index 0000000000..cc073a3500
--- /dev/null
+++ b/source4/scripting/devel/addlotscontacts
@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+#
+# Copyright (C) Matthieu Patou <mat@matws.net> 2010
+#
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+
+
+__docformat__ = "restructuredText"
+
+
+import optparse
+import sys
+# Allow to run from s4 source directory (without installing samba)
+sys.path.insert(0, "bin/python")
+
+import samba.getopt as options
+from samba.credentials import DONT_USE_KERBEROS
+from samba.auth import system_session
+from samba import param
+from samba.provision import find_provision_key_parameters
+from samba.upgradehelpers import (get_paths, get_ldbs)
+from ldb import SCOPE_BASE, Message, MessageElement, Dn, FLAG_MOD_ADD
+
+parser = optparse.OptionParser("chgtdcpass [options]")
+sambaopts = options.SambaOptions(parser)
+parser.add_option_group(sambaopts)
+parser.add_option_group(options.VersionOptions(parser))
+credopts = options.CredentialsOptions(parser)
+parser.add_option_group(credopts)
+
+(opts, args) = parser.parse_args()
+
+lp = sambaopts.get_loadparm()
+smbconf = lp.configfile
+creds = credopts.get_credentials(lp)
+creds.set_kerberos_state(DONT_USE_KERBEROS)
+
+if len(args) > 0:
+ num_contacts = int(args[0])
+else:
+ num_contacts = 10000
+
+if __name__ == '__main__':
+ paths = get_paths(param, smbconf=smbconf)
+ session = system_session()
+
+ ldbs = get_ldbs(paths, creds, session, lp)
+ ldbs.startTransactions()
+
+ names = find_provision_key_parameters(ldbs.sam, ldbs.secrets, ldbs.idmap,
+ paths, smbconf, lp)
+
+ contactdn = "OU=Contacts,%s" % str(names.domaindn)
+ res = ldbs.sam.search(expression="(dn=%s)" % contactdn,
+ base=str(names.domaindn),
+ scope=SCOPE_BASE)
+
+ if (len(res) == 0):
+ msg = Message()
+ msg.dn = Dn(ldbs.sam, contactdn)
+ msg["objectClass"] = MessageElement("organizationalUnit", FLAG_MOD_ADD,
+ "objectClass")
+
+ ldbs.sam.add(msg)
+
+ print "Creating %d contacts" % num_contacts
+ count = 0
+ increment = num_contacts / 10
+ if increment > 5000:
+ increment = 5000
+
+ while (count < num_contacts):
+ msg = Message()
+ msg.dn = Dn(ldbs.sam, "CN=contact%d,%s" % (count + 1, contactdn))
+ msg["objectClass"] = MessageElement("contact", FLAG_MOD_ADD,
+ "objectClass")
+
+ if count !=0 and (count % increment) == 0:
+ print "Added contacts: %d" % count
+
+ ldbs.sam.add(msg)
+ count += 1
+
+ ldbs.groupedCommit()