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 <cutils/sockets.h>
18 #include <errno.h>
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <unistd.h>
22 #include <string.h>
23
24 #ifndef _WIN32
25 #include <netinet/in.h>
26 #include <netinet/tcp.h>
27 #else
28 #include <ws2tcpip.h>
29 #endif
30
TcpStream(size_t bufSize)31 TcpStream::TcpStream(size_t bufSize) :
32 SocketStream(bufSize)
33 {
34 }
35
TcpStream(int sock,size_t bufSize)36 TcpStream::TcpStream(int sock, size_t bufSize) :
37 SocketStream(sock, bufSize)
38 {
39 // disable Nagle algorithm to improve bandwidth of small
40 // packets which are quite common in our implementation.
41 #ifdef _WIN32
42 DWORD flag;
43 #else
44 int flag;
45 #endif
46 flag = 1;
47 setsockopt( sock, IPPROTO_TCP, TCP_NODELAY, (const char*)&flag, sizeof(flag) );
48 }
49
listen(unsigned short port)50 int TcpStream::listen(unsigned short port)
51 {
52 m_sock = socket_loopback_server(port, SOCK_STREAM);
53 if (!valid()) return int(ERR_INVALID_SOCKET);
54
55 return 0;
56 }
57
accept()58 SocketStream * TcpStream::accept()
59 {
60 int clientSock = -1;
61
62 while (true) {
63 struct sockaddr_in addr;
64 socklen_t len = sizeof(addr);
65 clientSock = ::accept(m_sock, (sockaddr *)&addr, &len);
66
67 if (clientSock < 0 && errno == EINTR) {
68 continue;
69 }
70 break;
71 }
72
73 TcpStream *clientStream = NULL;
74
75 if (clientSock >= 0) {
76 clientStream = new TcpStream(clientSock, m_bufsize);
77 }
78 return clientStream;
79 }
80
connect(unsigned short port)81 int TcpStream::connect(unsigned short port)
82 {
83 return connect("127.0.0.1",port);
84 }
85
connect(const char * hostname,unsigned short port)86 int TcpStream::connect(const char* hostname, unsigned short port)
87 {
88 m_sock = socket_network_client(hostname, port, SOCK_STREAM);
89 if (!valid()) return -1;
90 return 0;
91 }
92