1 /* 2 * Copyright (C) 2010 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #define LOG_TAG "NetworkUtilities" 18 19 #include "NetworkUtilities.h" 20 #include "JNIHelp.h" 21 #include "JniConstants.h" 22 #include "ScopedLocalRef.h" 23 24 #include <arpa/inet.h> 25 #include <fcntl.h> 26 #include <stdio.h> 27 #include <string.h> 28 #include <sys/socket.h> 29 #include <sys/un.h> 30 31 jobject sockaddrToInetAddress(JNIEnv* env, const sockaddr_storage& ss, jint* port) { 32 // Convert IPv4-mapped IPv6 addresses to IPv4 addresses. 33 // The RI states "Java will never return an IPv4-mapped address". 34 const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss); 35 if (ss.ss_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sin6.sin6_addr)) { 36 // Copy the IPv6 address into the temporary sockaddr_storage. 37 sockaddr_storage tmp; 38 memset(&tmp, 0, sizeof(tmp)); 39 memcpy(&tmp, &ss, sizeof(sockaddr_in6)); 40 // Unmap it into an IPv4 address. 41 sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(tmp); 42 sin.sin_family = AF_INET; 43 sin.sin_port = sin6.sin6_port; 44 memcpy(&sin.sin_addr.s_addr, &sin6.sin6_addr.s6_addr[12], 4); 45 // Do the regular conversion using the unmapped address. 46 return sockaddrToInetAddress(env, tmp, port); 47 } 48 49 const void* rawAddress; 50 size_t addressLength; 51 int sin_port = 0; 52 int scope_id = 0; 53 if (ss.ss_family == AF_INET) { 54 const sockaddr_in& sin = reinterpret_cast<const sockaddr_in&>(ss); 55 rawAddress = &sin.sin_addr.s_addr; 56 addressLength = 4; 57 sin_port = ntohs(sin.sin_port); 58 } else if (ss.ss_family == AF_INET6) { 59 const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss); 60 rawAddress = &sin6.sin6_addr.s6_addr; 61 addressLength = 16; 62 sin_port = ntohs(sin6.sin6_port); 63 scope_id = sin6.sin6_scope_id; 64 } else if (ss.ss_family == AF_UNIX) { 65 const sockaddr_un& sun = reinterpret_cast<const sockaddr_un&>(ss); 66 rawAddress = &sun.sun_path; 67 addressLength = strlen(sun.sun_path); 68 } else { 69 // We can't throw SocketException. We aren't meant to see bad addresses, so seeing one 70 // really does imply an internal error. 71 jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException", 72 "sockaddrToInetAddress unsupported ss_family: %i", ss.ss_family); 73 return NULL; 74 } 75 if (port != NULL) { 76 *port = sin_port; 77 } 78 79 ScopedLocalRef<jbyteArray> byteArray(env, env->NewByteArray(addressLength)); 80 if (byteArray.get() == NULL) { 81 return NULL; 82 } 83 env->SetByteArrayRegion(byteArray.get(), 0, addressLength, 84 reinterpret_cast<const jbyte*>(rawAddress)); 85 86 if (ss.ss_family == AF_UNIX) { 87 // Note that we get here for AF_UNIX sockets on accept(2). The unix(7) man page claims 88 // that the peer's sun_path will contain the path, but in practice it doesn't, and the 89 // peer length is returned as 2 (meaning only the sun_family field was set). 90 static jmethodID ctor = env->GetMethodID(JniConstants::inetUnixAddressClass, "<init>", "([B)V"); 91 return env->NewObject(JniConstants::inetUnixAddressClass, ctor, byteArray.get()); 92 } 93 94 static jmethodID getByAddressMethod = env->GetStaticMethodID(JniConstants::inetAddressClass, 95 "getByAddress", "(Ljava/lang/String;[BI)Ljava/net/InetAddress;"); 96 if (getByAddressMethod == NULL) { 97 return NULL; 98 } 99 return env->CallStaticObjectMethod(JniConstants::inetAddressClass, getByAddressMethod, 100 NULL, byteArray.get(), scope_id); 101 } 102 103 static bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len, bool map) { 104 memset(&ss, 0, sizeof(ss)); 105 sa_len = 0; 106 107 if (inetAddress == NULL) { 108 jniThrowNullPointerException(env, NULL); 109 return false; 110 } 111 112 // Get the address family. 113 static jfieldID familyFid = env->GetFieldID(JniConstants::inetAddressClass, "family", "I"); 114 ss.ss_family = env->GetIntField(inetAddress, familyFid); 115 if (ss.ss_family == AF_UNSPEC) { 116 sa_len = sizeof(ss.ss_family); 117 return true; // Job done! 118 } 119 120 // Check this is an address family we support. 121 if (ss.ss_family != AF_INET && ss.ss_family != AF_INET6 && ss.ss_family != AF_UNIX) { 122 jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException", 123 "inetAddressToSockaddr bad family: %i", ss.ss_family); 124 return false; 125 } 126 127 // Get the byte array that stores the IP address bytes in the InetAddress. 128 static jfieldID bytesFid = env->GetFieldID(JniConstants::inetAddressClass, "ipaddress", "[B"); 129 ScopedLocalRef<jbyteArray> addressBytes(env, reinterpret_cast<jbyteArray>(env->GetObjectField(inetAddress, bytesFid))); 130 if (addressBytes.get() == NULL) { 131 jniThrowNullPointerException(env, NULL); 132 return false; 133 } 134 135 // Handle the AF_UNIX special case. 136 if (ss.ss_family == AF_UNIX) { 137 sockaddr_un& sun = reinterpret_cast<sockaddr_un&>(ss); 138 139 size_t path_length = env->GetArrayLength(addressBytes.get()); 140 if (path_length >= sizeof(sun.sun_path)) { 141 jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException", 142 "inetAddressToSockaddr path too long for AF_UNIX: %i", path_length); 143 return false; 144 } 145 146 // Copy the bytes... 147 jbyte* dst = reinterpret_cast<jbyte*>(&sun.sun_path); 148 memset(dst, 0, sizeof(sun.sun_path)); 149 env->GetByteArrayRegion(addressBytes.get(), 0, path_length, dst); 150 sa_len = sizeof(sun.sun_path); 151 return true; 152 } 153 154 // TODO: bionic's getnameinfo(3) seems to want its length parameter to be exactly 155 // sizeof(sockaddr_in) for an IPv4 address and sizeof (sockaddr_in6) for an 156 // IPv6 address. Fix getnameinfo so it accepts sizeof(sockaddr_storage), and 157 // then unconditionally set sa_len to sizeof(sockaddr_storage) instead of having 158 // to deal with this case by case. 159 160 // We use AF_INET6 sockets, so we want an IPv6 address (which may be a IPv4-mapped address). 161 sockaddr_in6& sin6 = reinterpret_cast<sockaddr_in6&>(ss); 162 sin6.sin6_port = htons(port); 163 if (ss.ss_family == AF_INET6) { 164 // IPv6 address. Copy the bytes... 165 jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr); 166 env->GetByteArrayRegion(addressBytes.get(), 0, 16, dst); 167 // ...and set the scope id... 168 static jfieldID scopeFid = env->GetFieldID(JniConstants::inet6AddressClass, "scope_id", "I"); 169 sin6.sin6_scope_id = env->GetIntField(inetAddress, scopeFid); 170 sa_len = sizeof(sockaddr_in6); 171 return true; 172 } 173 174 // Deal with Inet4Address instances. 175 if (map) { 176 // We should represent this Inet4Address as an IPv4-mapped IPv6 sockaddr_in6. 177 // Change the family... 178 sin6.sin6_family = AF_INET6; 179 // Copy the bytes... 180 jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr[12]); 181 env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst); 182 // INADDR_ANY and in6addr_any are both all-zeros... 183 if (!IN6_IS_ADDR_UNSPECIFIED(&sin6.sin6_addr)) { 184 // ...but all other IPv4-mapped addresses are ::ffff:a.b.c.d, so insert the ffff... 185 memset(&(sin6.sin6_addr.s6_addr[10]), 0xff, 2); 186 } 187 sa_len = sizeof(sockaddr_in6); 188 } else { 189 // We should represent this Inet4Address as an IPv4 sockaddr_in. 190 sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(ss); 191 sin.sin_port = htons(port); 192 jbyte* dst = reinterpret_cast<jbyte*>(&sin.sin_addr.s_addr); 193 env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst); 194 sa_len = sizeof(sockaddr_in); 195 } 196 return true; 197 } 198 199 bool inetAddressToSockaddrVerbatim(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) { 200 return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, false); 201 } 202 203 bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) { 204 return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, true); 205 } 206 207 bool setBlocking(int fd, bool blocking) { 208 int flags = fcntl(fd, F_GETFL); 209 if (flags == -1) { 210 return false; 211 } 212 213 if (!blocking) { 214 flags |= O_NONBLOCK; 215 } else { 216 flags &= ~O_NONBLOCK; 217 } 218 219 int rc = fcntl(fd, F_SETFL, flags); 220 return (rc != -1); 221 } 222