advproc.c

/**
@file		advproc.c
@date		2014/07/31
@author		WangChunyan
@version	1.0.0
@brief		proc文件系统相关操作接口

@note
创建proc文件，删除proc文件，读写proc文件操作
*/

#include "advkill.h"
#include "advproc.h"
#include "advconfparse.h"
#include "advhash.h"

extern struct advconf_hashtable *g_advconf_hashtable;

#ifdef ADVKILL_CHECK_MEM
extern unsigned long long int g_calloc_times;
extern unsigned long long int g_calloc_size;
extern unsigned long long int g_free_times;
extern unsigned long long int g_free_size;
#endif

static struct proc_dir_entry *proc_dir = NULL;	///< proc文件中文件夹
static struct proc_dir_entry *proc_advkill_conf = NULL; ///< proc文件中文件
static int advkill_conf_index;  
static int advkill_conf_next;
static char *advkillconfdata = NULL; ///< 内核中去广告配置缓冲区地址

/**
从用户空间将数据读取到内核空间，解析后保存在哈希表中

@param filp 文件结构体指针，本接口对此没有操作，为内核函数参数格式
@param buff 用户空间数据地址
@param len 用户空间数据长度
@param data 本接口对此没有操作，为内核函数参数格式
@return 返回读取的长度
*/
static ssize_t advkill_conf_write( struct file *filp, const char __user *buff, unsigned long len, void *data);

/**
从内核空间将去广告配置内存拷贝到用户空间

@param page 用户空间接收数据的地址
@param start 本接口对此没有操作，为内核函数参数格式
@param off 本接口对此没有操作，为内核函数参数格式
@param count 本接口对此没有操作，为内核函数参数格式
@param eof 本接口对此没有操作，为内核函数参数格式
@param data 本接口对此没有操作，为内核函数参数格式
@return 返回写入的长度
*/
static int advkill_conf_read( char *page, char **start, off_t off, int count, int *eof, void *data);

/**
从用户空间将数据读取到内核空间，解析后保存在哈希表中

@param filp 文件结构体指针，本接口对此没有操作，为内核函数参数格式
@param buff 用户空间数据地址
@param len 用户空间数据长度
@param data 本接口对此没有操作，为内核函数参数格式
@return 返回读取的长度
*/
static ssize_t advkill_conf_write( struct file *filp, const char __user *buff, unsigned long len, void *data)
{
	int space_available = (MAX_ADVKILL_CONF_LEN-advkill_conf_index)+1;

	if (len > space_available) 
	{
		memset(advkillconfdata, 0, MAX_ADVKILL_CONF_LEN);
		advkill_conf_index = 0;
		advkill_conf_next = 0;
	}

	if (copy_from_user(&advkillconfdata[advkill_conf_index], buff, len )) 
	{
		return -EFAULT;
	}

#ifdef ADVKILL_PRINT_DEBUG_INFO
	ADV_PRINT_INFO(&advkillconfdata[advkill_conf_index]);
#endif

	//parse advconf
	parse_advconf_line_data(&advkillconfdata[advkill_conf_index], g_advconf_hashtable, HOST_HASH_SIZE);

	advkill_conf_index += len;
	advkillconfdata[advkill_conf_index-1] = '\0';

	return len;
}

/**
从内核空间将去广告配置内存拷贝到用户空间

@param page 用户空间接收数据的地址
@param start 本接口对此没有操作，为内核函数参数格式
@param off 本接口对此没有操作，为内核函数参数格式
@param count 本接口对此没有操作，为内核函数参数格式
@param eof 本接口对此没有操作，为内核函数参数格式
@param data 本接口对此没有操作，为内核函数参数格式
@return 返回写入的长度
*/
static int advkill_conf_read( char *page, char **start, off_t off, int count, int *eof, void *data)
{
	int len;

	if (off > 0) 
	{
		*eof = 1;
		return 0;
	}
	/* Wrap-around */
	if (advkill_conf_next >= advkill_conf_index) 
		advkill_conf_next = 0;
	len = sprintf(page, "%s\n", &advkillconfdata[advkill_conf_next]);
	advkill_conf_next += len;

	return len;
}

/**
创建需要用到的proc文件

@return 成功返回 RR_OK，失败返回 RR_FAIL。
*/
int create_proc_file(void)
{
	int ret = ADV_KILL_OK;
	//web config
	advkillconfdata = (char *)ADVKILL_CALLOC(1, MAX_ADVKILL_CONF_LEN);
	if (advkillconfdata == NULL)
		return ADV_KILL_FAIL;
	
	proc_dir = proc_mkdir(ADV_KILL_PROC_DIRECTORY, NULL);
	if (proc_dir == NULL) 
	{
		ret = -ENOMEM;
		printk(KERN_ERR "Couldn't create proc dir[/proc/%s]\n", ADV_KILL_PROC_DIRECTORY);
		goto exit_fail;
	}
	else
	{
		proc_advkill_conf = create_proc_entry(ADV_KILL_PROC_FILE, 0644, proc_dir);
		if (proc_advkill_conf == NULL)
		{
			printk(KERN_ERR "Couldn't create proc entry[/proc/%s/%s]\n", ADV_KILL_PROC_DIRECTORY, ADV_KILL_PROC_FILE);
			ret = -ENOMEM;
			goto exit_fail;
		}
		proc_advkill_conf->read_proc = advkill_conf_read;
		proc_advkill_conf->write_proc = advkill_conf_write;
	}

	return ret;

exit_fail:
	if (advkillconfdata != NULL)
	{
		ADVKILL_FREE(advkillconfdata, MAX_ADVKILL_CONF_LEN);
		advkillconfdata = NULL;
	}
	if (proc_dir != NULL)
	{
		remove_proc_entry(ADV_KILL_PROC_FILE, proc_dir);
		proc_dir = NULL;
	}
	return ret;	
}

/**
删除proc文件
*/
void destroy_proc_file(void)
{
	if (proc_advkill_conf != NULL)
	{
		remove_proc_entry(ADV_KILL_PROC_FILE, proc_dir);
		proc_advkill_conf = NULL;
	}
	if (proc_dir != NULL)
	{
		remove_proc_entry(ADV_KILL_PROC_DIRECTORY, NULL);
		proc_dir = NULL;
	}
	if (advkillconfdata != NULL)
	{
		ADVKILL_FREE(advkillconfdata, MAX_ADVKILL_CONF_LEN);
		advkillconfdata = NULL;
	}
}