增加了自己的驱动学习...

2026-09-22 04:33:37 +08:00 · 2016-05-02 16:43:18 +08:00
parent 81386bf015
commit 9c310d57a0
25 changed files with 5673 additions and 0 deletions
@@ -0,0 +1,121 @@
+/*
+*  Author: HIT CS HDMC team.
+*  Create: 2010-3-12 8:20:01
+*  Last modified: 2010-6-13 14:13:47
+*/
+
+#ifndef _MEMORY_ENGINE_H
+#define _MEMORY_ENGINE_H
+
+/*
+*  common include header files
+*/
+
+
+#include <linux/version.h>
+#include <linux/kernel.h>
+#include <linux/module.h>
+#include <linux/string.h>
+#include <linux/kprobes.h>
+#include <linux/mm.h>
+#include <linux/sched.h>
+#include <linux/fs.h>
+#include <linux/mount.h>
+#include <linux/dcache.h>
+#include <linux/slab.h>
+#include <linux/highmem.h>
+#include <linux/proc_fs.h>
+#include <linux/uaccess.h>
+#include <linux/kallsyms.h>
+#include <linux/kprobes.h>
+#include <asm/pgtable.h>
+#include <asm/page.h>
+
+#include "common.h"
+
+
+#define MAX_LINE		        256						/// max length of a line
+#define DELIMITER		        '\n'					/// the char is used to split the taskinfo
+#define PERMISSION	            0644					/// proc node permission
+/*
+*	return values
+*/
+#define OK					     0
+#define FAIL					-1
+#define PGD_NONE 			    -2
+#define PUD_NONE			    -3
+#define PMD_NONE 			    -4
+#define PTE_NONE 			    -5
+#define PTE_NOT_PRESENT         -6
+
+
+
+/*
+*	request command
+*/
+#define REQUEST_TASK_INFO		1		/// get a task's memory map information
+#define REQUEST_V2P				2		/// convert a process's linear address to physical address
+#define REQUEST_KV2P			3		/// convert kernel virtual address to physical address
+#define REQUEST_KFUNC_VA		4		/// get kernel function's addr(kernel virtual address)
+#define REQUEST_READ_KFUNC	    5		/// 请求读取内核函数起始地址内容
+#define REQUEST_WRITE_KFUNC	    6		/// 请求改写内核函数起始地址内容
+///#define REQUEST_WRITE				10 	/// 请求改写指定物理地址内容，改为用户态实现此功能
+///#define REQUEST_MEM					11	/// 请求获取全部物理内存信息
+///#define REQUEST_ADDR_STOP		12	///
+
+/*
+*	ack signals
+*/
+#define ACK_TASK_INFO			REQUEST_TASK_INFO
+#define ACK_V2P					REQUEST_V2P
+#define ACK_KV2P				REQUEST_KV2P
+#define ACK_KFUNC_VA			REQUEST_KFUNC_VA
+#define ACK_READ_KFUNC		    REQUEST_READ_KFUNC
+#define ACK_WRITE_KFUNC		    REQUEST_WRITE_KFUNC
+///#define REQUEST_WRITE		REQUEST_WRITE
+///#define ACK_MEM				REQUEST_MEM
+///#define ACK_ADDR_STOP		REQUEST_ADDR_STOP
+
+/*
+*	utility functions
+*/
+static int handler_pre_kFunc(struct kprobe *p, struct pt_regs *regs);
+
+struct task_struct *findTaskByPid(pid_t pid);
+
+long v2p(struct mm_struct *pMM,unsigned long va,int *pStatus);
+
+long kv2p(unsigned long va,int *pStatus);
+
+long kFunc2v(char *funcName);
+
+struct vm_area_struct * getVMA(struct mm_struct *pMM,unsigned long va);
+
+pte_t * getPte(struct mm_struct *pMM,unsigned long va);
+
+int setVMAFlags(struct mm_struct *pMM,unsigned long va,int *pStatus,int flags);
+
+int setPageFlags(struct mm_struct *pMM,unsigned long va,int *pStatus,int flags);
+
+int getTaskInfo(struct task_struct *pTask, char *pData, int length);
+
+/*
+*	proc entry function
+*/
+int proc_write_pid(struct file *file,const char __user *buffer,unsigned long count,void * data);
+
+int proc_read_virtualAddr(char * page,char **start, off_t off, int count, int * eof,void * data);
+int proc_write_virtualAddr(struct file *file,const char *buffer,unsigned long count,void * data);
+
+int proc_write_ctl(struct file *file,const char *buffer,unsigned long count,void * data);
+
+int proc_read_signal(char * page,char **start, off_t off, int count, int * eof,void * data);
+int proc_write_signal(struct file *file,const char *buffer,unsigned long count,void * data);
+
+int proc_read_pa(char * page,char **start, off_t off, int count, int * eof,void * data);
+
+static int jforce_sig_info(int sig,struct siginfo *info,struct task_struct *t);
+
+
+#endif
+