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) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/api/messaging/native_message_process_host.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/bind.h" 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/platform_file.h" 114e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "base/process/kill.h" 12ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch#include "base/threading/sequenced_worker_pool.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/extensions/api/messaging/native_messaging_host_manifest.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/api/messaging/native_process_launcher.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_version_info.h" 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "extensions/common/constants.h" 183551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)#include "extensions/common/features/feature.h" 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/file_stream.h" 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/io_buffer.h" 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/net_errors.h" 222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/net_util.h" 237dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Maximum message size in bytes for messages received from Native Messaging 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// hosts. Message size is limited mainly to prevent Chrome from crashing when 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// native application misbehaves (e.g. starts writing garbage to the pipe). 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const size_t kMaximumMessageSize = 1024 * 1024; 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Message header contains 4-byte integer size of the message. 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const size_t kMessageHeaderSize = 4; 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Size of the buffer to be allocated for each read. 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const size_t kReadBufferSize = 4096; 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kFailedToStartError[] = "Failed to start native messaging host."; 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kInvalidNameError[] = 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "Invalid native messaging host name specified."; 414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)const char kNativeHostExited[] = "Native host has exited."; 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kNotFoundError[] = "Specified native messaging host not found."; 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kForbiddenError[] = 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "Access to the specified native messaging host is forbidden."; 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kHostInputOuputError[] = 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "Error when communicating with the native messaging host."; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)NativeMessageProcessHost::NativeMessageProcessHost( 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtr<Client> weak_client_ui, 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& source_extension_id, 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& native_host_name, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int destination_port, 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<NativeProcessLauncher> launcher) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : weak_client_ui_(weak_client_ui), 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_extension_id_(source_extension_id), 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) native_host_name_(native_host_name), 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) destination_port_(destination_port), 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) launcher_(launcher.Pass()), 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_(false), 644e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) process_handle_(base::kNullProcessHandle), 654e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) read_file_(base::kInvalidPlatformFileValue), 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_pending_(false), 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_pending_(false) { 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // It's safe to use base::Unretained() here because NativeMessagePort always 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // deletes us on the IO thread. 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::BrowserThread::PostTask(content::BrowserThread::IO, FROM_HERE, 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&NativeMessageProcessHost::LaunchHostProcess, 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Unretained(this))); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)NativeMessageProcessHost::~NativeMessageProcessHost() { 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(std::string()); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)scoped_ptr<NativeMessageProcessHost> NativeMessageProcessHost::Create( 84ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch gfx::NativeView native_view, 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::WeakPtr<Client> weak_client_ui, 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& source_extension_id, 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& native_host_name, 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int destination_port) { 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return CreateWithLauncher(weak_client_ui, source_extension_id, 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) native_host_name, destination_port, 91ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch NativeProcessLauncher::CreateDefault(native_view)); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)scoped_ptr<NativeMessageProcessHost> 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)NativeMessageProcessHost::CreateWithLauncher( 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::WeakPtr<Client> weak_client_ui, 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& source_extension_id, 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const std::string& native_host_name, 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int destination_port, 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<NativeProcessLauncher> launcher) { 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<NativeMessageProcessHost> process(new NativeMessageProcessHost( 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) weak_client_ui, source_extension_id, native_host_name, 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) destination_port, launcher.Pass())); 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return process.Pass(); 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::LaunchHostProcess() { 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GURL origin(std::string(kExtensionScheme) + "://" + source_extension_id_); 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) launcher_->Launch(origin, native_host_name_, 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&NativeMessageProcessHost::OnHostProcessLaunched, 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Unretained(this))); 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::OnHostProcessLaunched( 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NativeProcessLauncher::LaunchResult result, 1224e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) base::ProcessHandle process_handle, 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::PlatformFile read_file, 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::PlatformFile write_file) { 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) switch (result) { 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) case NativeProcessLauncher::RESULT_INVALID_NAME: 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kInvalidNameError); 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) case NativeProcessLauncher::RESULT_NOT_FOUND: 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kNotFoundError); 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) case NativeProcessLauncher::RESULT_FORBIDDEN: 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kForbiddenError); 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) case NativeProcessLauncher::RESULT_FAILED_TO_START: 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kFailedToStartError); 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) case NativeProcessLauncher::RESULT_SUCCESS: 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) break; 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) process_handle_ = process_handle; 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_file_ = read_file; 146ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch 147ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch scoped_refptr<base::TaskRunner> task_runner( 148ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch content::BrowserThread::GetBlockingPool()-> 149ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch GetTaskRunnerWithShutdownBehavior( 150ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)); 151ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_stream_.reset(new net::FileStream( 153ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch read_file, base::PLATFORM_FILE_READ | base::PLATFORM_FILE_ASYNC, NULL, 154ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch task_runner)); 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_stream_.reset(new net::FileStream( 156ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch write_file, base::PLATFORM_FILE_WRITE | base::PLATFORM_FILE_ASYNC, NULL, 157ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch task_runner)); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WaitRead(); 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoWrite(); 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::Send(const std::string& json) { 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (closed_) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Allocate new buffer for the message. 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<net::IOBufferWithSize> buffer = 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new net::IOBufferWithSize(json.size() + kMessageHeaderSize); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Copy size and content of the message to the buffer. 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) COMPILE_ASSERT(sizeof(uint32) == kMessageHeaderSize, incorrect_header_size); 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) *reinterpret_cast<uint32*>(buffer->data()) = json.size(); 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) memcpy(buffer->data() + kMessageHeaderSize, json.data(), json.size()); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Push new message to the write queue. 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_queue_.push(buffer); 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Send() may be called before the host process is started. In that case the 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // message will be written when OnHostProcessLaunched() is called. If it's 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // already started then write the message now. 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (write_stream_) 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoWrite(); 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(OS_POSIX) 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::OnFileCanReadWithoutBlocking(int fd) { 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(fd, read_file_); 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoRead(); 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::OnFileCanWriteWithoutBlocking(int fd) { 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // !defined(OS_POSIX) 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::ReadNowForTesting() { 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoRead(); 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::WaitRead() { 2044e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) if (closed_) 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!read_pending_); 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // On POSIX FileStream::Read() uses blocking thread pool, so it's better to 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // wait for the file to become readable before calling DoRead(). Otherwise it 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // would always be consuming one thread in the thread pool. On Windows 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // FileStream uses overlapped IO, so that optimization isn't necessary there. 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if defined(OS_POSIX) 21490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::current()->WatchFileDescriptor( 21590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) read_file_, false /* persistent */, base::MessageLoopForIO::WATCH_READ, 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &read_watcher_, this); 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#else // defined(OS_POSIX) 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoRead(); 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif // defined(!OS_POSIX) 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::DoRead() { 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2254e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) while (!closed_ && !read_pending_) { 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_buffer_ = new net::IOBuffer(kReadBufferSize); 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int result = read_stream_->Read( 228868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) read_buffer_.get(), 229868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kReadBufferSize, 230868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::Bind(&NativeMessageProcessHost::OnRead, base::Unretained(this))); 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HandleReadResult(result); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::OnRead(int result) { 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(read_pending_); 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_pending_ = false; 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HandleReadResult(result); 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WaitRead(); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::HandleReadResult(int result) { 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (closed_) 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (result > 0) { 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProcessIncomingData(read_buffer_->data(), result); 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else if (result == net::ERR_IO_PENDING) { 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_pending_ = true; 2544e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) } else if (result == 0 || result == net::ERR_CONNECTION_RESET) { 2554e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // On Windows we get net::ERR_CONNECTION_RESET for a broken pipe, while on 2564e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Posix read() returns 0 in that case. 2574e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) Close(kNativeHostExited); 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LOG(ERROR) << "Error when reading from Native Messaging host: " << result; 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kHostInputOuputError); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::ProcessIncomingData( 2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const char* data, int data_size) { 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) incoming_data_.append(data, data_size); 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (true) { 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (incoming_data_.size() < kMessageHeaderSize) 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t message_size = 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) *reinterpret_cast<const uint32*>(incoming_data_.data()); 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (message_size > kMaximumMessageSize) { 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LOG(ERROR) << "Native Messaging host tried sending a message that is " 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) << message_size << " bytes long."; 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kHostInputOuputError); 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (incoming_data_.size() < message_size + kMessageHeaderSize) 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE, 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&Client::PostMessageFromNativeProcess, weak_client_ui_, 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) destination_port_, 290eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch incoming_data_.substr(kMessageHeaderSize, message_size))); 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) incoming_data_.erase(0, kMessageHeaderSize + message_size); 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::DoWrite() { 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (!write_pending_ && !closed_) { 300868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!current_write_buffer_.get() || 301868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) !current_write_buffer_->BytesRemaining()) { 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (write_queue_.empty()) 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) current_write_buffer_ = new net::DrainableIOBuffer( 305868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) write_queue_.front().get(), write_queue_.front()->size()); 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_queue_.pop(); 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 309868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) int result = 310868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) write_stream_->Write(current_write_buffer_.get(), 311868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) current_write_buffer_->BytesRemaining(), 312868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::Bind(&NativeMessageProcessHost::OnWritten, 313868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::Unretained(this))); 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HandleWriteResult(result); 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::HandleWriteResult(int result) { 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (result <= 0) { 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (result == net::ERR_IO_PENDING) { 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_pending_ = true; 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LOG(ERROR) << "Error when writing to Native Messaging host: " << result; 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Close(kHostInputOuputError); 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) current_write_buffer_->DidConsume(result); 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::OnWritten(int result) { 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(write_pending_); 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_pending_ = false; 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) HandleWriteResult(result); 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DoWrite(); 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void NativeMessageProcessHost::Close(const std::string& error_message) { 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!closed_) { 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) closed_ = true; 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) read_stream_.reset(); 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) write_stream_.reset(); 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE, 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&Client::CloseChannel, weak_client_ui_, 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) destination_port_, error_message)); 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3554e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 3564e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) if (process_handle_ != base::kNullProcessHandle) { 3574e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Kill the host process if necessary to make sure we don't leave zombies. 3584e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // On OSX base::EnsureProcessTerminated() may block, so we have to post a 3594e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // task on the blocking pool. 3604e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#if defined(OS_MACOSX) 3614e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) content::BrowserThread::PostBlockingPoolTask( 3624e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) FROM_HERE, base::Bind(&base::EnsureProcessTerminated, process_handle_)); 3634e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#else 3644e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) base::EnsureProcessTerminated(process_handle_); 3654e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#endif 3664e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) process_handle_ = base::kNullProcessHandle; 3674e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) } 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 371