Lightemerald/windows-user-space-emulator
2
0
Fork 0
mirror of https://github.com/momo5502/emulator.git synced 2026-01-25 06:31:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Prepare support for more socket types

Browse Source
This commit is contained in:
momo5502
2024-11-08 20:41:06 +01:00
parent b367db3584
commit a63ae4e782
7 changed files with 251 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

160
src/windows-emulator/devices/afd_endpoint.cpp
View File

@@ -1,87 +1,125 @@
#include "afd_endpoint.hpp"
#include "afd_types.hpp"
#include "../windows_emulator.hpp"
#include <network/address.hpp>
#include <network/socket.hpp>
typedef LONG TDI_STATUS;
typedef PVOID CONNECTION_CONTEXT;
typedef struct _TDI_CONNECTION_INFORMATION
{
LONG UserDataLength;
PVOID UserData;
LONG OptionsLength;
PVOID Options;
LONG RemoteAddressLength;
PVOID RemoteAddress;
} TDI_CONNECTION_INFORMATION, *PTDI_CONNECTION_INFORMATION;
typedef struct _TDI_REQUEST
{
union
{
HANDLE AddressHandle;
CONNECTION_CONTEXT ConnectionContext;
HANDLE ControlChannel;
} Handle;
PVOID RequestNotifyObject;
PVOID RequestContext;
TDI_STATUS TdiStatus;
} TDI_REQUEST, *PTDI_REQUEST;
typedef struct _TDI_REQUEST_SEND_DATAGRAM
{
TDI_REQUEST Request;
PTDI_CONNECTION_INFORMATION SendDatagramInformation;
} TDI_REQUEST_SEND_DATAGRAM, *PTDI_REQUEST_SEND_DATAGRAM;
typedef struct _AFD_SEND_DATAGRAM_INFO
{
LPWSABUF BufferArray;
ULONG BufferCount;
ULONG AfdFlags;
TDI_REQUEST_SEND_DATAGRAM TdiRequest;
TDI_CONNECTION_INFORMATION TdiConnInfo;
} AFD_SEND_DATAGRAM_INFO, *PAFD_SEND_DATAGRAM_INFO;
namespace
{
struct afd_endpoint : stateless_device
struct afd_creation_data
{
network::socket s{AF_INET};
uint64_t unk1;
char afd_open_packet_xx[0x10];
uint64_t unk2;
int address_family;
int type;
int protocol;
// ...
};
afd_creation_data get_creation_data(const io_device_creation_data& data)
{
if (!data.buffer || data.length < sizeof(afd_creation_data))
{
throw std::runtime_error("Bad AFD creation data");
}
return emulator_object<afd_creation_data>{data.emu, data.buffer}.read();
}
struct afd_endpoint : io_device
{
std::optional<SOCKET> s{};
afd_endpoint()
{
network::initialize_wsa();
}
afd_endpoint(afd_endpoint&&) = delete;
afd_endpoint& operator=(afd_endpoint&&) = delete;
~afd_endpoint() override
{
if (this->s)
{
closesocket(*this->s);
}
}
void create(const io_device_creation_data& data) override
{
const auto creation_data = get_creation_data(data);
const auto sock = socket(creation_data.address_family, creation_data.type, creation_data.protocol);
if (sock == INVALID_SOCKET)
{
throw std::runtime_error("Failed to create socket!");
}
s = sock;
}
void deserialize(utils::buffer_deserializer&) override
{
// TODO
}
void serialize(utils::buffer_serializer&) const override
{
// TODO
}
NTSTATUS io_control(const io_device_context& c) override
{
c.io_status_block.write({});
if (_AFD_BASE(c.io_control_code) != FSCTL_AFD_BASE)
{
c.win_emu.logger.print(color::cyan, "Bad AFD IOCTL: %X\n", c.io_control_code);
return STATUS_NOT_SUPPORTED;
}
c.win_emu.logger.print(color::cyan, "AFD IOCTL: %X\n", c.io_control_code);
switch (c.io_control_code)
const auto request = _AFD_REQUEST(c.io_control_code);
switch (request)
{
case 0x12003:
case AFD_BIND:
return this->ioctl_bind(c);
case 0x12023:
case AFD_SEND_DATAGRAM:
return this->ioctl_send_datagram(c);
case 0x12047: // ?
case 0x1207B: // ?
case AFD_SET_CONTEXT:
return STATUS_SUCCESS;
case AFD_GET_INFORMATION:
return STATUS_SUCCESS;
}
return STATUS_SUCCESS;
c.win_emu.logger.print(color::gray, "Unsupported AFD IOCTL: %X\n", c.io_control_code);
return STATUS_NOT_SUPPORTED;
}
NTSTATUS ioctl_bind(const io_device_context& c)
NTSTATUS ioctl_bind(const io_device_context& c) const
{
std::vector<std::byte> data{};
data.resize(c.input_buffer_length);
c.emu.read_memory(c.input_buffer, data.data(), c.input_buffer_length);
utils::buffer_deserializer deserializer{data, true};
deserializer.read<uint32_t>(); // IDK :(
const network::address addr = deserializer.read<sockaddr_in>();
constexpr auto address_offset = 4;
if (!this->s.bind_port(addr))
if (data.size() < address_offset)
{
return STATUS_BUFFER_TOO_SMALL;
}
const auto* address = reinterpret_cast<const sockaddr*>(data.data() + address_offset);
const auto address_size = static_cast<int>(data.size() - address_offset);
const network::address addr(address, address_size);
if (bind(*this->s, &addr.get_addr(), addr.get_size()) == SOCKET_ERROR)
{
return STATUS_ADDRESS_ALREADY_ASSOCIATED;
}
@@ -89,7 +127,7 @@ namespace
return STATUS_SUCCESS;
}
NTSTATUS ioctl_send_datagram(const io_device_context& c)
NTSTATUS ioctl_send_datagram(const io_device_context& c) const
{
if (c.input_buffer_length < sizeof(AFD_SEND_DATAGRAM_INFO))
{
@@ -111,7 +149,11 @@ namespace
data.resize(buffer.len);
c.emu.read_memory(reinterpret_cast<uint64_t>(buffer.buf), data.data(), data.size());
if (!s.send(target, data.data(), data.size()))
const auto sent_data = sendto(*this->s, reinterpret_cast<const char*>(data.data()),
static_cast<int>(data.size()), 0 /* ? */, &target.get_addr(),
target.get_size());
if (sent_data < 0)
{
return STATUS_CONNECTION_REFUSED;
}
@@ -119,7 +161,7 @@ namespace
if (c.io_status_block)
{
IO_STATUS_BLOCK block{};
block.Information = data.size();
block.Information = static_cast<uint32_t>(sent_data);
c.io_status_block.write(block);
}