Fix memleak
[ira/wip.git] / source3 / utils / net_groupmap.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  RPC Pipe client / server routines
4  *  Copyright (C) Andrew Tridgell              1992-2000,
5  *  Copyright (C) Jean François Micouleau      1998-2001.
6  *  Copyright (C) Gerald Carter                2003.
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21  */
22
23
24 #include "includes.h"
25 #include "../utils/net.h"
26
27
28 /*********************************************************
29  utility function to parse an integer parameter from 
30  "parameter = value"
31 **********************************************************/
32 static uint32 get_int_param( const char* param )
33 {
34         char *p;
35         
36         p = strchr( param, '=' );
37         if ( !p )
38                 return 0;
39                 
40         return atoi(p+1);
41 }
42
43 /*********************************************************
44  utility function to parse an integer parameter from 
45  "parameter = value"
46 **********************************************************/
47 static char* get_string_param( const char* param )
48 {
49         char *p;
50         
51         p = strchr( param, '=' );
52         if ( !p )
53                 return NULL;
54                 
55         return (p+1);
56 }
57
58 /*********************************************************
59  Figure out if the input was an NT group or a SID string.
60  Return the SID.
61 **********************************************************/
62 static BOOL get_sid_from_input(DOM_SID *sid, char *input)
63 {
64         GROUP_MAP map;
65
66         if (StrnCaseCmp( input, "S-", 2)) {
67                 /* Perhaps its the NT group name? */
68                 if (!pdb_getgrnam(&map, input)) {
69                         printf("NT Group %s doesn't exist in mapping DB\n", input);
70                         return False;
71                 } else {
72                         *sid = map.sid;
73                 }
74         } else {
75                 if (!string_to_sid(sid, input)) {
76                         printf("converting sid %s from a string failed!\n", input);
77                         return False;
78                 }
79         }
80         return True;
81 }
82
83 /*********************************************************
84  Dump a GROUP_MAP entry to stdout (long or short listing)
85 **********************************************************/
86
87 static void print_map_entry ( GROUP_MAP map, BOOL long_list )
88 {
89         fstring string_sid;
90         fstring group_type;
91         
92         decode_sid_name_use(group_type, map.sid_name_use);
93         sid_to_string(string_sid, &map.sid);
94                 
95         if (!long_list)
96                 d_printf("%s (%s) -> %s\n", map.nt_name, string_sid, gidtoname(map.gid));
97         else {
98                 d_printf("%s\n", map.nt_name);
99                 d_printf("\tSID       : %s\n", string_sid);
100                 d_printf("\tUnix group: %s\n", gidtoname(map.gid));
101                 d_printf("\tGroup type: %s\n", group_type);
102                 d_printf("\tComment   : %s\n", map.comment);
103         }
104
105 }
106 /*********************************************************
107  List the groups.
108 **********************************************************/
109 static int net_groupmap_list(int argc, const char **argv)
110 {
111         int entries;
112         BOOL long_list = False;
113         int i;
114         fstring ntgroup = "";
115         fstring sid_string = "";
116         
117         /* get the options */
118         for ( i=0; i<argc; i++ ) {
119                 if ( !StrCaseCmp(argv[i], "verbose")) {
120                         long_list = True;
121                 }
122                 else if ( !StrnCaseCmp(argv[i], "ntgroup", strlen("ntgroup")) ) {
123                         fstrcpy( ntgroup, get_string_param( argv[i] ) );
124                         if ( !ntgroup[0] ) {
125                                 d_printf("must supply a name\n");
126                                 return -1;
127                         }               
128                 }
129                 else if ( !StrnCaseCmp(argv[i], "sid", strlen("sid")) ) {
130                         fstrcpy( sid_string, get_string_param( argv[i] ) );
131                         if ( !sid_string[0] ) {
132                                 d_printf("must supply a SID\n");
133                                 return -1;
134                         }               
135                 }
136                 else {
137                         d_printf("Bad option: %s\n", argv[i]);
138                         return -1;
139                 }
140         }
141
142         /* list a single group is given a name */
143         if ( ntgroup[0] || sid_string[0] ) {
144                 DOM_SID sid;
145                 GROUP_MAP map;
146                 
147                 if ( sid_string[0] )
148                         fstrcpy( ntgroup, sid_string);
149                         
150                 if (!get_sid_from_input(&sid, ntgroup)) {
151                         return -1;
152                 }
153
154                 /* Get the current mapping from the database */
155                 if(!pdb_getgrsid(&map, sid)) {
156                         d_printf("Failure to local group SID in the database\n");
157                         return -1;
158                 }
159         
160                 print_map_entry( map, long_list );
161         }
162         else {
163                 GROUP_MAP *map=NULL;
164                 /* enumerate all group mappings */
165                 if (!pdb_enum_group_mapping(SID_NAME_UNKNOWN, &map, &entries, ENUM_ALL_MAPPED))
166                         return -1;
167         
168                 for (i=0; i<entries; i++) {
169                         print_map_entry( map[i], long_list );
170                 }
171
172                 SAFE_FREE(map);
173         }
174
175         return 0;
176 }
177
178 /*********************************************************
179  Add a new group mapping entry
180 **********************************************************/
181
182 static int net_groupmap_add(int argc, const char **argv)
183 {
184         DOM_SID sid;
185         fstring ntgroup = "";
186         fstring unixgrp = "";
187         fstring string_sid = "";
188         fstring type = "";
189         fstring ntcomment = "";
190         enum SID_NAME_USE sid_type = SID_NAME_DOM_GRP;
191         uint32 rid = 0; 
192         gid_t gid;
193         int i;
194         
195         /* get the options */
196         for ( i=0; i<argc; i++ ) {
197                 if ( !StrnCaseCmp(argv[i], "rid", strlen("rid")) ) {
198                         rid = get_int_param(argv[i]);
199                         if ( rid < DOMAIN_GROUP_RID_ADMINS ) {
200                                 d_printf("RID must be greater than %d\n", (uint32)DOMAIN_GROUP_RID_ADMINS-1);
201                                 return -1;
202                         }
203                 }
204                 else if ( !StrnCaseCmp(argv[i], "unixgroup", strlen("unixgroup")) ) {
205                         fstrcpy( unixgrp, get_string_param( argv[i] ) );
206                         if ( !unixgrp[0] ) {
207                                 d_printf("must supply a name\n");
208                                 return -1;
209                         }               
210                 }
211                 else if ( !StrnCaseCmp(argv[i], "ntgroup", strlen("ntgroup")) ) {
212                         fstrcpy( ntgroup, get_string_param( argv[i] ) );
213                         if ( !ntgroup[0] ) {
214                                 d_printf("must supply a name\n");
215                                 return -1;
216                         }               
217                 }
218                 else if ( !StrnCaseCmp(argv[i], "sid", strlen("sid")) ) {
219                         fstrcpy( string_sid, get_string_param( argv[i] ) );
220                         if ( !string_sid[0] ) {
221                                 d_printf("must supply a SID\n");
222                                 return -1;
223                         }               
224                 }
225                 else if ( !StrnCaseCmp(argv[i], "comment", strlen("comment")) ) {
226                         fstrcpy( ntcomment, get_string_param( argv[i] ) );
227                         if ( !ntcomment[0] ) {
228                                 d_printf("must supply a comment string\n");
229                                 return -1;
230                         }                               
231                 }
232                 else if ( !StrnCaseCmp(argv[i], "type", strlen("type")) )  {
233                         fstrcpy( type, get_string_param( argv[i] ) );
234                         switch ( type[0] ) {
235                                 case 'b':
236                                 case 'B':
237                                         sid_type = SID_NAME_WKN_GRP;
238                                         break;
239                                 case 'd':
240                                 case 'D':
241                                         sid_type = SID_NAME_DOM_GRP;
242                                         break;
243                                 case 'l':
244                                 case 'L':
245                                         sid_type = SID_NAME_ALIAS;
246                                         break;
247                         }
248                 }
249                 else {
250                         d_printf("Bad option: %s\n", argv[i]);
251                         return -1;
252                 }
253         }
254
255         if ( !unixgrp[0] || (!rid && !string_sid[0]) ) {
256                 d_printf("Usage: net groupmap add {rid=<int>|sid=<string>} unixgroup=<string> [type=<domain|local|builtin>] [ntgroup=<string>] [comment=<string>]\n");
257                 return -1;
258         }
259         
260         /* append the rid to our own domain/machine SID if we don't have a full SID */
261         if ( !string_sid[0] ) {
262                 sid_copy(&sid, get_global_sam_sid());
263                 sid_append_rid(&sid, rid);
264                 sid_to_string(string_sid, &sid);
265         }
266
267         if (ntcomment[0])
268                 fstrcpy(ntcomment, "Local Unix group");
269                 
270         if ( (gid = nametogid(unixgrp)) == (gid_t)-1 ) {
271                 d_printf("Can't lookup UNIX group %s\n", ntgroup);
272                 return -1;
273         }
274         
275         if ( !ntgroup[0] )
276                 fstrcpy( ntgroup, unixgrp );
277                 
278         
279         if (!add_initial_entry(gid, string_sid, sid_type, ntgroup, ntcomment)) {
280                 d_printf("adding entry for group %s failed!\n", ntgroup);
281                 return -1;
282         }
283
284         d_printf("Successully added group %s to the mapping db\n", ntgroup);
285         return 0;
286 }
287
288 static int net_groupmap_modify(int argc, const char **argv)
289 {
290         DOM_SID sid;
291         GROUP_MAP map;
292         fstring ntcomment = "";
293         fstring type = "";
294         fstring ntgroup = "";
295         fstring unixgrp = "";
296         fstring sid_string = "";
297         enum SID_NAME_USE sid_type = SID_NAME_UNKNOWN;
298         int i;
299         gid_t gid;
300
301         /* get the options */
302         for ( i=0; i<argc; i++ ) {
303                 if ( !StrnCaseCmp(argv[i], "ntgroup", strlen("ntgroup")) ) {
304                         fstrcpy( ntgroup, get_string_param( argv[i] ) );
305                         if ( !ntgroup[0] ) {
306                                 d_printf("must supply a name\n");
307                                 return -1;
308                         }               
309                 }
310                 else if ( !StrnCaseCmp(argv[i], "sid", strlen("sid")) ) {
311                         fstrcpy( sid_string, get_string_param( argv[i] ) );
312                         if ( !sid_string[0] ) {
313                                 d_printf("must supply a name\n");
314                                 return -1;
315                         }               
316                 }
317                 else if ( !StrnCaseCmp(argv[i], "comment", strlen("comment")) ) {
318                         fstrcpy( ntcomment, get_string_param( argv[i] ) );
319                         if ( !ntcomment[0] ) {
320                                 d_printf("must supply a comment string\n");
321                                 return -1;
322                         }                               
323                 }
324                 else if ( !StrnCaseCmp(argv[i], "unixgroup", strlen("unixgroup")) ) {
325                         fstrcpy( unixgrp, get_string_param( argv[i] ) );
326                         if ( !unixgrp[0] ) {
327                                 d_printf("must supply a group name\n");
328                                 return -1;
329                         }                               
330                 }
331                 else if ( !StrnCaseCmp(argv[i], "type", strlen("type")) )  {
332                         fstrcpy( type, get_string_param( argv[i] ) );
333                         switch ( type[0] ) {
334                                 case 'd':
335                                 case 'D':
336                                         sid_type = SID_NAME_DOM_GRP;
337                                         break;
338                                 case 'l':
339                                 case 'L':
340                                         sid_type = SID_NAME_ALIAS;
341                                         break;
342                         }
343                 }
344                 else {
345                         d_printf("Bad option: %s\n", argv[i]);
346                         return -1;
347                 }
348         }
349         
350         if ( !ntgroup[0] && !sid_string[0] ) {
351                 d_printf("Usage: net groupmap modify {ntgroup=<string>|sid=<SID>} [comment=<string>] [unixgroup=<string>] [type=<domain|local>]\n");
352                 return -1;
353         }
354
355         /* give preference to the SID; if both the ntgroup name and SID
356            are defined, use the SID and assume that the group name could be a 
357            new name */
358                 
359         if ( sid_string[0] ) {  
360                 if (!get_sid_from_input(&sid, sid_string)) {
361                         return -1;
362                 }
363         }
364         else {
365                 if (!get_sid_from_input(&sid, ntgroup)) {
366                         return -1;
367                 }
368         }       
369
370         /* Get the current mapping from the database */
371         if(!pdb_getgrsid(&map, sid)) {
372                 d_printf("Failure to local group SID in the database\n");
373                 return -1;
374         }
375         
376         /*
377          * Allow changing of group type only between domain and local
378          * We disallow changing Builtin groups !!! (SID problem)
379          */ 
380         if (sid_type != SID_NAME_UNKNOWN) { 
381                 if (map.sid_name_use == SID_NAME_WKN_GRP) {
382                         d_printf("You can only change between domain and local groups.\n");
383                         return -1;
384                 }
385                 
386                 map.sid_name_use=sid_type;
387         }
388
389         /* Change comment if new one */
390         if ( ntcomment[0] )
391                 fstrcpy( map.comment, ntcomment );
392                 
393         if ( ntgroup[0] )
394                 fstrcpy( map.nt_name, ntgroup );
395                 
396         if ( unixgrp[0] ) {
397                 gid = nametogid( unixgrp );
398                 if ( gid == -1 ) {
399                         d_printf("Unable to lookup UNIX group %s.  Make sure the group exists.\n",
400                                 unixgrp);
401                         return -1;
402                 }
403                 
404                 map.gid = gid;
405         }
406
407         if ( !pdb_update_group_mapping_entry(&map) ) {
408                 d_printf("Could not update group database\n");
409                 return -1;
410         }
411         
412         d_printf("Updated mapping entry for %s\n", ntgroup);
413
414         return 0;
415 }
416
417 static int net_groupmap_delete(int argc, const char **argv)
418 {
419         DOM_SID sid;
420         fstring ntgroup = "";
421         fstring sid_string = "";
422         int i;
423
424         /* get the options */
425         for ( i=0; i<argc; i++ ) {
426                 if ( !StrnCaseCmp(argv[i], "ntgroup", strlen("ntgroup")) ) {
427                         fstrcpy( ntgroup, get_string_param( argv[i] ) );
428                         if ( !ntgroup[0] ) {
429                                 d_printf("must supply a name\n");
430                                 return -1;
431                         }               
432                 }
433                 else if ( !StrnCaseCmp(argv[i], "sid", strlen("sid")) ) {
434                         fstrcpy( sid_string, get_string_param( argv[i] ) );
435                         if ( !sid_string[0] ) {
436                                 d_printf("must supply a SID\n");
437                                 return -1;
438                         }               
439                 }
440                 else {
441                         d_printf("Bad option: %s\n", argv[i]);
442                         return -1;
443                 }
444         }
445         
446         if ( !ntgroup[0] && !sid_string[0]) {
447                 d_printf("Usage: net groupmap delete {ntgroup=<string>|sid=<SID>}\n");
448                 return -1;
449         }
450         
451         /* give preference to the SID if we have that */
452         
453         if ( sid_string[0] )
454                 fstrcpy( ntgroup, sid_string );
455                 
456         if ( !get_sid_from_input(&sid, ntgroup) ) {
457                 d_printf("Unable to resolve group %s to a SID\n", ntgroup);
458                 return -1;
459         }
460
461         if ( !pdb_delete_group_mapping_entry(sid) ) {
462                 printf("Failed to removing group %s from the mapping db!\n", ntgroup);
463                 return -1;
464         }
465
466         d_printf("Sucessfully removed %s from the mapping db\n", ntgroup);
467
468         return 0;
469 }
470
471 int net_help_groupmap(int argc, const char **argv)
472 {
473         d_printf("net groupmap add"\
474                 "\n  Create a new group mapping\n");
475         d_printf("net groupmap modify"\
476                 "\n  Update a group mapping\n");
477         d_printf("net groupmap delete"\
478                 "\n  Remove a group mapping\n");
479         d_printf("net groupmap list"\
480                 "\n  List current group map\n");
481         
482         return -1;
483 }
484
485
486 /***********************************************************
487  migrated functionality from smbgroupedit
488  **********************************************************/
489 int net_groupmap(int argc, const char **argv)
490 {
491         struct functable func[] = {
492                 {"add", net_groupmap_add},
493                 {"modify", net_groupmap_modify},
494                 {"delete", net_groupmap_delete},
495                 {"list", net_groupmap_list},
496                 {"help", net_help_groupmap},
497                 {NULL, NULL}
498         };
499
500         /* we shouldn't have silly checks like this */
501         if (getuid() != 0) {
502                 d_printf("You must be root to edit group mappings.\nExiting...\n");
503                 return -1;
504         }
505         
506         return net_run_function(argc, argv, func, net_help_groupmap);
507         if ( 0 == argc )
508                 return net_help_groupmap( argc, argv );
509
510         return net_help_groupmap( argc, argv );
511 }
512