/*
 * Scsi Host Layer for MPT (Message Passing Technology) based controllers
 *
 * This code is based on drivers/scsi/mpt3sas/mpt3sas_scsih.c
 * Copyright (C) 2012-2014  LSI Corporation
 * Copyright (C) 2013-2014 Avago Technologies
 *  (mailto: MPT-FusionLinux.pdl@avagotech.com)
 *
 * 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 (at your option) 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.
 *
 * NO WARRANTY
 * THE PROGRAM IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OR
 * CONDITIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED INCLUDING, WITHOUT
 * LIMITATION, ANY WARRANTIES OR CONDITIONS OF TITLE, NON-INFRINGEMENT,
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Each Recipient is
 * solely responsible for determining the appropriateness of using and
 * distributing the Program and assumes all risks associated with its
 * exercise of rights under this Agreement, including but not limited to
 * the risks and costs of program errors, damage to or loss of data,
 * programs or equipment, and unavailability or interruption of operations.

 * DISCLAIMER OF LIABILITY
 * NEITHER RECIPIENT NOR ANY CONTRIBUTORS SHALL HAVE ANY LIABILITY FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING WITHOUT LIMITATION LOST PROFITS), HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OR DISTRIBUTION OF THE PROGRAM OR THE EXERCISE OF ANY RIGHTS GRANTED
 * HEREUNDER, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES

 * 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
 * USA.
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/errno.h>
#include <linux/blkdev.h>
#include <linux/sched.h>
#include <linux/workqueue.h>
#include <linux/delay.h>
#include <linux/pci.h>
#include <linux/interrupt.h>
#include <linux/aer.h>
#include <linux/raid_class.h>
#include <linux/blk-mq-pci.h>
#include <asm/unaligned.h>

#include "mpt3sas_base.h"

#define RAID_CHANNEL 1

#define PCIE_CHANNEL 2

/* forward proto's */
static void _scsih_expander_node_remove(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_node *sas_expander);
static void _firmware_event_work(struct work_struct *work);

static void _scsih_remove_device(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_device *sas_device);
static int _scsih_add_device(struct MPT3SAS_ADAPTER *ioc, u16 handle,
        u8 retry_count, u8 is_pd);
static int _scsih_pcie_add_device(struct MPT3SAS_ADAPTER *ioc, u16 handle);
static void _scsih_pcie_device_remove_from_sml(struct MPT3SAS_ADAPTER *ioc,
        struct _pcie_device *pcie_device);
static void
_scsih_pcie_check_device(struct MPT3SAS_ADAPTER *ioc, u16 handle);
static u8 _scsih_check_for_pending_tm(struct MPT3SAS_ADAPTER *ioc, u16 smid);

/* global parameters */
LIST_HEAD(mpt3sas_ioc_list);
/* global ioc lock for list operations */
DEFINE_SPINLOCK(gioc_lock);

MODULE_AUTHOR(MPT3SAS_AUTHOR);
MODULE_DESCRIPTION(MPT3SAS_DESCRIPTION);
MODULE_LICENSE("GPL");
MODULE_VERSION(MPT3SAS_DRIVER_VERSION);
MODULE_ALIAS("mpt2sas");

/* local parameters */
static u8 scsi_io_cb_idx = -1;
static u8 tm_cb_idx = -1;
static u8 ctl_cb_idx = -1;
static u8 base_cb_idx = -1;
static u8 port_enable_cb_idx = -1;
static u8 transport_cb_idx = -1;
static u8 scsih_cb_idx = -1;
static u8 config_cb_idx = -1;
static int mpt2_ids;
static int mpt3_ids;

static u8 tm_tr_cb_idx = -1 ;
static u8 tm_tr_volume_cb_idx = -1 ;
static u8 tm_sas_control_cb_idx = -1;

/* command line options */
static u32 logging_level;
MODULE_PARM_DESC(logging_level,
        " bits for enabling additional logging info (default=0)");


static ushort max_sectors = 0xFFFF;
module_param(max_sectors, ushort, 0444);
MODULE_PARM_DESC(max_sectors, "max sectors, range 64 to 32767  default=32767");


static int missing_delay[2] = {-1, -1};
module_param_array(missing_delay, int, NULL, 0444);
MODULE_PARM_DESC(missing_delay, " device missing delay , io missing delay");

/* scsi-mid layer global parmeter is max_report_luns, which is 511 */
#define MPT3SAS_MAX_LUN (16895)
static u64 max_lun = MPT3SAS_MAX_LUN;
module_param(max_lun, ullong, 0444);
MODULE_PARM_DESC(max_lun, " max lun, default=16895 ");

static ushort hbas_to_enumerate;
module_param(hbas_to_enumerate, ushort, 0444);
MODULE_PARM_DESC(hbas_to_enumerate,
                " 0 - enumerates both SAS 2.0 & SAS 3.0 generation HBAs\n \
                  1 - enumerates only SAS 2.0 generation HBAs\n \
                  2 - enumerates only SAS 3.0 generation HBAs (default=0)");

/* diag_buffer_enable is bitwise
 * bit 0 set = TRACE
 * bit 1 set = SNAPSHOT
 * bit 2 set = EXTENDED
 *
 * Either bit can be set, or both
 */
static int diag_buffer_enable = -1;
module_param(diag_buffer_enable, int, 0444);
MODULE_PARM_DESC(diag_buffer_enable,
        " post diag buffers (TRACE=1/SNAPSHOT=2/EXTENDED=4/default=0)");
static int disable_discovery = -1;
module_param(disable_discovery, int, 0444);
MODULE_PARM_DESC(disable_discovery, " disable discovery ");


/* permit overriding the host protection capabilities mask (EEDP/T10 PI) */
static int prot_mask = -1;
module_param(prot_mask, int, 0444);
MODULE_PARM_DESC(prot_mask, " host protection capabilities mask, def=7 ");

static bool enable_sdev_max_qd;
module_param(enable_sdev_max_qd, bool, 0444);
MODULE_PARM_DESC(enable_sdev_max_qd,
        "Enable sdev max qd as can_queue, def=disabled(0)");

static int multipath_on_hba = -1;
module_param(multipath_on_hba, int, 0);
MODULE_PARM_DESC(multipath_on_hba,
        "Multipath support to add same target device\n\t\t"
        "as many times as it is visible to HBA from various paths\n\t\t"
        "(by default:\n\t\t"
        "\t SAS 2.0 & SAS 3.0 HBA - This will be disabled,\n\t\t"
        "\t SAS 3.5 HBA - This will be enabled)");

static int host_tagset_enable = 1;
module_param(host_tagset_enable, int, 0444);
MODULE_PARM_DESC(host_tagset_enable,
        "Shared host tagset enable/disable Default: enable(1)");

/* raid transport support */
static struct raid_template *mpt3sas_raid_template;
static struct raid_template *mpt2sas_raid_template;


/**
 * struct sense_info - common structure for obtaining sense keys
 * @skey: sense key
 * @asc: additional sense code
 * @ascq: additional sense code qualifier
 */
struct sense_info {
        u8 skey;
        u8 asc;
        u8 ascq;
};

#define MPT3SAS_PROCESS_TRIGGER_DIAG (0xFFFB)
#define MPT3SAS_TURN_ON_PFA_LED (0xFFFC)
#define MPT3SAS_PORT_ENABLE_COMPLETE (0xFFFD)
#define MPT3SAS_ABRT_TASK_SET (0xFFFE)
#define MPT3SAS_REMOVE_UNRESPONDING_DEVICES (0xFFFF)
/**
 * struct fw_event_work - firmware event struct
 * @list: link list framework
 * @work: work object (ioc->fault_reset_work_q)
 * @ioc: per adapter object
 * @device_handle: device handle
 * @VF_ID: virtual function id
 * @VP_ID: virtual port id
 * @ignore: flag meaning this event has been marked to ignore
 * @event: firmware event MPI2_EVENT_XXX defined in mpi2_ioc.h
 * @refcount: kref for this event
 * @event_data: reply event data payload follows
 *
 * This object stored on ioc->fw_event_list.
 */
struct fw_event_work {
        struct list_head        list;
        struct work_struct      work;

        struct MPT3SAS_ADAPTER *ioc;
        u16                     device_handle;
        u8                      VF_ID;
        u8                      VP_ID;
        u8                      ignore;
        u16                     event;
        struct kref             refcount;
        char                    event_data[] __aligned(4);
};

static void fw_event_work_free(struct kref *r)
{
        kfree(container_of(r, struct fw_event_work, refcount));
}

static void fw_event_work_get(struct fw_event_work *fw_work)
{
        kref_get(&fw_work->refcount);
}

static void fw_event_work_put(struct fw_event_work *fw_work)
{
        kref_put(&fw_work->refcount, fw_event_work_free);
}

static struct fw_event_work *alloc_fw_event_work(int len)
{
        struct fw_event_work *fw_event;

        fw_event = kzalloc(sizeof(*fw_event) + len, GFP_ATOMIC);
        if (!fw_event)
                return NULL;

        kref_init(&fw_event->refcount);
        return fw_event;
}

/**
 * struct _scsi_io_transfer - scsi io transfer
 * @handle: sas device handle (assigned by firmware)
 * @is_raid: flag set for hidden raid components
 * @dir: DMA_TO_DEVICE, DMA_FROM_DEVICE,
 * @data_length: data transfer length
 * @data_dma: dma pointer to data
 * @sense: sense data
 * @lun: lun number
 * @cdb_length: cdb length
 * @cdb: cdb contents
 * @timeout: timeout for this command
 * @VF_ID: virtual function id
 * @VP_ID: virtual port id
 * @valid_reply: flag set for reply message
 * @sense_length: sense length
 * @ioc_status: ioc status
 * @scsi_state: scsi state
 * @scsi_status: scsi staus
 * @log_info: log information
 * @transfer_length: data length transfer when there is a reply message
 *
 * Used for sending internal scsi commands to devices within this module.
 * Refer to _scsi_send_scsi_io().
 */
struct _scsi_io_transfer {
        u16     handle;
        u8      is_raid;
        enum dma_data_direction dir;
        u32     data_length;
        dma_addr_t data_dma;
        u8      sense[SCSI_SENSE_BUFFERSIZE];
        u32     lun;
        u8      cdb_length;
        u8      cdb[32];
        u8      timeout;
        u8      VF_ID;
        u8      VP_ID;
        u8      valid_reply;
  /* the following bits are only valid when 'valid_reply = 1' */
        u32     sense_length;
        u16     ioc_status;
        u8      scsi_state;
        u8      scsi_status;
        u32     log_info;
        u32     transfer_length;
};

/**
 * _scsih_set_debug_level - global setting of ioc->logging_level.
 * @val: ?
 * @kp: ?
 *
 * Note: The logging levels are defined in mpt3sas_debug.h.
 */
static int
_scsih_set_debug_level(const char *val, const struct kernel_param *kp)
{
        int ret = param_set_int(val, kp);
        struct MPT3SAS_ADAPTER *ioc;

        if (ret)
                return ret;

        pr_info("setting logging_level(0x%08x)\n", logging_level);
        spin_lock(&gioc_lock);
        list_for_each_entry(ioc, &mpt3sas_ioc_list, list)
                ioc->logging_level = logging_level;
        spin_unlock(&gioc_lock);
        return 0;
}
module_param_call(logging_level, _scsih_set_debug_level, param_get_int,
        &logging_level, 0644);

/**
 * _scsih_srch_boot_sas_address - search based on sas_address
 * @sas_address: sas address
 * @boot_device: boot device object from bios page 2
 *
 * Return: 1 when there's a match, 0 means no match.
 */
static inline int
_scsih_srch_boot_sas_address(u64 sas_address,
        Mpi2BootDeviceSasWwid_t *boot_device)
{
        return (sas_address == le64_to_cpu(boot_device->SASAddress)) ?  1 : 0;
}

/**
 * _scsih_srch_boot_device_name - search based on device name
 * @device_name: device name specified in INDENTIFY fram
 * @boot_device: boot device object from bios page 2
 *
 * Return: 1 when there's a match, 0 means no match.
 */
static inline int
_scsih_srch_boot_device_name(u64 device_name,
        Mpi2BootDeviceDeviceName_t *boot_device)
{
        return (device_name == le64_to_cpu(boot_device->DeviceName)) ? 1 : 0;
}

/**
 * _scsih_srch_boot_encl_slot - search based on enclosure_logical_id/slot
 * @enclosure_logical_id: enclosure logical id
 * @slot_number: slot number
 * @boot_device: boot device object from bios page 2
 *
 * Return: 1 when there's a match, 0 means no match.
 */
static inline int
_scsih_srch_boot_encl_slot(u64 enclosure_logical_id, u16 slot_number,
        Mpi2BootDeviceEnclosureSlot_t *boot_device)
{
        return (enclosure_logical_id == le64_to_cpu(boot_device->
            EnclosureLogicalID) && slot_number == le16_to_cpu(boot_device->
            SlotNumber)) ? 1 : 0;
}

/**
 * mpt3sas_get_port_by_id - get hba port entry corresponding to provided
 *                        port number from port list
 * @ioc: per adapter object
 * @port_id: port number
 * @bypass_dirty_port_flag: when set look the matching hba port entry even
 *                      if hba port entry is marked as dirty.
 *
 * Search for hba port entry corresponding to provided port number,
 * if available return port object otherwise return NULL.
 */
struct hba_port *
mpt3sas_get_port_by_id(struct MPT3SAS_ADAPTER *ioc,
        u8 port_id, u8 bypass_dirty_port_flag)
{
        struct hba_port *port, *port_next;

        /*
         * When multipath_on_hba is disabled then
         * search the hba_port entry using default
         * port id i.e. 255
         */
        if (!ioc->multipath_on_hba)
                port_id = MULTIPATH_DISABLED_PORT_ID;

        list_for_each_entry_safe(port, port_next,
            &ioc->port_table_list, list) {
                if (port->port_id != port_id)
                        continue;
                if (bypass_dirty_port_flag)
                        return port;
                if (port->flags & HBA_PORT_FLAG_DIRTY_PORT)
                        continue;
                return port;
        }

        /*
         * Allocate hba_port object for default port id (i.e. 255)
         * when multipath_on_hba is disabled for the HBA.
         * And add this object to port_table_list.
         */
        if (!ioc->multipath_on_hba) {
                port = kzalloc(sizeof(struct hba_port), GFP_ATOMIC);
                if (!port)
                        return NULL;

                port->port_id = port_id;
                ioc_info(ioc,
                   "hba_port entry: %p, port: %d is added to hba_port list\n",
                   port, port->port_id);
                list_add_tail(&port->list,
                    &ioc->port_table_list);
                return port;
        }
        return NULL;
}

/**
 * mpt3sas_get_vphy_by_phy - get virtual_phy object corresponding to phy number
 * @ioc: per adapter object
 * @port: hba_port object
 * @phy: phy number
 *
 * Return virtual_phy object corresponding to phy number.
 */
struct virtual_phy *
mpt3sas_get_vphy_by_phy(struct MPT3SAS_ADAPTER *ioc,
        struct hba_port *port, u32 phy)
{
        struct virtual_phy *vphy, *vphy_next;

        if (!port->vphys_mask)
                return NULL;

        list_for_each_entry_safe(vphy, vphy_next, &port->vphys_list, list) {
                if (vphy->phy_mask & (1 << phy))
                        return vphy;
        }
        return NULL;
}

/**
 * _scsih_is_boot_device - search for matching boot device.
 * @sas_address: sas address
 * @device_name: device name specified in INDENTIFY fram
 * @enclosure_logical_id: enclosure logical id
 * @slot: slot number
 * @form: specifies boot device form
 * @boot_device: boot device object from bios page 2
 *
 * Return: 1 when there's a match, 0 means no match.
 */
static int
_scsih_is_boot_device(u64 sas_address, u64 device_name,
        u64 enclosure_logical_id, u16 slot, u8 form,
        Mpi2BiosPage2BootDevice_t *boot_device)
{
        int rc = 0;

        switch (form) {
        case MPI2_BIOSPAGE2_FORM_SAS_WWID:
                if (!sas_address)
                        break;
                rc = _scsih_srch_boot_sas_address(
                    sas_address, &boot_device->SasWwid);
                break;
        case MPI2_BIOSPAGE2_FORM_ENCLOSURE_SLOT:
                if (!enclosure_logical_id)
                        break;
                rc = _scsih_srch_boot_encl_slot(
                    enclosure_logical_id,
                    slot, &boot_device->EnclosureSlot);
                break;
        case MPI2_BIOSPAGE2_FORM_DEVICE_NAME:
                if (!device_name)
                        break;
                rc = _scsih_srch_boot_device_name(
                    device_name, &boot_device->DeviceName);
                break;
        case MPI2_BIOSPAGE2_FORM_NO_DEVICE_SPECIFIED:
                break;
        }

        return rc;
}

/**
 * _scsih_get_sas_address - set the sas_address for given device handle
 * @ioc: ?
 * @handle: device handle
 * @sas_address: sas address
 *
 * Return: 0 success, non-zero when failure
 */
