/* SPDX-License-Identifier: GPL-2.0 */
#undef TRACE_SYSTEM
#define TRACE_SYSTEM icmp

#if !defined(_TRACE_ICMP_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_ICMP_H

#include <linux/icmp.h>
#include <linux/tracepoint.h>

TRACE_EVENT(icmp_send,

  TP_PROTO(const struct sk_buff *skb, int type, int code),

  TP_ARGS(skb, type, code),

  TP_STRUCT__entry(
   __field(const void *, skbaddr)
   __field(int, type)
   __field(int, code)
   __array(__u8, saddr, 4)
   __array(__u8, daddr, 4)
   __field(__u16, sport)
   __field(__u16, dport)
   __field(unsigned short, ulen)
  ),

  TP_fast_assign(
   struct iphdr *iph = ip_hdr(skb);
   struct udphdr *uh = udp_hdr(skb);
   int proto_4 = iph->protocol;
   __be32 *p32;

   __entry->skbaddr = skb;
   __entry->type = type;
   __entry->code = code;

   if (proto_4 != IPPROTO_UDP || (u8 *)uh < skb->head ||
    (u8 *)uh + sizeof(struct udphdr)
    > skb_tail_pointer(skb)) {
    __entry->sport = 0;
    __entry->dport = 0;
    __entry->ulen = 0;
   } else {
    __entry->sport = ntohs(uh->source);
    __entry->dport = ntohs(uh->dest);
    __entry->ulen = ntohs(uh->len);
   }

   p32 = (__be32 *) __entry->saddr;
   *p32 = iph->saddr;

   p32 = (__be32 *) __entry->daddr;
   *p32 = iph->daddr;
  ),

  TP_printk("icmp_send: type=%d, code=%d. From %pI4:%u to %pI4:%u ulen=%d skbaddr=%p",
   __entry->type, __entry->code,
   __entry->saddr, __entry->sport, __entry->daddr,
   __entry->dport, __entry->ulen, __entry->skbaddr)
);

#endif /* _TRACE_ICMP_H */

/* This part must be outside protection */
#include <trace/define_trace.h>