Skip to content

Commit

Permalink
Translate socket values
Browse files Browse the repository at this point in the history
  • Loading branch information
@momo5502
momo5502 committed Jan 26, 2025
1 parent 72a3123 commit a6362b0
Showing 1 changed file with 91 additions and 10 deletions.
101 changes: 91 additions & 10 deletions src/windows-emulator/devices/afd_endpoint.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,73 @@ namespace
// ...
};

const std::map<int, int> address_family_map{
{0, AF_UNSPEC}, //
{2, AF_INET}, //
{23, AF_INET6}, //
};

const std::map<int, int> socket_type_map{
{0, 0}, //
{1, SOCK_STREAM}, //
{2, SOCK_DGRAM}, //
{3, SOCK_RAW}, //
{4, SOCK_RDM}, //
};

const std::map<int, int> socket_protocol_map{
{0, 0}, //
{6, IPPROTO_TCP}, //
{17, IPPROTO_UDP}, //
{255, IPPROTO_RAW}, //
};

int translate_host_to_win_address_family(const int host_af)
{
for (auto& entry : address_family_map)
{
if (entry.second == host_af)
{
return entry.first;
}
}

throw std::runtime_error("Unknown host address family: " + std::to_string(host_af));
}

int translate_address_family(const int win_af)
{
const auto entry = address_family_map.find(win_af);
if (entry != address_family_map.end())
{
return entry->second;
}

throw std::runtime_error("Unknown address family: " + std::to_string(win_af));
}

int translate_type(const int win_type)
{
const auto entry = socket_type_map.find(win_type);
if (entry != socket_type_map.end())
{
return entry->second;
}

throw std::runtime_error("Unknown socket type: " + std::to_string(win_type));
}

int translate_protocol(const int win_protocol)
{
const auto entry = socket_protocol_map.find(win_protocol);
if (entry != socket_protocol_map.end())
{
return entry->second;
}

throw std::runtime_error("Unknown socket protocol: " + std::to_string(win_protocol));
}

afd_creation_data get_creation_data(windows_emulator& win_emu, const io_device_creation_data& data)
{
if (!data.buffer || data.length < sizeof(afd_creation_data))
Expand Down Expand Up @@ -216,8 +283,11 @@ namespace

const auto& data = *this->creation_data;

// TODO: values map to windows values; might not be the case for other platforms
const auto sock = socket(data.address_family, data.type, data.protocol);
const auto af = translate_address_family(data.address_family);
const auto type = translate_type(data.type);
const auto protocol = translate_protocol(data.protocol);

const auto sock = socket(af, type, protocol);
if (sock == INVALID_SOCKET)
{
throw std::runtime_error("Failed to create socket!");
Expand Down Expand Up @@ -339,7 +409,7 @@ namespace

NTSTATUS ioctl_bind(windows_emulator& win_emu, const io_device_context& c) const
{
const auto data = win_emu.emu().read_memory(c.input_buffer, c.input_buffer_length);
auto data = win_emu.emu().read_memory(c.input_buffer, c.input_buffer_length);

constexpr auto address_offset = 4;

Expand All @@ -348,9 +418,12 @@ namespace
return STATUS_BUFFER_TOO_SMALL;
}

const auto* address = reinterpret_cast<const sockaddr*>(data.data() + address_offset);
auto* address = reinterpret_cast<sockaddr*>(data.data() + address_offset);
const auto address_size = static_cast<socklen_t>(data.size() - address_offset);

address->sa_family =
static_cast<decltype(address->sa_family)>(translate_address_family(address->sa_family));

const network::address addr(address, address_size);

if (bind(*this->s_, &addr.get_addr(), addr.get_size()) == SOCKET_ERROR)
Expand Down Expand Up @@ -451,8 +524,10 @@ namespace
std::vector<char> data{};
data.resize(buffer.len);

const auto recevied_data = recvfrom(*this->s_, data.data(), static_cast<send_size>(data.size()), 0,
reinterpret_cast<sockaddr*>(address.data()), &fromlength);
auto* sender = reinterpret_cast<sockaddr*>(address.data());

const auto recevied_data =
recvfrom(*this->s_, data.data(), static_cast<send_size>(data.size()), 0, sender, &fromlength);

if (recevied_data < 0)
{
Expand All @@ -466,6 +541,10 @@ namespace
return STATUS_UNSUCCESSFUL;
}

// TODO: Translate rest of address struct?
sender->sa_family =
static_cast<decltype(sender->sa_family)>(translate_host_to_win_address_family(sender->sa_family));

const auto data_size = std::min(data.size(), static_cast<size_t>(recevied_data));
emu.write_memory(buffer.buf, data.data(), data_size);

Expand Down Expand Up @@ -497,11 +576,13 @@ namespace
const auto send_info = emu.read_memory<AFD_SEND_DATAGRAM_INFO<EmulatorTraits<Emu64>>>(c.input_buffer);
const auto buffer = emu.read_memory<EMU_WSABUF<EmulatorTraits<Emu64>>>(send_info.BufferArray);

const auto address = emu.read_memory(send_info.TdiConnInfo.RemoteAddress,
static_cast<size_t>(send_info.TdiConnInfo.RemoteAddressLength));
auto address_buffer = emu.read_memory(send_info.TdiConnInfo.RemoteAddress,
static_cast<size_t>(send_info.TdiConnInfo.RemoteAddressLength));
auto* address = reinterpret_cast<sockaddr*>(address_buffer.data());
address->sa_family =
static_cast<decltype(address->sa_family)>(translate_address_family(address->sa_family));

const network::address target(reinterpret_cast<const sockaddr*>(address.data()),
static_cast<socklen_t>(address.size()));
const network::address target(address, static_cast<socklen_t>(address_buffer.size()));

const auto data = emu.read_memory(buffer.buf, buffer.len);

Expand Down

0 comments on commit a6362b0

Please sign in to comment.