content/test/plugin/plugin_request_read_test.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/test/plugin/plugin_request_read_test.h"
   6
   7 #include "base/basictypes.h"
   8
   9 namespace NPAPIClient {
  10
  11 PluginRequestReadTest::PluginRequestReadTest(NPP id,
  12                                              NPNetscapeFuncs* host_functions)
  13     : PluginTest(id, host_functions),
  14       tests_started_(false),
  15       read_requested_(false) {
  16 }
  17
  18 PluginRequestReadTest::~PluginRequestReadTest() {
  19 }
  20
  21 NPError PluginRequestReadTest::New(uint16 mode, int16 argc, const char* argn[],
  22                              const char* argv[], NPSavedData* saved) {
  23   url_to_request_ = GetArgValue("url_to_request", argc, argn, argv);
  24   return PluginTest::New(mode, argc, argn, argv, saved);
  25 }
  26
  27 NPError PluginRequestReadTest::SetWindow(NPWindow* window) {
  28   if (!tests_started_) {
  29     tests_started_ = true;
  30     NPError result = HostFunctions()->geturl(id(),
  31                                              url_to_request_.c_str(),
  32                                              NULL);
  33     if (result != NPERR_NO_ERROR)
  34       SetError("Failed request anURL.");
  35   }
  36   return PluginTest::SetWindow(window);
  37 }
  38
  39 NPError PluginRequestReadTest::NewStream(NPMIMEType type, NPStream* stream,
  40                                          NPBool seekable, uint16* stream_type) {
  41   *stream_type = NP_SEEK;
  42   if (!read_requested_) {
  43     requested_ranges_.resize(1);
  44     requested_ranges_[0].offset = 4;
  45     requested_ranges_[0].length = 6;
  46     requested_ranges_[0].next = NULL;
  47     NPError result = HostFunctions()->requestread(stream,
  48                                                   &requested_ranges_[0]);
  49     if (result != NPERR_NO_ERROR)
  50       SetError("Failed request read from stream.");
  51     read_requested_ = true;
  52   }
  53   return PluginTest::NewStream(type, stream, seekable, stream_type);
  54 }
  55
  56 NPError PluginRequestReadTest::DestroyStream(NPStream *stream, NPError reason) {
  57   if (!requested_ranges_.empty())
  58     SetError("Some requested ranges are not received!");
  59   SignalTestCompleted();
  60   return PluginTest::DestroyStream(stream, reason);
  61 }
  62
  63 int32 PluginRequestReadTest::WriteReady(NPStream* stream) {
  64   int32 result = 0;
  65   for (size_t i = 0; i < requested_ranges_.size(); ++i)
  66     result += requested_ranges_[i].length;
  67   return result;
  68 }
  69
  70 int32 PluginRequestReadTest::Write(NPStream* stream, int32 offset, int32 len,
  71                                    void* buffer) {
  72   std::vector<NPByteRange>::iterator it;
  73   // Remove received range (or sub-range) from requested_ranges_, and
  74   // verify that we have received proper data.
  75
  76   for (it = requested_ranges_.begin(); it != requested_ranges_.end(); ++it) {
  77     if (it->offset == offset)
  78       break;
  79   }
  80   if (it == requested_ranges_.end()) {
  81     // It is Ok for browser to write some data from start of the stream before
  82     // we've issued any read requests.
  83     return len;
  84   }
  85   // Shrink range to mark area we have just received.
  86   it->offset += len;
  87   if (static_cast<int32>(it->length) < len)
  88     it->length = 0;
  89   else
  90     it->length -= len;
  91   if (it->length == 0)
  92     requested_ranges_.erase(it);
  93
  94   // Verify the data we got is right.  We expect a string like "01234...".
  95   const char* data = static_cast<const char*>(buffer);
  96   for (int32 i = 0; i < len; ++i) {
  97     char expected = '0' + static_cast<char>(offset + i);
  98     if (data[i] != expected) {
  99       SetError("Content mismatch between data and source!");
 100       break;
 101     }
 102   }
 103   if (requested_ranges_.empty())
 104     SignalTestCompleted();
 105
 106   return len;
 107 }
 108
 109 }  // namespace NPAPIClient
 110