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 19namespace chromeos { 20 21typedef 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). 26class 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_ 76