|
Java example source code file (Socket.java)
The Socket.java Java example source code/* * Copyright 2015 The Netty Project * * The Netty Project licenses this file to you 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. */ package io.netty.channel.unix; import io.netty.channel.ChannelException; import io.netty.util.CharsetUtil; import io.netty.util.internal.ThrowableUtil; import java.io.IOException; import java.net.Inet6Address; import java.net.InetAddress; import java.net.InetSocketAddress; import java.net.SocketAddress; import java.nio.ByteBuffer; import java.nio.channels.ClosedChannelException; import static io.netty.channel.unix.Errors.ERRNO_EAGAIN_NEGATIVE; import static io.netty.channel.unix.Errors.ERRNO_EINPROGRESS_NEGATIVE; import static io.netty.channel.unix.Errors.ERRNO_EWOULDBLOCK_NEGATIVE; import static io.netty.channel.unix.Errors.ioResult; import static io.netty.channel.unix.Errors.newConnectException; import static io.netty.channel.unix.Errors.newIOException; import static io.netty.channel.unix.NativeInetAddress.address; import static io.netty.channel.unix.NativeInetAddress.ipv4MappedIpv6Address; /** * Provides a JNI bridge to native socket operations. * <strong>Internal usage only! */ public final class Socket extends FileDescriptor { private static final ClosedChannelException SHUTDOWN_CLOSED_CHANNEL_EXCEPTION = ThrowableUtil.unknownStackTrace( new ClosedChannelException(), Socket.class, "shutdown(...)"); private static final ClosedChannelException SEND_TO_CLOSED_CHANNEL_EXCEPTION = ThrowableUtil.unknownStackTrace( new ClosedChannelException(), Socket.class, "sendTo(...)"); private static final ClosedChannelException SEND_TO_ADDRESS_CLOSED_CHANNEL_EXCEPTION = ThrowableUtil.unknownStackTrace(new ClosedChannelException(), Socket.class, "sendToAddress(...)"); private static final ClosedChannelException SEND_TO_ADDRESSES_CLOSED_CHANNEL_EXCEPTION = ThrowableUtil.unknownStackTrace(new ClosedChannelException(), Socket.class, "sendToAddresses(...)"); private static final Errors.NativeIoException SEND_TO_CONNECTION_RESET_EXCEPTION = ThrowableUtil.unknownStackTrace( Errors.newConnectionResetException("syscall:sendto(...)", Errors.ERRNO_EPIPE_NEGATIVE), Socket.class, "sendTo(...)"); private static final Errors.NativeIoException SEND_TO_ADDRESS_CONNECTION_RESET_EXCEPTION = ThrowableUtil.unknownStackTrace(Errors.newConnectionResetException("syscall:sendto(...)", Errors.ERRNO_EPIPE_NEGATIVE), Socket.class, "sendToAddress(...)"); private static final Errors.NativeIoException CONNECTION_RESET_EXCEPTION_SENDMSG = ThrowableUtil.unknownStackTrace( Errors.newConnectionResetException("syscall:sendmsg(...)", Errors.ERRNO_EPIPE_NEGATIVE), Socket.class, "sendToAddresses(...)"); private static final Errors.NativeIoException CONNECTION_NOT_CONNECTED_SHUTDOWN_EXCEPTION = ThrowableUtil.unknownStackTrace(Errors.newConnectionResetException("syscall:shutdown(...)", Errors.ERRNO_ENOTCONN_NEGATIVE), Socket.class, "shutdown(...)"); public Socket(int fd) { super(fd); } public void shutdown() throws IOException { shutdown(true, true); } public void shutdown(boolean read, boolean write) throws IOException { for (;;) { // We need to only shutdown what has not been shutdown yet, and if there is no change we should not // shutdown anything. This is because if the underlying FD is reused and we still have an object which // represents the previous incarnation of the FD we need to be sure we don't inadvertently shutdown the // "new" FD without explicitly having a change. final int oldState = this.state; int newState = oldState; if (read && !isInputShutdown(newState)) { newState = inputShutdown(newState); } if (write && !isOutputShutdown(newState)) { newState = outputShutdown(newState); } // If there is no change in state, then we should not take any action. if (newState == oldState) { return; } if (casState(oldState, newState)) { break; } } int res = shutdown(fd, read, write); if (res < 0) { ioResult("shutdown", res, CONNECTION_NOT_CONNECTED_SHUTDOWN_EXCEPTION, SHUTDOWN_CLOSED_CHANNEL_EXCEPTION); } } public boolean isShutdown() { int state = this.state; return isInputShutdown(state) && isOutputShutdown(state); } public boolean isInputShutdown() { return isInputShutdown(state); } public boolean isOutputShutdown() { return isOutputShutdown(state); } public int sendTo(ByteBuffer buf, int pos, int limit, InetAddress addr, int port) throws IOException { // just duplicate the toNativeInetAddress code here to minimize object creation as this method is expected // to be called frequently byte[] address; int scopeId; if (addr instanceof Inet6Address) { address = addr.getAddress(); scopeId = ((Inet6Address) addr).getScopeId(); } else { // convert to ipv4 mapped ipv6 address; scopeId = 0; address = ipv4MappedIpv6Address(addr.getAddress()); } int res = sendTo(fd, buf, pos, limit, address, scopeId, port); if (res >= 0) { return res; } return ioResult("sendTo", res, SEND_TO_CONNECTION_RESET_EXCEPTION, SEND_TO_CLOSED_CHANNEL_EXCEPTION); } public int sendToAddress(long memoryAddress, int pos, int limit, InetAddress addr, int port) throws IOException { // just duplicate the toNativeInetAddress code here to minimize object creation as this method is expected // to be called frequently byte[] address; int scopeId; if (addr instanceof Inet6Address) { address = addr.getAddress(); scopeId = ((Inet6Address) addr).getScopeId(); } else { // convert to ipv4 mapped ipv6 address; scopeId = 0; address = ipv4MappedIpv6Address(addr.getAddress()); } int res = sendToAddress(fd, memoryAddress, pos, limit, address, scopeId, port); if (res >= 0) { return res; } return ioResult("sendToAddress", res, SEND_TO_ADDRESS_CONNECTION_RESET_EXCEPTION, SEND_TO_ADDRESS_CLOSED_CHANNEL_EXCEPTION); } public int sendToAddresses(long memoryAddress, int length, InetAddress addr, int port) throws IOException { // just duplicate the toNativeInetAddress code here to minimize object creation as this method is expected // to be called frequently byte[] address; int scopeId; if (addr instanceof Inet6Address) { address = addr.getAddress(); scopeId = ((Inet6Address) addr).getScopeId(); } else { // convert to ipv4 mapped ipv6 address; scopeId = 0; address = ipv4MappedIpv6Address(addr.getAddress()); } int res = sendToAddresses(fd, memoryAddress, length, address, scopeId, port); if (res >= 0) { return res; } return ioResult("sendToAddresses", res, CONNECTION_RESET_EXCEPTION_SENDMSG, SEND_TO_ADDRESSES_CLOSED_CHANNEL_EXCEPTION); } public DatagramSocketAddress recvFrom(ByteBuffer buf, int pos, int limit) throws IOException { return recvFrom(fd, buf, pos, limit); } public DatagramSocketAddress recvFromAddress(long memoryAddress, int pos, int limit) throws IOException { return recvFromAddress(fd, memoryAddress, pos, limit); } public boolean connect(SocketAddress socketAddress) throws IOException { int res; if (socketAddress instanceof InetSocketAddress) { InetSocketAddress inetSocketAddress = (InetSocketAddress) socketAddress; NativeInetAddress address = NativeInetAddress.newInstance(inetSocketAddress.getAddress()); res = connect(fd, address.address, address.scopeId, inetSocketAddress.getPort()); } else if (socketAddress instanceof DomainSocketAddress) { DomainSocketAddress unixDomainSocketAddress = (DomainSocketAddress) socketAddress; res = connectDomainSocket(fd, unixDomainSocketAddress.path().getBytes(CharsetUtil.UTF_8)); } else { throw new Error("Unexpected SocketAddress implementation " + socketAddress); } if (res < 0) { if (res == ERRNO_EINPROGRESS_NEGATIVE) { // connect not complete yet need to wait for EPOLLOUT event return false; } throw newConnectException("connect", res); } return true; } public boolean finishConnect() throws IOException { int res = finishConnect(fd); if (res < 0) { if (res == ERRNO_EINPROGRESS_NEGATIVE) { // connect still in progress return false; } throw newConnectException("finishConnect", res); } return true; } public void bind(SocketAddress socketAddress) throws IOException { if (socketAddress instanceof InetSocketAddress) { InetSocketAddress addr = (InetSocketAddress) socketAddress; NativeInetAddress address = NativeInetAddress.newInstance(addr.getAddress()); int res = bind(fd, address.address, address.scopeId, addr.getPort()); if (res < 0) { throw newIOException("bind", res); } } else if (socketAddress instanceof DomainSocketAddress) { DomainSocketAddress addr = (DomainSocketAddress) socketAddress; int res = bindDomainSocket(fd, addr.path().getBytes(CharsetUtil.UTF_8)); if (res < 0) { throw newIOException("bind", res); } } else { throw new Error("Unexpected SocketAddress implementation " + socketAddress); } } public void listen(int backlog) throws IOException { int res = listen(fd, backlog); if (res < 0) { throw newIOException("listen", res); } } public int accept(byte[] addr) throws IOException { int res = accept(fd, addr); if (res >= 0) { return res; } if (res == ERRNO_EAGAIN_NEGATIVE || res == ERRNO_EWOULDBLOCK_NEGATIVE) { // Everything consumed so just return -1 here. return -1; } throw newIOException("accept", res); } public InetSocketAddress remoteAddress() { byte[] addr = remoteAddress(fd); // addr may be null if getpeername failed. // See https://github.com/netty/netty/issues/3328 if (addr == null) { return null; } return address(addr, 0, addr.length); } public InetSocketAddress localAddress() { byte[] addr = localAddress(fd); // addr may be null if getpeername failed. // See https://github.com/netty/netty/issues/3328 if (addr == null) { return null; } return address(addr, 0, addr.length); } public int getReceiveBufferSize() throws IOException { return getReceiveBufferSize(fd); } public int getSendBufferSize() throws IOException { return getSendBufferSize(fd); } public boolean isKeepAlive() throws IOException { return isKeepAlive(fd) != 0; } public boolean isTcpNoDelay() throws IOException { return isTcpNoDelay(fd) != 0; } public boolean isTcpCork() throws IOException { return isTcpCork(fd) != 0; } public int getSoLinger() throws IOException { return getSoLinger(fd); } public int getTcpDeferAccept() throws IOException { return getTcpDeferAccept(fd); } public boolean isTcpQuickAck() throws IOException { return isTcpQuickAck(fd) != 0; } public int getSoError() throws IOException { return getSoError(fd); } public void setKeepAlive(boolean keepAlive) throws IOException { setKeepAlive(fd, keepAlive ? 1 : 0); } public void setReceiveBufferSize(int receiveBufferSize) throws IOException { setReceiveBufferSize(fd, receiveBufferSize); } public void setSendBufferSize(int sendBufferSize) throws IOException { setSendBufferSize(fd, sendBufferSize); } public void setTcpNoDelay(boolean tcpNoDelay) throws IOException { setTcpNoDelay(fd, tcpNoDelay ? 1 : 0); } public void setTcpCork(boolean tcpCork) throws IOException { setTcpCork(fd, tcpCork ? 1 : 0); } public void setSoLinger(int soLinger) throws IOException { setSoLinger(fd, soLinger); } public void setTcpDeferAccept(int deferAccept) throws IOException { setTcpDeferAccept(fd, deferAccept); } public void setTcpQuickAck(boolean quickAck) throws IOException { setTcpQuickAck(fd, quickAck ? 1 : 0); } @Override public String toString() { return "Socket{" + "fd=" + fd + '}'; } public static Socket newSocketStream() { int res = newSocketStreamFd(); if (res < 0) { throw new ChannelException(newIOException("newSocketStream", res)); } return new Socket(res); } public static Socket newSocketDgram() { int res = newSocketDgramFd(); if (res < 0) { throw new ChannelException(newIOException("newSocketDgram", res)); } return new Socket(res); } public static Socket newSocketDomain() { int res = newSocketDomainFd(); if (res < 0) { throw new ChannelException(newIOException("newSocketDomain", res)); } return new Socket(res); } private static native int shutdown(int fd, boolean read, boolean write); private static native int connect(int fd, byte[] address, int scopeId, int port); private static native int connectDomainSocket(int fd, byte[] path); private static native int finishConnect(int fd); private static native int bind(int fd, byte[] address, int scopeId, int port); private static native int bindDomainSocket(int fd, byte[] path); private static native int listen(int fd, int backlog); private static native int accept(int fd, byte[] addr); private static native byte[] remoteAddress(int fd); private static native byte[] localAddress(int fd); private static native int sendTo( int fd, ByteBuffer buf, int pos, int limit, byte[] address, int scopeId, int port); private static native int sendToAddress( int fd, long memoryAddress, int pos, int limit, byte[] address, int scopeId, int port); private static native int sendToAddresses( int fd, long memoryAddress, int length, byte[] address, int scopeId, int port); private static native DatagramSocketAddress recvFrom( int fd, ByteBuffer buf, int pos, int limit) throws IOException; private static native DatagramSocketAddress recvFromAddress( int fd, long memoryAddress, int pos, int limit) throws IOException; private static native int newSocketStreamFd(); private static native int newSocketDgramFd(); private static native int newSocketDomainFd(); private static native int getReceiveBufferSize(int fd) throws IOException; private static native int getSendBufferSize(int fd) throws IOException; private static native int isKeepAlive(int fd) throws IOException; private static native int isTcpNoDelay(int fd) throws IOException; private static native int isTcpCork(int fd) throws IOException; private static native int getSoLinger(int fd) throws IOException; private static native int getSoError(int fd) throws IOException; private static native int getTcpDeferAccept(int fd) throws IOException; private static native int isTcpQuickAck(int fd) throws IOException; private static native void setKeepAlive(int fd, int keepAlive) throws IOException; private static native void setReceiveBufferSize(int fd, int receiveBufferSize) throws IOException; private static native void setSendBufferSize(int fd, int sendBufferSize) throws IOException; private static native void setTcpNoDelay(int fd, int tcpNoDelay) throws IOException; private static native void setTcpCork(int fd, int tcpCork) throws IOException; private static native void setSoLinger(int fd, int soLinger) throws IOException; private static native void setTcpDeferAccept(int fd, int deferAccept) throws IOException; private static native void setTcpQuickAck(int fd, int quickAck) throws IOException; } Other Java examples (source code examples)Here is a short list of links related to this Java Socket.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.