Make more noise in config parser
[mspang/pyceo.git] / src / krb5.c
1 #include <stdio.h>
2 #include <krb5.h>
3 #include <syslog.h>
4
5 #include "krb5.h"
6 #include "util.h"
7 #include "config.h"
8
9 extern char *prog;
10
11 krb5_context context;
12
13 static void com_err_hk(const char *whoami, long code, const char *fmt, va_list args) {
14     char message[4096];
15     char *msgp = message;
16
17     msgp += snprintf(msgp, sizeof(message) - 2 - (msgp - message), "%s ", error_message(code));
18     if (msgp - message > sizeof(message) - 2)
19         fatal("error message overflowed");
20
21     msgp += vsnprintf(msgp, sizeof(message) - 2 - (msgp - message), fmt, args);
22     if (msgp - message > sizeof(message) - 2)
23         fatal("error message overflowed");
24
25     *msgp++ = '\n';
26     *msgp++ = '\0';
27
28     logmsg(LOG_ERR, "fatal: %s", message);
29     exit(1);
30 }
31
32 void ceo_krb5_init() {
33     krb5_error_code retval;
34
35     set_com_err_hook(com_err_hk);
36
37     retval = krb5_init_context(&context);
38     if (retval)
39         com_err(prog, retval, "while initializing krb5");
40
41     retval = krb5_set_default_realm(context, realm);
42     if (retval)
43         com_err(prog, retval, "while setting default realm");
44 }
45
46 void ceo_krb5_auth(char *principal, char *ktname) {
47     krb5_error_code retval;
48     krb5_creds creds;
49     krb5_principal princ;
50     krb5_keytab keytab;
51     krb5_ccache cache;
52     krb5_get_init_creds_opt options;
53
54     krb5_get_init_creds_opt_init(&options);
55     memset(&creds, 0, sizeof(creds));
56
57     if ((retval = krb5_parse_name(context, principal, &princ)))
58         com_err(prog, retval, "while resolving user %s", admin_bind_userid);
59
60     if ((retval = krb5_cc_default(context, &cache)))
61         com_err(prog, retval, "while resolving credentials cache");
62
63     if ((retval = krb5_kt_resolve(context, ktname, &keytab)))
64         com_err(prog, retval, "while resolving keytab %s", admin_bind_keytab);
65
66     if ((retval = krb5_get_init_creds_keytab(context, &creds, princ, keytab, 0, NULL, &options)))
67         com_err(prog, retval, "while getting initial credentials");
68
69     if ((retval = krb5_cc_initialize(context, cache, princ)))
70         com_err(prog, retval, "while initializing credentials cache");
71
72     if ((retval = krb5_cc_store_cred(context, cache, &creds)))
73         com_err(prog, retval, "while storing credentials");
74
75     krb5_free_cred_contents(context, &creds);
76     krb5_kt_close(context, keytab);
77     krb5_free_principal(context, princ);
78     krb5_cc_close(context, cache);
79 }
80
81 void ceo_krb5_deauth() {
82     krb5_error_code retval;
83     krb5_ccache cache;
84
85     if ((retval = krb5_cc_default(context, &cache)))
86         com_err(prog, retval, "while resolving credentials cache");
87
88     if ((retval = krb5_cc_destroy(context, cache)))
89         com_err(prog, retval, "while destroying credentials cache");
90 }
91
92 void ceo_krb5_cleanup() {
93     krb5_free_context(context);
94 }
95
96 int ceo_read_password(char *password, unsigned int size, int use_stdin) {
97     int tries = 0;
98     unsigned int len;
99
100     do {
101         if (use_stdin) {
102             if (fgets(password, size, stdin) == NULL)
103                 fatal("eof while reading password");
104
105             size = strlen(password);
106
107             if (password[size - 1] == '\n')
108                 password[size - 1] = '\0';
109         } else {
110             len = size;
111             int retval = krb5_read_password(context, "New password", "Confirm password", password, &len);
112             if (retval == KRB5_LIBOS_PWDINTR) {
113                 error("interrupted");
114                 return -1;
115             } else if (retval == KRB5_LIBOS_BADPWDMATCH) {
116                 fputs("Passwords do not match.\n", stderr);
117             } else if (!password || !*password) {
118                 fputs("Please enter a password.\n", stderr);
119             }
120         }
121     } while (++tries < 3 && !*password);
122
123     if (!*password) {
124         error("maximum tries exceeded reading password");
125         return -1;
126     }
127
128     return 0;
129 }