my-task-manager/task.h at master · laoar/my-task-manager · GitHub

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
#ifndef _CPU_LIMIT_TASK_
#define _CPU_LIMIT_TASK_

#include <unistd.h>
#include <limits.h>
#include <dirent.h>
#include <libproc.h>
#include <signal.h>
#include <string.h>
#include <assert.h>
#include <sys/param.h>
#include <sys/queue.h>
#include <sys/time.h>

/*
   refer to struct proc_bsdinfo, which define in bsd/sys/proc_info.h.
   The max len of exe name.
*/

#define NAME_LEN 2*MAXCOMLEN
#define MIN_CPU_TIME 20
#define TIME_SLOT 100000  // 1ms.

struct task_control {
	uint64_t task_time;
	struct timeval cputime;
	double cpu_usage; /* value int range <0, 1>*/
};

struct allpid {
	int cnt;
	int *pid_list;
};

extern struct allpid pids;

struct pid_hit_list {
	TAILQ_ENTRY(pid_hit_list) pid_list;
	pid_t pid;
	struct proc_taskallinfo ti;
	/* the following 3 members are for managment, although they're already exist in ti.*/
	pid_t ppid;
	uint64_t start_time;
	uint64_t cputime;
	struct timeval last_update;
};


int get_allproc(struct allpid *pids);
int get_pidinfo(pid_t pid, struct proc_taskallinfo *task_info);
int pid_hit_list_init(char *task_name);
int task_manage(double limit);
void inc_nice(void);
#endif