Add CDM FileIO tests.

media/cdm/ppapi/clear_key_cdm.cc

 // Copyright 2013 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 "media/cdm/ppapi/clear_key_cdm.h"
 
 #include <algorithm>
 #include <sstream>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/debug/trace_event.h"
 #include "base/logging.h"
 #include "base/time/time.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decrypt_config.h"
+#include "media/cdm/ppapi/cdm_file_io_test.h"
 #include "media/cdm/ppapi/cdm_video_decoder.h"
 
 #if defined(CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER)
 #include "base/basictypes.h"
 const int64 kNoTimestamp = kint64min;
 #endif  // CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER
 
 #if defined(CLEAR_KEY_CDM_USE_FFMPEG_DECODER)
 #include "base/at_exit.h"
 #include "base/files/file_path.h"
@@ skipping 33 matching lines @@
 
 const char kClearKeyCdmVersion[] = "0.1.0.1";
 const char kExternalClearKeyKeySystem[] = "org.chromium.externalclearkey";
 const int64 kSecondsPerMinute = 60;
 const int64 kMsPerSecond = 1000;
 const int64 kInitialTimerDelayMs = 200;
 const int64 kMaxTimerDelayMs = 1 * kSecondsPerMinute * kMsPerSecond;
 // Heart beat message header. If a key message starts with |kHeartBeatHeader|,
 // it's a heart beat message. Otherwise, it's a key request.
 const char kHeartBeatHeader[] = "HEARTBEAT";
+// CDM file IO test result header.
+const char kCdmFileIOTestResultHeader[] = "CDMFILEIOTESTRESULT";
 
 // Copies |input_buffer| into a media::DecoderBuffer. If the |input_buffer| is
 // empty, an empty (end-of-stream) media::DecoderBuffer is returned.
 static scoped_refptr<media::DecoderBuffer> CopyDecoderBufferFrom(
     const cdm::InputBuffer& input_buffer) {
   if (!input_buffer.data) {
     DCHECK(!input_buffer.data_size);
     return media::DecoderBuffer::CreateEOSBuffer();
   }
 
@@ skipping 17 matching lines @@
       input_buffer.data_offset,
       subsamples));
 
   output_buffer->set_decrypt_config(decrypt_config.Pass());
   output_buffer->set_timestamp(
       base::TimeDelta::FromMicroseconds(input_buffer.timestamp));
 
   return output_buffer;
 }
 
+std::string GetCdmFileIOTestResultMessage(bool success) {
+  std::string message(kCdmFileIOTestResultHeader);
+  message += success ? '1' : '0';
+  return message;
+}
+
 template<typename Type>
 class ScopedResetter {
  public:
   explicit ScopedResetter(Type* object) : object_(object) {}
   ~ScopedResetter() { object_->Reset(); }
 
  private:
   Type* const object_;
 };
 
@@ skipping 112 matching lines @@
 
   if (client_.status() != (Client::kKeyMessage | Client::kSetSessionId)) {
     // Use values returned to client if possible.
     host_->SendKeyError(client_.session_id().data(),
                         client_.session_id().size(),
                         static_cast<cdm::MediaKeyError>(client_.error_code()),
                         client_.system_code());
     return cdm::kSessionError;
   }
 
+  std::string session_id = client_.session_id();
+
   host_->SendKeyMessage(
-      client_.session_id().data(), client_.session_id().size(),
+      session_id.data(), session_id.size(),
       reinterpret_cast<const char*>(&client_.key_message()[0]),
       client_.key_message().size(),
       client_.default_url().data(), client_.default_url().size());
 
-  // Only save the latest session ID for heartbeat messages.
-  heartbeat_session_id_ = client_.session_id();
+  // Save the latest session ID for heartbeat and file IO test messages.
+  last_session_id_ = client_.session_id();
+
+  StartCdmFileIOTest();
 
   return cdm::kSuccess;
 }
 
 cdm::Status ClearKeyCdm::AddKey(const char* session_id,
                                 uint32_t session_id_size,
                                 const uint8_t* key,
                                 uint32_t key_size,
                                 const uint8_t* key_id,
                                 uint32_t key_id_size) {
@@ skipping 45 matching lines @@
     heartbeat_message = next_heartbeat_message_;
   } else {
     heartbeat_message = "ERROR: Invalid timer context found!";
   }
 
   // This URL is only used for testing the code path for defaultURL.
   // There is no service at this URL, so applications should ignore it.
   const char url[] = "http://test.externalclearkey.chromium.org";
 
   host_->SendKeyMessage(
-      heartbeat_session_id_.data(), heartbeat_session_id_.size(),
+      last_session_id_.data(), last_session_id_.size(),
       heartbeat_message.data(), heartbeat_message.size(),
       url, arraysize(url) - 1);
 
   ScheduleNextHeartBeat();
 }
 
 static void CopyDecryptResults(
     media::Decryptor::Status* status_copy,
     scoped_refptr<media::DecoderBuffer>* buffer_copy,
     media::Decryptor::Status status,
@@ skipping 273 matching lines @@
 
   int samples_generated = GenerateFakeAudioFramesFromDuration(
       timestamp_in_microseconds - CurrentTimeStampInMicroseconds(),
       audio_frames);
   total_samples_generated_ += samples_generated;
 
   return samples_generated == 0 ? cdm::kNeedMoreData : cdm::kSuccess;
 }
 #endif  // CLEAR_KEY_CDM_USE_FAKE_AUDIO_DECODER
 
+void ClearKeyCdm::StartCdmFileIOTest() {
+  cdm_file_io_test_.reset(new CdmFileIOTest());
+
+  cdm::CdmFileIO* cdm_file_io = host_->GetCdmFileIO(cdm_file_io_test_.get());

[ddorwin, 2013/12/04 05:27:08]

Could/should we encapsulate this in the test file too? Why does the main CDM
need to know these details? I guest that could add Host versioning issues to
that file.

[xhwang, 2013/12/10 01:24:25]

Done.

+  if (!cdm_file_io) {
+    DVLOG(1) << "Cannot get CDM File IO interface";
+    OnCdmFileIOTestComplete(false);
+    return;
+  }
+
+  cdm_file_io_test_->Run(cdm_file_io,
+                         base::Bind(&ClearKeyCdm::OnCdmFileIOTestComplete,
+                                    base::Unretained(this)));
+}
+
+void ClearKeyCdm::OnCdmFileIOTestComplete(bool success) {
+  DVLOG(1) << __FUNCTION__ << ": " << success;
+  std::string message = GetCdmFileIOTestResultMessage(success);
+  host_->SendKeyMessage(last_session_id_.data(), last_session_id_.size(),
+                        message.data(), message.size(),
+                        NULL, 0);
+  cdm_file_io_test_.reset();
+}
+
 }  // namespace media