[mirror_ubuntu-eoan-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/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 *page2;
	unsigned char *page10;
	short page1_len;
	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 + 12 + ses_dev->page1[11];
	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[10]; 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 + 12 + ses_dev->page1[11];
	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[10]; 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;
}

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[2] = 0x02;
		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;
};

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;
}

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;
}

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 = scsi_get_vpd_page(sdev, 0x83);
	if (!buf)
		return;

	vpd_len = ((buf[2] << 8) | buf[3]) + 4;

	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);
}

#define INIT_ALLOC_SIZE 32

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, *desc_ptr = NULL,
		*addl_desc_ptr = NULL;
	struct ses_device *ses_dev;
	u32 result;
	int i, j, types, len, page7_len = 0, components = 0;
	int err = -ENOMEM;
	struct enclosure_device *edev;
	struct ses_component *scomp = NULL;

	if (!scsi_device_enclosure(sdev)) {
		/* not an enclosure, but might be in one */
		edev = enclosure_find(&sdev->host->shost_gendev);
		if (edev) {
			ses_match_to_enclosure(edev, sdev);
			put_device(&edev->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;

	if (hdr_buf[1] != 0) {
		/* FIXME: need subenclosure support; I've just never
		 * seen a device with subenclosures and it makes the
		 * traversal routines more complex */
		sdev_printk(KERN_ERR, sdev,
			"FIXME driver has no support for subenclosures (%d)\n",
			hdr_buf[1]);
		goto err_free;
	}

	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 = buf[10];

	type_ptr = buf + 12 + buf[11];

	for (i = 0; i < types; 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;
	}

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

	/* 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;
		addl_desc_ptr = NULL;
	} 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 + 12 + ses_dev->page1[11];
	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) {

				ecomp =	enclosure_component_register(edev,
							     components++,
							     type_ptr[0],
							     name);

				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);

	/* 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(struct device *cdev,
			    struct class_interface *intf)
{
	struct scsi_device *sdev = to_scsi_device(cdev->parent);
	struct enclosure_device *edev;
	struct ses_device *ses_dev;

	if (!scsi_device_enclosure(sdev))
		return;

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