MINA、Netty、Twisted一起學（十一）：SSL/TLS

本文轉載自查看原文 2017-02-27 16:08 1333 TCP/ 安全/ NIO/ Netty/ TLS/ MINA/ 網絡編程/ 加密/ SSL/ Twisted

什么是SSL/TLS

不使用SSL/TLS的網絡通信，一般都是明文傳輸，網絡傳輸內容在傳輸過程中很容易被竊聽甚至篡改，非常不安全。SSL/TLS協議就是為了解決這些安全問題而設計的。SSL/TLS協議位於TCP/IP協議之上，各個應用層協議之下，使網絡傳輸的內容通過加密算法加密，並且只有服務器和客戶端可以加密解密，中間人即使抓到數據包也無法解密獲取傳輸的內容，從而避免安全問題。例如廣泛使用的HTTPS協議即是在TCP協議和HTTP協議之間加了一層SSL/TLS協議。

SSL/TLS流程

使用SSL/TLS協議的服務器和客戶端開始通信之前，會先進行一個握手階段：

客戶端發出請求：這一步客戶端會生成一個隨機數傳給服務器；
服務器回應：這一步服務器會返回給客戶端一個服務器數字證書（證書中包含用於加密的公鑰），另外服務器也會生成一個隨機數給客戶端；
客戶端回應：這一步客戶端首先會校驗數字證書的合法性，然后會再生成一個隨機數，這個隨機數會使用第2步中的公鑰采用非對稱加密算法（例如RSA算法）進行加密后傳給服務器，密文只能通過服務器的私鑰來解密。
服務器最后回應：握手結束。

握手結束后，客戶端和服務器都有上面握手階段的三個隨機數。客戶端和服務器都通過這三個隨機生成一個密鑰，接下來所有的通信內容都使用這個密鑰通過對稱加密算法加密傳輸，服務器和客戶端才開始進行安全的通信。

如果看到這里還是一臉懵逼，可以參考SSL/TLS協議運行機制的概述更深入地了解SSL/TLS流程，本文不再過多介紹。

生成私鑰和證書

使用openssl來生成私鑰和證書：

openssl req -x509 -newkey rsa:2048 -nodes -days 365 -keyout private.pem -out cert.crt

運行以上命令后，會在當前目錄下生成一個私鑰文件（private.pem）和一個證書文件（cert.crt）。

生成的私鑰和證書Twisted、Netty可以直接使用，然而MINA對私鑰文件的格式的要求，要將pem格式轉換成der格式，實際上就是將文本文件私鑰轉成二進制文件私鑰。openssl將private.pem轉成private.der私鑰文件：

openssl pkcs8 -topk8 -inform PEM -in private.pem -outform DER -nocrypt -out private.der

SSL/TLS服務器

接下來在http://xxgblog.com/2014/08/21/mina-netty-twisted-2/一文的基礎上，加上SSL/TLS層。

MINA

MINA可以通過SslFilter來實現SSL/TLS，初始化SslFilter的代碼比較繁瑣：

public class MinaServer {

    public static void main(String[] args) throws Exception {


        String certPath = "/Users/wucao/Desktop/ssl/cert.crt";  // 證書
        String privateKeyPath = "/Users/wucao/Desktop/ssl/private.der";  // 私鑰

        // 證書
        // https://docs.oracle.com/javase/7/docs/api/java/security/cert/X509Certificate.html
        InputStream inStream = null;
        Certificate certificate = null;
        try {
            inStream = new FileInputStream(certPath);
            CertificateFactory cf = CertificateFactory.getInstance("X.509");
            certificate = cf.generateCertificate(inStream);
        } finally {
            if (inStream != null) {
                inStream.close();
            }
        }

        // 私鑰
        PKCS8EncodedKeySpec keySpec = new PKCS8EncodedKeySpec(Files.readAllBytes(new File(privateKeyPath).toPath()));
        PrivateKey privateKey = KeyFactory.getInstance("RSA").generatePrivate(keySpec);

        KeyStore ks = KeyStore.getInstance(KeyStore.getDefaultType());
        ks.load(null, null);
        Certificate[] certificates = {certificate};
        ks.setKeyEntry("key", privateKey, "".toCharArray(), certificates);

        KeyManagerFactory kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
        kmf.init(ks, "".toCharArray());

        SSLContext sslContext = SSLContext.getInstance("TLS");
        sslContext.init(kmf.getKeyManagers(), null, null);

        IoAcceptor acceptor = new NioSocketAcceptor();
        DefaultIoFilterChainBuilder chain = acceptor.getFilterChain();
        chain.addLast("ssl", new SslFilter(sslContext));  // SslFilter需要放在最前面
        chain.addLast("codec", new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"), "\r\n", "\r\n")));
        acceptor.setHandler(new TcpServerHandle());
        acceptor.bind(new InetSocketAddress(8080));
    }
}


class TcpServerHandle extends IoHandlerAdapter {

