#include <WS2tcpip.h>
#include<string>
#pragma warning (disable:4996)
static std::string get_ip_this_a() {
	WORD wVersionRequested = MAKEWORD(2, 2);
	WSADATA wsaData;
	if (WSAStartup(wVersionRequested, &wsaData) != 0)
		return "";
	char local[255] = { 0 };
	gethostname(local, sizeof(local));
	hostent* ph = gethostbyname(local);
	if (ph == NULL)
		return "";
	in_addr addr;
	memcpy(&addr, ph->h_addr_list[0], sizeof(in_addr));
	std::string localIP;
	localIP.assign(inet_ntoa(addr));
	WSACleanup();
	return localIP;
}

/*此页代码调试不好了，服了*/
static bool Ipv6_send_net_msg(const std::string& serverIP, u_short port, LPBYTE msg, size_t len) {
	//初始化
	WSADATA wsaData;
	int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
	if (iResult != 0) {
		return false;
	}

	// 创建 socket
	SOCKET sock = INVALID_SOCKET;
	sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (sock == INVALID_SOCKET) {
		WSACleanup();
		return false;
	}
	sockaddr_in server;
	server.sin_family = AF_INET;
	server.sin_port = htons(port);


	//判断是否是IPv4

	if (inet_pton(AF_INET, serverIP.data(), &server.sin_addr) == 1) {
		//IPV4不做特殊处理
	}
	else {
		sockaddr_in6 server6;
		server6.sin6_family = AF_INET6;
		server6.sin6_port = htons(port);
		//判断是否是IPv6
		if (InetPtonA(AF_INET6, serverIP.data(), &server6.sin6_addr) == 1) {
			server = reinterpret_cast<sockaddr_in&>(server6);
		}
		else {
			closesocket(sock);
			WSACleanup();
			return false;
		}
	}

	// 连接
	if (connect(sock, (sockaddr*)&server, sizeof(server)) == SOCKET_ERROR) {
		closesocket(sock);
		WSACleanup();
		return false;
	}

	// 发送
	int iSendResult = send(sock, (char*)msg, len, 0);
	if (iSendResult == SOCKET_ERROR) {
		closesocket(sock);
		WSACleanup();
		return false;
	}

	// 释放资源
	closesocket(sock);
	WSACleanup();
	return true;
}


#include "..\include_eapi_header.h"

// 调用格式: SDT_BOOL 发送局域网消息, 命令说明: "发送消息到指定的机器（即Windows信使服务，只有启动了Messenger服务的机器才能收到该命令发送的消息）。成功返回真；失败返回假。"
// 参数<1>: [发送目标 SDT_TEXT], 参数说明: "将要接收消息的目标机器，可以为机器名、IP地址、工作组名（具体由目标类型决定）"
// 参数<2>: 消息 SDT_TEXT, 参数说明: "将要发送的消息文本"
// 参数<3>: [保留参数 SDT_INT], 参数说明: NULL
EAPI_EXTERN_C void eapi_NetSendMessage_80_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	auto ip = args_to_sdata(pArgInf, 0);
	LPBYTE data = reinterpret_cast<LPBYTE>(pArgInf[2].m_pBin + sizeof(std::uint32_t) * 2);
	auto datalent = *reinterpret_cast<std::uint32_t*>(pArgInf[2].m_pBin + sizeof(std::uint32_t));
	if (datalent > 0 && data)
	{
		pRetData->m_bool = Ipv6_send_net_msg(ip.empty() ? get_ip_this_a() : std::string(ip), pArgInf[1].m_short, data, datalent);
		return;
	}
	pRetData->m_bool = false;
}