drivers/infiniband/core/iwpm_msg.c

   1 /*
   2  * Copyright (c) 2014 Intel Corporation. All rights reserved.
   3  * Copyright (c) 2014 Chelsio, Inc. All rights reserved.
   4  *
   5  * This software is available to you under a choice of one of two
   6  * licenses.  You may choose to be licensed under the terms of the GNU
   7  * General Public License (GPL) Version 2, available from the file
   8  * COPYING in the main directory of this source tree, or the
   9  * OpenIB.org BSD license below:
  10  *
  11  *     Redistribution and use in source and binary forms, with or
  12  *     without modification, are permitted provided that the following
  13  *     conditions are met:
  14  *
  15  *      - Redistributions of source code must retain the above
  16  *        copyright notice, this list of conditions and the following
  17  *        disclaimer.
  18  *
  19  *      - Redistributions in binary form must reproduce the above
  20  *        copyright notice, this list of conditions and the following
  21  *        disclaimer in the documentation and/or other materials
  22  *        provided with the distribution.
  23  *
  24  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  25  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  26  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  27  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  28  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  29  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  30  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  31  * SOFTWARE.
  32  */
  33
  34 #include "iwpm_util.h"
  35
  36 static const char iwpm_ulib_name[IWPM_ULIBNAME_SIZE] = "iWarpPortMapperUser";
  37 u16 iwpm_ulib_version = IWPM_UABI_VERSION_MIN;
  38 static int iwpm_user_pid = IWPM_PID_UNDEFINED;
  39 static atomic_t echo_nlmsg_seq;
  40
  41 /**
  42  * iwpm_valid_pid - Check if the userspace iwarp port mapper pid is valid
  43  *
  44  * Returns true if the pid is greater than zero, otherwise returns false
  45  */
  46 int iwpm_valid_pid(void)
  47 {
  48         return iwpm_user_pid > 0;
  49 }
  50
  51 /**
  52  * iwpm_register_pid - Send a netlink query to userspace
  53  *                     to get the iwarp port mapper pid
  54  * @pm_msg: Contains driver info to send to the userspace port mapper
  55  * @nl_client: The index of the netlink client
  56  *
  57  * nlmsg attributes:
  58  *      [IWPM_NLA_REG_PID_SEQ]
  59  *      [IWPM_NLA_REG_IF_NAME]
  60  *      [IWPM_NLA_REG_IBDEV_NAME]
  61  *      [IWPM_NLA_REG_ULIB_NAME]
  62  */
  63 int iwpm_register_pid(struct iwpm_dev_data *pm_msg, u8 nl_client)
  64 {
  65         struct sk_buff *skb = NULL;
  66         struct iwpm_nlmsg_request *nlmsg_request = NULL;
  67         struct nlmsghdr *nlh;
  68         u32 msg_seq;
  69         const char *err_str = "";
  70         int ret = -EINVAL;
  71
  72         if (!iwpm_valid_client(nl_client)) {
  73                 err_str = "Invalid port mapper client";
  74                 goto pid_query_error;
  75         }
  76         if (iwpm_check_registration(nl_client, IWPM_REG_VALID) ||
  77                         iwpm_user_pid == IWPM_PID_UNAVAILABLE)
  78                 return 0;
  79         skb = iwpm_create_nlmsg(RDMA_NL_IWPM_REG_PID, &nlh, nl_client);
  80         if (!skb) {
  81                 err_str = "Unable to create a nlmsg";
  82                 goto pid_query_error;
  83         }
  84         nlh->nlmsg_seq = iwpm_get_nlmsg_seq();
  85         nlmsg_request = iwpm_get_nlmsg_request(nlh->nlmsg_seq, nl_client, GFP_KERNEL);
  86         if (!nlmsg_request) {
  87                 err_str = "Unable to allocate netlink request";
  88                 goto pid_query_error;
  89         }
  90         msg_seq = atomic_read(&echo_nlmsg_seq);
  91
  92         /* fill in the pid request message */
  93         err_str = "Unable to put attribute of the nlmsg";
  94         ret = ibnl_put_attr(skb, nlh, sizeof(u32), &msg_seq, IWPM_NLA_REG_PID_SEQ);
  95         if (ret)
  96                 goto pid_query_error;
  97         ret = ibnl_put_attr(skb, nlh, IFNAMSIZ,
  98                             pm_msg->if_name, IWPM_NLA_REG_IF_NAME);
  99         if (ret)
 100                 goto pid_query_error;
 101         ret = ibnl_put_attr(skb, nlh, IWPM_DEVNAME_SIZE,
 102                                 pm_msg->dev_name, IWPM_NLA_REG_IBDEV_NAME);
 103         if (ret)
 104                 goto pid_query_error;
 105         ret = ibnl_put_attr(skb, nlh, IWPM_ULIBNAME_SIZE,
 106                                 (char *)iwpm_ulib_name, IWPM_NLA_REG_ULIB_NAME);
 107         if (ret)
 108                 goto pid_query_error;
 109
 110         nlmsg_end(skb, nlh);
 111
 112         pr_debug("%s: Multicasting a nlmsg (dev = %s ifname = %s iwpm = %s)\n",
 113                 __func__, pm_msg->dev_name, pm_msg->if_name, iwpm_ulib_name);
 114
 115         ret = rdma_nl_multicast(&init_net, skb, RDMA_NL_GROUP_IWPM, GFP_KERNEL);
 116         if (ret) {
 117                 skb = NULL; /* skb is freed in the netlink send-op handling */
 118                 iwpm_user_pid = IWPM_PID_UNAVAILABLE;
 119                 err_str = "Unable to send a nlmsg";
 120                 goto pid_query_error;
 121         }
 122         nlmsg_request->req_buffer = pm_msg;
 123         ret = iwpm_wait_complete_req(nlmsg_request);
 124         return ret;
 125 pid_query_error:
 126         pr_info("%s: %s (client = %d)\n", __func__, err_str, nl_client);
 127         dev_kfree_skb(skb);
 128         if (nlmsg_request)
 129                 iwpm_free_nlmsg_request(&nlmsg_request->kref);
 130         return ret;
 131 }
 132
 133 /**
 134  * iwpm_add_mapping - Send a netlink add mapping request to
 135  *                    the userspace port mapper
 136  * @pm_msg: Contains the local ip/tcp address info to send
 137  * @nl_client: The index of the netlink client
 138  *
 139  * nlmsg attributes:
 140  *      [IWPM_NLA_MANAGE_MAPPING_SEQ]
 141  *      [IWPM_NLA_MANAGE_ADDR]
 142  *      [IWPM_NLA_MANAGE_FLAGS]
 143  *
 144  * If the request is successful, the pm_msg stores
 145  * the port mapper response (mapped address info)
 146  */
 147 int iwpm_add_mapping(struct iwpm_sa_data *pm_msg, u8 nl_client)
 148 {
 149         struct sk_buff *skb = NULL;
 150         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 151         struct nlmsghdr *nlh;
 152         u32 msg_seq;
 153         const char *err_str = "";
 154         int ret = -EINVAL;
 155
 156         if (!iwpm_valid_client(nl_client)) {
 157                 err_str = "Invalid port mapper client";
 158                 goto add_mapping_error;
 159         }
 160         if (!iwpm_valid_pid())
 161                 return 0;
 162         if (!iwpm_check_registration(nl_client, IWPM_REG_VALID)) {
 163                 err_str = "Unregistered port mapper client";
 164                 goto add_mapping_error;
 165         }
 166         skb = iwpm_create_nlmsg(RDMA_NL_IWPM_ADD_MAPPING, &nlh, nl_client);
 167         if (!skb) {
 168                 err_str = "Unable to create a nlmsg";
 169                 goto add_mapping_error;
 170         }
 171         nlh->nlmsg_seq = iwpm_get_nlmsg_seq();
 172         nlmsg_request = iwpm_get_nlmsg_request(nlh->nlmsg_seq, nl_client, GFP_KERNEL);
 173         if (!nlmsg_request) {
 174                 err_str = "Unable to allocate netlink request";
 175                 goto add_mapping_error;
 176         }
 177         msg_seq = atomic_read(&echo_nlmsg_seq);
 178         /* fill in the add mapping message */
 179         err_str = "Unable to put attribute of the nlmsg";
 180         ret = ibnl_put_attr(skb, nlh, sizeof(u32), &msg_seq,
 181                                 IWPM_NLA_MANAGE_MAPPING_SEQ);
 182         if (ret)
 183                 goto add_mapping_error;
 184         ret = ibnl_put_attr(skb, nlh, sizeof(struct sockaddr_storage),
 185                                 &pm_msg->loc_addr, IWPM_NLA_MANAGE_ADDR);
 186         if (ret)
 187                 goto add_mapping_error;
 188
 189         /* If flags are required and we're not V4, then return a quiet error */
 190         if (pm_msg->flags && iwpm_ulib_version == IWPM_UABI_VERSION_MIN) {
 191                 ret = -EINVAL;
 192                 goto add_mapping_error_nowarn;
 193         }
 194         if (iwpm_ulib_version > IWPM_UABI_VERSION_MIN) {
 195                 ret = ibnl_put_attr(skb, nlh, sizeof(u32), &pm_msg->flags,
 196                                 IWPM_NLA_MANAGE_FLAGS);
 197                 if (ret)
 198                         goto add_mapping_error;
 199         }
 200
 201         nlmsg_end(skb, nlh);
 202         nlmsg_request->req_buffer = pm_msg;
 203
 204         ret = rdma_nl_unicast_wait(&init_net, skb, iwpm_user_pid);
 205         if (ret) {
 206                 skb = NULL; /* skb is freed in the netlink send-op handling */
 207                 iwpm_user_pid = IWPM_PID_UNDEFINED;
 208                 err_str = "Unable to send a nlmsg";
 209                 goto add_mapping_error;
 210         }
 211         ret = iwpm_wait_complete_req(nlmsg_request);
 212         return ret;
 213 add_mapping_error:
 214         pr_info("%s: %s (client = %d)\n", __func__, err_str, nl_client);
 215 add_mapping_error_nowarn:
 216         dev_kfree_skb(skb);
 217         if (nlmsg_request)
 218                 iwpm_free_nlmsg_request(&nlmsg_request->kref);
 219         return ret;
 220 }
 221
 222 /**
 223  * iwpm_add_and_query_mapping - Process the port mapper response to
 224  *                              iwpm_add_and_query_mapping request
 225  * @pm_msg: Contains the local ip/tcp address info to send
 226  * @nl_client: The index of the netlink client
 227  *
 228  * nlmsg attributes:
 229  *      [IWPM_NLA_QUERY_MAPPING_SEQ]
 230  *      [IWPM_NLA_QUERY_LOCAL_ADDR]
 231  *      [IWPM_NLA_QUERY_REMOTE_ADDR]
 232  *      [IWPM_NLA_QUERY_FLAGS]
 233  */
 234 int iwpm_add_and_query_mapping(struct iwpm_sa_data *pm_msg, u8 nl_client)
 235 {
 236         struct sk_buff *skb = NULL;
 237         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 238         struct nlmsghdr *nlh;
 239         u32 msg_seq;
 240         const char *err_str = "";
 241         int ret = -EINVAL;
 242
 243         if (!iwpm_valid_client(nl_client)) {
 244                 err_str = "Invalid port mapper client";
 245                 goto query_mapping_error;
 246         }
 247         if (!iwpm_valid_pid())
 248                 return 0;
 249         if (!iwpm_check_registration(nl_client, IWPM_REG_VALID)) {
 250                 err_str = "Unregistered port mapper client";
 251                 goto query_mapping_error;
 252         }
 253         ret = -ENOMEM;
 254         skb = iwpm_create_nlmsg(RDMA_NL_IWPM_QUERY_MAPPING, &nlh, nl_client);
 255         if (!skb) {
 256                 err_str = "Unable to create a nlmsg";
 257                 goto query_mapping_error;
 258         }
 259         nlh->nlmsg_seq = iwpm_get_nlmsg_seq();
 260         nlmsg_request = iwpm_get_nlmsg_request(nlh->nlmsg_seq,
 261                                 nl_client, GFP_KERNEL);
 262         if (!nlmsg_request) {
 263                 err_str = "Unable to allocate netlink request";
 264                 goto query_mapping_error;
 265         }
 266         msg_seq = atomic_read(&echo_nlmsg_seq);
 267
 268         /* fill in the query message */
 269         err_str = "Unable to put attribute of the nlmsg";
 270         ret = ibnl_put_attr(skb, nlh, sizeof(u32), &msg_seq,
 271                                 IWPM_NLA_QUERY_MAPPING_SEQ);
 272         if (ret)
 273                 goto query_mapping_error;
 274         ret = ibnl_put_attr(skb, nlh, sizeof(struct sockaddr_storage),
 275                                 &pm_msg->loc_addr, IWPM_NLA_QUERY_LOCAL_ADDR);
 276         if (ret)
 277                 goto query_mapping_error;
 278         ret = ibnl_put_attr(skb, nlh, sizeof(struct sockaddr_storage),
 279                                 &pm_msg->rem_addr, IWPM_NLA_QUERY_REMOTE_ADDR);
 280         if (ret)
 281                 goto query_mapping_error;
 282
 283         /* If flags are required and we're not V4, then return a quite error */
 284         if (pm_msg->flags && iwpm_ulib_version == IWPM_UABI_VERSION_MIN) {
 285                 ret = -EINVAL;
 286                 goto query_mapping_error_nowarn;
 287         }
 288         if (iwpm_ulib_version > IWPM_UABI_VERSION_MIN) {
 289                 ret = ibnl_put_attr(skb, nlh, sizeof(u32), &pm_msg->flags,
 290                                 IWPM_NLA_QUERY_FLAGS);
 291                 if (ret)
 292                         goto query_mapping_error;
 293         }
 294
 295         nlmsg_end(skb, nlh);
 296         nlmsg_request->req_buffer = pm_msg;
 297
 298         ret = rdma_nl_unicast_wait(&init_net, skb, iwpm_user_pid);
 299         if (ret) {
 300                 skb = NULL; /* skb is freed in the netlink send-op handling */
 301                 err_str = "Unable to send a nlmsg";
 302                 goto query_mapping_error;
 303         }
 304         ret = iwpm_wait_complete_req(nlmsg_request);
 305         return ret;
 306 query_mapping_error:
 307         pr_info("%s: %s (client = %d)\n", __func__, err_str, nl_client);
 308 query_mapping_error_nowarn:
 309         dev_kfree_skb(skb);
 310         if (nlmsg_request)
 311                 iwpm_free_nlmsg_request(&nlmsg_request->kref);
 312         return ret;
 313 }
 314
 315 /**
 316  * iwpm_remove_mapping - Send a netlink remove mapping request
 317  *                       to the userspace port mapper
 318  *
 319  * @local_addr: Local ip/tcp address to remove
 320  * @nl_client: The index of the netlink client
 321  *
 322  * nlmsg attributes:
 323  *      [IWPM_NLA_MANAGE_MAPPING_SEQ]
 324  *      [IWPM_NLA_MANAGE_ADDR]
 325  */
 326 int iwpm_remove_mapping(struct sockaddr_storage *local_addr, u8 nl_client)
 327 {
 328         struct sk_buff *skb = NULL;
 329         struct nlmsghdr *nlh;
 330         u32 msg_seq;
 331         const char *err_str = "";
 332         int ret = -EINVAL;
 333
 334         if (!iwpm_valid_client(nl_client)) {
 335                 err_str = "Invalid port mapper client";
 336                 goto remove_mapping_error;
 337         }
 338         if (!iwpm_valid_pid())
 339                 return 0;
 340         if (iwpm_check_registration(nl_client, IWPM_REG_UNDEF)) {
 341                 err_str = "Unregistered port mapper client";
 342                 goto remove_mapping_error;
 343         }
 344         skb = iwpm_create_nlmsg(RDMA_NL_IWPM_REMOVE_MAPPING, &nlh, nl_client);
 345         if (!skb) {
 346                 ret = -ENOMEM;
 347                 err_str = "Unable to create a nlmsg";
 348                 goto remove_mapping_error;
 349         }
 350         msg_seq = atomic_read(&echo_nlmsg_seq);
 351         nlh->nlmsg_seq = iwpm_get_nlmsg_seq();
 352         err_str = "Unable to put attribute of the nlmsg";
 353         ret = ibnl_put_attr(skb, nlh, sizeof(u32), &msg_seq,
 354                                 IWPM_NLA_MANAGE_MAPPING_SEQ);
 355         if (ret)
 356                 goto remove_mapping_error;
 357         ret = ibnl_put_attr(skb, nlh, sizeof(struct sockaddr_storage),
 358                                 local_addr, IWPM_NLA_MANAGE_ADDR);
 359         if (ret)
 360                 goto remove_mapping_error;
 361
 362         nlmsg_end(skb, nlh);
 363
 364         ret = rdma_nl_unicast_wait(&init_net, skb, iwpm_user_pid);
 365         if (ret) {
 366                 skb = NULL; /* skb is freed in the netlink send-op handling */
 367                 iwpm_user_pid = IWPM_PID_UNDEFINED;
 368                 err_str = "Unable to send a nlmsg";
 369                 goto remove_mapping_error;
 370         }
 371         iwpm_print_sockaddr(local_addr,
 372                         "remove_mapping: Local sockaddr:");
 373         return 0;
 374 remove_mapping_error:
 375         pr_info("%s: %s (client = %d)\n", __func__, err_str, nl_client);
 376         if (skb)
 377                 dev_kfree_skb_any(skb);
 378         return ret;
 379 }
 380
 381 /* netlink attribute policy for the received response to register pid request */
 382 static const struct nla_policy resp_reg_policy[IWPM_NLA_RREG_PID_MAX] = {
 383         [IWPM_NLA_RREG_PID_SEQ]     = { .type = NLA_U32 },
 384         [IWPM_NLA_RREG_IBDEV_NAME]  = { .type = NLA_STRING,
 385                                         .len = IWPM_DEVNAME_SIZE - 1 },
 386         [IWPM_NLA_RREG_ULIB_NAME]   = { .type = NLA_STRING,
 387                                         .len = IWPM_ULIBNAME_SIZE - 1 },
 388         [IWPM_NLA_RREG_ULIB_VER]    = { .type = NLA_U16 },
 389         [IWPM_NLA_RREG_PID_ERR]     = { .type = NLA_U16 }
 390 };
 391
 392 /**
 393  * iwpm_register_pid_cb - Process the port mapper response to
 394  *                        iwpm_register_pid query
 395  * @skb:
 396  * @cb: Contains the received message (payload and netlink header)
 397  *
 398  * If successful, the function receives the userspace port mapper pid
 399  * which is used in future communication with the port mapper
 400  */
 401 int iwpm_register_pid_cb(struct sk_buff *skb, struct netlink_callback *cb)
 402 {
 403         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 404         struct nlattr *nltb[IWPM_NLA_RREG_PID_MAX];
 405         struct iwpm_dev_data *pm_msg;
 406         char *dev_name, *iwpm_name;
 407         u32 msg_seq;
 408         u8 nl_client;
 409         u16 iwpm_version;
 410         const char *msg_type = "Register Pid response";
 411
 412         if (iwpm_parse_nlmsg(cb, IWPM_NLA_RREG_PID_MAX,
 413                                 resp_reg_policy, nltb, msg_type))
 414                 return -EINVAL;
 415
 416         msg_seq = nla_get_u32(nltb[IWPM_NLA_RREG_PID_SEQ]);
 417         nlmsg_request = iwpm_find_nlmsg_request(msg_seq);
 418         if (!nlmsg_request) {
 419                 pr_info("%s: Could not find a matching request (seq = %u)\n",
 420                                  __func__, msg_seq);
 421                 return -EINVAL;
 422         }
 423         pm_msg = nlmsg_request->req_buffer;
 424         nl_client = nlmsg_request->nl_client;
 425         dev_name = (char *)nla_data(nltb[IWPM_NLA_RREG_IBDEV_NAME]);
 426         iwpm_name = (char *)nla_data(nltb[IWPM_NLA_RREG_ULIB_NAME]);
 427         iwpm_version = nla_get_u16(nltb[IWPM_NLA_RREG_ULIB_VER]);
 428
 429         /* check device name, ulib name and version */
 430         if (strcmp(pm_msg->dev_name, dev_name) ||
 431                         strcmp(iwpm_ulib_name, iwpm_name) ||
 432                         iwpm_version < IWPM_UABI_VERSION_MIN) {
 433
 434                 pr_info("%s: Incorrect info (dev = %s name = %s version = %d)\n",
 435                                 __func__, dev_name, iwpm_name, iwpm_version);
 436                 nlmsg_request->err_code = IWPM_USER_LIB_INFO_ERR;
 437                 goto register_pid_response_exit;
 438         }
 439         iwpm_user_pid = cb->nlh->nlmsg_pid;
 440         iwpm_ulib_version = iwpm_version;
 441         if (iwpm_ulib_version < IWPM_UABI_VERSION)
 442                 pr_warn_once("%s: Down level iwpmd/pid %u.  Continuing...",
 443                         __func__, iwpm_user_pid);
 444         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 445         pr_debug("%s: iWarp Port Mapper (pid = %d) is available!\n",
 446                         __func__, iwpm_user_pid);
 447         if (iwpm_valid_client(nl_client))
 448                 iwpm_set_registration(nl_client, IWPM_REG_VALID);
 449 register_pid_response_exit:
 450         nlmsg_request->request_done = 1;
 451         /* always for found nlmsg_request */
 452         kref_put(&nlmsg_request->kref, iwpm_free_nlmsg_request);
 453         barrier();
 454         up(&nlmsg_request->sem);
 455         return 0;
 456 }
 457
 458 /* netlink attribute policy for the received response to add mapping request */
 459 static const struct nla_policy resp_add_policy[IWPM_NLA_RMANAGE_MAPPING_MAX] = {
 460         [IWPM_NLA_RMANAGE_MAPPING_SEQ]     = { .type = NLA_U32 },
 461         [IWPM_NLA_RMANAGE_ADDR]            = {
 462                                 .len = sizeof(struct sockaddr_storage) },
 463         [IWPM_NLA_RMANAGE_MAPPED_LOC_ADDR] = {
 464                                 .len = sizeof(struct sockaddr_storage) },
 465         [IWPM_NLA_RMANAGE_MAPPING_ERR]     = { .type = NLA_U16 }
 466 };
 467
 468 /**
 469  * iwpm_add_mapping_cb - Process the port mapper response to
 470  *                       iwpm_add_mapping request
 471  * @skb:
 472  * @cb: Contains the received message (payload and netlink header)
 473  */
 474 int iwpm_add_mapping_cb(struct sk_buff *skb, struct netlink_callback *cb)
 475 {
 476         struct iwpm_sa_data *pm_msg;
 477         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 478         struct nlattr *nltb[IWPM_NLA_RMANAGE_MAPPING_MAX];
 479         struct sockaddr_storage *local_sockaddr;
 480         struct sockaddr_storage *mapped_sockaddr;
 481         const char *msg_type;
 482         u32 msg_seq;
 483
 484         msg_type = "Add Mapping response";
 485         if (iwpm_parse_nlmsg(cb, IWPM_NLA_RMANAGE_MAPPING_MAX,
 486                                 resp_add_policy, nltb, msg_type))
 487                 return -EINVAL;
 488
 489         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 490
 491         msg_seq = nla_get_u32(nltb[IWPM_NLA_RMANAGE_MAPPING_SEQ]);
 492         nlmsg_request = iwpm_find_nlmsg_request(msg_seq);
 493         if (!nlmsg_request) {
 494                 pr_info("%s: Could not find a matching request (seq = %u)\n",
 495                                  __func__, msg_seq);
 496                 return -EINVAL;
 497         }
 498         pm_msg = nlmsg_request->req_buffer;
 499         local_sockaddr = (struct sockaddr_storage *)
 500                         nla_data(nltb[IWPM_NLA_RMANAGE_ADDR]);
 501         mapped_sockaddr = (struct sockaddr_storage *)
 502                         nla_data(nltb[IWPM_NLA_RMANAGE_MAPPED_LOC_ADDR]);
 503
 504         if (iwpm_compare_sockaddr(local_sockaddr, &pm_msg->loc_addr)) {
 505                 nlmsg_request->err_code = IWPM_USER_LIB_INFO_ERR;
 506                 goto add_mapping_response_exit;
 507         }
 508         if (mapped_sockaddr->ss_family != local_sockaddr->ss_family) {
 509                 pr_info("%s: Sockaddr family doesn't match the requested one\n",
 510                                 __func__);
 511                 nlmsg_request->err_code = IWPM_USER_LIB_INFO_ERR;
 512                 goto add_mapping_response_exit;
 513         }
 514         memcpy(&pm_msg->mapped_loc_addr, mapped_sockaddr,
 515                         sizeof(*mapped_sockaddr));
 516         iwpm_print_sockaddr(&pm_msg->loc_addr,
 517                         "add_mapping: Local sockaddr:");
 518         iwpm_print_sockaddr(&pm_msg->mapped_loc_addr,
 519                         "add_mapping: Mapped local sockaddr:");
 520
 521 add_mapping_response_exit:
 522         nlmsg_request->request_done = 1;
 523         /* always for found request */
 524         kref_put(&nlmsg_request->kref, iwpm_free_nlmsg_request);
 525         barrier();
 526         up(&nlmsg_request->sem);
 527         return 0;
 528 }
 529
 530 /* netlink attribute policy for the response to add and query mapping request
 531  * and response with remote address info */
 532 static const struct nla_policy resp_query_policy[IWPM_NLA_RQUERY_MAPPING_MAX] = {
 533         [IWPM_NLA_RQUERY_MAPPING_SEQ]     = { .type = NLA_U32 },
 534         [IWPM_NLA_RQUERY_LOCAL_ADDR]      = {
 535                                 .len = sizeof(struct sockaddr_storage) },
 536         [IWPM_NLA_RQUERY_REMOTE_ADDR]     = {
 537                                 .len = sizeof(struct sockaddr_storage) },
 538         [IWPM_NLA_RQUERY_MAPPED_LOC_ADDR] = {
 539                                 .len = sizeof(struct sockaddr_storage) },
 540         [IWPM_NLA_RQUERY_MAPPED_REM_ADDR] = {
 541                                 .len = sizeof(struct sockaddr_storage) },
 542         [IWPM_NLA_RQUERY_MAPPING_ERR]     = { .type = NLA_U16 }
 543 };
 544
 545 /**
 546  * iwpm_add_and_query_mapping_cb - Process the port mapper response to
 547  *                                 iwpm_add_and_query_mapping request
 548  * @skb:
 549  * @cb: Contains the received message (payload and netlink header)
 550  */
 551 int iwpm_add_and_query_mapping_cb(struct sk_buff *skb,
 552                                 struct netlink_callback *cb)
 553 {
 554         struct iwpm_sa_data *pm_msg;
 555         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 556         struct nlattr *nltb[IWPM_NLA_RQUERY_MAPPING_MAX];
 557         struct sockaddr_storage *local_sockaddr, *remote_sockaddr;
 558         struct sockaddr_storage *mapped_loc_sockaddr, *mapped_rem_sockaddr;
 559         const char *msg_type;
 560         u32 msg_seq;
 561         u16 err_code;
 562
 563         msg_type = "Query Mapping response";
 564         if (iwpm_parse_nlmsg(cb, IWPM_NLA_RQUERY_MAPPING_MAX,
 565                                 resp_query_policy, nltb, msg_type))
 566                 return -EINVAL;
 567         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 568
 569         msg_seq = nla_get_u32(nltb[IWPM_NLA_RQUERY_MAPPING_SEQ]);
 570         nlmsg_request = iwpm_find_nlmsg_request(msg_seq);
 571         if (!nlmsg_request) {
 572                 pr_info("%s: Could not find a matching request (seq = %u)\n",
 573                                  __func__, msg_seq);
 574                 return -EINVAL;
 575         }
 576         pm_msg = nlmsg_request->req_buffer;
 577         local_sockaddr = (struct sockaddr_storage *)
 578                         nla_data(nltb[IWPM_NLA_RQUERY_LOCAL_ADDR]);
 579         remote_sockaddr = (struct sockaddr_storage *)
 580                         nla_data(nltb[IWPM_NLA_RQUERY_REMOTE_ADDR]);
 581         mapped_loc_sockaddr = (struct sockaddr_storage *)
 582                         nla_data(nltb[IWPM_NLA_RQUERY_MAPPED_LOC_ADDR]);
 583         mapped_rem_sockaddr = (struct sockaddr_storage *)
 584                         nla_data(nltb[IWPM_NLA_RQUERY_MAPPED_REM_ADDR]);
 585
 586         err_code = nla_get_u16(nltb[IWPM_NLA_RQUERY_MAPPING_ERR]);
 587         if (err_code == IWPM_REMOTE_QUERY_REJECT) {
 588                 pr_info("%s: Received a Reject (pid = %u, echo seq = %u)\n",
 589                         __func__, cb->nlh->nlmsg_pid, msg_seq);
 590                 nlmsg_request->err_code = IWPM_REMOTE_QUERY_REJECT;
 591         }
 592         if (iwpm_compare_sockaddr(local_sockaddr, &pm_msg->loc_addr) ||
 593                 iwpm_compare_sockaddr(remote_sockaddr, &pm_msg->rem_addr)) {
 594                 pr_info("%s: Incorrect local sockaddr\n", __func__);
 595                 nlmsg_request->err_code = IWPM_USER_LIB_INFO_ERR;
 596                 goto query_mapping_response_exit;
 597         }
 598         if (mapped_loc_sockaddr->ss_family != local_sockaddr->ss_family ||
 599                 mapped_rem_sockaddr->ss_family != remote_sockaddr->ss_family) {
 600                 pr_info("%s: Sockaddr family doesn't match the requested one\n",
 601                                 __func__);
 602                 nlmsg_request->err_code = IWPM_USER_LIB_INFO_ERR;
 603                 goto query_mapping_response_exit;
 604         }
 605         memcpy(&pm_msg->mapped_loc_addr, mapped_loc_sockaddr,
 606                         sizeof(*mapped_loc_sockaddr));
 607         memcpy(&pm_msg->mapped_rem_addr, mapped_rem_sockaddr,
 608                         sizeof(*mapped_rem_sockaddr));
 609
 610         iwpm_print_sockaddr(&pm_msg->loc_addr,
 611                         "query_mapping: Local sockaddr:");
 612         iwpm_print_sockaddr(&pm_msg->mapped_loc_addr,
 613                         "query_mapping: Mapped local sockaddr:");
 614         iwpm_print_sockaddr(&pm_msg->rem_addr,
 615                         "query_mapping: Remote sockaddr:");
 616         iwpm_print_sockaddr(&pm_msg->mapped_rem_addr,
 617                         "query_mapping: Mapped remote sockaddr:");
 618 query_mapping_response_exit:
 619         nlmsg_request->request_done = 1;
 620         /* always for found request */
 621         kref_put(&nlmsg_request->kref, iwpm_free_nlmsg_request);
 622         barrier();
 623         up(&nlmsg_request->sem);
 624         return 0;
 625 }
 626
 627 /**
 628  * iwpm_remote_info_cb - Process remote connecting peer address info, which
 629  *                       the port mapper has received from the connecting peer
 630  * @skb:
 631  * @cb: Contains the received message (payload and netlink header)
 632  *
 633  * Stores the IPv4/IPv6 address info in a hash table
 634  */
 635 int iwpm_remote_info_cb(struct sk_buff *skb, struct netlink_callback *cb)
 636 {
 637         struct nlattr *nltb[IWPM_NLA_RQUERY_MAPPING_MAX];
 638         struct sockaddr_storage *local_sockaddr, *remote_sockaddr;
 639         struct sockaddr_storage *mapped_loc_sockaddr, *mapped_rem_sockaddr;
 640         struct iwpm_remote_info *rem_info;
 641         const char *msg_type;
 642         u8 nl_client;
 643         int ret = -EINVAL;
 644
 645         msg_type = "Remote Mapping info";
 646         if (iwpm_parse_nlmsg(cb, IWPM_NLA_RQUERY_MAPPING_MAX,
 647                                 resp_query_policy, nltb, msg_type))
 648                 return ret;
 649
 650         nl_client = RDMA_NL_GET_CLIENT(cb->nlh->nlmsg_type);
 651         if (!iwpm_valid_client(nl_client)) {
 652                 pr_info("%s: Invalid port mapper client = %d\n",
 653                                 __func__, nl_client);
 654                 return ret;
 655         }
 656         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 657
 658         local_sockaddr = (struct sockaddr_storage *)
 659                         nla_data(nltb[IWPM_NLA_RQUERY_LOCAL_ADDR]);
 660         remote_sockaddr = (struct sockaddr_storage *)
 661                         nla_data(nltb[IWPM_NLA_RQUERY_REMOTE_ADDR]);
 662         mapped_loc_sockaddr = (struct sockaddr_storage *)
 663                         nla_data(nltb[IWPM_NLA_RQUERY_MAPPED_LOC_ADDR]);
 664         mapped_rem_sockaddr = (struct sockaddr_storage *)
 665                         nla_data(nltb[IWPM_NLA_RQUERY_MAPPED_REM_ADDR]);
 666
 667         if (mapped_loc_sockaddr->ss_family != local_sockaddr->ss_family ||
 668                 mapped_rem_sockaddr->ss_family != remote_sockaddr->ss_family) {
 669                 pr_info("%s: Sockaddr family doesn't match the requested one\n",
 670                                 __func__);
 671                 return ret;
 672         }
 673         rem_info = kzalloc(sizeof(struct iwpm_remote_info), GFP_ATOMIC);
 674         if (!rem_info) {
 675                 ret = -ENOMEM;
 676                 return ret;
 677         }
 678         memcpy(&rem_info->mapped_loc_sockaddr, mapped_loc_sockaddr,
 679                sizeof(struct sockaddr_storage));
 680         memcpy(&rem_info->remote_sockaddr, remote_sockaddr,
 681                sizeof(struct sockaddr_storage));
 682         memcpy(&rem_info->mapped_rem_sockaddr, mapped_rem_sockaddr,
 683                sizeof(struct sockaddr_storage));
 684         rem_info->nl_client = nl_client;
 685
 686         iwpm_add_remote_info(rem_info);
 687
 688         iwpm_print_sockaddr(local_sockaddr,
 689                         "remote_info: Local sockaddr:");
 690         iwpm_print_sockaddr(mapped_loc_sockaddr,
 691                         "remote_info: Mapped local sockaddr:");
 692         iwpm_print_sockaddr(remote_sockaddr,
 693                         "remote_info: Remote sockaddr:");
 694         iwpm_print_sockaddr(mapped_rem_sockaddr,
 695                         "remote_info: Mapped remote sockaddr:");
 696         return ret;
 697 }
 698
 699 /* netlink attribute policy for the received request for mapping info */
 700 static const struct nla_policy resp_mapinfo_policy[IWPM_NLA_MAPINFO_REQ_MAX] = {
 701         [IWPM_NLA_MAPINFO_ULIB_NAME] = { .type = NLA_STRING,
 702                                         .len = IWPM_ULIBNAME_SIZE - 1 },
 703         [IWPM_NLA_MAPINFO_ULIB_VER]  = { .type = NLA_U16 }
 704 };
 705
 706 /**
 707  * iwpm_mapping_info_cb - Process a notification that the userspace
 708  *                        port mapper daemon is started
 709  * @skb:
 710  * @cb: Contains the received message (payload and netlink header)
 711  *
 712  * Using the received port mapper pid, send all the local mapping
 713  * info records to the userspace port mapper
 714  */
 715 int iwpm_mapping_info_cb(struct sk_buff *skb, struct netlink_callback *cb)
 716 {
 717         struct nlattr *nltb[IWPM_NLA_MAPINFO_REQ_MAX];
 718         const char *msg_type = "Mapping Info response";
 719         u8 nl_client;
 720         char *iwpm_name;
 721         u16 iwpm_version;
 722         int ret = -EINVAL;
 723
 724         if (iwpm_parse_nlmsg(cb, IWPM_NLA_MAPINFO_REQ_MAX,
 725                                 resp_mapinfo_policy, nltb, msg_type)) {
 726                 pr_info("%s: Unable to parse nlmsg\n", __func__);
 727                 return ret;
 728         }
 729         iwpm_name = (char *)nla_data(nltb[IWPM_NLA_MAPINFO_ULIB_NAME]);
 730         iwpm_version = nla_get_u16(nltb[IWPM_NLA_MAPINFO_ULIB_VER]);
 731         if (strcmp(iwpm_ulib_name, iwpm_name) ||
 732                         iwpm_version < IWPM_UABI_VERSION_MIN) {
 733                 pr_info("%s: Invalid port mapper name = %s version = %d\n",
 734                                 __func__, iwpm_name, iwpm_version);
 735                 return ret;
 736         }
 737         nl_client = RDMA_NL_GET_CLIENT(cb->nlh->nlmsg_type);
 738         if (!iwpm_valid_client(nl_client)) {
 739                 pr_info("%s: Invalid port mapper client = %d\n",
 740                                 __func__, nl_client);
 741                 return ret;
 742         }
 743         iwpm_set_registration(nl_client, IWPM_REG_INCOMPL);
 744         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 745         iwpm_user_pid = cb->nlh->nlmsg_pid;
 746
 747         if (iwpm_ulib_version < IWPM_UABI_VERSION)
 748                 pr_warn_once("%s: Down level iwpmd/pid %u.  Continuing...",
 749                         __func__, iwpm_user_pid);
 750
 751         if (!iwpm_mapinfo_available())
 752                 return 0;
 753         pr_debug("%s: iWarp Port Mapper (pid = %d) is available!\n",
 754                  __func__, iwpm_user_pid);
 755         ret = iwpm_send_mapinfo(nl_client, iwpm_user_pid);
 756         return ret;
 757 }
 758
 759 /* netlink attribute policy for the received mapping info ack */
 760 static const struct nla_policy ack_mapinfo_policy[IWPM_NLA_MAPINFO_NUM_MAX] = {
 761         [IWPM_NLA_MAPINFO_SEQ]    =   { .type = NLA_U32 },
 762         [IWPM_NLA_MAPINFO_SEND_NUM] = { .type = NLA_U32 },
 763         [IWPM_NLA_MAPINFO_ACK_NUM] =  { .type = NLA_U32 }
 764 };
 765
 766 /**
 767  * iwpm_ack_mapping_info_cb - Process the port mapper ack for
 768  *                            the provided local mapping info records
 769  * @skb:
 770  * @cb: Contains the received message (payload and netlink header)
 771  */
 772 int iwpm_ack_mapping_info_cb(struct sk_buff *skb, struct netlink_callback *cb)
 773 {
 774         struct nlattr *nltb[IWPM_NLA_MAPINFO_NUM_MAX];
 775         u32 mapinfo_send, mapinfo_ack;
 776         const char *msg_type = "Mapping Info Ack";
 777
 778         if (iwpm_parse_nlmsg(cb, IWPM_NLA_MAPINFO_NUM_MAX,
 779                                 ack_mapinfo_policy, nltb, msg_type))
 780                 return -EINVAL;
 781         mapinfo_send = nla_get_u32(nltb[IWPM_NLA_MAPINFO_SEND_NUM]);
 782         mapinfo_ack = nla_get_u32(nltb[IWPM_NLA_MAPINFO_ACK_NUM]);
 783         if (mapinfo_ack != mapinfo_send)
 784                 pr_info("%s: Invalid mapinfo number (sent = %u ack-ed = %u)\n",
 785                         __func__, mapinfo_send, mapinfo_ack);
 786         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 787         return 0;
 788 }
 789
 790 /* netlink attribute policy for the received port mapper error message */
 791 static const struct nla_policy map_error_policy[IWPM_NLA_ERR_MAX] = {
 792         [IWPM_NLA_ERR_SEQ]        = { .type = NLA_U32 },
 793         [IWPM_NLA_ERR_CODE]       = { .type = NLA_U16 },
 794 };
 795
 796 /**
 797  * iwpm_mapping_error_cb - Process port mapper notification for error
 798  *
 799  * @skb:
 800  * @cb: Contains the received message (payload and netlink header)
 801  */
 802 int iwpm_mapping_error_cb(struct sk_buff *skb, struct netlink_callback *cb)
 803 {
 804         struct iwpm_nlmsg_request *nlmsg_request = NULL;
 805         int nl_client = RDMA_NL_GET_CLIENT(cb->nlh->nlmsg_type);
 806         struct nlattr *nltb[IWPM_NLA_ERR_MAX];
 807         u32 msg_seq;
 808         u16 err_code;
 809         const char *msg_type = "Mapping Error Msg";
 810
 811         if (iwpm_parse_nlmsg(cb, IWPM_NLA_ERR_MAX,
 812                                 map_error_policy, nltb, msg_type))
 813                 return -EINVAL;
 814
 815         msg_seq = nla_get_u32(nltb[IWPM_NLA_ERR_SEQ]);
 816         err_code = nla_get_u16(nltb[IWPM_NLA_ERR_CODE]);
 817         pr_info("%s: Received msg seq = %u err code = %u client = %d\n",
 818                                 __func__, msg_seq, err_code, nl_client);
 819         /* look for nlmsg_request */
 820         nlmsg_request = iwpm_find_nlmsg_request(msg_seq);
 821         if (!nlmsg_request) {
 822                 /* not all errors have associated requests */
 823                 pr_debug("Could not find matching req (seq = %u)\n", msg_seq);
 824                 return 0;
 825         }
 826         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 827         nlmsg_request->err_code = err_code;
 828         nlmsg_request->request_done = 1;
 829         /* always for found request */
 830         kref_put(&nlmsg_request->kref, iwpm_free_nlmsg_request);
 831         barrier();
 832         up(&nlmsg_request->sem);
 833         return 0;
 834 }
 835
 836 /* netlink attribute policy for the received hello request */
 837 static const struct nla_policy hello_policy[IWPM_NLA_HELLO_MAX] = {
 838         [IWPM_NLA_HELLO_ABI_VERSION]     = { .type = NLA_U16 }
 839 };
 840
 841 /**
 842  * iwpm_hello_cb - Process a hello message from iwpmd
 843  *
 844  * @skb:
 845  * @cb: Contains the received message (payload and netlink header)
 846  *
 847  * Using the received port mapper pid, send the kernel's abi_version
 848  * after adjusting it to support the iwpmd version.
 849  */
 850 int iwpm_hello_cb(struct sk_buff *skb, struct netlink_callback *cb)
 851 {
 852         struct nlattr *nltb[IWPM_NLA_HELLO_MAX];
 853         const char *msg_type = "Hello request";
 854         u8 nl_client;
 855         u16 abi_version;
 856         int ret = -EINVAL;
 857
 858         if (iwpm_parse_nlmsg(cb, IWPM_NLA_HELLO_MAX, hello_policy, nltb,
 859                              msg_type)) {
 860                 pr_info("%s: Unable to parse nlmsg\n", __func__);
 861                 return ret;
 862         }
 863         abi_version = nla_get_u16(nltb[IWPM_NLA_HELLO_ABI_VERSION]);
 864         nl_client = RDMA_NL_GET_CLIENT(cb->nlh->nlmsg_type);
 865         if (!iwpm_valid_client(nl_client)) {
 866                 pr_info("%s: Invalid port mapper client = %d\n",
 867                                 __func__, nl_client);
 868                 return ret;
 869         }
 870         iwpm_set_registration(nl_client, IWPM_REG_INCOMPL);
 871         atomic_set(&echo_nlmsg_seq, cb->nlh->nlmsg_seq);
 872         iwpm_ulib_version = min_t(u16, IWPM_UABI_VERSION, abi_version);
 873         pr_debug("Using ABI version %u\n", iwpm_ulib_version);
 874         iwpm_user_pid = cb->nlh->nlmsg_pid;
 875         ret = iwpm_send_hello(nl_client, iwpm_user_pid, iwpm_ulib_version);
 876         return ret;
 877 }