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> 40233c1955SLeon Romanovsky #include "core_priv.h" 41b2cbae2cSRoland Dreier 42b2cbae2cSRoland Dreier struct ibnl_client { 43b2cbae2cSRoland Dreier struct list_head list; 44b2cbae2cSRoland Dreier int index; 45b2cbae2cSRoland Dreier int nops; 46b2cbae2cSRoland Dreier const struct ibnl_client_cbs *cb_table; 47b2cbae2cSRoland Dreier }; 48b2cbae2cSRoland Dreier 49b2cbae2cSRoland Dreier static DEFINE_MUTEX(ibnl_mutex); 50b2cbae2cSRoland Dreier static struct sock *nls; 51b2cbae2cSRoland Dreier static LIST_HEAD(client_list); 52b2cbae2cSRoland Dreier 53bc10ed7dSKaike Wan int ibnl_chk_listeners(unsigned int group) 54bc10ed7dSKaike Wan { 55bc10ed7dSKaike Wan if (netlink_has_listeners(nls, group) == 0) 56bc10ed7dSKaike Wan return -1; 57bc10ed7dSKaike Wan return 0; 58bc10ed7dSKaike Wan } 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 1492d4bc933SJohannes Berg static int ibnl_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh, 1502d4bc933SJohannes Berg struct netlink_ext_ack *extack) 151b2cbae2cSRoland Dreier { 152b2cbae2cSRoland Dreier struct ibnl_client *client; 153b2cbae2cSRoland Dreier int type = nlh->nlmsg_type; 154b2cbae2cSRoland Dreier int index = RDMA_NL_GET_CLIENT(type); 1551ae5ccc7SMark Bloch unsigned int op = RDMA_NL_GET_OP(type); 156b2cbae2cSRoland Dreier 157b2cbae2cSRoland Dreier list_for_each_entry(client, &client_list, list) { 158b2cbae2cSRoland Dreier if (client->index == index) { 1591ae5ccc7SMark Bloch if (op >= client->nops || !client->cb_table[op].dump) 160b2cbae2cSRoland Dreier return -EINVAL; 16180d326faSPablo Neira Ayuso 162bc10ed7dSKaike Wan /* 163bc10ed7dSKaike Wan * For response or local service set_timeout request, 164bc10ed7dSKaike Wan * there is no need to use netlink_dump_start. 165bc10ed7dSKaike Wan */ 166bc10ed7dSKaike Wan if (!(nlh->nlmsg_flags & NLM_F_REQUEST) || 167bc10ed7dSKaike Wan (index == RDMA_NL_LS && 168bc10ed7dSKaike Wan op == RDMA_NL_LS_OP_SET_TIMEOUT)) { 169bc10ed7dSKaike Wan struct netlink_callback cb = { 170bc10ed7dSKaike Wan .skb = skb, 171bc10ed7dSKaike Wan .nlh = nlh, 172bc10ed7dSKaike Wan .dump = client->cb_table[op].dump, 173bc10ed7dSKaike Wan .module = client->cb_table[op].module, 174bc10ed7dSKaike Wan }; 175bc10ed7dSKaike Wan 176bc10ed7dSKaike Wan return cb.dump(skb, &cb); 177bc10ed7dSKaike Wan } 178bc10ed7dSKaike Wan 17980d326faSPablo Neira Ayuso { 18080d326faSPablo Neira Ayuso struct netlink_dump_control c = { 18180d326faSPablo Neira Ayuso .dump = client->cb_table[op].dump, 182809d5fc9SGao feng .module = client->cb_table[op].module, 18380d326faSPablo Neira Ayuso }; 18480d326faSPablo Neira Ayuso return netlink_dump_start(nls, skb, nlh, &c); 18580d326faSPablo Neira Ayuso } 186b2cbae2cSRoland Dreier } 187b2cbae2cSRoland Dreier } 188b2cbae2cSRoland Dreier 189b2cbae2cSRoland Dreier pr_info("Index %d wasn't found in client list\n", index); 190b2cbae2cSRoland Dreier return -EINVAL; 191b2cbae2cSRoland Dreier } 192b2cbae2cSRoland Dreier 193bc10ed7dSKaike Wan static void ibnl_rcv_reply_skb(struct sk_buff *skb) 194bc10ed7dSKaike Wan { 195bc10ed7dSKaike Wan struct nlmsghdr *nlh; 196bc10ed7dSKaike Wan int msglen; 197bc10ed7dSKaike Wan 198bc10ed7dSKaike Wan /* 199bc10ed7dSKaike Wan * Process responses until there is no more message or the first 200bc10ed7dSKaike Wan * request. Generally speaking, it is not recommended to mix responses 201bc10ed7dSKaike Wan * with requests. 202bc10ed7dSKaike Wan */ 203bc10ed7dSKaike Wan while (skb->len >= nlmsg_total_size(0)) { 204bc10ed7dSKaike Wan nlh = nlmsg_hdr(skb); 205bc10ed7dSKaike Wan 206bc10ed7dSKaike Wan if (nlh->nlmsg_len < NLMSG_HDRLEN || skb->len < nlh->nlmsg_len) 207bc10ed7dSKaike Wan return; 208bc10ed7dSKaike Wan 209bc10ed7dSKaike Wan /* Handle response only */ 210bc10ed7dSKaike Wan if (nlh->nlmsg_flags & NLM_F_REQUEST) 211bc10ed7dSKaike Wan return; 212bc10ed7dSKaike Wan 2132d4bc933SJohannes Berg ibnl_rcv_msg(skb, nlh, NULL); 214bc10ed7dSKaike Wan 215bc10ed7dSKaike Wan msglen = NLMSG_ALIGN(nlh->nlmsg_len); 216bc10ed7dSKaike Wan if (msglen > skb->len) 217bc10ed7dSKaike Wan msglen = skb->len; 218bc10ed7dSKaike Wan skb_pull(skb, msglen); 219bc10ed7dSKaike Wan } 220bc10ed7dSKaike Wan } 221bc10ed7dSKaike Wan 222b2cbae2cSRoland Dreier static void ibnl_rcv(struct sk_buff *skb) 223b2cbae2cSRoland Dreier { 224b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 225bc10ed7dSKaike Wan ibnl_rcv_reply_skb(skb); 226b2cbae2cSRoland Dreier netlink_rcv_skb(skb, &ibnl_rcv_msg); 227b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 228b2cbae2cSRoland Dreier } 229b2cbae2cSRoland Dreier 23030dc5e63STatyana Nikolova int ibnl_unicast(struct sk_buff *skb, struct nlmsghdr *nlh, 23130dc5e63STatyana Nikolova __u32 pid) 23230dc5e63STatyana Nikolova { 233cea05eadSMustafa Ismail int err; 234cea05eadSMustafa Ismail 2359047811bSIsmail, Mustafa err = netlink_unicast(nls, skb, pid, MSG_DONTWAIT); 236cea05eadSMustafa Ismail return (err < 0) ? err : 0; 23730dc5e63STatyana Nikolova } 23830dc5e63STatyana Nikolova EXPORT_SYMBOL(ibnl_unicast); 23930dc5e63STatyana Nikolova 2409047811bSIsmail, Mustafa int ibnl_unicast_wait(struct sk_buff *skb, struct nlmsghdr *nlh, 2419047811bSIsmail, Mustafa __u32 pid) 2429047811bSIsmail, Mustafa { 2439047811bSIsmail, Mustafa int err; 2449047811bSIsmail, Mustafa 2459047811bSIsmail, Mustafa err = netlink_unicast(nls, skb, pid, 0); 2469047811bSIsmail, Mustafa return (err < 0) ? err : 0; 2479047811bSIsmail, Mustafa } 2489047811bSIsmail, Mustafa EXPORT_SYMBOL(ibnl_unicast_wait); 2499047811bSIsmail, Mustafa 25030dc5e63STatyana Nikolova int ibnl_multicast(struct sk_buff *skb, struct nlmsghdr *nlh, 25130dc5e63STatyana Nikolova unsigned int group, gfp_t flags) 25230dc5e63STatyana Nikolova { 25330dc5e63STatyana Nikolova return nlmsg_multicast(nls, skb, 0, group, flags); 25430dc5e63STatyana Nikolova } 25530dc5e63STatyana Nikolova EXPORT_SYMBOL(ibnl_multicast); 25630dc5e63STatyana Nikolova 257b2cbae2cSRoland Dreier int __init ibnl_init(void) 258b2cbae2cSRoland Dreier { 259a31f2d17SPablo Neira Ayuso struct netlink_kernel_cfg cfg = { 260a31f2d17SPablo Neira Ayuso .input = ibnl_rcv, 261a31f2d17SPablo Neira Ayuso }; 262a31f2d17SPablo Neira Ayuso 2639f00d977SPablo Neira Ayuso nls = netlink_kernel_create(&init_net, NETLINK_RDMA, &cfg); 264b2cbae2cSRoland Dreier if (!nls) { 265b2cbae2cSRoland Dreier pr_warn("Failed to create netlink socket\n"); 266b2cbae2cSRoland Dreier return -ENOMEM; 267b2cbae2cSRoland Dreier } 268b2cbae2cSRoland Dreier 269cea05eadSMustafa Ismail nls->sk_sndtimeo = 10 * HZ; 270b2cbae2cSRoland Dreier return 0; 271b2cbae2cSRoland Dreier } 272b2cbae2cSRoland Dreier 273b2cbae2cSRoland Dreier void ibnl_cleanup(void) 274b2cbae2cSRoland Dreier { 275b2cbae2cSRoland Dreier struct ibnl_client *cur, *next; 276b2cbae2cSRoland Dreier 277b2cbae2cSRoland Dreier mutex_lock(&ibnl_mutex); 278b2cbae2cSRoland Dreier list_for_each_entry_safe(cur, next, &client_list, list) { 279b2cbae2cSRoland Dreier list_del(&(cur->list)); 280b2cbae2cSRoland Dreier kfree(cur); 281b2cbae2cSRoland Dreier } 282b2cbae2cSRoland Dreier mutex_unlock(&ibnl_mutex); 283b2cbae2cSRoland Dreier 284b2cbae2cSRoland Dreier netlink_kernel_release(nls); 285b2cbae2cSRoland Dreier } 286