import java.net.InetAddress;
import java.net.Socket;
import java.util.concurrent.TimeUnit;
-import java.util.logging.Level;
-import java.util.logging.Logger;
import net.pterodactylus.irc.event.DccDownloadFailed;
import net.pterodactylus.irc.event.DccDownloadFinished;
import com.google.common.eventbus.EventBus;
import com.google.common.io.Closeables;
import com.google.common.util.concurrent.AbstractExecutionThreadService;
+import org.apache.log4j.Logger;
/**
* Service that receives a file offered by a {@link DccSendReceived}.
* The output stream to write the file to
*/
public DccReceiver(EventBus eventBus, InetAddress inetAddress, int port, String filename, long size, OutputStream outputStream) {
+ this(eventBus, inetAddress, port, filename, 0, size, outputStream);
+ }
+
+ /**
+ * Creates a new DCC receiver.
+ *
+ * @param inetAddress
+ * The address to connect to
+ * @param port
+ * The port number to connect to
+ * @param filename
+ * The name of the file being downloaded
+ * @param startOffset
+ * The offset at which the download starts in case of a resume
+ * @param size
+ * The size of the file being downloaded, or {@code -1} if the size is not
+ * known
+ * @param outputStream
+ * The output stream to write the file to
+ */
+ public DccReceiver(EventBus eventBus, InetAddress inetAddress, int port, String filename, long startOffset, long size, OutputStream outputStream) {
this.eventBus = eventBus;
this.inetAddress = inetAddress;
this.port = port;
this.filename = filename;
+ this.progress = startOffset;
this.size = size;
this.outputStream = outputStream;
}
* @return The current rate of the download, in bytes/second
*/
public long currentRate() {
- return inputStream.getCurrentRate();
+ return (inputStream != null) ? inputStream.getCurrentRate() : 0;
}
/**
* @return The overall rate of the download, in bytes/second
*/
public long overallRate() {
- return inputStream.getOverallRate();
+ return (inputStream != null) ? inputStream.getOverallRate() : 0;
}
//
Socket socket = null;
try {
socket = new Socket(inetAddress, port);
+ socket.setSoTimeout((int) TimeUnit.MINUTES.toMillis(3));
InputStream socketInputStream = socket.getInputStream();
inputStream = new BandwidthCountingInputStream(socketInputStream, 5, TimeUnit.SECONDS);
byte[] buffer = new byte[65536];
progress += r;
}
outputStream.flush();
- eventBus.post(new DccDownloadFinished(this));
+ if ((size == -1) || (progress == size)) {
+ eventBus.post(new DccDownloadFinished(this));
+ } else {
+ eventBus.post(new DccDownloadFailed(this, new IOException("Download aborted.")));
+ }
} catch (IOException ioe1) {
- logger.log(Level.WARNING, "I/O error while receiving DCC!", ioe1);
+ logger.warn("I/O error while receiving DCC!", ioe1);
eventBus.post(new DccDownloadFailed(this, ioe1));
} finally {
Closeables.close(inputStream, true);