Linux Audio

Check our new training course

Loading...
v6.8
  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * PS3 Disk Storage Driver
  4 *
  5 * Copyright (C) 2007 Sony Computer Entertainment Inc.
  6 * Copyright 2007 Sony Corp.
 
 
 
 
 
 
 
 
 
 
 
 
 
  7 */
  8
  9#include <linux/ata.h>
 10#include <linux/blk-mq.h>
 11#include <linux/slab.h>
 12#include <linux/module.h>
 13
 14#include <asm/lv1call.h>
 15#include <asm/ps3stor.h>
 16#include <asm/firmware.h>
 17
 18
 19#define DEVICE_NAME		"ps3disk"
 20
 21#define BOUNCE_SIZE		(64*1024)
 22
 23#define PS3DISK_MAX_DISKS	16
 24#define PS3DISK_MINORS		16
 25
 26
 27#define PS3DISK_NAME		"ps3d%c"
 28
 29
 30struct ps3disk_private {
 31	spinlock_t lock;		/* Request queue spinlock */
 32	struct blk_mq_tag_set tag_set;
 33	struct gendisk *gendisk;
 34	unsigned int blocking_factor;
 35	struct request *req;
 36	u64 raw_capacity;
 37	unsigned char model[ATA_ID_PROD_LEN+1];
 38};
 39
 40
 41#define LV1_STORAGE_SEND_ATA_COMMAND	(2)
 42#define LV1_STORAGE_ATA_HDDOUT		(0x23)
 43
 44struct lv1_ata_cmnd_block {
 45	u16	features;
 46	u16	sector_count;
 47	u16	LBA_low;
 48	u16	LBA_mid;
 49	u16	LBA_high;
 50	u8	device;
 51	u8	command;
 52	u32	is_ext;
 53	u32	proto;
 54	u32	in_out;
 55	u32	size;
 56	u64	buffer;
 57	u32	arglen;
 58};
 59
 60enum lv1_ata_proto {
 61	NON_DATA_PROTO     = 0,
 62	PIO_DATA_IN_PROTO  = 1,
 63	PIO_DATA_OUT_PROTO = 2,
 64	DMA_PROTO = 3
 65};
 66
 67enum lv1_ata_in_out {
 68	DIR_WRITE = 0,			/* memory -> device */
 69	DIR_READ = 1			/* device -> memory */
 70};
 71
 72static int ps3disk_major;
 73
 74
 75static const struct block_device_operations ps3disk_fops = {
 76	.owner		= THIS_MODULE,
 77};
 78
 79
 80static void ps3disk_scatter_gather(struct ps3_storage_device *dev,
 81				   struct request *req, int gather)
 82{
 83	unsigned int offset = 0;
 84	struct req_iterator iter;
 85	struct bio_vec bvec;
 
 
 
 86
 87	rq_for_each_segment(bvec, req, iter) {
 
 
 
 
 
 
 
 
 88		if (gather)
 89			memcpy_from_bvec(dev->bounce_buf + offset, &bvec);
 90		else
 91			memcpy_to_bvec(&bvec, dev->bounce_buf + offset);
 
 
 
 
 92	}
 93}
 94
 95static blk_status_t ps3disk_submit_request_sg(struct ps3_storage_device *dev,
 96					      struct request *req)
 97{
 98	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
 99	int write = rq_data_dir(req), res;
100	const char *op = write ? "write" : "read";
101	u64 start_sector, sectors;
102	unsigned int region_id = dev->regions[dev->region_idx].id;
103
104#ifdef DEBUG
105	unsigned int n = 0;
106	struct bio_vec bv;
107	struct req_iterator iter;
108
109	rq_for_each_segment(bv, req, iter)
110		n++;
111	dev_dbg(&dev->sbd.core,
112		"%s:%u: %s req has %u bvecs for %u sectors\n",
113		__func__, __LINE__, op, n, blk_rq_sectors(req));
114#endif
115
116	start_sector = blk_rq_pos(req) * priv->blocking_factor;
117	sectors = blk_rq_sectors(req) * priv->blocking_factor;
118	dev_dbg(&dev->sbd.core, "%s:%u: %s %llu sectors starting at %llu\n",
119		__func__, __LINE__, op, sectors, start_sector);
120
121	if (write) {
122		ps3disk_scatter_gather(dev, req, 1);
123
124		res = lv1_storage_write(dev->sbd.dev_id, region_id,
125					start_sector, sectors, 0,
126					dev->bounce_lpar, &dev->tag);
127	} else {
128		res = lv1_storage_read(dev->sbd.dev_id, region_id,
129				       start_sector, sectors, 0,
130				       dev->bounce_lpar, &dev->tag);
131	}
132	if (res) {
133		dev_err(&dev->sbd.core, "%s:%u: %s failed %d\n", __func__,
134			__LINE__, op, res);
135		return BLK_STS_IOERR;
 
136	}
137
138	priv->req = req;
139	return BLK_STS_OK;
140}
141
142static blk_status_t ps3disk_submit_flush_request(struct ps3_storage_device *dev,
143						 struct request *req)
144{
145	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
146	u64 res;
147
148	dev_dbg(&dev->sbd.core, "%s:%u: flush request\n", __func__, __LINE__);
149
150	res = lv1_storage_send_device_command(dev->sbd.dev_id,
151					      LV1_STORAGE_ATA_HDDOUT, 0, 0, 0,
152					      0, &dev->tag);
153	if (res) {
154		dev_err(&dev->sbd.core, "%s:%u: sync cache failed 0x%llx\n",
155			__func__, __LINE__, res);
156		return BLK_STS_IOERR;
 
157	}
158
159	priv->req = req;
160	return BLK_STS_OK;
161}
162
163static blk_status_t ps3disk_do_request(struct ps3_storage_device *dev,
164				       struct request *req)
165{
 
 
166	dev_dbg(&dev->sbd.core, "%s:%u\n", __func__, __LINE__);
167
168	switch (req_op(req)) {
169	case REQ_OP_FLUSH:
170		return ps3disk_submit_flush_request(dev, req);
171	case REQ_OP_READ:
172	case REQ_OP_WRITE:
173		return ps3disk_submit_request_sg(dev, req);
174	default:
175		blk_dump_rq_flags(req, DEVICE_NAME " bad request");
176		return BLK_STS_IOERR;
 
 
 
177	}
178}
179
180static blk_status_t ps3disk_queue_rq(struct blk_mq_hw_ctx *hctx,
181				     const struct blk_mq_queue_data *bd)
182{
183	struct request_queue *q = hctx->queue;
184	struct ps3_storage_device *dev = q->queuedata;
185	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
186	blk_status_t ret;
187
188	blk_mq_start_request(bd->rq);
189
190	spin_lock_irq(&priv->lock);
191	ret = ps3disk_do_request(dev, bd->rq);
192	spin_unlock_irq(&priv->lock);
 
193
194	return ret;
195}
196
197static irqreturn_t ps3disk_interrupt(int irq, void *data)
198{
199	struct ps3_storage_device *dev = data;
200	struct ps3disk_private *priv;
201	struct request *req;
202	int res, read;
203	blk_status_t error;
204	u64 tag, status;
205	const char *op;
206
207	res = lv1_storage_get_async_status(dev->sbd.dev_id, &tag, &status);
208
209	if (tag != dev->tag)
210		dev_err(&dev->sbd.core,
211			"%s:%u: tag mismatch, got %llx, expected %llx\n",
212			__func__, __LINE__, tag, dev->tag);
213
214	if (res) {
215		dev_err(&dev->sbd.core, "%s:%u: res=%d status=0x%llx\n",
216			__func__, __LINE__, res, status);
217		return IRQ_HANDLED;
218	}
219
220	priv = ps3_system_bus_get_drvdata(&dev->sbd);
221	req = priv->req;
222	if (!req) {
223		dev_dbg(&dev->sbd.core,
224			"%s:%u non-block layer request completed\n", __func__,
225			__LINE__);
226		dev->lv1_status = status;
227		complete(&dev->done);
228		return IRQ_HANDLED;
229	}
230
231	if (req_op(req) == REQ_OP_FLUSH) {
232		read = 0;
233		op = "flush";
234	} else {
235		read = !rq_data_dir(req);
236		op = read ? "read" : "write";
237	}
238	if (status) {
239		dev_dbg(&dev->sbd.core, "%s:%u: %s failed 0x%llx\n", __func__,
240			__LINE__, op, status);
241		error = BLK_STS_IOERR;
242	} else {
243		dev_dbg(&dev->sbd.core, "%s:%u: %s completed\n", __func__,
244			__LINE__, op);
245		error = 0;
246		if (read)
247			ps3disk_scatter_gather(dev, req, 0);
248	}
249
250	spin_lock(&priv->lock);
 
251	priv->req = NULL;
252	blk_mq_end_request(req, error);
253	spin_unlock(&priv->lock);
254
255	blk_mq_run_hw_queues(priv->gendisk->queue, true);
256	return IRQ_HANDLED;
257}
258
259static int ps3disk_sync_cache(struct ps3_storage_device *dev)
260{
261	u64 res;
262
263	dev_dbg(&dev->sbd.core, "%s:%u: sync cache\n", __func__, __LINE__);
264
265	res = ps3stor_send_command(dev, LV1_STORAGE_ATA_HDDOUT, 0, 0, 0, 0);
266	if (res) {
267		dev_err(&dev->sbd.core, "%s:%u: sync cache failed 0x%llx\n",
268			__func__, __LINE__, res);
269		return -EIO;
270	}
271	return 0;
272}
273
274
275/* ATA helpers copied from drivers/ata/libata-core.c */
276
277static void swap_buf_le16(u16 *buf, unsigned int buf_words)
278{
279#ifdef __BIG_ENDIAN
280	unsigned int i;
281
282	for (i = 0; i < buf_words; i++)
283		buf[i] = le16_to_cpu(buf[i]);
284#endif /* __BIG_ENDIAN */
285}
286
287static u64 ata_id_n_sectors(const u16 *id)
288{
289	if (ata_id_has_lba(id)) {
290		if (ata_id_has_lba48(id))
291			return ata_id_u64(id, 100);
292		else
293			return ata_id_u32(id, 60);
294	} else {
295		if (ata_id_current_chs_valid(id))
296			return ata_id_u32(id, 57);
297		else
298			return id[1] * id[3] * id[6];
299	}
300}
301
302static void ata_id_string(const u16 *id, unsigned char *s, unsigned int ofs,
303			  unsigned int len)
304{
305	unsigned int c;
306
307	while (len > 0) {
308		c = id[ofs] >> 8;
309		*s = c;
310		s++;
311
312		c = id[ofs] & 0xff;
313		*s = c;
314		s++;
315
316		ofs++;
317		len -= 2;
318	}
319}
320
321static void ata_id_c_string(const u16 *id, unsigned char *s, unsigned int ofs,
322			    unsigned int len)
323{
324	unsigned char *p;
325
326	WARN_ON(!(len & 1));
327
328	ata_id_string(id, s, ofs, len - 1);
329
330	p = s + strnlen(s, len - 1);
331	while (p > s && p[-1] == ' ')
332		p--;
333	*p = '\0';
334}
335
336static int ps3disk_identify(struct ps3_storage_device *dev)
337{
338	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
339	struct lv1_ata_cmnd_block ata_cmnd;
340	u16 *id = dev->bounce_buf;
341	u64 res;
342
343	dev_dbg(&dev->sbd.core, "%s:%u: identify disk\n", __func__, __LINE__);
344
345	memset(&ata_cmnd, 0, sizeof(struct lv1_ata_cmnd_block));
346	ata_cmnd.command = ATA_CMD_ID_ATA;
347	ata_cmnd.sector_count = 1;
348	ata_cmnd.size = ata_cmnd.arglen = ATA_ID_WORDS * 2;
349	ata_cmnd.buffer = dev->bounce_lpar;
350	ata_cmnd.proto = PIO_DATA_IN_PROTO;
351	ata_cmnd.in_out = DIR_READ;
352
353	res = ps3stor_send_command(dev, LV1_STORAGE_SEND_ATA_COMMAND,
354				   ps3_mm_phys_to_lpar(__pa(&ata_cmnd)),
355				   sizeof(ata_cmnd), ata_cmnd.buffer,
356				   ata_cmnd.arglen);
357	if (res) {
358		dev_err(&dev->sbd.core, "%s:%u: identify disk failed 0x%llx\n",
359			__func__, __LINE__, res);
360		return -EIO;
361	}
362
363	swap_buf_le16(id, ATA_ID_WORDS);
364
365	/* All we're interested in are raw capacity and model name */
366	priv->raw_capacity = ata_id_n_sectors(id);
367	ata_id_c_string(id, priv->model, ATA_ID_PROD, sizeof(priv->model));
368	return 0;
369}
370
371static unsigned long ps3disk_mask;
372
373static DEFINE_MUTEX(ps3disk_mask_mutex);
374
375static const struct blk_mq_ops ps3disk_mq_ops = {
376	.queue_rq	= ps3disk_queue_rq,
377};
378
379static int ps3disk_probe(struct ps3_system_bus_device *_dev)
380{
381	struct ps3_storage_device *dev = to_ps3_storage_device(&_dev->core);
382	struct ps3disk_private *priv;
383	int error;
384	unsigned int devidx;
385	struct request_queue *queue;
386	struct gendisk *gendisk;
387
388	if (dev->blk_size < 512) {
389		dev_err(&dev->sbd.core,
390			"%s:%u: cannot handle block size %llu\n", __func__,
391			__LINE__, dev->blk_size);
392		return -EINVAL;
393	}
394
395	BUILD_BUG_ON(PS3DISK_MAX_DISKS > BITS_PER_LONG);
396	mutex_lock(&ps3disk_mask_mutex);
397	devidx = find_first_zero_bit(&ps3disk_mask, PS3DISK_MAX_DISKS);
398	if (devidx >= PS3DISK_MAX_DISKS) {
399		dev_err(&dev->sbd.core, "%s:%u: Too many disks\n", __func__,
400			__LINE__);
401		mutex_unlock(&ps3disk_mask_mutex);
402		return -ENOSPC;
403	}
404	__set_bit(devidx, &ps3disk_mask);
405	mutex_unlock(&ps3disk_mask_mutex);
406
407	priv = kzalloc(sizeof(*priv), GFP_KERNEL);
408	if (!priv) {
409		error = -ENOMEM;
410		goto fail;
411	}
412
413	ps3_system_bus_set_drvdata(_dev, priv);
414	spin_lock_init(&priv->lock);
415
416	dev->bounce_size = BOUNCE_SIZE;
417	dev->bounce_buf = kmalloc(BOUNCE_SIZE, GFP_DMA);
418	if (!dev->bounce_buf) {
419		error = -ENOMEM;
420		goto fail_free_priv;
421	}
422
423	error = ps3stor_setup(dev, ps3disk_interrupt);
424	if (error)
425		goto fail_free_bounce;
426
427	ps3disk_identify(dev);
428
429	error = blk_mq_alloc_sq_tag_set(&priv->tag_set, &ps3disk_mq_ops, 1,
430					BLK_MQ_F_SHOULD_MERGE);
431	if (error)
432		goto fail_teardown;
433
434	gendisk = blk_mq_alloc_disk(&priv->tag_set, dev);
435	if (IS_ERR(gendisk)) {
436		dev_err(&dev->sbd.core, "%s:%u: blk_mq_alloc_disk failed\n",
437			__func__, __LINE__);
438		error = PTR_ERR(gendisk);
439		goto fail_free_tag_set;
440	}
441
442	queue = gendisk->queue;
 
 
 
443
444	blk_queue_max_hw_sectors(queue, dev->bounce_size >> 9);
 
445	blk_queue_dma_alignment(queue, dev->blk_size-1);
446	blk_queue_logical_block_size(queue, dev->blk_size);
447
448	blk_queue_write_cache(queue, true, false);
449
450	blk_queue_max_segments(queue, -1);
451	blk_queue_max_segment_size(queue, dev->bounce_size);
452
 
 
 
 
 
 
 
 
453	priv->gendisk = gendisk;
454	gendisk->major = ps3disk_major;
455	gendisk->first_minor = devidx * PS3DISK_MINORS;
456	gendisk->minors = PS3DISK_MINORS;
457	gendisk->fops = &ps3disk_fops;
 
458	gendisk->private_data = dev;
 
459	snprintf(gendisk->disk_name, sizeof(gendisk->disk_name), PS3DISK_NAME,
460		 devidx+'a');
461	priv->blocking_factor = dev->blk_size >> 9;
462	set_capacity(gendisk,
463		     dev->regions[dev->region_idx].size*priv->blocking_factor);
464
465	dev_info(&dev->sbd.core,
466		 "%s is a %s (%llu MiB total, %llu MiB for OtherOS)\n",
467		 gendisk->disk_name, priv->model, priv->raw_capacity >> 11,
468		 get_capacity(gendisk) >> 11);
469
470	error = device_add_disk(&dev->sbd.core, gendisk, NULL);
471	if (error)
472		goto fail_cleanup_disk;
473
474	return 0;
475fail_cleanup_disk:
476	put_disk(gendisk);
477fail_free_tag_set:
478	blk_mq_free_tag_set(&priv->tag_set);
479fail_teardown:
480	ps3stor_teardown(dev);
481fail_free_bounce:
482	kfree(dev->bounce_buf);
483fail_free_priv:
484	kfree(priv);
485	ps3_system_bus_set_drvdata(_dev, NULL);
486fail:
487	mutex_lock(&ps3disk_mask_mutex);
488	__clear_bit(devidx, &ps3disk_mask);
489	mutex_unlock(&ps3disk_mask_mutex);
490	return error;
491}
492
493static void ps3disk_remove(struct ps3_system_bus_device *_dev)
494{
495	struct ps3_storage_device *dev = to_ps3_storage_device(&_dev->core);
496	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
497
498	mutex_lock(&ps3disk_mask_mutex);
499	__clear_bit(MINOR(disk_devt(priv->gendisk)) / PS3DISK_MINORS,
500		    &ps3disk_mask);
501	mutex_unlock(&ps3disk_mask_mutex);
502	del_gendisk(priv->gendisk);
 
503	put_disk(priv->gendisk);
504	blk_mq_free_tag_set(&priv->tag_set);
505	dev_notice(&dev->sbd.core, "Synchronizing disk cache\n");
506	ps3disk_sync_cache(dev);
507	ps3stor_teardown(dev);
508	kfree(dev->bounce_buf);
509	kfree(priv);
510	ps3_system_bus_set_drvdata(_dev, NULL);
 
511}
512
513static struct ps3_system_bus_driver ps3disk = {
514	.match_id	= PS3_MATCH_ID_STOR_DISK,
515	.core.name	= DEVICE_NAME,
516	.core.owner	= THIS_MODULE,
517	.probe		= ps3disk_probe,
518	.remove		= ps3disk_remove,
519	.shutdown	= ps3disk_remove,
520};
521
522
523static int __init ps3disk_init(void)
524{
525	int error;
526
527	if (!firmware_has_feature(FW_FEATURE_PS3_LV1))
528		return -ENODEV;
529
530	error = register_blkdev(0, DEVICE_NAME);
531	if (error <= 0) {
532		printk(KERN_ERR "%s:%u: register_blkdev failed %d\n", __func__,
533		       __LINE__, error);
534		return error;
535	}
536	ps3disk_major = error;
537
538	pr_info("%s:%u: registered block device major %d\n", __func__,
539		__LINE__, ps3disk_major);
540
541	error = ps3_system_bus_driver_register(&ps3disk);
542	if (error)
543		unregister_blkdev(ps3disk_major, DEVICE_NAME);
544
545	return error;
546}
547
548static void __exit ps3disk_exit(void)
549{
550	ps3_system_bus_driver_unregister(&ps3disk);
551	unregister_blkdev(ps3disk_major, DEVICE_NAME);
552}
553
554module_init(ps3disk_init);
555module_exit(ps3disk_exit);
556
557MODULE_LICENSE("GPL");
558MODULE_DESCRIPTION("PS3 Disk Storage Driver");
559MODULE_AUTHOR("Sony Corporation");
560MODULE_ALIAS(PS3_MODULE_ALIAS_STOR_DISK);
v3.1
 
  1/*
  2 * PS3 Disk Storage Driver
  3 *
  4 * Copyright (C) 2007 Sony Computer Entertainment Inc.
  5 * Copyright 2007 Sony Corp.
  6 *
  7 * This program is free software; you can redistribute it and/or modify it
  8 * under the terms of the GNU General Public License as published
  9 * by the Free Software Foundation; version 2 of the License.
 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, write to the Free Software Foundation, Inc.,
 18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 19 */
 20
 21#include <linux/ata.h>
 22#include <linux/blkdev.h>
 23#include <linux/slab.h>
 
 24
 25#include <asm/lv1call.h>
 26#include <asm/ps3stor.h>
 27#include <asm/firmware.h>
 28
 29
 30#define DEVICE_NAME		"ps3disk"
 31
 32#define BOUNCE_SIZE		(64*1024)
 33
 34#define PS3DISK_MAX_DISKS	16
 35#define PS3DISK_MINORS		16
 36
 37
 38#define PS3DISK_NAME		"ps3d%c"
 39
 40
 41struct ps3disk_private {
 42	spinlock_t lock;		/* Request queue spinlock */
 43	struct request_queue *queue;
 44	struct gendisk *gendisk;
 45	unsigned int blocking_factor;
 46	struct request *req;
 47	u64 raw_capacity;
 48	unsigned char model[ATA_ID_PROD_LEN+1];
 49};
 50
 51
 52#define LV1_STORAGE_SEND_ATA_COMMAND	(2)
 53#define LV1_STORAGE_ATA_HDDOUT		(0x23)
 54
 55struct lv1_ata_cmnd_block {
 56	u16	features;
 57	u16	sector_count;
 58	u16	LBA_low;
 59	u16	LBA_mid;
 60	u16	LBA_high;
 61	u8	device;
 62	u8	command;
 63	u32	is_ext;
 64	u32	proto;
 65	u32	in_out;
 66	u32	size;
 67	u64	buffer;
 68	u32	arglen;
 69};
 70
 71enum lv1_ata_proto {
 72	NON_DATA_PROTO     = 0,
 73	PIO_DATA_IN_PROTO  = 1,
 74	PIO_DATA_OUT_PROTO = 2,
 75	DMA_PROTO = 3
 76};
 77
 78enum lv1_ata_in_out {
 79	DIR_WRITE = 0,			/* memory -> device */
 80	DIR_READ = 1			/* device -> memory */
 81};
 82
 83static int ps3disk_major;
 84
 85
 86static const struct block_device_operations ps3disk_fops = {
 87	.owner		= THIS_MODULE,
 88};
 89
 90
 91static void ps3disk_scatter_gather(struct ps3_storage_device *dev,
 92				   struct request *req, int gather)
 93{
 94	unsigned int offset = 0;
 95	struct req_iterator iter;
 96	struct bio_vec *bvec;
 97	unsigned int i = 0;
 98	size_t size;
 99	void *buf;
100
101	rq_for_each_segment(bvec, req, iter) {
102		unsigned long flags;
103		dev_dbg(&dev->sbd.core,
104			"%s:%u: bio %u: %u segs %u sectors from %lu\n",
105			__func__, __LINE__, i, bio_segments(iter.bio),
106			bio_sectors(iter.bio), iter.bio->bi_sector);
107
108		size = bvec->bv_len;
109		buf = bvec_kmap_irq(bvec, &flags);
110		if (gather)
111			memcpy(dev->bounce_buf+offset, buf, size);
112		else
113			memcpy(buf, dev->bounce_buf+offset, size);
114		offset += size;
115		flush_kernel_dcache_page(bvec->bv_page);
116		bvec_kunmap_irq(buf, &flags);
117		i++;
118	}
119}
120
121static int ps3disk_submit_request_sg(struct ps3_storage_device *dev,
122				     struct request *req)
123{
124	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
125	int write = rq_data_dir(req), res;
126	const char *op = write ? "write" : "read";
127	u64 start_sector, sectors;
128	unsigned int region_id = dev->regions[dev->region_idx].id;
129
130#ifdef DEBUG
131	unsigned int n = 0;
132	struct bio_vec *bv;
133	struct req_iterator iter;
134
135	rq_for_each_segment(bv, req, iter)
136		n++;
137	dev_dbg(&dev->sbd.core,
138		"%s:%u: %s req has %u bvecs for %u sectors\n",
139		__func__, __LINE__, op, n, blk_rq_sectors(req));
140#endif
141
142	start_sector = blk_rq_pos(req) * priv->blocking_factor;
143	sectors = blk_rq_sectors(req) * priv->blocking_factor;
144	dev_dbg(&dev->sbd.core, "%s:%u: %s %llu sectors starting at %llu\n",
145		__func__, __LINE__, op, sectors, start_sector);
146
147	if (write) {
148		ps3disk_scatter_gather(dev, req, 1);
149
150		res = lv1_storage_write(dev->sbd.dev_id, region_id,
151					start_sector, sectors, 0,
152					dev->bounce_lpar, &dev->tag);
153	} else {
154		res = lv1_storage_read(dev->sbd.dev_id, region_id,
155				       start_sector, sectors, 0,
156				       dev->bounce_lpar, &dev->tag);
157	}
158	if (res) {
159		dev_err(&dev->sbd.core, "%s:%u: %s failed %d\n", __func__,
160			__LINE__, op, res);
161		__blk_end_request_all(req, -EIO);
162		return 0;
163	}
164
165	priv->req = req;
166	return 1;
167}
168
169static int ps3disk_submit_flush_request(struct ps3_storage_device *dev,
170					struct request *req)
171{
172	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
173	u64 res;
174
175	dev_dbg(&dev->sbd.core, "%s:%u: flush request\n", __func__, __LINE__);
176
177	res = lv1_storage_send_device_command(dev->sbd.dev_id,
178					      LV1_STORAGE_ATA_HDDOUT, 0, 0, 0,
179					      0, &dev->tag);
180	if (res) {
181		dev_err(&dev->sbd.core, "%s:%u: sync cache failed 0x%llx\n",
182			__func__, __LINE__, res);
183		__blk_end_request_all(req, -EIO);
184		return 0;
185	}
186
187	priv->req = req;
188	return 1;
189}
190
191static void ps3disk_do_request(struct ps3_storage_device *dev,
192			       struct request_queue *q)
193{
194	struct request *req;
195
196	dev_dbg(&dev->sbd.core, "%s:%u\n", __func__, __LINE__);
197
198	while ((req = blk_fetch_request(q))) {
199		if (req->cmd_flags & REQ_FLUSH) {
200			if (ps3disk_submit_flush_request(dev, req))
201				break;
202		} else if (req->cmd_type == REQ_TYPE_FS) {
203			if (ps3disk_submit_request_sg(dev, req))
204				break;
205		} else {
206			blk_dump_rq_flags(req, DEVICE_NAME " bad request");
207			__blk_end_request_all(req, -EIO);
208			continue;
209		}
210	}
211}
212
213static void ps3disk_request(struct request_queue *q)
 
