LCOV - code coverage report
Current view: top level - libavformat - tta.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 70 99 70.7 %
Date: 2017-12-18 20:14:19 Functions: 3 4 75.0 %

          Line data    Source code
       1             : /*
       2             :  * TTA demuxer
       3             :  * Copyright (c) 2006 Alex Beregszaszi
       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/crc.h"
      23             : #include "libavutil/dict.h"
      24             : #include "libavutil/intreadwrite.h"
      25             : 
      26             : #include "apetag.h"
      27             : #include "avformat.h"
      28             : #include "avio_internal.h"
      29             : #include "internal.h"
      30             : #include "id3v1.h"
      31             : 
      32             : typedef struct TTAContext {
      33             :     int totalframes, currentframe;
      34             :     int frame_size;
      35             :     int last_frame_size;
      36             : } TTAContext;
      37             : 
      38        6130 : static int tta_probe(AVProbeData *p)
      39             : {
      40        6134 :     if (AV_RL32(&p->buf[0]) == MKTAG('T', 'T', 'A', '1') &&
      41           9 :         (AV_RL16(&p->buf[4]) == 1 || AV_RL16(&p->buf[4]) == 2) &&
      42           8 :         AV_RL16(&p->buf[6]) > 0 &&
      43           8 :         AV_RL16(&p->buf[8]) > 0 &&
      44           4 :         AV_RL32(&p->buf[10]) > 0)
      45           4 :         return AVPROBE_SCORE_EXTENSION + 30;
      46        6126 :     return 0;
      47             : }
      48             : 
      49           4 : static int tta_read_header(AVFormatContext *s)
      50             : {
      51           4 :     TTAContext *c = s->priv_data;
      52             :     AVStream *st;
      53             :     int i, channels, bps, samplerate;
      54             :     int64_t framepos, start_offset;
      55             :     uint32_t nb_samples, crc;
      56             : 
      57           4 :     ff_id3v1_read(s);
      58             : 
      59           4 :     start_offset = avio_tell(s->pb);
      60           4 :     if (start_offset < 0)
      61           0 :         return start_offset;
      62           4 :     ffio_init_checksum(s->pb, ff_crcEDB88320_update, UINT32_MAX);
      63           4 :     if (avio_rl32(s->pb) != AV_RL32("TTA1"))
      64           0 :         return AVERROR_INVALIDDATA;
      65             : 
      66           4 :     avio_skip(s->pb, 2); // FIXME: flags
      67           4 :     channels = avio_rl16(s->pb);
      68           4 :     bps = avio_rl16(s->pb);
      69           4 :     samplerate = avio_rl32(s->pb);
      70           4 :     if(samplerate <= 0 || samplerate > 1000000){
      71           0 :         av_log(s, AV_LOG_ERROR, "nonsense samplerate\n");
      72           0 :         return AVERROR_INVALIDDATA;
      73             :     }
      74             : 
      75           4 :     nb_samples = avio_rl32(s->pb);
      76           4 :     if (!nb_samples) {
      77           0 :         av_log(s, AV_LOG_ERROR, "invalid number of samples\n");
      78           0 :         return AVERROR_INVALIDDATA;
      79             :     }
      80             : 
      81           4 :     crc = ffio_get_checksum(s->pb) ^ UINT32_MAX;
      82           4 :     if (crc != avio_rl32(s->pb) && s->error_recognition & AV_EF_CRCCHECK) {
      83           0 :         av_log(s, AV_LOG_ERROR, "Header CRC error\n");
      84           0 :         return AVERROR_INVALIDDATA;
      85             :     }
      86             : 
      87           4 :     c->frame_size      = samplerate * 256 / 245;
      88           4 :     c->last_frame_size = nb_samples % c->frame_size;
      89           4 :     if (!c->last_frame_size)
      90           0 :         c->last_frame_size = c->frame_size;
      91           4 :     c->totalframes = nb_samples / c->frame_size + (c->last_frame_size < c->frame_size);
      92           4 :     c->currentframe = 0;
      93             : 
      94           4 :     if(c->totalframes >= UINT_MAX/sizeof(uint32_t) || c->totalframes <= 0){
      95           0 :         av_log(s, AV_LOG_ERROR, "totalframes %d invalid\n", c->totalframes);
      96           0 :         return AVERROR_INVALIDDATA;
      97             :     }
      98             : 
      99           4 :     st = avformat_new_stream(s, NULL);
     100           4 :     if (!st)
     101           0 :         return AVERROR(ENOMEM);
     102             : 
     103           4 :     avpriv_set_pts_info(st, 64, 1, samplerate);
     104           4 :     st->start_time = 0;
     105           4 :     st->duration = nb_samples;
     106             : 
     107           4 :     framepos = avio_tell(s->pb);
     108           4 :     if (framepos < 0)
     109           0 :         return framepos;
     110           4 :     framepos += 4 * c->totalframes + 4;
     111             : 
     112           4 :     if (ff_alloc_extradata(st->codecpar, avio_tell(s->pb) - start_offset))
     113           0 :         return AVERROR(ENOMEM);
     114             : 
     115           4 :     avio_seek(s->pb, start_offset, SEEK_SET);
     116           4 :     avio_read(s->pb, st->codecpar->extradata, st->codecpar->extradata_size);
     117             : 
     118           4 :     ffio_init_checksum(s->pb, ff_crcEDB88320_update, UINT32_MAX);
     119          31 :     for (i = 0; i < c->totalframes; i++) {
     120          27 :         uint32_t size = avio_rl32(s->pb);
     121             :         int r;
     122          27 :         if ((r = av_add_index_entry(st, framepos, i * c->frame_size, size, 0,
     123             :                                     AVINDEX_KEYFRAME)) < 0)
     124           0 :             return r;
     125          27 :         framepos += size;
     126             :     }
     127           4 :     crc = ffio_get_checksum(s->pb) ^ UINT32_MAX;
     128           4 :     if (crc != avio_rl32(s->pb) && s->error_recognition & AV_EF_CRCCHECK) {
     129           0 :         av_log(s, AV_LOG_ERROR, "Seek table CRC error\n");
     130           0 :         return AVERROR_INVALIDDATA;
     131             :     }
     132             : 
     133           4 :     st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
     134           4 :     st->codecpar->codec_id = AV_CODEC_ID_TTA;
     135           4 :     st->codecpar->channels = channels;
     136           4 :     st->codecpar->sample_rate = samplerate;
     137           4 :     st->codecpar->bits_per_coded_sample = bps;
     138             : 
     139           4 :     if (s->pb->seekable & AVIO_SEEKABLE_NORMAL) {
     140           4 :         int64_t pos = avio_tell(s->pb);
     141           4 :         ff_ape_parse_tag(s);
     142           4 :         avio_seek(s->pb, pos, SEEK_SET);
     143             :     }
     144             : 
     145           4 :     return 0;
     146             : }
     147             : 
     148          31 : static int tta_read_packet(AVFormatContext *s, AVPacket *pkt)
     149             : {
     150          31 :     TTAContext *c = s->priv_data;
     151          31 :     AVStream *st = s->streams[0];
     152             :     int size, ret;
     153             : 
     154             :     // FIXME!
     155          31 :     if (c->currentframe >= c->totalframes)
     156           4 :         return AVERROR_EOF;
     157             : 
     158          27 :     if (st->nb_index_entries < c->totalframes) {
     159           0 :         av_log(s, AV_LOG_ERROR, "Index entry disappeared\n");
     160           0 :         return AVERROR_INVALIDDATA;
     161             :     }
     162             : 
     163          27 :     size = st->index_entries[c->currentframe].size;
     164             : 
     165          27 :     ret = av_get_packet(s->pb, pkt, size);
     166          27 :     pkt->dts = st->index_entries[c->currentframe++].timestamp;
     167          50 :     pkt->duration = c->currentframe == c->totalframes ? c->last_frame_size :
     168          23 :                                                         c->frame_size;
     169          27 :     return ret;
     170             : }
     171             : 
     172           0 : static int tta_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
     173             : {
     174           0 :     TTAContext *c = s->priv_data;
     175           0 :     AVStream *st = s->streams[stream_index];
     176           0 :     int index = av_index_search_timestamp(st, timestamp, flags);
     177           0 :     if (index < 0)
     178           0 :         return -1;
     179           0 :     if (avio_seek(s->pb, st->index_entries[index].pos, SEEK_SET) < 0)
     180           0 :         return -1;
     181             : 
     182           0 :     c->currentframe = index;
     183             : 
     184           0 :     return 0;
     185             : }
     186             : 
     187             : AVInputFormat ff_tta_demuxer = {
     188             :     .name           = "tta",
     189             :     .long_name      = NULL_IF_CONFIG_SMALL("TTA (True Audio)"),
     190             :     .priv_data_size = sizeof(TTAContext),
     191             :     .read_probe     = tta_probe,
     192             :     .read_header    = tta_read_header,
     193             :     .read_packet    = tta_read_packet,
     194             :     .read_seek      = tta_read_seek,
     195             :     .extensions     = "tta",
     196             : };

Generated by: LCOV version 1.13