/* 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>