Provide optimized assembly for PPOLL blocking syscall on ARM

fako1024 · Sep 7, 2023 · 8e23f5f · 8e23f5f
1 parent e02f6c9
commit 8e23f5f
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 6 deletions.
diff --git a/event/poll_amd64.s b/event/poll_amd64.s
@@ -1,13 +1,15 @@
 #include "textflag.h"
 
+#define SYS__PPOLL 0x10f
+
 // func pollBlock(fds *unix.PollFd, nfds int) (err syscall.Errno)
 TEXT ·pollBlock(SB),NOSPLIT,$0-24
 	CALL	runtime·entersyscallblock(SB)	// Call blocking SYSCALL directive from runtime package
 	MOVQ	fds+0(FP), DI					// PollFDs parameter
 	MOVQ	nfds+8(FP), SI					// Put nFDs parameter
 	MOVQ	$0x0, DX						// Put timeout parameter (set to NULL)
 	MOVQ	$0x0, R10                      	// Put sigmask parameter (skip)
-	MOVQ	$0x10f, AX						// Prepare / perform ppoll() SYSCALL
+	MOVQ	$SYS__PPOLL, AX					// Prepare / perform ppoll() SYSCALL
 	SYSCALL
 	CMPQ	AX, $0xfffffffffffff002			// No error / EINTR
 	JLS		success							// Jump to success

diff --git a/event/poll_arm.s b/event/poll_arm.s
@@ -0,0 +1,24 @@
+#include "textflag.h"
+
+#define SYS__PPOLL 0x150
+
+// func pollBlock(fds *unix.PollFd, nfds int) (err syscall.Errno)
+TEXT ·pollBlock(SB),NOSPLIT,$0-12
+	BL  	runtime·entersyscallblock(SB)	// Call blocking SYSCALL directive from runtime package
+	MOVW	$SYS__PPOLL, R7			// Prepare / perform ppoll() SYSCALL    
+	MOVW	fds+0(FP), R0			// PollFDs parameter
+	MOVW	nfds+4(FP), R1		        // Put nFDs parameter
+	MOVW	$0x0, R2			// Put timeout parameter (set to NULL)
+	MOVW	$0x0, R3                      	// Put sigmask parameter (skip)
+	SWI     $0
+        CMP     $0xfffff002, R0		        // No error / EINTR
+	BLS	success			        // Jump to success
+	RSB     $0, R0, R0			// Negate SYSCALL errno
+	MOVW	R0, err+8(FP)			// Store error code in err return value
+	BL  	runtime·exitsyscall(SB)		// Finalize SYSCALL using the directive from runtime package
+	RET					// Return
+success:
+	MOVW    $0, R0
+	MOVW	R0, err+8(FP)			// Store NULL error code in err return value
+	BL	runtime·exitsyscall(SB)		// Finalize SYSCALL using the directive from runtime package
+	RET					// Return
diff --git a/event/poll_arm64.s b/event/poll_arm64.s
@@ -1,13 +1,15 @@
 #include "textflag.h"
 
+#define SYS__PPOLL 0x49
+
 // func pollBlock(fds *unix.PollFd, nfds int) (err syscall.Errno)
 TEXT ·pollBlock(SB),NOSPLIT,$0-24
 	BL  	runtime·entersyscallblock(SB)	// Call blocking SYSCALL directive from runtime package
 	MOVD	fds+0(FP), R0					// PollFDs parameter
 	MOVD	nfds+8(FP), R1					// Put nFDs parameter
 	MOVD	$0x0, R2						// Put timeout parameter (set to NULL)
 	MOVD	$0x0, R3                      	// Put sigmask parameter (skip)
-	MOVD	$0x49, R8						// Prepare / perform ppoll() SYSCALL
+	MOVD	$SYS__PPOLL, R8					// Prepare / perform ppoll() SYSCALL
 	SVC
 	CMP     $0xfffffffffffff002, R0			// No error / EINTR
 	BLS		success							// Jump to success

diff --git a/event/poll_asm.go b/event/poll_asm.go
@@ -1,5 +1,5 @@
-//go:build (linux && amd64) || (linux && arm64)
-// +build linux,amd64 linux,arm64
+//go:build (linux && amd64) || (linux && arm64) || (linux && arm)
+// +build linux,amd64 linux,arm64 linux,arm
 
 package event
 

diff --git a/event/poll_default.go b/event/poll_default.go
@@ -1,5 +1,5 @@
-//go:build linux && !amd64 && !arm64
-// +build linux,!amd64,!arm64
+//go:build linux && !amd64 && !arm64 && !arm
+// +build linux,!amd64,!arm64,!arm
 
 package event