Skip to content

Add Custom IP/Port, Support Xbox360/X1S/XSS Vibration, Auto Disconnect for timeout 15s #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
QeeAI wants to merge 5 commits into
base: master
Choose a base branch
from
Open

Add Custom IP/Port, Support Xbox360/X1S/XSS Vibration, Auto Disconnect for timeout 15s #2

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
1bb3655
Update v1.12
QeeAI Jan 31, 2023
d1c2bdf
Update v1.13
QeeAI Feb 1, 2023
ab1b4f5
GamePads depended By Client Mac Address and Xbox controller Index Id
QeeAI Feb 4, 2023
1fadff1
Added check valid for client mac address.
QeeAI Feb 5, 2023
408d0b0
Added check valid for client mac address.
QeeAI Feb 5, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
200 changes: 168 additions & 32 deletions NetInput.Capture/NetInput.Capture.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,57 +12,184 @@
#include <string>
#include <stdint.h>
#include <fstream>
#include <nb30.h>
#pragma comment(lib, "Netapi32.lib")

typedef struct _ASTAT_
{
ADAPTER_STATUS adapt;
NAME_BUFFER NameBuff[30];
} ASTAT, * PASTAT;


#define PAD_SERVER_SIZE 5
#define PAD_CLIENT_SIZE 24
#define MAC_SIZE 6

#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[MAC_SIZE];

bool checkvalidmac(uint8_t hMac[])
{
for (int i = 0; i < MAC_SIZE; i++) {
if (hMac[i] > 0x00u && hMac[i] < uint8_t(0xFFu))
return true;
}
return false;
}


bool 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 < MAC_SIZE; j++) {
hmac[j] = int(Adapter.adapt.adapter_address[j]);
}

if (checkvalidmac(hmac))
return true;
}
}
return false;
}


void CheckControllerMessage()
{
while (true)
{
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_SERVER_SIZE){
uint8_t i = (uint8_t)packet[1];
if ((i >= 0) && (i < XUSER_MAX_COUNT)) {
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("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] == (uint8_t)PAD_VIBRA) {
XINPUT_VIBRATION Vibration;
Vibration.wLeftMotorSpeed = packet[2] << 8;
Vibration.wRightMotorSpeed = packet[3] << 8;
XInputSetState(i, &Vibration);
}
}
}
}
}


void SendResetControllers()
{
uint8_t packet[] = { 0xFFu };
uint8_t packet[1+ MAC_SIZE];
packet[0] = (uint8_t)PAD_RESET;
memcpy(packet + 1, &hmac, MAC_SIZE);
if (sendto(sock, (const char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, sizeof(addr)) != sizeof(packet))
printf("Failed to send reset message.\n");
}

void PollControllers()
{
XINPUT_STATE state;
for (uint32_t i = 0u; i < XUSER_MAX_COUNT; i++)
for (uint8_t i = 0u; i < XUSER_MAX_COUNT; i++)
{
memset(&state, 0, sizeof(XINPUT_STATE));
if (XInputGetState(i, &state) != ERROR_SUCCESS)
continue;
continue;

if (time(NULL) - updatetime[i] >= 5)
{
uint8_t packet[MAC_SIZE+2];
packet[0] = (uint8_t)PAD_REG;
packet[1] = (uint8_t)i;
memcpy(packet + 2, &hmac, MAC_SIZE);
if (sendto(sock, (const char*)&packet, sizeof(packet), 0, (struct sockaddr*)&addr, sizeof(addr)) != sizeof(packet))
printf("Failed to checktimeOut of client controller %u .\n",i);
updatetime[i] = time(NULL);
}

if (memcmp(lastSentInputStates + i, &state, sizeof(XINPUT_STATE)) == 0)
continue;
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, MAC_SIZE);
memcpy(packet + MAC_SIZE + 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);
}
}

int main()
int main(int argc, char* argv[])
{
std::string ip;

std::ifstream input_file("target.txt");
if (input_file.is_open())
{
printf("Reading ip from target.txt.\n");
ip = std::string((std::istreambuf_iterator<char>(input_file)), std::istreambuf_iterator<char>());

if (inet_pton(AF_INET, ip.c_str(), &(addr.sin_addr)) != 1)
{
std::cout << ip << " is not a valid ip, please correct target.txt\n";
return -1;
}
int port = 4313;

if (argc > 1) {
ip = argv[1];
}

if (argc > 2) {
sscanf_s(argv[2], "%d", &port);
}

if (ip.empty())
{
while (true)
Expand All @@ -77,12 +204,24 @@ int main()
}
}

printf("Target is %s.\n", ip.c_str());
if (inet_pton(AF_INET, ip.c_str(), &(addr.sin_addr)) != 1)
{
std::cout << ip << " is not a valid ip, please try again ! \n";
return -1;
}

CoInitialize(NULL);
if (!getMac()) {
printf("Failed to get mac address .\n");
return -1;
}

printf("Starting networking...\n");
printf("Connected to server %s:%d.\n", ip.c_str(),port);

CoInitialize(NULL);

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)
Expand All @@ -99,23 +238,20 @@ int main()
}

addr.sin_family = AF_INET;
addr.sin_port = htons(4313);
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

printf("Waiting for gamepad input, press ESC to exit...\n");
memset(lastSentInputStates, 0, sizeof(lastSentInputStates));
while (true)
{
if (GetKeyState(VK_ESCAPE) & 0x8000)
break;

PollControllers();
std::this_thread::sleep_for(std::chrono::milliseconds(1));
}

printf("Wait For Closing...\n");
SendResetControllers();
closesocket(sock);
WSACleanup();
Expand Down
Loading