| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "quiche/quic/core/quic_socket_address_coder.h" |
| |
| #include <cstring> |
| #include <string> |
| #include <vector> |
| |
| #include "quiche/quic/platform/api/quic_ip_address_family.h" |
| |
| namespace quic { |
| |
| namespace { |
| |
| // For convenience, the values of these constants match the values of AF_INET |
| // and AF_INET6 on Linux. |
| const uint16_t kIPv4 = 2; |
| const uint16_t kIPv6 = 10; |
| |
| } // namespace |
| |
| QuicSocketAddressCoder::QuicSocketAddressCoder() {} |
| |
| QuicSocketAddressCoder::QuicSocketAddressCoder(const QuicSocketAddress& address) |
| : address_(address) {} |
| |
| QuicSocketAddressCoder::~QuicSocketAddressCoder() {} |
| |
| std::string QuicSocketAddressCoder::Encode() const { |
| std::string serialized; |
| uint16_t address_family; |
| switch (address_.host().address_family()) { |
| case IpAddressFamily::IP_V4: |
| address_family = kIPv4; |
| break; |
| case IpAddressFamily::IP_V6: |
| address_family = kIPv6; |
| break; |
| default: |
| return serialized; |
| } |
| serialized.append(reinterpret_cast<const char*>(&address_family), |
| sizeof(address_family)); |
| serialized.append(address_.host().ToPackedString()); |
| uint16_t port = address_.port(); |
| serialized.append(reinterpret_cast<const char*>(&port), sizeof(port)); |
| return serialized; |
| } |
| |
| bool QuicSocketAddressCoder::Decode(const char* data, size_t length) { |
| uint16_t address_family; |
| if (length < sizeof(address_family)) { |
| return false; |
| } |
| memcpy(&address_family, data, sizeof(address_family)); |
| data += sizeof(address_family); |
| length -= sizeof(address_family); |
| |
| size_t ip_length; |
| switch (address_family) { |
| case kIPv4: |
| ip_length = QuicIpAddress::kIPv4AddressSize; |
| break; |
| case kIPv6: |
| ip_length = QuicIpAddress::kIPv6AddressSize; |
| break; |
| default: |
| return false; |
| } |
| if (length < ip_length) { |
| return false; |
| } |
| std::vector<uint8_t> ip(ip_length); |
| memcpy(&ip[0], data, ip_length); |
| data += ip_length; |
| length -= ip_length; |
| |
| uint16_t port; |
| if (length != sizeof(port)) { |
| return false; |
| } |
| memcpy(&port, data, length); |
| |
| QuicIpAddress ip_address; |
| ip_address.FromPackedString(reinterpret_cast<const char*>(&ip[0]), ip_length); |
| address_ = QuicSocketAddress(ip_address, port); |
| return true; |
| } |
| |
| } // namespace quic |