161e121047645122c47714fcda684d0ee67f444afWon Kang/* 261e121047645122c47714fcda684d0ee67f444afWon Kang * Copyright (c) 2012 GCT Semiconductor, Inc. All rights reserved. 361e121047645122c47714fcda684d0ee67f444afWon Kang * 461e121047645122c47714fcda684d0ee67f444afWon Kang * This software is licensed under the terms of the GNU General Public 561e121047645122c47714fcda684d0ee67f444afWon Kang * License version 2, as published by the Free Software Foundation, and 661e121047645122c47714fcda684d0ee67f444afWon Kang * may be copied, distributed, and modified under those terms. 761e121047645122c47714fcda684d0ee67f444afWon Kang * 861e121047645122c47714fcda684d0ee67f444afWon Kang * This program is distributed in the hope that it will be useful, 961e121047645122c47714fcda684d0ee67f444afWon Kang * but WITHOUT ANY WARRANTY; without even the implied warranty of 1061e121047645122c47714fcda684d0ee67f444afWon Kang * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 1161e121047645122c47714fcda684d0ee67f444afWon Kang * GNU General Public License for more details. 1261e121047645122c47714fcda684d0ee67f444afWon Kang */ 1361e121047645122c47714fcda684d0ee67f444afWon Kang 140ec473b5267d01762b610d57e168b0401a137f3aJoe Perches#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 150ec473b5267d01762b610d57e168b0401a137f3aJoe Perches 1661e121047645122c47714fcda684d0ee67f444afWon Kang#include <linux/export.h> 1761e121047645122c47714fcda684d0ee67f444afWon Kang#include <linux/etherdevice.h> 1861e121047645122c47714fcda684d0ee67f444afWon Kang#include <linux/netlink.h> 1961e121047645122c47714fcda684d0ee67f444afWon Kang#include <asm/byteorder.h> 2061e121047645122c47714fcda684d0ee67f444afWon Kang#include <net/sock.h> 2161e121047645122c47714fcda684d0ee67f444afWon Kang 2261e121047645122c47714fcda684d0ee67f444afWon Kang#include "netlink_k.h" 2361e121047645122c47714fcda684d0ee67f444afWon Kang 2461e121047645122c47714fcda684d0ee67f444afWon Kang#if defined(DEFINE_MUTEX) 2561e121047645122c47714fcda684d0ee67f444afWon Kangstatic DEFINE_MUTEX(netlink_mutex); 2661e121047645122c47714fcda684d0ee67f444afWon Kang#else 2761e121047645122c47714fcda684d0ee67f444afWon Kangstatic struct semaphore netlink_mutex; 2861e121047645122c47714fcda684d0ee67f444afWon Kang#define mutex_lock(x) down(x) 2961e121047645122c47714fcda684d0ee67f444afWon Kang#define mutex_unlock(x) up(x) 3061e121047645122c47714fcda684d0ee67f444afWon Kang#endif 3161e121047645122c47714fcda684d0ee67f444afWon Kang 3261e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_MAX_GROUP 30 3361e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_IFINDEX_LEN sizeof(int) 3461e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_NLMSG_SPACE(len) (NLMSG_SPACE(len) + ND_IFINDEX_LEN) 35a67083725cfd56021eb80af8fc7c3f282ef270d2Daniel Ngu#define ND_NLMSG_DATA(nlh) ((void *)((char *)NLMSG_DATA(nlh) + \ 36a67083725cfd56021eb80af8fc7c3f282ef270d2Daniel Ngu ND_IFINDEX_LEN)) 3761e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_NLMSG_S_LEN(len) (len+ND_IFINDEX_LEN) 3861e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_NLMSG_R_LEN(nlh) (nlh->nlmsg_len-ND_IFINDEX_LEN) 3961e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_NLMSG_IFIDX(nlh) NLMSG_DATA(nlh) 4061e121047645122c47714fcda684d0ee67f444afWon Kang#define ND_MAX_MSG_LEN (1024 * 32) 4161e121047645122c47714fcda684d0ee67f444afWon Kang 4261e121047645122c47714fcda684d0ee67f444afWon Kangstatic void (*rcv_cb)(struct net_device *dev, u16 type, void *msg, int len); 4361e121047645122c47714fcda684d0ee67f444afWon Kang 4461e121047645122c47714fcda684d0ee67f444afWon Kangstatic void netlink_rcv_cb(struct sk_buff *skb) 4561e121047645122c47714fcda684d0ee67f444afWon Kang{ 4661e121047645122c47714fcda684d0ee67f444afWon Kang struct nlmsghdr *nlh; 4761e121047645122c47714fcda684d0ee67f444afWon Kang struct net_device *dev; 4861e121047645122c47714fcda684d0ee67f444afWon Kang u32 mlen; 4961e121047645122c47714fcda684d0ee67f444afWon Kang void *msg; 5061e121047645122c47714fcda684d0ee67f444afWon Kang int ifindex; 5161e121047645122c47714fcda684d0ee67f444afWon Kang 5261e121047645122c47714fcda684d0ee67f444afWon Kang if (!rcv_cb) { 530ec473b5267d01762b610d57e168b0401a137f3aJoe Perches pr_err("nl cb - unregistered\n"); 5461e121047645122c47714fcda684d0ee67f444afWon Kang return; 5561e121047645122c47714fcda684d0ee67f444afWon Kang } 5661e121047645122c47714fcda684d0ee67f444afWon Kang 5758419939bc2c4d33e367ecfe87269b4951b8f557Himangi Saraogi if (skb->len < NLMSG_HDRLEN) { 580ec473b5267d01762b610d57e168b0401a137f3aJoe Perches pr_err("nl cb - invalid skb length\n"); 5961e121047645122c47714fcda684d0ee67f444afWon Kang return; 6061e121047645122c47714fcda684d0ee67f444afWon Kang } 6161e121047645122c47714fcda684d0ee67f444afWon Kang 6261e121047645122c47714fcda684d0ee67f444afWon Kang nlh = (struct nlmsghdr *)skb->data; 6361e121047645122c47714fcda684d0ee67f444afWon Kang 6461e121047645122c47714fcda684d0ee67f444afWon Kang if (skb->len < nlh->nlmsg_len || nlh->nlmsg_len > ND_MAX_MSG_LEN) { 650ec473b5267d01762b610d57e168b0401a137f3aJoe Perches pr_err("nl cb - invalid length (%d,%d)\n", 6661e121047645122c47714fcda684d0ee67f444afWon Kang skb->len, nlh->nlmsg_len); 6761e121047645122c47714fcda684d0ee67f444afWon Kang return; 6861e121047645122c47714fcda684d0ee67f444afWon Kang } 6961e121047645122c47714fcda684d0ee67f444afWon Kang 7061e121047645122c47714fcda684d0ee67f444afWon Kang memcpy(&ifindex, ND_NLMSG_IFIDX(nlh), ND_IFINDEX_LEN); 7161e121047645122c47714fcda684d0ee67f444afWon Kang msg = ND_NLMSG_DATA(nlh); 7261e121047645122c47714fcda684d0ee67f444afWon Kang mlen = ND_NLMSG_R_LEN(nlh); 7361e121047645122c47714fcda684d0ee67f444afWon Kang 7461e121047645122c47714fcda684d0ee67f444afWon Kang dev = dev_get_by_index(&init_net, ifindex); 7561e121047645122c47714fcda684d0ee67f444afWon Kang if (dev) { 7661e121047645122c47714fcda684d0ee67f444afWon Kang rcv_cb(dev, nlh->nlmsg_type, msg, mlen); 7761e121047645122c47714fcda684d0ee67f444afWon Kang dev_put(dev); 7861e121047645122c47714fcda684d0ee67f444afWon Kang } else { 790ec473b5267d01762b610d57e168b0401a137f3aJoe Perches pr_err("nl cb - dev (%d) not found\n", ifindex); 8061e121047645122c47714fcda684d0ee67f444afWon Kang } 8161e121047645122c47714fcda684d0ee67f444afWon Kang} 8261e121047645122c47714fcda684d0ee67f444afWon Kang 8361e121047645122c47714fcda684d0ee67f444afWon Kangstatic void netlink_rcv(struct sk_buff *skb) 8461e121047645122c47714fcda684d0ee67f444afWon Kang{ 8561e121047645122c47714fcda684d0ee67f444afWon Kang mutex_lock(&netlink_mutex); 8661e121047645122c47714fcda684d0ee67f444afWon Kang netlink_rcv_cb(skb); 8761e121047645122c47714fcda684d0ee67f444afWon Kang mutex_unlock(&netlink_mutex); 8861e121047645122c47714fcda684d0ee67f444afWon Kang} 8961e121047645122c47714fcda684d0ee67f444afWon Kang 9061e121047645122c47714fcda684d0ee67f444afWon Kangstruct sock *netlink_init(int unit, 9161e121047645122c47714fcda684d0ee67f444afWon Kang void (*cb)(struct net_device *dev, u16 type, void *msg, int len)) 9261e121047645122c47714fcda684d0ee67f444afWon Kang{ 9361e121047645122c47714fcda684d0ee67f444afWon Kang struct sock *sock; 9461e121047645122c47714fcda684d0ee67f444afWon Kang struct netlink_kernel_cfg cfg = { 9561e121047645122c47714fcda684d0ee67f444afWon Kang .input = netlink_rcv, 9661e121047645122c47714fcda684d0ee67f444afWon Kang }; 9761e121047645122c47714fcda684d0ee67f444afWon Kang 9861e121047645122c47714fcda684d0ee67f444afWon Kang#if !defined(DEFINE_MUTEX) 9961e121047645122c47714fcda684d0ee67f444afWon Kang init_MUTEX(&netlink_mutex); 10061e121047645122c47714fcda684d0ee67f444afWon Kang#endif 10161e121047645122c47714fcda684d0ee67f444afWon Kang 10261e121047645122c47714fcda684d0ee67f444afWon Kang sock = netlink_kernel_create(&init_net, unit, &cfg); 10361e121047645122c47714fcda684d0ee67f444afWon Kang 10461e121047645122c47714fcda684d0ee67f444afWon Kang if (sock) 10561e121047645122c47714fcda684d0ee67f444afWon Kang rcv_cb = cb; 10661e121047645122c47714fcda684d0ee67f444afWon Kang 10761e121047645122c47714fcda684d0ee67f444afWon Kang return sock; 10861e121047645122c47714fcda684d0ee67f444afWon Kang} 10961e121047645122c47714fcda684d0ee67f444afWon Kang 11061e121047645122c47714fcda684d0ee67f444afWon Kangvoid netlink_exit(struct sock *sock) 11161e121047645122c47714fcda684d0ee67f444afWon Kang{ 11261e121047645122c47714fcda684d0ee67f444afWon Kang sock_release(sock->sk_socket); 11361e121047645122c47714fcda684d0ee67f444afWon Kang} 11461e121047645122c47714fcda684d0ee67f444afWon Kang 11561e121047645122c47714fcda684d0ee67f444afWon Kangint netlink_send(struct sock *sock, int group, u16 type, void *msg, int len) 11661e121047645122c47714fcda684d0ee67f444afWon Kang{ 11761e121047645122c47714fcda684d0ee67f444afWon Kang static u32 seq; 11861e121047645122c47714fcda684d0ee67f444afWon Kang struct sk_buff *skb = NULL; 11961e121047645122c47714fcda684d0ee67f444afWon Kang struct nlmsghdr *nlh; 12061e121047645122c47714fcda684d0ee67f444afWon Kang int ret = 0; 12161e121047645122c47714fcda684d0ee67f444afWon Kang 12261e121047645122c47714fcda684d0ee67f444afWon Kang if (group > ND_MAX_GROUP) 12361e121047645122c47714fcda684d0ee67f444afWon Kang return -EINVAL; 12461e121047645122c47714fcda684d0ee67f444afWon Kang 12561e121047645122c47714fcda684d0ee67f444afWon Kang if (!netlink_has_listeners(sock, group+1)) 12661e121047645122c47714fcda684d0ee67f444afWon Kang return -ESRCH; 12761e121047645122c47714fcda684d0ee67f444afWon Kang 12861e121047645122c47714fcda684d0ee67f444afWon Kang skb = alloc_skb(NLMSG_SPACE(len), GFP_ATOMIC); 12961e121047645122c47714fcda684d0ee67f444afWon Kang if (!skb) 13061e121047645122c47714fcda684d0ee67f444afWon Kang return -ENOMEM; 13161e121047645122c47714fcda684d0ee67f444afWon Kang 13261e121047645122c47714fcda684d0ee67f444afWon Kang seq++; 13361e121047645122c47714fcda684d0ee67f444afWon Kang 13461e121047645122c47714fcda684d0ee67f444afWon Kang nlh = nlmsg_put(skb, 0, seq, type, len, 0); 13561e121047645122c47714fcda684d0ee67f444afWon Kang memcpy(NLMSG_DATA(nlh), msg, len); 13661e121047645122c47714fcda684d0ee67f444afWon Kang NETLINK_CB(skb).portid = 0; 13761e121047645122c47714fcda684d0ee67f444afWon Kang NETLINK_CB(skb).dst_group = 0; 13861e121047645122c47714fcda684d0ee67f444afWon Kang 13961e121047645122c47714fcda684d0ee67f444afWon Kang ret = netlink_broadcast(sock, skb, 0, group+1, GFP_ATOMIC); 14061e121047645122c47714fcda684d0ee67f444afWon Kang if (!ret) 14161e121047645122c47714fcda684d0ee67f444afWon Kang return len; 14261e121047645122c47714fcda684d0ee67f444afWon Kang 14361e121047645122c47714fcda684d0ee67f444afWon Kang if (ret != -ESRCH) 1440ec473b5267d01762b610d57e168b0401a137f3aJoe Perches pr_err("nl broadcast g=%d, t=%d, l=%d, r=%d\n", 14561e121047645122c47714fcda684d0ee67f444afWon Kang group, type, len, ret); 14661e121047645122c47714fcda684d0ee67f444afWon Kang else if (netlink_has_listeners(sock, group+1)) 14761e121047645122c47714fcda684d0ee67f444afWon Kang return -EAGAIN; 14861e121047645122c47714fcda684d0ee67f444afWon Kang 14961e121047645122c47714fcda684d0ee67f444afWon Kang return ret; 15061e121047645122c47714fcda684d0ee67f444afWon Kang} 151