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