1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
|
/* Copyright (C) 2008 Vincent Penquerc'h.
This file is part of the Kate codec library.
Written by Vincent Penquerc'h.
Use, distribution and reproduction of this library is governed
by a BSD style source license included with this source in the
file 'COPYING'. Please read these terms before distributing. */
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#if defined WIN32 || defined _WIN32 || defined MSDOS || defined __CYGWIN__ || defined __EMX__ || defined OS2
#include <io.h>
#include <fcntl.h>
#endif
#if defined WIN32 || defined _WIN32
#include <process.h>
#endif
#include <stdarg.h>
#include <stdio.h>
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#include <errno.h>
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#include <ogg/ogg.h>
#include "kate/oggkate.h"
#include "kate_internal.h"
#include "ksrt.h"
#include "klrc.h"
#include "kkate.h"
#include "kfuzz.h"
#include "kstream.h"
#include "kutil.h"
#include "kread.h"
static int read_raw_packet(FILE *f,char **buffer,ogg_int64_t bytes)
{
size_t ret;
*buffer=(char*)kate_realloc(*buffer,bytes);
if (!*buffer) return -1;
ret=read(fileno(f),*buffer,bytes);
if (ret<(size_t)bytes) return -1;
return 0;
}
int read_raw_size_and_packet(FILE *f,char **buffer,ogg_int64_t *bytes)
{
size_t bytes_read;
int ret;
/* all subsequent packets are prefixed with 64 bits (signed) of the packet length in bytes */
bytes_read=read(fileno(f),bytes,8);
if (bytes_read!=8 || *bytes<=0) {
fprintf(stderr,"failed to read raw kate packet size (read %lu, bytes %"PRId64")\n",(unsigned long)bytes_read,*bytes);
return -1;
}
ret=read_raw_packet(f,buffer,*bytes);
if (ret<0) {
fprintf(stderr,"failed to read raw kate packet (%"PRId64" bytes)\n",*bytes);
return -1;
}
return 0;
}
FILE *open_and_probe_stream(const char *filename)
{
FILE *fin;
if (!filename || !strcmp(filename,"-")) {
#if defined WIN32 || defined _WIN32
_setmode(_fileno(stdin),_O_BINARY);
#else
#if defined MSDOS || defined __CYGWIN__ || defined __EMX__ || defined OS2 || defined __BORLANDC__
setmode(fileno(stdin),_O_BINARY);
#endif
#endif
fin=stdin;
}
else {
fin=fopen(filename,"rb");
if (!fin) {
fprintf(stderr,"%s: %s\n",filename,strerror(errno));
return NULL;
}
setvbuf(fin,NULL,_IONBF,0);
}
return fin;
}
int parse_ogg_stream(FILE *f,const char *pre_buffer,size_t pre_bytes,ogg_parser_funcs funcs,kate_uintptr_t data)
{
char *ptr=NULL;
ogg_sync_state oy;
ogg_page og;
ogg_int64_t bytes_read;
static const size_t buffer_size=4096;
size_t offset=0;
int ret;
long last_returned;
long leftover_size=0;
int in_hole=0;
long first_hole_size=0;
long hole_offset;
ogg_sync_init(&oy);
/* first add the buffer, if any */
if (pre_buffer && pre_bytes) {
ptr=ogg_sync_buffer(&oy,pre_bytes);
if (!ptr) {
fprintf(stderr,"Failed to get sync buffer for %lu bytes\n",(unsigned long)pre_bytes);
goto error;
}
memcpy(ptr,pre_buffer,pre_bytes);
ogg_sync_wrote(&oy,pre_bytes);
}
while (1) {
ptr=ogg_sync_buffer(&oy,buffer_size);
if (!ptr) {
fprintf(stderr,"Failed to get sync buffer for %lu bytes\n",(unsigned long)buffer_size);
goto error;
}
bytes_read=read(fileno(f),ptr,buffer_size);
if (bytes_read==0) {
break;
}
ogg_sync_wrote(&oy,bytes_read);
last_returned=0;
while ((ret=ogg_sync_pageout(&oy,&og))!=0) {
if (ret>0) {
/* new page */
if (in_hole) {
long hole_size=first_hole_size+leftover_size+oy.returned-(og.header_len+og.body_len);
if (funcs.on_hole) {
ret=(*funcs.on_hole)(data,hole_offset,hole_size);
if (ret<0) goto error;
}
offset+=hole_size;
}
if (funcs.on_page) {
ret=(*funcs.on_page)(data,offset,&og);
if (ret<0) goto error;
}
in_hole=0;
offset+=og.header_len+og.body_len;
}
else {
/* hole in data */
first_hole_size=oy.returned-last_returned;
hole_offset=offset;
in_hole=1;
}
leftover_size=0;
last_returned=oy.returned;
}
leftover_size+=oy.fill-last_returned;
}
/* detect holes at end of file */
if (leftover_size>0) {
if (funcs.on_hole) {
ret=(*funcs.on_hole)(data,offset,leftover_size);
if (ret<0) goto error;
}
}
ogg_sync_clear(&oy);
return 0;
error:
ogg_sync_clear(&oy);
return -1;
}
|