X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Firc%2FConnection.java;h=15c8ff50793487eeb6bec3b283a81bbfe1acee05;hb=5cc5fd2f2cc010e77448ca044c7679739add966f;hp=d12736375f7ac4d79023fa1cc4b8773179ad780b;hpb=207761aba7e6ddf5e736058f15e844a4df0d1f55;p=xudocci.git

diff --git a/src/main/java/net/pterodactylus/irc/Connection.java b/src/main/java/net/pterodactylus/irc/Connection.java
index d127363..15c8ff5 100644
--- a/src/main/java/net/pterodactylus/irc/Connection.java
+++ b/src/main/java/net/pterodactylus/irc/Connection.java
@@ -27,7 +27,9 @@ import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
+import java.net.InetAddress;
 import java.net.Socket;
+import java.net.UnknownHostException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
@@ -37,6 +39,7 @@ import java.util.logging.Logger;
 import javax.net.SocketFactory;
 
 import net.pterodactylus.irc.event.ChannelJoined;
+import net.pterodactylus.irc.event.ChannelLeft;
 import net.pterodactylus.irc.event.ChannelMessageReceived;
 import net.pterodactylus.irc.event.ChannelNicknames;
 import net.pterodactylus.irc.event.ChannelNotJoined;
@@ -44,6 +47,7 @@ import net.pterodactylus.irc.event.ChannelNotJoined.Reason;
 import net.pterodactylus.irc.event.ChannelTopic;
 import net.pterodactylus.irc.event.ConnectionEstablished;
 import net.pterodactylus.irc.event.ConnectionFailed;
+import net.pterodactylus.irc.event.DccSendReceived;
 import net.pterodactylus.irc.event.MotdReceived;
 import net.pterodactylus.irc.event.NicknameInUseReceived;
 import net.pterodactylus.irc.event.NoNicknameGivenReceived;
@@ -56,6 +60,8 @@ import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 import com.google.common.eventbus.EventBus;
 import com.google.common.io.Closeables;
+import com.google.common.primitives.Ints;
+import com.google.common.primitives.Longs;
 import com.google.common.util.concurrent.AbstractExecutionThreadService;
 import com.google.common.util.concurrent.Service;
 
@@ -196,6 +202,19 @@ public class Connection extends AbstractExecutionThreadService implements Servic
 	//
 
 	/**
+	 * Checks whether the given source is the client represented by this
+	 * connection.
+	 *
+	 * @param source
+	 * 		The source to check
+	 * @return {@code true} if this connection represents the given source, {@code
+	 *         false} otherwise
+	 */
+	public boolean isSource(Source source) {
+		return source.nick().isPresent() && source.nick().get().equals(nickname);
+	}
+
+	/**
 	 * Joins the given channel.
 	 *
 	 * @param channel
@@ -207,6 +226,21 @@ public class Connection extends AbstractExecutionThreadService implements Servic
 		connectionHandler.sendCommand("JOIN", channel);
 	}
 
+	/**
+	 * Sends a message to the given recipient, which may be a channel or another
+	 * nickname.
+	 *
+	 * @param recipient
+	 * 		The recipient of the message
+	 * @param message
+	 * 		The message
+	 * @throws IOException
+	 * 		if an I/O error occurs
+	 */
+	public void sendMessage(String recipient, String message) throws IOException {
+		connectionHandler.sendCommand("PRIVMSG", recipient, message);
+	}
+
 	//
 	// ABSTRACTEXECUTIONTHREADSERVICE METHODS
 	//
@@ -261,7 +295,23 @@ public class Connection extends AbstractExecutionThreadService implements Servic
 				if (command.equalsIgnoreCase("PRIVMSG")) {
 					String recipient = parameters.get(0);
 					String message = parameters.get(1);
-					if (!channelTypes.contains(recipient.charAt(0))) {
+					if (message.startsWith("\u0001") && message.endsWith("\u0001")) {
+						/* CTCP! */
+						String[] messageWords = message.substring(1, message.length() - 1).split(" +");
+						String ctcpCommand = messageWords[0];
+						if (ctcpCommand.equalsIgnoreCase("DCC")) {
+							if (messageWords[1].equalsIgnoreCase("SEND")) {
+								Optional<InetAddress> inetAddress = parseInetAddress(messageWords[3]);
+								Optional<Integer> port = Optional.fromNullable(Ints.tryParse(messageWords[4]));
+								long fileSize = Optional.fromNullable(Longs.tryParse(messageWords[5])).or(-1L);
+								if (inetAddress.isPresent() && port.isPresent()) {
+									eventBus.post(new DccSendReceived(this, reply.source().get(), messageWords[2], inetAddress.get(), port.get(), fileSize));
+								} else {
+									logger.warning(String.format("Received malformed DCC SEND: “%s”", message));
+								}
+							}
+						}
+					} else if (!channelTypes.contains(recipient.charAt(0))) {
 						eventBus.post(new PrivateMessageReceived(this, reply.source().get(), message));
 					} else {
 						eventBus.post(new ChannelMessageReceived(this, recipient, reply.source().get(), message));
@@ -339,6 +389,8 @@ public class Connection extends AbstractExecutionThreadService implements Servic
 				} else if (command.equals("366")) {
 					eventBus.post(new ChannelNicknames(this, parameters.get(1), nicks));
 					nicks.clear();
+				} else if (command.equalsIgnoreCase("PART")) {
+					eventBus.post(new ChannelLeft(this, parameters.get(0), reply.source().get(), parameters.get(1)));
 
 				/* common channel join errors. */
 				} else if (command.equals("474")) {
@@ -400,6 +452,32 @@ public class Connection extends AbstractExecutionThreadService implements Servic
 		return Optional.absent();
 	}
 
+	/**
+	 * Parses the given {@code ip} and returns an {@link InetAddress} from it.
+	 *
+	 * @param ip
+	 * 		The IP to parse
+	 * @return The parsed inet address, or {@link Optional#absent()} if no inet
+	 *         address could be parsed
+	 */
+	private Optional<InetAddress> parseInetAddress(String ip) {
+		Long ipNumber = Longs.tryParse(ip);
+		if (ipNumber == null) {
+			return Optional.absent();
+		}
+
+		StringBuilder hostname = new StringBuilder(15);
+		hostname.append((ipNumber >>> 24) & 0xff).append('.');
+		hostname.append((ipNumber >>> 16) & 0xff).append('.');
+		hostname.append((ipNumber >>> 8) & 0xff).append('.');
+		hostname.append(ipNumber & 0xff);
+		try {
+			return Optional.of(InetAddress.getByName(hostname.toString()));
+		} catch (UnknownHostException uhe1) {
+			return Optional.absent();
+		}
+	}
+
 	/** Handles input and output for the connection. */
 	private class ConnectionHandler implements Closeable {