[mirror_ubuntu-bionic-kernel.git] / drivers / scsi / ses.c

/*
 * SCSI Enclosure Services
 *
 * Copyright (C) 2008 James Bottomley <James.Bottomley@HansenPartnership.com>
 *
**-----------------------------------------------------------------------------
**
**  This program is free software; you can redistribute it and/or
**  modify it under the terms of the GNU General Public License
**  version 2 as published by the Free Software Foundation.
**
**  This program is distributed in the hope that it will be useful,
**  but WITHOUT ANY WARRANTY; without even the implied warranty of
**  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
**  GNU General Public License for more details.
**
**  You should have received a copy of the GNU General Public License
**  along with this program; if not, write to the Free Software
**  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
**
**-----------------------------------------------------------------------------
*/

#include <linux/slab.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/enclosure.h>

#include <scsi/scsi.h>
#include <scsi/scsi_cmnd.h>
#include <scsi/scsi_dbg.h>
#include <scsi/scsi_device.h>
#include <scsi/scsi_driver.h>
#include <scsi/scsi_host.h>

struct ses_device {
	unsigned char *page1;
	unsigned char *page1_types;
	unsigned char *page2;
	unsigned char *page10;
	short page1_len;
	short page1_num_types;
	short page2_len;
	short page10_len;
};

struct ses_component {
	u64 addr;
	unsigned char *desc;
};

static int ses_probe(struct device *dev)
{
	struct scsi_device *sdev = to_scsi_device(dev);
	int err = -ENODEV;

	if (sdev->type != TYPE_ENCLOSURE)
		goto out;

	err = 0;
	sdev_printk(KERN_NOTICE, sdev, "Attached Enclosure device\n");

 out:
	return err;
}

#define SES_TIMEOUT (30 * HZ)
#define SES_RETRIES 3

static int ses_recv_diag(struct scsi_device *sdev, int page_code,
			 void *buf, int bufflen)
{
	unsigned char cmd[] = {
		RECEIVE_DIAGNOSTIC,
		1,		/* Set PCV bit */
		page_code,
		bufflen >> 8,
		bufflen & 0xff,
		0
	};

	return scsi_execute_req(sdev, cmd, DMA_FROM_DEVICE, buf, bufflen,
				NULL, SES_TIMEOUT, SES_RETRIES, NULL);
}

static int ses_send_diag(struct scsi_device *sdev, int page_code,
			 void *buf, int bufflen)
{
	u32 result;

	unsigned char cmd[] = {
		SEND_DIAGNOSTIC,
		0x10,		/* Set PF bit */
		0,
		bufflen >> 8,
		bufflen & 0xff,
		0
	};

	result = scsi_execute_req(sdev, cmd, DMA_TO_DEVICE, buf, bufflen,
				  NULL, SES_TIMEOUT, SES_RETRIES, NULL);
	if (result)
		sdev_printk(KERN_ERR, sdev, "SEND DIAGNOSTIC result: %8x\n",
			    result);
	return result;
}

static int ses_set_page2_descriptor(struct enclosure_device *edev,
				      struct enclosure_component *ecomp,
				      unsigned char *desc)
{
	int i, j, count = 0, descriptor = ecomp->number;
	struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
	struct ses_device *ses_dev = edev->scratch;
	unsigned char *type_ptr = ses_dev->page1_types;
	unsigned char *desc_ptr = ses_dev->page2 + 8;

	/* Clear everything */
	memset(desc_ptr, 0, ses_dev->page2_len - 8);
	for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
		for (j = 0; j < type_ptr[1]; j++) {
			desc_ptr += 4;
			if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
			    type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
				continue;
			if (count++ == descriptor) {
				memcpy(desc_ptr, desc, 4);
				/* set select */
				desc_ptr[0] |= 0x80;
				/* clear reserved, just in case */
				desc_ptr[0] &= 0xf0;
			}
		}
	}

	return ses_send_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len);
}

static unsigned char *ses_get_page2_descriptor(struct enclosure_device *edev,
				      struct enclosure_component *ecomp)
{
	int i, j, count = 0, descriptor = ecomp->number;
	struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
	struct ses_device *ses_dev = edev->scratch;
	unsigned char *type_ptr = ses_dev->page1_types;
	unsigned char *desc_ptr = ses_dev->page2 + 8;

	ses_recv_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len);

	for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
		for (j = 0; j < type_ptr[1]; j++) {
			desc_ptr += 4;
			if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
			    type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
				continue;
			if (count++ == descriptor)
				return desc_ptr;
		}
	}
	return NULL;
}

/* For device slot and array device slot elements, byte 3 bit 6
 * is "fault sensed" while byte 3 bit 5 is "fault reqstd". As this
 * code stands these bits are shifted 4 positions right so in
 * sysfs they will appear as bits 2 and 1 respectively. Strange. */
static void ses_get_fault(struct enclosure_device *edev,
			  struct enclosure_component *ecomp)
{
	unsigned char *desc;

	desc = ses_get_page2_descriptor(edev, ecomp);
	if (desc)
		ecomp->fault = (desc[3] & 0x60) >> 4;
}

static int ses_set_fault(struct enclosure_device *edev,
			  struct enclosure_component *ecomp,
			 enum enclosure_component_setting val)
{
	unsigned char desc[4] = {0 };

	switch (val) {
	case ENCLOSURE_SETTING_DISABLED:
		/* zero is disabled */
		break;
	case ENCLOSURE_SETTING_ENABLED:
		desc[3] = 0x20;
		break;
	default:
		/* SES doesn't do the SGPIO blink settings */
		return -EINVAL;
	}

	return ses_set_page2_descriptor(edev, ecomp, desc);
}

static void ses_get_status(struct enclosure_device *edev,
			   struct enclosure_component *ecomp)
{
	unsigned char *desc;

	desc = ses_get_page2_descriptor(edev, ecomp);
	if (desc)
		ecomp->status = (desc[0] & 0x0f);
}

static void ses_get_locate(struct enclosure_device *edev,
			   struct enclosure_component *ecomp)
{
	unsigned char *desc;

	desc = ses_get_page2_descriptor(edev, ecomp);
	if (desc)
		ecomp->locate = (desc[2] & 0x02) ? 1 : 0;
}

