- 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