1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

SYSCALL_DEFINE2(getrusage, int, who, struct rusage __user *, ru)
{
	struct rusage r;

	if (who != RUSAGE_SELF && who != RUSAGE_CHILDREN &&
	    who != RUSAGE_THREAD)
		return -EINVAL;

	getrusage(current, who, &r);
	return copy_to_user(ru, &r, sizeof(r)) ? -EFAULT : 0;
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

void getrusage(struct task_struct *p, int who, struct rusage *r)
{
	struct task_struct *t;
	unsigned long flags;
	u64 tgutime, tgstime, utime, stime;
	unsigned long maxrss = 0;

	memset((char *)r, 0, sizeof (*r));
	utime = stime = 0;

	if (who == RUSAGE_THREAD) {
		task_cputime_adjusted(current, &utime, &stime);
		accumulate_thread_rusage(p, r);
		maxrss = p->signal->maxrss;
		goto out;
	}

	if (!lock_task_sighand(p, &flags))
		return;

	switch (who) {
	case RUSAGE_BOTH:
	case RUSAGE_CHILDREN:
		utime = p->signal->cutime;
		stime = p->signal->cstime;
		r->ru_nvcsw = p->signal->cnvcsw;
		r->ru_nivcsw = p->signal->cnivcsw;
		r->ru_minflt = p->signal->cmin_flt;
		r->ru_majflt = p->signal->cmaj_flt;
		r->ru_inblock = p->signal->cinblock;
		r->ru_oublock = p->signal->coublock;
		maxrss = p->signal->cmaxrss;

		if (who == RUSAGE_CHILDREN)
			break;

	case RUSAGE_SELF:
		thread_group_cputime_adjusted(p, &tgutime, &tgstime);
		utime += tgutime;
		stime += tgstime;
		r->ru_nvcsw += p->signal->nvcsw;
		r->ru_nivcsw += p->signal->nivcsw;
		r->ru_minflt += p->signal->min_flt;
		r->ru_majflt += p->signal->maj_flt;
		r->ru_inblock += p->signal->inblock;
		r->ru_oublock += p->signal->oublock;
		if (maxrss < p->signal->maxrss)
			maxrss = p->signal->maxrss;
		t = p;
		do {
			accumulate_thread_rusage(t, r);
		} while_each_thread(p, t);
		break;

	default:
		BUG();
	}
	unlock_task_sighand(p, &flags);

out:
	r->ru_utime = ns_to_timeval(utime);
	r->ru_stime = ns_to_timeval(stime);

	if (who != RUSAGE_CHILDREN) {
		struct mm_struct *mm = get_task_mm(p);

		if (mm) {
			setmax_mm_hiwater_rss(&maxrss, mm);
			mmput(mm);
		}
	}
	r->ru_maxrss = maxrss * (PAGE_SIZE / 1024); /* convert pages to KBs */
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

// Customed getrusage for the kernel
SYSCALL_DEFINE2(my_getrusage,pid_t,p_pid,struct rusage __user *, u_rusage)
{
	struct task_struct *p;
	struct rusage *k_rusage;
	p=pid_task(find_get_pid(p_pid),PIDTYPE_PID);
	if(p==NULL)
	{
		printk(KERN_ALERT"Could not find a process by pid %d",p_pid);
		return -EFAULT;
	}else{
		printk(KERN_ALERT"PID[%d]@[%s]",p->pid,p->comm);
	}

	k_rusage=kmalloc(sizeof(struct rusage),GFP_KERNEL);
	if(k_rusage==NULL)
	{
		printk(KERN_ALERT"Could not allocate memory for k_rusage");
		return -EFAULT;
	}

	getrusage(p, RUSAGE_SELF, k_rusage);

	if(copy_to_user(u_rusage, k_rusage, sizeof(struct rusage))!=0){
		printk(KERN_ALERT"Could not copy usage to user");
		return -EFAULT;
	}
	return 0;
}

1
2

#define __NR_my_getrusage 295
__SYSCALL(__NR_my_getrusage, sys_my_getrusage)

1

asmlinkage long sys_my_getrusage(pid_t p_pid, struct rusage __user *u_rusage);

1

make openeuler_defconfig

1

make -j24

1

make modules_install && make install

1

reboot

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

#include <stdio.h>
#include <stdlib.h>
#include <linux/kernel.h>
#include <sys/syscall.h>
#include <sys/resource.h>
#include <unistd.h>

#define MY_GETRUSAGE 295

void printUsage(struct rusage *usage);

int main(){
    pid_t pid;
    struct rusage *usage = NULL;

    printf("Enter the pid of the process: \n");
    scanf("%d", &pid);

    printf("The pid received is %d\n", pid);

    usage=malloc(sizeof(struct rusage));
    if(usage==NULL){
        printf("Error in allocating memory to 'usage'\n");
        return -1;
    }

    syscall(MY_GETRUSAGE, pid, usage);

    if(usage == NULL){
        printf("Error in the system call\n");
    }
    else{
        printf("The system call was successful\n");
        printUsage(usage);
    }
    return 0;
}

void printUsage(struct rusage *usage){
    printf("\nfunction printUsage() called\n");
    printf("%-15s %-15s %-20s %-20s %-15s %-15s\n", "[User Time]", "[System Time]", "[Page Faults|MIN]", "[Page Faults|MAX]", "[Block IN]","[Block OUT]");

    printf("%-15.3f %-15.3f %-20ld %-20ld %-15ld %-15ld\n",
           usage->ru_utime.tv_sec + usage->ru_utime.tv_usec*1.0,
           usage->ru_stime.tv_sec + usage->ru_stime.tv_usec*1.0,
           usage->ru_minflt,
           usage->ru_majflt,
           usage->ru_inblock,
           usage->ru_oublock);
}

1
2
3
4
5

# 编译
gcc getrusage.c -o getrusage.o

# 运行
./getrusage.o

1

yum install -y openssl-devel bison flex libelf-devel

1
2
3
4
5
6

copy_from_user(void *to, const void __user *from, unsigned long n)
{
	if (check_copy_size(to, n, false))
		n = _copy_from_user(to, from, n);
	return n;
}

1
2
3

fdisk -l #查看磁盘信息

umount /dev/vdb1  /mnt # 将vdb1设备挂载到/mnt目录下

1

insmod ./test.ko p_pid=1234