initial commit

git-svn-id: http://svn.miranda-ng.org/main/trunk@2 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c

1 files changed, 203 insertions, 0 deletions

diff --git a/src/modules/netlib/netlibsock.cpp b/src/modules/netlib/netlibsock.cpp
new file mode 100644
index 0000000000..875c47b0a1
--- /dev/null
+++ b/src/modules/netlib/netlibsock.cpp
@@ -0,0 +1,203 @@
+/*

+

+Miranda IM: the free IM client for Microsoft* Windows*

+

+Copyright 2000-2009 Miranda ICQ/IM project,

+all portions of this codebase are copyrighted to the people

+listed in contributors.txt.

+

+This program 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 program 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 program; if not, write to the Free Software

+Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

+*/

+#include "commonheaders.h"

+#include "netlib.h"

+

+extern HANDLE hConnectionHeaderMutex,hSendEvent,hRecvEvent;

+

+INT_PTR NetlibSend(WPARAM wParam,LPARAM lParam)

+{

+	struct NetlibConnection *nlc=(struct NetlibConnection*)wParam;

+	NETLIBBUFFER *nlb=(NETLIBBUFFER*)lParam;

+	INT_PTR result;

+

+	if ( nlb == NULL ) {

+		SetLastError(ERROR_INVALID_PARAMETER);

+		return SOCKET_ERROR;

+	}

+

+	if ( !NetlibEnterNestedCS( nlc, NLNCS_SEND ))

+		return SOCKET_ERROR;

+

+	if ( nlc->usingHttpGateway && !( nlb->flags & MSG_RAW )) {

+		if ( !( nlb->flags & MSG_NOHTTPGATEWAYWRAP ) && nlc->nlu->user.pfnHttpGatewayWrapSend ) {

+			NetlibDumpData( nlc, ( PBYTE )nlb->buf, nlb->len, 1, nlb->flags );

+			result = nlc->nlu->user.pfnHttpGatewayWrapSend(( HANDLE )nlc, ( PBYTE )nlb->buf, nlb->len, nlb->flags | MSG_NOHTTPGATEWAYWRAP, NetlibSend );

+		}

+		else result = NetlibHttpGatewayPost( nlc, nlb->buf, nlb->len, nlb->flags );

+	}

+	else {

+		NetlibDumpData( nlc, ( PBYTE )nlb->buf, nlb->len, 1, nlb->flags );

+		if (nlc->hSsl)

+			result = si.write( nlc->hSsl, nlb->buf, nlb->len );

+		else

+			result = send( nlc->s, nlb->buf, nlb->len, nlb->flags & 0xFFFF );

+	}

+	NetlibLeaveNestedCS( &nlc->ncsSend );

+

+	if ((( THook* )hSendEvent)->subscriberCount ) {

+		NETLIBNOTIFY nln = { nlb, result };

+		CallHookSubscribers( hSendEvent, (WPARAM)&nln, (LPARAM)&nlc->nlu->user );

+	}

+	return result;

+}

+

+INT_PTR NetlibRecv(WPARAM wParam,LPARAM lParam)

+{

+	struct NetlibConnection *nlc = (struct NetlibConnection*)wParam;

+	NETLIBBUFFER* nlb = ( NETLIBBUFFER* )lParam;

+	int recvResult;

+

+	if ( nlb == NULL ) {

+		SetLastError( ERROR_INVALID_PARAMETER );

+		return SOCKET_ERROR;

+	}

+

+	if ( !NetlibEnterNestedCS( nlc, NLNCS_RECV ))

+		return SOCKET_ERROR;

+

+	if ( nlc->usingHttpGateway && !( nlb->flags & MSG_RAW ))

+		recvResult = NetlibHttpGatewayRecv( nlc, nlb->buf, nlb->len, nlb->flags );

+	else

+	{

+		if (nlc->hSsl)

+			recvResult = si.read( nlc->hSsl, nlb->buf, nlb->len, (nlb->flags & MSG_PEEK) != 0 );

+		else

+			recvResult = recv( nlc->s, nlb->buf, nlb->len, nlb->flags & 0xFFFF );

+	}

+	NetlibLeaveNestedCS( &nlc->ncsRecv );

+	if (recvResult <= 0) 

+		return recvResult;

+

+	NetlibDumpData(nlc, (PBYTE)nlb->buf, recvResult, 0, nlb->flags);

+

+	if ((nlb->flags & MSG_PEEK) == 0 && ((THook*)hRecvEvent)->subscriberCount)

+	{

+		NETLIBNOTIFY nln = { nlb, recvResult };

+		CallHookSubscribers(hRecvEvent, (WPARAM)&nln, (LPARAM)&nlc->nlu->user);

+	}

+	return recvResult;

+}

+

+static int ConnectionListToSocketList(HANDLE *hConns, fd_set *fd, int& pending)

