1a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// Copyright 2014 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) 5a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "extensions/browser/api/socket/socket.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 87d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/lazy_instance.h" 9a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "extensions/browser/api/api_resource_manager.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/address_list.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/ip_endpoint.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/socket.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kSocketTypeNotSupported[] = "Socket type does not support this API"; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 20a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)static base::LazyInstance< 21a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) BrowserContextKeyedAPIFactory<ApiResourceManager<Socket> > > g_factory = 22a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) LAZY_INSTANCE_INITIALIZER; 237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// static 257d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)template <> 26a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)BrowserContextKeyedAPIFactory<ApiResourceManager<Socket> >* 277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)ApiResourceManager<Socket>::GetFactoryInstance() { 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return g_factory.Pointer(); 297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} 307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Socket::Socket(const std::string& owner_extension_id) 32a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) : ApiResource(owner_extension_id), is_connected_(false) {} 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Socket::~Socket() { 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Derived destructors should make sure the socket has been closed. 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!is_connected_); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void Socket::Write(scoped_refptr<net::IOBuffer> io_buffer, 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int byte_count, 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) { 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!callback.is_null()); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write_queue_.push(WriteRequest(io_buffer, byte_count, callback)); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteData(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void Socket::WriteData() { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // IO is pending. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (io_buffer_write_.get()) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteRequest& request = write_queue_.front(); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(request.byte_count >= request.bytes_written); 55a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) io_buffer_write_ = new net::WrappedIOBuffer(request.io_buffer->data() + 56a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) request.bytes_written); 57a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int result = 58a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) WriteImpl(io_buffer_write_.get(), 59a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) request.byte_count - request.bytes_written, 60a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::Bind(&Socket::OnWriteComplete, base::Unretained(this))); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result != net::ERR_IO_PENDING) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OnWriteComplete(result); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void Socket::OnWriteComplete(int result) { 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io_buffer_write_ = NULL; 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteRequest& request = write_queue_.front(); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result >= 0) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request.bytes_written += result; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (request.bytes_written < request.byte_count) { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteData(); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(request.bytes_written == request.byte_count); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = request.bytes_written; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request.callback.Run(result); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write_queue_.pop(); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!write_queue_.empty()) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteData(); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 88a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)bool Socket::SetKeepAlive(bool enable, int delay) { return false; } 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 90a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)bool Socket::SetNoDelay(bool no_delay) { return false; } 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 92a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)int Socket::Listen(const std::string& address, 93a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int port, 94a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int backlog, 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string* error_msg) { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *error_msg = kSocketTypeNotSupported; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return net::ERR_FAILED; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void Socket::Accept(const AcceptCompletionCallback& callback) { 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) callback.Run(net::ERR_FAILED, NULL); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Socket::StringAndPortToIPEndPoint(const std::string& ip_address_str, 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int port, 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::IPEndPoint* ip_end_point) { 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(ip_end_point); 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::IPAddressNumber ip_number; 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!net::ParseIPLiteralToNumber(ip_address_str, &ip_number)) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *ip_end_point = net::IPEndPoint(ip_number, port); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Socket::StringAndPortToAddressList(const std::string& ip_address_str, 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int port, 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::AddressList* address_list) { 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(address_list); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::IPAddressNumber ip_number; 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!net::ParseIPLiteralToNumber(ip_address_str, &ip_number)) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *address_list = net::AddressList::CreateFromIPAddress(ip_number, port); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void Socket::IPEndPointToStringAndPort(const net::IPEndPoint& address, 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string* ip_address_str, 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* port) { 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(ip_address_str); 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(port); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *ip_address_str = address.ToStringWithoutPort(); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ip_address_str->empty()) { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *port = 0; 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *port = address.port(); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Socket::WriteRequest::WriteRequest(scoped_refptr<net::IOBuffer> io_buffer, 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int byte_count, 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback) 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : io_buffer(io_buffer), 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) byte_count(byte_count), 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) callback(callback), 148a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) bytes_written(0) {} 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 150a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)Socket::WriteRequest::~WriteRequest() {} 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 153