static int
_scsih_get_sas_address(struct MPT3SAS_ADAPTER *ioc, u16 handle,
        u64 *sas_address)
{
        Mpi2SasDevicePage0_t sas_device_pg0;
        Mpi2ConfigReply_t mpi_reply;
        u32 ioc_status;

        *sas_address = 0;

        if ((mpt3sas_config_get_sas_device_pg0(ioc, &mpi_reply, &sas_device_pg0,
            MPI2_SAS_DEVICE_PGAD_FORM_HANDLE, handle))) {
                ioc_err(ioc, "failure at %s:%d/%s()!\n",
                        __FILE__, __LINE__, __func__);
                return -ENXIO;
        }

        ioc_status = le16_to_cpu(mpi_reply.IOCStatus) & MPI2_IOCSTATUS_MASK;
        if (ioc_status == MPI2_IOCSTATUS_SUCCESS) {
                /* For HBA, vSES doesn't return HBA SAS address. Instead return
                 * vSES's sas address.
                 */
                if ((handle <= ioc->sas_hba.num_phys) &&
                   (!(le32_to_cpu(sas_device_pg0.DeviceInfo) &
                   MPI2_SAS_DEVICE_INFO_SEP)))
                        *sas_address = ioc->sas_hba.sas_address;
                else
                        *sas_address = le64_to_cpu(sas_device_pg0.SASAddress);
                return 0;
        }

        /* we hit this because the given parent handle doesn't exist */
        if (ioc_status == MPI2_IOCSTATUS_CONFIG_INVALID_PAGE)
                return -ENXIO;

        /* else error case */
        ioc_err(ioc, "handle(0x%04x), ioc_status(0x%04x), failure at %s:%d/%s()!\n",
                handle, ioc_status, __FILE__, __LINE__, __func__);
        return -EIO;
}

/**
 * _scsih_determine_boot_device - determine boot device.
 * @ioc: per adapter object
 * @device: sas_device or pcie_device object
 * @channel: SAS or PCIe channel
 *
 * Determines whether this device should be first reported device to
 * to scsi-ml or sas transport, this purpose is for persistent boot device.
 * There are primary, alternate, and current entries in bios page 2. The order
 * priority is primary, alternate, then current.  This routine saves
 * the corresponding device object.
 * The saved data to be used later in _scsih_probe_boot_devices().
 */
static void
_scsih_determine_boot_device(struct MPT3SAS_ADAPTER *ioc, void *device,
        u32 channel)
{
        struct _sas_device *sas_device;
        struct _pcie_device *pcie_device;
        struct _raid_device *raid_device;
        u64 sas_address;
        u64 device_name;
        u64 enclosure_logical_id;
        u16 slot;

         /* only process this function when driver loads */
        if (!ioc->is_driver_loading)
                return;

         /* no Bios, return immediately */
        if (!ioc->bios_pg3.BiosVersion)
                return;

        if (channel == RAID_CHANNEL) {
                raid_device = device;
                sas_address = raid_device->wwid;
                device_name = 0;
                enclosure_logical_id = 0;
                slot = 0;
        } else if (channel == PCIE_CHANNEL) {
                pcie_device = device;
                sas_address = pcie_device->wwid;
                device_name = 0;
                enclosure_logical_id = 0;
                slot = 0;
        } else {
                sas_device = device;
                sas_address = sas_device->sas_address;
                device_name = sas_device->device_name;
                enclosure_logical_id = sas_device->enclosure_logical_id;
                slot = sas_device->slot;
        }

        if (!ioc->req_boot_device.device) {
                if (_scsih_is_boot_device(sas_address, device_name,
                    enclosure_logical_id, slot,
                    (ioc->bios_pg2.ReqBootDeviceForm &
                    MPI2_BIOSPAGE2_FORM_MASK),
                    &ioc->bios_pg2.RequestedBootDevice)) {
                        dinitprintk(ioc,
                                    ioc_info(ioc, "%s: req_boot_device(0x%016llx)\n",
                                             __func__, (u64)sas_address));
                        ioc->req_boot_device.device = device;
                        ioc->req_boot_device.channel = channel;
                }
        }

        if (!ioc->req_alt_boot_device.device) {
                if (_scsih_is_boot_device(sas_address, device_name,
                    enclosure_logical_id, slot,
                    (ioc->bios_pg2.ReqAltBootDeviceForm &
                    MPI2_BIOSPAGE2_FORM_MASK),
                    &ioc->bios_pg2.RequestedAltBootDevice)) {
                        dinitprintk(ioc,
                                    ioc_info(ioc, "%s: req_alt_boot_device(0x%016llx)\n",
                                             __func__, (u64)sas_address));
                        ioc->req_alt_boot_device.device = device;
                        ioc->req_alt_boot_device.channel = channel;
                }
        }

        if (!ioc->current_boot_device.device) {
                if (_scsih_is_boot_device(sas_address, device_name,
                    enclosure_logical_id, slot,
                    (ioc->bios_pg2.CurrentBootDeviceForm &
                    MPI2_BIOSPAGE2_FORM_MASK),
                    &ioc->bios_pg2.CurrentBootDevice)) {
                        dinitprintk(ioc,
                                    ioc_info(ioc, "%s: current_boot_device(0x%016llx)\n",
                                             __func__, (u64)sas_address));
                        ioc->current_boot_device.device = device;
                        ioc->current_boot_device.channel = channel;
                }
        }
}

static struct _sas_device *
__mpt3sas_get_sdev_from_target(struct MPT3SAS_ADAPTER *ioc,
                struct MPT3SAS_TARGET *tgt_priv)
{
        struct _sas_device *ret;

        assert_spin_locked(&ioc->sas_device_lock);

        ret = tgt_priv->sas_dev;
        if (ret)
                sas_device_get(ret);

        return ret;
}

static struct _sas_device *
mpt3sas_get_sdev_from_target(struct MPT3SAS_ADAPTER *ioc,
                struct MPT3SAS_TARGET *tgt_priv)
{
        struct _sas_device *ret;
        unsigned long flags;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        ret = __mpt3sas_get_sdev_from_target(ioc, tgt_priv);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        return ret;
}

static struct _pcie_device *
__mpt3sas_get_pdev_from_target(struct MPT3SAS_ADAPTER *ioc,
        struct MPT3SAS_TARGET *tgt_priv)
{
        struct _pcie_device *ret;

        assert_spin_locked(&ioc->pcie_device_lock);

        ret = tgt_priv->pcie_dev;
        if (ret)
                pcie_device_get(ret);

        return ret;
}

/**
 * mpt3sas_get_pdev_from_target - pcie device search
 * @ioc: per adapter object
 * @tgt_priv: starget private object
 *
 * Context: This function will acquire ioc->pcie_device_lock and will release
 * before returning the pcie_device object.
 *
 * This searches for pcie_device from target, then return pcie_device object.
 */
static struct _pcie_device *
mpt3sas_get_pdev_from_target(struct MPT3SAS_ADAPTER *ioc,
        struct MPT3SAS_TARGET *tgt_priv)
{
        struct _pcie_device *ret;
        unsigned long flags;

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        ret = __mpt3sas_get_pdev_from_target(ioc, tgt_priv);
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        return ret;
}


/**
 * __mpt3sas_get_sdev_by_rphy - sas device search
 * @ioc: per adapter object
 * @rphy: sas_rphy pointer
 *
 * Context: This function will acquire ioc->sas_device_lock and will release
 * before returning the sas_device object.
 *
 * This searches for sas_device from rphy object
 * then return sas_device object.
 */
struct _sas_device *
__mpt3sas_get_sdev_by_rphy(struct MPT3SAS_ADAPTER *ioc,
        struct sas_rphy *rphy)
{
        struct _sas_device *sas_device;

        assert_spin_locked(&ioc->sas_device_lock);

        list_for_each_entry(sas_device, &ioc->sas_device_list, list) {
                if (sas_device->rphy != rphy)
                        continue;
                sas_device_get(sas_device);
                return sas_device;
        }

        sas_device = NULL;
        list_for_each_entry(sas_device, &ioc->sas_device_init_list, list) {
                if (sas_device->rphy != rphy)
                        continue;
                sas_device_get(sas_device);
                return sas_device;
        }

        return NULL;
}

/**
 * mpt3sas_get_sdev_by_addr - get _sas_device object corresponding to provided
 *                              sas address from sas_device_list list
 * @ioc: per adapter object
 * @port: port number
 *
 * Search for _sas_device object corresponding to provided sas address,
 * if available return _sas_device object address otherwise return NULL.
 */
struct _sas_device *
__mpt3sas_get_sdev_by_addr(struct MPT3SAS_ADAPTER *ioc,
        u64 sas_address, struct hba_port *port)
{
        struct _sas_device *sas_device;

        if (!port)
                return NULL;

        assert_spin_locked(&ioc->sas_device_lock);

        list_for_each_entry(sas_device, &ioc->sas_device_list, list) {
                if (sas_device->sas_address != sas_address)
                        continue;
                if (sas_device->port != port)
                        continue;
                sas_device_get(sas_device);
                return sas_device;
        }

        list_for_each_entry(sas_device, &ioc->sas_device_init_list, list) {
                if (sas_device->sas_address != sas_address)
                        continue;
                if (sas_device->port != port)
                        continue;
                sas_device_get(sas_device);
                return sas_device;
        }

        return NULL;
}

/**
 * mpt3sas_get_sdev_by_addr - sas device search
 * @ioc: per adapter object
 * @sas_address: sas address
 * @port: hba port entry
 * Context: Calling function should acquire ioc->sas_device_lock
 *
 * This searches for sas_device based on sas_address & port number,
 * then return sas_device object.
 */
struct _sas_device *
mpt3sas_get_sdev_by_addr(struct MPT3SAS_ADAPTER *ioc,
        u64 sas_address, struct hba_port *port)
{
        struct _sas_device *sas_device;
        unsigned long flags;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_by_addr(ioc,
            sas_address, port);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        return sas_device;
}

static struct _sas_device *
__mpt3sas_get_sdev_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _sas_device *sas_device;

        assert_spin_locked(&ioc->sas_device_lock);

        list_for_each_entry(sas_device, &ioc->sas_device_list, list)
                if (sas_device->handle == handle)
                        goto found_device;

        list_for_each_entry(sas_device, &ioc->sas_device_init_list, list)
                if (sas_device->handle == handle)
                        goto found_device;

        return NULL;

found_device:
        sas_device_get(sas_device);
        return sas_device;
}

/**
 * mpt3sas_get_sdev_by_handle - sas device search
 * @ioc: per adapter object
 * @handle: sas device handle (assigned by firmware)
 * Context: Calling function should acquire ioc->sas_device_lock
 *
 * This searches for sas_device based on sas_address, then return sas_device
 * object.
 */
struct _sas_device *
mpt3sas_get_sdev_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _sas_device *sas_device;
        unsigned long flags;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_by_handle(ioc, handle);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        return sas_device;
}

/**
 * _scsih_display_enclosure_chassis_info - display device location info
 * @ioc: per adapter object
 * @sas_device: per sas device object
 * @sdev: scsi device struct
 * @starget: scsi target struct
 */
static void
_scsih_display_enclosure_chassis_info(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_device *sas_device, struct scsi_device *sdev,
        struct scsi_target *starget)
{
        if (sdev) {
                if (sas_device->enclosure_handle != 0)
                        sdev_printk(KERN_INFO, sdev,
                            "enclosure logical id (0x%016llx), slot(%d) \n",
                            (unsigned long long)
                            sas_device->enclosure_logical_id,
                            sas_device->slot);
                if (sas_device->connector_name[0] != '\0')
                        sdev_printk(KERN_INFO, sdev,
                            "enclosure level(0x%04x), connector name( %s)\n",
                            sas_device->enclosure_level,
                            sas_device->connector_name);
                if (sas_device->is_chassis_slot_valid)
                        sdev_printk(KERN_INFO, sdev, "chassis slot(0x%04x)\n",
                            sas_device->chassis_slot);
        } else if (starget) {
                if (sas_device->enclosure_handle != 0)
                        starget_printk(KERN_INFO, starget,
                            "enclosure logical id(0x%016llx), slot(%d) \n",
                            (unsigned long long)
                            sas_device->enclosure_logical_id,
                            sas_device->slot);
                if (sas_device->connector_name[0] != '\0')
                        starget_printk(KERN_INFO, starget,
                            "enclosure level(0x%04x), connector name( %s)\n",
                            sas_device->enclosure_level,
                            sas_device->connector_name);
                if (sas_device->is_chassis_slot_valid)
                        starget_printk(KERN_INFO, starget,
                            "chassis slot(0x%04x)\n",
                            sas_device->chassis_slot);
        } else {
                if (sas_device->enclosure_handle != 0)
                        ioc_info(ioc, "enclosure logical id(0x%016llx), slot(%d)\n",
                                 (u64)sas_device->enclosure_logical_id,
                                 sas_device->slot);
                if (sas_device->connector_name[0] != '\0')
                        ioc_info(ioc, "enclosure level(0x%04x), connector name( %s)\n",
                                 sas_device->enclosure_level,
                                 sas_device->connector_name);
                if (sas_device->is_chassis_slot_valid)
                        ioc_info(ioc, "chassis slot(0x%04x)\n",
                                 sas_device->chassis_slot);
        }
}

/**
 * _scsih_sas_device_remove - remove sas_device from list.
 * @ioc: per adapter object
 * @sas_device: the sas_device object
 * Context: This function will acquire ioc->sas_device_lock.
 *
 * If sas_device is on the list, remove it and decrement its reference count.
 */
