git.sesse.net Git - ffmpeg/blob - libavformat/httpauth.c

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