Find and fix more GPL2 -> GPL3.
[ira/wip.git] / examples / scripts / users_and_groups / createdomobj.pl
1 #!/usr/bin/perl
2
3 #
4 # createdomobj.pl
5 #
6 #    create single or continuously numbered domain 
7 #    users/groups/aliases via rpc
8 #
9 # Copyright (C) Michael Adam <obnox@samba.org> 2007
10 #
11 # This program is free software; you can redistribute it and/or modify it
12 # under the terms of the GNU General Public License as published by the Free
13 # Software Foundation; either version 3 of the License, or (at your option)
14 # any later version.
15 #
16 # This program is distributed in the hope that it will be useful, but WITHOUT
17 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
18 # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
19 # more details.
20 #
21 # You should have received a copy of the GNU General Public License along with
22 # this program; if not, write to the Free Software Foundation, Inc., 675
23 # Mass Ave, Cambridge, MA 02139, USA.
24 #
25
26 #
27 # WARNING: This script is still rather crude.
28 #
29
30 use strict;
31 use Getopt::Std;
32
33
34 my $target_type = "group";              # what type of object to create
35 my $rpc_cmd     = "createdom".$target_type;
36 my $rpccli_cmd  = "rpcclient";
37
38 # defaults:
39
40 my $server;
41 my $num_targets = 1;
42 my $startnum;                           # if empty, don't add numbers to prefix
43 my $prefix      = $target_type;         # name-prefix
44 my $path;                               # path to rpcclient command
45 my $rpccli_path = $rpccli_cmd;
46 my $creds;
47
48 sub usage {
49         print "USAGE: $0 [-h] -S server -U user\%pass [-p prefix] \\\n"
50                 . "\t[-t {alias|group|user}] [-s startnum] [-n numobjs] [-P path] \n";
51 }
52
53 # parse commandline:
54
55 my %options = ();
56 getopts("U:t:S:s:n:p:P:h", \%options);
57
58 if (exists($options{h})) {
59         usage();
60         exit 0;
61 }
62
63 if (exists($options{t})) {
64         $target_type = $options{t};
65         if ($target_type !~ /^(alias|user|group)$/) {
66                 print "ERROR: invalid target type given\n";
67                 usage();
68                 exit 1;
69         }
70         $rpc_cmd = "createdom".$target_type;
71 }
72
73 if (exists($options{U})) {
74         $creds = "-U $options{U}";
75         if ($creds !~ '%') {
76                 print "ERROR: you need to specify credentials in the form -U user\%pass\n";
77                 usage();
78                 exit 1;
79         }
80 }
81 else {
82         print "ERROR: mandatory argument '-U' missing\n";
83         usage();
84         exit 1;
85 }
86
87 if (exists($options{S})) {
88         $server = $options{S};
89 }
90 else {
91         print "ERROR: madatory argument '-S' missing\n";
92         usage();
93         exit 1;
94 }
95
96 if (exists($options{s})) {
97         $startnum = $options{s};
98 }
99
100 if (exists($options{n})) {
101         $num_targets = $options{n};
102 }
103
104 if (exists($options{p})) {
105         $prefix = $options{p};
106 }
107
108 if (exists($options{P})) {
109         $path = $options{p};
110         $rpccli_path = "$path/$rpccli_cmd";
111 }
112
113 if (@ARGV) {
114         print "ERROR: junk on the command line ('" . join(" ", @ARGV) . "')...\n";
115         usage();
116         exit 1;
117 }
118
119 # utility functions:
120
121 sub open_rpc_pipe {
122         print "opening rpc pipe\n";
123         open(IPC, "| $rpccli_cmd $server $creds -d0") or
124                 die "error opening rpc pipe.";
125 }
126
127 sub close_rpc_pipe {
128         print "closing rpc pipe\n";
129         close(IPC);
130 }
131
132 sub do_create {
133         my $target_name = shift;
134         print "creating $target_type $target_name\n";
135         print IPC "$rpc_cmd $target_name\n";
136 }
137
138 # main:
139
140 open_rpc_pipe();
141
142 if ("x$startnum" eq "x") {
143         do_create($prefix);
144 }
145 else {
146         for (my $num = 1; $num <= $num_targets; ++$num) {
147                 do_create(sprintf "%s%.05d", $prefix, $startnum + $num - 1);
148                 if (($num) % 500 == 0) {
149                         printf("500 ".$target_type."s created\n");
150                         close_rpc_pipe();
151                         sleep 2;
152                         open_rpc_pipe();
153                 }
154         }
155 }
156
157 close_rpc_pipe();
158