Blame src/socket/ntapi_sc_connect_v2.c

dd89bb
/********************************************************/
dd89bb
/*  ntapi: Native API core library                      */
64e606
/*  Copyright (C) 2013--2021  SysDeer Technologies, LLC */
dd89bb
/*  Released under GPLv2 and GPLv3; see COPYING.NTAPI.  */
dd89bb
/********************************************************/
dd89bb
dd89bb
#include <psxtypes/psxtypes.h>
dd89bb
#include <ntapi/nt_object.h>
dd89bb
#include <ntapi/nt_file.h>
dd89bb
#include <ntapi/nt_socket.h>
dd89bb
#include <ntapi/ntapi.h>
dd89bb
#include "ntapi_impl.h"
dd89bb
dd89bb
typedef struct _nt_afd_connect_request {
dd89bb
	uintptr_t	unknown[2];
dd89bb
	void *		paddr;
dd89bb
	nt_sockaddr	addr;
dd89bb
} nt_afd_connect_request;
dd89bb
dd89bb
typedef struct __addr_memcpy {
dd89bb
	uint64_t	d0;
dd89bb
	uint64_t	d1;
d02e49
	uint64_t	d2;
d02e49
	uint32_t	d3;
dd89bb
} _addr_memcpy;
dd89bb
dd89bb
dd89bb
int32_t __cdecl __ntapi_sc_connect_v2(
dd89bb
	__in		nt_socket *		hssocket,
dd89bb
	__in		nt_sockaddr *		addr,
dd89bb
	__in		uintptr_t		addrlen,
dd89bb
	__in		uintptr_t		service_flags	__optional,
2e04e9
	__out		nt_iosb volatile *	iosb		__optional)
dd89bb
{
2e04e9
	nt_iosb volatile	siosb;
dd89bb
	nt_afd_connect_request	afd_connect_req;
dd89bb
dd89bb
	_addr_memcpy *		src;
dd89bb
	_addr_memcpy *		dst;
dd89bb
c713d8
	(void)service_flags;
c713d8
d02e49
	if ((addrlen != sizeof(nt_sockaddr_in4))
d02e49
	 && (addrlen != sizeof(nt_sockaddr_in6)))
d02e49
		return NT_STATUS_INVALID_PARAMETER_3;
d02e49
dd89bb
	iosb = iosb ? iosb : &siosb;
dd89bb
dd89bb
	/* afd_connect_req */
dd89bb
	afd_connect_req.unknown[0] = 0;
dd89bb
	afd_connect_req.unknown[1] = 0;
dd89bb
dd89bb
	src = (_addr_memcpy *)addr;
dd89bb
	dst = (_addr_memcpy *)&(afd_connect_req.addr);
dd89bb
dd89bb
	dst->d0 = src->d0;
dd89bb
	dst->d1 = src->d1;
dd89bb
d02e49
	if (addrlen == sizeof(nt_sockaddr_in6)) {
d02e49
		dst->d2 = src->d2;
d02e49
		dst->d3 = src->d3;
c308d4
	} else {
c308d4
		dst->d2 = 0;
c308d4
		dst->d3 = 0;
d02e49
	}
d02e49
dd89bb
	afd_connect_req.paddr = &(afd_connect_req.addr);
dd89bb
	afd_connect_req.addr.sa_addr_in4.sa_family = hssocket->domain;
dd89bb
dd89bb
	hssocket->iostatus = __ntapi->zw_device_io_control_file(
dd89bb
			hssocket->hsocket,
dd89bb
			hssocket->hevent,
dd89bb
			0,
dd89bb
			0,
dd89bb
			iosb,
dd89bb
			NT_AFD_IOCTL_CONNECT,
dd89bb
			&afd_connect_req,
dd89bb
			sizeof(afd_connect_req),
dd89bb
			(void *)0,
dd89bb
			0);
dd89bb
dd89bb
	return hssocket->iostatus
dd89bb
		? __ntapi->sc_wait(hssocket,iosb,0)
dd89bb
		: NT_STATUS_SUCCESS;
dd89bb
}