static int ses_set_locate(struct enclosure_device *edev,
			  struct enclosure_component *ecomp,
			  enum enclosure_component_setting val)
{
	unsigned char desc[4] = {0 };

	switch (val) {
	case ENCLOSURE_SETTING_DISABLED:
		/* zero is disabled */
		break;
	case ENCLOSURE_SETTING_ENABLED:
		desc[2] = 0x02;
		break;
	default:
		/* SES doesn't do the SGPIO blink settings */
		return -EINVAL;
	}
	return ses_set_page2_descriptor(edev, ecomp, desc);
}

static int ses_set_active(struct enclosure_device *edev,
			  struct enclosure_component *ecomp,
			  enum enclosure_component_setting val)
{
	unsigned char desc[4] = {0 };

	switch (val) {
	case ENCLOSURE_SETTING_DISABLED:
		/* zero is disabled */
		ecomp->active = 0;
		break;
	case ENCLOSURE_SETTING_ENABLED:
		desc[2] = 0x80;
		ecomp->active = 1;
		break;
	default:
		/* SES doesn't do the SGPIO blink settings */
		return -EINVAL;
	}
	return ses_set_page2_descriptor(edev, ecomp, desc);
}

static struct enclosure_component_callbacks ses_enclosure_callbacks = {
	.get_fault		= ses_get_fault,
	.set_fault		= ses_set_fault,
	.get_status		= ses_get_status,
	.get_locate		= ses_get_locate,
	.set_locate		= ses_set_locate,
	.set_active		= ses_set_active,
};

struct ses_host_edev {
	struct Scsi_Host *shost;
	struct enclosure_device *edev;
};

#if 0
int ses_match_host(struct enclosure_device *edev, void *data)
{
	struct ses_host_edev *sed = data;
	struct scsi_device *sdev;

	if (!scsi_is_sdev_device(edev->edev.parent))
		return 0;

	sdev = to_scsi_device(edev->edev.parent);

	if (sdev->host != sed->shost)
		return 0;

	sed->edev = edev;
	return 1;
}
#endif  /*  0  */

static void ses_process_descriptor(struct enclosure_component *ecomp,
				   unsigned char *desc)
{
	int eip = desc[0] & 0x10;
	int invalid = desc[0] & 0x80;
	enum scsi_protocol proto = desc[0] & 0x0f;
	u64 addr = 0;
	struct ses_component *scomp = ecomp->scratch;
	unsigned char *d;

	scomp->desc = desc;

	if (invalid)
		return;

	switch (proto) {
	case SCSI_PROTOCOL_SAS:
		if (eip)
			d = desc + 8;
		else
			d = desc + 4;
		/* only take the phy0 addr */
		addr = (u64)d[12] << 56 |
			(u64)d[13] << 48 |
			(u64)d[14] << 40 |
			(u64)d[15] << 32 |
			(u64)d[16] << 24 |
			(u64)d[17] << 16 |
			(u64)d[18] << 8 |
			(u64)d[19];
		break;
	default:
		/* FIXME: Need to add more protocols than just SAS */
		break;
	}
	scomp->addr = addr;
}

struct efd {
	u64 addr;
	struct device *dev;
};

static int ses_enclosure_find_by_addr(struct enclosure_device *edev,
				      void *data)
{
	struct efd *efd = data;
	int i;
	struct ses_component *scomp;

	if (!edev->component[0].scratch)
		return 0;

	for (i = 0; i < edev->components; i++) {
		scomp = edev->component[i].scratch;
		if (scomp->addr != efd->addr)
			continue;

		enclosure_add_device(edev, i, efd->dev);
		return 1;
	}
	return 0;
}

#define INIT_ALLOC_SIZE 32

static void ses_enclosure_data_process(struct enclosure_device *edev,
				       struct scsi_device *sdev,
				       int create)
{
	u32 result;
	unsigned char *buf = NULL, *type_ptr, *desc_ptr, *addl_desc_ptr = NULL;
	int i, j, page7_len, len, components;
	struct ses_device *ses_dev = edev->scratch;
	int types = ses_dev->page1_num_types;
	unsigned char *hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);

	if (!hdr_buf)
		goto simple_populate;

	/* re-read page 10 */
	if (ses_dev->page10)
		ses_recv_diag(sdev, 10, ses_dev->page10, ses_dev->page10_len);
	/* Page 7 for the descriptors is optional */
	result = ses_recv_diag(sdev, 7, hdr_buf, INIT_ALLOC_SIZE);
	if (result)
		goto simple_populate;

	page7_len = len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	/* add 1 for trailing '\0' we'll use */
	buf = kzalloc(len + 1, GFP_KERNEL);
	if (!buf)
		goto simple_populate;
	result = ses_recv_diag(sdev, 7, buf, len);
	if (result) {
 simple_populate:
		kfree(buf);
		buf = NULL;
		desc_ptr = NULL;
		len = 0;
		page7_len = 0;
	} else {
		desc_ptr = buf + 8;
		len = (desc_ptr[2] << 8) + desc_ptr[3];
		/* skip past overall descriptor */
		desc_ptr += len + 4;
	}
	if (ses_dev->page10)
		addl_desc_ptr = ses_dev->page10 + 8;
	type_ptr = ses_dev->page1_types;
	components = 0;
	for (i = 0; i < types; i++, type_ptr += 4) {
		for (j = 0; j < type_ptr[1]; j++) {
			char *name = NULL;
			struct enclosure_component *ecomp;

			if (desc_ptr) {
				if (desc_ptr >= buf + page7_len) {
					desc_ptr = NULL;
				} else {
					len = (desc_ptr[2] << 8) + desc_ptr[3];
					desc_ptr += 4;
					/* Add trailing zero - pushes into
					 * reserved space */
					desc_ptr[len] = '\0';
					name = desc_ptr;
				}
			}
			if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE ||
			    type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE) {

				if (create)
					ecomp =	enclosure_component_register(edev,
									     components++,
									     type_ptr[0],
									     name);
				else
					ecomp = &edev->component[components++];

				if (!IS_ERR(ecomp) && addl_desc_ptr)
					ses_process_descriptor(ecomp,
							       addl_desc_ptr);
			}
			if (desc_ptr)
				desc_ptr += len;

			if (addl_desc_ptr)
				addl_desc_ptr += addl_desc_ptr[1] + 2;

		}
	}
	kfree(buf);
	kfree(hdr_buf);
}

