Make CEO non-SUID
[mspang/pyceo.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, quota_ok;
50     int id;
51     char homedir[1024];
52
53     logmsg("adding uid=%s cn=%s program=%s by %s", userid, name, program, user);
54
55     if (setreuid(0, 0))
56         fatalpe("setreuid");
57
58     if (!force && getpwnam(userid) != NULL)
59         deny("user %s already exists", userid);
60
61     if (ceo_read_password(password, sizeof(password), use_stdin))
62         return;
63
64     ceo_krb5_init();
65     ceo_ldap_init();
66     ceo_kadm_init();
67
68     if (ceo_user_exists(userid))
69         deny("user %s already exists in LDAP", userid);
70
71     if ((id = ceo_new_uid(member_min_id, member_max_id)) <= 0)
72         fatal("no available uids in range [%d, %d]", member_min_id, member_max_id);
73
74     snprintf(homedir, sizeof(homedir), "%s/%s", member_home, userid);
75
76     krb_ok = ceo_del_princ(userid);
77     krb_ok = krb_ok || ceo_add_princ(userid, password);
78     if (!krb_ok)
79         logmsg("successfully created principal for %s", userid);
80
81     user_ok = krb_ok || ceo_add_user(userid, users_base, "member", name, homedir,
82             member_shell, id, "program", program, NULL);
83     if (!user_ok)
84         logmsg("successfully created account for %s", userid);
85
86     group_ok = user_ok || ceo_add_group(userid, groups_base, id);
87     if (!group_ok)
88         logmsg("successfully created group for %s", userid);
89
90     home_ok = user_ok || ceo_create_home(homedir, id, id);
91     if (!home_ok)
92         logmsg("successfully created home directory for %s", userid);
93
94     quota_ok = user_ok || ceo_set_quota(quota_prototype, id);
95     if (!quota_ok)
96         logmsg("successfully set quota for %s", userid);
97
98     logmsg("done uid=%s", userid);
99
100     if (!no_notify && !user_ok) {
101         int pid;
102         int hkp[2];
103         FILE *hkf;
104         int status;
105
106         if (pipe(hkp))
107             errorpe("pipe");
108
109         fflush(stdout);
110         fflush(stderr);
111
112         pid = fork();
113
114         if (!pid) {
115             fclose(stdout);
116             fclose(stderr);
117             close(hkp[1]);
118             dup2(hkp[0], 0);
119             exit(execl(notify_hook, notify_hook, prog, user, userid, name, program, NULL));
120         }
121
122         hkf = fdopen(hkp[1], "w");
123
124         if (group_ok)
125             fprintf(hkf, "failed to create group\n");
126         if (home_ok)
127             fprintf(hkf, "failed to create home directory\n");
128         if (quota_ok)
129             fprintf(hkf, "failed to set quota\n");
130         if (!group_ok && !home_ok && !quota_ok)
131             fprintf(hkf, "all failures went undetected\n");
132
133         fclose(hkf);
134
135         waitpid(pid, &status, 0);
136
137         if (WIFEXITED(status) && WEXITSTATUS(status))
138             logmsg("hook %s exited with status %d", notify_hook, WEXITSTATUS(status));
139         else if (WIFSIGNALED(status))
140             logmsg("hook %s killed by signal %d", notify_hook, WTERMSIG(status));
141     }
142
143     ceo_kadm_cleanup();
144     ceo_ldap_cleanup();
145     ceo_krb5_cleanup();
146
147     return !krb_ok && !user_ok && !group_ok && !home_ok && quota_ok;
148 }
149
150 int main(int argc, char *argv[]) {
151     int opt;
152
153     openlog(prog, 0, LOG_AUTHPRIV);
154
155     configure();
156
157     prog = basename(argv[0]);
158     user = ceo_get_user();
159     privileged = ceo_get_privileged();
160
161     while ((opt = getopt_long(argc, argv, "", opts, NULL)) != -1) {
162         switch (opt) {
163             case 's':
164                 use_stdin = 1;
165                 break;
166             case 'f':
167                 if (!privileged)
168                     deny("not privileged enough to force");
169                 force = 1;
170                 break;
171             case 'q':
172                 if (!privileged)
173                     deny("not privileged enough to suppress notifications");
174                 no_notify = 1;
175                 break;
176             case '?':
177                 usage();
178                 break;
179             default:
180                 fatal("error parsing arguments");
181         }
182     }
183
184     if (argc - optind != 2 && argc - optind != 3)
185         usage();
186
187     userid = argv[optind++];
188     name = argv[optind++];
189
190     if (argc - optind)
191         program = argv[optind++];
192
193     return addmember();
194 }