| 76 | } |
| 77 | |
| 78 | func (pd *IPDecoder) Decode(packet RawPacket) (*IP, error) { |
| 79 | // Should decode to IP layer |
| 80 | decoded, err := pd.decodeByVersion(packet.Data) |
| 81 | if err != nil { |
| 82 | return nil, err |
| 83 | } |
| 84 | |
| 85 | for _, layerType := range decoded { |
| 86 | switch layerType { |
| 87 | case layers.LayerTypeIPv4: |
| 88 | return newIPv4(pd.ipv4) |
| 89 | case layers.LayerTypeIPv6: |
| 90 | return newIPv6(pd.ipv6) |
| 91 | } |
| 92 | } |
| 93 | return nil, fmt.Errorf("no ip layer is decoded") |
| 94 | } |
| 95 | |
| 96 | func (pd *IPDecoder) decodeByVersion(packet []byte) ([]gopacket.LayerType, error) { |
| 97 | version, err := FindIPVersion(packet) |