com.virjar.sekiro.netty.client.ClientIdleCheckHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sekiro-api Show documentation

ratel api,used for developer on ratel system,an extension for xposed framewrok,ratel api compatable with original xposed framework

There is a newer version: 1.0.5

Show newest version

package com.virjar.sekiro.netty.client;


import com.virjar.sekiro.Constants;
import com.virjar.sekiro.api.SekiroClient;
import com.virjar.sekiro.log.SekiroLogger;
import com.virjar.sekiro.netty.protocol.SekiroNatMessage;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.timeout.IdleStateEvent;
import io.netty.handler.timeout.IdleStateHandler;

public class ClientIdleCheckHandler extends IdleStateHandler {

    private SekiroClient sekiroClient;

    public ClientIdleCheckHandler(SekiroClient sekiroClient) {
        super(Constants.READ_IDLE_TIME, Constants.WRITE_IDLE_TIME, 0);
        this.sekiroClient = sekiroClient;
    }

    @Override
    protected void channelIdle(ChannelHandlerContext ctx, IdleStateEvent evt) throws Exception {
        Channel cmdChannel = sekiroClient.getCmdChannel();
        if (ctx.channel() == cmdChannel) {
            //如果不是cmd channel的事件，直接丢弃
            handleIdle(ctx, evt);
        } else {
            ctx.close();
        }
        super.channelIdle(ctx, evt);
    }

    private void handleIdle(ChannelHandlerContext ctx, IdleStateEvent evt) {
        SekiroLogger.info("idle event:" + evt.state() + " connectionInfo:" + sekiroClient.getSekiroConnectDescription());
        if (IdleStateEvent.FIRST_WRITER_IDLE_STATE_EVENT == evt) {
            SekiroLogger.info("write idle, write a heartbeat message to server");
            SekiroNatMessage sekiroNatMessage = new SekiroNatMessage();
            sekiroNatMessage.setType(SekiroNatMessage.TYPE_HEARTBEAT);
            ctx.channel().writeAndFlush(sekiroNatMessage);
        } else if (IdleStateEvent.FIRST_READER_IDLE_STATE_EVENT == evt) {
            SekiroLogger.info("read timeout,close channel");
            ctx.channel().close();
            SekiroLogger.info("the cmd channel lost,restart client");
            sekiroClient.connectNatServer();
        }
    }
}