/* 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, *uh = NULL;
			__be32 *p32;

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

			if (iph->protocol == IPPROTO_UDP)
				uh = skb_header_pointer(skb,
							skb_network_offset(skb) +
							(iph->ihl << 2),
							sizeof(_uh), &_uh);

			if (!uh) {
				__entry->sport = 0;
				__entry->dport = 0;
				__entry->ulen = 0;
			} else {
				__entry->sport = ntohs(uh->source);
				__entry->dport = ntohs(uh->dest);
				__entry->ulen = udp_get_len_short(uh);
			}

			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>