/********************************************************/

/*  ntapi: Native API core library                      */

/*  Copyright (C) 2013--2021  Z. Gilboa                 */

/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */

/********************************************************/

#include <psxtypes/psxtypes.h>

#include <ntapi/nt_object.h>

#include <ntapi/nt_file.h>

#include <ntapi/nt_socket.h>

#include <ntapi/ntapi.h>

#include "ntapi_impl.h"

typedef struct _nt_afd_connect_request {

	uintptr_t	unknown[2];

	void *		paddr;

	nt_sockaddr	addr;

} nt_afd_connect_request;

typedef struct __addr_memcpy {

	uint64_t	d0;

	uint64_t	d1;

	uint64_t	d2;

	uint32_t	d3;

} _addr_memcpy;

int32_t __cdecl __ntapi_sc_connect_v2(

	__in		nt_socket *		hssocket,

	__in		nt_sockaddr *		addr,

	__in		uintptr_t		addrlen,

	__in		uintptr_t		service_flags	__optional,

	__out		nt_iosb volatile *	iosb		__optional)

{

	nt_iosb volatile	siosb;

	nt_afd_connect_request	afd_connect_req;

	_addr_memcpy *		src;

	_addr_memcpy *		dst;

	(void)service_flags;

	if ((addrlen != sizeof(nt_sockaddr_in4))

	 && (addrlen != sizeof(nt_sockaddr_in6)))

		return NT_STATUS_INVALID_PARAMETER_3;

	iosb = iosb ? iosb : &siosb;

	/* afd_connect_req */

	afd_connect_req.unknown[0] = 0;

	afd_connect_req.unknown[1] = 0;

	src = (_addr_memcpy *)addr;

	dst = (_addr_memcpy *)&(afd_connect_req.addr);

	dst->d0 = src->d0;

	dst->d1 = src->d1;

	if (addrlen == sizeof(nt_sockaddr_in6)) {

		dst->d2 = src->d2;

		dst->d3 = src->d3;

	} else {

		dst->d2 = 0;

		dst->d3 = 0;

	}

	afd_connect_req.paddr = &(afd_connect_req.addr);

	afd_connect_req.addr.sa_addr_in4.sa_family = hssocket->domain;

	hssocket->iostatus = __ntapi->zw_device_io_control_file(

			hssocket->hsocket,

			hssocket->hevent,

			0,

			0,

			iosb,

			NT_AFD_IOCTL_CONNECT,

			&afd_connect_req,

			sizeof(afd_connect_req),

			(void *)0,

			0);

	return hssocket->iostatus

		? __ntapi->sc_wait(hssocket,iosb,0)

		: NT_STATUS_SUCCESS;

}