Chromium Code Reviews| Index: media/filters/h264_parser_fuzzertest.cc |
| diff --git a/media/filters/h264_parser_fuzzertest.cc b/media/filters/h264_parser_fuzzertest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..7c98ed94692ab31074db9003260ff574dbf19169 |
| --- /dev/null |
| +++ b/media/filters/h264_parser_fuzzertest.cc |
| @@ -0,0 +1,56 @@ |
| +// Copyright 2016 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 <stddef.h> |
| + |
| +#include "base/numerics/safe_conversions.h" |
| +#include "media/filters/h264_parser.h" |
| + |
| +// Entry point for LibFuzzer. |
| +extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
| + if (size > 0) { |
|
DaleCurtis
2016/03/05 01:30:06
Invert conditional, return 0 if !size.
jrummell
2016/03/07 20:07:28
Done.
|
| + media::H264Parser parser; |
| + parser.SetStream(data, base::checked_cast<off_t>(size)); |
| + |
| + // Parse until the end of stream/unsupported stream/error in stream is |
| + // found. |
| + while (true) { |
| + media::H264NALU nalu; |
| + media::H264Parser::Result res = parser.AdvanceToNextNALU(&nalu); |
| + if (res != media::H264Parser::kOk) |
| + break; |
| + |
| + switch (nalu.nal_unit_type) { |
| + case media::H264NALU::kIDRSlice: |
| + case media::H264NALU::kNonIDRSlice: { |
| + media::H264SliceHeader shdr; |
| + res = parser.ParseSliceHeader(nalu, &shdr); |
| + } break; |
|
DaleCurtis
2016/03/05 01:30:06
put break in the {} block.
jrummell
2016/03/07 20:07:28
Done.
|
| + |
| + case media::H264NALU::kSPS: { |
| + int id; |
| + res = parser.ParseSPS(&id); |
| + } break; |
| + |
| + case media::H264NALU::kPPS: { |
| + int id; |
| + res = parser.ParsePPS(&id); |
| + } break; |
| + |
| + case media::H264NALU::kSEIMessage: { |
| + media::H264SEIMessage sei_msg; |
| + res = parser.ParseSEI(&sei_msg); |
| + } break; |
| + |
| + default: |
| + // Skip any other NALU. |
| + break; |
| + } |
| + if (res != media::H264Parser::kOk) |
| + break; |
| + } |
| + } |
| + |
| + return 0; |
| +} |