Java：如何在"regular" TCP Socket和SSLSocket之间进行抽象

Question

编辑：删除startHandshake();因为它与问题无关且很少需要(例如，在我的情况下不是(

我有一个相当具体和罕见的客户端 - 服务器协议(通过TCP(。
我已经使用SSLSocket实现了它。
现在，我预见到我可能需要在未加密的连接上使用相同的协议。

我的问题是实现协议的类有一个字段：public SSLSocket currentSocket;
(然后我的客户端类中的方法执行各种.read(), .write(), flush()...(

我想过更改字段类型，如下所示：public Socket currentSocket;
但是，问题是我的连接过程不兼容：

public static void connect () {
currentSocket = SslUtils.getSSLsocket(host, port, keystoreFile, keystorePass, pkPass);
...

• java.net.Socket的默认构造函数显然不接受密钥库的东西

我不想仅仅为了这种差异而重新实现我的整个客户端......

• 我的一个想法是，当我需要一个明文套接字时，创建一个没有加密的SSLSocket。
  我不知道这是否是一种专业的方法，或者它是否会起作用(服务器在新用例中会期望一个明文客户端套接字(

• 我的另一个想法是定义两个字段，一个用于纯文本套接字，一个用于 SSL 套接字，然后根据需要使用逻辑将输入/输出流链接到正确的字段。但是，这将导致"挂起"字段。如果您使用 SSL，则会Socket plaintextSocket一个冗余字段，反之亦然...

有没有办法让我的currentSocket字段更抽象，以便我可以在同一客户端中定义它，然后根据已知变量(如needSSLsocket=true(指示略有不同的客户端代码路径进行实例化和连接？

Answer 1

SSLSocket扩展Socket，因此您可以将SSLSocket对象分配给Socket变量。 您将currentSocket字段更改为Socket是正确的。 只需在需要时使用另一个变量来处理SSLSocket，例如：

public static void connect () {
if (needSSLsocket) {
SSLSocket ssl = SslUtils.getSSLsocket(host, port, keystoreFile, keystorePass, pkPass);
ssl.startHandshake();
...
currentSocket = ssl;
/* or:
currentSocket = SslUtils.getSSLsocket(host, port, keystoreFile, keystorePass, pkPass);
((SSLSocket) currentSocket).startHandshake();
...
*/
} else {
currentSocket = new Socket(host, port);
}
...
}