static void ses_match_to_enclosure(struct enclosure_device *edev,
				   struct scsi_device *sdev)
{
	unsigned char *buf;
	unsigned char *desc;
	unsigned int vpd_len;
	struct efd efd = {
		.addr = 0,
	};

	buf = kmalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
	if (!buf || scsi_get_vpd_page(sdev, 0x83, buf, INIT_ALLOC_SIZE))
		goto free;

	ses_enclosure_data_process(edev, to_scsi_device(edev->edev.parent), 0);

	vpd_len = ((buf[2] << 8) | buf[3]) + 4;
	kfree(buf);
	buf = kmalloc(vpd_len, GFP_KERNEL);
	if (!buf ||scsi_get_vpd_page(sdev, 0x83, buf, vpd_len))
		goto free;

	desc = buf + 4;
	while (desc < buf + vpd_len) {
		enum scsi_protocol proto = desc[0] >> 4;
		u8 code_set = desc[0] & 0x0f;
		u8 piv = desc[1] & 0x80;
		u8 assoc = (desc[1] & 0x30) >> 4;
		u8 type = desc[1] & 0x0f;
		u8 len = desc[3];

		if (piv && code_set == 1 && assoc == 1
		    && proto == SCSI_PROTOCOL_SAS && type == 3 && len == 8)
			efd.addr = (u64)desc[4] << 56 |
				(u64)desc[5] << 48 |
				(u64)desc[6] << 40 |
				(u64)desc[7] << 32 |
				(u64)desc[8] << 24 |
				(u64)desc[9] << 16 |
				(u64)desc[10] << 8 |
				(u64)desc[11];

		desc += len + 4;
	}
	if (!efd.addr)
		goto free;

	efd.dev = &sdev->sdev_gendev;

	enclosure_for_each_device(ses_enclosure_find_by_addr, &efd);
 free:
	kfree(buf);
}

static int ses_intf_add(struct device *cdev,
			struct class_interface *intf)
{
	struct scsi_device *sdev = to_scsi_device(cdev->parent);
	struct scsi_device *tmp_sdev;
	unsigned char *buf = NULL, *hdr_buf, *type_ptr;
	struct ses_device *ses_dev;
	u32 result;
	int i, types, len, components = 0;
	int err = -ENOMEM;
	int num_enclosures;
	struct enclosure_device *edev;
	struct ses_component *scomp = NULL;

	if (!scsi_device_enclosure(sdev)) {
		/* not an enclosure, but might be in one */
		struct enclosure_device *prev = NULL;

		while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
			ses_match_to_enclosure(edev, sdev);
			prev = edev;
		}
		return -ENODEV;
	}

	/* TYPE_ENCLOSURE prints a message in probe */
	if (sdev->type != TYPE_ENCLOSURE)
		sdev_printk(KERN_NOTICE, sdev, "Embedded Enclosure Device\n");

	ses_dev = kzalloc(sizeof(*ses_dev), GFP_KERNEL);
	hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
	if (!hdr_buf || !ses_dev)
		goto err_init_free;

	result = ses_recv_diag(sdev, 1, hdr_buf, INIT_ALLOC_SIZE);
	if (result)
		goto recv_failed;

	len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	buf = kzalloc(len, GFP_KERNEL);
	if (!buf)
		goto err_free;

	result = ses_recv_diag(sdev, 1, buf, len);
	if (result)
		goto recv_failed;

	types = 0;

	/* we always have one main enclosure and the rest are referred
	 * to as secondary subenclosures */
	num_enclosures = buf[1] + 1;

	/* begin at the enclosure descriptor */
	type_ptr = buf + 8;
	/* skip all the enclosure descriptors */
	for (i = 0; i < num_enclosures && type_ptr < buf + len; i++) {
		types += type_ptr[2];
		type_ptr += type_ptr[3] + 4;
	}

	ses_dev->page1_types = type_ptr;
	ses_dev->page1_num_types = types;

	for (i = 0; i < types && type_ptr < buf + len; i++, type_ptr += 4) {
		if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE ||
		    type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE)
			components += type_ptr[1];
	}
	ses_dev->page1 = buf;
	ses_dev->page1_len = len;
	buf = NULL;

	result = ses_recv_diag(sdev, 2, hdr_buf, INIT_ALLOC_SIZE);
	if (result)
		goto recv_failed;

	len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	buf = kzalloc(len, GFP_KERNEL);
	if (!buf)
		goto err_free;

	/* make sure getting page 2 actually works */
	result = ses_recv_diag(sdev, 2, buf, len);
	if (result)
		goto recv_failed;
	ses_dev->page2 = buf;
	ses_dev->page2_len = len;
	buf = NULL;

	/* The additional information page --- allows us
	 * to match up the devices */
	result = ses_recv_diag(sdev, 10, hdr_buf, INIT_ALLOC_SIZE);
	if (!result) {

		len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
		buf = kzalloc(len, GFP_KERNEL);
		if (!buf)
			goto err_free;

		result = ses_recv_diag(sdev, 10, buf, len);
		if (result)
			goto recv_failed;
		ses_dev->page10 = buf;
		ses_dev->page10_len = len;
		buf = NULL;
	}
	scomp = kzalloc(sizeof(struct ses_component) * components, GFP_KERNEL);
	if (!scomp)
		goto err_free;

	edev = enclosure_register(cdev->parent, dev_name(&sdev->sdev_gendev),
				  components, &ses_enclosure_callbacks);
	if (IS_ERR(edev)) {
		err = PTR_ERR(edev);
		goto err_free;
	}

	kfree(hdr_buf);

	edev->scratch = ses_dev;
	for (i = 0; i < components; i++)
		edev->component[i].scratch = scomp + i;

	ses_enclosure_data_process(edev, sdev, 1);

	/* see if there are any devices matching before
	 * we found the enclosure */
	shost_for_each_device(tmp_sdev, sdev->host) {
		if (tmp_sdev->lun != 0 || scsi_device_enclosure(tmp_sdev))
			continue;
		ses_match_to_enclosure(edev, tmp_sdev);
	}

	return 0;

 recv_failed:
	sdev_printk(KERN_ERR, sdev, "Failed to get diagnostic page 0x%x\n",
		    result);
	err = -ENODEV;
 err_free:
	kfree(buf);
	kfree(scomp);
	kfree(ses_dev->page10);
	kfree(ses_dev->page2);
	kfree(ses_dev->page1);
 err_init_free:
	kfree(ses_dev);
	kfree(hdr_buf);
	sdev_printk(KERN_ERR, sdev, "Failed to bind enclosure %d\n", err);
	return err;
}