214{
 
215	struct ps3_storage_device *dev = q->queuedata;
216	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
 
 
 
217
218	if (priv->req) {
219		dev_dbg(&dev->sbd.core, "%s:%u busy\n", __func__, __LINE__);
220		return;
221	}
222
223	ps3disk_do_request(dev, q);
224}
225
226static irqreturn_t ps3disk_interrupt(int irq, void *data)
227{
228	struct ps3_storage_device *dev = data;
229	struct ps3disk_private *priv;
230	struct request *req;
231	int res, read, error;
 
232	u64 tag, status;
233	const char *op;
234
235	res = lv1_storage_get_async_status(dev->sbd.dev_id, &tag, &status);
236
237	if (tag != dev->tag)
238		dev_err(&dev->sbd.core,
239			"%s:%u: tag mismatch, got %llx, expected %llx\n",
240			__func__, __LINE__, tag, dev->tag);
241
242	if (res) {
243		dev_err(&dev->sbd.core, "%s:%u: res=%d status=0x%llx\n",
244			__func__, __LINE__, res, status);
245		return IRQ_HANDLED;
246	}
247
248	priv = ps3_system_bus_get_drvdata(&dev->sbd);
249	req = priv->req;
250	if (!req) {
251		dev_dbg(&dev->sbd.core,
252			"%s:%u non-block layer request completed\n", __func__,
253			__LINE__);
254		dev->lv1_status = status;
255		complete(&dev->done);
256		return IRQ_HANDLED;
257	}
258
259	if (req->cmd_flags & REQ_FLUSH) {
260		read = 0;
261		op = "flush";
262	} else {
263		read = !rq_data_dir(req);
264		op = read ? "read" : "write";
265	}
266	if (status) {
267		dev_dbg(&dev->sbd.core, "%s:%u: %s failed 0x%llx\n", __func__,
268			__LINE__, op, status);
269		error = -EIO;
270	} else {
271		dev_dbg(&dev->sbd.core, "%s:%u: %s completed\n", __func__,
272			__LINE__, op);
273		error = 0;
274		if (read)
275			ps3disk_scatter_gather(dev, req, 0);
276	}
277
278	spin_lock(&priv->lock);
279	__blk_end_request_all(req, error);
280	priv->req = NULL;
281	ps3disk_do_request(dev, priv->queue);
282	spin_unlock(&priv->lock);
283
 
284	return IRQ_HANDLED;
285}
286
287static int ps3disk_sync_cache(struct ps3_storage_device *dev)
288{
289	u64 res;
290
291	dev_dbg(&dev->sbd.core, "%s:%u: sync cache\n", __func__, __LINE__);
292
293	res = ps3stor_send_command(dev, LV1_STORAGE_ATA_HDDOUT, 0, 0, 0, 0);
294	if (res) {
295		dev_err(&dev->sbd.core, "%s:%u: sync cache failed 0x%llx\n",
296			__func__, __LINE__, res);
297		return -EIO;
298	}
299	return 0;
300}
301
302
303/* ATA helpers copied from drivers/ata/libata-core.c */
304
305static void swap_buf_le16(u16 *buf, unsigned int buf_words)
306{
307#ifdef __BIG_ENDIAN
308	unsigned int i;
309
310	for (i = 0; i < buf_words; i++)
311		buf[i] = le16_to_cpu(buf[i]);
312#endif /* __BIG_ENDIAN */
313}
314
315static u64 ata_id_n_sectors(const u16 *id)
316{
317	if (ata_id_has_lba(id)) {
318		if (ata_id_has_lba48(id))
319			return ata_id_u64(id, 100);
320		else
321			return ata_id_u32(id, 60);
322	} else {
323		if (ata_id_current_chs_valid(id))
324			return ata_id_u32(id, 57);
325		else
326			return id[1] * id[3] * id[6];
327	}
328}
329
330static void ata_id_string(const u16 *id, unsigned char *s, unsigned int ofs,
331			  unsigned int len)
332{
333	unsigned int c;
334
335	while (len > 0) {
336		c = id[ofs] >> 8;
337		*s = c;
338		s++;
339
340		c = id[ofs] & 0xff;
341		*s = c;
342		s++;
343
344		ofs++;
345		len -= 2;
346	}
347}
348
349static void ata_id_c_string(const u16 *id, unsigned char *s, unsigned int ofs,
350			    unsigned int len)
351{
352	unsigned char *p;
353
354	WARN_ON(!(len & 1));
355
356	ata_id_string(id, s, ofs, len - 1);
357
358	p = s + strnlen(s, len - 1);
359	while (p > s && p[-1] == ' ')
360		p--;
361	*p = '\0';
362}
363
364static int ps3disk_identify(struct ps3_storage_device *dev)
365{
366	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
367	struct lv1_ata_cmnd_block ata_cmnd;
368	u16 *id = dev->bounce_buf;
369	u64 res;
370
371	dev_dbg(&dev->sbd.core, "%s:%u: identify disk\n", __func__, __LINE__);
372
373	memset(&ata_cmnd, 0, sizeof(struct lv1_ata_cmnd_block));
374	ata_cmnd.command = ATA_CMD_ID_ATA;
375	ata_cmnd.sector_count = 1;
376	ata_cmnd.size = ata_cmnd.arglen = ATA_ID_WORDS * 2;
377	ata_cmnd.buffer = dev->bounce_lpar;
378	ata_cmnd.proto = PIO_DATA_IN_PROTO;
379	ata_cmnd.in_out = DIR_READ;
380
381	res = ps3stor_send_command(dev, LV1_STORAGE_SEND_ATA_COMMAND,
382				   ps3_mm_phys_to_lpar(__pa(&ata_cmnd)),
383				   sizeof(ata_cmnd), ata_cmnd.buffer,
384				   ata_cmnd.arglen);
385	if (res) {
386		dev_err(&dev->sbd.core, "%s:%u: identify disk failed 0x%llx\n",
387			__func__, __LINE__, res);
388		return -EIO;
389	}
390
391	swap_buf_le16(id, ATA_ID_WORDS);
392
393	/* All we're interested in are raw capacity and model name */
394	priv->raw_capacity = ata_id_n_sectors(id);
395	ata_id_c_string(id, priv->model, ATA_ID_PROD, sizeof(priv->model));
396	return 0;
397}
398
399static unsigned long ps3disk_mask;
400
401static DEFINE_MUTEX(ps3disk_mask_mutex);
402
403static int __devinit ps3disk_probe(struct ps3_system_bus_device *_dev)
 
 
 
 
404{
405	struct ps3_storage_device *dev = to_ps3_storage_device(&_dev->core);
406	struct ps3disk_private *priv;
407	int error;
408	unsigned int devidx;
409	struct request_queue *queue;
410	struct gendisk *gendisk;
411
412	if (dev->blk_size < 512) {
413		dev_err(&dev->sbd.core,
414			"%s:%u: cannot handle block size %llu\n", __func__,
415			__LINE__, dev->blk_size);
416		return -EINVAL;
417	}
418
419	BUILD_BUG_ON(PS3DISK_MAX_DISKS > BITS_PER_LONG);
420	mutex_lock(&ps3disk_mask_mutex);
421	devidx = find_first_zero_bit(&ps3disk_mask, PS3DISK_MAX_DISKS);
422	if (devidx >= PS3DISK_MAX_DISKS) {
423		dev_err(&dev->sbd.core, "%s:%u: Too many disks\n", __func__,
424			__LINE__);
425		mutex_unlock(&ps3disk_mask_mutex);
426		return -ENOSPC;
427	}
428	__set_bit(devidx, &ps3disk_mask);
429	mutex_unlock(&ps3disk_mask_mutex);
430
431	priv = kzalloc(sizeof(*priv), GFP_KERNEL);
432	if (!priv) {
433		error = -ENOMEM;
434		goto fail;
435	}
436
437	ps3_system_bus_set_drvdata(_dev, priv);
438	spin_lock_init(&priv->lock);
439
440	dev->bounce_size = BOUNCE_SIZE;
441	dev->bounce_buf = kmalloc(BOUNCE_SIZE, GFP_DMA);
442	if (!dev->bounce_buf) {
443		error = -ENOMEM;
444		goto fail_free_priv;
445	}
446
447	error = ps3stor_setup(dev, ps3disk_interrupt);
448	if (error)
449		goto fail_free_bounce;
450
451	ps3disk_identify(dev);
452
453	queue = blk_init_queue(ps3disk_request, &priv->lock);
454	if (!queue) {
455		dev_err(&dev->sbd.core, "%s:%u: blk_init_queue failed\n",
 
 
 
 
 
456			__func__, __LINE__);
457		error = -ENOMEM;
458		goto fail_teardown;
459	}
460
461	priv->queue = queue;
462	queue->queuedata = dev;
463
464	blk_queue_bounce_limit(queue, BLK_BOUNCE_HIGH);
465
466	blk_queue_max_hw_sectors(queue, dev->bounce_size >> 9);
467	blk_queue_segment_boundary(queue, -1UL);
468	blk_queue_dma_alignment(queue, dev->blk_size-1);
469	blk_queue_logical_block_size(queue, dev->blk_size);
470
471	blk_queue_flush(queue, REQ_FLUSH);
472
473	blk_queue_max_segments(queue, -1);
474	blk_queue_max_segment_size(queue, dev->bounce_size);
475
476	gendisk = alloc_disk(PS3DISK_MINORS);
477	if (!gendisk) {
478		dev_err(&dev->sbd.core, "%s:%u: alloc_disk failed\n", __func__,
479			__LINE__);
480		error = -ENOMEM;
481		goto fail_cleanup_queue;
482	}
483
484	priv->gendisk = gendisk;
485	gendisk->major = ps3disk_major;
486	gendisk->first_minor = devidx * PS3DISK_MINORS;
 
487	gendisk->fops = &ps3disk_fops;
488	gendisk->queue = queue;
489	gendisk->private_data = dev;
490	gendisk->driverfs_dev = &dev->sbd.core;
491	snprintf(gendisk->disk_name, sizeof(gendisk->disk_name), PS3DISK_NAME,
492		 devidx+'a');
493	priv->blocking_factor = dev->blk_size >> 9;
494	set_capacity(gendisk,
495		     dev->regions[dev->region_idx].size*priv->blocking_factor);
496
497	dev_info(&dev->sbd.core,
498		 "%s is a %s (%llu MiB total, %lu MiB for OtherOS)\n",
499		 gendisk->disk_name, priv->model, priv->raw_capacity >> 11,
500		 get_capacity(gendisk) >> 11);
501
502	add_disk(gendisk);
 
 
 
503	return 0;
504
505fail_cleanup_queue:
506	blk_cleanup_queue(queue);
 
507fail_teardown:
508	ps3stor_teardown(dev);
509fail_free_bounce:
510	kfree(dev->bounce_buf);
511fail_free_priv:
512	kfree(priv);
513	ps3_system_bus_set_drvdata(_dev, NULL);
514fail:
515	mutex_lock(&ps3disk_mask_mutex);
516	__clear_bit(devidx, &ps3disk_mask);
517	mutex_unlock(&ps3disk_mask_mutex);
518	return error;
519}
520
521static int ps3disk_remove(struct ps3_system_bus_device *_dev)
522{
523	struct ps3_storage_device *dev = to_ps3_storage_device(&_dev->core);
524	struct ps3disk_private *priv = ps3_system_bus_get_drvdata(&dev->sbd);
525
526	mutex_lock(&ps3disk_mask_mutex);
527	__clear_bit(MINOR(disk_devt(priv->gendisk)) / PS3DISK_MINORS,
528		    &ps3disk_mask);
529	mutex_unlock(&ps3disk_mask_mutex);
530	del_gendisk(priv->gendisk);
531	blk_cleanup_queue(priv->queue);
532	put_disk(priv->gendisk);
 
533	dev_notice(&dev->sbd.core, "Synchronizing disk cache\n");
534	ps3disk_sync_cache(dev);
535	ps3stor_teardown(dev);
536	kfree(dev->bounce_buf);
537	kfree(priv);
538	ps3_system_bus_set_drvdata(_dev, NULL);
539	return 0;
540}
541
542static struct ps3_system_bus_driver ps3disk = {
543	.match_id	= PS3_MATCH_ID_STOR_DISK,
544	.core.name	= DEVICE_NAME,
545	.core.owner	= THIS_MODULE,
546	.probe		= ps3disk_probe,
547	.remove		= ps3disk_remove,
548	.shutdown	= ps3disk_remove,
549};
550
551
552static int __init ps3disk_init(void)
553{
554	int error;
555
556	if (!firmware_has_feature(FW_FEATURE_PS3_LV1))
557		return -ENODEV;
558
559	error = register_blkdev(0, DEVICE_NAME);
560	if (error <= 0) {
561		printk(KERN_ERR "%s:%u: register_blkdev failed %d\n", __func__,
562		       __LINE__, error);
563		return error;
564	}
565	ps3disk_major = error;
566
567	pr_info("%s:%u: registered block device major %d\n", __func__,
568		__LINE__, ps3disk_major);
569
570	error = ps3_system_bus_driver_register(&ps3disk);
571	if (error)
572		unregister_blkdev(ps3disk_major, DEVICE_NAME);
573
574	return error;
575}
576
577static void __exit ps3disk_exit(void)
578{
579	ps3_system_bus_driver_unregister(&ps3disk);
580	unregister_blkdev(ps3disk_major, DEVICE_NAME);
581}
582
583module_init(ps3disk_init);
584module_exit(ps3disk_exit);
585
586MODULE_LICENSE("GPL");
587MODULE_DESCRIPTION("PS3 Disk Storage Driver");
588MODULE_AUTHOR("Sony Corporation");
589MODULE_ALIAS(PS3_MODULE_ALIAS_STOR_DISK);