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