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

Unified Diff: net/filter/fuzzed_source_stream.h

Issue 2460633002: Add a fuzzer for GzipSourceStream (Closed)
Patch Set: Address comments Created 4 years, 2 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: 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..79810eb06afd315ac6f16208755a7cce29c5aca7
--- /dev/null
+++ b/net/filter/fuzzed_source_stream.h
@@ -0,0 +1,54 @@
+// 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 "base/macros.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,
mmenke 2016/10/31 14:59:52 Should probably include <string> here instead of t
xunjieli 2016/10/31 15:14:56 Done. Thanks for point that out. I recently learne
+ scoped_refptr<IOBuffer> read_buf,
mmenke 2016/10/31 14:59:52 include base/memory/ref_counted.h
xunjieli 2016/10/31 15:14:56 Done.
+ 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_

Powered by Google App Engine
This is Rietveld 408576698