Examples with EzyMessage com.tvd12.ezyfox.codec.EzyMessage used on opensource projects

Example 11 with EzyMessage

use of com.tvd12.ezyfox.codec.EzyMessage in project ezyfox-server by youngmonkeys.

the class EzySimpleNioUdpDataHandlerTest method handleUdpHandshakeContentLengthLessThan11.

@Test
public void handleUdpHandshakeContentLengthLessThan11() {
    // given
    EzySimpleNioUdpDataHandler sut = new EzySimpleNioUdpDataHandler(1);
    DatagramChannel channel = mock(DatagramChannel.class);
    InetSocketAddress address = new InetSocketAddress(3005);
    EzyMessage message = mock(EzyMessage.class);
    when(message.getContent()).thenReturn(new byte[0]);
    // when
    MethodInvoker.create().object(sut).method("handleUdpHandshake").param(DatagramChannel.class, channel).param(InetSocketAddress.class, address).param(EzyMessage.class, message).call();
    // then
    verify(message, times(1)).getContent();
    sut.destroy();
}

Example 12 with EzyMessage

use of com.tvd12.ezyfox.codec.EzyMessage in project ezyfox-server by youngmonkeys.

the class EzySimpleNioUdpDataHandlerTest method handleUdpHandshakeSessionIsNull.

@SuppressWarnings("rawtypes")
@Test
public void handleUdpHandshakeSessionIsNull() {
    // given
    EzySimpleNioUdpDataHandler sut = new EzySimpleNioUdpDataHandler(1);
    DatagramChannel channel = mock(DatagramChannel.class);
    InetSocketAddress address = new InetSocketAddress(3005);
    EzyMessage message = mock(EzyMessage.class);
    String sessionToken = RandomUtil.randomAlphabetString(8);
    long sessionId = 12345L;
    int tokenSize = sessionToken.length();
    int messageSize = 0;
    // sessionIdSize
    messageSize += 8;
    // tokenLengthSize
    messageSize += 2;
    // messageSize
    messageSize += tokenSize;
    ByteBuffer buffer = ByteBuffer.allocate(messageSize);
    buffer.putLong(sessionId);
    buffer.putShort((short) tokenSize);
    buffer.put(sessionToken.getBytes());
    buffer.flip();
    byte[] bytes = EzyByteBuffers.getBytes(buffer);
    when(message.getContent()).thenReturn(bytes);
    EzySessionManager sessionManager = mock(EzySessionManager.class);
    sut.setSessionManager(sessionManager);
    // when
    MethodInvoker.create().object(sut).method("handleUdpHandshake").param(DatagramChannel.class, channel).param(InetSocketAddress.class, address).param(EzyMessage.class, message).call();
    // then
    verify(message, times(1)).getContent();
    verify(sessionManager, times(1)).getSession(sessionId);
    sut.destroy();
}

Example 13 with EzyMessage

use of com.tvd12.ezyfox.codec.EzyMessage in project ezyfox-server by youngmonkeys.

the class EzySimpleNioUdpDataHandler method handleUdpHandshake.

protected void handleUdpHandshake(DatagramChannel channel, InetSocketAddress address, EzyMessage message) throws Exception {
    byte[] content = message.getContent();
    if (content.length < 11) {
        return;
    }
    byte[] sessionIdBytes = EzyBytes.copy(content, 0, 8);
    long sessionId = EzyLongs.bin2long(sessionIdBytes);
    byte[] tokenSizeBytes = EzyBytes.copy(content, 8, 2);
    int tokenSize = EzyInts.bin2int(tokenSizeBytes);
    if (tokenSize > 512) {
        return;
    }
    byte[] tokenBytes = EzyBytes.copy(content, 10, tokenSize);
    EzySession session = sessionManager.getSession(sessionId);
    if (session == null) {
        return;
    }
    int responseCode = 498;
    String token = new String(tokenBytes);
    String sessionToken = session.getToken();
    if (sessionToken.equals(token)) {
        responseCode = 200;
        SocketAddress oldUdpAddress = session.getUdpClientAddress();
        handlerGroupManager.unmapHandlerGroup(oldUdpAddress);
        handlerGroupManager.mapSocketChannel(address, session);
        ((EzyDatagramChannelAware) session).setDatagramChannel(channel);
        ((EzyUdpClientAddressAware) session).setUdpClientAddress(address);
        ((EzyDatagramChannelPoolAware) session).setDatagramChannelPool(datagramChannelPool);
    }
    response(session, responseCode);
}