Files
2017-10-21 22:02:32 +00:00

68 lines
1.4 KiB
C

#include <linux/module.h>
#include <linux/init.h>
#include <linux/fs.h>
#include <linux/uaccess.h>
/* read write buffer */
static char wbuf[] = "this is a kernel read write test";
static char rbuf[1024];
#define FILE_NAME "/tmp/vfs_rw_test"
/*
* 在内核中一般不容易生成用户空间的指针
* 或者不方便独立使用用户空间内存
* 而vfs_read,vfs_write的参数buffer都是用户空间指针
* 所以需要调用set_fs目的是为了让内核改变对内存地址检查的处理方式
* set_fs参数只有两个USER_DS和KERNEL_DS
* USER_DS代表用户空间,set_fs
* KERNEL_DS代表内核空间
*/
static int vfs_rw_init(void)
{
struct file *fp;
mm_segment_t fs;
loff_t pos;
printk("%s, %d\n", __FUNCTION__, __LINE__);
/* open file */
fp = filp_open(FILE_NAME, O_RDWR | O_CREAT, 0644);
if (IS_ERR(fp))
{
printk("create file error\n");
return -1;
}
/* write file */
/* get current fs*/
fs = get_fs();
/* 即对内核空间地址检查并做变换 */
set_fs(KERNEL_DS);
pos = 0;
printk("=> %s\n", wbuf);
vfs_write(fp, wbuf, sizeof(wbuf), &pos);
/* read file */
pos = 0;
vfs_read(fp, rbuf, sizeof(wbuf), &pos);
printk("<= %s\n", rbuf);
/* close file */
filp_close(fp, NULL);
set_fs(fs);
return 0;
}
static void vfs_rw_exit(void)
{
printk("%s, %d\n", __FUNCTION__, __LINE__);
}
module_init(vfs_rw_init);
module_exit(vfs_rw_exit);
MODULE_LICENSE("GPL");