2 * XdccDownloader - DccReceiver.java - Copyright © 2013 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.irc;
20 import java.io.IOException;
21 import java.io.InputStream;
22 import java.io.OutputStream;
23 import java.net.InetAddress;
24 import java.net.Socket;
25 import java.util.concurrent.TimeUnit;
26 import java.util.logging.Level;
27 import java.util.logging.Logger;
29 import net.pterodactylus.irc.event.DccSendReceived;
30 import net.pterodactylus.xdcc.util.io.BandwidthCountingInputStream;
32 import com.google.common.io.Closeables;
33 import com.google.common.util.concurrent.AbstractExecutionThreadService;
36 * Service that receives a file offered by a {@link DccSendReceived}.
38 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
40 public class DccReceiver extends AbstractExecutionThreadService {
43 private static final Logger logger = Logger.getLogger(DccReceiver.class.getName());
45 /** The address to connect to. */
46 private final InetAddress inetAddress;
48 /** The port number to connect to. */
49 private final int port;
51 /** The name of the file being offered. */
52 private final String filename;
54 /** The size of the file being offered. */
55 private final long size;
57 /** The output stream to write the file to. */
58 private final OutputStream outputStream;
60 /** The number of bytes already written. */
61 private long progress;
63 /** The bandwidth-measuring input stream. */
64 private BandwidthCountingInputStream inputStream;
67 * Creates a new DCC receiver.
70 * The address to connect to
72 * The port number to connect to
74 * The name of the file being downloaded
76 * The size of the file being downloaded, or {@code -1} if the size is not
79 * The output stream to write the file to
81 public DccReceiver(InetAddress inetAddress, int port, String filename, long size, OutputStream outputStream) {
82 this.inetAddress = inetAddress;
84 this.filename = filename;
86 this.outputStream = outputStream;
94 * Returns the name of the file being downloaded. The name is not used by the
95 * DCC receiver, it only serves as a kind of identifier.
97 * @return The name of the file being downloaded
99 public String filename() {
104 * Returns the size of the file being downloaded. If the size of the file is
105 * not known, {@code -1} is returned.
107 * @return The size of the file being downloaded, or {@code -1} if the size is
115 * Returns the number of bytes that have already been downloaded.
117 * @return The number of bytes that have already been downloaded
119 public long progress() {
124 * Returns the current rate of the download.
126 * @return The current rate of the download, in bytes/second
128 public long currentRate() {
129 return inputStream.getCurrentRate();
133 * Returns the overall rate of the download.
135 * @return The overall rate of the download, in bytes/second
137 public long overallRate() {
138 return inputStream.getOverallRate();
142 // ABSTRACTEXECUTIONTHREADSERVICE METHODS
146 protected void run() throws IOException {
147 Socket socket = null;
149 socket = new Socket(inetAddress, port);
150 InputStream socketInputStream = socket.getInputStream();
151 inputStream = new BandwidthCountingInputStream(socketInputStream, 5, TimeUnit.SECONDS);
152 byte[] buffer = new byte[65536];
153 while (isRunning() && ((size == -1) || (progress < size))) {
154 int r = inputStream.read(buffer);
159 outputStream.write(buffer, 0, r);
162 outputStream.flush();
163 } catch (IOException ioe1) {
164 logger.log(Level.WARNING, "I/O error while receiving DCC!", ioe1);
166 Closeables.close(inputStream, true);