pcap.codec.TransportLayer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pcap-codec Show documentation

Packet Codec

There is a newer version: 1.5.1

/** This code is licenced under the GPL version 2. */
package pcap.codec;

import java.util.HashMap;
import java.util.Map;
import pcap.common.annotation.Inclubating;
import pcap.common.memory.Memory;
import pcap.common.util.NamedNumber;

/** @author Ardika Rommy Sanjaya */
@Inclubating
public final class TransportLayer extends NamedNumber {

  public static final TransportLayer ICMP =
      new TransportLayer((byte) 1, "Internet Control Message Protocol Version 4");

  public static final TransportLayer IPV6 = new TransportLayer((byte) 41, "IPv6 Header.");

  public static final TransportLayer IPV6_ICMP =
      new TransportLayer((byte) 58, "Internet Control Message Protocol Version 6");

  public static final TransportLayer IGMP =
      new TransportLayer((byte) 2, "Internet Group Management Protocol");

  public static final TransportLayer TCP =
      new TransportLayer((byte) 6, "Transmission Control Protocol");

  public static final TransportLayer UDP = new TransportLayer((byte) 17, "User Datagram Protocol");

  public static final TransportLayer UNKNOWN = new TransportLayer((byte) -1, "Unknown");

  private static final Map REGISTRY = new HashMap<>();

  private static final Map BUILDER = new HashMap<>();

  public TransportLayer(int value, String name) {
    super((byte) value, name);
  }

  public static TransportLayer valueOf(final Byte value) {
    TransportLayer transportLayer = REGISTRY.get(value);
    if (transportLayer == null) {
      return UNKNOWN;
    } else {
      return transportLayer;
    }
  }

  /**
   * @param type type.
   * @param packetBuilder packet builder.
   */
  public static synchronized void register(
      TransportLayer type, AbstractPacket.Builder packetBuilder) {
    BUILDER.put(type.value(), packetBuilder);
    REGISTRY.put(type.value(), type);
  }

  public Packet newInstance(Memory buffer) {
    AbstractPacket.Builder packetBuilder = BUILDER.get(this.value());
    if (packetBuilder == null) {
      if (buffer == null || buffer.capacity() <= 0) {
        return null;
      }
      return new UnknownPacket.Builder().build(buffer);
    }
    return packetBuilder.build(buffer);
  }
}