drivers/usb/storage/realtek_cr.c

   1 /* Driver for Realtek RTS51xx USB card reader
   2  *
   3  * Copyright(c) 2009 Realtek Semiconductor Corp. All rights reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; either version 2, or (at your option) any
   8  * later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, see <http://www.gnu.org/licenses/>.
  17  *
  18  * Author:
  19  *   wwang (wei_wang@realsil.com.cn)
  20  *   No. 450, Shenhu Road, Suzhou Industry Park, Suzhou, China
  21  */
  22
  23 #include <linux/module.h>
  24 #include <linux/blkdev.h>
  25 #include <linux/kthread.h>
  26 #include <linux/sched.h>
  27 #include <linux/kernel.h>
  28
  29 #include <scsi/scsi.h>
  30 #include <scsi/scsi_cmnd.h>
  31 #include <scsi/scsi_device.h>
  32 #include <linux/cdrom.h>
  33
  34 #include <linux/usb.h>
  35 #include <linux/slab.h>
  36 #include <linux/usb_usual.h>
  37
  38 #include "usb.h"
  39 #include "transport.h"
  40 #include "protocol.h"
  41 #include "debug.h"
  42 #include "scsiglue.h"
  43
  44 #define DRV_NAME "ums-realtek"
  45
  46 MODULE_DESCRIPTION("Driver for Realtek USB Card Reader");
  47 MODULE_AUTHOR("wwang <wei_wang@realsil.com.cn>");
  48 MODULE_LICENSE("GPL");
  49 MODULE_VERSION("1.03");
  50
  51 static int auto_delink_en = 1;
  52 module_param(auto_delink_en, int, S_IRUGO | S_IWUSR);
  53 MODULE_PARM_DESC(auto_delink_en, "enable auto delink");
  54
  55 #ifdef CONFIG_REALTEK_AUTOPM
  56 static int ss_en = 1;
  57 module_param(ss_en, int, S_IRUGO | S_IWUSR);
  58 MODULE_PARM_DESC(ss_en, "enable selective suspend");
  59
  60 static int ss_delay = 50;
  61 module_param(ss_delay, int, S_IRUGO | S_IWUSR);
  62 MODULE_PARM_DESC(ss_delay,
  63                  "seconds to delay before entering selective suspend");
  64
  65 enum RTS51X_STAT {
  66         RTS51X_STAT_INIT,
  67         RTS51X_STAT_IDLE,
  68         RTS51X_STAT_RUN,
  69         RTS51X_STAT_SS
  70 };
  71
  72 #define POLLING_INTERVAL        50
  73
  74 #define rts51x_set_stat(chip, stat)     \
  75         ((chip)->state = (enum RTS51X_STAT)(stat))
  76 #define rts51x_get_stat(chip)           ((chip)->state)
  77
  78 #define SET_LUN_READY(chip, lun)        ((chip)->lun_ready |= ((u8)1 << (lun)))
  79 #define CLR_LUN_READY(chip, lun)        ((chip)->lun_ready &= ~((u8)1 << (lun)))
  80 #define TST_LUN_READY(chip, lun)        ((chip)->lun_ready & ((u8)1 << (lun)))
  81
  82 #endif
  83
  84 struct rts51x_status {
  85         u16 vid;
  86         u16 pid;
  87         u8 cur_lun;
  88         u8 card_type;
  89         u8 total_lun;
  90         u16 fw_ver;
  91         u8 phy_exist;
  92         u8 multi_flag;
  93         u8 multi_card;
  94         u8 log_exist;
  95         union {
  96                 u8 detailed_type1;
  97                 u8 detailed_type2;
  98         } detailed_type;
  99         u8 function[2];
 100 };
 101
 102 struct rts51x_chip {
 103         u16 vendor_id;
 104         u16 product_id;
 105         char max_lun;
 106
 107         struct rts51x_status *status;
 108         int status_len;
 109
 110         u32 flag;
 111         struct us_data *us;
 112
 113 #ifdef CONFIG_REALTEK_AUTOPM
 114         struct timer_list rts51x_suspend_timer;
 115         unsigned long timer_expires;
 116         int pwr_state;
 117         u8 lun_ready;
 118         enum RTS51X_STAT state;
 119         int support_auto_delink;
 120 #endif
 121         /* used to back up the protocol chosen in probe1 phase */
 122         proto_cmnd proto_handler_backup;
 123 };
 124
 125 /* flag definition */
 126 #define FLIDX_AUTO_DELINK               0x01
 127
 128 #define SCSI_LUN(srb)                   ((srb)->device->lun)
 129
 130 /* Bit Operation */
 131 #define SET_BIT(data, idx)              ((data) |= 1 << (idx))
 132 #define CLR_BIT(data, idx)              ((data) &= ~(1 << (idx)))
 133 #define CHK_BIT(data, idx)              ((data) & (1 << (idx)))
 134
 135 #define SET_AUTO_DELINK(chip)           ((chip)->flag |= FLIDX_AUTO_DELINK)
 136 #define CLR_AUTO_DELINK(chip)           ((chip)->flag &= ~FLIDX_AUTO_DELINK)
 137 #define CHK_AUTO_DELINK(chip)           ((chip)->flag & FLIDX_AUTO_DELINK)
 138
 139 #define RTS51X_GET_VID(chip)            ((chip)->vendor_id)
 140 #define RTS51X_GET_PID(chip)            ((chip)->product_id)
 141
 142 #define VENDOR_ID(chip)                 ((chip)->status[0].vid)
 143 #define PRODUCT_ID(chip)                ((chip)->status[0].pid)
 144 #define FW_VERSION(chip)                ((chip)->status[0].fw_ver)
 145 #define STATUS_LEN(chip)                ((chip)->status_len)
 146
 147 #define STATUS_SUCCESS          0
 148 #define STATUS_FAIL             1
 149
 150 /* Check card reader function */
 151 #define SUPPORT_DETAILED_TYPE1(chip)    \
 152                 CHK_BIT((chip)->status[0].function[0], 1)
 153 #define SUPPORT_OT(chip)                \
 154                 CHK_BIT((chip)->status[0].function[0], 2)
 155 #define SUPPORT_OC(chip)                \
 156                 CHK_BIT((chip)->status[0].function[0], 3)
 157 #define SUPPORT_AUTO_DELINK(chip)       \
 158                 CHK_BIT((chip)->status[0].function[0], 4)
 159 #define SUPPORT_SDIO(chip)              \
 160                 CHK_BIT((chip)->status[0].function[1], 0)
 161 #define SUPPORT_DETAILED_TYPE2(chip)    \
 162                 CHK_BIT((chip)->status[0].function[1], 1)
 163
 164 #define CHECK_PID(chip, pid)            (RTS51X_GET_PID(chip) == (pid))
 165 #define CHECK_FW_VER(chip, fw_ver)      (FW_VERSION(chip) == (fw_ver))
 166 #define CHECK_ID(chip, pid, fw_ver)     \
 167                 (CHECK_PID((chip), (pid)) && CHECK_FW_VER((chip), (fw_ver)))
 168
 169 static int init_realtek_cr(struct us_data *us);
 170
 171 /*
 172  * The table of devices
 173  */
 174 #define UNUSUAL_DEV(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax, \
 175                     vendorName, productName, useProtocol, useTransport, \
 176                     initFunction, flags) \
 177 {\
 178         USB_DEVICE_VER(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax), \
 179         .driver_info = (flags) \
 180 }
 181
 182 static const struct usb_device_id realtek_cr_ids[] = {
 183 #       include "unusual_realtek.h"
 184         {}                      /* Terminating entry */
 185 };
 186
 187 MODULE_DEVICE_TABLE(usb, realtek_cr_ids);
 188
 189 #undef UNUSUAL_DEV
 190
 191 /*
 192  * The flags table
 193  */
 194 #define UNUSUAL_DEV(idVendor, idProduct, bcdDeviceMin, bcdDeviceMax, \
 195                     vendor_name, product_name, use_protocol, use_transport, \
 196                     init_function, Flags) \
 197 { \
 198         .vendorName = vendor_name,      \
 199         .productName = product_name,    \
 200         .useProtocol = use_protocol,    \
 201         .useTransport = use_transport,  \
 202         .initFunction = init_function,  \
 203 }
 204
 205 static struct us_unusual_dev realtek_cr_unusual_dev_list[] = {
 206 #       include "unusual_realtek.h"
 207         {}                      /* Terminating entry */
 208 };
 209
 210 #undef UNUSUAL_DEV
 211
 212 static int rts51x_bulk_transport(struct us_data *us, u8 lun,
 213                                  u8 *cmd, int cmd_len, u8 *buf, int buf_len,
 214                                  enum dma_data_direction dir, int *act_len)
 215 {
 216         struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *)us->iobuf;
 217         struct bulk_cs_wrap *bcs = (struct bulk_cs_wrap *)us->iobuf;
 218         int result;
 219         unsigned int residue;
 220         unsigned int cswlen;
 221         unsigned int cbwlen = US_BULK_CB_WRAP_LEN;
 222
 223         /* set up the command wrapper */
 224         bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
 225         bcb->DataTransferLength = cpu_to_le32(buf_len);
 226         bcb->Flags = (dir == DMA_FROM_DEVICE) ? US_BULK_FLAG_IN : 0;
 227         bcb->Tag = ++us->tag;
 228         bcb->Lun = lun;
 229         bcb->Length = cmd_len;
 230
 231         /* copy the command payload */
 232         memset(bcb->CDB, 0, sizeof(bcb->CDB));
 233         memcpy(bcb->CDB, cmd, bcb->Length);
 234
 235         /* send it to out endpoint */
 236         result = usb_stor_bulk_transfer_buf(us, us->send_bulk_pipe,
 237                                             bcb, cbwlen, NULL);
 238         if (result != USB_STOR_XFER_GOOD)
 239                 return USB_STOR_TRANSPORT_ERROR;
 240
 241         /* DATA STAGE */
 242         /* send/receive data payload, if there is any */
 243
 244         if (buf && buf_len) {
 245                 unsigned int pipe = (dir == DMA_FROM_DEVICE) ?
 246                     us->recv_bulk_pipe : us->send_bulk_pipe;
 247                 result = usb_stor_bulk_transfer_buf(us, pipe,
 248                                                     buf, buf_len, NULL);
 249                 if (result == USB_STOR_XFER_ERROR)
 250                         return USB_STOR_TRANSPORT_ERROR;
 251         }
 252
 253         /* get CSW for device status */
 254         result = usb_stor_bulk_transfer_buf(us, us->recv_bulk_pipe,
 255                                             bcs, US_BULK_CS_WRAP_LEN, &cswlen);
 256         if (result != USB_STOR_XFER_GOOD)
 257                 return USB_STOR_TRANSPORT_ERROR;
 258
 259         /* check bulk status */
 260         if (bcs->Signature != cpu_to_le32(US_BULK_CS_SIGN)) {
 261                 usb_stor_dbg(us, "Signature mismatch: got %08X, expecting %08X\n",
 262                              le32_to_cpu(bcs->Signature), US_BULK_CS_SIGN);
 263                 return USB_STOR_TRANSPORT_ERROR;
 264         }
 265
 266         residue = bcs->Residue;
 267         if (bcs->Tag != us->tag)
 268                 return USB_STOR_TRANSPORT_ERROR;
 269
 270         /* try to compute the actual residue, based on how much data
 271          * was really transferred and what the device tells us */
 272         if (residue)
 273                 residue = residue < buf_len ? residue : buf_len;
 274
 275         if (act_len)
 276                 *act_len = buf_len - residue;
 277
 278         /* based on the status code, we report good or bad */
 279         switch (bcs->Status) {
 280         case US_BULK_STAT_OK:
 281                 /* command good -- note that data could be short */
 282                 return USB_STOR_TRANSPORT_GOOD;
 283
 284         case US_BULK_STAT_FAIL:
 285                 /* command failed */
 286                 return USB_STOR_TRANSPORT_FAILED;
 287
 288         case US_BULK_STAT_PHASE:
 289                 /* phase error -- note that a transport reset will be
 290                  * invoked by the invoke_transport() function
 291                  */
 292                 return USB_STOR_TRANSPORT_ERROR;
 293         }
 294
 295         /* we should never get here, but if we do, we're in trouble */
 296         return USB_STOR_TRANSPORT_ERROR;
 297 }
 298
 299 static int rts51x_bulk_transport_special(struct us_data *us, u8 lun,
 300                                  u8 *cmd, int cmd_len, u8 *buf, int buf_len,
 301                                  enum dma_data_direction dir, int *act_len)
 302 {
 303         struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
 304         struct bulk_cs_wrap *bcs = (struct bulk_cs_wrap *) us->iobuf;
 305         int result;
 306         unsigned int cswlen;
 307         unsigned int cbwlen = US_BULK_CB_WRAP_LEN;
 308
 309         /* set up the command wrapper */
 310         bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
 311         bcb->DataTransferLength = cpu_to_le32(buf_len);
 312         bcb->Flags = (dir == DMA_FROM_DEVICE) ? US_BULK_FLAG_IN : 0;
 313         bcb->Tag = ++us->tag;
 314         bcb->Lun = lun;
 315         bcb->Length = cmd_len;
 316
 317         /* copy the command payload */
 318         memset(bcb->CDB, 0, sizeof(bcb->CDB));
 319         memcpy(bcb->CDB, cmd, bcb->Length);
 320
 321         /* send it to out endpoint */
 322         result = usb_stor_bulk_transfer_buf(us, us->send_bulk_pipe,
 323                                 bcb, cbwlen, NULL);
 324         if (result != USB_STOR_XFER_GOOD)
 325                 return USB_STOR_TRANSPORT_ERROR;
 326
 327         /* DATA STAGE */
 328         /* send/receive data payload, if there is any */
 329
 330         if (buf && buf_len) {
 331                 unsigned int pipe = (dir == DMA_FROM_DEVICE) ?
 332                                 us->recv_bulk_pipe : us->send_bulk_pipe;
 333                 result = usb_stor_bulk_transfer_buf(us, pipe,
 334                                 buf, buf_len, NULL);
 335                 if (result == USB_STOR_XFER_ERROR)
 336                         return USB_STOR_TRANSPORT_ERROR;
 337         }
 338
 339         /* get CSW for device status */
 340         result = usb_bulk_msg(us->pusb_dev, us->recv_bulk_pipe, bcs,
 341                         US_BULK_CS_WRAP_LEN, &cswlen, 250);
 342         return result;
 343 }
 344
 345 /* Determine what the maximum LUN supported is */
 346 static int rts51x_get_max_lun(struct us_data *us)
 347 {
 348         int result;
 349
 350         /* issue the command */
 351         us->iobuf[0] = 0;
 352         result = usb_stor_control_msg(us, us->recv_ctrl_pipe,
 353                                       US_BULK_GET_MAX_LUN,
 354                                       USB_DIR_IN | USB_TYPE_CLASS |
 355                                       USB_RECIP_INTERFACE,
 356                                       0, us->ifnum, us->iobuf, 1, 10 * HZ);
 357
 358         usb_stor_dbg(us, "GetMaxLUN command result is %d, data is %d\n",
 359                      result, us->iobuf[0]);
 360
 361         /* if we have a successful request, return the result */
 362         if (result > 0)
 363                 return us->iobuf[0];
 364
 365         return 0;
 366 }
 367
 368 static int rts51x_read_mem(struct us_data *us, u16 addr, u8 *data, u16 len)
 369 {
 370         int retval;
 371         u8 cmnd[12] = { 0 };
 372         u8 *buf;
 373
 374         buf = kmalloc(len, GFP_NOIO);
 375         if (buf == NULL)
 376                 return USB_STOR_TRANSPORT_ERROR;
 377
 378         usb_stor_dbg(us, "addr = 0x%x, len = %d\n", addr, len);
 379
 380         cmnd[0] = 0xF0;
 381         cmnd[1] = 0x0D;
 382         cmnd[2] = (u8) (addr >> 8);
 383         cmnd[3] = (u8) addr;
 384         cmnd[4] = (u8) (len >> 8);
 385         cmnd[5] = (u8) len;
 386
 387         retval = rts51x_bulk_transport(us, 0, cmnd, 12,
 388                                        buf, len, DMA_FROM_DEVICE, NULL);
 389         if (retval != USB_STOR_TRANSPORT_GOOD) {
 390                 kfree(buf);
 391                 return -EIO;
 392         }
 393
 394         memcpy(data, buf, len);
 395         kfree(buf);
 396         return 0;
 397 }
 398
 399 static int rts51x_write_mem(struct us_data *us, u16 addr, u8 *data, u16 len)
 400 {
 401         int retval;
 402         u8 cmnd[12] = { 0 };
 403         u8 *buf;
 404
 405         buf = kmemdup(data, len, GFP_NOIO);
 406         if (buf == NULL)
 407                 return USB_STOR_TRANSPORT_ERROR;
 408
 409         usb_stor_dbg(us, "addr = 0x%x, len = %d\n", addr, len);
 410
 411         cmnd[0] = 0xF0;
 412         cmnd[1] = 0x0E;
 413         cmnd[2] = (u8) (addr >> 8);
 414         cmnd[3] = (u8) addr;
 415         cmnd[4] = (u8) (len >> 8);
 416         cmnd[5] = (u8) len;
 417
 418         retval = rts51x_bulk_transport(us, 0, cmnd, 12,
 419                                        buf, len, DMA_TO_DEVICE, NULL);
 420         kfree(buf);
 421         if (retval != USB_STOR_TRANSPORT_GOOD)
 422                 return -EIO;
 423
 424         return 0;
 425 }
 426
 427 static int rts51x_read_status(struct us_data *us,
 428                               u8 lun, u8 *status, int len, int *actlen)
 429 {
 430         int retval;
 431         u8 cmnd[12] = { 0 };
 432         u8 *buf;
 433
 434         buf = kmalloc(len, GFP_NOIO);
 435         if (buf == NULL)
 436                 return USB_STOR_TRANSPORT_ERROR;
 437
 438         usb_stor_dbg(us, "lun = %d\n", lun);
 439
 440         cmnd[0] = 0xF0;
 441         cmnd[1] = 0x09;
 442
 443         retval = rts51x_bulk_transport(us, lun, cmnd, 12,
 444                                        buf, len, DMA_FROM_DEVICE, actlen);
 445         if (retval != USB_STOR_TRANSPORT_GOOD) {
 446                 kfree(buf);
 447                 return -EIO;
 448         }
 449
 450         memcpy(status, buf, len);
 451         kfree(buf);
 452         return 0;
 453 }
 454
 455 static int rts51x_check_status(struct us_data *us, u8 lun)
 456 {
 457         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 458         int retval;
 459         u8 buf[16];
 460
 461         retval = rts51x_read_status(us, lun, buf, 16, &(chip->status_len));
 462         if (retval != STATUS_SUCCESS)
 463                 return -EIO;
 464
 465         usb_stor_dbg(us, "chip->status_len = %d\n", chip->status_len);
 466
 467         chip->status[lun].vid = ((u16) buf[0] << 8) | buf[1];
 468         chip->status[lun].pid = ((u16) buf[2] << 8) | buf[3];
 469         chip->status[lun].cur_lun = buf[4];
 470         chip->status[lun].card_type = buf[5];
 471         chip->status[lun].total_lun = buf[6];
 472         chip->status[lun].fw_ver = ((u16) buf[7] << 8) | buf[8];
 473         chip->status[lun].phy_exist = buf[9];
 474         chip->status[lun].multi_flag = buf[10];
 475         chip->status[lun].multi_card = buf[11];
 476         chip->status[lun].log_exist = buf[12];
 477         if (chip->status_len == 16) {
 478                 chip->status[lun].detailed_type.detailed_type1 = buf[13];
 479                 chip->status[lun].function[0] = buf[14];
 480                 chip->status[lun].function[1] = buf[15];
 481         }
 482
 483         return 0;
 484 }
 485
 486 static int enable_oscillator(struct us_data *us)
 487 {
 488         int retval;
 489         u8 value;
 490
 491         retval = rts51x_read_mem(us, 0xFE77, &value, 1);
 492         if (retval < 0)
 493                 return -EIO;
 494
 495         value |= 0x04;
 496         retval = rts51x_write_mem(us, 0xFE77, &value, 1);
 497         if (retval < 0)
 498                 return -EIO;
 499
 500         retval = rts51x_read_mem(us, 0xFE77, &value, 1);
 501         if (retval < 0)
 502                 return -EIO;
 503
 504         if (!(value & 0x04))
 505                 return -EIO;
 506
 507         return 0;
 508 }
 509
 510 static int __do_config_autodelink(struct us_data *us, u8 *data, u16 len)
 511 {
 512         int retval;
 513         u8 cmnd[12] = {0};
 514         u8 *buf;
 515
 516         usb_stor_dbg(us, "addr = 0xfe47, len = %d\n", len);
 517
 518         buf = kmemdup(data, len, GFP_NOIO);
 519         if (!buf)
 520                 return USB_STOR_TRANSPORT_ERROR;
 521
 522         cmnd[0] = 0xF0;
 523         cmnd[1] = 0x0E;
 524         cmnd[2] = 0xfe;
 525         cmnd[3] = 0x47;
 526         cmnd[4] = (u8)(len >> 8);
 527         cmnd[5] = (u8)len;
 528
 529         retval = rts51x_bulk_transport_special(us, 0, cmnd, 12, buf, len, DMA_TO_DEVICE, NULL);
 530         kfree(buf);
 531         if (retval != USB_STOR_TRANSPORT_GOOD) {
 532                 return -EIO;
 533         }
 534
 535         return 0;
 536 }
 537
 538 static int do_config_autodelink(struct us_data *us, int enable, int force)
 539 {
 540         int retval;
 541         u8 value;
 542
 543         retval = rts51x_read_mem(us, 0xFE47, &value, 1);
 544         if (retval < 0)
 545                 return -EIO;
 546
 547         if (enable) {
 548                 if (force)
 549                         value |= 0x03;
 550                 else
 551                         value |= 0x01;
 552         } else {
 553                 value &= ~0x03;
 554         }
 555
 556         usb_stor_dbg(us, "set 0xfe47 to 0x%x\n", value);
 557
 558         /* retval = rts51x_write_mem(us, 0xFE47, &value, 1); */
 559         retval = __do_config_autodelink(us, &value, 1);
 560         if (retval < 0)
 561                 return -EIO;
 562
 563         return 0;
 564 }
 565
 566 static int config_autodelink_after_power_on(struct us_data *us)
 567 {
 568         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 569         int retval;
 570         u8 value;
 571
 572         if (!CHK_AUTO_DELINK(chip))
 573                 return 0;
 574
 575         retval = rts51x_read_mem(us, 0xFE47, &value, 1);
 576         if (retval < 0)
 577                 return -EIO;
 578
 579         if (auto_delink_en) {
 580                 CLR_BIT(value, 0);
 581                 CLR_BIT(value, 1);
 582                 SET_BIT(value, 2);
 583
 584                 if (CHECK_ID(chip, 0x0138, 0x3882))
 585                         CLR_BIT(value, 2);
 586
 587                 SET_BIT(value, 7);
 588
 589                 /* retval = rts51x_write_mem(us, 0xFE47, &value, 1); */
 590                 retval = __do_config_autodelink(us, &value, 1);
 591                 if (retval < 0)
 592                         return -EIO;
 593
 594                 retval = enable_oscillator(us);
 595                 if (retval == 0)
 596                         (void)do_config_autodelink(us, 1, 0);
 597         } else {
 598                 /* Autodelink controlled by firmware */
 599
 600                 SET_BIT(value, 2);
 601
 602                 if (CHECK_ID(chip, 0x0138, 0x3882))
 603                         CLR_BIT(value, 2);
 604
 605                 if (CHECK_ID(chip, 0x0159, 0x5889) ||
 606                     CHECK_ID(chip, 0x0138, 0x3880)) {
 607                         CLR_BIT(value, 0);
 608                         CLR_BIT(value, 7);
 609                 }
 610
 611                 /* retval = rts51x_write_mem(us, 0xFE47, &value, 1); */
 612                 retval = __do_config_autodelink(us, &value, 1);
 613                 if (retval < 0)
 614                         return -EIO;
 615
 616                 if (CHECK_ID(chip, 0x0159, 0x5888)) {
 617                         value = 0xFF;
 618                         retval = rts51x_write_mem(us, 0xFE79, &value, 1);
 619                         if (retval < 0)
 620                                 return -EIO;
 621
 622                         value = 0x01;
 623                         retval = rts51x_write_mem(us, 0x48, &value, 1);
 624                         if (retval < 0)
 625                                 return -EIO;
 626                 }
 627         }
 628
 629         return 0;
 630 }
 631
 632 #ifdef CONFIG_PM
 633 static int config_autodelink_before_power_down(struct us_data *us)
 634 {
 635         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 636         int retval;
 637         u8 value;
 638
 639         if (!CHK_AUTO_DELINK(chip))
 640                 return 0;
 641
 642         if (auto_delink_en) {
 643                 retval = rts51x_read_mem(us, 0xFE77, &value, 1);
 644                 if (retval < 0)
 645                         return -EIO;
 646
 647                 SET_BIT(value, 2);
 648                 retval = rts51x_write_mem(us, 0xFE77, &value, 1);
 649                 if (retval < 0)
 650                         return -EIO;
 651
 652                 if (CHECK_ID(chip, 0x0159, 0x5888)) {
 653                         value = 0x01;
 654                         retval = rts51x_write_mem(us, 0x48, &value, 1);
 655                         if (retval < 0)
 656                                 return -EIO;
 657                 }
 658
 659                 retval = rts51x_read_mem(us, 0xFE47, &value, 1);
 660                 if (retval < 0)
 661                         return -EIO;
 662
 663                 SET_BIT(value, 0);
 664                 if (CHECK_ID(chip, 0x0138, 0x3882))
 665                         SET_BIT(value, 2);
 666                 retval = rts51x_write_mem(us, 0xFE77, &value, 1);
 667                 if (retval < 0)
 668                         return -EIO;
 669         } else {
 670                 if (CHECK_ID(chip, 0x0159, 0x5889) ||
 671                     CHECK_ID(chip, 0x0138, 0x3880) ||
 672                     CHECK_ID(chip, 0x0138, 0x3882)) {
 673                         retval = rts51x_read_mem(us, 0xFE47, &value, 1);
 674                         if (retval < 0)
 675                                 return -EIO;
 676
 677                         if (CHECK_ID(chip, 0x0159, 0x5889) ||
 678                             CHECK_ID(chip, 0x0138, 0x3880)) {
 679                                 SET_BIT(value, 0);
 680                                 SET_BIT(value, 7);
 681                         }
 682
 683                         if (CHECK_ID(chip, 0x0138, 0x3882))
 684                                 SET_BIT(value, 2);
 685
 686                         /* retval = rts51x_write_mem(us, 0xFE47, &value, 1); */
 687                         retval = __do_config_autodelink(us, &value, 1);
 688                         if (retval < 0)
 689                                 return -EIO;
 690                 }
 691
 692                 if (CHECK_ID(chip, 0x0159, 0x5888)) {
 693                         value = 0x01;
 694                         retval = rts51x_write_mem(us, 0x48, &value, 1);
 695                         if (retval < 0)
 696                                 return -EIO;
 697                 }
 698         }
 699
 700         return 0;
 701 }
 702
 703 static void fw5895_init(struct us_data *us)
 704 {
 705         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 706         int retval;
 707         u8 val;
 708
 709         if ((PRODUCT_ID(chip) != 0x0158) || (FW_VERSION(chip) != 0x5895)) {
 710                 usb_stor_dbg(us, "Not the specified device, return immediately!\n");
 711         } else {
 712                 retval = rts51x_read_mem(us, 0xFD6F, &val, 1);
 713                 if (retval == STATUS_SUCCESS && (val & 0x1F) == 0) {
 714                         val = 0x1F;
 715                         retval = rts51x_write_mem(us, 0xFD70, &val, 1);
 716                         if (retval != STATUS_SUCCESS)
 717                                 usb_stor_dbg(us, "Write memory fail\n");
 718                 } else {
 719                         usb_stor_dbg(us, "Read memory fail, OR (val & 0x1F) != 0\n");
 720                 }
 721         }
 722 }
 723 #endif
 724
 725 #ifdef CONFIG_REALTEK_AUTOPM
 726 static void fw5895_set_mmc_wp(struct us_data *us)
 727 {
 728         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 729         int retval;
 730         u8 buf[13];
 731
 732         if ((PRODUCT_ID(chip) != 0x0158) || (FW_VERSION(chip) != 0x5895)) {
 733                 usb_stor_dbg(us, "Not the specified device, return immediately!\n");
 734         } else {
 735                 retval = rts51x_read_mem(us, 0xFD6F, buf, 1);
 736                 if (retval == STATUS_SUCCESS && (buf[0] & 0x24) == 0x24) {
 737                         /* SD Exist and SD WP */
 738                         retval = rts51x_read_mem(us, 0xD04E, buf, 1);
 739                         if (retval == STATUS_SUCCESS) {
 740                                 buf[0] |= 0x04;
 741                                 retval = rts51x_write_mem(us, 0xFD70, buf, 1);
 742                                 if (retval != STATUS_SUCCESS)
 743                                         usb_stor_dbg(us, "Write memory fail\n");
 744                         } else {
 745                                 usb_stor_dbg(us, "Read memory fail\n");
 746                         }
 747                 } else {
 748                         usb_stor_dbg(us, "Read memory fail, OR (buf[0]&0x24)!=0x24\n");
 749                 }
 750         }
 751 }
 752
 753 static void rts51x_modi_suspend_timer(struct rts51x_chip *chip)
 754 {
 755         struct us_data *us = chip->us;
 756
 757         usb_stor_dbg(us, "state:%d\n", rts51x_get_stat(chip));
 758
 759         chip->timer_expires = jiffies + msecs_to_jiffies(1000*ss_delay);
 760         mod_timer(&chip->rts51x_suspend_timer, chip->timer_expires);
 761 }
 762
 763 static void rts51x_suspend_timer_fn(unsigned long data)
 764 {
 765         struct rts51x_chip *chip = (struct rts51x_chip *)data;
 766         struct us_data *us = chip->us;
 767
 768         switch (rts51x_get_stat(chip)) {
 769         case RTS51X_STAT_INIT:
 770         case RTS51X_STAT_RUN:
 771                 rts51x_modi_suspend_timer(chip);
 772                 break;
 773         case RTS51X_STAT_IDLE:
 774         case RTS51X_STAT_SS:
 775                 usb_stor_dbg(us, "RTS51X_STAT_SS, intf->pm_usage_cnt:%d, power.usage:%d\n",
 776                              atomic_read(&us->pusb_intf->pm_usage_cnt),
 777                              atomic_read(&us->pusb_intf->dev.power.usage_count));
 778
 779                 if (atomic_read(&us->pusb_intf->pm_usage_cnt) > 0) {
 780                         usb_stor_dbg(us, "Ready to enter SS state\n");
 781                         rts51x_set_stat(chip, RTS51X_STAT_SS);
 782                         /* ignore mass storage interface's children */
 783                         pm_suspend_ignore_children(&us->pusb_intf->dev, true);
 784                         usb_autopm_put_interface_async(us->pusb_intf);
 785                         usb_stor_dbg(us, "RTS51X_STAT_SS 01, intf->pm_usage_cnt:%d, power.usage:%d\n",
 786                                      atomic_read(&us->pusb_intf->pm_usage_cnt),
 787                                      atomic_read(&us->pusb_intf->dev.power.usage_count));
 788                 }
 789                 break;
 790         default:
 791                 usb_stor_dbg(us, "Unknown state !!!\n");
 792                 break;
 793         }
 794 }
 795
 796 static inline int working_scsi(struct scsi_cmnd *srb)
 797 {
 798         if ((srb->cmnd[0] == TEST_UNIT_READY) ||
 799             (srb->cmnd[0] == ALLOW_MEDIUM_REMOVAL)) {
 800                 return 0;
 801         }
 802
 803         return 1;
 804 }
 805
 806 static void rts51x_invoke_transport(struct scsi_cmnd *srb, struct us_data *us)
 807 {
 808         struct rts51x_chip *chip = (struct rts51x_chip *)(us->extra);
 809         static int card_first_show = 1;
 810         static u8 media_not_present[] = { 0x70, 0, 0x02, 0, 0, 0, 0,
 811                 10, 0, 0, 0, 0, 0x3A, 0, 0, 0, 0, 0
 812         };
 813         static u8 invalid_cmd_field[] = { 0x70, 0, 0x05, 0, 0, 0, 0,
 814                 10, 0, 0, 0, 0, 0x24, 0, 0, 0, 0, 0
 815         };
 816         int ret;
 817
 818         if (working_scsi(srb)) {
 819                 usb_stor_dbg(us, "working scsi, intf->pm_usage_cnt:%d, power.usage:%d\n",
 820                              atomic_read(&us->pusb_intf->pm_usage_cnt),
 821                              atomic_read(&us->pusb_intf->dev.power.usage_count));
 822
 823                 if (atomic_read(&us->pusb_intf->pm_usage_cnt) <= 0) {
 824                         ret = usb_autopm_get_interface(us->pusb_intf);
 825                         usb_stor_dbg(us, "working scsi, ret=%d\n", ret);
 826                 }
 827                 if (rts51x_get_stat(chip) != RTS51X_STAT_RUN)
 828                         rts51x_set_stat(chip, RTS51X_STAT_RUN);
 829                 chip->proto_handler_backup(srb, us);
 830         } else {
 831                 if (rts51x_get_stat(chip) == RTS51X_STAT_SS) {
 832                         usb_stor_dbg(us, "NOT working scsi\n");
 833                         if ((srb->cmnd[0] == TEST_UNIT_READY) &&
 834                             (chip->pwr_state == US_SUSPEND)) {
 835                                 if (TST_LUN_READY(chip, srb->device->lun)) {
 836                                         srb->result = SAM_STAT_GOOD;
 837                                 } else {
 838                                         srb->result = SAM_STAT_CHECK_CONDITION;
 839                                         memcpy(srb->sense_buffer,
 840                                                media_not_present,
 841                                                US_SENSE_SIZE);
 842                                 }
 843                                 usb_stor_dbg(us, "TEST_UNIT_READY\n");
 844                                 goto out;
 845                         }
 846                         if (srb->cmnd[0] == ALLOW_MEDIUM_REMOVAL) {
 847                                 int prevent = srb->cmnd[4] & 0x1;
 848                                 if (prevent) {
 849                                         srb->result = SAM_STAT_CHECK_CONDITION;
 850                                         memcpy(srb->sense_buffer,
 851                                                invalid_cmd_field,
 852                                                US_SENSE_SIZE);
 853                                 } else {
 854                                         srb->result = SAM_STAT_GOOD;
 855                                 }
 856                                 usb_stor_dbg(us, "ALLOW_MEDIUM_REMOVAL\n");
 857                                 goto out;
 858                         }
 859                 } else {
 860                         usb_stor_dbg(us, "NOT working scsi, not SS\n");
 861                         chip->proto_handler_backup(srb, us);
 862                         /* Check whether card is plugged in */
 863                         if (srb->cmnd[0] == TEST_UNIT_READY) {
 864                                 if (srb->result == SAM_STAT_GOOD) {
 865                                         SET_LUN_READY(chip, srb->device->lun);
 866                                         if (card_first_show) {
 867                                                 card_first_show = 0;
 868                                                 fw5895_set_mmc_wp(us);
 869                                         }
 870                                 } else {
 871                                         CLR_LUN_READY(chip, srb->device->lun);
 872                                         card_first_show = 1;
 873                                 }
 874                         }
 875                         if (rts51x_get_stat(chip) != RTS51X_STAT_IDLE)
 876                                 rts51x_set_stat(chip, RTS51X_STAT_IDLE);
 877                 }
 878         }
 879 out:
 880         usb_stor_dbg(us, "state:%d\n", rts51x_get_stat(chip));
 881         if (rts51x_get_stat(chip) == RTS51X_STAT_RUN)
 882                 rts51x_modi_suspend_timer(chip);
 883 }
 884
 885 static int realtek_cr_autosuspend_setup(struct us_data *us)
 886 {
 887         struct rts51x_chip *chip;
 888         struct rts51x_status *status = NULL;
 889         u8 buf[16];
 890         int retval;
 891
 892         chip = (struct rts51x_chip *)us->extra;
 893         chip->support_auto_delink = 0;
 894         chip->pwr_state = US_RESUME;
 895         chip->lun_ready = 0;
 896         rts51x_set_stat(chip, RTS51X_STAT_INIT);
 897
 898         retval = rts51x_read_status(us, 0, buf, 16, &(chip->status_len));
 899         if (retval != STATUS_SUCCESS) {
 900                 usb_stor_dbg(us, "Read status fail\n");
 901                 return -EIO;
 902         }
 903         status = chip->status;
 904         status->vid = ((u16) buf[0] << 8) | buf[1];
 905         status->pid = ((u16) buf[2] << 8) | buf[3];
 906         status->cur_lun = buf[4];
 907         status->card_type = buf[5];
 908         status->total_lun = buf[6];
 909         status->fw_ver = ((u16) buf[7] << 8) | buf[8];
 910         status->phy_exist = buf[9];
 911         status->multi_flag = buf[10];
 912         status->multi_card = buf[11];
 913         status->log_exist = buf[12];
 914         if (chip->status_len == 16) {
 915                 status->detailed_type.detailed_type1 = buf[13];
 916                 status->function[0] = buf[14];
 917                 status->function[1] = buf[15];
 918         }
 919
 920         /* back up the proto_handler in us->extra */
 921         chip = (struct rts51x_chip *)(us->extra);
 922         chip->proto_handler_backup = us->proto_handler;
 923         /* Set the autosuspend_delay to 0 */
 924         pm_runtime_set_autosuspend_delay(&us->pusb_dev->dev, 0);
 925         /* override us->proto_handler setted in get_protocol() */
 926         us->proto_handler = rts51x_invoke_transport;
 927
 928         chip->timer_expires = 0;
 929         setup_timer(&chip->rts51x_suspend_timer, rts51x_suspend_timer_fn,
 930                         (unsigned long)chip);
 931         fw5895_init(us);
 932
 933         /* enable autosuspend function of the usb device */
 934         usb_enable_autosuspend(us->pusb_dev);
 935
 936         return 0;
 937 }
 938 #endif
 939
 940 static void realtek_cr_destructor(void *extra)
 941 {
 942         struct rts51x_chip *chip = extra;
 943
 944         if (!chip)
 945                 return;
 946
 947 #ifdef CONFIG_REALTEK_AUTOPM
 948         if (ss_en) {
 949                 del_timer(&chip->rts51x_suspend_timer);
 950                 chip->timer_expires = 0;
 951         }
 952 #endif
 953         kfree(chip->status);
 954 }
 955
 956 #ifdef CONFIG_PM
 957 static int realtek_cr_suspend(struct usb_interface *iface, pm_message_t message)
 958 {
 959         struct us_data *us = usb_get_intfdata(iface);
 960
 961         /* wait until no command is running */
 962         mutex_lock(&us->dev_mutex);
 963
 964         config_autodelink_before_power_down(us);
 965
 966         mutex_unlock(&us->dev_mutex);
 967
 968         return 0;
 969 }
 970
 971 static int realtek_cr_resume(struct usb_interface *iface)
 972 {
 973         struct us_data *us = usb_get_intfdata(iface);
 974
 975         fw5895_init(us);
 976         config_autodelink_after_power_on(us);
 977
 978         return 0;
 979 }
 980 #else
 981 #define realtek_cr_suspend      NULL
 982 #define realtek_cr_resume       NULL
 983 #endif
 984
 985 static int init_realtek_cr(struct us_data *us)
 986 {
 987         struct rts51x_chip *chip;
 988         int size, i, retval;
 989
 990         chip = kzalloc(sizeof(struct rts51x_chip), GFP_KERNEL);
 991         if (!chip)
 992                 return -ENOMEM;
 993
 994         us->extra = chip;
 995         us->extra_destructor = realtek_cr_destructor;
 996         us->max_lun = chip->max_lun = rts51x_get_max_lun(us);
 997         chip->us = us;
 998
 999         usb_stor_dbg(us, "chip->max_lun = %d\n", chip->max_lun);
1000
1001         size = (chip->max_lun + 1) * sizeof(struct rts51x_status);
1002         chip->status = kzalloc(size, GFP_KERNEL);
1003         if (!chip->status)
1004                 goto INIT_FAIL;
1005
1006         for (i = 0; i <= (int)(chip->max_lun); i++) {
1007                 retval = rts51x_check_status(us, (u8) i);
1008                 if (retval < 0)
1009                         goto INIT_FAIL;
1010         }
1011
1012         if (CHECK_FW_VER(chip, 0x5888) || CHECK_FW_VER(chip, 0x5889) ||
1013             CHECK_FW_VER(chip, 0x5901))
1014                 SET_AUTO_DELINK(chip);
1015         if (STATUS_LEN(chip) == 16) {
1016                 if (SUPPORT_AUTO_DELINK(chip))
1017                         SET_AUTO_DELINK(chip);
1018         }
1019 #ifdef CONFIG_REALTEK_AUTOPM
1020         if (ss_en)
1021                 realtek_cr_autosuspend_setup(us);
1022 #endif
1023
1024         usb_stor_dbg(us, "chip->flag = 0x%x\n", chip->flag);
1025
1026         (void)config_autodelink_after_power_on(us);
1027
1028         return 0;
1029
1030 INIT_FAIL:
1031         if (us->extra) {
1032                 kfree(chip->status);
1033                 kfree(us->extra);
1034                 us->extra = NULL;
1035         }
1036
1037         return -EIO;
1038 }
1039
1040 static struct scsi_host_template realtek_cr_host_template;
1041
1042 static int realtek_cr_probe(struct usb_interface *intf,
1043                             const struct usb_device_id *id)
1044 {
1045         struct us_data *us;
1046         int result;
1047
1048         dev_dbg(&intf->dev, "Probe Realtek Card Reader!\n");
1049
1050         result = usb_stor_probe1(&us, intf, id,
1051                                  (id - realtek_cr_ids) +
1052                                  realtek_cr_unusual_dev_list,
1053                                  &realtek_cr_host_template);
1054         if (result)
1055                 return result;
1056
1057         result = usb_stor_probe2(us);
1058
1059         return result;
1060 }
1061
1062 static struct usb_driver realtek_cr_driver = {
1063         .name = DRV_NAME,
1064         .probe = realtek_cr_probe,
1065         .disconnect = usb_stor_disconnect,
1066         /* .suspend =      usb_stor_suspend, */
1067         /* .resume =       usb_stor_resume, */
1068         .reset_resume = usb_stor_reset_resume,
1069         .suspend = realtek_cr_suspend,
1070         .resume = realtek_cr_resume,
1071         .pre_reset = usb_stor_pre_reset,
1072         .post_reset = usb_stor_post_reset,
1073         .id_table = realtek_cr_ids,
1074         .soft_unbind = 1,
1075         .supports_autosuspend = 1,
1076         .no_dynamic_id = 1,
1077 };
1078
1079 module_usb_stor_driver(realtek_cr_driver, realtek_cr_host_template, DRV_NAME);