static int ses_remove(struct device *dev)
{
	return 0;
}

static void ses_intf_remove_component(struct scsi_device *sdev)
{
	struct enclosure_device *edev, *prev = NULL;

	while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
		prev = edev;
		if (!enclosure_remove_device(edev, &sdev->sdev_gendev))
			break;
	}
	if (edev)
		put_device(&edev->edev);
}

static void ses_intf_remove_enclosure(struct scsi_device *sdev)
{
	struct enclosure_device *edev;
	struct ses_device *ses_dev;

	/*  exact match to this enclosure */
	edev = enclosure_find(&sdev->sdev_gendev, NULL);
	if (!edev)
		return;

	ses_dev = edev->scratch;
	edev->scratch = NULL;

	kfree(ses_dev->page10);
	kfree(ses_dev->page1);
	kfree(ses_dev->page2);
	kfree(ses_dev);

	kfree(edev->component[0].scratch);

	put_device(&edev->edev);
	enclosure_unregister(edev);
}

static void ses_intf_remove(struct device *cdev,
			    struct class_interface *intf)
{
	struct scsi_device *sdev = to_scsi_device(cdev->parent);

	if (!scsi_device_enclosure(sdev))
		ses_intf_remove_component(sdev);
	else
		ses_intf_remove_enclosure(sdev);
}

static struct class_interface ses_interface = {
	.add_dev	= ses_intf_add,
	.remove_dev	= ses_intf_remove,
};

static struct scsi_driver ses_template = {
	.owner			= THIS_MODULE,
	.gendrv = {
		.name		= "ses",
		.probe		= ses_probe,
		.remove		= ses_remove,
	},
};

static int __init ses_init(void)
{
	int err;

	err = scsi_register_interface(&ses_interface);
	if (err)
		return err;

	err = scsi_register_driver(&ses_template.gendrv);
	if (err)
		goto out_unreg;

	return 0;

 out_unreg:
	scsi_unregister_interface(&ses_interface);
	return err;
}

static void __exit ses_exit(void)
{
	scsi_unregister_driver(&ses_template.gendrv);
	scsi_unregister_interface(&ses_interface);
}

module_init(ses_init);
module_exit(ses_exit);

MODULE_ALIAS_SCSI_DEVICE(TYPE_ENCLOSURE);

MODULE_AUTHOR("James Bottomley");
MODULE_DESCRIPTION("SCSI Enclosure Services (ses) driver");
MODULE_LICENSE("GPL v2");
Commit	Line	Data
9927c688 JB	1	/*
	2	* SCSI Enclosure Services
	3	*
	4	* Copyright (C) 2008 James Bottomley <James.Bottomley@HansenPartnership.com>
	5	*
	6	**-----------------------------------------------------------------------------
	7	**
	8	** This program is free software; you can redistribute it and/or
	9	** modify it under the terms of the GNU General Public License
	10	** version 2 as published by the Free Software Foundation.
	11	**
	12	** This program is distributed in the hope that it will be useful,
	13	** but WITHOUT ANY WARRANTY; without even the implied warranty of
	14	** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	15	** GNU General Public License for more details.
	16	**
	17	** You should have received a copy of the GNU General Public License
	18	** along with this program; if not, write to the Free Software
	19	** Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
	20	**
	21	**-----------------------------------------------------------------------------
	22	*/
	23
5a0e3ad6	24	#include <linux/slab.h>
9927c688 JB	25	#include <linux/module.h>
	26	#include <linux/kernel.h>
	27	#include <linux/enclosure.h>
	28
	29	#include <scsi/scsi.h>
	30	#include <scsi/scsi_cmnd.h>
	31	#include <scsi/scsi_dbg.h>
	32	#include <scsi/scsi_device.h>
	33	#include <scsi/scsi_driver.h>
	34	#include <scsi/scsi_host.h>
	35
	36	struct ses_device {
691b4773	37	unsigned char *page1;
8c3adc79	38	unsigned char *page1_types;
691b4773 YL	39	unsigned char *page2;
691b4773 YL	40	unsigned char *page10;
9927c688	41	short page1_len;
8c3adc79	42	short page1_num_types;
9927c688 JB	43	short page2_len;
	44	short page10_len;
	45	};
	46
	47	struct ses_component {
	48	u64 addr;
	49	unsigned char *desc;
	50	};
	51
	52	static int ses_probe(struct device *dev)
	53	{
	54	struct scsi_device *sdev = to_scsi_device(dev);
	55	int err = -ENODEV;
	56
	57	if (sdev->type != TYPE_ENCLOSURE)
	58	goto out;
	59
	60	err = 0;
	61	sdev_printk(KERN_NOTICE, sdev, "Attached Enclosure device\n");
	62
	63	out:
	64	return err;
	65	}
	66
