FFmpeg coverage

Line	Branch	Exec	Source
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 "libavutil/mem.h"
26			#include "internal.h"
27			#include "libavutil/random_seed.h"
28			#include "libavutil/md5.h"
29			#include "urldecode.h"
30
31		✗	static void handle_basic_params(void *context, const char *key,
32			int key_len, char **dest, int *dest_len)
33			{
34		✗	HTTPAuthState *state = context;
35		✗	if (!strncmp(key, "realm=", key_len)) {
36		✗	*dest = state->realm;
37		✗	*dest_len = sizeof(state->realm);
38			}
39		✗	}
40
41		✗	static void handle_digest_params(void *context, const char *key,
42			int key_len, char **dest, int *dest_len)
43			{
44		✗	HTTPAuthState *state = context;
45		✗	DigestParams *digest = &state->digest_params;
46
47		✗	if (!strncmp(key, "realm=", key_len)) {
48		✗	*dest = state->realm;
49		✗	*dest_len = sizeof(state->realm);
50		✗	} else if (!strncmp(key, "nonce=", key_len)) {
51		✗	*dest = digest->nonce;
52		✗	*dest_len = sizeof(digest->nonce);
53		✗	} else if (!strncmp(key, "opaque=", key_len)) {
54		✗	*dest = digest->opaque;
55		✗	*dest_len = sizeof(digest->opaque);
56		✗	} else if (!strncmp(key, "algorithm=", key_len)) {
57		✗	*dest = digest->algorithm;
58		✗	*dest_len = sizeof(digest->algorithm);
59		✗	} else if (!strncmp(key, "qop=", key_len)) {
60		✗	*dest = digest->qop;
61		✗	*dest_len = sizeof(digest->qop);
62		✗	} else if (!strncmp(key, "stale=", key_len)) {
63		✗	*dest = digest->stale;
64		✗	*dest_len = sizeof(digest->stale);
65			}
66		✗	}
67
68		✗	static void handle_digest_update(void *context, const char *key,
69			int key_len, char **dest, int *dest_len)
70			{
71		✗	HTTPAuthState *state = context;
72		✗	DigestParams *digest = &state->digest_params;
73
74		✗	if (!strncmp(key, "nextnonce=", key_len)) {
75		✗	*dest = digest->nonce;
76		✗	*dest_len = sizeof(digest->nonce);
77			}
78		✗	}
79
80		✗	static void choose_qop(char *qop, int size)
81			{
82		✗	char *ptr = strstr(qop, "auth");
83		✗	char *end = ptr + strlen("auth");
84
85		✗	if (ptr && (!*end || av_isspace(*end) || *end == ',') &&
86		✗	(ptr == qop || av_isspace(ptr[-1]) || ptr[-1] == ',')) {
87		✗	av_strlcpy(qop, "auth", size);
88			} else {
89		✗	qop[0] = 0;
90			}
91		✗	}
92
93		✗	void ff_http_auth_handle_header(HTTPAuthState *state, const char *key,
94			const char *value)
95			{
96		✗	if (!av_strcasecmp(key, "WWW-Authenticate") || !av_strcasecmp(key, "Proxy-Authenticate")) {
97			const char *p;
98		✗	if (av_stristart(value, "Basic ", &p) &&
99		✗	state->auth_type <= HTTP_AUTH_BASIC) {
100		✗	state->auth_type = HTTP_AUTH_BASIC;
101		✗	state->realm[0] = 0;
102		✗	state->stale = 0;
103		✗	ff_parse_key_value(p, handle_basic_params, state);
104		✗	} else if (av_stristart(value, "Digest ", &p) &&
105		✗	state->auth_type <= HTTP_AUTH_DIGEST) {
106		✗	state->auth_type = HTTP_AUTH_DIGEST;
107		✗	memset(&state->digest_params, 0, sizeof(DigestParams));
108		✗	state->realm[0] = 0;
109		✗	state->stale = 0;
110		✗	ff_parse_key_value(p, handle_digest_params, state);
111		✗	choose_qop(state->digest_params.qop,
112			sizeof(state->digest_params.qop));
113		✗	if (!av_strcasecmp(state->digest_params.stale, "true"))
114		✗	state->stale = 1;
115			}
116		✗	} else if (!av_strcasecmp(key, "Authentication-Info")) {
117		✗	ff_parse_key_value(value, handle_digest_update, state);
118			}
119		✗	}
120
121
122		✗	static void update_md5_strings(struct AVMD5 *md5ctx, ...)
123			{
124			va_list vl;
125
126		✗	va_start(vl, md5ctx);
127		✗	while (1) {
128		✗	const char* str = va_arg(vl, const char*);
129		✗	if (!str)
130		✗	break;
131		✗	av_md5_update(md5ctx, str, strlen(str));
132			}
133		✗	va_end(vl);
134		✗	}
135
136			/* Generate a digest reply, according to RFC 2617. */
137		✗	static char *make_digest_auth(HTTPAuthState *state, const char *username,
138			const char *password, const char *uri,
139			const char *method)
140			{
141		✗	DigestParams *digest = &state->digest_params;
142			int len;
143			uint32_t cnonce_buf[2];
144			char cnonce[17];
145			char nc[9];
146			int i;
147			char A1hash[33], A2hash[33], response[33];
148			struct AVMD5 *md5ctx;
149			uint8_t hash[16];
150			char *authstr;
151
152		✗	digest->nc++;
153		✗	snprintf(nc, sizeof(nc), "%08x", digest->nc);
154
155			/* Generate a client nonce. */
156		✗	for (i = 0; i < 2; i++)
157		✗	cnonce_buf[i] = av_get_random_seed();
158		✗	ff_data_to_hex(cnonce, (const uint8_t*) cnonce_buf, sizeof(cnonce_buf), 1);
159
160		✗	md5ctx = av_md5_alloc();
161		✗	if (!md5ctx)
162		✗	return NULL;
163
164		✗	av_md5_init(md5ctx);
165		✗	update_md5_strings(md5ctx, username, ":", state->realm, ":", password, NULL);
166		✗	av_md5_final(md5ctx, hash);
167		✗	ff_data_to_hex(A1hash, hash, 16, 1);
168
169		✗	if (!strcmp(digest->algorithm, "") || !strcmp(digest->algorithm, "MD5")) {
170		✗	} else if (!strcmp(digest->algorithm, "MD5-sess")) {
171		✗	av_md5_init(md5ctx);
172		✗	update_md5_strings(md5ctx, A1hash, ":", digest->nonce, ":", cnonce, NULL);
173		✗	av_md5_final(md5ctx, hash);
174		✗	ff_data_to_hex(A1hash, hash, 16, 1);
175			} else {
176			/* Unsupported algorithm */
177		✗	av_free(md5ctx);
178		✗	return NULL;
179			}
180
181		✗	av_md5_init(md5ctx);
182		✗	update_md5_strings(md5ctx, method, ":", uri, NULL);
183		✗	av_md5_final(md5ctx, hash);
184		✗	ff_data_to_hex(A2hash, hash, 16, 1);
185
186		✗	av_md5_init(md5ctx);
187		✗	update_md5_strings(md5ctx, A1hash, ":", digest->nonce, NULL);
188		✗	if (!strcmp(digest->qop, "auth") || !strcmp(digest->qop, "auth-int")) {
189		✗	update_md5_strings(md5ctx, ":", nc, ":", cnonce, ":", digest->qop, NULL);
190			}
191		✗	update_md5_strings(md5ctx, ":", A2hash, NULL);
192		✗	av_md5_final(md5ctx, hash);
193		✗	ff_data_to_hex(response, hash, 16, 1);
194
195		✗	av_free(md5ctx);
196
197		✗	if (!strcmp(digest->qop, "") || !strcmp(digest->qop, "auth")) {
198		✗	} else if (!strcmp(digest->qop, "auth-int")) {
199			/* qop=auth-int not supported */
200		✗	return NULL;
201			} else {
202			/* Unsupported qop value. */
203		✗	return NULL;
204			}
205
206		✗	len = strlen(username) + strlen(state->realm) + strlen(digest->nonce) +
207		✗	strlen(uri) + strlen(response) + strlen(digest->algorithm) +
208		✗	strlen(digest->opaque) + strlen(digest->qop) + strlen(cnonce) +
209		✗	strlen(nc) + 150;
210
211		✗	authstr = av_malloc(len);
212		✗	if (!authstr)
213		✗	return NULL;
214		✗	snprintf(authstr, len, "Authorization: Digest ");
215
216			/* TODO: Escape the quoted strings properly. */
217		✗	av_strlcatf(authstr, len, "username=\"%s\"", username);
218		✗	av_strlcatf(authstr, len, ", realm=\"%s\"", state->realm);
219		✗	av_strlcatf(authstr, len, ", nonce=\"%s\"", digest->nonce);
220		✗	av_strlcatf(authstr, len, ", uri=\"%s\"", uri);
221		✗	av_strlcatf(authstr, len, ", response=\"%s\"", response);
222
223			// we are violating the RFC and use "" because all others seem to do that too.
224		✗	if (digest->algorithm[0])
225		✗	av_strlcatf(authstr, len, ", algorithm=\"%s\"", digest->algorithm);
226
227		✗	if (digest->opaque[0])
228		✗	av_strlcatf(authstr, len, ", opaque=\"%s\"", digest->opaque);
229		✗	if (digest->qop[0]) {
230		✗	av_strlcatf(authstr, len, ", qop=\"%s\"", digest->qop);
231		✗	av_strlcatf(authstr, len, ", cnonce=\"%s\"", cnonce);
232		✗	av_strlcatf(authstr, len, ", nc=%s", nc);
233			}
234
235		✗	av_strlcatf(authstr, len, "\r\n");
236
237		✗	return authstr;
238			}
239
240		✗	char *ff_http_auth_create_response(HTTPAuthState *state, const char *auth,
241			const char *path, const char *method)
242			{
243		✗	char *authstr = NULL;
244
245			/* Clear the stale flag, we assume the auth is ok now. It is reset
246			* by the server headers if there's a new issue. */
247		✗	state->stale = 0;
248		✗	if (!auth || !strchr(auth, ':'))
249		✗	return NULL;
250
251		✗	if (state->auth_type == HTTP_AUTH_BASIC) {
252			int auth_b64_len, len;
253		✗	char *ptr, *decoded_auth = ff_urldecode(auth, 0);
254
255		✗	if (!decoded_auth)
256		✗	return NULL;
257
258		✗	auth_b64_len = AV_BASE64_SIZE(strlen(decoded_auth));
259		✗	len = auth_b64_len + 30;
260
261		✗	authstr = av_malloc(len);
262		✗	if (!authstr) {
263		✗	av_free(decoded_auth);
264		✗	return NULL;
265			}
266
267		✗	snprintf(authstr, len, "Authorization: Basic ");
268		✗	ptr = authstr + strlen(authstr);
269		✗	av_base64_encode(ptr, auth_b64_len, decoded_auth, strlen(decoded_auth));
270		✗	av_strlcat(ptr, "\r\n", len - (ptr - authstr));
271		✗	av_free(decoded_auth);
272		✗	} else if (state->auth_type == HTTP_AUTH_DIGEST) {
273		✗	char *username = ff_urldecode(auth, 0), *password;
274
275		✗	if (!username)
276		✗	return NULL;
277
278		✗	if ((password = strchr(username, ':'))) {
279		✗	*password++ = 0;
280		✗	authstr = make_digest_auth(state, username, password, path, method);
281			}
282		✗	av_free(username);
283			}
284		✗	return authstr;
285			}
286