OLD | NEW |
| (Empty) |
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 #include "chrome/browser/extensions/api/socket/socket.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/lazy_instance.h" | |
9 #include "chrome/browser/extensions/api/api_resource_manager.h" | |
10 #include "net/base/address_list.h" | |
11 #include "net/base/io_buffer.h" | |
12 #include "net/base/ip_endpoint.h" | |
13 #include "net/base/net_errors.h" | |
14 #include "net/socket/socket.h" | |
15 | |
16 namespace extensions { | |
17 | |
18 const char kSocketTypeNotSupported[] = "Socket type does not support this API"; | |
19 | |
20 static base::LazyInstance< | |
21 BrowserContextKeyedAPIFactory<ApiResourceManager<Socket> > > g_factory = | |
22 LAZY_INSTANCE_INITIALIZER; | |
23 | |
24 // static | |
25 template <> | |
26 BrowserContextKeyedAPIFactory<ApiResourceManager<Socket> >* | |
27 ApiResourceManager<Socket>::GetFactoryInstance() { | |
28 return g_factory.Pointer(); | |
29 } | |
30 | |
31 Socket::Socket(const std::string& owner_extension_id) | |
32 : ApiResource(owner_extension_id), is_connected_(false) { | |
33 } | |
34 | |
35 Socket::~Socket() { | |
36 // Derived destructors should make sure the socket has been closed. | |
37 DCHECK(!is_connected_); | |
38 } | |
39 | |
40 void Socket::Write(scoped_refptr<net::IOBuffer> io_buffer, | |
41 int byte_count, | |
42 const CompletionCallback& callback) { | |
43 DCHECK(!callback.is_null()); | |
44 write_queue_.push(WriteRequest(io_buffer, byte_count, callback)); | |
45 WriteData(); | |
46 } | |
47 | |
48 void Socket::WriteData() { | |
49 // IO is pending. | |
50 if (io_buffer_write_.get()) | |
51 return; | |
52 | |
53 WriteRequest& request = write_queue_.front(); | |
54 | |
55 DCHECK(request.byte_count >= request.bytes_written); | |
56 io_buffer_write_ = new net::WrappedIOBuffer( | |
57 request.io_buffer->data() + request.bytes_written); | |
58 int result = WriteImpl( | |
59 io_buffer_write_.get(), | |
60 request.byte_count - request.bytes_written, | |
61 base::Bind(&Socket::OnWriteComplete, base::Unretained(this))); | |
62 | |
63 if (result != net::ERR_IO_PENDING) | |
64 OnWriteComplete(result); | |
65 } | |
66 | |
67 void Socket::OnWriteComplete(int result) { | |
68 io_buffer_write_ = NULL; | |
69 | |
70 WriteRequest& request = write_queue_.front(); | |
71 | |
72 if (result >= 0) { | |
73 request.bytes_written += result; | |
74 if (request.bytes_written < request.byte_count) { | |
75 WriteData(); | |
76 return; | |
77 } | |
78 DCHECK(request.bytes_written == request.byte_count); | |
79 result = request.bytes_written; | |
80 } | |
81 | |
82 request.callback.Run(result); | |
83 write_queue_.pop(); | |
84 | |
85 if (!write_queue_.empty()) | |
86 WriteData(); | |
87 } | |
88 | |
89 bool Socket::SetKeepAlive(bool enable, int delay) { | |
90 return false; | |
91 } | |
92 | |
93 bool Socket::SetNoDelay(bool no_delay) { | |
94 return false; | |
95 } | |
96 | |
97 int Socket::Listen(const std::string& address, int port, int backlog, | |
98 std::string* error_msg) { | |
99 *error_msg = kSocketTypeNotSupported; | |
100 return net::ERR_FAILED; | |
101 } | |
102 | |
103 void Socket::Accept(const AcceptCompletionCallback& callback) { | |
104 callback.Run(net::ERR_FAILED, NULL); | |
105 } | |
106 | |
107 // static | |
108 bool Socket::StringAndPortToIPEndPoint(const std::string& ip_address_str, | |
109 int port, | |
110 net::IPEndPoint* ip_end_point) { | |
111 DCHECK(ip_end_point); | |
112 net::IPAddressNumber ip_number; | |
113 if (!net::ParseIPLiteralToNumber(ip_address_str, &ip_number)) | |
114 return false; | |
115 | |
116 *ip_end_point = net::IPEndPoint(ip_number, port); | |
117 return true; | |
118 } | |
119 | |
120 bool Socket::StringAndPortToAddressList(const std::string& ip_address_str, | |
121 int port, | |
122 net::AddressList* address_list) { | |
123 DCHECK(address_list); | |
124 net::IPAddressNumber ip_number; | |
125 if (!net::ParseIPLiteralToNumber(ip_address_str, &ip_number)) | |
126 return false; | |
127 | |
128 *address_list = net::AddressList::CreateFromIPAddress(ip_number, port); | |
129 return true; | |
130 } | |
131 | |
132 void Socket::IPEndPointToStringAndPort(const net::IPEndPoint& address, | |
133 std::string* ip_address_str, | |
134 int* port) { | |
135 DCHECK(ip_address_str); | |
136 DCHECK(port); | |
137 *ip_address_str = address.ToStringWithoutPort(); | |
138 if (ip_address_str->empty()) { | |
139 *port = 0; | |
140 } else { | |
141 *port = address.port(); | |
142 } | |
143 } | |
144 | |
145 Socket::WriteRequest::WriteRequest(scoped_refptr<net::IOBuffer> io_buffer, | |
146 int byte_count, | |
147 const CompletionCallback& callback) | |
148 : io_buffer(io_buffer), | |
149 byte_count(byte_count), | |
150 callback(callback), | |
151 bytes_written(0) { | |
152 } | |
153 | |
154 Socket::WriteRequest::~WriteRequest() { } | |
155 | |
156 } // namespace extensions | |
OLD | NEW |