2 ldb database library - command line handling for ldb tools
4 Copyright (C) Andrew Tridgell 2005
6 ** NOTE! The following LGPL license applies to the ldb
7 ** library. This does NOT imply that all of Samba is released
10 This library is free software; you can redistribute it and/or
11 modify it under the terms of the GNU Lesser General Public
12 License as published by the Free Software Foundation; either
13 version 2 of the License, or (at your option) any later version.
15 This library is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 Lesser General Public License for more details.
20 You should have received a copy of the GNU Lesser General Public
21 License along with this library; if not, write to the Free Software
22 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 #include "ldb/include/includes.h"
27 #include "ldb/tools/cmdline.h"
30 #include "lib/cmdline/popt_common.h"
31 #include "auth/auth.h"
35 process command line options
37 struct ldb_cmdline
*ldb_cmdline_process(struct ldb_context
*ldb
, int argc
, const char **argv
,
40 struct ldb_cmdline options
, *ret
=NULL
;
47 struct poptOption popt_options
[] = {
49 { "url", 'H', POPT_ARG_STRING
, &options
.url
, 0, "database URL", "URL" },
50 { "basedn", 'b', POPT_ARG_STRING
, &options
.basedn
, 0, "base DN", "DN" },
51 { "editor", 'e', POPT_ARG_STRING
, &options
.editor
, 0, "external editor", "PROGRAM" },
52 { "scope", 's', POPT_ARG_STRING
, NULL
, 's', "search scope", "SCOPE" },
53 { "verbose", 'v', POPT_ARG_NONE
, NULL
, 'v', "increase verbosity", NULL
},
54 { "interactive", 'i', POPT_ARG_NONE
, &options
.interactive
, 0, "input from stdin", NULL
},
55 { "recursive", 'r', POPT_ARG_NONE
, &options
.recursive
, 0, "recursive delete", NULL
},
56 { "num-searches", 0, POPT_ARG_INT
, &options
.num_searches
, 0, "number of test searches", NULL
},
57 { "num-records", 0, POPT_ARG_INT
, &options
.num_records
, 0, "number of test records", NULL
},
58 { "all", 'a', POPT_ARG_NONE
, &options
.all_records
, 0, "objectClass=*", NULL
},
59 { "nosync", 0, POPT_ARG_NONE
, &options
.nosync
, 0, "non-synchronous transactions", NULL
},
60 { "sorted", 'S', POPT_ARG_NONE
, &options
.sorted
, 0, "sort attributes", NULL
},
61 { "sasl-mechanism", 0, POPT_ARG_STRING
, &options
.sasl_mechanism
, 0, "choose SASL mechanism", "MECHANISM" },
62 { "input", 'I', POPT_ARG_STRING
, &options
.input
, 0, "Input File", "Input" },
63 { "output", 'O', POPT_ARG_STRING
, &options
.output
, 0, "Output File", "Output" },
64 { NULL
, 'o', POPT_ARG_STRING
, NULL
, 'o', "ldb_connect option", "OPTION" },
65 { "controls", 0, POPT_ARG_STRING
, NULL
, 'c', "controls", NULL
},
68 POPT_COMMON_CREDENTIALS
77 r
= ldb_register_samba_handlers(ldb
);
84 ret
= talloc_zero(ldb
, struct ldb_cmdline
);
93 options
.url
= getenv("LDB_URL");
95 /* and editor (used by ldbedit) */
96 options
.editor
= getenv("VISUAL");
97 if (!options
.editor
) {
98 options
.editor
= getenv("EDITOR");
100 if (!options
.editor
) {
101 options
.editor
= "vi";
104 options
.scope
= LDB_SCOPE_DEFAULT
;
106 pc
= poptGetContext(argv
[0], argc
, argv
, popt_options
,
107 POPT_CONTEXT_KEEP_FIRST
);
109 while((opt
= poptGetNextOpt(pc
)) != -1) {
112 const char *arg
= poptGetOptArg(pc
);
113 if (strcmp(arg
, "base") == 0) {
114 options
.scope
= LDB_SCOPE_BASE
;
115 } else if (strcmp(arg
, "sub") == 0) {
116 options
.scope
= LDB_SCOPE_SUBTREE
;
117 } else if (strcmp(arg
, "one") == 0) {
118 options
.scope
= LDB_SCOPE_ONELEVEL
;
120 fprintf(stderr
, "Invalid scope '%s'\n", arg
);
131 options
.options
= talloc_realloc(ret
, options
.options
,
132 const char *, num_options
+3);
133 if (options
.options
== NULL
) {
137 options
.options
[num_options
] = poptGetOptArg(pc
);
138 options
.options
[num_options
+1] = NULL
;
143 const char *cs
= poptGetOptArg(pc
);
147 for (p
= cs
, cc
= 1; (q
= strchr(p
, ',')); cc
++, p
= q
+ 1) ;
149 options
.controls
= talloc_array(ret
, char *, cc
+ 1);
150 if (options
.controls
== NULL
) {
154 for (p
= cs
, cc
= 0; p
!= NULL
; cc
++) {
159 options
.controls
[cc
] = talloc_strdup(options
.controls
, p
);
162 options
.controls
[cc
] = talloc_strndup(options
.controls
, p
, t
-p
);
166 options
.controls
[cc
] = NULL
;
171 fprintf(stderr
, "Invalid option %s: %s\n",
172 poptBadOption(pc
, 0), poptStrerror(opt
));
178 /* setup the remaining options for the main program to use */
179 options
.argv
= poptGetArgs(pc
);
182 while (options
.argv
[options
.argc
]) options
.argc
++;
187 /* all utils need some option */
188 if (ret
->url
== NULL
) {
189 fprintf(stderr
, "You must supply a url with -H or with $LDB_URL\n");
194 if (strcmp(ret
->url
, "NONE") != 0) {
196 if (options
.nosync
) {
197 flags
|= LDB_FLG_NOSYNC
;
201 if (ldb_set_opaque(ldb
, "sessionInfo", system_session(ldb
))) {
204 if (ldb_set_opaque(ldb
, "credentials", cmdline_credentials
)) {
208 if (ldb_connect(ldb
, ret
->url
, flags
, ret
->options
) != 0) {
209 fprintf(stderr
, "Failed to connect to %s - %s\n",
210 ret
->url
, ldb_errstring(ldb
));