ATSParser.cpp revision 29357bc2c0dd7c43ad3bd0c8e3efa4e6fd9bfd47
1/* 2 * Copyright (C) 2010 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17//#define LOG_NDEBUG 0 18#define LOG_TAG "ATSParser" 19#include <utils/Log.h> 20 21#include "ATSParser.h" 22 23#include "AnotherPacketSource.h" 24#include "ESQueue.h" 25#include "include/avc_utils.h" 26 27#include <media/stagefright/foundation/ABitReader.h> 28#include <media/stagefright/foundation/ABuffer.h> 29#include <media/stagefright/foundation/ADebug.h> 30#include <media/stagefright/foundation/AMessage.h> 31#include <media/stagefright/foundation/hexdump.h> 32#include <media/stagefright/MediaDefs.h> 33#include <media/stagefright/MediaErrors.h> 34#include <media/stagefright/MetaData.h> 35#include <media/IStreamSource.h> 36#include <utils/KeyedVector.h> 37 38namespace android { 39 40// I want the expression "y" evaluated even if verbose logging is off. 41#define MY_LOGV(x, y) \ 42 do { unsigned tmp = y; ALOGV(x, tmp); } while (0) 43 44static const size_t kTSPacketSize = 188; 45 46struct ATSParser::Program : public RefBase { 47 Program(ATSParser *parser, unsigned programNumber, unsigned programMapPID); 48 49 bool parsePID( 50 unsigned pid, unsigned payload_unit_start_indicator, 51 ABitReader *br, status_t *err); 52 53 void signalDiscontinuity( 54 DiscontinuityType type, const sp<AMessage> &extra); 55 56 void signalEOS(status_t finalResult); 57 58 sp<MediaSource> getSource(SourceType type); 59 60 int64_t convertPTSToTimestamp(uint64_t PTS); 61 62 bool PTSTimeDeltaEstablished() const { 63 return mFirstPTSValid; 64 } 65 66 unsigned number() const { return mProgramNumber; } 67 68 void updateProgramMapPID(unsigned programMapPID) { 69 mProgramMapPID = programMapPID; 70 } 71 72private: 73 ATSParser *mParser; 74 unsigned mProgramNumber; 75 unsigned mProgramMapPID; 76 KeyedVector<unsigned, sp<Stream> > mStreams; 77 bool mFirstPTSValid; 78 uint64_t mFirstPTS; 79 80 status_t parseProgramMap(ABitReader *br); 81 82 DISALLOW_EVIL_CONSTRUCTORS(Program); 83}; 84 85struct ATSParser::Stream : public RefBase { 86 Stream(Program *program, unsigned elementaryPID, unsigned streamType); 87 88 unsigned type() const { return mStreamType; } 89 unsigned pid() const { return mElementaryPID; } 90 void setPID(unsigned pid) { mElementaryPID = pid; } 91 92 status_t parse( 93 unsigned payload_unit_start_indicator, 94 ABitReader *br); 95 96 void signalDiscontinuity( 97 DiscontinuityType type, const sp<AMessage> &extra); 98 99 void signalEOS(status_t finalResult); 100 101 sp<MediaSource> getSource(SourceType type); 102 103protected: 104 virtual ~Stream(); 105 106private: 107 Program *mProgram; 108 unsigned mElementaryPID; 109 unsigned mStreamType; 110 111 sp<ABuffer> mBuffer; 112 sp<AnotherPacketSource> mSource; 113 bool mPayloadStarted; 114 115 ElementaryStreamQueue *mQueue; 116 117 status_t flush(); 118 status_t parsePES(ABitReader *br); 119 120 void onPayloadData( 121 unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 122 const uint8_t *data, size_t size); 123 124 void extractAACFrames(const sp<ABuffer> &buffer); 125 126 bool isAudio() const; 127 bool isVideo() const; 128 129 DISALLOW_EVIL_CONSTRUCTORS(Stream); 130}; 131 132//////////////////////////////////////////////////////////////////////////////// 133 134ATSParser::Program::Program( 135 ATSParser *parser, unsigned programNumber, unsigned programMapPID) 136 : mParser(parser), 137 mProgramNumber(programNumber), 138 mProgramMapPID(programMapPID), 139 mFirstPTSValid(false), 140 mFirstPTS(0) { 141 ALOGV("new program number %u", programNumber); 142} 143 144bool ATSParser::Program::parsePID( 145 unsigned pid, unsigned payload_unit_start_indicator, 146 ABitReader *br, status_t *err) { 147 *err = OK; 148 149 if (pid == mProgramMapPID) { 150 if (payload_unit_start_indicator) { 151 unsigned skip = br->getBits(8); 152 br->skipBits(skip * 8); 153 } 154 155 *err = parseProgramMap(br); 156 157 return true; 158 } 159 160 ssize_t index = mStreams.indexOfKey(pid); 161 if (index < 0) { 162 return false; 163 } 164 165 *err = mStreams.editValueAt(index)->parse( 166 payload_unit_start_indicator, br); 167 168 return true; 169} 170 171void ATSParser::Program::signalDiscontinuity( 172 DiscontinuityType type, const sp<AMessage> &extra) { 173 for (size_t i = 0; i < mStreams.size(); ++i) { 174 mStreams.editValueAt(i)->signalDiscontinuity(type, extra); 175 } 176} 177 178void ATSParser::Program::signalEOS(status_t finalResult) { 179 for (size_t i = 0; i < mStreams.size(); ++i) { 180 mStreams.editValueAt(i)->signalEOS(finalResult); 181 } 182} 183 184struct StreamInfo { 185 unsigned mType; 186 unsigned mPID; 187}; 188 189status_t ATSParser::Program::parseProgramMap(ABitReader *br) { 190 unsigned table_id = br->getBits(8); 191 ALOGV(" table_id = %u", table_id); 192 CHECK_EQ(table_id, 0x02u); 193 194 unsigned section_syntax_indicator = br->getBits(1); 195 ALOGV(" section_syntax_indicator = %u", section_syntax_indicator); 196 CHECK_EQ(section_syntax_indicator, 1u); 197 198 CHECK_EQ(br->getBits(1), 0u); 199 MY_LOGV(" reserved = %u", br->getBits(2)); 200 201 unsigned section_length = br->getBits(12); 202 ALOGV(" section_length = %u", section_length); 203 CHECK_EQ(section_length & 0xc00, 0u); 204 CHECK_LE(section_length, 1021u); 205 206 MY_LOGV(" program_number = %u", br->getBits(16)); 207 MY_LOGV(" reserved = %u", br->getBits(2)); 208 MY_LOGV(" version_number = %u", br->getBits(5)); 209 MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 210 MY_LOGV(" section_number = %u", br->getBits(8)); 211 MY_LOGV(" last_section_number = %u", br->getBits(8)); 212 MY_LOGV(" reserved = %u", br->getBits(3)); 213 MY_LOGV(" PCR_PID = 0x%04x", br->getBits(13)); 214 MY_LOGV(" reserved = %u", br->getBits(4)); 215 216 unsigned program_info_length = br->getBits(12); 217 ALOGV(" program_info_length = %u", program_info_length); 218 CHECK_EQ(program_info_length & 0xc00, 0u); 219 220 br->skipBits(program_info_length * 8); // skip descriptors 221 222 Vector<StreamInfo> infos; 223 224 // infoBytesRemaining is the number of bytes that make up the 225 // variable length section of ES_infos. It does not include the 226 // final CRC. 227 size_t infoBytesRemaining = section_length - 9 - program_info_length - 4; 228 229 while (infoBytesRemaining > 0) { 230 CHECK_GE(infoBytesRemaining, 5u); 231 232 unsigned streamType = br->getBits(8); 233 ALOGV(" stream_type = 0x%02x", streamType); 234 235 MY_LOGV(" reserved = %u", br->getBits(3)); 236 237 unsigned elementaryPID = br->getBits(13); 238 ALOGV(" elementary_PID = 0x%04x", elementaryPID); 239 240 MY_LOGV(" reserved = %u", br->getBits(4)); 241 242 unsigned ES_info_length = br->getBits(12); 243 ALOGV(" ES_info_length = %u", ES_info_length); 244 CHECK_EQ(ES_info_length & 0xc00, 0u); 245 246 CHECK_GE(infoBytesRemaining - 5, ES_info_length); 247 248#if 0 249 br->skipBits(ES_info_length * 8); // skip descriptors 250#else 251 unsigned info_bytes_remaining = ES_info_length; 252 while (info_bytes_remaining >= 2) { 253 MY_LOGV(" tag = 0x%02x", br->getBits(8)); 254 255 unsigned descLength = br->getBits(8); 256 ALOGV(" len = %u", descLength); 257 258 CHECK_GE(info_bytes_remaining, 2 + descLength); 259 260 br->skipBits(descLength * 8); 261 262 info_bytes_remaining -= descLength + 2; 263 } 264 CHECK_EQ(info_bytes_remaining, 0u); 265#endif 266 267 StreamInfo info; 268 info.mType = streamType; 269 info.mPID = elementaryPID; 270 infos.push(info); 271 272 infoBytesRemaining -= 5 + ES_info_length; 273 } 274 275 CHECK_EQ(infoBytesRemaining, 0u); 276 MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 277 278 bool PIDsChanged = false; 279 for (size_t i = 0; i < infos.size(); ++i) { 280 StreamInfo &info = infos.editItemAt(i); 281 282 ssize_t index = mStreams.indexOfKey(info.mPID); 283 284 if (index >= 0 && mStreams.editValueAt(index)->type() != info.mType) { 285 ALOGI("uh oh. stream PIDs have changed."); 286 PIDsChanged = true; 287 break; 288 } 289 } 290 291 if (PIDsChanged) { 292#if 0 293 ALOGI("before:"); 294 for (size_t i = 0; i < mStreams.size(); ++i) { 295 sp<Stream> stream = mStreams.editValueAt(i); 296 297 ALOGI("PID 0x%08x => type 0x%02x", stream->pid(), stream->type()); 298 } 299 300 ALOGI("after:"); 301 for (size_t i = 0; i < infos.size(); ++i) { 302 StreamInfo &info = infos.editItemAt(i); 303 304 ALOGI("PID 0x%08x => type 0x%02x", info.mPID, info.mType); 305 } 306#endif 307 308 // The only case we can recover from is if we have two streams 309 // and they switched PIDs. 310 311 bool success = false; 312 313 if (mStreams.size() == 2 && infos.size() == 2) { 314 const StreamInfo &info1 = infos.itemAt(0); 315 const StreamInfo &info2 = infos.itemAt(1); 316 317 sp<Stream> s1 = mStreams.editValueAt(0); 318 sp<Stream> s2 = mStreams.editValueAt(1); 319 320 bool caseA = 321 info1.mPID == s1->pid() && info1.mType == s2->type() 322 && info2.mPID == s2->pid() && info2.mType == s1->type(); 323 324 bool caseB = 325 info1.mPID == s2->pid() && info1.mType == s1->type() 326 && info2.mPID == s1->pid() && info2.mType == s2->type(); 327 328 if (caseA || caseB) { 329 unsigned pid1 = s1->pid(); 330 unsigned pid2 = s2->pid(); 331 s1->setPID(pid2); 332 s2->setPID(pid1); 333 334 mStreams.clear(); 335 mStreams.add(s1->pid(), s1); 336 mStreams.add(s2->pid(), s2); 337 338 success = true; 339 } 340 } 341 342 if (!success) { 343 ALOGI("Stream PIDs changed and we cannot recover."); 344 return ERROR_MALFORMED; 345 } 346 } 347 348 for (size_t i = 0; i < infos.size(); ++i) { 349 StreamInfo &info = infos.editItemAt(i); 350 351 ssize_t index = mStreams.indexOfKey(info.mPID); 352 353 if (index < 0) { 354 sp<Stream> stream = new Stream(this, info.mPID, info.mType); 355 mStreams.add(info.mPID, stream); 356 } 357 } 358 359 return OK; 360} 361 362sp<MediaSource> ATSParser::Program::getSource(SourceType type) { 363 size_t index = (type == AUDIO) ? 0 : 0; 364 365 for (size_t i = 0; i < mStreams.size(); ++i) { 366 sp<MediaSource> source = mStreams.editValueAt(i)->getSource(type); 367 if (source != NULL) { 368 if (index == 0) { 369 return source; 370 } 371 --index; 372 } 373 } 374 375 return NULL; 376} 377 378int64_t ATSParser::Program::convertPTSToTimestamp(uint64_t PTS) { 379 if (!(mParser->mFlags & TS_TIMESTAMPS_ARE_ABSOLUTE)) { 380 if (!mFirstPTSValid) { 381 mFirstPTSValid = true; 382 mFirstPTS = PTS; 383 PTS = 0; 384 } else if (PTS < mFirstPTS) { 385 PTS = 0; 386 } else { 387 PTS -= mFirstPTS; 388 } 389 } 390 391 return (PTS * 100) / 9; 392} 393 394//////////////////////////////////////////////////////////////////////////////// 395 396ATSParser::Stream::Stream( 397 Program *program, unsigned elementaryPID, unsigned streamType) 398 : mProgram(program), 399 mElementaryPID(elementaryPID), 400 mStreamType(streamType), 401 mPayloadStarted(false), 402 mQueue(NULL) { 403 switch (mStreamType) { 404 case STREAMTYPE_H264: 405 mQueue = new ElementaryStreamQueue(ElementaryStreamQueue::H264); 406 break; 407 case STREAMTYPE_MPEG2_AUDIO_ADTS: 408 mQueue = new ElementaryStreamQueue(ElementaryStreamQueue::AAC); 409 break; 410 case STREAMTYPE_MPEG1_AUDIO: 411 case STREAMTYPE_MPEG2_AUDIO: 412 mQueue = new ElementaryStreamQueue( 413 ElementaryStreamQueue::MPEG_AUDIO); 414 break; 415 416 case STREAMTYPE_MPEG1_VIDEO: 417 case STREAMTYPE_MPEG2_VIDEO: 418 mQueue = new ElementaryStreamQueue( 419 ElementaryStreamQueue::MPEG_VIDEO); 420 break; 421 422 case STREAMTYPE_MPEG4_VIDEO: 423 mQueue = new ElementaryStreamQueue( 424 ElementaryStreamQueue::MPEG4_VIDEO); 425 break; 426 427 default: 428 break; 429 } 430 431 ALOGV("new stream PID 0x%02x, type 0x%02x", elementaryPID, streamType); 432 433 if (mQueue != NULL) { 434 mBuffer = new ABuffer(192 * 1024); 435 mBuffer->setRange(0, 0); 436 } 437} 438 439ATSParser::Stream::~Stream() { 440 delete mQueue; 441 mQueue = NULL; 442} 443 444status_t ATSParser::Stream::parse( 445 unsigned payload_unit_start_indicator, ABitReader *br) { 446 if (mQueue == NULL) { 447 return OK; 448 } 449 450 if (payload_unit_start_indicator) { 451 if (mPayloadStarted) { 452 // Otherwise we run the danger of receiving the trailing bytes 453 // of a PES packet that we never saw the start of and assuming 454 // we have a a complete PES packet. 455 456 status_t err = flush(); 457 458 if (err != OK) { 459 return err; 460 } 461 } 462 463 mPayloadStarted = true; 464 } 465 466 if (!mPayloadStarted) { 467 return OK; 468 } 469 470 size_t payloadSizeBits = br->numBitsLeft(); 471 CHECK_EQ(payloadSizeBits % 8, 0u); 472 473 size_t neededSize = mBuffer->size() + payloadSizeBits / 8; 474 if (mBuffer->capacity() < neededSize) { 475 // Increment in multiples of 64K. 476 neededSize = (neededSize + 65535) & ~65535; 477 478 ALOGI("resizing buffer to %d bytes", neededSize); 479 480 sp<ABuffer> newBuffer = new ABuffer(neededSize); 481 memcpy(newBuffer->data(), mBuffer->data(), mBuffer->size()); 482 newBuffer->setRange(0, mBuffer->size()); 483 mBuffer = newBuffer; 484 } 485 486 memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8); 487 mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8); 488 489 return OK; 490} 491 492bool ATSParser::Stream::isVideo() const { 493 switch (mStreamType) { 494 case STREAMTYPE_H264: 495 case STREAMTYPE_MPEG1_VIDEO: 496 case STREAMTYPE_MPEG2_VIDEO: 497 case STREAMTYPE_MPEG4_VIDEO: 498 return true; 499 500 default: 501 return false; 502 } 503} 504 505bool ATSParser::Stream::isAudio() const { 506 switch (mStreamType) { 507 case STREAMTYPE_MPEG1_AUDIO: 508 case STREAMTYPE_MPEG2_AUDIO: 509 case STREAMTYPE_MPEG2_AUDIO_ADTS: 510 return true; 511 512 default: 513 return false; 514 } 515} 516 517void ATSParser::Stream::signalDiscontinuity( 518 DiscontinuityType type, const sp<AMessage> &extra) { 519 if (mQueue == NULL) { 520 return; 521 } 522 523 mPayloadStarted = false; 524 mBuffer->setRange(0, 0); 525 526 bool clearFormat = false; 527 if (isAudio()) { 528 if (type & DISCONTINUITY_AUDIO_FORMAT) { 529 clearFormat = true; 530 } 531 } else { 532 if (type & DISCONTINUITY_VIDEO_FORMAT) { 533 clearFormat = true; 534 } 535 } 536 537 mQueue->clear(clearFormat); 538 539 if (type & DISCONTINUITY_TIME) { 540 uint64_t resumeAtPTS; 541 if (extra != NULL 542 && extra->findInt64( 543 IStreamListener::kKeyResumeAtPTS, 544 (int64_t *)&resumeAtPTS)) { 545 int64_t resumeAtMediaTimeUs = 546 mProgram->convertPTSToTimestamp(resumeAtPTS); 547 548 extra->setInt64("resume-at-mediatimeUs", resumeAtMediaTimeUs); 549 } 550 } 551 552 if (mSource != NULL) { 553 mSource->queueDiscontinuity(type, extra); 554 } 555} 556 557void ATSParser::Stream::signalEOS(status_t finalResult) { 558 if (mSource != NULL) { 559 mSource->signalEOS(finalResult); 560 } 561} 562 563status_t ATSParser::Stream::parsePES(ABitReader *br) { 564 unsigned packet_startcode_prefix = br->getBits(24); 565 566 ALOGV("packet_startcode_prefix = 0x%08x", packet_startcode_prefix); 567 568 if (packet_startcode_prefix != 1) { 569 ALOGV("Supposedly payload_unit_start=1 unit does not start " 570 "with startcode."); 571 572 return ERROR_MALFORMED; 573 } 574 575 CHECK_EQ(packet_startcode_prefix, 0x000001u); 576 577 unsigned stream_id = br->getBits(8); 578 ALOGV("stream_id = 0x%02x", stream_id); 579 580 unsigned PES_packet_length = br->getBits(16); 581 ALOGV("PES_packet_length = %u", PES_packet_length); 582 583 if (stream_id != 0xbc // program_stream_map 584 && stream_id != 0xbe // padding_stream 585 && stream_id != 0xbf // private_stream_2 586 && stream_id != 0xf0 // ECM 587 && stream_id != 0xf1 // EMM 588 && stream_id != 0xff // program_stream_directory 589 && stream_id != 0xf2 // DSMCC 590 && stream_id != 0xf8) { // H.222.1 type E 591 CHECK_EQ(br->getBits(2), 2u); 592 593 MY_LOGV("PES_scrambling_control = %u", br->getBits(2)); 594 MY_LOGV("PES_priority = %u", br->getBits(1)); 595 MY_LOGV("data_alignment_indicator = %u", br->getBits(1)); 596 MY_LOGV("copyright = %u", br->getBits(1)); 597 MY_LOGV("original_or_copy = %u", br->getBits(1)); 598 599 unsigned PTS_DTS_flags = br->getBits(2); 600 ALOGV("PTS_DTS_flags = %u", PTS_DTS_flags); 601 602 unsigned ESCR_flag = br->getBits(1); 603 ALOGV("ESCR_flag = %u", ESCR_flag); 604 605 unsigned ES_rate_flag = br->getBits(1); 606 ALOGV("ES_rate_flag = %u", ES_rate_flag); 607 608 unsigned DSM_trick_mode_flag = br->getBits(1); 609 ALOGV("DSM_trick_mode_flag = %u", DSM_trick_mode_flag); 610 611 unsigned additional_copy_info_flag = br->getBits(1); 612 ALOGV("additional_copy_info_flag = %u", additional_copy_info_flag); 613 614 MY_LOGV("PES_CRC_flag = %u", br->getBits(1)); 615 MY_LOGV("PES_extension_flag = %u", br->getBits(1)); 616 617 unsigned PES_header_data_length = br->getBits(8); 618 ALOGV("PES_header_data_length = %u", PES_header_data_length); 619 620 unsigned optional_bytes_remaining = PES_header_data_length; 621 622 uint64_t PTS = 0, DTS = 0; 623 624 if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) { 625 CHECK_GE(optional_bytes_remaining, 5u); 626 627 CHECK_EQ(br->getBits(4), PTS_DTS_flags); 628 629 PTS = ((uint64_t)br->getBits(3)) << 30; 630 CHECK_EQ(br->getBits(1), 1u); 631 PTS |= ((uint64_t)br->getBits(15)) << 15; 632 CHECK_EQ(br->getBits(1), 1u); 633 PTS |= br->getBits(15); 634 CHECK_EQ(br->getBits(1), 1u); 635 636 ALOGV("PTS = %llu", PTS); 637 // ALOGI("PTS = %.2f secs", PTS / 90000.0f); 638 639 optional_bytes_remaining -= 5; 640 641 if (PTS_DTS_flags == 3) { 642 CHECK_GE(optional_bytes_remaining, 5u); 643 644 CHECK_EQ(br->getBits(4), 1u); 645 646 DTS = ((uint64_t)br->getBits(3)) << 30; 647 CHECK_EQ(br->getBits(1), 1u); 648 DTS |= ((uint64_t)br->getBits(15)) << 15; 649 CHECK_EQ(br->getBits(1), 1u); 650 DTS |= br->getBits(15); 651 CHECK_EQ(br->getBits(1), 1u); 652 653 ALOGV("DTS = %llu", DTS); 654 655 optional_bytes_remaining -= 5; 656 } 657 } 658 659 if (ESCR_flag) { 660 CHECK_GE(optional_bytes_remaining, 6u); 661 662 br->getBits(2); 663 664 uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30; 665 CHECK_EQ(br->getBits(1), 1u); 666 ESCR |= ((uint64_t)br->getBits(15)) << 15; 667 CHECK_EQ(br->getBits(1), 1u); 668 ESCR |= br->getBits(15); 669 CHECK_EQ(br->getBits(1), 1u); 670 671 ALOGV("ESCR = %llu", ESCR); 672 MY_LOGV("ESCR_extension = %u", br->getBits(9)); 673 674 CHECK_EQ(br->getBits(1), 1u); 675 676 optional_bytes_remaining -= 6; 677 } 678 679 if (ES_rate_flag) { 680 CHECK_GE(optional_bytes_remaining, 3u); 681 682 CHECK_EQ(br->getBits(1), 1u); 683 MY_LOGV("ES_rate = %u", br->getBits(22)); 684 CHECK_EQ(br->getBits(1), 1u); 685 686 optional_bytes_remaining -= 3; 687 } 688 689 br->skipBits(optional_bytes_remaining * 8); 690 691 // ES data follows. 692 693 if (PES_packet_length != 0) { 694 CHECK_GE(PES_packet_length, PES_header_data_length + 3); 695 696 unsigned dataLength = 697 PES_packet_length - 3 - PES_header_data_length; 698 699 if (br->numBitsLeft() < dataLength * 8) { 700 ALOGE("PES packet does not carry enough data to contain " 701 "payload. (numBitsLeft = %d, required = %d)", 702 br->numBitsLeft(), dataLength * 8); 703 704 return ERROR_MALFORMED; 705 } 706 707 CHECK_GE(br->numBitsLeft(), dataLength * 8); 708 709 onPayloadData( 710 PTS_DTS_flags, PTS, DTS, br->data(), dataLength); 711 712 br->skipBits(dataLength * 8); 713 } else { 714 onPayloadData( 715 PTS_DTS_flags, PTS, DTS, 716 br->data(), br->numBitsLeft() / 8); 717 718 size_t payloadSizeBits = br->numBitsLeft(); 719 CHECK_EQ(payloadSizeBits % 8, 0u); 720 721 ALOGV("There's %d bytes of payload.", payloadSizeBits / 8); 722 } 723 } else if (stream_id == 0xbe) { // padding_stream 724 CHECK_NE(PES_packet_length, 0u); 725 br->skipBits(PES_packet_length * 8); 726 } else { 727 CHECK_NE(PES_packet_length, 0u); 728 br->skipBits(PES_packet_length * 8); 729 } 730 731 return OK; 732} 733 734status_t ATSParser::Stream::flush() { 735 if (mBuffer->size() == 0) { 736 return OK; 737 } 738 739 ALOGV("flushing stream 0x%04x size = %d", mElementaryPID, mBuffer->size()); 740 741 ABitReader br(mBuffer->data(), mBuffer->size()); 742 743 status_t err = parsePES(&br); 744 745 mBuffer->setRange(0, 0); 746 747 return err; 748} 749 750void ATSParser::Stream::onPayloadData( 751 unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 752 const uint8_t *data, size_t size) { 753 ALOGV("onPayloadData mStreamType=0x%02x", mStreamType); 754 755 int64_t timeUs = 0ll; // no presentation timestamp available. 756 if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) { 757 timeUs = mProgram->convertPTSToTimestamp(PTS); 758 } 759 760 status_t err = mQueue->appendData(data, size, timeUs); 761 762 if (err != OK) { 763 return; 764 } 765 766 sp<ABuffer> accessUnit; 767 while ((accessUnit = mQueue->dequeueAccessUnit()) != NULL) { 768 if (mSource == NULL) { 769 sp<MetaData> meta = mQueue->getFormat(); 770 771 if (meta != NULL) { 772 ALOGV("Stream PID 0x%08x of type 0x%02x now has data.", 773 mElementaryPID, mStreamType); 774 775 mSource = new AnotherPacketSource(meta); 776 mSource->queueAccessUnit(accessUnit); 777 } 778 } else if (mQueue->getFormat() != NULL) { 779 // After a discontinuity we invalidate the queue's format 780 // and won't enqueue any access units to the source until 781 // the queue has reestablished the new format. 782 783 if (mSource->getFormat() == NULL) { 784 mSource->setFormat(mQueue->getFormat()); 785 } 786 mSource->queueAccessUnit(accessUnit); 787 } 788 } 789} 790 791sp<MediaSource> ATSParser::Stream::getSource(SourceType type) { 792 switch (type) { 793 case VIDEO: 794 { 795 if (isVideo()) { 796 return mSource; 797 } 798 break; 799 } 800 801 case AUDIO: 802 { 803 if (isAudio()) { 804 return mSource; 805 } 806 break; 807 } 808 809 default: 810 break; 811 } 812 813 return NULL; 814} 815 816//////////////////////////////////////////////////////////////////////////////// 817 818ATSParser::ATSParser(uint32_t flags) 819 : mFlags(flags) { 820} 821 822ATSParser::~ATSParser() { 823} 824 825status_t ATSParser::feedTSPacket(const void *data, size_t size) { 826 CHECK_EQ(size, kTSPacketSize); 827 828 ABitReader br((const uint8_t *)data, kTSPacketSize); 829 return parseTS(&br); 830} 831 832void ATSParser::signalDiscontinuity( 833 DiscontinuityType type, const sp<AMessage> &extra) { 834 for (size_t i = 0; i < mPrograms.size(); ++i) { 835 mPrograms.editItemAt(i)->signalDiscontinuity(type, extra); 836 } 837} 838 839void ATSParser::signalEOS(status_t finalResult) { 840 CHECK_NE(finalResult, (status_t)OK); 841 842 for (size_t i = 0; i < mPrograms.size(); ++i) { 843 mPrograms.editItemAt(i)->signalEOS(finalResult); 844 } 845} 846 847void ATSParser::parseProgramAssociationTable(ABitReader *br) { 848 unsigned table_id = br->getBits(8); 849 ALOGV(" table_id = %u", table_id); 850 CHECK_EQ(table_id, 0x00u); 851 852 unsigned section_syntax_indictor = br->getBits(1); 853 ALOGV(" section_syntax_indictor = %u", section_syntax_indictor); 854 CHECK_EQ(section_syntax_indictor, 1u); 855 856 CHECK_EQ(br->getBits(1), 0u); 857 MY_LOGV(" reserved = %u", br->getBits(2)); 858 859 unsigned section_length = br->getBits(12); 860 ALOGV(" section_length = %u", section_length); 861 CHECK_EQ(section_length & 0xc00, 0u); 862 863 MY_LOGV(" transport_stream_id = %u", br->getBits(16)); 864 MY_LOGV(" reserved = %u", br->getBits(2)); 865 MY_LOGV(" version_number = %u", br->getBits(5)); 866 MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 867 MY_LOGV(" section_number = %u", br->getBits(8)); 868 MY_LOGV(" last_section_number = %u", br->getBits(8)); 869 870 size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */); 871 CHECK_EQ((numProgramBytes % 4), 0u); 872 873 for (size_t i = 0; i < numProgramBytes / 4; ++i) { 874 unsigned program_number = br->getBits(16); 875 ALOGV(" program_number = %u", program_number); 876 877 MY_LOGV(" reserved = %u", br->getBits(3)); 878 879 if (program_number == 0) { 880 MY_LOGV(" network_PID = 0x%04x", br->getBits(13)); 881 } else { 882 unsigned programMapPID = br->getBits(13); 883 884 ALOGV(" program_map_PID = 0x%04x", programMapPID); 885 886 bool found = false; 887 for (size_t index = 0; index < mPrograms.size(); ++index) { 888 const sp<Program> &program = mPrograms.itemAt(index); 889 890 if (program->number() == program_number) { 891 program->updateProgramMapPID(programMapPID); 892 found = true; 893 break; 894 } 895 } 896 897 if (!found) { 898 mPrograms.push( 899 new Program(this, program_number, programMapPID)); 900 } 901 } 902 } 903 904 MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 905} 906 907status_t ATSParser::parsePID( 908 ABitReader *br, unsigned PID, 909 unsigned payload_unit_start_indicator) { 910 if (PID == 0) { 911 if (payload_unit_start_indicator) { 912 unsigned skip = br->getBits(8); 913 br->skipBits(skip * 8); 914 } 915 parseProgramAssociationTable(br); 916 return OK; 917 } 918 919 bool handled = false; 920 for (size_t i = 0; i < mPrograms.size(); ++i) { 921 status_t err; 922 if (mPrograms.editItemAt(i)->parsePID( 923 PID, payload_unit_start_indicator, br, &err)) { 924 if (err != OK) { 925 return err; 926 } 927 928 handled = true; 929 break; 930 } 931 } 932 933 if (!handled) { 934 ALOGV("PID 0x%04x not handled.", PID); 935 } 936 937 return OK; 938} 939 940void ATSParser::parseAdaptationField(ABitReader *br) { 941 unsigned adaptation_field_length = br->getBits(8); 942 if (adaptation_field_length > 0) { 943 br->skipBits(adaptation_field_length * 8); // XXX 944 } 945} 946 947status_t ATSParser::parseTS(ABitReader *br) { 948 ALOGV("---"); 949 950 unsigned sync_byte = br->getBits(8); 951 CHECK_EQ(sync_byte, 0x47u); 952 953 MY_LOGV("transport_error_indicator = %u", br->getBits(1)); 954 955 unsigned payload_unit_start_indicator = br->getBits(1); 956 ALOGV("payload_unit_start_indicator = %u", payload_unit_start_indicator); 957 958 MY_LOGV("transport_priority = %u", br->getBits(1)); 959 960 unsigned PID = br->getBits(13); 961 ALOGV("PID = 0x%04x", PID); 962 963 MY_LOGV("transport_scrambling_control = %u", br->getBits(2)); 964 965 unsigned adaptation_field_control = br->getBits(2); 966 ALOGV("adaptation_field_control = %u", adaptation_field_control); 967 968 unsigned continuity_counter = br->getBits(4); 969 ALOGV("continuity_counter = %u", continuity_counter); 970 971 // ALOGI("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter); 972 973 if (adaptation_field_control == 2 || adaptation_field_control == 3) { 974 parseAdaptationField(br); 975 } 976 977 if (adaptation_field_control == 1 || adaptation_field_control == 3) { 978 return parsePID(br, PID, payload_unit_start_indicator); 979 } 980 981 return OK; 982} 983 984sp<MediaSource> ATSParser::getSource(SourceType type) { 985 int which = -1; // any 986 987 for (size_t i = 0; i < mPrograms.size(); ++i) { 988 const sp<Program> &program = mPrograms.editItemAt(i); 989 990 if (which >= 0 && (int)program->number() != which) { 991 continue; 992 } 993 994 sp<MediaSource> source = program->getSource(type); 995 996 if (source != NULL) { 997 return source; 998 } 999 } 1000 1001 return NULL; 1002} 1003 1004bool ATSParser::PTSTimeDeltaEstablished() { 1005 if (mPrograms.isEmpty()) { 1006 return false; 1007 } 1008 1009 return mPrograms.editItemAt(0)->PTSTimeDeltaEstablished(); 1010} 1011 1012} // namespace android 1013