[SCSI] aacraid: check buffer address in aac_internal_transfer
[linux-2.6] / drivers / scsi / aacraid / commctrl.c
1 /*
2  *      Adaptec AAC series RAID controller driver
3  *      (c) Copyright 2001 Red Hat Inc. <alan@redhat.com>
4  *
5  * based on the old aacraid driver that is..
6  * Adaptec aacraid device driver for Linux.
7  *
8  * Copyright (c) 2000-2007 Adaptec, Inc. (aacraid@adaptec.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 as published by
12  * the Free Software Foundation; either version 2, or (at your option)
13  * any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; see the file COPYING.  If not, write to
22  * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
23  *
24  * Module Name:
25  *  commctrl.c
26  *
27  * Abstract: Contains all routines for control of the AFA comm layer
28  *
29  */
30
31 #include <linux/kernel.h>
32 #include <linux/init.h>
33 #include <linux/types.h>
34 #include <linux/pci.h>
35 #include <linux/spinlock.h>
36 #include <linux/slab.h>
37 #include <linux/completion.h>
38 #include <linux/dma-mapping.h>
39 #include <linux/blkdev.h>
40 #include <linux/delay.h> /* ssleep prototype */
41 #include <linux/kthread.h>
42 #include <asm/semaphore.h>
43 #include <asm/uaccess.h>
44
45 #include "aacraid.h"
46
47 /**
48  *      ioctl_send_fib  -       send a FIB from userspace
49  *      @dev:   adapter is being processed
50  *      @arg:   arguments to the ioctl call
51  *      
52  *      This routine sends a fib to the adapter on behalf of a user level
53  *      program.
54  */
55 # define AAC_DEBUG_PREAMBLE     KERN_INFO
56 # define AAC_DEBUG_POSTAMBLE
57  
58 static int ioctl_send_fib(struct aac_dev * dev, void __user *arg)
59 {
60         struct hw_fib * kfib;
61         struct fib *fibptr;
62         struct hw_fib * hw_fib = (struct hw_fib *)0;
63         dma_addr_t hw_fib_pa = (dma_addr_t)0LL;
64         unsigned size;
65         int retval;
66
67         if (dev->in_reset) {
68                 return -EBUSY;
69         }
70         fibptr = aac_fib_alloc(dev);
71         if(fibptr == NULL) {
72                 return -ENOMEM;
73         }
74                 
75         kfib = fibptr->hw_fib_va;
76         /*
77          *      First copy in the header so that we can check the size field.
78          */
79         if (copy_from_user((void *)kfib, arg, sizeof(struct aac_fibhdr))) {
80                 aac_fib_free(fibptr);
81                 return -EFAULT;
82         }
83         /*
84          *      Since we copy based on the fib header size, make sure that we
85          *      will not overrun the buffer when we copy the memory. Return
86          *      an error if we would.
87          */
88         size = le16_to_cpu(kfib->header.Size) + sizeof(struct aac_fibhdr);
89         if (size < le16_to_cpu(kfib->header.SenderSize))
90                 size = le16_to_cpu(kfib->header.SenderSize);
91         if (size > dev->max_fib_size) {
92                 if (size > 2048) {
93                         retval = -EINVAL;
94                         goto cleanup;
95                 }
96                 /* Highjack the hw_fib */
97                 hw_fib = fibptr->hw_fib_va;
98                 hw_fib_pa = fibptr->hw_fib_pa;
99                 fibptr->hw_fib_va = kfib = pci_alloc_consistent(dev->pdev, size, &fibptr->hw_fib_pa);
100                 memset(((char *)kfib) + dev->max_fib_size, 0, size - dev->max_fib_size);
101                 memcpy(kfib, hw_fib, dev->max_fib_size);
102         }
103
104         if (copy_from_user(kfib, arg, size)) {
105                 retval = -EFAULT;
106                 goto cleanup;
107         }
108
109         if (kfib->header.Command == cpu_to_le16(TakeABreakPt)) {
110                 aac_adapter_interrupt(dev);
111                 /*
112                  * Since we didn't really send a fib, zero out the state to allow 
113                  * cleanup code not to assert.
114                  */
115                 kfib->header.XferState = 0;
116         } else {
117                 retval = aac_fib_send(le16_to_cpu(kfib->header.Command), fibptr,
118                                 le16_to_cpu(kfib->header.Size) , FsaNormal,
119                                 1, 1, NULL, NULL);
120                 if (retval) {
121                         goto cleanup;
122                 }
123                 if (aac_fib_complete(fibptr) != 0) {
124                         retval = -EINVAL;
125                         goto cleanup;
126                 }
127         }
128         /*
129          *      Make sure that the size returned by the adapter (which includes
130          *      the header) is less than or equal to the size of a fib, so we
131          *      don't corrupt application data. Then copy that size to the user
132          *      buffer. (Don't try to add the header information again, since it
133          *      was already included by the adapter.)
134          */
135
136         retval = 0;
137         if (copy_to_user(arg, (void *)kfib, size))
138                 retval = -EFAULT;
139 cleanup:
140         if (hw_fib) {
141                 pci_free_consistent(dev->pdev, size, kfib, fibptr->hw_fib_pa);
142                 fibptr->hw_fib_pa = hw_fib_pa;
143                 fibptr->hw_fib_va = hw_fib;
144         }
145         if (retval != -EINTR)
146                 aac_fib_free(fibptr);
147         return retval;
148 }
149
150 /**
151  *      open_getadapter_fib     -       Get the next fib
152  *
153  *      This routine will get the next Fib, if available, from the AdapterFibContext
154  *      passed in from the user.
155  */
156
157 static int open_getadapter_fib(struct aac_dev * dev, void __user *arg)
158 {
159         struct aac_fib_context * fibctx;
160         int status;
161
162         fibctx = kmalloc(sizeof(struct aac_fib_context), GFP_KERNEL);
163         if (fibctx == NULL) {
164                 status = -ENOMEM;
165         } else {
166                 unsigned long flags;
167                 struct list_head * entry;
168                 struct aac_fib_context * context;
169
170                 fibctx->type = FSAFS_NTC_GET_ADAPTER_FIB_CONTEXT;
171                 fibctx->size = sizeof(struct aac_fib_context);
172                 /*
173                  *      Yes yes, I know this could be an index, but we have a
174                  * better guarantee of uniqueness for the locked loop below.
175                  * Without the aid of a persistent history, this also helps
176                  * reduce the chance that the opaque context would be reused.
177                  */
178                 fibctx->unique = (u32)((ulong)fibctx & 0xFFFFFFFF);
179                 /*
180                  *      Initialize the mutex used to wait for the next AIF.
181                  */
182                 init_MUTEX_LOCKED(&fibctx->wait_sem);
183                 fibctx->wait = 0;
184                 /*
185                  *      Initialize the fibs and set the count of fibs on
186                  *      the list to 0.
187                  */
188                 fibctx->count = 0;
189                 INIT_LIST_HEAD(&fibctx->fib_list);
190                 fibctx->jiffies = jiffies/HZ;
191                 /*
192                  *      Now add this context onto the adapter's 
193                  *      AdapterFibContext list.
194                  */
195                 spin_lock_irqsave(&dev->fib_lock, flags);
196                 /* Ensure that we have a unique identifier */
197                 entry = dev->fib_list.next;
198                 while (entry != &dev->fib_list) {
199                         context = list_entry(entry, struct aac_fib_context, next);
200                         if (context->unique == fibctx->unique) {
201                                 /* Not unique (32 bits) */
202                                 fibctx->unique++;
203                                 entry = dev->fib_list.next;
204                         } else {
205                                 entry = entry->next;
206                         }
207                 }
208                 list_add_tail(&fibctx->next, &dev->fib_list);
209                 spin_unlock_irqrestore(&dev->fib_lock, flags);
210                 if (copy_to_user(arg,  &fibctx->unique, 
211                                                 sizeof(fibctx->unique))) {
212                         status = -EFAULT;
213                 } else {
214                         status = 0;
215                 }       
216         }
217         return status;
218 }
219
220 /**
221  *      next_getadapter_fib     -       get the next fib
222  *      @dev: adapter to use
223  *      @arg: ioctl argument
224  *      
225  *      This routine will get the next Fib, if available, from the AdapterFibContext
226  *      passed in from the user.
227  */
228
229 static int next_getadapter_fib(struct aac_dev * dev, void __user *arg)
230 {
231         struct fib_ioctl f;
232         struct fib *fib;
233         struct aac_fib_context *fibctx;
234         int status;
235         struct list_head * entry;
236         unsigned long flags;
237         
238         if(copy_from_user((void *)&f, arg, sizeof(struct fib_ioctl)))
239                 return -EFAULT;
240         /*
241          *      Verify that the HANDLE passed in was a valid AdapterFibContext
242          *
243          *      Search the list of AdapterFibContext addresses on the adapter
244          *      to be sure this is a valid address
245          */
246         entry = dev->fib_list.next;
247         fibctx = NULL;
248
249         while (entry != &dev->fib_list) {
250                 fibctx = list_entry(entry, struct aac_fib_context, next);
251                 /*
252                  *      Extract the AdapterFibContext from the Input parameters.
253                  */
254                 if (fibctx->unique == f.fibctx) {   /* We found a winner */
255                         break;
256                 }
257                 entry = entry->next;
258                 fibctx = NULL;
259         }
260         if (!fibctx) {
261                 dprintk ((KERN_INFO "Fib Context not found\n"));
262                 return -EINVAL;
263         }
264
265         if((fibctx->type != FSAFS_NTC_GET_ADAPTER_FIB_CONTEXT) ||
266                  (fibctx->size != sizeof(struct aac_fib_context))) {
267                 dprintk ((KERN_INFO "Fib Context corrupt?\n"));
268                 return -EINVAL;
269         }
270         status = 0;
271         spin_lock_irqsave(&dev->fib_lock, flags);
272         /*
273          *      If there are no fibs to send back, then either wait or return
274          *      -EAGAIN
275          */
276 return_fib:
277         if (!list_empty(&fibctx->fib_list)) {
278                 struct list_head * entry;
279                 /*
280                  *      Pull the next fib from the fibs
281                  */
282                 entry = fibctx->fib_list.next;
283                 list_del(entry);
284                 
285                 fib = list_entry(entry, struct fib, fiblink);
286                 fibctx->count--;
287                 spin_unlock_irqrestore(&dev->fib_lock, flags);
288                 if (copy_to_user(f.fib, fib->hw_fib_va, sizeof(struct hw_fib))) {
289                         kfree(fib->hw_fib_va);
290                         kfree(fib);
291                         return -EFAULT;
292                 }       
293                 /*
294                  *      Free the space occupied by this copy of the fib.
295                  */
296                 kfree(fib->hw_fib_va);
297                 kfree(fib);
298                 status = 0;
299         } else {
300                 spin_unlock_irqrestore(&dev->fib_lock, flags);
301                 /* If someone killed the AIF aacraid thread, restart it */
302                 status = !dev->aif_thread;
303                 if (status && !dev->in_reset && dev->queues && dev->fsa_dev) {
304                         /* Be paranoid, be very paranoid! */
305                         kthread_stop(dev->thread);
306                         ssleep(1);
307                         dev->aif_thread = 0;
308                         dev->thread = kthread_run(aac_command_thread, dev, dev->name);
309                         ssleep(1);
310                 }
311                 if (f.wait) {
312                         if(down_interruptible(&fibctx->wait_sem) < 0) {
313                                 status = -EINTR;
314                         } else {
315                                 /* Lock again and retry */
316                                 spin_lock_irqsave(&dev->fib_lock, flags);
317                                 goto return_fib;
318                         }
319                 } else {
320                         status = -EAGAIN;
321                 }       
322         }
323         fibctx->jiffies = jiffies/HZ;
324         return status;
325 }
326
327 int aac_close_fib_context(struct aac_dev * dev, struct aac_fib_context * fibctx)
328 {
329         struct fib *fib;
330
331         /*
332          *      First free any FIBs that have not been consumed.
333          */
334         while (!list_empty(&fibctx->fib_list)) {
335                 struct list_head * entry;
336                 /*
337                  *      Pull the next fib from the fibs
338                  */
339                 entry = fibctx->fib_list.next;
340                 list_del(entry);
341                 fib = list_entry(entry, struct fib, fiblink);
342                 fibctx->count--;
343                 /*
344                  *      Free the space occupied by this copy of the fib.
345                  */
346                 kfree(fib->hw_fib_va);
347                 kfree(fib);
348         }
349         /*
350          *      Remove the Context from the AdapterFibContext List
351          */
352         list_del(&fibctx->next);
353         /*
354          *      Invalidate context
355          */
356         fibctx->type = 0;
357         /*
358          *      Free the space occupied by the Context
359          */
360         kfree(fibctx);
361         return 0;
362 }
363
364 /**
365  *      close_getadapter_fib    -       close down user fib context
366  *      @dev: adapter
367  *      @arg: ioctl arguments
368  *
369  *      This routine will close down the fibctx passed in from the user.
370  */
371  
372 static int close_getadapter_fib(struct aac_dev * dev, void __user *arg)
373 {
374         struct aac_fib_context *fibctx;
375         int status;
376         unsigned long flags;
377         struct list_head * entry;
378
379         /*
380          *      Verify that the HANDLE passed in was a valid AdapterFibContext
381          *
382          *      Search the list of AdapterFibContext addresses on the adapter
383          *      to be sure this is a valid address
384          */
385
386         entry = dev->fib_list.next;
387         fibctx = NULL;
388
389         while(entry != &dev->fib_list) {
390                 fibctx = list_entry(entry, struct aac_fib_context, next);
391                 /*
392                  *      Extract the fibctx from the input parameters
393                  */
394                 if (fibctx->unique == (u32)(unsigned long)arg) {   
395                         /* We found a winner */
396                         break;
397                 }
398                 entry = entry->next;
399                 fibctx = NULL;
400         }
401
402         if (!fibctx)
403                 return 0; /* Already gone */
404
405         if((fibctx->type != FSAFS_NTC_GET_ADAPTER_FIB_CONTEXT) ||
406                  (fibctx->size != sizeof(struct aac_fib_context)))
407                 return -EINVAL;
408         spin_lock_irqsave(&dev->fib_lock, flags);
409         status = aac_close_fib_context(dev, fibctx);
410         spin_unlock_irqrestore(&dev->fib_lock, flags);
411         return status;
412 }
413
414 /**
415  *      check_revision  -       close down user fib context
416  *      @dev: adapter
417  *      @arg: ioctl arguments
418  *
419  *      This routine returns the driver version.
420  *      Under Linux, there have been no version incompatibilities, so this is 
421  *      simple!
422  */
423
424 static int check_revision(struct aac_dev *dev, void __user *arg)
425 {
426         struct revision response;
427         char *driver_version = aac_driver_version;
428         u32 version;
429
430         response.compat = 1;
431         version = (simple_strtol(driver_version, 
432                                 &driver_version, 10) << 24) | 0x00000400;
433         version += simple_strtol(driver_version + 1, &driver_version, 10) << 16;
434         version += simple_strtol(driver_version + 1, NULL, 10);
435         response.version = cpu_to_le32(version);
436 #       if (defined(AAC_DRIVER_BUILD))
437                 response.build = cpu_to_le32(AAC_DRIVER_BUILD);
438 #       else
439                 response.build = cpu_to_le32(9999);
440 #       endif
441
442         if (copy_to_user(arg, &response, sizeof(response)))
443                 return -EFAULT;
444         return 0;
445 }
446
447
448 /**
449  *
450  * aac_send_raw_scb
451  *
452  */
453
454 static int aac_send_raw_srb(struct aac_dev* dev, void __user * arg)
455 {
456         struct fib* srbfib;
457         int status;
458         struct aac_srb *srbcmd = NULL;
459         struct user_aac_srb *user_srbcmd = NULL;
460         struct user_aac_srb __user *user_srb = arg;
461         struct aac_srb_reply __user *user_reply;
462         struct aac_srb_reply* reply;
463         u32 fibsize = 0;
464         u32 flags = 0;
465         s32 rcode = 0;
466         u32 data_dir;
467         void __user *sg_user[32];
468         void *sg_list[32];
469         u32   sg_indx = 0;
470         u32 byte_count = 0;
471         u32 actual_fibsize64, actual_fibsize = 0;
472         int i;
473
474
475         if (dev->in_reset) {
476                 dprintk((KERN_DEBUG"aacraid: send raw srb -EBUSY\n"));
477                 return -EBUSY;
478         }
479         if (!capable(CAP_SYS_ADMIN)){
480                 dprintk((KERN_DEBUG"aacraid: No permission to send raw srb\n")); 
481                 return -EPERM;
482         }
483         /*
484          *      Allocate and initialize a Fib then setup a SRB command
485          */
486         if (!(srbfib = aac_fib_alloc(dev))) {
487                 return -ENOMEM;
488         }
489         aac_fib_init(srbfib);
490
491         srbcmd = (struct aac_srb*) fib_data(srbfib);
492
493         memset(sg_list, 0, sizeof(sg_list)); /* cleanup may take issue */
494         if(copy_from_user(&fibsize, &user_srb->count,sizeof(u32))){
495                 dprintk((KERN_DEBUG"aacraid: Could not copy data size from user\n")); 
496                 rcode = -EFAULT;
497                 goto cleanup;
498         }
499
500         if (fibsize > (dev->max_fib_size - sizeof(struct aac_fibhdr))) {
501                 rcode = -EINVAL;
502                 goto cleanup;
503         }
504
505         user_srbcmd = kmalloc(fibsize, GFP_KERNEL);
506         if (!user_srbcmd) {
507                 dprintk((KERN_DEBUG"aacraid: Could not make a copy of the srb\n"));
508                 rcode = -ENOMEM;
509                 goto cleanup;
510         }
511         if(copy_from_user(user_srbcmd, user_srb,fibsize)){
512                 dprintk((KERN_DEBUG"aacraid: Could not copy srb from user\n")); 
513                 rcode = -EFAULT;
514                 goto cleanup;
515         }
516
517         user_reply = arg+fibsize;
518
519         flags = user_srbcmd->flags; /* from user in cpu order */
520         // Fix up srb for endian and force some values
521
522         srbcmd->function = cpu_to_le32(SRBF_ExecuteScsi);       // Force this
523         srbcmd->channel  = cpu_to_le32(user_srbcmd->channel);
524         srbcmd->id       = cpu_to_le32(user_srbcmd->id);
525         srbcmd->lun      = cpu_to_le32(user_srbcmd->lun);
526         srbcmd->timeout  = cpu_to_le32(user_srbcmd->timeout);
527         srbcmd->flags    = cpu_to_le32(flags);
528         srbcmd->retry_limit = 0; // Obsolete parameter
529         srbcmd->cdb_size = cpu_to_le32(user_srbcmd->cdb_size);
530         memcpy(srbcmd->cdb, user_srbcmd->cdb, sizeof(srbcmd->cdb));
531         
532         switch (flags & (SRB_DataIn | SRB_DataOut)) {
533         case SRB_DataOut:
534                 data_dir = DMA_TO_DEVICE;
535                 break;
536         case (SRB_DataIn | SRB_DataOut):
537                 data_dir = DMA_BIDIRECTIONAL;
538                 break;
539         case SRB_DataIn:
540                 data_dir = DMA_FROM_DEVICE;
541                 break;
542         default:
543                 data_dir = DMA_NONE;
544         }
545         if (user_srbcmd->sg.count > ARRAY_SIZE(sg_list)) {
546                 dprintk((KERN_DEBUG"aacraid: too many sg entries %d\n",
547                   le32_to_cpu(srbcmd->sg.count)));
548                 rcode = -EINVAL;
549                 goto cleanup;
550         }
551         actual_fibsize = sizeof(struct aac_srb) - sizeof(struct sgentry) +
552                 ((user_srbcmd->sg.count & 0xff) * sizeof(struct sgentry));
553         actual_fibsize64 = actual_fibsize + (user_srbcmd->sg.count & 0xff) *
554           (sizeof(struct sgentry64) - sizeof(struct sgentry));
555         /* User made a mistake - should not continue */
556         if ((actual_fibsize != fibsize) && (actual_fibsize64 != fibsize)) {
557                 dprintk((KERN_DEBUG"aacraid: Bad Size specified in "
558                   "Raw SRB command calculated fibsize=%lu;%lu "
559                   "user_srbcmd->sg.count=%d aac_srb=%lu sgentry=%lu;%lu "
560                   "issued fibsize=%d\n",
561                   actual_fibsize, actual_fibsize64, user_srbcmd->sg.count,
562                   sizeof(struct aac_srb), sizeof(struct sgentry),
563                   sizeof(struct sgentry64), fibsize));
564                 rcode = -EINVAL;
565                 goto cleanup;
566         }
567         if ((data_dir == DMA_NONE) && user_srbcmd->sg.count) {
568                 dprintk((KERN_DEBUG"aacraid: SG with no direction specified in Raw SRB command\n"));
569                 rcode = -EINVAL;
570                 goto cleanup;
571         }
572         byte_count = 0;
573         if (dev->adapter_info.options & AAC_OPT_SGMAP_HOST64) {
574                 struct user_sgmap64* upsg = (struct user_sgmap64*)&user_srbcmd->sg;
575                 struct sgmap64* psg = (struct sgmap64*)&srbcmd->sg;
576
577                 /*
578                  * This should also catch if user used the 32 bit sgmap
579                  */
580                 if (actual_fibsize64 == fibsize) {
581                         actual_fibsize = actual_fibsize64;
582                         for (i = 0; i < upsg->count; i++) {
583                                 u64 addr;
584                                 void* p;
585                                 /* Does this really need to be GFP_DMA? */
586                                 p = kmalloc(upsg->sg[i].count,GFP_KERNEL|__GFP_DMA);
587                                 if(p == 0) {
588                                         dprintk((KERN_DEBUG"aacraid: Could not allocate SG buffer - size = %d buffer number %d of %d\n",
589                                           upsg->sg[i].count,i,upsg->count));
590                                         rcode = -ENOMEM;
591                                         goto cleanup;
592                                 }
593                                 addr = (u64)upsg->sg[i].addr[0];
594                                 addr += ((u64)upsg->sg[i].addr[1]) << 32;
595                                 sg_user[i] = (void __user *)(ptrdiff_t)addr;
596                                 sg_list[i] = p; // save so we can clean up later
597                                 sg_indx = i;
598
599                                 if( flags & SRB_DataOut ){
600                                         if(copy_from_user(p,sg_user[i],upsg->sg[i].count)){
601                                                 dprintk((KERN_DEBUG"aacraid: Could not copy sg data from user\n"));
602                                                 rcode = -EFAULT;
603                                                 goto cleanup;
604                                         }
605                                 }
606                                 addr = pci_map_single(dev->pdev, p, upsg->sg[i].count, data_dir);
607
608                                 psg->sg[i].addr[0] = cpu_to_le32(addr & 0xffffffff);
609                                 psg->sg[i].addr[1] = cpu_to_le32(addr>>32);
610                                 byte_count += upsg->sg[i].count;
611                                 psg->sg[i].count = cpu_to_le32(upsg->sg[i].count);
612                         }
613                 } else {
614                         struct user_sgmap* usg;
615                         usg = kmalloc(actual_fibsize - sizeof(struct aac_srb)
616                           + sizeof(struct sgmap), GFP_KERNEL);
617                         if (!usg) {
618                                 dprintk((KERN_DEBUG"aacraid: Allocation error in Raw SRB command\n"));
619                                 rcode = -ENOMEM;
620                                 goto cleanup;
621                         }
622                         memcpy (usg, upsg, actual_fibsize - sizeof(struct aac_srb)
623                           + sizeof(struct sgmap));
624                         actual_fibsize = actual_fibsize64;
625
626                         for (i = 0; i < usg->count; i++) {
627                                 u64 addr;
628                                 void* p;
629                                 /* Does this really need to be GFP_DMA? */
630                                 p = kmalloc(usg->sg[i].count,GFP_KERNEL|__GFP_DMA);
631                                 if(p == 0) {
632                                         kfree (usg);
633                                         dprintk((KERN_DEBUG"aacraid: Could not allocate SG buffer - size = %d buffer number %d of %d\n",
634                                           usg->sg[i].count,i,usg->count));
635                                         rcode = -ENOMEM;
636                                         goto cleanup;
637                                 }
638                                 sg_user[i] = (void __user *)(ptrdiff_t)usg->sg[i].addr;
639                                 sg_list[i] = p; // save so we can clean up later
640                                 sg_indx = i;
641
642                                 if( flags & SRB_DataOut ){
643                                         if(copy_from_user(p,sg_user[i],upsg->sg[i].count)){
644                                                 kfree (usg);
645                                                 dprintk((KERN_DEBUG"aacraid: Could not copy sg data from user\n"));
646                                                 rcode = -EFAULT;
647                                                 goto cleanup;
648                                         }
649                                 }
650                                 addr = pci_map_single(dev->pdev, p, usg->sg[i].count, data_dir);
651
652                                 psg->sg[i].addr[0] = cpu_to_le32(addr & 0xffffffff);
653                                 psg->sg[i].addr[1] = cpu_to_le32(addr>>32);
654                                 byte_count += usg->sg[i].count;
655                                 psg->sg[i].count = cpu_to_le32(usg->sg[i].count);
656                         }
657                         kfree (usg);
658                 }
659                 srbcmd->count = cpu_to_le32(byte_count);
660                 psg->count = cpu_to_le32(sg_indx+1);
661                 status = aac_fib_send(ScsiPortCommand64, srbfib, actual_fibsize, FsaNormal, 1, 1,NULL,NULL);
662         } else {
663                 struct user_sgmap* upsg = &user_srbcmd->sg;
664                 struct sgmap* psg = &srbcmd->sg;
665
666                 if (actual_fibsize64 == fibsize) {
667                         struct user_sgmap64* usg = (struct user_sgmap64 *)upsg;
668                         for (i = 0; i < upsg->count; i++) {
669                                 u64 addr;
670                                 void* p;
671                                 /* Does this really need to be GFP_DMA? */
672                                 p = kmalloc(usg->sg[i].count,GFP_KERNEL|__GFP_DMA);
673                                 if(p == 0) {
674                                         dprintk((KERN_DEBUG"aacraid: Could not allocate SG buffer - size = %d buffer number %d of %d\n",
675                                           usg->sg[i].count,i,usg->count));
676                                         rcode = -ENOMEM;
677                                         goto cleanup;
678                                 }
679                                 addr = (u64)usg->sg[i].addr[0];
680                                 addr += ((u64)usg->sg[i].addr[1]) << 32;
681                                 sg_user[i] = (void __user *)(ptrdiff_t)addr;
682                                 sg_list[i] = p; // save so we can clean up later
683                                 sg_indx = i;
684
685                                 if( flags & SRB_DataOut ){
686                                         if(copy_from_user(p,sg_user[i],usg->sg[i].count)){
687                                                 dprintk((KERN_DEBUG"aacraid: Could not copy sg data from user\n"));
688                                                 rcode = -EFAULT;
689                                                 goto cleanup;
690                                         }
691                                 }
692                                 addr = pci_map_single(dev->pdev, p, usg->sg[i].count, data_dir);
693
694                                 psg->sg[i].addr = cpu_to_le32(addr & 0xffffffff);
695                                 byte_count += usg->sg[i].count;
696                                 psg->sg[i].count = cpu_to_le32(usg->sg[i].count);
697                         }
698                 } else {
699                         for (i = 0; i < upsg->count; i++) {
700                                 dma_addr_t addr;
701                                 void* p;
702                                 p = kmalloc(upsg->sg[i].count, GFP_KERNEL);
703                                 if(p == 0) {
704                                         dprintk((KERN_DEBUG"aacraid: Could not allocate SG buffer - size = %d buffer number %d of %d\n",
705                                           upsg->sg[i].count, i, upsg->count));
706                                         rcode = -ENOMEM;
707                                         goto cleanup;
708                                 }
709                                 sg_user[i] = (void __user *)(ptrdiff_t)upsg->sg[i].addr;
710                                 sg_list[i] = p; // save so we can clean up later
711                                 sg_indx = i;
712
713                                 if( flags & SRB_DataOut ){
714                                         if(copy_from_user(p, sg_user[i],
715                                                         upsg->sg[i].count)) {
716                                                 dprintk((KERN_DEBUG"aacraid: Could not copy sg data from user\n"));
717                                                 rcode = -EFAULT;
718                                                 goto cleanup;
719                                         }
720                                 }
721                                 addr = pci_map_single(dev->pdev, p,
722                                         upsg->sg[i].count, data_dir);
723
724                                 psg->sg[i].addr = cpu_to_le32(addr);
725                                 byte_count += upsg->sg[i].count;
726                                 psg->sg[i].count = cpu_to_le32(upsg->sg[i].count);
727                         }
728                 }
729                 srbcmd->count = cpu_to_le32(byte_count);
730                 psg->count = cpu_to_le32(sg_indx+1);
731                 status = aac_fib_send(ScsiPortCommand, srbfib, actual_fibsize, FsaNormal, 1, 1, NULL, NULL);
732         }
733         if (status == -EINTR) {
734                 rcode = -EINTR;
735                 goto cleanup;
736         }
737
738         if (status != 0){
739                 dprintk((KERN_DEBUG"aacraid: Could not send raw srb fib to hba\n")); 
740                 rcode = -ENXIO;
741                 goto cleanup;
742         }
743
744         if( flags & SRB_DataIn ) {
745                 for(i = 0 ; i <= sg_indx; i++){
746                         byte_count = le32_to_cpu(
747                           (dev->adapter_info.options & AAC_OPT_SGMAP_HOST64)
748                               ? ((struct sgmap64*)&srbcmd->sg)->sg[i].count
749                               : srbcmd->sg.sg[i].count);
750                         if(copy_to_user(sg_user[i], sg_list[i], byte_count)){
751                                 dprintk((KERN_DEBUG"aacraid: Could not copy sg data to user\n")); 
752                                 rcode = -EFAULT;
753                                 goto cleanup;
754
755                         }
756                 }
757         }
758
759         reply = (struct aac_srb_reply *) fib_data(srbfib);
760         if(copy_to_user(user_reply,reply,sizeof(struct aac_srb_reply))){
761                 dprintk((KERN_DEBUG"aacraid: Could not copy reply to user\n")); 
762                 rcode = -EFAULT;
763                 goto cleanup;
764         }
765
766 cleanup:
767         kfree(user_srbcmd);
768         for(i=0; i <= sg_indx; i++){
769                 kfree(sg_list[i]);
770         }
771         if (rcode != -EINTR) {
772                 aac_fib_complete(srbfib);
773                 aac_fib_free(srbfib);
774         }
775
776         return rcode;
777 }
778
779 struct aac_pci_info {
780         u32 bus;
781         u32 slot;
782 };
783
784
785 static int aac_get_pci_info(struct aac_dev* dev, void __user *arg)
786 {
787         struct aac_pci_info pci_info;
788
789         pci_info.bus = dev->pdev->bus->number;
790         pci_info.slot = PCI_SLOT(dev->pdev->devfn);
791
792        if (copy_to_user(arg, &pci_info, sizeof(struct aac_pci_info))) {
793                dprintk((KERN_DEBUG "aacraid: Could not copy pci info\n"));
794                return -EFAULT;
795         }
796         return 0;
797 }
798  
799
800 int aac_do_ioctl(struct aac_dev * dev, int cmd, void __user *arg)
801 {
802         int status;
803         
804         /*
805          *      HBA gets first crack
806          */
807          
808         status = aac_dev_ioctl(dev, cmd, arg);
809         if(status != -ENOTTY)
810                 return status;
811
812         switch (cmd) {
813         case FSACTL_MINIPORT_REV_CHECK:
814                 status = check_revision(dev, arg);
815                 break;
816         case FSACTL_SEND_LARGE_FIB:
817         case FSACTL_SENDFIB:
818                 status = ioctl_send_fib(dev, arg);
819                 break;
820         case FSACTL_OPEN_GET_ADAPTER_FIB:
821                 status = open_getadapter_fib(dev, arg);
822                 break;
823         case FSACTL_GET_NEXT_ADAPTER_FIB:
824                 status = next_getadapter_fib(dev, arg);
825                 break;
826         case FSACTL_CLOSE_GET_ADAPTER_FIB:
827                 status = close_getadapter_fib(dev, arg);
828                 break;
829         case FSACTL_SEND_RAW_SRB:
830                 status = aac_send_raw_srb(dev,arg);
831                 break;
832         case FSACTL_GET_PCI_INFO:
833                 status = aac_get_pci_info(dev,arg);
834                 break;
835         default:
836                 status = -ENOTTY;
837                 break;  
838         }
839         return status;
840 }
841