/*
 *  libsocket - BSD socket like library for DJGPP
 *  Copyright 1997, 1998 by Indrek Mandre
 *  Copyright 1997-2000 by Richard Dawe
 *
 *  Portions of libsocket Copyright 1985-1993 Regents of the University of 
 *  California.
 *  Portions of libsocket Copyright 1991, 1992 Free Software Foundation, Inc.
 *  Portions of libsocket Copyright 1997, 1998 by the Regdos Group.
 *
 *  This library is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU Library General Public License as published
 *  by the Free Software Foundation; either version 2 of the License, or (at
 *  your option) any later version.
 *
 *  This library 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
 *  Library General Public License for more details.
 * 
 *  You should have received a copy of the GNU Library General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

/* INFO: Comments starting with 'IM' indicate that Indrek Mandre made changes.
 * Comments starting wiht 'RD' indicate that Richard Dawe made changes. If no
 * comment is given, then the code was written by Indrek Mandre. */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <dpmi.h>
#include <sys/movedata.h>
#include <sys/farptr.h>
#include <sys/segments.h>

#include <sys/fsext.h>

#include <sys/socket.h>

#include <lsck/if.h>
#include "wsock.h"
#include "farptrx.h"
#include "wsockvxd.h"
#include "w_errors.h"

/* ----------------
 * - __wsock_recv -
 * ---------------- */

ssize_t __wsock_recv (LSCK_SOCKET * lsd, void *buf, size_t len,
                      unsigned int flags)
{
	LSCK_SOCKET_WSOCK *wsock = (LSCK_SOCKET_WSOCK *) lsd->idata;
	WSOCK_RECV_PARAMS params;

	/* RD: Protect against the use of MSG_PEEK. It's very buggy with
	 * Winsock. */
	/* RD: Later: OK, maybe that was a bit harsh. But it may not work! */
	/*if (flags & MSG_PEEK) {
	  errno = EOPNOTSUPP;
	  return (-1);
	  }*/

	/* RD: Check that the request buffer length isn't bigger than the
	 * transfer buffer SocketD. If so, only transfer the size of
	 SocketD. */
	if (len > _SocketD.size) len = _SocketD.size;

	bzero (&params, sizeof (params));

	params.Buffer = (void *) ((SocketD << 16));
	params.Address = NULL;
	params.Socket = (void *) wsock->_Socket;
	params.BufferLength = len;
	params.Flags = flags;
	params.AddressLength = 0;
	params.BytesReceived = 0;
	params.ApcRoutine = NULL;
	params.ApcContext = 0;
	params.Timeout = 0;

	_farpokex (SocketP, 0, &params, sizeof (WSOCK_RECV_PARAMS));

	__wsock_callvxd (WSOCK_RECV_CMD);

	if ((_VXDError == WSAEWOULDBLOCK)
	    || (_VXDError == 0xFFFF)) {    /* Which?? */
		if (lsd->blocking) {
			if (__wsock_select_wait (lsd, FD_READ) == -1)
				return -1;

			_farpokex (SocketP, 0,
				   &params, sizeof (WSOCK_RECV_PARAMS));

			__wsock_callvxd (WSOCK_RECV_CMD);
		} else {
			return (-1);
		}
	}
	
	if (_VXDError) return -1;

	_farpeekx (SocketP, 0, &params, sizeof (WSOCK_RECV_PARAMS));
	movedata (SocketD, 0, _my_ds (), (int) buf, params.BytesReceived);

	return params.BytesReceived;
}

/* --------------------
 * - __wsock_recvfrom -
 * -------------------- */

ssize_t __wsock_recvfrom (LSCK_SOCKET * lsd, void *buf, size_t len,
                          unsigned int flags,
		          struct sockaddr *from, size_t *fromlen)
{	
	LSCK_SOCKET_WSOCK *wsock = (LSCK_SOCKET_WSOCK *) lsd->idata;
	WSOCK_RECV_PARAMS params;

	/* RD: Presume that Winsock checks that the address length is long
	 * enough & will return the appropriate error. */

	/* RD: Protect against the use of MSG_PEEK. It's very buggy with
	 * Winsock. */
	/* RD: Later: OK, maybe that was a bit harsh. But it may not work! */
	/*if (flags & MSG_PEEK) {
		errno = EOPNOTSUPP;
		return (-1);
	}*/
	
	/* RD: Check that the request buffer length isn't bigger than the
	 * transfer buffer SocketD. If so, only transfer the size of
	 * SocketD. */
	if (len > _SocketD.size) len = _SocketD.size;

	bzero (&params, sizeof (params));

	params.Buffer = (void *) (SocketD << 16);
	params.Address = (void *) (SocketP << 16) + (10 * 4);
	params.Socket = (void *) wsock->_Socket;
	params.BufferLength = len;
	params.Flags = flags;
	params.AddressLength = *fromlen;
	params.BytesReceived = 0;
	params.ApcRoutine = NULL;
	params.ApcContext = 0;
	params.Timeout = 0;

	_farpokex (SocketP, 0, &params, sizeof (WSOCK_RECV_PARAMS));
	/* RD: I don't think this should be filled on entry. */
	/*_farpokex ( SocketP, 10 * 4, from, *fromlen );*/

	__wsock_callvxd (WSOCK_RECV_CMD);

	/* RD: NB: errno set automatically */
	if (_VXDError == WSAEWOULDBLOCK || _VXDError == 0xFFFF) { /* Which?? */
		/* Blocking */
		if (lsd->blocking) {
			if (__wsock_select_wait (lsd, FD_READ) == -1)
				return -1;

			_farpokex (SocketP, 0, &params,
				   sizeof (WSOCK_RECV_PARAMS));
			/* RD: I don't think this should be filled on entry. */
			/*_farpokex ( SocketP, 10 * 4, from, *fromlen );*/

			__wsock_callvxd (WSOCK_RECV_CMD);

			/* Non-blocking */
		} else
			return (-1);
	}
	
	if (_VXDError) return -1;

	_farpeekx (SocketP, 0, &params, sizeof (WSOCK_RECV_PARAMS));
	_farpeekx (SocketD, 0, buf, params.BytesReceived);

	/* RD: Only copy this if from is non-NULL. */
	if ((from != NULL) && (fromlen != NULL)) {
		*fromlen = params.AddressLength;
		_farpeekx (SocketP, 10 * 4, from, *fromlen);
	}
	return params.BytesReceived;
}