Re: [PATCH] 2.5: (better) syscalls for setting task affinity

[Robert Love <rml@tech9.net>]

[-- Attachment #1: Type: text/plain, Size: 483 bytes --]

On Wed, 2002-02-27 at 18:45, Robert Love wrote:

> The attached patch implements a syscall interface for setting and
> retrieving a task's CPU affinity (task->cpus_allowed):

I should posted a test program to demonstrate the syscalls - find such
an attachment below.  It demonstrates using sched_get_affinity to find
the length of the cpu bitmask and then get and set a new value.

`pid_t p' is the process in question.
`unsigned long new_mask' is the new bitmask.

	Robert Love





[-- Attachment #2: affinity.c --]
[-- Type: text/x-c, Size: 1075 bytes --]

/*
 * Example of sched_set_affinity and sched_get_affinity
 * 	Robert Love, 20020227
 */

#include <stdio.h>
#include <stdlib.h>
#include <linux/unistd.h>
#include <unistd.h>

#define __NR_sched_set_affinity 239
#define __NR_sched_get_affinity 240

_syscall3 (int, sched_set_affinity, pid_t, pid, unsigned int, len, unsigned long *, new_mask_ptr)
_syscall3 (int, sched_get_affinity, pid_t, pid, unsigned int *, user_len_ptr, unsigned long *, user_mask_ptr)

int main(int argc, char * argv[])
{
	unsigned long new_mask = 2;
	unsigned int len;
	unsigned long cur_mask;
	pid_t p = getpid();
	int ret;

	ret = sched_get_affinity(p, &len, NULL);
	printf(" len = %u\n", len);

	ret = sched_get_affinity(p, &len, &cur_mask);
	printf(" sched_get_affinity = %d, cur_mask = %ld\n", ret, cur_mask);

	ret = sched_set_affinity(p, len, &new_mask);
	printf(" sched_set_affinity = %d, new_mask = %ld\n", ret, new_mask);

	ret = sched_get_affinity(p, &len, &cur_mask);
	printf(" sched_get_affinity = %d, cur_mask = %ld\n", ret, cur_mask);

	return 0;
}