c95e62ce	67	#define SES_TIMEOUT (30 * HZ)
9927c688 JB	68	#define SES_RETRIES 3
	69
	70	static int ses_recv_diag(struct scsi_device *sdev, int page_code,
	71	void *buf, int bufflen)
	72	{
691b4773	73	unsigned char cmd[] = {
9927c688 JB	74	RECEIVE_DIAGNOSTIC,
	75	1, /* Set PCV bit */
	76	page_code,
	77	bufflen >> 8,
	78	bufflen & 0xff,
	79	0
	80	};
	81
	82	return scsi_execute_req(sdev, cmd, DMA_FROM_DEVICE, buf, bufflen,
f4f4e47e	83	NULL, SES_TIMEOUT, SES_RETRIES, NULL);
9927c688 JB	84	}
	85
	86	static int ses_send_diag(struct scsi_device *sdev, int page_code,
	87	void *buf, int bufflen)
	88	{
	89	u32 result;
	90
691b4773	91	unsigned char cmd[] = {
9927c688 JB	92	SEND_DIAGNOSTIC,
	93	0x10, /* Set PF bit */
	94	0,
	95	bufflen >> 8,
	96	bufflen & 0xff,
	97	0
	98	};
	99
	100	result = scsi_execute_req(sdev, cmd, DMA_TO_DEVICE, buf, bufflen,
f4f4e47e	101	NULL, SES_TIMEOUT, SES_RETRIES, NULL);
9927c688 JB	102	if (result)
	103	sdev_printk(KERN_ERR, sdev, "SEND DIAGNOSTIC result: %8x\n",
	104	result);
	105	return result;
	106	}
	107
	108	static int ses_set_page2_descriptor(struct enclosure_device *edev,
	109	struct enclosure_component *ecomp,
691b4773	110	unsigned char *desc)
9927c688 JB	111	{
9927c688 JB	112	int i, j, count = 0, descriptor = ecomp->number;
ee959b00	113	struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
9927c688	114	struct ses_device *ses_dev = edev->scratch;
8c3adc79	115	unsigned char *type_ptr = ses_dev->page1_types;
691b4773	116	unsigned char *desc_ptr = ses_dev->page2 + 8;
9927c688 JB	117
	118	/* Clear everything */
	119	memset(desc_ptr, 0, ses_dev->page2_len - 8);
8c3adc79	120	for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
9927c688 JB	121	for (j = 0; j < type_ptr[1]; j++) {
	122	desc_ptr += 4;
	123	if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
	124	type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
	125	continue;
	126	if (count++ == descriptor) {
	127	memcpy(desc_ptr, desc, 4);
	128	/* set select */
	129	desc_ptr[0] \|= 0x80;
	130	/* clear reserved, just in case */
	131	desc_ptr[0] &= 0xf0;
	132	}
	133	}
	134	}
	135
	136	return ses_send_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len);
	137	}
	138
691b4773	139	static unsigned char ses_get_page2_descriptor(struct enclosure_device edev,
9927c688 JB	140	struct enclosure_component *ecomp)
	141	{
	142	int i, j, count = 0, descriptor = ecomp->number;
ee959b00	143	struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
9927c688	144	struct ses_device *ses_dev = edev->scratch;
8c3adc79	145	unsigned char *type_ptr = ses_dev->page1_types;
691b4773	146	unsigned char *desc_ptr = ses_dev->page2 + 8;
9927c688 JB	147
	148	ses_recv_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len);
	149
8c3adc79	150	for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
9927c688 JB	151	for (j = 0; j < type_ptr[1]; j++) {
	152	desc_ptr += 4;
	153	if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
	154	type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
	155	continue;
	156	if (count++ == descriptor)
	157	return desc_ptr;
	158	}
	159	}
	160	return NULL;
	161	}
	162
2a350cab DG	163	/* For device slot and array device slot elements, byte 3 bit 6
	164	* is "fault sensed" while byte 3 bit 5 is "fault reqstd". As this
	165	* code stands these bits are shifted 4 positions right so in
	166	* sysfs they will appear as bits 2 and 1 respectively. Strange. */
9927c688 JB	167	static void ses_get_fault(struct enclosure_device *edev,
	168	struct enclosure_component *ecomp)
	169	{
691b4773	170	unsigned char *desc;
9927c688 JB	171
9927c688 JB	172	desc = ses_get_page2_descriptor(edev, ecomp);
691b4773 YL	173	if (desc)
691b4773 YL	174	ecomp->fault = (desc[3] & 0x60) >> 4;
9927c688 JB	175	}
	176
	177	static int ses_set_fault(struct enclosure_device *edev,
	178	struct enclosure_component *ecomp,
	179	enum enclosure_component_setting val)
	180	{
691b4773	181	unsigned char desc[4] = {0 };
9927c688 JB	182
	183	switch (val) {
	184	case ENCLOSURE_SETTING_DISABLED:
	185	/* zero is disabled */
	186	break;
	187	case ENCLOSURE_SETTING_ENABLED:
2a350cab	188	desc[3] = 0x20;
9927c688 JB	189	break;
	190	default:
	191	/* SES doesn't do the SGPIO blink settings */
	192	return -EINVAL;
	193	}
	194
	195	return ses_set_page2_descriptor(edev, ecomp, desc);
	196	}
	197
	198	static void ses_get_status(struct enclosure_device *edev,
	199	struct enclosure_component *ecomp)
	200	{
691b4773	201	unsigned char *desc;
9927c688 JB	202
9927c688 JB	203	desc = ses_get_page2_descriptor(edev, ecomp);
691b4773 YL	204	if (desc)
691b4773 YL	205	ecomp->status = (desc[0] & 0x0f);
9927c688 JB	206	}
	207
	208	static void ses_get_locate(struct enclosure_device *edev,
	209	struct enclosure_component *ecomp)
	210	{
691b4773	211	unsigned char *desc;
9927c688 JB	212
9927c688 JB	213	desc = ses_get_page2_descriptor(edev, ecomp);
691b4773 YL	214	if (desc)
691b4773 YL	215	ecomp->locate = (desc[2] & 0x02) ? 1 : 0;
9927c688 JB	216	}
	217
	218	static int ses_set_locate(struct enclosure_device *edev,
	219	struct enclosure_component *ecomp,
	220	enum enclosure_component_setting val)
	221	{
691b4773	222	unsigned char desc[4] = {0 };
9927c688 JB	223
	224	switch (val) {
	225	case ENCLOSURE_SETTING_DISABLED:
	226	/* zero is disabled */
	227	break;
	228	case ENCLOSURE_SETTING_ENABLED:
	229	desc[2] = 0x02;
	230	break;
	231	default:
	232	/* SES doesn't do the SGPIO blink settings */
	233	return -EINVAL;
	234	}
	235	return ses_set_page2_descriptor(edev, ecomp, desc);
	236	}
	237
	238	static int ses_set_active(struct enclosure_device *edev,
	239	struct enclosure_component *ecomp,
	240	enum enclosure_component_setting val)
	241	{
691b4773	242	unsigned char desc[4] = {0 };
9927c688 JB	243
	244	switch (val) {
	245	case ENCLOSURE_SETTING_DISABLED:
	246	/* zero is disabled */
	247	ecomp->active = 0;
	248	break;
	249	case ENCLOSURE_SETTING_ENABLED:
	250	desc[2] = 0x80;
	251	ecomp->active = 1;
	252	break;
	253	default:
	254	/* SES doesn't do the SGPIO blink settings */
	255	return -EINVAL;
	256	}
	257	return ses_set_page2_descriptor(edev, ecomp, desc);
	258	}
	259
	260	static struct enclosure_component_callbacks ses_enclosure_callbacks = {
	261	.get_fault = ses_get_fault,
	262	.set_fault = ses_set_fault,
	263	.get_status = ses_get_status,
	264	.get_locate = ses_get_locate,
	265	.set_locate = ses_set_locate,
	266	.set_active = ses_set_active,
	267	};
	268
	269	struct ses_host_edev {
	270	struct Scsi_Host *shost;
	271	struct enclosure_device *edev;
	272	};
	273
