kekePower/rdesktop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

win32 bits

Browse Source 
git-svn-id: svn://svn.code.sf.net/p/rdesktop/code/trunk/rdesktop@1348 423420c4-83ab-492f-b58f-81f9feb106b5
This commit is contained in:
Jay Sorg 2006-12-23 03:23:59 +00:00
parent b924730d4f
commit c1aa0656a4
1 changed files with 126 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

183
tcp.c
View File

@ -18,6 +18,7 @@
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/ */
#ifndef _WIN32
#include <unistd.h> /* select read write close */ #include <unistd.h> /* select read write close */
#include <sys/socket.h> /* socket connect setsockopt */ #include <sys/socket.h> /* socket connect setsockopt */
#include <sys/time.h> /* timeval */ #include <sys/time.h> /* timeval */
@ -26,19 +27,52 @@
#include <netinet/tcp.h> /* TCP_NODELAY */ #include <netinet/tcp.h> /* TCP_NODELAY */
#include <arpa/inet.h> /* inet_addr */ #include <arpa/inet.h> /* inet_addr */
#include <errno.h> /* errno */ #include <errno.h> /* errno */
#endif
#include "rdesktop.h" #include "rdesktop.h"
#ifdef _WIN32
#define socklen_t int
#define TCP_CLOSE(_sck) closesocket(_sck)
#define TCP_STRERROR "tcp error"
#define TCP_BLOCKS (WSAGetLastError() == WSAEWOULDBLOCK)
#else
#define TCP_CLOSE(_sck) close(_sck)
#define TCP_STRERROR strerror(errno)
#define TCP_BLOCKS (errno == EWOULDBLOCK)
#endif
#ifndef INADDR_NONE #ifndef INADDR_NONE
#define INADDR_NONE ((unsigned long) -1) #define INADDR_NONE ((unsigned long) -1)
#endif #endif
static int sock; static int g_sock;
static struct stream in; static struct stream g_in;
#ifndef WITH_SCARD #ifndef WITH_SCARD
static struct stream out; static struct stream g_out;
#endif #endif
int g_tcp_port_rdp = TCP_PORT_RDP; int g_tcp_port_rdp = TCP_PORT_RDP;
/* wait till socket is ready to write or timeout */
static BOOL
tcp_can_send(int sck, int millis)
{
fd_set wfds;
struct timeval time;
int sel_count;
time.tv_sec = millis / 1000;
time.tv_usec = (millis * 1000) % 1000000;
FD_ZERO(&wfds);
FD_SET(sck, &wfds);
sel_count = select(sck + 1, 0, &wfds, 0, &time);
if (sel_count > 0)
{
return True;
}
return False;
}
/* Initialise TCP transport data packet */ /* Initialise TCP transport data packet */
STREAM STREAM
tcp_init(uint32 maxlen) tcp_init(uint32 maxlen)
@ -49,7 +83,7 @@ tcp_init(uint32 maxlen)
scard_lock(SCARD_LOCK_TCP); scard_lock(SCARD_LOCK_TCP);
result = scard_tcp_init(); result = scard_tcp_init();
#else #else
result = &out; result = &g_out;
#endif #endif
if (maxlen > result->size) if (maxlen > result->size)
@ -78,13 +112,20 @@ tcp_send(STREAM s)
#endif #endif
while (total < length) while (total < length)
{ {
sent = send(sock, s->data + total, length - total, 0); sent = send(g_sock, s->data + total, length - total, 0);
if (sent <= 0) if (sent <= 0)
{ {
error("send: %s\n", strerror(errno)); if (sent == -1 && TCP_BLOCKS)
{
tcp_can_send(g_sock, 100);
sent = 0;
}
else
{
error("send: %s\n", TCP_STRERROR);
return; return;
} }
}
total += sent; total += sent;
} }
#ifdef WITH_SCARD #ifdef WITH_SCARD
@ -96,19 +137,19 @@ tcp_send(STREAM s)
STREAM STREAM
tcp_recv(STREAM s, uint32 length) tcp_recv(STREAM s, uint32 length)
{ {
unsigned int new_length, end_offset, p_offset; uint32 new_length, end_offset, p_offset;
int rcvd = 0; int rcvd = 0;
if (s == NULL) if (s == NULL)
{ {
/* read into "new" stream */ /* read into "new" stream */
if (length > in.size) if (length > g_in.size)
{ {
in.data = (uint8 *) xrealloc(in.data, length); g_in.data = (uint8 *) xrealloc(g_in.data, length);
in.size = length; g_in.size = length;
} }
in.end = in.p = in.data; g_in.end = g_in.p = g_in.data;
s = &in; s = &g_in;
} }
else else
{ {
@ -127,16 +168,23 @@ tcp_recv(STREAM s, uint32 length)
while (length > 0) while (length > 0)
{ {
if (!ui_select(sock)) if (!ui_select(g_sock))
/* User quit */ /* User quit */
return NULL; return NULL;
rcvd = recv(sock, s->end, length, 0); rcvd = recv(g_sock, s->end, length, 0);
if (rcvd < 0) if (rcvd < 0)
{ {
error("recv: %s\n", strerror(errno)); if (rcvd == -1 && TCP_BLOCKS)
{
rcvd = 0;
}
else
{
error("recv: %s\n", TCP_STRERROR);
return NULL; return NULL;
} }
}
else if (rcvd == 0) else if (rcvd == 0)
{ {
error("Connection closed\n"); error("Connection closed\n");
@ -154,7 +202,8 @@ tcp_recv(STREAM s, uint32 length)
BOOL BOOL
tcp_connect(char *server) tcp_connect(char *server)
{ {
int true_value = 1; socklen_t option_len;
uint32 option_value;
#ifdef IPv6 #ifdef IPv6
@ -175,22 +224,22 @@ tcp_connect(char *server)
} }
ressave = res; ressave = res;
sock = -1; g_sock = -1;
while (res) while (res)
{ {
sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol); g_sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (!(sock < 0)) if (!(g_sock < 0))
{ {
if (connect(sock, res->ai_addr, res->ai_addrlen) == 0) if (connect(g_sock, res->ai_addr, res->ai_addrlen) == 0)
break; break;
close(sock); TCP_CLOSE(g_sock);
sock = -1; g_sock = -1;
} }
res = res->ai_next; res = res->ai_next;
} }
freeaddrinfo(ressave); freeaddrinfo(ressave);
if (sock == -1) if (g_sock == -1)
{ {
error("%s: unable to connect\n", server); error("%s: unable to connect\n", server);
return False; return False;
@ -211,34 +260,46 @@ tcp_connect(char *server)
return False; return False;
} }
if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) if ((g_sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{ {
error("socket: %s\n", strerror(errno)); error("socket: %s\n", TCP_STRERROR);
return False; return False;
} }
servaddr.sin_family = AF_INET; servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(g_tcp_port_rdp); servaddr.sin_port = htons((uint16) g_tcp_port_rdp);
if (connect(sock, (struct sockaddr *) &servaddr, sizeof(struct sockaddr)) < 0) if (connect(g_sock, (struct sockaddr *) &servaddr, sizeof(struct sockaddr)) < 0)
{ {
error("connect: %s\n", strerror(errno)); error("connect: %s\n", TCP_STRERROR);
close(sock); TCP_CLOSE(g_sock);
return False; return False;
} }
#endif /* IPv6 */ #endif /* IPv6 */
setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (void *) &true_value, sizeof(true_value)); option_value = 1;
option_len = sizeof(option_value);
setsockopt(g_sock, IPPROTO_TCP, TCP_NODELAY, (void *) &option_value, option_len);
/* receive buffer must be a least 16 K */
if (getsockopt(g_sock, SOL_SOCKET, SO_RCVBUF, (void *) &option_value, &option_len) == 0)
{
if (option_value < (1024 * 16))
{
option_value = 1024 * 16;
option_len = sizeof(option_value);
setsockopt(g_sock, SOL_SOCKET, SO_RCVBUF, (void *) &option_value, option_len);
}
}
in.size = 4096; g_in.size = 4096;
in.data = (uint8 *) xmalloc(in.size); g_in.data = (uint8 *) xmalloc(g_in.size);
#ifdef WITH_SCARD #ifdef WITH_SCARD
scard_tcp_connect(); scard_tcp_connect();
#else #else
out.size = 4096; g_out.size = 4096;
out.data = (uint8 *) xmalloc(out.size); g_out.data = (uint8 *) xmalloc(g_out.size);
#endif #endif
return True; return True;
@ -248,7 +309,7 @@ tcp_connect(char *server)
void void
tcp_disconnect(void) tcp_disconnect(void)
{ {
close(sock); TCP_CLOSE(g_sock);
} }
char * char *
@ -257,9 +318,9 @@ tcp_get_address()
static char ipaddr[32]; static char ipaddr[32];
struct sockaddr_in sockaddr; struct sockaddr_in sockaddr;
socklen_t len = sizeof(sockaddr); socklen_t len = sizeof(sockaddr);
if (getsockname(sock, (struct sockaddr *) &sockaddr, &len) == 0) if (getsockname(g_sock, (struct sockaddr *) &sockaddr, &len) == 0)
{ {
unsigned char *ip = (unsigned char *) &sockaddr.sin_addr; uint8 *ip = (uint8 *) &sockaddr.sin_addr;
sprintf(ipaddr, "%d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3]); sprintf(ipaddr, "%d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3]);
} }
else else
@ -272,35 +333,35 @@ tcp_get_address()
void void
tcp_reset_state(void) tcp_reset_state(void)
{ {
sock = -1; /* reset socket */ g_sock = -1; /* reset socket */
/* Clear the incoming stream */ /* Clear the incoming stream */
if (in.data != NULL) if (g_in.data != NULL)
xfree(in.data); xfree(g_in.data);
in.p = NULL; g_in.p = NULL;
in.end = NULL; g_in.end = NULL;
in.data = NULL; g_in.data = NULL;
in.size = 0; g_in.size = 0;
in.iso_hdr = NULL; g_in.iso_hdr = NULL;
in.mcs_hdr = NULL; g_in.mcs_hdr = NULL;
in.sec_hdr = NULL; g_in.sec_hdr = NULL;
in.rdp_hdr = NULL; g_in.rdp_hdr = NULL;
in.channel_hdr = NULL; g_in.channel_hdr = NULL;
/* Clear the outgoing stream(s) */ /* Clear the outgoing stream(s) */
#ifdef WITH_SCARD #ifdef WITH_SCARD
scard_tcp_reset_state(); scard_tcp_reset_state();
#else #else
if (out.data != NULL) if (g_out.data != NULL)
xfree(out.data); xfree(g_out.data);
out.p = NULL; g_out.p = NULL;
out.end = NULL; g_out.end = NULL;
out.data = NULL; g_out.data = NULL;
out.size = 0; g_out.size = 0;
out.iso_hdr = NULL; g_out.iso_hdr = NULL;
out.mcs_hdr = NULL; g_out.mcs_hdr = NULL;
out.sec_hdr = NULL; g_out.sec_hdr = NULL;
out.rdp_hdr = NULL; g_out.rdp_hdr = NULL;
out.channel_hdr = NULL; g_out.channel_hdr = NULL;
#endif #endif
} }