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

   1 /* Electronic Arts Multimedia File Demuxer
   2  * Copyright (c) 2004  The FFmpeg project
   3  * Copyright (c) 2006-2008 Peter Ross
   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 /**
  23  * @file
  24  * Electronic Arts Multimedia file demuxer (WVE/UV2/etc.)
  25  * by Robin Kay (komadori at gekkou.co.uk)
  26  */
  27
  28 #include <inttypes.h>
  29
  30 #include "libavutil/intreadwrite.h"
  31 #include "avformat.h"
  32 #include "internal.h"
  33
  34 #define SCHl_TAG MKTAG('S', 'C', 'H', 'l')
  35 #define SEAD_TAG MKTAG('S', 'E', 'A', 'D')  /* Sxxx header */
  36 #define SNDC_TAG MKTAG('S', 'N', 'D', 'C')  /* Sxxx data */
  37 #define SEND_TAG MKTAG('S', 'E', 'N', 'D')  /* Sxxx end */
  38 #define SHEN_TAG MKTAG('S', 'H', 'E', 'N')  /* SxEN header */
  39 #define SDEN_TAG MKTAG('S', 'D', 'E', 'N')  /* SxEN data */
  40 #define SEEN_TAG MKTAG('S', 'E', 'E', 'N')  /* SxEN end */
  41 #define ISNh_TAG MKTAG('1', 'S', 'N', 'h')  /* 1SNx header */
  42 #define EACS_TAG MKTAG('E', 'A', 'C', 'S')
  43 #define ISNd_TAG MKTAG('1', 'S', 'N', 'd')  /* 1SNx data */
  44 #define ISNe_TAG MKTAG('1', 'S', 'N', 'e')  /* 1SNx end */
  45 #define PT00_TAG MKTAG('P', 'T', 0x0, 0x0)
  46 #define GSTR_TAG MKTAG('G', 'S', 'T', 'R')
  47 #define SCDl_TAG MKTAG('S', 'C', 'D', 'l')
  48 #define SCEl_TAG MKTAG('S', 'C', 'E', 'l')
  49 #define kVGT_TAG MKTAG('k', 'V', 'G', 'T')  /* TGV I-frame */
  50 #define fVGT_TAG MKTAG('f', 'V', 'G', 'T')  /* TGV P-frame */
  51 #define mTCD_TAG MKTAG('m', 'T', 'C', 'D')  /* MDEC */
  52 #define MADk_TAG MKTAG('M', 'A', 'D', 'k')  /* MAD I-frame */
  53 #define MADm_TAG MKTAG('M', 'A', 'D', 'm')  /* MAD P-frame */
  54 #define MADe_TAG MKTAG('M', 'A', 'D', 'e')  /* MAD lqp-frame */
  55 #define MPCh_TAG MKTAG('M', 'P', 'C', 'h')  /* MPEG-2 */
  56 #define TGQs_TAG MKTAG('T', 'G', 'Q', 's')  /* TGQ I-frame (appears in .TGQ files) */
  57 #define pQGT_TAG MKTAG('p', 'Q', 'G', 'T')  /* TGQ I-frame (appears in .UV files) */
  58 #define pIQT_TAG MKTAG('p', 'I', 'Q', 'T')  /* TQI/UV2 I-frame (.UV2/.WVE) */
  59 #define MVhd_TAG MKTAG('M', 'V', 'h', 'd')
  60 #define MV0K_TAG MKTAG('M', 'V', '0', 'K')
  61 #define MV0F_TAG MKTAG('M', 'V', '0', 'F')
  62 #define AVhd_TAG MKTAG('A', 'V', 'h', 'd')
  63 #define AV0K_TAG MKTAG('A', 'V', '0', 'K')
  64 #define AV0F_TAG MKTAG('A', 'V', '0', 'F')
  65 #define MVIh_TAG MKTAG('M', 'V', 'I', 'h')  /* CMV header */
  66 #define MVIf_TAG MKTAG('M', 'V', 'I', 'f')  /* CMV I-frame */
  67 #define AVP6_TAG MKTAG('A', 'V', 'P', '6')
  68
  69 typedef struct VideoProperties {
  70     enum AVCodecID codec;
  71     AVRational time_base;
  72     int width, height;
  73     int nb_frames;
  74     int stream_index;
  75 } VideoProperties;
  76
  77 typedef struct EaDemuxContext {
  78     int big_endian;
  79
  80     VideoProperties video, alpha;
  81
  82     enum AVCodecID audio_codec;
  83     int audio_stream_index;
  84
  85     int bytes;
  86     int sample_rate;
  87     int num_channels;
  88     int num_samples;
  89
  90     int platform;
  91 } EaDemuxContext;
  92
  93 static uint32_t read_arbitrary(AVIOContext *pb)
  94 {
  95     uint8_t size, byte;
  96     int i;
  97     uint32_t word;
  98
  99     size = avio_r8(pb);
 100
 101     word = 0;
 102     for (i = 0; i < size; i++) {
 103         byte   = avio_r8(pb);
 104         word <<= 8;
 105         word  |= byte;
 106     }
 107
 108     return word;
 109 }
 110
 111 static int process_audio_header_elements(AVFormatContext *s)
 112 {
 113     EaDemuxContext *ea = s->priv_data;
 114     AVIOContext    *pb = s->pb;
 115     int in_header = 1;
 116     int compression_type = -1, revision = -1, revision2 = -1;
 117
 118     ea->bytes        = 2;
 119     ea->sample_rate  = -1;
 120     ea->num_channels = 1;
 121
 122     while (!avio_feof(pb) && in_header) {
 123         int in_subheader;
 124         uint8_t byte;
 125         byte = avio_r8(pb);
 126
 127         switch (byte) {
 128         case 0xFD:
 129             av_log(s, AV_LOG_DEBUG, "entered audio subheader\n");
 130             in_subheader = 1;
 131             while (!avio_feof(pb) && in_subheader) {
 132                 uint8_t subbyte;
 133                 subbyte = avio_r8(pb);
 134
 135                 switch (subbyte) {
 136                 case 0x80:
 137                     revision = read_arbitrary(pb);
 138                     av_log(s, AV_LOG_DEBUG,
 139                            "revision (element 0x80) set to 0x%08x\n", revision);
 140                     break;
 141                 case 0x82:
 142                     ea->num_channels = read_arbitrary(pb);
 143                     av_log(s, AV_LOG_DEBUG,
 144                            "num_channels (element 0x82) set to 0x%08x\n",
 145                            ea->num_channels);
 146                     break;
 147                 case 0x83:
 148                     compression_type = read_arbitrary(pb);
 149                     av_log(s, AV_LOG_DEBUG,
 150                            "compression_type (element 0x83) set to 0x%08x\n",
 151                            compression_type);
 152                     break;
 153                 case 0x84:
 154                     ea->sample_rate = read_arbitrary(pb);
 155                     av_log(s, AV_LOG_DEBUG,
 156                            "sample_rate (element 0x84) set to %i\n",
 157                            ea->sample_rate);
 158                     break;
 159                 case 0x85:
 160                     ea->num_samples = read_arbitrary(pb);
 161                     av_log(s, AV_LOG_DEBUG,
 162                            "num_samples (element 0x85) set to 0x%08x\n",
 163                            ea->num_samples);
 164                     break;
 165                 case 0x8A:
 166                     av_log(s, AV_LOG_DEBUG,
 167                            "element 0x%02x set to 0x%08"PRIx32"\n",
 168                            subbyte, read_arbitrary(pb));
 169                     av_log(s, AV_LOG_DEBUG, "exited audio subheader\n");
 170                     in_subheader = 0;
 171                     break;
 172                 case 0xA0:
 173                     revision2 = read_arbitrary(pb);
 174                     av_log(s, AV_LOG_DEBUG,
 175                            "revision2 (element 0xA0) set to 0x%08x\n",
 176                            revision2);
 177                     break;
 178                 case 0xFF:
 179                     av_log(s, AV_LOG_DEBUG,
 180                            "end of header block reached (within audio subheader)\n");
 181                     in_subheader = 0;
 182                     in_header    = 0;
 183                     break;
 184                 default:
 185                     av_log(s, AV_LOG_DEBUG,
 186                            "element 0x%02x set to 0x%08"PRIx32"\n",
 187                            subbyte, read_arbitrary(pb));
 188                     break;
 189                 }
 190             }
 191             break;
 192         case 0xFF:
 193             av_log(s, AV_LOG_DEBUG, "end of header block reached\n");
 194             in_header = 0;
 195             break;
 196         default:
 197             av_log(s, AV_LOG_DEBUG,
 198                    "header element 0x%02x set to 0x%08"PRIx32"\n",
 199                    byte, read_arbitrary(pb));
 200             break;
 201         }
 202     }
 203
 204     switch (compression_type) {
 205     case  0:
 206         ea->audio_codec = AV_CODEC_ID_PCM_S16LE;
 207         break;
 208     case  7:
 209         ea->audio_codec = AV_CODEC_ID_ADPCM_EA;
 210         break;
 211     case -1:
 212         switch (revision) {
 213         case  1:
 214             ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R1;
 215             break;
 216         case  2:
 217             ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R2;
 218             break;
 219         case  3:
 220             ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R3;
 221             break;
 222         case -1:
 223             break;
 224         default:
 225             avpriv_request_sample(s, "stream type; revision=%i", revision);
 226             return 0;
 227         }
 228         switch (revision2) {
 229         case  8:
 230             ea->audio_codec = AV_CODEC_ID_PCM_S16LE_PLANAR;
 231             break;
 232         case 10:
 233             switch (revision) {
 234             case -1:
 235             case  2: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R1; break;
 236             case  3: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R2; break;
 237             default:
 238                 avpriv_request_sample(s, "stream type; revision=%i, revision2=%i", revision, revision2);
 239                 return 0;
 240             }
 241             break;
 242         case 15:
 243         case 16:
 244             ea->audio_codec = AV_CODEC_ID_MP3;
 245             break;
 246         case -1:
 247             break;
 248         default:
 249             ea->audio_codec = AV_CODEC_ID_NONE;
 250             avpriv_request_sample(s, "stream type; revision2=%i", revision2);
 251             return 0;
 252         }
 253         break;
 254     default:
 255         avpriv_request_sample(s,
 256                               "stream type; compression_type=%i",
 257                               compression_type);
 258         return 0;
 259     }
 260
 261     if (ea->audio_codec == AV_CODEC_ID_NONE && ea->platform == 0x01)
 262         ea->audio_codec = AV_CODEC_ID_ADPCM_PSX;
 263     if (ea->sample_rate == -1)
 264         ea->sample_rate = revision == 3 ? 48000 : 22050;
 265
 266     return 1;
 267 }
 268
 269 static void process_audio_header_eacs(AVFormatContext *s)
 270 {
 271     EaDemuxContext *ea = s->priv_data;
 272     AVIOContext *pb    = s->pb;
 273     int compression_type;
 274
 275     ea->sample_rate  = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb);
 276     ea->bytes        = avio_r8(pb);   /* 1=8-bit, 2=16-bit */
 277     ea->num_channels = avio_r8(pb);
 278     compression_type = avio_r8(pb);
 279     avio_skip(pb, 13);
 280
 281     switch (compression_type) {
 282     case 0:
 283         switch (ea->bytes) {
 284         case 1:
 285             ea->audio_codec = AV_CODEC_ID_PCM_S8;
 286             break;
 287         case 2:
 288             ea->audio_codec = AV_CODEC_ID_PCM_S16LE;
 289             break;
 290         }
 291         break;
 292     case 1:
 293         ea->audio_codec = AV_CODEC_ID_PCM_MULAW;
 294         ea->bytes       = 1;
 295         break;
 296     case 2:
 297         ea->audio_codec = AV_CODEC_ID_ADPCM_IMA_EA_EACS;
 298         break;
 299     default:
 300         avpriv_request_sample(s,
 301                               "stream type; audio compression_type=%i",
 302                               compression_type);
 303     }
 304 }
 305
 306 static void process_audio_header_sead(AVFormatContext *s)
 307 {
 308     EaDemuxContext *ea = s->priv_data;
 309     AVIOContext *pb    = s->pb;
 310
 311     ea->sample_rate  = avio_rl32(pb);
 312     ea->bytes        = avio_rl32(pb);  /* 1=8-bit, 2=16-bit */
 313     ea->num_channels = avio_rl32(pb);
 314     ea->audio_codec  = AV_CODEC_ID_ADPCM_IMA_EA_SEAD;
 315 }
 316
 317 static void process_video_header_mdec(AVFormatContext *s, VideoProperties *video)
 318 {
 319     AVIOContext *pb    = s->pb;
 320     avio_skip(pb, 4);
 321     video->width       = avio_rl16(pb);
 322     video->height      = avio_rl16(pb);
 323     video->time_base   = (AVRational) { 1, 15 };
 324     video->codec = AV_CODEC_ID_MDEC;
 325 }
 326
 327 static int process_video_header_vp6(AVFormatContext *s, VideoProperties *video)
 328 {
 329     AVIOContext *pb = s->pb;
 330
 331     avio_skip(pb, 8);
 332     video->nb_frames = avio_rl32(pb);
 333     avio_skip(pb, 4);
 334     video->time_base.den = avio_rl32(pb);
 335     video->time_base.num = avio_rl32(pb);
 336     if (video->time_base.den <= 0 || video->time_base.num <= 0) {
 337         av_log(s, AV_LOG_ERROR, "Timebase is invalid\n");
 338         return AVERROR_INVALIDDATA;
 339     }
 340     video->codec   = AV_CODEC_ID_VP6;
 341
 342     return 1;
 343 }
 344
 345 static void process_video_header_cmv(AVFormatContext *s, VideoProperties *video)
 346 {
 347     int fps;
 348
 349     avio_skip(s->pb, 10);
 350     fps = avio_rl16(s->pb);
 351     if (fps)
 352         video->time_base = (AVRational) { 1, fps };
 353     video->codec = AV_CODEC_ID_CMV;
 354 }
 355
 356 /* Process EA file header.
 357  * Return 1 if the EA file is valid and successfully opened, 0 otherwise. */
 358 static int process_ea_header(AVFormatContext *s)
 359 {
 360     uint32_t blockid, size = 0;
 361     EaDemuxContext *ea = s->priv_data;
 362     AVIOContext *pb    = s->pb;
 363     int i;
 364
 365     for (i = 0; i < 5 && (!ea->audio_codec || !ea->video.codec); i++) {
 366         uint64_t startpos     = avio_tell(pb);
 367         int err               = 0;
 368
 369         blockid = avio_rl32(pb);
 370         size    = avio_rl32(pb);
 371         if (i == 0)
 372             ea->big_endian = size > av_bswap32(size);
 373         if (ea->big_endian)
 374             size = av_bswap32(size);
 375
 376         if (size < 8) {
 377             av_log(s, AV_LOG_ERROR, "chunk size too small\n");
 378             return AVERROR_INVALIDDATA;
 379         }
 380
 381         switch (blockid) {
 382         case ISNh_TAG:
 383             if (avio_rl32(pb) != EACS_TAG) {
 384                 avpriv_request_sample(s, "unknown 1SNh headerid");
 385                 return 0;
 386             }
 387             process_audio_header_eacs(s);
 388             break;
 389
 390         case SCHl_TAG:
 391         case SHEN_TAG:
 392             blockid = avio_rl32(pb);
 393             if (blockid == GSTR_TAG) {
 394                 avio_skip(pb, 4);
 395             } else if ((blockid & 0xFF) != (PT00_TAG & 0xFF)) {
 396                 blockid = avio_rl32(pb);
 397             }
 398             ea->platform = (blockid >> 16) & 0xFF;
 399             err = process_audio_header_elements(s);
 400             break;
 401
 402         case SEAD_TAG:
 403             process_audio_header_sead(s);
 404             break;
 405
 406         case MVIh_TAG:
 407             process_video_header_cmv(s, &ea->video);
 408             break;
 409
 410         case kVGT_TAG:
 411             ea->video.codec = AV_CODEC_ID_TGV;
 412             break;
 413
 414         case mTCD_TAG:
 415             process_video_header_mdec(s, &ea->video);
 416             break;
 417
 418         case MPCh_TAG:
 419             ea->video.codec = AV_CODEC_ID_MPEG2VIDEO;
 420             break;
 421
 422         case pQGT_TAG:
 423         case TGQs_TAG:
 424             ea->video.codec = AV_CODEC_ID_TGQ;
 425             ea->video.time_base   = (AVRational) { 1, 15 };
 426             break;
 427
 428         case pIQT_TAG:
 429             ea->video.codec = AV_CODEC_ID_TQI;
 430             ea->video.time_base   = (AVRational) { 1, 15 };
 431             break;
 432
 433         case MADk_TAG:
 434             ea->video.codec = AV_CODEC_ID_MAD;
 435             avio_skip(pb, 6);
 436             ea->video.time_base = (AVRational) { avio_rl16(pb), 1000 };
 437             break;
 438
 439         case MVhd_TAG:
 440             err = process_video_header_vp6(s, &ea->video);
 441             break;
 442
 443         case AVhd_TAG:
 444             err = process_video_header_vp6(s, &ea->alpha);
 445             break;
 446         }
 447
 448         if (err < 0) {
 449             av_log(s, AV_LOG_ERROR, "error parsing header: %i\n", err);
 450             return err;
 451         }
 452
 453         avio_seek(pb, startpos + size, SEEK_SET);
 454     }
 455
 456     avio_seek(pb, 0, SEEK_SET);
 457
 458     return 1;
 459 }
 460
 461 static int ea_probe(const AVProbeData *p)
 462 {
 463     unsigned big_endian, size;
 464
 465     switch (AV_RL32(&p->buf[0])) {
 466     case ISNh_TAG:
 467     case SCHl_TAG:
 468     case SEAD_TAG:
 469     case SHEN_TAG:
 470     case kVGT_TAG:
 471     case MADk_TAG:
 472     case MPCh_TAG:
 473     case MVhd_TAG:
 474     case MVIh_TAG:
 475     case AVP6_TAG:
 476         break;
 477     default:
 478         return 0;
 479     }
 480     size = AV_RL32(&p->buf[4]);
 481     big_endian = size > 0x000FFFFF;
 482     if (big_endian)
 483         size = av_bswap32(size);
 484     if (size > 0xfffff || size < 8)
 485         return 0;
 486
 487     return AVPROBE_SCORE_MAX;
 488 }
 489
 490 static int init_video_stream(AVFormatContext *s, VideoProperties *video)
 491 {
 492     AVStream *st;
 493
 494     if (!video->codec)
 495         return 0;
 496
 497     /* initialize the video decoder stream */
 498     st = avformat_new_stream(s, NULL);
 499     if (!st)
 500         return AVERROR(ENOMEM);
 501     video->stream_index = st->index;
 502     st->codecpar->codec_type  = AVMEDIA_TYPE_VIDEO;
 503     st->codecpar->codec_id    = video->codec;
 504     // parsing is necessary to make FFmpeg generate correct timestamps
 505     if (st->codecpar->codec_id == AV_CODEC_ID_MPEG2VIDEO)
 506         st->need_parsing = AVSTREAM_PARSE_HEADERS;
 507     st->codecpar->codec_tag   = 0; /* no fourcc */
 508     st->codecpar->width       = video->width;
 509     st->codecpar->height      = video->height;
 510     st->duration           = st->nb_frames = video->nb_frames;
 511     if (video->time_base.num)
 512         avpriv_set_pts_info(st, 64, video->time_base.num, video->time_base.den);
 513     st->r_frame_rate       =
 514     st->avg_frame_rate     = av_inv_q(video->time_base);
 515     return 0;
 516 }
 517
 518 static int ea_read_header(AVFormatContext *s)
 519 {
 520     EaDemuxContext *ea = s->priv_data;
 521     AVStream *st;
 522
 523     if (process_ea_header(s)<=0)
 524         return AVERROR(EIO);
 525
 526     if (init_video_stream(s, &ea->video) || init_video_stream(s, &ea->alpha))
 527         return AVERROR(ENOMEM);
 528
 529     if (ea->audio_codec) {
 530         if (ea->num_channels <= 0 || ea->num_channels > 2) {
 531             av_log(s, AV_LOG_WARNING,
 532                    "Unsupported number of channels: %d\n", ea->num_channels);
 533             goto no_audio;
 534         }
 535         if (ea->sample_rate <= 0) {
 536             av_log(s, AV_LOG_ERROR,
 537                    "Unsupported sample rate: %d\n", ea->sample_rate);
 538             goto no_audio;
 539         }
 540         if (ea->bytes <= 0 || ea->bytes > 2) {
 541             av_log(s, AV_LOG_ERROR,
 542                    "Invalid number of bytes per sample: %d\n", ea->bytes);
 543             goto no_audio;
 544         }
 545
 546         /* initialize the audio decoder stream */
 547         st = avformat_new_stream(s, NULL);
 548         if (!st)
 549             return AVERROR(ENOMEM);
 550         avpriv_set_pts_info(st, 33, 1, ea->sample_rate);
 551         st->codecpar->codec_type            = AVMEDIA_TYPE_AUDIO;
 552         st->codecpar->codec_id              = ea->audio_codec;
 553         st->codecpar->codec_tag             = 0;   /* no tag */
 554         st->codecpar->channels              = ea->num_channels;
 555         st->codecpar->sample_rate           = ea->sample_rate;
 556         st->codecpar->bits_per_coded_sample = ea->bytes * 8;
 557         st->codecpar->bit_rate              = (int64_t)st->codecpar->channels *
 558                                               st->codecpar->sample_rate *
 559                                               st->codecpar->bits_per_coded_sample / 4;
 560         st->codecpar->block_align           = st->codecpar->channels *
 561                                               st->codecpar->bits_per_coded_sample;
 562         ea->audio_stream_index           = st->index;
 563         st->start_time                   = 0;
 564         return 0;
 565     }
 566 no_audio:
 567     ea->audio_codec = AV_CODEC_ID_NONE;
 568
 569     if (!ea->video.codec)
 570         return AVERROR_INVALIDDATA;
 571     return 0;
 572 }
 573
 574 static int ea_read_packet(AVFormatContext *s, AVPacket *pkt)
 575 {
 576     EaDemuxContext *ea = s->priv_data;
 577     AVIOContext *pb    = s->pb;
 578     int partial_packet = 0;
 579     int hit_end = 0;
 580     unsigned int chunk_type, chunk_size;
 581     int ret = 0, packet_read = 0, key = 0;
 582     int av_uninit(num_samples);
 583
 584     while ((!packet_read && !hit_end) || partial_packet) {
 585         if (avio_feof(pb))
 586             return AVERROR_EOF;
 587         chunk_type = avio_rl32(pb);
 588         chunk_size = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb);
 589         if (chunk_size < 8)
 590             return AVERROR_INVALIDDATA;
 591         chunk_size -= 8;
 592
 593         switch (chunk_type) {
 594         /* audio data */
 595         case ISNh_TAG:
 596             /* header chunk also contains data; skip over the header portion */
 597             if (chunk_size < 32)
 598                 return AVERROR_INVALIDDATA;
 599             avio_skip(pb, 32);
 600             chunk_size -= 32;
 601         case ISNd_TAG:
 602         case SCDl_TAG:
 603         case SNDC_TAG:
 604         case SDEN_TAG:
 605             if (!ea->audio_codec) {
 606                 avio_skip(pb, chunk_size);
 607                 break;
 608             } else if (ea->audio_codec == AV_CODEC_ID_PCM_S16LE_PLANAR ||
 609                        ea->audio_codec == AV_CODEC_ID_MP3) {
 610                 if (chunk_size < 12)
 611                     return AVERROR_INVALIDDATA;
 612                 num_samples = avio_rl32(pb);
 613                 avio_skip(pb, 8);
 614                 chunk_size -= 12;
 615             } else if (ea->audio_codec == AV_CODEC_ID_ADPCM_PSX) {
 616                 if (chunk_size < 8)
 617                     return AVERROR_INVALIDDATA;
 618                 avio_skip(pb, 8);
 619                 chunk_size -= 8;
 620             }
 621
 622             if (partial_packet) {
 623                 avpriv_request_sample(s, "video header followed by audio packet");
 624                 av_packet_unref(pkt);
 625                 partial_packet = 0;
 626             }
 627
 628             if (!chunk_size)
 629                 continue;
 630
 631             ret = av_get_packet(pb, pkt, chunk_size);
 632             if (ret < 0)
 633                 return ret;
 634             pkt->stream_index = ea->audio_stream_index;
 635
 636             switch (ea->audio_codec) {
 637             case AV_CODEC_ID_ADPCM_EA:
 638             case AV_CODEC_ID_ADPCM_EA_R1:
 639             case AV_CODEC_ID_ADPCM_EA_R2:
 640             case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
 641             case AV_CODEC_ID_ADPCM_EA_R3:
 642                 if (pkt->size < 4) {
 643                     av_log(s, AV_LOG_ERROR, "Packet is too short\n");
 644                     return AVERROR_INVALIDDATA;
 645                 }
 646                 if (ea->audio_codec == AV_CODEC_ID_ADPCM_EA_R3)
 647                     pkt->duration = AV_RB32(pkt->data);
 648                 else
 649                     pkt->duration = AV_RL32(pkt->data);
 650                 break;
 651             case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
 652                 pkt->duration = ret * 2 / ea->num_channels;
 653                 break;
 654             case AV_CODEC_ID_PCM_S16LE_PLANAR:
 655             case AV_CODEC_ID_MP3:
 656                 pkt->duration = num_samples;
 657                 break;
 658             case AV_CODEC_ID_ADPCM_PSX:
 659                 pkt->duration = chunk_size / (16 * ea->num_channels) * 28;
 660                 break;
 661             default:
 662                 pkt->duration = chunk_size / (ea->bytes * ea->num_channels);
 663             }
 664
 665             packet_read = 1;
 666             break;
 667
 668         /* ending tag */
 669         case 0:
 670         case ISNe_TAG:
 671         case SCEl_TAG:
 672         case SEND_TAG:
 673         case SEEN_TAG:
 674             while (!avio_feof(pb)) {
 675                 int tag = avio_rl32(pb);
 676
 677                 if (tag == ISNh_TAG ||
 678                     tag == SCHl_TAG ||
 679                     tag == SEAD_TAG ||
 680                     tag == SHEN_TAG) {
 681                     avio_skip(pb, -4);
 682                     break;
 683                 }
 684             }
 685             if (avio_feof(pb))
 686                 ret = AVERROR_EOF;
 687             hit_end = 1;
 688             break;
 689
 690         case MVIh_TAG:
 691         case kVGT_TAG:
 692         case pQGT_TAG:
 693         case TGQs_TAG:
 694         case MADk_TAG:
 695             key = AV_PKT_FLAG_KEY;
 696         case MVIf_TAG:
 697         case fVGT_TAG:
 698         case MADm_TAG:
 699         case MADe_TAG:
 700             if (chunk_size > INT_MAX - 8)
 701                 return AVERROR_INVALIDDATA;
 702             avio_seek(pb, -8, SEEK_CUR);    // include chunk preamble
 703             chunk_size += 8;
 704             goto get_video_packet;
 705
 706         case mTCD_TAG:
 707             if (chunk_size < 8)
 708                 return AVERROR_INVALIDDATA;
 709
 710             avio_skip(pb, 8);               // skip ea DCT header
 711             chunk_size -= 8;
 712             goto get_video_packet;
 713
 714         case MV0K_TAG:
 715         case AV0K_TAG:
 716         case MPCh_TAG:
 717         case pIQT_TAG:
 718             key = AV_PKT_FLAG_KEY;
 719         case MV0F_TAG:
 720         case AV0F_TAG:
 721 get_video_packet:
 722             if (!chunk_size)
 723                 continue;
 724
 725             if (partial_packet) {
 726                 ret = av_append_packet(pb, pkt, chunk_size);
 727             } else
 728                 ret = av_get_packet(pb, pkt, chunk_size);
 729             if (ret < 0) {
 730                 packet_read = 1;
 731                 partial_packet = 0;
 732                 break;
 733             }
 734             partial_packet = chunk_type == MVIh_TAG;
 735             if (chunk_type == AV0K_TAG || chunk_type == AV0F_TAG)
 736                 pkt->stream_index = ea->alpha.stream_index;
 737             else
 738                 pkt->stream_index = ea->video.stream_index;
 739             pkt->flags       |= key;
 740             packet_read       = 1;
 741             break;
 742
 743         default:
 744             avio_skip(pb, chunk_size);
 745             break;
 746         }
 747     }
 748
 749     if (ret >= 0 && hit_end && !packet_read)
 750         return AVERROR(EAGAIN);
 751
 752     return ret;
 753 }
 754
 755 const AVInputFormat ff_ea_demuxer = {
 756     .name           = "ea",
 757     .long_name      = NULL_IF_CONFIG_SMALL("Electronic Arts Multimedia"),
 758     .priv_data_size = sizeof(EaDemuxContext),
 759     .read_probe     = ea_probe,
 760     .read_header    = ea_read_header,
 761     .read_packet    = ea_read_packet,
 762 };