aboutsummaryrefslogtreecommitdiff
path: root/src/org/xbill/DNS/TCPClient.java
diff options
context:
space:
mode:
authorShuyi Chen <shuyichen@google.com>2013-05-22 14:51:55 -0700
committerShuyi Chen <shuyichen@google.com>2013-05-22 17:19:30 -0700
commitd7955ce24d294fb2014c59d11fca184471056f44 (patch)
treee260500b0b7639127038495d46a0ad6dcbb6d96c /src/org/xbill/DNS/TCPClient.java
parent8f4ce9ea0de51fee918bffe19c434612d6bbb2d7 (diff)
downloadsmack-main.tar.gz
Add android smack source.HEADandroid-wear-5.0.0_r1android-sdk-4.4.2_r1.0.1android-sdk-4.4.2_r1android-l-preview_r2android-cts-5.1_r9android-cts-5.1_r8android-cts-5.1_r7android-cts-5.1_r6android-cts-5.1_r5android-cts-5.1_r4android-cts-5.1_r3android-cts-5.1_r28android-cts-5.1_r27android-cts-5.1_r26android-cts-5.1_r25android-cts-5.1_r24android-cts-5.1_r23android-cts-5.1_r22android-cts-5.1_r21android-cts-5.1_r20android-cts-5.1_r2android-cts-5.1_r19android-cts-5.1_r18android-cts-5.1_r17android-cts-5.1_r16android-cts-5.1_r15android-cts-5.1_r14android-cts-5.1_r13android-cts-5.1_r10android-cts-5.1_r1android-cts-5.0_r9android-cts-5.0_r8android-cts-5.0_r7android-cts-5.0_r6android-cts-5.0_r5android-cts-5.0_r4android-cts-5.0_r3android-cts-4.4_r4android-cts-4.4_r1android-5.1.1_r9android-5.1.1_r8android-5.1.1_r7android-5.1.1_r6android-5.1.1_r5android-5.1.1_r4android-5.1.1_r38android-5.1.1_r37android-5.1.1_r36android-5.1.1_r35android-5.1.1_r34android-5.1.1_r33android-5.1.1_r30android-5.1.1_r3android-5.1.1_r29android-5.1.1_r28android-5.1.1_r26android-5.1.1_r25android-5.1.1_r24android-5.1.1_r23android-5.1.1_r22android-5.1.1_r20android-5.1.1_r2android-5.1.1_r19android-5.1.1_r18android-5.1.1_r17android-5.1.1_r16android-5.1.1_r15android-5.1.1_r14android-5.1.1_r13android-5.1.1_r12android-5.1.1_r10android-5.1.1_r1android-5.1.0_r5android-5.1.0_r4android-5.1.0_r3android-5.1.0_r1android-5.0.2_r3android-5.0.2_r1android-5.0.1_r1android-5.0.0_r7android-5.0.0_r6android-5.0.0_r5.1android-5.0.0_r5android-5.0.0_r4android-5.0.0_r3android-5.0.0_r2android-5.0.0_r1android-4.4w_r1android-4.4_r1.2.0.1android-4.4_r1.2android-4.4_r1.1.0.1android-4.4_r1.1android-4.4_r1.0.1android-4.4_r1android-4.4_r0.9android-4.4_r0.8android-4.4_r0.7android-4.4.4_r2.0.1android-4.4.4_r2android-4.4.4_r1.0.1android-4.4.4_r1android-4.4.3_r1.1.0.1android-4.4.3_r1.1android-4.4.3_r1.0.1android-4.4.3_r1android-4.4.2_r2.0.1android-4.4.2_r2android-4.4.2_r1.0.1android-4.4.2_r1android-4.4.1_r1.0.1android-4.4.1_r1android-4.3_r3.1android-4.3_r3android-4.3_r2.3android-4.3_r2.2android-4.3_r2.1android-4.3_r2android-4.3_r1.1android-4.3_r1android-4.3_r0.9.1android-4.3_r0.9android-4.3.1_r1tools_r22.2mastermainlollipop-wear-releaselollipop-releaselollipop-mr1-wfc-releaselollipop-mr1-releaselollipop-mr1-fi-releaselollipop-mr1-devlollipop-mr1-cts-releaselollipop-devlollipop-cts-releasel-previewkitkat-wearkitkat-releasekitkat-mr2.2-releasekitkat-mr2.1-releasekitkat-mr2-releasekitkat-mr1.1-releasekitkat-mr1-releasekitkat-devkitkat-cts-releasekitkat-cts-devjb-mr2.0.0-releasejb-mr2.0-releasejb-mr2-releasejb-mr2-devidea133-weekly-releaseidea133
Change-Id: I49ce97136c17173c4ae3965c694af6e7bc49897d
Diffstat (limited to 'src/org/xbill/DNS/TCPClient.java')
-rw-r--r--src/org/xbill/DNS/TCPClient.java132
1 files changed, 132 insertions, 0 deletions
diff --git a/src/org/xbill/DNS/TCPClient.java b/src/org/xbill/DNS/TCPClient.java
new file mode 100644
index 0000000..1f17d72
--- /dev/null
+++ b/src/org/xbill/DNS/TCPClient.java
@@ -0,0 +1,132 @@
+// Copyright (c) 2005 Brian Wellington (bwelling@xbill.org)
+
+package org.xbill.DNS;
+
+import java.io.*;
+import java.net.*;
+import java.nio.*;
+import java.nio.channels.*;
+
+final class TCPClient extends Client {
+
+public
+TCPClient(long endTime) throws IOException {
+ super(SocketChannel.open(), endTime);
+}
+
+void
+bind(SocketAddress addr) throws IOException {
+ SocketChannel channel = (SocketChannel) key.channel();
+ channel.socket().bind(addr);
+}
+
+void
+connect(SocketAddress addr) throws IOException {
+ SocketChannel channel = (SocketChannel) key.channel();
+ if (channel.connect(addr))
+ return;
+ key.interestOps(SelectionKey.OP_CONNECT);
+ try {
+ while (!channel.finishConnect()) {
+ if (!key.isConnectable())
+ blockUntil(key, endTime);
+ }
+ }
+ finally {
+ if (key.isValid())
+ key.interestOps(0);
+ }
+}
+
+void
+send(byte [] data) throws IOException {
+ SocketChannel channel = (SocketChannel) key.channel();
+ verboseLog("TCP write", data);
+ byte [] lengthArray = new byte[2];
+ lengthArray[0] = (byte)(data.length >>> 8);
+ lengthArray[1] = (byte)(data.length & 0xFF);
+ ByteBuffer [] buffers = new ByteBuffer[2];
+ buffers[0] = ByteBuffer.wrap(lengthArray);
+ buffers[1] = ByteBuffer.wrap(data);
+ int nsent = 0;
+ key.interestOps(SelectionKey.OP_WRITE);
+ try {
+ while (nsent < data.length + 2) {
+ if (key.isWritable()) {
+ long n = channel.write(buffers);
+ if (n < 0)
+ throw new EOFException();
+ nsent += (int) n;
+ if (nsent < data.length + 2 &&
+ System.currentTimeMillis() > endTime)
+ throw new SocketTimeoutException();
+ } else
+ blockUntil(key, endTime);
+ }
+ }
+ finally {
+ if (key.isValid())
+ key.interestOps(0);
+ }
+}
+
+private byte []
+_recv(int length) throws IOException {
+ SocketChannel channel = (SocketChannel) key.channel();
+ int nrecvd = 0;
+ byte [] data = new byte[length];
+ ByteBuffer buffer = ByteBuffer.wrap(data);
+ key.interestOps(SelectionKey.OP_READ);
+ try {
+ while (nrecvd < length) {
+ if (key.isReadable()) {
+ long n = channel.read(buffer);
+ if (n < 0)
+ throw new EOFException();
+ nrecvd += (int) n;
+ if (nrecvd < length &&
+ System.currentTimeMillis() > endTime)
+ throw new SocketTimeoutException();
+ } else
+ blockUntil(key, endTime);
+ }
+ }
+ finally {
+ if (key.isValid())
+ key.interestOps(0);
+ }
+ return data;
+}
+
+byte []
+recv() throws IOException {
+ byte [] buf = _recv(2);
+ int length = ((buf[0] & 0xFF) << 8) + (buf[1] & 0xFF);
+ byte [] data = _recv(length);
+ verboseLog("TCP read", data);
+ return data;
+}
+
+static byte []
+sendrecv(SocketAddress local, SocketAddress remote, byte [] data, long endTime)
+throws IOException
+{
+ TCPClient client = new TCPClient(endTime);
+ try {
+ if (local != null)
+ client.bind(local);
+ client.connect(remote);
+ client.send(data);
+ return client.recv();
+ }
+ finally {
+ client.cleanup();
+ }
+}
+
+static byte []
+sendrecv(SocketAddress addr, byte [] data, long endTime) throws IOException {
+ return sendrecv(null, addr, data, endTime);
+}
+
+}