--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/windows/native/sun/nio/ch/DatagramChannelImpl.c Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,303 @@
+/*
+ * Copyright 2001-2007 Sun Microsystems, Inc. All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation. Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun in the LICENSE file that accompanied this code.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+#include "jni.h"
+#include "jni_util.h"
+#include "jvm.h"
+#include "jlong.h"
+#include <io.h>
+#include "sun_nio_ch_DatagramChannelImpl.h"
+#include "nio.h"
+#include "nio_util.h"
+#include "net_util.h"
+#include <winsock2.h>
+
+static jfieldID isa_addrID; /* address in java.net.InetSocketAddress */
+static jfieldID isa_portID; /* port in java.net.InetSocketAddress */
+static jfieldID dci_senderID; /* sender in sun.nio.ch.DatagramChannelImpl */
+static jfieldID dci_senderAddrID; /* sender InetAddress in sun.nio.ch.DatagramChannelImpl */
+static jfieldID dci_senderPortID; /* sender port in sun.nio.ch.DatagramChannelImpl */
+static jfieldID ia_addrID;
+static jfieldID ia_famID;
+static jclass isa_class; /* java.net.InetSocketAddress */
+static jclass ia_class;
+static jmethodID isa_ctorID; /* .InetSocketAddress(InetAddress, int) */
+static jmethodID ia_ctorID;
+
+/*
+ * Returns JNI_TRUE if DatagramChannelImpl has already cached an
+ * InetAddress/port corresponding to the socket address.
+ */
+static jboolean isSenderCached(JNIEnv *env, jobject this, struct sockaddr_in *sa) {
+ jobject senderAddr;
+
+ /* shouldn't happen until we have dual IPv4/IPv6 stack (post-XP ?) */
+ if (sa->sin_family != AF_INET) {
+ return JNI_FALSE;
+ }
+
+ /*
+ * Compare source address to cached InetAddress
+ */
+ senderAddr = (*env)->GetObjectField(env, this, dci_senderAddrID);
+ if (senderAddr == NULL) {
+ return JNI_FALSE;
+ }
+ if ((jint)ntohl(sa->sin_addr.s_addr) !=
+ (*env)->GetIntField(env, senderAddr, ia_addrID)) {
+ return JNI_FALSE;
+ }
+
+ /*
+ * Compare source port to cached port
+ */
+ if ((jint)ntohs(sa->sin_port) !=
+ (*env)->GetIntField(env, this, dci_senderPortID)) {
+ return JNI_FALSE;
+ }
+ return JNI_TRUE;
+}
+
+JNIEXPORT void JNICALL
+Java_sun_nio_ch_DatagramChannelImpl_initIDs(JNIEnv *env, jclass clazz)
+{
+ clazz = (*env)->FindClass(env, "java/net/InetSocketAddress");
+ isa_class = (*env)->NewGlobalRef(env, clazz);
+ isa_ctorID = (*env)->GetMethodID(env, clazz, "<init>",
+ "(Ljava/net/InetAddress;I)V");
+ isa_addrID = (*env)->GetFieldID(env, clazz, "addr",
+ "Ljava/net/InetAddress;");
+ isa_portID = (*env)->GetFieldID(env, clazz, "port", "I");
+
+ clazz = (*env)->FindClass(env, "sun/nio/ch/DatagramChannelImpl");
+ dci_senderID = (*env)->GetFieldID(env, clazz, "sender",
+ "Ljava/net/SocketAddress;");
+ dci_senderAddrID = (*env)->GetFieldID(env, clazz,
+ "cachedSenderInetAddress",
+ "Ljava/net/InetAddress;");
+ dci_senderPortID = (*env)->GetFieldID(env, clazz,
+ "cachedSenderPort", "I");
+ clazz = (*env)->FindClass(env, "java/net/Inet4Address");
+ ia_class = (*env)->NewGlobalRef(env, clazz);
+ ia_addrID = (*env)->GetFieldID(env, clazz, "address", "I");
+ ia_famID = (*env)->GetFieldID(env, clazz, "family", "I");
+ ia_ctorID = (*env)->GetMethodID(env, clazz, "<init>", "()V");
+}
+
+/*
+ * Return JNI_TRUE if this Windows edition supports ICMP Port Unreachable
+ */
+__inline static jboolean supportPortUnreachable() {
+ static jboolean initDone;
+ static jboolean portUnreachableSupported;
+
+ if (!initDone) {
+ OSVERSIONINFO ver;
+ ver.dwOSVersionInfoSize = sizeof(ver);
+ GetVersionEx(&ver);
+ if (ver.dwPlatformId == VER_PLATFORM_WIN32_NT && ver.dwMajorVersion >= 5) {
+ portUnreachableSupported = JNI_TRUE;
+ } else {
+ portUnreachableSupported = JNI_FALSE;
+ }
+ initDone = JNI_TRUE;
+ }
+ return portUnreachableSupported;
+}
+
+/*
+ * This function "purges" all outstanding ICMP port unreachable packets
+ * outstanding on a socket and returns JNI_TRUE if any ICMP messages
+ * have been purged. The rational for purging is to emulate normal BSD
+ * behaviour whereby receiving a "connection reset" status resets the
+ * socket.
+ */
+jboolean purgeOutstandingICMP(JNIEnv *env, jclass clazz, jint fd)
+{
+ jboolean got_icmp = JNI_FALSE;
+ char buf[1];
+ fd_set tbl;
+ struct timeval t = { 0, 0 };
+ struct sockaddr_in rmtaddr;
+ int addrlen = sizeof(rmtaddr);
+
+ /*
+ * A no-op if this OS doesn't support it.
+ */
+ if (!supportPortUnreachable()) {
+ return JNI_FALSE;
+ }
+
+ /*
+ * Peek at the queue to see if there is an ICMP port unreachable. If there
+ * is then receive it.
+ */
+ FD_ZERO(&tbl);
+ FD_SET((u_int)fd, &tbl);
+ while(1) {
+ if (select(/*ignored*/fd+1, &tbl, 0, 0, &t) <= 0) {
+ break;
+ }
+ if (recvfrom(fd, buf, 1, MSG_PEEK,
+ (struct sockaddr *)&rmtaddr, &addrlen) != SOCKET_ERROR) {
+ break;
+ }
+ if (WSAGetLastError() != WSAECONNRESET) {
+ /* some other error - we don't care here */
+ break;
+ }
+
+ recvfrom(fd, buf, 1, 0, (struct sockaddr *)&rmtaddr, &addrlen);
+ got_icmp = JNI_TRUE;
+ }
+
+ return got_icmp;
+}
+
+JNIEXPORT void JNICALL
+Java_sun_nio_ch_DatagramChannelImpl_disconnect0(JNIEnv *env, jobject this,
+ jobject fdo)
+{
+ jint fd = fdval(env, fdo);
+ int rv = 0;
+ struct sockaddr_in psa;
+ int sa_len = sizeof(psa);
+
+ memset(&psa, 0, sa_len);
+
+ rv = connect((SOCKET)fd, (struct sockaddr *)&psa, sa_len);
+ if (rv == SOCKET_ERROR) {
+ handleSocketError(env, WSAGetLastError());
+ }
+}
+
+JNIEXPORT jint JNICALL
+Java_sun_nio_ch_DatagramChannelImpl_receive0(JNIEnv *env, jobject this,
+ jobject fdo, jlong address,
+ jint len, jboolean connected)
+{
+ jint fd = fdval(env, fdo);
+ void *buf = (void *)jlong_to_ptr(address);
+ struct sockaddr_in psa;
+ int sa_len = sizeof(psa);
+ BOOL retry = FALSE;
+ jint n;
+
+ do {
+ retry = FALSE;
+ n = recvfrom((SOCKET)fd,
+ (char *)buf,
+ len,
+ 0,
+ (struct sockaddr *)&psa,
+ &sa_len);
+
+ if (n == SOCKET_ERROR) {
+ int theErr = (jint)WSAGetLastError();
+ if (theErr == WSAEMSGSIZE) {
+ /* Spec says the rest of the data will be discarded... */
+ n = len;
+ } else if (theErr == WSAECONNRESET) {
+ purgeOutstandingICMP(env, this, fd);
+ if (connected == JNI_FALSE) {
+ retry = TRUE;
+ } else {
+ JNU_ThrowByName(env, JNU_JAVANETPKG "PortUnreachableException", 0);
+ return IOS_THROWN;
+ }
+ } else if (theErr == WSAEWOULDBLOCK) {
+ return IOS_UNAVAILABLE;
+ } else return handleSocketError(env, theErr);
+ }
+ } while (retry);
+
+ if (!isSenderCached(env, this, &psa)) {
+ int port = ntohs(psa.sin_port);
+ jobject ia = (*env)->NewObject(env, ia_class, ia_ctorID);
+ jobject isa = NULL;
+
+ if (psa.sin_family != AF_INET) {
+ JNU_ThrowByName(env, JNU_JAVANETPKG "SocketException",
+ "Protocol family unavailable");
+ }
+
+ if (ia != NULL) {
+ // populate InetAddress (assumes AF_INET)
+ (*env)->SetIntField(env, ia, ia_addrID, ntohl(psa.sin_addr.s_addr));
+
+ // create InetSocketAddress
+ isa = (*env)->NewObject(env, isa_class, isa_ctorID, ia, port);
+ }
+
+ if (isa == NULL) {
+ JNU_ThrowOutOfMemoryError(env, "heap allocation failed");
+ return IOS_THROWN;
+ }
+
+ // update cachedSenderInetAddress/cachedSenderPort
+ (*env)->SetObjectField(env, this, dci_senderAddrID, ia);
+ (*env)->SetIntField(env, this, dci_senderPortID, port);
+
+ // update sender
+ (*env)->SetObjectField(env, this, dci_senderID, isa);
+ }
+ return n;
+}
+
+JNIEXPORT jint JNICALL
+Java_sun_nio_ch_DatagramChannelImpl_send0(JNIEnv *env, jobject this,
+ jobject fdo, jlong address,
+ jint len, jobject dest)
+{
+ jint fd = fdval(env, fdo);
+ void *buf = (void *)jlong_to_ptr(address);
+ SOCKETADDRESS psa;
+ int sa_len = sizeof(psa);
+ jint rv = 0;
+ jobject destAddress = (*env)->GetObjectField(env, dest, isa_addrID);
+ jint destPort = (*env)->GetIntField(env, dest, isa_portID);
+
+
+ if (NET_InetAddressToSockaddr(env, destAddress, destPort,
+ (struct sockaddr *)&psa,
+ &sa_len, JNI_FALSE) != 0) {
+ return IOS_THROWN;
+ }
+
+ rv = sendto((SOCKET)fd,
+ buf,
+ len,
+ 0,
+ (struct sockaddr *)&psa,
+ sa_len);
+ if (rv == SOCKET_ERROR) {
+ int theErr = (jint)WSAGetLastError();
+ if (theErr == WSAEWOULDBLOCK) {
+ return IOS_UNAVAILABLE;
+ }
+ return handleSocketError(env, (jint)WSAGetLastError());
+ }
+ return rv;
+}