secgateway/Platform/modules/cfgrcv/cfgrcv_kinit.c

#include <linux/module.h>
#include <linux/netlink.h>
#include <linux/netfilter.h>
#include <linux/ip.h>
#include <uapi/linux/netfilter_ipv4.h>
#include <uapi/linux/ip.h>
#include <net/netlink.h>
#include <net/net_namespace.h>

#include "../netlink_api/libnetlink_k.h"
#include "../../../Common/commuapinl.h"



struct netlinkk_cfg g_upmnlcfg;

int cfgnl_unicast(struct sk_buff *skb, u32 portid)
{
	return(commnl_unicast(g_upmnlcfg.sk, skb, portid));
}


#if 0
int nl_upm_data_ready(struct sk_buff *skb, struct nlmsghdr *nlh)
{
    void *payload;
    struct sk_buff *out_skb;
    void *out_payload;
    struct nlmsghdr *out_nlh;
    int payload_len; // with padding, but ok for echo 
    
	printk(KERN_DEBUG "nl_upm_data_ready() begin.\n");

    switch(nlh->nlmsg_type)
    {
        case COMMNMSG_POLICYCONF:/**/

			payload = nlmsg_data(nlh);
			payload_len = nlmsg_len(nlh);
			printk(KERN_INFO "Recievid: %s, From: %d\n", (char *)payload, nlh->nlmsg_pid);
			
			out_skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL); //分配足以存放默认大小的sk_buff
			if (!out_skb) goto failure;
			
			out_nlh = nlmsg_put(out_skb, 0, 0, COMMNMSG_POLICYCONF, payload_len, 0); //填充协议头数据
			if (!out_nlh) goto failure;
			
			//out_payload = nlmsg_data(out_nlh);
			//nla_put_u32(out_skb,PDELIVERY_DUMP_PKTNUM_ATTR,g_nlcfg->pkt_delev_num);
			
			commnl_unicast(g_upmnlcfg.sk, out_skb, nlh->nlmsg_pid);
            break;

        default:
            printk(KERN_INFO "libnetlink Unknow msgtype recieved!\n");
			return 0;
    }
		
	printk(KERN_DEBUG "nl_upm_data_ready() end.\n");
	
    return 0;
failure:
    printk(KERN_INFO " failed in fun dataready!\n");

	return 0;
}
#endif

int cfgrcv_debugfs(struct sk_buff *skb, struct nlmsghdr *nlh,struct netlink_ext_ack *extack)
{
	int ret = 0;

    printk(KERN_INFO "cfgrcv_debugfs, From pid: %d\n", nlh->nlmsg_pid);

#ifdef NLDEBUG_ACK_COOKIES
	ret = debugfs_pkt_num_stati(&g_upmnlcfg, nlh,extack);
#else
	ret = debugfs_pkt_num_stati_witisend(&g_upmnlcfg, nlh,extack);
#endif

	printk("*****************biduichengong***************\n");
	return ret;

}


int __init upm_init(void)
{
	int ret = -1;
	
	printk(KERN_CRIT "nl_upm initialed ok!\n");

	/*init for pdelivery module*/
	g_upmnlcfg.groups = 0;
	g_upmnlcfg.subscriptions = NETLINK_COMMCFG;/*创建配置处理通道*/
	ret = libnetlinkk_init_byproto(&g_upmnlcfg);
	if(ret < 0)
	{
		printk (KERN_CRIT "upm_init netlink init fail!.\n");
		return ret;
	}	

	/*do msg process register*/
	cfg_msgtype_register(COMMNMSG_CFG_DEBUGFS,cfgrcv_debugfs,NULL,NULL);
	
	return 0;
}

void __exit upm_exit(void)
{
    printk(KERN_CRIT "nl_upm existing...\n");
    libnetlinkk_exit(&g_upmnlcfg);

	return;
}

EXPORT_SYMBOL_GPL(cfgnl_unicast);

module_init(upm_init);
module_exit(upm_exit);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("a simple example for upm(user policy manage) netlink protocal family");
MODULE_AUTHOR("RSLjdkt");