static void
_scsih_sas_device_remove(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_device *sas_device)
{
        unsigned long flags;

        if (!sas_device)
                return;
        ioc_info(ioc, "removing handle(0x%04x), sas_addr(0x%016llx)\n",
                 sas_device->handle, (u64)sas_device->sas_address);

        _scsih_display_enclosure_chassis_info(ioc, sas_device, NULL, NULL);

        /*
         * The lock serializes access to the list, but we still need to verify
         * that nobody removed the entry while we were waiting on the lock.
         */
        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        if (!list_empty(&sas_device->list)) {
                list_del_init(&sas_device->list);
                sas_device_put(sas_device);
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
}

/**
 * _scsih_device_remove_by_handle - removing device object by handle
 * @ioc: per adapter object
 * @handle: device handle
 */
static void
_scsih_device_remove_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _sas_device *sas_device;
        unsigned long flags;

        if (ioc->shost_recovery)
                return;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_by_handle(ioc, handle);
        if (sas_device) {
                list_del_init(&sas_device->list);
                sas_device_put(sas_device);
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
        if (sas_device) {
                _scsih_remove_device(ioc, sas_device);
                sas_device_put(sas_device);
        }
}

/**
 * mpt3sas_device_remove_by_sas_address - removing device object by
 *                                      sas address & port number
 * @ioc: per adapter object
 * @sas_address: device sas_address
 * @port: hba port entry
 *
 * Return nothing.
 */
void
mpt3sas_device_remove_by_sas_address(struct MPT3SAS_ADAPTER *ioc,
        u64 sas_address, struct hba_port *port)
{
        struct _sas_device *sas_device;
        unsigned long flags;

        if (ioc->shost_recovery)
                return;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_by_addr(ioc, sas_address, port);
        if (sas_device) {

                list_del_init(&sas_device->list);
                sas_device_put(sas_device);
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
        if (sas_device) {
                _scsih_remove_device(ioc, sas_device);
                sas_device_put(sas_device);
        }
}

/**
 * _scsih_sas_device_add - insert sas_device to the list.
 * @ioc: per adapter object
 * @sas_device: the sas_device object
 * Context: This function will acquire ioc->sas_device_lock.
 *
 * Adding new object to the ioc->sas_device_list.
 */
static void
_scsih_sas_device_add(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_device *sas_device)
{
        unsigned long flags;

        dewtprintk(ioc,
                   ioc_info(ioc, "%s: handle(0x%04x), sas_addr(0x%016llx)\n",
                            __func__, sas_device->handle,
                            (u64)sas_device->sas_address));

        dewtprintk(ioc, _scsih_display_enclosure_chassis_info(ioc, sas_device,
            NULL, NULL));

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device_get(sas_device);
        list_add_tail(&sas_device->list, &ioc->sas_device_list);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        if (ioc->hide_drives) {
                clear_bit(sas_device->handle, ioc->pend_os_device_add);
                return;
        }

        if (!mpt3sas_transport_port_add(ioc, sas_device->handle,
             sas_device->sas_address_parent, sas_device->port)) {
                _scsih_sas_device_remove(ioc, sas_device);
        } else if (!sas_device->starget) {
                /*
                 * When asyn scanning is enabled, its not possible to remove
                 * devices while scanning is turned on due to an oops in
                 * scsi_sysfs_add_sdev()->add_device()->sysfs_addrm_start()
                 */
                if (!ioc->is_driver_loading) {
                        mpt3sas_transport_port_remove(ioc,
                            sas_device->sas_address,
                            sas_device->sas_address_parent,
                            sas_device->port);
                        _scsih_sas_device_remove(ioc, sas_device);
                }
        } else
                clear_bit(sas_device->handle, ioc->pend_os_device_add);
}

/**
 * _scsih_sas_device_init_add - insert sas_device to the list.
 * @ioc: per adapter object
 * @sas_device: the sas_device object
 * Context: This function will acquire ioc->sas_device_lock.
 *
 * Adding new object at driver load time to the ioc->sas_device_init_list.
 */
static void
_scsih_sas_device_init_add(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_device *sas_device)
{
        unsigned long flags;

        dewtprintk(ioc,
                   ioc_info(ioc, "%s: handle(0x%04x), sas_addr(0x%016llx)\n",
                            __func__, sas_device->handle,
                            (u64)sas_device->sas_address));

        dewtprintk(ioc, _scsih_display_enclosure_chassis_info(ioc, sas_device,
            NULL, NULL));

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device_get(sas_device);
        list_add_tail(&sas_device->list, &ioc->sas_device_init_list);
        _scsih_determine_boot_device(ioc, sas_device, 0);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
}


static struct _pcie_device *
__mpt3sas_get_pdev_by_wwid(struct MPT3SAS_ADAPTER *ioc, u64 wwid)
{
        struct _pcie_device *pcie_device;

        assert_spin_locked(&ioc->pcie_device_lock);

        list_for_each_entry(pcie_device, &ioc->pcie_device_list, list)
                if (pcie_device->wwid == wwid)
                        goto found_device;

        list_for_each_entry(pcie_device, &ioc->pcie_device_init_list, list)
                if (pcie_device->wwid == wwid)
                        goto found_device;

        return NULL;

found_device:
        pcie_device_get(pcie_device);
        return pcie_device;
}


/**
 * mpt3sas_get_pdev_by_wwid - pcie device search
 * @ioc: per adapter object
 * @wwid: wwid
 *
 * Context: This function will acquire ioc->pcie_device_lock and will release
 * before returning the pcie_device object.
 *
 * This searches for pcie_device based on wwid, then return pcie_device object.
 */
static struct _pcie_device *
mpt3sas_get_pdev_by_wwid(struct MPT3SAS_ADAPTER *ioc, u64 wwid)
{
        struct _pcie_device *pcie_device;
        unsigned long flags;

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        pcie_device = __mpt3sas_get_pdev_by_wwid(ioc, wwid);
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        return pcie_device;
}


static struct _pcie_device *
__mpt3sas_get_pdev_by_idchannel(struct MPT3SAS_ADAPTER *ioc, int id,
        int channel)
{
        struct _pcie_device *pcie_device;

        assert_spin_locked(&ioc->pcie_device_lock);

        list_for_each_entry(pcie_device, &ioc->pcie_device_list, list)
                if (pcie_device->id == id && pcie_device->channel == channel)
                        goto found_device;

        list_for_each_entry(pcie_device, &ioc->pcie_device_init_list, list)
                if (pcie_device->id == id && pcie_device->channel == channel)
                        goto found_device;

        return NULL;

found_device:
        pcie_device_get(pcie_device);
        return pcie_device;
}

static struct _pcie_device *
__mpt3sas_get_pdev_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _pcie_device *pcie_device;

        assert_spin_locked(&ioc->pcie_device_lock);

        list_for_each_entry(pcie_device, &ioc->pcie_device_list, list)
                if (pcie_device->handle == handle)
                        goto found_device;

        list_for_each_entry(pcie_device, &ioc->pcie_device_init_list, list)
                if (pcie_device->handle == handle)
                        goto found_device;

        return NULL;

found_device:
        pcie_device_get(pcie_device);
        return pcie_device;
}


/**
 * mpt3sas_get_pdev_by_handle - pcie device search
 * @ioc: per adapter object
 * @handle: Firmware device handle
 *
 * Context: This function will acquire ioc->pcie_device_lock and will release
 * before returning the pcie_device object.
 *
 * This searches for pcie_device based on handle, then return pcie_device
 * object.
 */
struct _pcie_device *
mpt3sas_get_pdev_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _pcie_device *pcie_device;
        unsigned long flags;

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        pcie_device = __mpt3sas_get_pdev_by_handle(ioc, handle);
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        return pcie_device;
}

/**
 * _scsih_set_nvme_max_shutdown_latency - Update max_shutdown_latency.
 * @ioc: per adapter object
 * Context: This function will acquire ioc->pcie_device_lock
 *
 * Update ioc->max_shutdown_latency to that NVMe drives RTD3 Entry Latency
 * which has reported maximum among all available NVMe drives.
 * Minimum max_shutdown_latency will be six seconds.
 */
static void
_scsih_set_nvme_max_shutdown_latency(struct MPT3SAS_ADAPTER *ioc)
{
        struct _pcie_device *pcie_device;
        unsigned long flags;
        u16 shutdown_latency = IO_UNIT_CONTROL_SHUTDOWN_TIMEOUT;

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        list_for_each_entry(pcie_device, &ioc->pcie_device_list, list) {
                if (pcie_device->shutdown_latency) {
                        if (shutdown_latency < pcie_device->shutdown_latency)
                                shutdown_latency =
                                        pcie_device->shutdown_latency;
                }
        }
        ioc->max_shutdown_latency = shutdown_latency;
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
}

/**
 * _scsih_pcie_device_remove - remove pcie_device from list.
 * @ioc: per adapter object
 * @pcie_device: the pcie_device object
 * Context: This function will acquire ioc->pcie_device_lock.
 *
 * If pcie_device is on the list, remove it and decrement its reference count.
 */
static void
_scsih_pcie_device_remove(struct MPT3SAS_ADAPTER *ioc,
        struct _pcie_device *pcie_device)
{
        unsigned long flags;
        int was_on_pcie_device_list = 0;
        u8 update_latency = 0;

        if (!pcie_device)
                return;
        ioc_info(ioc, "removing handle(0x%04x), wwid(0x%016llx)\n",
                 pcie_device->handle, (u64)pcie_device->wwid);
        if (pcie_device->enclosure_handle != 0)
                ioc_info(ioc, "removing enclosure logical id(0x%016llx), slot(%d)\n",
                         (u64)pcie_device->enclosure_logical_id,
                         pcie_device->slot);
        if (pcie_device->connector_name[0] != '\0')
                ioc_info(ioc, "removing enclosure level(0x%04x), connector name( %s)\n",
                         pcie_device->enclosure_level,
                         pcie_device->connector_name);

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        if (!list_empty(&pcie_device->list)) {
                list_del_init(&pcie_device->list);
                was_on_pcie_device_list = 1;
        }
        if (pcie_device->shutdown_latency == ioc->max_shutdown_latency)
                update_latency = 1;
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
        if (was_on_pcie_device_list) {
                kfree(pcie_device->serial_number);
                pcie_device_put(pcie_device);
        }

        /*
         * This device's RTD3 Entry Latency matches IOC's
         * max_shutdown_latency. Recalculate IOC's max_shutdown_latency
         * from the available drives as current drive is getting removed.
         */
        if (update_latency)
                _scsih_set_nvme_max_shutdown_latency(ioc);
}


/**
 * _scsih_pcie_device_remove_by_handle - removing pcie device object by handle
 * @ioc: per adapter object
 * @handle: device handle
 */
static void
_scsih_pcie_device_remove_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _pcie_device *pcie_device;
        unsigned long flags;
        int was_on_pcie_device_list = 0;
        u8 update_latency = 0;

        if (ioc->shost_recovery)
                return;

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        pcie_device = __mpt3sas_get_pdev_by_handle(ioc, handle);
        if (pcie_device) {
                if (!list_empty(&pcie_device->list)) {
                        list_del_init(&pcie_device->list);
                        was_on_pcie_device_list = 1;
                        pcie_device_put(pcie_device);
                }
                if (pcie_device->shutdown_latency == ioc->max_shutdown_latency)
                        update_latency = 1;
        }
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
        if (was_on_pcie_device_list) {
                _scsih_pcie_device_remove_from_sml(ioc, pcie_device);
                pcie_device_put(pcie_device);
        }

        /*
         * This device's RTD3 Entry Latency matches IOC's
         * max_shutdown_latency. Recalculate IOC's max_shutdown_latency
         * from the available drives as current drive is getting removed.
         */
        if (update_latency)
                _scsih_set_nvme_max_shutdown_latency(ioc);
}

/**
 * _scsih_pcie_device_add - add pcie_device object
 * @ioc: per adapter object
 * @pcie_device: pcie_device object
 *
 * This is added to the pcie_device_list link list.
 */
static void
_scsih_pcie_device_add(struct MPT3SAS_ADAPTER *ioc,
        struct _pcie_device *pcie_device)
{
        unsigned long flags;

        dewtprintk(ioc,
                   ioc_info(ioc, "%s: handle (0x%04x), wwid(0x%016llx)\n",
                            __func__,
                            pcie_device->handle, (u64)pcie_device->wwid));
        if (pcie_device->enclosure_handle != 0)
                dewtprintk(ioc,
                           ioc_info(ioc, "%s: enclosure logical id(0x%016llx), slot( %d)\n",
                                    __func__,
                                    (u64)pcie_device->enclosure_logical_id,
                                    pcie_device->slot));
        if (pcie_device->connector_name[0] != '\0')
                dewtprintk(ioc,
                           ioc_info(ioc, "%s: enclosure level(0x%04x), connector name( %s)\n",
                                    __func__, pcie_device->enclosure_level,
                                    pcie_device->connector_name));

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        pcie_device_get(pcie_device);
        list_add_tail(&pcie_device->list, &ioc->pcie_device_list);
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        if (pcie_device->access_status ==
            MPI26_PCIEDEV0_ASTATUS_DEVICE_BLOCKED) {
                clear_bit(pcie_device->handle, ioc->pend_os_device_add);
                return;
        }
        if (scsi_add_device(ioc->shost, PCIE_CHANNEL, pcie_device->id, 0)) {
                _scsih_pcie_device_remove(ioc, pcie_device);
        } else if (!pcie_device->starget) {
                if (!ioc->is_driver_loading) {
/*TODO-- Need to find out whether this condition will occur or not*/
                        clear_bit(pcie_device->handle, ioc->pend_os_device_add);
                }
        } else
                clear_bit(pcie_device->handle, ioc->pend_os_device_add);
}

/*
 * _scsih_pcie_device_init_add - insert pcie_device to the init list.
 * @ioc: per adapter object
 * @pcie_device: the pcie_device object
 * Context: This function will acquire ioc->pcie_device_lock.
 *
 * Adding new object at driver load time to the ioc->pcie_device_init_list.
 */
static void
_scsih_pcie_device_init_add(struct MPT3SAS_ADAPTER *ioc,
                                struct _pcie_device *pcie_device)
{
        unsigned long flags;

        dewtprintk(ioc,
                   ioc_info(ioc, "%s: handle (0x%04x), wwid(0x%016llx)\n",
                            __func__,
                            pcie_device->handle, (u64)pcie_device->wwid));
        if (pcie_device->enclosure_handle != 0)
                dewtprintk(ioc,
                           ioc_info(ioc, "%s: enclosure logical id(0x%016llx), slot( %d)\n",
                                    __func__,
                                    (u64)pcie_device->enclosure_logical_id,
                                    pcie_device->slot));
        if (pcie_device->connector_name[0] != '\0')
                dewtprintk(ioc,
                           ioc_info(ioc, "%s: enclosure level(0x%04x), connector name( %s)\n",
                                    __func__, pcie_device->enclosure_level,
                                    pcie_device->connector_name));

        spin_lock_irqsave(&ioc->pcie_device_lock, flags);
        pcie_device_get(pcie_device);
        list_add_tail(&pcie_device->list, &ioc->pcie_device_init_list);
        if (pcie_device->access_status !=
            MPI26_PCIEDEV0_ASTATUS_DEVICE_BLOCKED)
                _scsih_determine_boot_device(ioc, pcie_device, PCIE_CHANNEL);
        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
}
/**
 * _scsih_raid_device_find_by_id - raid device search
 * @ioc: per adapter object
 * @id: sas device target id
 * @channel: sas device channel
 * Context: Calling function should acquire ioc->raid_device_lock
 *
 * This searches for raid_device based on target id, then return raid_device
 * object.
 */
static struct _raid_device *
_scsih_raid_device_find_by_id(struct MPT3SAS_ADAPTER *ioc, int id, int channel)
{
        struct _raid_device *raid_device, *r;

        r = NULL;
        list_for_each_entry(raid_device, &ioc->raid_device_list, list) {
                if (raid_device->id == id && raid_device->channel == channel) {
                        r = raid_device;
                        goto out;
                }
        }

 out:
        return r;
}

/**
 * mpt3sas_raid_device_find_by_handle - raid device search
 * @ioc: per adapter object
 * @handle: sas device handle (assigned by firmware)
 * Context: Calling function should acquire ioc->raid_device_lock
 *
 * This searches for raid_device based on handle, then return raid_device
 * object.
 */
struct _raid_device *
mpt3sas_raid_device_find_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _raid_device *raid_device, *r;

        r = NULL;
        list_for_each_entry(raid_device, &ioc->raid_device_list, list) {
                if (raid_device->handle != handle)
                        continue;
                r = raid_device;
                goto out;
        }

 out:
        return r;
}

/**
 * _scsih_raid_device_find_by_wwid - raid device search
 * @ioc: per adapter object
 * @wwid: ?
 * Context: Calling function should acquire ioc->raid_device_lock
 *
 * This searches for raid_device based on wwid, then return raid_device
 * object.
 */
static struct _raid_device *
_scsih_raid_device_find_by_wwid(struct MPT3SAS_ADAPTER *ioc, u64 wwid)
{
        struct _raid_device *raid_device, *r;

        r = NULL;
        list_for_each_entry(raid_device, &ioc->raid_device_list, list) {
                if (raid_device->wwid != wwid)
                        continue;
                r = raid_device;
                goto out;
        }

 out:
        return r;
}

/**
 * _scsih_raid_device_add - add raid_device object
 * @ioc: per adapter object
 * @raid_device: raid_device object
 *
 * This is added to the raid_device_list link list.
 */
static void
_scsih_raid_device_add(struct MPT3SAS_ADAPTER *ioc,
        struct _raid_device *raid_device)
{
        unsigned long flags;

        dewtprintk(ioc,
                   ioc_info(ioc, "%s: handle(0x%04x), wwid(0x%016llx)\n",
                            __func__,
                            raid_device->handle, (u64)raid_device->wwid));

        spin_lock_irqsave(&ioc->raid_device_lock, flags);
        list_add_tail(&raid_device->list, &ioc->raid_device_list);
        spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
}

/**
 * _scsih_raid_device_remove - delete raid_device object
 * @ioc: per adapter object
 * @raid_device: raid_device object
 *
 */
static void
_scsih_raid_device_remove(struct MPT3SAS_ADAPTER *ioc,
        struct _raid_device *raid_device)
{
        unsigned long flags;

        spin_lock_irqsave(&ioc->raid_device_lock, flags);
        list_del(&raid_device->list);
        kfree(raid_device);
        spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
}

/**
 * mpt3sas_scsih_expander_find_by_handle - expander device search
 * @ioc: per adapter object
 * @handle: expander handle (assigned by firmware)
 * Context: Calling function should acquire ioc->sas_device_lock
 *
 * This searches for expander device based on handle, then returns the
 * sas_node object.
 */
struct _sas_node *
mpt3sas_scsih_expander_find_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _sas_node *sas_expander, *r;

        r = NULL;
        list_for_each_entry(sas_expander, &ioc->sas_expander_list, list) {
                if (sas_expander->handle != handle)
                        continue;
                r = sas_expander;
                goto out;
        }
 out:
        return r;
}

/**
 * mpt3sas_scsih_enclosure_find_by_handle - exclosure device search
 * @ioc: per adapter object
 * @handle: enclosure handle (assigned by firmware)
 * Context: Calling function should acquire ioc->sas_device_lock
 *
 * This searches for enclosure device based on handle, then returns the
 * enclosure object.
 */
static struct _enclosure_node *
mpt3sas_scsih_enclosure_find_by_handle(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct _enclosure_node *enclosure_dev, *r;

        r = NULL;
        list_for_each_entry(enclosure_dev, &ioc->enclosure_list, list) {
                if (le16_to_cpu(enclosure_dev->pg0.EnclosureHandle) != handle)
                        continue;
                r = enclosure_dev;
                goto out;
        }
out:
        return r;
}
/**
 * mpt3sas_scsih_expander_find_by_sas_address - expander device search
 * @ioc: per adapter object
 * @sas_address: sas address
 * @port: hba port entry
 * Context: Calling function should acquire ioc->sas_node_lock.
 *
 * This searches for expander device based on sas_address & port number,
 * then returns the sas_node object.
 */
