ide: use rq->nr_sectors in task_end_request()
[safe/jmp/linux-2.6] / drivers / ide / ide-taskfile.c
1 /*
2  * linux/drivers/ide/ide-taskfile.c     Version 0.38    March 05, 2003
3  *
4  *  Copyright (C) 2000-2002     Michael Cornwell <cornwell@acm.org>
5  *  Copyright (C) 2000-2002     Andre Hedrick <andre@linux-ide.org>
6  *  Copyright (C) 2001-2002     Klaus Smolin
7  *                                      IBM Storage Technology Division
8  *  Copyright (C) 2003-2004     Bartlomiej Zolnierkiewicz
9  *
10  *  The big the bad and the ugly.
11  */
12
13 #include <linux/module.h>
14 #include <linux/types.h>
15 #include <linux/string.h>
16 #include <linux/kernel.h>
17 #include <linux/timer.h>
18 #include <linux/mm.h>
19 #include <linux/sched.h>
20 #include <linux/interrupt.h>
21 #include <linux/major.h>
22 #include <linux/errno.h>
23 #include <linux/genhd.h>
24 #include <linux/blkpg.h>
25 #include <linux/slab.h>
26 #include <linux/pci.h>
27 #include <linux/delay.h>
28 #include <linux/hdreg.h>
29 #include <linux/ide.h>
30 #include <linux/bitops.h>
31 #include <linux/scatterlist.h>
32
33 #include <asm/byteorder.h>
34 #include <asm/irq.h>
35 #include <asm/uaccess.h>
36 #include <asm/io.h>
37
38 void ide_tf_load(ide_drive_t *drive, ide_task_t *task)
39 {
40         ide_hwif_t *hwif = drive->hwif;
41         struct ide_taskfile *tf = &task->tf;
42         u8 HIHI = (task->tf_flags & IDE_TFLAG_LBA48) ? 0xE0 : 0xEF;
43
44         if (task->tf_flags & IDE_TFLAG_FLAGGED)
45                 HIHI = 0xFF;
46
47 #ifdef DEBUG
48         printk("%s: tf: feat 0x%02x nsect 0x%02x lbal 0x%02x "
49                 "lbam 0x%02x lbah 0x%02x dev 0x%02x cmd 0x%02x\n",
50                 drive->name, tf->feature, tf->nsect, tf->lbal,
51                 tf->lbam, tf->lbah, tf->device, tf->command);
52         printk("%s: hob: nsect 0x%02x lbal 0x%02x "
53                 "lbam 0x%02x lbah 0x%02x\n",
54                 drive->name, tf->hob_nsect, tf->hob_lbal,
55                 tf->hob_lbam, tf->hob_lbah);
56 #endif
57
58         ide_set_irq(drive, 1);
59
60         if ((task->tf_flags & IDE_TFLAG_NO_SELECT_MASK) == 0)
61                 SELECT_MASK(drive, 0);
62
63         if (task->tf_flags & IDE_TFLAG_OUT_DATA)
64                 hwif->OUTW((tf->hob_data << 8) | tf->data, IDE_DATA_REG);
65
66         if (task->tf_flags & IDE_TFLAG_OUT_HOB_FEATURE)
67                 hwif->OUTB(tf->hob_feature, IDE_FEATURE_REG);
68         if (task->tf_flags & IDE_TFLAG_OUT_HOB_NSECT)
69                 hwif->OUTB(tf->hob_nsect, IDE_NSECTOR_REG);
70         if (task->tf_flags & IDE_TFLAG_OUT_HOB_LBAL)
71                 hwif->OUTB(tf->hob_lbal, IDE_SECTOR_REG);
72         if (task->tf_flags & IDE_TFLAG_OUT_HOB_LBAM)
73                 hwif->OUTB(tf->hob_lbam, IDE_LCYL_REG);
74         if (task->tf_flags & IDE_TFLAG_OUT_HOB_LBAH)
75                 hwif->OUTB(tf->hob_lbah, IDE_HCYL_REG);
76
77         if (task->tf_flags & IDE_TFLAG_OUT_FEATURE)
78                 hwif->OUTB(tf->feature, IDE_FEATURE_REG);
79         if (task->tf_flags & IDE_TFLAG_OUT_NSECT)
80                 hwif->OUTB(tf->nsect, IDE_NSECTOR_REG);
81         if (task->tf_flags & IDE_TFLAG_OUT_LBAL)
82                 hwif->OUTB(tf->lbal, IDE_SECTOR_REG);
83         if (task->tf_flags & IDE_TFLAG_OUT_LBAM)
84                 hwif->OUTB(tf->lbam, IDE_LCYL_REG);
85         if (task->tf_flags & IDE_TFLAG_OUT_LBAH)
86                 hwif->OUTB(tf->lbah, IDE_HCYL_REG);
87
88         if (task->tf_flags & IDE_TFLAG_OUT_DEVICE)
89                 hwif->OUTB((tf->device & HIHI) | drive->select.all, IDE_SELECT_REG);
90 }
91
92 int taskfile_lib_get_identify (ide_drive_t *drive, u8 *buf)
93 {
94         ide_task_t args;
95
96         memset(&args, 0, sizeof(ide_task_t));
97         args.tf.nsect = 0x01;
98         if (drive->media == ide_disk)
99                 args.tf.command = WIN_IDENTIFY;
100         else
101                 args.tf.command = WIN_PIDENTIFY;
102         args.tf_flags   = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
103         args.data_phase = TASKFILE_IN;
104         return ide_raw_taskfile(drive, &args, buf, 1);
105 }
106
107 static int inline task_dma_ok(ide_task_t *task)
108 {
109         if (blk_fs_request(task->rq) || (task->tf_flags & IDE_TFLAG_FLAGGED))
110                 return 1;
111
112         switch (task->tf.command) {
113                 case WIN_WRITEDMA_ONCE:
114                 case WIN_WRITEDMA:
115                 case WIN_WRITEDMA_EXT:
116                 case WIN_READDMA_ONCE:
117                 case WIN_READDMA:
118                 case WIN_READDMA_EXT:
119                 case WIN_IDENTIFY_DMA:
120                         return 1;
121         }
122
123         return 0;
124 }
125
126 static ide_startstop_t task_no_data_intr(ide_drive_t *);
127 static ide_startstop_t set_geometry_intr(ide_drive_t *);
128 static ide_startstop_t recal_intr(ide_drive_t *);
129 static ide_startstop_t set_multmode_intr(ide_drive_t *);
130 static ide_startstop_t pre_task_out_intr(ide_drive_t *, struct request *);
131 static ide_startstop_t task_in_intr(ide_drive_t *);
132
133 ide_startstop_t do_rw_taskfile (ide_drive_t *drive, ide_task_t *task)
134 {
135         ide_hwif_t *hwif        = HWIF(drive);
136         struct ide_taskfile *tf = &task->tf;
137         ide_handler_t *handler = NULL;
138
139         if (task->data_phase == TASKFILE_MULTI_IN ||
140             task->data_phase == TASKFILE_MULTI_OUT) {
141                 if (!drive->mult_count) {
142                         printk(KERN_ERR "%s: multimode not set!\n",
143                                         drive->name);
144                         return ide_stopped;
145                 }
146         }
147
148         if (task->tf_flags & IDE_TFLAG_FLAGGED)
149                 task->tf_flags |= IDE_TFLAG_FLAGGED_SET_IN_FLAGS;
150
151         if ((task->tf_flags & IDE_TFLAG_DMA_PIO_FALLBACK) == 0)
152                 ide_tf_load(drive, task);
153
154         switch (task->data_phase) {
155         case TASKFILE_MULTI_OUT:
156         case TASKFILE_OUT:
157                 hwif->OUTBSYNC(drive, tf->command, IDE_COMMAND_REG);
158                 ndelay(400);    /* FIXME */
159                 return pre_task_out_intr(drive, task->rq);
160         case TASKFILE_MULTI_IN:
161         case TASKFILE_IN:
162                 handler = task_in_intr;
163                 /* fall-through */
164         case TASKFILE_NO_DATA:
165                 if (handler == NULL)
166                         handler = task_no_data_intr;
167                 /* WIN_{SPECIFY,RESTORE,SETMULT} use custom handlers */
168                 if (task->tf_flags & IDE_TFLAG_CUSTOM_HANDLER) {
169                         switch (tf->command) {
170                         case WIN_SPECIFY: handler = set_geometry_intr;  break;
171                         case WIN_RESTORE: handler = recal_intr;         break;
172                         case WIN_SETMULT: handler = set_multmode_intr;  break;
173                         }
174                 }
175                 ide_execute_command(drive, tf->command, handler,
176                                     WAIT_WORSTCASE, NULL);
177                 return ide_started;
178         default:
179                 if (task_dma_ok(task) == 0 || drive->using_dma == 0 ||
180                     hwif->dma_setup(drive))
181                         return ide_stopped;
182                 hwif->dma_exec_cmd(drive, tf->command);
183                 hwif->dma_start(drive);
184                 return ide_started;
185         }
186 }
187 EXPORT_SYMBOL_GPL(do_rw_taskfile);
188
189 /*
190  * set_multmode_intr() is invoked on completion of a WIN_SETMULT cmd.
191  */
192 static ide_startstop_t set_multmode_intr(ide_drive_t *drive)
193 {
194         ide_hwif_t *hwif = HWIF(drive);
195         u8 stat;
196
197         if (OK_STAT(stat = hwif->INB(IDE_STATUS_REG),READY_STAT,BAD_STAT)) {
198                 drive->mult_count = drive->mult_req;
199         } else {
200                 drive->mult_req = drive->mult_count = 0;
201                 drive->special.b.recalibrate = 1;
202                 (void) ide_dump_status(drive, "set_multmode", stat);
203         }
204         return ide_stopped;
205 }
206
207 /*
208  * set_geometry_intr() is invoked on completion of a WIN_SPECIFY cmd.
209  */
210 static ide_startstop_t set_geometry_intr(ide_drive_t *drive)
211 {
212         ide_hwif_t *hwif = HWIF(drive);
213         int retries = 5;
214         u8 stat;
215
216         while (((stat = hwif->INB(IDE_STATUS_REG)) & BUSY_STAT) && retries--)
217                 udelay(10);
218
219         if (OK_STAT(stat, READY_STAT, BAD_STAT))
220                 return ide_stopped;
221
222         if (stat & (ERR_STAT|DRQ_STAT))
223                 return ide_error(drive, "set_geometry_intr", stat);
224
225         BUG_ON(HWGROUP(drive)->handler != NULL);
226         ide_set_handler(drive, &set_geometry_intr, WAIT_WORSTCASE, NULL);
227         return ide_started;
228 }
229
230 /*
231  * recal_intr() is invoked on completion of a WIN_RESTORE (recalibrate) cmd.
232  */
233 static ide_startstop_t recal_intr(ide_drive_t *drive)
234 {
235         ide_hwif_t *hwif = HWIF(drive);
236         u8 stat;
237
238         if (!OK_STAT(stat = hwif->INB(IDE_STATUS_REG), READY_STAT, BAD_STAT))
239                 return ide_error(drive, "recal_intr", stat);
240         return ide_stopped;
241 }
242
243 /*
244  * Handler for commands without a data phase
245  */
246 static ide_startstop_t task_no_data_intr(ide_drive_t *drive)
247 {
248         ide_task_t *args        = HWGROUP(drive)->rq->special;
249         ide_hwif_t *hwif        = HWIF(drive);
250         u8 stat;
251
252         local_irq_enable_in_hardirq();
253         if (!OK_STAT(stat = hwif->INB(IDE_STATUS_REG),READY_STAT,BAD_STAT)) {
254                 return ide_error(drive, "task_no_data_intr", stat);
255                 /* calls ide_end_drive_cmd */
256         }
257         if (args)
258                 ide_end_drive_cmd(drive, stat, hwif->INB(IDE_ERROR_REG));
259
260         return ide_stopped;
261 }
262
263 static u8 wait_drive_not_busy(ide_drive_t *drive)
264 {
265         ide_hwif_t *hwif = HWIF(drive);
266         int retries;
267         u8 stat;
268
269         /*
270          * Last sector was transfered, wait until drive is ready.
271          * This can take up to 10 usec, but we will wait max 1 ms
272          * (drive_cmd_intr() waits that long).
273          */
274         for (retries = 0; retries < 100; retries++) {
275                 if ((stat = hwif->INB(IDE_STATUS_REG)) & BUSY_STAT)
276                         udelay(10);
277                 else
278                         break;
279         }
280
281         if (stat & BUSY_STAT)
282                 printk(KERN_ERR "%s: drive still BUSY!\n", drive->name);
283
284         return stat;
285 }
286
287 static void ide_pio_sector(ide_drive_t *drive, unsigned int write)
288 {
289         ide_hwif_t *hwif = drive->hwif;
290         struct scatterlist *sg = hwif->sg_table;
291         struct scatterlist *cursg = hwif->cursg;
292         struct page *page;
293 #ifdef CONFIG_HIGHMEM
294         unsigned long flags;
295 #endif
296         unsigned int offset;
297         u8 *buf;
298
299         cursg = hwif->cursg;
300         if (!cursg) {
301                 cursg = sg;
302                 hwif->cursg = sg;
303         }
304
305         page = sg_page(cursg);
306         offset = cursg->offset + hwif->cursg_ofs * SECTOR_SIZE;
307
308         /* get the current page and offset */
309         page = nth_page(page, (offset >> PAGE_SHIFT));
310         offset %= PAGE_SIZE;
311
312 #ifdef CONFIG_HIGHMEM
313         local_irq_save(flags);
314 #endif
315         buf = kmap_atomic(page, KM_BIO_SRC_IRQ) + offset;
316
317         hwif->nleft--;
318         hwif->cursg_ofs++;
319
320         if ((hwif->cursg_ofs * SECTOR_SIZE) == cursg->length) {
321                 hwif->cursg = sg_next(hwif->cursg);
322                 hwif->cursg_ofs = 0;
323         }
324
325         /* do the actual data transfer */
326         if (write)
327                 hwif->ata_output_data(drive, buf, SECTOR_WORDS);
328         else
329                 hwif->ata_input_data(drive, buf, SECTOR_WORDS);
330
331         kunmap_atomic(buf, KM_BIO_SRC_IRQ);
332 #ifdef CONFIG_HIGHMEM
333         local_irq_restore(flags);
334 #endif
335 }
336
337 static void ide_pio_multi(ide_drive_t *drive, unsigned int write)
338 {
339         unsigned int nsect;
340
341         nsect = min_t(unsigned int, drive->hwif->nleft, drive->mult_count);
342         while (nsect--)
343                 ide_pio_sector(drive, write);
344 }
345
346 static void ide_pio_datablock(ide_drive_t *drive, struct request *rq,
347                                      unsigned int write)
348 {
349         u8 saved_io_32bit = drive->io_32bit;
350
351         if (rq->bio)    /* fs request */
352                 rq->errors = 0;
353
354         if (rq->cmd_type == REQ_TYPE_ATA_TASKFILE) {
355                 ide_task_t *task = rq->special;
356
357                 if (task->tf_flags & IDE_TFLAG_IO_16BIT)
358                         drive->io_32bit = 0;
359         }
360
361         touch_softlockup_watchdog();
362
363         switch (drive->hwif->data_phase) {
364         case TASKFILE_MULTI_IN:
365         case TASKFILE_MULTI_OUT:
366                 ide_pio_multi(drive, write);
367                 break;
368         default:
369                 ide_pio_sector(drive, write);
370                 break;
371         }
372
373         drive->io_32bit = saved_io_32bit;
374 }
375
376 static ide_startstop_t task_error(ide_drive_t *drive, struct request *rq,
377                                   const char *s, u8 stat)
378 {
379         if (rq->bio) {
380                 ide_hwif_t *hwif = drive->hwif;
381                 int sectors = hwif->nsect - hwif->nleft;
382
383                 switch (hwif->data_phase) {
384                 case TASKFILE_IN:
385                         if (hwif->nleft)
386                                 break;
387                         /* fall through */
388                 case TASKFILE_OUT:
389                         sectors--;
390                         break;
391                 case TASKFILE_MULTI_IN:
392                         if (hwif->nleft)
393                                 break;
394                         /* fall through */
395                 case TASKFILE_MULTI_OUT:
396                         sectors -= drive->mult_count;
397                 default:
398                         break;
399                 }
400
401                 if (sectors > 0) {
402                         ide_driver_t *drv;
403
404                         drv = *(ide_driver_t **)rq->rq_disk->private_data;
405                         drv->end_request(drive, 1, sectors);
406                 }
407         }
408         return ide_error(drive, s, stat);
409 }
410
411 static void task_end_request(ide_drive_t *drive, struct request *rq, u8 stat)
412 {
413         if (rq->cmd_type == REQ_TYPE_ATA_TASKFILE) {
414                 ide_task_t *task = rq->special;
415
416                 if (task->tf_flags & IDE_TFLAG_FLAGGED) {
417                         u8 err = drive->hwif->INB(IDE_ERROR_REG);
418                         ide_end_drive_cmd(drive, stat, err);
419                         return;
420                 }
421         }
422
423         if (rq->rq_disk) {
424                 ide_driver_t *drv;
425
426                 drv = *(ide_driver_t **)rq->rq_disk->private_data;;
427                 drv->end_request(drive, 1, rq->nr_sectors);
428         } else
429                 ide_end_request(drive, 1, rq->nr_sectors);
430 }
431
432 /*
433  * Handler for command with PIO data-in phase (Read/Read Multiple).
434  */
435 static ide_startstop_t task_in_intr(ide_drive_t *drive)
436 {
437         ide_hwif_t *hwif = drive->hwif;
438         struct request *rq = HWGROUP(drive)->rq;
439         u8 stat = hwif->INB(IDE_STATUS_REG);
440
441         /* new way for dealing with premature shared PCI interrupts */
442         if (!OK_STAT(stat, DATA_READY, BAD_R_STAT)) {
443                 if (stat & (ERR_STAT | DRQ_STAT))
444                         return task_error(drive, rq, __FUNCTION__, stat);
445                 /* No data yet, so wait for another IRQ. */
446                 ide_set_handler(drive, &task_in_intr, WAIT_WORSTCASE, NULL);
447                 return ide_started;
448         }
449
450         ide_pio_datablock(drive, rq, 0);
451
452         /* If it was the last datablock check status and finish transfer. */
453         if (!hwif->nleft) {
454                 stat = wait_drive_not_busy(drive);
455                 if (!OK_STAT(stat, 0, BAD_STAT))
456                         return task_error(drive, rq, __FUNCTION__, stat);
457                 task_end_request(drive, rq, stat);
458                 return ide_stopped;
459         }
460
461         /* Still data left to transfer. */
462         ide_set_handler(drive, &task_in_intr, WAIT_WORSTCASE, NULL);
463
464         return ide_started;
465 }
466
467 /*
468  * Handler for command with PIO data-out phase (Write/Write Multiple).
469  */
470 static ide_startstop_t task_out_intr (ide_drive_t *drive)
471 {
472         ide_hwif_t *hwif = drive->hwif;
473         struct request *rq = HWGROUP(drive)->rq;
474         u8 stat = hwif->INB(IDE_STATUS_REG);
475
476         if (!OK_STAT(stat, DRIVE_READY, drive->bad_wstat))
477                 return task_error(drive, rq, __FUNCTION__, stat);
478
479         /* Deal with unexpected ATA data phase. */
480         if (((stat & DRQ_STAT) == 0) ^ !hwif->nleft)
481                 return task_error(drive, rq, __FUNCTION__, stat);
482
483         if (!hwif->nleft) {
484                 task_end_request(drive, rq, stat);
485                 return ide_stopped;
486         }
487
488         /* Still data left to transfer. */
489         ide_pio_datablock(drive, rq, 1);
490         ide_set_handler(drive, &task_out_intr, WAIT_WORSTCASE, NULL);
491
492         return ide_started;
493 }
494
495 static ide_startstop_t pre_task_out_intr(ide_drive_t *drive, struct request *rq)
496 {
497         ide_startstop_t startstop;
498
499         if (ide_wait_stat(&startstop, drive, DATA_READY,
500                           drive->bad_wstat, WAIT_DRQ)) {
501                 printk(KERN_ERR "%s: no DRQ after issuing %sWRITE%s\n",
502                                 drive->name,
503                                 drive->hwif->data_phase ? "MULT" : "",
504                                 drive->addressing ? "_EXT" : "");
505                 return startstop;
506         }
507
508         if (!drive->unmask)
509                 local_irq_disable();
510
511         ide_set_handler(drive, &task_out_intr, WAIT_WORSTCASE, NULL);
512         ide_pio_datablock(drive, rq, 1);
513
514         return ide_started;
515 }
516
517 int ide_raw_taskfile(ide_drive_t *drive, ide_task_t *task, u8 *buf, u16 nsect)
518 {
519         struct request rq;
520
521         memset(&rq, 0, sizeof(rq));
522         rq.ref_count = 1;
523         rq.cmd_type = REQ_TYPE_ATA_TASKFILE;
524         rq.buffer = buf;
525
526         /*
527          * (ks) We transfer currently only whole sectors.
528          * This is suffient for now.  But, it would be great,
529          * if we would find a solution to transfer any size.
530          * To support special commands like READ LONG.
531          */
532         rq.hard_nr_sectors = rq.nr_sectors = nsect;
533         rq.hard_cur_sectors = rq.current_nr_sectors = nsect;
534
535         if (task->tf_flags & IDE_TFLAG_WRITE)
536                 rq.cmd_flags |= REQ_RW;
537
538         rq.special = task;
539         task->rq = &rq;
540
541         return ide_do_drive_cmd(drive, &rq, ide_wait);
542 }
543
544 EXPORT_SYMBOL(ide_raw_taskfile);
545
546 int ide_no_data_taskfile(ide_drive_t *drive, ide_task_t *task)
547 {
548         task->data_phase = TASKFILE_NO_DATA;
549
550         return ide_raw_taskfile(drive, task, NULL, 0);
551 }
552 EXPORT_SYMBOL_GPL(ide_no_data_taskfile);
553
554 #ifdef CONFIG_IDE_TASK_IOCTL
555 int ide_taskfile_ioctl (ide_drive_t *drive, unsigned int cmd, unsigned long arg)
556 {
557         ide_task_request_t      *req_task;
558         ide_task_t              args;
559         u8 *outbuf              = NULL;
560         u8 *inbuf               = NULL;
561         u8 *data_buf            = NULL;
562         int err                 = 0;
563         int tasksize            = sizeof(struct ide_task_request_s);
564         unsigned int taskin     = 0;
565         unsigned int taskout    = 0;
566         u16 nsect               = 0;
567         char __user *buf = (char __user *)arg;
568
569 //      printk("IDE Taskfile ...\n");
570
571         req_task = kzalloc(tasksize, GFP_KERNEL);
572         if (req_task == NULL) return -ENOMEM;
573         if (copy_from_user(req_task, buf, tasksize)) {
574                 kfree(req_task);
575                 return -EFAULT;
576         }
577
578         taskout = req_task->out_size;
579         taskin  = req_task->in_size;
580         
581         if (taskin > 65536 || taskout > 65536) {
582                 err = -EINVAL;
583                 goto abort;
584         }
585
586         if (taskout) {
587                 int outtotal = tasksize;
588                 outbuf = kzalloc(taskout, GFP_KERNEL);
589                 if (outbuf == NULL) {
590                         err = -ENOMEM;
591                         goto abort;
592                 }
593                 if (copy_from_user(outbuf, buf + outtotal, taskout)) {
594                         err = -EFAULT;
595                         goto abort;
596                 }
597         }
598
599         if (taskin) {
600                 int intotal = tasksize + taskout;
601                 inbuf = kzalloc(taskin, GFP_KERNEL);
602                 if (inbuf == NULL) {
603                         err = -ENOMEM;
604                         goto abort;
605                 }
606                 if (copy_from_user(inbuf, buf + intotal, taskin)) {
607                         err = -EFAULT;
608                         goto abort;
609                 }
610         }
611
612         memset(&args, 0, sizeof(ide_task_t));
613
614         memcpy(&args.tf_array[0], req_task->hob_ports, HDIO_DRIVE_HOB_HDR_SIZE - 2);
615         memcpy(&args.tf_array[6], req_task->io_ports, HDIO_DRIVE_TASK_HDR_SIZE);
616
617         args.data_phase = req_task->data_phase;
618
619         args.tf_flags = IDE_TFLAG_IO_16BIT | IDE_TFLAG_DEVICE |
620                         IDE_TFLAG_IN_TF;
621         if (drive->addressing == 1)
622                 args.tf_flags |= (IDE_TFLAG_LBA48 | IDE_TFLAG_IN_HOB);
623
624         if (req_task->out_flags.all) {
625                 args.tf_flags |= IDE_TFLAG_FLAGGED;
626
627                 if (req_task->out_flags.b.data)
628                         args.tf_flags |= IDE_TFLAG_OUT_DATA;
629
630                 if (req_task->out_flags.b.nsector_hob)
631                         args.tf_flags |= IDE_TFLAG_OUT_HOB_NSECT;
632                 if (req_task->out_flags.b.sector_hob)
633                         args.tf_flags |= IDE_TFLAG_OUT_HOB_LBAL;
634                 if (req_task->out_flags.b.lcyl_hob)
635                         args.tf_flags |= IDE_TFLAG_OUT_HOB_LBAM;
636                 if (req_task->out_flags.b.hcyl_hob)
637                         args.tf_flags |= IDE_TFLAG_OUT_HOB_LBAH;
638
639                 if (req_task->out_flags.b.error_feature)
640                         args.tf_flags |= IDE_TFLAG_OUT_FEATURE;
641                 if (req_task->out_flags.b.nsector)
642                         args.tf_flags |= IDE_TFLAG_OUT_NSECT;
643                 if (req_task->out_flags.b.sector)
644                         args.tf_flags |= IDE_TFLAG_OUT_LBAL;
645                 if (req_task->out_flags.b.lcyl)
646                         args.tf_flags |= IDE_TFLAG_OUT_LBAM;
647                 if (req_task->out_flags.b.hcyl)
648                         args.tf_flags |= IDE_TFLAG_OUT_LBAH;
649         } else {
650                 args.tf_flags |= IDE_TFLAG_OUT_TF;
651                 if (args.tf_flags & IDE_TFLAG_LBA48)
652                         args.tf_flags |= IDE_TFLAG_OUT_HOB;
653         }
654
655         if (req_task->in_flags.b.data)
656                 args.tf_flags |= IDE_TFLAG_IN_DATA;
657
658         switch(req_task->data_phase) {
659                 case TASKFILE_MULTI_OUT:
660                         if (!drive->mult_count) {
661                                 /* (hs): give up if multcount is not set */
662                                 printk(KERN_ERR "%s: %s Multimode Write " \
663                                         "multcount is not set\n",
664                                         drive->name, __FUNCTION__);
665                                 err = -EPERM;
666                                 goto abort;
667                         }
668                         /* fall through */
669                 case TASKFILE_OUT:
670                         /* fall through */
671                 case TASKFILE_OUT_DMAQ:
672                 case TASKFILE_OUT_DMA:
673                         nsect = taskout / SECTOR_SIZE;
674                         data_buf = outbuf;
675                         break;
676                 case TASKFILE_MULTI_IN:
677                         if (!drive->mult_count) {
678                                 /* (hs): give up if multcount is not set */
679                                 printk(KERN_ERR "%s: %s Multimode Read failure " \
680                                         "multcount is not set\n",
681                                         drive->name, __FUNCTION__);
682                                 err = -EPERM;
683                                 goto abort;
684                         }
685                         /* fall through */
686                 case TASKFILE_IN:
687                         /* fall through */
688                 case TASKFILE_IN_DMAQ:
689                 case TASKFILE_IN_DMA:
690                         nsect = taskin / SECTOR_SIZE;
691                         data_buf = inbuf;
692                         break;
693                 case TASKFILE_NO_DATA:
694                         break;
695                 default:
696                         err = -EFAULT;
697                         goto abort;
698         }
699
700         if (req_task->req_cmd == IDE_DRIVE_TASK_NO_DATA)
701                 nsect = 0;
702         else if (!nsect) {
703                 nsect = (args.tf.hob_nsect << 8) | args.tf.nsect;
704
705                 if (!nsect) {
706                         printk(KERN_ERR "%s: in/out command without data\n",
707                                         drive->name);
708                         err = -EFAULT;
709                         goto abort;
710                 }
711         }
712
713         if (req_task->req_cmd == IDE_DRIVE_TASK_RAW_WRITE)
714                 args.tf_flags |= IDE_TFLAG_WRITE;
715
716         err = ide_raw_taskfile(drive, &args, data_buf, nsect);
717
718         memcpy(req_task->hob_ports, &args.tf_array[0], HDIO_DRIVE_HOB_HDR_SIZE - 2);
719         memcpy(req_task->io_ports, &args.tf_array[6], HDIO_DRIVE_TASK_HDR_SIZE);
720
721         if ((args.tf_flags & IDE_TFLAG_FLAGGED_SET_IN_FLAGS) &&
722             req_task->in_flags.all == 0) {
723                 req_task->in_flags.all = IDE_TASKFILE_STD_IN_FLAGS;
724                 if (drive->addressing == 1)
725                         req_task->in_flags.all |= (IDE_HOB_STD_IN_FLAGS << 8);
726         }
727
728         if (copy_to_user(buf, req_task, tasksize)) {
729                 err = -EFAULT;
730                 goto abort;
731         }
732         if (taskout) {
733                 int outtotal = tasksize;
734                 if (copy_to_user(buf + outtotal, outbuf, taskout)) {
735                         err = -EFAULT;
736                         goto abort;
737                 }
738         }
739         if (taskin) {
740                 int intotal = tasksize + taskout;
741                 if (copy_to_user(buf + intotal, inbuf, taskin)) {
742                         err = -EFAULT;
743                         goto abort;
744                 }
745         }
746 abort:
747         kfree(req_task);
748         kfree(outbuf);
749         kfree(inbuf);
750
751 //      printk("IDE Taskfile ioctl ended. rc = %i\n", err);
752
753         return err;
754 }
755 #endif
756
757 int ide_wait_cmd (ide_drive_t *drive, u8 cmd, u8 nsect, u8 feature, u8 sectors, u8 *buf)
758 {
759         struct request rq;
760         u8 buffer[4];
761
762         if (!buf)
763                 buf = buffer;
764         memset(buf, 0, 4 + SECTOR_WORDS * 4 * sectors);
765         ide_init_drive_cmd(&rq);
766         rq.buffer = buf;
767         *buf++ = cmd;
768         *buf++ = nsect;
769         *buf++ = feature;
770         *buf++ = sectors;
771         return ide_do_drive_cmd(drive, &rq, ide_wait);
772 }
773
774 int ide_cmd_ioctl (ide_drive_t *drive, unsigned int cmd, unsigned long arg)
775 {
776         int err = 0;
777         u8 args[4], *argbuf = args;
778         u8 xfer_rate = 0;
779         int argsize = 4;
780         ide_task_t tfargs;
781         struct ide_taskfile *tf = &tfargs.tf;
782
783         if (NULL == (void *) arg) {
784                 struct request rq;
785                 ide_init_drive_cmd(&rq);
786                 return ide_do_drive_cmd(drive, &rq, ide_wait);
787         }
788
789         if (copy_from_user(args, (void __user *)arg, 4))
790                 return -EFAULT;
791
792         memset(&tfargs, 0, sizeof(ide_task_t));
793         tf->feature = args[2];
794         tf->nsect   = args[3];
795         tf->lbal    = args[1];
796         tf->command = args[0];
797
798         if (args[3]) {
799                 argsize = 4 + (SECTOR_WORDS * 4 * args[3]);
800                 argbuf = kzalloc(argsize, GFP_KERNEL);
801                 if (argbuf == NULL)
802                         return -ENOMEM;
803         }
804         if (set_transfer(drive, &tfargs)) {
805                 xfer_rate = args[1];
806                 if (ide_ata66_check(drive, &tfargs))
807                         goto abort;
808         }
809
810         err = ide_wait_cmd(drive, args[0], args[1], args[2], args[3], argbuf);
811
812         if (!err && xfer_rate) {
813                 /* active-retuning-calls future */
814                 ide_set_xfer_rate(drive, xfer_rate);
815                 ide_driveid_update(drive);
816         }
817 abort:
818         if (copy_to_user((void __user *)arg, argbuf, argsize))
819                 err = -EFAULT;
820         if (argsize > 4)
821                 kfree(argbuf);
822         return err;
823 }
824
825 int ide_task_ioctl (ide_drive_t *drive, unsigned int cmd, unsigned long arg)
826 {
827         void __user *p = (void __user *)arg;
828         int err = 0;
829         u8 args[7];
830         ide_task_t task;
831
832         if (copy_from_user(args, p, 7))
833                 return -EFAULT;
834
835         memset(&task, 0, sizeof(task));
836         memcpy(&task.tf_array[7], &args[1], 6);
837         task.tf.command = args[0];
838         task.tf_flags = IDE_TFLAG_TF | IDE_TFLAG_DEVICE;
839
840         err = ide_no_data_taskfile(drive, &task);
841
842         args[0] = task.tf.command;
843         memcpy(&args[1], &task.tf_array[7], 6);
844
845         if (copy_to_user(p, args, 7))
846                 err = -EFAULT;
847
848         return err;
849 }