diff options
author | Shuyi Chen <shuyichen@google.com> | 2013-05-22 14:51:55 -0700 |
---|---|---|
committer | Shuyi Chen <shuyichen@google.com> | 2013-05-22 17:19:30 -0700 |
commit | d7955ce24d294fb2014c59d11fca184471056f44 (patch) | |
tree | e260500b0b7639127038495d46a0ad6dcbb6d96c /src/org/xbill/DNS/UDPClient.java | |
parent | 8f4ce9ea0de51fee918bffe19c434612d6bbb2d7 (diff) | |
download | smack-android-4.4.1_r1.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/UDPClient.java')
-rw-r--r-- | src/org/xbill/DNS/UDPClient.java | 164 |
1 files changed, 164 insertions, 0 deletions
diff --git a/src/org/xbill/DNS/UDPClient.java b/src/org/xbill/DNS/UDPClient.java new file mode 100644 index 0000000..e752ce4 --- /dev/null +++ b/src/org/xbill/DNS/UDPClient.java @@ -0,0 +1,164 @@ +// Copyright (c) 2005 Brian Wellington (bwelling@xbill.org) + +package org.xbill.DNS; + +import java.io.*; +import java.net.*; +import java.security.SecureRandom; +import java.nio.*; +import java.nio.channels.*; + +final class UDPClient extends Client { + +private static final int EPHEMERAL_START = 1024; +private static final int EPHEMERAL_STOP = 65535; +private static final int EPHEMERAL_RANGE = EPHEMERAL_STOP - EPHEMERAL_START; + +private static SecureRandom prng = new SecureRandom(); +private static volatile boolean prng_initializing = true; + +/* + * On some platforms (Windows), the SecureRandom module initialization involves + * a call to InetAddress.getLocalHost(), which can end up here if using a + * dnsjava name service provider. + * + * This can cause problems in multiple ways. + * - If the SecureRandom seed generation process calls into here, and this + * module attempts to seed the local SecureRandom object, the thread hangs. + * - If something else calls InetAddress.getLocalHost(), and that causes this + * module to seed the local SecureRandom object, the thread hangs. + * + * To avoid both of these, check at initialization time to see if InetAddress + * is in the call chain. If so, initialize the SecureRandom object in a new + * thread, and disable port randomization until it completes. + */ +static { + new Thread(new Runnable() { + public void run() { + int n = prng.nextInt(); + prng_initializing = false; + }}).start(); +} + +private boolean bound = false; + +public +UDPClient(long endTime) throws IOException { + super(DatagramChannel.open(), endTime); +} + +private void +bind_random(InetSocketAddress addr) throws IOException +{ + if (prng_initializing) { + try { + Thread.sleep(2); + } + catch (InterruptedException e) { + } + if (prng_initializing) + return; + } + + DatagramChannel channel = (DatagramChannel) key.channel(); + InetSocketAddress temp; + + for (int i = 0; i < 1024; i++) { + try { + int port = prng.nextInt(EPHEMERAL_RANGE) + + EPHEMERAL_START; + if (addr != null) + temp = new InetSocketAddress(addr.getAddress(), + port); + else + temp = new InetSocketAddress(port); + channel.socket().bind(temp); + bound = true; + return; + } + catch (SocketException e) { + } + } +} + +void +bind(SocketAddress addr) throws IOException { + if (addr == null || + (addr instanceof InetSocketAddress && + ((InetSocketAddress)addr).getPort() == 0)) + { + bind_random((InetSocketAddress) addr); + if (bound) + return; + } + + if (addr != null) { + DatagramChannel channel = (DatagramChannel) key.channel(); + channel.socket().bind(addr); + bound = true; + } +} + +void +connect(SocketAddress addr) throws IOException { + if (!bound) + bind(null); + DatagramChannel channel = (DatagramChannel) key.channel(); + channel.connect(addr); +} + +void +send(byte [] data) throws IOException { + DatagramChannel channel = (DatagramChannel) key.channel(); + verboseLog("UDP write", data); + channel.write(ByteBuffer.wrap(data)); +} + +byte [] +recv(int max) throws IOException { + DatagramChannel channel = (DatagramChannel) key.channel(); + byte [] temp = new byte[max]; + key.interestOps(SelectionKey.OP_READ); + try { + while (!key.isReadable()) + blockUntil(key, endTime); + } + finally { + if (key.isValid()) + key.interestOps(0); + } + long ret = channel.read(ByteBuffer.wrap(temp)); + if (ret <= 0) + throw new EOFException(); + int len = (int) ret; + byte [] data = new byte[len]; + System.arraycopy(temp, 0, data, 0, len); + verboseLog("UDP read", data); + return data; +} + +static byte [] +sendrecv(SocketAddress local, SocketAddress remote, byte [] data, int max, + long endTime) +throws IOException +{ + UDPClient client = new UDPClient(endTime); + try { + client.bind(local); + client.connect(remote); + client.send(data); + return client.recv(max); + } + finally { + client.cleanup(); + } +} + +static byte [] +sendrecv(SocketAddress addr, byte [] data, int max, long endTime) +throws IOException +{ + return sendrecv(null, addr, data, max, endTime); +} + +} |