Added KUTrace Support

Moved sched_litmus back to the top scheduling class

3 files changed, 45 insertions, 0 deletions

diff --git a/net/ipv4/tcp_input.c b/net/ipv4/tcp_input.c
index a2e52ad7cdab..87641ed2c172 100644
--- a/net/ipv4/tcp_input.c
+++ b/net/ipv4/tcp_input.c
@@ -80,6 +80,8 @@
 #include <linux/jump_label_ratelimit.h>
 #include <net/busy_poll.h>
 
+#include <linux/kutrace.h>
+
 int sysctl_tcp_max_orphans __read_mostly = NR_FILE;
 
 #define FLAG_DATA               0x01 /* Incoming frame contained data.          */
@@ -5547,6 +5549,19 @@ void tcp_rcv_established(struct sock *sk, struct sk_buff *skb)
         struct tcp_sock *tp = tcp_sk(sk);
         unsigned int len = skb->len;
 
+        /* dsites 2020.10.28 */
+        /* Apply quick filter and if it passes, make a KUtrace entry for */
+        /* rx packet. Use XOR of first 32 bytes as the recorded argument value */
+        if (kutrace_tracing && ((20 + 32) <= len)) {
+                int ku_hdr_len = th->doff << 2;
+                const u64 *ku_payload = (u64*)(skb->data + ku_hdr_len);
+                int ku_payloadlen = len - ku_hdr_len;
+                if (32 <= ku_payloadlen) {
+                        kutrace_pkttrace(KUTRACE_RX_PKT, ku_payload);
+                }
+        }
+
+
         /* TCP congestion window tracking */
         trace_tcp_probe(sk, skb);
 
diff --git a/net/ipv4/tcp_output.c b/net/ipv4/tcp_output.c
index 0488607c5cd3..1a1df10a99b9 100644
--- a/net/ipv4/tcp_output.c
+++ b/net/ipv4/tcp_output.c
@@ -46,6 +46,8 @@
 
 #include <trace/events/tcp.h>
 
+#include <linux/kutrace.h>
+
 /* Refresh clocks of a TCP socket,
  * ensuring monotically increasing values.
  */
@@ -1166,6 +1168,14 @@ static int __tcp_transmit_skb(struct sock *sk, struct sk_buff *skb,
 
         tcp_add_tx_delay(skb, tp);
 
+        /* dsites 2020.11.11 */
+        /* Apply quick filter and if it passes, make a KUtrace entry for tx packet. */
+        /* Use XOR of first 32 bytes as the recorded argument value */
+        if (kutrace_tracing && ((tcp_header_size + 32) <= skb->len)) {
+                const u64 *ku_payload = (u64*)(skb->data + tcp_header_size);
+                kutrace_pkttrace(KUTRACE_TX_PKT, ku_payload);
+        }
+
         err = icsk->icsk_af_ops->queue_xmit(sk, skb, &inet->cork.fl);
 
         if (unlikely(err > 0)) {
diff --git a/net/ipv4/udp.c b/net/ipv4/udp.c
index 1d58ce829dca..da6e0b21b339 100644
--- a/net/ipv4/udp.c
+++ b/net/ipv4/udp.c
@@ -113,6 +113,8 @@
 #include <net/addrconf.h>
 #include <net/udp_tunnel.h>
 
+#include <linux/kutrace.h>
+
 struct udp_table udp_table __read_mostly;
 EXPORT_SYMBOL(udp_table);
 
@@ -888,6 +890,15 @@ csum_partial:
                 uh->check = CSUM_MANGLED_0;
 
 send:
+
+        /* dsites 2020.11.11 */
+        /* Apply quick filter and if it passes, make a KUtrace entry for tx packet. */
+        /* Use XOR of first 32 bytes as the recorded argument value */
+        if (kutrace_tracing && ((8 + 32) <= len)) {
+                const u64 *ku_payload = (u64*)(skb->data);
+                kutrace_pkttrace(KUTRACE_TX_PKT, ku_payload);
+        }
+
         err = ip_send_skb(sock_net(sk), skb);
         if (err) {
                 if (err == -ENOBUFS && !inet->recverr) {
@@ -2289,6 +2300,15 @@ int __udp4_lib_rcv(struct sk_buff *skb, struct udp_table *udptable,
         if (udp4_csum_init(skb, uh, proto))
                 goto csum_error;
 
+        /* dsites 2020.11.02 */
+        /* Apply quick filter and if it passes, make a KUtrace entry for rx packet. */
+        /* Use XOR of first 32 bytes as the recorded argument value */
+        if (kutrace_tracing && ((8 + 32) <= ulen)) {
+                int ku_hdr_len = 8;
+                const u64 *ku_payload = (u64*)((u8*)(uh) + ku_hdr_len);
+                kutrace_pkttrace(KUTRACE_RX_PKT, ku_payload);
+        }
+
         sk = skb_steal_sock(skb);
         if (sk) {
                 struct dst_entry *dst = skb_dst(skb);