135 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			135 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
| /*
 | |
|  * Copyright (C) 2017 The Android Open Source Project
 | |
|  *
 | |
|  * Licensed under the Apache License, Version 2.0 (the "License");
 | |
|  * you may not use this file except in compliance with the License.
 | |
|  * You may obtain a copy of the License at
 | |
|  *
 | |
|  *      http://www.apache.org/licenses/LICENSE-2.0
 | |
|  *
 | |
|  * Unless required by applicable law or agreed to in writing, software
 | |
|  * distributed under the License is distributed on an "AS IS" BASIS,
 | |
|  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
|  * See the License for the specific language governing permissions and
 | |
|  * limitations under the License.
 | |
|  */
 | |
| 
 | |
| #include "DnsTlsServer.h"
 | |
| 
 | |
| #include <algorithm>
 | |
| 
 | |
| #include <netdutils/InternetAddresses.h>
 | |
| 
 | |
| namespace {
 | |
| 
 | |
| // Returns a tuple of references to the elements of a.
 | |
| auto make_tie(const sockaddr_in& a) {
 | |
|     return std::tie(a.sin_port, a.sin_addr.s_addr);
 | |
| }
 | |
| 
 | |
| // Returns a tuple of references to the elements of a.
 | |
| auto make_tie(const sockaddr_in6& a) {
 | |
|     // Skip flowinfo, which is not relevant.
 | |
|     return std::tie(
 | |
|         a.sin6_port,
 | |
|         a.sin6_addr,
 | |
|         a.sin6_scope_id
 | |
|     );
 | |
| }
 | |
| 
 | |
| } // namespace
 | |
| 
 | |
| // These binary operators make sockaddr_storage comparable.  They need to be
 | |
| // in the global namespace so that the std::tuple < and == operators can see them.
 | |
| static bool operator <(const in6_addr& x, const in6_addr& y) {
 | |
|     return std::lexicographical_compare(
 | |
|             std::begin(x.s6_addr), std::end(x.s6_addr),
 | |
|             std::begin(y.s6_addr), std::end(y.s6_addr));
 | |
| }
 | |
| 
 | |
| static bool operator ==(const in6_addr& x, const in6_addr& y) {
 | |
|     return std::equal(
 | |
|             std::begin(x.s6_addr), std::end(x.s6_addr),
 | |
|             std::begin(y.s6_addr), std::end(y.s6_addr));
 | |
| }
 | |
| 
 | |
| static bool operator <(const sockaddr_storage& x, const sockaddr_storage& y) {
 | |
|     if (x.ss_family != y.ss_family) {
 | |
|         return x.ss_family < y.ss_family;
 | |
|     }
 | |
|     // Same address family.
 | |
|     if (x.ss_family == AF_INET) {
 | |
|         const sockaddr_in& x_sin = reinterpret_cast<const sockaddr_in&>(x);
 | |
|         const sockaddr_in& y_sin = reinterpret_cast<const sockaddr_in&>(y);
 | |
|         return make_tie(x_sin) < make_tie(y_sin);
 | |
|     } else if (x.ss_family == AF_INET6) {
 | |
|         const sockaddr_in6& x_sin6 = reinterpret_cast<const sockaddr_in6&>(x);
 | |
|         const sockaddr_in6& y_sin6 = reinterpret_cast<const sockaddr_in6&>(y);
 | |
|         return make_tie(x_sin6) < make_tie(y_sin6);
 | |
|     }
 | |
|     return false;  // Unknown address type.  This is an error.
 | |
| }
 | |
| 
 | |
| static bool operator ==(const sockaddr_storage& x, const sockaddr_storage& y) {
 | |
|     if (x.ss_family != y.ss_family) {
 | |
|         return false;
 | |
|     }
 | |
|     // Same address family.
 | |
|     if (x.ss_family == AF_INET) {
 | |
|         const sockaddr_in& x_sin = reinterpret_cast<const sockaddr_in&>(x);
 | |
|         const sockaddr_in& y_sin = reinterpret_cast<const sockaddr_in&>(y);
 | |
|         return make_tie(x_sin) == make_tie(y_sin);
 | |
|     } else if (x.ss_family == AF_INET6) {
 | |
|         const sockaddr_in6& x_sin6 = reinterpret_cast<const sockaddr_in6&>(x);
 | |
|         const sockaddr_in6& y_sin6 = reinterpret_cast<const sockaddr_in6&>(y);
 | |
|         return make_tie(x_sin6) == make_tie(y_sin6);
 | |
|     }
 | |
|     return false;  // Unknown address type.  This is an error.
 | |
| }
 | |
| 
 | |
| namespace android {
 | |
| namespace net {
 | |
| 
 | |
| // This comparison ignores ports and certificates.
 | |
| bool AddressComparator::operator() (const DnsTlsServer& x, const DnsTlsServer& y) const {
 | |
|     if (x.ss.ss_family != y.ss.ss_family) {
 | |
|         return x.ss.ss_family < y.ss.ss_family;
 | |
|     }
 | |
|     // Same address family.
 | |
|     if (x.ss.ss_family == AF_INET) {
 | |
|         const sockaddr_in& x_sin = reinterpret_cast<const sockaddr_in&>(x.ss);
 | |
|         const sockaddr_in& y_sin = reinterpret_cast<const sockaddr_in&>(y.ss);
 | |
|         return x_sin.sin_addr.s_addr < y_sin.sin_addr.s_addr;
 | |
|     } else if (x.ss.ss_family == AF_INET6) {
 | |
|         const sockaddr_in6& x_sin6 = reinterpret_cast<const sockaddr_in6&>(x.ss);
 | |
|         const sockaddr_in6& y_sin6 = reinterpret_cast<const sockaddr_in6&>(y.ss);
 | |
|         return std::tie(x_sin6.sin6_addr, x_sin6.sin6_scope_id) <
 | |
|                 std::tie(y_sin6.sin6_addr, y_sin6.sin6_scope_id);
 | |
|     }
 | |
|     return false;  // Unknown address type.  This is an error.
 | |
| }
 | |
| 
 | |
| // Returns a tuple of references to the elements of s.
 | |
| auto make_tie(const DnsTlsServer& s) {
 | |
|     return std::tie(s.ss, s.name, s.protocol);
 | |
| }
 | |
| 
 | |
| bool DnsTlsServer::operator <(const DnsTlsServer& other) const {
 | |
|     return make_tie(*this) < make_tie(other);
 | |
| }
 | |
| 
 | |
| bool DnsTlsServer::operator ==(const DnsTlsServer& other) const {
 | |
|     return make_tie(*this) == make_tie(other);
 | |
| }
 | |
| 
 | |
| bool DnsTlsServer::wasExplicitlyConfigured() const {
 | |
|     return !name.empty();
 | |
| }
 | |
| 
 | |
| std::string DnsTlsServer::toIpString() const {
 | |
|     return netdutils::IPSockAddr::toIPSockAddr(ss).ip().toString();
 | |
| }
 | |
| 
 | |
| }  // namespace net
 | |
| }  // namespace android
 |