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