Home | History | Annotate | Download | only in importer 
      1 // Copyright (c) 2011 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 #ifndef CHROME_BROWSER_IMPORTER_FIREFOX_IMPORTER_UNITTEST_UTILS_H_
      6 #define CHROME_BROWSER_IMPORTER_FIREFOX_IMPORTER_UNITTEST_UTILS_H_
      7 #pragma once
      8 
      9 #include "base/basictypes.h"
     10 #include "base/memory/scoped_ptr.h"
     11 #include "base/process_util.h"
     12 #include "chrome/browser/importer/nss_decryptor.h"
     13 
     14 class FFDecryptorServerChannelListener;
     15 namespace IPC {
     16   class Channel;
     17 }  // namespace IPC
     18 class MessageLoopForIO;
     19 
     20 // On OS X NSSDecryptor needs to run in a separate process. To allow us to use
     21 // the same unit test on all platforms we use a proxy class which spawns a
     22 // child process to do decryption on OS X, and calls through directly
     23 // to NSSDecryptor on other platforms.
     24 // On OS X:
     25 // 2 IPC messages are sent for every method of NSSDecryptor, one containing the
     26 // input arguments sent from Server->Child and one coming back containing
     27 // the return argument e.g.
     28 //
     29 // -> Msg_Decryptor_Init(dll_path, db_path)
     30 // <- Msg_Decryptor_InitReturnCode(bool)
     31 class FFUnitTestDecryptorProxy {
     32  public:
     33   FFUnitTestDecryptorProxy();
     34   virtual ~FFUnitTestDecryptorProxy();
     35 
     36   // Initialize a decryptor, returns true if the object was
     37   // constructed successfully.
     38   bool Setup(const FilePath& nss_path);
     39 
     40   // This match the parallel functions in NSSDecryptor.
     41   bool DecryptorInit(const FilePath& dll_path, const FilePath& db_path);
     42   string16 Decrypt(const std::string& crypt);
     43 
     44  private:
     45 #if defined(OS_MACOSX)
     46   // Blocks until either a timeout is reached, or until the client process
     47   // responds to an IPC message.
     48   // Returns true if a reply was received successfully and false if the
     49   // the operation timed out.
     50   bool WaitForClientResponse();
     51 
     52   base::ProcessHandle child_process_;
     53   scoped_ptr<IPC::Channel> channel_;
     54   scoped_ptr<FFDecryptorServerChannelListener> listener_;
     55   scoped_ptr<MessageLoopForIO> message_loop_;
     56 #else
     57   NSSDecryptor decryptor_;
     58 #endif  // !OS_MACOSX
     59   DISALLOW_COPY_AND_ASSIGN(FFUnitTestDecryptorProxy);
     60 };
     61 
     62 // On Non-OSX platforms FFUnitTestDecryptorProxy simply calls through to
     63 // NSSDecryptor.
     64 #if !defined(OS_MACOSX)
     65 FFUnitTestDecryptorProxy::FFUnitTestDecryptorProxy() {
     66 }
     67 
     68 FFUnitTestDecryptorProxy::~FFUnitTestDecryptorProxy() {
     69 }
     70 
     71 bool FFUnitTestDecryptorProxy::Setup(const FilePath& nss_path) {
     72   return true;
     73 }
     74 
     75 bool FFUnitTestDecryptorProxy::DecryptorInit(const FilePath& dll_path,
     76                                              const FilePath& db_path) {
     77   return decryptor_.Init(dll_path, db_path);
     78 }
     79 
     80 string16 FFUnitTestDecryptorProxy::Decrypt(const std::string& crypt) {
     81   return decryptor_.Decrypt(crypt);
     82 }
     83 #endif  // !OS_MACOSX
     84 
     85 #endif  // CHROME_BROWSER_IMPORTER_FIREFOX_IMPORTER_UNITTEST_UTILS_H_
     86