| /* |
| * Copyright (C) 2008 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <arpa/inet.h> |
| #include <errno.h> |
| #include <netdb.h> |
| #include <netinet/in.h> |
| #include <netinet/tcp.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <sys/socket.h> |
| #include <sys/types.h> |
| #include <unistd.h> |
| |
| #include "base/logging.h" |
| #include "base/stringprintf.h" |
| #include "jdwp/jdwp_priv.h" |
| |
| #define kBasePort 8000 |
| #define kMaxPort 8040 |
| |
| namespace art { |
| |
| namespace JDWP { |
| |
| /* |
| * JDWP network state. |
| * |
| * We only talk to one debugger at a time. |
| */ |
| struct JdwpSocketState : public JdwpNetStateBase { |
| uint16_t listenPort; |
| int listenSock; /* listen for connection from debugger */ |
| |
| explicit JdwpSocketState(JdwpState* state) : JdwpNetStateBase(state) { |
| listenPort = 0; |
| listenSock = -1; |
| } |
| |
| virtual bool Accept(); |
| virtual bool Establish(const JdwpOptions*); |
| virtual void Shutdown(); |
| virtual bool ProcessIncoming(); |
| |
| private: |
| in_addr remote_addr_; |
| uint16_t remote_port_; |
| }; |
| |
| static JdwpSocketState* SocketStartup(JdwpState* state, uint16_t port, bool probe); |
| |
| /* |
| * Set up some stuff for transport=dt_socket. |
| */ |
| bool InitSocketTransport(JdwpState* state, const JdwpOptions* options) { |
| uint16_t port = options->port; |
| |
| if (options->server) { |
| if (options->port != 0) { |
| /* try only the specified port */ |
| state->netState = SocketStartup(state, port, false); |
| } else { |
| /* scan through a range of ports, binding to the first available */ |
| for (port = kBasePort; port <= kMaxPort; port++) { |
| state->netState = SocketStartup(state, port, true); |
| if (state->netState != NULL) { |
| break; |
| } |
| } |
| } |
| if (state->netState == NULL) { |
| LOG(ERROR) << "JDWP net startup failed (req port=" << options->port << ")"; |
| return false; |
| } |
| } else { |
| state->netState = SocketStartup(state, 0, false); |
| } |
| |
| if (options->suspend) { |
| LOG(INFO) << "JDWP will wait for debugger on port " << port; |
| } else { |
| LOG(INFO) << "JDWP will " << (options->server ? "listen" : "connect") << " on port " << port; |
| } |
| |
| return true; |
| } |
| |
| /* |
| * Initialize JDWP stuff. |
| * |
| * Allocates a new state structure. If "port" is non-zero, this also |
| * tries to bind to a listen port. If "port" is zero, we assume |
| * we're preparing for an outbound connection, and return without binding |
| * to anything. |
| * |
| * This may be called several times if we're probing for a port. |
| * |
| * Returns 0 on success. |
| */ |
| static JdwpSocketState* SocketStartup(JdwpState* state, uint16_t port, bool probe) { |
| JdwpSocketState* netState = new JdwpSocketState(state); |
| if (port == 0) { |
| return netState; |
| } |
| |
| netState->listenSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); |
| if (netState->listenSock < 0) { |
| PLOG(probe ? ERROR : FATAL) << "Socket create failed"; |
| goto fail; |
| } |
| |
| /* allow immediate re-use */ |
| { |
| int one = 1; |
| if (setsockopt(netState->listenSock, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) < 0) { |
| PLOG(probe ? ERROR : FATAL) << "setsockopt(SO_REUSEADDR) failed"; |
| goto fail; |
| } |
| } |
| |
| union { |
| sockaddr_in addrInet; |
| sockaddr addrPlain; |
| } addr; |
| addr.addrInet.sin_family = AF_INET; |
| addr.addrInet.sin_port = htons(port); |
| inet_aton("127.0.0.1", &addr.addrInet.sin_addr); |
| |
| if (bind(netState->listenSock, &addr.addrPlain, sizeof(addr)) != 0) { |
| PLOG(probe ? ERROR : FATAL) << "Attempt to bind to port " << port << " failed"; |
| goto fail; |
| } |
| |
| netState->listenPort = port; |
| |
| if (listen(netState->listenSock, 5) != 0) { |
| PLOG(probe ? ERROR : FATAL) << "Listen failed"; |
| goto fail; |
| } |
| |
| return netState; |
| |
| fail: |
| netState->Shutdown(); |
| delete netState; |
| return NULL; |
| } |
| |
| /* |
| * Shut down JDWP listener. Don't free state. |
| * |
| * This may be called from a non-JDWP thread as part of shutting the |
| * JDWP thread down. |
| * |
| * (This is currently called several times during startup as we probe |
| * for an open port.) |
| */ |
| void JdwpSocketState::Shutdown() { |
| int listenSock = this->listenSock; |
| int clientSock = this->clientSock; |
| |
| /* clear these out so it doesn't wake up and try to reuse them */ |
| this->listenSock = this->clientSock = -1; |
| |
| /* "shutdown" dislodges blocking read() and accept() calls */ |
| if (listenSock != -1) { |
| shutdown(listenSock, SHUT_RDWR); |
| close(listenSock); |
| } |
| if (clientSock != -1) { |
| shutdown(clientSock, SHUT_RDWR); |
| close(clientSock); |
| } |
| |
| WakePipe(); |
| } |
| |
| /* |
| * Disable the TCP Nagle algorithm, which delays transmission of outbound |
| * packets until the previous transmissions have been acked. JDWP does a |
| * lot of back-and-forth with small packets, so this may help. |
| */ |
| static int SetNoDelay(int fd) { |
| int on = 1; |
| int cc = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &on, sizeof(on)); |
| CHECK_EQ(cc, 0); |
| return cc; |
| } |
| |
| /* |
| * Accept a connection. This will block waiting for somebody to show up. |
| * If that's not desirable, use checkConnection() to make sure something |
| * is pending. |
| */ |
| bool JdwpSocketState::Accept() { |
| union { |
| sockaddr_in addrInet; |
| sockaddr addrPlain; |
| } addr; |
| socklen_t addrlen; |
| int sock; |
| |
| if (listenSock < 0) { |
| return false; /* you're not listening! */ |
| } |
| |
| CHECK_EQ(clientSock, -1); /* must not already be talking */ |
| |
| addrlen = sizeof(addr); |
| do { |
| sock = accept(listenSock, &addr.addrPlain, &addrlen); |
| if (sock < 0 && errno != EINTR) { |
| // When we call shutdown() on the socket, accept() returns with |
| // EINVAL. Don't gripe about it. |
| if (errno == EINVAL) { |
| if (VLOG_IS_ON(jdwp)) { |
| PLOG(ERROR) << "accept failed"; |
| } |
| } else { |
| PLOG(ERROR) << "accept failed"; |
| return false; |
| } |
| } |
| } while (sock < 0); |
| |
| remote_addr_ = addr.addrInet.sin_addr; |
| remote_port_ = ntohs(addr.addrInet.sin_port); |
| VLOG(jdwp) << "+++ accepted connection from " << inet_ntoa(remote_addr_) << ":" << remote_port_; |
| |
| clientSock = sock; |
| SetAwaitingHandshake(true); |
| input_count_ = 0; |
| |
| VLOG(jdwp) << "Setting TCP_NODELAY on accepted socket"; |
| SetNoDelay(clientSock); |
| |
| if (!MakePipe()) { |
| return false; |
| } |
| |
| return true; |
| } |
| |
| /* |
| * Create a connection to a waiting debugger. |
| */ |
| bool JdwpSocketState::Establish(const JdwpOptions* options) { |
| union { |
| sockaddr_in addrInet; |
| sockaddr addrPlain; |
| } addr; |
| hostent* pEntry; |
| |
| CHECK(!options->server); |
| CHECK(!options->host.empty()); |
| CHECK_NE(options->port, 0); |
| |
| /* |
| * Start by resolving the host name. |
| */ |
| #ifdef HAVE_GETHOSTBYNAME_R |
| hostent he; |
| char auxBuf[128]; |
| int error; |
| int cc = gethostbyname_r(options->host.c_str(), &he, auxBuf, sizeof(auxBuf), &pEntry, &error); |
| if (cc != 0) { |
| LOG(WARNING) << "gethostbyname_r('" << options->host << "') failed: " << hstrerror(error); |
| return false; |
| } |
| #else |
| h_errno = 0; |
| pEntry = gethostbyname(options->host.c_str()); |
| if (pEntry == NULL) { |
| PLOG(WARNING) << "gethostbyname('" << options->host << "') failed"; |
| return false; |
| } |
| #endif |
| |
| /* copy it out ASAP to minimize risk of multithreaded annoyances */ |
| memcpy(&addr.addrInet.sin_addr, pEntry->h_addr, pEntry->h_length); |
| addr.addrInet.sin_family = pEntry->h_addrtype; |
| |
| addr.addrInet.sin_port = htons(options->port); |
| |
| LOG(INFO) << "Connecting out to " << inet_ntoa(addr.addrInet.sin_addr) << ":" << ntohs(addr.addrInet.sin_port); |
| |
| /* |
| * Create a socket. |
| */ |
| clientSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); |
| if (clientSock < 0) { |
| PLOG(ERROR) << "Unable to create socket"; |
| return false; |
| } |
| |
| /* |
| * Try to connect. |
| */ |
| if (connect(clientSock, &addr.addrPlain, sizeof(addr)) != 0) { |
| PLOG(ERROR) << "Unable to connect to " << inet_ntoa(addr.addrInet.sin_addr) << ":" << ntohs(addr.addrInet.sin_port); |
| close(clientSock); |
| clientSock = -1; |
| return false; |
| } |
| |
| LOG(INFO) << "Connection established to " << options->host << " (" << inet_ntoa(addr.addrInet.sin_addr) << ":" << ntohs(addr.addrInet.sin_port) << ")"; |
| SetAwaitingHandshake(true); |
| input_count_ = 0; |
| |
| SetNoDelay(clientSock); |
| |
| if (!MakePipe()) { |
| return false; |
| } |
| |
| return true; |
| } |
| |
| /* |
| * Process incoming data. If no data is available, this will block until |
| * some arrives. |
| * |
| * If we get a full packet, handle it. |
| * |
| * To take some of the mystery out of life, we want to reject incoming |
| * connections if we already have a debugger attached. If we don't, the |
| * debugger will just mysteriously hang until it times out. We could just |
| * close the listen socket, but there's a good chance we won't be able to |
| * bind to the same port again, which would confuse utilities. |
| * |
| * Returns "false" on error (indicating that the connection has been severed), |
| * "true" if things are still okay. |
| */ |
| bool JdwpSocketState::ProcessIncoming() { |
| int readCount; |
| |
| CHECK_NE(clientSock, -1); |
| |
| if (!HaveFullPacket()) { |
| /* read some more, looping until we have data */ |
| errno = 0; |
| while (1) { |
| int selCount; |
| fd_set readfds; |
| int maxfd = -1; |
| int fd; |
| |
| FD_ZERO(&readfds); |
| |
| /* configure fds; note these may get zapped by another thread */ |
| fd = listenSock; |
| if (fd >= 0) { |
| FD_SET(fd, &readfds); |
| if (maxfd < fd) { |
| maxfd = fd; |
| } |
| } |
| fd = clientSock; |
| if (fd >= 0) { |
| FD_SET(fd, &readfds); |
| if (maxfd < fd) { |
| maxfd = fd; |
| } |
| } |
| fd = wake_pipe_[0]; |
| if (fd >= 0) { |
| FD_SET(fd, &readfds); |
| if (maxfd < fd) { |
| maxfd = fd; |
| } |
| } else { |
| LOG(INFO) << "NOTE: entering select w/o wakepipe"; |
| } |
| |
| if (maxfd < 0) { |
| VLOG(jdwp) << "+++ all fds are closed"; |
| return false; |
| } |
| |
| /* |
| * Select blocks until it sees activity on the file descriptors. |
| * Closing the local file descriptor does not count as activity, |
| * so we can't rely on that to wake us up (it works for read() |
| * and accept(), but not select()). |
| * |
| * We can do one of three things: (1) send a signal and catch |
| * EINTR, (2) open an additional fd ("wake pipe") and write to |
| * it when it's time to exit, or (3) time out periodically and |
| * re-issue the select. We're currently using #2, as it's more |
| * reliable than #1 and generally better than #3. Wastes two fds. |
| */ |
| selCount = select(maxfd+1, &readfds, NULL, NULL, NULL); |
| if (selCount < 0) { |
| if (errno == EINTR) { |
| continue; |
| } |
| PLOG(ERROR) << "select failed"; |
| goto fail; |
| } |
| |
| if (wake_pipe_[0] >= 0 && FD_ISSET(wake_pipe_[0], &readfds)) { |
| if (listenSock >= 0) { |
| LOG(ERROR) << "Exit wake set, but not exiting?"; |
| } else { |
| VLOG(jdwp) << "Got wake-up signal, bailing out of select"; |
| } |
| goto fail; |
| } |
| if (listenSock >= 0 && FD_ISSET(listenSock, &readfds)) { |
| LOG(INFO) << "Ignoring second debugger -- accepting and dropping"; |
| union { |
| sockaddr_in addrInet; |
| sockaddr addrPlain; |
| } addr; |
| socklen_t addrlen; |
| int tmpSock; |
| tmpSock = accept(listenSock, &addr.addrPlain, &addrlen); |
| if (tmpSock < 0) { |
| LOG(INFO) << "Weird -- accept failed"; |
| } else { |
| close(tmpSock); |
| } |
| } |
| if (clientSock >= 0 && FD_ISSET(clientSock, &readfds)) { |
| readCount = read(clientSock, input_buffer_ + input_count_, sizeof(input_buffer_) - input_count_); |
| if (readCount < 0) { |
| /* read failed */ |
| if (errno != EINTR) { |
| goto fail; |
| } |
| VLOG(jdwp) << "+++ EINTR hit"; |
| return true; |
| } else if (readCount == 0) { |
| /* EOF hit -- far end went away */ |
| VLOG(jdwp) << "+++ peer disconnected"; |
| goto fail; |
| } else { |
| break; |
| } |
| } |
| } |
| |
| input_count_ += readCount; |
| if (!HaveFullPacket()) { |
| return true; /* still not there yet */ |
| } |
| } |
| |
| /* |
| * Special-case the initial handshake. For some bizarre reason we're |
| * expected to emulate bad tty settings by echoing the request back |
| * exactly as it was sent. Note the handshake is always initiated by |
| * the debugger, no matter who connects to whom. |
| * |
| * Other than this one case, the protocol [claims to be] stateless. |
| */ |
| if (IsAwaitingHandshake()) { |
| if (memcmp(input_buffer_, kMagicHandshake, kMagicHandshakeLen) != 0) { |
| LOG(ERROR) << StringPrintf("ERROR: bad handshake '%.14s'", input_buffer_); |
| goto fail; |
| } |
| |
| errno = 0; |
| int cc = TEMP_FAILURE_RETRY(write(clientSock, input_buffer_, kMagicHandshakeLen)); |
| if (cc != kMagicHandshakeLen) { |
| PLOG(ERROR) << "Failed writing handshake bytes (" << cc << " of " << kMagicHandshakeLen << ")"; |
| goto fail; |
| } |
| |
| ConsumeBytes(kMagicHandshakeLen); |
| SetAwaitingHandshake(false); |
| VLOG(jdwp) << "+++ handshake complete"; |
| return true; |
| } |
| |
| /* |
| * Handle this packet. |
| */ |
| return state_->HandlePacket(); |
| |
| fail: |
| Close(); |
| return false; |
| } |
| |
| } // namespace JDWP |
| |
| } // namespace art |