|
|
1.1 ! root 1: /* ! 2: * Copyright (C) 2007 Michael Brown <[email protected]>. ! 3: * ! 4: * This program is free software; you can redistribute it and/or ! 5: * modify it under the terms of the GNU General Public License as ! 6: * published by the Free Software Foundation; either version 2 of the ! 7: * License, or any later version. ! 8: * ! 9: * This program is distributed in the hope that it will be useful, but ! 10: * WITHOUT ANY WARRANTY; without even the implied warranty of ! 11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU ! 12: * General Public License for more details. ! 13: * ! 14: * You should have received a copy of the GNU General Public License ! 15: * along with this program; if not, write to the Free Software ! 16: * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. ! 17: */ ! 18: ! 19: FILE_LICENCE ( GPL2_OR_LATER ); ! 20: ! 21: #include <stdint.h> ! 22: #include <string.h> ! 23: #include <byteswap.h> ! 24: #include <errno.h> ! 25: #include <ipxe/list.h> ! 26: #include <ipxe/infiniband.h> ! 27: #include <ipxe/ib_mi.h> ! 28: #include <ipxe/ib_mcast.h> ! 29: ! 30: /** @file ! 31: * ! 32: * Infiniband multicast groups ! 33: * ! 34: */ ! 35: ! 36: /** ! 37: * Generate multicast membership MAD ! 38: * ! 39: * @v ibdev Infiniband device ! 40: * @v gid Multicast GID ! 41: * @v join Join (rather than leave) group ! 42: * @v mad MAD to fill in ! 43: */ ! 44: static void ib_mcast_mad ( struct ib_device *ibdev, union ib_gid *gid, ! 45: int join, union ib_mad *mad ) { ! 46: struct ib_mad_sa *sa = &mad->sa; ! 47: ! 48: /* Construct multicast membership record request */ ! 49: memset ( sa, 0, sizeof ( *sa ) ); ! 50: sa->mad_hdr.mgmt_class = IB_MGMT_CLASS_SUBN_ADM; ! 51: sa->mad_hdr.class_version = IB_SA_CLASS_VERSION; ! 52: sa->mad_hdr.method = ! 53: ( join ? IB_MGMT_METHOD_SET : IB_MGMT_METHOD_DELETE ); ! 54: sa->mad_hdr.attr_id = htons ( IB_SA_ATTR_MC_MEMBER_REC ); ! 55: sa->sa_hdr.comp_mask[1] = ! 56: htonl ( IB_SA_MCMEMBER_REC_MGID | IB_SA_MCMEMBER_REC_PORT_GID | ! 57: IB_SA_MCMEMBER_REC_JOIN_STATE ); ! 58: sa->sa_data.mc_member_record.scope__join_state = 1; ! 59: memcpy ( &sa->sa_data.mc_member_record.mgid, gid, ! 60: sizeof ( sa->sa_data.mc_member_record.mgid ) ); ! 61: memcpy ( &sa->sa_data.mc_member_record.port_gid, &ibdev->gid, ! 62: sizeof ( sa->sa_data.mc_member_record.port_gid ) ); ! 63: } ! 64: ! 65: /** ! 66: * Handle multicast membership record join response ! 67: * ! 68: * @v ibdev Infiniband device ! 69: * @v mi Management interface ! 70: * @v madx Management transaction ! 71: * @v rc Status code ! 72: * @v mad Received MAD (or NULL on error) ! 73: * @v av Source address vector (or NULL on error) ! 74: */ ! 75: static void ib_mcast_complete ( struct ib_device *ibdev, ! 76: struct ib_mad_interface *mi __unused, ! 77: struct ib_mad_transaction *madx, ! 78: int rc, union ib_mad *mad, ! 79: struct ib_address_vector *av __unused ) { ! 80: struct ib_mc_membership *membership = ib_madx_get_ownerdata ( madx ); ! 81: struct ib_queue_pair *qp = membership->qp; ! 82: union ib_gid *gid = &membership->gid; ! 83: struct ib_mc_member_record *mc_member_record = ! 84: &mad->sa.sa_data.mc_member_record; ! 85: int joined; ! 86: unsigned long qkey; ! 87: ! 88: /* Report failures */ ! 89: if ( ( rc == 0 ) && ( mad->hdr.status != htons ( IB_MGMT_STATUS_OK ) )) ! 90: rc = -ENOTCONN; ! 91: if ( rc != 0 ) { ! 92: DBGC ( ibdev, "IBDEV %p QPN %lx join failed: %s\n", ! 93: ibdev, qp->qpn, strerror ( rc ) ); ! 94: goto out; ! 95: } ! 96: ! 97: /* Extract values from MAD */ ! 98: joined = ( mad->hdr.method == IB_MGMT_METHOD_GET_RESP ); ! 99: qkey = ntohl ( mc_member_record->qkey ); ! 100: DBGC ( ibdev, "IBDEV %p QPN %lx %s " IB_GID_FMT " qkey %lx\n", ! 101: ibdev, qp->qpn, ( joined ? "joined" : "left" ), ! 102: IB_GID_ARGS ( gid ), qkey ); ! 103: ! 104: /* Set queue key */ ! 105: qp->qkey = qkey; ! 106: if ( ( rc = ib_modify_qp ( ibdev, qp ) ) != 0 ) { ! 107: DBGC ( ibdev, "IBDEV %p QPN %lx could not modify qkey: %s\n", ! 108: ibdev, qp->qpn, strerror ( rc ) ); ! 109: goto out; ! 110: } ! 111: ! 112: out: ! 113: /* Destroy the completed transaction */ ! 114: ib_destroy_madx ( ibdev, mi, madx ); ! 115: membership->madx = NULL; ! 116: ! 117: /* Hand off to upper completion handler */ ! 118: membership->complete ( ibdev, qp, membership, rc, mad ); ! 119: } ! 120: ! 121: /** Multicast membership management transaction completion operations */ ! 122: static struct ib_mad_transaction_operations ib_mcast_op = { ! 123: .complete = ib_mcast_complete, ! 124: }; ! 125: ! 126: /** ! 127: * Join multicast group ! 128: * ! 129: * @v ibdev Infiniband device ! 130: * @v qp Queue pair ! 131: * @v membership Multicast group membership ! 132: * @v gid Multicast GID to join ! 133: * @v joined Join completion handler ! 134: * @ret rc Return status code ! 135: */ ! 136: int ib_mcast_join ( struct ib_device *ibdev, struct ib_queue_pair *qp, ! 137: struct ib_mc_membership *membership, union ib_gid *gid, ! 138: void ( * complete ) ( struct ib_device *ibdev, ! 139: struct ib_queue_pair *qp, ! 140: struct ib_mc_membership *membership, ! 141: int rc, union ib_mad *mad ) ) { ! 142: union ib_mad mad; ! 143: int rc; ! 144: ! 145: DBGC ( ibdev, "IBDEV %p QPN %lx joining " IB_GID_FMT "\n", ! 146: ibdev, qp->qpn, IB_GID_ARGS ( gid ) ); ! 147: ! 148: /* Initialise structure */ ! 149: membership->qp = qp; ! 150: memcpy ( &membership->gid, gid, sizeof ( membership->gid ) ); ! 151: membership->complete = complete; ! 152: ! 153: /* Attach queue pair to multicast GID */ ! 154: if ( ( rc = ib_mcast_attach ( ibdev, qp, gid ) ) != 0 ) { ! 155: DBGC ( ibdev, "IBDEV %p QPN %lx could not attach: %s\n", ! 156: ibdev, qp->qpn, strerror ( rc ) ); ! 157: goto err_mcast_attach; ! 158: } ! 159: ! 160: /* Initiate multicast membership join */ ! 161: ib_mcast_mad ( ibdev, gid, 1, &mad ); ! 162: membership->madx = ib_create_madx ( ibdev, ibdev->gsi, &mad, NULL, ! 163: &ib_mcast_op ); ! 164: if ( ! membership->madx ) { ! 165: DBGC ( ibdev, "IBDEV %p QPN %lx could not create join " ! 166: "transaction\n", ibdev, qp->qpn ); ! 167: rc = -ENOMEM; ! 168: goto err_create_madx; ! 169: } ! 170: ib_madx_set_ownerdata ( membership->madx, membership ); ! 171: ! 172: return 0; ! 173: ! 174: ib_destroy_madx ( ibdev, ibdev->gsi, membership->madx ); ! 175: err_create_madx: ! 176: ib_mcast_detach ( ibdev, qp, gid ); ! 177: err_mcast_attach: ! 178: return rc; ! 179: } ! 180: ! 181: /** ! 182: * Leave multicast group ! 183: * ! 184: * @v ibdev Infiniband device ! 185: * @v qp Queue pair ! 186: * @v membership Multicast group membership ! 187: */ ! 188: void ib_mcast_leave ( struct ib_device *ibdev, struct ib_queue_pair *qp, ! 189: struct ib_mc_membership *membership ) { ! 190: union ib_gid *gid = &membership->gid; ! 191: union ib_mad mad; ! 192: int rc; ! 193: ! 194: DBGC ( ibdev, "IBDEV %p QPN %lx leaving " IB_GID_FMT "\n", ! 195: ibdev, qp->qpn, IB_GID_ARGS ( gid ) ); ! 196: ! 197: /* Detach from multicast GID */ ! 198: ib_mcast_detach ( ibdev, qp, &membership->gid ); ! 199: ! 200: /* Cancel multicast membership join, if applicable */ ! 201: if ( membership->madx ) { ! 202: ib_destroy_madx ( ibdev, ibdev->gsi, membership->madx ); ! 203: membership->madx = NULL; ! 204: } ! 205: ! 206: /* Send a single group leave MAD */ ! 207: ib_mcast_mad ( ibdev, &membership->gid, 0, &mad ); ! 208: if ( ( rc = ib_mi_send ( ibdev, ibdev->gsi, &mad, NULL ) ) != 0 ) { ! 209: DBGC ( ibdev, "IBDEV %p QPN %lx could not send leave request: " ! 210: "%s\n", ibdev, qp->qpn, strerror ( rc ) ); ! 211: } ! 212: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.