Add support for http digest authentication
[ffmpeg.git] / libavformat / httpauth.c
1 /*
2  * HTTP authentication
3  * Copyright (c) 2010 Martin Storsjo
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "httpauth.h"
23 #include "libavutil/base64.h"
24 #include "libavutil/avstring.h"
25 #include "internal.h"
26 #include "libavutil/random_seed.h"
27 #include "libavutil/md5.h"
28 #include "avformat.h"
29 #include <ctype.h>
30
31 static void parse_key_value(const char *params,
32                             void (*callback_get_buf)(HTTPAuthState *state,
33                             const char *key, int key_len,
34                             char **dest, int *dest_len), HTTPAuthState *state)
35 {
36     const char *ptr = params;
37
38     /* Parse key=value pairs. */
39     for (;;) {
40         const char *key;
41         char *dest = NULL, *dest_end;
42         int key_len, dest_len = 0;
43
44         /* Skip whitespace and potential commas. */
45         while (*ptr && (isspace(*ptr) || *ptr == ','))
46             ptr++;
47         if (!*ptr)
48             break;
49
50         key = ptr;
51
52         if (!(ptr = strchr(key, '=')))
53             break;
54         ptr++;
55         key_len = ptr - key;
56
57         callback_get_buf(state, key, key_len, &dest, &dest_len);
58         dest_end = dest + dest_len - 1;
59
60         if (*ptr == '\"') {
61             ptr++;
62             while (*ptr && *ptr != '\"') {
63                 if (*ptr == '\\') {
64                     if (!ptr[1])
65                         break;
66                     if (dest && dest < dest_end)
67                         *dest++ = ptr[1];
68                     ptr += 2;
69                 } else {
70                     if (dest && dest < dest_end)
71                         *dest++ = *ptr;
72                     ptr++;
73                 }
74             }
75             if (*ptr == '\"')
76                 ptr++;
77         } else {
78             for (; *ptr && !(isspace(*ptr) || *ptr == ','); ptr++)
79                 if (dest && dest < dest_end)
80                     *dest++ = *ptr;
81         }
82         if (dest)
83             *dest = 0;
84     }
85 }
86
87 static void handle_basic_params(HTTPAuthState *state, const char *key,
88                                 int key_len, char **dest, int *dest_len)
89 {
90     if (!strncmp(key, "realm=", key_len)) {
91         *dest     =        state->realm;
92         *dest_len = sizeof(state->realm);
93     }
94 }
95
96 static void handle_digest_params(HTTPAuthState *state, const char *key,
97                                  int key_len, char **dest, int *dest_len)
98 {
99     DigestParams *digest = &state->digest_params;
100
101     if (!strncmp(key, "realm=", key_len)) {
102         *dest     =        state->realm;
103         *dest_len = sizeof(state->realm);
104     } else if (!strncmp(key, "nonce=", key_len)) {
105         *dest     =        digest->nonce;
106         *dest_len = sizeof(digest->nonce);
107     } else if (!strncmp(key, "opaque=", key_len)) {
108         *dest     =        digest->opaque;
109         *dest_len = sizeof(digest->opaque);
110     } else if (!strncmp(key, "algorithm=", key_len)) {
111         *dest     =        digest->algorithm;
112         *dest_len = sizeof(digest->algorithm);
113     } else if (!strncmp(key, "qop=", key_len)) {
114         *dest     =        digest->qop;
115         *dest_len = sizeof(digest->qop);
116     }
117 }
118
119 static void handle_digest_update(HTTPAuthState *state, const char *key,
120                                  int key_len, char **dest, int *dest_len)
121 {
122     DigestParams *digest = &state->digest_params;
123
124     if (!strncmp(key, "nextnonce=", key_len)) {
125         *dest     =        digest->nonce;
126         *dest_len = sizeof(digest->nonce);
127     }
128 }
129
130 static void choose_qop(char *qop, int size)
131 {
132     char *ptr = strstr(qop, "auth");
133     char *end = ptr + strlen("auth");
134
135     if (ptr && (!*end || isspace(*end) || *end == ',') &&
136         (ptr == qop || isspace(ptr[-1]) || ptr[-1] == ',')) {
137         av_strlcpy(qop, "auth", size);
138     } else {
139         qop[0] = 0;
140     }
141 }
142
143 void ff_http_auth_handle_header(HTTPAuthState *state, const char *key,
144                                 const char *value)
145 {
146     if (!state)
147         return;
148
149     if (!strcmp(key, "WWW-Authenticate")) {
150         const char *p;
151         if (av_stristart(value, "Basic ", &p) &&
152             state->auth_type <= HTTP_AUTH_BASIC) {
153             state->auth_type = HTTP_AUTH_BASIC;
154             state->realm[0] = 0;
155             parse_key_value(p, handle_basic_params, state);
156         } else if (av_stristart(value, "Digest ", &p) &&
157                    state->auth_type <= HTTP_AUTH_DIGEST) {
158             state->auth_type = HTTP_AUTH_DIGEST;
159             memset(&state->digest_params, 0, sizeof(DigestParams));
160             state->realm[0] = 0;
161             parse_key_value(p, handle_digest_params, state);
162             choose_qop(state->digest_params.qop,
163                        sizeof(state->digest_params.qop));
164         }
165     } else if (!strcmp(key, "Authentication-Info")) {
166         parse_key_value(value, handle_digest_update, state);
167     }
168 }
169
170
171 static void update_md5_strings(struct AVMD5 *md5ctx, ...)
172 {
173     va_list vl;
174
175     va_start(vl, md5ctx);
176     while (1) {
177         const char* str = va_arg(vl, const char*);
178         if (!str)
179             break;
180         av_md5_update(md5ctx, str, strlen(str));
181     }
182     va_end(vl);
183 }
184
185 /* Generate a digest reply, according to RFC 2617. */
186 static char *make_digest_auth(HTTPAuthState *state, const char *username,
187                               const char *password, const char *uri,
188                               const char *method)
189 {
190     DigestParams *digest = &state->digest_params;
191     int len;
192     uint32_t cnonce_buf[2];
193     char cnonce[9];
194     char nc[9];
195     int i;
196     char A1hash[33], A2hash[33], response[33];
197     struct AVMD5 *md5ctx;
198     uint8_t hash[16];
199     char *authstr;
200
201     digest->nc++;
202     snprintf(nc, sizeof(nc), "%08x", digest->nc);
203
204     /* Generate a client nonce. */
205     for (i = 0; i < 2; i++)
206         cnonce_buf[i] = ff_random_get_seed();
207     ff_data_to_hex(cnonce, (const uint8_t*) cnonce_buf, sizeof(cnonce_buf), 1);
208     cnonce[2*sizeof(cnonce_buf)] = 0;
209
210     md5ctx = av_malloc(av_md5_size);
211     if (!md5ctx)
212         return NULL;
213
214     av_md5_init(md5ctx);
215     update_md5_strings(md5ctx, username, ":", state->realm, ":", password, NULL);
216     av_md5_final(md5ctx, hash);
217     ff_data_to_hex(A1hash, hash, 16, 1);
218     A1hash[32] = 0;
219
220     if (!strcmp(digest->algorithm, "") || !strcmp(digest->algorithm, "MD5")) {
221     } else if (!strcmp(digest->algorithm, "MD5-sess")) {
222         av_md5_init(md5ctx);
223         update_md5_strings(md5ctx, A1hash, ":", digest->nonce, ":", cnonce, NULL);
224         av_md5_final(md5ctx, hash);
225         ff_data_to_hex(A1hash, hash, 16, 1);
226         A1hash[32] = 0;
227     } else {
228         /* Unsupported algorithm */
229         av_free(md5ctx);
230         return NULL;
231     }
232
233     av_md5_init(md5ctx);
234     update_md5_strings(md5ctx, method, ":", uri, NULL);
235     av_md5_final(md5ctx, hash);
236     ff_data_to_hex(A2hash, hash, 16, 1);
237     A2hash[32] = 0;
238
239     av_md5_init(md5ctx);
240     update_md5_strings(md5ctx, A1hash, ":", digest->nonce, NULL);
241     if (!strcmp(digest->qop, "auth") || !strcmp(digest->qop, "auth-int")) {
242         update_md5_strings(md5ctx, ":", nc, ":", cnonce, ":", digest->qop, NULL);
243     }
244     update_md5_strings(md5ctx, ":", A2hash, NULL);
245     av_md5_final(md5ctx, hash);
246     ff_data_to_hex(response, hash, 16, 1);
247     response[32] = 0;
248
249     av_free(md5ctx);
250
251     if (!strcmp(digest->qop, "") || !strcmp(digest->qop, "auth")) {
252     } else if (!strcmp(digest->qop, "auth-int")) {
253         /* qop=auth-int not supported */
254         return NULL;
255     } else {
256         /* Unsupported qop value. */
257         return NULL;
258     }
259
260     len = strlen(username) + strlen(state->realm) + strlen(digest->nonce) +
261               strlen(uri) + strlen(response) + strlen(digest->algorithm) +
262               strlen(digest->opaque) + strlen(digest->qop) + strlen(cnonce) +
263               strlen(nc) + 150;
264
265     authstr = av_malloc(len);
266     if (!authstr)
267         return NULL;
268     snprintf(authstr, len, "Authorization: Digest ");
269
270     /* TODO: Escape the quoted strings properly. */
271     av_strlcatf(authstr, len, "username=\"%s\"",   username);
272     av_strlcatf(authstr, len, ",realm=\"%s\"",     state->realm);
273     av_strlcatf(authstr, len, ",nonce=\"%s\"",     digest->nonce);
274     av_strlcatf(authstr, len, ",uri=\"%s\"",       uri);
275     av_strlcatf(authstr, len, ",response=\"%s\"",  response);
276     if (digest->algorithm[0])
277         av_strlcatf(authstr, len, ",algorithm=%s",  digest->algorithm);
278     if (digest->opaque[0])
279         av_strlcatf(authstr, len, ",opaque=\"%s\"", digest->opaque);
280     if (digest->qop[0]) {
281         av_strlcatf(authstr, len, ",qop=\"%s\"",    digest->qop);
282         av_strlcatf(authstr, len, ",cnonce=\"%s\"", cnonce);
283         av_strlcatf(authstr, len, ",nc=%s",         nc);
284     }
285
286     av_strlcatf(authstr, len, "\r\n");
287
288     return authstr;
289 }
290
291 char *ff_http_auth_create_response(HTTPAuthState *state, const char *auth,
292                                    const char *path, const char *method)
293 {
294     char *authstr = NULL;
295
296     if (!auth || !strchr(auth, ':'))
297         return NULL;
298
299     if (state->auth_type == HTTP_AUTH_BASIC) {
300         int auth_b64_len = (strlen(auth) + 2) / 3 * 4 + 1;
301         int len = auth_b64_len + 30;
302         char *ptr;
303         authstr = av_malloc(len);
304         if (!authstr)
305             return NULL;
306         snprintf(authstr, len, "Authorization: Basic ");
307         ptr = authstr + strlen(authstr);
308         av_base64_encode(ptr, auth_b64_len, auth, strlen(auth));
309         av_strlcat(ptr, "\r\n", len);
310     } else if (state->auth_type == HTTP_AUTH_DIGEST) {
311         char *username = av_strdup(auth), *password;
312
313         if (!username)
314             return NULL;
315
316         if ((password = strchr(username, ':'))) {
317             *password++ = 0;
318             authstr = make_digest_auth(state, username, password, path, method);
319         }
320         av_free(username);
321     }
322     return authstr;
323 }
324