drivers/infiniband/core/security.c

   1 /*
   2  * Copyright (c) 2016 Mellanox Technologies Ltd.  All rights reserved.
   3  *
   4  * This software is available to you under a choice of one of two
   5  * licenses.  You may choose to be licensed under the terms of the GNU
   6  * General Public License (GPL) Version 2, available from the file
   7  * COPYING in the main directory of this source tree, or the
   8  * OpenIB.org BSD license below:
   9  *
  10  *     Redistribution and use in source and binary forms, with or
  11  *     without modification, are permitted provided that the following
  12  *     conditions are met:
  13  *
  14  *      - Redistributions of source code must retain the above
  15  *        copyright notice, this list of conditions and the following
  16  *        disclaimer.
  17  *
  18  *      - Redistributions in binary form must reproduce the above
  19  *        copyright notice, this list of conditions and the following
  20  *        disclaimer in the documentation and/or other materials
  21  *        provided with the distribution.
  22  *
  23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  30  * SOFTWARE.
  31  */
  32
  33 #include <linux/security.h>
  34 #include <linux/completion.h>
  35 #include <linux/list.h>
  36
  37 #include <rdma/ib_verbs.h>
  38 #include <rdma/ib_cache.h>
  39 #include "core_priv.h"
  40 #include "mad_priv.h"
  41
  42 static LIST_HEAD(mad_agent_list);
  43 /* Lock to protect mad_agent_list */
  44 static DEFINE_SPINLOCK(mad_agent_list_lock);
  45
  46 static struct pkey_index_qp_list *get_pkey_idx_qp_list(struct ib_port_pkey *pp)
  47 {
  48         struct pkey_index_qp_list *pkey = NULL;
  49         struct pkey_index_qp_list *tmp_pkey;
  50         struct ib_device *dev = pp->sec->dev;
  51
  52         spin_lock(&dev->port_data[pp->port_num].pkey_list_lock);
  53         list_for_each_entry (tmp_pkey, &dev->port_data[pp->port_num].pkey_list,
  54                              pkey_index_list) {
  55                 if (tmp_pkey->pkey_index == pp->pkey_index) {
  56                         pkey = tmp_pkey;
  57                         break;
  58                 }
  59         }
  60         spin_unlock(&dev->port_data[pp->port_num].pkey_list_lock);
  61         return pkey;
  62 }
  63
  64 static int get_pkey_and_subnet_prefix(struct ib_port_pkey *pp,
  65                                       u16 *pkey,
  66                                       u64 *subnet_prefix)
  67 {
  68         struct ib_device *dev = pp->sec->dev;
  69         int ret;
  70
  71         ret = ib_get_cached_pkey(dev, pp->port_num, pp->pkey_index, pkey);
  72         if (ret)
  73                 return ret;
  74
  75         ret = ib_get_cached_subnet_prefix(dev, pp->port_num, subnet_prefix);
  76
  77         return ret;
  78 }
  79
  80 static int enforce_qp_pkey_security(u16 pkey,
  81                                     u64 subnet_prefix,
  82                                     struct ib_qp_security *qp_sec)
  83 {
  84         struct ib_qp_security *shared_qp_sec;
  85         int ret;
  86
  87         ret = security_ib_pkey_access(qp_sec->security, subnet_prefix, pkey);
  88         if (ret)
  89                 return ret;
  90
  91         list_for_each_entry(shared_qp_sec,
  92                             &qp_sec->shared_qp_list,
  93                             shared_qp_list) {
  94                 ret = security_ib_pkey_access(shared_qp_sec->security,
  95                                               subnet_prefix,
  96                                               pkey);
  97                 if (ret)
  98                         return ret;
  99         }
 100         return 0;
 101 }
 102
 103 /* The caller of this function must hold the QP security
 104  * mutex of the QP of the security structure in *pps.
 105  *
 106  * It takes separate ports_pkeys and security structure
 107  * because in some cases the pps will be for a new settings
 108  * or the pps will be for the real QP and security structure
 109  * will be for a shared QP.
 110  */
 111 static int check_qp_port_pkey_settings(struct ib_ports_pkeys *pps,
 112                                        struct ib_qp_security *sec)
 113 {
 114         u64 subnet_prefix;
 115         u16 pkey;
 116         int ret = 0;
 117
 118         if (!pps)
 119                 return 0;
 120
 121         if (pps->main.state != IB_PORT_PKEY_NOT_VALID) {
 122                 ret = get_pkey_and_subnet_prefix(&pps->main,
 123                                                  &pkey,
 124                                                  &subnet_prefix);
 125                 if (ret)
 126                         return ret;
 127
 128                 ret = enforce_qp_pkey_security(pkey,
 129                                                subnet_prefix,
 130                                                sec);
 131                 if (ret)
 132                         return ret;
 133         }
 134
 135         if (pps->alt.state != IB_PORT_PKEY_NOT_VALID) {
 136                 ret = get_pkey_and_subnet_prefix(&pps->alt,
 137                                                  &pkey,
 138                                                  &subnet_prefix);
 139                 if (ret)
 140                         return ret;
 141
 142                 ret = enforce_qp_pkey_security(pkey,
 143                                                subnet_prefix,
 144                                                sec);
 145         }
 146
 147         return ret;
 148 }
 149
 150 /* The caller of this function must hold the QP security
 151  * mutex.
 152  */
 153 static void qp_to_error(struct ib_qp_security *sec)
 154 {
 155         struct ib_qp_security *shared_qp_sec;
 156         struct ib_qp_attr attr = {
 157                 .qp_state = IB_QPS_ERR
 158         };
 159         struct ib_event event = {
 160                 .event = IB_EVENT_QP_FATAL
 161         };
 162
 163         /* If the QP is in the process of being destroyed
 164          * the qp pointer in the security structure is
 165          * undefined.  It cannot be modified now.
 166          */
 167         if (sec->destroying)
 168                 return;
 169
 170         ib_modify_qp(sec->qp,
 171                      &attr,
 172                      IB_QP_STATE);
 173
 174         if (sec->qp->event_handler && sec->qp->qp_context) {
 175                 event.element.qp = sec->qp;
 176                 sec->qp->event_handler(&event,
 177                                        sec->qp->qp_context);
 178         }
 179
 180         list_for_each_entry(shared_qp_sec,
 181                             &sec->shared_qp_list,
 182                             shared_qp_list) {
 183                 struct ib_qp *qp = shared_qp_sec->qp;
 184
 185                 if (qp->event_handler && qp->qp_context) {
 186                         event.element.qp = qp;
 187                         event.device = qp->device;
 188                         qp->event_handler(&event,
 189                                           qp->qp_context);
 190                 }
 191         }
 192 }
 193
 194 static inline void check_pkey_qps(struct pkey_index_qp_list *pkey,
 195                                   struct ib_device *device,
 196                                   u8 port_num,
 197                                   u64 subnet_prefix)
 198 {
 199         struct ib_port_pkey *pp, *tmp_pp;
 200         bool comp;
 201         LIST_HEAD(to_error_list);
 202         u16 pkey_val;
 203
 204         if (!ib_get_cached_pkey(device,
 205                                 port_num,
 206                                 pkey->pkey_index,
 207                                 &pkey_val)) {
 208                 spin_lock(&pkey->qp_list_lock);
 209                 list_for_each_entry(pp, &pkey->qp_list, qp_list) {
 210                         if (atomic_read(&pp->sec->error_list_count))
 211                                 continue;
 212
 213                         if (enforce_qp_pkey_security(pkey_val,
 214                                                      subnet_prefix,
 215                                                      pp->sec)) {
 216                                 atomic_inc(&pp->sec->error_list_count);
 217                                 list_add(&pp->to_error_list,
 218                                          &to_error_list);
 219                         }
 220                 }
 221                 spin_unlock(&pkey->qp_list_lock);
 222         }
 223
 224         list_for_each_entry_safe(pp,
 225                                  tmp_pp,
 226                                  &to_error_list,
 227                                  to_error_list) {
 228                 mutex_lock(&pp->sec->mutex);
 229                 qp_to_error(pp->sec);
 230                 list_del(&pp->to_error_list);
 231                 atomic_dec(&pp->sec->error_list_count);
 232                 comp = pp->sec->destroying;
 233                 mutex_unlock(&pp->sec->mutex);
 234
 235                 if (comp)
 236                         complete(&pp->sec->error_complete);
 237         }
 238 }
 239
 240 /* The caller of this function must hold the QP security
 241  * mutex.
 242  */
 243 static int port_pkey_list_insert(struct ib_port_pkey *pp)
 244 {
 245         struct pkey_index_qp_list *tmp_pkey;
 246         struct pkey_index_qp_list *pkey;
 247         struct ib_device *dev;
 248         u8 port_num = pp->port_num;
 249         int ret = 0;
 250
 251         if (pp->state != IB_PORT_PKEY_VALID)
 252                 return 0;
 253
 254         dev = pp->sec->dev;
 255
 256         pkey = get_pkey_idx_qp_list(pp);
 257
 258         if (!pkey) {
 259                 bool found = false;
 260
 261                 pkey = kzalloc(sizeof(*pkey), GFP_KERNEL);
 262                 if (!pkey)
 263                         return -ENOMEM;
 264
 265                 spin_lock(&dev->port_data[port_num].pkey_list_lock);
 266                 /* Check for the PKey again.  A racing process may
 267                  * have created it.
 268                  */
 269                 list_for_each_entry(tmp_pkey,
 270                                     &dev->port_data[port_num].pkey_list,
 271                                     pkey_index_list) {
 272                         if (tmp_pkey->pkey_index == pp->pkey_index) {
 273                                 kfree(pkey);
 274                                 pkey = tmp_pkey;
 275                                 found = true;
 276                                 break;
 277                         }
 278                 }
 279
 280                 if (!found) {
 281                         pkey->pkey_index = pp->pkey_index;
 282                         spin_lock_init(&pkey->qp_list_lock);
 283                         INIT_LIST_HEAD(&pkey->qp_list);
 284                         list_add(&pkey->pkey_index_list,
 285                                  &dev->port_data[port_num].pkey_list);
 286                 }
 287                 spin_unlock(&dev->port_data[port_num].pkey_list_lock);
 288         }
 289
 290         spin_lock(&pkey->qp_list_lock);
 291         list_add(&pp->qp_list, &pkey->qp_list);
 292         spin_unlock(&pkey->qp_list_lock);
 293
 294         pp->state = IB_PORT_PKEY_LISTED;
 295
 296         return ret;
 297 }
 298
 299 /* The caller of this function must hold the QP security
 300  * mutex.
 301  */
 302 static void port_pkey_list_remove(struct ib_port_pkey *pp)
 303 {
 304         struct pkey_index_qp_list *pkey;
 305
 306         if (pp->state != IB_PORT_PKEY_LISTED)
 307                 return;
 308
 309         pkey = get_pkey_idx_qp_list(pp);
 310
 311         spin_lock(&pkey->qp_list_lock);
 312         list_del(&pp->qp_list);
 313         spin_unlock(&pkey->qp_list_lock);
 314
 315         /* The setting may still be valid, i.e. after
 316          * a destroy has failed for example.
 317          */
 318         pp->state = IB_PORT_PKEY_VALID;
 319 }
 320
 321 static void destroy_qp_security(struct ib_qp_security *sec)
 322 {
 323         security_ib_free_security(sec->security);
 324         kfree(sec->ports_pkeys);
 325         kfree(sec);
 326 }
 327
 328 /* The caller of this function must hold the QP security
 329  * mutex.
 330  */
 331 static struct ib_ports_pkeys *get_new_pps(const struct ib_qp *qp,
 332                                           const struct ib_qp_attr *qp_attr,
 333                                           int qp_attr_mask)
 334 {
 335         struct ib_ports_pkeys *new_pps;
 336         struct ib_ports_pkeys *qp_pps = qp->qp_sec->ports_pkeys;
 337
 338         new_pps = kzalloc(sizeof(*new_pps), GFP_KERNEL);
 339         if (!new_pps)
 340                 return NULL;
 341
 342         if (qp_attr_mask & (IB_QP_PKEY_INDEX | IB_QP_PORT)) {
 343                 if (!qp_pps) {
 344                         new_pps->main.port_num = qp_attr->port_num;
 345                         new_pps->main.pkey_index = qp_attr->pkey_index;
 346                 } else {
 347                         new_pps->main.port_num = (qp_attr_mask & IB_QP_PORT) ?
 348                                                   qp_attr->port_num :
 349                                                   qp_pps->main.port_num;
 350
 351                         new_pps->main.pkey_index =
 352                                         (qp_attr_mask & IB_QP_PKEY_INDEX) ?
 353                                          qp_attr->pkey_index :
 354                                          qp_pps->main.pkey_index;
 355                 }
 356                 new_pps->main.state = IB_PORT_PKEY_VALID;
 357         } else if (qp_pps) {
 358                 new_pps->main.port_num = qp_pps->main.port_num;
 359                 new_pps->main.pkey_index = qp_pps->main.pkey_index;
 360                 if (qp_pps->main.state != IB_PORT_PKEY_NOT_VALID)
 361                         new_pps->main.state = IB_PORT_PKEY_VALID;
 362         }
 363
 364         if (qp_attr_mask & IB_QP_ALT_PATH) {
 365                 new_pps->alt.port_num = qp_attr->alt_port_num;
 366                 new_pps->alt.pkey_index = qp_attr->alt_pkey_index;
 367                 new_pps->alt.state = IB_PORT_PKEY_VALID;
 368         } else if (qp_pps) {
 369                 new_pps->alt.port_num = qp_pps->alt.port_num;
 370                 new_pps->alt.pkey_index = qp_pps->alt.pkey_index;
 371                 if (qp_pps->alt.state != IB_PORT_PKEY_NOT_VALID)
 372                         new_pps->alt.state = IB_PORT_PKEY_VALID;
 373         }
 374
 375         new_pps->main.sec = qp->qp_sec;
 376         new_pps->alt.sec = qp->qp_sec;
 377         return new_pps;
 378 }
 379
 380 int ib_open_shared_qp_security(struct ib_qp *qp, struct ib_device *dev)
 381 {
 382         struct ib_qp *real_qp = qp->real_qp;
 383         int ret;
 384
 385         ret = ib_create_qp_security(qp, dev);
 386
 387         if (ret)
 388                 return ret;
 389
 390         if (!qp->qp_sec)
 391                 return 0;
 392
 393         mutex_lock(&real_qp->qp_sec->mutex);
 394         ret = check_qp_port_pkey_settings(real_qp->qp_sec->ports_pkeys,
 395                                           qp->qp_sec);
 396
 397         if (ret)
 398                 goto ret;
 399
 400         if (qp != real_qp)
 401                 list_add(&qp->qp_sec->shared_qp_list,
 402                          &real_qp->qp_sec->shared_qp_list);
 403 ret:
 404         mutex_unlock(&real_qp->qp_sec->mutex);
 405         if (ret)
 406                 destroy_qp_security(qp->qp_sec);
 407
 408         return ret;
 409 }
 410
 411 void ib_close_shared_qp_security(struct ib_qp_security *sec)
 412 {
 413         struct ib_qp *real_qp = sec->qp->real_qp;
 414
 415         mutex_lock(&real_qp->qp_sec->mutex);
 416         list_del(&sec->shared_qp_list);
 417         mutex_unlock(&real_qp->qp_sec->mutex);
 418
 419         destroy_qp_security(sec);
 420 }
 421
 422 int ib_create_qp_security(struct ib_qp *qp, struct ib_device *dev)
 423 {
 424         unsigned int i;
 425         bool is_ib = false;
 426         int ret;
 427
 428         rdma_for_each_port (dev, i) {
 429                 is_ib = rdma_protocol_ib(dev, i);
 430                 if (is_ib)
 431                         break;
 432         }
 433
 434         /* If this isn't an IB device don't create the security context */
 435         if (!is_ib)
 436                 return 0;
 437
 438         qp->qp_sec = kzalloc(sizeof(*qp->qp_sec), GFP_KERNEL);
 439         if (!qp->qp_sec)
 440                 return -ENOMEM;
 441
 442         qp->qp_sec->qp = qp;
 443         qp->qp_sec->dev = dev;
 444         mutex_init(&qp->qp_sec->mutex);
 445         INIT_LIST_HEAD(&qp->qp_sec->shared_qp_list);
 446         atomic_set(&qp->qp_sec->error_list_count, 0);
 447         init_completion(&qp->qp_sec->error_complete);
 448         ret = security_ib_alloc_security(&qp->qp_sec->security);
 449         if (ret) {
 450                 kfree(qp->qp_sec);
 451                 qp->qp_sec = NULL;
 452         }
 453
 454         return ret;
 455 }
 456 EXPORT_SYMBOL(ib_create_qp_security);
 457
 458 void ib_destroy_qp_security_begin(struct ib_qp_security *sec)
 459 {
 460         /* Return if not IB */
 461         if (!sec)
 462                 return;
 463
 464         mutex_lock(&sec->mutex);
 465
 466         /* Remove the QP from the lists so it won't get added to
 467          * a to_error_list during the destroy process.
 468          */
 469         if (sec->ports_pkeys) {
 470                 port_pkey_list_remove(&sec->ports_pkeys->main);
 471                 port_pkey_list_remove(&sec->ports_pkeys->alt);
 472         }
 473
 474         /* If the QP is already in one or more of those lists
 475          * the destroying flag will ensure the to error flow
 476          * doesn't operate on an undefined QP.
 477          */
 478         sec->destroying = true;
 479
 480         /* Record the error list count to know how many completions
 481          * to wait for.
 482          */
 483         sec->error_comps_pending = atomic_read(&sec->error_list_count);
 484
 485         mutex_unlock(&sec->mutex);
 486 }
 487
 488 void ib_destroy_qp_security_abort(struct ib_qp_security *sec)
 489 {
 490         int ret;
 491         int i;
 492
 493         /* Return if not IB */
 494         if (!sec)
 495                 return;
 496
 497         /* If a concurrent cache update is in progress this
 498          * QP security could be marked for an error state
 499          * transition.  Wait for this to complete.
 500          */
 501         for (i = 0; i < sec->error_comps_pending; i++)
 502                 wait_for_completion(&sec->error_complete);
 503
 504         mutex_lock(&sec->mutex);
 505         sec->destroying = false;
 506
 507         /* Restore the position in the lists and verify
 508          * access is still allowed in case a cache update
 509          * occurred while attempting to destroy.
 510          *
 511          * Because these setting were listed already
 512          * and removed during ib_destroy_qp_security_begin
 513          * we know the pkey_index_qp_list for the PKey
 514          * already exists so port_pkey_list_insert won't fail.
 515          */
 516         if (sec->ports_pkeys) {
 517                 port_pkey_list_insert(&sec->ports_pkeys->main);
 518                 port_pkey_list_insert(&sec->ports_pkeys->alt);
 519         }
 520
 521         ret = check_qp_port_pkey_settings(sec->ports_pkeys, sec);
 522         if (ret)
 523                 qp_to_error(sec);
 524
 525         mutex_unlock(&sec->mutex);
 526 }
 527
 528 void ib_destroy_qp_security_end(struct ib_qp_security *sec)
 529 {
 530         int i;
 531
 532         /* Return if not IB */
 533         if (!sec)
 534                 return;
 535
 536         /* If a concurrent cache update is occurring we must
 537          * wait until this QP security structure is processed
 538          * in the QP to error flow before destroying it because
 539          * the to_error_list is in use.
 540          */
 541         for (i = 0; i < sec->error_comps_pending; i++)
 542                 wait_for_completion(&sec->error_complete);
 543
 544         destroy_qp_security(sec);
 545 }
 546
 547 void ib_security_cache_change(struct ib_device *device,
 548                               u8 port_num,
 549                               u64 subnet_prefix)
 550 {
 551         struct pkey_index_qp_list *pkey;
 552
 553         list_for_each_entry (pkey, &device->port_data[port_num].pkey_list,
 554                              pkey_index_list) {
 555                 check_pkey_qps(pkey,
 556                                device,
 557                                port_num,
 558                                subnet_prefix);
 559         }
 560 }
 561
 562 void ib_security_release_port_pkey_list(struct ib_device *device)
 563 {
 564         struct pkey_index_qp_list *pkey, *tmp_pkey;
 565         unsigned int i;
 566
 567         rdma_for_each_port (device, i) {
 568                 list_for_each_entry_safe(pkey,
 569                                          tmp_pkey,
 570                                          &device->port_data[i].pkey_list,
 571                                          pkey_index_list) {
 572                         list_del(&pkey->pkey_index_list);
 573                         kfree(pkey);
 574                 }
 575         }
 576 }
 577
 578 int ib_security_modify_qp(struct ib_qp *qp,
 579                           struct ib_qp_attr *qp_attr,
 580                           int qp_attr_mask,
 581                           struct ib_udata *udata)
 582 {
 583         int ret = 0;
 584         struct ib_ports_pkeys *tmp_pps;
 585         struct ib_ports_pkeys *new_pps = NULL;
 586         struct ib_qp *real_qp = qp->real_qp;
 587         bool special_qp = (real_qp->qp_type == IB_QPT_SMI ||
 588                            real_qp->qp_type == IB_QPT_GSI ||
 589                            real_qp->qp_type >= IB_QPT_RESERVED1);
 590         bool pps_change = ((qp_attr_mask & (IB_QP_PKEY_INDEX | IB_QP_PORT)) ||
 591                            (qp_attr_mask & IB_QP_ALT_PATH));
 592
 593         WARN_ONCE((qp_attr_mask & IB_QP_PORT &&
 594                    rdma_protocol_ib(real_qp->device, qp_attr->port_num) &&
 595                    !real_qp->qp_sec),
 596                    "%s: QP security is not initialized for IB QP: %d\n",
 597                    __func__, real_qp->qp_num);
 598
 599         /* The port/pkey settings are maintained only for the real QP. Open
 600          * handles on the real QP will be in the shared_qp_list. When
 601          * enforcing security on the real QP all the shared QPs will be
 602          * checked as well.
 603          */
 604
 605         if (pps_change && !special_qp && real_qp->qp_sec) {
 606                 mutex_lock(&real_qp->qp_sec->mutex);
 607                 new_pps = get_new_pps(real_qp,
 608                                       qp_attr,
 609                                       qp_attr_mask);
 610                 if (!new_pps) {
 611                         mutex_unlock(&real_qp->qp_sec->mutex);
 612                         return -ENOMEM;
 613                 }
 614                 /* Add this QP to the lists for the new port
 615                  * and pkey settings before checking for permission
 616                  * in case there is a concurrent cache update
 617                  * occurring.  Walking the list for a cache change
 618                  * doesn't acquire the security mutex unless it's
 619                  * sending the QP to error.
 620                  */
 621                 ret = port_pkey_list_insert(&new_pps->main);
 622
 623                 if (!ret)
 624                         ret = port_pkey_list_insert(&new_pps->alt);
 625
 626                 if (!ret)
 627                         ret = check_qp_port_pkey_settings(new_pps,
 628                                                           real_qp->qp_sec);
 629         }
 630
 631         if (!ret)
 632                 ret = real_qp->device->ops.modify_qp(real_qp,
 633                                                      qp_attr,
 634                                                      qp_attr_mask,
 635                                                      udata);
 636
 637         if (new_pps) {
 638                 /* Clean up the lists and free the appropriate
 639                  * ports_pkeys structure.
 640                  */
 641                 if (ret) {
 642                         tmp_pps = new_pps;
 643                 } else {
 644                         tmp_pps = real_qp->qp_sec->ports_pkeys;
 645                         real_qp->qp_sec->ports_pkeys = new_pps;
 646                 }
 647
 648                 if (tmp_pps) {
 649                         port_pkey_list_remove(&tmp_pps->main);
 650                         port_pkey_list_remove(&tmp_pps->alt);
 651                 }
 652                 kfree(tmp_pps);
 653                 mutex_unlock(&real_qp->qp_sec->mutex);
 654         }
 655         return ret;
 656 }
 657
 658 static int ib_security_pkey_access(struct ib_device *dev,
 659                                    u8 port_num,
 660                                    u16 pkey_index,
 661                                    void *sec)
 662 {
 663         u64 subnet_prefix;
 664         u16 pkey;
 665         int ret;
 666
 667         if (!rdma_protocol_ib(dev, port_num))
 668                 return 0;
 669
 670         ret = ib_get_cached_pkey(dev, port_num, pkey_index, &pkey);
 671         if (ret)
 672                 return ret;
 673
 674         ret = ib_get_cached_subnet_prefix(dev, port_num, &subnet_prefix);
 675
 676         if (ret)
 677                 return ret;
 678
 679         return security_ib_pkey_access(sec, subnet_prefix, pkey);
 680 }
 681
 682 void ib_mad_agent_security_change(void)
 683 {
 684         struct ib_mad_agent *ag;
 685
 686         spin_lock(&mad_agent_list_lock);
 687         list_for_each_entry(ag,
 688                             &mad_agent_list,
 689                             mad_agent_sec_list)
 690                 WRITE_ONCE(ag->smp_allowed,
 691                            !security_ib_endport_manage_subnet(ag->security,
 692                                 dev_name(&ag->device->dev), ag->port_num));
 693         spin_unlock(&mad_agent_list_lock);
 694 }
 695
 696 int ib_mad_agent_security_setup(struct ib_mad_agent *agent,
 697                                 enum ib_qp_type qp_type)
 698 {
 699         int ret;
 700
 701         if (!rdma_protocol_ib(agent->device, agent->port_num))
 702                 return 0;
 703
 704         INIT_LIST_HEAD(&agent->mad_agent_sec_list);
 705
 706         ret = security_ib_alloc_security(&agent->security);
 707         if (ret)
 708                 return ret;
 709
 710         if (qp_type != IB_QPT_SMI)
 711                 return 0;
 712
 713         spin_lock(&mad_agent_list_lock);
 714         ret = security_ib_endport_manage_subnet(agent->security,
 715                                                 dev_name(&agent->device->dev),
 716                                                 agent->port_num);
 717         if (ret)
 718                 goto free_security;
 719
 720         WRITE_ONCE(agent->smp_allowed, true);
 721         list_add(&agent->mad_agent_sec_list, &mad_agent_list);
 722         spin_unlock(&mad_agent_list_lock);
 723         return 0;
 724
 725 free_security:
 726         spin_unlock(&mad_agent_list_lock);
 727         security_ib_free_security(agent->security);
 728         return ret;
 729 }
 730
 731 void ib_mad_agent_security_cleanup(struct ib_mad_agent *agent)
 732 {
 733         if (!rdma_protocol_ib(agent->device, agent->port_num))
 734                 return;
 735
 736         if (agent->qp->qp_type == IB_QPT_SMI) {
 737                 spin_lock(&mad_agent_list_lock);
 738                 list_del(&agent->mad_agent_sec_list);
 739                 spin_unlock(&mad_agent_list_lock);
 740         }
 741
 742         security_ib_free_security(agent->security);
 743 }
 744
 745 int ib_mad_enforce_security(struct ib_mad_agent_private *map, u16 pkey_index)
 746 {
 747         if (!rdma_protocol_ib(map->agent.device, map->agent.port_num))
 748                 return 0;
 749
 750         if (map->agent.qp->qp_type == IB_QPT_SMI) {
 751                 if (!READ_ONCE(map->agent.smp_allowed))
 752                         return -EACCES;
 753                 return 0;
 754         }
 755
 756         return ib_security_pkey_access(map->agent.device,
 757                                        map->agent.port_num,
 758                                        pkey_index,
 759                                        map->agent.security);
 760 }