123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157 |
- // Copyright 2019 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include <fuzzer/FuzzedDataProvider.h>
- #include <stddef.h>
- #include <stdint.h>
- #include <random>
- #include "third_party/libyuv/include/libyuv.h"
- void FillBufferWithRandomData(uint8_t* dst, size_t len, std::minstd_rand0 rng) {
- size_t i;
- for (i = 0; i + 3 < len; i += 4) {
- *reinterpret_cast<uint32_t*>(dst) = rng();
- dst += 4;
- }
- for (; i < len; ++i) {
- *dst++ = rng();
- }
- }
- static void Scale(bool is420,
- int src_width,
- int src_height,
- int dst_width,
- int dst_height,
- int filter_num,
- std::string seed_str) {
- int src_width_uv, src_height_uv;
- if (is420) {
- src_width_uv = (std::abs(src_width) + 1) >> 1;
- src_height_uv = (std::abs(src_height) + 1) >> 1;
- } else {
- src_width_uv = (std::abs(src_width));
- src_height_uv = (std::abs(src_height));
- }
- size_t src_y_plane_size = (std::abs(src_width)) * (std::abs(src_height));
- size_t src_uv_plane_size = (src_width_uv) * (src_height_uv);
- int src_stride_y = std::abs(src_width);
- int src_stride_uv = src_width_uv;
- uint8_t* src_y = reinterpret_cast<uint8_t*>(malloc(src_y_plane_size));
- uint8_t* src_u = reinterpret_cast<uint8_t*>(malloc(src_uv_plane_size));
- uint8_t* src_v = reinterpret_cast<uint8_t*>(malloc(src_uv_plane_size));
- uint16_t* p_src_y_16 =
- reinterpret_cast<uint16_t*>(malloc(src_y_plane_size * 2));
- uint16_t* p_src_u_16 =
- reinterpret_cast<uint16_t*>(malloc(src_uv_plane_size * 2));
- uint16_t* p_src_v_16 =
- reinterpret_cast<uint16_t*>(malloc(src_uv_plane_size * 2));
- std::seed_seq seed(seed_str.begin(), seed_str.end());
- std::minstd_rand0 rng(seed);
- FillBufferWithRandomData(src_y, src_y_plane_size, rng);
- FillBufferWithRandomData(src_u, src_uv_plane_size, rng);
- FillBufferWithRandomData(src_v, src_uv_plane_size, rng);
- for (size_t i = 0; i < src_y_plane_size; ++i) {
- p_src_y_16[i] = src_y[i];
- }
- for (size_t i = 0; i < src_uv_plane_size; ++i) {
- p_src_u_16[i] = src_u[i];
- p_src_v_16[i] = src_v[i];
- }
- int dst_width_uv, dst_height_uv;
- if (is420) {
- dst_width_uv = (dst_width + 1) >> 1;
- dst_height_uv = (dst_height + 1) >> 1;
- } else {
- dst_width_uv = dst_width;
- dst_height_uv = dst_height;
- }
- size_t dst_y_plane_size = (dst_width) * (dst_height);
- size_t dst_uv_plane_size = (dst_width_uv) * (dst_height_uv);
- int dst_stride_y = dst_width;
- int dst_stride_uv = dst_width_uv;
- uint8_t* dst_y_c = reinterpret_cast<uint8_t*>(malloc(dst_y_plane_size));
- uint8_t* dst_u_c = reinterpret_cast<uint8_t*>(malloc(dst_uv_plane_size));
- uint8_t* dst_v_c = reinterpret_cast<uint8_t*>(malloc(dst_uv_plane_size));
- uint16_t* p_dst_y_16 =
- reinterpret_cast<uint16_t*>(malloc(dst_y_plane_size * 2));
- uint16_t* p_dst_u_16 =
- reinterpret_cast<uint16_t*>(malloc(dst_uv_plane_size * 2));
- uint16_t* p_dst_v_16 =
- reinterpret_cast<uint16_t*>(malloc(dst_uv_plane_size * 2));
- if (is420) {
- I420Scale(src_y, src_stride_y, src_u, src_stride_uv, src_v, src_stride_uv,
- src_width, src_height, dst_y_c, dst_stride_y, dst_u_c,
- dst_stride_uv, dst_v_c, dst_stride_uv, dst_width, dst_height,
- static_cast<libyuv::FilterMode>(filter_num));
- I420Scale_16(p_src_y_16, src_stride_y, p_src_u_16, src_stride_uv,
- p_src_v_16, src_stride_uv, src_width, src_height, p_dst_y_16,
- dst_stride_y, p_dst_u_16, dst_stride_uv, p_dst_v_16,
- dst_stride_uv, dst_width, dst_height,
- static_cast<libyuv::FilterMode>(filter_num));
- } else {
- I444Scale(src_y, src_stride_y, src_u, src_stride_uv, src_v, src_stride_uv,
- src_width, src_height, dst_y_c, dst_stride_y, dst_u_c,
- dst_stride_uv, dst_v_c, dst_stride_uv, dst_width, dst_height,
- static_cast<libyuv::FilterMode>(filter_num));
- I444Scale_16(p_src_y_16, src_stride_y, p_src_u_16, src_stride_uv,
- p_src_v_16, src_stride_uv, src_width, src_height, p_dst_y_16,
- dst_stride_y, p_dst_u_16, dst_stride_uv, p_dst_v_16,
- dst_stride_uv, dst_width, dst_height,
- static_cast<libyuv::FilterMode>(filter_num));
- }
- free(src_y);
- free(src_u);
- free(src_v);
- free(p_src_y_16);
- free(p_src_u_16);
- free(p_src_v_16);
- free(dst_y_c);
- free(dst_u_c);
- free(dst_v_c);
- free(p_dst_y_16);
- free(p_dst_u_16);
- free(p_dst_v_16);
- }
- extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
- FuzzedDataProvider provider(data, size);
- // Limit width and height for performance.
- int src_width = provider.ConsumeIntegralInRange<int>(1, 256);
- int src_height = provider.ConsumeIntegralInRange<int>(1, 256);
- int filter_num =
- provider.ConsumeIntegralInRange<int>(0, libyuv::FilterMode::kFilterBox);
- int dst_width = provider.ConsumeIntegralInRange<int>(1, 256);
- int dst_height = provider.ConsumeIntegralInRange<int>(1, 256);
- std::string seed = provider.ConsumeRemainingBytesAsString();
- Scale(true, src_width, src_height, dst_width, dst_height, filter_num, seed);
- Scale(false, src_width, src_height, dst_width, dst_height, filter_num, seed);
- return 0;
- }
|