unix: add SchedSetAttr and SchedGetAttr for Linux

Add wrappers for sched_getattr(2) and sched_setattr(2), as well as various SCHED_ values usable for these. The kludge in linux/types.go is needed so we can include both linux/sched/types.h (for struct sched_attr) and sched.h (for a few defines from include/bits/cpu-set.h). Unfortunately, they both define struct sched_param, thus the need to mask one of the definitions. Change-Id: I3e13cf49ccef7ae81a75d33826d18de84a52106d Signed-off-by: Kir Kolyshkin <[email protected]> Reviewed-on: https://go-review.googlesource.com/c/sys/+/516756 Reviewed-by: Ian Lance Taylor <[email protected]> Auto-Submit: Ian Lance Taylor <[email protected]> Run-TryBot: Ian Lance Taylor <[email protected]> Reviewed-by: Michael Knyszek <[email protected]> TryBot-Result: Gopher Robot <[email protected]> Run-TryBot: Ian Lance Taylor <[email protected]>
golang · Aug 9, 2023 · ee57887 · ee57887
1 parent 60ecf13
commit ee57887
Show file tree

Hide file tree

Showing 6 changed files with 85 additions and 0 deletions.
diff --git a/unix/linux/types.go b/unix/linux/types.go
@@ -138,6 +138,11 @@ struct termios2 {
 #include <linux/random.h>
 #include <linux/rtc.h>
 #include <linux/rtnetlink.h>
+// This is to avoid a conflict of struct sched_param being defined by
+// both the kernel and the glibc (sched.h) headers.
+#define sched_param kernel_sched_param
+#include <linux/sched/types.h>
+#undef kernel_sched_param
 #include <linux/shm.h>
 #include <linux/socket.h>
 #include <linux/stat.h>
@@ -5796,3 +5801,7 @@ const (
 	RISCV_HWPROBE_MISALIGNED_UNSUPPORTED = C.RISCV_HWPROBE_MISALIGNED_UNSUPPORTED
 	RISCV_HWPROBE_MISALIGNED_MASK        = C.RISCV_HWPROBE_MISALIGNED_MASK
 )
+
+type SchedAttr C.struct_sched_attr
+
+const SizeofSchedAttr = C.sizeof_struct_sched_attr
diff --git a/unix/mkerrors.sh b/unix/mkerrors.sh
@@ -583,6 +583,7 @@ ccflags="$@"
 		$2 ~ /^PERF_/ ||
 		$2 ~ /^SECCOMP_MODE_/ ||
 		$2 ~ /^SEEK_/ ||
+		$2 ~ /^SCHED_/ ||
 		$2 ~ /^SPLICE_/ ||
 		$2 ~ /^SYNC_FILE_RANGE_/ ||
 		$2 !~ /IOC_MAGIC/ &&

diff --git a/unix/syscall_linux.go b/unix/syscall_linux.go
@@ -2471,6 +2471,29 @@ func Pselect(nfd int, r *FdSet, w *FdSet, e *FdSet, timeout *Timespec, sigmask *
 	return pselect6(nfd, r, w, e, mutableTimeout, kernelMask)
 }
 
+//sys	schedSetattr(pid int, attr *SchedAttr, flags uint) (err error)
+//sys	schedGetattr(pid int, attr *SchedAttr, size uint, flags uint) (err error)
+
+// SchedSetAttr is a wrapper for sched_setattr(2) syscall.
+// https://man7.org/linux/man-pages/man2/sched_setattr.2.html
+func SchedSetAttr(pid int, attr *SchedAttr, flags uint) error {
+	if attr == nil {
+		return EINVAL
+	}
+	attr.Size = SizeofSchedAttr
+	return schedSetattr(pid, attr, flags)
+}
+
+// SchedGetAttr is a wrapper for sched_getattr(2) syscall.
+// https://man7.org/linux/man-pages/man2/sched_getattr.2.html
+func SchedGetAttr(pid int, flags uint) (*SchedAttr, error) {
+	attr := &SchedAttr{}
+	if err := schedGetattr(pid, attr, SizeofSchedAttr, flags); err != nil {
+		return nil, err
+	}
+	return attr, nil
+}
+
 /*
  * Unimplemented
  */

diff --git a/unix/zerrors_linux.go b/unix/zerrors_linux.go
diff --git a/unix/zsyscall_linux.go b/unix/zsyscall_linux.go
diff --git a/unix/ztypes_linux.go b/unix/ztypes_linux.go