chromeos/process_proxy/process_proxy_registry.h

   1 // Copyright (c) 2012 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 CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_
   6 #define CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_
   7
   8 #include <map>
   9
  10 #include "base/callback.h"
  11 #include "base/lazy_instance.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/memory/ref_counted.h"
  14 #include "base/threading/non_thread_safe.h"
  15 #include "base/threading/thread.h"
  16 #include "chromeos/chromeos_export.h"
  17 #include "chromeos/process_proxy/process_proxy.h"
  18
  19 namespace chromeos {
  20
  21 typedef base::Callback<void(pid_t, const std::string&, const std::string&)>
  22       ProcessOutputCallbackWithPid;
  23
  24 // Keeps track of all created ProcessProxies. It is created lazily and should
  25 // live on a single thread (where all methods must be called).
  26 class CHROMEOS_EXPORT ProcessProxyRegistry : public base::NonThreadSafe {
  27  public:
  28   // Info we need about a ProcessProxy instance.
  29   struct ProcessProxyInfo {
  30     scoped_refptr<ProcessProxy> proxy;
  31     scoped_ptr<base::Thread> watcher_thread;
  32     ProcessOutputCallbackWithPid callback;
  33     pid_t process_id;
  34
  35     ProcessProxyInfo();
  36     // This is to make map::insert happy, we don't init anything.
  37     ProcessProxyInfo(const ProcessProxyInfo& other);
  38     ~ProcessProxyInfo();
  39   };
  40
  41   static ProcessProxyRegistry* Get();
  42
  43   // Starts new ProcessProxy (which starts new process).
  44   bool OpenProcess(const std::string& command, pid_t* pid,
  45                    const ProcessOutputCallbackWithPid& callback);
  46   // Sends data to the process with id |pid|.
  47   bool SendInput(pid_t pid, const std::string& data);
  48   // Stops the process with id |pid|.
  49   bool CloseProcess(pid_t pid);
  50   // Reports terminal resize to process proxy.
  51   bool OnTerminalResize(pid_t pid, int width, int height);
  52
  53   // Currently used for testing.
  54   void SetOutputCallback(const ProcessOutputCallback& callback);
  55
  56  private:
  57   friend struct ::base::DefaultLazyInstanceTraits<ProcessProxyRegistry>;
  58
  59   ProcessProxyRegistry();
  60   ~ProcessProxyRegistry();
  61
  62   // Gets called when output gets detected.
  63   void OnProcessOutput(pid_t pid,
  64                        ProcessOutputType type,
  65                        const std::string& data);
  66
  67   // Map of all existing ProcessProxies.
  68   std::map<pid_t, ProcessProxyInfo> proxy_map_;
  69
  70   DISALLOW_COPY_AND_ASSIGN(ProcessProxyRegistry);
  71 };
  72
  73 }  // namespace chromeos
  74
  75 #endif  // CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_