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