鏇存柊浜嗗唴瀛樼鐞嗕箣椤靛紡绠＄悊鏈哄埗...

2026-09-22 04:33:37 +08:00 · 2016-08-07 16:35:13 +08:00
parent 0a5d402edf
commit 212a9fac6e
59 changed files with 686 additions and 686 deletions
@@ -0,0 +1,69 @@
+# ------------------------------------------------------------------------------
+#
+# Makefile for the LDD-LinuxDeviceDrivers print_task.
+#
+# Author: gatieme
+# Create: 2016-07-29 15:50:46
+# Last modified: 2016-07-29 16:10:29
+# Description:
+# 	This program is loaded as a kernel(v2.6.18 or later) module.
+# 	Use "make install" to load it into kernel.
+# 	Use "make remove" to remove the module out of kernel.
+#
+# ------------------------------------------------------------------------------
+
+
+ROOT=..
+#PLATFORM=$(shell $(ROOT)/systype.sh)
+#include $(ROOT)/Make.defines.$(PLATFORM)
+
+MODULE_NAME := print_task
+
+ifneq ($(KERNELRELEASE),)
+
+#CFG_FLAGS += -O2  -I./
+#EXTRA_CFLAGS  += $(C_FLAGS) $(CFG_INC) $(CFG_INC)
+
+
+
+obj-m := $(MODULE_NAME).o #print_task.o
+
+else
+
+KERNELDIR ?= /lib/modules/$(shell uname -r)/build
+
+PWD := $(shell pwd)
+
+modules:
+	make -C $(KERNELDIR) M=$(PWD) modules
+
+modules_install:
+	make -C $(KERNELDIR) M=$(PWD) modules_install
+
+
+
+insmod:
+	sudo insmod $(MODULE_NAME).ko
+
+reinsmod:
+	sudo rmmod $(MODULE_NAME)
+	sudo insmod $(MODULE_NAME).ko
+
+github:
+	cd $(ROOT) && make github
+
+rmmod:
+	sudo rmmod $(MODULE_NAME)
+
+test :
+	sudo ../injector/memInjector -l stack -m random -t word_0 --time 1 --timeout 3 -p 1
+
+clean:
+	make -C $(KERNELDIR) M=$(PWD) clean
+	rm -f modules.order Module.symvers Module.markers
+
+.PHNOY:
+	modules modules_install clean
+
+endif
+
@@ -0,0 +1,173 @@
+#include <linux/module.h>
+#include <linux/kernel.h>
+#include <linux/init.h>
+#include <linux/moduleparam.h>
+#include <linux/sched.h>
+#include <linux/list.h>
+#include <linux/mm.h>
+#include <linux/mm_types.h>
+
+
+#include <linux/sched.h>
+#include <linux/sched/rt.h>
+
+
+
+#if !defined(NICE_TO_PRIO)
+#define NICE_TO_PRIO(nice)      ((nice) + DEFAULT_PRIO)
+#endif
+
+#if !defined(PRIO_TO_NICE)
+#define PRIO_TO_NICE(prio)      ((prio) - DEFAULT_PRIO)
+#endif
+
+#ifndef offsetof
+#define offsetof(type, field)   ((long) &((type *)0)->field)
+#endif   /* offsetof */
+
+#ifndef container_of
+#define container_of(ptr, type, member) ({                      \
+        const typeof( ((type *)0)->member ) *__mptr = (ptr);    \
+        (type *)( (char *)__mptr - offsetof(type,member) );})
+#endif
+
+
+static int pid = 1;
+module_param(pid,int,0644);
+
+
+static inline int print_task_policy(int policy)
+{
+    printk("POLICY = ");
+    switch(policy)
+    {
+        case SCHED_FIFO     :
+            printk("Real-Time FIFO SCHEDULER TASK\n");
+            break;
+        case SCHED_RR       :
+            printk("Real-Time RR SCHEDULER TASK\n");
+            break;
+        case SCHED_DEADLINE :
+            printk("Real-Time EDF SCHEDULER TASK\n");
+            break;
+        case SCHED_NORMAL   :
+            printk("Normal NORMAL SCHEDULER TASK\n");
+            break;
+        case SCHED_BATCH    :
+            printk("Normal BATCH SCHEDULER TASK\n");
+            break;
+        case SCHED_IDLE     :
+            printk("IDLE SCHEDULER TASK\n");
+            break;
+    }
+
+    return policy;
+}
+
+
+#if 0
+extern const struct sched_class stop_sched_class;
+extern const struct sched_class dl_sched_class;
+extern const struct sched_class rt_sched_class;
+extern const struct sched_class fair_sched_class;
+extern const struct sched_class idle_sched_class;
+
+
+void print_task_sched_class(const struct sched_class  *schedclass)
+{
+    if(schedclass == &stop_sched_class)
+    {
+        printk("STOP SCHED CLASS\n");
+    }
+    else if(schedclass == &dl_sched_class)
+    {
+        printk("EDF SCHED CLASS\n");
+    }
+    else if(schedclass == &rt_sched_class)
+    {
+        printk("RT SCHED CLASS\n");
+    }
+    else if(schedclass == &fair_sched_class)
+    {
+        printk("CFS SCHED CLASS\n");
+    }
+    else if(schedclass == &idle_sched_class)
+    {
+        printk("IDLE SCHED CLASS\n");
+    }
+    else
+    {
+        printk("unknown shced class\n");
+    }
+}
+#endif
+
+void print_task_priority(struct task_struct *ptask)
+{
+    //  priority
+    if(rt_task(ptask) == 1)
+    {
+        printk("Real Time Process\n");
+        printk("rt_priority = %d\n", ptask->rt_priority);
+    }
+    else
+    {
+        printk("Normal Process\n");
+        printk("static prio = %d, NICE(%d)\n", ptask->static_prio, PRIO_TO_NICE(ptask->static_prio));
+    }
+    printk("normal_prio = %d\n", ptask->normal_prio);
+    printk("prio = %d\n", ptask->prio);
+}
+
+
+
+
+void print_task_struct(struct task_struct *ptask)
+{
+    printk("flag = 0x%x\n", ptask->flags);
+
+    //  priority
+    print_task_priority(ptask);
+
+    //  policy
+    print_task_policy(ptask->policy);
+    printk("cpus_allow = 0x%x\n", ptask->nr_cpus_allowed);
+
+    // sched class
+    //print_task_sched_class(ptask->sched_class);
+}
+
+
+
+static void print_task(int pid)
+{
+	struct task_struct *ptask;
+	struct pid *k;
+
+	k = find_vpid(pid);
+	ptask = pid_task(k, PIDTYPE_PID);
+
+    print_task_struct(ptask);
+
+
+    return ;
+}
+
+static int __init print_task_init(void)
+{
+	printk(KERN_INFO"------------------------\n");
+    print_task(pid);
+	printk(KERN_INFO"------------------------\n");
+
+    return 0;
+}
+
+static void __exit print_task_exit(void)
+{
+   printk(KERN_INFO"exit ---------------------!\n");
+}
+
+module_init(print_task_init);
+module_exit(print_task_exit);
+
+MODULE_LICENSE("GPL");