123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- /*
- * Copyright (c) 2016, Alliance for Open Media. All rights reserved
- *
- * This source code is subject to the terms of the BSD 2 Clause License and
- * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
- * was not distributed with this source code in the LICENSE file, you can
- * obtain it at www.aomedia.org/license/software. If the Alliance for Open
- * Media Patent License 1.0 was not distributed with this source code in the
- * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
- */
- #include "common/ivfdec.h"
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include "aom_ports/mem_ops.h"
- #include "aom_ports/sanitizer.h"
- #include "tools_common.h"
- static const char *IVF_SIGNATURE = "DKIF";
- static void fix_framerate(int *num, int *den) {
- if (*den <= 0 || *den >= 1000000000 || *num <= 0 || *num >= 1000) {
- // framerate seems to be invalid, just default to 30fps.
- *num = 30;
- *den = 1;
- }
- }
- int file_is_ivf(struct AvxInputContext *input_ctx) {
- unsigned char raw_hdr[32];
- int is_ivf = 0;
- if (buffer_input(input_ctx, 32, raw_hdr, /*buffered=*/true) == 32) {
- if (memcmp(IVF_SIGNATURE, raw_hdr, 4) == 0) {
- is_ivf = 1;
- if (mem_get_le16(raw_hdr + 4) != 0) {
- fprintf(stderr,
- "Error: Unrecognized IVF version! This file may not"
- " decode properly.\n");
- }
- input_ctx->fourcc = mem_get_le32(raw_hdr + 8);
- input_ctx->width = mem_get_le16(raw_hdr + 12);
- input_ctx->height = mem_get_le16(raw_hdr + 14);
- input_ctx->framerate.numerator = mem_get_le32(raw_hdr + 16);
- input_ctx->framerate.denominator = mem_get_le32(raw_hdr + 20);
- fix_framerate(&input_ctx->framerate.numerator,
- &input_ctx->framerate.denominator);
- }
- }
- if (!is_ivf) {
- rewind_detect(input_ctx);
- }
- return is_ivf;
- }
- int ivf_read_frame(struct AvxInputContext *input_ctx, uint8_t **buffer,
- size_t *bytes_read, size_t *buffer_size,
- aom_codec_pts_t *pts) {
- unsigned char raw_header[IVF_FRAME_HDR_SZ] = { 0 };
- size_t frame_size = 0;
- if (read_from_input(input_ctx, IVF_FRAME_HDR_SZ, raw_header) !=
- IVF_FRAME_HDR_SZ) {
- if (!input_eof(input_ctx))
- fprintf(stderr, "Warning: Failed to read frame size\n");
- } else {
- frame_size = mem_get_le32(raw_header);
- if (frame_size > 256 * 1024 * 1024) {
- fprintf(stderr, "Warning: Read invalid frame size (%u)\n",
- (unsigned int)frame_size);
- frame_size = 0;
- }
- if (frame_size > *buffer_size) {
- uint8_t *new_buffer = (uint8_t *)realloc(*buffer, 2 * frame_size);
- if (new_buffer) {
- *buffer = new_buffer;
- *buffer_size = 2 * frame_size;
- } else {
- fprintf(stderr, "Warning: Failed to allocate compressed data buffer\n");
- frame_size = 0;
- }
- }
- if (pts) {
- *pts = mem_get_le32(&raw_header[4]);
- *pts += ((aom_codec_pts_t)mem_get_le32(&raw_header[8]) << 32);
- }
- }
- if (!input_eof(input_ctx)) {
- ASAN_UNPOISON_MEMORY_REGION(*buffer, *buffer_size);
- if (read_from_input(input_ctx, frame_size, *buffer) != frame_size) {
- fprintf(stderr, "Warning: Failed to read full frame\n");
- return 1;
- }
- ASAN_POISON_MEMORY_REGION(*buffer + frame_size, *buffer_size - frame_size);
- *bytes_read = frame_size;
- return 0;
- }
- return 1;
- }
|