struct _sas_node *
mpt3sas_scsih_expander_find_by_sas_address(struct MPT3SAS_ADAPTER *ioc,
        u64 sas_address, struct hba_port *port)
{
        struct _sas_node *sas_expander, *r = NULL;

        if (!port)
                return r;

        list_for_each_entry(sas_expander, &ioc->sas_expander_list, list) {
                if (sas_expander->sas_address != sas_address)
                        continue;
                if (sas_expander->port != port)
                        continue;
                r = sas_expander;
                goto out;
        }
 out:
        return r;
}

/**
 * _scsih_expander_node_add - insert expander device to the list.
 * @ioc: per adapter object
 * @sas_expander: the sas_device object
 * Context: This function will acquire ioc->sas_node_lock.
 *
 * Adding new object to the ioc->sas_expander_list.
 */
static void
_scsih_expander_node_add(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_node *sas_expander)
{
        unsigned long flags;

        spin_lock_irqsave(&ioc->sas_node_lock, flags);
        list_add_tail(&sas_expander->list, &ioc->sas_expander_list);
        spin_unlock_irqrestore(&ioc->sas_node_lock, flags);
}

/**
 * _scsih_is_end_device - determines if device is an end device
 * @device_info: bitfield providing information about the device.
 * Context: none
 *
 * Return: 1 if end device.
 */
static int
_scsih_is_end_device(u32 device_info)
{
        if (device_info & MPI2_SAS_DEVICE_INFO_END_DEVICE &&
                ((device_info & MPI2_SAS_DEVICE_INFO_SSP_TARGET) |
                (device_info & MPI2_SAS_DEVICE_INFO_STP_TARGET) |
                (device_info & MPI2_SAS_DEVICE_INFO_SATA_DEVICE)))
                return 1;
        else
                return 0;
}

/**
 * _scsih_is_nvme_pciescsi_device - determines if
 *                      device is an pcie nvme/scsi device
 * @device_info: bitfield providing information about the device.
 * Context: none
 *
 * Returns 1 if device is pcie device type nvme/scsi.
 */
static int
_scsih_is_nvme_pciescsi_device(u32 device_info)
{
        if (((device_info & MPI26_PCIE_DEVINFO_MASK_DEVICE_TYPE)
            == MPI26_PCIE_DEVINFO_NVME) ||
            ((device_info & MPI26_PCIE_DEVINFO_MASK_DEVICE_TYPE)
            == MPI26_PCIE_DEVINFO_SCSI))
                return 1;
        else
                return 0;
}

/**
 * _scsih_scsi_lookup_find_by_target - search for matching channel:id
 * @ioc: per adapter object
 * @id: target id
 * @channel: channel
 * Context: This function will acquire ioc->scsi_lookup_lock.
 *
 * This will search for a matching channel:id in the scsi_lookup array,
 * returning 1 if found.
 */
static u8
_scsih_scsi_lookup_find_by_target(struct MPT3SAS_ADAPTER *ioc, int id,
        int channel)
{
        int smid;
        struct scsi_cmnd *scmd;

        for (smid = 1;
             smid <= ioc->shost->can_queue; smid++) {
                scmd = mpt3sas_scsih_scsi_lookup_get(ioc, smid);
                if (!scmd)
                        continue;
                if (scmd->device->id == id &&
                    scmd->device->channel == channel)
                        return 1;
        }
        return 0;
}

/**
 * _scsih_scsi_lookup_find_by_lun - search for matching channel:id:lun
 * @ioc: per adapter object
 * @id: target id
 * @lun: lun number
 * @channel: channel
 * Context: This function will acquire ioc->scsi_lookup_lock.
 *
 * This will search for a matching channel:id:lun in the scsi_lookup array,
 * returning 1 if found.
 */
static u8
_scsih_scsi_lookup_find_by_lun(struct MPT3SAS_ADAPTER *ioc, int id,
        unsigned int lun, int channel)
{
        int smid;
        struct scsi_cmnd *scmd;

        for (smid = 1; smid <= ioc->shost->can_queue; smid++) {

                scmd = mpt3sas_scsih_scsi_lookup_get(ioc, smid);
                if (!scmd)
                        continue;
                if (scmd->device->id == id &&
                    scmd->device->channel == channel &&
                    scmd->device->lun == lun)
                        return 1;
        }
        return 0;
}

/**
 * mpt3sas_scsih_scsi_lookup_get - returns scmd entry
 * @ioc: per adapter object
 * @smid: system request message index
 *
 * Return: the smid stored scmd pointer.
 * Then will dereference the stored scmd pointer.
 */
struct scsi_cmnd *
mpt3sas_scsih_scsi_lookup_get(struct MPT3SAS_ADAPTER *ioc, u16 smid)
{
        struct scsi_cmnd *scmd = NULL;
        struct scsiio_tracker *st;
        Mpi25SCSIIORequest_t *mpi_request;
        u16 tag = smid - 1;

        if (smid > 0  &&
            smid <= ioc->scsiio_depth - INTERNAL_SCSIIO_CMDS_COUNT) {
                u32 unique_tag =
                    ioc->io_queue_num[tag] << BLK_MQ_UNIQUE_TAG_BITS | tag;

                mpi_request = mpt3sas_base_get_msg_frame(ioc, smid);

                /*
                 * If SCSI IO request is outstanding at driver level then
                 * DevHandle filed must be non-zero. If DevHandle is zero
                 * then it means that this smid is free at driver level,
                 * so return NULL.
                 */
                if (!mpi_request->DevHandle)
                        return scmd;

                scmd = scsi_host_find_tag(ioc->shost, unique_tag);
                if (scmd) {
                        st = scsi_cmd_priv(scmd);
                        if (st->cb_idx == 0xFF || st->smid == 0)
                                scmd = NULL;
                }
        }
        return scmd;
}

/**
 * scsih_change_queue_depth - setting device queue depth
 * @sdev: scsi device struct
 * @qdepth: requested queue depth
 *
 * Return: queue depth.
 */
static int
scsih_change_queue_depth(struct scsi_device *sdev, int qdepth)
{
        struct Scsi_Host *shost = sdev->host;
        int max_depth;
        struct MPT3SAS_ADAPTER *ioc = shost_priv(shost);
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct _sas_device *sas_device;
        unsigned long flags;

        max_depth = shost->can_queue;

        /*
         * limit max device queue for SATA to 32 if enable_sdev_max_qd
         * is disabled.
         */
        if (ioc->enable_sdev_max_qd)
                goto not_sata;

        sas_device_priv_data = sdev->hostdata;
        if (!sas_device_priv_data)
                goto not_sata;
        sas_target_priv_data = sas_device_priv_data->sas_target;
        if (!sas_target_priv_data)
                goto not_sata;
        if ((sas_target_priv_data->flags & MPT_TARGET_FLAGS_VOLUME))
                goto not_sata;

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_from_target(ioc, sas_target_priv_data);
        if (sas_device) {
                if (sas_device->device_info & MPI2_SAS_DEVICE_INFO_SATA_DEVICE)
                        max_depth = MPT3SAS_SATA_QUEUE_DEPTH;

                sas_device_put(sas_device);
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

 not_sata:

        if (!sdev->tagged_supported)
                max_depth = 1;
        if (qdepth > max_depth)
                qdepth = max_depth;
        scsi_change_queue_depth(sdev, qdepth);
        sdev_printk(KERN_INFO, sdev,
            "qdepth(%d), tagged(%d), scsi_level(%d), cmd_que(%d)\n",
            sdev->queue_depth, sdev->tagged_supported,
            sdev->scsi_level, ((sdev->inquiry[7] & 2) >> 1));
        return sdev->queue_depth;
}

/**
 * mpt3sas_scsih_change_queue_depth - setting device queue depth
 * @sdev: scsi device struct
 * @qdepth: requested queue depth
 *
 * Returns nothing.
 */
void
mpt3sas_scsih_change_queue_depth(struct scsi_device *sdev, int qdepth)
{
        struct Scsi_Host *shost = sdev->host;
        struct MPT3SAS_ADAPTER *ioc = shost_priv(shost);

        if (ioc->enable_sdev_max_qd)
                qdepth = shost->can_queue;

        scsih_change_queue_depth(sdev, qdepth);
}

/**
 * scsih_target_alloc - target add routine
 * @starget: scsi target struct
 *
 * Return: 0 if ok. Any other return is assumed to be an error and
 * the device is ignored.
 */
static int
scsih_target_alloc(struct scsi_target *starget)
{
        struct Scsi_Host *shost = dev_to_shost(&starget->dev);
        struct MPT3SAS_ADAPTER *ioc = shost_priv(shost);
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct _sas_device *sas_device;
        struct _raid_device *raid_device;
        struct _pcie_device *pcie_device;
        unsigned long flags;
        struct sas_rphy *rphy;

        sas_target_priv_data = kzalloc(sizeof(*sas_target_priv_data),
                                       GFP_KERNEL);
        if (!sas_target_priv_data)
                return -ENOMEM;

        starget->hostdata = sas_target_priv_data;
        sas_target_priv_data->starget = starget;
        sas_target_priv_data->handle = MPT3SAS_INVALID_DEVICE_HANDLE;

        /* RAID volumes */
        if (starget->channel == RAID_CHANNEL) {
                spin_lock_irqsave(&ioc->raid_device_lock, flags);
                raid_device = _scsih_raid_device_find_by_id(ioc, starget->id,
                    starget->channel);
                if (raid_device) {
                        sas_target_priv_data->handle = raid_device->handle;
                        sas_target_priv_data->sas_address = raid_device->wwid;
                        sas_target_priv_data->flags |= MPT_TARGET_FLAGS_VOLUME;
                        if (ioc->is_warpdrive)
                                sas_target_priv_data->raid_device = raid_device;
                        raid_device->starget = starget;
                }
                spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
                return 0;
        }

        /* PCIe devices */
        if (starget->channel == PCIE_CHANNEL) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_by_idchannel(ioc, starget->id,
                        starget->channel);
                if (pcie_device) {
                        sas_target_priv_data->handle = pcie_device->handle;
                        sas_target_priv_data->sas_address = pcie_device->wwid;
                        sas_target_priv_data->port = NULL;
                        sas_target_priv_data->pcie_dev = pcie_device;
                        pcie_device->starget = starget;
                        pcie_device->id = starget->id;
                        pcie_device->channel = starget->channel;
                        sas_target_priv_data->flags |=
                                MPT_TARGET_FLAGS_PCIE_DEVICE;
                        if (pcie_device->fast_path)
                                sas_target_priv_data->flags |=
                                        MPT_TARGET_FASTPATH_IO;
                }
                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
                return 0;
        }

        /* sas/sata devices */
        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        rphy = dev_to_rphy(starget->dev.parent);
        sas_device = __mpt3sas_get_sdev_by_rphy(ioc, rphy);

        if (sas_device) {
                sas_target_priv_data->handle = sas_device->handle;
                sas_target_priv_data->sas_address = sas_device->sas_address;
                sas_target_priv_data->port = sas_device->port;
                sas_target_priv_data->sas_dev = sas_device;
                sas_device->starget = starget;
                sas_device->id = starget->id;
                sas_device->channel = starget->channel;
                if (test_bit(sas_device->handle, ioc->pd_handles))
                        sas_target_priv_data->flags |=
                            MPT_TARGET_FLAGS_RAID_COMPONENT;
                if (sas_device->fast_path)
                        sas_target_priv_data->flags |=
                                        MPT_TARGET_FASTPATH_IO;
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        return 0;
}

/**
 * scsih_target_destroy - target destroy routine
 * @starget: scsi target struct
 */
static void
scsih_target_destroy(struct scsi_target *starget)
{
        struct Scsi_Host *shost = dev_to_shost(&starget->dev);
        struct MPT3SAS_ADAPTER *ioc = shost_priv(shost);
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct _sas_device *sas_device;
        struct _raid_device *raid_device;
        struct _pcie_device *pcie_device;
        unsigned long flags;

        sas_target_priv_data = starget->hostdata;
        if (!sas_target_priv_data)
                return;

        if (starget->channel == RAID_CHANNEL) {
                spin_lock_irqsave(&ioc->raid_device_lock, flags);
                raid_device = _scsih_raid_device_find_by_id(ioc, starget->id,
                    starget->channel);
                if (raid_device) {
                        raid_device->starget = NULL;
                        raid_device->sdev = NULL;
                }
                spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
                goto out;
        }

        if (starget->channel == PCIE_CHANNEL) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_from_target(ioc,
                                                        sas_target_priv_data);
                if (pcie_device && (pcie_device->starget == starget) &&
                        (pcie_device->id == starget->id) &&
                        (pcie_device->channel == starget->channel))
                        pcie_device->starget = NULL;

                if (pcie_device) {
                        /*
                         * Corresponding get() is in _scsih_target_alloc()
                         */
                        sas_target_priv_data->pcie_dev = NULL;
                        pcie_device_put(pcie_device);
                        pcie_device_put(pcie_device);
                }
                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

                goto out;
        }

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_from_target(ioc, sas_target_priv_data);
        if (sas_device && (sas_device->starget == starget) &&
            (sas_device->id == starget->id) &&
            (sas_device->channel == starget->channel))
                sas_device->starget = NULL;

        if (sas_device) {
                /*
                 * Corresponding get() is in _scsih_target_alloc()
                 */
                sas_target_priv_data->sas_dev = NULL;
                sas_device_put(sas_device);

                sas_device_put(sas_device);
        }
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

 out:
        kfree(sas_target_priv_data);
        starget->hostdata = NULL;
}

/**
 * scsih_slave_alloc - device add routine
 * @sdev: scsi device struct
 *
 * Return: 0 if ok. Any other return is assumed to be an error and
 * the device is ignored.
 */
static int
scsih_slave_alloc(struct scsi_device *sdev)
{
        struct Scsi_Host *shost;
        struct MPT3SAS_ADAPTER *ioc;
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_target *starget;
        struct _raid_device *raid_device;
        struct _sas_device *sas_device;
        struct _pcie_device *pcie_device;
        unsigned long flags;

        sas_device_priv_data = kzalloc(sizeof(*sas_device_priv_data),
                                       GFP_KERNEL);
        if (!sas_device_priv_data)
                return -ENOMEM;

        sas_device_priv_data->lun = sdev->lun;
        sas_device_priv_data->flags = MPT_DEVICE_FLAGS_INIT;

        starget = scsi_target(sdev);
        sas_target_priv_data = starget->hostdata;
        sas_target_priv_data->num_luns++;
        sas_device_priv_data->sas_target = sas_target_priv_data;
        sdev->hostdata = sas_device_priv_data;
        if ((sas_target_priv_data->flags & MPT_TARGET_FLAGS_RAID_COMPONENT))
                sdev->no_uld_attach = 1;

        shost = dev_to_shost(&starget->dev);
        ioc = shost_priv(shost);
        if (starget->channel == RAID_CHANNEL) {
                spin_lock_irqsave(&ioc->raid_device_lock, flags);
                raid_device = _scsih_raid_device_find_by_id(ioc,
                    starget->id, starget->channel);
                if (raid_device)
                        raid_device->sdev = sdev; /* raid is single lun */
                spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
        }
        if (starget->channel == PCIE_CHANNEL) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_by_wwid(ioc,
                                sas_target_priv_data->sas_address);
                if (pcie_device && (pcie_device->starget == NULL)) {
                        sdev_printk(KERN_INFO, sdev,
                            "%s : pcie_device->starget set to starget @ %d\n",
                            __func__, __LINE__);
                        pcie_device->starget = starget;
                }

                if (pcie_device)
                        pcie_device_put(pcie_device);
                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        } else  if (!(sas_target_priv_data->flags & MPT_TARGET_FLAGS_VOLUME)) {
                spin_lock_irqsave(&ioc->sas_device_lock, flags);
                sas_device = __mpt3sas_get_sdev_by_addr(ioc,
                    sas_target_priv_data->sas_address,
                    sas_target_priv_data->port);
                if (sas_device && (sas_device->starget == NULL)) {
                        sdev_printk(KERN_INFO, sdev,
                        "%s : sas_device->starget set to starget @ %d\n",
                             __func__, __LINE__);
                        sas_device->starget = starget;
                }

                if (sas_device)
                        sas_device_put(sas_device);

                spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
        }

        return 0;
}

/**
 * scsih_slave_destroy - device destroy routine
 * @sdev: scsi device struct
 */
static void
scsih_slave_destroy(struct scsi_device *sdev)
{
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct scsi_target *starget;
        struct Scsi_Host *shost;
        struct MPT3SAS_ADAPTER *ioc;
        struct _sas_device *sas_device;
        struct _pcie_device *pcie_device;
        unsigned long flags;

        if (!sdev->hostdata)
                return;

        starget = scsi_target(sdev);
        sas_target_priv_data = starget->hostdata;
        sas_target_priv_data->num_luns--;

        shost = dev_to_shost(&starget->dev);
        ioc = shost_priv(shost);

        if (sas_target_priv_data->flags & MPT_TARGET_FLAGS_PCIE_DEVICE) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_from_target(ioc,
                                sas_target_priv_data);
                if (pcie_device && !sas_target_priv_data->num_luns)
                        pcie_device->starget = NULL;

                if (pcie_device)
                        pcie_device_put(pcie_device);

                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        } else if (!(sas_target_priv_data->flags & MPT_TARGET_FLAGS_VOLUME)) {
                spin_lock_irqsave(&ioc->sas_device_lock, flags);
                sas_device = __mpt3sas_get_sdev_from_target(ioc,
                                sas_target_priv_data);
                if (sas_device && !sas_target_priv_data->num_luns)
                        sas_device->starget = NULL;

                if (sas_device)
                        sas_device_put(sas_device);
                spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
        }

        kfree(sdev->hostdata);
        sdev->hostdata = NULL;
}

