Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1184)

Unified Diff: media/filters/h264_parser_fuzzertest.cc

Issue 1770583002: Add fuzzer test for various media parsers (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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;
+}

Powered by Google App Engine
This is Rietveld 408576698