Vitaly Buka | cad20f0 | 2015-10-16 17:27:15 -0700 | [diff] [blame] | 1 | // Copyright 2015 The Android Open Source Project |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 14 | |
| 15 | #include <arpa/inet.h> |
| 16 | #include <map> |
| 17 | #include <netdb.h> |
| 18 | #include <string> |
| 19 | #include <sys/socket.h> |
| 20 | #include <sys/types.h> |
| 21 | #include <unistd.h> |
| 22 | |
| 23 | #include <base/bind.h> |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 24 | #include <base/bind_helpers.h> |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 25 | #include <base/files/file_util.h> |
Alex Vakulenko | 71b0d78 | 2015-09-16 14:28:01 -0700 | [diff] [blame] | 26 | #include <base/message_loop/message_loop.h> |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 27 | #include <base/strings/stringprintf.h> |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 28 | #include <brillo/bind_lambda.h> |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 29 | #include <brillo/streams/file_stream.h> |
| 30 | #include <brillo/streams/tls_stream.h> |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 31 | |
| 32 | #include "buffet/socket_stream.h" |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 33 | #include "buffet/weave_error_conversion.h" |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 34 | |
| 35 | namespace buffet { |
| 36 | |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 37 | using weave::provider::Network; |
| 38 | |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 39 | namespace { |
| 40 | |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 41 | std::string GetIPAddress(const sockaddr* sa) { |
| 42 | std::string addr; |
| 43 | char str[INET6_ADDRSTRLEN] = {}; |
| 44 | switch (sa->sa_family) { |
| 45 | case AF_INET: |
| 46 | if (inet_ntop(AF_INET, |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 47 | &(reinterpret_cast<const sockaddr_in*>(sa)->sin_addr), str, |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 48 | sizeof(str))) { |
| 49 | addr = str; |
| 50 | } |
| 51 | break; |
| 52 | |
| 53 | case AF_INET6: |
| 54 | if (inet_ntop(AF_INET6, |
Alex Vakulenko | 0022b75 | 2015-10-02 11:09:59 -0700 | [diff] [blame] | 55 | &(reinterpret_cast<const sockaddr_in6*>(sa)->sin6_addr), |
| 56 | str, sizeof(str))) { |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 57 | addr = str; |
| 58 | } |
| 59 | break; |
| 60 | } |
| 61 | if (addr.empty()) |
| 62 | addr = base::StringPrintf("<Unknown address family: %d>", sa->sa_family); |
| 63 | return addr; |
| 64 | } |
| 65 | |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 66 | int ConnectSocket(const std::string& host, uint16_t port) { |
| 67 | std::string service = std::to_string(port); |
| 68 | addrinfo hints = {0, AF_UNSPEC, SOCK_STREAM}; |
| 69 | addrinfo* result = nullptr; |
| 70 | if (getaddrinfo(host.c_str(), service.c_str(), &hints, &result)) { |
| 71 | PLOG(WARNING) << "Failed to resolve host name: " << host; |
| 72 | return -1; |
| 73 | } |
| 74 | |
| 75 | int socket_fd = -1; |
| 76 | for (const addrinfo* info = result; info != nullptr; info = info->ai_next) { |
| 77 | socket_fd = socket(info->ai_family, info->ai_socktype, info->ai_protocol); |
| 78 | if (socket_fd < 0) |
| 79 | continue; |
| 80 | |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 81 | std::string addr = GetIPAddress(info->ai_addr); |
| 82 | LOG(INFO) << "Connecting to address: " << addr; |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 83 | if (connect(socket_fd, info->ai_addr, info->ai_addrlen) == 0) |
| 84 | break; // Success. |
| 85 | |
Alex Vakulenko | fa1de83 | 2015-09-29 15:51:57 -0700 | [diff] [blame] | 86 | PLOG(WARNING) << "Failed to connect to address: " << addr; |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 87 | close(socket_fd); |
| 88 | socket_fd = -1; |
| 89 | } |
| 90 | |
| 91 | freeaddrinfo(result); |
| 92 | return socket_fd; |
| 93 | } |
| 94 | |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 95 | void OnSuccess(const Network::OpenSslSocketCallback& callback, |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame] | 96 | brillo::StreamPtr tls_stream) { |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 97 | callback.Run( |
| 98 | std::unique_ptr<weave::Stream>{new SocketStream{std::move(tls_stream)}}, |
| 99 | nullptr); |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 100 | } |
| 101 | |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 102 | void OnError(const weave::DoneCallback& callback, |
| 103 | const brillo::Error* brillo_error) { |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 104 | weave::ErrorPtr error; |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 105 | ConvertError(*brillo_error, &error); |
| 106 | callback.Run(std::move(error)); |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 107 | } |
| 108 | |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 109 | } // namespace |
| 110 | |
Alex Vakulenko | 0fef815 | 2015-09-25 08:45:22 -0700 | [diff] [blame] | 111 | void SocketStream::Read(void* buffer, |
| 112 | size_t size_to_read, |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 113 | const ReadCallback& callback) { |
| 114 | brillo::ErrorPtr brillo_error; |
| 115 | if (!ptr_->ReadAsync( |
| 116 | buffer, size_to_read, |
| 117 | base::Bind([](const ReadCallback& callback, |
| 118 | size_t size) { callback.Run(size, nullptr); }, |
| 119 | callback), |
| 120 | base::Bind(&OnError, base::Bind(callback, 0)), &brillo_error)) { |
Alex Vakulenko | 71b0d78 | 2015-09-16 14:28:01 -0700 | [diff] [blame] | 121 | weave::ErrorPtr error; |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 122 | ConvertError(*brillo_error, &error); |
Alex Vakulenko | 71b0d78 | 2015-09-16 14:28:01 -0700 | [diff] [blame] | 123 | base::MessageLoop::current()->PostTask( |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 124 | FROM_HERE, base::Bind(callback, 0, base::Passed(&error))); |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 125 | } |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 126 | } |
| 127 | |
Alex Vakulenko | 0fef815 | 2015-09-25 08:45:22 -0700 | [diff] [blame] | 128 | void SocketStream::Write(const void* buffer, |
| 129 | size_t size_to_write, |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 130 | const WriteCallback& callback) { |
| 131 | brillo::ErrorPtr brillo_error; |
| 132 | if (!ptr_->WriteAllAsync(buffer, size_to_write, base::Bind(callback, nullptr), |
| 133 | base::Bind(&OnError, callback), &brillo_error)) { |
Alex Vakulenko | 71b0d78 | 2015-09-16 14:28:01 -0700 | [diff] [blame] | 134 | weave::ErrorPtr error; |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 135 | ConvertError(*brillo_error, &error); |
Alex Vakulenko | 71b0d78 | 2015-09-16 14:28:01 -0700 | [diff] [blame] | 136 | base::MessageLoop::current()->PostTask( |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 137 | FROM_HERE, base::Bind(callback, base::Passed(&error))); |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 138 | } |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 139 | } |
| 140 | |
Alex Vakulenko | 0fef815 | 2015-09-25 08:45:22 -0700 | [diff] [blame] | 141 | void SocketStream::CancelPendingOperations() { |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 142 | ptr_->CancelPendingAsyncOperations(); |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 143 | } |
| 144 | |
| 145 | std::unique_ptr<weave::Stream> SocketStream::ConnectBlocking( |
| 146 | const std::string& host, |
| 147 | uint16_t port) { |
| 148 | int socket_fd = ConnectSocket(host, port); |
| 149 | if (socket_fd <= 0) |
| 150 | return nullptr; |
| 151 | |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 152 | auto ptr_ = brillo::FileStream::FromFileDescriptor(socket_fd, true, nullptr); |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 153 | if (ptr_) |
| 154 | return std::unique_ptr<Stream>{new SocketStream{std::move(ptr_)}}; |
| 155 | |
| 156 | close(socket_fd); |
| 157 | return nullptr; |
| 158 | } |
| 159 | |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 160 | void SocketStream::TlsConnect(std::unique_ptr<Stream> socket, |
| 161 | const std::string& host, |
| 162 | const Network::OpenSslSocketCallback& callback) { |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 163 | SocketStream* stream = static_cast<SocketStream*>(socket.get()); |
Alex Vakulenko | 94f8eba | 2015-10-14 08:52:45 -0700 | [diff] [blame] | 164 | brillo::TlsStream::Connect( |
| 165 | std::move(stream->ptr_), host, base::Bind(&OnSuccess, callback), |
| 166 | base::Bind(&OnError, base::Bind(callback, nullptr))); |
Vitaly Buka | 493f604 | 2015-08-12 16:17:16 -0700 | [diff] [blame] | 167 | } |
| 168 | |
| 169 | } // namespace buffet |