luni/src/main/native/NetworkUtilities.cpp - platform/libcore - Git at Google

 /*
  * Copyright (C) 2010 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.
  */

 #define LOG_TAG "NetworkUtilities"

 #include "NetworkUtilities.h"

 #include <arpa/inet.h>
 #include <fcntl.h>
 #include <stdio.h>
 #include <string.h>
 #include <sys/socket.h>
 #include <sys/un.h>

 #include <nativehelper/JNIHelp.h>
 #include <nativehelper/ScopedLocalRef.h>

 #include "JniConstants.h"

 jobject sockaddrToInetAddress(JNIEnv* env, const sockaddr_storage& ss, jint* port) {
     // Convert IPv4-mapped IPv6 addresses to IPv4 addresses.
     // The RI states "Java will never return an IPv4-mapped address".
     const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss);
     if (ss.ss_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sin6.sin6_addr)) {
         // Copy the IPv6 address into the temporary sockaddr_storage.
         sockaddr_storage tmp;
         memset(&tmp, 0, sizeof(tmp));
         memcpy(&tmp, &ss, sizeof(sockaddr_in6));
         // Unmap it into an IPv4 address.
         sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(tmp);
         sin.sin_family = AF_INET;
         sin.sin_port = sin6.sin6_port;
         memcpy(&sin.sin_addr.s_addr, &sin6.sin6_addr.s6_addr[12], 4);
         // Do the regular conversion using the unmapped address.
         return sockaddrToInetAddress(env, tmp, port);
     }

     const void* rawAddress;
     size_t addressLength;
     int sin_port = 0;
     int scope_id = 0;
     if (ss.ss_family == AF_INET) {
         const sockaddr_in& sin = reinterpret_cast<const sockaddr_in&>(ss);
         rawAddress = &sin.sin_addr.s_addr;
         addressLength = 4;
         sin_port = ntohs(sin.sin_port);
     } else if (ss.ss_family == AF_INET6) {
         const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss);
         rawAddress = &sin6.sin6_addr.s6_addr;
         addressLength = 16;
         sin_port = ntohs(sin6.sin6_port);
         scope_id = sin6.sin6_scope_id;
     } else {
         // We can't throw SocketException. We aren't meant to see bad addresses, so seeing one
         // really does imply an internal error.
         jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException",
                              "sockaddrToInetAddress unsupported ss_family: %i", ss.ss_family);
         return NULL;
     }
     if (port != NULL) {
         *port = sin_port;
     }

     ScopedLocalRef<jbyteArray> byteArray(env, env->NewByteArray(addressLength));
     if (byteArray.get() == NULL) {
         return NULL;
     }
     env->SetByteArrayRegion(byteArray.get(), 0, addressLength,
             reinterpret_cast<const jbyte*>(rawAddress));

     static jmethodID getByAddressMethod = env->GetStaticMethodID(JniConstants::GetInetAddressClass(env),
             "getByAddress", "(Ljava/lang/String;[BI)Ljava/net/InetAddress;");
     if (getByAddressMethod == NULL) {
         return NULL;
     }
     return env->CallStaticObjectMethod(JniConstants::GetInetAddressClass(env), getByAddressMethod,
             NULL, byteArray.get(), scope_id);
 }

 static bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len, bool map) {
     memset(&ss, 0, sizeof(ss));
     sa_len = 0;

     if (inetAddress == NULL) {
         jniThrowNullPointerException(env, NULL);
         return false;
     }

     // Get holder.
     static jfieldID holderFid = env->GetFieldID(JniConstants::GetInetAddressClass(env), "holder", "Ljava/net/InetAddress$InetAddressHolder;");
     if (holderFid == NULL) {
         return false;
     }
     ScopedLocalRef<jobject> holder(env, env->GetObjectField(inetAddress, holderFid));
     // Get the address family.
     static jfieldID familyFid = env->GetFieldID(JniConstants::GetInetAddressHolderClass(env), "family", "I");
     if (familyFid == NULL) {
         return false;
     }
     ss.ss_family = env->GetIntField(holder.get(), familyFid);
     if (ss.ss_family == AF_UNSPEC) {
         sa_len = sizeof(ss.ss_family);
         return true; // Job done!
     }

     // Check this is an address family we support.
     if (ss.ss_family != AF_INET && ss.ss_family != AF_INET6) {
         jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException",
                 "inetAddressToSockaddr bad family: %i", ss.ss_family);
         return false;
     }

     // Get the byte array that stores the IP address bytes in the InetAddress.
     static jmethodID bytesMid = env->GetMethodID(JniConstants::GetInetAddressClass(env), "getAddress", "()[B");
     if (bytesMid == NULL) {
         return false;
     }
     ScopedLocalRef<jbyteArray> addressBytes(env, reinterpret_cast<jbyteArray>(env->CallObjectMethod(inetAddress, bytesMid)));
     if (env->ExceptionCheck()) {
         return false;
     }
     if (addressBytes.get() == NULL) {
         jniThrowNullPointerException(env, NULL);
         return false;
     }

     // TODO: bionic's getnameinfo(3) seems to want its length parameter to be exactly
     // sizeof(sockaddr_in) for an IPv4 address and sizeof (sockaddr_in6) for an
     // IPv6 address. Fix getnameinfo so it accepts sizeof(sockaddr_storage), and
     // then unconditionally set sa_len to sizeof(sockaddr_storage) instead of having
     // to deal with this case by case.

     // We use AF_INET6 sockets, so we want an IPv6 address (which may be a IPv4-mapped address).
     sockaddr_in6& sin6 = reinterpret_cast<sockaddr_in6&>(ss);
     sin6.sin6_port = htons(port);
     if (ss.ss_family == AF_INET6) {
         // IPv6 address. Copy the bytes...
         jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr);
         env->GetByteArrayRegion(addressBytes.get(), 0, 16, dst);
         // ...and set the scope id...
         static jfieldID holder6Fid = env->GetFieldID(JniConstants::GetInet6AddressClass(env),
                                                      "holder6",
                                                      "Ljava/net/Inet6Address$Inet6AddressHolder;");
         if (holder6Fid == NULL) {
             return false;
         }
         ScopedLocalRef<jobject> holder6(env, env->GetObjectField(inetAddress, holder6Fid));
         static jfieldID scopeFid = env->GetFieldID(JniConstants::GetInet6AddressHolderClass(env),
                                                    "scope_id",
                                                    "I");
         sin6.sin6_scope_id = env->GetIntField(holder6.get(), scopeFid);
         sa_len = sizeof(sockaddr_in6);
         return true;
     }

     // Deal with Inet4Address instances.
     if (map) {
         // We should represent this Inet4Address as an IPv4-mapped IPv6 sockaddr_in6.
         // Change the family...
         sin6.sin6_family = AF_INET6;
         // Copy the bytes...
         jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr[12]);
         env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst);
         // INADDR_ANY and in6addr_any are both all-zeros...
         if (!IN6_IS_ADDR_UNSPECIFIED(&sin6.sin6_addr)) {
             // ...but all other IPv4-mapped addresses are ::ffff:a.b.c.d, so insert the ffff...
             memset(&(sin6.sin6_addr.s6_addr[10]), 0xff, 2);
         }
         sa_len = sizeof(sockaddr_in6);
     } else {
         // We should represent this Inet4Address as an IPv4 sockaddr_in.
         sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(ss);
         sin.sin_port = htons(port);
         jbyte* dst = reinterpret_cast<jbyte*>(&sin.sin_addr.s_addr);
         env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst);
         sa_len = sizeof(sockaddr_in);
     }
     return true;
 }

 bool inetAddressToSockaddrVerbatim(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) {
     return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, false);
 }

 bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) {
     return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, true);
 }

 bool setBlocking(int fd, bool blocking) {
     int flags = fcntl(fd, F_GETFL);
     if (flags == -1) {
         return false;
     }

     if (!blocking) {
         flags |= O_NONBLOCK;
     } else {
         flags &= ~O_NONBLOCK;
     }

     int rc = fcntl(fd, F_SETFL, flags);
     return (rc != -1);
 }
	/*
	* Copyright (C) 2010 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.
	*/

	#define LOG_TAG "NetworkUtilities"

	#include "NetworkUtilities.h"

	#include <arpa/inet.h>
	#include <fcntl.h>
	#include <stdio.h>
	#include <string.h>
	#include <sys/socket.h>
	#include <sys/un.h>

	#include <nativehelper/JNIHelp.h>
	#include <nativehelper/ScopedLocalRef.h>

	#include "JniConstants.h"

	jobject sockaddrToInetAddress(JNIEnv* env, const sockaddr_storage& ss, jint* port) {
	// Convert IPv4-mapped IPv6 addresses to IPv4 addresses.
	// The RI states "Java will never return an IPv4-mapped address".
	const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss);
	if (ss.ss_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sin6.sin6_addr)) {
	// Copy the IPv6 address into the temporary sockaddr_storage.
	sockaddr_storage tmp;
	memset(&tmp, 0, sizeof(tmp));
	memcpy(&tmp, &ss, sizeof(sockaddr_in6));
	// Unmap it into an IPv4 address.
	sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(tmp);
	sin.sin_family = AF_INET;
	sin.sin_port = sin6.sin6_port;
	memcpy(&sin.sin_addr.s_addr, &sin6.sin6_addr.s6_addr[12], 4);
	// Do the regular conversion using the unmapped address.
	return sockaddrToInetAddress(env, tmp, port);
	}

	const void* rawAddress;
	size_t addressLength;
	int sin_port = 0;
	int scope_id = 0;
	if (ss.ss_family == AF_INET) {
	const sockaddr_in& sin = reinterpret_cast<const sockaddr_in&>(ss);
	rawAddress = &sin.sin_addr.s_addr;
	addressLength = 4;
	sin_port = ntohs(sin.sin_port);
	} else if (ss.ss_family == AF_INET6) {
	const sockaddr_in6& sin6 = reinterpret_cast<const sockaddr_in6&>(ss);
	rawAddress = &sin6.sin6_addr.s6_addr;
	addressLength = 16;
	sin_port = ntohs(sin6.sin6_port);
	scope_id = sin6.sin6_scope_id;
	} else {
	// We can't throw SocketException. We aren't meant to see bad addresses, so seeing one
	// really does imply an internal error.
	jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException",
	"sockaddrToInetAddress unsupported ss_family: %i", ss.ss_family);
	return NULL;
	}
	if (port != NULL) {
	*port = sin_port;
	}

	ScopedLocalRef<jbyteArray> byteArray(env, env->NewByteArray(addressLength));
	if (byteArray.get() == NULL) {
	return NULL;
	}
	env->SetByteArrayRegion(byteArray.get(), 0, addressLength,
	reinterpret_cast<const jbyte*>(rawAddress));

	static jmethodID getByAddressMethod = env->GetStaticMethodID(JniConstants::GetInetAddressClass(env),
	"getByAddress", "(Ljava/lang/String;[BI)Ljava/net/InetAddress;");
	if (getByAddressMethod == NULL) {
	return NULL;
	}
	return env->CallStaticObjectMethod(JniConstants::GetInetAddressClass(env), getByAddressMethod,
	NULL, byteArray.get(), scope_id);
	}

	static bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len, bool map) {
	memset(&ss, 0, sizeof(ss));
	sa_len = 0;

	if (inetAddress == NULL) {
	jniThrowNullPointerException(env, NULL);
	return false;
	}

	// Get holder.
	static jfieldID holderFid = env->GetFieldID(JniConstants::GetInetAddressClass(env), "holder", "Ljava/net/InetAddress$InetAddressHolder;");
	if (holderFid == NULL) {
	return false;
	}
	ScopedLocalRef<jobject> holder(env, env->GetObjectField(inetAddress, holderFid));
	// Get the address family.
	static jfieldID familyFid = env->GetFieldID(JniConstants::GetInetAddressHolderClass(env), "family", "I");
	if (familyFid == NULL) {
	return false;
	}
	ss.ss_family = env->GetIntField(holder.get(), familyFid);
	if (ss.ss_family == AF_UNSPEC) {
	sa_len = sizeof(ss.ss_family);
	return true; // Job done!
	}

	// Check this is an address family we support.
	if (ss.ss_family != AF_INET && ss.ss_family != AF_INET6) {
	jniThrowExceptionFmt(env, "java/lang/IllegalArgumentException",
	"inetAddressToSockaddr bad family: %i", ss.ss_family);
	return false;
	}

	// Get the byte array that stores the IP address bytes in the InetAddress.
	static jmethodID bytesMid = env->GetMethodID(JniConstants::GetInetAddressClass(env), "getAddress", "()[B");
	if (bytesMid == NULL) {
	return false;
	}
	ScopedLocalRef<jbyteArray> addressBytes(env, reinterpret_cast<jbyteArray>(env->CallObjectMethod(inetAddress, bytesMid)));
	if (env->ExceptionCheck()) {
	return false;
	}
	if (addressBytes.get() == NULL) {
	jniThrowNullPointerException(env, NULL);
	return false;
	}

	// TODO: bionic's getnameinfo(3) seems to want its length parameter to be exactly
	// sizeof(sockaddr_in) for an IPv4 address and sizeof (sockaddr_in6) for an
	// IPv6 address. Fix getnameinfo so it accepts sizeof(sockaddr_storage), and
	// then unconditionally set sa_len to sizeof(sockaddr_storage) instead of having
	// to deal with this case by case.

	// We use AF_INET6 sockets, so we want an IPv6 address (which may be a IPv4-mapped address).
	sockaddr_in6& sin6 = reinterpret_cast<sockaddr_in6&>(ss);
	sin6.sin6_port = htons(port);
	if (ss.ss_family == AF_INET6) {
	// IPv6 address. Copy the bytes...
	jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr);
	env->GetByteArrayRegion(addressBytes.get(), 0, 16, dst);
	// ...and set the scope id...
	static jfieldID holder6Fid = env->GetFieldID(JniConstants::GetInet6AddressClass(env),
	"holder6",
	"Ljava/net/Inet6Address$Inet6AddressHolder;");
	if (holder6Fid == NULL) {
	return false;
	}
	ScopedLocalRef<jobject> holder6(env, env->GetObjectField(inetAddress, holder6Fid));
	static jfieldID scopeFid = env->GetFieldID(JniConstants::GetInet6AddressHolderClass(env),
	"scope_id",
	"I");
	sin6.sin6_scope_id = env->GetIntField(holder6.get(), scopeFid);
	sa_len = sizeof(sockaddr_in6);
	return true;
	}

	// Deal with Inet4Address instances.
	if (map) {
	// We should represent this Inet4Address as an IPv4-mapped IPv6 sockaddr_in6.
	// Change the family...
	sin6.sin6_family = AF_INET6;
	// Copy the bytes...
	jbyte* dst = reinterpret_cast<jbyte*>(&sin6.sin6_addr.s6_addr[12]);
	env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst);
	// INADDR_ANY and in6addr_any are both all-zeros...
	if (!IN6_IS_ADDR_UNSPECIFIED(&sin6.sin6_addr)) {
	// ...but all other IPv4-mapped addresses are ::ffff:a.b.c.d, so insert the ffff...
	memset(&(sin6.sin6_addr.s6_addr[10]), 0xff, 2);
	}
	sa_len = sizeof(sockaddr_in6);
	} else {
	// We should represent this Inet4Address as an IPv4 sockaddr_in.
	sockaddr_in& sin = reinterpret_cast<sockaddr_in&>(ss);
	sin.sin_port = htons(port);
	jbyte* dst = reinterpret_cast<jbyte*>(&sin.sin_addr.s_addr);
	env->GetByteArrayRegion(addressBytes.get(), 0, 4, dst);
	sa_len = sizeof(sockaddr_in);
	}
	return true;
	}

	bool inetAddressToSockaddrVerbatim(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) {
	return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, false);
	}

	bool inetAddressToSockaddr(JNIEnv* env, jobject inetAddress, int port, sockaddr_storage& ss, socklen_t& sa_len) {
	return inetAddressToSockaddr(env, inetAddress, port, ss, sa_len, true);
	}

	bool setBlocking(int fd, bool blocking) {
	int flags = fcntl(fd, F_GETFL);
	if (flags == -1) {
	return false;
	}

	if (!blocking) {
	flags \|= O_NONBLOCK;
	} else {
	flags &= ~O_NONBLOCK;
	}

	int rc = fcntl(fd, F_SETFL, flags);
	return (rc != -1);
	}