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

   1 /*
   2  * Format register and lookup
   3  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
   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 "libavutil/avstring.h"
  23 #include "libavutil/bprint.h"
  24 #include "libavutil/opt.h"
  25 #include "libavutil/thread.h"
  26
  27 #include "avio_internal.h"
  28 #include "avformat.h"
  29 #include "id3v2.h"
  30 #include "internal.h"
  31
  32
  33 /**
  34  * @file
  35  * Format register and lookup
  36  */
  37 /** head of registered input format linked list */
  38 static AVInputFormat *first_iformat = NULL;
  39 /** head of registered output format linked list */
  40 static AVOutputFormat *first_oformat = NULL;
  41
  42 static AVInputFormat **last_iformat = &first_iformat;
  43 static AVOutputFormat **last_oformat = &first_oformat;
  44
  45 AVInputFormat *av_iformat_next(const AVInputFormat *f)
  46 {
  47     if (f)
  48         return f->next;
  49     else
  50         return first_iformat;
  51 }
  52
  53 AVOutputFormat *av_oformat_next(const AVOutputFormat *f)
  54 {
  55     if (f)
  56         return f->next;
  57     else
  58         return first_oformat;
  59 }
  60
  61 static AVMutex iformat_register_mutex = AV_MUTEX_INITIALIZER;
  62
  63 void av_register_input_format(AVInputFormat *format)
  64 {
  65     AVInputFormat **p;
  66
  67     ff_mutex_lock(&iformat_register_mutex);
  68     p = last_iformat;
  69
  70     while (*p)
  71         p = &(*p)->next;
  72     *p = format;
  73     format->next = NULL;
  74     last_iformat = &format->next;
  75
  76     ff_mutex_unlock(&iformat_register_mutex);
  77 }
  78
  79 static AVMutex oformat_register_mutex = AV_MUTEX_INITIALIZER;
  80
  81 void av_register_output_format(AVOutputFormat *format)
  82 {
  83     AVOutputFormat **p;
  84
  85     ff_mutex_lock(&oformat_register_mutex);
  86     p = last_oformat;
  87
  88     while (*p)
  89         p = &(*p)->next;
  90     *p = format;
  91     format->next = NULL;
  92     last_oformat = &format->next;
  93
  94     ff_mutex_unlock(&oformat_register_mutex);
  95 }
  96
  97 int av_match_ext(const char *filename, const char *extensions)
  98 {
  99     const char *ext;
 100
 101     if (!filename)
 102         return 0;
 103
 104     ext = strrchr(filename, '.');
 105     if (ext)
 106         return av_match_name(ext + 1, extensions);
 107     return 0;
 108 }
 109
 110 AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
 111                                 const char *mime_type)
 112 {
 113     AVOutputFormat *fmt = NULL, *fmt_found;
 114     int score_max, score;
 115
 116     /* specific test for image sequences */
 117 #if CONFIG_IMAGE2_MUXER
 118     if (!short_name && filename &&
 119         av_filename_number_test(filename) &&
 120         ff_guess_image2_codec(filename) != AV_CODEC_ID_NONE) {
 121         return av_guess_format("image2", NULL, NULL);
 122     }
 123 #endif
 124     /* Find the proper file type. */
 125     fmt_found = NULL;
 126     score_max = 0;
 127     while ((fmt = av_oformat_next(fmt))) {
 128         score = 0;
 129         if (fmt->name && short_name && av_match_name(short_name, fmt->name))
 130             score += 100;
 131         if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
 132             score += 10;
 133         if (filename && fmt->extensions &&
 134             av_match_ext(filename, fmt->extensions)) {
 135             score += 5;
 136         }
 137         if (score > score_max) {
 138             score_max = score;
 139             fmt_found = fmt;
 140         }
 141     }
 142     return fmt_found;
 143 }
 144
 145 enum AVCodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
 146                               const char *filename, const char *mime_type,
 147                               enum AVMediaType type)
 148 {
 149     if (av_match_name("segment", fmt->name) || av_match_name("ssegment", fmt->name)) {
 150         AVOutputFormat *fmt2 = av_guess_format(NULL, filename, NULL);
 151         if (fmt2)
 152             fmt = fmt2;
 153     }
 154
 155     if (type == AVMEDIA_TYPE_VIDEO) {
 156         enum AVCodecID codec_id = AV_CODEC_ID_NONE;
 157
 158 #if CONFIG_IMAGE2_MUXER
 159         if (!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")) {
 160             codec_id = ff_guess_image2_codec(filename);
 161         }
 162 #endif
 163         if (codec_id == AV_CODEC_ID_NONE)
 164             codec_id = fmt->video_codec;
 165         return codec_id;
 166     } else if (type == AVMEDIA_TYPE_AUDIO)
 167         return fmt->audio_codec;
 168     else if (type == AVMEDIA_TYPE_SUBTITLE)
 169         return fmt->subtitle_codec;
 170     else if (type == AVMEDIA_TYPE_DATA)
 171         return fmt->data_codec;
 172     else
 173         return AV_CODEC_ID_NONE;
 174 }
 175
 176 AVInputFormat *av_find_input_format(const char *short_name)
 177 {
 178     AVInputFormat *fmt = NULL;
 179     while ((fmt = av_iformat_next(fmt)))
 180         if (av_match_name(short_name, fmt->name))
 181             return fmt;
 182     return NULL;
 183 }
 184
 185 AVInputFormat *av_probe_input_format3(AVProbeData *pd, int is_opened,
 186                                       int *score_ret)
 187 {
 188     AVProbeData lpd = *pd;
 189     AVInputFormat *fmt1 = NULL, *fmt;
 190     int score, score_max = 0;
 191     const static uint8_t zerobuffer[AVPROBE_PADDING_SIZE];
 192     enum nodat {
 193         NO_ID3,
 194         ID3_ALMOST_GREATER_PROBE,
 195         ID3_GREATER_PROBE,
 196         ID3_GREATER_MAX_PROBE,
 197     } nodat = NO_ID3;
 198
 199     if (!lpd.buf)
 200         lpd.buf = (unsigned char *) zerobuffer;
 201
 202     if (lpd.buf_size > 10 && ff_id3v2_match(lpd.buf, ID3v2_DEFAULT_MAGIC)) {
 203         int id3len = ff_id3v2_tag_len(lpd.buf);
 204         if (lpd.buf_size > id3len + 16) {
 205             if (lpd.buf_size < 2LL*id3len + 16)
 206                 nodat = ID3_ALMOST_GREATER_PROBE;
 207             lpd.buf      += id3len;
 208             lpd.buf_size -= id3len;
 209         } else if (id3len >= PROBE_BUF_MAX) {
 210             nodat = ID3_GREATER_MAX_PROBE;
 211         } else
 212             nodat = ID3_GREATER_PROBE;
 213     }
 214
 215     fmt = NULL;
 216     while ((fmt1 = av_iformat_next(fmt1))) {
 217         if (!is_opened == !(fmt1->flags & AVFMT_NOFILE) && strcmp(fmt1->name, "image2"))
 218             continue;
 219         score = 0;
 220         if (fmt1->read_probe) {
 221             score = fmt1->read_probe(&lpd);
 222             if (score)
 223                 av_log(NULL, AV_LOG_TRACE, "Probing %s score:%d size:%d\n", fmt1->name, score, lpd.buf_size);
 224             if (fmt1->extensions && av_match_ext(lpd.filename, fmt1->extensions)) {
 225                 switch (nodat) {
 226                 case NO_ID3:
 227                     score = FFMAX(score, 1);
 228                     break;
 229                 case ID3_GREATER_PROBE:
 230                 case ID3_ALMOST_GREATER_PROBE:
 231                     score = FFMAX(score, AVPROBE_SCORE_EXTENSION / 2 - 1);
 232                     break;
 233                 case ID3_GREATER_MAX_PROBE:
 234                     score = FFMAX(score, AVPROBE_SCORE_EXTENSION);
 235                     break;
 236                 }
 237             }
 238         } else if (fmt1->extensions) {
 239             if (av_match_ext(lpd.filename, fmt1->extensions))
 240                 score = AVPROBE_SCORE_EXTENSION;
 241         }
 242         if (av_match_name(lpd.mime_type, fmt1->mime_type)) {
 243             if (AVPROBE_SCORE_MIME > score) {
 244                 av_log(NULL, AV_LOG_DEBUG, "Probing %s score:%d increased to %d due to MIME type\n", fmt1->name, score, AVPROBE_SCORE_MIME);
 245                 score = AVPROBE_SCORE_MIME;
 246             }
 247         }
 248         if (score > score_max) {
 249             score_max = score;
 250             fmt       = fmt1;
 251         } else if (score == score_max)
 252             fmt = NULL;
 253     }
 254     if (nodat == ID3_GREATER_PROBE)
 255         score_max = FFMIN(AVPROBE_SCORE_EXTENSION / 2 - 1, score_max);
 256     *score_ret = score_max;
 257
 258     return fmt;
 259 }
 260
 261 AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)
 262 {
 263     int score_ret;
 264     AVInputFormat *fmt = av_probe_input_format3(pd, is_opened, &score_ret);
 265     if (score_ret > *score_max) {
 266         *score_max = score_ret;
 267         return fmt;
 268     } else
 269         return NULL;
 270 }
 271
 272 AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened)
 273 {
 274     int score = 0;
 275     return av_probe_input_format2(pd, is_opened, &score);
 276 }
 277
 278 int av_probe_input_buffer2(AVIOContext *pb, AVInputFormat **fmt,
 279                           const char *filename, void *logctx,
 280                           unsigned int offset, unsigned int max_probe_size)
 281 {
 282     AVProbeData pd = { filename ? filename : "" };
 283     uint8_t *buf = NULL;
 284     int ret = 0, probe_size, buf_offset = 0;
 285     int score = 0;
 286     int ret2;
 287
 288     if (!max_probe_size)
 289         max_probe_size = PROBE_BUF_MAX;
 290     else if (max_probe_size < PROBE_BUF_MIN) {
 291         av_log(logctx, AV_LOG_ERROR,
 292                "Specified probe size value %u cannot be < %u\n", max_probe_size, PROBE_BUF_MIN);
 293         return AVERROR(EINVAL);
 294     }
 295
 296     if (offset >= max_probe_size)
 297         return AVERROR(EINVAL);
 298
 299     if (pb->av_class) {
 300         uint8_t *mime_type_opt = NULL;
 301         char *semi;
 302         av_opt_get(pb, "mime_type", AV_OPT_SEARCH_CHILDREN, &mime_type_opt);
 303         pd.mime_type = (const char *)mime_type_opt;
 304         semi = pd.mime_type ? strchr(pd.mime_type, ';') : NULL;
 305         if (semi) {
 306             *semi = '\0';
 307         }
 308     }
 309 #if 0
 310     if (!*fmt && pb->av_class && av_opt_get(pb, "mime_type", AV_OPT_SEARCH_CHILDREN, &mime_type) >= 0 && mime_type) {
 311         if (!av_strcasecmp(mime_type, "audio/aacp")) {
 312             *fmt = av_find_input_format("aac");
 313         }
 314         av_freep(&mime_type);
 315     }
 316 #endif
 317
 318     for (probe_size = PROBE_BUF_MIN; probe_size <= max_probe_size && !*fmt;
 319          probe_size = FFMIN(probe_size << 1,
 320                             FFMAX(max_probe_size, probe_size + 1))) {
 321         score = probe_size < max_probe_size ? AVPROBE_SCORE_RETRY : 0;
 322
 323         /* Read probe data. */
 324         if ((ret = av_reallocp(&buf, probe_size + AVPROBE_PADDING_SIZE)) < 0)
 325             goto fail;
 326         if ((ret = avio_read(pb, buf + buf_offset,
 327                              probe_size - buf_offset)) < 0) {
 328             /* Fail if error was not end of file, otherwise, lower score. */
 329             if (ret != AVERROR_EOF)
 330                 goto fail;
 331
 332             score = 0;
 333             ret   = 0;          /* error was end of file, nothing read */
 334         }
 335         buf_offset += ret;
 336         if (buf_offset < offset)
 337             continue;
 338         pd.buf_size = buf_offset - offset;
 339         pd.buf = &buf[offset];
 340
 341         memset(pd.buf + pd.buf_size, 0, AVPROBE_PADDING_SIZE);
 342
 343         /* Guess file format. */
 344         *fmt = av_probe_input_format2(&pd, 1, &score);
 345         if (*fmt) {
 346             /* This can only be true in the last iteration. */
 347             if (score <= AVPROBE_SCORE_RETRY) {
 348                 av_log(logctx, AV_LOG_WARNING,
 349                        "Format %s detected only with low score of %d, "
 350                        "misdetection possible!\n", (*fmt)->name, score);
 351             } else
 352                 av_log(logctx, AV_LOG_DEBUG,
 353                        "Format %s probed with size=%d and score=%d\n",
 354                        (*fmt)->name, probe_size, score);
 355 #if 0
 356             FILE *f = fopen("probestat.tmp", "ab");
 357             fprintf(f, "probe_size:%d format:%s score:%d filename:%s\n", probe_size, (*fmt)->name, score, filename);
 358             fclose(f);
 359 #endif
 360         }
 361     }
 362
 363     if (!*fmt)
 364         ret = AVERROR_INVALIDDATA;
 365
 366 fail:
 367     /* Rewind. Reuse probe buffer to avoid seeking. */
 368     ret2 = ffio_rewind_with_probe_data(pb, &buf, buf_offset);
 369     if (ret >= 0)
 370         ret = ret2;
 371
 372     av_freep(&pd.mime_type);
 373     return ret < 0 ? ret : score;
 374 }
 375
 376 int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt,
 377                           const char *filename, void *logctx,
 378                           unsigned int offset, unsigned int max_probe_size)
 379 {
 380     int ret = av_probe_input_buffer2(pb, fmt, filename, logctx, offset, max_probe_size);
 381     return ret < 0 ? ret : 0;
 382 }