+{

+	struct NetlibConnection *nlcCheck;

+	int i;

+

+	FD_ZERO(fd);

+	for(i=0;hConns[i] && hConns[i]!=INVALID_HANDLE_VALUE && i<FD_SETSIZE;i++) {

+		nlcCheck=(struct NetlibConnection*)hConns[i];

+		if (nlcCheck->handleType!=NLH_CONNECTION && nlcCheck->handleType!=NLH_BOUNDPORT) {

+			SetLastError(ERROR_INVALID_DATA);

+			return 0;

+		}

+		FD_SET(nlcCheck->s,fd);

+		if ( si.pending( nlcCheck->hSsl ))

+			pending++;

+	}

+	return 1;

+}

+

+INT_PTR NetlibSelect(WPARAM,LPARAM lParam)

+{

+	NETLIBSELECT *nls=(NETLIBSELECT*)lParam;

+	if (nls==NULL || nls->cbSize!=sizeof(NETLIBSELECT)) {

+		SetLastError(ERROR_INVALID_PARAMETER);

+		return SOCKET_ERROR;

+	}

+

+	TIMEVAL tv;

+	tv.tv_sec=nls->dwTimeout/1000;

+	tv.tv_usec=(nls->dwTimeout%1000)*1000;

+

+	int pending = 0;

+	fd_set readfd, writefd, exceptfd;

+	WaitForSingleObject(hConnectionHeaderMutex,INFINITE);

+	if (!ConnectionListToSocketList(nls->hReadConns,&readfd,pending)

+	   || !ConnectionListToSocketList(nls->hWriteConns,&writefd,pending)

+	   || !ConnectionListToSocketList(nls->hExceptConns,&exceptfd,pending)) {

+		ReleaseMutex(hConnectionHeaderMutex);

+		return SOCKET_ERROR;

+	}

+	ReleaseMutex(hConnectionHeaderMutex);

+	if (pending)

+		return 1;

+

+	return select(0,&readfd,&writefd,&exceptfd,nls->dwTimeout==INFINITE?NULL:&tv);

+}

+

+INT_PTR NetlibSelectEx(WPARAM, LPARAM lParam)

+{

+	NETLIBSELECTEX *nls=(NETLIBSELECTEX*)lParam;

+	if (nls==NULL || nls->cbSize!=sizeof(NETLIBSELECTEX)) {

+		SetLastError(ERROR_INVALID_PARAMETER);

+		return SOCKET_ERROR;

+	}

+

+	TIMEVAL tv;

+	tv.tv_sec=nls->dwTimeout/1000;

+	tv.tv_usec=(nls->dwTimeout%1000)*1000;

+	WaitForSingleObject(hConnectionHeaderMutex,INFINITE);

+

+	int pending = 0;

+	fd_set readfd,writefd,exceptfd;

+	if (!ConnectionListToSocketList(nls->hReadConns,&readfd,pending)

+	   || !ConnectionListToSocketList(nls->hWriteConns,&writefd,pending)

+	   || !ConnectionListToSocketList(nls->hExceptConns,&exceptfd,pending)) {

+		ReleaseMutex(hConnectionHeaderMutex);

+		return SOCKET_ERROR;

+	}

+	ReleaseMutex(hConnectionHeaderMutex);

+

+	int rc = (pending) ? pending : select(0,&readfd,&writefd,&exceptfd,nls->dwTimeout==INFINITE?NULL:&tv);

+

+	WaitForSingleObject(hConnectionHeaderMutex,INFINITE);

+	/* go thru each passed HCONN array and grab its socket handle, then give it to FD_ISSET()

+	to see if an event happened for that socket, if it has it will be returned as TRUE (otherwise not)

+	This happens for read/write/except */

+	struct NetlibConnection *conn=NULL;

+	int j;

+	for (j=0; j<FD_SETSIZE; j++) {

+		conn=(struct NetlibConnection*)nls->hReadConns[j];

+		if (conn==NULL || conn==INVALID_HANDLE_VALUE) break;

+

+		if (si.pending(conn->hSsl))

+			nls->hReadStatus[j] = TRUE;

+		if (conn->usingHttpGateway && conn->nlhpi.szHttpGetUrl == NULL && conn->dataBuffer == NULL)

+			nls->hReadStatus[j] = (conn->pHttpProxyPacketQueue != NULL);

+		else

+			nls->hReadStatus[j] = FD_ISSET(conn->s,&readfd);

+	}

+	for (j=0; j<FD_SETSIZE; j++) {

+		conn=(struct NetlibConnection*)nls->hWriteConns[j];

+		if (conn==NULL || conn==INVALID_HANDLE_VALUE) break;

+		nls->hWriteStatus[j] = FD_ISSET(conn->s,&writefd);

+	}

+	for (j=0; j<FD_SETSIZE; j++) {

+		conn=(struct NetlibConnection*)nls->hExceptConns[j];

+		if (conn==NULL || conn==INVALID_HANDLE_VALUE) break;

+		nls->hExceptStatus[j] = FD_ISSET(conn->s,&exceptfd);

+	}

+	ReleaseMutex(hConnectionHeaderMutex);

+	return rc;

+}