+++ /dev/null
-/*
- * Copyright (C) 1999 AT&T Laboratories Cambridge. All Rights Reserved.
- *
- * This is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This software is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
-
-/*
- * sockets.c - functions to deal with sockets.
- */
-
-#ifdef __STRICT_ANSI__
-#define _BSD_SOURCE
-#endif
-#include <unistd.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <assert.h>
-#include <rfb/rfbclient.h>
-#ifdef WIN32
-#include <winsock2.h>
-#define EWOULDBLOCK WSAEWOULDBLOCK
-#define close closesocket
-#define read(sock,buf,len) recv(sock,buf,len,0)
-#define write(sock,buf,len) send(sock,buf,len,0)
-#else
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <netinet/tcp.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#endif
-
-void PrintInHex(char *buf, int len);
-
-rfbBool errorMessageOnReadFailure = TRUE;
-
-/*
- * ReadFromRFBServer is called whenever we want to read some data from the RFB
- * server. It is non-trivial for two reasons:
- *
- * 1. For efficiency it performs some intelligent buffering, avoiding invoking
- * the read() system call too often. For small chunks of data, it simply
- * copies the data out of an internal buffer. For large amounts of data it
- * reads directly into the buffer provided by the caller.
- *
- * 2. Whenever read() would block, it invokes the Xt event dispatching
- * mechanism to process X events. In fact, this is the only place these
- * events are processed, as there is no XtAppMainLoop in the program.
- */
-
-rfbBool
-ReadFromRFBServer(rfbClient* client, char *out, unsigned int n)
-{
-#undef DEBUG_READ_EXACT
-#ifdef DEBUG_READ_EXACT
- char* oout=out;
- int nn=n;
- rfbClientLog("ReadFromRFBServer %d bytes\n",n);
-#endif
- if (client->serverPort==-1) {
- /* vncrec playing */
- rfbVNCRec* rec = client->vncRec;
- struct timeval tv;
-
- if (rec->readTimestamp) {
- rec->readTimestamp = FALSE;
- if (!fread(&tv,sizeof(struct timeval),1,rec->file))
- return FALSE;
-
- tv.tv_sec = rfbClientSwap32IfLE (tv.tv_sec);
- tv.tv_usec = rfbClientSwap32IfLE (tv.tv_usec);
-
- if (rec->tv.tv_sec!=0 && !rec->doNotSleep) {
- struct timeval diff;
- diff.tv_sec = tv.tv_sec - rec->tv.tv_sec;
- diff.tv_usec = tv.tv_usec - rec->tv.tv_usec;
- if(diff.tv_usec<0) {
- diff.tv_sec--;
- diff.tv_usec+=1000000;
- }
-#ifndef __MINGW32__
- sleep (diff.tv_sec);
- usleep (diff.tv_usec);
-#else
- Sleep (diff.tv_sec * 1000 + diff.tv_usec/1000);
-#endif
- }
-
- rec->tv=tv;
- }
-
- return (fread(out,1,n,rec->file)<0?FALSE:TRUE);
- }
-
- if (n <= client->buffered) {
- memcpy(out, client->bufoutptr, n);
- client->bufoutptr += n;
- client->buffered -= n;
-#ifdef DEBUG_READ_EXACT
- goto hexdump;
-#endif
- return TRUE;
- }
-
- memcpy(out, client->bufoutptr, client->buffered);
-
- out += client->buffered;
- n -= client->buffered;
-
- client->bufoutptr = client->buf;
- client->buffered = 0;
-
- if (n <= RFB_BUF_SIZE) {
-
- while (client->buffered < n) {
- int i = read(client->sock, client->buf + client->buffered, RFB_BUF_SIZE - client->buffered);
- if (i <= 0) {
- if (i < 0) {
-#ifdef WIN32
- errno=WSAGetLastError();
-#endif
- if (errno == EWOULDBLOCK || errno == EAGAIN) {
- /* TODO:
- ProcessXtEvents();
- */
- i = 0;
- } else {
- rfbClientErr("read (%d: %s)\n",errno,strerror(errno));
- return FALSE;
- }
- } else {
- if (errorMessageOnReadFailure) {
- rfbClientLog("VNC server closed connection\n");
- }
- return FALSE;
- }
- }
- client->buffered += i;
- }
-
- memcpy(out, client->bufoutptr, n);
- client->bufoutptr += n;
- client->buffered -= n;
-
- } else {
-
- while (n > 0) {
- int i = read(client->sock, out, n);
- if (i <= 0) {
- if (i < 0) {
-#ifdef WIN32
- errno=WSAGetLastError();
-#endif
- if (errno == EWOULDBLOCK || errno == EAGAIN) {
- /* TODO:
- ProcessXtEvents();
- */
- i = 0;
- } else {
- rfbClientErr("read (%s)\n",strerror(errno));
- return FALSE;
- }
- } else {
- if (errorMessageOnReadFailure) {
- rfbClientLog("VNC server closed connection\n");
- }
- return FALSE;
- }
- }
- out += i;
- n -= i;
- }
- }
-
-#ifdef DEBUG_READ_EXACT
-hexdump:
- { int ii;
- for(ii=0;ii<nn;ii++)
- fprintf(stderr,"%02x ",(unsigned char)oout[ii]);
- fprintf(stderr,"\n");
- }
-#endif
-
- return TRUE;
-}
-
-
-/*
- * Write an exact number of bytes, and don't return until you've sent them.
- */
-
-rfbBool
-WriteToRFBServer(rfbClient* client, char *buf, int n)
-{
- fd_set fds;
- int i = 0;
- int j;
-
- if (client->serverPort==-1)
- return TRUE; /* vncrec playing */
-
- while (i < n) {
- j = write(client->sock, buf + i, (n - i));
- if (j <= 0) {
- if (j < 0) {
- if (errno == EWOULDBLOCK ||
-#ifdef LIBVNCSERVER_ENOENT_WORKAROUND
- errno == ENOENT ||
-#endif
- errno == EAGAIN) {
- FD_ZERO(&fds);
- FD_SET(client->sock,&fds);
-
- if (select(client->sock+1, NULL, &fds, NULL, NULL) <= 0) {
- rfbClientErr("select\n");
- return FALSE;
- }
- j = 0;
- } else {
- rfbClientErr("write\n");
- return FALSE;
- }
- } else {
- rfbClientLog("write failed\n");
- return FALSE;
- }
- }
- i += j;
- }
- return TRUE;
-}
-
-
-/*
- * ConnectToTcpAddr connects to the given TCP port.
- */
-
-int
-ConnectClientToTcpAddr(unsigned int host, int port)
-{
- int sock;
- struct sockaddr_in addr;
- int one = 1;
-
-#ifdef WIN32
- WSADATA trash;
- static rfbBool WSAinitted=FALSE;
- if(!WSAinitted) {
- WSAinitted=TRUE;
- int i=WSAStartup(MAKEWORD(2,0),&trash);
- if(i!=0) {
- rfbClientErr("Couldn't init Windows Sockets\n");
- return -1;
- }
- }
-#endif
-
- addr.sin_family = AF_INET;
- addr.sin_port = htons(port);
- addr.sin_addr.s_addr = host;
-
- sock = socket(AF_INET, SOCK_STREAM, 0);
- if (sock < 0) {
-#ifdef WIN32
- errno=WSAGetLastError();
-#endif
- rfbClientErr("ConnectToTcpAddr: socket (%s)\n",strerror(errno));
- return -1;
- }
-
- if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
- rfbClientErr("ConnectToTcpAddr: connect\n");
- close(sock);
- return -1;
- }
-
- if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY,
- (char *)&one, sizeof(one)) < 0) {
- rfbClientErr("ConnectToTcpAddr: setsockopt\n");
- close(sock);
- return -1;
- }
-
- return sock;
-}
-
-
-
-/*
- * FindFreeTcpPort tries to find unused TCP port in the range
- * (TUNNEL_PORT_OFFSET, TUNNEL_PORT_OFFSET + 99]. Returns 0 on failure.
- */
-
-int
-FindFreeTcpPort(void)
-{
- int sock, port;
- struct sockaddr_in addr;
-
- addr.sin_family = AF_INET;
- addr.sin_addr.s_addr = htonl(INADDR_ANY);
-
- sock = socket(AF_INET, SOCK_STREAM, 0);
- if (sock < 0) {
- rfbClientErr(": FindFreeTcpPort: socket\n");
- return 0;
- }
-
- for (port = TUNNEL_PORT_OFFSET + 99; port > TUNNEL_PORT_OFFSET; port--) {
- addr.sin_port = htons((unsigned short)port);
- if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) == 0) {
- close(sock);
- return port;
- }
- }
-
- close(sock);
- return 0;
-}
-
-
-/*
- * ListenAtTcpPort starts listening at the given TCP port.
- */
-
-int
-ListenAtTcpPort(int port)
-{
- int sock;
- struct sockaddr_in addr;
- int one = 1;
-
- addr.sin_family = AF_INET;
- addr.sin_port = htons(port);
- addr.sin_addr.s_addr = htonl(INADDR_ANY);
-
- sock = socket(AF_INET, SOCK_STREAM, 0);
- if (sock < 0) {
- rfbClientErr("ListenAtTcpPort: socket\n");
- return -1;
- }
-
- if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
- (const char *)&one, sizeof(one)) < 0) {
- rfbClientErr("ListenAtTcpPort: setsockopt\n");
- close(sock);
- return -1;
- }
-
- if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
- rfbClientErr("ListenAtTcpPort: bind\n");
- close(sock);
- return -1;
- }
-
- if (listen(sock, 5) < 0) {
- rfbClientErr("ListenAtTcpPort: listen\n");
- close(sock);
- return -1;
- }
-
- return sock;
-}
-
-
-/*
- * AcceptTcpConnection accepts a TCP connection.
- */
-
-int
-AcceptTcpConnection(int listenSock)
-{
- int sock;
- struct sockaddr_in addr;
- int addrlen = sizeof(addr);
- int one = 1;
-
- sock = accept(listenSock, (struct sockaddr *) &addr, &addrlen);
- if (sock < 0) {
- rfbClientErr("AcceptTcpConnection: accept\n");
- return -1;
- }
-
- if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY,
- (char *)&one, sizeof(one)) < 0) {
- rfbClientErr("AcceptTcpConnection: setsockopt\n");
- close(sock);
- return -1;
- }
-
- return sock;
-}
-
-
-/*
- * SetNonBlocking sets a socket into non-blocking mode.
- */
-
-rfbBool
-SetNonBlocking(int sock)
-{
-#ifndef __MINGW32__
- if (fcntl(sock, F_SETFL, O_NONBLOCK) < 0) {
- rfbClientErr("AcceptTcpConnection: fcntl\n");
- return FALSE;
- }
-#else
- rfbClientErr("O_NONBLOCK on MinGW32 NOT IMPLEMENTED\n");
-#endif
- return TRUE;
-}
-
-
-/*
- * StringToIPAddr - convert a host string to an IP address.
- */
-
-rfbBool
-StringToIPAddr(const char *str, unsigned int *addr)
-{
- struct hostent *hp;
-
- if (strcmp(str,"") == 0) {
- *addr = 0; /* local */
- return TRUE;
- }
-
- *addr = inet_addr(str);
-
- if (*addr != -1)
- return TRUE;
-
- hp = gethostbyname(str);
-
- if (hp) {
- *addr = *(unsigned int *)hp->h_addr;
- return TRUE;
- }
-
- return FALSE;
-}
-
-
-/*
- * Test if the other end of a socket is on the same machine.
- */
-
-rfbBool
-SameMachine(int sock)
-{
- struct sockaddr_in peeraddr, myaddr;
- int addrlen = sizeof(struct sockaddr_in);
-
- getpeername(sock, (struct sockaddr *)&peeraddr, &addrlen);
- getsockname(sock, (struct sockaddr *)&myaddr, &addrlen);
-
- return (peeraddr.sin_addr.s_addr == myaddr.sin_addr.s_addr);
-}
-
-
-/*
- * Print out the contents of a packet for debugging.
- */
-
-void
-PrintInHex(char *buf, int len)
-{
- int i, j;
- char c, str[17];
-
- str[16] = 0;
-
- rfbClientLog("ReadExact: ");
-
- for (i = 0; i < len; i++)
- {
- if ((i % 16 == 0) && (i != 0)) {
- rfbClientLog(" ");
- }
- c = buf[i];
- str[i % 16] = (((c > 31) && (c < 127)) ? c : '.');
- rfbClientLog("%02x ",(unsigned char)c);
- if ((i % 4) == 3)
- rfbClientLog(" ");
- if ((i % 16) == 15)
- {
- rfbClientLog("%s\n",str);
- }
- }
- if ((i % 16) != 0)
- {
- for (j = i % 16; j < 16; j++)
- {
- rfbClientLog(" ");
- if ((j % 4) == 3) rfbClientLog(" ");
- }
- str[i % 16] = 0;
- rfbClientLog("%s\n",str);
- }
-
- fflush(stderr);
-}
-
-int WaitForMessage(rfbClient* client,unsigned int usecs)
-{
- fd_set fds;
- struct timeval timeout;
- int num;
-
- if (client->serverPort==-1)
- /* playing back vncrec file */
- return 1;
-
- timeout.tv_sec=(usecs/1000000);
- timeout.tv_usec=(usecs%1000000);
-
- FD_ZERO(&fds);
- FD_SET(client->sock,&fds);
-
- num=select(client->sock+1, &fds, NULL, NULL, &timeout);
- if(num<0)
- rfbClientLog("Waiting for message failed: %d (%s)\n",errno,strerror(errno));
-
- return num;
-}
-
-