Cleaned up client encryption logic.

This commit is contained in:
Andrew Lalis 2021-09-11 23:43:15 +02:00
parent ec3bfbbc09
commit 5568136d86
2 changed files with 12 additions and 23 deletions

View File

@ -34,8 +34,8 @@ import java.util.Map;
public class ConcordClient implements Runnable {
private final Socket socket;
private InputStream in;
private OutputStream out;
private final InputStream in;
private final OutputStream out;
private final Serializer serializer;
@Getter
@ -48,9 +48,14 @@ public class ConcordClient implements Runnable {
public ConcordClient(String host, int port, String nickname, Path tokensFile) throws IOException {
this.eventManager = new EventManager(this);
this.socket = new Socket(host, port);
this.in = this.socket.getInputStream();
this.out = this.socket.getOutputStream();
this.serializer = new Serializer();
try {
var streams = Encryption.upgrade(socket.getInputStream(), socket.getOutputStream(), this.serializer);
this.in = streams.first();
this.out = streams.second();
} catch (GeneralSecurityException e) {
throw new IOException("Could not establish secure connection to the server.", e);
}
this.model = this.initializeConnectionToServer(nickname, tokensFile);
// Add event listeners.
@ -75,7 +80,6 @@ public class ConcordClient implements Runnable {
* messages, or if the server sends an unexpected response.
*/
private ClientModel initializeConnectionToServer(String nickname, Path tokensFile) throws IOException {
this.establishEncryption();
String token = this.getSessionToken(tokensFile);
this.serializer.writeMessage(new Identification(nickname, token), this.out);
Message reply = this.serializer.readMessage(this.in);
@ -90,24 +94,6 @@ public class ConcordClient implements Runnable {
}
}
/**
* Establishes an encrypted connection to the server. This should be the
* first method which interacts with the server, since it sends and receives
* specific key information, and all subsequent traffic should be encrypted.
* @throws IOException If encryption could not be established.
*/
private void establishEncryption() throws IOException {
try {
System.out.println("Initializing end-to-end encryption with the server...");
var streams = Encryption.upgrade(this.in, this.out, this.serializer);
this.in = streams.first();
this.out = streams.second();
System.out.println("Successfully established cipher streams.");
} catch (GeneralSecurityException e) {
throw new IOException(e);
}
}
public void sendMessage(Message message) throws IOException {
this.serializer.writeMessage(message, this.out);
}

View File

@ -19,6 +19,9 @@ import java.util.Arrays;
import java.util.Collections;
import java.util.List;
/**
* Utility class for handling the establishment of encrypted communication.
*/
public class Encryption {
public static Pair<CipherInputStream, CipherOutputStream> upgrade(
InputStream in,