Move common libraries from root to lib/.
[kai/samba-autobuild/.git] / lib / nss_wrapper / nss_wrapper.pl
1 #!/usr/bin/perl
2 #
3
4 use strict;
5
6 use Getopt::Long;
7 use Cwd qw(abs_path);
8
9 my $opt_help = 0;
10 my $opt_path = undef;
11 my $opt_action = undef;
12 my $opt_type = undef;
13 my $opt_name = undef;
14
15 my $passwdfn = undef;
16 my $groupfn = undef;
17 my $actionfn = undef;
18
19 sub passwd_add($$);
20 sub passwd_delete($$);
21 sub group_add($$);
22 sub group_delete($$);
23
24 my $result = GetOptions(
25         'help|h|?'      => \$opt_help,
26         'path=s'        => \$opt_path,
27         'action=s'      => \$opt_action,
28         'type=s'        => \$opt_type,
29         'name=s'        => \$opt_name
30 );
31
32 sub usage($;$)
33 {
34         my ($ret, $msg) = @_;
35
36         print $msg."\n\n" if defined($msg);
37
38         print "usage:
39
40         --help|-h|-?            Show this help.
41
42         --path <path>           Path of the 'passwd' or 'group' file.
43
44         --type <type>           Only 'passwd' is supported yet,
45                                 but 'group' and maybe 'member' will be added
46                                 in future.
47
48         --action <action>       'add' or 'delete'.
49
50         --name <name>           The name of the object.
51 ";
52         exit($ret);
53 }
54
55 usage(1) if (not $result);
56
57 usage(0) if ($opt_help);
58
59 if (not defined($opt_path)) {
60         usage(1, "missing: --path <path>");
61 }
62 if ($opt_path eq "" or $opt_path eq "/") {
63         usage(1, "invalid: --path <path>: '$opt_path'");
64 }
65 my $opt_fullpath = abs_path($opt_path);
66 if (not defined($opt_fullpath)) {
67         usage(1, "invalid: --path <path>: '$opt_path'");
68 }
69
70
71 if (not defined($opt_action)) {
72         usage(1, "missing: --action [add|delete]");
73 }
74 if ($opt_action eq "add") {
75         $passwdfn = \&passwd_add;
76         $groupfn = \&group_add;
77 } elsif ($opt_action eq "delete") {
78         $passwdfn = \&passwd_delete;
79         $groupfn = \&group_delete;
80 } else {
81         usage(1, "invalid: --action [add|delete]: '$opt_action'");
82 }
83
84 if (not defined($opt_type)) {
85         usage(1, "missing: --type [passwd|group]");
86 }
87 if ($opt_type eq "passwd") {
88         $actionfn = $passwdfn;
89 } elsif ($opt_type eq "group") {
90         $actionfn = $groupfn;
91 } else {
92         usage(1, "invalid: --type [passwd|group]: '$opt_type'")
93 }
94
95 if (not defined($opt_name)) {
96         usage(1, "missing: --name <name>");
97 }
98 if ($opt_name eq "") {
99         usage(1, "invalid: --name <name>");
100 }
101
102 exit $actionfn->($opt_fullpath, $opt_name);
103
104 sub passwd_add_entry($$);
105
106 sub passwd_load($)
107 {
108         my ($path) = @_;
109         my @lines;
110         my $passwd = undef;
111
112         open(PWD, "<$path") or die("Unable to open '$path' for read");
113         @lines = <PWD>;
114         close(PWD);
115
116         $passwd->{array} = ();
117         $passwd->{name} = {};
118         $passwd->{uid} = {};
119         $passwd->{path} = $path;
120
121         foreach my $line (@lines) {
122                 passwd_add_entry($passwd, $line);
123         }
124
125         return $passwd;
126 }
127
128 sub passwd_lookup_name($$)
129 {
130         my ($passwd, $name) = @_;
131
132         return undef unless defined($passwd->{name}{$name});
133
134         return $passwd->{name}{$name};
135 }
136
137 sub passwd_lookup_uid($$)
138 {
139         my ($passwd, $uid) = @_;
140
141         return undef unless defined($passwd->{uid}{$uid});
142
143         return $passwd->{uid}{$uid};
144 }
145
146 sub passwd_get_free_uid($)
147 {
148         my ($passwd) = @_;
149         my $uid = 1000;
150
151         while (passwd_lookup_uid($passwd, $uid)) {
152                 $uid++;
153         }
154
155         return $uid;
156 }
157
158 sub passwd_add_entry($$)
159 {
160         my ($passwd, $str) = @_;
161
162         chomp $str;
163         my @e = split(':', $str);
164
165         push(@{$passwd->{array}}, \@e);
166         $passwd->{name}{$e[0]} = \@e;
167         $passwd->{uid}{$e[2]} = \@e;
168 }
169
170 sub passwd_remove_entry($$)
171 {
172         my ($passwd, $eref) = @_;
173
174         for(my $i; defined($passwd->{array}[$i]); $i++) {
175                 if ($eref == $passwd->{array}[$i]) {
176                         $passwd->{array}[$i] = undef;
177                 }
178         }
179
180         delete $passwd->{name}{${$eref}[0]};
181         delete $passwd->{uid}{${$eref}[2]};
182 }
183
184 sub passwd_save($)
185 {
186         my ($passwd) = @_;
187         my @lines = ();
188         my $path = $passwd->{path};
189         my $tmppath = $path.$$;
190
191         foreach my $eref (@{$passwd->{array}}) {
192                 next unless defined($eref);
193
194                 my $line = join(':', @{$eref});
195                 push(@lines, $line);
196         }
197
198         open(PWD, ">$tmppath") or die("Unable to open '$tmppath' for write");
199         print PWD join("\n", @lines)."\n";
200         close(PWD);
201         rename($tmppath, $path) or die("Unable to rename $tmppath => $path");
202 }
203
204 sub passwd_add($$)
205 {
206         my ($path, $name) = @_;
207
208         #print "passwd_add: '$name' in '$path'\n";
209
210         my $passwd = passwd_load($path);
211
212         my $e = passwd_lookup_name($passwd, $name);
213         die("account[$name] already exists in '$path'") if defined($e);
214
215         my $uid = passwd_get_free_uid($passwd);
216         my $gid = 65534;# nogroup gid
217
218         my $pwent = $name.":x:".$uid.":".$gid.":".$name." gecos:/nodir:/bin/false";
219
220         passwd_add_entry($passwd, $pwent);
221
222         passwd_save($passwd);
223
224         return 0;
225 }
226
227 sub passwd_delete($$)
228 {
229         my ($path, $name) = @_;
230
231         #print "passwd_delete: '$name' in '$path'\n";
232
233         my $passwd = passwd_load($path);
234
235         my $e = passwd_lookup_name($passwd, $name);
236         die("account[$name] does not exists in '$path'") unless defined($e);
237
238         passwd_remove_entry($passwd, $e);
239
240         passwd_save($passwd);
241
242         return 0;
243 }
244
245 sub group_add($$)
246 {
247         my ($path, $name) = @_;
248
249         #print "group_add: '$name' in '$path'\n";
250
251         die("group_add: not implemented yet!");
252
253         return 0;
254 }
255
256 sub group_delete($$)
257 {
258         my ($path, $name) = @_;
259
260         #print "group_delete: '$name' in '$path'\n";
261
262         die("group_delete: not implemented yet!");
263
264         return 0;
265 }