- enhanced intranet portscan speed (better time-out) git-svn-id: https://svn.berlios.de/svnroot/repos/yacy/trunk@7227 6c8d7289-2bf4-0310-a012-ef5d649a1542pull/1/head
orbiter
15 years ago
parent
b9f405d1e8
commit
091dd3f6ec
@ -0,0 +1,325 @@
|
||||
/**
|
||||
* TimeoutRequest
|
||||
* Copyright 2010 by Michael Peter Christen, mc@yacy.net, Frankfurt a. M., Germany
|
||||
* First released 08.10.2007 at http://yacy.net
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Lesser General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2.1 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library 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
|
||||
* Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with this program in the file lgpl21.txt
|
||||
* If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package net.yacy.cora.protocol;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.InetAddress;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.Socket;
|
||||
import java.net.UnknownHostException;
|
||||
import java.util.concurrent.Callable;
|
||||
import java.util.concurrent.CancellationException;
|
||||
import java.util.concurrent.ExecutionException;
|
||||
import java.util.concurrent.ExecutorService;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.Future;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.concurrent.TimeoutException;
|
||||
|
||||
import jcifs.smb.SmbException;
|
||||
import jcifs.smb.SmbFile;
|
||||
|
||||
import net.yacy.cora.document.MultiProtocolURI;
|
||||
|
||||
/**
|
||||
* TimeoutRequest is a class that can apply a timeout on method calls that may block
|
||||
* for undefined time. Some network operations can only be accessed without a given
|
||||
* time-out value. Using this class all network operations may be timed out.
|
||||
* This class provides also some static methods that give already solutions for typical
|
||||
* network operations that should be timed-out, like dns resolving and reverse domain name resolving.
|
||||
*/
|
||||
public class TimeoutRequest<E> {
|
||||
|
||||
private Callable<E> call;
|
||||
|
||||
/**
|
||||
* initialize the TimeoutRequest with a callable method
|
||||
*/
|
||||
public TimeoutRequest(Callable<E> call) {
|
||||
this.call = call;
|
||||
}
|
||||
|
||||
/**
|
||||
* call the method using a time-out
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws ExecutionException
|
||||
*/
|
||||
public E call(long timeout) throws ExecutionException {
|
||||
ExecutorService service = Executors.newSingleThreadExecutor();
|
||||
final Future<E> taskFuture = service.submit(this.call);
|
||||
Runnable t = new Runnable() {
|
||||
public void run() { taskFuture.cancel(true); }
|
||||
};
|
||||
service.execute(t);
|
||||
service.shutdown();
|
||||
try {
|
||||
return taskFuture.get(timeout, TimeUnit.MILLISECONDS);
|
||||
} catch (CancellationException e) {
|
||||
// callable was interrupted
|
||||
throw new ExecutionException(e);
|
||||
} catch (InterruptedException e) {
|
||||
// service was shutdown
|
||||
throw new ExecutionException(e);
|
||||
} catch (ExecutionException e) {
|
||||
// callable failed unexpectedly
|
||||
throw e;
|
||||
} catch (TimeoutException e) {
|
||||
// time-out
|
||||
throw new ExecutionException(e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* ping a remote server using a given uri and a time-out
|
||||
* @param uri
|
||||
* @param timeout
|
||||
* @return true if the server exists and replies within the given time-out
|
||||
* @throws ExecutionException
|
||||
*/
|
||||
public static boolean ping(final MultiProtocolURI uri, final int timeout) throws ExecutionException {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() {
|
||||
try {
|
||||
Socket socket = new Socket();
|
||||
socket.connect(new InetSocketAddress(uri.getHost(), uri.getPort()), timeout);
|
||||
if (socket.isConnected()) {
|
||||
socket.close();
|
||||
return Boolean.TRUE;
|
||||
}
|
||||
return Boolean.FALSE;
|
||||
} catch (UnknownHostException e) {
|
||||
return Boolean.FALSE;
|
||||
} catch (IOException e) {
|
||||
return Boolean.FALSE;
|
||||
}
|
||||
}
|
||||
}).call(timeout).booleanValue();
|
||||
}
|
||||
|
||||
/**
|
||||
* do a DNS lookup within a given time
|
||||
* @param host
|
||||
* @param timeout
|
||||
* @return the InetAddress for a given domain name
|
||||
* @throws ExecutionException
|
||||
*/
|
||||
public static InetAddress getByName(final String host, final long timeout) throws ExecutionException {
|
||||
return new TimeoutRequest<InetAddress>(new Callable<InetAddress>() {
|
||||
public InetAddress call() {
|
||||
try {
|
||||
return InetAddress.getByName(host);
|
||||
} catch (UnknownHostException e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}).call(timeout);
|
||||
}
|
||||
|
||||
/**
|
||||
* perform a reverse domain name lookup for a given InetAddress within a given timeout
|
||||
* @param i
|
||||
* @param timeout
|
||||
* @return the host name of a given InetAddress
|
||||
* @throws ExecutionException
|
||||
*/
|
||||
public static String getHostName(final InetAddress i, final long timeout) throws ExecutionException {
|
||||
return new TimeoutRequest<String>(new Callable<String>() {
|
||||
public String call() { return i.getHostName(); }
|
||||
}).call(timeout);
|
||||
}
|
||||
|
||||
/**
|
||||
* check if a smb file exists
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static boolean exists(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() { try {
|
||||
return file.exists();
|
||||
} catch (SmbException e) {
|
||||
return Boolean.FALSE;
|
||||
} }
|
||||
}).call(timeout).booleanValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* check if a smb file can be read
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static boolean canRead(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() { try {
|
||||
return file.canRead();
|
||||
} catch (SmbException e) {
|
||||
return Boolean.FALSE;
|
||||
} }
|
||||
}).call(timeout).booleanValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* check if a smb file ran be written
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static boolean canWrite(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() { try {
|
||||
return file.canWrite();
|
||||
} catch (SmbException e) {
|
||||
return Boolean.FALSE;
|
||||
} }
|
||||
}).call(timeout).booleanValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* check if a smb file is hidden
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static boolean isHidden(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() { try {
|
||||
return file.isHidden();
|
||||
} catch (SmbException e) {
|
||||
return Boolean.FALSE;
|
||||
} }
|
||||
}).call(timeout).booleanValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* check if a smb file is a directory
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static boolean isDirectory(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Boolean>(new Callable<Boolean>() {
|
||||
public Boolean call() { try {
|
||||
return file.isDirectory();
|
||||
} catch (SmbException e) {
|
||||
return Boolean.FALSE;
|
||||
} }
|
||||
}).call(timeout).booleanValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* get the size of a smb file
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static long length(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Long>(new Callable<Long>() {
|
||||
public Long call() { try {
|
||||
return file.length();
|
||||
} catch (SmbException e) {
|
||||
return Long.valueOf(0);
|
||||
} }
|
||||
}).call(timeout).longValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* get last-modified time of a smb file
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static long lastModified(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<Long>(new Callable<Long>() {
|
||||
public Long call() { try {
|
||||
return file.lastModified();
|
||||
} catch (SmbException e) {
|
||||
return Long.valueOf(0);
|
||||
} }
|
||||
}).call(timeout).longValue();
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* get list of a smb directory
|
||||
* @param file
|
||||
* @param timeout
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
public static String[] list(final SmbFile file, final long timeout) throws IOException {
|
||||
try {
|
||||
return new TimeoutRequest<String[]>(new Callable<String[]>() {
|
||||
public String[] call() { try {
|
||||
return file.list();
|
||||
} catch (SmbException e) {
|
||||
return null;
|
||||
} }
|
||||
}).call(timeout);
|
||||
} catch (ExecutionException e) {
|
||||
throw new IOException(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public static void main(String[] args) {
|
||||
try {
|
||||
System.out.println(getByName("yacy.net", 100));
|
||||
} catch (ExecutionException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue