void IPv4AddressToString(IPv4Address address, std::u16string& buffer) { const char* const InvalidIPv4AddressStr = "Invalid IPv4 address!"; const auto& addressBytes = address.Bytes; const in_addr addr = { static_cast<uchar>(addressBytes[0]), static_cast<uchar>(addressBytes[1]), static_cast<uchar>(addressBytes[2]), static_cast<uchar>(addressBytes[3]) }; const auto& ntFunctions = GetNtFunctions(); //If the length of the buffer pointed to by the AddressString parameter is not large enough //to receive the string representation of the IPv4 address and port, RtlIpv4AddressToStringEx //returns ERROR_INVALID_PARAMETER and sets the AddressStringLength parameter to the buffer length required. ulong bufferLength = {}; auto errCode = ntFunctions.RtlIpv4AddressToStringEx(&addr, 0, nullptr, &bufferLength); //如果 bufferLength 在调用后依旧为 0 说明 address 有问题。 Try<std::invalid_argument>(bufferLength == 0, InvalidIPv4AddressStr); buffer.resize(static_cast<std::size_t>(bufferLength)); errCode = ntFunctions.RtlIpv4AddressToStringEx(&addr, 0, reinterpret_cast<wchar_t*>(&buffer[0]), &bufferLength); //如果有错误,前面必然已经抛出异常,故这里必然是 Success。 assert(errCode == NtFunctions::NtSuccess); assert(bufferLength > 0); //在写入的 buffer 中包含 '\0',会导致判等出现奇怪的结果,故舍弃掉结尾的空字符。 buffer.pop_back(); }
void IPv4StringToAddress(const char16_t* ipString, gsl::span<std::uint8_t> bytes, std::uint16_t& port) { const auto& ntFunctions = GetNtFunctions(); const auto ipStr = reinterpret_cast<const wchar_t*>(ipString); ::in_addr v4Addr = {}; ushort tempPort = {}; Try<std::invalid_argument>(ntFunctions.RtlIpv4StringToAddressEx(ipStr, 0, &v4Addr, &tempPort) == NtFunctions::NtSuccess, "Invalid IPv4 address string!"); bytes[0] = static_cast<std::uint8_t>(v4Addr.s_net); bytes[1] = static_cast<std::uint8_t>(v4Addr.s_host); bytes[2] = static_cast<std::uint8_t>(v4Addr.s_lh); bytes[3] = static_cast<std::uint8_t>(v4Addr.s_impno); port = static_cast<std::uint16_t>(tempPort); }
void IPv6StringToAddress(const char16_t* ipString, gsl::span<std::uint8_t> bytes, std::uint32_t& scope, std::uint16_t& port) { const auto& ntFunctions = GetNtFunctions(); const auto ipStr = reinterpret_cast<const wchar_t*>(ipString); ::in6_addr v6Addr = {}; ulong tempScope = {}; ushort tempPort = {}; Try<std::invalid_argument>(ntFunctions.RtlIpv6StringToAddressEx(ipStr, &v6Addr, &tempScope, &tempPort) == NtFunctions::NtSuccess, "Invalid IPv4 address string!"); auto& addrBytes = v6Addr.s6_addr; for (std::size_t i = 0; i < kIPv6AddressBytes; ++i) bytes[i] = static_cast<std::uint8_t>(addrBytes[i]); scope = static_cast<std::uint32_t>(tempScope); port = static_cast<std::uint16_t>(tempPort); }
BOOL Startup() { NTSTATUS ret; _wfopen_s(&fp, L"C:\\Log\\logger.txt", L"wb"); if (!GetNtFunctions()) return FALSE; if (fp) { if (LdrRegisterDllNotification) { ret = LdrRegisterDllNotification(0, &MyDllNotification, NULL, &Cookie); if (ret != STATUS_SUCCESS) return FALSE; else return TRUE; } } return FALSE; }
void IPv6AddressToString(IPv6Address address, std::uint32_t scopeId, std::u16string& buffer) { const char* const InvalidIPv6AddressStr = "Invalid IPv6 address!"; ::in6_addr v6Addr; auto& desAddrBytes = v6Addr.s6_addr; auto& srcAddrBytes = address.Bytes; for (std::size_t i = 0; i < kIPv4AddressBytes; ++i) desAddrBytes[i] = static_cast<uchar>(srcAddrBytes[i]); const auto& ntFunctions = GetNtFunctions(); ulong bufferLength = {}; auto errCode = ntFunctions.RtlIpv6AddressToStringEx(&v6Addr, static_cast<ulong>(scopeId), 0, nullptr, &bufferLength); Try<std::invalid_argument>(bufferLength != 0, InvalidIPv6AddressStr); buffer.resize(static_cast<std::size_t>(bufferLength)); errCode = ntFunctions.RtlIpv6AddressToStringEx(&v6Addr, static_cast<ulong>(scopeId), 0, reinterpret_cast<wchar_t*>(&buffer[0]), &bufferLength); assert(errCode == NtFunctions::NtSuccess); buffer.pop_back(); }