From 017aad9c2d4f5621a4902d707e371fd32ae187b1 Mon Sep 17 00:00:00 2001
From: Roberto Scolaro <roberto.scolaro21@gmail.com>
Date: Tue, 7 May 2024 09:51:06 +0000
Subject: [PATCH] fix(bpf): fix bpf on aarch64

Signed-off-by: Roberto Scolaro <roberto.scolaro21@gmail.com>
---
 driver/bpf/fillers.h                          | 22 +++++++++++++++++++
 .../helpers/store/auxmap_store_params.h       |  5 +++++
 2 files changed, 27 insertions(+)

diff --git a/driver/bpf/fillers.h b/driver/bpf/fillers.h
index 311e1856a6..9da334155d 100644
--- a/driver/bpf/fillers.h
+++ b/driver/bpf/fillers.h
@@ -578,29 +578,37 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 						       int flags)
 {
 	const struct iovec *iov;
+#if defined(CONFIG_X86_64)
 	const struct compat_iovec *compat_iov;
+#endif
 	int res = PPM_SUCCESS;
 	unsigned int copylen;
 	long size = 0;
 	int j;
 
+#if defined(CONFIG_X86_64)
 	if (!bpf_in_ia32_syscall())
+#endif
 	{
 		copylen = iovcnt * sizeof(struct iovec);
 		iov = (const struct iovec *)data->tmp_scratch;
 	}
+#if defined(CONFIG_X86_64)
 	else
 	{
 		copylen = iovcnt * sizeof(struct compat_iovec);
 		compat_iov = (const struct compat_iovec *)data->tmp_scratch;
 	}
+#endif
 
 	if (copylen > SCRATCH_SIZE_MAX)
 	{
 		return PPM_FAILURE_FRAME_SCRATCH_MAP_FULL;
 	}
 
+#if defined(CONFIG_X86_64)
 	if (!bpf_in_ia32_syscall())
+#endif
 	{
 #ifdef BPF_FORBIDS_ZERO_ACCESS
 		if (copylen)
@@ -614,6 +622,7 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 #endif
 			return PPM_FAILURE_INVALID_USER_MEMORY;
 	}
+#if defined(CONFIG_X86_64)
 	else
 	{
 #ifdef BPF_FORBIDS_ZERO_ACCESS
@@ -628,6 +637,7 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 #endif
 			return PPM_FAILURE_INVALID_USER_MEMORY;
 	}
+#endif
 
 
 	#pragma unroll
@@ -638,14 +648,18 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 		if (size == LONG_MAX)
 			break;
 
+#if defined(CONFIG_X86_64)
 		if (!bpf_in_ia32_syscall())
+#endif
 		{
 			size += iov[j].iov_len;
 		}
+#if defined(CONFIG_X86_64)
 		else
 		{
 			size += compat_iov[j].iov_len;
 		}
+#endif
 	}
 
 	if ((flags & PRB_FLAG_IS_WRITE) == 0)
@@ -674,13 +688,16 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 				if (off > SCRATCH_SIZE_HALF)
 					break;
 
+#if defined(CONFIG_X86_64)
 				if (!bpf_in_ia32_syscall())
+#endif
 				{
 					if (iov[j].iov_len <= remaining)
 						to_read = iov[j].iov_len;
 					else
 						to_read = remaining;
 				}
+#if defined(CONFIG_X86_64)
 				else
 				{
 					if (compat_iov[j].iov_len <= remaining)
@@ -688,11 +705,14 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 					else
 						to_read = remaining;
 				}
+#endif
 
 				if (to_read > SCRATCH_SIZE_HALF)
 					to_read = SCRATCH_SIZE_HALF;
 
+#if defined(CONFIG_X86_64)
 				if (!bpf_in_ia32_syscall())
+#endif
 				{
 	#ifdef BPF_FORBIDS_ZERO_ACCESS
 					if (to_read)
@@ -706,6 +726,7 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 	#endif
 						return PPM_FAILURE_INVALID_USER_MEMORY;
 				}
+#if defined(CONFIG_X86_64)
 				else
 				{
 
@@ -722,6 +743,7 @@ static __always_inline int bpf_parse_readv_writev_bufs(struct filler_data *data,
 						return PPM_FAILURE_INVALID_USER_MEMORY;
 
 				}
+#endif
 
 				remaining -= to_read;
 				off += to_read;
diff --git a/driver/modern_bpf/helpers/store/auxmap_store_params.h b/driver/modern_bpf/helpers/store/auxmap_store_params.h
index 2b5939b3d2..ac179cee66 100644
--- a/driver/modern_bpf/helpers/store/auxmap_store_params.h
+++ b/driver/modern_bpf/helpers/store/auxmap_store_params.h
@@ -627,6 +627,11 @@ static __always_inline void auxmap__store_sockaddr_param(struct auxiliary_map *a
 			start_reading_point = (unsigned long)sockaddr_un->sun_path;
 		}
 
+		// The addrlen is used has hard limit. So we should use add 1 for the `\0`
+		addrlen -= (FAMILY_SIZE + 1);
+		if(sockaddr_un->sun_path[addrlen-1] != '\0')
+			addrlen += 1;
+
 		/* Pack the sockaddr info:
 		 * - socket family.
 		 * - socket_unix_path (sun_path).