pyldb: avoid segfault when adding an element with no name
[kai/samba-autobuild/.git] / source4 / scripting / bin / mymachinepw
1 #!/usr/bin/env python3
2
3 # Unix SMB/CIFS implementation.
4 # Copyright (C) Volker Lendecke 2008
5 # Copyright (C) Stefan Metzmacher 2008
6 #
7 # Extract our own machine pw from secrets.ldb
8 #
9 # This program is free software; you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation; either version 3 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 # GNU General Public License for more details.
18 #
19 # You should have received a copy of the GNU General Public License
20 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 #
22 import samba.param as param, ldb, sys, getopt
23
24 optlist, args = getopt.getopt(sys.argv[1:], "s:")
25
26 conf = param.LoadParm()
27 loaded = False
28
29 for o, v in optlist:
30     if o == "-s":
31         if not conf.load(v):
32             print(v + " not found")
33             exit(1)
34         loaded = True
35
36 if not loaded:
37     conf.load_default()
38
39 path=conf.get("private dir") + "/secrets.ldb"
40 netbios=conf.get("netbios name")
41
42 secrets = ldb.Ldb(path)
43
44 search = ("(&(objectclass=primaryDomain)(samaccountname=" +
45       netbios + "$))")
46
47 msg = secrets.search(expression=search, attrs=['secret'])
48
49 if not msg:
50     print("Error:")
51     print("Password for host[%s] not found in path[%s]." % (netbios, path))
52     print("You may want to pass the smb.conf location via the -s option.")
53     exit(1)
54
55 password=msg[0]['secret'][0]
56
57 print(password)
58 exit(0)