org.tron.p2p.connection.message.Message Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of libp2p Show documentation

libp2p is a p2p network SDK implemented in java language.

The newest version!

package org.tron.p2p.connection.message;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import org.apache.commons.lang3.ArrayUtils;
import org.tron.p2p.connection.message.handshake.HelloMessage;
import org.tron.p2p.connection.message.keepalive.PingMessage;
import org.tron.p2p.connection.message.keepalive.PongMessage;
import org.tron.p2p.exception.P2pException;

public abstract class Message {

  protected MessageType type;
  protected byte[] data;

  public Message(MessageType type, byte[] data) {
    this.type = type;
    this.data = data;
  }

  public MessageType getType() {
    return this.type;
  }

  public byte[] getData() {
    return this.data;
  }

  public ByteBuf getSendData() {
    return Unpooled.wrappedBuffer(ArrayUtils.add(this.data, 0, type.getType()));
  }

  public abstract boolean valid();

  public static Message parse(byte[] encode) throws P2pException {
    byte type = encode[0];
    try {
      byte[] data = ArrayUtils.subarray(encode, 1, encode.length);
      Message message;
      switch (MessageType.fromByte(type)) {
        case KEEP_ALIVE_PING:
          message = new PingMessage(data);
          break;
        case KEEP_ALIVE_PONG:
          message = new PongMessage(data);
          break;
        case HANDSHAKE_HELLO:
          message = new HelloMessage(data);
          break;
        default:
          throw new P2pException(P2pException.TypeEnum.NO_SUCH_MESSAGE, "type=" + type);
      }
      if (!message.valid()) {
        throw new P2pException(P2pException.TypeEnum.BAD_MESSAGE, "type=" + type);
      }
      return message;
    } catch (P2pException p2pException) {
      throw p2pException;
    } catch (Exception e) {
      throw new P2pException(P2pException.TypeEnum.BAD_MESSAGE, "type:" + type);
    }
  }

}