Mercurial > hg > openjdk7.svn
view j2se/src/windows/native/sun/nio/ch/DatagramChannelImpl.c @ 3:64ed597c0ad3 trunk
[svn] Load openjdk/jdk7/b15 into jdk/trunk.
author | xiomara |
---|---|
date | Thu, 05 Jul 2007 23:47:33 +0000 |
parents | 193df1943809 |
children |
line wrap: on
line source
/* * 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; }