Actually store basic_hash in the session data
[mod_auth_gssapi.git] / src / sessions.c
1 /* Copyright (C) 2014 mod_auth_gssapi authors - See COPYING for (C) terms */
2
3 #include "mod_auth_gssapi.h"
4 #include "asn1c/GSSSessionData.h"
5
6 APLOG_USE_MODULE(auth_gssapi);
7
8 static APR_OPTIONAL_FN_TYPE(ap_session_load) *mag_sess_load_fn = NULL;
9 static APR_OPTIONAL_FN_TYPE(ap_session_get) *mag_sess_get_fn = NULL;
10 static APR_OPTIONAL_FN_TYPE(ap_session_set) *mag_sess_set_fn = NULL;
11
12 void mag_post_config_session(void)
13 {
14     mag_sess_load_fn = APR_RETRIEVE_OPTIONAL_FN(ap_session_load);
15     mag_sess_get_fn = APR_RETRIEVE_OPTIONAL_FN(ap_session_get);
16     mag_sess_set_fn = APR_RETRIEVE_OPTIONAL_FN(ap_session_set);
17 }
18
19 static apr_status_t mag_session_load(request_rec *req, session_rec **sess)
20 {
21     if (mag_sess_load_fn) {
22         return mag_sess_load_fn(req, sess);
23     }
24     return DECLINED;
25 }
26
27 static apr_status_t mag_session_get(request_rec *req, session_rec *sess,
28                                     const char *key, const char **value)
29 {
30     if (mag_sess_get_fn) {
31         return mag_sess_get_fn(req, sess, key, value);
32     }
33     return DECLINED;
34 }
35
36 static apr_status_t mag_session_set(request_rec *req, session_rec *sess,
37                                     const char *key, const char *value)
38 {
39     if (mag_sess_set_fn) {
40         return mag_sess_set_fn(req, sess, key, value);
41     }
42     return DECLINED;
43 }
44
45 static bool encode_GSSSessionData(apr_pool_t *mempool,
46                                   GSSSessionData_t *gsessdata,
47                                   unsigned char **buf, int *len)
48 {
49     asn_enc_rval_t rval;
50     unsigned char *buffer = NULL;
51     size_t buflen;
52     bool ret = false;
53
54     /* dry run to compute the size */
55     rval = der_encode(&asn_DEF_GSSSessionData, gsessdata, NULL, NULL);
56     if (rval.encoded == -1) goto done;
57
58     buflen = rval.encoded;
59     buffer = apr_pcalloc(mempool, buflen);
60
61     /* now for real */
62     rval = der_encode_to_buffer(&asn_DEF_GSSSessionData,
63                                 gsessdata, buffer, buflen);
64     if (rval.encoded == -1) goto done;
65
66     *buf = buffer;
67     *len = buflen;
68     ret = true;
69
70 done:
71     return ret;
72 }
73
74 static GSSSessionData_t *decode_GSSSessionData(void *buf, size_t len)
75 {
76     GSSSessionData_t *gsessdata = NULL;
77     asn_dec_rval_t rval;
78
79     rval = ber_decode(NULL, &asn_DEF_GSSSessionData,
80                       (void **)&gsessdata, buf, len);
81     if (rval.code == RC_OK) {
82         return gsessdata;
83     }
84     return NULL;
85 }
86
87 #define MAG_BEARER_KEY "MagBearerToken"
88
89 void mag_check_session(request_rec *req,
90                        struct mag_config *cfg, struct mag_conn **conn)
91 {
92     struct mag_conn *mc;
93     apr_status_t rc;
94     session_rec *sess = NULL;
95     const char *sessval = NULL;
96     int declen;
97     struct databuf ctxbuf = { 0 };
98     struct databuf cipherbuf = { 0 };
99     GSSSessionData_t *gsessdata;
100     time_t expiration;
101
102     rc = mag_session_load(req, &sess);
103     if (rc != OK || sess == NULL) {
104         ap_log_rerror(APLOG_MARK, APLOG_INFO|APLOG_NOERRNO, 0, req,
105                       "Sessions not available, no cookies!");
106         return;
107     }
108
109     mc = *conn;
110     if (!mc) {
111         mc = apr_pcalloc(req->pool, sizeof(struct mag_conn));
112         if (!mc) return;
113
114         mc->parent = req->pool;
115         *conn = mc;
116     }
117
118     rc = mag_session_get(req, sess, MAG_BEARER_KEY, &sessval);
119     if (rc != OK) {
120         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
121                       "Failed to get session data!");
122         return;
123     }
124     if (!sessval) {
125         /* no session established, just return */
126         return;
127     }
128
129     if (!cfg->mag_skey) {
130         ap_log_rerror(APLOG_MARK, APLOG_INFO, 0, req,
131                       "Session key not available, no cookies!");
132         /* we do not have a key, just return */
133         return;
134     }
135
136     /* decode it */
137     declen = apr_base64_decode_len(sessval);
138     cipherbuf.value = apr_palloc(req->pool, declen);
139     if (!cipherbuf.value) return;
140     cipherbuf.length = (int)apr_base64_decode((char *)cipherbuf.value, sessval);
141
142     rc = UNSEAL_BUFFER(req->pool, cfg->mag_skey, &cipherbuf, &ctxbuf);
143     if (rc != OK) {
144         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
145                       "Failed to unseal session data!");
146         return;
147     }
148
149     gsessdata = decode_GSSSessionData(ctxbuf.value, ctxbuf.length);
150     if (!gsessdata) {
151         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
152                       "Failed to unpack session data!");
153         return;
154     }
155
156     /* booleans */
157     if (gsessdata->established != 0) mc->established = true;
158     if (gsessdata->delegated != 0) mc->delegated = true;
159
160     /* get time */
161     expiration = gsessdata->expiration;
162     if (expiration < time(NULL)) {
163         /* credentials fully expired, return nothing */
164         goto done;
165     }
166
167     /* user name */
168     mc->user_name = apr_pstrndup(mc->parent,
169                                  (char *)gsessdata->username.buf,
170                                  gsessdata->username.size);
171     if (!mc->user_name) goto done;
172
173     /* gssapi name */
174     mc->gss_name = apr_pstrndup(mc->parent,
175                                 (char *)gsessdata->gssname.buf,
176                                 gsessdata->gssname.size);
177     if (!mc->gss_name) goto done;
178
179     mc->basic_hash.length = gsessdata->basichash.size;
180     mc->basic_hash.value = apr_palloc(mc->parent, mc->basic_hash.length);
181     memcpy(mc->basic_hash.value,
182            gsessdata->basichash.buf, gsessdata->basichash.size);
183
184     /* OK we have a valid token */
185     mc->established = true;
186
187 done:
188     ASN_STRUCT_FREE(asn_DEF_GSSSessionData, gsessdata);
189 }
190
191 void mag_attempt_session(request_rec *req,
192                          struct mag_config *cfg, struct mag_conn *mc)
193 {
194     session_rec *sess = NULL;
195     struct databuf plainbuf = { 0 };
196     struct databuf cipherbuf = { 0 };
197     struct databuf ctxbuf = { 0 };
198     GSSSessionData_t gsessdata = { 0 };
199     apr_status_t rc;
200     bool ret;
201
202     /* we save the session only if the authentication is established */
203
204     if (!mc->established) return;
205     rc = mag_session_load(req, &sess);
206     if (rc != OK || sess == NULL) {
207         ap_log_rerror(APLOG_MARK, APLOG_INFO, 0, req,
208                       "Sessions not available, can't send cookies!");
209         return;
210     }
211
212     if (!cfg->mag_skey) {
213         ap_log_rerror(APLOG_MARK, APLOG_INFO, 0, req,
214                       "Session key not available, generating new one.");
215         rc = SEAL_KEY_CREATE(cfg->pool, &cfg->mag_skey, NULL);
216         if (rc != OK) {
217             ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
218                           "Failed to create sealing key!");
219             return;
220         }
221     }
222
223     gsessdata.established = mc->established?1:0;
224     gsessdata.delegated = mc->delegated?1:0;
225     gsessdata.expiration = mc->expiration;
226     if (OCTET_STRING_fromString(&gsessdata.username, mc->user_name) != 0)
227         goto done;
228     if (OCTET_STRING_fromString(&gsessdata.gssname, mc->gss_name) != 0)
229         goto done;
230     if (OCTET_STRING_fromBuf(&gsessdata.basichash,
231                              (const char *)mc->basic_hash.value,
232                              mc->basic_hash.length) != 0)
233         goto done;
234     ret = encode_GSSSessionData(req->pool, &gsessdata,
235                                 &plainbuf.value, &plainbuf.length);
236     if (ret == false) {
237         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
238                       "Failed to pack session data!");
239         goto done;
240     }
241
242     rc = SEAL_BUFFER(req->pool, cfg->mag_skey, &plainbuf, &cipherbuf);
243     if (rc != OK) {
244         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
245                       "Failed to seal session data!");
246         goto done;
247     }
248
249     ctxbuf.length = apr_base64_encode_len(cipherbuf.length);
250     ctxbuf.value = apr_pcalloc(req->pool, ctxbuf.length);
251     if (!ctxbuf.value) goto done;
252
253     ctxbuf.length = apr_base64_encode((char *)ctxbuf.value,
254                                       (char *)cipherbuf.value,
255                                       cipherbuf.length);
256
257     rc = mag_session_set(req, sess, MAG_BEARER_KEY, (char *)ctxbuf.value);
258     if (rc != OK) {
259         ap_log_rerror(APLOG_MARK, APLOG_ERR|APLOG_NOERRNO, 0, req,
260                       "Failed to set session data!");
261     }
262
263 done:
264     ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_GSSSessionData, &gsessdata);
265 }
266
267 static int mag_basic_hmac(struct seal_key *key, unsigned char *mac,
268                           gss_buffer_desc user, gss_buffer_desc pwd)
269 {
270     struct databuf hmacbuf = { mac, 0 };
271     int data_size = user.length + pwd.length + 1;
272     unsigned char data[data_size];
273     struct databuf databuf = { data, data_size };
274
275     memcpy(data, user.value, user.length);
276     data[user.length] = '\0';
277     memcpy(&data[user.length + 1], pwd.value, pwd.length);
278
279     return HMAC_BUFFER(key, &databuf, &hmacbuf);
280 }
281
282 bool mag_basic_check(struct mag_config *cfg, struct mag_conn *mc,
283                      gss_buffer_desc user, gss_buffer_desc pwd)
284 {
285     int mac_size = get_mac_size(cfg->mag_skey);
286     unsigned char mac[mac_size];
287     int ret, i, j;
288     bool res = false;
289
290     if (mac_size == 0) return false;
291     if (mc->basic_hash.value == NULL) return false;
292
293     ret = mag_basic_hmac(cfg->mag_skey, mac, user, pwd);
294     if (ret != 0) goto done;
295
296     for (i = 0, j = 0; i < mac_size; i++) {
297         if (mc->basic_hash.value[i] != mac[i]) j++;
298     }
299     if (j == 0) res = true;
300
301 done:
302     if (res == false) {
303         mc->basic_hash.value = NULL;
304         mc->basic_hash.length = 0;
305     }
306     return res;
307 }
308
309 void mag_basic_cache(struct mag_config *cfg, struct mag_conn *mc,
310                      gss_buffer_desc user, gss_buffer_desc pwd)
311 {
312     int mac_size = get_mac_size(cfg->mag_skey);
313     unsigned char mac[mac_size];
314     int ret;
315
316     ret = mag_basic_hmac(cfg->mag_skey, mac, user, pwd);
317     if (ret != 0) return;
318
319     mc->basic_hash.length = mac_size;
320     mc->basic_hash.value = apr_palloc(mc->parent, mac_size);
321     memcpy(mc->basic_hash.value, mac, mac_size);
322 }