GamePads depended By Client Mac Address and Xbox controller Index Id

Author: QeeAI

NetInput.Capture/NetInput.Capture.cpp

@@ -12,44 +12,100 @@
 #include <string>
 #include <stdint.h>
 #include <fstream>
+#include <nb30.h>
+#pragma comment(lib, "Netapi32.lib")
 
-#define PAD_OUT_SIZE 5
-#define PAD_ONLINE 0x00
-#define PAD_VIBRA 0x01
+typedef struct _ASTAT_
+{
+	ADAPTER_STATUS adapt;
+	NAME_BUFFER NameBuff[30];
+} ASTAT, * PASTAT;
+
+
+#define PAD_SERVER_SIZE 5
+#define PAD_CLIENT_SIZE 24
+
+#define PAD_ONLINE 0x00u
+#define PAD_VIBRA 0x01u
+#define PAD_REG 0x03u
+#define PAD_STATE 0x04u
+#define PAD_RESET 0xFFu
 
 SOCKET sock;
 struct sockaddr_in addr;
 XINPUT_STATE lastSentInputStates[XUSER_MAX_COUNT];
 time_t updatetime[XUSER_MAX_COUNT];
+uint8_t hmac[6];
+
+void getMac()
+{
+	ASTAT Adapter;
+	NCB Ncb;
+	UCHAR uRetCode;
+	LANA_ENUM lenum;
+	int i = 0;
+
+	memset(&Ncb, 0, sizeof(Ncb));
+	Ncb.ncb_command = NCBENUM;
+	Ncb.ncb_buffer = (UCHAR*)&lenum;
+	Ncb.ncb_length = sizeof(lenum);
+
+	uRetCode = Netbios(&Ncb);
+
+	for (i = 0; i < lenum.length; i++)
+	{
+		memset(&Ncb, 0, sizeof(Ncb));
+		Ncb.ncb_command = NCBRESET;
+		Ncb.ncb_lana_num = lenum.lana[i];
+		uRetCode = Netbios(&Ncb);
+
+		memset(&Ncb, 0, sizeof(Ncb));
+		Ncb.ncb_command = NCBASTAT;
+		Ncb.ncb_lana_num = lenum.lana[i];
+		strcpy_s((char*)Ncb.ncb_callname,  sizeof(Ncb.ncb_callname), "*");
+		Ncb.ncb_buffer = (unsigned char*)&Adapter;
+		Ncb.ncb_length = sizeof(Adapter);
+		uRetCode = Netbios(&Ncb);
+
+		if (uRetCode == 0)
+		{
+			for (int j = 0; j < sizeof(hmac); j++) {
+				hmac[j] = int(Adapter.adapt.adapter_address[j]);
+			}		
+			break;
+		}
+	}
+}
+
 
 void CheckControllerMessage()
 { 	
 	while (true)
 	{		
-		char packet[PAD_OUT_SIZE];
+		char packet[PAD_SERVER_SIZE];
 		int addr_size = sizeof(addr);
 		int bytesReceived = recvfrom(sock, (char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, &addr_size);	
 
-		if (bytesReceived == PAD_OUT_SIZE){	
+		if (bytesReceived == PAD_SERVER_SIZE){
 			uint8_t i = (uint8_t)packet[1];
 			if ((i >= 0) && (i < XUSER_MAX_COUNT)) {
-				if (packet[0] == PAD_ONLINE) {
+				if (packet[0] == (uint8_t)PAD_ONLINE) {
 					XINPUT_VIBRATION Vibration;
 
 					uint8_t pindex = (uint8_t)packet[4];
 					Vibration.wLeftMotorSpeed = 0;
 					Vibration.wRightMotorSpeed = -1;
 					XInputSetState(i, &Vibration);
 
-					printf("Get Local controller %u as Online controller %u.\n", i,pindex);
+					printf("Connected client controller %u to server controller %u.\n", i,pindex);
 					std::this_thread::sleep_for(std::chrono::milliseconds(250));
 
 					Vibration.wLeftMotorSpeed = 0;
 					Vibration.wRightMotorSpeed = 0;
 					XInputSetState(i, &Vibration);
 
 				}
-				else if (packet[0] == PAD_VIBRA) {
+				else if (packet[0] == (uint8_t)PAD_VIBRA) {
 					XINPUT_VIBRATION Vibration;
 					Vibration.wLeftMotorSpeed = packet[2] << 8;
 					Vibration.wRightMotorSpeed = packet[3] << 8;
@@ -63,7 +119,9 @@ void CheckControllerMessage()
 
 void SendResetControllers()
 {
-	uint8_t packet[] = { 0xFFu };
+	uint8_t packet[1+sizeof(hmac)];
+	packet[0] = (uint8_t)PAD_RESET;
+	memcpy(packet + 1, &hmac, sizeof(hmac));
 	if (sendto(sock, (const char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, sizeof(addr)) != sizeof(packet))
 		printf("Failed to send reset message.\n");
 }
@@ -79,22 +137,29 @@ void PollControllers()
 
 		if (time(NULL) - updatetime[i] >= 5)
 		{
-			uint8_t packet[] = { 0xFEu ,i };
+			uint8_t packet[sizeof(hmac)+2];
+			packet[0] = (uint8_t)PAD_REG;
+			packet[1] = (uint8_t)i;
+			memcpy(packet + 2, &hmac, sizeof(hmac));		
 			if (sendto(sock, (const char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, sizeof(addr)) != sizeof(packet))
-				printf("Failed to CheckUpdateTimeOut!\n");
+				printf("Failed to checktimeOut of client controller %u .\n",i);
 			updatetime[i] = time(NULL);
 		}
 
 		if (memcmp(lastSentInputStates + i, &state, sizeof(XINPUT_STATE)) == 0)
 		  continue;		
 
-		uint8_t packet[sizeof(XINPUT_STATE) + 1];
-		packet[0] = (uint8_t)i;
-		memcpy(packet + 1, &state, sizeof(XINPUT_STATE));
+		uint8_t packet[PAD_CLIENT_SIZE];
+		
+		packet[0] = (uint8_t)PAD_STATE;
+		packet[1] = (uint8_t)i;
+
+		memcpy(packet + 2, &hmac, sizeof(hmac));	
+		memcpy(packet + sizeof(hmac) + 2, &state, sizeof(XINPUT_STATE));	
 		memcpy(lastSentInputStates + i, &state, sizeof(XINPUT_STATE));
 
 		if (sendto(sock, (const char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, sizeof(addr)) != sizeof(packet))
-			printf("Failed to send update message of controller %u.\n", i);
+			printf("Failed to send update message of client controller %u.\n", i);
 	}
 }
 
@@ -131,12 +196,14 @@ int main(int argc, char* argv[])
 		return -1;
 	}
 
-	printf("Target is %s:%d.\n", ip.c_str(),port);
+	printf("Connected to server %s:%d.\n", ip.c_str(),port);
 
 	CoInitialize(NULL);
-
-	printf("Starting networking...\n");
-
+	
+	getMac();
+	printf("Starting networking at %02X:%02X:%02X:%02X:%02X:%02X ...\n",
+		  hmac[0], hmac[1], hmac[2], hmac[3], hmac[4], hmac[5]);
+	
 	WSADATA wsaData;
 	int wsaStartupResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
 	if (wsaStartupResult != 0) 
@@ -155,9 +222,6 @@ int main(int argc, char* argv[])
 	addr.sin_family = AF_INET;
 	addr.sin_port = htons(port);
 
-	printf("Sending reset...\n");
-	SendResetControllers();
-	printf("Done.\n");
 	printf("Waiting for gamepad input, press Ctrl+C to exit...\n");
 
 	std::thread message(CheckControllerMessage); // new thread for message