00001
00002
00003
00004
00005
00006
00007
00033 #include <stdio.h>
00034 #include "oggdec.h"
00035 #include "avformat.h"
00036
00037 #define MAX_PAGE_SIZE 65307
00038 #define DECODER_BUFFER_SIZE MAX_PAGE_SIZE
00039
00040 static ogg_codec_t *ogg_codecs[] = {
00041 &vorbis_codec,
00042 &theora_codec,
00043 &flac_codec,
00044 &old_flac_codec,
00045 &ogm_video_codec,
00046 &ogm_audio_codec,
00047 &ogm_text_codec,
00048 &ogm_old_codec,
00049 NULL
00050 };
00051
00052
00053 static int
00054 ogg_save (AVFormatContext * s)
00055 {
00056 ogg_t *ogg = s->priv_data;
00057 ogg_state_t *ost =
00058 av_malloc(sizeof (*ost) + (ogg->nstreams-1) * sizeof (*ogg->streams));
00059 int i;
00060 ost->pos = url_ftell (s->pb);
00061 ost->curidx = ogg->curidx;
00062 ost->next = ogg->state;
00063 ost->nstreams = ogg->nstreams;
00064 memcpy(ost->streams, ogg->streams, ogg->nstreams * sizeof(*ogg->streams));
00065
00066 for (i = 0; i < ogg->nstreams; i++){
00067 ogg_stream_t *os = ogg->streams + i;
00068 os->buf = av_malloc (os->bufsize);
00069 memset (os->buf, 0, os->bufsize);
00070 memcpy (os->buf, ost->streams[i].buf, os->bufpos);
00071 }
00072
00073 ogg->state = ost;
00074
00075 return 0;
00076 }
00077
00078 static int
00079 ogg_restore (AVFormatContext * s, int discard)
00080 {
00081 ogg_t *ogg = s->priv_data;
00082 ByteIOContext *bc = s->pb;
00083 ogg_state_t *ost = ogg->state;
00084 int i;
00085
00086 if (!ost)
00087 return 0;
00088
00089 ogg->state = ost->next;
00090
00091 if (!discard){
00092 for (i = 0; i < ogg->nstreams; i++)
00093 av_free (ogg->streams[i].buf);
00094
00095 url_fseek (bc, ost->pos, SEEK_SET);
00096 ogg->curidx = ost->curidx;
00097 ogg->nstreams = ost->nstreams;
00098 memcpy(ogg->streams, ost->streams,
00099 ost->nstreams * sizeof(*ogg->streams));
00100 }
00101
00102 av_free (ost);
00103
00104 return 0;
00105 }
00106
00107 static int
00108 ogg_reset (ogg_t * ogg)
00109 {
00110 int i;
00111
00112 for (i = 0; i < ogg->nstreams; i++){
00113 ogg_stream_t *os = ogg->streams + i;
00114 os->bufpos = 0;
00115 os->pstart = 0;
00116 os->psize = 0;
00117 os->granule = -1;
00118 os->lastgp = -1;
00119 os->nsegs = 0;
00120 os->segp = 0;
00121 }
00122
00123 ogg->curidx = -1;
00124
00125 return 0;
00126 }
00127
00128 static ogg_codec_t *
00129 ogg_find_codec (uint8_t * buf, int size)
00130 {
00131 int i;
00132
00133 for (i = 0; ogg_codecs[i]; i++)
00134 if (size >= ogg_codecs[i]->magicsize &&
00135 !memcmp (buf, ogg_codecs[i]->magic, ogg_codecs[i]->magicsize))
00136 return ogg_codecs[i];
00137
00138 return NULL;
00139 }
00140
00141 static int
00142 ogg_find_stream (ogg_t * ogg, int serial)
00143 {
00144 int i;
00145
00146 for (i = 0; i < ogg->nstreams; i++)
00147 if (ogg->streams[i].serial == serial)
00148 return i;
00149
00150 return -1;
00151 }
00152
00153 static int
00154 ogg_new_stream (AVFormatContext * s, uint32_t serial)
00155 {
00156
00157 ogg_t *ogg = s->priv_data;
00158 int idx = ogg->nstreams++;
00159 AVStream *st;
00160 ogg_stream_t *os;
00161
00162 ogg->streams = av_realloc (ogg->streams,
00163 ogg->nstreams * sizeof (*ogg->streams));
00164 memset (ogg->streams + idx, 0, sizeof (*ogg->streams));
00165 os = ogg->streams + idx;
00166 os->serial = serial;
00167 os->bufsize = DECODER_BUFFER_SIZE;
00168 os->buf = av_malloc(os->bufsize);
00169 os->header = -1;
00170
00171 st = av_new_stream (s, idx);
00172 if (!st)
00173 return AVERROR(ENOMEM);
00174
00175 av_set_pts_info(st, 64, 1, 1000000);
00176
00177 return idx;
00178 }
00179
00180 static int
00181 ogg_new_buf(ogg_t *ogg, int idx)
00182 {
00183 ogg_stream_t *os = ogg->streams + idx;
00184 uint8_t *nb = av_malloc(os->bufsize);
00185 int size = os->bufpos - os->pstart;
00186 if(os->buf){
00187 memcpy(nb, os->buf + os->pstart, size);
00188 av_free(os->buf);
00189 }
00190 os->buf = nb;
00191 os->bufpos = size;
00192 os->pstart = 0;
00193
00194 return 0;
00195 }
00196
00197 static int
00198 ogg_read_page (AVFormatContext * s, int *str)
00199 {
00200 ByteIOContext *bc = s->pb;
00201 ogg_t *ogg = s->priv_data;
00202 ogg_stream_t *os;
00203 int i = 0;
00204 int flags, nsegs;
00205 uint64_t gp;
00206 uint32_t serial;
00207 uint32_t seq;
00208 uint32_t crc;
00209 int size, idx;
00210 uint8_t sync[4];
00211 int sp = 0;
00212
00213 if (get_buffer (bc, sync, 4) < 4)
00214 return -1;
00215
00216 do{
00217 int c;
00218
00219 if (sync[sp & 3] == 'O' &&
00220 sync[(sp + 1) & 3] == 'g' &&
00221 sync[(sp + 2) & 3] == 'g' && sync[(sp + 3) & 3] == 'S')
00222 break;
00223
00224 c = url_fgetc (bc);
00225 if (c < 0)
00226 return -1;
00227 sync[sp++ & 3] = c;
00228 }while (i++ < MAX_PAGE_SIZE);
00229
00230 if (i >= MAX_PAGE_SIZE){
00231 av_log (s, AV_LOG_INFO, "ogg, can't find sync word\n");
00232 return -1;
00233 }
00234
00235 if (url_fgetc (bc) != 0)
00236 return -1;
00237
00238 flags = url_fgetc (bc);
00239 gp = get_le64 (bc);
00240 serial = get_le32 (bc);
00241 seq = get_le32 (bc);
00242 crc = get_le32 (bc);
00243 nsegs = url_fgetc (bc);
00244
00245 idx = ogg_find_stream (ogg, serial);
00246 if (idx < 0){
00247 idx = ogg_new_stream (s, serial);
00248 if (idx < 0)
00249 return -1;
00250 }
00251
00252 os = ogg->streams + idx;
00253
00254 if(os->psize > 0)
00255 ogg_new_buf(ogg, idx);
00256
00257 if (get_buffer (bc, os->segments, nsegs) < nsegs)
00258 return -1;
00259
00260 os->nsegs = nsegs;
00261 os->segp = 0;
00262
00263 size = 0;
00264 for (i = 0; i < nsegs; i++)
00265 size += os->segments[i];
00266
00267 if (flags & OGG_FLAG_CONT){
00268 if (!os->psize){
00269 while (os->segp < os->nsegs){
00270 int seg = os->segments[os->segp++];
00271 os->pstart += seg;
00272 if (seg < 255)
00273 break;
00274 }
00275 }
00276 }else{
00277 os->psize = 0;
00278 }
00279
00280 if (os->bufsize - os->bufpos < size){
00281 uint8_t *nb = av_malloc (os->bufsize *= 2);
00282 memcpy (nb, os->buf, os->bufpos);
00283 av_free (os->buf);
00284 os->buf = nb;
00285 }
00286
00287 if (get_buffer (bc, os->buf + os->bufpos, size) < size)
00288 return -1;
00289
00290 os->lastgp = os->granule;
00291 os->bufpos += size;
00292 os->granule = gp;
00293 os->flags = flags;
00294
00295 if (str)
00296 *str = idx;
00297
00298 return 0;
00299 }
00300
00301 static int
00302 ogg_packet (AVFormatContext * s, int *str, int *dstart, int *dsize)
00303 {
00304 ogg_t *ogg = s->priv_data;
00305 int idx;
00306 ogg_stream_t *os;
00307 int complete = 0;
00308 int segp = 0, psize = 0;
00309
00310 #if 0
00311 av_log (s, AV_LOG_DEBUG, "ogg_packet: curidx=%i\n", ogg->curidx);
00312 #endif
00313
00314 do{
00315 idx = ogg->curidx;
00316
00317 while (idx < 0){
00318 if (ogg_read_page (s, &idx) < 0)
00319 return -1;
00320 }
00321
00322 os = ogg->streams + idx;
00323
00324 #if 0
00325 av_log (s, AV_LOG_DEBUG,
00326 "ogg_packet: idx=%d pstart=%d psize=%d segp=%d nsegs=%d\n",
00327 idx, os->pstart, os->psize, os->segp, os->nsegs);
00328 #endif
00329
00330 if (!os->codec){
00331 if (os->header < 0){
00332 os->codec = ogg_find_codec (os->buf, os->bufpos);
00333 if (!os->codec){
00334 os->header = 0;
00335 return 0;
00336 }
00337 }else{
00338 return 0;
00339 }
00340 }
00341
00342 segp = os->segp;
00343 psize = os->psize;
00344
00345 while (os->segp < os->nsegs){
00346 int ss = os->segments[os->segp++];
00347 os->psize += ss;
00348 if (ss < 255){
00349 complete = 1;
00350 break;
00351 }
00352 }
00353
00354 if (!complete && os->segp == os->nsegs){
00355 ogg->curidx = -1;
00356 }
00357 }while (!complete);
00358
00359 #if 0
00360 av_log (s, AV_LOG_DEBUG,
00361 "ogg_packet: idx %i, frame size %i, start %i\n",
00362 idx, os->psize, os->pstart);
00363 #endif
00364
00365 ogg->curidx = idx;
00366
00367 if (os->header < 0){
00368 int hdr = os->codec->header (s, idx);
00369 if (!hdr){
00370 os->header = os->seq;
00371 os->segp = segp;
00372 os->psize = psize;
00373 ogg->headers = 1;
00374 }else{
00375 os->pstart += os->psize;
00376 os->psize = 0;
00377 }
00378 }
00379
00380 if (os->header > -1 && os->seq > os->header){
00381 os->pflags = 0;
00382 if (os->codec && os->codec->packet)
00383 os->codec->packet (s, idx);
00384 if (str)
00385 *str = idx;
00386 if (dstart)
00387 *dstart = os->pstart;
00388 if (dsize)
00389 *dsize = os->psize;
00390 os->pstart += os->psize;
00391 os->psize = 0;
00392 }
00393
00394 os->seq++;
00395 if (os->segp == os->nsegs)
00396 ogg->curidx = -1;
00397
00398 return 0;
00399 }
00400
00401 static int
00402 ogg_get_headers (AVFormatContext * s)
00403 {
00404 ogg_t *ogg = s->priv_data;
00405
00406 do{
00407 if (ogg_packet (s, NULL, NULL, NULL) < 0)
00408 return -1;
00409 }while (!ogg->headers);
00410
00411 #if 0
00412 av_log (s, AV_LOG_DEBUG, "found headers\n");
00413 #endif
00414
00415 return 0;
00416 }
00417
00418 static uint64_t
00419 ogg_gptopts (AVFormatContext * s, int i, uint64_t gp)
00420 {
00421 ogg_t *ogg = s->priv_data;
00422 ogg_stream_t *os = ogg->streams + i;
00423 uint64_t pts = AV_NOPTS_VALUE;
00424
00425 if(os->codec->gptopts){
00426 pts = os->codec->gptopts(s, i, gp);
00427 } else {
00428 pts = gp;
00429 }
00430
00431 return pts;
00432 }
00433
00434
00435 static int
00436 ogg_get_length (AVFormatContext * s)
00437 {
00438 ogg_t *ogg = s->priv_data;
00439 int idx = -1, i;
00440 offset_t size, end;
00441
00442 if(url_is_streamed(s->pb))
00443 return 0;
00444
00445
00446 if (s->duration != AV_NOPTS_VALUE)
00447 return 0;
00448
00449 size = url_fsize(s->pb);
00450 if(size < 0)
00451 return 0;
00452 end = size > MAX_PAGE_SIZE? size - MAX_PAGE_SIZE: size;
00453
00454 ogg_save (s);
00455 url_fseek (s->pb, end, SEEK_SET);
00456
00457 while (!ogg_read_page (s, &i)){
00458 if (ogg->streams[i].granule != -1 && ogg->streams[i].granule != 0 &&
00459 ogg->streams[i].codec)
00460 idx = i;
00461 }
00462
00463 if (idx != -1){
00464 s->streams[idx]->duration =
00465 ogg_gptopts (s, idx, ogg->streams[idx].granule);
00466 }
00467
00468 ogg->size = size;
00469 ogg_restore (s, 0);
00470 ogg_save (s);
00471 while (!ogg_read_page (s, &i)) {
00472 if (i == idx && ogg->streams[i].granule != -1 && ogg->streams[i].granule != 0)
00473 break;
00474 }
00475 if (i == idx) {
00476 s->streams[idx]->start_time = ogg_gptopts (s, idx, ogg->streams[idx].granule);
00477 s->streams[idx]->duration -= s->streams[idx]->start_time;
00478 }
00479 ogg_restore (s, 0);
00480
00481 return 0;
00482 }
00483
00484
00485 static int
00486 ogg_read_header (AVFormatContext * s, AVFormatParameters * ap)
00487 {
00488 ogg_t *ogg = s->priv_data;
00489 int i;
00490 ogg->curidx = -1;
00491
00492 if (ogg_get_headers (s) < 0){
00493 return -1;
00494 }
00495
00496 for (i = 0; i < ogg->nstreams; i++)
00497 if (ogg->streams[i].header < 0)
00498 ogg->streams[i].codec = NULL;
00499
00500
00501 ogg_get_length (s);
00502
00503
00504 return 0;
00505 }
00506
00507
00508 static int
00509 ogg_read_packet (AVFormatContext * s, AVPacket * pkt)
00510 {
00511 ogg_t *ogg;
00512 ogg_stream_t *os;
00513 int idx = -1;
00514 int pstart, psize;
00515
00516
00517 do{
00518 if (ogg_packet (s, &idx, &pstart, &psize) < 0)
00519 return AVERROR(EIO);
00520 }while (idx < 0 || !s->streams[idx]);
00521
00522 ogg = s->priv_data;
00523 os = ogg->streams + idx;
00524
00525
00526 if (av_new_packet (pkt, psize) < 0)
00527 return AVERROR(EIO);
00528 pkt->stream_index = idx;
00529 memcpy (pkt->data, os->buf + pstart, psize);
00530 if (os->lastgp != -1LL){
00531 pkt->pts = ogg_gptopts (s, idx, os->lastgp);
00532 os->lastgp = -1;
00533 }
00534
00535 pkt->flags = os->pflags;
00536
00537 return psize;
00538 }
00539
00540
00541 static int
00542 ogg_read_close (AVFormatContext * s)
00543 {
00544 ogg_t *ogg = s->priv_data;
00545 int i;
00546
00547 for (i = 0; i < ogg->nstreams; i++){
00548 av_free (ogg->streams[i].buf);
00549 av_free (ogg->streams[i].private);
00550 }
00551 av_free (ogg->streams);
00552 return 0;
00553 }
00554
00555
00556 static int64_t
00557 ogg_read_timestamp (AVFormatContext * s, int stream_index, int64_t * pos_arg,
00558 int64_t pos_limit)
00559 {
00560 ogg_t *ogg = s->priv_data;
00561 ByteIOContext *bc = s->pb;
00562 int64_t pts = AV_NOPTS_VALUE;
00563 int i;
00564 url_fseek(bc, *pos_arg, SEEK_SET);
00565 while (url_ftell(bc) < pos_limit && !ogg_read_page (s, &i)) {
00566 if (ogg->streams[i].granule != -1 && ogg->streams[i].granule != 0 &&
00567 ogg->streams[i].codec && i == stream_index) {
00568 pts = ogg_gptopts(s, i, ogg->streams[i].granule);
00569
00570
00571 *pos_arg = url_ftell(bc);
00572 break;
00573 }
00574 }
00575 ogg_reset(ogg);
00576 return pts;
00577 }
00578
00579 static int ogg_probe(AVProbeData *p)
00580 {
00581 if (p->buf[0] == 'O' && p->buf[1] == 'g' &&
00582 p->buf[2] == 'g' && p->buf[3] == 'S' &&
00583 p->buf[4] == 0x0 && p->buf[5] <= 0x7 )
00584 return AVPROBE_SCORE_MAX;
00585 else
00586 return 0;
00587 }
00588
00589 AVInputFormat ogg_demuxer = {
00590 "ogg",
00591 "Ogg",
00592 sizeof (ogg_t),
00593 ogg_probe,
00594 ogg_read_header,
00595 ogg_read_packet,
00596 ogg_read_close,
00597 NULL,
00598 ogg_read_timestamp,
00599 .extensions = "ogg",
00600 };