/**
 * _scsih_display_sata_capabilities - sata capabilities
 * @ioc: per adapter object
 * @handle: device handle
 * @sdev: scsi device struct
 */
static void
_scsih_display_sata_capabilities(struct MPT3SAS_ADAPTER *ioc,
        u16 handle, struct scsi_device *sdev)
{
        Mpi2ConfigReply_t mpi_reply;
        Mpi2SasDevicePage0_t sas_device_pg0;
        u32 ioc_status;
        u16 flags;
        u32 device_info;

        if ((mpt3sas_config_get_sas_device_pg0(ioc, &mpi_reply, &sas_device_pg0,
            MPI2_SAS_DEVICE_PGAD_FORM_HANDLE, handle))) {
                ioc_err(ioc, "failure at %s:%d/%s()!\n",
                        __FILE__, __LINE__, __func__);
                return;
        }

        ioc_status = le16_to_cpu(mpi_reply.IOCStatus) &
            MPI2_IOCSTATUS_MASK;
        if (ioc_status != MPI2_IOCSTATUS_SUCCESS) {
                ioc_err(ioc, "failure at %s:%d/%s()!\n",
                        __FILE__, __LINE__, __func__);
                return;
        }

        flags = le16_to_cpu(sas_device_pg0.Flags);
        device_info = le32_to_cpu(sas_device_pg0.DeviceInfo);

        sdev_printk(KERN_INFO, sdev,
            "atapi(%s), ncq(%s), asyn_notify(%s), smart(%s), fua(%s), "
            "sw_preserve(%s)\n",
            (device_info & MPI2_SAS_DEVICE_INFO_ATAPI_DEVICE) ? "y" : "n",
            (flags & MPI2_SAS_DEVICE0_FLAGS_SATA_NCQ_SUPPORTED) ? "y" : "n",
            (flags & MPI2_SAS_DEVICE0_FLAGS_SATA_ASYNCHRONOUS_NOTIFY) ? "y" :
            "n",
            (flags & MPI2_SAS_DEVICE0_FLAGS_SATA_SMART_SUPPORTED) ? "y" : "n",
            (flags & MPI2_SAS_DEVICE0_FLAGS_SATA_FUA_SUPPORTED) ? "y" : "n",
            (flags & MPI2_SAS_DEVICE0_FLAGS_SATA_SW_PRESERVE) ? "y" : "n");
}

/*
 * raid transport support -
 * Enabled for SLES11 and newer, in older kernels the driver will panic when
 * unloading the driver followed by a load - I believe that the subroutine
 * raid_class_release() is not cleaning up properly.
 */

/**
 * scsih_is_raid - return boolean indicating device is raid volume
 * @dev: the device struct object
 */
static int
scsih_is_raid(struct device *dev)
{
        struct scsi_device *sdev = to_scsi_device(dev);
        struct MPT3SAS_ADAPTER *ioc = shost_priv(sdev->host);

        if (ioc->is_warpdrive)
                return 0;
        return (sdev->channel == RAID_CHANNEL) ? 1 : 0;
}

static int
scsih_is_nvme(struct device *dev)
{
        struct scsi_device *sdev = to_scsi_device(dev);

        return (sdev->channel == PCIE_CHANNEL) ? 1 : 0;
}

/**
 * scsih_get_resync - get raid volume resync percent complete
 * @dev: the device struct object
 */
static void
scsih_get_resync(struct device *dev)
{
        struct scsi_device *sdev = to_scsi_device(dev);
        struct MPT3SAS_ADAPTER *ioc = shost_priv(sdev->host);
        static struct _raid_device *raid_device;
        unsigned long flags;
        Mpi2RaidVolPage0_t vol_pg0;
        Mpi2ConfigReply_t mpi_reply;
        u32 volume_status_flags;
        u8 percent_complete;
        u16 handle;

        percent_complete = 0;
        handle = 0;
        if (ioc->is_warpdrive)
                goto out;

        spin_lock_irqsave(&ioc->raid_device_lock, flags);
        raid_device = _scsih_raid_device_find_by_id(ioc, sdev->id,
            sdev->channel);
        if (raid_device) {
                handle = raid_device->handle;
                percent_complete = raid_device->percent_complete;
        }
        spin_unlock_irqrestore(&ioc->raid_device_lock, flags);

        if (!handle)
                goto out;

        if (mpt3sas_config_get_raid_volume_pg0(ioc, &mpi_reply, &vol_pg0,
             MPI2_RAID_VOLUME_PGAD_FORM_HANDLE, handle,
             sizeof(Mpi2RaidVolPage0_t))) {
                ioc_err(ioc, "failure at %s:%d/%s()!\n",
                        __FILE__, __LINE__, __func__);
                percent_complete = 0;
                goto out;
        }

        volume_status_flags = le32_to_cpu(vol_pg0.VolumeStatusFlags);
        if (!(volume_status_flags &
            MPI2_RAIDVOL0_STATUS_FLAG_RESYNC_IN_PROGRESS))
                percent_complete = 0;

 out:

        switch (ioc->hba_mpi_version_belonged) {
        case MPI2_VERSION:
                raid_set_resync(mpt2sas_raid_template, dev, percent_complete);
                break;
        case MPI25_VERSION:
        case MPI26_VERSION:
                raid_set_resync(mpt3sas_raid_template, dev, percent_complete);
                break;
        }
}

/**
 * scsih_get_state - get raid volume level
 * @dev: the device struct object
 */
static void
scsih_get_state(struct device *dev)
{
        struct scsi_device *sdev = to_scsi_device(dev);
        struct MPT3SAS_ADAPTER *ioc = shost_priv(sdev->host);
        static struct _raid_device *raid_device;
        unsigned long flags;
        Mpi2RaidVolPage0_t vol_pg0;
        Mpi2ConfigReply_t mpi_reply;
        u32 volstate;
        enum raid_state state = RAID_STATE_UNKNOWN;
        u16 handle = 0;

        spin_lock_irqsave(&ioc->raid_device_lock, flags);
        raid_device = _scsih_raid_device_find_by_id(ioc, sdev->id,
            sdev->channel);
        if (raid_device)
                handle = raid_device->handle;
        spin_unlock_irqrestore(&ioc->raid_device_lock, flags);

        if (!raid_device)
                goto out;

        if (mpt3sas_config_get_raid_volume_pg0(ioc, &mpi_reply, &vol_pg0,
             MPI2_RAID_VOLUME_PGAD_FORM_HANDLE, handle,
             sizeof(Mpi2RaidVolPage0_t))) {
                ioc_err(ioc, "failure at %s:%d/%s()!\n",
                        __FILE__, __LINE__, __func__);
                goto out;
        }

        volstate = le32_to_cpu(vol_pg0.VolumeStatusFlags);
        if (volstate & MPI2_RAIDVOL0_STATUS_FLAG_RESYNC_IN_PROGRESS) {
                state = RAID_STATE_RESYNCING;
                goto out;
        }

        switch (vol_pg0.VolumeState) {
        case MPI2_RAID_VOL_STATE_OPTIMAL:
        case MPI2_RAID_VOL_STATE_ONLINE:
                state = RAID_STATE_ACTIVE;
                break;
        case  MPI2_RAID_VOL_STATE_DEGRADED:
                state = RAID_STATE_DEGRADED;
                break;
        case MPI2_RAID_VOL_STATE_FAILED:
        case MPI2_RAID_VOL_STATE_MISSING:
                state = RAID_STATE_OFFLINE;
                break;
        }
 out:
        switch (ioc->hba_mpi_version_belonged) {
        case MPI2_VERSION:
                raid_set_state(mpt2sas_raid_template, dev, state);
                break;
        case MPI25_VERSION:
        case MPI26_VERSION:
                raid_set_state(mpt3sas_raid_template, dev, state);
                break;
        }
}

/**
 * _scsih_set_level - set raid level
 * @ioc: ?
 * @sdev: scsi device struct
 * @volume_type: volume type
 */
static void
_scsih_set_level(struct MPT3SAS_ADAPTER *ioc,
        struct scsi_device *sdev, u8 volume_type)
{
        enum raid_level level = RAID_LEVEL_UNKNOWN;

        switch (volume_type) {
        case MPI2_RAID_VOL_TYPE_RAID0:
                level = RAID_LEVEL_0;
                break;
        case MPI2_RAID_VOL_TYPE_RAID10:
                level = RAID_LEVEL_10;
                break;
        case MPI2_RAID_VOL_TYPE_RAID1E:
                level = RAID_LEVEL_1E;
                break;
        case MPI2_RAID_VOL_TYPE_RAID1:
                level = RAID_LEVEL_1;
                break;
        }

        switch (ioc->hba_mpi_version_belonged) {
        case MPI2_VERSION:
                raid_set_level(mpt2sas_raid_template,
                        &sdev->sdev_gendev, level);
                break;
        case MPI25_VERSION:
        case MPI26_VERSION:
                raid_set_level(mpt3sas_raid_template,
                        &sdev->sdev_gendev, level);
                break;
        }
}


/**
 * _scsih_get_volume_capabilities - volume capabilities
 * @ioc: per adapter object
 * @raid_device: the raid_device object
 *
 * Return: 0 for success, else 1
 */
static int
_scsih_get_volume_capabilities(struct MPT3SAS_ADAPTER *ioc,
        struct _raid_device *raid_device)
{
        Mpi2RaidVolPage0_t *vol_pg0;
        Mpi2RaidPhysDiskPage0_t pd_pg0;
        Mpi2SasDevicePage0_t sas_device_pg0;
        Mpi2ConfigReply_t mpi_reply;
        u16 sz;
        u8 num_pds;

        if ((mpt3sas_config_get_number_pds(ioc, raid_device->handle,
            &num_pds)) || !num_pds) {
                dfailprintk(ioc,
                            ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                     __FILE__, __LINE__, __func__));
                return 1;
        }

        raid_device->num_pds = num_pds;
        sz = offsetof(Mpi2RaidVolPage0_t, PhysDisk) + (num_pds *
            sizeof(Mpi2RaidVol0PhysDisk_t));
        vol_pg0 = kzalloc(sz, GFP_KERNEL);
        if (!vol_pg0) {
                dfailprintk(ioc,
                            ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                     __FILE__, __LINE__, __func__));
                return 1;
        }

        if ((mpt3sas_config_get_raid_volume_pg0(ioc, &mpi_reply, vol_pg0,
             MPI2_RAID_VOLUME_PGAD_FORM_HANDLE, raid_device->handle, sz))) {
                dfailprintk(ioc,
                            ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                     __FILE__, __LINE__, __func__));
                kfree(vol_pg0);
                return 1;
        }

        raid_device->volume_type = vol_pg0->VolumeType;

        /* figure out what the underlying devices are by
         * obtaining the device_info bits for the 1st device
         */
        if (!(mpt3sas_config_get_phys_disk_pg0(ioc, &mpi_reply,
            &pd_pg0, MPI2_PHYSDISK_PGAD_FORM_PHYSDISKNUM,
            vol_pg0->PhysDisk[0].PhysDiskNum))) {
                if (!(mpt3sas_config_get_sas_device_pg0(ioc, &mpi_reply,
                    &sas_device_pg0, MPI2_SAS_DEVICE_PGAD_FORM_HANDLE,
                    le16_to_cpu(pd_pg0.DevHandle)))) {
                        raid_device->device_info =
                            le32_to_cpu(sas_device_pg0.DeviceInfo);
                }
        }

        kfree(vol_pg0);
        return 0;
}

/**
 * _scsih_enable_tlr - setting TLR flags
 * @ioc: per adapter object
 * @sdev: scsi device struct
 *
 * Enabling Transaction Layer Retries for tape devices when
 * vpd page 0x90 is present
 *
 */
static void
_scsih_enable_tlr(struct MPT3SAS_ADAPTER *ioc, struct scsi_device *sdev)
{

        /* only for TAPE */
        if (sdev->type != TYPE_TAPE)
                return;

        if (!(ioc->facts.IOCCapabilities & MPI2_IOCFACTS_CAPABILITY_TLR))
                return;

        sas_enable_tlr(sdev);
        sdev_printk(KERN_INFO, sdev, "TLR %s\n",
            sas_is_tlr_enabled(sdev) ? "Enabled" : "Disabled");
        return;

}

/**
 * scsih_slave_configure - device configure routine.
 * @sdev: scsi device struct
 *
 * Return: 0 if ok. Any other return is assumed to be an error and
 * the device is ignored.
 */
