scsi: ses: Have midlayer retry scsi_execute_cmd() errors
[linux-block.git] / drivers / scsi / ses.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * SCSI Enclosure Services
4  *
5  * Copyright (C) 2008 James Bottomley <James.Bottomley@HansenPartnership.com>
6  */
7
8 #include <linux/slab.h>
9 #include <linux/module.h>
10 #include <linux/kernel.h>
11 #include <linux/enclosure.h>
12 #include <asm/unaligned.h>
13
14 #include <scsi/scsi.h>
15 #include <scsi/scsi_cmnd.h>
16 #include <scsi/scsi_dbg.h>
17 #include <scsi/scsi_device.h>
18 #include <scsi/scsi_driver.h>
19 #include <scsi/scsi_host.h>
20
21 #include <scsi/scsi_transport_sas.h>
22
23 struct ses_device {
24         unsigned char *page1;
25         unsigned char *page1_types;
26         unsigned char *page2;
27         unsigned char *page10;
28         short page1_len;
29         short page1_num_types;
30         short page2_len;
31         short page10_len;
32 };
33
34 struct ses_component {
35         u64 addr;
36 };
37
38 static bool ses_page2_supported(struct enclosure_device *edev)
39 {
40         struct ses_device *ses_dev = edev->scratch;
41
42         return (ses_dev->page2 != NULL);
43 }
44
45 static int ses_probe(struct device *dev)
46 {
47         struct scsi_device *sdev = to_scsi_device(dev);
48         int err = -ENODEV;
49
50         if (sdev->type != TYPE_ENCLOSURE)
51                 goto out;
52
53         err = 0;
54         sdev_printk(KERN_NOTICE, sdev, "Attached Enclosure device\n");
55
56  out:
57         return err;
58 }
59
60 #define SES_TIMEOUT (30 * HZ)
61 #define SES_RETRIES 3
62
63 static void init_device_slot_control(unsigned char *dest_desc,
64                                      struct enclosure_component *ecomp,
65                                      unsigned char *status)
66 {
67         memcpy(dest_desc, status, 4);
68         dest_desc[0] = 0;
69         /* only clear byte 1 for ENCLOSURE_COMPONENT_DEVICE */
70         if (ecomp->type == ENCLOSURE_COMPONENT_DEVICE)
71                 dest_desc[1] = 0;
72         dest_desc[2] &= 0xde;
73         dest_desc[3] &= 0x3c;
74 }
75
76
77 static int ses_recv_diag(struct scsi_device *sdev, int page_code,
78                          void *buf, int bufflen)
79 {
80         int ret;
81         unsigned char cmd[] = {
82                 RECEIVE_DIAGNOSTIC,
83                 1,              /* Set PCV bit */
84                 page_code,
85                 bufflen >> 8,
86                 bufflen & 0xff,
87                 0
88         };
89         unsigned char recv_page_code;
90         struct scsi_failure failure_defs[] = {
91                 {
92                         .sense = UNIT_ATTENTION,
93                         .asc = 0x29,
94                         .ascq = SCMD_FAILURE_ASCQ_ANY,
95                         .allowed = SES_RETRIES,
96                         .result = SAM_STAT_CHECK_CONDITION,
97                 },
98                 {
99                         .sense = NOT_READY,
100                         .asc = SCMD_FAILURE_ASC_ANY,
101                         .ascq = SCMD_FAILURE_ASCQ_ANY,
102                         .allowed = SES_RETRIES,
103                         .result = SAM_STAT_CHECK_CONDITION,
104                 },
105                 {}
106         };
107         struct scsi_failures failures = {
108                 .failure_definitions = failure_defs,
109         };
110         const struct scsi_exec_args exec_args = {
111                 .failures = &failures,
112         };
113
114         ret = scsi_execute_cmd(sdev, cmd, REQ_OP_DRV_IN, buf, bufflen,
115                                SES_TIMEOUT, 1, &exec_args);
116         if (unlikely(ret))
117                 return ret;
118
119         recv_page_code = ((unsigned char *)buf)[0];
120
121         if (likely(recv_page_code == page_code))
122                 return ret;
123
124         /* successful diagnostic but wrong page code.  This happens to some
125          * USB devices, just print a message and pretend there was an error */
126
127         sdev_printk(KERN_ERR, sdev,
128                     "Wrong diagnostic page; asked for %d got %u\n",
129                     page_code, recv_page_code);
130
131         return -EINVAL;
132 }
133
134 static int ses_send_diag(struct scsi_device *sdev, int page_code,
135                          void *buf, int bufflen)
136 {
137         int result;
138
139         unsigned char cmd[] = {
140                 SEND_DIAGNOSTIC,
141                 0x10,           /* Set PF bit */
142                 0,
143                 bufflen >> 8,
144                 bufflen & 0xff,
145                 0
146         };
147         struct scsi_failure failure_defs[] = {
148                 {
149                         .sense = UNIT_ATTENTION,
150                         .asc = 0x29,
151                         .ascq = SCMD_FAILURE_ASCQ_ANY,
152                         .allowed = SES_RETRIES,
153                         .result = SAM_STAT_CHECK_CONDITION,
154                 },
155                 {
156                         .sense = NOT_READY,
157                         .asc = SCMD_FAILURE_ASC_ANY,
158                         .ascq = SCMD_FAILURE_ASCQ_ANY,
159                         .allowed = SES_RETRIES,
160                         .result = SAM_STAT_CHECK_CONDITION,
161                 },
162                 {}
163         };
164         struct scsi_failures failures = {
165                 .failure_definitions = failure_defs,
166         };
167         const struct scsi_exec_args exec_args = {
168                 .failures = &failures,
169         };
170
171         result = scsi_execute_cmd(sdev, cmd, REQ_OP_DRV_OUT, buf, bufflen,
172                                   SES_TIMEOUT, 1, &exec_args);
173         if (result)
174                 sdev_printk(KERN_ERR, sdev, "SEND DIAGNOSTIC result: %8x\n",
175                             result);
176         return result;
177 }
178
179 static int ses_set_page2_descriptor(struct enclosure_device *edev,
180                                       struct enclosure_component *ecomp,
181                                       unsigned char *desc)
182 {
183         int i, j, count = 0, descriptor = ecomp->number;
184         struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
185         struct ses_device *ses_dev = edev->scratch;
186         unsigned char *type_ptr = ses_dev->page1_types;
187         unsigned char *desc_ptr = ses_dev->page2 + 8;
188
189         /* Clear everything */
190         memset(desc_ptr, 0, ses_dev->page2_len - 8);
191         for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
192                 for (j = 0; j < type_ptr[1]; j++) {
193                         desc_ptr += 4;
194                         if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
195                             type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
196                                 continue;
197                         if (count++ == descriptor) {
198                                 memcpy(desc_ptr, desc, 4);
199                                 /* set select */
200                                 desc_ptr[0] |= 0x80;
201                                 /* clear reserved, just in case */
202                                 desc_ptr[0] &= 0xf0;
203                         }
204                 }
205         }
206
207         return ses_send_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len);
208 }
209
210 static unsigned char *ses_get_page2_descriptor(struct enclosure_device *edev,
211                                       struct enclosure_component *ecomp)
212 {
213         int i, j, count = 0, descriptor = ecomp->number;
214         struct scsi_device *sdev = to_scsi_device(edev->edev.parent);
215         struct ses_device *ses_dev = edev->scratch;
216         unsigned char *type_ptr = ses_dev->page1_types;
217         unsigned char *desc_ptr = ses_dev->page2 + 8;
218
219         if (ses_recv_diag(sdev, 2, ses_dev->page2, ses_dev->page2_len) < 0)
220                 return NULL;
221
222         for (i = 0; i < ses_dev->page1_num_types; i++, type_ptr += 4) {
223                 for (j = 0; j < type_ptr[1]; j++) {
224                         desc_ptr += 4;
225                         if (type_ptr[0] != ENCLOSURE_COMPONENT_DEVICE &&
226                             type_ptr[0] != ENCLOSURE_COMPONENT_ARRAY_DEVICE)
227                                 continue;
228                         if (count++ == descriptor)
229                                 return desc_ptr;
230                 }
231         }
232         return NULL;
233 }
234
235 /* For device slot and array device slot elements, byte 3 bit 6
236  * is "fault sensed" while byte 3 bit 5 is "fault reqstd". As this
237  * code stands these bits are shifted 4 positions right so in
238  * sysfs they will appear as bits 2 and 1 respectively. Strange. */
239 static void ses_get_fault(struct enclosure_device *edev,
240                           struct enclosure_component *ecomp)
241 {
242         unsigned char *desc;
243
244         if (!ses_page2_supported(edev)) {
245                 ecomp->fault = 0;
246                 return;
247         }
248         desc = ses_get_page2_descriptor(edev, ecomp);
249         if (desc)
250                 ecomp->fault = (desc[3] & 0x60) >> 4;
251 }
252
253 static int ses_set_fault(struct enclosure_device *edev,
254                           struct enclosure_component *ecomp,
255                          enum enclosure_component_setting val)
256 {
257         unsigned char desc[4];
258         unsigned char *desc_ptr;
259
260         if (!ses_page2_supported(edev))
261                 return -EINVAL;
262
263         desc_ptr = ses_get_page2_descriptor(edev, ecomp);
264
265         if (!desc_ptr)
266                 return -EIO;
267
268         init_device_slot_control(desc, ecomp, desc_ptr);
269
270         switch (val) {
271         case ENCLOSURE_SETTING_DISABLED:
272                 desc[3] &= 0xdf;
273                 break;
274         case ENCLOSURE_SETTING_ENABLED:
275                 desc[3] |= 0x20;
276                 break;
277         default:
278                 /* SES doesn't do the SGPIO blink settings */
279                 return -EINVAL;
280         }
281
282         return ses_set_page2_descriptor(edev, ecomp, desc);
283 }
284
285 static void ses_get_status(struct enclosure_device *edev,
286                            struct enclosure_component *ecomp)
287 {
288         unsigned char *desc;
289
290         if (!ses_page2_supported(edev)) {
291                 ecomp->status = 0;
292                 return;
293         }
294         desc = ses_get_page2_descriptor(edev, ecomp);
295         if (desc)
296                 ecomp->status = (desc[0] & 0x0f);
297 }
298
299 static void ses_get_locate(struct enclosure_device *edev,
300                            struct enclosure_component *ecomp)
301 {
302         unsigned char *desc;
303
304         if (!ses_page2_supported(edev)) {
305                 ecomp->locate = 0;
306                 return;
307         }
308         desc = ses_get_page2_descriptor(edev, ecomp);
309         if (desc)
310                 ecomp->locate = (desc[2] & 0x02) ? 1 : 0;
311 }
312
313 static int ses_set_locate(struct enclosure_device *edev,
314                           struct enclosure_component *ecomp,
315                           enum enclosure_component_setting val)
316 {
317         unsigned char desc[4];
318         unsigned char *desc_ptr;
319
320         if (!ses_page2_supported(edev))
321                 return -EINVAL;
322
323         desc_ptr = ses_get_page2_descriptor(edev, ecomp);
324
325         if (!desc_ptr)
326                 return -EIO;
327
328         init_device_slot_control(desc, ecomp, desc_ptr);
329
330         switch (val) {
331         case ENCLOSURE_SETTING_DISABLED:
332                 desc[2] &= 0xfd;
333                 break;
334         case ENCLOSURE_SETTING_ENABLED:
335                 desc[2] |= 0x02;
336                 break;
337         default:
338                 /* SES doesn't do the SGPIO blink settings */
339                 return -EINVAL;
340         }
341         return ses_set_page2_descriptor(edev, ecomp, desc);
342 }
343
344 static int ses_set_active(struct enclosure_device *edev,
345                           struct enclosure_component *ecomp,
346                           enum enclosure_component_setting val)
347 {
348         unsigned char desc[4];
349         unsigned char *desc_ptr;
350
351         if (!ses_page2_supported(edev))
352                 return -EINVAL;
353
354         desc_ptr = ses_get_page2_descriptor(edev, ecomp);
355
356         if (!desc_ptr)
357                 return -EIO;
358
359         init_device_slot_control(desc, ecomp, desc_ptr);
360
361         switch (val) {
362         case ENCLOSURE_SETTING_DISABLED:
363                 desc[2] &= 0x7f;
364                 ecomp->active = 0;
365                 break;
366         case ENCLOSURE_SETTING_ENABLED:
367                 desc[2] |= 0x80;
368                 ecomp->active = 1;
369                 break;
370         default:
371                 /* SES doesn't do the SGPIO blink settings */
372                 return -EINVAL;
373         }
374         return ses_set_page2_descriptor(edev, ecomp, desc);
375 }
376
377 static int ses_show_id(struct enclosure_device *edev, char *buf)
378 {
379         struct ses_device *ses_dev = edev->scratch;
380         unsigned long long id = get_unaligned_be64(ses_dev->page1+8+4);
381
382         return sprintf(buf, "%#llx\n", id);
383 }
384
385 static void ses_get_power_status(struct enclosure_device *edev,
386                                  struct enclosure_component *ecomp)
387 {
388         unsigned char *desc;
389
390         if (!ses_page2_supported(edev)) {
391                 ecomp->power_status = 0;
392                 return;
393         }
394
395         desc = ses_get_page2_descriptor(edev, ecomp);
396         if (desc)
397                 ecomp->power_status = (desc[3] & 0x10) ? 0 : 1;
398 }
399
400 static int ses_set_power_status(struct enclosure_device *edev,
401                                 struct enclosure_component *ecomp,
402                                 int val)
403 {
404         unsigned char desc[4];
405         unsigned char *desc_ptr;
406
407         if (!ses_page2_supported(edev))
408                 return -EINVAL;
409
410         desc_ptr = ses_get_page2_descriptor(edev, ecomp);
411
412         if (!desc_ptr)
413                 return -EIO;
414
415         init_device_slot_control(desc, ecomp, desc_ptr);
416
417         switch (val) {
418         /* power = 1 is device_off = 0 and vice versa */
419         case 0:
420                 desc[3] |= 0x10;
421                 break;
422         case 1:
423                 desc[3] &= 0xef;
424                 break;
425         default:
426                 return -EINVAL;
427         }
428         ecomp->power_status = val;
429         return ses_set_page2_descriptor(edev, ecomp, desc);
430 }
431
432 static struct enclosure_component_callbacks ses_enclosure_callbacks = {
433         .get_fault              = ses_get_fault,
434         .set_fault              = ses_set_fault,
435         .get_status             = ses_get_status,
436         .get_locate             = ses_get_locate,
437         .set_locate             = ses_set_locate,
438         .get_power_status       = ses_get_power_status,
439         .set_power_status       = ses_set_power_status,
440         .set_active             = ses_set_active,
441         .show_id                = ses_show_id,
442 };
443
444 struct ses_host_edev {
445         struct Scsi_Host *shost;
446         struct enclosure_device *edev;
447 };
448
449 #if 0
450 int ses_match_host(struct enclosure_device *edev, void *data)
451 {
452         struct ses_host_edev *sed = data;
453         struct scsi_device *sdev;
454
455         if (!scsi_is_sdev_device(edev->edev.parent))
456                 return 0;
457
458         sdev = to_scsi_device(edev->edev.parent);
459
460         if (sdev->host != sed->shost)
461                 return 0;
462
463         sed->edev = edev;
464         return 1;
465 }
466 #endif  /*  0  */
467
468 static int ses_process_descriptor(struct enclosure_component *ecomp,
469                                    unsigned char *desc, int max_desc_len)
470 {
471         int eip = desc[0] & 0x10;
472         int invalid = desc[0] & 0x80;
473         enum scsi_protocol proto = desc[0] & 0x0f;
474         u64 addr = 0;
475         int slot = -1;
476         struct ses_component *scomp = ecomp->scratch;
477         unsigned char *d;
478
479         if (invalid)
480                 return 0;
481
482         switch (proto) {
483         case SCSI_PROTOCOL_FCP:
484                 if (eip) {
485                         if (max_desc_len <= 7)
486                                 return 1;
487                         d = desc + 4;
488                         slot = d[3];
489                 }
490                 break;
491         case SCSI_PROTOCOL_SAS:
492
493                 if (eip) {
494                         if (max_desc_len <= 27)
495                                 return 1;
496                         d = desc + 4;
497                         slot = d[3];
498                         d = desc + 8;
499                 } else {
500                         if (max_desc_len <= 23)
501                                 return 1;
502                         d = desc + 4;
503                 }
504
505
506                 /* only take the phy0 addr */
507                 addr = (u64)d[12] << 56 |
508                         (u64)d[13] << 48 |
509                         (u64)d[14] << 40 |
510                         (u64)d[15] << 32 |
511                         (u64)d[16] << 24 |
512                         (u64)d[17] << 16 |
513                         (u64)d[18] << 8 |
514                         (u64)d[19];
515                 break;
516         default:
517                 /* FIXME: Need to add more protocols than just SAS */
518                 break;
519         }
520         ecomp->slot = slot;
521         scomp->addr = addr;
522
523         return 0;
524 }
525
526 struct efd {
527         u64 addr;
528         struct device *dev;
529 };
530
531 static int ses_enclosure_find_by_addr(struct enclosure_device *edev,
532                                       void *data)
533 {
534         struct efd *efd = data;
535         int i;
536         struct ses_component *scomp;
537
538         for (i = 0; i < edev->components; i++) {
539                 scomp = edev->component[i].scratch;
540                 if (scomp->addr != efd->addr)
541                         continue;
542
543                 if (enclosure_add_device(edev, i, efd->dev) == 0)
544                         kobject_uevent(&efd->dev->kobj, KOBJ_CHANGE);
545                 return 1;
546         }
547         return 0;
548 }
549
550 #define INIT_ALLOC_SIZE 32
551
552 static void ses_enclosure_data_process(struct enclosure_device *edev,
553                                        struct scsi_device *sdev,
554                                        int create)
555 {
556         u32 result;
557         unsigned char *buf = NULL, *type_ptr, *desc_ptr, *addl_desc_ptr = NULL;
558         int i, j, page7_len, len, components;
559         struct ses_device *ses_dev = edev->scratch;
560         int types = ses_dev->page1_num_types;
561         unsigned char *hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
562
563         if (!hdr_buf)
564                 goto simple_populate;
565
566         /* re-read page 10 */
567         if (ses_dev->page10)
568                 ses_recv_diag(sdev, 10, ses_dev->page10, ses_dev->page10_len);
569         /* Page 7 for the descriptors is optional */
570         result = ses_recv_diag(sdev, 7, hdr_buf, INIT_ALLOC_SIZE);
571         if (result)
572                 goto simple_populate;
573
574         page7_len = len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
575         /* add 1 for trailing '\0' we'll use */
576         buf = kzalloc(len + 1, GFP_KERNEL);
577         if (!buf)
578                 goto simple_populate;
579         result = ses_recv_diag(sdev, 7, buf, len);
580         if (result) {
581  simple_populate:
582                 kfree(buf);
583                 buf = NULL;
584                 desc_ptr = NULL;
585                 len = 0;
586                 page7_len = 0;
587         } else {
588                 desc_ptr = buf + 8;
589                 len = (desc_ptr[2] << 8) + desc_ptr[3];
590                 /* skip past overall descriptor */
591                 desc_ptr += len + 4;
592         }
593         if (ses_dev->page10 && ses_dev->page10_len > 9)
594                 addl_desc_ptr = ses_dev->page10 + 8;
595         type_ptr = ses_dev->page1_types;
596         components = 0;
597         for (i = 0; i < types; i++, type_ptr += 4) {
598                 for (j = 0; j < type_ptr[1]; j++) {
599                         char *name = NULL;
600                         struct enclosure_component *ecomp;
601                         int max_desc_len;
602
603                         if (desc_ptr) {
604                                 if (desc_ptr + 3 >= buf + page7_len) {
605                                         desc_ptr = NULL;
606                                 } else {
607                                         len = (desc_ptr[2] << 8) + desc_ptr[3];
608                                         desc_ptr += 4;
609                                         if (desc_ptr + len > buf + page7_len)
610                                                 desc_ptr = NULL;
611                                         else {
612                                                 /* Add trailing zero - pushes into
613                                                  * reserved space */
614                                                 desc_ptr[len] = '\0';
615                                                 name = desc_ptr;
616                                         }
617                                 }
618                         }
619                         if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE ||
620                             type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE) {
621
622                                 if (create)
623                                         ecomp = enclosure_component_alloc(
624                                                 edev,
625                                                 components++,
626                                                 type_ptr[0],
627                                                 name);
628                                 else if (components < edev->components)
629                                         ecomp = &edev->component[components++];
630                                 else
631                                         ecomp = ERR_PTR(-EINVAL);
632
633                                 if (!IS_ERR(ecomp)) {
634                                         if (addl_desc_ptr) {
635                                                 max_desc_len = ses_dev->page10_len -
636                                                     (addl_desc_ptr - ses_dev->page10);
637                                                 if (ses_process_descriptor(ecomp,
638                                                     addl_desc_ptr,
639                                                     max_desc_len))
640                                                         addl_desc_ptr = NULL;
641                                         }
642                                         if (create)
643                                                 enclosure_component_register(
644                                                         ecomp);
645                                 }
646                         }
647                         if (desc_ptr)
648                                 desc_ptr += len;
649
650                         if (addl_desc_ptr &&
651                             /* only find additional descriptions for specific devices */
652                             (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE ||
653                              type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE ||
654                              type_ptr[0] == ENCLOSURE_COMPONENT_SAS_EXPANDER ||
655                              /* these elements are optional */
656                              type_ptr[0] == ENCLOSURE_COMPONENT_SCSI_TARGET_PORT ||
657                              type_ptr[0] == ENCLOSURE_COMPONENT_SCSI_INITIATOR_PORT ||
658                              type_ptr[0] == ENCLOSURE_COMPONENT_CONTROLLER_ELECTRONICS)) {
659                                 addl_desc_ptr += addl_desc_ptr[1] + 2;
660                                 if (addl_desc_ptr + 1 >= ses_dev->page10 + ses_dev->page10_len)
661                                         addl_desc_ptr = NULL;
662                         }
663                 }
664         }
665         kfree(buf);
666         kfree(hdr_buf);
667 }
668
669 static void ses_match_to_enclosure(struct enclosure_device *edev,
670                                    struct scsi_device *sdev,
671                                    int refresh)
672 {
673         struct scsi_device *edev_sdev = to_scsi_device(edev->edev.parent);
674         struct efd efd = {
675                 .addr = 0,
676         };
677
678         if (refresh)
679                 ses_enclosure_data_process(edev, edev_sdev, 0);
680
681         if (scsi_is_sas_rphy(sdev->sdev_target->dev.parent))
682                 efd.addr = sas_get_address(sdev);
683
684         if (efd.addr) {
685                 efd.dev = &sdev->sdev_gendev;
686
687                 enclosure_for_each_device(ses_enclosure_find_by_addr, &efd);
688         }
689 }
690
691 static int ses_intf_add(struct device *cdev)
692 {
693         struct scsi_device *sdev = to_scsi_device(cdev->parent);
694         struct scsi_device *tmp_sdev;
695         unsigned char *buf = NULL, *hdr_buf, *type_ptr, page;
696         struct ses_device *ses_dev;
697         u32 result;
698         int i, types, len, components = 0;
699         int err = -ENOMEM;
700         int num_enclosures;
701         struct enclosure_device *edev;
702         struct ses_component *scomp = NULL;
703
704         if (!scsi_device_enclosure(sdev)) {
705                 /* not an enclosure, but might be in one */
706                 struct enclosure_device *prev = NULL;
707
708                 while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
709                         ses_match_to_enclosure(edev, sdev, 1);
710                         prev = edev;
711                 }
712                 return -ENODEV;
713         }
714
715         /* TYPE_ENCLOSURE prints a message in probe */
716         if (sdev->type != TYPE_ENCLOSURE)
717                 sdev_printk(KERN_NOTICE, sdev, "Embedded Enclosure Device\n");
718
719         ses_dev = kzalloc(sizeof(*ses_dev), GFP_KERNEL);
720         hdr_buf = kzalloc(INIT_ALLOC_SIZE, GFP_KERNEL);
721         if (!hdr_buf || !ses_dev)
722                 goto err_init_free;
723
724         page = 1;
725         result = ses_recv_diag(sdev, page, hdr_buf, INIT_ALLOC_SIZE);
726         if (result)
727                 goto recv_failed;
728
729         len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
730         buf = kzalloc(len, GFP_KERNEL);
731         if (!buf)
732                 goto err_free;
733
734         result = ses_recv_diag(sdev, page, buf, len);
735         if (result)
736                 goto recv_failed;
737
738         types = 0;
739
740         /* we always have one main enclosure and the rest are referred
741          * to as secondary subenclosures */
742         num_enclosures = buf[1] + 1;
743
744         /* begin at the enclosure descriptor */
745         type_ptr = buf + 8;
746         /* skip all the enclosure descriptors */
747         for (i = 0; i < num_enclosures && type_ptr < buf + len; i++) {
748                 types += type_ptr[2];
749                 type_ptr += type_ptr[3] + 4;
750         }
751
752         ses_dev->page1_types = type_ptr;
753         ses_dev->page1_num_types = types;
754
755         for (i = 0; i < types && type_ptr < buf + len; i++, type_ptr += 4) {
756                 if (type_ptr[0] == ENCLOSURE_COMPONENT_DEVICE ||
757                     type_ptr[0] == ENCLOSURE_COMPONENT_ARRAY_DEVICE)
758                         components += type_ptr[1];
759         }
760
761         ses_dev->page1 = buf;
762         ses_dev->page1_len = len;
763         buf = NULL;
764
765         page = 2;
766         result = ses_recv_diag(sdev, page, hdr_buf, INIT_ALLOC_SIZE);
767         if (result)
768                 goto page2_not_supported;
769
770         len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
771         buf = kzalloc(len, GFP_KERNEL);
772         if (!buf)
773                 goto err_free;
774
775         /* make sure getting page 2 actually works */
776         result = ses_recv_diag(sdev, 2, buf, len);
777         if (result)
778                 goto recv_failed;
779         ses_dev->page2 = buf;
780         ses_dev->page2_len = len;
781         buf = NULL;
782
783         /* The additional information page --- allows us
784          * to match up the devices */
785         page = 10;
786         result = ses_recv_diag(sdev, page, hdr_buf, INIT_ALLOC_SIZE);
787         if (!result) {
788
789                 len = (hdr_buf[2] << 8) + hdr_buf[3] + 4;
790                 buf = kzalloc(len, GFP_KERNEL);
791                 if (!buf)
792                         goto err_free;
793
794                 result = ses_recv_diag(sdev, page, buf, len);
795                 if (result)
796                         goto recv_failed;
797                 ses_dev->page10 = buf;
798                 ses_dev->page10_len = len;
799                 buf = NULL;
800         }
801 page2_not_supported:
802         if (components > 0) {
803                 scomp = kcalloc(components, sizeof(struct ses_component), GFP_KERNEL);
804                 if (!scomp)
805                         goto err_free;
806         }
807
808         edev = enclosure_register(cdev->parent, dev_name(&sdev->sdev_gendev),
809                                   components, &ses_enclosure_callbacks);
810         if (IS_ERR(edev)) {
811                 err = PTR_ERR(edev);
812                 goto err_free;
813         }
814
815         kfree(hdr_buf);
816
817         edev->scratch = ses_dev;
818         for (i = 0; i < components; i++)
819                 edev->component[i].scratch = scomp + i;
820
821         ses_enclosure_data_process(edev, sdev, 1);
822
823         /* see if there are any devices matching before
824          * we found the enclosure */
825         shost_for_each_device(tmp_sdev, sdev->host) {
826                 if (tmp_sdev->lun != 0 || scsi_device_enclosure(tmp_sdev))
827                         continue;
828                 ses_match_to_enclosure(edev, tmp_sdev, 0);
829         }
830
831         return 0;
832
833  recv_failed:
834         sdev_printk(KERN_ERR, sdev, "Failed to get diagnostic page 0x%x\n",
835                     page);
836         err = -ENODEV;
837  err_free:
838         kfree(buf);
839         kfree(scomp);
840         kfree(ses_dev->page10);
841         kfree(ses_dev->page2);
842         kfree(ses_dev->page1);
843  err_init_free:
844         kfree(ses_dev);
845         kfree(hdr_buf);
846         sdev_printk(KERN_ERR, sdev, "Failed to bind enclosure %d\n", err);
847         return err;
848 }
849
850 static int ses_remove(struct device *dev)
851 {
852         return 0;
853 }
854
855 static void ses_intf_remove_component(struct scsi_device *sdev)
856 {
857         struct enclosure_device *edev, *prev = NULL;
858
859         while ((edev = enclosure_find(&sdev->host->shost_gendev, prev)) != NULL) {
860                 prev = edev;
861                 if (!enclosure_remove_device(edev, &sdev->sdev_gendev))
862                         break;
863         }
864         if (edev)
865                 put_device(&edev->edev);
866 }
867
868 static void ses_intf_remove_enclosure(struct scsi_device *sdev)
869 {
870         struct enclosure_device *edev;
871         struct ses_device *ses_dev;
872
873         /*  exact match to this enclosure */
874         edev = enclosure_find(&sdev->sdev_gendev, NULL);
875         if (!edev)
876                 return;
877
878         ses_dev = edev->scratch;
879         edev->scratch = NULL;
880
881         kfree(ses_dev->page10);
882         kfree(ses_dev->page1);
883         kfree(ses_dev->page2);
884         kfree(ses_dev);
885
886         if (edev->components)
887                 kfree(edev->component[0].scratch);
888
889         put_device(&edev->edev);
890         enclosure_unregister(edev);
891 }
892
893 static void ses_intf_remove(struct device *cdev)
894 {
895         struct scsi_device *sdev = to_scsi_device(cdev->parent);
896
897         if (!scsi_device_enclosure(sdev))
898                 ses_intf_remove_component(sdev);
899         else
900                 ses_intf_remove_enclosure(sdev);
901 }
902
903 static struct class_interface ses_interface = {
904         .add_dev        = ses_intf_add,
905         .remove_dev     = ses_intf_remove,
906 };
907
908 static struct scsi_driver ses_template = {
909         .gendrv = {
910                 .name           = "ses",
911                 .owner          = THIS_MODULE,
912                 .probe          = ses_probe,
913                 .remove         = ses_remove,
914         },
915 };
916
917 static int __init ses_init(void)
918 {
919         int err;
920
921         err = scsi_register_interface(&ses_interface);
922         if (err)
923                 return err;
924
925         err = scsi_register_driver(&ses_template.gendrv);
926         if (err)
927                 goto out_unreg;
928
929         return 0;
930
931  out_unreg:
932         scsi_unregister_interface(&ses_interface);
933         return err;
934 }
935
936 static void __exit ses_exit(void)
937 {
938         scsi_unregister_driver(&ses_template.gendrv);
939         scsi_unregister_interface(&ses_interface);
940 }
941
942 module_init(ses_init);
943 module_exit(ses_exit);
944
945 MODULE_ALIAS_SCSI_DEVICE(TYPE_ENCLOSURE);
946
947 MODULE_AUTHOR("James Bottomley");
948 MODULE_DESCRIPTION("SCSI Enclosure Services (ses) driver");
949 MODULE_LICENSE("GPL v2");