refactor: replace plain assignments with built-in memcpy function

daeuniverse · Sep 13, 2024 · 398628f · 398628f
1 parent 9701f5a
commit 398628f
Showing 1 changed file with 8 additions and 16 deletions.
diff --git a/control/kern/tproxy.c b/control/kern/tproxy.c
@@ -1855,14 +1855,10 @@ int local_tcp_sockops(struct bpf_sock_ops *skops)
 		tuple.dip.u6_addr32[2] = bpf_htonl(0x0000ffff);
 		tuple.dip.u6_addr32[3] = skops->remote_ip4;
 	} else if (skops->family == AF_INET6) {
-		tuple.sip.u6_addr32[3] = skops->local_ip6[3];
-		tuple.sip.u6_addr32[2] = skops->local_ip6[2];
-		tuple.sip.u6_addr32[1] = skops->local_ip6[1];
-		tuple.sip.u6_addr32[0] = skops->local_ip6[0];
-		tuple.dip.u6_addr32[3] = skops->remote_ip6[3];
-		tuple.dip.u6_addr32[2] = skops->remote_ip6[2];
-		tuple.dip.u6_addr32[1] = skops->remote_ip6[1];
-		tuple.dip.u6_addr32[0] = skops->remote_ip6[0];
+		__builtin_memcpy(tuple.sip.u6_addr32, skops->local_ip6,
+				 sizeof(skops->local_ip6));
+		__builtin_memcpy(tuple.dip.u6_addr32, skops->remote_ip6,
+				 sizeof(skops->remote_ip6));
 	} else {
 		return 0;
 	}
@@ -1933,14 +1929,10 @@ int sk_msg_fast_redirect(struct sk_msg_md *msg)
 		rev_tuple.dip.u6_addr32[2] = bpf_htonl(0x0000ffff);
 		rev_tuple.dip.u6_addr32[3] = msg->local_ip4;
 	} else if (msg->family == AF_INET6) {
-		rev_tuple.sip.u6_addr32[3] = msg->remote_ip6[3];
-		rev_tuple.sip.u6_addr32[2] = msg->remote_ip6[2];
-		rev_tuple.sip.u6_addr32[1] = msg->remote_ip6[1];
-		rev_tuple.sip.u6_addr32[0] = msg->remote_ip6[0];
-		rev_tuple.dip.u6_addr32[3] = msg->local_ip6[3];
-		rev_tuple.dip.u6_addr32[2] = msg->local_ip6[2];
-		rev_tuple.dip.u6_addr32[1] = msg->local_ip6[1];
-		rev_tuple.dip.u6_addr32[0] = msg->local_ip6[0];
+		__builtin_memcpy(rev_tuple.sip.u6_addr32, msg->remote_ip6,
+				 sizeof(msg->remote_ip6));
+		__builtin_memcpy(rev_tuple.dip.u6_addr32, msg->local_ip6,
+				 sizeof(msg->local_ip6));
 	} else {
 		return SK_PASS;
 	}