Merge branch 'amd-iommu/fixes' of git://git.kernel.org/pub/scm/linux/kernel/git/joro...
[linux-2.6] / drivers / block / mg_disk.c
1 /*
2  *  drivers/block/mg_disk.c
3  *
4  *  Support for the mGine m[g]flash IO mode.
5  *  Based on legacy hd.c
6  *
7  * (c) 2008 mGine Co.,LTD
8  * (c) 2008 unsik Kim <donari75@gmail.com>
9  *
10  *  This program is free software; you can redistribute it and/or modify
11  *  it under the terms of the GNU General Public License version 2 as
12  *  published by the Free Software Foundation.
13  */
14
15 #include <linux/kernel.h>
16 #include <linux/module.h>
17 #include <linux/fs.h>
18 #include <linux/blkdev.h>
19 #include <linux/hdreg.h>
20 #include <linux/ata.h>
21 #include <linux/interrupt.h>
22 #include <linux/delay.h>
23 #include <linux/platform_device.h>
24 #include <linux/gpio.h>
25 #include <linux/mg_disk.h>
26
27 #define MG_RES_SEC (CONFIG_MG_DISK_RES << 1)
28
29 /* name for block device */
30 #define MG_DISK_NAME "mgd"
31
32 #define MG_DISK_MAJ 0
33 #define MG_DISK_MAX_PART 16
34 #define MG_SECTOR_SIZE 512
35 #define MG_MAX_SECTS 256
36
37 /* Register offsets */
38 #define MG_BUFF_OFFSET                  0x8000
39 #define MG_STORAGE_BUFFER_SIZE          0x200
40 #define MG_REG_OFFSET                   0xC000
41 #define MG_REG_FEATURE                  (MG_REG_OFFSET + 2)     /* write case */
42 #define MG_REG_ERROR                    (MG_REG_OFFSET + 2)     /* read case */
43 #define MG_REG_SECT_CNT                 (MG_REG_OFFSET + 4)
44 #define MG_REG_SECT_NUM                 (MG_REG_OFFSET + 6)
45 #define MG_REG_CYL_LOW                  (MG_REG_OFFSET + 8)
46 #define MG_REG_CYL_HIGH                 (MG_REG_OFFSET + 0xA)
47 #define MG_REG_DRV_HEAD                 (MG_REG_OFFSET + 0xC)
48 #define MG_REG_COMMAND                  (MG_REG_OFFSET + 0xE)   /* write case */
49 #define MG_REG_STATUS                   (MG_REG_OFFSET + 0xE)   /* read  case */
50 #define MG_REG_DRV_CTRL                 (MG_REG_OFFSET + 0x10)
51 #define MG_REG_BURST_CTRL               (MG_REG_OFFSET + 0x12)
52
53 /* handy status */
54 #define MG_STAT_READY   (ATA_DRDY | ATA_DSC)
55 #define MG_READY_OK(s)  (((s) & (MG_STAT_READY | (ATA_BUSY | ATA_DF | \
56                                  ATA_ERR))) == MG_STAT_READY)
57
58 /* error code for others */
59 #define MG_ERR_NONE             0
60 #define MG_ERR_TIMEOUT          0x100
61 #define MG_ERR_INIT_STAT        0x101
62 #define MG_ERR_TRANSLATION      0x102
63 #define MG_ERR_CTRL_RST         0x103
64 #define MG_ERR_INV_STAT         0x104
65 #define MG_ERR_RSTOUT           0x105
66
67 #define MG_MAX_ERRORS   6       /* Max read/write errors */
68
69 /* command */
70 #define MG_CMD_RD 0x20
71 #define MG_CMD_WR 0x30
72 #define MG_CMD_SLEEP 0x99
73 #define MG_CMD_WAKEUP 0xC3
74 #define MG_CMD_ID 0xEC
75 #define MG_CMD_WR_CONF 0x3C
76 #define MG_CMD_RD_CONF 0x40
77
78 /* operation mode */
79 #define MG_OP_CASCADE (1 << 0)
80 #define MG_OP_CASCADE_SYNC_RD (1 << 1)
81 #define MG_OP_CASCADE_SYNC_WR (1 << 2)
82 #define MG_OP_INTERLEAVE (1 << 3)
83
84 /* synchronous */
85 #define MG_BURST_LAT_4 (3 << 4)
86 #define MG_BURST_LAT_5 (4 << 4)
87 #define MG_BURST_LAT_6 (5 << 4)
88 #define MG_BURST_LAT_7 (6 << 4)
89 #define MG_BURST_LAT_8 (7 << 4)
90 #define MG_BURST_LEN_4 (1 << 1)
91 #define MG_BURST_LEN_8 (2 << 1)
92 #define MG_BURST_LEN_16 (3 << 1)
93 #define MG_BURST_LEN_32 (4 << 1)
94 #define MG_BURST_LEN_CONT (0 << 1)
95
96 /* timeout value (unit: ms) */
97 #define MG_TMAX_CONF_TO_CMD     1
98 #define MG_TMAX_WAIT_RD_DRQ     10
99 #define MG_TMAX_WAIT_WR_DRQ     500
100 #define MG_TMAX_RST_TO_BUSY     10
101 #define MG_TMAX_HDRST_TO_RDY    500
102 #define MG_TMAX_SWRST_TO_RDY    500
103 #define MG_TMAX_RSTOUT          3000
104
105 #define MG_DEV_MASK (MG_BOOT_DEV | MG_STORAGE_DEV | MG_STORAGE_DEV_SKIP_RST)
106
107 /* main structure for mflash driver */
108 struct mg_host {
109         struct device *dev;
110
111         struct request_queue *breq;
112         struct request *req;
113         spinlock_t lock;
114         struct gendisk *gd;
115
116         struct timer_list timer;
117         void (*mg_do_intr) (struct mg_host *);
118
119         u16 id[ATA_ID_WORDS];
120
121         u16 cyls;
122         u16 heads;
123         u16 sectors;
124         u32 n_sectors;
125         u32 nres_sectors;
126
127         void __iomem *dev_base;
128         unsigned int irq;
129         unsigned int rst;
130         unsigned int rstout;
131
132         u32 major;
133         u32 error;
134 };
135
136 /*
137  * Debugging macro and defines
138  */
139 #undef DO_MG_DEBUG
140 #ifdef DO_MG_DEBUG
141 #  define MG_DBG(fmt, args...) \
142         printk(KERN_DEBUG "%s:%d "fmt, __func__, __LINE__, ##args)
143 #else /* CONFIG_MG_DEBUG */
144 #  define MG_DBG(fmt, args...) do { } while (0)
145 #endif /* CONFIG_MG_DEBUG */
146
147 static void mg_request(struct request_queue *);
148
149 static bool mg_end_request(struct mg_host *host, int err, unsigned int nr_bytes)
150 {
151         if (__blk_end_request(host->req, err, nr_bytes))
152                 return true;
153
154         host->req = NULL;
155         return false;
156 }
157
158 static bool mg_end_request_cur(struct mg_host *host, int err)
159 {
160         return mg_end_request(host, err, blk_rq_cur_bytes(host->req));
161 }
162
163 static void mg_dump_status(const char *msg, unsigned int stat,
164                 struct mg_host *host)
165 {
166         char *name = MG_DISK_NAME;
167
168         if (host->req)
169                 name = host->req->rq_disk->disk_name;
170
171         printk(KERN_ERR "%s: %s: status=0x%02x { ", name, msg, stat & 0xff);
172         if (stat & ATA_BUSY)
173                 printk("Busy ");
174         if (stat & ATA_DRDY)
175                 printk("DriveReady ");
176         if (stat & ATA_DF)
177                 printk("WriteFault ");
178         if (stat & ATA_DSC)
179                 printk("SeekComplete ");
180         if (stat & ATA_DRQ)
181                 printk("DataRequest ");
182         if (stat & ATA_CORR)
183                 printk("CorrectedError ");
184         if (stat & ATA_ERR)
185                 printk("Error ");
186         printk("}\n");
187         if ((stat & ATA_ERR) == 0) {
188                 host->error = 0;
189         } else {
190                 host->error = inb((unsigned long)host->dev_base + MG_REG_ERROR);
191                 printk(KERN_ERR "%s: %s: error=0x%02x { ", name, msg,
192                                 host->error & 0xff);
193                 if (host->error & ATA_BBK)
194                         printk("BadSector ");
195                 if (host->error & ATA_UNC)
196                         printk("UncorrectableError ");
197                 if (host->error & ATA_IDNF)
198                         printk("SectorIdNotFound ");
199                 if (host->error & ATA_ABORTED)
200                         printk("DriveStatusError ");
201                 if (host->error & ATA_AMNF)
202                         printk("AddrMarkNotFound ");
203                 printk("}");
204                 if (host->error & (ATA_BBK | ATA_UNC | ATA_IDNF | ATA_AMNF)) {
205                         if (host->req)
206                                 printk(", sector=%u",
207                                        (unsigned int)blk_rq_pos(host->req));
208                 }
209                 printk("\n");
210         }
211 }
212
213 static unsigned int mg_wait(struct mg_host *host, u32 expect, u32 msec)
214 {
215         u8 status;
216         unsigned long expire, cur_jiffies;
217         struct mg_drv_data *prv_data = host->dev->platform_data;
218
219         host->error = MG_ERR_NONE;
220         expire = jiffies + msecs_to_jiffies(msec);
221
222         status = inb((unsigned long)host->dev_base + MG_REG_STATUS);
223
224         do {
225                 cur_jiffies = jiffies;
226                 if (status & ATA_BUSY) {
227                         if (expect == ATA_BUSY)
228                                 break;
229                 } else {
230                         /* Check the error condition! */
231                         if (status & ATA_ERR) {
232                                 mg_dump_status("mg_wait", status, host);
233                                 break;
234                         }
235
236                         if (expect == MG_STAT_READY)
237                                 if (MG_READY_OK(status))
238                                         break;
239
240                         if (expect == ATA_DRQ)
241                                 if (status & ATA_DRQ)
242                                         break;
243                 }
244                 if (!msec) {
245                         mg_dump_status("not ready", status, host);
246                         return MG_ERR_INV_STAT;
247                 }
248                 if (prv_data->use_polling)
249                         msleep(1);
250
251                 status = inb((unsigned long)host->dev_base + MG_REG_STATUS);
252         } while (time_before(cur_jiffies, expire));
253
254         if (time_after_eq(cur_jiffies, expire) && msec)
255                 host->error = MG_ERR_TIMEOUT;
256
257         return host->error;
258 }
259
260 static unsigned int mg_wait_rstout(u32 rstout, u32 msec)
261 {
262         unsigned long expire;
263
264         expire = jiffies + msecs_to_jiffies(msec);
265         while (time_before(jiffies, expire)) {
266                 if (gpio_get_value(rstout) == 1)
267                         return MG_ERR_NONE;
268                 msleep(10);
269         }
270
271         return MG_ERR_RSTOUT;
272 }
273
274 static void mg_unexpected_intr(struct mg_host *host)
275 {
276         u32 status = inb((unsigned long)host->dev_base + MG_REG_STATUS);
277
278         mg_dump_status("mg_unexpected_intr", status, host);
279 }
280
281 static irqreturn_t mg_irq(int irq, void *dev_id)
282 {
283         struct mg_host *host = dev_id;
284         void (*handler)(struct mg_host *) = host->mg_do_intr;
285
286         spin_lock(&host->lock);
287
288         host->mg_do_intr = NULL;
289         del_timer(&host->timer);
290         if (!handler)
291                 handler = mg_unexpected_intr;
292         handler(host);
293
294         spin_unlock(&host->lock);
295
296         return IRQ_HANDLED;
297 }
298
299 /* local copy of ata_id_string() */
300 static void mg_id_string(const u16 *id, unsigned char *s,
301                          unsigned int ofs, unsigned int len)
302 {
303         unsigned int c;
304
305         BUG_ON(len & 1);
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
321 /* local copy of ata_id_c_string() */
322 static void mg_id_c_string(const u16 *id, unsigned char *s,
323                            unsigned int ofs, unsigned int len)
324 {
325         unsigned char *p;
326
327         mg_id_string(id, s, ofs, len - 1);
328
329         p = s + strnlen(s, len - 1);
330         while (p > s && p[-1] == ' ')
331                 p--;
332         *p = '\0';
333 }
334
335 static int mg_get_disk_id(struct mg_host *host)
336 {
337         u32 i;
338         s32 err;
339         const u16 *id = host->id;
340         struct mg_drv_data *prv_data = host->dev->platform_data;
341         char fwrev[ATA_ID_FW_REV_LEN + 1];
342         char model[ATA_ID_PROD_LEN + 1];
343         char serial[ATA_ID_SERNO_LEN + 1];
344
345         if (!prv_data->use_polling)
346                 outb(ATA_NIEN, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
347
348         outb(MG_CMD_ID, (unsigned long)host->dev_base + MG_REG_COMMAND);
349         err = mg_wait(host, ATA_DRQ, MG_TMAX_WAIT_RD_DRQ);
350         if (err)
351                 return err;
352
353         for (i = 0; i < (MG_SECTOR_SIZE >> 1); i++)
354                 host->id[i] = le16_to_cpu(inw((unsigned long)host->dev_base +
355                                         MG_BUFF_OFFSET + i * 2));
356
357         outb(MG_CMD_RD_CONF, (unsigned long)host->dev_base + MG_REG_COMMAND);
358         err = mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD);
359         if (err)
360                 return err;
361
362         if ((id[ATA_ID_FIELD_VALID] & 1) == 0)
363                 return MG_ERR_TRANSLATION;
364
365         host->n_sectors = ata_id_u32(id, ATA_ID_LBA_CAPACITY);
366         host->cyls = id[ATA_ID_CYLS];
367         host->heads = id[ATA_ID_HEADS];
368         host->sectors = id[ATA_ID_SECTORS];
369
370         if (MG_RES_SEC && host->heads && host->sectors) {
371                 /* modify cyls, n_sectors */
372                 host->cyls = (host->n_sectors - MG_RES_SEC) /
373                         host->heads / host->sectors;
374                 host->nres_sectors = host->n_sectors - host->cyls *
375                         host->heads * host->sectors;
376                 host->n_sectors -= host->nres_sectors;
377         }
378
379         mg_id_c_string(id, fwrev, ATA_ID_FW_REV, sizeof(fwrev));
380         mg_id_c_string(id, model, ATA_ID_PROD, sizeof(model));
381         mg_id_c_string(id, serial, ATA_ID_SERNO, sizeof(serial));
382         printk(KERN_INFO "mg_disk: model: %s\n", model);
383         printk(KERN_INFO "mg_disk: firm: %.8s\n", fwrev);
384         printk(KERN_INFO "mg_disk: serial: %s\n", serial);
385         printk(KERN_INFO "mg_disk: %d + reserved %d sectors\n",
386                         host->n_sectors, host->nres_sectors);
387
388         if (!prv_data->use_polling)
389                 outb(0, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
390
391         return err;
392 }
393
394
395 static int mg_disk_init(struct mg_host *host)
396 {
397         struct mg_drv_data *prv_data = host->dev->platform_data;
398         s32 err;
399         u8 init_status;
400
401         /* hdd rst low */
402         gpio_set_value(host->rst, 0);
403         err = mg_wait(host, ATA_BUSY, MG_TMAX_RST_TO_BUSY);
404         if (err)
405                 return err;
406
407         /* hdd rst high */
408         gpio_set_value(host->rst, 1);
409         err = mg_wait(host, MG_STAT_READY, MG_TMAX_HDRST_TO_RDY);
410         if (err)
411                 return err;
412
413         /* soft reset on */
414         outb(ATA_SRST | (prv_data->use_polling ? ATA_NIEN : 0),
415                         (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
416         err = mg_wait(host, ATA_BUSY, MG_TMAX_RST_TO_BUSY);
417         if (err)
418                 return err;
419
420         /* soft reset off */
421         outb(prv_data->use_polling ? ATA_NIEN : 0,
422                         (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
423         err = mg_wait(host, MG_STAT_READY, MG_TMAX_SWRST_TO_RDY);
424         if (err)
425                 return err;
426
427         init_status = inb((unsigned long)host->dev_base + MG_REG_STATUS) & 0xf;
428
429         if (init_status == 0xf)
430                 return MG_ERR_INIT_STAT;
431
432         return err;
433 }
434
435 static void mg_bad_rw_intr(struct mg_host *host)
436 {
437         if (host->req)
438                 if (++host->req->errors >= MG_MAX_ERRORS ||
439                     host->error == MG_ERR_TIMEOUT)
440                         mg_end_request_cur(host, -EIO);
441 }
442
443 static unsigned int mg_out(struct mg_host *host,
444                 unsigned int sect_num,
445                 unsigned int sect_cnt,
446                 unsigned int cmd,
447                 void (*intr_addr)(struct mg_host *))
448 {
449         struct mg_drv_data *prv_data = host->dev->platform_data;
450
451         if (mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD))
452                 return host->error;
453
454         if (!prv_data->use_polling) {
455                 host->mg_do_intr = intr_addr;
456                 mod_timer(&host->timer, jiffies + 3 * HZ);
457         }
458         if (MG_RES_SEC)
459                 sect_num += MG_RES_SEC;
460         outb((u8)sect_cnt, (unsigned long)host->dev_base + MG_REG_SECT_CNT);
461         outb((u8)sect_num, (unsigned long)host->dev_base + MG_REG_SECT_NUM);
462         outb((u8)(sect_num >> 8), (unsigned long)host->dev_base +
463                         MG_REG_CYL_LOW);
464         outb((u8)(sect_num >> 16), (unsigned long)host->dev_base +
465                         MG_REG_CYL_HIGH);
466         outb((u8)((sect_num >> 24) | ATA_LBA | ATA_DEVICE_OBS),
467                         (unsigned long)host->dev_base + MG_REG_DRV_HEAD);
468         outb(cmd, (unsigned long)host->dev_base + MG_REG_COMMAND);
469         return MG_ERR_NONE;
470 }
471
472 static void mg_read(struct request *req)
473 {
474         u32 j;
475         struct mg_host *host = req->rq_disk->private_data;
476
477         if (mg_out(host, blk_rq_pos(req), blk_rq_sectors(req),
478                    MG_CMD_RD, NULL) != MG_ERR_NONE)
479                 mg_bad_rw_intr(host);
480
481         MG_DBG("requested %d sects (from %ld), buffer=0x%p\n",
482                blk_rq_sectors(req), blk_rq_pos(req), req->buffer);
483
484         do {
485                 u16 *buff = (u16 *)req->buffer;
486
487                 if (mg_wait(host, ATA_DRQ,
488                             MG_TMAX_WAIT_RD_DRQ) != MG_ERR_NONE) {
489                         mg_bad_rw_intr(host);
490                         return;
491                 }
492                 for (j = 0; j < MG_SECTOR_SIZE >> 1; j++)
493                         *buff++ = inw((unsigned long)host->dev_base +
494                                       MG_BUFF_OFFSET + (j << 1));
495
496                 outb(MG_CMD_RD_CONF, (unsigned long)host->dev_base +
497                                 MG_REG_COMMAND);
498         } while (mg_end_request(host, 0, MG_SECTOR_SIZE));
499 }
500
501 static void mg_write(struct request *req)
502 {
503         u32 j;
504         struct mg_host *host = req->rq_disk->private_data;
505
506         if (mg_out(host, blk_rq_pos(req), blk_rq_sectors(req),
507                    MG_CMD_WR, NULL) != MG_ERR_NONE) {
508                 mg_bad_rw_intr(host);
509                 return;
510         }
511
512         MG_DBG("requested %d sects (from %ld), buffer=0x%p\n",
513                blk_rq_sectors(req), blk_rq_pos(req), req->buffer);
514
515         do {
516                 u16 *buff = (u16 *)req->buffer;
517
518         if (mg_wait(host, ATA_DRQ, MG_TMAX_WAIT_WR_DRQ) != MG_ERR_NONE) {
519                         mg_bad_rw_intr(host);
520                         return;
521                 }
522                 for (j = 0; j < MG_SECTOR_SIZE >> 1; j++)
523                         outw(*buff++, (unsigned long)host->dev_base +
524                                       MG_BUFF_OFFSET + (j << 1));
525
526                 outb(MG_CMD_WR_CONF, (unsigned long)host->dev_base +
527                                 MG_REG_COMMAND);
528         } while (mg_end_request(host, 0, MG_SECTOR_SIZE));
529 }
530
531 static void mg_read_intr(struct mg_host *host)
532 {
533         struct request *req = host->req;
534         u32 i;
535         u16 *buff;
536
537         /* check status */
538         do {
539                 i = inb((unsigned long)host->dev_base + MG_REG_STATUS);
540                 if (i & ATA_BUSY)
541                         break;
542                 if (!MG_READY_OK(i))
543                         break;
544                 if (i & ATA_DRQ)
545                         goto ok_to_read;
546         } while (0);
547         mg_dump_status("mg_read_intr", i, host);
548         mg_bad_rw_intr(host);
549         mg_request(host->breq);
550         return;
551
552 ok_to_read:
553         /* get current segment of request */
554         buff = (u16 *)req->buffer;
555
556         /* read 1 sector */
557         for (i = 0; i < MG_SECTOR_SIZE >> 1; i++)
558                 *buff++ = inw((unsigned long)host->dev_base + MG_BUFF_OFFSET +
559                               (i << 1));
560
561         MG_DBG("sector %ld, remaining=%ld, buffer=0x%p\n",
562                blk_rq_pos(req), blk_rq_sectors(req) - 1, req->buffer);
563
564         /* send read confirm */
565         outb(MG_CMD_RD_CONF, (unsigned long)host->dev_base + MG_REG_COMMAND);
566
567         if (mg_end_request(host, 0, MG_SECTOR_SIZE)) {
568                 /* set handler if read remains */
569                 host->mg_do_intr = mg_read_intr;
570                 mod_timer(&host->timer, jiffies + 3 * HZ);
571         } else /* goto next request */
572                 mg_request(host->breq);
573 }
574
575 static void mg_write_intr(struct mg_host *host)
576 {
577         struct request *req = host->req;
578         u32 i, j;
579         u16 *buff;
580         bool rem;
581
582         /* check status */
583         do {
584                 i = inb((unsigned long)host->dev_base + MG_REG_STATUS);
585                 if (i & ATA_BUSY)
586                         break;
587                 if (!MG_READY_OK(i))
588                         break;
589                 if ((blk_rq_sectors(req) <= 1) || (i & ATA_DRQ))
590                         goto ok_to_write;
591         } while (0);
592         mg_dump_status("mg_write_intr", i, host);
593         mg_bad_rw_intr(host);
594         mg_request(host->breq);
595         return;
596
597 ok_to_write:
598         if ((rem = mg_end_request(host, 0, MG_SECTOR_SIZE))) {
599                 /* write 1 sector and set handler if remains */
600                 buff = (u16 *)req->buffer;
601                 for (j = 0; j < MG_STORAGE_BUFFER_SIZE >> 1; j++) {
602                         outw(*buff, (unsigned long)host->dev_base +
603                                         MG_BUFF_OFFSET + (j << 1));
604                         buff++;
605                 }
606                 MG_DBG("sector %ld, remaining=%ld, buffer=0x%p\n",
607                        blk_rq_pos(req), blk_rq_sectors(req), req->buffer);
608                 host->mg_do_intr = mg_write_intr;
609                 mod_timer(&host->timer, jiffies + 3 * HZ);
610         }
611
612         /* send write confirm */
613         outb(MG_CMD_WR_CONF, (unsigned long)host->dev_base + MG_REG_COMMAND);
614
615         if (!rem)
616                 mg_request(host->breq);
617 }
618
619 void mg_times_out(unsigned long data)
620 {
621         struct mg_host *host = (struct mg_host *)data;
622         char *name;
623
624         spin_lock_irq(&host->lock);
625
626         if (!host->req)
627                 goto out_unlock;
628
629         host->mg_do_intr = NULL;
630
631         name = host->req->rq_disk->disk_name;
632         printk(KERN_DEBUG "%s: timeout\n", name);
633
634         host->error = MG_ERR_TIMEOUT;
635         mg_bad_rw_intr(host);
636
637 out_unlock:
638         mg_request(host->breq);
639         spin_unlock_irq(&host->lock);
640 }
641
642 static void mg_request_poll(struct request_queue *q)
643 {
644         struct mg_host *host = q->queuedata;
645
646         while (1) {
647                 if (!host->req) {
648                         host->req = blk_fetch_request(q);
649                         if (!host->req)
650                                 break;
651                 }
652
653                 if (unlikely(!blk_fs_request(host->req))) {
654                         mg_end_request_cur(host, -EIO);
655                         continue;
656                 }
657
658                 if (rq_data_dir(host->req) == READ)
659                         mg_read(host->req);
660                 else
661                         mg_write(host->req);
662         }
663 }
664
665 static unsigned int mg_issue_req(struct request *req,
666                 struct mg_host *host,
667                 unsigned int sect_num,
668                 unsigned int sect_cnt)
669 {
670         u16 *buff;
671         u32 i;
672
673         switch (rq_data_dir(req)) {
674         case READ:
675                 if (mg_out(host, sect_num, sect_cnt, MG_CMD_RD, &mg_read_intr)
676                                 != MG_ERR_NONE) {
677                         mg_bad_rw_intr(host);
678                         return host->error;
679                 }
680                 break;
681         case WRITE:
682                 /* TODO : handler */
683                 outb(ATA_NIEN, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
684                 if (mg_out(host, sect_num, sect_cnt, MG_CMD_WR, &mg_write_intr)
685                                 != MG_ERR_NONE) {
686                         mg_bad_rw_intr(host);
687                         return host->error;
688                 }
689                 del_timer(&host->timer);
690                 mg_wait(host, ATA_DRQ, MG_TMAX_WAIT_WR_DRQ);
691                 outb(0, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
692                 if (host->error) {
693                         mg_bad_rw_intr(host);
694                         return host->error;
695                 }
696                 buff = (u16 *)req->buffer;
697                 for (i = 0; i < MG_SECTOR_SIZE >> 1; i++) {
698                         outw(*buff, (unsigned long)host->dev_base +
699                                         MG_BUFF_OFFSET + (i << 1));
700                         buff++;
701                 }
702                 mod_timer(&host->timer, jiffies + 3 * HZ);
703                 outb(MG_CMD_WR_CONF, (unsigned long)host->dev_base +
704                                 MG_REG_COMMAND);
705                 break;
706         }
707         return MG_ERR_NONE;
708 }
709
710 /* This function also called from IRQ context */
711 static void mg_request(struct request_queue *q)
712 {
713         struct mg_host *host = q->queuedata;
714         struct request *req;
715         u32 sect_num, sect_cnt;
716
717         while (1) {
718                 if (!host->req) {
719                         host->req = blk_fetch_request(q);
720                         if (!host->req)
721                                 break;
722                 }
723                 req = host->req;
724
725                 /* check unwanted request call */
726                 if (host->mg_do_intr)
727                         return;
728
729                 del_timer(&host->timer);
730
731                 sect_num = blk_rq_pos(req);
732                 /* deal whole segments */
733                 sect_cnt = blk_rq_sectors(req);
734
735                 /* sanity check */
736                 if (sect_num >= get_capacity(req->rq_disk) ||
737                                 ((sect_num + sect_cnt) >
738                                  get_capacity(req->rq_disk))) {
739                         printk(KERN_WARNING
740                                         "%s: bad access: sector=%d, count=%d\n",
741                                         req->rq_disk->disk_name,
742                                         sect_num, sect_cnt);
743                         mg_end_request_cur(host, -EIO);
744                         continue;
745                 }
746
747                 if (unlikely(!blk_fs_request(req))) {
748                         mg_end_request_cur(host, -EIO);
749                         continue;
750                 }
751
752                 if (!mg_issue_req(req, host, sect_num, sect_cnt))
753                         return;
754         }
755 }
756
757 static int mg_getgeo(struct block_device *bdev, struct hd_geometry *geo)
758 {
759         struct mg_host *host = bdev->bd_disk->private_data;
760
761         geo->cylinders = (unsigned short)host->cyls;
762         geo->heads = (unsigned char)host->heads;
763         geo->sectors = (unsigned char)host->sectors;
764         return 0;
765 }
766
767 static struct block_device_operations mg_disk_ops = {
768         .getgeo = mg_getgeo
769 };
770
771 static int mg_suspend(struct platform_device *plat_dev, pm_message_t state)
772 {
773         struct mg_drv_data *prv_data = plat_dev->dev.platform_data;
774         struct mg_host *host = prv_data->host;
775
776         if (mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD))
777                 return -EIO;
778
779         if (!prv_data->use_polling)
780                 outb(ATA_NIEN, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
781
782         outb(MG_CMD_SLEEP, (unsigned long)host->dev_base + MG_REG_COMMAND);
783         /* wait until mflash deep sleep */
784         msleep(1);
785
786         if (mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD)) {
787                 if (!prv_data->use_polling)
788                         outb(0, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
789                 return -EIO;
790         }
791
792         return 0;
793 }
794
795 static int mg_resume(struct platform_device *plat_dev)
796 {
797         struct mg_drv_data *prv_data = plat_dev->dev.platform_data;
798         struct mg_host *host = prv_data->host;
799
800         if (mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD))
801                 return -EIO;
802
803         outb(MG_CMD_WAKEUP, (unsigned long)host->dev_base + MG_REG_COMMAND);
804         /* wait until mflash wakeup */
805         msleep(1);
806
807         if (mg_wait(host, MG_STAT_READY, MG_TMAX_CONF_TO_CMD))
808                 return -EIO;
809
810         if (!prv_data->use_polling)
811                 outb(0, (unsigned long)host->dev_base + MG_REG_DRV_CTRL);
812
813         return 0;
814 }
815
816 static int mg_probe(struct platform_device *plat_dev)
817 {
818         struct mg_host *host;
819         struct resource *rsc;
820         struct mg_drv_data *prv_data = plat_dev->dev.platform_data;
821         int err = 0;
822
823         if (!prv_data) {
824                 printk(KERN_ERR "%s:%d fail (no driver_data)\n",
825                                 __func__, __LINE__);
826                 err = -EINVAL;
827                 goto probe_err;
828         }
829
830         /* alloc mg_host */
831         host = kzalloc(sizeof(struct mg_host), GFP_KERNEL);
832         if (!host) {
833                 printk(KERN_ERR "%s:%d fail (no memory for mg_host)\n",
834                                 __func__, __LINE__);
835                 err = -ENOMEM;
836                 goto probe_err;
837         }
838         host->major = MG_DISK_MAJ;
839
840         /* link each other */
841         prv_data->host = host;
842         host->dev = &plat_dev->dev;
843
844         /* io remap */
845         rsc = platform_get_resource(plat_dev, IORESOURCE_MEM, 0);
846         if (!rsc) {
847                 printk(KERN_ERR "%s:%d platform_get_resource fail\n",
848                                 __func__, __LINE__);
849                 err = -EINVAL;
850                 goto probe_err_2;
851         }
852         host->dev_base = ioremap(rsc->start , rsc->end + 1);
853         if (!host->dev_base) {
854                 printk(KERN_ERR "%s:%d ioremap fail\n",
855                                 __func__, __LINE__);
856                 err = -EIO;
857                 goto probe_err_2;
858         }
859         MG_DBG("dev_base = 0x%x\n", (u32)host->dev_base);
860
861         /* get reset pin */
862         rsc = platform_get_resource_byname(plat_dev, IORESOURCE_IO,
863                         MG_RST_PIN);
864         if (!rsc) {
865                 printk(KERN_ERR "%s:%d get reset pin fail\n",
866                                 __func__, __LINE__);
867                 err = -EIO;
868                 goto probe_err_3;
869         }
870         host->rst = rsc->start;
871
872         /* init rst pin */
873         err = gpio_request(host->rst, MG_RST_PIN);
874         if (err)
875                 goto probe_err_3;
876         gpio_direction_output(host->rst, 1);
877
878         /* reset out pin */
879         if (!(prv_data->dev_attr & MG_DEV_MASK))
880                 goto probe_err_3a;
881
882         if (prv_data->dev_attr != MG_BOOT_DEV) {
883                 rsc = platform_get_resource_byname(plat_dev, IORESOURCE_IO,
884                                 MG_RSTOUT_PIN);
885                 if (!rsc) {
886                         printk(KERN_ERR "%s:%d get reset-out pin fail\n",
887                                         __func__, __LINE__);
888                         err = -EIO;
889                         goto probe_err_3a;
890                 }
891                 host->rstout = rsc->start;
892                 err = gpio_request(host->rstout, MG_RSTOUT_PIN);
893                 if (err)
894                         goto probe_err_3a;
895                 gpio_direction_input(host->rstout);
896         }
897
898         /* disk reset */
899         if (prv_data->dev_attr == MG_STORAGE_DEV) {
900                 /* If POR seq. not yet finised, wait */
901                 err = mg_wait_rstout(host->rstout, MG_TMAX_RSTOUT);
902                 if (err)
903                         goto probe_err_3b;
904                 err = mg_disk_init(host);
905                 if (err) {
906                         printk(KERN_ERR "%s:%d fail (err code : %d)\n",
907                                         __func__, __LINE__, err);
908                         err = -EIO;
909                         goto probe_err_3b;
910                 }
911         }
912
913         /* get irq resource */
914         if (!prv_data->use_polling) {
915                 host->irq = platform_get_irq(plat_dev, 0);
916                 if (host->irq == -ENXIO) {
917                         err = host->irq;
918                         goto probe_err_3b;
919                 }
920                 err = request_irq(host->irq, mg_irq,
921                                 IRQF_DISABLED | IRQF_TRIGGER_RISING,
922                                 MG_DEV_NAME, host);
923                 if (err) {
924                         printk(KERN_ERR "%s:%d fail (request_irq err=%d)\n",
925                                         __func__, __LINE__, err);
926                         goto probe_err_3b;
927                 }
928
929         }
930
931         /* get disk id */
932         err = mg_get_disk_id(host);
933         if (err) {
934                 printk(KERN_ERR "%s:%d fail (err code : %d)\n",
935                                 __func__, __LINE__, err);
936                 err = -EIO;
937                 goto probe_err_4;
938         }
939
940         err = register_blkdev(host->major, MG_DISK_NAME);
941         if (err < 0) {
942                 printk(KERN_ERR "%s:%d register_blkdev fail (err code : %d)\n",
943                                 __func__, __LINE__, err);
944                 goto probe_err_4;
945         }
946         if (!host->major)
947                 host->major = err;
948
949         spin_lock_init(&host->lock);
950
951         if (prv_data->use_polling)
952                 host->breq = blk_init_queue(mg_request_poll, &host->lock);
953         else
954                 host->breq = blk_init_queue(mg_request, &host->lock);
955
956         if (!host->breq) {
957                 err = -ENOMEM;
958                 printk(KERN_ERR "%s:%d (blk_init_queue) fail\n",
959                                 __func__, __LINE__);
960                 goto probe_err_5;
961         }
962         host->breq->queuedata = host;
963
964         /* mflash is random device, thanx for the noop */
965         elevator_exit(host->breq->elevator);
966         err = elevator_init(host->breq, "noop");
967         if (err) {
968                 printk(KERN_ERR "%s:%d (elevator_init) fail\n",
969                                 __func__, __LINE__);
970                 goto probe_err_6;
971         }
972         blk_queue_max_sectors(host->breq, MG_MAX_SECTS);
973         blk_queue_logical_block_size(host->breq, MG_SECTOR_SIZE);
974
975         init_timer(&host->timer);
976         host->timer.function = mg_times_out;
977         host->timer.data = (unsigned long)host;
978
979         host->gd = alloc_disk(MG_DISK_MAX_PART);
980         if (!host->gd) {
981                 printk(KERN_ERR "%s:%d (alloc_disk) fail\n",
982                                 __func__, __LINE__);
983                 err = -ENOMEM;
984                 goto probe_err_7;
985         }
986         host->gd->major = host->major;
987         host->gd->first_minor = 0;
988         host->gd->fops = &mg_disk_ops;
989         host->gd->queue = host->breq;
990         host->gd->private_data = host;
991         sprintf(host->gd->disk_name, MG_DISK_NAME"a");
992
993         set_capacity(host->gd, host->n_sectors);
994
995         add_disk(host->gd);
996
997         return err;
998
999 probe_err_7:
1000         del_timer_sync(&host->timer);
1001 probe_err_6:
1002         blk_cleanup_queue(host->breq);
1003 probe_err_5:
1004         unregister_blkdev(MG_DISK_MAJ, MG_DISK_NAME);
1005 probe_err_4:
1006         if (!prv_data->use_polling)
1007                 free_irq(host->irq, host);
1008 probe_err_3b:
1009         gpio_free(host->rstout);
1010 probe_err_3a:
1011         gpio_free(host->rst);
1012 probe_err_3:
1013         iounmap(host->dev_base);
1014 probe_err_2:
1015         kfree(host);
1016 probe_err:
1017         return err;
1018 }
1019
1020 static int mg_remove(struct platform_device *plat_dev)
1021 {
1022         struct mg_drv_data *prv_data = plat_dev->dev.platform_data;
1023         struct mg_host *host = prv_data->host;
1024         int err = 0;
1025
1026         /* delete timer */
1027         del_timer_sync(&host->timer);
1028
1029         /* remove disk */
1030         if (host->gd) {
1031                 del_gendisk(host->gd);
1032                 put_disk(host->gd);
1033         }
1034         /* remove queue */
1035         if (host->breq)
1036                 blk_cleanup_queue(host->breq);
1037
1038         /* unregister blk device */
1039         unregister_blkdev(host->major, MG_DISK_NAME);
1040
1041         /* free irq */
1042         if (!prv_data->use_polling)
1043                 free_irq(host->irq, host);
1044
1045         /* free reset-out pin */
1046         if (prv_data->dev_attr != MG_BOOT_DEV)
1047                 gpio_free(host->rstout);
1048
1049         /* free rst pin */
1050         if (host->rst)
1051                 gpio_free(host->rst);
1052
1053         /* unmap io */
1054         if (host->dev_base)
1055                 iounmap(host->dev_base);
1056
1057         /* free mg_host */
1058         kfree(host);
1059
1060         return err;
1061 }
1062
1063 static struct platform_driver mg_disk_driver = {
1064         .probe = mg_probe,
1065         .remove = mg_remove,
1066         .suspend = mg_suspend,
1067         .resume = mg_resume,
1068         .driver = {
1069                 .name = MG_DEV_NAME,
1070                 .owner = THIS_MODULE,
1071         }
1072 };
1073
1074 /****************************************************************************
1075  *
1076  * Module stuff
1077  *
1078  ****************************************************************************/
1079
1080 static int __init mg_init(void)
1081 {
1082         printk(KERN_INFO "mGine mflash driver, (c) 2008 mGine Co.\n");
1083         return platform_driver_register(&mg_disk_driver);
1084 }
1085
1086 static void __exit mg_exit(void)
1087 {
1088         printk(KERN_INFO "mflash driver : bye bye\n");
1089         platform_driver_unregister(&mg_disk_driver);
1090 }
1091
1092 module_init(mg_init);
1093 module_exit(mg_exit);
1094
1095 MODULE_LICENSE("GPL");
1096 MODULE_AUTHOR("unsik Kim <donari75@gmail.com>");
1097 MODULE_DESCRIPTION("mGine m[g]flash device driver");