1、源代码文件
#include <linux/init.h>
#include <linux/module.h>
#include <linux/slab.h>
static char *buf1 = NULL;
static char *buf2 = NULL;
static int __init alloc_mem_init(void)
{
int ret;
buf1 = kmalloc(100, GFP_KERNEL); // 在内核空间分配100个字节的内存
if(!buf1){
ret = -ENOMEM;
pr_err("allocate memory failed with kmalloc\n");
return ret;
}
memset(buf1, 0, 100); // 清空这100个字节的内存
strcpy(buf1, "--- kmalloc memory ok ---");// 把字符串拷贝到buf1中
printk(KERN_INFO "buf1 = %s\n", buf1);
buf2 = get_zeroed_page(GFP_KERNEL); // 分配一页内存并清空这一页内存
if(!buf2){
ret = -ENOMEM;
pr_err("allocate memory failed with get_zeroed_page\n");
return ret;
}
strcpy(buf2, "--- get zeroed page memory ok ---");// 把字符串拷贝到buf1中
printk(KERN_INFO "buf2 = %s\n", buf2);
printk(KERN_INFO "alloc_mem_init ok\n");
return 0;
}
static void __init alloc_mem_exit(void)
{
if(buf1){
kfree(buf1); // 释放分配的内存
buf1 = NULL;
}
if(buf2){
free_page(buf2); // 释放分配的一页内存
buf2 = NULL;
}
printk(KERN_INFO "alloc_mem_exit!\n");
}
MODULE_LICENSE("GPL v2");
MODULE_VERSION("v1.0");
MODULE_DESCRIPTION("Linux Kernel Memory Allocate Module");
MODULE_AUTHOR("David");
MODULE_ALIAS("Mermory Allocate Module");
module_init(alloc_mem_init);
module_exit(alloc_mem_exit);
2、Makefile
ifeq ($(KERNELRELEASE),)
KERNELDIR ?= /lib/modules/$(shell uname -r)/build
PWD := $(shell pwd)
all:
$(MAKE) -C $(KERNELDIR) M=$(PWD) modules
clean:
$(MAKE) -C $(KERNELDIR) M=$(PWD) clean
else
obj-m := alloc_mem.o
endif
3、验证测试
1)make
2)sudo insmod alloc_mem.ko
3)dmesg
buf1 = --- kmalloc memory ok ---
buf2 = --- get zeroed page memory ok ---
alloc_mem_init ok
4)sudo rmmod alloc_mem
5)dmesg
alloc_mem_exit!