address_utils.h 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. // Copyright (c) 2019 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef NET_QUIC_ADDRESS_UTILS_H_
  5. #define NET_QUIC_ADDRESS_UTILS_H_
  6. #include <string.h>
  7. #include "net/base/ip_address.h"
  8. #include "net/base/ip_endpoint.h"
  9. #include "net/third_party/quiche/src/quiche/quic/platform/api/quic_ip_address.h"
  10. #include "net/third_party/quiche/src/quiche/quic/platform/api/quic_ip_address_family.h"
  11. #include "net/third_party/quiche/src/quiche/quic/platform/api/quic_socket_address.h"
  12. namespace net {
  13. inline IPEndPoint ToIPEndPoint(quic::QuicSocketAddress address) {
  14. if (!address.IsInitialized()) {
  15. return IPEndPoint();
  16. }
  17. IPEndPoint result;
  18. sockaddr_storage storage = address.generic_address();
  19. const bool success = result.FromSockAddr(
  20. reinterpret_cast<const sockaddr*>(&storage), sizeof(storage));
  21. DCHECK(success);
  22. return result;
  23. }
  24. inline IPAddress ToIPAddress(quic::QuicIpAddress address) {
  25. if (!address.IsInitialized()) {
  26. return IPAddress();
  27. }
  28. switch (address.address_family()) {
  29. case quic::IpAddressFamily::IP_V4: {
  30. in_addr raw_address = address.GetIPv4();
  31. return IPAddress(reinterpret_cast<const uint8_t*>(&raw_address),
  32. sizeof(raw_address));
  33. }
  34. case quic::IpAddressFamily::IP_V6: {
  35. in6_addr raw_address = address.GetIPv6();
  36. return IPAddress(reinterpret_cast<const uint8_t*>(&raw_address),
  37. sizeof(raw_address));
  38. }
  39. default:
  40. DCHECK_EQ(address.address_family(), quic::IpAddressFamily::IP_UNSPEC);
  41. return IPAddress();
  42. }
  43. }
  44. inline quic::QuicSocketAddress ToQuicSocketAddress(IPEndPoint address) {
  45. if (address.address().empty()) {
  46. return quic::QuicSocketAddress();
  47. }
  48. sockaddr_storage result;
  49. socklen_t size = sizeof(result);
  50. if (!address.ToSockAddr(reinterpret_cast<sockaddr*>(&result), &size)) {
  51. return quic::QuicSocketAddress();
  52. }
  53. return quic::QuicSocketAddress(result);
  54. }
  55. inline quic::QuicIpAddress ToQuicIpAddress(net::IPAddress address) {
  56. if (address.IsIPv4()) {
  57. in_addr result;
  58. static_assert(sizeof(result) == IPAddress::kIPv4AddressSize,
  59. "Address size mismatch");
  60. memcpy(&result, address.bytes().data(), IPAddress::kIPv4AddressSize);
  61. return quic::QuicIpAddress(result);
  62. }
  63. if (address.IsIPv6()) {
  64. in6_addr result;
  65. static_assert(sizeof(result) == IPAddress::kIPv6AddressSize,
  66. "Address size mismatch");
  67. memcpy(&result, address.bytes().data(), IPAddress::kIPv6AddressSize);
  68. return quic::QuicIpAddress(result);
  69. }
  70. DCHECK(address.empty());
  71. return quic::QuicIpAddress();
  72. }
  73. } // namespace net
  74. #endif // NET_QUIC_ADDRESS_UTILS_H_