e0aae1a5	274	#if 0
9927c688 JB	275	int ses_match_host(struct enclosure_device edev, void data)
	276	{
	277	struct ses_host_edev *sed = data;
	278	struct scsi_device *sdev;
	279
ee959b00	280	if (!scsi_is_sdev_device(edev->edev.parent))
9927c688 JB	281	return 0;
9927c688 JB	282
ee959b00	283	sdev = to_scsi_device(edev->edev.parent);
9927c688 JB	284
	285	if (sdev->host != sed->shost)
	286	return 0;
	287
	288	sed->edev = edev;
	289	return 1;
	290	}
e0aae1a5	291	#endif /* 0 */
9927c688 JB	292
	293	static void ses_process_descriptor(struct enclosure_component *ecomp,
	294	unsigned char *desc)
	295	{
	296	int eip = desc[0] & 0x10;
	297	int invalid = desc[0] & 0x80;
	298	enum scsi_protocol proto = desc[0] & 0x0f;
	299	u64 addr = 0;
	300	struct ses_component *scomp = ecomp->scratch;
	301	unsigned char *d;
	302
	303	scomp->desc = desc;
	304
	305	if (invalid)
	306	return;
	307
	308	switch (proto) {
	309	case SCSI_PROTOCOL_SAS:
	310	if (eip)
	311	d = desc + 8;
	312	else
	313	d = desc + 4;
	314	/* only take the phy0 addr */
	315	addr = (u64)d[12] << 56 \|
	316	(u64)d[13] << 48 \|
	317	(u64)d[14] << 40 \|
	318	(u64)d[15] << 32 \|
	319	(u64)d[16] << 24 \|
	320	(u64)d[17] << 16 \|
	321	(u64)d[18] << 8 \|
	322	(u64)d[19];
	323	break;
	324	default:
	325	/* FIXME: Need to add more protocols than just SAS */
	326	break;
	327	}
	328	scomp->addr = addr;
	329	}
	330
	331	struct efd {
	332	u64 addr;
	333	struct device *dev;
	334	};
	335
	336	static int ses_enclosure_find_by_addr(struct enclosure_device *edev,
	337	void *data)
	338	{
	339	struct efd *efd = data;
	340	int i;
	341	struct ses_component *scomp;
	342
	343	if (!edev->component[0].scratch)
	344	return 0;
	345
	346	for (i = 0; i < edev->components; i++) {
	347	scomp = edev->component[i].scratch;
	348	if (scomp->addr != efd->addr)
	349	continue;
	350
	351	enclosure_add_device(edev, i, efd->dev);
	352	return 1;
	353	}
	354	return 0;
	355	}
356
21fab1d0 JB	357	#define INIT_ALLOC_SIZE 32
	358
	359	static void ses_enclosure_data_process(struct enclosure_device *edev,
	360	struct scsi_device *sdev,
	361	int create)
	362	{
	363	u32 result;
	364	unsigned char buf = NULL, type_ptr, desc_ptr, addl_desc_ptr = NULL;
	365	int i, j, page7_len, len, components;
	366	struct ses_device *ses_dev = edev->scratch;
8c3adc79	367	int types = ses_dev->page1_num_types;
21fab1d0 JB	368	unsigned char *hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
	369
	370	if (!hdr_buf)
	371	goto simple_populate;
	372
	373	/* re-read page 10 */
	374	if (ses_dev->page10)
	375	ses_recv_diag(sdev, 10, ses_dev->page10, ses_dev->page10_len);
	376	/* Page 7 for the descriptors is optional */
	377	result = ses_recv_diag(sdev, 7, hdr_buf, INIT_ALLOC_SIZE);
	378	if (result)
	379	goto simple_populate;
	380
	381	page7_len = len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	382	/* add 1 for trailing '\0' we'll use */
	383	buf = kzalloc(len + 1, GFP_KERNEL);
	384	if (!buf)
	385	goto simple_populate;
	386	result = ses_recv_diag(sdev, 7, buf, len);
	387	if (result) {
	388	simple_populate:
	389	kfree(buf);
	390	buf = NULL;
	391	desc_ptr = NULL;
	392	len = 0;
	393	page7_len = 0;
	394	} else {
	395	desc_ptr = buf + 8;
	396	len = (desc_ptr[2] << 8) + desc_ptr[3];
	397	/* skip past overall descriptor */
	398	desc_ptr += len + 4;
21fab1d0	399	}
877a5597 JH	400	if (ses_dev->page10)
877a5597 JH	401	addl_desc_ptr = ses_dev->page10 + 8;
8c3adc79	402	type_ptr = ses_dev->page1_types;
21fab1d0 JB	403	components = 0;
	404	for (i = 0; i < types; i++, type_ptr += 4) {
	405	for (j = 0; j < type_ptr[1]; j++) {
	406	char *name = NULL;
	407	struct enclosure_component *ecomp;
	408
	409	if (desc_ptr) {
	410	if (desc_ptr >= buf + page7_len) {
	411	desc_ptr = NULL;
	412	} else {
	413	len = (desc_ptr[2] << 8) + desc_ptr[3];
	414	desc_ptr += 4;
	415	/* Add trailing zero - pushes into
	416	* reserved space */
	417	desc_ptr[len] = '\0';
	418	name = desc_ptr;
	419	}
	420	}
	421	if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE \|\|
	422	type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE) {
	423
	424	if (create)
	425	ecomp = enclosure_component_register(edev,
	426	components++,
	427	type_ptr[0],
	428	name);
	429	else
	430	ecomp = &edev->component[components++];
	431
	432	if (!IS_ERR(ecomp) && addl_desc_ptr)
	433	ses_process_descriptor(ecomp,
	434	addl_desc_ptr);
	435	}
	436	if (desc_ptr)
	437	desc_ptr += len;
	438
	439	if (addl_desc_ptr)
	440	addl_desc_ptr += addl_desc_ptr[1] + 2;
	441
	442	}
	443	}
	444	kfree(buf);
	445	kfree(hdr_buf);
	446	}
	447
