/********************************************************/
/*  ntapi: Native API core library                      */
/*  Copyright (C) 2013--2017  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_socket_attr {
	uint32_t	datagram;
	uint32_t	unknown;
	uint32_t	domain;
	uint32_t	type;
	uint32_t	protocol;
} nt_socket_attr;

typedef struct _nt_afd_socket_ea {
	uint32_t	next_entry_offset;
	unsigned char	ea_flags;
	unsigned char	ea_name_length;
	uint16_t	ea_value_length;
	char		afd_open_packet[16];
	nt_socket_attr	sattr;
	uint32_t	ea_ext[4];
} nt_afd_socket_ea;

int32_t __cdecl __ntapi_sc_socket_v2(
	__out	nt_socket *		hssocket,
	__in	int32_t			domain,
	__in	int32_t			type,
	__in	uint32_t		protocol,
	__in	uint32_t		desired_access	__optional,
	__in	nt_sqos *		sqos		__optional,
	__out	nt_io_status_block *	iosb		__optional)
{
	int32_t			status;
	nt_object_attributes	oa;
	nt_io_status_block	siosb;
	nt_sqos			ssqos;
	nt_unicode_string	nt_afdep;
	uint32_t		ea_length;
	uint16_t		sdomain;
	uint16_t		stype;
	void *			_hsocket;

	wchar16_t afd_end_point[] = {
		'\\','D','e','v','i','c','e',
		'\\','A','f','d',
		'\\','E','n','d','P','o','i','n','t',
		0};

	nt_afd_socket_ea afd_ea = {
		0,
		0,
		0x0f,
		0x20,
		{'A','f','d','O','p','e','n','P','a','c','k','e','t','X','X',0},
		{0,0,0,0,0},
		{0}};

	if (domain >= 0x10000)
		return NT_STATUS_INVALID_PARAMETER_2;
	else if (type > 0x10000)
		return NT_STATUS_INVALID_PARAMETER_3;

	ea_length = sizeof(afd_ea);

	afd_ea.sattr.domain	= domain;
	afd_ea.sattr.type	= type;
	afd_ea.sattr.protocol	= protocol;

	afd_ea.sattr.datagram	= (type == NT_SOCK_DGRAM) ? protocol : 0;

	__ntapi->rtl_init_unicode_string(&nt_afdep,afd_end_point);

	if (!desired_access)
		desired_access = NT_GENERIC_READ \
				| NT_GENERIC_WRITE \
				| NT_SEC_SYNCHRONIZE \
				| NT_SEC_WRITE_DAC;

	if (!sqos) {
		ssqos.length 			= sizeof(ssqos);
		ssqos.impersonation_level	= NT_SECURITY_IMPERSONATION;
		ssqos.context_tracking_mode	= NT_SECURITY_TRACKING_DYNAMIC;
		ssqos.effective_only		= 1;
		sqos 				= &ssqos;
	}

	oa.len		= sizeof(oa);
	oa.root_dir	= (void *)0;
	oa.obj_name	= &nt_afdep;
	oa.obj_attr	= NT_OBJ_CASE_INSENSITIVE | NT_OBJ_INHERIT;
	oa.sec_desc	= (nt_security_descriptor *)0;
	oa.sec_qos	= sqos;

	iosb = iosb ? iosb : &siosb;

	if ((status = __ntapi->zw_create_file(
			&_hsocket,
			desired_access,
			&oa,
			iosb,
			0,
			0,
			NT_FILE_SHARE_READ | NT_FILE_SHARE_WRITE,
			NT_FILE_OPEN_IF,
			0,
			&afd_ea,
			ea_length)))
		return status;

	oa.obj_name = 0;
	oa.obj_attr = 0;

	sdomain = (uint16_t)domain;
	stype   = (uint16_t)type;

	hssocket->hsocket	= _hsocket;
	hssocket->ntflags	= 0;
	hssocket->domain 	= sdomain;
	hssocket->type		= stype;
	hssocket->protocol	= protocol;
	hssocket->timeout.quad	= 0;
	hssocket->iostatus	= NT_STATUS_SUCCESS;
	hssocket->waitstatus	= NT_STATUS_SUCCESS;

	return status;
}