h265_decoder_fuzzertest.cc 3.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <stddef.h>
  5. #include "base/numerics/safe_conversions.h"
  6. #include "media/base/decoder_buffer.h"
  7. #include "media/base/video_codecs.h"
  8. #include "media/base/video_types.h"
  9. #include "media/gpu/h265_decoder.h"
  10. namespace {
  11. class FakeH265Accelerator : public media::H265Decoder::H265Accelerator {
  12. public:
  13. FakeH265Accelerator() = default;
  14. FakeH265Accelerator(const FakeH265Accelerator&) = delete;
  15. FakeH265Accelerator& operator=(const FakeH265Accelerator&) = delete;
  16. ~FakeH265Accelerator() override = default;
  17. // media::H265Decoder::H265Accelerator
  18. scoped_refptr<media::H265Picture> CreateH265Picture() override {
  19. return new media::H265Picture();
  20. }
  21. Status SubmitFrameMetadata(const media::H265SPS* sps,
  22. const media::H265PPS* pps,
  23. const media::H265SliceHeader* slice_hdr,
  24. const media::H265Picture::Vector& ref_pic_list,
  25. scoped_refptr<media::H265Picture> pic) override {
  26. return Status::kOk;
  27. }
  28. Status SubmitSlice(
  29. const media::H265SPS* sps,
  30. const media::H265PPS* pps,
  31. const media::H265SliceHeader* slice_hdr,
  32. const media::H265Picture::Vector& ref_pic_list0,
  33. const media::H265Picture::Vector& ref_pic_list1,
  34. const media::H265Picture::Vector& ref_pic_set_lt_curr,
  35. const media::H265Picture::Vector& ref_pic_set_st_curr_after,
  36. const media::H265Picture::Vector& ref_pic_set_st_curr_before,
  37. scoped_refptr<media::H265Picture> pic,
  38. const uint8_t* data,
  39. size_t size,
  40. const std::vector<media::SubsampleEntry>& subsamples) override {
  41. return Status::kOk;
  42. }
  43. Status SubmitDecode(scoped_refptr<media::H265Picture> pic) override {
  44. return Status::kOk;
  45. }
  46. bool OutputPicture(scoped_refptr<media::H265Picture> pic) override {
  47. return true;
  48. }
  49. void Reset() override {}
  50. Status SetStream(base::span<const uint8_t> stream,
  51. const media::DecryptConfig* decrypt_config) override {
  52. return Status::kOk;
  53. }
  54. bool IsChromaSamplingSupported(media::VideoChromaSampling format) override {
  55. return format == media::VideoChromaSampling::k420;
  56. }
  57. };
  58. } // namespace
  59. // Entry point for LibFuzzer.
  60. extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  61. if (!size)
  62. return 0;
  63. media::H265Decoder decoder(std::make_unique<FakeH265Accelerator>(),
  64. media::HEVCPROFILE_MAIN);
  65. scoped_refptr<media::DecoderBuffer> decoder_buffer =
  66. media::DecoderBuffer::CopyFrom(data, size);
  67. decoder.SetStream(1, *decoder_buffer);
  68. // Decode should consume all the data unless it returns kConfigChange, and in
  69. // that case it needs to be called again.
  70. while (true) {
  71. if (decoder.Decode() != media::AcceleratedVideoDecoder::kConfigChange)
  72. break;
  73. }
  74. return 0;
  75. }