Patch from tito to add argument checking.
[platform/upstream/busybox.git] / loginutils / addgroup.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * addgroup - add users to /etc/passwd and /etc/shadow
4  *
5  * Copyright (C) 1999 by Lineo, inc. and John Beppu
6  * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org>
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 GNU
16  * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21  *
22  */
23
24 #include <errno.h>
25 #include <fcntl.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <sys/param.h>
31 #include <sys/stat.h>
32 #include <sys/types.h>
33 #include <unistd.h>
34 #include "busybox.h"
35 #include "pwd_.h"
36 #include "grp_.h"
37
38
39 /* structs __________________________ */
40
41 /* data _____________________________ */
42
43 /* defaults : should this be in an external file? */
44 static const char default_passwd[] = "x";
45
46
47 /* make sure gr_name isn't taken, make sure gid is kosher
48  * return 1 on failure */
49 static int group_study(const char *filename, struct group *g)
50 {
51         FILE *etc_group;
52         gid_t desired;
53
54         struct group *grp;
55         const int max = 65000;
56
57         etc_group = bb_xfopen(filename, "r");
58
59         /* make sure gr_name isn't taken, make sure gid is kosher */
60         desired = g->gr_gid;
61         while ((grp = fgetgrent(etc_group))) {
62                 if ((strcmp(grp->gr_name, g->gr_name)) == 0) {
63                         bb_error_msg_and_die("%s: group already in use\n", g->gr_name);
64                 }
65                 if ((desired) && grp->gr_gid == desired) {
66                         bb_error_msg_and_die("%d: gid has already been allocated\n",
67                                                           desired);
68                 }
69                 if ((grp->gr_gid > g->gr_gid) && (grp->gr_gid < max)) {
70                         g->gr_gid = grp->gr_gid;
71                 }
72         }
73         fclose(etc_group);
74
75         /* gid */
76         if (desired) {
77                 g->gr_gid = desired;
78         } else {
79                 g->gr_gid++;
80         }
81         /* return 1; */
82         return 0;
83 }
84
85 /* append a new user to the passwd file */
86 static int addgroup(const char *filename, char *group, gid_t gid, const char *user)
87 {
88         FILE *etc_group;
89
90 #ifdef CONFIG_FEATURE_SHADOWPASSWDS
91         FILE *etc_gshadow;
92 #endif
93
94         struct group gr;
95
96         /* group:passwd:gid:userlist */
97         static const char entryfmt[] = "%s:%s:%d:%s\n";
98
99         /* make sure gid and group haven't already been allocated */
100         gr.gr_gid = gid;
101         gr.gr_name = group;
102         if (group_study(filename, &gr))
103                 return 1;
104
105         /* add entry to group */
106         etc_group = bb_xfopen(filename, "a");
107
108         fprintf(etc_group, entryfmt, group, default_passwd, gr.gr_gid, user);
109         fclose(etc_group);
110
111
112 #ifdef CONFIG_FEATURE_SHADOWPASSWDS
113         /* add entry to gshadow if necessary */
114         if (access(bb_path_gshadow_file, F_OK|W_OK) == 0) {
115                 etc_gshadow = bb_xfopen(bb_path_gshadow_file, "a");
116                 fprintf(etc_gshadow, "%s:!::\n", group);
117                 fclose(etc_gshadow);
118         }
119 #endif
120
121         /* return 1; */
122         return 0;
123 }
124
125 #ifndef CONFIG_ADDUSER
126 static inline void if_i_am_not_root(void)
127 {
128         if (geteuid()) {
129                 bb_error_msg_and_die( "Only root may add a user or group to the system.");
130         }
131 }
132 #else
133 extern void if_i_am_not_root(void);
134 #endif
135
136 /*
137  * addgroup will take a login_name as its first parameter.
138  *
139  * gid
140  *
141  * can be customized via command-line parameters.
142  * ________________________________________________________________________ */
143 int addgroup_main(int argc, char **argv)
144 {
145         char *group;
146         char *user;
147         gid_t gid = 0;
148
149         /* get remaining args */
150         if(bb_getopt_ulflags(argc, argv, "g:", &group)) {
151                 gid = bb_xgetlarg(group, 10, 0, LONG_MAX);
152         }
153
154         if (optind < argc) {
155                 group = argv[optind];
156                 optind++;
157         } else {
158                 bb_show_usage();
159         }
160
161         if (optind < argc) {
162                 user = argv[optind];
163         } else {
164                 user = "";
165         }
166         
167         if_i_am_not_root();
168
169         /* werk */
170         return addgroup(bb_path_group_file, group, gid, user);
171 }