Home | History | Annotate | Download | only in plugin 
      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 that data, which we got, is right.
     95   const char* data = static_cast<const char*>(buffer);
     96   for (int32 i = 0; i < len; ++i) {
     97     int cur_offset = offset + i;
     98     char expected = '0' + cur_offset;
     99     if (data[i] != expected) {
    100       SetError("Content mismatch between data and source!");
    101       break;
    102     }
    103   }
    104   if (requested_ranges_.empty())
    105     SignalTestCompleted();
    106 
    107   return len;
    108 }
    109 
    110 }  // namespace NPAPIClient
    111 
    112