9927c688 JB	448	static void ses_match_to_enclosure(struct enclosure_device *edev,
	449	struct scsi_device *sdev)
	450	{
40c3460f	451	unsigned char *buf;
9927c688	452	unsigned char *desc;
40c3460f	453	unsigned int vpd_len;
9927c688 JB	454	struct efd efd = {
	455	.addr = 0,
	456	};
9927c688	457
e3deec09 JB	458	buf = kmalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
	459	if (!buf \|\| scsi_get_vpd_page(sdev, 0x83, buf, INIT_ALLOC_SIZE))
	460	goto free;
9927c688	461
21fab1d0 JB	462	ses_enclosure_data_process(edev, to_scsi_device(edev->edev.parent), 0);
21fab1d0 JB	463
40c3460f	464	vpd_len = ((buf[2] << 8) \| buf[3]) + 4;
e3deec09 JB	465	kfree(buf);
	466	buf = kmalloc(vpd_len, GFP_KERNEL);
	467	if (!buf \|\|scsi_get_vpd_page(sdev, 0x83, buf, vpd_len))
	468	goto free;
671a99c8	469
9927c688	470	desc = buf + 4;
671a99c8	471	while (desc < buf + vpd_len) {
9927c688 JB	472	enum scsi_protocol proto = desc[0] >> 4;
	473	u8 code_set = desc[0] & 0x0f;
	474	u8 piv = desc[1] & 0x80;
	475	u8 assoc = (desc[1] & 0x30) >> 4;
	476	u8 type = desc[1] & 0x0f;
	477	u8 len = desc[3];
	478
b3f1f9aa	479	if (piv && code_set == 1 && assoc == 1
9927c688 JB	480	&& proto == SCSI_PROTOCOL_SAS && type == 3 && len == 8)
	481	efd.addr = (u64)desc[4] << 56 \|
	482	(u64)desc[5] << 48 \|
	483	(u64)desc[6] << 40 \|
	484	(u64)desc[7] << 32 \|
	485	(u64)desc[8] << 24 \|
	486	(u64)desc[9] << 16 \|
	487	(u64)desc[10] << 8 \|
	488	(u64)desc[11];
	489
	490	desc += len + 4;
	491	}
	492	if (!efd.addr)
	493	goto free;
	494
	495	efd.dev = &sdev->sdev_gendev;
	496
	497	enclosure_for_each_device(ses_enclosure_find_by_addr, &efd);
	498	free:
	499	kfree(buf);
	500	}
	501
ee959b00	502	static int ses_intf_add(struct device *cdev,
9927c688 JB	503	struct class_interface *intf)
9927c688 JB	504	{
ee959b00	505	struct scsi_device *sdev = to_scsi_device(cdev->parent);
9927c688	506	struct scsi_device *tmp_sdev;
21fab1d0	507	unsigned char buf = NULL, hdr_buf, *type_ptr;
9927c688 JB	508	struct ses_device *ses_dev;
9927c688 JB	509	u32 result;
21fab1d0	510	int i, types, len, components = 0;
9927c688	511	int err = -ENOMEM;
8c3adc79	512	int num_enclosures;
9927c688	513	struct enclosure_device *edev;
7c46c20a	514	struct ses_component *scomp = NULL;
9927c688 JB	515
	516	if (!scsi_device_enclosure(sdev)) {
	517	/* not an enclosure, but might be in one */
163f52b6 JB	518	struct enclosure_device *prev = NULL;
	519
	520	while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
9927c688	521	ses_match_to_enclosure(edev, sdev);
163f52b6	522	prev = edev;
9927c688 JB	523	}
	524	return -ENODEV;
	525	}
	526
	527	/* TYPE_ENCLOSURE prints a message in probe */
	528	if (sdev->type != TYPE_ENCLOSURE)
	529	sdev_printk(KERN_NOTICE, sdev, "Embedded Enclosure Device\n");
	530
	531	ses_dev = kzalloc(sizeof(*ses_dev), GFP_KERNEL);
	532	hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
	533	if (!hdr_buf \|\| !ses_dev)
	534	goto err_init_free;
	535
	536	result = ses_recv_diag(sdev, 1, hdr_buf, INIT_ALLOC_SIZE);
	537	if (result)
	538	goto recv_failed;
	539
9927c688 JB	540	len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	541	buf = kzalloc(len, GFP_KERNEL);
	542	if (!buf)
	543	goto err_free;
	544
9927c688 JB	545	result = ses_recv_diag(sdev, 1, buf, len);
	546	if (result)
	547	goto recv_failed;
	548
8c3adc79	549	types = 0;
9927c688	550
8c3adc79 JB	551	/* we always have one main enclosure and the rest are referred
	552	* to as secondary subenclosures */
	553	num_enclosures = buf[1] + 1;
9927c688	554
8c3adc79 JB	555	/* begin at the enclosure descriptor */
	556	type_ptr = buf + 8;
	557	/* skip all the enclosure descriptors */
	558	for (i = 0; i < num_enclosures && type_ptr < buf + len; i++) {
	559	types += type_ptr[2];
	560	type_ptr += type_ptr[3] + 4;
	561	}
	562
	563	ses_dev->page1_types = type_ptr;
	564	ses_dev->page1_num_types = types;
	565
	566	for (i = 0; i < types && type_ptr < buf + len; i++, type_ptr += 4) {
9927c688 JB	567	if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE \|\|
	568	type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE)
	569	components += type_ptr[1];
	570	}
7c46c20a YL	571	ses_dev->page1 = buf;
	572	ses_dev->page1_len = len;
	573	buf = NULL;