static int
scsih_slave_configure(struct scsi_device *sdev)
{
        struct Scsi_Host *shost = sdev->host;
        struct MPT3SAS_ADAPTER *ioc = shost_priv(shost);
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct MPT3SAS_TARGET *sas_target_priv_data;
        struct _sas_device *sas_device;
        struct _pcie_device *pcie_device;
        struct _raid_device *raid_device;
        unsigned long flags;
        int qdepth;
        u8 ssp_target = 0;
        char *ds = "";
        char *r_level = "";
        u16 handle, volume_handle = 0;
        u64 volume_wwid = 0;

        qdepth = 1;
        sas_device_priv_data = sdev->hostdata;
        sas_device_priv_data->configured_lun = 1;
        sas_device_priv_data->flags &= ~MPT_DEVICE_FLAGS_INIT;
        sas_target_priv_data = sas_device_priv_data->sas_target;
        handle = sas_target_priv_data->handle;

        /* raid volume handling */
        if (sas_target_priv_data->flags & MPT_TARGET_FLAGS_VOLUME) {

                spin_lock_irqsave(&ioc->raid_device_lock, flags);
                raid_device = mpt3sas_raid_device_find_by_handle(ioc, handle);
                spin_unlock_irqrestore(&ioc->raid_device_lock, flags);
                if (!raid_device) {
                        dfailprintk(ioc,
                                    ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                             __FILE__, __LINE__, __func__));
                        return 1;
                }

                if (_scsih_get_volume_capabilities(ioc, raid_device)) {
                        dfailprintk(ioc,
                                    ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                             __FILE__, __LINE__, __func__));
                        return 1;
                }

                /*
                 * WARPDRIVE: Initialize the required data for Direct IO
                 */
                mpt3sas_init_warpdrive_properties(ioc, raid_device);

                /* RAID Queue Depth Support
                 * IS volume = underlying qdepth of drive type, either
                 *    MPT3SAS_SAS_QUEUE_DEPTH or MPT3SAS_SATA_QUEUE_DEPTH
                 * IM/IME/R10 = 128 (MPT3SAS_RAID_QUEUE_DEPTH)
                 */
                if (raid_device->device_info &
                    MPI2_SAS_DEVICE_INFO_SSP_TARGET) {
                        qdepth = MPT3SAS_SAS_QUEUE_DEPTH;
                        ds = "SSP";
                } else {
                        qdepth = MPT3SAS_SATA_QUEUE_DEPTH;
                        if (raid_device->device_info &
                            MPI2_SAS_DEVICE_INFO_SATA_DEVICE)
                                ds = "SATA";
                        else
                                ds = "STP";
                }

                switch (raid_device->volume_type) {
                case MPI2_RAID_VOL_TYPE_RAID0:
                        r_level = "RAID0";
                        break;
                case MPI2_RAID_VOL_TYPE_RAID1E:
                        qdepth = MPT3SAS_RAID_QUEUE_DEPTH;
                        if (ioc->manu_pg10.OEMIdentifier &&
                            (le32_to_cpu(ioc->manu_pg10.GenericFlags0) &
                            MFG10_GF0_R10_DISPLAY) &&
                            !(raid_device->num_pds % 2))
                                r_level = "RAID10";
                        else
                                r_level = "RAID1E";
                        break;
                case MPI2_RAID_VOL_TYPE_RAID1:
                        qdepth = MPT3SAS_RAID_QUEUE_DEPTH;
                        r_level = "RAID1";
                        break;
                case MPI2_RAID_VOL_TYPE_RAID10:
                        qdepth = MPT3SAS_RAID_QUEUE_DEPTH;
                        r_level = "RAID10";
                        break;
                case MPI2_RAID_VOL_TYPE_UNKNOWN:
                default:
                        qdepth = MPT3SAS_RAID_QUEUE_DEPTH;
                        r_level = "RAIDX";
                        break;
                }

                if (!ioc->hide_ir_msg)
                        sdev_printk(KERN_INFO, sdev,
                           "%s: handle(0x%04x), wwid(0x%016llx),"
                            " pd_count(%d), type(%s)\n",
                            r_level, raid_device->handle,
                            (unsigned long long)raid_device->wwid,
                            raid_device->num_pds, ds);

                if (shost->max_sectors > MPT3SAS_RAID_MAX_SECTORS) {
                        blk_queue_max_hw_sectors(sdev->request_queue,
                                                MPT3SAS_RAID_MAX_SECTORS);
                        sdev_printk(KERN_INFO, sdev,
                                        "Set queue's max_sector to: %u\n",
                                                MPT3SAS_RAID_MAX_SECTORS);
                }

                mpt3sas_scsih_change_queue_depth(sdev, qdepth);

                /* raid transport support */
                if (!ioc->is_warpdrive)
                        _scsih_set_level(ioc, sdev, raid_device->volume_type);
                return 0;
        }

        /* non-raid handling */
        if (sas_target_priv_data->flags & MPT_TARGET_FLAGS_RAID_COMPONENT) {
                if (mpt3sas_config_get_volume_handle(ioc, handle,
                    &volume_handle)) {
                        dfailprintk(ioc,
                                    ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                             __FILE__, __LINE__, __func__));
                        return 1;
                }
                if (volume_handle && mpt3sas_config_get_volume_wwid(ioc,
                    volume_handle, &volume_wwid)) {
                        dfailprintk(ioc,
                                    ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                             __FILE__, __LINE__, __func__));
                        return 1;
                }
        }

        /* PCIe handling */
        if (sas_target_priv_data->flags & MPT_TARGET_FLAGS_PCIE_DEVICE) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_by_wwid(ioc,
                                sas_device_priv_data->sas_target->sas_address);
                if (!pcie_device) {
                        spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
                        dfailprintk(ioc,
                                    ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                             __FILE__, __LINE__, __func__));
                        return 1;
                }

                qdepth = MPT3SAS_NVME_QUEUE_DEPTH;
                ds = "NVMe";
                sdev_printk(KERN_INFO, sdev,
                        "%s: handle(0x%04x), wwid(0x%016llx), port(%d)\n",
                        ds, handle, (unsigned long long)pcie_device->wwid,
                        pcie_device->port_num);
                if (pcie_device->enclosure_handle != 0)
                        sdev_printk(KERN_INFO, sdev,
                        "%s: enclosure logical id(0x%016llx), slot(%d)\n",
                        ds,
                        (unsigned long long)pcie_device->enclosure_logical_id,
                        pcie_device->slot);
                if (pcie_device->connector_name[0] != '\0')
                        sdev_printk(KERN_INFO, sdev,
                                "%s: enclosure level(0x%04x),"
                                "connector name( %s)\n", ds,
                                pcie_device->enclosure_level,
                                pcie_device->connector_name);

                if (pcie_device->nvme_mdts)
                        blk_queue_max_hw_sectors(sdev->request_queue,
                                        pcie_device->nvme_mdts/512);

                pcie_device_put(pcie_device);
                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);
                mpt3sas_scsih_change_queue_depth(sdev, qdepth);
                /* Enable QUEUE_FLAG_NOMERGES flag, so that IOs won't be
                 ** merged and can eliminate holes created during merging
                 ** operation.
                 **/
                blk_queue_flag_set(QUEUE_FLAG_NOMERGES,
                                sdev->request_queue);
                blk_queue_virt_boundary(sdev->request_queue,
                                ioc->page_size - 1);
                return 0;
        }

        spin_lock_irqsave(&ioc->sas_device_lock, flags);
        sas_device = __mpt3sas_get_sdev_by_addr(ioc,
           sas_device_priv_data->sas_target->sas_address,
           sas_device_priv_data->sas_target->port);
        if (!sas_device) {
                spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
                dfailprintk(ioc,
                            ioc_warn(ioc, "failure at %s:%d/%s()!\n",
                                     __FILE__, __LINE__, __func__));
                return 1;
        }

        sas_device->volume_handle = volume_handle;
        sas_device->volume_wwid = volume_wwid;
        if (sas_device->device_info & MPI2_SAS_DEVICE_INFO_SSP_TARGET) {
                qdepth = MPT3SAS_SAS_QUEUE_DEPTH;
                ssp_target = 1;
                if (sas_device->device_info &
                                MPI2_SAS_DEVICE_INFO_SEP) {
                        sdev_printk(KERN_WARNING, sdev,
                        "set ignore_delay_remove for handle(0x%04x)\n",
                        sas_device_priv_data->sas_target->handle);
                        sas_device_priv_data->ignore_delay_remove = 1;
                        ds = "SES";
                } else
                        ds = "SSP";
        } else {
                qdepth = MPT3SAS_SATA_QUEUE_DEPTH;
                if (sas_device->device_info & MPI2_SAS_DEVICE_INFO_STP_TARGET)
                        ds = "STP";
                else if (sas_device->device_info &
                    MPI2_SAS_DEVICE_INFO_SATA_DEVICE)
                        ds = "SATA";
        }

        sdev_printk(KERN_INFO, sdev, "%s: handle(0x%04x), " \
            "sas_addr(0x%016llx), phy(%d), device_name(0x%016llx)\n",
            ds, handle, (unsigned long long)sas_device->sas_address,
            sas_device->phy, (unsigned long long)sas_device->device_name);

        _scsih_display_enclosure_chassis_info(NULL, sas_device, sdev, NULL);

        sas_device_put(sas_device);
        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);

        if (!ssp_target)
                _scsih_display_sata_capabilities(ioc, handle, sdev);


        mpt3sas_scsih_change_queue_depth(sdev, qdepth);

        if (ssp_target) {
                sas_read_port_mode_page(sdev);
                _scsih_enable_tlr(ioc, sdev);
        }

        return 0;
}

/**
 * scsih_bios_param - fetch head, sector, cylinder info for a disk
 * @sdev: scsi device struct
 * @bdev: pointer to block device context
 * @capacity: device size (in 512 byte sectors)
 * @params: three element array to place output:
 *              params[0] number of heads (max 255)
 *              params[1] number of sectors (max 63)
 *              params[2] number of cylinders
 */
static int
scsih_bios_param(struct scsi_device *sdev, struct block_device *bdev,
        sector_t capacity, int params[])
{
        int             heads;
        int             sectors;
        sector_t        cylinders;
        ulong           dummy;

        heads = 64;
        sectors = 32;

        dummy = heads * sectors;
        cylinders = capacity;
        sector_div(cylinders, dummy);

        /*
         * Handle extended translation size for logical drives
         * > 1Gb
         */
        if ((ulong)capacity >= 0x200000) {
                heads = 255;
                sectors = 63;
                dummy = heads * sectors;
                cylinders = capacity;
                sector_div(cylinders, dummy);
        }

        /* return result */
        params[0] = heads;
        params[1] = sectors;
        params[2] = cylinders;

        return 0;
}

/**
 * _scsih_response_code - translation of device response code
 * @ioc: per adapter object
 * @response_code: response code returned by the device
 */
static void
_scsih_response_code(struct MPT3SAS_ADAPTER *ioc, u8 response_code)
{
        char *desc;

        switch (response_code) {
        case MPI2_SCSITASKMGMT_RSP_TM_COMPLETE:
                desc = "task management request completed";
                break;
        case MPI2_SCSITASKMGMT_RSP_INVALID_FRAME:
                desc = "invalid frame";
                break;
        case MPI2_SCSITASKMGMT_RSP_TM_NOT_SUPPORTED:
                desc = "task management request not supported";
                break;
        case MPI2_SCSITASKMGMT_RSP_TM_FAILED:
                desc = "task management request failed";
                break;
        case MPI2_SCSITASKMGMT_RSP_TM_SUCCEEDED:
                desc = "task management request succeeded";
                break;
        case MPI2_SCSITASKMGMT_RSP_TM_INVALID_LUN:
                desc = "invalid lun";
                break;
        case 0xA:
                desc = "overlapped tag attempted";
                break;
        case MPI2_SCSITASKMGMT_RSP_IO_QUEUED_ON_IOC:
                desc = "task queued, however not sent to target";
                break;
        default:
                desc = "unknown";
                break;
        }
        ioc_warn(ioc, "response_code(0x%01x): %s\n", response_code, desc);
}

/**
 * _scsih_tm_done - tm completion routine
 * @ioc: per adapter object
 * @smid: system request message index
 * @msix_index: MSIX table index supplied by the OS
 * @reply: reply message frame(lower 32bit addr)
 * Context: none.
 *
 * The callback handler when using scsih_issue_tm.
 *
 * Return: 1 meaning mf should be freed from _base_interrupt
 *         0 means the mf is freed from this function.
 */
static u8
_scsih_tm_done(struct MPT3SAS_ADAPTER *ioc, u16 smid, u8 msix_index, u32 reply)
{
        MPI2DefaultReply_t *mpi_reply;

        if (ioc->tm_cmds.status == MPT3_CMD_NOT_USED)
                return 1;
        if (ioc->tm_cmds.smid != smid)
                return 1;
        ioc->tm_cmds.status |= MPT3_CMD_COMPLETE;
        mpi_reply =  mpt3sas_base_get_reply_virt_addr(ioc, reply);
        if (mpi_reply) {
                memcpy(ioc->tm_cmds.reply, mpi_reply, mpi_reply->MsgLength*4);
                ioc->tm_cmds.status |= MPT3_CMD_REPLY_VALID;
        }
        ioc->tm_cmds.status &= ~MPT3_CMD_PENDING;
        complete(&ioc->tm_cmds.done);
        return 1;
}

/**
 * mpt3sas_scsih_set_tm_flag - set per target tm_busy
 * @ioc: per adapter object
 * @handle: device handle
 *
 * During taskmangement request, we need to freeze the device queue.
 */
void
mpt3sas_scsih_set_tm_flag(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;
        u8 skip = 0;

        shost_for_each_device(sdev, ioc->shost) {
                if (skip)
                        continue;
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (sas_device_priv_data->sas_target->handle == handle) {
                        sas_device_priv_data->sas_target->tm_busy = 1;
                        skip = 1;
                        ioc->ignore_loginfos = 1;
                }
        }
}

/**
 * mpt3sas_scsih_clear_tm_flag - clear per target tm_busy
 * @ioc: per adapter object
 * @handle: device handle
 *
 * During taskmangement request, we need to freeze the device queue.
 */
void
mpt3sas_scsih_clear_tm_flag(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;
        u8 skip = 0;

        shost_for_each_device(sdev, ioc->shost) {
                if (skip)
                        continue;
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (sas_device_priv_data->sas_target->handle == handle) {
                        sas_device_priv_data->sas_target->tm_busy = 0;
                        skip = 1;
                        ioc->ignore_loginfos = 0;
                }
        }
}

/**
 * scsih_tm_cmd_map_status - map the target reset & LUN reset TM status
 * @ioc: per adapter object
 * @channel: the channel assigned by the OS
 * @id: the id assigned by the OS
 * @lun: lun number
 * @type: MPI2_SCSITASKMGMT_TASKTYPE__XXX (defined in mpi2_init.h)
 * @smid_task: smid assigned to the task
 *
 * Look whether TM has aborted the timed out SCSI command, if
 * TM has aborted the IO then return SUCCESS else return FAILED.
 */
static int
scsih_tm_cmd_map_status(struct MPT3SAS_ADAPTER *ioc, uint channel,
        uint id, uint lun, u8 type, u16 smid_task)
{

        if (smid_task <= ioc->shost->can_queue) {
                switch (type) {
                case MPI2_SCSITASKMGMT_TASKTYPE_TARGET_RESET:
                        if (!(_scsih_scsi_lookup_find_by_target(ioc,
                            id, channel)))
                                return SUCCESS;
                        break;
                case MPI2_SCSITASKMGMT_TASKTYPE_ABRT_TASK_SET:
                case MPI2_SCSITASKMGMT_TASKTYPE_LOGICAL_UNIT_RESET:
                        if (!(_scsih_scsi_lookup_find_by_lun(ioc, id,
                            lun, channel)))
                                return SUCCESS;
                        break;
                default:
                        return SUCCESS;
                }
        } else if (smid_task == ioc->scsih_cmds.smid) {
                if ((ioc->scsih_cmds.status & MPT3_CMD_COMPLETE) ||
                    (ioc->scsih_cmds.status & MPT3_CMD_NOT_USED))
                        return SUCCESS;
        } else if (smid_task == ioc->ctl_cmds.smid) {
                if ((ioc->ctl_cmds.status & MPT3_CMD_COMPLETE) ||
                    (ioc->ctl_cmds.status & MPT3_CMD_NOT_USED))
                        return SUCCESS;
        }

        return FAILED;
}

/**
 * scsih_tm_post_processing - post processing of target & LUN reset
 * @ioc: per adapter object
 * @handle: device handle
 * @channel: the channel assigned by the OS
 * @id: the id assigned by the OS
 * @lun: lun number
 * @type: MPI2_SCSITASKMGMT_TASKTYPE__XXX (defined in mpi2_init.h)
 * @smid_task: smid assigned to the task
 *
 * Post processing of target & LUN reset. Due to interrupt latency
 * issue it possible that interrupt for aborted IO might not be
 * received yet. So before returning failure status, poll the
 * reply descriptor pools for the reply of timed out SCSI command.
 * Return FAILED status if reply for timed out is not received
 * otherwise return SUCCESS.
 */
static int
scsih_tm_post_processing(struct MPT3SAS_ADAPTER *ioc, u16 handle,
        uint channel, uint id, uint lun, u8 type, u16 smid_task)
{
        int rc;

        rc = scsih_tm_cmd_map_status(ioc, channel, id, lun, type, smid_task);

        if (rc == SUCCESS)
                return rc;

        ioc_info(ioc,
            "Poll ReplyDescriptor queues for completion of"
            " smid(%d), task_type(0x%02x), handle(0x%04x)\n",
            smid_task, type, handle);

        /*
         * Due to interrupt latency issues, driver may receive interrupt for
         * TM first and then for aborted SCSI IO command. So, poll all the
         * ReplyDescriptor pools before returning the FAILED status to SML.
         */
        mpt3sas_base_mask_interrupts(ioc);
        mpt3sas_base_sync_reply_irqs(ioc, 1);
        mpt3sas_base_unmask_interrupts(ioc);

        return scsih_tm_cmd_map_status(ioc, channel, id, lun, type, smid_task);
}

/**
 * mpt3sas_scsih_issue_tm - main routine for sending tm requests
 * @ioc: per adapter struct
 * @handle: device handle
 * @channel: the channel assigned by the OS
 * @id: the id assigned by the OS
 * @lun: lun number
 * @type: MPI2_SCSITASKMGMT_TASKTYPE__XXX (defined in mpi2_init.h)
 * @smid_task: smid assigned to the task
 * @msix_task: MSIX table index supplied by the OS
 * @timeout: timeout in seconds
 * @tr_method: Target Reset Method
 * Context: user
 *
 * A generic API for sending task management requests to firmware.
 *
 * The callback index is set inside `ioc->tm_cb_idx`.
 * The caller is responsible to check for outstanding commands.
 *
 * Return: SUCCESS or FAILED.
 */
