You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
2.8 KiB
124 lines
2.8 KiB
/*
|
|
* Copyright (C) 2011 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 "TcpStream.h"
|
|
#include <cutils/sockets.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#ifndef _WIN32
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
#else
|
|
#include <ws2tcpip.h>
|
|
#endif
|
|
|
|
static int _socket_loopback_server(int port, int type)
|
|
{
|
|
struct sockaddr_in addr;
|
|
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = htons(port);
|
|
addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
|
|
|
|
int s = socket(AF_INET, type, 0);
|
|
if (s < 0)
|
|
return -1;
|
|
|
|
int n = 1;
|
|
setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (const char *) &n, sizeof(n));
|
|
|
|
if (bind(s, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) < 0) {
|
|
close(s);
|
|
return -1;
|
|
}
|
|
|
|
if (type == SOCK_STREAM) {
|
|
if (listen(s, 4) < 0) {
|
|
close(s);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return s;
|
|
}
|
|
|
|
TcpStream::TcpStream(size_t bufSize) :
|
|
SocketStream(bufSize)
|
|
{
|
|
}
|
|
|
|
TcpStream::TcpStream(int sock, size_t bufSize) :
|
|
SocketStream(sock, bufSize)
|
|
{
|
|
// disable Nagle algorithm to improve bandwidth of small
|
|
// packets which are quite common in our implementation.
|
|
#ifdef _WIN32
|
|
DWORD flag;
|
|
#else
|
|
int flag;
|
|
#endif
|
|
flag = 1;
|
|
setsockopt( sock, IPPROTO_TCP, TCP_NODELAY, (const char*)&flag, sizeof(flag) );
|
|
}
|
|
|
|
int TcpStream::listen(unsigned short port)
|
|
{
|
|
m_sock = _socket_loopback_server(port, SOCK_STREAM);
|
|
if (!valid()) return int(ERR_INVALID_SOCKET);
|
|
|
|
return 0;
|
|
}
|
|
|
|
SocketStream * TcpStream::accept()
|
|
{
|
|
int clientSock = -1;
|
|
|
|
while (true) {
|
|
struct sockaddr_in addr;
|
|
socklen_t len = sizeof(addr);
|
|
clientSock = ::accept(m_sock, (sockaddr *)&addr, &len);
|
|
|
|
if (clientSock < 0 && errno == EINTR) {
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
TcpStream *clientStream = NULL;
|
|
|
|
if (clientSock >= 0) {
|
|
clientStream = new TcpStream(clientSock, m_bufsize);
|
|
}
|
|
return clientStream;
|
|
}
|
|
|
|
int TcpStream::connect(unsigned short port)
|
|
{
|
|
return connect("127.0.0.1",port);
|
|
}
|
|
|
|
int TcpStream::connect(const char* hostname, unsigned short port)
|
|
{
|
|
m_sock = socket_network_client(hostname, port, SOCK_STREAM);
|
|
if (!valid()) return -1;
|
|
return 0;
|
|
}
|