Fixes a crash caused due to a call to NPP_DestroyStream occuring in the conte...

webkit/glue/plugins/test/plugin_geturl_test.cc

 // Copyright (c) 2006-2008 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 "webkit/glue/plugins/test/plugin_geturl_test.h"
 
 #include <stdio.h>
 
 #include "base/basictypes.h"
 #include "base/file_util.h"
@@ skipping 16 matching lines @@
 
 // The maximum chunk size of stream data.
 #define STREAM_CHUNK 197
 
 namespace NPAPIClient {
 
 PluginGetURLTest::PluginGetURLTest(NPP id, NPNetscapeFuncs *host_functions)
   : PluginTest(id, host_functions),
     tests_started_(false),
     tests_in_progress_(0),
-    test_file_(NULL) {
+    test_file_(NULL),
+    expect_404_response_(false),
+    npn_evaluate_context_(false) {
+}
+
+NPError PluginGetURLTest::New(uint16 mode, int16 argc, const char* argn[],
+                              const char* argv[], NPSavedData* saved) {
+  const char* page_not_found_url = GetArgValue("page_not_found_url", argc,
+                                               argn, argv);
+  if (page_not_found_url) {
+    page_not_found_url_ = page_not_found_url;
+    expect_404_response_ = true;
+  }
+
+  return PluginTest::New(mode, argc, argn, argv, saved);
 }
 
 NPError PluginGetURLTest::SetWindow(NPWindow* pNPWindow) {
   if (!tests_started_) {
     tests_started_ = true;
 
     tests_in_progress_++;
 
+    if (expect_404_response_) {
+      HostFunctions()->geturl(id(), page_not_found_url_.c_str(), NULL);
+      return NPERR_NO_ERROR;
+    }
+
     std::string url = SELF_URL;
     HostFunctions()->geturlnotify(id(), url.c_str(), NULL,
                                   reinterpret_cast<void*>(SELF_URL_STREAM_ID));
 
     tests_in_progress_++;
     std::string bogus_url = BOGUS_URL;
     HostFunctions()->geturlnotify(id(), bogus_url.c_str(), NULL,
                                   reinterpret_cast<void*>(BOGUS_URL_STREAM_ID));
   }
   return NPERR_NO_ERROR;
 }
 
 NPError PluginGetURLTest::NewStream(NPMIMEType type, NPStream* stream,
                               NPBool seekable, uint16* stype) {
   if (stream == NULL)
     SetError("NewStream got null stream");
 
+  if (test_completed()) {
+    return PluginTest::NewStream(type, stream, seekable, stype);
+  }
+
   COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData),
                  cast_validity_check);
+
+  if (expect_404_response_) {
+    NPObject *window_obj = NULL;
+    HostFunctions()->getvalue(id(), NPNVWindowNPObject, &window_obj);
+    if (!window_obj) {
+      SetError("Failed to get NPObject for plugin instance2");
+      SignalTestCompleted();
+      return NPERR_NO_ERROR;
+    }
+
+    std::string script = "javascript:alert('Hi there from plugin');";
+    NPString script_string;
+    script_string.UTF8Characters = script.c_str();
+    script_string.UTF8Length = static_cast<unsigned int>(script.length());
+    NPVariant result_var;
+
+    npn_evaluate_context_ = true;
+    HostFunctions()->evaluate(id(), window_obj, &script_string, &result_var);
+    npn_evaluate_context_ = false;
+    return NPERR_NO_ERROR;
+  }
+
   unsigned long stream_id = reinterpret_cast<unsigned long>(
       stream->notifyData);
+
   switch (stream_id) {
     case SELF_URL_STREAM_ID:
       break;
     case FETCHED_URL_STREAM_ID:
       {
         std::string filename = self_url_;
         if (filename.find("file:///", 0) != 0) {
           SetError("Test expects a file-url.");
           break;
         }
@@ skipping 10 matching lines @@
       SetError("Unexpected NewStream for BOGUS_URL");
       break;
     default:
       SetError("Unexpected NewStream callback");
       break;
   }
   return NPERR_NO_ERROR;
 }
 
 int32 PluginGetURLTest::WriteReady(NPStream *stream) {
+  if (test_completed()) {
+    return PluginTest::WriteReady(stream);
+  }
+
   COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData),
                  cast_validity_check);
   unsigned long stream_id = reinterpret_cast<unsigned long>(
       stream->notifyData);
   if (stream_id == BOGUS_URL_STREAM_ID)
     SetError("Received WriteReady for BOGUS_URL");
 
   return STREAM_CHUNK;
 }
 
 int32 PluginGetURLTest::Write(NPStream *stream, int32 offset, int32 len,
                               void *buffer) {
+  if (test_completed()) {
+    return PluginTest::Write(stream, offset, len, buffer);
+  }
+
   if (stream == NULL)
     SetError("Write got null stream");
   if (len < 0 || len > STREAM_CHUNK)
     SetError("Write got bogus stream chunk size");
 
   COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData),
                  cast_validity_check);
   unsigned long stream_id = reinterpret_cast<unsigned long>(
       stream->notifyData);
   switch (stream_id) {
@@ skipping 18 matching lines @@
     default:
       SetError("Unexpected write callback");
       break;
   }
   // Pretend that we took all the data.
   return len;
 }
 
 
 NPError PluginGetURLTest::DestroyStream(NPStream *stream, NPError reason) {
+  if (test_completed()) {
+    return PluginTest::DestroyStream(stream, reason);
+  }
+
   if (stream == NULL)
     SetError("NewStream got null stream");
 
   COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData),
                  cast_validity_check);
+
+  if (expect_404_response_) {
+    if (npn_evaluate_context_) {
+      SetError("Received destroyStream in the context of NPN_Evaluate.");
+    }
+
+    SignalTestCompleted();
+    return NPERR_NO_ERROR;
+  }
+
   unsigned long stream_id =
       reinterpret_cast<unsigned long>(stream->notifyData);
   switch (stream_id) {
     case SELF_URL_STREAM_ID:
       // don't care
       break;
     case FETCHED_URL_STREAM_ID:
       {
         char read_buffer[STREAM_CHUNK];
         size_t bytes = fread(read_buffer, 1, sizeof(read_buffer), test_file_);
@@ skipping 65 matching lines @@
     default:
       SetError("Unexpected NewStream callback");
       break;
   }
 
   if (tests_in_progress_ == 0)
       SignalTestCompleted();
 }
 
 } // namespace NPAPIClient