9927c688 JB	574
	575	result = ses_recv_diag(sdev, 2, hdr_buf, INIT_ALLOC_SIZE);
	576	if (result)
	577	goto recv_failed;
	578
	579	len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	580	buf = kzalloc(len, GFP_KERNEL);
	581	if (!buf)
	582	goto err_free;
	583
	584	/* make sure getting page 2 actually works */
	585	result = ses_recv_diag(sdev, 2, buf, len);
	586	if (result)
	587	goto recv_failed;
	588	ses_dev->page2 = buf;
	589	ses_dev->page2_len = len;
7c46c20a	590	buf = NULL;
9927c688 JB	591
	592	/* The additional information page --- allows us
	593	* to match up the devices */
	594	result = ses_recv_diag(sdev, 10, hdr_buf, INIT_ALLOC_SIZE);
691b4773 YL	595	if (!result) {
	596
	597	len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
	598	buf = kzalloc(len, GFP_KERNEL);
	599	if (!buf)
	600	goto err_free;
	601
	602	result = ses_recv_diag(sdev, 10, buf, len);
	603	if (result)
	604	goto recv_failed;
	605	ses_dev->page10 = buf;
	606	ses_dev->page10_len = len;
	607	buf = NULL;
	608	}
7c46c20a	609	scomp = kzalloc(sizeof(struct ses_component) * components, GFP_KERNEL);
9927c688	610	if (!scomp)
7c46c20a	611	goto err_free;
9927c688	612
71610f55	613	edev = enclosure_register(cdev->parent, dev_name(&sdev->sdev_gendev),
9927c688 JB	614	components, &ses_enclosure_callbacks);
	615	if (IS_ERR(edev)) {
	616	err = PTR_ERR(edev);
	617	goto err_free;
	618	}
	619
9b3a6549 JL	620	kfree(hdr_buf);
9b3a6549 JL	621
9927c688 JB	622	edev->scratch = ses_dev;
9927c688 JB	623	for (i = 0; i < components; i++)
7c46c20a	624	edev->component[i].scratch = scomp + i;
9927c688	625
21fab1d0	626	ses_enclosure_data_process(edev, sdev, 1);
9927c688 JB	627
	628	/* see if there are any devices matching before
	629	* we found the enclosure */
	630	shost_for_each_device(tmp_sdev, sdev->host) {
	631	if (tmp_sdev->lun != 0 \|\| scsi_device_enclosure(tmp_sdev))
	632	continue;
	633	ses_match_to_enclosure(edev, tmp_sdev);
	634	}
	635
	636	return 0;
	637
	638	recv_failed:
	639	sdev_printk(KERN_ERR, sdev, "Failed to get diagnostic page 0x%x\n",
	640	result);
	641	err = -ENODEV;
	642	err_free:
	643	kfree(buf);
7c46c20a	644	kfree(scomp);
9927c688 JB	645	kfree(ses_dev->page10);
	646	kfree(ses_dev->page2);
	647	kfree(ses_dev->page1);
	648	err_init_free:
	649	kfree(ses_dev);
	650	kfree(hdr_buf);
	651	sdev_printk(KERN_ERR, sdev, "Failed to bind enclosure %d\n", err);
	652	return err;
	653	}
	654
	655	static int ses_remove(struct device *dev)
	656	{
	657	return 0;
	658	}
	659
43d8eb9c JB	660	static void ses_intf_remove_component(struct scsi_device *sdev)
	661	{
	662	struct enclosure_device edev, prev = NULL;
	663
	664	while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
	665	prev = edev;
	666	if (!enclosure_remove_device(edev, &sdev->sdev_gendev))
	667	break;
	668	}
	669	if (edev)
	670	put_device(&edev->edev);
	671	}
	672
	673	static void ses_intf_remove_enclosure(struct scsi_device *sdev)
9927c688	674	{
9927c688 JB	675	struct enclosure_device *edev;
	676	struct ses_device *ses_dev;
	677
163f52b6	678	/* exact match to this enclosure */
43d8eb9c	679	edev = enclosure_find(&sdev->sdev_gendev, NULL);
9927c688 JB	680	if (!edev)
	681	return;
	682
	683	ses_dev = edev->scratch;
	684	edev->scratch = NULL;
	685
7c46c20a	686	kfree(ses_dev->page10);
9927c688 JB	687	kfree(ses_dev->page1);
	688	kfree(ses_dev->page2);
	689	kfree(ses_dev);
	690
	691	kfree(edev->component[0].scratch);
	692
ee959b00	693	put_device(&edev->edev);
9927c688 JB	694	enclosure_unregister(edev);
	695	}
	696
43d8eb9c JB	697	static void ses_intf_remove(struct device *cdev,
	698	struct class_interface *intf)
	699	{
	700	struct scsi_device *sdev = to_scsi_device(cdev->parent);
	701
	702	if (!scsi_device_enclosure(sdev))
	703	ses_intf_remove_component(sdev);
	704	else
	705	ses_intf_remove_enclosure(sdev);
	706	}
	707
9927c688	708	static struct class_interface ses_interface = {
ee959b00 TJ	709	.add_dev = ses_intf_add,
ee959b00 TJ	710	.remove_dev = ses_intf_remove,
9927c688 JB	711	};
	712
	713	static struct scsi_driver ses_template = {
	714	.owner = THIS_MODULE,
	715	.gendrv = {
	716	.name = "ses",
	717	.probe = ses_probe,
	718	.remove = ses_remove,
	719	},
	720	};
	721
	722	static int __init ses_init(void)
	723	{
	724	int err;
	725
	726	err = scsi_register_interface(&ses_interface);
	727	if (err)
	728	return err;
	729
	730	err = scsi_register_driver(&ses_template.gendrv);
	731	if (err)
	732	goto out_unreg;
	733
	734	return 0;
	735
	736	out_unreg:
	737	scsi_unregister_interface(&ses_interface);
	738	return err;
	739	}
	740
	741	static void __exit ses_exit(void)
	742	{
	743	scsi_unregister_driver(&ses_template.gendrv);
	744	scsi_unregister_interface(&ses_interface);
	745	}
	746
	747	module_init(ses_init);
	748	module_exit(ses_exit);
	749
	750	MODULE_ALIAS_SCSI_DEVICE(TYPE_ENCLOSURE);
	751
	752	MODULE_AUTHOR("James Bottomley");
	753	MODULE_DESCRIPTION("SCSI Enclosure Services (ses) driver");
	754	MODULE_LICENSE("GPL v2");