    @Override
    public void exceptionCaught(IoSession session, Throwable cause)
            throws Exception {
        cause.printStackTrace();
    }

    @Override
    public void messageReceived(IoSession session, Object message)
            throws Exception {
        String line = (String) message;
        System.out.println("messageReceived:" + line);
    }

    @Override
    public void sessionCreated(IoSession session) throws Exception {
        System.out.println("sessionCreated");
    }

    @Override
    public void sessionClosed(IoSession session) throws Exception {
        System.out.println("sessionClosed");
    }
}

Netty

Netty通過添加一個SslHandler來實現SSL/TLS，相對MINA來說代碼就比較簡潔：

public class NettyServer {

    public static void main(String[] args) throws InterruptedException, SSLException {

        File certificate = new File("/Users/wucao/Desktop/ssl/cert.crt");  // 證書
        File privateKey = new File("/Users/wucao/Desktop/ssl/private.pem");  // 私鑰
        final SslContext sslContext = SslContextBuilder.forServer(certificate, privateKey).build();

        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        public void initChannel(SocketChannel ch)
                                throws Exception {
                            ChannelPipeline pipeline = ch.pipeline();

                            // SslHandler要放在最前面
                            SslHandler sslHandler = sslContext.newHandler(ch.alloc());
                            pipeline.addLast(sslHandler);

                            pipeline.addLast(new LineBasedFrameDecoder(80));
                            pipeline.addLast(new StringDecoder(CharsetUtil.UTF_8));

                            pipeline.addLast(new TcpServerHandler());
                        }
                    });
            ChannelFuture f = b.bind(8080).sync();
            f.channel().closeFuture().sync();
        } finally {
            workerGroup.shutdownGracefully();
            bossGroup.shutdownGracefully();
        }
    }

}

class TcpServerHandler extends ChannelInboundHandlerAdapter {

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) {
        String line = (String) msg;
        System.out.println("channelRead:" + line);
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) {
        System.out.println("channelActive");
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) {
        System.out.println("channelInactive");
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        ctx.close();
    }
}

Twisted

Twisted實現SSL/TLS也是非常簡單的，將reactor.listenTCP替換為reactor.listenSSL即可

# -*- coding:utf-8 –*-

from twisted.protocols.basic import LineOnlyReceiver
from twisted.internet.protocol import Factory
from twisted.internet import reactor, ssl

sslContext = ssl.DefaultOpenSSLContextFactory(
    '/Users/wucao/Desktop/ssl/private.pem',  # 私鑰
    '/Users/wucao/Desktop/ssl/cert.crt',  # 公鑰
)

class TcpServerHandle(LineOnlyReceiver):

    def connectionMade(self):
        print 'connectionMade'

    def connectionLost(self, reason):
        print 'connectionLost'

    def lineReceived(self, data):
        print 'lineReceived:' + data

factory = Factory()
factory.protocol = TcpServerHandle
reactor.listenSSL(8080, factory, sslContext)
reactor.run()

SSL/TLS客戶端

這里還是使用Java來寫一個SSL/TLS客戶端，用來測試以上三個服務器程序。需要注意的是，在上面SSL/TLS流程的介紹中，SSL/TLS握手階段的第2步服務器會將證書傳給客戶端，第3步客戶端會校驗證書的合法性，所以下面的代碼首先會讓客戶端信任openssl生成的證書，才能正確的完成SSL/TLS握手。

public class SSLClient {

    public static void main(String args[]) throws Exception {

        // 客戶端信任改證書，將用於校驗服務器傳過來的證書的合法性
        String certPath = "/Users/wucao/Desktop/ssl/cert.crt";
        InputStream inStream = null;
        Certificate certificate = null;
        try {
            inStream = new FileInputStream(certPath);
            CertificateFactory cf = CertificateFactory.getInstance("X.509");
            certificate = cf.generateCertificate(inStream);
        } finally {
            if (inStream != null) {
                inStream.close();
            }
        }

        KeyStore ks = KeyStore.getInstance(KeyStore.getDefaultType());
        ks.load(null, null);
        ks.setCertificateEntry("cert", certificate);

        TrustManagerFactory tmf = TrustManagerFactory.getInstance("sunx509");
        tmf.init(ks);

        SSLContext sslContext = SSLContext.getInstance("TLS");
        sslContext.init(null, tmf.getTrustManagers(), null);

        SSLSocketFactory socketFactory = sslContext.getSocketFactory();

        Socket socket = null;
        OutputStream out = null;

        try {

            socket = socketFactory.createSocket("localhost", 8080);
            out = socket.getOutputStream();

            // 請求服務器
            String lines = "床前明月光\r\n疑是地上霜\r\n舉頭望明月\r\n低頭思故鄉\r\n";
            byte[] outputBytes = lines.getBytes("UTF-8");
            out.write(outputBytes);
            out.flush();

        } finally {
            // 關閉連接
            out.close();
            socket.close();
        }

    }
}