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