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

Unified Diff: net/filter/fuzzed_source_stream.h

Issue 2460633002: Add a fuzzer for GzipSourceStream (Closed)
Patch Set: Address comments Created 4 years, 1 month 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
« no previous file with comments | « net/BUILD.gn ('k') | net/filter/fuzzed_source_stream.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/filter/fuzzed_source_stream.h
diff --git a/net/filter/fuzzed_source_stream.h b/net/filter/fuzzed_source_stream.h
new file mode 100644
index 0000000000000000000000000000000000000000..6a969474899abedcdb7909efb1b5cbb8ba516f77
--- /dev/null
+++ b/net/filter/fuzzed_source_stream.h
@@ -0,0 +1,57 @@
+// 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.
+
+#ifndef NET_FILTER_FUZZED_SOURCE_STREAM_H_
+#define NET_FILTER_FUZZED_SOURCE_STREAM_H_
+
+#include <string>
+
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "net/base/completion_callback.h"
+#include "net/filter/source_stream.h"
+
+namespace base {
+class FuzzedDataProvider;
+} // namespace base
+
+namespace net {
+
+class IOBuffer;
+
+// A SourceStream implementation used in tests. This allows tests to specify
+// what data to return for each Read() call.
+class FuzzedSourceStream : public SourceStream {
+ public:
+ // |data_provider| is used to determine behavior of the FuzzedSourceStream.
+ // It must remain valid until after the FuzzedSocket is destroyed.
+ explicit FuzzedSourceStream(base::FuzzedDataProvider* data_provider);
+ ~FuzzedSourceStream() override;
+
+ // SourceStream implementation
+ int Read(IOBuffer* dest_buffer,
+ int buffer_size,
+ const CompletionCallback& callback) override;
+ std::string Description() const override;
+
+ private:
+ void OnReadComplete(const CompletionCallback& callback,
+ const std::string& fuzzed_data,
+ scoped_refptr<IOBuffer> read_buf,
+ int result);
+
+ base::FuzzedDataProvider* data_provider_;
+
+ // Whether there is a pending Read().
+ bool read_pending_;
+
+ // Last result returned by Read() is an error or 0.
+ bool end_returned_;
+
+ DISALLOW_COPY_AND_ASSIGN(FuzzedSourceStream);
+};
+
+} // namespace net
+
+#endif // NET_FILTER_FUZZED_SOURCE_STREAM_H_
« no previous file with comments | « net/BUILD.gn ('k') | net/filter/fuzzed_source_stream.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698