FreeRDP/libfreerdp-utils/tcp.c

191 lines
3.6 KiB
C
Raw Normal View History

2012-03-06 23:41:07 +04:00
/**
* FreeRDP: A Remote Desktop Protocol Client
* TCP Utils
*
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
*
* 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 <freerdp/utils/tcp.h>
2012-03-07 00:30:24 +04:00
#include <freerdp/utils/print.h>
2012-03-06 23:41:07 +04:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <errno.h>
#include <fcntl.h>
#ifndef _WIN32
2012-03-06 23:41:07 +04:00
#include <netdb.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <net/if.h>
#ifdef __APPLE__
#ifndef TCP_KEEPIDLE
#define TCP_KEEPIDLE TCP_KEEPALIVE
#endif
#endif
#else /* ifdef _WIN32 */
#include <freerdp/utils/windows.h>
2012-03-06 23:41:07 +04:00
#define SHUT_RDWR SD_BOTH
#define close(_fd) closesocket(_fd)
#endif
2012-03-07 00:30:24 +04:00
#ifndef MSG_NOSIGNAL
#define MSG_NOSIGNAL 0
2012-03-07 00:30:24 +04:00
#endif
2012-03-06 23:41:07 +04:00
int freerdp_tcp_connect(const char* hostname, int port)
{
int status;
int sockfd;
char servname[10];
struct addrinfo* ai;
struct addrinfo* res;
struct addrinfo hints = { 0 };
memset(&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
snprintf(servname, sizeof(servname), "%d", port);
status = getaddrinfo(hostname, servname, &hints, &res);
if (status != 0)
{
printf("tcp_connect: getaddrinfo (%s)\n", gai_strerror(status));
return -1;
}
sockfd = -1;
for (ai = res; ai; ai = ai->ai_next)
{
sockfd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
if (sockfd < 0)
continue;
if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) == 0)
{
printf("connected to %s:%s\n", hostname, servname);
break;
}
close(sockfd);
sockfd = -1;
}
freeaddrinfo(res);
if (sockfd == -1)
{
printf("unable to connect to %s:%s\n", hostname, servname);
return -1;
}
return sockfd;
}
int freerdp_tcp_read(int sockfd, uint8* data, int length)
{
int status;
status = recv(sockfd, data, length, 0);
if (status == 0)
{
return -1; /* peer disconnected */
}
else if (status < 0)
{
#ifdef _WIN32
int wsa_error = WSAGetLastError();
/* No data available */
if (wsa_error == WSAEWOULDBLOCK)
return 0;
printf("recv() error: %d\n", wsa_error);
#else
/* No data available */
if (errno == EAGAIN || errno == EWOULDBLOCK)
return 0;
perror("recv");
#endif
return -1;
}
return status;
}
int freerdp_tcp_write(int sockfd, uint8* data, int length)
{
int status;
status = send(sockfd, data, length, MSG_NOSIGNAL);
if (status < 0)
{
#ifdef _WIN32
int wsa_error = WSAGetLastError();
/* No data available */
if (wsa_error == WSAEWOULDBLOCK)
status = 0;
else
perror("send");
#else
if (errno == EAGAIN || errno == EWOULDBLOCK)
status = 0;
else
perror("send");
#endif
}
return status;
}
int freerdp_tcp_disconnect(int sockfd)
{
if (sockfd != -1)
{
shutdown(sockfd, SHUT_RDWR);
close(sockfd);
}
return 0;
}
2012-03-18 21:13:34 +04:00
int freerdp_tcp_set_no_delay(int sockfd, boolean no_delay)
{
uint32 option_value;
socklen_t option_len;
option_value = no_delay;
option_len = sizeof(option_value);
setsockopt(sockfd, IPPROTO_TCP, TCP_NODELAY, (void*) &option_value, option_len);
return 0;
}