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