int
mpt3sas_scsih_issue_tm(struct MPT3SAS_ADAPTER *ioc, u16 handle, uint channel,
        uint id, u64 lun, u8 type, u16 smid_task, u16 msix_task,
        u8 timeout, u8 tr_method)
{
        Mpi2SCSITaskManagementRequest_t *mpi_request;
        Mpi2SCSITaskManagementReply_t *mpi_reply;
        Mpi25SCSIIORequest_t *request;
        u16 smid = 0;
        u32 ioc_state;
        int rc;
        u8 issue_reset = 0;

        lockdep_assert_held(&ioc->tm_cmds.mutex);

        if (ioc->tm_cmds.status != MPT3_CMD_NOT_USED) {
                ioc_info(ioc, "%s: tm_cmd busy!!!\n", __func__);
                return FAILED;
        }

        if (ioc->shost_recovery || ioc->remove_host ||
            ioc->pci_error_recovery) {
                ioc_info(ioc, "%s: host reset in progress!\n", __func__);
                return FAILED;
        }

        ioc_state = mpt3sas_base_get_iocstate(ioc, 0);
        if (ioc_state & MPI2_DOORBELL_USED) {
                dhsprintk(ioc, ioc_info(ioc, "unexpected doorbell active!\n"));
                rc = mpt3sas_base_hard_reset_handler(ioc, FORCE_BIG_HAMMER);
                return (!rc) ? SUCCESS : FAILED;
        }

        if ((ioc_state & MPI2_IOC_STATE_MASK) == MPI2_IOC_STATE_FAULT) {
                mpt3sas_print_fault_code(ioc, ioc_state &
                    MPI2_DOORBELL_DATA_MASK);
                rc = mpt3sas_base_hard_reset_handler(ioc, FORCE_BIG_HAMMER);
                return (!rc) ? SUCCESS : FAILED;
        } else if ((ioc_state & MPI2_IOC_STATE_MASK) ==
            MPI2_IOC_STATE_COREDUMP) {
                mpt3sas_print_coredump_info(ioc, ioc_state &
                    MPI2_DOORBELL_DATA_MASK);
                rc = mpt3sas_base_hard_reset_handler(ioc, FORCE_BIG_HAMMER);
                return (!rc) ? SUCCESS : FAILED;
        }

        smid = mpt3sas_base_get_smid_hpr(ioc, ioc->tm_cb_idx);
        if (!smid) {
                ioc_err(ioc, "%s: failed obtaining a smid\n", __func__);
                return FAILED;
        }

        dtmprintk(ioc,
                  ioc_info(ioc, "sending tm: handle(0x%04x), task_type(0x%02x), smid(%d), timeout(%d), tr_method(0x%x)\n",
                           handle, type, smid_task, timeout, tr_method));
        ioc->tm_cmds.status = MPT3_CMD_PENDING;
        mpi_request = mpt3sas_base_get_msg_frame(ioc, smid);
        ioc->tm_cmds.smid = smid;
        memset(mpi_request, 0, sizeof(Mpi2SCSITaskManagementRequest_t));
        memset(ioc->tm_cmds.reply, 0, sizeof(Mpi2SCSITaskManagementReply_t));
        mpi_request->Function = MPI2_FUNCTION_SCSI_TASK_MGMT;
        mpi_request->DevHandle = cpu_to_le16(handle);
        mpi_request->TaskType = type;
        if (type == MPI2_SCSITASKMGMT_TASKTYPE_ABORT_TASK ||
            type == MPI2_SCSITASKMGMT_TASKTYPE_QUERY_TASK)
                mpi_request->MsgFlags = tr_method;
        mpi_request->TaskMID = cpu_to_le16(smid_task);
        int_to_scsilun(lun, (struct scsi_lun *)mpi_request->LUN);
        mpt3sas_scsih_set_tm_flag(ioc, handle);
        init_completion(&ioc->tm_cmds.done);
        ioc->put_smid_hi_priority(ioc, smid, msix_task);
        wait_for_completion_timeout(&ioc->tm_cmds.done, timeout*HZ);
        if (!(ioc->tm_cmds.status & MPT3_CMD_COMPLETE)) {
                mpt3sas_check_cmd_timeout(ioc,
                    ioc->tm_cmds.status, mpi_request,
                    sizeof(Mpi2SCSITaskManagementRequest_t)/4, issue_reset);
                if (issue_reset) {
                        rc = mpt3sas_base_hard_reset_handler(ioc,
                                        FORCE_BIG_HAMMER);
                        rc = (!rc) ? SUCCESS : FAILED;
                        goto out;
                }
        }

        /* sync IRQs in case those were busy during flush. */
        mpt3sas_base_sync_reply_irqs(ioc, 0);

        if (ioc->tm_cmds.status & MPT3_CMD_REPLY_VALID) {
                mpt3sas_trigger_master(ioc, MASTER_TRIGGER_TASK_MANAGMENT);
                mpi_reply = ioc->tm_cmds.reply;
                dtmprintk(ioc,
                          ioc_info(ioc, "complete tm: ioc_status(0x%04x), loginfo(0x%08x), term_count(0x%08x)\n",
                                   le16_to_cpu(mpi_reply->IOCStatus),
                                   le32_to_cpu(mpi_reply->IOCLogInfo),
                                   le32_to_cpu(mpi_reply->TerminationCount)));
                if (ioc->logging_level & MPT_DEBUG_TM) {
                        _scsih_response_code(ioc, mpi_reply->ResponseCode);
                        if (mpi_reply->IOCStatus)
                                _debug_dump_mf(mpi_request,
                                    sizeof(Mpi2SCSITaskManagementRequest_t)/4);
                }
        }

        switch (type) {
        case MPI2_SCSITASKMGMT_TASKTYPE_ABORT_TASK:
                rc = SUCCESS;
                /*
                 * If DevHandle filed in smid_task's entry of request pool
                 * doesn't match with device handle on which this task abort
                 * TM is received then it means that TM has successfully
                 * aborted the timed out command. Since smid_task's entry in
                 * request pool will be memset to zero once the timed out
                 * command is returned to the SML. If the command is not
                 * aborted then smid_task’s entry won’t be cleared and it
                 * will have same DevHandle value on which this task abort TM
                 * is received and driver will return the TM status as FAILED.
                 */
                request = mpt3sas_base_get_msg_frame(ioc, smid_task);
                if (le16_to_cpu(request->DevHandle) != handle)
                        break;

                ioc_info(ioc, "Task abort tm failed: handle(0x%04x),"
                    "timeout(%d) tr_method(0x%x) smid(%d) msix_index(%d)\n",
                    handle, timeout, tr_method, smid_task, msix_task);
                rc = FAILED;
                break;

        case MPI2_SCSITASKMGMT_TASKTYPE_TARGET_RESET:
        case MPI2_SCSITASKMGMT_TASKTYPE_ABRT_TASK_SET:
        case MPI2_SCSITASKMGMT_TASKTYPE_LOGICAL_UNIT_RESET:
                rc = scsih_tm_post_processing(ioc, handle, channel, id, lun,
                    type, smid_task);
                break;
        case MPI2_SCSITASKMGMT_TASKTYPE_QUERY_TASK:
                rc = SUCCESS;
                break;
        default:
                rc = FAILED;
                break;
        }

out:
        mpt3sas_scsih_clear_tm_flag(ioc, handle);
        ioc->tm_cmds.status = MPT3_CMD_NOT_USED;
        return rc;
}

int mpt3sas_scsih_issue_locked_tm(struct MPT3SAS_ADAPTER *ioc, u16 handle,
                uint channel, uint id, u64 lun, u8 type, u16 smid_task,
                u16 msix_task, u8 timeout, u8 tr_method)
{
        int ret;

        mutex_lock(&ioc->tm_cmds.mutex);
        ret = mpt3sas_scsih_issue_tm(ioc, handle, channel, id, lun, type,
                        smid_task, msix_task, timeout, tr_method);
        mutex_unlock(&ioc->tm_cmds.mutex);

        return ret;
}

/**
 * _scsih_tm_display_info - displays info about the device
 * @ioc: per adapter struct
 * @scmd: pointer to scsi command object
 *
 * Called by task management callback handlers.
 */
static void
_scsih_tm_display_info(struct MPT3SAS_ADAPTER *ioc, struct scsi_cmnd *scmd)
{
        struct scsi_target *starget = scmd->device->sdev_target;
        struct MPT3SAS_TARGET *priv_target = starget->hostdata;
        struct _sas_device *sas_device = NULL;
        struct _pcie_device *pcie_device = NULL;
        unsigned long flags;
        char *device_str = NULL;

        if (!priv_target)
                return;
        if (ioc->hide_ir_msg)
                device_str = "WarpDrive";
        else
                device_str = "volume";

        scsi_print_command(scmd);
        if (priv_target->flags & MPT_TARGET_FLAGS_VOLUME) {
                starget_printk(KERN_INFO, starget,
                        "%s handle(0x%04x), %s wwid(0x%016llx)\n",
                        device_str, priv_target->handle,
                    device_str, (unsigned long long)priv_target->sas_address);

        } else if (priv_target->flags & MPT_TARGET_FLAGS_PCIE_DEVICE) {
                spin_lock_irqsave(&ioc->pcie_device_lock, flags);
                pcie_device = __mpt3sas_get_pdev_from_target(ioc, priv_target);
                if (pcie_device) {
                        starget_printk(KERN_INFO, starget,
                                "handle(0x%04x), wwid(0x%016llx), port(%d)\n",
                                pcie_device->handle,
                                (unsigned long long)pcie_device->wwid,
                                pcie_device->port_num);
                        if (pcie_device->enclosure_handle != 0)
                                starget_printk(KERN_INFO, starget,
                                        "enclosure logical id(0x%016llx), slot(%d)\n",
                                        (unsigned long long)
                                        pcie_device->enclosure_logical_id,
                                        pcie_device->slot);
                        if (pcie_device->connector_name[0] != '\0')
                                starget_printk(KERN_INFO, starget,
                                        "enclosure level(0x%04x), connector name( %s)\n",
                                        pcie_device->enclosure_level,
                                        pcie_device->connector_name);
                        pcie_device_put(pcie_device);
                }
                spin_unlock_irqrestore(&ioc->pcie_device_lock, flags);

        } else {
                spin_lock_irqsave(&ioc->sas_device_lock, flags);
                sas_device = __mpt3sas_get_sdev_from_target(ioc, priv_target);
                if (sas_device) {
                        if (priv_target->flags &
                            MPT_TARGET_FLAGS_RAID_COMPONENT) {
                                starget_printk(KERN_INFO, starget,
                                    "volume handle(0x%04x), "
                                    "volume wwid(0x%016llx)\n",
                                    sas_device->volume_handle,
                                   (unsigned long long)sas_device->volume_wwid);
                        }
                        starget_printk(KERN_INFO, starget,
                            "handle(0x%04x), sas_address(0x%016llx), phy(%d)\n",
                            sas_device->handle,
                            (unsigned long long)sas_device->sas_address,
                            sas_device->phy);

                        _scsih_display_enclosure_chassis_info(NULL, sas_device,
                            NULL, starget);

                        sas_device_put(sas_device);
                }
                spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
        }
}

/**
 * scsih_abort - eh threads main abort routine
 * @scmd: pointer to scsi command object
 *
 * Return: SUCCESS if command aborted else FAILED
 */
static int
scsih_abort(struct scsi_cmnd *scmd)
{
        struct MPT3SAS_ADAPTER *ioc = shost_priv(scmd->device->host);
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsiio_tracker *st = scsi_cmd_priv(scmd);
        u16 handle;
        int r;

        u8 timeout = 30;
        struct _pcie_device *pcie_device = NULL;
        sdev_printk(KERN_INFO, scmd->device, "attempting task abort!"
            "scmd(0x%p), outstanding for %u ms & timeout %u ms\n",
            scmd, jiffies_to_msecs(jiffies - scmd->jiffies_at_alloc),
            (scmd->request->timeout / HZ) * 1000);
        _scsih_tm_display_info(ioc, scmd);

        sas_device_priv_data = scmd->device->hostdata;
        if (!sas_device_priv_data || !sas_device_priv_data->sas_target ||
            ioc->remove_host) {
                sdev_printk(KERN_INFO, scmd->device,
                    "device been deleted! scmd(0x%p)\n", scmd);
                scmd->result = DID_NO_CONNECT << 16;
                scmd->scsi_done(scmd);
                r = SUCCESS;
                goto out;
        }

        /* check for completed command */
        if (st == NULL || st->cb_idx == 0xFF) {
                sdev_printk(KERN_INFO, scmd->device, "No reference found at "
                    "driver, assuming scmd(0x%p) might have completed\n", scmd);
                scmd->result = DID_RESET << 16;
                r = SUCCESS;
                goto out;
        }

        /* for hidden raid components and volumes this is not supported */
        if (sas_device_priv_data->sas_target->flags &
            MPT_TARGET_FLAGS_RAID_COMPONENT ||
            sas_device_priv_data->sas_target->flags & MPT_TARGET_FLAGS_VOLUME) {
                scmd->result = DID_RESET << 16;
                r = FAILED;
                goto out;
        }

        mpt3sas_halt_firmware(ioc);

        handle = sas_device_priv_data->sas_target->handle;
        pcie_device = mpt3sas_get_pdev_by_handle(ioc, handle);
        if (pcie_device && (!ioc->tm_custom_handling) &&
            (!(mpt3sas_scsih_is_pcie_scsi_device(pcie_device->device_info))))
                timeout = ioc->nvme_abort_timeout;
        r = mpt3sas_scsih_issue_locked_tm(ioc, handle, scmd->device->channel,
                scmd->device->id, scmd->device->lun,
                MPI2_SCSITASKMGMT_TASKTYPE_ABORT_TASK,
                st->smid, st->msix_io, timeout, 0);
        /* Command must be cleared after abort */
        if (r == SUCCESS && st->cb_idx != 0xFF)
                r = FAILED;
 out:
        sdev_printk(KERN_INFO, scmd->device, "task abort: %s scmd(0x%p)\n",
            ((r == SUCCESS) ? "SUCCESS" : "FAILED"), scmd);
        if (pcie_device)
                pcie_device_put(pcie_device);
        return r;
}

/**
 * scsih_dev_reset - eh threads main device reset routine
 * @scmd: pointer to scsi command object
 *
 * Return: SUCCESS if command aborted else FAILED
 */
static int
scsih_dev_reset(struct scsi_cmnd *scmd)
{
        struct MPT3SAS_ADAPTER *ioc = shost_priv(scmd->device->host);
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct _sas_device *sas_device = NULL;
        struct _pcie_device *pcie_device = NULL;
        u16     handle;
        u8      tr_method = 0;
        u8      tr_timeout = 30;
        int r;

        struct scsi_target *starget = scmd->device->sdev_target;
        struct MPT3SAS_TARGET *target_priv_data = starget->hostdata;

        sdev_printk(KERN_INFO, scmd->device,
            "attempting device reset! scmd(0x%p)\n", scmd);
        _scsih_tm_display_info(ioc, scmd);

        sas_device_priv_data = scmd->device->hostdata;
        if (!sas_device_priv_data || !sas_device_priv_data->sas_target ||
            ioc->remove_host) {
                sdev_printk(KERN_INFO, scmd->device,
                    "device been deleted! scmd(0x%p)\n", scmd);
                scmd->result = DID_NO_CONNECT << 16;
                scmd->scsi_done(scmd);
                r = SUCCESS;
                goto out;
        }

        /* for hidden raid components obtain the volume_handle */
        handle = 0;
        if (sas_device_priv_data->sas_target->flags &
            MPT_TARGET_FLAGS_RAID_COMPONENT) {
                sas_device = mpt3sas_get_sdev_from_target(ioc,
                                target_priv_data);
                if (sas_device)
                        handle = sas_device->volume_handle;
        } else
                handle = sas_device_priv_data->sas_target->handle;

        if (!handle) {
                scmd->result = DID_RESET << 16;
                r = FAILED;
                goto out;
        }

        pcie_device = mpt3sas_get_pdev_by_handle(ioc, handle);

        if (pcie_device && (!ioc->tm_custom_handling) &&
            (!(mpt3sas_scsih_is_pcie_scsi_device(pcie_device->device_info)))) {
                tr_timeout = pcie_device->reset_timeout;
                tr_method = MPI26_SCSITASKMGMT_MSGFLAGS_PROTOCOL_LVL_RST_PCIE;
        } else
                tr_method = MPI2_SCSITASKMGMT_MSGFLAGS_LINK_RESET;

        r = mpt3sas_scsih_issue_locked_tm(ioc, handle, scmd->device->channel,
                scmd->device->id, scmd->device->lun,
                MPI2_SCSITASKMGMT_TASKTYPE_LOGICAL_UNIT_RESET, 0, 0,
                tr_timeout, tr_method);
        /* Check for busy commands after reset */
        if (r == SUCCESS && scsi_device_busy(scmd->device))
                r = FAILED;
 out:
        sdev_printk(KERN_INFO, scmd->device, "device reset: %s scmd(0x%p)\n",
            ((r == SUCCESS) ? "SUCCESS" : "FAILED"), scmd);

        if (sas_device)
                sas_device_put(sas_device);
        if (pcie_device)
                pcie_device_put(pcie_device);

        return r;
}

/**
 * scsih_target_reset - eh threads main target reset routine
 * @scmd: pointer to scsi command object
 *
 * Return: SUCCESS if command aborted else FAILED
 */
static int
scsih_target_reset(struct scsi_cmnd *scmd)
{
        struct MPT3SAS_ADAPTER *ioc = shost_priv(scmd->device->host);
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct _sas_device *sas_device = NULL;
        struct _pcie_device *pcie_device = NULL;
        u16     handle;
        u8      tr_method = 0;
        u8      tr_timeout = 30;
        int r;
        struct scsi_target *starget = scmd->device->sdev_target;
        struct MPT3SAS_TARGET *target_priv_data = starget->hostdata;

        starget_printk(KERN_INFO, starget,
            "attempting target reset! scmd(0x%p)\n", scmd);
        _scsih_tm_display_info(ioc, scmd);

        sas_device_priv_data = scmd->device->hostdata;
        if (!sas_device_priv_data || !sas_device_priv_data->sas_target ||
            ioc->remove_host) {
                starget_printk(KERN_INFO, starget,
                    "target been deleted! scmd(0x%p)\n", scmd);
                scmd->result = DID_NO_CONNECT << 16;
                scmd->scsi_done(scmd);
                r = SUCCESS;
                goto out;
        }

        /* for hidden raid components obtain the volume_handle */
        handle = 0;
        if (sas_device_priv_data->sas_target->flags &
            MPT_TARGET_FLAGS_RAID_COMPONENT) {
                sas_device = mpt3sas_get_sdev_from_target(ioc,
                                target_priv_data);
                if (sas_device)
                        handle = sas_device->volume_handle;
        } else
                handle = sas_device_priv_data->sas_target->handle;

        if (!handle) {
                scmd->result = DID_RESET << 16;
                r = FAILED;
                goto out;
        }

        pcie_device = mpt3sas_get_pdev_by_handle(ioc, handle);

        if (pcie_device && (!ioc->tm_custom_handling) &&
            (!(mpt3sas_scsih_is_pcie_scsi_device(pcie_device->device_info)))) {
                tr_timeout = pcie_device->reset_timeout;
                tr_method = MPI26_SCSITASKMGMT_MSGFLAGS_PROTOCOL_LVL_RST_PCIE;
        } else
                tr_method = MPI2_SCSITASKMGMT_MSGFLAGS_LINK_RESET;
        r = mpt3sas_scsih_issue_locked_tm(ioc, handle, scmd->device->channel,
                scmd->device->id, 0,
                MPI2_SCSITASKMGMT_TASKTYPE_TARGET_RESET, 0, 0,
            tr_timeout, tr_method);
        /* Check for busy commands after reset */
        if (r == SUCCESS && atomic_read(&starget->target_busy))
                r = FAILED;
 out:
        starget_printk(KERN_INFO, starget, "target reset: %s scmd(0x%p)\n",
            ((r == SUCCESS) ? "SUCCESS" : "FAILED"), scmd);

        if (sas_device)
                sas_device_put(sas_device);
        if (pcie_device)
                pcie_device_put(pcie_device);
        return r;
}


