1233d2500723e5594f3e7c70896ffeeef32b9c950ywan/* 2233d2500723e5594f3e7c70896ffeeef32b9c950ywan * Copyright © 2010 Mozilla Foundation 3233d2500723e5594f3e7c70896ffeeef32b9c950ywan * 4233d2500723e5594f3e7c70896ffeeef32b9c950ywan * This program is made available under an ISC-style license. See the 5233d2500723e5594f3e7c70896ffeeef32b9c950ywan * accompanying file LICENSE for details. 6233d2500723e5594f3e7c70896ffeeef32b9c950ywan */ 7233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <assert.h> 8233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <stdarg.h> 9233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <stdio.h> 10233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <stdlib.h> 11233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <stdint.h> 12233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "nestegg/nestegg.h" 13233d2500723e5594f3e7c70896ffeeef32b9c950ywan 14233d2500723e5594f3e7c70896ffeeef32b9c950ywan#undef DEBUG 15233d2500723e5594f3e7c70896ffeeef32b9c950ywan#define SEEK_TEST 16233d2500723e5594f3e7c70896ffeeef32b9c950ywan 17233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic int 18233d2500723e5594f3e7c70896ffeeef32b9c950ywanstdio_read(void * p, size_t length, void * file) 19233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 20233d2500723e5594f3e7c70896ffeeef32b9c950ywan size_t r; 21233d2500723e5594f3e7c70896ffeeef32b9c950ywan FILE * fp = file; 22233d2500723e5594f3e7c70896ffeeef32b9c950ywan 23233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = fread(p, length, 1, fp); 24233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 0 && feof(fp)) 25233d2500723e5594f3e7c70896ffeeef32b9c950ywan return 0; 26233d2500723e5594f3e7c70896ffeeef32b9c950ywan return r == 0 ? -1 : 1; 27233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 28233d2500723e5594f3e7c70896ffeeef32b9c950ywan 29233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic int 30233d2500723e5594f3e7c70896ffeeef32b9c950ywanstdio_seek(int64_t offset, int whence, void * file) 31233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 32233d2500723e5594f3e7c70896ffeeef32b9c950ywan FILE * fp = file; 33233d2500723e5594f3e7c70896ffeeef32b9c950ywan return fseek(fp, offset, whence); 34233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 35233d2500723e5594f3e7c70896ffeeef32b9c950ywan 36233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic int64_t 37233d2500723e5594f3e7c70896ffeeef32b9c950ywanstdio_tell(void * fp) 38233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 39233d2500723e5594f3e7c70896ffeeef32b9c950ywan return ftell(fp); 40233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 41233d2500723e5594f3e7c70896ffeeef32b9c950ywan 42233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic void 43233d2500723e5594f3e7c70896ffeeef32b9c950ywanlog_callback(nestegg * ctx, unsigned int severity, char const * fmt, ...) 44233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 45233d2500723e5594f3e7c70896ffeeef32b9c950ywan va_list ap; 46233d2500723e5594f3e7c70896ffeeef32b9c950ywan char const * sev = NULL; 47233d2500723e5594f3e7c70896ffeeef32b9c950ywan 48233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if !defined(DEBUG) 49233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (severity < NESTEGG_LOG_WARNING) 50233d2500723e5594f3e7c70896ffeeef32b9c950ywan return; 51233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 52233d2500723e5594f3e7c70896ffeeef32b9c950ywan 53233d2500723e5594f3e7c70896ffeeef32b9c950ywan switch (severity) { 54233d2500723e5594f3e7c70896ffeeef32b9c950ywan case NESTEGG_LOG_DEBUG: 55233d2500723e5594f3e7c70896ffeeef32b9c950ywan sev = "debug: "; 56233d2500723e5594f3e7c70896ffeeef32b9c950ywan break; 57233d2500723e5594f3e7c70896ffeeef32b9c950ywan case NESTEGG_LOG_WARNING: 58233d2500723e5594f3e7c70896ffeeef32b9c950ywan sev = "warning: "; 59233d2500723e5594f3e7c70896ffeeef32b9c950ywan break; 60233d2500723e5594f3e7c70896ffeeef32b9c950ywan case NESTEGG_LOG_CRITICAL: 61233d2500723e5594f3e7c70896ffeeef32b9c950ywan sev = "critical:"; 62233d2500723e5594f3e7c70896ffeeef32b9c950ywan break; 63233d2500723e5594f3e7c70896ffeeef32b9c950ywan default: 64233d2500723e5594f3e7c70896ffeeef32b9c950ywan sev = "unknown: "; 65233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 66233d2500723e5594f3e7c70896ffeeef32b9c950ywan 67233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "%p %s ", (void *) ctx, sev); 68233d2500723e5594f3e7c70896ffeeef32b9c950ywan 69233d2500723e5594f3e7c70896ffeeef32b9c950ywan va_start(ap, fmt); 70233d2500723e5594f3e7c70896ffeeef32b9c950ywan vfprintf(stderr, fmt, ap); 71233d2500723e5594f3e7c70896ffeeef32b9c950ywan va_end(ap); 72233d2500723e5594f3e7c70896ffeeef32b9c950ywan 73233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "\n"); 74233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 75233d2500723e5594f3e7c70896ffeeef32b9c950ywan 76233d2500723e5594f3e7c70896ffeeef32b9c950ywanint 77233d2500723e5594f3e7c70896ffeeef32b9c950ywanmain(int argc, char * argv[]) 78233d2500723e5594f3e7c70896ffeeef32b9c950ywan{ 79233d2500723e5594f3e7c70896ffeeef32b9c950ywan FILE * fp; 80233d2500723e5594f3e7c70896ffeeef32b9c950ywan int r, type; 81233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg * ctx; 82233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_audio_params aparams; 83233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet * pkt; 84233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_video_params vparams; 85233d2500723e5594f3e7c70896ffeeef32b9c950ywan size_t length, size; 86233d2500723e5594f3e7c70896ffeeef32b9c950ywan uint64_t duration, tstamp, pkt_tstamp; 87233d2500723e5594f3e7c70896ffeeef32b9c950ywan unsigned char * codec_data, * ptr; 88233d2500723e5594f3e7c70896ffeeef32b9c950ywan unsigned int cnt, i, j, track, tracks, pkt_cnt, pkt_track; 89233d2500723e5594f3e7c70896ffeeef32b9c950ywan unsigned int data_items = 0; 90233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_io io = { 91233d2500723e5594f3e7c70896ffeeef32b9c950ywan stdio_read, 92233d2500723e5594f3e7c70896ffeeef32b9c950ywan stdio_seek, 93233d2500723e5594f3e7c70896ffeeef32b9c950ywan stdio_tell, 94233d2500723e5594f3e7c70896ffeeef32b9c950ywan NULL 95233d2500723e5594f3e7c70896ffeeef32b9c950ywan }; 96233d2500723e5594f3e7c70896ffeeef32b9c950ywan 97233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (argc != 2) 98233d2500723e5594f3e7c70896ffeeef32b9c950ywan return EXIT_FAILURE; 99233d2500723e5594f3e7c70896ffeeef32b9c950ywan 100233d2500723e5594f3e7c70896ffeeef32b9c950ywan fp = fopen(argv[1], "rb"); 101233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (!fp) 102233d2500723e5594f3e7c70896ffeeef32b9c950ywan return EXIT_FAILURE; 103233d2500723e5594f3e7c70896ffeeef32b9c950ywan 104233d2500723e5594f3e7c70896ffeeef32b9c950ywan io.userdata = fp; 105233d2500723e5594f3e7c70896ffeeef32b9c950ywan 106233d2500723e5594f3e7c70896ffeeef32b9c950ywan ctx = NULL; 107233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_init(&ctx, io, log_callback, -1); 108233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r != 0) 109233d2500723e5594f3e7c70896ffeeef32b9c950ywan return EXIT_FAILURE; 110233d2500723e5594f3e7c70896ffeeef32b9c950ywan 111233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_track_count(ctx, &tracks); 112233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_duration(ctx, &duration); 113233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 114233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "media has %u tracks and duration %fs\n", tracks, duration / 1e9); 115233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 116233d2500723e5594f3e7c70896ffeeef32b9c950ywan 117233d2500723e5594f3e7c70896ffeeef32b9c950ywan for (i = 0; i < tracks; ++i) { 118233d2500723e5594f3e7c70896ffeeef32b9c950ywan type = nestegg_track_type(ctx, i); 119233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 120233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "track %u: type: %d codec: %d", i, 121233d2500723e5594f3e7c70896ffeeef32b9c950ywan type, nestegg_track_codec_id(ctx, i)); 122233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 123233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_track_codec_data_count(ctx, i, &data_items); 124233d2500723e5594f3e7c70896ffeeef32b9c950ywan for (j = 0; j < data_items; ++j) { 125233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_track_codec_data(ctx, i, j, &codec_data, &length); 126233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 127233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, " (%p, %u)", codec_data, (unsigned int) length); 128233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 129233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 130233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (type == NESTEGG_TRACK_VIDEO) { 131233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_track_video_params(ctx, i, &vparams); 132233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 133233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, " video: %ux%u (d: %ux%u %ux%ux%ux%u)", 134233d2500723e5594f3e7c70896ffeeef32b9c950ywan vparams.width, vparams.height, 135233d2500723e5594f3e7c70896ffeeef32b9c950ywan vparams.display_width, vparams.display_height, 136233d2500723e5594f3e7c70896ffeeef32b9c950ywan vparams.crop_top, vparams.crop_left, vparams.crop_bottom, vparams.crop_right); 137233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 138233d2500723e5594f3e7c70896ffeeef32b9c950ywan } else if (type == NESTEGG_TRACK_AUDIO) { 139233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_track_audio_params(ctx, i, &aparams); 140233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 141233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, " audio: %.2fhz %u bit %u channels", 142233d2500723e5594f3e7c70896ffeeef32b9c950ywan aparams.rate, aparams.depth, aparams.channels); 143233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 144233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 145233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 146233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "\n"); 147233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 148233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 149233d2500723e5594f3e7c70896ffeeef32b9c950ywan 150233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(SEEK_TEST) 151233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 152233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "seek to middle\n"); 153233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 154233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_track_seek(ctx, 0, duration / 2); 155233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 0) { 156233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 157233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "middle "); 158233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 159233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_read_packet(ctx, &pkt); 160233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 1) { 161233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_track(pkt, &track); 162233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_count(pkt, &cnt); 163233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_tstamp(pkt, &tstamp); 164233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 165233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 166233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 167233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_free_packet(pkt); 168233d2500723e5594f3e7c70896ffeeef32b9c950ywan } else { 169233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 170233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "middle seek failed\n"); 171233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 172233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 173233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 174233d2500723e5594f3e7c70896ffeeef32b9c950ywan 175233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 176233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "seek to ~end\n"); 177233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 178233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_track_seek(ctx, 0, duration - (duration / 10)); 179233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 0) { 180233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 181233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "end "); 182233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 183233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_read_packet(ctx, &pkt); 184233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 1) { 185233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_track(pkt, &track); 186233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_count(pkt, &cnt); 187233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_tstamp(pkt, &tstamp); 188233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 189233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 190233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 191233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_free_packet(pkt); 192233d2500723e5594f3e7c70896ffeeef32b9c950ywan } else { 193233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 194233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "end seek failed\n"); 195233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 196233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 197233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 198233d2500723e5594f3e7c70896ffeeef32b9c950ywan 199233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 200233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "seek to ~start\n"); 201233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 202233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_track_seek(ctx, 0, duration / 10); 203233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 0) { 204233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 205233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "start "); 206233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 207233d2500723e5594f3e7c70896ffeeef32b9c950ywan r = nestegg_read_packet(ctx, &pkt); 208233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (r == 1) { 209233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_track(pkt, &track); 210233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_count(pkt, &cnt); 211233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_tstamp(pkt, &tstamp); 212233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 213233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "* t %u pts %f frames %u\n", track, tstamp / 1e9, cnt); 214233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 215233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_free_packet(pkt); 216233d2500723e5594f3e7c70896ffeeef32b9c950ywan } else { 217233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 218233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "start seek failed\n"); 219233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 220233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 221233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 222233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 223233d2500723e5594f3e7c70896ffeeef32b9c950ywan 224233d2500723e5594f3e7c70896ffeeef32b9c950ywan while (nestegg_read_packet(ctx, &pkt) > 0) { 225233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_track(pkt, &pkt_track); 226233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_count(pkt, &pkt_cnt); 227233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_tstamp(pkt, &pkt_tstamp); 228233d2500723e5594f3e7c70896ffeeef32b9c950ywan 229233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 230233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "t %u pts %f frames %u: ", pkt_track, pkt_tstamp / 1e9, pkt_cnt); 231233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 232233d2500723e5594f3e7c70896ffeeef32b9c950ywan 233233d2500723e5594f3e7c70896ffeeef32b9c950ywan for (i = 0; i < pkt_cnt; ++i) { 234233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_packet_data(pkt, i, &ptr, &size); 235233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 236233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "%u ", (unsigned int) size); 237233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 238233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 239233d2500723e5594f3e7c70896ffeeef32b9c950ywan#if defined(DEBUG) 240233d2500723e5594f3e7c70896ffeeef32b9c950ywan fprintf(stderr, "\n"); 241233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif 242233d2500723e5594f3e7c70896ffeeef32b9c950ywan 243233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_free_packet(pkt); 244233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 245233d2500723e5594f3e7c70896ffeeef32b9c950ywan 246233d2500723e5594f3e7c70896ffeeef32b9c950ywan nestegg_destroy(ctx); 247233d2500723e5594f3e7c70896ffeeef32b9c950ywan fclose(fp); 248233d2500723e5594f3e7c70896ffeeef32b9c950ywan 249233d2500723e5594f3e7c70896ffeeef32b9c950ywan return EXIT_SUCCESS; 250233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 251