1b2cbae2cSRoland Dreier /* 2b2cbae2cSRoland Dreier * Copyright (c) 2010 Voltaire Inc. All rights reserved. 3b2cbae2cSRoland Dreier * 4b2cbae2cSRoland Dreier * This software is available to you under a choice of one of two 5b2cbae2cSRoland Dreier * licenses. You may choose to be licensed under the terms of the GNU 6b2cbae2cSRoland Dreier * General Public License (GPL) Version 2, available from the file 7b2cbae2cSRoland Dreier * COPYING in the main directory of this source tree, or the 8b2cbae2cSRoland Dreier * OpenIB.org BSD license below: 9b2cbae2cSRoland Dreier * 10b2cbae2cSRoland Dreier * Redistribution and use in source and binary forms, with or 11b2cbae2cSRoland Dreier * without modification, are permitted provided that the following 12b2cbae2cSRoland Dreier * conditions are met: 13b2cbae2cSRoland Dreier * 14b2cbae2cSRoland Dreier * - Redistributions of source code must retain the above 15b2cbae2cSRoland Dreier * copyright notice, this list of conditions and the following 16b2cbae2cSRoland Dreier * disclaimer. 17b2cbae2cSRoland Dreier * 18b2cbae2cSRoland Dreier * - Redistributions in binary form must reproduce the above 19b2cbae2cSRoland Dreier * copyright notice, this list of conditions and the following 20b2cbae2cSRoland Dreier * disclaimer in the documentation and/or other materials 21b2cbae2cSRoland Dreier * provided with the distribution. 22b2cbae2cSRoland Dreier * 23b2cbae2cSRoland Dreier * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 24b2cbae2cSRoland Dreier * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 25b2cbae2cSRoland Dreier * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 26b2cbae2cSRoland Dreier * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS 27b2cbae2cSRoland Dreier * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN 28b2cbae2cSRoland Dreier * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN 29b2cbae2cSRoland Dreier * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 30b2cbae2cSRoland Dreier * SOFTWARE. 31b2cbae2cSRoland Dreier */ 32b2cbae2cSRoland Dreier 33b2cbae2cSRoland Dreier #define pr_fmt(fmt) "%s:%s: " fmt, KBUILD_MODNAME, __func__ 34b2cbae2cSRoland Dreier 35b108d976SPaul Gortmaker #include <linux/export.h> 36b2cbae2cSRoland Dreier #include <net/netlink.h> 37b2cbae2cSRoland Dreier #include <net/net_namespace.h> 38b2cbae2cSRoland Dreier #include <net/sock.h> 39b2cbae2cSRoland Dreier #include <rdma/rdma_netlink.h> 40b2cbae2cSRoland Dreier 41b2cbae2cSRoland Dreier struct ibnl_client { 42b2cbae2cSRoland Dreier struct list_head list; 43b2cbae2cSRoland Dreier int index; 44b2cbae2cSRoland Dreier int nops; 45b2cbae2cSRoland Dreier const struct ibnl_client_cbs *cb_table; 46b2cbae2cSRoland Dreier }; 47b2cbae2cSRoland Dreier 48b2cbae2cSRoland Dreier static DEFINE_MUTEX(ibnl_mutex); 49b2cbae2cSRoland Dreier static struct sock *nls; 50b2cbae2cSRoland Dreier static LIST_HEAD(client_list); 51b2cbae2cSRoland Dreier 52bc10ed7dSKaike Wan int ibnl_chk_listeners(unsigned int group) 53bc10ed7dSKaike Wan { 54bc10ed7dSKaike Wan if (netlink_has_listeners(nls, group) == 0) 55bc10ed7dSKaike Wan return -1; 56bc10ed7dSKaike Wan return 0; 57bc10ed7dSKaike Wan } 58bc10ed7dSKaike Wan EXPORT_SYMBOL(ibnl_chk_listeners); 59bc10ed7dSKaike Wan 60b2cbae2cSRoland Dreier int ibnl_add_client(int index, int nops, 61b2cbae2cSRoland Dreier const struct ibnl_client_cbs cb_table[]) 62b2cbae2cSRoland Dreier { 63b2cbae2cSRoland Dreier struct ibnl_client *cur; 64b2cbae2cSRoland Dreier struct ibnl_client *nl_client; 65b2cbae2cSRoland Dreier 66b2cbae2cSRoland Dreier nl_client = kmalloc(sizeof *nl_client, GFP_KERNEL); 67b2cbae2cSRoland Dreier if (!nl_client) 68b2cbae2cSRoland Dreier return -ENOMEM; 69b2cbae2cSRoland Dreier 70b2cbae2cSRoland Dreier nl_client->index = index; 71b2cbae2cSRoland Dreier nl_client->nops = nops; 72b2cbae2cSRoland Dreier nl_client->cb_table = cb_table; 73b2cbae2cSRoland Dreier 74b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 75b2cbae2cSRoland Dreier 76b2cbae2cSRoland Dreier list_for_each_entry(cur, &client_list, list) { 77b2cbae2cSRoland Dreier if (cur->index == index) { 78b2cbae2cSRoland Dreier pr_warn("Client for %d already exists\n", index); 79b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 80b2cbae2cSRoland Dreier kfree(nl_client); 81b2cbae2cSRoland Dreier return -EINVAL; 82b2cbae2cSRoland Dreier } 83b2cbae2cSRoland Dreier } 84b2cbae2cSRoland Dreier 85b2cbae2cSRoland Dreier list_add_tail(&nl_client->list, &client_list); 86b2cbae2cSRoland Dreier 87b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 88b2cbae2cSRoland Dreier 89b2cbae2cSRoland Dreier return 0; 90b2cbae2cSRoland Dreier } 91b2cbae2cSRoland Dreier EXPORT_SYMBOL(ibnl_add_client); 92b2cbae2cSRoland Dreier 93b2cbae2cSRoland Dreier int ibnl_remove_client(int index) 94b2cbae2cSRoland Dreier { 95b2cbae2cSRoland Dreier struct ibnl_client *cur, *next; 96b2cbae2cSRoland Dreier 97b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 98b2cbae2cSRoland Dreier list_for_each_entry_safe(cur, next, &client_list, list) { 99b2cbae2cSRoland Dreier if (cur->index == index) { 100b2cbae2cSRoland Dreier list_del(&(cur->list)); 101b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 102b2cbae2cSRoland Dreier kfree(cur); 103b2cbae2cSRoland Dreier return 0; 104b2cbae2cSRoland Dreier } 105b2cbae2cSRoland Dreier } 106b2cbae2cSRoland Dreier pr_warn("Can't remove callback for client idx %d. Not found\n", index); 107b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 108b2cbae2cSRoland Dreier 109b2cbae2cSRoland Dreier return -EINVAL; 110b2cbae2cSRoland Dreier } 111b2cbae2cSRoland Dreier EXPORT_SYMBOL(ibnl_remove_client); 112b2cbae2cSRoland Dreier 113b2cbae2cSRoland Dreier void *ibnl_put_msg(struct sk_buff *skb, struct nlmsghdr **nlh, int seq, 11430dc5e63STatyana Nikolova int len, int client, int op, int flags) 115b2cbae2cSRoland Dreier { 116b2cbae2cSRoland Dreier unsigned char *prev_tail; 117b2cbae2cSRoland Dreier 118b2cbae2cSRoland Dreier prev_tail = skb_tail_pointer(skb); 119e0527334SDavid S. Miller *nlh = nlmsg_put(skb, 0, seq, RDMA_NL_GET_TYPE(client, op), 12030dc5e63STatyana Nikolova len, flags); 121e0527334SDavid S. Miller if (!*nlh) 122e0527334SDavid S. Miller goto out_nlmsg_trim; 123b2cbae2cSRoland Dreier (*nlh)->nlmsg_len = skb_tail_pointer(skb) - prev_tail; 124e0527334SDavid S. Miller return nlmsg_data(*nlh); 125b2cbae2cSRoland Dreier 126e0527334SDavid S. Miller out_nlmsg_trim: 127b2cbae2cSRoland Dreier nlmsg_trim(skb, prev_tail); 128b2cbae2cSRoland Dreier return NULL; 129b2cbae2cSRoland Dreier } 130b2cbae2cSRoland Dreier EXPORT_SYMBOL(ibnl_put_msg); 131b2cbae2cSRoland Dreier 132b2cbae2cSRoland Dreier int ibnl_put_attr(struct sk_buff *skb, struct nlmsghdr *nlh, 133b2cbae2cSRoland Dreier int len, void *data, int type) 134b2cbae2cSRoland Dreier { 135b2cbae2cSRoland Dreier unsigned char *prev_tail; 136b2cbae2cSRoland Dreier 137b2cbae2cSRoland Dreier prev_tail = skb_tail_pointer(skb); 1384e24ffa4SDavid S. Miller if (nla_put(skb, type, len, data)) 1394e24ffa4SDavid S. Miller goto nla_put_failure; 140b2cbae2cSRoland Dreier nlh->nlmsg_len += skb_tail_pointer(skb) - prev_tail; 141b2cbae2cSRoland Dreier return 0; 142b2cbae2cSRoland Dreier 143b2cbae2cSRoland Dreier nla_put_failure: 144b2cbae2cSRoland Dreier nlmsg_trim(skb, prev_tail - nlh->nlmsg_len); 145b2cbae2cSRoland Dreier return -EMSGSIZE; 146b2cbae2cSRoland Dreier } 147b2cbae2cSRoland Dreier EXPORT_SYMBOL(ibnl_put_attr); 148b2cbae2cSRoland Dreier 149b2cbae2cSRoland Dreier static int ibnl_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh) 150b2cbae2cSRoland Dreier { 151b2cbae2cSRoland Dreier struct ibnl_client *client; 152b2cbae2cSRoland Dreier int type = nlh->nlmsg_type; 153b2cbae2cSRoland Dreier int index = RDMA_NL_GET_CLIENT(type); 1541ae5ccc7SMark Bloch unsigned int op = RDMA_NL_GET_OP(type); 155b2cbae2cSRoland Dreier 156b2cbae2cSRoland Dreier list_for_each_entry(client, &client_list, list) { 157b2cbae2cSRoland Dreier if (client->index == index) { 1581ae5ccc7SMark Bloch if (op >= client->nops || !client->cb_table[op].dump) 159b2cbae2cSRoland Dreier return -EINVAL; 16080d326faSPablo Neira Ayuso 161bc10ed7dSKaike Wan /* 162bc10ed7dSKaike Wan * For response or local service set_timeout request, 163bc10ed7dSKaike Wan * there is no need to use netlink_dump_start. 164bc10ed7dSKaike Wan */ 165bc10ed7dSKaike Wan if (!(nlh->nlmsg_flags & NLM_F_REQUEST) || 166bc10ed7dSKaike Wan (index == RDMA_NL_LS && 167bc10ed7dSKaike Wan op == RDMA_NL_LS_OP_SET_TIMEOUT)) { 168bc10ed7dSKaike Wan struct netlink_callback cb = { 169bc10ed7dSKaike Wan .skb = skb, 170bc10ed7dSKaike Wan .nlh = nlh, 171bc10ed7dSKaike Wan .dump = client->cb_table[op].dump, 172bc10ed7dSKaike Wan .module = client->cb_table[op].module, 173bc10ed7dSKaike Wan }; 174bc10ed7dSKaike Wan 175bc10ed7dSKaike Wan return cb.dump(skb, &cb); 176bc10ed7dSKaike Wan } 177bc10ed7dSKaike Wan 17880d326faSPablo Neira Ayuso { 17980d326faSPablo Neira Ayuso struct netlink_dump_control c = { 18080d326faSPablo Neira Ayuso .dump = client->cb_table[op].dump, 181809d5fc9SGao feng .module = client->cb_table[op].module, 18280d326faSPablo Neira Ayuso }; 18380d326faSPablo Neira Ayuso return netlink_dump_start(nls, skb, nlh, &c); 18480d326faSPablo Neira Ayuso } 185b2cbae2cSRoland Dreier } 186b2cbae2cSRoland Dreier } 187b2cbae2cSRoland Dreier 188b2cbae2cSRoland Dreier pr_info("Index %d wasn't found in client list\n", index); 189b2cbae2cSRoland Dreier return -EINVAL; 190b2cbae2cSRoland Dreier } 191b2cbae2cSRoland Dreier 192bc10ed7dSKaike Wan static void ibnl_rcv_reply_skb(struct sk_buff *skb) 193bc10ed7dSKaike Wan { 194bc10ed7dSKaike Wan struct nlmsghdr *nlh; 195bc10ed7dSKaike Wan int msglen; 196bc10ed7dSKaike Wan 197bc10ed7dSKaike Wan /* 198bc10ed7dSKaike Wan * Process responses until there is no more message or the first 199bc10ed7dSKaike Wan * request. Generally speaking, it is not recommended to mix responses 200bc10ed7dSKaike Wan * with requests. 201bc10ed7dSKaike Wan */ 202bc10ed7dSKaike Wan while (skb->len >= nlmsg_total_size(0)) { 203bc10ed7dSKaike Wan nlh = nlmsg_hdr(skb); 204bc10ed7dSKaike Wan 205bc10ed7dSKaike Wan if (nlh->nlmsg_len < NLMSG_HDRLEN || skb->len < nlh->nlmsg_len) 206bc10ed7dSKaike Wan return; 207bc10ed7dSKaike Wan 208bc10ed7dSKaike Wan /* Handle response only */ 209bc10ed7dSKaike Wan if (nlh->nlmsg_flags & NLM_F_REQUEST) 210bc10ed7dSKaike Wan return; 211bc10ed7dSKaike Wan 212bc10ed7dSKaike Wan ibnl_rcv_msg(skb, nlh); 213bc10ed7dSKaike Wan 214bc10ed7dSKaike Wan msglen = NLMSG_ALIGN(nlh->nlmsg_len); 215bc10ed7dSKaike Wan if (msglen > skb->len) 216bc10ed7dSKaike Wan msglen = skb->len; 217bc10ed7dSKaike Wan skb_pull(skb, msglen); 218bc10ed7dSKaike Wan } 219bc10ed7dSKaike Wan } 220bc10ed7dSKaike Wan 221b2cbae2cSRoland Dreier static void ibnl_rcv(struct sk_buff *skb) 222b2cbae2cSRoland Dreier { 223b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 224bc10ed7dSKaike Wan ibnl_rcv_reply_skb(skb); 225b2cbae2cSRoland Dreier netlink_rcv_skb(skb, &ibnl_rcv_msg); 226b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 227b2cbae2cSRoland Dreier } 228b2cbae2cSRoland Dreier 22930dc5e63STatyana Nikolova int ibnl_unicast(struct sk_buff *skb, struct nlmsghdr *nlh, 23030dc5e63STatyana Nikolova __u32 pid) 23130dc5e63STatyana Nikolova { 23230dc5e63STatyana Nikolova return nlmsg_unicast(nls, skb, pid); 23330dc5e63STatyana Nikolova } 23430dc5e63STatyana Nikolova EXPORT_SYMBOL(ibnl_unicast); 23530dc5e63STatyana Nikolova 23630dc5e63STatyana Nikolova int ibnl_multicast(struct sk_buff *skb, struct nlmsghdr *nlh, 23730dc5e63STatyana Nikolova unsigned int group, gfp_t flags) 23830dc5e63STatyana Nikolova { 23930dc5e63STatyana Nikolova return nlmsg_multicast(nls, skb, 0, group, flags); 24030dc5e63STatyana Nikolova } 24130dc5e63STatyana Nikolova EXPORT_SYMBOL(ibnl_multicast); 24230dc5e63STatyana Nikolova 243b2cbae2cSRoland Dreier int __init ibnl_init(void) 244b2cbae2cSRoland Dreier { 245a31f2d17SPablo Neira Ayuso struct netlink_kernel_cfg cfg = { 246a31f2d17SPablo Neira Ayuso .input = ibnl_rcv, 247a31f2d17SPablo Neira Ayuso }; 248a31f2d17SPablo Neira Ayuso 2499f00d977SPablo Neira Ayuso nls = netlink_kernel_create(&init_net, NETLINK_RDMA, &cfg); 250b2cbae2cSRoland Dreier if (!nls) { 251b2cbae2cSRoland Dreier pr_warn("Failed to create netlink socket\n"); 252b2cbae2cSRoland Dreier return -ENOMEM; 253b2cbae2cSRoland Dreier } 254b2cbae2cSRoland Dreier 255b2cbae2cSRoland Dreier return 0; 256b2cbae2cSRoland Dreier } 257b2cbae2cSRoland Dreier 258b2cbae2cSRoland Dreier void ibnl_cleanup(void) 259b2cbae2cSRoland Dreier { 260b2cbae2cSRoland Dreier struct ibnl_client *cur, *next; 261b2cbae2cSRoland Dreier 262b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 263b2cbae2cSRoland Dreier list_for_each_entry_safe(cur, next, &client_list, list) { 264b2cbae2cSRoland Dreier list_del(&(cur->list)); 265b2cbae2cSRoland Dreier kfree(cur); 266b2cbae2cSRoland Dreier } 267b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 268b2cbae2cSRoland Dreier 269b2cbae2cSRoland Dreier netlink_kernel_release(nls); 270b2cbae2cSRoland Dreier } 271