USB: gadgetfs race fix
[linux-2.6] / drivers / scsi / libsrp.c
1 /*
2  * SCSI RDAM Protocol lib functions
3  *
4  * Copyright (C) 2006 FUJITA Tomonori <tomof@acm.org>
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License as
8  * published by the Free Software Foundation; either version 2 of the
9  * License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
19  * 02110-1301 USA
20  */
21 #include <linux/err.h>
22 #include <linux/kfifo.h>
23 #include <linux/scatterlist.h>
24 #include <linux/dma-mapping.h>
25 #include <linux/pci.h>
26 #include <scsi/scsi.h>
27 #include <scsi/scsi_cmnd.h>
28 #include <scsi/scsi_tcq.h>
29 #include <scsi/scsi_tgt.h>
30 #include <scsi/srp.h>
31 #include <scsi/libsrp.h>
32
33 enum srp_task_attributes {
34         SRP_SIMPLE_TASK = 0,
35         SRP_HEAD_TASK = 1,
36         SRP_ORDERED_TASK = 2,
37         SRP_ACA_TASK = 4
38 };
39
40 /* tmp - will replace with SCSI logging stuff */
41 #define eprintk(fmt, args...)                                   \
42 do {                                                            \
43         printk("%s(%d) " fmt, __FUNCTION__, __LINE__, ##args);  \
44 } while (0)
45 /* #define dprintk eprintk */
46 #define dprintk(fmt, args...)
47
48 static int srp_iu_pool_alloc(struct srp_queue *q, size_t max,
49                              struct srp_buf **ring)
50 {
51         int i;
52         struct iu_entry *iue;
53
54         q->pool = kcalloc(max, sizeof(struct iu_entry *), GFP_KERNEL);
55         if (!q->pool)
56                 return -ENOMEM;
57         q->items = kcalloc(max, sizeof(struct iu_entry), GFP_KERNEL);
58         if (!q->items)
59                 goto free_pool;
60
61         spin_lock_init(&q->lock);
62         q->queue = kfifo_init((void *) q->pool, max * sizeof(void *),
63                               GFP_KERNEL, &q->lock);
64         if (IS_ERR(q->queue))
65                 goto free_item;
66
67         for (i = 0, iue = q->items; i < max; i++) {
68                 __kfifo_put(q->queue, (void *) &iue, sizeof(void *));
69                 iue->sbuf = ring[i];
70                 iue++;
71         }
72         return 0;
73
74 free_item:
75         kfree(q->items);
76 free_pool:
77         kfree(q->pool);
78         return -ENOMEM;
79 }
80
81 static void srp_iu_pool_free(struct srp_queue *q)
82 {
83         kfree(q->items);
84         kfree(q->pool);
85 }
86
87 static struct srp_buf **srp_ring_alloc(struct device *dev,
88                                        size_t max, size_t size)
89 {
90         int i;
91         struct srp_buf **ring;
92
93         ring = kcalloc(max, sizeof(struct srp_buf *), GFP_KERNEL);
94         if (!ring)
95                 return NULL;
96
97         for (i = 0; i < max; i++) {
98                 ring[i] = kzalloc(sizeof(struct srp_buf), GFP_KERNEL);
99                 if (!ring[i])
100                         goto out;
101                 ring[i]->buf = dma_alloc_coherent(dev, size, &ring[i]->dma,
102                                                   GFP_KERNEL);
103                 if (!ring[i]->buf)
104                         goto out;
105         }
106         return ring;
107
108 out:
109         for (i = 0; i < max && ring[i]; i++) {
110                 if (ring[i]->buf)
111                         dma_free_coherent(dev, size, ring[i]->buf, ring[i]->dma);
112                 kfree(ring[i]);
113         }
114         kfree(ring);
115
116         return NULL;
117 }
118
119 static void srp_ring_free(struct device *dev, struct srp_buf **ring, size_t max,
120                           size_t size)
121 {
122         int i;
123
124         for (i = 0; i < max; i++) {
125                 dma_free_coherent(dev, size, ring[i]->buf, ring[i]->dma);
126                 kfree(ring[i]);
127         }
128 }
129
130 int srp_target_alloc(struct srp_target *target, struct device *dev,
131                      size_t nr, size_t iu_size)
132 {
133         int err;
134
135         spin_lock_init(&target->lock);
136         INIT_LIST_HEAD(&target->cmd_queue);
137
138         target->dev = dev;
139         target->dev->driver_data = target;
140
141         target->srp_iu_size = iu_size;
142         target->rx_ring_size = nr;
143         target->rx_ring = srp_ring_alloc(target->dev, nr, iu_size);
144         if (!target->rx_ring)
145                 return -ENOMEM;
146         err = srp_iu_pool_alloc(&target->iu_queue, nr, target->rx_ring);
147         if (err)
148                 goto free_ring;
149
150         return 0;
151
152 free_ring:
153         srp_ring_free(target->dev, target->rx_ring, nr, iu_size);
154         return -ENOMEM;
155 }
156 EXPORT_SYMBOL_GPL(srp_target_alloc);
157
158 void srp_target_free(struct srp_target *target)
159 {
160         srp_ring_free(target->dev, target->rx_ring, target->rx_ring_size,
161                       target->srp_iu_size);
162         srp_iu_pool_free(&target->iu_queue);
163 }
164 EXPORT_SYMBOL_GPL(srp_target_free);
165
166 struct iu_entry *srp_iu_get(struct srp_target *target)
167 {
168         struct iu_entry *iue = NULL;
169
170         kfifo_get(target->iu_queue.queue, (void *) &iue, sizeof(void *));
171         if (!iue)
172                 return iue;
173         iue->target = target;
174         INIT_LIST_HEAD(&iue->ilist);
175         iue->flags = 0;
176         return iue;
177 }
178 EXPORT_SYMBOL_GPL(srp_iu_get);
179
180 void srp_iu_put(struct iu_entry *iue)
181 {
182         kfifo_put(iue->target->iu_queue.queue, (void *) &iue, sizeof(void *));
183 }
184 EXPORT_SYMBOL_GPL(srp_iu_put);
185
186 static int srp_direct_data(struct scsi_cmnd *sc, struct srp_direct_buf *md,
187                            enum dma_data_direction dir, srp_rdma_t rdma_io,
188                            int dma_map, int ext_desc)
189 {
190         struct iu_entry *iue = NULL;
191         struct scatterlist *sg = NULL;
192         int err, nsg = 0, len;
193
194         if (dma_map) {
195                 iue = (struct iu_entry *) sc->SCp.ptr;
196                 sg = sc->request_buffer;
197
198                 dprintk("%p %u %u %d\n", iue, sc->request_bufflen,
199                         md->len, sc->use_sg);
200
201                 nsg = dma_map_sg(iue->target->dev, sg, sc->use_sg,
202                                  DMA_BIDIRECTIONAL);
203                 if (!nsg) {
204                         printk("fail to map %p %d\n", iue, sc->use_sg);
205                         return 0;
206                 }
207                 len = min(sc->request_bufflen, md->len);
208         } else
209                 len = md->len;
210
211         err = rdma_io(sc, sg, nsg, md, 1, dir, len);
212
213         if (dma_map)
214                 dma_unmap_sg(iue->target->dev, sg, nsg, DMA_BIDIRECTIONAL);
215
216         return err;
217 }
218
219 static int srp_indirect_data(struct scsi_cmnd *sc, struct srp_cmd *cmd,
220                              struct srp_indirect_buf *id,
221                              enum dma_data_direction dir, srp_rdma_t rdma_io,
222                              int dma_map, int ext_desc)
223 {
224         struct iu_entry *iue = NULL;
225         struct srp_direct_buf *md = NULL;
226         struct scatterlist dummy, *sg = NULL;
227         dma_addr_t token = 0;
228         long err;
229         unsigned int done = 0;
230         int nmd, nsg = 0, len;
231
232         if (dma_map || ext_desc) {
233                 iue = (struct iu_entry *) sc->SCp.ptr;
234                 sg = sc->request_buffer;
235
236                 dprintk("%p %u %u %d %d\n",
237                         iue, sc->request_bufflen, id->len,
238                         cmd->data_in_desc_cnt, cmd->data_out_desc_cnt);
239         }
240
241         nmd = id->table_desc.len / sizeof(struct srp_direct_buf);
242
243         if ((dir == DMA_FROM_DEVICE && nmd == cmd->data_in_desc_cnt) ||
244             (dir == DMA_TO_DEVICE && nmd == cmd->data_out_desc_cnt)) {
245                 md = &id->desc_list[0];
246                 goto rdma;
247         }
248
249         if (ext_desc && dma_map) {
250                 md = dma_alloc_coherent(iue->target->dev, id->table_desc.len,
251                                 &token, GFP_KERNEL);
252                 if (!md) {
253                         eprintk("Can't get dma memory %u\n", id->table_desc.len);
254                         return -ENOMEM;
255                 }
256
257                 sg_init_one(&dummy, md, id->table_desc.len);
258                 sg_dma_address(&dummy) = token;
259                 err = rdma_io(sc, &dummy, 1, &id->table_desc, 1, DMA_TO_DEVICE,
260                               id->table_desc.len);
261                 if (err < 0) {
262                         eprintk("Error copying indirect table %ld\n", err);
263                         goto free_mem;
264                 }
265         } else {
266                 eprintk("This command uses external indirect buffer\n");
267                 return -EINVAL;
268         }
269
270 rdma:
271         if (dma_map) {
272                 nsg = dma_map_sg(iue->target->dev, sg, sc->use_sg, DMA_BIDIRECTIONAL);
273                 if (!nsg) {
274                         eprintk("fail to map %p %d\n", iue, sc->use_sg);
275                         goto free_mem;
276                 }
277                 len = min(sc->request_bufflen, id->len);
278         } else
279                 len = id->len;
280
281         err = rdma_io(sc, sg, nsg, md, nmd, dir, len);
282
283         if (dma_map)
284                 dma_unmap_sg(iue->target->dev, sg, nsg, DMA_BIDIRECTIONAL);
285
286 free_mem:
287         if (token && dma_map)
288                 dma_free_coherent(iue->target->dev, id->table_desc.len, md, token);
289
290         return done;
291 }
292
293 static int data_out_desc_size(struct srp_cmd *cmd)
294 {
295         int size = 0;
296         u8 fmt = cmd->buf_fmt >> 4;
297
298         switch (fmt) {
299         case SRP_NO_DATA_DESC:
300                 break;
301         case SRP_DATA_DESC_DIRECT:
302                 size = sizeof(struct srp_direct_buf);
303                 break;
304         case SRP_DATA_DESC_INDIRECT:
305                 size = sizeof(struct srp_indirect_buf) +
306                         sizeof(struct srp_direct_buf) * cmd->data_out_desc_cnt;
307                 break;
308         default:
309                 eprintk("client error. Invalid data_out_format %x\n", fmt);
310                 break;
311         }
312         return size;
313 }
314
315 /*
316  * TODO: this can be called multiple times for a single command if it
317  * has very long data.
318  */
319 int srp_transfer_data(struct scsi_cmnd *sc, struct srp_cmd *cmd,
320                       srp_rdma_t rdma_io, int dma_map, int ext_desc)
321 {
322         struct srp_direct_buf *md;
323         struct srp_indirect_buf *id;
324         enum dma_data_direction dir;
325         int offset, err = 0;
326         u8 format;
327
328         offset = cmd->add_cdb_len * 4;
329
330         dir = srp_cmd_direction(cmd);
331         if (dir == DMA_FROM_DEVICE)
332                 offset += data_out_desc_size(cmd);
333
334         if (dir == DMA_TO_DEVICE)
335                 format = cmd->buf_fmt >> 4;
336         else
337                 format = cmd->buf_fmt & ((1U << 4) - 1);
338
339         switch (format) {
340         case SRP_NO_DATA_DESC:
341                 break;
342         case SRP_DATA_DESC_DIRECT:
343                 md = (struct srp_direct_buf *)
344                         (cmd->add_data + offset);
345                 err = srp_direct_data(sc, md, dir, rdma_io, dma_map, ext_desc);
346                 break;
347         case SRP_DATA_DESC_INDIRECT:
348                 id = (struct srp_indirect_buf *)
349                         (cmd->add_data + offset);
350                 err = srp_indirect_data(sc, cmd, id, dir, rdma_io, dma_map,
351                                         ext_desc);
352                 break;
353         default:
354                 eprintk("Unknown format %d %x\n", dir, format);
355                 break;
356         }
357
358         return err;
359 }
360 EXPORT_SYMBOL_GPL(srp_transfer_data);
361
362 static int vscsis_data_length(struct srp_cmd *cmd, enum dma_data_direction dir)
363 {
364         struct srp_direct_buf *md;
365         struct srp_indirect_buf *id;
366         int len = 0, offset = cmd->add_cdb_len * 4;
367         u8 fmt;
368
369         if (dir == DMA_TO_DEVICE)
370                 fmt = cmd->buf_fmt >> 4;
371         else {
372                 fmt = cmd->buf_fmt & ((1U << 4) - 1);
373                 offset += data_out_desc_size(cmd);
374         }
375
376         switch (fmt) {
377         case SRP_NO_DATA_DESC:
378                 break;
379         case SRP_DATA_DESC_DIRECT:
380                 md = (struct srp_direct_buf *) (cmd->add_data + offset);
381                 len = md->len;
382                 break;
383         case SRP_DATA_DESC_INDIRECT:
384                 id = (struct srp_indirect_buf *) (cmd->add_data + offset);
385                 len = id->len;
386                 break;
387         default:
388                 eprintk("invalid data format %x\n", fmt);
389                 break;
390         }
391         return len;
392 }
393
394 int srp_cmd_queue(struct Scsi_Host *shost, struct srp_cmd *cmd, void *info,
395                   u64 addr)
396 {
397         enum dma_data_direction dir;
398         struct scsi_cmnd *sc;
399         int tag, len, err;
400
401         switch (cmd->task_attr) {
402         case SRP_SIMPLE_TASK:
403                 tag = MSG_SIMPLE_TAG;
404                 break;
405         case SRP_ORDERED_TASK:
406                 tag = MSG_ORDERED_TAG;
407                 break;
408         case SRP_HEAD_TASK:
409                 tag = MSG_HEAD_TAG;
410                 break;
411         default:
412                 eprintk("Task attribute %d not supported\n", cmd->task_attr);
413                 tag = MSG_ORDERED_TAG;
414         }
415
416         dir = srp_cmd_direction(cmd);
417         len = vscsis_data_length(cmd, dir);
418
419         dprintk("%p %x %lx %d %d %d %llx\n", info, cmd->cdb[0],
420                 cmd->lun, dir, len, tag, (unsigned long long) cmd->tag);
421
422         sc = scsi_host_get_command(shost, dir, GFP_KERNEL);
423         if (!sc)
424                 return -ENOMEM;
425
426         sc->SCp.ptr = info;
427         memcpy(sc->cmnd, cmd->cdb, MAX_COMMAND_SIZE);
428         sc->request_bufflen = len;
429         sc->request_buffer = (void *) (unsigned long) addr;
430         sc->tag = tag;
431         err = scsi_tgt_queue_command(sc, (struct scsi_lun *) &cmd->lun, cmd->tag);
432         if (err)
433                 scsi_host_put_command(shost, sc);
434
435         return err;
436 }
437 EXPORT_SYMBOL_GPL(srp_cmd_queue);
438
439 MODULE_DESCRIPTION("SCSI RDAM Protocol lib functions");
440 MODULE_AUTHOR("FUJITA Tomonori");
441 MODULE_LICENSE("GPL");