1 /*
2  *  Sony MemoryStick Pro storage support
3  *
4  *  Copyright (C) 2007 Alex Dubov <oakad@yahoo.com>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  *
10  * Special thanks to Carlos Corbacho for providing various MemoryStick cards
11  * that made this driver possible.
12  *
13  */
14 
15 #include <linux/blkdev.h>
16 #include <linux/idr.h>
17 #include <linux/hdreg.h>
18 #include <linux/kthread.h>
19 #include <linux/delay.h>
20 #include <linux/memstick.h>
21 
22 #define DRIVER_NAME "mspro_block"
23 
24 static int major;
25 module_param(major, int, 0644);
26 
27 #define MSPRO_BLOCK_MAX_SEGS  32
28 #define MSPRO_BLOCK_MAX_PAGES ((2 << 16) - 1)
29 
30 #define MSPRO_BLOCK_SIGNATURE        0xa5c3
31 #define MSPRO_BLOCK_MAX_ATTRIBUTES   41
32 
33 #define MSPRO_BLOCK_PART_SHIFT 3
34 
35 enum {
36 	MSPRO_BLOCK_ID_SYSINFO         = 0x10,
37 	MSPRO_BLOCK_ID_MODELNAME       = 0x15,
38 	MSPRO_BLOCK_ID_MBR             = 0x20,
39 	MSPRO_BLOCK_ID_PBR16           = 0x21,
40 	MSPRO_BLOCK_ID_PBR32           = 0x22,
41 	MSPRO_BLOCK_ID_SPECFILEVALUES1 = 0x25,
42 	MSPRO_BLOCK_ID_SPECFILEVALUES2 = 0x26,
43 	MSPRO_BLOCK_ID_DEVINFO         = 0x30
44 };
45 
46 struct mspro_sys_attr {
47 	size_t                  size;
48 	void                    *data;
49 	unsigned char           id;
50 	char                    name[32];
51 	struct device_attribute dev_attr;
52 };
53 
54 struct mspro_attr_entry {
55 	unsigned int  address;
56 	unsigned int  size;
57 	unsigned char id;
58 	unsigned char reserved[3];
59 } __attribute__((packed));
60 
61 struct mspro_attribute {
62 	unsigned short          signature;
63 	unsigned short          version;
64 	unsigned char           count;
65 	unsigned char           reserved[11];
66 	struct mspro_attr_entry entries[];
67 } __attribute__((packed));
68 
69 struct mspro_sys_info {
70 	unsigned char  class;
71 	unsigned char  reserved0;
72 	unsigned short block_size;
73 	unsigned short block_count;
74 	unsigned short user_block_count;
75 	unsigned short page_size;
76 	unsigned char  reserved1[2];
77 	unsigned char  assembly_date[8];
78 	unsigned int   serial_number;
79 	unsigned char  assembly_maker_code;
80 	unsigned char  assembly_model_code[3];
81 	unsigned short memory_maker_code;
82 	unsigned short memory_model_code;
83 	unsigned char  reserved2[4];
84 	unsigned char  vcc;
85 	unsigned char  vpp;
86 	unsigned short controller_number;
87 	unsigned short controller_function;
88 	unsigned short start_sector;
89 	unsigned short unit_size;
90 	unsigned char  ms_sub_class;
91 	unsigned char  reserved3[4];
92 	unsigned char  interface_type;
93 	unsigned short controller_code;
94 	unsigned char  format_type;
95 	unsigned char  reserved4;
96 	unsigned char  device_type;
97 	unsigned char  reserved5[7];
98 	unsigned char  mspro_id[16];
99 	unsigned char  reserved6[16];
100 } __attribute__((packed));
101 
102 struct mspro_mbr {
103 	unsigned char boot_partition;
104 	unsigned char start_head;
105 	unsigned char start_sector;
106 	unsigned char start_cylinder;
107 	unsigned char partition_type;
108 	unsigned char end_head;
109 	unsigned char end_sector;
110 	unsigned char end_cylinder;
111 	unsigned int  start_sectors;
112 	unsigned int  sectors_per_partition;
113 } __attribute__((packed));
114 
115 struct mspro_specfile {
116 	char           name[8];
117 	char           ext[3];
118 	unsigned char  attr;
119 	unsigned char  reserved[10];
120 	unsigned short time;
121 	unsigned short date;
122 	unsigned short cluster;
123 	unsigned int   size;
124 } __attribute__((packed));
125 
126 struct mspro_devinfo {
127 	unsigned short cylinders;
128 	unsigned short heads;
129 	unsigned short bytes_per_track;
130 	unsigned short bytes_per_sector;
131 	unsigned short sectors_per_track;
132 	unsigned char  reserved[6];
133 } __attribute__((packed));
134 
135 struct mspro_block_data {
136 	struct memstick_dev   *card;
137 	unsigned int          usage_count;
138 	unsigned int          caps;
139 	struct gendisk        *disk;
140 	struct request_queue  *queue;
141 	struct request        *block_req;
142 	spinlock_t            q_lock;
143 
144 	unsigned short        page_size;
145 	unsigned short        cylinders;
146 	unsigned short        heads;
147 	unsigned short        sectors_per_track;
148 
149 	unsigned char         system;
150 	unsigned char         read_only:1,
151 			      eject:1,
152 			      has_request:1,
153 			      data_dir:1,
154 			      active:1;
155 	unsigned char         transfer_cmd;
156 
157 	int                   (*mrq_handler)(struct memstick_dev *card,
158 					     struct memstick_request **mrq);
159 
160 	struct attribute_group attr_group;
161 
162 	struct scatterlist    req_sg[MSPRO_BLOCK_MAX_SEGS];
163 	unsigned int          seg_count;
164 	unsigned int          current_seg;
165 	unsigned int          current_page;
166 };
167 
168 static DEFINE_IDR(mspro_block_disk_idr);
169 static DEFINE_MUTEX(mspro_block_disk_lock);
170 
171 static int mspro_block_complete_req(struct memstick_dev *card, int error);
172 
173 /*** Block device ***/
174 
175 static int mspro_block_bd_open(struct block_device *bdev, fmode_t mode)
176 {
177 	struct gendisk *disk = bdev->bd_disk;
178 	struct mspro_block_data *msb = disk->private_data;
179 	int rc = -ENXIO;
180 
181 	mutex_lock(&mspro_block_disk_lock);
182 
183 	if (msb && msb->card) {
184 		msb->usage_count++;
185 		if ((mode & FMODE_WRITE) && msb->read_only)
186 			rc = -EROFS;
187 		else
188 			rc = 0;
189 	}
190 
191 	mutex_unlock(&mspro_block_disk_lock);
192 
193 	return rc;
194 }
195 
196 
197 static int mspro_block_disk_release(struct gendisk *disk)
198 {
199 	struct mspro_block_data *msb = disk->private_data;
200 	int disk_id = MINOR(disk_devt(disk)) >> MSPRO_BLOCK_PART_SHIFT;
201 
202 	mutex_lock(&mspro_block_disk_lock);
203 
204 	if (msb) {
205 		if (msb->usage_count)
206 			msb->usage_count--;
207 
208 		if (!msb->usage_count) {
209 			kfree(msb);
210 			disk->private_data = NULL;
211 			idr_remove(&mspro_block_disk_idr, disk_id);
212 			put_disk(disk);
213 		}
214 	}
215 
216 	mutex_unlock(&mspro_block_disk_lock);
217 
218 	return 0;
219 }
220 
221 static int mspro_block_bd_release(struct gendisk *disk, fmode_t mode)
222 {
223 	return mspro_block_disk_release(disk);
224 }
225 
226 static int mspro_block_bd_getgeo(struct block_device *bdev,
227 				 struct hd_geometry *geo)
228 {
229 	struct mspro_block_data *msb = bdev->bd_disk->private_data;
230 
231 	geo->heads = msb->heads;
232 	geo->sectors = msb->sectors_per_track;
233 	geo->cylinders = msb->cylinders;
234 
235 	return 0;
236 }
237 
238 static struct block_device_operations ms_block_bdops = {
239 	.open    = mspro_block_bd_open,
240 	.release = mspro_block_bd_release,
241 	.getgeo  = mspro_block_bd_getgeo,
242 	.owner   = THIS_MODULE
243 };
244 
245 /*** Information ***/
246 
247 static struct mspro_sys_attr *mspro_from_sysfs_attr(struct attribute *attr)
248 {
249 	struct device_attribute *dev_attr
250 		= container_of(attr, struct device_attribute, attr);
251 	return container_of(dev_attr, struct mspro_sys_attr, dev_attr);
252 }
253 
254 static const char *mspro_block_attr_name(unsigned char tag)
255 {
256 	switch (tag) {
257 	case MSPRO_BLOCK_ID_SYSINFO:
258 		return "attr_sysinfo";
259 	case MSPRO_BLOCK_ID_MODELNAME:
260 		return "attr_modelname";
261 	case MSPRO_BLOCK_ID_MBR:
262 		return "attr_mbr";
263 	case MSPRO_BLOCK_ID_PBR16:
264 		return "attr_pbr16";
265 	case MSPRO_BLOCK_ID_PBR32:
266 		return "attr_pbr32";
267 	case MSPRO_BLOCK_ID_SPECFILEVALUES1:
268 		return "attr_specfilevalues1";
269 	case MSPRO_BLOCK_ID_SPECFILEVALUES2:
270 		return "attr_specfilevalues2";
271 	case MSPRO_BLOCK_ID_DEVINFO:
272 		return "attr_devinfo";
273 	default:
274 		return NULL;
275 	};
276 }
277 
278 typedef ssize_t (*sysfs_show_t)(struct device *dev,
279 				struct device_attribute *attr,
280 				char *buffer);
281 
282 static ssize_t mspro_block_attr_show_default(struct device *dev,
283 					     struct device_attribute *attr,
284 					     char *buffer)
285 {
286 	struct mspro_sys_attr *s_attr = container_of(attr,
287 						     struct mspro_sys_attr,
288 						     dev_attr);
289 
290 	ssize_t cnt, rc = 0;
291 
292 	for (cnt = 0; cnt < s_attr->size; cnt++) {
293 		if (cnt && !(cnt % 16)) {
294 			if (PAGE_SIZE - rc)
295 				buffer[rc++] = '\n';
296 		}
297 
298 		rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "%02x ",
299 				((unsigned char *)s_attr->data)[cnt]);
300 	}
301 	return rc;
302 }
303 
304 static ssize_t mspro_block_attr_show_sysinfo(struct device *dev,
305 					     struct device_attribute *attr,
306 					     char *buffer)
307 {
308 	struct mspro_sys_attr *x_attr = container_of(attr,
309 						     struct mspro_sys_attr,
310 						     dev_attr);
311 	struct mspro_sys_info *x_sys = x_attr->data;
312 	ssize_t rc = 0;
313 	int date_tz = 0, date_tz_f = 0;
314 
315 	if (x_sys->assembly_date[0] > 0x80U) {
316 		date_tz = (~x_sys->assembly_date[0]) + 1;
317 		date_tz_f = date_tz & 3;
318 		date_tz >>= 2;
319 		date_tz = -date_tz;
320 		date_tz_f *= 15;
321 	} else if (x_sys->assembly_date[0] < 0x80U) {
322 		date_tz = x_sys->assembly_date[0];
323 		date_tz_f = date_tz & 3;
324 		date_tz >>= 2;
325 		date_tz_f *= 15;
326 	}
327 
328 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "class: %x\n",
329 			x_sys->class);
330 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "block size: %x\n",
331 			be16_to_cpu(x_sys->block_size));
332 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "block count: %x\n",
333 			be16_to_cpu(x_sys->block_count));
334 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "user block count: %x\n",
335 			be16_to_cpu(x_sys->user_block_count));
336 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "page size: %x\n",
337 			be16_to_cpu(x_sys->page_size));
338 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "assembly date: "
339 			"GMT%+d:%d %04u-%02u-%02u %02u:%02u:%02u\n",
340 			date_tz, date_tz_f,
341 			be16_to_cpu(*(unsigned short *)
342 				    &x_sys->assembly_date[1]),
343 			x_sys->assembly_date[3], x_sys->assembly_date[4],
344 			x_sys->assembly_date[5], x_sys->assembly_date[6],
345 			x_sys->assembly_date[7]);
346 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "serial number: %x\n",
347 			be32_to_cpu(x_sys->serial_number));
348 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc,
349 			"assembly maker code: %x\n",
350 			x_sys->assembly_maker_code);
351 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "assembly model code: "
352 			"%02x%02x%02x\n", x_sys->assembly_model_code[0],
353 			x_sys->assembly_model_code[1],
354 			x_sys->assembly_model_code[2]);
355 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "memory maker code: %x\n",
356 			be16_to_cpu(x_sys->memory_maker_code));
357 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "memory model code: %x\n",
358 			be16_to_cpu(x_sys->memory_model_code));
359 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "vcc: %x\n",
360 			x_sys->vcc);
361 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "vpp: %x\n",
362 			x_sys->vpp);
363 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "controller number: %x\n",
364 			be16_to_cpu(x_sys->controller_number));
365 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc,
366 			"controller function: %x\n",
367 			be16_to_cpu(x_sys->controller_function));
368 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start sector: %x\n",
369 			be16_to_cpu(x_sys->start_sector));
370 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "unit size: %x\n",
371 			be16_to_cpu(x_sys->unit_size));
372 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "sub class: %x\n",
373 			x_sys->ms_sub_class);
374 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "interface type: %x\n",
375 			x_sys->interface_type);
376 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "controller code: %x\n",
377 			be16_to_cpu(x_sys->controller_code));
378 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "format type: %x\n",
379 			x_sys->format_type);
380 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "device type: %x\n",
381 			x_sys->device_type);
382 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "mspro id: %s\n",
383 			x_sys->mspro_id);
384 	return rc;
385 }
386 
387 static ssize_t mspro_block_attr_show_modelname(struct device *dev,
388 					       struct device_attribute *attr,
389 					       char *buffer)
390 {
391 	struct mspro_sys_attr *s_attr = container_of(attr,
392 						     struct mspro_sys_attr,
393 						     dev_attr);
394 
395 	return scnprintf(buffer, PAGE_SIZE, "%s", (char *)s_attr->data);
396 }
397 
398 static ssize_t mspro_block_attr_show_mbr(struct device *dev,
399 					 struct device_attribute *attr,
400 					 char *buffer)
401 {
402 	struct mspro_sys_attr *x_attr = container_of(attr,
403 						     struct mspro_sys_attr,
404 						     dev_attr);
405 	struct mspro_mbr *x_mbr = x_attr->data;
406 	ssize_t rc = 0;
407 
408 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "boot partition: %x\n",
409 			x_mbr->boot_partition);
410 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start head: %x\n",
411 			x_mbr->start_head);
412 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start sector: %x\n",
413 			x_mbr->start_sector);
414 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start cylinder: %x\n",
415 			x_mbr->start_cylinder);
416 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "partition type: %x\n",
417 			x_mbr->partition_type);
418 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "end head: %x\n",
419 			x_mbr->end_head);
420 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "end sector: %x\n",
421 			x_mbr->end_sector);
422 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "end cylinder: %x\n",
423 			x_mbr->end_cylinder);
424 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start sectors: %x\n",
425 			x_mbr->start_sectors);
426 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc,
427 			"sectors per partition: %x\n",
428 			x_mbr->sectors_per_partition);
429 	return rc;
430 }
431 
432 static ssize_t mspro_block_attr_show_specfile(struct device *dev,
433 					      struct device_attribute *attr,
434 					      char *buffer)
435 {
436 	struct mspro_sys_attr *x_attr = container_of(attr,
437 						     struct mspro_sys_attr,
438 						     dev_attr);
439 	struct mspro_specfile *x_spfile = x_attr->data;
440 	char name[9], ext[4];
441 	ssize_t rc = 0;
442 
443 	memcpy(name, x_spfile->name, 8);
444 	name[8] = 0;
445 	memcpy(ext, x_spfile->ext, 3);
446 	ext[3] = 0;
447 
448 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "name: %s\n", name);
449 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "ext: %s\n", ext);
450 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "attribute: %x\n",
451 			x_spfile->attr);
452 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "time: %d:%d:%d\n",
453 			x_spfile->time >> 11,
454 			(x_spfile->time >> 5) & 0x3f,
455 			(x_spfile->time & 0x1f) * 2);
456 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "date: %d-%d-%d\n",
457 			(x_spfile->date >> 9) + 1980,
458 			(x_spfile->date >> 5) & 0xf,
459 			x_spfile->date & 0x1f);
460 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "start cluster: %x\n",
461 			x_spfile->cluster);
462 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "size: %x\n",
463 			x_spfile->size);
464 	return rc;
465 }
466 
467 static ssize_t mspro_block_attr_show_devinfo(struct device *dev,
468 					     struct device_attribute *attr,
469 					     char *buffer)
470 {
471 	struct mspro_sys_attr *x_attr = container_of(attr,
472 						     struct mspro_sys_attr,
473 						     dev_attr);
474 	struct mspro_devinfo *x_devinfo = x_attr->data;
475 	ssize_t rc = 0;
476 
477 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "cylinders: %x\n",
478 			be16_to_cpu(x_devinfo->cylinders));
479 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "heads: %x\n",
480 			be16_to_cpu(x_devinfo->heads));
481 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "bytes per track: %x\n",
482 			be16_to_cpu(x_devinfo->bytes_per_track));
483 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "bytes per sector: %x\n",
484 			be16_to_cpu(x_devinfo->bytes_per_sector));
485 	rc += scnprintf(buffer + rc, PAGE_SIZE - rc, "sectors per track: %x\n",
486 			be16_to_cpu(x_devinfo->sectors_per_track));
487 	return rc;
488 }
489 
490 static sysfs_show_t mspro_block_attr_show(unsigned char tag)
491 {
492 	switch (tag) {
493 	case MSPRO_BLOCK_ID_SYSINFO:
494 		return mspro_block_attr_show_sysinfo;
495 	case MSPRO_BLOCK_ID_MODELNAME:
496 		return mspro_block_attr_show_modelname;
497 	case MSPRO_BLOCK_ID_MBR:
498 		return mspro_block_attr_show_mbr;
499 	case MSPRO_BLOCK_ID_SPECFILEVALUES1:
500 	case MSPRO_BLOCK_ID_SPECFILEVALUES2:
501 		return mspro_block_attr_show_specfile;
502 	case MSPRO_BLOCK_ID_DEVINFO:
503 		return mspro_block_attr_show_devinfo;
504 	default:
505 		return mspro_block_attr_show_default;
506 	}
507 }
508 
509 /*** Protocol handlers ***/
510 
511 /*
512  * Functions prefixed with "h_" are protocol callbacks. They can be called from
513  * interrupt context. Return value of 0 means that request processing is still
514  * ongoing, while special error value of -EAGAIN means that current request is
515  * finished (and request processor should come back some time later).
516  */
517 
518 static int h_mspro_block_req_init(struct memstick_dev *card,
519 				  struct memstick_request **mrq)
520 {
521 	struct mspro_block_data *msb = memstick_get_drvdata(card);
522 
523 	*mrq = &card->current_mrq;
524 	card->next_request = msb->mrq_handler;
525 	return 0;
526 }
527 
528 static int h_mspro_block_default(struct memstick_dev *card,
529 				 struct memstick_request **mrq)
530 {
531 	return mspro_block_complete_req(card, (*mrq)->error);
532 }
533 
534 static int h_mspro_block_default_bad(struct memstick_dev *card,
535 				     struct memstick_request **mrq)
536 {
537 	return -ENXIO;
538 }
539 
540 static int h_mspro_block_get_ro(struct memstick_dev *card,
541 				struct memstick_request **mrq)
542 {
543 	struct mspro_block_data *msb = memstick_get_drvdata(card);
544 
545 	if (!(*mrq)->error) {
546 		if ((*mrq)->data[offsetof(struct ms_status_register, status0)]
547 		    & MEMSTICK_STATUS0_WP)
548 			msb->read_only = 1;
549 		else
550 			msb->read_only = 0;
551 	}
552 
553 	return mspro_block_complete_req(card, (*mrq)->error);
554 }
555 
556 static int h_mspro_block_wait_for_ced(struct memstick_dev *card,
557 				      struct memstick_request **mrq)
558 {
559 	dev_dbg(&card->dev, "wait for ced: value %x\n", (*mrq)->data[0]);
560 
561 	if (!(*mrq)->error) {
562 		if ((*mrq)->data[0] & (MEMSTICK_INT_CMDNAK | MEMSTICK_INT_ERR))
563 			(*mrq)->error = -EFAULT;
564 		else if (!((*mrq)->data[0] & MEMSTICK_INT_CED))
565 			return 0;
566 	}
567 
568 	return mspro_block_complete_req(card, (*mrq)->error);
569 }
570 
571 static int h_mspro_block_transfer_data(struct memstick_dev *card,
572 				       struct memstick_request **mrq)
573 {
574 	struct mspro_block_data *msb = memstick_get_drvdata(card);
575 	unsigned char t_val = 0;
576 	struct scatterlist t_sg = { 0 };
577 	size_t t_offset;
578 
579 	if ((*mrq)->error)
580 		return mspro_block_complete_req(card, (*mrq)->error);
581 
582 	switch ((*mrq)->tpc) {
583 	case MS_TPC_WRITE_REG:
584 		memstick_init_req(*mrq, MS_TPC_SET_CMD, &msb->transfer_cmd, 1);
585 		(*mrq)->need_card_int = 1;
586 		return 0;
587 	case MS_TPC_SET_CMD:
588 		t_val = (*mrq)->int_reg;
589 		memstick_init_req(*mrq, MS_TPC_GET_INT, NULL, 1);
590 		if (msb->caps & MEMSTICK_CAP_AUTO_GET_INT)
591 			goto has_int_reg;
592 		return 0;
593 	case MS_TPC_GET_INT:
594 		t_val = (*mrq)->data[0];
595 has_int_reg:
596 		if (t_val & (MEMSTICK_INT_CMDNAK | MEMSTICK_INT_ERR)) {
597 			t_val = MSPRO_CMD_STOP;
598 			memstick_init_req(*mrq, MS_TPC_SET_CMD, &t_val, 1);
599 			card->next_request = h_mspro_block_default;
600 			return 0;
601 		}
602 
603 		if (msb->current_page
604 		    == (msb->req_sg[msb->current_seg].length
605 			/ msb->page_size)) {
606 			msb->current_page = 0;
607 			msb->current_seg++;
608 
609 			if (msb->current_seg == msb->seg_count) {
610 				if (t_val & MEMSTICK_INT_CED) {
611 					return mspro_block_complete_req(card,
612 									0);
613 				} else {
614 					card->next_request
615 						= h_mspro_block_wait_for_ced;
616 					memstick_init_req(*mrq, MS_TPC_GET_INT,
617 							  NULL, 1);
618 					return 0;
619 				}
620 			}
621 		}
622 
623 		if (!(t_val & MEMSTICK_INT_BREQ)) {
624 			memstick_init_req(*mrq, MS_TPC_GET_INT, NULL, 1);
625 			return 0;
626 		}
627 
628 		t_offset = msb->req_sg[msb->current_seg].offset;
629 		t_offset += msb->current_page * msb->page_size;
630 
631 		sg_set_page(&t_sg,
632 			    nth_page(sg_page(&(msb->req_sg[msb->current_seg])),
633 				     t_offset >> PAGE_SHIFT),
634 			    msb->page_size, offset_in_page(t_offset));
635 
636 		memstick_init_req_sg(*mrq, msb->data_dir == READ
637 					   ? MS_TPC_READ_LONG_DATA
638 					   : MS_TPC_WRITE_LONG_DATA,
639 				     &t_sg);
640 		(*mrq)->need_card_int = 1;
641 		return 0;
642 	case MS_TPC_READ_LONG_DATA:
643 	case MS_TPC_WRITE_LONG_DATA:
644 		msb->current_page++;
645 		if (msb->caps & MEMSTICK_CAP_AUTO_GET_INT) {
646 			t_val = (*mrq)->int_reg;
647 			goto has_int_reg;
648 		} else {
649 			memstick_init_req(*mrq, MS_TPC_GET_INT, NULL, 1);
650 			return 0;
651 		}
652 
653 	default:
654 		BUG();
655 	}
656 }
657 
658 /*** Data transfer ***/
659 
660 static int mspro_block_issue_req(struct memstick_dev *card, int chunk)
661 {
662 	struct mspro_block_data *msb = memstick_get_drvdata(card);
663 	sector_t t_sec;
664 	unsigned int count;
665 	struct mspro_param_register param;
666 
667 try_again:
668 	while (chunk) {
669 		msb->current_page = 0;
670 		msb->current_seg = 0;
671 		msb->seg_count = blk_rq_map_sg(msb->block_req->q,
672 					       msb->block_req,
673 					       msb->req_sg);
674 
675 		if (!msb->seg_count) {
676 			chunk = __blk_end_request(msb->block_req, -ENOMEM,
677 					blk_rq_cur_bytes(msb->block_req));
678 			continue;
679 		}
680 
681 		t_sec = msb->block_req->sector << 9;
682 		sector_div(t_sec, msb->page_size);
683 
684 		count = msb->block_req->nr_sectors << 9;
685 		count /= msb->page_size;
686 
687 		param.system = msb->system;
688 		param.data_count = cpu_to_be16(count);
689 		param.data_address = cpu_to_be32((uint32_t)t_sec);
690 		param.tpc_param = 0;
691 
692 		msb->data_dir = rq_data_dir(msb->block_req);
693 		msb->transfer_cmd = msb->data_dir == READ
694 				    ? MSPRO_CMD_READ_DATA
695 				    : MSPRO_CMD_WRITE_DATA;
696 
697 		dev_dbg(&card->dev, "data transfer: cmd %x, "
698 			"lba %x, count %x\n", msb->transfer_cmd,
699 			be32_to_cpu(param.data_address), count);
700 
701 		card->next_request = h_mspro_block_req_init;
702 		msb->mrq_handler = h_mspro_block_transfer_data;
703 		memstick_init_req(&card->current_mrq, MS_TPC_WRITE_REG,
704 				  &param, sizeof(param));
705 		memstick_new_req(card->host);
706 		return 0;
707 	}
708 
709 	dev_dbg(&card->dev, "elv_next\n");
710 	msb->block_req = elv_next_request(msb->queue);
711 	if (!msb->block_req) {
712 		dev_dbg(&card->dev, "issue end\n");
713 		return -EAGAIN;
714 	}
715 
716 	dev_dbg(&card->dev, "trying again\n");
717 	chunk = 1;
718 	goto try_again;
719 }
720 
721 static int mspro_block_complete_req(struct memstick_dev *card, int error)
722 {
723 	struct mspro_block_data *msb = memstick_get_drvdata(card);
724 	int chunk, cnt;
725 	unsigned int t_len = 0;
726 	unsigned long flags;
727 
728 	spin_lock_irqsave(&msb->q_lock, flags);
729 	dev_dbg(&card->dev, "complete %d, %d\n", msb->has_request ? 1 : 0,
730 		error);
731 
732 	if (msb->has_request) {
733 		/* Nothing to do - not really an error */
734 		if (error == -EAGAIN)
735 			error = 0;
736 
737 		if (error || (card->current_mrq.tpc == MSPRO_CMD_STOP)) {
738 			if (msb->data_dir == READ) {
739 				for (cnt = 0; cnt < msb->current_seg; cnt++)
740 					t_len += msb->req_sg[cnt].length
741 						 / msb->page_size;
742 
743 					if (msb->current_page)
744 						t_len += msb->current_page - 1;
745 
746 					t_len *= msb->page_size;
747 			}
748 		} else
749 			t_len = msb->block_req->nr_sectors << 9;
750 
751 		dev_dbg(&card->dev, "transferred %x (%d)\n", t_len, error);
752 
753 		if (error && !t_len)
754 			t_len = blk_rq_cur_bytes(msb->block_req);
755 
756 		chunk = __blk_end_request(msb->block_req, error, t_len);
757 
758 		error = mspro_block_issue_req(card, chunk);
759 
760 		if (!error)
761 			goto out;
762 		else
763 			msb->has_request = 0;
764 	} else {
765 		if (!error)
766 			error = -EAGAIN;
767 	}
768 
769 	card->next_request = h_mspro_block_default_bad;
770 	complete_all(&card->mrq_complete);
771 out:
772 	spin_unlock_irqrestore(&msb->q_lock, flags);
773 	return error;
774 }
775 
776 static void mspro_block_stop(struct memstick_dev *card)
777 {
778 	struct mspro_block_data *msb = memstick_get_drvdata(card);
779 	int rc = 0;
780 	unsigned long flags;
781 
782 	while (1) {
783 		spin_lock_irqsave(&msb->q_lock, flags);
784 		if (!msb->has_request) {
785 			blk_stop_queue(msb->queue);
786 			rc = 1;
787 		}
788 		spin_unlock_irqrestore(&msb->q_lock, flags);
789 
790 		if (rc)
791 			break;
792 
793 		wait_for_completion(&card->mrq_complete);
794 	}
795 }
796 
797 static void mspro_block_start(struct memstick_dev *card)
798 {
799 	struct mspro_block_data *msb = memstick_get_drvdata(card);
800 	unsigned long flags;
801 
802 	spin_lock_irqsave(&msb->q_lock, flags);
803 	blk_start_queue(msb->queue);
804 	spin_unlock_irqrestore(&msb->q_lock, flags);
805 }
806 
807 static int mspro_block_prepare_req(struct request_queue *q, struct request *req)
808 {
809 	if (!blk_fs_request(req) && !blk_pc_request(req)) {
810 		blk_dump_rq_flags(req, "MSPro unsupported request");
811 		return BLKPREP_KILL;
812 	}
813 
814 	req->cmd_flags |= REQ_DONTPREP;
815 
816 	return BLKPREP_OK;
817 }
818 
819 static void mspro_block_submit_req(struct request_queue *q)
820 {
821 	struct memstick_dev *card = q->queuedata;
822 	struct mspro_block_data *msb = memstick_get_drvdata(card);
823 	struct request *req = NULL;
824 
825 	if (msb->has_request)
826 		return;
827 
828 	if (msb->eject) {
829 		while ((req = elv_next_request(q)) != NULL)
830 			__blk_end_request(req, -ENODEV, blk_rq_bytes(req));
831 
832 		return;
833 	}
834 
835 	msb->has_request = 1;
836 	if (mspro_block_issue_req(card, 0))
837 		msb->has_request = 0;
838 }
839 
840 /*** Initialization ***/
841 
842 static int mspro_block_wait_for_ced(struct memstick_dev *card)
843 {
844 	struct mspro_block_data *msb = memstick_get_drvdata(card);
845 
846 	card->next_request = h_mspro_block_req_init;
847 	msb->mrq_handler = h_mspro_block_wait_for_ced;
848 	memstick_init_req(&card->current_mrq, MS_TPC_GET_INT, NULL, 1);
849 	memstick_new_req(card->host);
850 	wait_for_completion(&card->mrq_complete);
851 	return card->current_mrq.error;
852 }
853 
854 static int mspro_block_set_interface(struct memstick_dev *card,
855 				     unsigned char sys_reg)
856 {
857 	struct memstick_host *host = card->host;
858 	struct mspro_block_data *msb = memstick_get_drvdata(card);
859 	struct mspro_param_register param = {
860 		.system = sys_reg,
861 		.data_count = 0,
862 		.data_address = 0,
863 		.tpc_param = 0
864 	};
865 
866 	card->next_request = h_mspro_block_req_init;
867 	msb->mrq_handler = h_mspro_block_default;
868 	memstick_init_req(&card->current_mrq, MS_TPC_WRITE_REG, &param,
869 			  sizeof(param));
870 	memstick_new_req(host);
871 	wait_for_completion(&card->mrq_complete);
872 	return card->current_mrq.error;
873 }
874 
875 static int mspro_block_switch_interface(struct memstick_dev *card)
876 {
877 	struct memstick_host *host = card->host;
878 	struct mspro_block_data *msb = memstick_get_drvdata(card);
879 	int rc = 0;
880 
881 try_again:
882 	if (msb->caps & MEMSTICK_CAP_PAR4)
883 		rc = mspro_block_set_interface(card, MEMSTICK_SYS_PAR4);
884 	else
885 		return 0;
886 
887 	if (rc) {
888 		printk(KERN_WARNING
889 		       "%s: could not switch to 4-bit mode, error %d\n",
890 		       card->dev.bus_id, rc);
891 		return 0;
892 	}
893 
894 	msb->system = MEMSTICK_SYS_PAR4;
895 	host->set_param(host, MEMSTICK_INTERFACE, MEMSTICK_PAR4);
896 	printk(KERN_INFO "%s: switching to 4-bit parallel mode\n",
897 	       card->dev.bus_id);
898 
899 	if (msb->caps & MEMSTICK_CAP_PAR8) {
900 		rc = mspro_block_set_interface(card, MEMSTICK_SYS_PAR8);
901 
902 		if (!rc) {
903 			msb->system = MEMSTICK_SYS_PAR8;
904 			host->set_param(host, MEMSTICK_INTERFACE,
905 					MEMSTICK_PAR8);
906 			printk(KERN_INFO
907 			       "%s: switching to 8-bit parallel mode\n",
908 			       card->dev.bus_id);
909 		} else
910 			printk(KERN_WARNING
911 			       "%s: could not switch to 8-bit mode, error %d\n",
912 			       card->dev.bus_id, rc);
913 	}
914 
915 	card->next_request = h_mspro_block_req_init;
916 	msb->mrq_handler = h_mspro_block_default;
917 	memstick_init_req(&card->current_mrq, MS_TPC_GET_INT, NULL, 1);
918 	memstick_new_req(card->host);
919 	wait_for_completion(&card->mrq_complete);
920 	rc = card->current_mrq.error;
921 
922 	if (rc) {
923 		printk(KERN_WARNING
924 		       "%s: interface error, trying to fall back to serial\n",
925 		       card->dev.bus_id);
926 		msb->system = MEMSTICK_SYS_SERIAL;
927 		host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
928 		msleep(10);
929 		host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_ON);
930 		host->set_param(host, MEMSTICK_INTERFACE, MEMSTICK_SERIAL);
931 
932 		rc = memstick_set_rw_addr(card);
933 		if (!rc)
934 			rc = mspro_block_set_interface(card, msb->system);
935 
936 		if (!rc) {
937 			msleep(150);
938 			rc = mspro_block_wait_for_ced(card);
939 			if (rc)
940 				return rc;
941 
942 			if (msb->caps & MEMSTICK_CAP_PAR8) {
943 				msb->caps &= ~MEMSTICK_CAP_PAR8;
944 				goto try_again;
945 			}
946 		}
947 	}
948 	return rc;
949 }
950 
951 /* Memory allocated for attributes by this function should be freed by
952  * mspro_block_data_clear, no matter if the initialization process succeded
953  * or failed.
954  */
955 static int mspro_block_read_attributes(struct memstick_dev *card)
956 {
957 	struct mspro_block_data *msb = memstick_get_drvdata(card);
958 	struct mspro_param_register param = {
959 		.system = msb->system,
960 		.data_count = cpu_to_be16(1),
961 		.data_address = 0,
962 		.tpc_param = 0
963 	};
964 	struct mspro_attribute *attr = NULL;
965 	struct mspro_sys_attr *s_attr = NULL;
966 	unsigned char *buffer = NULL;
967 	int cnt, rc, attr_count;
968 	unsigned int addr;
969 	unsigned short page_count;
970 
971 	attr = kmalloc(msb->page_size, GFP_KERNEL);
972 	if (!attr)
973 		return -ENOMEM;
974 
975 	sg_init_one(&msb->req_sg[0], attr, msb->page_size);
976 	msb->seg_count = 1;
977 	msb->current_seg = 0;
978 	msb->current_page = 0;
979 	msb->data_dir = READ;
980 	msb->transfer_cmd = MSPRO_CMD_READ_ATRB;
981 
982 	card->next_request = h_mspro_block_req_init;
983 	msb->mrq_handler = h_mspro_block_transfer_data;
984 	memstick_init_req(&card->current_mrq, MS_TPC_WRITE_REG, &param,
985 			  sizeof(param));
986 	memstick_new_req(card->host);
987 	wait_for_completion(&card->mrq_complete);
988 	if (card->current_mrq.error) {
989 		rc = card->current_mrq.error;
990 		goto out_free_attr;
991 	}
992 
993 	if (be16_to_cpu(attr->signature) != MSPRO_BLOCK_SIGNATURE) {
994 		printk(KERN_ERR "%s: unrecognized device signature %x\n",
995 		       card->dev.bus_id, be16_to_cpu(attr->signature));
996 		rc = -ENODEV;
997 		goto out_free_attr;
998 	}
999 
1000 	if (attr->count > MSPRO_BLOCK_MAX_ATTRIBUTES) {
1001 		printk(KERN_WARNING "%s: way too many attribute entries\n",
1002 		       card->dev.bus_id);
1003 		attr_count = MSPRO_BLOCK_MAX_ATTRIBUTES;
1004 	} else
1005 		attr_count = attr->count;
1006 
1007 	msb->attr_group.attrs = kzalloc((attr_count + 1)
1008 					* sizeof(struct attribute),
1009 					GFP_KERNEL);
1010 	if (!msb->attr_group.attrs) {
1011 		rc = -ENOMEM;
1012 		goto out_free_attr;
1013 	}
1014 	msb->attr_group.name = "media_attributes";
1015 
1016 	buffer = kmalloc(msb->page_size, GFP_KERNEL);
1017 	if (!buffer) {
1018 		rc = -ENOMEM;
1019 		goto out_free_attr;
1020 	}
1021 	memcpy(buffer, (char *)attr, msb->page_size);
1022 	page_count = 1;
1023 
1024 	for (cnt = 0; cnt < attr_count; ++cnt) {
1025 		s_attr = kzalloc(sizeof(struct mspro_sys_attr), GFP_KERNEL);
1026 		if (!s_attr) {
1027 			rc = -ENOMEM;
1028 			goto out_free_buffer;
1029 		}
1030 
1031 		msb->attr_group.attrs[cnt] = &s_attr->dev_attr.attr;
1032 		addr = be32_to_cpu(attr->entries[cnt].address);
1033 		rc = be32_to_cpu(attr->entries[cnt].size);
1034 		dev_dbg(&card->dev, "adding attribute %d: id %x, address %x, "
1035 			"size %x\n", cnt, attr->entries[cnt].id, addr, rc);
1036 		s_attr->id = attr->entries[cnt].id;
1037 		if (mspro_block_attr_name(s_attr->id))
1038 			snprintf(s_attr->name, sizeof(s_attr->name), "%s",
1039 				 mspro_block_attr_name(attr->entries[cnt].id));
1040 		else
1041 			snprintf(s_attr->name, sizeof(s_attr->name),
1042 				 "attr_x%02x", attr->entries[cnt].id);
1043 
1044 		s_attr->dev_attr.attr.name = s_attr->name;
1045 		s_attr->dev_attr.attr.mode = S_IRUGO;
1046 		s_attr->dev_attr.show = mspro_block_attr_show(s_attr->id);
1047 
1048 		if (!rc)
1049 			continue;
1050 
1051 		s_attr->size = rc;
1052 		s_attr->data = kmalloc(rc, GFP_KERNEL);
1053 		if (!s_attr->data) {
1054 			rc = -ENOMEM;
1055 			goto out_free_buffer;
1056 		}
1057 
1058 		if (((addr / msb->page_size)
1059 		     == be32_to_cpu(param.data_address))
1060 		    && (((addr + rc - 1) / msb->page_size)
1061 			== be32_to_cpu(param.data_address))) {
1062 			memcpy(s_attr->data, buffer + addr % msb->page_size,
1063 			       rc);
1064 			continue;
1065 		}
1066 
1067 		if (page_count <= (rc / msb->page_size)) {
1068 			kfree(buffer);
1069 			page_count = (rc / msb->page_size) + 1;
1070 			buffer = kmalloc(page_count * msb->page_size,
1071 					 GFP_KERNEL);
1072 			if (!buffer) {
1073 				rc = -ENOMEM;
1074 				goto out_free_attr;
1075 			}
1076 		}
1077 
1078 		param.system = msb->system;
1079 		param.data_count = cpu_to_be16((rc / msb->page_size) + 1);
1080 		param.data_address = cpu_to_be32(addr / msb->page_size);
1081 		param.tpc_param = 0;
1082 
1083 		sg_init_one(&msb->req_sg[0], buffer,
1084 			    be16_to_cpu(param.data_count) * msb->page_size);
1085 		msb->seg_count = 1;
1086 		msb->current_seg = 0;
1087 		msb->current_page = 0;
1088 		msb->data_dir = READ;
1089 		msb->transfer_cmd = MSPRO_CMD_READ_ATRB;
1090 
1091 		dev_dbg(&card->dev, "reading attribute pages %x, %x\n",
1092 			be32_to_cpu(param.data_address),
1093 			be16_to_cpu(param.data_count));
1094 
1095 		card->next_request = h_mspro_block_req_init;
1096 		msb->mrq_handler = h_mspro_block_transfer_data;
1097 		memstick_init_req(&card->current_mrq, MS_TPC_WRITE_REG,
1098 				  (char *)&param, sizeof(param));
1099 		memstick_new_req(card->host);
1100 		wait_for_completion(&card->mrq_complete);
1101 		if (card->current_mrq.error) {
1102 			rc = card->current_mrq.error;
1103 			goto out_free_buffer;
1104 		}
1105 
1106 		memcpy(s_attr->data, buffer + addr % msb->page_size, rc);
1107 	}
1108 
1109 	rc = 0;
1110 out_free_buffer:
1111 	kfree(buffer);
1112 out_free_attr:
1113 	kfree(attr);
1114 	return rc;
1115 }
1116 
1117 static int mspro_block_init_card(struct memstick_dev *card)
1118 {
1119 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1120 	struct memstick_host *host = card->host;
1121 	int rc = 0;
1122 
1123 	msb->system = MEMSTICK_SYS_SERIAL;
1124 	card->reg_addr.r_offset = offsetof(struct mspro_register, status);
1125 	card->reg_addr.r_length = sizeof(struct ms_status_register);
1126 	card->reg_addr.w_offset = offsetof(struct mspro_register, param);
1127 	card->reg_addr.w_length = sizeof(struct mspro_param_register);
1128 
1129 	if (memstick_set_rw_addr(card))
1130 		return -EIO;
1131 
1132 	msb->caps = host->caps;
1133 
1134 	msleep(150);
1135 	rc = mspro_block_wait_for_ced(card);
1136 	if (rc)
1137 		return rc;
1138 
1139 	rc = mspro_block_switch_interface(card);
1140 	if (rc)
1141 		return rc;
1142 
1143 	dev_dbg(&card->dev, "card activated\n");
1144 	if (msb->system != MEMSTICK_SYS_SERIAL)
1145 		msb->caps |= MEMSTICK_CAP_AUTO_GET_INT;
1146 
1147 	card->next_request = h_mspro_block_req_init;
1148 	msb->mrq_handler = h_mspro_block_get_ro;
1149 	memstick_init_req(&card->current_mrq, MS_TPC_READ_REG, NULL,
1150 			  sizeof(struct ms_status_register));
1151 	memstick_new_req(card->host);
1152 	wait_for_completion(&card->mrq_complete);
1153 	if (card->current_mrq.error)
1154 		return card->current_mrq.error;
1155 
1156 	dev_dbg(&card->dev, "card r/w status %d\n", msb->read_only ? 0 : 1);
1157 
1158 	msb->page_size = 512;
1159 	rc = mspro_block_read_attributes(card);
1160 	if (rc)
1161 		return rc;
1162 
1163 	dev_dbg(&card->dev, "attributes loaded\n");
1164 	return 0;
1165 
1166 }
1167 
1168 static int mspro_block_init_disk(struct memstick_dev *card)
1169 {
1170 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1171 	struct memstick_host *host = card->host;
1172 	struct mspro_devinfo *dev_info = NULL;
1173 	struct mspro_sys_info *sys_info = NULL;
1174 	struct mspro_sys_attr *s_attr = NULL;
1175 	int rc, disk_id;
1176 	u64 limit = BLK_BOUNCE_HIGH;
1177 	unsigned long capacity;
1178 
1179 	if (host->dev.dma_mask && *(host->dev.dma_mask))
1180 		limit = *(host->dev.dma_mask);
1181 
1182 	for (rc = 0; msb->attr_group.attrs[rc]; ++rc) {
1183 		s_attr = mspro_from_sysfs_attr(msb->attr_group.attrs[rc]);
1184 
1185 		if (s_attr->id == MSPRO_BLOCK_ID_DEVINFO)
1186 			dev_info = s_attr->data;
1187 		else if (s_attr->id == MSPRO_BLOCK_ID_SYSINFO)
1188 			sys_info = s_attr->data;
1189 	}
1190 
1191 	if (!dev_info || !sys_info)
1192 		return -ENODEV;
1193 
1194 	msb->cylinders = be16_to_cpu(dev_info->cylinders);
1195 	msb->heads = be16_to_cpu(dev_info->heads);
1196 	msb->sectors_per_track = be16_to_cpu(dev_info->sectors_per_track);
1197 
1198 	msb->page_size = be16_to_cpu(sys_info->unit_size);
1199 
1200 	if (!idr_pre_get(&mspro_block_disk_idr, GFP_KERNEL))
1201 		return -ENOMEM;
1202 
1203 	mutex_lock(&mspro_block_disk_lock);
1204 	rc = idr_get_new(&mspro_block_disk_idr, card, &disk_id);
1205 	mutex_unlock(&mspro_block_disk_lock);
1206 
1207 	if (rc)
1208 		return rc;
1209 
1210 	if ((disk_id << MSPRO_BLOCK_PART_SHIFT) > 255) {
1211 		rc = -ENOSPC;
1212 		goto out_release_id;
1213 	}
1214 
1215 	msb->disk = alloc_disk(1 << MSPRO_BLOCK_PART_SHIFT);
1216 	if (!msb->disk) {
1217 		rc = -ENOMEM;
1218 		goto out_release_id;
1219 	}
1220 
1221 	msb->queue = blk_init_queue(mspro_block_submit_req, &msb->q_lock);
1222 	if (!msb->queue) {
1223 		rc = -ENOMEM;
1224 		goto out_put_disk;
1225 	}
1226 
1227 	msb->queue->queuedata = card;
1228 	blk_queue_prep_rq(msb->queue, mspro_block_prepare_req);
1229 
1230 	blk_queue_bounce_limit(msb->queue, limit);
1231 	blk_queue_max_sectors(msb->queue, MSPRO_BLOCK_MAX_PAGES);
1232 	blk_queue_max_phys_segments(msb->queue, MSPRO_BLOCK_MAX_SEGS);
1233 	blk_queue_max_hw_segments(msb->queue, MSPRO_BLOCK_MAX_SEGS);
1234 	blk_queue_max_segment_size(msb->queue,
1235 				   MSPRO_BLOCK_MAX_PAGES * msb->page_size);
1236 
1237 	msb->disk->major = major;
1238 	msb->disk->first_minor = disk_id << MSPRO_BLOCK_PART_SHIFT;
1239 	msb->disk->fops = &ms_block_bdops;
1240 	msb->usage_count = 1;
1241 	msb->disk->private_data = msb;
1242 	msb->disk->queue = msb->queue;
1243 	msb->disk->driverfs_dev = &card->dev;
1244 
1245 	sprintf(msb->disk->disk_name, "mspblk%d", disk_id);
1246 
1247 	blk_queue_hardsect_size(msb->queue, msb->page_size);
1248 
1249 	capacity = be16_to_cpu(sys_info->user_block_count);
1250 	capacity *= be16_to_cpu(sys_info->block_size);
1251 	capacity *= msb->page_size >> 9;
1252 	set_capacity(msb->disk, capacity);
1253 	dev_dbg(&card->dev, "capacity set %ld\n", capacity);
1254 
1255 	add_disk(msb->disk);
1256 	msb->active = 1;
1257 	return 0;
1258 
1259 out_put_disk:
1260 	put_disk(msb->disk);
1261 out_release_id:
1262 	mutex_lock(&mspro_block_disk_lock);
1263 	idr_remove(&mspro_block_disk_idr, disk_id);
1264 	mutex_unlock(&mspro_block_disk_lock);
1265 	return rc;
1266 }
1267 
1268 static void mspro_block_data_clear(struct mspro_block_data *msb)
1269 {
1270 	int cnt;
1271 	struct mspro_sys_attr *s_attr;
1272 
1273 	if (msb->attr_group.attrs) {
1274 		for (cnt = 0; msb->attr_group.attrs[cnt]; ++cnt) {
1275 			s_attr = mspro_from_sysfs_attr(msb->attr_group
1276 							   .attrs[cnt]);
1277 			kfree(s_attr->data);
1278 			kfree(s_attr);
1279 		}
1280 		kfree(msb->attr_group.attrs);
1281 	}
1282 
1283 	msb->card = NULL;
1284 }
1285 
1286 static int mspro_block_check_card(struct memstick_dev *card)
1287 {
1288 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1289 
1290 	return (msb->active == 1);
1291 }
1292 
1293 static int mspro_block_probe(struct memstick_dev *card)
1294 {
1295 	struct mspro_block_data *msb;
1296 	int rc = 0;
1297 
1298 	msb = kzalloc(sizeof(struct mspro_block_data), GFP_KERNEL);
1299 	if (!msb)
1300 		return -ENOMEM;
1301 	memstick_set_drvdata(card, msb);
1302 	msb->card = card;
1303 	spin_lock_init(&msb->q_lock);
1304 
1305 	rc = mspro_block_init_card(card);
1306 
1307 	if (rc)
1308 		goto out_free;
1309 
1310 	rc = sysfs_create_group(&card->dev.kobj, &msb->attr_group);
1311 	if (rc)
1312 		goto out_free;
1313 
1314 	rc = mspro_block_init_disk(card);
1315 	if (!rc) {
1316 		card->check = mspro_block_check_card;
1317 		card->stop = mspro_block_stop;
1318 		card->start = mspro_block_start;
1319 		return 0;
1320 	}
1321 
1322 	sysfs_remove_group(&card->dev.kobj, &msb->attr_group);
1323 out_free:
1324 	memstick_set_drvdata(card, NULL);
1325 	mspro_block_data_clear(msb);
1326 	kfree(msb);
1327 	return rc;
1328 }
1329 
1330 static void mspro_block_remove(struct memstick_dev *card)
1331 {
1332 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1333 	unsigned long flags;
1334 
1335 	del_gendisk(msb->disk);
1336 	dev_dbg(&card->dev, "mspro block remove\n");
1337 	spin_lock_irqsave(&msb->q_lock, flags);
1338 	msb->eject = 1;
1339 	blk_start_queue(msb->queue);
1340 	spin_unlock_irqrestore(&msb->q_lock, flags);
1341 
1342 	blk_cleanup_queue(msb->queue);
1343 	msb->queue = NULL;
1344 
1345 	sysfs_remove_group(&card->dev.kobj, &msb->attr_group);
1346 
1347 	mutex_lock(&mspro_block_disk_lock);
1348 	mspro_block_data_clear(msb);
1349 	mutex_unlock(&mspro_block_disk_lock);
1350 
1351 	mspro_block_disk_release(msb->disk);
1352 	memstick_set_drvdata(card, NULL);
1353 }
1354 
1355 #ifdef CONFIG_PM
1356 
1357 static int mspro_block_suspend(struct memstick_dev *card, pm_message_t state)
1358 {
1359 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1360 	unsigned long flags;
1361 
1362 	spin_lock_irqsave(&msb->q_lock, flags);
1363 	blk_stop_queue(msb->queue);
1364 	msb->active = 0;
1365 	spin_unlock_irqrestore(&msb->q_lock, flags);
1366 
1367 	return 0;
1368 }
1369 
1370 static int mspro_block_resume(struct memstick_dev *card)
1371 {
1372 	struct mspro_block_data *msb = memstick_get_drvdata(card);
1373 	unsigned long flags;
1374 	int rc = 0;
1375 
1376 #ifdef CONFIG_MEMSTICK_UNSAFE_RESUME
1377 
1378 	struct mspro_block_data *new_msb;
1379 	struct memstick_host *host = card->host;
1380 	struct mspro_sys_attr *s_attr, *r_attr;
1381 	unsigned char cnt;
1382 
1383 	mutex_lock(&host->lock);
1384 	new_msb = kzalloc(sizeof(struct mspro_block_data), GFP_KERNEL);
1385 	if (!new_msb) {
1386 		rc = -ENOMEM;
1387 		goto out_unlock;
1388 	}
1389 
1390 	new_msb->card = card;
1391 	memstick_set_drvdata(card, new_msb);
1392 	if (mspro_block_init_card(card))
1393 		goto out_free;
1394 
1395 	for (cnt = 0; new_msb->attr_group.attrs[cnt]
1396 		      && msb->attr_group.attrs[cnt]; ++cnt) {
1397 		s_attr = mspro_from_sysfs_attr(new_msb->attr_group.attrs[cnt]);
1398 		r_attr = mspro_from_sysfs_attr(msb->attr_group.attrs[cnt]);
1399 
1400 		if (s_attr->id == MSPRO_BLOCK_ID_SYSINFO
1401 		    && r_attr->id == s_attr->id) {
1402 			if (memcmp(s_attr->data, r_attr->data, s_attr->size))
1403 				break;
1404 
1405 			msb->active = 1;
1406 			break;
1407 		}
1408 	}
1409 
1410 out_free:
1411 	memstick_set_drvdata(card, msb);
1412 	mspro_block_data_clear(new_msb);
1413 	kfree(new_msb);
1414 out_unlock:
1415 	mutex_unlock(&host->lock);
1416 
1417 #endif /* CONFIG_MEMSTICK_UNSAFE_RESUME */
1418 
1419 	spin_lock_irqsave(&msb->q_lock, flags);
1420 	blk_start_queue(msb->queue);
1421 	spin_unlock_irqrestore(&msb->q_lock, flags);
1422 	return rc;
1423 }
1424 
1425 #else
1426 
1427 #define mspro_block_suspend NULL
1428 #define mspro_block_resume NULL
1429 
1430 #endif /* CONFIG_PM */
1431 
1432 static struct memstick_device_id mspro_block_id_tbl[] = {
1433 	{MEMSTICK_MATCH_ALL, MEMSTICK_TYPE_PRO, MEMSTICK_CATEGORY_STORAGE_DUO,
1434 	 MEMSTICK_CLASS_DUO},
1435 	{}
1436 };
1437 
1438 
1439 static struct memstick_driver mspro_block_driver = {
1440 	.driver = {
1441 		.name  = DRIVER_NAME,
1442 		.owner = THIS_MODULE
1443 	},
1444 	.id_table = mspro_block_id_tbl,
1445 	.probe    = mspro_block_probe,
1446 	.remove   = mspro_block_remove,
1447 	.suspend  = mspro_block_suspend,
1448 	.resume   = mspro_block_resume
1449 };
1450 
1451 static int __init mspro_block_init(void)
1452 {
1453 	int rc = -ENOMEM;
1454 
1455 	rc = register_blkdev(major, DRIVER_NAME);
1456 	if (rc < 0) {
1457 		printk(KERN_ERR DRIVER_NAME ": failed to register "
1458 		       "major %d, error %d\n", major, rc);
1459 		return rc;
1460 	}
1461 	if (!major)
1462 		major = rc;
1463 
1464 	rc = memstick_register_driver(&mspro_block_driver);
1465 	if (rc)
1466 		unregister_blkdev(major, DRIVER_NAME);
1467 	return rc;
1468 }
1469 
1470 static void __exit mspro_block_exit(void)
1471 {
1472 	memstick_unregister_driver(&mspro_block_driver);
1473 	unregister_blkdev(major, DRIVER_NAME);
1474 	idr_destroy(&mspro_block_disk_idr);
1475 }
1476 
1477 module_init(mspro_block_init);
1478 module_exit(mspro_block_exit);
1479 
1480 MODULE_LICENSE("GPL");
1481 MODULE_AUTHOR("Alex Dubov");
1482 MODULE_DESCRIPTION("Sony MemoryStickPro block device driver");
1483 MODULE_DEVICE_TABLE(memstick, mspro_block_id_tbl);
1484