/**
 * scsih_host_reset - eh threads main host reset routine
 * @scmd: pointer to scsi command object
 *
 * Return: SUCCESS if command aborted else FAILED
 */
static int
scsih_host_reset(struct scsi_cmnd *scmd)
{
        struct MPT3SAS_ADAPTER *ioc = shost_priv(scmd->device->host);
        int r, retval;

        ioc_info(ioc, "attempting host reset! scmd(0x%p)\n", scmd);
        scsi_print_command(scmd);

        if (ioc->is_driver_loading || ioc->remove_host) {
                ioc_info(ioc, "Blocking the host reset\n");
                r = FAILED;
                goto out;
        }

        retval = mpt3sas_base_hard_reset_handler(ioc, FORCE_BIG_HAMMER);
        r = (retval < 0) ? FAILED : SUCCESS;
out:
        ioc_info(ioc, "host reset: %s scmd(0x%p)\n",
                 r == SUCCESS ? "SUCCESS" : "FAILED", scmd);

        return r;
}

/**
 * _scsih_fw_event_add - insert and queue up fw_event
 * @ioc: per adapter object
 * @fw_event: object describing the event
 * Context: This function will acquire ioc->fw_event_lock.
 *
 * This adds the firmware event object into link list, then queues it up to
 * be processed from user context.
 */
static void
_scsih_fw_event_add(struct MPT3SAS_ADAPTER *ioc, struct fw_event_work *fw_event)
{
        unsigned long flags;

        if (ioc->firmware_event_thread == NULL)
                return;

        spin_lock_irqsave(&ioc->fw_event_lock, flags);
        fw_event_work_get(fw_event);
        INIT_LIST_HEAD(&fw_event->list);
        list_add_tail(&fw_event->list, &ioc->fw_event_list);
        INIT_WORK(&fw_event->work, _firmware_event_work);
        fw_event_work_get(fw_event);
        queue_work(ioc->firmware_event_thread, &fw_event->work);
        spin_unlock_irqrestore(&ioc->fw_event_lock, flags);
}

/**
 * _scsih_fw_event_del_from_list - delete fw_event from the list
 * @ioc: per adapter object
 * @fw_event: object describing the event
 * Context: This function will acquire ioc->fw_event_lock.
 *
 * If the fw_event is on the fw_event_list, remove it and do a put.
 */
static void
_scsih_fw_event_del_from_list(struct MPT3SAS_ADAPTER *ioc, struct fw_event_work
        *fw_event)
{
        unsigned long flags;

        spin_lock_irqsave(&ioc->fw_event_lock, flags);
        if (!list_empty(&fw_event->list)) {
                list_del_init(&fw_event->list);
                fw_event_work_put(fw_event);
        }
        spin_unlock_irqrestore(&ioc->fw_event_lock, flags);
}


 /**
 * mpt3sas_send_trigger_data_event - send event for processing trigger data
 * @ioc: per adapter object
 * @event_data: trigger event data
 */
void
mpt3sas_send_trigger_data_event(struct MPT3SAS_ADAPTER *ioc,
        struct SL_WH_TRIGGERS_EVENT_DATA_T *event_data)
{
        struct fw_event_work *fw_event;
        u16 sz;

        if (ioc->is_driver_loading)
                return;
        sz = sizeof(*event_data);
        fw_event = alloc_fw_event_work(sz);
        if (!fw_event)
                return;
        fw_event->event = MPT3SAS_PROCESS_TRIGGER_DIAG;
        fw_event->ioc = ioc;
        memcpy(fw_event->event_data, event_data, sizeof(*event_data));
        _scsih_fw_event_add(ioc, fw_event);
        fw_event_work_put(fw_event);
}

/**
 * _scsih_error_recovery_delete_devices - remove devices not responding
 * @ioc: per adapter object
 */
static void
_scsih_error_recovery_delete_devices(struct MPT3SAS_ADAPTER *ioc)
{
        struct fw_event_work *fw_event;

        if (ioc->is_driver_loading)
                return;
        fw_event = alloc_fw_event_work(0);
        if (!fw_event)
                return;
        fw_event->event = MPT3SAS_REMOVE_UNRESPONDING_DEVICES;
        fw_event->ioc = ioc;
        _scsih_fw_event_add(ioc, fw_event);
        fw_event_work_put(fw_event);
}

/**
 * mpt3sas_port_enable_complete - port enable completed (fake event)
 * @ioc: per adapter object
 */
void
mpt3sas_port_enable_complete(struct MPT3SAS_ADAPTER *ioc)
{
        struct fw_event_work *fw_event;

        fw_event = alloc_fw_event_work(0);
        if (!fw_event)
                return;
        fw_event->event = MPT3SAS_PORT_ENABLE_COMPLETE;
        fw_event->ioc = ioc;
        _scsih_fw_event_add(ioc, fw_event);
        fw_event_work_put(fw_event);
}

static struct fw_event_work *dequeue_next_fw_event(struct MPT3SAS_ADAPTER *ioc)
{
        unsigned long flags;
        struct fw_event_work *fw_event = NULL;

        spin_lock_irqsave(&ioc->fw_event_lock, flags);
        if (!list_empty(&ioc->fw_event_list)) {
                fw_event = list_first_entry(&ioc->fw_event_list,
                                struct fw_event_work, list);
                list_del_init(&fw_event->list);
        }
        spin_unlock_irqrestore(&ioc->fw_event_lock, flags);

        return fw_event;
}

/**
 * _scsih_fw_event_cleanup_queue - cleanup event queue
 * @ioc: per adapter object
 *
 * Walk the firmware event queue, either killing timers, or waiting
 * for outstanding events to complete
 *
 * Context: task, can sleep
 */
static void
_scsih_fw_event_cleanup_queue(struct MPT3SAS_ADAPTER *ioc)
{
        struct fw_event_work *fw_event;

        if ((list_empty(&ioc->fw_event_list) && !ioc->current_event) ||
            !ioc->firmware_event_thread)
                return;

        ioc->fw_events_cleanup = 1;
        while ((fw_event = dequeue_next_fw_event(ioc)) ||
             (fw_event = ioc->current_event)) {
                /*
                 * Wait on the fw_event to complete. If this returns 1, then
                 * the event was never executed, and we need a put for the
                 * reference the work had on the fw_event.
                 *
                 * If it did execute, we wait for it to finish, and the put will
                 * happen from _firmware_event_work()
                 */
                if (cancel_work_sync(&fw_event->work))
                        fw_event_work_put(fw_event);

                fw_event_work_put(fw_event);
        }
        ioc->fw_events_cleanup = 0;
}

/**
 * _scsih_internal_device_block - block the sdev device
 * @sdev: per device object
 * @sas_device_priv_data : per device driver private data
 *
 * make sure device is blocked without error, if not
 * print an error
 */
static void
_scsih_internal_device_block(struct scsi_device *sdev,
                        struct MPT3SAS_DEVICE *sas_device_priv_data)
{
        int r = 0;

        sdev_printk(KERN_INFO, sdev, "device_block, handle(0x%04x)\n",
            sas_device_priv_data->sas_target->handle);
        sas_device_priv_data->block = 1;

        r = scsi_internal_device_block_nowait(sdev);
        if (r == -EINVAL)
                sdev_printk(KERN_WARNING, sdev,
                    "device_block failed with return(%d) for handle(0x%04x)\n",
                    r, sas_device_priv_data->sas_target->handle);
}

/**
 * _scsih_internal_device_unblock - unblock the sdev device
 * @sdev: per device object
 * @sas_device_priv_data : per device driver private data
 * make sure device is unblocked without error, if not retry
 * by blocking and then unblocking
 */

static void
_scsih_internal_device_unblock(struct scsi_device *sdev,
                        struct MPT3SAS_DEVICE *sas_device_priv_data)
{
        int r = 0;

        sdev_printk(KERN_WARNING, sdev, "device_unblock and setting to running, "
            "handle(0x%04x)\n", sas_device_priv_data->sas_target->handle);
        sas_device_priv_data->block = 0;
        r = scsi_internal_device_unblock_nowait(sdev, SDEV_RUNNING);
        if (r == -EINVAL) {
                /* The device has been set to SDEV_RUNNING by SD layer during
                 * device addition but the request queue is still stopped by
                 * our earlier block call. We need to perform a block again
                 * to get the device to SDEV_BLOCK and then to SDEV_RUNNING */

                sdev_printk(KERN_WARNING, sdev,
                    "device_unblock failed with return(%d) for handle(0x%04x) "
                    "performing a block followed by an unblock\n",
                    r, sas_device_priv_data->sas_target->handle);
                sas_device_priv_data->block = 1;
                r = scsi_internal_device_block_nowait(sdev);
                if (r)
                        sdev_printk(KERN_WARNING, sdev, "retried device_block "
                            "failed with return(%d) for handle(0x%04x)\n",
                            r, sas_device_priv_data->sas_target->handle);

                sas_device_priv_data->block = 0;
                r = scsi_internal_device_unblock_nowait(sdev, SDEV_RUNNING);
                if (r)
                        sdev_printk(KERN_WARNING, sdev, "retried device_unblock"
                            " failed with return(%d) for handle(0x%04x)\n",
                            r, sas_device_priv_data->sas_target->handle);
        }
}

/**
 * _scsih_ublock_io_all_device - unblock every device
 * @ioc: per adapter object
 *
 * change the device state from block to running
 */
static void
_scsih_ublock_io_all_device(struct MPT3SAS_ADAPTER *ioc)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;

        shost_for_each_device(sdev, ioc->shost) {
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (!sas_device_priv_data->block)
                        continue;

                dewtprintk(ioc, sdev_printk(KERN_INFO, sdev,
                        "device_running, handle(0x%04x)\n",
                    sas_device_priv_data->sas_target->handle));
                _scsih_internal_device_unblock(sdev, sas_device_priv_data);
        }
}


/**
 * _scsih_ublock_io_device - prepare device to be deleted
 * @ioc: per adapter object
 * @sas_address: sas address
 * @port: hba port entry
 *
 * unblock then put device in offline state
 */
static void
_scsih_ublock_io_device(struct MPT3SAS_ADAPTER *ioc,
        u64 sas_address, struct hba_port *port)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;

        shost_for_each_device(sdev, ioc->shost) {
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (sas_device_priv_data->sas_target->sas_address
                    != sas_address)
                        continue;
                if (sas_device_priv_data->sas_target->port != port)
                        continue;
                if (sas_device_priv_data->block)
                        _scsih_internal_device_unblock(sdev,
                                sas_device_priv_data);
        }
}

/**
 * _scsih_block_io_all_device - set the device state to SDEV_BLOCK
 * @ioc: per adapter object
 *
 * During device pull we need to appropriately set the sdev state.
 */
static void
_scsih_block_io_all_device(struct MPT3SAS_ADAPTER *ioc)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;

        shost_for_each_device(sdev, ioc->shost) {
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (sas_device_priv_data->block)
                        continue;
                if (sas_device_priv_data->ignore_delay_remove) {
                        sdev_printk(KERN_INFO, sdev,
                        "%s skip device_block for SES handle(0x%04x)\n",
                        __func__, sas_device_priv_data->sas_target->handle);
                        continue;
                }
                _scsih_internal_device_block(sdev, sas_device_priv_data);
        }
}

/**
 * _scsih_block_io_device - set the device state to SDEV_BLOCK
 * @ioc: per adapter object
 * @handle: device handle
 *
 * During device pull we need to appropriately set the sdev state.
 */
static void
_scsih_block_io_device(struct MPT3SAS_ADAPTER *ioc, u16 handle)
{
        struct MPT3SAS_DEVICE *sas_device_priv_data;
        struct scsi_device *sdev;
        struct _sas_device *sas_device;

        sas_device = mpt3sas_get_sdev_by_handle(ioc, handle);

        shost_for_each_device(sdev, ioc->shost) {
                sas_device_priv_data = sdev->hostdata;
                if (!sas_device_priv_data)
                        continue;
                if (sas_device_priv_data->sas_target->handle != handle)
                        continue;
                if (sas_device_priv_data->block)
                        continue;
                if (sas_device && sas_device->pend_sas_rphy_add)
                        continue;
                if (sas_device_priv_data->ignore_delay_remove) {
                        sdev_printk(KERN_INFO, sdev,
                        "%s skip device_block for SES handle(0x%04x)\n",
                        __func__, sas_device_priv_data->sas_target->handle);
                        continue;
                }
                _scsih_internal_device_block(sdev, sas_device_priv_data);
        }

        if (sas_device)
                sas_device_put(sas_device);
}

/**
 * _scsih_block_io_to_children_attached_to_ex
 * @ioc: per adapter object
 * @sas_expander: the sas_device object
 *
 * This routine set sdev state to SDEV_BLOCK for all devices
 * attached to this expander. This function called when expander is
 * pulled.
 */
static void
_scsih_block_io_to_children_attached_to_ex(struct MPT3SAS_ADAPTER *ioc,
        struct _sas_node *sas_expander)
{
        struct _sas_port *mpt3sas_port;
        struct _sas_device *sas_device;
        struct _sas_node *expander_sibling;
        unsigned long flags;

        if (!sas_expander)
                return;

        list_for_each_entry(mpt3sas_port,
           &sas_expander->sas_port_list, port_list) {
                if (mpt3sas_port->remote_identify.device_type ==
                    SAS_END_DEVICE) {
                        spin_lock_irqsave(&ioc->sas_device_lock, flags);
                        sas_device = __mpt3sas_get_sdev_by_addr(ioc,
                            mpt3sas_port->remote_identify.sas_address,
                            mpt3sas_port->hba_port);
                        if (sas_device) {
                                set_bit(sas_device->handle,
                                                ioc->blocking_handles);
                                sas_device_put(sas_device);
                        }
                        spin_unlock_irqrestore(&ioc->sas_device_lock, flags);
                }
        }

        list_for_each_entry(mpt3sas_port,
           &sas_expander->sas_port_list, port_list) {

                if (mpt3sas_port->remote_identify.device_type ==
                    SAS_EDGE_EXPANDER_DEVICE ||
                    mpt3sas_port->remote_identify.device_type ==
                    SAS_FANOUT_EXPANDER_DEVICE) {
                        expander_sibling =
                            mpt3sas_scsih_expander_find_by_sas_address(
                            ioc, mpt3sas_port->remote_identify.sas_address,
                            mpt3sas_port->hba_port);
                        _scsih_block_io_to_children_attached_to_ex(ioc,
                            expander_sibling);
                }
        }
}

/**
 * _scsih_block_io_to_children_attached_directly
 * @ioc: per adapter object
 * @event_data: topology change event data
 *
 * This routine set sdev state to SDEV_BLOCK for all devices
 * direct attached during device pull.
 */
static void
_scsih_block_io_to_children_attached_directly(struct MPT3SAS_ADAPTER *ioc,
        Mpi2EventDataSasTopologyChangeList_t *event_data)
{
        int i;
        u16 handle;
        u16 reason_code;

        for (i = 0; i < event_data->NumEntries; i++) {
                handle = le16_to_cpu(event_data->PHY[i].AttachedDevHandle);
                if (!handle)
                        continue;
                reason_code = event_data->PHY[i].PhyStatus &
                    MPI2_EVENT_SAS_TOPO_RC_MASK;
                if (reason_code == MPI2_EVENT_SAS_TOPO_RC_DELAY_NOT_RESPONDING)
                        _scsih_block_io_device(ioc, handle);
        }
}

/**
 * _scsih_block_io_to_pcie_children_attached_directly
 * @ioc: per adapter object
 * @event_data: topology change event data
 *
 * This routine set sdev state to SDEV_BLOCK for all devices
 * direct attached during device pull/reconnect.
 */
static void
_scsih_block_io_to_pcie_children_attached_directly(struct MPT3SAS_ADAPTER *ioc,
                Mpi26EventDataPCIeTopologyChangeList_t *event_data)