Add regression test for 301510

content/test/plugin/plugin_geturl_test.cc

 // Copyright (c) 2011 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 "content/test/plugin/plugin_geturl_test.h"
 
 #include <stdio.h>
 
 #include "base/basictypes.h"
 #include "base/file_util.h"
@@ skipping 11 matching lines @@
 // The identifier for the fetched url stream.
 #define FETCHED_URL_STREAM_ID 2
 
 // url for testing GetURL with a bogus URL.
 #define BOGUS_URL "bogoproto:///x:/asdf.xysdhffieasdf.asdhj/"
 
 // url for testing redirect notifications sent to plugins.
 #define REDIRECT_SRC_URL \
     "http://mock.http/npapi/plugin_read_page_redirect_src.html"
 
-// The notification id for the redirect notification url.
-#define REDIRECT_SRC_URL_NOTIFICATION_ID 4
+// The notification id for the redirect notification url that we will cancel.
+#define REDIRECT_SRC_URL_NOTIFICATION_CANCEL_ID 4
+
+// The notification id for the redirect notification url that we will accept.
+#define REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID 5
 
 // The identifier for the bogus url stream.
 #define BOGUS_URL_STREAM_ID 3
 
 // 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),
     expect_404_response_(false),
     npn_evaluate_context_(false),
     handle_url_redirects_(false),
-    received_url_redirect_notification_(false),
+    received_url_redirect_cancel_notification_(false),
+    received_url_redirect_allow_notification_(false),
     check_cookies_(false) {
 }
 
 PluginGetURLTest::~PluginGetURLTest() {}
 
 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) {
@@ skipping 45 matching lines @@
       HostFunctions()->geturl(id(), fail_write_url_.c_str(), NULL);
       return NPERR_NO_ERROR;
     } else if (!referrer_target_url_.empty()) {
       HostFunctions()->pushpopupsenabledstate(id(), true);
       HostFunctions()->geturl(id(), referrer_target_url_.c_str(), "_blank");
       HostFunctions()->poppopupsenabledstate(id());
       return NPERR_NO_ERROR;
     } else if (handle_url_redirects_) {
       HostFunctions()->geturlnotify(
           id(), REDIRECT_SRC_URL, NULL,
-          reinterpret_cast<void*>(REDIRECT_SRC_URL_NOTIFICATION_ID));
+          reinterpret_cast<void*>(REDIRECT_SRC_URL_NOTIFICATION_CANCEL_ID));
       return NPERR_NO_ERROR;
     } else if (check_cookies_) {
       HostFunctions()->geturlnotify(
           id(),
           "plugin_ref_target_page.html",
           NULL,
           reinterpret_cast<void*>(SELF_URL_STREAM_ID));
       return NPERR_NO_ERROR;
     }
 
@@ skipping 82 matching lines @@
 
         test_file_ = file_util::OpenFile(path, "r");
         if (!test_file_) {
           SetError("Could not open source file");
         }
       }
       break;
     case BOGUS_URL_STREAM_ID:
       SetError("Unexpected NewStream for BOGUS_URL");
       break;
-    case REDIRECT_SRC_URL_NOTIFICATION_ID:
+    case REDIRECT_SRC_URL_NOTIFICATION_CANCEL_ID:
       SetError("Should not redirect to URL when plugin denied it.");
       break;
+    case REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID:
+      break;
     default:
       SetError("Unexpected NewStream callback");
       break;
   }
   return NPERR_NO_ERROR;
 }
 
 int32 PluginGetURLTest::WriteReady(NPStream *stream) {
   if (test_completed()) {
     return PluginTest::WriteReady(stream);
@@ skipping 54 matching lines @@
         // bytes.  But this is not likely.
         if (bytes != len)
           SetError("Did not read correct bytelength from source file");
         if (memcmp(read_buffer, buffer, len))
           SetError("Content mismatch between data and source!");
       }
       break;
     case BOGUS_URL_STREAM_ID:
       SetError("Unexpected write callback for BOGUS_URL");
       break;
+    case REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID:
+      break;
     default:
       SetError("Unexpected write callback");
       break;
   }
   // Pretend that we took all the data.
   return len;
 }
 
 
 NPError PluginGetURLTest::DestroyStream(NPStream *stream, NPError reason) {
@@ skipping 26 matching lines @@
     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 REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID:
+      break;
     case FETCHED_URL_STREAM_ID:
       {
         char read_buffer[STREAM_CHUNK];
         size_t bytes = fread(read_buffer, 1, sizeof(read_buffer), test_file_);
         if (bytes != 0)
           SetError("Data and source mismatch on length");
         file_util::CloseFile(test_file_);
       }
       break;
     default:
@@ skipping 54 matching lines @@
       tests_in_progress_--;
       break;
     case BOGUS_URL_STREAM_ID:
       if (reason != NPRES_NETWORK_ERR) {
         std::string err = "BOGUS_URL received unexpected URLNotify status: ";
         err.append(base::IntToString(reason));
         SetError(err);
       }
       tests_in_progress_--;
       break;
-    case REDIRECT_SRC_URL_NOTIFICATION_ID: {
-      if (!received_url_redirect_notification_) {
-        SetError("Failed to receive URLRedirect notification");
+    case REDIRECT_SRC_URL_NOTIFICATION_CANCEL_ID: {
+      if (!received_url_redirect_cancel_notification_) {
+        SetError("Failed to receive URLRedirect notification for cancel");
+      }
+      if (reason != NPRES_NETWORK_ERR)  {
+        SetError("Redirected URL didn't get canceled");
+      }
+      break;
+    }
+    case REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID: {
+      if (!received_url_redirect_allow_notification_) {
+        SetError("Failed to receive URLRedirect notification for allow");
+      }
+      if (reason != NPRES_DONE) {
+        SetError("Redirected URL didn't complete successfully");
       }
       tests_in_progress_--;
       break;
     }
     default:
       SetError("Unexpected NewStream callback");
       break;
   }
 
   if (tests_in_progress_ == 0)
       SignalTestCompleted();
 }
 
 void PluginGetURLTest::URLRedirectNotify(const char* url,
                                          int32_t status,
                                          void* notify_data) {
-  if (!base::strcasecmp(url, "http://mock.http/npapi/plugin_read_page.html")) {
-    received_url_redirect_notification_ = true;
-    // Disallow redirect notification.
-    HostFunctions()->urlredirectresponse(id(), notify_data, false);
+  unsigned long stream_id = reinterpret_cast<unsigned long>(notify_data);
+  if (stream_id == REDIRECT_SRC_URL_NOTIFICATION_CANCEL_ID) {
+    if (!base::strcasecmp(url,
+                          "http://mock.http/npapi/plugin_read_page.html")) {
+      received_url_redirect_cancel_notification_ = true;
+      // Disallow redirect notification.
+      HostFunctions()->urlredirectresponse(id(), notify_data, false);
+
+      // Now start a request that we will allow to redirect.
+      HostFunctions()->geturlnotify(
+        id(), REDIRECT_SRC_URL, NULL,
+        reinterpret_cast<void*>(REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID));
+    }
+  } else if (stream_id == REDIRECT_SRC_URL_NOTIFICATION_ALLOW_ID) {
+    received_url_redirect_allow_notification_ = true;
+    HostFunctions()->urlredirectresponse(id(), notify_data, true);
   }
 }
 
 } // namespace NPAPIClient