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