Prevent access to null handler in RTCDataChannel.

third_party/WebKit/Source/modules/peerconnection/RTCDataChannelTest.cpp

 // Copyright 2015 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 "modules/peerconnection/RTCDataChannel.h"
 
+#include <string>
 #include "bindings/core/v8/ExceptionState.h"
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/DOMException.h"
 #include "core/events/Event.h"
 #include "platform/heap/Heap.h"
 #include "platform/wtf/PtrUtil.h"
 #include "platform/wtf/RefPtr.h"
 #include "platform/wtf/text/WTFString.h"
 #include "public/platform/WebRTCDataChannelHandler.h"
 #include "public/platform/WebVector.h"
@@ skipping 122 matching lines @@
   EXPECT_EQ(4U, channel->scheduled_events_.size());  // No new events.
 
   handler->DrainBuffer(1);
   EXPECT_EQ(97U, channel->bufferedAmount());
   EXPECT_EQ(5U, channel->scheduled_events_.size());  // New event.
   EXPECT_EQ(
       "bufferedamountlow",
       std::string(channel->scheduled_events_.back()->type().Utf8().data()));
 }
 
+TEST(RTCDataChannelTest, SendAfterContextDestroyed) {
+  MockHandler* handler = new MockHandler();
+  RTCDataChannel* channel = RTCDataChannel::Create(0, WTF::WrapUnique(handler));
+  handler->ChangeState(WebRTCDataChannelHandlerClient::kReadyStateOpen);
+  channel->ContextDestroyed(nullptr);
+
+  String message(std::string(100, 'A').c_str());
+  DummyExceptionStateForTesting exception_state;
+  channel->send(message, exception_state);
+
+  EXPECT_TRUE(exception_state.HadException());
+}
+
+TEST(RTCDataChannelTest, CloseAfterContextDestroyed) {
+  MockHandler* handler = new MockHandler();
+  RTCDataChannel* channel = RTCDataChannel::Create(0, WTF::WrapUnique(handler));
+  handler->ChangeState(WebRTCDataChannelHandlerClient::kReadyStateOpen);
+  channel->ContextDestroyed(nullptr);
+  channel->close();
+  EXPECT_EQ(String::FromUTF8("closed"), channel->readyState());
+}
+
 }  // namespace blink