EpollSocketTcpMd5Test Class — netty Architecture

Source Code

transport-native-epoll/src/test/java/io/netty/channel/epoll/EpollSocketTcpMd5Test.java lines 42–152

public class EpollSocketTcpMd5Test {
    private static final byte[] SERVER_KEY = "abc".getBytes(CharsetUtil.US_ASCII);
    private static final byte[] BAD_KEY = "def".getBytes(CharsetUtil.US_ASCII);
    private static EventLoopGroup GROUP;
    private EpollServerSocketChannel server;

    @BeforeAll
    public static void beforeClass() {
        GROUP = new MultiThreadIoEventLoopGroup(1, EpollIoHandler.newFactory());
    }

    @AfterAll
    public static void afterClass() {
        GROUP.shutdownGracefully();
    }

    @BeforeEach
    public void setup() {
        ServerBootstrap bootstrap = new ServerBootstrap();
        server = (EpollServerSocketChannel) bootstrap.group(GROUP)
                .channel(EpollServerSocketChannel.class)
                .childHandler(new ChannelInboundHandlerAdapter())
                .bind(new InetSocketAddress(NetUtil.LOCALHOST4, 0)).syncUninterruptibly().channel();
    }

    @AfterEach
    public void teardown() {
        server.close().syncUninterruptibly();
    }

    @Test
    public void testServerSocketChannelOption() throws Exception {
        try {
            server.config().setOption(EpollChannelOption.TCP_MD5SIG,
                    Collections.<InetAddress, byte[]>singletonMap(NetUtil.LOCALHOST4, SERVER_KEY));
            server.config().setOption(EpollChannelOption.TCP_MD5SIG, Collections.<InetAddress, byte[]>emptyMap());
        } catch (ChannelException ce) {
            handleException(ce);
        }
    }

    private static void handleException(ChannelException ce) {
        if (ce.getMessage().contains("Protocol not available")) {
            // Some operating systems don't allow or support the TCP_MD5SIG option.
            // Abort the test (instead of failing) if that's the case.
            Assumptions.abort(ce.getMessage());
        }
        throw ce;
    }

    @Test
    public void testServerOption() throws Exception {
        ServerBootstrap bootstrap = new ServerBootstrap();
        EpollServerSocketChannel ch = (EpollServerSocketChannel) bootstrap.group(GROUP)
                .channel(EpollServerSocketChannel.class)
                .childHandler(new ChannelInboundHandlerAdapter())
                .bind(new InetSocketAddress(0)).syncUninterruptibly().channel();

        try {
            ch.config().setOption(EpollChannelOption.TCP_MD5SIG,
                    Collections.<InetAddress, byte[]>singletonMap(NetUtil.LOCALHOST4, SERVER_KEY));
            ch.config().setOption(EpollChannelOption.TCP_MD5SIG, Collections.<InetAddress, byte[]>emptyMap());
        } catch (ChannelException ce) {
            handleException(ce);
        } finally {
            ch.close().syncUninterruptibly();
        }
    }

    @Test
    public void testKeyMismatch() throws Exception {
        try {
            server.config().setOption(EpollChannelOption.TCP_MD5SIG,
                    Collections.<InetAddress, byte[]>singletonMap(NetUtil.LOCALHOST4, SERVER_KEY));
        } catch (ChannelException ce) {
            handleException(ce);
        }

        assertThrows(ConnectTimeoutException.class, new Executable() {
            @Override
            public void execute() throws Throwable {