483 lines
9.8 KiB
C
483 lines
9.8 KiB
C
#ifdef __KERNEL__
|
||
|
||
#include <linux/init.h>
|
||
#include <linux/module.h>
|
||
#include <linux/fs.h>
|
||
#include <linux/gfp.h>
|
||
#include <linux/mm.h>
|
||
#include <linux/device.h>
|
||
#include <linux/in.h>
|
||
#include <linux/proc_fs.h>
|
||
#include <linux/seq_file.h>
|
||
#include <linux/spinlock.h>
|
||
|
||
#include "../proc_api/proc_api.h"
|
||
#include "../../common/cJSON/cJSON.h"
|
||
#include "obj_api.h"
|
||
|
||
/**
|
||
* @def OBJ_DEV_NAME
|
||
* @brief
|
||
*/
|
||
#define OBJ_DEV_NAME ("isg_objs") ///< 设备节点名称
|
||
|
||
/**
|
||
* @def VERSION
|
||
* @brief
|
||
*/
|
||
#define VERSION ("obj0.0.0.2")
|
||
|
||
/**
|
||
* @def SHM_MEM_SIZE
|
||
* @brief
|
||
*/
|
||
#define SHM_MEM_SIZE (1024 * 1024) ///< 1M字节共享内存
|
||
|
||
/**
|
||
* @var typedef {anonOBJECT_DRV} OBJECT_DRV
|
||
* @brief
|
||
*/
|
||
/**
|
||
* @var typedef {anonOBJECT_DRV} POBJECT_DRV
|
||
* @brief
|
||
*/
|
||
/**
|
||
* @struct {anonOBJECT_DRV}
|
||
* @brief
|
||
*/
|
||
typedef struct {
|
||
struct class *dev_class; ///< 设备类，用于自动在 /dev 下创建文件节点
|
||
int dev_major; ///< 设备主版本号
|
||
atomic_t ref_count; ///< 驱动程序打开计数
|
||
unsigned int mem_size; ///< 共享内存大小
|
||
unsigned char *mem_buf; ///< 共享内存指针
|
||
} OBJECT_DRV, *POBJECT_DRV;
|
||
|
||
/**
|
||
* @var g_obj_device
|
||
* @brief
|
||
* @see POBJECT_DRV
|
||
*/
|
||
static POBJECT_DRV g_obj_device = NULL;
|
||
DEFINE_RWLOCK(g_obj_lock);
|
||
|
||
/**
|
||
* @var g_obj_data
|
||
* @brief
|
||
* @see PCMHI_OBJECT
|
||
*/
|
||
PCMHI_OBJECT g_obj_data = NULL;
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @param buf
|
||
* @param size
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
void obj_upgrade_dev_buffer(unsigned char *buf, unsigned int size)
|
||
{
|
||
if(buf && size < g_obj_device->mem_size - 1) {
|
||
memcpy(g_obj_device->mem_buf, buf, size);
|
||
g_obj_device->mem_buf[size] = 0;
|
||
} else {
|
||
printk(KERN_ERR "Buffer = %p, size = %u\n", buf, size);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @brief 根据对象名称获取对象
|
||
*
|
||
* @param name: 对象名称
|
||
* @param type: 对象类型
|
||
*
|
||
* @return: 对象指针 或 NULL 空指针
|
||
* @date 2019/08/07
|
||
*/
|
||
PCMHI_OBJECT get_object(const char *name, int type)
|
||
{
|
||
PCMHI_OBJECT obj, tmp;
|
||
PSERVER_OBJ_CONTENT req = NULL;
|
||
|
||
HASH_FIND_STR(g_obj_data, name, obj);
|
||
|
||
if(!obj) {
|
||
return NULL;
|
||
}
|
||
|
||
if(type >= OBJ_TYPE_ADDR && type < OBJ_TYPE_MAX) {
|
||
if(obj->type != type) {
|
||
return NULL;
|
||
}
|
||
}
|
||
|
||
return obj;
|
||
}
|
||
|
||
/**
|
||
* @brief 判断当前服务是否在对象中
|
||
*
|
||
* @param obj: 对象指针
|
||
* @param port: 协议端口
|
||
* @param proType: 协议类型
|
||
*
|
||
* @return: 存在：0， 不存在：1
|
||
* @date 2019/08/07
|
||
*/
|
||
int item_belong_server_obj(PCMHI_OBJECT obj, __be16 port, int proType)
|
||
{
|
||
PSERVER_OBJ_CONTENT req = NULL;
|
||
|
||
if(!obj) {
|
||
return 0;
|
||
}
|
||
|
||
list_for_each_entry(req, &obj->data->content, list) {
|
||
if(proType == req->pro_type && port >= req->min_port && port <= req->max_port) {
|
||
return 1;
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
/**
|
||
* @brief 判断当前时间是否在对象中
|
||
*
|
||
* @param obj: 对象指针
|
||
* @param tm: 时间戳
|
||
*
|
||
* @return: 存在：0， 不存在：1
|
||
* @date 2019/08/07
|
||
*/
|
||
int item_belong_dt_obj(PCMHI_OBJECT obj, unsigned long tm)
|
||
{
|
||
PDT_OBJ_CONTENT req = NULL;
|
||
|
||
if(!obj) {
|
||
return 0;
|
||
}
|
||
|
||
list_for_each_entry(req, &obj->data->content, list) {
|
||
if(MODE_NONE == req->rep_mode && tm >= req->min_time && tm <= req->max_time) {
|
||
return 1;
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
/**
|
||
* @brief 判断IPv4地址是否在对象中
|
||
*
|
||
* @param obj: 对象指针
|
||
* @param addr: IPv4地址
|
||
*
|
||
* @return: 存在：0， 不存在：1
|
||
* @date 2019/08/07
|
||
*/
|
||
int ipv4_belong_addr_obj(PCMHI_OBJECT obj, __be32 addr)
|
||
{
|
||
|
||
}
|
||
|
||
/**
|
||
* @brief 判断IPv6地址是否在对象中
|
||
*
|
||
* @param obj: 对象指针
|
||
* @param addr: IPv6地址
|
||
*
|
||
* @return: 存在：0， 不存在：1
|
||
* @date 2019/08/07
|
||
*/
|
||
int ipv6_belong_addr_obj(PCMHI_OBJECT obj, unsigned char addr[16])
|
||
{
|
||
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static int obj_proc_init(void)
|
||
{
|
||
return server_proc_init();
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static int obj_proc_uninit(void)
|
||
{
|
||
return server_proc_uninit();
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
const char *obj_version(void)
|
||
{
|
||
return VERSION;
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
PCMHI_OBJECT new_object(void)
|
||
{
|
||
PCMHI_OBJECT obj = (PCMHI_OBJECT)kmalloc(sizeof(CMHI_OBJECT), GFP_KERNEL);
|
||
|
||
if(obj == NULL) {
|
||
printk(KERN_ERR "Malloc CMHI_OBJECT Error\n");
|
||
} else {
|
||
memset(obj, 0, sizeof(CMHI_OBJECT));
|
||
rwlock_init(&obj->lock);
|
||
atomic_set(&obj->ref_count, 0);
|
||
}
|
||
|
||
return obj;
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @param pObj
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
void free_object(PCMHI_OBJECT pObj)
|
||
{
|
||
if(pObj) {
|
||
kfree(pObj);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
void object_init(void)
|
||
{
|
||
PCMHI_OBJECT pObj = create_server_object("list1");
|
||
|
||
if(!pObj) {
|
||
printk(KERN_ERR "%s(%d): Create Object Error\n", __FUNCTION__, __LINE__);
|
||
return;
|
||
}
|
||
|
||
HASH_ADD_STR(g_obj_data, name, pObj);
|
||
|
||
if(add_server_obj_data(pObj, 10000, 20000, IPPROTO_IP) != 0) {
|
||
printk(KERN_ERR "%s(%d): Add Server Object error\n", __FUNCTION__, __LINE__);
|
||
return;
|
||
}
|
||
|
||
if(add_server_obj_data(pObj, 80, 80, IPPROTO_TCP) != 0) {
|
||
printk(KERN_ERR "%s(%d): Add Server Object error\n", __FUNCTION__, __LINE__);
|
||
return;
|
||
}
|
||
|
||
if(add_server_obj_data(pObj, 443, 443, IPPROTO_UDP) != 0) {
|
||
printk(KERN_ERR "%s(%d): Add Server Object error\n", __FUNCTION__, __LINE__);
|
||
return;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
void object_uninit(void)
|
||
{
|
||
PCMHI_OBJECT obj, tmp;
|
||
|
||
HASH_ITER(hh, g_obj_data, obj, tmp) {
|
||
HASH_DEL(g_obj_data, obj);
|
||
cleanup_server_object(obj);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @param inode
|
||
* @param filp
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static int obj_open(struct inode *inode, struct file *filp)
|
||
{
|
||
// 增加引用计数
|
||
atomic_inc(&g_obj_device->ref_count);
|
||
return 0;
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @param inode
|
||
* @param filp
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static int obj_release(struct inode *inode, struct file *filp)
|
||
{
|
||
// 减少引用计数
|
||
atomic_dec(&g_obj_device->ref_count);
|
||
return 0;
|
||
}
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @param fd
|
||
* @param vma
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
int obj_mmap(struct file *fd, struct vm_area_struct *vma)
|
||
{
|
||
// 重新映射内存页表
|
||
int ret = remap_pfn_range(vma, vma->vm_start,
|
||
virt_to_phys(g_obj_device->mem_buf) >> PAGE_SHIFT,
|
||
g_obj_device->mem_size,
|
||
PAGE_SHARED);
|
||
return ret;
|
||
}
|
||
|
||
/**
|
||
* @var obj_fops
|
||
* @brief
|
||
*/
|
||
static struct file_operations obj_fops = {
|
||
.open = obj_open,
|
||
.release = obj_release,
|
||
.mmap = obj_mmap,
|
||
};
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static int object_module_init(void)
|
||
{
|
||
struct page *p = NULL;
|
||
dev_t dev = 0;
|
||
|
||
printk(KERN_INFO "Hello ISG objects manager version: %s\n", VERSION);
|
||
|
||
// 自动分配设备版本号
|
||
if(alloc_chrdev_region(&dev, 0, 1, OBJ_DEV_NAME) != 0) {
|
||
printk(KERN_ERR "Alloc driver dev id error\n");
|
||
return -ENODEV;
|
||
}
|
||
|
||
// 分配驱动程序结构内存
|
||
g_obj_device = (POBJECT_DRV)kmalloc(sizeof(OBJECT_DRV), GFP_KERNEL);
|
||
|
||
if(g_obj_device == NULL) {
|
||
printk(KERN_ERR "Create isg objects error\n");
|
||
return -ENOMEM;
|
||
}
|
||
|
||
memset(g_obj_device, 0, sizeof(OBJECT_DRV));
|
||
g_obj_device->dev_major = MAJOR(dev); // 保存驱动版本号
|
||
atomic_set(&g_obj_device->ref_count, 0);
|
||
|
||
// 初始化共享内存
|
||
g_obj_device->mem_size = PAGE_ALIGN(SHM_MEM_SIZE);
|
||
g_obj_device->mem_buf = (unsigned char *)__get_free_pages(GFP_KERNEL, get_order(g_obj_device->mem_size));
|
||
|
||
if(g_obj_device->mem_buf == NULL) {
|
||
printk(KERN_ERR "Malloc %d pages error\n", get_order(g_obj_device->mem_size));
|
||
kfree(g_obj_device);
|
||
return -ENOMEM;
|
||
}
|
||
|
||
for(p = virt_to_page(g_obj_device->mem_buf);
|
||
p < virt_to_page(g_obj_device->mem_buf + g_obj_device->mem_size);
|
||
p++) {
|
||
SetPageReserved(virt_to_page(p));
|
||
}
|
||
|
||
memset(g_obj_device->mem_buf, 0, g_obj_device->mem_size);
|
||
// 注册字符设备
|
||
register_chrdev(g_obj_device->dev_major, OBJ_DEV_NAME, &obj_fops);
|
||
// 创建字符设备类
|
||
g_obj_device->dev_class = class_create(THIS_MODULE, "obj");
|
||
// 创建字符设备节点
|
||
device_create(g_obj_device->dev_class, NULL, MKDEV(g_obj_device->dev_major, 0),
|
||
NULL, "isg_objs/dev%d", 0);
|
||
|
||
// 初始化 proc 接口
|
||
obj_proc_init();
|
||
object_init();
|
||
|
||
return 0;
|
||
}
|
||
module_init(object_module_init);
|
||
|
||
/**
|
||
* @brief
|
||
*
|
||
* @return
|
||
* @date 2019/08/07
|
||
*/
|
||
static void __exit object_module_exit(void)
|
||
{
|
||
struct page *p = NULL;
|
||
|
||
printk(KERN_INFO "Bye ISG objects manager version: %s\n", VERSION);
|
||
|
||
obj_proc_uninit();
|
||
object_uninit();
|
||
|
||
if(atomic_read(&g_obj_device->ref_count) != 0) {
|
||
printk(KERN_ERR "Device used, please close it first.\n");
|
||
return;
|
||
}
|
||
|
||
if(g_obj_device->mem_buf) {
|
||
for(p = virt_to_page(g_obj_device->mem_buf);
|
||
p < virt_to_page(g_obj_device->mem_buf + g_obj_device->mem_size);
|
||
p++) {
|
||
ClearPageReserved(virt_to_page(g_obj_device->mem_buf));
|
||
}
|
||
|
||
free_pages((unsigned long)g_obj_device->mem_buf, get_order(g_obj_device->mem_size));
|
||
}
|
||
|
||
device_destroy(g_obj_device->dev_class, MKDEV(g_obj_device->dev_major, 0));
|
||
class_destroy(g_obj_device->dev_class);
|
||
unregister_chrdev(g_obj_device->dev_major, OBJ_DEV_NAME);
|
||
|
||
kfree(g_obj_device);
|
||
}
|
||
module_exit(object_module_exit);
|
||
|
||
MODULE_LICENSE("Dual BSD/GPL");
|
||
#endif // __KERNEL__
|