217 lines
3.8 KiB
C
217 lines
3.8 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
* 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.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <freerdp/utils/tcp.h>
|
|
#include <freerdp/utils/print.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <time.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
|
|
#ifndef _WIN32
|
|
|
|
#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 <winpr/windows.h>
|
|
#define SHUT_RDWR SD_BOTH
|
|
#define close(_fd) closesocket(_fd)
|
|
#endif
|
|
|
|
#ifndef MSG_NOSIGNAL
|
|
#define MSG_NOSIGNAL 0
|
|
|
|
#endif
|
|
|
|
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, BYTE* 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, BYTE* 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;
|
|
}
|
|
|
|
int freerdp_tcp_set_no_delay(int sockfd, BOOL 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;
|
|
}
|
|
|
|
int freerdp_wsa_startup()
|
|
{
|
|
#ifdef _WIN32
|
|
WSADATA wsaData;
|
|
return WSAStartup(0x101, &wsaData);
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
int freerdp_wsa_cleanup()
|
|
{
|
|
#ifdef _WIN32
|
|
return WSACleanup();
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|