aboutsummaryrefslogtreecommitdiffstats
path: root/judge/judgm_mod.c
blob: 092902a1ec06320ccd300a8d86181ee94b166e9b (plain) (blame)
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
#include<linux/module.h>
#include<linux/kernel.h>
#include<linux/kdev_t.h>
#include<linux/device.h>
#include<linux/cdev.h>
#include<linux/fs.h>

#include"judgm_mod.h"
#include"judgm.h"
#include"judgx_com.h"

static int __init mod_init(){
    alloc_chrdev_region(&mod_dev,0,1,"judgm");
    mod_class = class_create(THIS_MODULE,"chardev");
    device_create(mod_class,NULL,mod_dev,NULL,"judgm");
    cdev_init(&mod_cdev,&mod_fops);
    cdev_add(&mod_cdev,mod_dev,1);

    judgm_proc_init();
    judgm_security_hook();
    judgm_syscall_hook();

    pr_alert("judgm:Init\n");
    return 0;
}
static void __exit mod_exit(){
    cdev_del(&mod_cdev);
    device_destroy(mod_class,mod_dev);
    class_destroy(mod_class);
    unregister_chrdev_region(mod_dev,1);

    judgm_proc_exit();
    judgm_syscall_unhook();
    judgm_security_unhook();

    pr_alert("judgm:Exit\n");
}
module_init(mod_init);
module_exit(mod_exit);
MODULE_LICENSE("GPL");

static long mod_ioctl(struct file *file,unsigned int cmd,unsigned long arg){
    int ret;

    ret = -1;
    switch(cmd){
        case IOCTL_PROC_ADD:
            ret = judgm_proc_add(arg);
            break;
        case IOCTL_PROC_GET:
            ret = judgm_proc_get(arg); 
            break;
        case IOCTL_PROC_KILL:
            ret = judgm_proc_kill(arg); 
            break;
        case IOCTL_PROC_DEL:
            ret = judgm_proc_del(arg);
            break;
    }

    return ret;
}