OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "extensions/browser/api/socket/socket_api.h" | 5 #include "extensions/browser/api/socket/socket_api.h" |
6 | 6 |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/containers/hash_tables.h" | 10 #include "base/containers/hash_tables.h" |
11 #include "content/public/browser/browser_context.h" | 11 #include "content/public/browser/browser_context.h" |
12 #include "content/public/browser/resource_context.h" | 12 #include "content/public/browser/resource_context.h" |
13 #include "extensions/browser/api/dns/host_resolver_wrapper.h" | 13 #include "extensions/browser/api/dns/host_resolver_wrapper.h" |
14 #include "extensions/browser/api/socket/socket.h" | 14 #include "extensions/browser/api/socket/socket.h" |
15 #include "extensions/browser/api/socket/tcp_socket.h" | 15 #include "extensions/browser/api/socket/tcp_socket.h" |
| 16 #include "extensions/browser/api/socket/tls_socket.h" |
16 #include "extensions/browser/api/socket/udp_socket.h" | 17 #include "extensions/browser/api/socket/udp_socket.h" |
17 #include "extensions/browser/extension_system.h" | 18 #include "extensions/browser/extension_system.h" |
18 #include "extensions/common/extension.h" | 19 #include "extensions/common/extension.h" |
19 #include "extensions/common/permissions/permissions_data.h" | 20 #include "extensions/common/permissions/permissions_data.h" |
20 #include "extensions/common/permissions/socket_permission.h" | 21 #include "extensions/common/permissions/socket_permission.h" |
21 #include "net/base/host_port_pair.h" | 22 #include "net/base/host_port_pair.h" |
22 #include "net/base/io_buffer.h" | 23 #include "net/base/io_buffer.h" |
23 #include "net/base/ip_endpoint.h" | 24 #include "net/base/ip_endpoint.h" |
24 #include "net/base/net_errors.h" | 25 #include "net/base/net_errors.h" |
25 #include "net/base/net_log.h" | 26 #include "net/base/net_log.h" |
26 #include "net/base/net_util.h" | 27 #include "net/base/net_util.h" |
| 28 #include "net/url_request/url_request_context.h" |
| 29 #include "net/url_request/url_request_context_getter.h" |
27 | 30 |
28 namespace extensions { | 31 namespace extensions { |
29 | 32 |
30 using content::SocketPermissionRequest; | 33 using content::SocketPermissionRequest; |
31 | 34 |
32 const char kAddressKey[] = "address"; | 35 const char kAddressKey[] = "address"; |
33 const char kPortKey[] = "port"; | 36 const char kPortKey[] = "port"; |
34 const char kBytesWrittenKey[] = "bytesWritten"; | 37 const char kBytesWrittenKey[] = "bytesWritten"; |
35 const char kDataKey[] = "data"; | 38 const char kDataKey[] = "data"; |
36 const char kResultCodeKey[] = "resultCode"; | 39 const char kResultCodeKey[] = "resultCode"; |
37 const char kSocketIdKey[] = "socketId"; | 40 const char kSocketIdKey[] = "socketId"; |
38 | 41 |
39 const char kSocketNotFoundError[] = "Socket not found"; | 42 const char kSocketNotFoundError[] = "Socket not found"; |
40 const char kDnsLookupFailedError[] = "DNS resolution failed"; | 43 const char kDnsLookupFailedError[] = "DNS resolution failed"; |
41 const char kPermissionError[] = "App does not have permission"; | 44 const char kPermissionError[] = "App does not have permission"; |
42 const char kNetworkListError[] = "Network lookup failed or unsupported"; | 45 const char kNetworkListError[] = "Network lookup failed or unsupported"; |
43 const char kTCPSocketBindError[] = | 46 const char kTCPSocketBindError[] = |
44 "TCP socket does not support bind. For TCP server please use listen."; | 47 "TCP socket does not support bind. For TCP server please use listen."; |
45 const char kMulticastSocketTypeError[] = "Only UDP socket supports multicast."; | 48 const char kMulticastSocketTypeError[] = "Only UDP socket supports multicast."; |
| 49 const char kSecureSocketTypeError[] = "Only TCP sockets are supported for TLS."; |
| 50 const char kSocketNotConnectedError[] = "Socket not connected"; |
46 const char kWildcardAddress[] = "*"; | 51 const char kWildcardAddress[] = "*"; |
47 const int kWildcardPort = 0; | 52 const int kWildcardPort = 0; |
48 | 53 |
49 SocketAsyncApiFunction::SocketAsyncApiFunction() {} | 54 SocketAsyncApiFunction::SocketAsyncApiFunction() {} |
50 | 55 |
51 SocketAsyncApiFunction::~SocketAsyncApiFunction() {} | 56 SocketAsyncApiFunction::~SocketAsyncApiFunction() {} |
52 | 57 |
53 bool SocketAsyncApiFunction::PrePrepare() { | 58 bool SocketAsyncApiFunction::PrePrepare() { |
54 manager_ = CreateSocketResourceManager(); | 59 manager_ = CreateSocketResourceManager(); |
55 return manager_->SetBrowserContext(browser_context()); | 60 return manager_->SetBrowserContext(browser_context()); |
56 } | 61 } |
57 | 62 |
58 bool SocketAsyncApiFunction::Respond() { return error_.empty(); } | 63 bool SocketAsyncApiFunction::Respond() { return error_.empty(); } |
59 | 64 |
60 scoped_ptr<SocketResourceManagerInterface> | 65 scoped_ptr<SocketResourceManagerInterface> |
61 SocketAsyncApiFunction::CreateSocketResourceManager() { | 66 SocketAsyncApiFunction::CreateSocketResourceManager() { |
62 return scoped_ptr<SocketResourceManagerInterface>( | 67 return scoped_ptr<SocketResourceManagerInterface>( |
63 new SocketResourceManager<Socket>()).Pass(); | 68 new SocketResourceManager<Socket>()).Pass(); |
64 } | 69 } |
65 | 70 |
66 int SocketAsyncApiFunction::AddSocket(Socket* socket) { | 71 int SocketAsyncApiFunction::AddSocket(Socket* socket) { |
67 return manager_->Add(socket); | 72 return manager_->Add(socket); |
68 } | 73 } |
69 | 74 |
70 Socket* SocketAsyncApiFunction::GetSocket(int api_resource_id) { | 75 Socket* SocketAsyncApiFunction::GetSocket(int api_resource_id) { |
71 return manager_->Get(extension_->id(), api_resource_id); | 76 return manager_->Get(extension_->id(), api_resource_id); |
72 } | 77 } |
73 | 78 |
| 79 void SocketAsyncApiFunction::ReplaceSocket(int api_resource_id, |
| 80 Socket* socket) { |
| 81 manager_->Replace(extension_->id(), api_resource_id, socket); |
| 82 } |
| 83 |
74 base::hash_set<int>* SocketAsyncApiFunction::GetSocketIds() { | 84 base::hash_set<int>* SocketAsyncApiFunction::GetSocketIds() { |
75 return manager_->GetResourceIds(extension_->id()); | 85 return manager_->GetResourceIds(extension_->id()); |
76 } | 86 } |
77 | 87 |
78 void SocketAsyncApiFunction::RemoveSocket(int api_resource_id) { | 88 void SocketAsyncApiFunction::RemoveSocket(int api_resource_id) { |
79 manager_->Remove(extension_->id(), api_resource_id); | 89 manager_->Remove(extension_->id(), api_resource_id); |
80 } | 90 } |
81 | 91 |
82 SocketExtensionWithDnsLookupFunction::SocketExtensionWithDnsLookupFunction() | 92 SocketExtensionWithDnsLookupFunction::SocketExtensionWithDnsLookupFunction() |
83 : resource_context_(NULL), | 93 : resource_context_(NULL), |
(...skipping 103 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
187 | 197 |
188 void SocketConnectFunction::AsyncWorkStart() { | 198 void SocketConnectFunction::AsyncWorkStart() { |
189 socket_ = GetSocket(socket_id_); | 199 socket_ = GetSocket(socket_id_); |
190 if (!socket_) { | 200 if (!socket_) { |
191 error_ = kSocketNotFoundError; | 201 error_ = kSocketNotFoundError; |
192 SetResult(new base::FundamentalValue(-1)); | 202 SetResult(new base::FundamentalValue(-1)); |
193 AsyncWorkCompleted(); | 203 AsyncWorkCompleted(); |
194 return; | 204 return; |
195 } | 205 } |
196 | 206 |
| 207 socket_->set_hostname(hostname_); |
| 208 |
197 SocketPermissionRequest::OperationType operation_type; | 209 SocketPermissionRequest::OperationType operation_type; |
198 switch (socket_->GetSocketType()) { | 210 switch (socket_->GetSocketType()) { |
199 case Socket::TYPE_TCP: | 211 case Socket::TYPE_TCP: |
200 operation_type = SocketPermissionRequest::TCP_CONNECT; | 212 operation_type = SocketPermissionRequest::TCP_CONNECT; |
201 break; | 213 break; |
202 case Socket::TYPE_UDP: | 214 case Socket::TYPE_UDP: |
203 operation_type = SocketPermissionRequest::UDP_SEND_TO; | 215 operation_type = SocketPermissionRequest::UDP_SEND_TO; |
204 break; | 216 break; |
205 default: | 217 default: |
206 NOTREACHED() << "Unknown socket type."; | 218 NOTREACHED() << "Unknown socket type."; |
(...skipping 673 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
880 SetResult(new base::FundamentalValue(result)); | 892 SetResult(new base::FundamentalValue(result)); |
881 return; | 893 return; |
882 } | 894 } |
883 | 895 |
884 base::ListValue* values = new base::ListValue(); | 896 base::ListValue* values = new base::ListValue(); |
885 values->AppendStrings((std::vector<std::string>&)static_cast<UDPSocket*>( | 897 values->AppendStrings((std::vector<std::string>&)static_cast<UDPSocket*>( |
886 socket)->GetJoinedGroups()); | 898 socket)->GetJoinedGroups()); |
887 SetResult(values); | 899 SetResult(values); |
888 } | 900 } |
889 | 901 |
| 902 SocketSecureFunction::SocketSecureFunction() { |
| 903 } |
| 904 |
| 905 SocketSecureFunction::~SocketSecureFunction() { |
| 906 } |
| 907 |
| 908 bool SocketSecureFunction::Prepare() { |
| 909 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); |
| 910 params_ = core_api::socket::Secure::Params::Create(*args_); |
| 911 EXTENSION_FUNCTION_VALIDATE(params_.get()); |
| 912 url_request_getter_ = browser_context()->GetRequestContext(); |
| 913 return true; |
| 914 } |
| 915 |
| 916 // Override the regular implementation, which would call AsyncWorkCompleted |
| 917 // immediately after Work(). |
| 918 void SocketSecureFunction::AsyncWorkStart() { |
| 919 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); |
| 920 |
| 921 Socket* socket = GetSocket(params_->socket_id); |
| 922 if (!socket) { |
| 923 SetResult(new base::FundamentalValue(net::ERR_INVALID_ARGUMENT)); |
| 924 error_ = kSocketNotFoundError; |
| 925 AsyncWorkCompleted(); |
| 926 return; |
| 927 } |
| 928 |
| 929 // Make sure that the socket is a TCP client socket. |
| 930 if (socket->GetSocketType() != Socket::TYPE_TCP || |
| 931 static_cast<TCPSocket*>(socket)->ClientStream() == NULL) { |
| 932 SetResult(new base::FundamentalValue(net::ERR_INVALID_ARGUMENT)); |
| 933 error_ = kSecureSocketTypeError; |
| 934 AsyncWorkCompleted(); |
| 935 return; |
| 936 } |
| 937 |
| 938 if (!socket->IsConnected()) { |
| 939 SetResult(new base::FundamentalValue(net::ERR_INVALID_ARGUMENT)); |
| 940 error_ = kSocketNotConnectedError; |
| 941 AsyncWorkCompleted(); |
| 942 return; |
| 943 } |
| 944 |
| 945 net::URLRequestContext* url_request_context = |
| 946 url_request_getter_->GetURLRequestContext(); |
| 947 |
| 948 TLSSocket::UpgradeSocketToTLS( |
| 949 socket, |
| 950 url_request_context->ssl_config_service(), |
| 951 url_request_context->cert_verifier(), |
| 952 url_request_context->transport_security_state(), |
| 953 extension_id(), |
| 954 params_->options.get(), |
| 955 base::Bind(&SocketSecureFunction::TlsConnectDone, this)); |
| 956 } |
| 957 |
| 958 void SocketSecureFunction::TlsConnectDone(scoped_ptr<TLSSocket> socket, |
| 959 int result) { |
| 960 // if an error occurred, socket MUST be NULL. |
| 961 DCHECK(result == net::OK || socket == NULL); |
| 962 |
| 963 if (socket && result == net::OK) { |
| 964 ReplaceSocket(params_->socket_id, socket.release()); |
| 965 } else { |
| 966 RemoveSocket(params_->socket_id); |
| 967 error_ = net::ErrorToString(result); |
| 968 } |
| 969 |
| 970 results_ = core_api::socket::Secure::Results::Create(result); |
| 971 AsyncWorkCompleted(); |
| 972 } |
| 973 |
890 } // namespace extensions | 974 } // namespace extensions |
OLD | NEW |