generic-library/vpx
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge "Adding raw_read_frame() function to vpxdec."

Browse Source
This commit is contained in:
Dmitry Kovalev 2014-01-13 11:46:01 -08:00 committed by Gerrit Code Review
commit 9c9fdf30e6
1 changed files with 53 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
vpxdec.c
View File

@ -167,65 +167,69 @@ void usage_exit() {
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
static int read_frame(struct VpxDecInputContext *input, static int raw_read_frame(struct VpxInputContext *input_ctx, uint8_t **buffer,
uint8_t **buf, size_t *bytes_read, size_t *buffer_size) {
size_t *bytes_in_buffer,
size_t *buffer_size) {
char raw_hdr[RAW_FRAME_HDR_SZ]; char raw_hdr[RAW_FRAME_HDR_SZ];
size_t bytes_to_read = 0; size_t frame_size = 0;
FILE *infile = input->vpx_input_ctx->file; FILE *infile = input_ctx->file;
enum VideoFileType kind = input->vpx_input_ctx->file_type;
if (kind == FILE_TYPE_WEBM) {
return webm_read_frame(input->webm_ctx,
buf, bytes_in_buffer, buffer_size);
} else if (kind == FILE_TYPE_RAW) {
if (fread(raw_hdr, RAW_FRAME_HDR_SZ, 1, infile) != 1) { if (fread(raw_hdr, RAW_FRAME_HDR_SZ, 1, infile) != 1) {
if (!feof(infile)) if (!feof(infile))
warn("Failed to read RAW frame size\n"); warn("Failed to read RAW frame size\n");
} else { } else {
const int kCorruptFrameThreshold = 256 * 1024 * 1024; const int kCorruptFrameThreshold = 256 * 1024 * 1024;
const int kFrameTooSmallThreshold = 256 * 1024; const int kFrameTooSmallThreshold = 256 * 1024;
bytes_to_read = mem_get_le32(raw_hdr); frame_size = mem_get_le32(raw_hdr);
if (bytes_to_read > kCorruptFrameThreshold) { if (frame_size > kCorruptFrameThreshold) {
warn("Read invalid frame size (%u)\n", (unsigned int)bytes_to_read); warn("Read invalid frame size (%u)\n", (unsigned int)frame_size);
bytes_to_read = 0; frame_size = 0;
} }
if (kind == FILE_TYPE_RAW && bytes_to_read < kFrameTooSmallThreshold) { if (frame_size < kFrameTooSmallThreshold) {
warn("Warning: Read invalid frame size (%u) - not a raw file?\n", warn("Warning: Read invalid frame size (%u) - not a raw file?\n",
(unsigned int)bytes_to_read); (unsigned int)frame_size);
} }
if (bytes_to_read > *buffer_size) { if (frame_size > *buffer_size) {
uint8_t *new_buf = realloc(*buf, 2 * bytes_to_read); uint8_t *new_buf = realloc(*buffer, 2 * frame_size);
if (new_buf) { if (new_buf) {
*buf = new_buf; *buffer = new_buf;
*buffer_size = 2 * bytes_to_read; *buffer_size = 2 * frame_size;
} else { } else {
warn("Failed to allocate compressed data buffer\n"); warn("Failed to allocate compressed data buffer\n");
bytes_to_read = 0; frame_size = 0;
} }
} }
} }
if (!feof(infile)) { if (!feof(infile)) {
if (fread(*buf, 1, bytes_to_read, infile) != bytes_to_read) { if (fread(*buffer, 1, frame_size, infile) != frame_size) {
warn("Failed to read full frame\n"); warn("Failed to read full frame\n");
return 1; return 1;
} }
*bytes_in_buffer = bytes_to_read; *bytes_read = frame_size;
} }
return 0; return 0;
} else if (kind == FILE_TYPE_IVF) {
return ivf_read_frame(input->vpx_input_ctx,
buf, bytes_in_buffer, buffer_size);
} }
static int read_frame(struct VpxDecInputContext *input, uint8_t **buf,
size_t *bytes_in_buffer, size_t *buffer_size) {
switch (input->vpx_input_ctx->file_type) {
case FILE_TYPE_WEBM:
return webm_read_frame(input->webm_ctx,
buf, bytes_in_buffer, buffer_size);
case FILE_TYPE_RAW:
return raw_read_frame(input->vpx_input_ctx,
buf, bytes_in_buffer, buffer_size);
case FILE_TYPE_IVF:
return ivf_read_frame(input->vpx_input_ctx,
buf, bytes_in_buffer, buffer_size);
default:
return 1; return 1;
} }
}
void *out_open(const char *out_fn, int do_md5) { void *out_open(const char *out_fn, int do_md5) {
void *out = NULL; void *out = NULL;