1 /* 2 * Copyright (C) 2011 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #include "TcpStream.h" 17 #include "emugl/common/sockets.h" 18 19 #include <errno.h> 20 #include <stdio.h> 21 #include <stdlib.h> 22 #include <unistd.h> 23 #include <string.h> 24 25 #ifndef _WIN32 26 #include <netinet/in.h> 27 #include <netinet/tcp.h> 28 #else 29 #include <ws2tcpip.h> 30 #endif 31 32 #define LISTEN_BACKLOG 4 33 34 TcpStream::TcpStream(size_t bufSize) : SocketStream(bufSize) {} 35 36 TcpStream::TcpStream(int sock, size_t bufSize) : 37 SocketStream(sock, bufSize) { 38 // disable Nagle algorithm to improve bandwidth of small 39 // packets which are quite common in our implementation. 40 emugl::socketTcpDisableNagle(sock); 41 } 42 43 int TcpStream::listen(char addrstr[MAX_ADDRSTR_LEN]) { 44 m_sock = emugl::socketTcpLoopbackServer(0, SOCK_STREAM); 45 if (!valid()) 46 return int(ERR_INVALID_SOCKET); 47 48 int port = emugl::socketGetPort(m_sock); 49 if (port < 0) { 50 ::close(m_sock); 51 return int(ERR_INVALID_SOCKET); 52 } 53 54 snprintf(addrstr, MAX_ADDRSTR_LEN - 1, "%hu", port); 55 addrstr[MAX_ADDRSTR_LEN-1] = '\0'; 56 57 return 0; 58 } 59 60 SocketStream * TcpStream::accept() { 61 int clientSock = emugl::socketAccept(m_sock); 62 if (clientSock < 0) 63 return NULL; 64 65 return new TcpStream(clientSock, m_bufsize); 66 } 67 68 int TcpStream::connect(const char* addr) { 69 int port = atoi(addr); 70 m_sock = emugl::socketTcpLoopbackClient(port, SOCK_STREAM); 71 return valid() ? 0 : -1; 72 } 73 74 int TcpStream::connect(const char* hostname, unsigned short port) 75 { 76 m_sock = emugl::socketTcpClient(hostname, port, SOCK_STREAM); 77 return valid() ? 0 : -1; 78 } 79