c2586abffa612d08b273fec2f43d69b1e1407a27
[public/pyceo-broken.git] / src / addmember.c
1 #include <unistd.h>
2 #include <sys/types.h>
3 #include <sys/wait.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <getopt.h>
7 #include <ctype.h>
8 #include <pwd.h>
9 #include <grp.h>
10 #include <errno.h>
11 #include <libgen.h>
12 #include <syslog.h>
13
14 #include "util.h"
15 #include "common.h"
16 #include "config.h"
17 #include "ldap.h"
18 #include "krb5.h"
19 #include "kadm.h"
20 #include "addhomedir.h"
21
22 char *prog = NULL;
23 char *user = NULL;
24 int privileged = 0;
25
26 static int force = 0;
27 static int no_notify = 0;
28
29 static int use_stdin = 0;
30
31 static char *name = NULL;
32 static char *userid = NULL;
33 static char *program = NULL;
34 static char password[1024];
35
36 static struct option opts[] = {
37     { "force", 0, NULL, 'f' },
38     { "no-notify", 0, NULL, 'q' },
39     { "stdin", 0, NULL, 's' },
40     { NULL, 0, NULL, '\0' },
41 };
42
43 static void usage() {
44     fprintf(stderr, "Usage: %s userid realname [program]\n", prog);
45     exit(2);
46 }
47
48 int addmember() {
49     int krb_ok, user_ok, group_ok, home_ok;
50     int id;
51     char homedir[1024];
52     char acl_s[1024] = {0};
53
54     logmsg("adding uid=%s cn=%s program=%s by %s", userid, name, program, user);
55
56     if (setreuid(0, 0))
57         fatalpe("setreuid");
58     if (setregid(0, 0))
59         fatalpe("setregid");
60
61     if (!force && getpwnam(userid) != NULL)
62         deny("user %s already exists", userid);
63
64     snprintf(homedir, sizeof(homedir), "%s/%s", member_home, userid);
65
66     if (ceo_read_password(password, sizeof(password), use_stdin))
67         return 1;
68
69     ceo_krb5_init();
70     ceo_ldap_init();
71     ceo_kadm_init();
72
73     if (ceo_user_exists(userid))
74         deny("user %s already exists in LDAP", userid);
75
76     if ((id = ceo_new_uid(member_min_id, member_max_id)) <= 0)
77         fatal("no available uids in range [%d, %d]", member_min_id, member_max_id);
78
79     if (*member_home_acl) {
80         snprintf(acl_s, sizeof(acl_s), member_home_acl, userid);
81     }
82
83     krb_ok = ceo_del_princ(userid);
84     krb_ok = krb_ok || ceo_add_princ(userid, password);
85     if (!krb_ok)
86         logmsg("successfully created principal for %s", userid);
87
88     user_ok = krb_ok || ceo_add_user(userid, users_base, "member", name, homedir,
89             member_shell, id, "program", program, NULL);
90     if (!user_ok)
91         logmsg("successfully created account for %s", userid);
92
93     group_ok = user_ok || ceo_add_group(userid, groups_base, id);
94     if (!group_ok)
95         logmsg("successfully created group for %s", userid);
96
97     home_ok = user_ok || ceo_create_home(homedir, refquota, id, id, homedir_mode, acl_s);
98     if (!home_ok)
99         logmsg("successfully created home directory for %s", userid);
100
101     logmsg("done uid=%s", userid);
102
103     if (!no_notify && !user_ok) {
104         int pid;
105         int hkp[2];
106         FILE *hkf;
107         int status;
108
109         if (pipe(hkp))
110             errorpe("pipe");
111
112         fflush(stdout);
113         fflush(stderr);
114
115         pid = fork();
116
117         if (!pid) {
118             fclose(stdout);
119             fclose(stderr);
120             close(hkp[1]);
121             dup2(hkp[0], 0);
122             exit(execl(notify_hook, notify_hook, prog, user, userid, name, program, NULL));
123         }
124
125         hkf = fdopen(hkp[1], "w");
126
127         if (group_ok)
128             fprintf(hkf, "failed to create group\n");
129         if (home_ok)
130             fprintf(hkf, "failed to create home directory\n");
131         if (!group_ok && !home_ok)
132             fprintf(hkf, "all failures went undetected\n");
133
134         fclose(hkf);
135
136         waitpid(pid, &status, 0);
137
138         if (WIFEXITED(status) && WEXITSTATUS(status))
139             logmsg("hook %s exited with status %d", notify_hook, WEXITSTATUS(status));
140         else if (WIFSIGNALED(status))
141             logmsg("hook %s killed by signal %d", notify_hook, WTERMSIG(status));
142     }
143
144     ceo_kadm_cleanup();
145     ceo_ldap_cleanup();
146     ceo_krb5_cleanup();
147
148     return krb_ok || user_ok || group_ok || home_ok;
149 }
150
151 int main(int argc, char *argv[]) {
152     int opt;
153
154     prog = basename(argv[0]);
155     openlog(prog, 0, LOG_AUTHPRIV);
156
157     configure();
158
159     user = ceo_get_user();
160     privileged = ceo_get_privileged();
161
162     while ((opt = getopt_long(argc, argv, "", opts, NULL)) != -1) {
163         switch (opt) {
164             case 's':
165                 use_stdin = 1;
166                 break;
167             case 'f':
168                 if (!privileged)
169                     deny("not privileged enough to force");
170                 force = 1;
171                 break;
172             case 'q':
173                 if (!privileged)
174                     deny("not privileged enough to suppress notifications");
175                 no_notify = 1;
176                 break;
177             case '?':
178                 usage();
179                 break;
180             default:
181                 fatal("error parsing arguments");
182         }
183     }
184
185     if (argc - optind != 2 && argc - optind != 3)
186         usage();
187
188     userid = argv[optind++];
189     name = argv[optind++];
190
191     if (argc - optind)
192         program = argv[optind++];
193
194     return addmember();
195 }