Free everything before exiting
[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 "config.h"
16 #include "ldap.h"
17 #include "krb5.h"
18 #include "kadm.h"
19 #include "ceo.pb-c.h"
20
21 char *prog = NULL;
22
23 static int use_stdin = 0;
24
25 static char *name = NULL;
26 static char *userid = NULL;
27 static char *program = NULL;
28 static char password[1024];
29
30 static struct option opts[] = {
31     { "stdin", 0, NULL, 's' },
32     { NULL, 0, NULL, '\0' },
33 };
34
35 const char *default_lib_dir = "/usr/lib/ceod";
36 const char *lib_dir;
37
38 static void usage() {
39     fprintf(stderr, "Usage: %s userid realname [program]\n", prog);
40     exit(2);
41 }
42
43 int addmember(void) {
44     struct strbuf preq = STRBUF_INIT;
45     struct strbuf pret = STRBUF_INIT;
46     char cpath[1024];
47     char *cargv[] = { "ceoc", "adduser", NULL };
48
49     if (snprintf(cpath, sizeof(cpath), "%s/ceoc", lib_dir) >= sizeof(cpath))
50         fatal("path too long");
51
52     if (ceo_read_password(password, sizeof(password), use_stdin))
53         return 1;
54
55     Ceo__AddUser req;
56     ceo__add_user__init(&req);
57
58     req.username = userid;
59     req.password = password;
60     req.program  = program;
61     req.realname = name;
62     req.type = CEO__ADD_USER__TYPE__MEMBER;
63
64     strbuf_grow(&preq, ceo__add_user__get_packed_size(&req));
65     strbuf_setlen(&preq, ceo__add_user__pack(&req, (uint8_t *)preq.buf));
66
67     if (spawnvem(cpath, cargv, environ, &preq, &pret, 0))
68         return 1;
69
70     Ceo__AddUserResponse *ret = ceo__add_user_response__unpack(&protobuf_c_default_allocator,
71                                                                pret.len, (uint8_t *)pret.buf);
72     if (!ret)
73         fatal("failed to unpack response");
74
75     for (int i = 0; i < ret->n_messages; i++) {
76         if (ret->messages[i]->status)
77             error("%s", ret->messages[i]->message);
78         else
79             notice("%s", ret->messages[i]->message);
80     }
81
82     ceo__add_user_response__free_unpacked(ret, &protobuf_c_default_allocator);
83     strbuf_release(&preq);
84     strbuf_release(&pret);
85
86     return 0;
87 }
88
89 int main(int argc, char *argv[]) {
90     int opt;
91     int ret;
92
93     prog = xstrdup(basename(argv[0]));
94     init_log(prog, 0, LOG_AUTHPRIV);
95
96     configure();
97
98     while ((opt = getopt_long(argc, argv, "", opts, NULL)) != -1) {
99         switch (opt) {
100             case 's':
101                 use_stdin = 1;
102                 break;
103             case '?':
104                 usage();
105                 break;
106             default:
107                 fatal("error parsing arguments");
108         }
109     }
110
111     if (argc - optind != 2 && argc - optind != 3)
112         usage();
113
114     userid = argv[optind++];
115     name = argv[optind++];
116
117     if (argc - optind)
118         program = argv[optind++];
119
120     lib_dir = getenv("CEO_LIB_DIR") ?: default_lib_dir;
121
122     ret = addmember();
123
124     free_config();
125     free(prog);
126
127     return ret;
128 }