236 lines
6.7 KiB
C
236 lines
6.7 KiB
C
|
/*
|
||
|
* Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
|
||
|
*
|
||
|
* This program is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU General Public License as
|
||
|
* published by the Free Software Foundation; either version 2 of the
|
||
|
* License, or any later version.
|
||
|
*
|
||
|
* This program is distributed in the hope that it will be useful, but
|
||
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU General Public License
|
||
|
* along with this program; if not, write to the Free Software
|
||
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||
|
*/
|
||
|
|
||
|
FILE_LICENCE ( GPL2_OR_LATER );
|
||
|
|
||
|
#include <stdint.h>
|
||
|
#include <string.h>
|
||
|
#include <byteswap.h>
|
||
|
#include <errno.h>
|
||
|
#include <gpxe/list.h>
|
||
|
#include <gpxe/infiniband.h>
|
||
|
#include <gpxe/ib_gma.h>
|
||
|
#include <gpxe/ib_mcast.h>
|
||
|
|
||
|
/** @file
|
||
|
*
|
||
|
* Infiniband multicast groups
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
/**
|
||
|
* Transmit multicast group membership request
|
||
|
*
|
||
|
* @v ibdev Infiniband device
|
||
|
* @v gid Multicast GID
|
||
|
* @v join Join (rather than leave) group
|
||
|
* @ret rc Return status code
|
||
|
*/
|
||
|
static int ib_mc_member_request ( struct ib_device *ibdev, struct ib_gid *gid,
|
||
|
int join ) {
|
||
|
union ib_mad mad;
|
||
|
struct ib_mad_sa *sa = &mad.sa;
|
||
|
int rc;
|
||
|
|
||
|
/* Construct multicast membership record request */
|
||
|
memset ( sa, 0, sizeof ( *sa ) );
|
||
|
sa->mad_hdr.base_version = IB_MGMT_BASE_VERSION;
|
||
|
sa->mad_hdr.mgmt_class = IB_MGMT_CLASS_SUBN_ADM;
|
||
|
sa->mad_hdr.class_version = IB_SA_CLASS_VERSION;
|
||
|
sa->mad_hdr.method =
|
||
|
( join ? IB_MGMT_METHOD_SET : IB_MGMT_METHOD_DELETE );
|
||
|
sa->mad_hdr.attr_id = htons ( IB_SA_ATTR_MC_MEMBER_REC );
|
||
|
sa->sa_hdr.comp_mask[1] =
|
||
|
htonl ( IB_SA_MCMEMBER_REC_MGID | IB_SA_MCMEMBER_REC_PORT_GID |
|
||
|
IB_SA_MCMEMBER_REC_JOIN_STATE );
|
||
|
sa->sa_data.mc_member_record.scope__join_state = 1;
|
||
|
memcpy ( &sa->sa_data.mc_member_record.mgid, gid,
|
||
|
sizeof ( sa->sa_data.mc_member_record.mgid ) );
|
||
|
memcpy ( &sa->sa_data.mc_member_record.port_gid, &ibdev->gid,
|
||
|
sizeof ( sa->sa_data.mc_member_record.port_gid ) );
|
||
|
|
||
|
/* Issue multicast membership record request */
|
||
|
if ( ( rc = ib_gma_request ( &ibdev->gma, &mad, NULL,
|
||
|
join ) ) != 0 ) {
|
||
|
DBGC ( ibdev, "IBDEV %p could not join group: %s\n",
|
||
|
ibdev, strerror ( rc ) );
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Join multicast group
|
||
|
*
|
||
|
* @v ibdev Infiniband device
|
||
|
* @v qp Queue pair
|
||
|
* @v gid Multicast GID
|
||
|
* @ret rc Return status code
|
||
|
*/
|
||
|
int ib_mcast_join ( struct ib_device *ibdev, struct ib_queue_pair *qp,
|
||
|
struct ib_gid *gid ) {
|
||
|
int rc;
|
||
|
|
||
|
DBGC ( ibdev, "IBDEV %p QPN %lx joining %08x:%08x:%08x:%08x\n",
|
||
|
ibdev, qp->qpn, ntohl ( gid->u.dwords[0] ),
|
||
|
ntohl ( gid->u.dwords[1] ), ntohl ( gid->u.dwords[2] ),
|
||
|
ntohl ( gid->u.dwords[3] ) );
|
||
|
|
||
|
/* Attach queue pair to multicast GID */
|
||
|
if ( ( rc = ib_mcast_attach ( ibdev, qp, gid ) ) != 0 ) {
|
||
|
DBGC ( ibdev, "IBDEV %p could not attach: %s\n",
|
||
|
ibdev, strerror ( rc ) );
|
||
|
goto err_mcast_attach;
|
||
|
}
|
||
|
|
||
|
/* Initiate multicast membership join */
|
||
|
if ( ( rc = ib_mc_member_request ( ibdev, gid, 1 ) ) != 0 )
|
||
|
goto err_mc_member_record;
|
||
|
|
||
|
return 0;
|
||
|
|
||
|
err_mc_member_record:
|
||
|
ib_mcast_detach ( ibdev, qp, gid );
|
||
|
err_mcast_attach:
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Leave multicast group
|
||
|
*
|
||
|
* @v ibdev Infiniband device
|
||
|
* @v qp Queue pair
|
||
|
* @v gid Multicast GID
|
||
|
*/
|
||
|
void ib_mcast_leave ( struct ib_device *ibdev, struct ib_queue_pair *qp,
|
||
|
struct ib_gid *gid ) {
|
||
|
|
||
|
DBGC ( ibdev, "IBDEV %p QPN %lx leaving %08x:%08x:%08x:%08x\n",
|
||
|
ibdev, qp->qpn, ntohl ( gid->u.dwords[0] ),
|
||
|
ntohl ( gid->u.dwords[1] ), ntohl ( gid->u.dwords[2] ),
|
||
|
ntohl ( gid->u.dwords[3] ) );
|
||
|
|
||
|
/* Detach queue pair from multicast GID */
|
||
|
ib_mcast_detach ( ibdev, qp, gid );
|
||
|
|
||
|
/* Initiate multicast membership leave */
|
||
|
ib_mc_member_request ( ibdev, gid, 0 );
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Handle multicast membership record join response
|
||
|
*
|
||
|
* @v ibdev Infiniband device
|
||
|
* @v mad MAD
|
||
|
* @ret rc Return status code
|
||
|
*/
|
||
|
static int ib_handle_mc_member_join ( struct ib_device *ibdev,
|
||
|
union ib_mad *mad ) {
|
||
|
struct ib_mc_member_record *mc_member_record =
|
||
|
&mad->sa.sa_data.mc_member_record;
|
||
|
struct ib_queue_pair *qp;
|
||
|
struct ib_gid *gid;
|
||
|
unsigned long qkey;
|
||
|
int rc;
|
||
|
|
||
|
/* Ignore if not a success */
|
||
|
if ( mad->hdr.status != htons ( IB_MGMT_STATUS_OK ) ) {
|
||
|
DBGC ( ibdev, "IBDEV %p join failed with status %04x\n",
|
||
|
ibdev, ntohs ( mad->hdr.status ) );
|
||
|
return -EINVAL;
|
||
|
}
|
||
|
|
||
|
/* Extract MAD parameters */
|
||
|
gid = &mc_member_record->mgid;
|
||
|
qkey = ntohl ( mc_member_record->qkey );
|
||
|
|
||
|
/* Locate matching queue pair */
|
||
|
qp = ib_find_qp_mgid ( ibdev, gid );
|
||
|
if ( ! qp ) {
|
||
|
DBGC ( ibdev, "IBDEV %p has no QP to join "
|
||
|
"%08x:%08x:%08x:%08x\n", ibdev,
|
||
|
ntohl ( gid->u.dwords[0] ),
|
||
|
ntohl ( gid->u.dwords[1] ),
|
||
|
ntohl ( gid->u.dwords[2] ),
|
||
|
ntohl ( gid->u.dwords[3] ) );
|
||
|
return -ENOENT;
|
||
|
}
|
||
|
DBGC ( ibdev, "IBDEV %p QPN %lx joined %08x:%08x:%08x:%08x qkey "
|
||
|
"%lx\n", ibdev, qp->qpn,
|
||
|
ntohl ( gid->u.dwords[0] ), ntohl ( gid->u.dwords[1] ),
|
||
|
ntohl ( gid->u.dwords[2] ), ntohl ( gid->u.dwords[3] ),
|
||
|
qkey );
|
||
|
|
||
|
/* Set queue key */
|
||
|
if ( ( rc = ib_modify_qp ( ibdev, qp, IB_MODIFY_QKEY, qkey ) ) != 0 ) {
|
||
|
DBGC ( ibdev, "IBDEV %p QPN %lx could not modify qkey: %s\n",
|
||
|
ibdev, qp->qpn, strerror ( rc ) );
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Handle multicast membership record leave response
|
||
|
*
|
||
|
* @v ibdev Infiniband device
|
||
|
* @v mad MAD
|
||
|
* @ret rc Return status code
|
||
|
*/
|
||
|
static int ib_handle_mc_member_leave ( struct ib_device *ibdev,
|
||
|
union ib_mad *mad ) {
|
||
|
struct ib_mc_member_record *mc_member_record =
|
||
|
&mad->sa.sa_data.mc_member_record;
|
||
|
struct ib_gid *gid;
|
||
|
|
||
|
/* Ignore if not a success */
|
||
|
if ( mad->hdr.status != htons ( IB_MGMT_STATUS_OK ) ) {
|
||
|
DBGC ( ibdev, "IBDEV %p leave failed with status %04x\n",
|
||
|
ibdev, ntohs ( mad->hdr.status ) );
|
||
|
return -EINVAL;
|
||
|
}
|
||
|
|
||
|
/* Extract MAD parameters */
|
||
|
gid = &mc_member_record->mgid;
|
||
|
DBGC ( ibdev, "IBDEV %p left %08x:%08x:%08x:%08x\n", ibdev,
|
||
|
ntohl ( gid->u.dwords[0] ), ntohl ( gid->u.dwords[1] ),
|
||
|
ntohl ( gid->u.dwords[2] ), ntohl ( gid->u.dwords[3] ) );
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/** Multicast membership record response handler */
|
||
|
struct ib_mad_handler ib_mc_member_record_handlers[] __ib_mad_handler = {
|
||
|
{
|
||
|
.mgmt_class = IB_MGMT_CLASS_SUBN_ADM,
|
||
|
.class_version = IB_SA_CLASS_VERSION,
|
||
|
.method = IB_MGMT_METHOD_GET_RESP,
|
||
|
.attr_id = htons ( IB_SA_ATTR_MC_MEMBER_REC ),
|
||
|
.handle = ib_handle_mc_member_join,
|
||
|
},
|
||
|
{
|
||
|
.mgmt_class = IB_MGMT_CLASS_SUBN_ADM,
|
||
|
.class_version = IB_SA_CLASS_VERSION,
|
||
|
.method = IB_SA_METHOD_DELETE_RESP,
|
||
|
.attr_id = htons ( IB_SA_ATTR_MC_MEMBER_REC ),
|
||
|
.handle = ib_handle_mc_member_leave,
|
||
|
},
|
||
|
};
|