15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#ifndef CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_ 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#define CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <map> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/lazy_instance.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/threading/non_thread_safe.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/thread.h" 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chromeos/chromeos_export.h" 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chromeos/process_proxy/process_proxy.h" 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace chromeos { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef base::Callback<void(pid_t, const std::string&, const std::string&)> 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessOutputCallbackWithPid; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Keeps track of all created ProcessProxies. It is created lazily and should 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// live on a single thread (where all methods must be called). 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class CHROMEOS_EXPORT ProcessProxyRegistry : public base::NonThreadSafe { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Info we need about a ProcessProxy instance. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct ProcessProxyInfo { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<ProcessProxy> proxy; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Thread> watcher_thread; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessOutputCallbackWithPid callback; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_t process_id; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessProxyInfo(); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is to make map::insert happy, we don't init anything. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessProxyInfo(const ProcessProxyInfo& other); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ProcessProxyInfo(); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static ProcessProxyRegistry* Get(); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts new ProcessProxy (which starts new process). 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool OpenProcess(const std::string& command, pid_t* pid, 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const ProcessOutputCallbackWithPid& callback); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sends data to the process with id |pid|. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool SendInput(pid_t pid, const std::string& data); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stops the process with id |pid|. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool CloseProcess(pid_t pid); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Reports terminal resize to process proxy. 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool OnTerminalResize(pid_t pid, int width, int height); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Currently used for testing. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOutputCallback(const ProcessOutputCallback& callback); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend struct ::base::DefaultLazyInstanceTraits<ProcessProxyRegistry>; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessProxyRegistry(); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ProcessProxyRegistry(); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Gets called when output gets detected. 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnProcessOutput(pid_t pid, 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessOutputType type, 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& data); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Map of all existing ProcessProxies. 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<pid_t, ProcessProxyInfo> proxy_map_; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ProcessProxyRegistry); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace chromeos 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // CHROMEOS_PROCESS_PROXY_PROCESS_PROXY_REGISTRY_H_ 76