Loading...
1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Sony MemoryStick support
4 *
5 * Copyright (C) 2007 Alex Dubov <oakad@yahoo.com>
6 *
7 * Special thanks to Carlos Corbacho for providing various MemoryStick cards
8 * that made this driver possible.
9 */
10
11#include <linux/memstick.h>
12#include <linux/idr.h>
13#include <linux/fs.h>
14#include <linux/delay.h>
15#include <linux/slab.h>
16#include <linux/module.h>
17#include <linux/pm_runtime.h>
18
19#define DRIVER_NAME "memstick"
20
21static unsigned int cmd_retries = 3;
22module_param(cmd_retries, uint, 0644);
23
24static struct workqueue_struct *workqueue;
25static DEFINE_IDR(memstick_host_idr);
26static DEFINE_SPINLOCK(memstick_host_lock);
27
28static int memstick_dev_match(struct memstick_dev *card,
29 struct memstick_device_id *id)
30{
31 if (id->match_flags & MEMSTICK_MATCH_ALL) {
32 if ((id->type == card->id.type)
33 && (id->category == card->id.category)
34 && (id->class == card->id.class))
35 return 1;
36 }
37
38 return 0;
39}
40
41static int memstick_bus_match(struct device *dev, struct device_driver *drv)
42{
43 struct memstick_dev *card = container_of(dev, struct memstick_dev,
44 dev);
45 struct memstick_driver *ms_drv = container_of(drv,
46 struct memstick_driver,
47 driver);
48 struct memstick_device_id *ids = ms_drv->id_table;
49
50 if (ids) {
51 while (ids->match_flags) {
52 if (memstick_dev_match(card, ids))
53 return 1;
54 ++ids;
55 }
56 }
57 return 0;
58}
59
60static int memstick_uevent(const struct device *dev, struct kobj_uevent_env *env)
61{
62 const struct memstick_dev *card = container_of_const(dev, struct memstick_dev,
63 dev);
64
65 if (add_uevent_var(env, "MEMSTICK_TYPE=%02X", card->id.type))
66 return -ENOMEM;
67
68 if (add_uevent_var(env, "MEMSTICK_CATEGORY=%02X", card->id.category))
69 return -ENOMEM;
70
71 if (add_uevent_var(env, "MEMSTICK_CLASS=%02X", card->id.class))
72 return -ENOMEM;
73
74 return 0;
75}
76
77static int memstick_device_probe(struct device *dev)
78{
79 struct memstick_dev *card = container_of(dev, struct memstick_dev,
80 dev);
81 struct memstick_driver *drv = container_of(dev->driver,
82 struct memstick_driver,
83 driver);
84 int rc = -ENODEV;
85
86 if (dev->driver && drv->probe) {
87 rc = drv->probe(card);
88 if (!rc)
89 get_device(dev);
90 }
91 return rc;
92}
93
94static void memstick_device_remove(struct device *dev)
95{
96 struct memstick_dev *card = container_of(dev, struct memstick_dev,
97 dev);
98 struct memstick_driver *drv = container_of(dev->driver,
99 struct memstick_driver,
100 driver);
101
102 if (dev->driver && drv->remove) {
103 drv->remove(card);
104 card->dev.driver = NULL;
105 }
106
107 put_device(dev);
108}
109
110#ifdef CONFIG_PM
111
112static int memstick_device_suspend(struct device *dev, pm_message_t state)
113{
114 struct memstick_dev *card = container_of(dev, struct memstick_dev,
115 dev);
116 struct memstick_driver *drv = container_of(dev->driver,
117 struct memstick_driver,
118 driver);
119
120 if (dev->driver && drv->suspend)
121 return drv->suspend(card, state);
122 return 0;
123}
124
125static int memstick_device_resume(struct device *dev)
126{
127 struct memstick_dev *card = container_of(dev, struct memstick_dev,
128 dev);
129 struct memstick_driver *drv = container_of(dev->driver,
130 struct memstick_driver,
131 driver);
132
133 if (dev->driver && drv->resume)
134 return drv->resume(card);
135 return 0;
136}
137
138#else
139
140#define memstick_device_suspend NULL
141#define memstick_device_resume NULL
142
143#endif /* CONFIG_PM */
144
145#define MEMSTICK_ATTR(name, format) \
146static ssize_t name##_show(struct device *dev, struct device_attribute *attr, \
147 char *buf) \
148{ \
149 struct memstick_dev *card = container_of(dev, struct memstick_dev, \
150 dev); \
151 return sprintf(buf, format, card->id.name); \
152} \
153static DEVICE_ATTR_RO(name);
154
155MEMSTICK_ATTR(type, "%02X");
156MEMSTICK_ATTR(category, "%02X");
157MEMSTICK_ATTR(class, "%02X");
158
159static struct attribute *memstick_dev_attrs[] = {
160 &dev_attr_type.attr,
161 &dev_attr_category.attr,
162 &dev_attr_class.attr,
163 NULL,
164};
165ATTRIBUTE_GROUPS(memstick_dev);
166
167static struct bus_type memstick_bus_type = {
168 .name = "memstick",
169 .dev_groups = memstick_dev_groups,
170 .match = memstick_bus_match,
171 .uevent = memstick_uevent,
172 .probe = memstick_device_probe,
173 .remove = memstick_device_remove,
174 .suspend = memstick_device_suspend,
175 .resume = memstick_device_resume
176};
177
178static void memstick_free(struct device *dev)
179{
180 struct memstick_host *host = container_of(dev, struct memstick_host,
181 dev);
182 kfree(host);
183}
184
185static struct class memstick_host_class = {
186 .name = "memstick_host",
187 .dev_release = memstick_free
188};
189
190static void memstick_free_card(struct device *dev)
191{
192 struct memstick_dev *card = container_of(dev, struct memstick_dev,
193 dev);
194 kfree(card);
195}
196
197static int memstick_dummy_check(struct memstick_dev *card)
198{
199 return 0;
200}
201
202/**
203 * memstick_detect_change - schedule media detection on memstick host
204 * @host - host to use
205 */
206void memstick_detect_change(struct memstick_host *host)
207{
208 queue_work(workqueue, &host->media_checker);
209}
210EXPORT_SYMBOL(memstick_detect_change);
211
212/**
213 * memstick_next_req - called by host driver to obtain next request to process
214 * @host - host to use
215 * @mrq - pointer to stick the request to
216 *
217 * Host calls this function from idle state (*mrq == NULL) or after finishing
218 * previous request (*mrq should point to it). If previous request was
219 * unsuccessful, it is retried for predetermined number of times. Return value
220 * of 0 means that new request was assigned to the host.
221 */
222int memstick_next_req(struct memstick_host *host, struct memstick_request **mrq)
223{
224 int rc = -ENXIO;
225
226 if ((*mrq) && (*mrq)->error && host->retries) {
227 (*mrq)->error = rc;
228 host->retries--;
229 return 0;
230 }
231
232 if (host->card && host->card->next_request)
233 rc = host->card->next_request(host->card, mrq);
234
235 if (!rc)
236 host->retries = cmd_retries > 1 ? cmd_retries - 1 : 1;
237 else
238 *mrq = NULL;
239
240 return rc;
241}
242EXPORT_SYMBOL(memstick_next_req);
243
244/**
245 * memstick_new_req - notify the host that some requests are pending
246 * @host - host to use
247 */
248void memstick_new_req(struct memstick_host *host)
249{
250 if (host->card) {
251 host->retries = cmd_retries;
252 reinit_completion(&host->card->mrq_complete);
253 host->request(host);
254 }
255}
256EXPORT_SYMBOL(memstick_new_req);
257
258/**
259 * memstick_init_req_sg - set request fields needed for bulk data transfer
260 * @mrq - request to use
261 * @tpc - memstick Transport Protocol Command
262 * @sg - TPC argument
263 */
264void memstick_init_req_sg(struct memstick_request *mrq, unsigned char tpc,
265 const struct scatterlist *sg)
266{
267 mrq->tpc = tpc;
268 if (tpc & 8)
269 mrq->data_dir = WRITE;
270 else
271 mrq->data_dir = READ;
272
273 mrq->sg = *sg;
274 mrq->long_data = 1;
275
276 if (tpc == MS_TPC_SET_CMD || tpc == MS_TPC_EX_SET_CMD)
277 mrq->need_card_int = 1;
278 else
279 mrq->need_card_int = 0;
280}
281EXPORT_SYMBOL(memstick_init_req_sg);
282
283/**
284 * memstick_init_req - set request fields needed for short data transfer
285 * @mrq - request to use
286 * @tpc - memstick Transport Protocol Command
287 * @buf - TPC argument buffer
288 * @length - TPC argument size
289 *
290 * The intended use of this function (transfer of data items several bytes
291 * in size) allows us to just copy the value between request structure and
292 * user supplied buffer.
293 */
294void memstick_init_req(struct memstick_request *mrq, unsigned char tpc,
295 const void *buf, size_t length)
296{
297 mrq->tpc = tpc;
298 if (tpc & 8)
299 mrq->data_dir = WRITE;
300 else
301 mrq->data_dir = READ;
302
303 mrq->data_len = length > sizeof(mrq->data) ? sizeof(mrq->data) : length;
304 if (mrq->data_dir == WRITE)
305 memcpy(mrq->data, buf, mrq->data_len);
306
307 mrq->long_data = 0;
308
309 if (tpc == MS_TPC_SET_CMD || tpc == MS_TPC_EX_SET_CMD)
310 mrq->need_card_int = 1;
311 else
312 mrq->need_card_int = 0;
313}
314EXPORT_SYMBOL(memstick_init_req);
315
316/*
317 * Functions prefixed with "h_" are protocol callbacks. They can be called from
318 * interrupt context. Return value of 0 means that request processing is still
319 * ongoing, while special error value of -EAGAIN means that current request is
320 * finished (and request processor should come back some time later).
321 */
322
323static int h_memstick_read_dev_id(struct memstick_dev *card,
324 struct memstick_request **mrq)
325{
326 struct ms_id_register id_reg;
327
328 if (!(*mrq)) {
329 memstick_init_req(&card->current_mrq, MS_TPC_READ_REG, &id_reg,
330 sizeof(struct ms_id_register));
331 *mrq = &card->current_mrq;
332 return 0;
333 }
334 if (!(*mrq)->error) {
335 memcpy(&id_reg, (*mrq)->data, sizeof(id_reg));
336 card->id.match_flags = MEMSTICK_MATCH_ALL;
337 card->id.type = id_reg.type;
338 card->id.category = id_reg.category;
339 card->id.class = id_reg.class;
340 dev_dbg(&card->dev, "if_mode = %02x\n", id_reg.if_mode);
341 }
342 complete(&card->mrq_complete);
343 return -EAGAIN;
344}
345
346static int h_memstick_set_rw_addr(struct memstick_dev *card,
347 struct memstick_request **mrq)
348{
349 if (!(*mrq)) {
350 memstick_init_req(&card->current_mrq, MS_TPC_SET_RW_REG_ADRS,
351 (char *)&card->reg_addr,
352 sizeof(card->reg_addr));
353 *mrq = &card->current_mrq;
354 return 0;
355 } else {
356 complete(&card->mrq_complete);
357 return -EAGAIN;
358 }
359}
360
361/**
362 * memstick_set_rw_addr - issue SET_RW_REG_ADDR request and wait for it to
363 * complete
364 * @card - media device to use
365 */
366int memstick_set_rw_addr(struct memstick_dev *card)
367{
368 card->next_request = h_memstick_set_rw_addr;
369 memstick_new_req(card->host);
370 wait_for_completion(&card->mrq_complete);
371
372 return card->current_mrq.error;
373}
374EXPORT_SYMBOL(memstick_set_rw_addr);
375
376static struct memstick_dev *memstick_alloc_card(struct memstick_host *host)
377{
378 struct memstick_dev *card = kzalloc(sizeof(struct memstick_dev),
379 GFP_KERNEL);
380 struct memstick_dev *old_card = host->card;
381 struct ms_id_register id_reg;
382
383 if (card) {
384 card->host = host;
385 dev_set_name(&card->dev, "%s", dev_name(&host->dev));
386 card->dev.parent = &host->dev;
387 card->dev.bus = &memstick_bus_type;
388 card->dev.release = memstick_free_card;
389 card->check = memstick_dummy_check;
390
391 card->reg_addr.r_offset = offsetof(struct ms_register, id);
392 card->reg_addr.r_length = sizeof(id_reg);
393 card->reg_addr.w_offset = offsetof(struct ms_register, id);
394 card->reg_addr.w_length = sizeof(id_reg);
395
396 init_completion(&card->mrq_complete);
397
398 host->card = card;
399 if (memstick_set_rw_addr(card))
400 goto err_out;
401
402 card->next_request = h_memstick_read_dev_id;
403 memstick_new_req(host);
404 wait_for_completion(&card->mrq_complete);
405
406 if (card->current_mrq.error)
407 goto err_out;
408 }
409 host->card = old_card;
410 return card;
411err_out:
412 host->card = old_card;
413 kfree_const(card->dev.kobj.name);
414 kfree(card);
415 return NULL;
416}
417
418static int memstick_power_on(struct memstick_host *host)
419{
420 int rc = host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_ON);
421
422 if (!rc)
423 rc = host->set_param(host, MEMSTICK_INTERFACE, MEMSTICK_SERIAL);
424
425 return rc;
426}
427
428static void memstick_check(struct work_struct *work)
429{
430 struct memstick_host *host = container_of(work, struct memstick_host,
431 media_checker);
432 struct memstick_dev *card;
433
434 dev_dbg(&host->dev, "memstick_check started\n");
435 pm_runtime_get_noresume(host->dev.parent);
436 mutex_lock(&host->lock);
437 if (!host->card) {
438 if (memstick_power_on(host))
439 goto out_power_off;
440 } else if (host->card->stop)
441 host->card->stop(host->card);
442
443 if (host->removing)
444 goto out_power_off;
445
446 card = memstick_alloc_card(host);
447
448 if (!card) {
449 if (host->card) {
450 device_unregister(&host->card->dev);
451 host->card = NULL;
452 }
453 } else {
454 dev_dbg(&host->dev, "new card %02x, %02x, %02x\n",
455 card->id.type, card->id.category, card->id.class);
456 if (host->card) {
457 if (memstick_set_rw_addr(host->card)
458 || !memstick_dev_match(host->card, &card->id)
459 || !(host->card->check(host->card))) {
460 device_unregister(&host->card->dev);
461 host->card = NULL;
462 } else if (host->card->start)
463 host->card->start(host->card);
464 }
465
466 if (!host->card) {
467 host->card = card;
468 if (device_register(&card->dev)) {
469 put_device(&card->dev);
470 host->card = NULL;
471 }
472 } else {
473 kfree_const(card->dev.kobj.name);
474 kfree(card);
475 }
476 }
477
478out_power_off:
479 if (!host->card)
480 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
481
482 mutex_unlock(&host->lock);
483 pm_runtime_put(host->dev.parent);
484 dev_dbg(&host->dev, "memstick_check finished\n");
485}
486
487/**
488 * memstick_alloc_host - allocate a memstick_host structure
489 * @extra: size of the user private data to allocate
490 * @dev: parent device of the host
491 */
492struct memstick_host *memstick_alloc_host(unsigned int extra,
493 struct device *dev)
494{
495 struct memstick_host *host;
496
497 host = kzalloc(sizeof(struct memstick_host) + extra, GFP_KERNEL);
498 if (host) {
499 mutex_init(&host->lock);
500 INIT_WORK(&host->media_checker, memstick_check);
501 host->dev.class = &memstick_host_class;
502 host->dev.parent = dev;
503 device_initialize(&host->dev);
504 }
505 return host;
506}
507EXPORT_SYMBOL(memstick_alloc_host);
508
509/**
510 * memstick_add_host - start request processing on memstick host
511 * @host - host to use
512 */
513int memstick_add_host(struct memstick_host *host)
514{
515 int rc;
516
517 idr_preload(GFP_KERNEL);
518 spin_lock(&memstick_host_lock);
519
520 rc = idr_alloc(&memstick_host_idr, host, 0, 0, GFP_NOWAIT);
521 if (rc >= 0)
522 host->id = rc;
523
524 spin_unlock(&memstick_host_lock);
525 idr_preload_end();
526 if (rc < 0)
527 return rc;
528
529 dev_set_name(&host->dev, "memstick%u", host->id);
530
531 rc = device_add(&host->dev);
532 if (rc) {
533 spin_lock(&memstick_host_lock);
534 idr_remove(&memstick_host_idr, host->id);
535 spin_unlock(&memstick_host_lock);
536 return rc;
537 }
538
539 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
540 memstick_detect_change(host);
541 return 0;
542}
543EXPORT_SYMBOL(memstick_add_host);
544
545/**
546 * memstick_remove_host - stop request processing on memstick host
547 * @host - host to use
548 */
549void memstick_remove_host(struct memstick_host *host)
550{
551 host->removing = 1;
552 flush_workqueue(workqueue);
553 mutex_lock(&host->lock);
554 if (host->card)
555 device_unregister(&host->card->dev);
556 host->card = NULL;
557 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
558 mutex_unlock(&host->lock);
559
560 spin_lock(&memstick_host_lock);
561 idr_remove(&memstick_host_idr, host->id);
562 spin_unlock(&memstick_host_lock);
563 device_del(&host->dev);
564}
565EXPORT_SYMBOL(memstick_remove_host);
566
567/**
568 * memstick_free_host - free memstick host
569 * @host - host to use
570 */
571void memstick_free_host(struct memstick_host *host)
572{
573 mutex_destroy(&host->lock);
574 put_device(&host->dev);
575}
576EXPORT_SYMBOL(memstick_free_host);
577
578/**
579 * memstick_suspend_host - notify bus driver of host suspension
580 * @host - host to use
581 */
582void memstick_suspend_host(struct memstick_host *host)
583{
584 mutex_lock(&host->lock);
585 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
586 mutex_unlock(&host->lock);
587}
588EXPORT_SYMBOL(memstick_suspend_host);
589
590/**
591 * memstick_resume_host - notify bus driver of host resumption
592 * @host - host to use
593 */
594void memstick_resume_host(struct memstick_host *host)
595{
596 int rc = 0;
597
598 mutex_lock(&host->lock);
599 if (host->card)
600 rc = memstick_power_on(host);
601 mutex_unlock(&host->lock);
602
603 if (!rc)
604 memstick_detect_change(host);
605}
606EXPORT_SYMBOL(memstick_resume_host);
607
608int memstick_register_driver(struct memstick_driver *drv)
609{
610 drv->driver.bus = &memstick_bus_type;
611
612 return driver_register(&drv->driver);
613}
614EXPORT_SYMBOL(memstick_register_driver);
615
616void memstick_unregister_driver(struct memstick_driver *drv)
617{
618 driver_unregister(&drv->driver);
619}
620EXPORT_SYMBOL(memstick_unregister_driver);
621
622
623static int __init memstick_init(void)
624{
625 int rc;
626
627 workqueue = create_freezable_workqueue("kmemstick");
628 if (!workqueue)
629 return -ENOMEM;
630
631 rc = bus_register(&memstick_bus_type);
632 if (rc)
633 goto error_destroy_workqueue;
634
635 rc = class_register(&memstick_host_class);
636 if (rc)
637 goto error_bus_unregister;
638
639 return 0;
640
641error_bus_unregister:
642 bus_unregister(&memstick_bus_type);
643error_destroy_workqueue:
644 destroy_workqueue(workqueue);
645
646 return rc;
647}
648
649static void __exit memstick_exit(void)
650{
651 class_unregister(&memstick_host_class);
652 bus_unregister(&memstick_bus_type);
653 destroy_workqueue(workqueue);
654 idr_destroy(&memstick_host_idr);
655}
656
657module_init(memstick_init);
658module_exit(memstick_exit);
659
660MODULE_AUTHOR("Alex Dubov");
661MODULE_LICENSE("GPL");
662MODULE_DESCRIPTION("Sony MemoryStick core driver");
1/*
2 * Sony MemoryStick support
3 *
4 * Copyright (C) 2007 Alex Dubov <oakad@yahoo.com>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 as
8 * published by the Free Software Foundation.
9 *
10 * Special thanks to Carlos Corbacho for providing various MemoryStick cards
11 * that made this driver possible.
12 *
13 */
14
15#include <linux/memstick.h>
16#include <linux/idr.h>
17#include <linux/fs.h>
18#include <linux/delay.h>
19#include <linux/slab.h>
20#include <linux/module.h>
21
22#define DRIVER_NAME "memstick"
23
24static unsigned int cmd_retries = 3;
25module_param(cmd_retries, uint, 0644);
26
27static struct workqueue_struct *workqueue;
28static DEFINE_IDR(memstick_host_idr);
29static DEFINE_SPINLOCK(memstick_host_lock);
30
31static int memstick_dev_match(struct memstick_dev *card,
32 struct memstick_device_id *id)
33{
34 if (id->match_flags & MEMSTICK_MATCH_ALL) {
35 if ((id->type == card->id.type)
36 && (id->category == card->id.category)
37 && (id->class == card->id.class))
38 return 1;
39 }
40
41 return 0;
42}
43
44static int memstick_bus_match(struct device *dev, struct device_driver *drv)
45{
46 struct memstick_dev *card = container_of(dev, struct memstick_dev,
47 dev);
48 struct memstick_driver *ms_drv = container_of(drv,
49 struct memstick_driver,
50 driver);
51 struct memstick_device_id *ids = ms_drv->id_table;
52
53 if (ids) {
54 while (ids->match_flags) {
55 if (memstick_dev_match(card, ids))
56 return 1;
57 ++ids;
58 }
59 }
60 return 0;
61}
62
63static int memstick_uevent(struct device *dev, struct kobj_uevent_env *env)
64{
65 struct memstick_dev *card = container_of(dev, struct memstick_dev,
66 dev);
67
68 if (add_uevent_var(env, "MEMSTICK_TYPE=%02X", card->id.type))
69 return -ENOMEM;
70
71 if (add_uevent_var(env, "MEMSTICK_CATEGORY=%02X", card->id.category))
72 return -ENOMEM;
73
74 if (add_uevent_var(env, "MEMSTICK_CLASS=%02X", card->id.class))
75 return -ENOMEM;
76
77 return 0;
78}
79
80static int memstick_device_probe(struct device *dev)
81{
82 struct memstick_dev *card = container_of(dev, struct memstick_dev,
83 dev);
84 struct memstick_driver *drv = container_of(dev->driver,
85 struct memstick_driver,
86 driver);
87 int rc = -ENODEV;
88
89 if (dev->driver && drv->probe) {
90 rc = drv->probe(card);
91 if (!rc)
92 get_device(dev);
93 }
94 return rc;
95}
96
97static int memstick_device_remove(struct device *dev)
98{
99 struct memstick_dev *card = container_of(dev, struct memstick_dev,
100 dev);
101 struct memstick_driver *drv = container_of(dev->driver,
102 struct memstick_driver,
103 driver);
104
105 if (dev->driver && drv->remove) {
106 drv->remove(card);
107 card->dev.driver = NULL;
108 }
109
110 put_device(dev);
111 return 0;
112}
113
114#ifdef CONFIG_PM
115
116static int memstick_device_suspend(struct device *dev, pm_message_t state)
117{
118 struct memstick_dev *card = container_of(dev, struct memstick_dev,
119 dev);
120 struct memstick_driver *drv = container_of(dev->driver,
121 struct memstick_driver,
122 driver);
123
124 if (dev->driver && drv->suspend)
125 return drv->suspend(card, state);
126 return 0;
127}
128
129static int memstick_device_resume(struct device *dev)
130{
131 struct memstick_dev *card = container_of(dev, struct memstick_dev,
132 dev);
133 struct memstick_driver *drv = container_of(dev->driver,
134 struct memstick_driver,
135 driver);
136
137 if (dev->driver && drv->resume)
138 return drv->resume(card);
139 return 0;
140}
141
142#else
143
144#define memstick_device_suspend NULL
145#define memstick_device_resume NULL
146
147#endif /* CONFIG_PM */
148
149#define MEMSTICK_ATTR(name, format) \
150static ssize_t name##_show(struct device *dev, struct device_attribute *attr, \
151 char *buf) \
152{ \
153 struct memstick_dev *card = container_of(dev, struct memstick_dev, \
154 dev); \
155 return sprintf(buf, format, card->id.name); \
156}
157
158MEMSTICK_ATTR(type, "%02X");
159MEMSTICK_ATTR(category, "%02X");
160MEMSTICK_ATTR(class, "%02X");
161
162#define MEMSTICK_ATTR_RO(name) __ATTR(name, S_IRUGO, name##_show, NULL)
163
164static struct device_attribute memstick_dev_attrs[] = {
165 MEMSTICK_ATTR_RO(type),
166 MEMSTICK_ATTR_RO(category),
167 MEMSTICK_ATTR_RO(class),
168 __ATTR_NULL
169};
170
171static struct bus_type memstick_bus_type = {
172 .name = "memstick",
173 .dev_attrs = memstick_dev_attrs,
174 .match = memstick_bus_match,
175 .uevent = memstick_uevent,
176 .probe = memstick_device_probe,
177 .remove = memstick_device_remove,
178 .suspend = memstick_device_suspend,
179 .resume = memstick_device_resume
180};
181
182static void memstick_free(struct device *dev)
183{
184 struct memstick_host *host = container_of(dev, struct memstick_host,
185 dev);
186 kfree(host);
187}
188
189static struct class memstick_host_class = {
190 .name = "memstick_host",
191 .dev_release = memstick_free
192};
193
194static void memstick_free_card(struct device *dev)
195{
196 struct memstick_dev *card = container_of(dev, struct memstick_dev,
197 dev);
198 kfree(card);
199}
200
201static int memstick_dummy_check(struct memstick_dev *card)
202{
203 return 0;
204}
205
206/**
207 * memstick_detect_change - schedule media detection on memstick host
208 * @host - host to use
209 */
210void memstick_detect_change(struct memstick_host *host)
211{
212 queue_work(workqueue, &host->media_checker);
213}
214EXPORT_SYMBOL(memstick_detect_change);
215
216/**
217 * memstick_next_req - called by host driver to obtain next request to process
218 * @host - host to use
219 * @mrq - pointer to stick the request to
220 *
221 * Host calls this function from idle state (*mrq == NULL) or after finishing
222 * previous request (*mrq should point to it). If previous request was
223 * unsuccessful, it is retried for predetermined number of times. Return value
224 * of 0 means that new request was assigned to the host.
225 */
226int memstick_next_req(struct memstick_host *host, struct memstick_request **mrq)
227{
228 int rc = -ENXIO;
229
230 if ((*mrq) && (*mrq)->error && host->retries) {
231 (*mrq)->error = rc;
232 host->retries--;
233 return 0;
234 }
235
236 if (host->card && host->card->next_request)
237 rc = host->card->next_request(host->card, mrq);
238
239 if (!rc)
240 host->retries = cmd_retries > 1 ? cmd_retries - 1 : 1;
241 else
242 *mrq = NULL;
243
244 return rc;
245}
246EXPORT_SYMBOL(memstick_next_req);
247
248/**
249 * memstick_new_req - notify the host that some requests are pending
250 * @host - host to use
251 */
252void memstick_new_req(struct memstick_host *host)
253{
254 if (host->card) {
255 host->retries = cmd_retries;
256 INIT_COMPLETION(host->card->mrq_complete);
257 host->request(host);
258 }
259}
260EXPORT_SYMBOL(memstick_new_req);
261
262/**
263 * memstick_init_req_sg - set request fields needed for bulk data transfer
264 * @mrq - request to use
265 * @tpc - memstick Transport Protocol Command
266 * @sg - TPC argument
267 */
268void memstick_init_req_sg(struct memstick_request *mrq, unsigned char tpc,
269 const struct scatterlist *sg)
270{
271 mrq->tpc = tpc;
272 if (tpc & 8)
273 mrq->data_dir = WRITE;
274 else
275 mrq->data_dir = READ;
276
277 mrq->sg = *sg;
278 mrq->long_data = 1;
279
280 if (tpc == MS_TPC_SET_CMD || tpc == MS_TPC_EX_SET_CMD)
281 mrq->need_card_int = 1;
282 else
283 mrq->need_card_int = 0;
284}
285EXPORT_SYMBOL(memstick_init_req_sg);
286
287/**
288 * memstick_init_req - set request fields needed for short data transfer
289 * @mrq - request to use
290 * @tpc - memstick Transport Protocol Command
291 * @buf - TPC argument buffer
292 * @length - TPC argument size
293 *
294 * The intended use of this function (transfer of data items several bytes
295 * in size) allows us to just copy the value between request structure and
296 * user supplied buffer.
297 */
298void memstick_init_req(struct memstick_request *mrq, unsigned char tpc,
299 const void *buf, size_t length)
300{
301 mrq->tpc = tpc;
302 if (tpc & 8)
303 mrq->data_dir = WRITE;
304 else
305 mrq->data_dir = READ;
306
307 mrq->data_len = length > sizeof(mrq->data) ? sizeof(mrq->data) : length;
308 if (mrq->data_dir == WRITE)
309 memcpy(mrq->data, buf, mrq->data_len);
310
311 mrq->long_data = 0;
312
313 if (tpc == MS_TPC_SET_CMD || tpc == MS_TPC_EX_SET_CMD)
314 mrq->need_card_int = 1;
315 else
316 mrq->need_card_int = 0;
317}
318EXPORT_SYMBOL(memstick_init_req);
319
320/*
321 * Functions prefixed with "h_" are protocol callbacks. They can be called from
322 * interrupt context. Return value of 0 means that request processing is still
323 * ongoing, while special error value of -EAGAIN means that current request is
324 * finished (and request processor should come back some time later).
325 */
326
327static int h_memstick_read_dev_id(struct memstick_dev *card,
328 struct memstick_request **mrq)
329{
330 struct ms_id_register id_reg;
331
332 if (!(*mrq)) {
333 memstick_init_req(&card->current_mrq, MS_TPC_READ_REG, NULL,
334 sizeof(struct ms_id_register));
335 *mrq = &card->current_mrq;
336 return 0;
337 } else {
338 if (!(*mrq)->error) {
339 memcpy(&id_reg, (*mrq)->data, sizeof(id_reg));
340 card->id.match_flags = MEMSTICK_MATCH_ALL;
341 card->id.type = id_reg.type;
342 card->id.category = id_reg.category;
343 card->id.class = id_reg.class;
344 dev_dbg(&card->dev, "if_mode = %02x\n", id_reg.if_mode);
345 }
346 complete(&card->mrq_complete);
347 return -EAGAIN;
348 }
349}
350
351static int h_memstick_set_rw_addr(struct memstick_dev *card,
352 struct memstick_request **mrq)
353{
354 if (!(*mrq)) {
355 memstick_init_req(&card->current_mrq, MS_TPC_SET_RW_REG_ADRS,
356 (char *)&card->reg_addr,
357 sizeof(card->reg_addr));
358 *mrq = &card->current_mrq;
359 return 0;
360 } else {
361 complete(&card->mrq_complete);
362 return -EAGAIN;
363 }
364}
365
366/**
367 * memstick_set_rw_addr - issue SET_RW_REG_ADDR request and wait for it to
368 * complete
369 * @card - media device to use
370 */
371int memstick_set_rw_addr(struct memstick_dev *card)
372{
373 card->next_request = h_memstick_set_rw_addr;
374 memstick_new_req(card->host);
375 wait_for_completion(&card->mrq_complete);
376
377 return card->current_mrq.error;
378}
379EXPORT_SYMBOL(memstick_set_rw_addr);
380
381static struct memstick_dev *memstick_alloc_card(struct memstick_host *host)
382{
383 struct memstick_dev *card = kzalloc(sizeof(struct memstick_dev),
384 GFP_KERNEL);
385 struct memstick_dev *old_card = host->card;
386 struct ms_id_register id_reg;
387
388 if (card) {
389 card->host = host;
390 dev_set_name(&card->dev, "%s", dev_name(&host->dev));
391 card->dev.parent = &host->dev;
392 card->dev.bus = &memstick_bus_type;
393 card->dev.release = memstick_free_card;
394 card->check = memstick_dummy_check;
395
396 card->reg_addr.r_offset = offsetof(struct ms_register, id);
397 card->reg_addr.r_length = sizeof(id_reg);
398 card->reg_addr.w_offset = offsetof(struct ms_register, id);
399 card->reg_addr.w_length = sizeof(id_reg);
400
401 init_completion(&card->mrq_complete);
402
403 host->card = card;
404 if (memstick_set_rw_addr(card))
405 goto err_out;
406
407 card->next_request = h_memstick_read_dev_id;
408 memstick_new_req(host);
409 wait_for_completion(&card->mrq_complete);
410
411 if (card->current_mrq.error)
412 goto err_out;
413 }
414 host->card = old_card;
415 return card;
416err_out:
417 host->card = old_card;
418 kfree(card);
419 return NULL;
420}
421
422static int memstick_power_on(struct memstick_host *host)
423{
424 int rc = host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_ON);
425
426 if (!rc)
427 rc = host->set_param(host, MEMSTICK_INTERFACE, MEMSTICK_SERIAL);
428
429 return rc;
430}
431
432static void memstick_check(struct work_struct *work)
433{
434 struct memstick_host *host = container_of(work, struct memstick_host,
435 media_checker);
436 struct memstick_dev *card;
437
438 dev_dbg(&host->dev, "memstick_check started\n");
439 mutex_lock(&host->lock);
440 if (!host->card) {
441 if (memstick_power_on(host))
442 goto out_power_off;
443 } else if (host->card->stop)
444 host->card->stop(host->card);
445
446 card = memstick_alloc_card(host);
447
448 if (!card) {
449 if (host->card) {
450 device_unregister(&host->card->dev);
451 host->card = NULL;
452 }
453 } else {
454 dev_dbg(&host->dev, "new card %02x, %02x, %02x\n",
455 card->id.type, card->id.category, card->id.class);
456 if (host->card) {
457 if (memstick_set_rw_addr(host->card)
458 || !memstick_dev_match(host->card, &card->id)
459 || !(host->card->check(host->card))) {
460 device_unregister(&host->card->dev);
461 host->card = NULL;
462 } else if (host->card->start)
463 host->card->start(host->card);
464 }
465
466 if (!host->card) {
467 host->card = card;
468 if (device_register(&card->dev)) {
469 put_device(&card->dev);
470 kfree(host->card);
471 host->card = NULL;
472 }
473 } else
474 kfree(card);
475 }
476
477out_power_off:
478 if (!host->card)
479 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
480
481 mutex_unlock(&host->lock);
482 dev_dbg(&host->dev, "memstick_check finished\n");
483}
484
485/**
486 * memstick_alloc_host - allocate a memstick_host structure
487 * @extra: size of the user private data to allocate
488 * @dev: parent device of the host
489 */
490struct memstick_host *memstick_alloc_host(unsigned int extra,
491 struct device *dev)
492{
493 struct memstick_host *host;
494
495 host = kzalloc(sizeof(struct memstick_host) + extra, GFP_KERNEL);
496 if (host) {
497 mutex_init(&host->lock);
498 INIT_WORK(&host->media_checker, memstick_check);
499 host->dev.class = &memstick_host_class;
500 host->dev.parent = dev;
501 device_initialize(&host->dev);
502 }
503 return host;
504}
505EXPORT_SYMBOL(memstick_alloc_host);
506
507/**
508 * memstick_add_host - start request processing on memstick host
509 * @host - host to use
510 */
511int memstick_add_host(struct memstick_host *host)
512{
513 int rc;
514
515 while (1) {
516 if (!idr_pre_get(&memstick_host_idr, GFP_KERNEL))
517 return -ENOMEM;
518
519 spin_lock(&memstick_host_lock);
520 rc = idr_get_new(&memstick_host_idr, host, &host->id);
521 spin_unlock(&memstick_host_lock);
522 if (!rc)
523 break;
524 else if (rc != -EAGAIN)
525 return rc;
526 }
527
528 dev_set_name(&host->dev, "memstick%u", host->id);
529
530 rc = device_add(&host->dev);
531 if (rc) {
532 spin_lock(&memstick_host_lock);
533 idr_remove(&memstick_host_idr, host->id);
534 spin_unlock(&memstick_host_lock);
535 return rc;
536 }
537
538 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
539 memstick_detect_change(host);
540 return 0;
541}
542EXPORT_SYMBOL(memstick_add_host);
543
544/**
545 * memstick_remove_host - stop request processing on memstick host
546 * @host - host to use
547 */
548void memstick_remove_host(struct memstick_host *host)
549{
550 flush_workqueue(workqueue);
551 mutex_lock(&host->lock);
552 if (host->card)
553 device_unregister(&host->card->dev);
554 host->card = NULL;
555 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
556 mutex_unlock(&host->lock);
557
558 spin_lock(&memstick_host_lock);
559 idr_remove(&memstick_host_idr, host->id);
560 spin_unlock(&memstick_host_lock);
561 device_del(&host->dev);
562}
563EXPORT_SYMBOL(memstick_remove_host);
564
565/**
566 * memstick_free_host - free memstick host
567 * @host - host to use
568 */
569void memstick_free_host(struct memstick_host *host)
570{
571 mutex_destroy(&host->lock);
572 put_device(&host->dev);
573}
574EXPORT_SYMBOL(memstick_free_host);
575
576/**
577 * memstick_suspend_host - notify bus driver of host suspension
578 * @host - host to use
579 */
580void memstick_suspend_host(struct memstick_host *host)
581{
582 mutex_lock(&host->lock);
583 host->set_param(host, MEMSTICK_POWER, MEMSTICK_POWER_OFF);
584 mutex_unlock(&host->lock);
585}
586EXPORT_SYMBOL(memstick_suspend_host);
587
588/**
589 * memstick_resume_host - notify bus driver of host resumption
590 * @host - host to use
591 */
592void memstick_resume_host(struct memstick_host *host)
593{
594 int rc = 0;
595
596 mutex_lock(&host->lock);
597 if (host->card)
598 rc = memstick_power_on(host);
599 mutex_unlock(&host->lock);
600
601 if (!rc)
602 memstick_detect_change(host);
603}
604EXPORT_SYMBOL(memstick_resume_host);
605
606int memstick_register_driver(struct memstick_driver *drv)
607{
608 drv->driver.bus = &memstick_bus_type;
609
610 return driver_register(&drv->driver);
611}
612EXPORT_SYMBOL(memstick_register_driver);
613
614void memstick_unregister_driver(struct memstick_driver *drv)
615{
616 driver_unregister(&drv->driver);
617}
618EXPORT_SYMBOL(memstick_unregister_driver);
619
620
621static int __init memstick_init(void)
622{
623 int rc;
624
625 workqueue = create_freezable_workqueue("kmemstick");
626 if (!workqueue)
627 return -ENOMEM;
628
629 rc = bus_register(&memstick_bus_type);
630 if (!rc)
631 rc = class_register(&memstick_host_class);
632
633 if (!rc)
634 return 0;
635
636 bus_unregister(&memstick_bus_type);
637 destroy_workqueue(workqueue);
638
639 return rc;
640}
641
642static void __exit memstick_exit(void)
643{
644 class_unregister(&memstick_host_class);
645 bus_unregister(&memstick_bus_type);
646 destroy_workqueue(workqueue);
647 idr_destroy(&memstick_host_idr);
648}
649
650module_init(memstick_init);
651module_exit(memstick_exit);
652
653MODULE_AUTHOR("Alex Dubov");
654MODULE_LICENSE("GPL");
655MODULE_DESCRIPTION("Sony MemoryStick core driver");