Everything seems to work except switching bridges.

This commit is contained in:
md_5 2012-10-05 15:24:58 +10:00
parent 093f45463a
commit 68e1445b1e
7 changed files with 43 additions and 41 deletions

View File

@ -172,24 +172,21 @@ public class Configuration {
}
}
public InetSocketAddress getHostFor(String user, String requestedHost) {
String entry = user + ";" + requestedHost;
String hostLine;
public String getServerFor(String user, String requestedHost) {
String server;
if (forcedServers.containsKey(requestedHost)) {
hostLine = servers.get(forcedServers.get(requestedHost));
server = servers.get(forcedServers.get(requestedHost));
} else {
hostLine = reconnectLocations.get(entry);
server = reconnectLocations.get(user);
}
if (hostLine == null) {
hostLine = servers.get(defaultServerName);
if (server == null) {
server = servers.get(defaultServerName);
}
return Util.getAddr(hostLine);
return server;
}
public void setHostFor(UserConnection user, String host) {
String entry = user.username + ";" + Util.getAddr(user.handshake.host);
reconnectLocations.put(entry, host);
public void setHostFor(UserConnection user, String server) {
reconnectLocations.put(user.username, server);
}
public InetSocketAddress getServer(String name) {
@ -198,7 +195,7 @@ public class Configuration {
if (server != null) {
return Util.getAddr(server);
} else {
return Util.getAddr(servers.get(defaultServerName));
return getServer(null);
}
}

View File

@ -57,7 +57,9 @@ public class EncryptionUtil {
return new PacketFDEncryptionRequest(hash, pubKey, verify);
}
public static SecretKey getSecret(PacketFCEncryptionResponse resp, PacketFDEncryptionRequest request) throws Exception {
public static SecretKey getSecret(PacketFCEncryptionResponse resp, PacketFDEncryptionRequest request) throws BadPaddingException, IllegalBlockSizeException,
IllegalStateException, InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException {
Cipher cipher = Cipher.getInstance("RSA");
cipher.init(Cipher.DECRYPT_MODE, keys.getPrivate());
byte[] decrypted = cipher.doFinal(resp.verifyToken);
@ -113,7 +115,7 @@ public class EncryptionUtil {
return hasher.doFinal(b);
}
public static byte[] getShared(SecretKey key, PublicKey pubkey) throws BadPaddingException, IllegalBlockSizeException, InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException {
public static byte[] getShared(SecretKey key, PublicKey pubkey) throws BadPaddingException, IllegalBlockSizeException, InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException {
Cipher cipher = Cipher.getInstance("RSA");
cipher.init(Cipher.ENCRYPT_MODE, pubkey);
return cipher.doFinal(key.getEncoded());

View File

@ -1,6 +1,5 @@
package net.md_5.bungee;
import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
@ -53,7 +52,7 @@ public class InitialHandler implements Runnable {
UserConnection userCon = new UserConnection(socket, in, out, handshake);
userCon.register();
userCon.connect(BungeeCord.instance.config.getHostFor(handshake.username, handshake.host));
userCon.connect(BungeeCord.instance.config.getServerFor(handshake.username, handshake.host));
break;
case 0xFE:
throw new KickException(BungeeCord.instance.config.motd + ChatColor.COLOR_CHAR + BungeeCord.instance.connections.size() + ChatColor.COLOR_CHAR + BungeeCord.instance.config.maxPlayers);

View File

@ -17,14 +17,16 @@ import org.bouncycastle.crypto.io.CipherOutputStream;
public class ServerConnection extends GenericConnection {
public final String name;
public final Packet1Login loginPacket;
public ServerConnection(Socket socket, PacketInputStream in, OutputStream out, Packet1Login loginPacket) {
public ServerConnection(String name, Socket socket, PacketInputStream in, OutputStream out, Packet1Login loginPacket) {
super(socket, in, out);
this.name = name;
this.loginPacket = loginPacket;
}
public static ServerConnection connect(InetSocketAddress address, Packet2Handshake handshake, boolean retry) {
public static ServerConnection connect(String name, InetSocketAddress address, Packet2Handshake handshake, boolean retry) {
try {
Socket socket = new Socket();
socket.connect(address, BungeeCord.instance.config.timeout);
@ -57,13 +59,13 @@ public class ServerConnection extends GenericConnection {
}
Packet1Login login = new Packet1Login(loginResponse);
return new ServerConnection(socket, in, out, login);
return new ServerConnection(name, socket, in, out, login);
} catch (KickException ex) {
throw ex;
} catch (Exception ex) {
InetSocketAddress def = BungeeCord.instance.config.getServer(null);
if (retry && address != def) {
return connect(def, handshake, false);
return connect(name, def, handshake, false);
} else {
throw new RuntimeException("Could not connect to target server");
}

View File

@ -37,19 +37,16 @@ public class UserConnection extends GenericConnection implements CommandSender {
public void connect(String server) {
InetSocketAddress addr = BungeeCord.instance.config.getServer(server);
if (addr == null) {
sendMessage(ChatColor.RED + "Specified server does not exist");
if (addr.equals(curServer())) {
sendMessage(ChatColor.RED + "You are already on this server");
return;
}
connect(addr);
connect(server, addr);
}
public void connect(InetSocketAddress serverAddr) {
private void connect(String name, InetSocketAddress serverAddr) {
try {
ServerConnection newServer = ServerConnection.connect(serverAddr, handshake, true);
if (downBridge != null) {
downBridge.interrupt();
}
ServerConnection newServer = ServerConnection.connect(name, serverAddr, handshake, server == null);
if (server == null) {
clientEntityId = newServer.loginPacket.entityId;
out.write(newServer.loginPacket.getPacket());
@ -59,10 +56,10 @@ public class UserConnection extends GenericConnection implements CommandSender {
server.disconnect("Quitting");
out.write(new Packet9Respawn((byte) 1, (byte) 0, (byte) 0, (short) 256, "DEFAULT").getPacket());
out.write(new Packet9Respawn((byte) -1, (byte) 0, (byte) 0, (short) 256, "DEFAULT").getPacket());
out.write(new Packet46GameState((byte) 2, (byte) 0).getPacket());
Packet1Login login = newServer.loginPacket;
serverEntityId = login.entityId;
out.write(new Packet9Respawn(login.dimension, login.difficulty, login.gameMode, (short) 256, login.levelType).getPacket());
out.write(new Packet46GameState((byte) 2, (byte) 0).getPacket());
if (heldItem != null) {
newServer.out.write(heldItem.getPacket());
}
@ -85,20 +82,21 @@ public class UserConnection extends GenericConnection implements CommandSender {
BungeeCord.instance.connections.put(username, this);
}
private InetSocketAddress curServer() {
return (server == null) ? null : new InetSocketAddress(server.socket.getInetAddress(), server.socket.getPort());
}
private void destory(String reason) {
if (BungeeCord.instance.isRunning) {
BungeeCord.instance.connections.remove(username);
}
if (upBridge != null) {
upBridge.interrupt();
}
if (downBridge != null) {
downBridge.interrupt();
}
disconnect(reason);
if (server != null) {
server.disconnect("Quitting");
}
if (server != null) {
BungeeCord.instance.config.setHostFor(this, server.name);
}
}
@Override
@ -114,7 +112,7 @@ public class UserConnection extends GenericConnection implements CommandSender {
@Override
public void run() {
while (!interrupted()) {
while (!socket.isClosed()) {
try {
byte[] packet = in.readPacket();
boolean sendPacket = true;
@ -151,7 +149,7 @@ public class UserConnection extends GenericConnection implements CommandSender {
@Override
public void run() {
while (!interrupted()) {
while (!server.socket.isClosed()) {
try {
byte[] packet = server.in.readPacket();
boolean sendPacket = true;

View File

@ -13,8 +13,8 @@ public class CommandServer extends Command {
return;
}
UserConnection con = (UserConnection) sender;
Collection<String> servers = BungeeCord.instance.config.servers.keySet();
if (args.length <= 0) {
Collection<String> servers = BungeeCord.instance.config.servers.keySet();
StringBuilder serverList = new StringBuilder();
for (String server : servers) {
serverList.append(server);
@ -24,7 +24,11 @@ public class CommandServer extends Command {
con.sendMessage(ChatColor.GOLD + "You may connect to the following servers at this time: " + serverList.toString());
} else {
String server = args[0];
con.connect(server);
if (!servers.contains(server)) {
con.sendMessage(ChatColor.RED + "The specified server does not exist");
} else {
con.connect(server);
}
}
}
}

View File

@ -14,6 +14,6 @@ public class PacketCDClientStatus extends DefinedPacket {
*/
public PacketCDClientStatus(byte payload) {
super(0xCD);
super.writeByte(payload);
writeByte(payload);
}
}