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