| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 | |
| 5 #include "ppapi/shared_impl/private/flash_net_address_impl.h" | |
| 6 | |
| 7 #include <string.h> | |
| 8 | |
| 9 #include <string> | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "build/build_config.h" | |
| 13 #include "net/base/net_util.h" | |
| 14 #include "net/base/sys_addrinfo.h" | |
| 15 #include "net/base/sys_byteorder.h" | |
| 16 #include "ppapi/c/pp_var.h" | |
| 17 #include "ppapi/c/private/ppb_flash_net_address.h" | |
| 18 #include "ppapi/shared_impl/var.h" | |
| 19 #include "ppapi/thunk/thunk.h" | |
| 20 | |
| 21 // The Flash net address interface doesn't have a normal C -> C++ thunk since it | |
| 22 // doesn't actually have any proxy wrapping or associated objects; it's just a | |
| 23 // call into base. So we implement the entire interface here, using the thunk | |
| 24 // namespace so it magically gets hooked up in the proper places. | |
| 25 | |
| 26 namespace ppapi { | |
| 27 | |
| 28 namespace { | |
| 29 | |
| 30 #if defined(OS_WIN) | |
| 31 // The type of |sockaddr::sa_family|. | |
| 32 typedef ADDRESS_FAMILY sa_family_t; | |
| 33 #endif | |
| 34 | |
| 35 inline sa_family_t GetFamily(const PP_Flash_NetAddress& addr) { | |
| 36 return reinterpret_cast<const sockaddr*>(addr.data)->sa_family; | |
| 37 } | |
| 38 | |
| 39 PP_Bool AreHostsEqual(const PP_Flash_NetAddress* addr1, | |
| 40 const PP_Flash_NetAddress* addr2) { | |
| 41 if (!FlashNetAddressImpl::ValidateNetAddress(*addr1) || | |
| 42 !FlashNetAddressImpl::ValidateNetAddress(*addr2)) | |
| 43 return PP_FALSE; | |
| 44 | |
| 45 if (GetFamily(*addr1) != GetFamily(*addr2)) | |
| 46 return PP_FALSE; | |
| 47 | |
| 48 if (GetFamily(*addr1) == AF_INET) { | |
| 49 const sockaddr_in* a1 = reinterpret_cast<const sockaddr_in*>(addr1->data); | |
| 50 const sockaddr_in* a2 = reinterpret_cast<const sockaddr_in*>(addr2->data); | |
| 51 return PP_FromBool(a1->sin_addr.s_addr == a2->sin_addr.s_addr); | |
| 52 } | |
| 53 | |
| 54 if (GetFamily(*addr1) == AF_INET6) { | |
| 55 const sockaddr_in6* a1 = reinterpret_cast<const sockaddr_in6*>(addr1->data); | |
| 56 const sockaddr_in6* a2 = reinterpret_cast<const sockaddr_in6*>(addr2->data); | |
| 57 return PP_FromBool(a1->sin6_flowinfo == a2->sin6_flowinfo && | |
| 58 memcmp(&a1->sin6_addr, &a2->sin6_addr, | |
| 59 sizeof(a1->sin6_addr)) == 0 && | |
| 60 a1->sin6_scope_id == a2->sin6_scope_id); | |
| 61 } | |
| 62 | |
| 63 return PP_FALSE; | |
| 64 } | |
| 65 | |
| 66 PP_Bool AreEqual(const PP_Flash_NetAddress* addr1, | |
| 67 const PP_Flash_NetAddress* addr2) { | |
| 68 // |AreHostsEqual()| will also validate the addresses and return false if | |
| 69 // either is invalid. | |
| 70 if (!AreHostsEqual(addr1, addr2)) | |
| 71 return PP_FALSE; | |
| 72 | |
| 73 // Note: Here, we know that |addr1| and |addr2| have the same family. | |
| 74 if (GetFamily(*addr1) == AF_INET) { | |
| 75 const sockaddr_in* a1 = reinterpret_cast<const sockaddr_in*>(addr1->data); | |
| 76 const sockaddr_in* a2 = reinterpret_cast<const sockaddr_in*>(addr2->data); | |
| 77 return PP_FromBool(a1->sin_port == a2->sin_port); | |
| 78 } | |
| 79 | |
| 80 if (GetFamily(*addr1) == AF_INET6) { | |
| 81 const sockaddr_in6* a1 = reinterpret_cast<const sockaddr_in6*>(addr1->data); | |
| 82 const sockaddr_in6* a2 = reinterpret_cast<const sockaddr_in6*>(addr2->data); | |
| 83 return PP_FromBool(a1->sin6_port == a2->sin6_port); | |
| 84 } | |
| 85 | |
| 86 return PP_FALSE; | |
| 87 } | |
| 88 | |
| 89 PP_Var Describe(PP_Module module, | |
| 90 const struct PP_Flash_NetAddress* addr, | |
| 91 PP_Bool include_port) { | |
| 92 if (!FlashNetAddressImpl::ValidateNetAddress(*addr)) | |
| 93 return PP_MakeUndefined(); | |
| 94 | |
| 95 const sockaddr* a = reinterpret_cast<const sockaddr*>(addr->data); | |
| 96 socklen_t l = addr->size; | |
| 97 std::string description = | |
| 98 include_port ? net::NetAddressToStringWithPort(a, l) : | |
| 99 net::NetAddressToString(a, l); | |
| 100 return StringVar::StringToPPVar(module, description); | |
| 101 } | |
| 102 | |
| 103 PP_Bool ReplacePort(const struct PP_Flash_NetAddress* src_addr, | |
| 104 uint16_t port, | |
| 105 struct PP_Flash_NetAddress* dest_addr) { | |
| 106 if (!FlashNetAddressImpl::ValidateNetAddress(*src_addr)) | |
| 107 return PP_FALSE; | |
| 108 | |
| 109 if (GetFamily(*src_addr) == AF_INET) { | |
| 110 memmove(dest_addr, src_addr, sizeof(*src_addr)); | |
| 111 reinterpret_cast<sockaddr_in*>(dest_addr->data)->sin_port = htons(port); | |
| 112 return PP_TRUE; | |
| 113 } | |
| 114 | |
| 115 if (GetFamily(*src_addr) == AF_INET6) { | |
| 116 memmove(dest_addr, src_addr, sizeof(*src_addr)); | |
| 117 reinterpret_cast<sockaddr_in6*>(dest_addr->data)->sin6_port = htons(port); | |
| 118 return PP_TRUE; | |
| 119 } | |
| 120 | |
| 121 return PP_FALSE; | |
| 122 } | |
| 123 | |
| 124 void GetAnyAddress(PP_Bool is_ipv6, struct PP_Flash_NetAddress* addr) { | |
| 125 memset(addr->data, 0, arraysize(addr->data) * sizeof(addr->data[0])); | |
| 126 if (is_ipv6) { | |
| 127 sockaddr_in6* a = reinterpret_cast<sockaddr_in6*>(addr->data); | |
| 128 addr->size = sizeof(*a); | |
| 129 a->sin6_family = AF_INET6; | |
| 130 a->sin6_addr = in6addr_any; | |
| 131 } else { | |
| 132 sockaddr_in* a = reinterpret_cast<sockaddr_in*>(addr->data); | |
| 133 addr->size = sizeof(*a); | |
| 134 a->sin_family = AF_INET; | |
| 135 a->sin_addr.s_addr = INADDR_ANY; | |
| 136 } | |
| 137 } | |
| 138 | |
| 139 const PPB_Flash_NetAddress flash_net_address_interface = { | |
| 140 &AreEqual, | |
| 141 &AreHostsEqual, | |
| 142 &Describe, | |
| 143 &ReplacePort, | |
| 144 &GetAnyAddress | |
| 145 }; | |
| 146 | |
| 147 } // namespace | |
| 148 | |
| 149 namespace thunk { | |
| 150 | |
| 151 PPAPI_THUNK_EXPORT const PPB_Flash_NetAddress* GetPPB_Flash_NetAddress_Thunk() { | |
| 152 return &flash_net_address_interface; | |
| 153 } | |
| 154 | |
| 155 } // namespace thunk | |
| 156 | |
| 157 // static | |
| 158 bool FlashNetAddressImpl::ValidateNetAddress(const PP_Flash_NetAddress& addr) { | |
| 159 if (addr.size < sizeof(reinterpret_cast<sockaddr*>(0)->sa_family)) | |
| 160 return false; | |
| 161 | |
| 162 // TODO(viettrungluu): more careful validation? | |
| 163 // Just do a size check for AF_INET. | |
| 164 if (GetFamily(addr) == AF_INET && addr.size >= sizeof(sockaddr_in)) | |
| 165 return true; | |
| 166 | |
| 167 // Ditto for AF_INET6. | |
| 168 if (GetFamily(addr) == AF_INET6 && addr.size >= sizeof(sockaddr_in6)) | |
| 169 return true; | |
| 170 | |
| 171 // Reject everything else. | |
| 172 return false; | |
| 173 } | |
| 174 | |
| 175 } // namespace ppapi | |
| OLD | NEW |