libFuzzer for blink::MHTMLParser

third_party/WebKit/Source/platform/testing/TestingPlatformSupport.cpp

 /*
  * Copyright (C) 2014 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 12 matching lines @@
  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "platform/testing/TestingPlatformSupport.h"
 
+#include "base/command_line.h"
+#include "base/memory/discardable_memory_allocator.h"
+#include "base/metrics/statistics_recorder.h"
+#include "base/test/icu_test_util.h"
+#include "base/test/test_discardable_memory_allocator.h"
+#include "cc/blink/web_compositor_support_impl.h"
+#include "platform/EventTracer.h"
+#include "platform/HTTPNames.h"
+#include "platform/heap/Heap.h"
+#include "wtf/CryptographicallyRandomNumber.h"
+#include "wtf/CurrentTime.h"
 #include "wtf/PtrUtil.h"
+#include "wtf/WTF.h"
+#include "wtf/allocator/Partitions.h"
 #include <memory>
 
 namespace blink {
 
+namespace {
+
+double dummyCurrentTime()
+{
+    return 0.0;
+}
+
+class DummyThread final : public blink::WebThread {
+public:
+    bool isCurrentThread() const override { return true; }
+    blink::WebScheduler* scheduler() const override { return nullptr; }
+};
+
+class DummyPlatform final : public blink::Platform {
+public:
+    DummyPlatform() { }
+
+    blink::WebThread* currentThread() override
+    {
+        static DummyThread dummyThread;
+        return &dummyThread;
+    };
+};
+
+} // namespace
+
 TestingPlatformSupport::TestingPlatformSupport()
     : TestingPlatformSupport(TestingPlatformSupport::Config())
 {
 }
 
 TestingPlatformSupport::TestingPlatformSupport(const Config& config)
     : m_config(config)
     , m_oldPlatform(Platform::current())
 {
     ASSERT(m_oldPlatform);
@@ skipping 138 matching lines @@
 WebThread* TestingPlatformSupportWithMockScheduler::currentThread()
 {
     return m_mockWebThread.get();
 }
 
 TestingPlatformMockScheduler* TestingPlatformSupportWithMockScheduler::mockWebScheduler()
 {
     return m_mockWebThread->mockWebScheduler();
 }
 
+class ScopedUnittestsEnvironmentSetup::Impl {
+public:
+    Impl(int argc, char** argv);
+    ~Impl();
+private:
+    base::TestDiscardableMemoryAllocator m_discardableMemoryAllocator;
+    std::unique_ptr<DummyPlatform> m_platform;
+
+    cc_blink::WebCompositorSupportImpl m_compositorSupport;

[esprehn, 2016/08/10 04:05:29]

Just put all of this in the header, there's no need to hide it in the Impl class

[Łukasz Anforowicz, 2016/08/10 19:31:17]

Done.

I need to note that I've also wrapped more things in std::unique_ptr:

- I think I should have done that before - to preserve the relative order of 1)
base::CommandLine::Init call and 2) construction of
base::TestDiscardableMemoryAllocator / cc_blink::WebCompositorSupportImpl / etc.

- Wrapping in std::unique_ptr allows to forward declare specific base and cc
classes (from TestingPlatformSupport.h) instead of including full base / cc
headers

+    TestingPlatformSupport::Config m_testingPlatformConfig;
+    std::unique_ptr<TestingPlatformSupport> m_testingPlatformSupport;
+};
+
+ScopedUnittestsEnvironmentSetup::Impl::Impl(int argc, char** argv)
+{
+    base::CommandLine::Init(argc, argv);
+
+    base::test::InitializeICUForTesting();
+
+    base::DiscardableMemoryAllocator::SetInstance(&m_discardableMemoryAllocator);
+    base::StatisticsRecorder::Initialize();
+
+    m_platform = wrapUnique(new DummyPlatform);
+    Platform::setCurrentPlatformForTesting(m_platform.get());
+
+    WTF::Partitions::initialize(nullptr);
+    WTF::setTimeFunctionsForTesting(dummyCurrentTime);
+    WTF::initialize(nullptr);
+
+    m_testingPlatformConfig.compositorSupport = &m_compositorSupport;
+    m_testingPlatformSupport = wrapUnique(new TestingPlatformSupport(m_testingPlatformConfig));
+
+    ProcessHeap::init();
+    ThreadState::attachMainThread();
+    ThreadState::current()->registerTraceDOMWrappers(nullptr, nullptr, nullptr);
+    EventTracer::initialize();
+    HTTPNames::init();
+}
+
+ScopedUnittestsEnvironmentSetup::Impl::~Impl()
+{
+    blink::ThreadState::detachMainThread();
+    blink::ProcessHeap::shutdown();
+    m_testingPlatformSupport.reset();
+
+    WTF::shutdown();
+    WTF::Partitions::shutdown();
+}
+
+ScopedUnittestsEnvironmentSetup::ScopedUnittestsEnvironmentSetup(int argc, char** argv)
+{
+    m_impl = wrapUnique(new Impl(argc, argv));
+}
+
+ScopedUnittestsEnvironmentSetup::~ScopedUnittestsEnvironmentSetup()
+{
+    m_impl.reset();

[esprehn, 2016/08/10 04:05:29]

manually clearing a unique_ptr in the destructor is very weird, why is this
needed?

[Łukasz Anforowicz, 2016/08/10 19:31:17]

You're right - this needs at least an explicit comment.  I am explicitly calling
reset to preserve the ordering of destruction from the code in RunAllTests.cpp
that is being replaced by ScopedUnittestsEnvironmentSetup class.

I've added a comment here to point out that we want to tear down testing
platform before shutting down WTF layer.  Does this help?

+}
+
 } // namespace blink