]> pilppa.org Git - linux-2.6-omap-h63xx.git/blob - drivers/firewire/fw-cdev.c
firewire: cdev: extend transaction payload size check
[linux-2.6-omap-h63xx.git] / drivers / firewire / fw-cdev.c
1 /*
2  * Char device for device raw access
3  *
4  * Copyright (C) 2005-2007  Kristian Hoegsberg <krh@bitplanet.net>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU 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 Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19  */
20
21 #include <linux/compat.h>
22 #include <linux/delay.h>
23 #include <linux/device.h>
24 #include <linux/errno.h>
25 #include <linux/firewire-cdev.h>
26 #include <linux/idr.h>
27 #include <linux/jiffies.h>
28 #include <linux/kernel.h>
29 #include <linux/kref.h>
30 #include <linux/mm.h>
31 #include <linux/module.h>
32 #include <linux/mutex.h>
33 #include <linux/poll.h>
34 #include <linux/preempt.h>
35 #include <linux/spinlock.h>
36 #include <linux/time.h>
37 #include <linux/vmalloc.h>
38 #include <linux/wait.h>
39 #include <linux/workqueue.h>
40
41 #include <asm/system.h>
42 #include <asm/uaccess.h>
43
44 #include "fw-device.h"
45 #include "fw-topology.h"
46 #include "fw-transaction.h"
47
48 struct client {
49         u32 version;
50         struct fw_device *device;
51
52         spinlock_t lock;
53         bool in_shutdown;
54         struct idr resource_idr;
55         struct list_head event_list;
56         wait_queue_head_t wait;
57         u64 bus_reset_closure;
58
59         struct fw_iso_context *iso_context;
60         u64 iso_closure;
61         struct fw_iso_buffer buffer;
62         unsigned long vm_start;
63
64         struct list_head link;
65         struct kref kref;
66 };
67
68 static inline void client_get(struct client *client)
69 {
70         kref_get(&client->kref);
71 }
72
73 static void client_release(struct kref *kref)
74 {
75         struct client *client = container_of(kref, struct client, kref);
76
77         fw_device_put(client->device);
78         kfree(client);
79 }
80
81 static void client_put(struct client *client)
82 {
83         kref_put(&client->kref, client_release);
84 }
85
86 struct client_resource;
87 typedef void (*client_resource_release_fn_t)(struct client *,
88                                              struct client_resource *);
89 struct client_resource {
90         client_resource_release_fn_t release;
91         int handle;
92 };
93
94 struct address_handler_resource {
95         struct client_resource resource;
96         struct fw_address_handler handler;
97         __u64 closure;
98         struct client *client;
99 };
100
101 struct outbound_transaction_resource {
102         struct client_resource resource;
103         struct fw_transaction transaction;
104 };
105
106 struct inbound_transaction_resource {
107         struct client_resource resource;
108         struct fw_request *request;
109         void *data;
110         size_t length;
111 };
112
113 struct descriptor_resource {
114         struct client_resource resource;
115         struct fw_descriptor descriptor;
116         u32 data[0];
117 };
118
119 struct iso_resource {
120         struct client_resource resource;
121         struct client *client;
122         /* Schedule work and access todo only with client->lock held. */
123         struct delayed_work work;
124         enum {ISO_RES_ALLOC, ISO_RES_REALLOC, ISO_RES_DEALLOC,
125               ISO_RES_ALLOC_ONCE, ISO_RES_DEALLOC_ONCE,} todo;
126         int generation;
127         u64 channels;
128         s32 bandwidth;
129         struct iso_resource_event *e_alloc, *e_dealloc;
130 };
131
132 static int schedule_iso_resource(struct iso_resource *);
133 static void release_iso_resource(struct client *, struct client_resource *);
134
135 /*
136  * dequeue_event() just kfree()'s the event, so the event has to be
137  * the first field in a struct XYZ_event.
138  */
139 struct event {
140         struct { void *data; size_t size; } v[2];
141         struct list_head link;
142 };
143
144 struct bus_reset_event {
145         struct event event;
146         struct fw_cdev_event_bus_reset reset;
147 };
148
149 struct outbound_transaction_event {
150         struct event event;
151         struct client *client;
152         struct outbound_transaction_resource r;
153         struct fw_cdev_event_response response;
154 };
155
156 struct inbound_transaction_event {
157         struct event event;
158         struct fw_cdev_event_request request;
159 };
160
161 struct iso_interrupt_event {
162         struct event event;
163         struct fw_cdev_event_iso_interrupt interrupt;
164 };
165
166 struct iso_resource_event {
167         struct event event;
168         struct fw_cdev_event_iso_resource resource;
169 };
170
171 static inline void __user *u64_to_uptr(__u64 value)
172 {
173         return (void __user *)(unsigned long)value;
174 }
175
176 static inline __u64 uptr_to_u64(void __user *ptr)
177 {
178         return (__u64)(unsigned long)ptr;
179 }
180
181 static int fw_device_op_open(struct inode *inode, struct file *file)
182 {
183         struct fw_device *device;
184         struct client *client;
185
186         device = fw_device_get_by_devt(inode->i_rdev);
187         if (device == NULL)
188                 return -ENODEV;
189
190         if (fw_device_is_shutdown(device)) {
191                 fw_device_put(device);
192                 return -ENODEV;
193         }
194
195         client = kzalloc(sizeof(*client), GFP_KERNEL);
196         if (client == NULL) {
197                 fw_device_put(device);
198                 return -ENOMEM;
199         }
200
201         client->device = device;
202         spin_lock_init(&client->lock);
203         idr_init(&client->resource_idr);
204         INIT_LIST_HEAD(&client->event_list);
205         init_waitqueue_head(&client->wait);
206         kref_init(&client->kref);
207
208         file->private_data = client;
209
210         mutex_lock(&device->client_list_mutex);
211         list_add_tail(&client->link, &device->client_list);
212         mutex_unlock(&device->client_list_mutex);
213
214         return 0;
215 }
216
217 static void queue_event(struct client *client, struct event *event,
218                         void *data0, size_t size0, void *data1, size_t size1)
219 {
220         unsigned long flags;
221
222         event->v[0].data = data0;
223         event->v[0].size = size0;
224         event->v[1].data = data1;
225         event->v[1].size = size1;
226
227         spin_lock_irqsave(&client->lock, flags);
228         if (client->in_shutdown)
229                 kfree(event);
230         else
231                 list_add_tail(&event->link, &client->event_list);
232         spin_unlock_irqrestore(&client->lock, flags);
233
234         wake_up_interruptible(&client->wait);
235 }
236
237 static int dequeue_event(struct client *client,
238                          char __user *buffer, size_t count)
239 {
240         unsigned long flags;
241         struct event *event;
242         size_t size, total;
243         int i, ret;
244
245         ret = wait_event_interruptible(client->wait,
246                         !list_empty(&client->event_list) ||
247                         fw_device_is_shutdown(client->device));
248         if (ret < 0)
249                 return ret;
250
251         if (list_empty(&client->event_list) &&
252                        fw_device_is_shutdown(client->device))
253                 return -ENODEV;
254
255         spin_lock_irqsave(&client->lock, flags);
256         event = list_first_entry(&client->event_list, struct event, link);
257         list_del(&event->link);
258         spin_unlock_irqrestore(&client->lock, flags);
259
260         total = 0;
261         for (i = 0; i < ARRAY_SIZE(event->v) && total < count; i++) {
262                 size = min(event->v[i].size, count - total);
263                 if (copy_to_user(buffer + total, event->v[i].data, size)) {
264                         ret = -EFAULT;
265                         goto out;
266                 }
267                 total += size;
268         }
269         ret = total;
270
271  out:
272         kfree(event);
273
274         return ret;
275 }
276
277 static ssize_t fw_device_op_read(struct file *file, char __user *buffer,
278                                  size_t count, loff_t *offset)
279 {
280         struct client *client = file->private_data;
281
282         return dequeue_event(client, buffer, count);
283 }
284
285 static void fill_bus_reset_event(struct fw_cdev_event_bus_reset *event,
286                                  struct client *client)
287 {
288         struct fw_card *card = client->device->card;
289         unsigned long flags;
290
291         spin_lock_irqsave(&card->lock, flags);
292
293         event->closure       = client->bus_reset_closure;
294         event->type          = FW_CDEV_EVENT_BUS_RESET;
295         event->generation    = client->device->generation;
296         event->node_id       = client->device->node_id;
297         event->local_node_id = card->local_node->node_id;
298         event->bm_node_id    = 0; /* FIXME: We don't track the BM. */
299         event->irm_node_id   = card->irm_node->node_id;
300         event->root_node_id  = card->root_node->node_id;
301
302         spin_unlock_irqrestore(&card->lock, flags);
303 }
304
305 static void for_each_client(struct fw_device *device,
306                             void (*callback)(struct client *client))
307 {
308         struct client *c;
309
310         mutex_lock(&device->client_list_mutex);
311         list_for_each_entry(c, &device->client_list, link)
312                 callback(c);
313         mutex_unlock(&device->client_list_mutex);
314 }
315
316 static int schedule_reallocations(int id, void *p, void *data)
317 {
318         struct client_resource *r = p;
319
320         if (r->release == release_iso_resource)
321                 schedule_iso_resource(container_of(r,
322                                         struct iso_resource, resource));
323         return 0;
324 }
325
326 static void queue_bus_reset_event(struct client *client)
327 {
328         struct bus_reset_event *e;
329
330         e = kzalloc(sizeof(*e), GFP_KERNEL);
331         if (e == NULL) {
332                 fw_notify("Out of memory when allocating bus reset event\n");
333                 return;
334         }
335
336         fill_bus_reset_event(&e->reset, client);
337
338         queue_event(client, &e->event,
339                     &e->reset, sizeof(e->reset), NULL, 0);
340
341         spin_lock_irq(&client->lock);
342         idr_for_each(&client->resource_idr, schedule_reallocations, client);
343         spin_unlock_irq(&client->lock);
344 }
345
346 void fw_device_cdev_update(struct fw_device *device)
347 {
348         for_each_client(device, queue_bus_reset_event);
349 }
350
351 static void wake_up_client(struct client *client)
352 {
353         wake_up_interruptible(&client->wait);
354 }
355
356 void fw_device_cdev_remove(struct fw_device *device)
357 {
358         for_each_client(device, wake_up_client);
359 }
360
361 static int ioctl_get_info(struct client *client, void *buffer)
362 {
363         struct fw_cdev_get_info *get_info = buffer;
364         struct fw_cdev_event_bus_reset bus_reset;
365         unsigned long ret = 0;
366
367         client->version = get_info->version;
368         get_info->version = FW_CDEV_VERSION;
369         get_info->card = client->device->card->index;
370
371         down_read(&fw_device_rwsem);
372
373         if (get_info->rom != 0) {
374                 void __user *uptr = u64_to_uptr(get_info->rom);
375                 size_t want = get_info->rom_length;
376                 size_t have = client->device->config_rom_length * 4;
377
378                 ret = copy_to_user(uptr, client->device->config_rom,
379                                    min(want, have));
380         }
381         get_info->rom_length = client->device->config_rom_length * 4;
382
383         up_read(&fw_device_rwsem);
384
385         if (ret != 0)
386                 return -EFAULT;
387
388         client->bus_reset_closure = get_info->bus_reset_closure;
389         if (get_info->bus_reset != 0) {
390                 void __user *uptr = u64_to_uptr(get_info->bus_reset);
391
392                 fill_bus_reset_event(&bus_reset, client);
393                 if (copy_to_user(uptr, &bus_reset, sizeof(bus_reset)))
394                         return -EFAULT;
395         }
396
397         return 0;
398 }
399
400 static int add_client_resource(struct client *client,
401                                struct client_resource *resource, gfp_t gfp_mask)
402 {
403         unsigned long flags;
404         int ret;
405
406  retry:
407         if (idr_pre_get(&client->resource_idr, gfp_mask) == 0)
408                 return -ENOMEM;
409
410         spin_lock_irqsave(&client->lock, flags);
411         if (client->in_shutdown)
412                 ret = -ECANCELED;
413         else
414                 ret = idr_get_new(&client->resource_idr, resource,
415                                   &resource->handle);
416         if (ret >= 0) {
417                 client_get(client);
418                 if (resource->release == release_iso_resource)
419                         schedule_iso_resource(container_of(resource,
420                                                 struct iso_resource, resource));
421         }
422         spin_unlock_irqrestore(&client->lock, flags);
423
424         if (ret == -EAGAIN)
425                 goto retry;
426
427         return ret < 0 ? ret : 0;
428 }
429
430 static int release_client_resource(struct client *client, u32 handle,
431                                    client_resource_release_fn_t release,
432                                    struct client_resource **resource)
433 {
434         struct client_resource *r;
435         unsigned long flags;
436
437         spin_lock_irqsave(&client->lock, flags);
438         if (client->in_shutdown)
439                 r = NULL;
440         else
441                 r = idr_find(&client->resource_idr, handle);
442         if (r && r->release == release)
443                 idr_remove(&client->resource_idr, handle);
444         spin_unlock_irqrestore(&client->lock, flags);
445
446         if (!(r && r->release == release))
447                 return -EINVAL;
448
449         if (resource)
450                 *resource = r;
451         else
452                 r->release(client, r);
453
454         client_put(client);
455
456         return 0;
457 }
458
459 static void release_transaction(struct client *client,
460                                 struct client_resource *resource)
461 {
462         struct outbound_transaction_resource *r = container_of(resource,
463                         struct outbound_transaction_resource, resource);
464
465         fw_cancel_transaction(client->device->card, &r->transaction);
466 }
467
468 static void complete_transaction(struct fw_card *card, int rcode,
469                                  void *payload, size_t length, void *data)
470 {
471         struct outbound_transaction_event *e = data;
472         struct fw_cdev_event_response *rsp = &e->response;
473         struct client *client = e->client;
474         unsigned long flags;
475
476         if (length < rsp->length)
477                 rsp->length = length;
478         if (rcode == RCODE_COMPLETE)
479                 memcpy(rsp->data, payload, rsp->length);
480
481         spin_lock_irqsave(&client->lock, flags);
482         /*
483          * 1. If called while in shutdown, the idr tree must be left untouched.
484          *    The idr handle will be removed and the client reference will be
485          *    dropped later.
486          * 2. If the call chain was release_client_resource ->
487          *    release_transaction -> complete_transaction (instead of a normal
488          *    conclusion of the transaction), i.e. if this resource was already
489          *    unregistered from the idr, the client reference will be dropped
490          *    by release_client_resource and we must not drop it here.
491          */
492         if (!client->in_shutdown &&
493             idr_find(&client->resource_idr, e->r.resource.handle)) {
494                 idr_remove(&client->resource_idr, e->r.resource.handle);
495                 /* Drop the idr's reference */
496                 client_put(client);
497         }
498         spin_unlock_irqrestore(&client->lock, flags);
499
500         rsp->type = FW_CDEV_EVENT_RESPONSE;
501         rsp->rcode = rcode;
502
503         /*
504          * In the case that sizeof(*rsp) doesn't align with the position of the
505          * data, and the read is short, preserve an extra copy of the data
506          * to stay compatible with a pre-2.6.27 bug.  Since the bug is harmless
507          * for short reads and some apps depended on it, this is both safe
508          * and prudent for compatibility.
509          */
510         if (rsp->length <= sizeof(*rsp) - offsetof(typeof(*rsp), data))
511                 queue_event(client, &e->event, rsp, sizeof(*rsp),
512                             rsp->data, rsp->length);
513         else
514                 queue_event(client, &e->event, rsp, sizeof(*rsp) + rsp->length,
515                             NULL, 0);
516
517         /* Drop the transaction callback's reference */
518         client_put(client);
519 }
520
521 static int init_request(struct client *client,
522                         struct fw_cdev_send_request *request,
523                         int destination_id, int speed)
524 {
525         struct outbound_transaction_event *e;
526         int ret;
527
528         if (request->length > 4096 || request->length > 512 << speed)
529                 return -EIO;
530
531         e = kmalloc(sizeof(*e) + request->length, GFP_KERNEL);
532         if (e == NULL)
533                 return -ENOMEM;
534
535         e->client = client;
536         e->response.length = request->length;
537         e->response.closure = request->closure;
538
539         if (request->data &&
540             copy_from_user(e->response.data,
541                            u64_to_uptr(request->data), request->length)) {
542                 ret = -EFAULT;
543                 goto failed;
544         }
545
546         e->r.resource.release = release_transaction;
547         ret = add_client_resource(client, &e->r.resource, GFP_KERNEL);
548         if (ret < 0)
549                 goto failed;
550
551         /* Get a reference for the transaction callback */
552         client_get(client);
553
554         fw_send_request(client->device->card, &e->r.transaction,
555                         request->tcode & 0x1f, destination_id,
556                         request->generation, speed, request->offset,
557                         e->response.data, request->length,
558                         complete_transaction, e);
559
560         if (request->data)
561                 return sizeof(request) + request->length;
562         else
563                 return sizeof(request);
564  failed:
565         kfree(e);
566
567         return ret;
568 }
569
570 static int ioctl_send_request(struct client *client, void *buffer)
571 {
572         struct fw_cdev_send_request *request = buffer;
573
574         switch (request->tcode) {
575         case TCODE_WRITE_QUADLET_REQUEST:
576         case TCODE_WRITE_BLOCK_REQUEST:
577         case TCODE_READ_QUADLET_REQUEST:
578         case TCODE_READ_BLOCK_REQUEST:
579         case TCODE_LOCK_MASK_SWAP:
580         case TCODE_LOCK_COMPARE_SWAP:
581         case TCODE_LOCK_FETCH_ADD:
582         case TCODE_LOCK_LITTLE_ADD:
583         case TCODE_LOCK_BOUNDED_ADD:
584         case TCODE_LOCK_WRAP_ADD:
585         case TCODE_LOCK_VENDOR_DEPENDENT:
586                 break;
587         default:
588                 return -EINVAL;
589         }
590
591         return init_request(client, request, client->device->node->node_id,
592                             client->device->max_speed);
593 }
594
595 static void release_request(struct client *client,
596                             struct client_resource *resource)
597 {
598         struct inbound_transaction_resource *r = container_of(resource,
599                         struct inbound_transaction_resource, resource);
600
601         fw_send_response(client->device->card, r->request,
602                          RCODE_CONFLICT_ERROR);
603         kfree(r);
604 }
605
606 static void handle_request(struct fw_card *card, struct fw_request *request,
607                            int tcode, int destination, int source,
608                            int generation, int speed,
609                            unsigned long long offset,
610                            void *payload, size_t length, void *callback_data)
611 {
612         struct address_handler_resource *handler = callback_data;
613         struct inbound_transaction_resource *r;
614         struct inbound_transaction_event *e;
615         int ret;
616
617         r = kmalloc(sizeof(*r), GFP_ATOMIC);
618         e = kmalloc(sizeof(*e), GFP_ATOMIC);
619         if (r == NULL || e == NULL)
620                 goto failed;
621
622         r->request = request;
623         r->data    = payload;
624         r->length  = length;
625
626         r->resource.release = release_request;
627         ret = add_client_resource(handler->client, &r->resource, GFP_ATOMIC);
628         if (ret < 0)
629                 goto failed;
630
631         e->request.type    = FW_CDEV_EVENT_REQUEST;
632         e->request.tcode   = tcode;
633         e->request.offset  = offset;
634         e->request.length  = length;
635         e->request.handle  = r->resource.handle;
636         e->request.closure = handler->closure;
637
638         queue_event(handler->client, &e->event,
639                     &e->request, sizeof(e->request), payload, length);
640         return;
641
642  failed:
643         kfree(r);
644         kfree(e);
645         fw_send_response(card, request, RCODE_CONFLICT_ERROR);
646 }
647
648 static void release_address_handler(struct client *client,
649                                     struct client_resource *resource)
650 {
651         struct address_handler_resource *r =
652             container_of(resource, struct address_handler_resource, resource);
653
654         fw_core_remove_address_handler(&r->handler);
655         kfree(r);
656 }
657
658 static int ioctl_allocate(struct client *client, void *buffer)
659 {
660         struct fw_cdev_allocate *request = buffer;
661         struct address_handler_resource *r;
662         struct fw_address_region region;
663         int ret;
664
665         r = kmalloc(sizeof(*r), GFP_KERNEL);
666         if (r == NULL)
667                 return -ENOMEM;
668
669         region.start = request->offset;
670         region.end = request->offset + request->length;
671         r->handler.length = request->length;
672         r->handler.address_callback = handle_request;
673         r->handler.callback_data = r;
674         r->closure = request->closure;
675         r->client = client;
676
677         ret = fw_core_add_address_handler(&r->handler, &region);
678         if (ret < 0) {
679                 kfree(r);
680                 return ret;
681         }
682
683         r->resource.release = release_address_handler;
684         ret = add_client_resource(client, &r->resource, GFP_KERNEL);
685         if (ret < 0) {
686                 release_address_handler(client, &r->resource);
687                 return ret;
688         }
689         request->handle = r->resource.handle;
690
691         return 0;
692 }
693
694 static int ioctl_deallocate(struct client *client, void *buffer)
695 {
696         struct fw_cdev_deallocate *request = buffer;
697
698         return release_client_resource(client, request->handle,
699                                        release_address_handler, NULL);
700 }
701
702 static int ioctl_send_response(struct client *client, void *buffer)
703 {
704         struct fw_cdev_send_response *request = buffer;
705         struct client_resource *resource;
706         struct inbound_transaction_resource *r;
707
708         if (release_client_resource(client, request->handle,
709                                     release_request, &resource) < 0)
710                 return -EINVAL;
711
712         r = container_of(resource, struct inbound_transaction_resource,
713                          resource);
714         if (request->length < r->length)
715                 r->length = request->length;
716         if (copy_from_user(r->data, u64_to_uptr(request->data), r->length))
717                 return -EFAULT;
718
719         fw_send_response(client->device->card, r->request, request->rcode);
720         kfree(r);
721
722         return 0;
723 }
724
725 static int ioctl_initiate_bus_reset(struct client *client, void *buffer)
726 {
727         struct fw_cdev_initiate_bus_reset *request = buffer;
728         int short_reset;
729
730         short_reset = (request->type == FW_CDEV_SHORT_RESET);
731
732         return fw_core_initiate_bus_reset(client->device->card, short_reset);
733 }
734
735 static void release_descriptor(struct client *client,
736                                struct client_resource *resource)
737 {
738         struct descriptor_resource *r =
739                 container_of(resource, struct descriptor_resource, resource);
740
741         fw_core_remove_descriptor(&r->descriptor);
742         kfree(r);
743 }
744
745 static int ioctl_add_descriptor(struct client *client, void *buffer)
746 {
747         struct fw_cdev_add_descriptor *request = buffer;
748         struct descriptor_resource *r;
749         int ret;
750
751         if (request->length > 256)
752                 return -EINVAL;
753
754         r = kmalloc(sizeof(*r) + request->length * 4, GFP_KERNEL);
755         if (r == NULL)
756                 return -ENOMEM;
757
758         if (copy_from_user(r->data,
759                            u64_to_uptr(request->data), request->length * 4)) {
760                 ret = -EFAULT;
761                 goto failed;
762         }
763
764         r->descriptor.length    = request->length;
765         r->descriptor.immediate = request->immediate;
766         r->descriptor.key       = request->key;
767         r->descriptor.data      = r->data;
768
769         ret = fw_core_add_descriptor(&r->descriptor);
770         if (ret < 0)
771                 goto failed;
772
773         r->resource.release = release_descriptor;
774         ret = add_client_resource(client, &r->resource, GFP_KERNEL);
775         if (ret < 0) {
776                 fw_core_remove_descriptor(&r->descriptor);
777                 goto failed;
778         }
779         request->handle = r->resource.handle;
780
781         return 0;
782  failed:
783         kfree(r);
784
785         return ret;
786 }
787
788 static int ioctl_remove_descriptor(struct client *client, void *buffer)
789 {
790         struct fw_cdev_remove_descriptor *request = buffer;
791
792         return release_client_resource(client, request->handle,
793                                        release_descriptor, NULL);
794 }
795
796 static void iso_callback(struct fw_iso_context *context, u32 cycle,
797                          size_t header_length, void *header, void *data)
798 {
799         struct client *client = data;
800         struct iso_interrupt_event *e;
801
802         e = kzalloc(sizeof(*e) + header_length, GFP_ATOMIC);
803         if (e == NULL)
804                 return;
805
806         e->interrupt.type      = FW_CDEV_EVENT_ISO_INTERRUPT;
807         e->interrupt.closure   = client->iso_closure;
808         e->interrupt.cycle     = cycle;
809         e->interrupt.header_length = header_length;
810         memcpy(e->interrupt.header, header, header_length);
811         queue_event(client, &e->event, &e->interrupt,
812                     sizeof(e->interrupt) + header_length, NULL, 0);
813 }
814
815 static int ioctl_create_iso_context(struct client *client, void *buffer)
816 {
817         struct fw_cdev_create_iso_context *request = buffer;
818         struct fw_iso_context *context;
819
820         /* We only support one context at this time. */
821         if (client->iso_context != NULL)
822                 return -EBUSY;
823
824         if (request->channel > 63)
825                 return -EINVAL;
826
827         switch (request->type) {
828         case FW_ISO_CONTEXT_RECEIVE:
829                 if (request->header_size < 4 || (request->header_size & 3))
830                         return -EINVAL;
831
832                 break;
833
834         case FW_ISO_CONTEXT_TRANSMIT:
835                 if (request->speed > SCODE_3200)
836                         return -EINVAL;
837
838                 break;
839
840         default:
841                 return -EINVAL;
842         }
843
844         context =  fw_iso_context_create(client->device->card,
845                                          request->type,
846                                          request->channel,
847                                          request->speed,
848                                          request->header_size,
849                                          iso_callback, client);
850         if (IS_ERR(context))
851                 return PTR_ERR(context);
852
853         client->iso_closure = request->closure;
854         client->iso_context = context;
855
856         /* We only support one context at this time. */
857         request->handle = 0;
858
859         return 0;
860 }
861
862 /* Macros for decoding the iso packet control header. */
863 #define GET_PAYLOAD_LENGTH(v)   ((v) & 0xffff)
864 #define GET_INTERRUPT(v)        (((v) >> 16) & 0x01)
865 #define GET_SKIP(v)             (((v) >> 17) & 0x01)
866 #define GET_TAG(v)              (((v) >> 18) & 0x03)
867 #define GET_SY(v)               (((v) >> 20) & 0x0f)
868 #define GET_HEADER_LENGTH(v)    (((v) >> 24) & 0xff)
869
870 static int ioctl_queue_iso(struct client *client, void *buffer)
871 {
872         struct fw_cdev_queue_iso *request = buffer;
873         struct fw_cdev_iso_packet __user *p, *end, *next;
874         struct fw_iso_context *ctx = client->iso_context;
875         unsigned long payload, buffer_end, header_length;
876         u32 control;
877         int count;
878         struct {
879                 struct fw_iso_packet packet;
880                 u8 header[256];
881         } u;
882
883         if (ctx == NULL || request->handle != 0)
884                 return -EINVAL;
885
886         /*
887          * If the user passes a non-NULL data pointer, has mmap()'ed
888          * the iso buffer, and the pointer points inside the buffer,
889          * we setup the payload pointers accordingly.  Otherwise we
890          * set them both to 0, which will still let packets with
891          * payload_length == 0 through.  In other words, if no packets
892          * use the indirect payload, the iso buffer need not be mapped
893          * and the request->data pointer is ignored.
894          */
895
896         payload = (unsigned long)request->data - client->vm_start;
897         buffer_end = client->buffer.page_count << PAGE_SHIFT;
898         if (request->data == 0 || client->buffer.pages == NULL ||
899             payload >= buffer_end) {
900                 payload = 0;
901                 buffer_end = 0;
902         }
903
904         p = (struct fw_cdev_iso_packet __user *)u64_to_uptr(request->packets);
905
906         if (!access_ok(VERIFY_READ, p, request->size))
907                 return -EFAULT;
908
909         end = (void __user *)p + request->size;
910         count = 0;
911         while (p < end) {
912                 if (get_user(control, &p->control))
913                         return -EFAULT;
914                 u.packet.payload_length = GET_PAYLOAD_LENGTH(control);
915                 u.packet.interrupt = GET_INTERRUPT(control);
916                 u.packet.skip = GET_SKIP(control);
917                 u.packet.tag = GET_TAG(control);
918                 u.packet.sy = GET_SY(control);
919                 u.packet.header_length = GET_HEADER_LENGTH(control);
920
921                 if (ctx->type == FW_ISO_CONTEXT_TRANSMIT) {
922                         header_length = u.packet.header_length;
923                 } else {
924                         /*
925                          * We require that header_length is a multiple of
926                          * the fixed header size, ctx->header_size.
927                          */
928                         if (ctx->header_size == 0) {
929                                 if (u.packet.header_length > 0)
930                                         return -EINVAL;
931                         } else if (u.packet.header_length % ctx->header_size != 0) {
932                                 return -EINVAL;
933                         }
934                         header_length = 0;
935                 }
936
937                 next = (struct fw_cdev_iso_packet __user *)
938                         &p->header[header_length / 4];
939                 if (next > end)
940                         return -EINVAL;
941                 if (__copy_from_user
942                     (u.packet.header, p->header, header_length))
943                         return -EFAULT;
944                 if (u.packet.skip && ctx->type == FW_ISO_CONTEXT_TRANSMIT &&
945                     u.packet.header_length + u.packet.payload_length > 0)
946                         return -EINVAL;
947                 if (payload + u.packet.payload_length > buffer_end)
948                         return -EINVAL;
949
950                 if (fw_iso_context_queue(ctx, &u.packet,
951                                          &client->buffer, payload))
952                         break;
953
954                 p = next;
955                 payload += u.packet.payload_length;
956                 count++;
957         }
958
959         request->size    -= uptr_to_u64(p) - request->packets;
960         request->packets  = uptr_to_u64(p);
961         request->data     = client->vm_start + payload;
962
963         return count;
964 }
965
966 static int ioctl_start_iso(struct client *client, void *buffer)
967 {
968         struct fw_cdev_start_iso *request = buffer;
969
970         if (client->iso_context == NULL || request->handle != 0)
971                 return -EINVAL;
972
973         if (client->iso_context->type == FW_ISO_CONTEXT_RECEIVE) {
974                 if (request->tags == 0 || request->tags > 15)
975                         return -EINVAL;
976
977                 if (request->sync > 15)
978                         return -EINVAL;
979         }
980
981         return fw_iso_context_start(client->iso_context, request->cycle,
982                                     request->sync, request->tags);
983 }
984
985 static int ioctl_stop_iso(struct client *client, void *buffer)
986 {
987         struct fw_cdev_stop_iso *request = buffer;
988
989         if (client->iso_context == NULL || request->handle != 0)
990                 return -EINVAL;
991
992         return fw_iso_context_stop(client->iso_context);
993 }
994
995 static int ioctl_get_cycle_timer(struct client *client, void *buffer)
996 {
997         struct fw_cdev_get_cycle_timer *request = buffer;
998         struct fw_card *card = client->device->card;
999         unsigned long long bus_time;
1000         struct timeval tv;
1001         unsigned long flags;
1002
1003         preempt_disable();
1004         local_irq_save(flags);
1005
1006         bus_time = card->driver->get_bus_time(card);
1007         do_gettimeofday(&tv);
1008
1009         local_irq_restore(flags);
1010         preempt_enable();
1011
1012         request->local_time = tv.tv_sec * 1000000ULL + tv.tv_usec;
1013         request->cycle_timer = bus_time & 0xffffffff;
1014         return 0;
1015 }
1016
1017 static void iso_resource_work(struct work_struct *work)
1018 {
1019         struct iso_resource_event *e;
1020         struct iso_resource *r =
1021                         container_of(work, struct iso_resource, work.work);
1022         struct client *client = r->client;
1023         int generation, channel, bandwidth, todo;
1024         bool skip, free, success;
1025
1026         spin_lock_irq(&client->lock);
1027         generation = client->device->generation;
1028         todo = r->todo;
1029         /* Allow 1000ms grace period for other reallocations. */
1030         if (todo == ISO_RES_ALLOC &&
1031             time_is_after_jiffies(client->device->card->reset_jiffies + HZ)) {
1032                 if (schedule_delayed_work(&r->work, DIV_ROUND_UP(HZ, 3)))
1033                         client_get(client);
1034                 skip = true;
1035         } else {
1036                 /* We could be called twice within the same generation. */
1037                 skip = todo == ISO_RES_REALLOC &&
1038                        r->generation == generation;
1039         }
1040         free = todo == ISO_RES_DEALLOC ||
1041                todo == ISO_RES_ALLOC_ONCE ||
1042                todo == ISO_RES_DEALLOC_ONCE;
1043         r->generation = generation;
1044         spin_unlock_irq(&client->lock);
1045
1046         if (skip)
1047                 goto out;
1048
1049         bandwidth = r->bandwidth;
1050
1051         fw_iso_resource_manage(client->device->card, generation,
1052                         r->channels, &channel, &bandwidth,
1053                         todo == ISO_RES_ALLOC ||
1054                         todo == ISO_RES_REALLOC ||
1055                         todo == ISO_RES_ALLOC_ONCE);
1056         /*
1057          * Is this generation outdated already?  As long as this resource sticks
1058          * in the idr, it will be scheduled again for a newer generation or at
1059          * shutdown.
1060          */
1061         if (channel == -EAGAIN &&
1062             (todo == ISO_RES_ALLOC || todo == ISO_RES_REALLOC))
1063                 goto out;
1064
1065         success = channel >= 0 || bandwidth > 0;
1066
1067         spin_lock_irq(&client->lock);
1068         /*
1069          * Transit from allocation to reallocation, except if the client
1070          * requested deallocation in the meantime.
1071          */
1072         if (r->todo == ISO_RES_ALLOC)
1073                 r->todo = ISO_RES_REALLOC;
1074         /*
1075          * Allocation or reallocation failure?  Pull this resource out of the
1076          * idr and prepare for deletion, unless the client is shutting down.
1077          */
1078         if (r->todo == ISO_RES_REALLOC && !success &&
1079             !client->in_shutdown &&
1080             idr_find(&client->resource_idr, r->resource.handle)) {
1081                 idr_remove(&client->resource_idr, r->resource.handle);
1082                 client_put(client);
1083                 free = true;
1084         }
1085         spin_unlock_irq(&client->lock);
1086
1087         if (todo == ISO_RES_ALLOC && channel >= 0)
1088                 r->channels = 1ULL << (63 - channel);
1089
1090         if (todo == ISO_RES_REALLOC && success)
1091                 goto out;
1092
1093         if (todo == ISO_RES_ALLOC || todo == ISO_RES_ALLOC_ONCE) {
1094                 e = r->e_alloc;
1095                 r->e_alloc = NULL;
1096         } else {
1097                 e = r->e_dealloc;
1098                 r->e_dealloc = NULL;
1099         }
1100         e->resource.handle      = r->resource.handle;
1101         e->resource.channel     = channel;
1102         e->resource.bandwidth   = bandwidth;
1103
1104         queue_event(client, &e->event,
1105                     &e->resource, sizeof(e->resource), NULL, 0);
1106
1107         if (free) {
1108                 cancel_delayed_work(&r->work);
1109                 kfree(r->e_alloc);
1110                 kfree(r->e_dealloc);
1111                 kfree(r);
1112         }
1113  out:
1114         client_put(client);
1115 }
1116
1117 static int schedule_iso_resource(struct iso_resource *r)
1118 {
1119         int scheduled;
1120
1121         client_get(r->client);
1122
1123         scheduled = schedule_delayed_work(&r->work, 0);
1124         if (!scheduled)
1125                 client_put(r->client);
1126
1127         return scheduled;
1128 }
1129
1130 static void release_iso_resource(struct client *client,
1131                                  struct client_resource *resource)
1132 {
1133         struct iso_resource *r =
1134                 container_of(resource, struct iso_resource, resource);
1135
1136         spin_lock_irq(&client->lock);
1137         r->todo = ISO_RES_DEALLOC;
1138         schedule_iso_resource(r);
1139         spin_unlock_irq(&client->lock);
1140 }
1141
1142 static int init_iso_resource(struct client *client,
1143                 struct fw_cdev_allocate_iso_resource *request, int todo)
1144 {
1145         struct iso_resource_event *e1, *e2;
1146         struct iso_resource *r;
1147         int ret;
1148
1149         if ((request->channels == 0 && request->bandwidth == 0) ||
1150             request->bandwidth > BANDWIDTH_AVAILABLE_INITIAL ||
1151             request->bandwidth < 0)
1152                 return -EINVAL;
1153
1154         r  = kmalloc(sizeof(*r), GFP_KERNEL);
1155         e1 = kmalloc(sizeof(*e1), GFP_KERNEL);
1156         e2 = kmalloc(sizeof(*e2), GFP_KERNEL);
1157         if (r == NULL || e1 == NULL || e2 == NULL) {
1158                 ret = -ENOMEM;
1159                 goto fail;
1160         }
1161
1162         INIT_DELAYED_WORK(&r->work, iso_resource_work);
1163         r->client       = client;
1164         r->todo         = todo;
1165         r->generation   = -1;
1166         r->channels     = request->channels;
1167         r->bandwidth    = request->bandwidth;
1168         r->e_alloc      = e1;
1169         r->e_dealloc    = e2;
1170
1171         e1->resource.closure    = request->closure;
1172         e1->resource.type       = FW_CDEV_EVENT_ISO_RESOURCE_ALLOCATED;
1173         e2->resource.closure    = request->closure;
1174         e2->resource.type       = FW_CDEV_EVENT_ISO_RESOURCE_DEALLOCATED;
1175
1176         if (todo == ISO_RES_ALLOC) {
1177                 r->resource.release = release_iso_resource;
1178                 ret = add_client_resource(client, &r->resource, GFP_KERNEL);
1179         } else {
1180                 r->resource.release = NULL;
1181                 r->resource.handle = -1;
1182                 ret = schedule_iso_resource(r) ? 0 : -ENOMEM;
1183         }
1184         if (ret < 0)
1185                 goto fail;
1186         request->handle = r->resource.handle;
1187
1188         return 0;
1189  fail:
1190         kfree(r);
1191         kfree(e1);
1192         kfree(e2);
1193
1194         return ret;
1195 }
1196
1197 static int ioctl_allocate_iso_resource(struct client *client, void *buffer)
1198 {
1199         struct fw_cdev_allocate_iso_resource *request = buffer;
1200
1201         return init_iso_resource(client, request, ISO_RES_ALLOC);
1202 }
1203
1204 static int ioctl_deallocate_iso_resource(struct client *client, void *buffer)
1205 {
1206         struct fw_cdev_deallocate *request = buffer;
1207
1208         return release_client_resource(client, request->handle,
1209                                        release_iso_resource, NULL);
1210 }
1211
1212 static int ioctl_allocate_iso_resource_once(struct client *client, void *buffer)
1213 {
1214         struct fw_cdev_allocate_iso_resource *request = buffer;
1215
1216         return init_iso_resource(client, request, ISO_RES_ALLOC_ONCE);
1217 }
1218
1219 static int ioctl_deallocate_iso_resource_once(struct client *client, void *buffer)
1220 {
1221         struct fw_cdev_allocate_iso_resource *request = buffer;
1222
1223         return init_iso_resource(client, request, ISO_RES_DEALLOC_ONCE);
1224 }
1225
1226 static int ioctl_get_speed(struct client *client, void *buffer)
1227 {
1228         struct fw_cdev_get_speed *request = buffer;
1229
1230         request->max_speed = client->device->max_speed;
1231
1232         return 0;
1233 }
1234
1235 static int ioctl_send_broadcast_request(struct client *client, void *buffer)
1236 {
1237         struct fw_cdev_send_request *request = buffer;
1238
1239         switch (request->tcode) {
1240         case TCODE_WRITE_QUADLET_REQUEST:
1241         case TCODE_WRITE_BLOCK_REQUEST:
1242                 break;
1243         default:
1244                 return -EINVAL;
1245         }
1246
1247         /* Security policy: Only allow accesses to Units Space. */
1248         if (request->offset < CSR_REGISTER_BASE + CSR_CONFIG_ROM_END)
1249                 return -EACCES;
1250
1251         return init_request(client, request, LOCAL_BUS | 0x3f, SCODE_100);
1252 }
1253
1254 static int (* const ioctl_handlers[])(struct client *client, void *buffer) = {
1255         ioctl_get_info,
1256         ioctl_send_request,
1257         ioctl_allocate,
1258         ioctl_deallocate,
1259         ioctl_send_response,
1260         ioctl_initiate_bus_reset,
1261         ioctl_add_descriptor,
1262         ioctl_remove_descriptor,
1263         ioctl_create_iso_context,
1264         ioctl_queue_iso,
1265         ioctl_start_iso,
1266         ioctl_stop_iso,
1267         ioctl_get_cycle_timer,
1268         ioctl_allocate_iso_resource,
1269         ioctl_deallocate_iso_resource,
1270         ioctl_allocate_iso_resource_once,
1271         ioctl_deallocate_iso_resource_once,
1272         ioctl_get_speed,
1273         ioctl_send_broadcast_request,
1274 };
1275
1276 static int dispatch_ioctl(struct client *client,
1277                           unsigned int cmd, void __user *arg)
1278 {
1279         char buffer[256];
1280         int ret;
1281
1282         if (_IOC_TYPE(cmd) != '#' ||
1283             _IOC_NR(cmd) >= ARRAY_SIZE(ioctl_handlers))
1284                 return -EINVAL;
1285
1286         if (_IOC_DIR(cmd) & _IOC_WRITE) {
1287                 if (_IOC_SIZE(cmd) > sizeof(buffer) ||
1288                     copy_from_user(buffer, arg, _IOC_SIZE(cmd)))
1289                         return -EFAULT;
1290         }
1291
1292         ret = ioctl_handlers[_IOC_NR(cmd)](client, buffer);
1293         if (ret < 0)
1294                 return ret;
1295
1296         if (_IOC_DIR(cmd) & _IOC_READ) {
1297                 if (_IOC_SIZE(cmd) > sizeof(buffer) ||
1298                     copy_to_user(arg, buffer, _IOC_SIZE(cmd)))
1299                         return -EFAULT;
1300         }
1301
1302         return ret;
1303 }
1304
1305 static long fw_device_op_ioctl(struct file *file,
1306                                unsigned int cmd, unsigned long arg)
1307 {
1308         struct client *client = file->private_data;
1309
1310         if (fw_device_is_shutdown(client->device))
1311                 return -ENODEV;
1312
1313         return dispatch_ioctl(client, cmd, (void __user *) arg);
1314 }
1315
1316 #ifdef CONFIG_COMPAT
1317 static long fw_device_op_compat_ioctl(struct file *file,
1318                                       unsigned int cmd, unsigned long arg)
1319 {
1320         struct client *client = file->private_data;
1321
1322         if (fw_device_is_shutdown(client->device))
1323                 return -ENODEV;
1324
1325         return dispatch_ioctl(client, cmd, compat_ptr(arg));
1326 }
1327 #endif
1328
1329 static int fw_device_op_mmap(struct file *file, struct vm_area_struct *vma)
1330 {
1331         struct client *client = file->private_data;
1332         enum dma_data_direction direction;
1333         unsigned long size;
1334         int page_count, ret;
1335
1336         if (fw_device_is_shutdown(client->device))
1337                 return -ENODEV;
1338
1339         /* FIXME: We could support multiple buffers, but we don't. */
1340         if (client->buffer.pages != NULL)
1341                 return -EBUSY;
1342
1343         if (!(vma->vm_flags & VM_SHARED))
1344                 return -EINVAL;
1345
1346         if (vma->vm_start & ~PAGE_MASK)
1347                 return -EINVAL;
1348
1349         client->vm_start = vma->vm_start;
1350         size = vma->vm_end - vma->vm_start;
1351         page_count = size >> PAGE_SHIFT;
1352         if (size & ~PAGE_MASK)
1353                 return -EINVAL;
1354
1355         if (vma->vm_flags & VM_WRITE)
1356                 direction = DMA_TO_DEVICE;
1357         else
1358                 direction = DMA_FROM_DEVICE;
1359
1360         ret = fw_iso_buffer_init(&client->buffer, client->device->card,
1361                                  page_count, direction);
1362         if (ret < 0)
1363                 return ret;
1364
1365         ret = fw_iso_buffer_map(&client->buffer, vma);
1366         if (ret < 0)
1367                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1368
1369         return ret;
1370 }
1371
1372 static int shutdown_resource(int id, void *p, void *data)
1373 {
1374         struct client_resource *r = p;
1375         struct client *client = data;
1376
1377         r->release(client, r);
1378         client_put(client);
1379
1380         return 0;
1381 }
1382
1383 static int fw_device_op_release(struct inode *inode, struct file *file)
1384 {
1385         struct client *client = file->private_data;
1386         struct event *e, *next_e;
1387         unsigned long flags;
1388
1389         mutex_lock(&client->device->client_list_mutex);
1390         list_del(&client->link);
1391         mutex_unlock(&client->device->client_list_mutex);
1392
1393         if (client->buffer.pages)
1394                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1395
1396         if (client->iso_context)
1397                 fw_iso_context_destroy(client->iso_context);
1398
1399         /* Freeze client->resource_idr and client->event_list */
1400         spin_lock_irqsave(&client->lock, flags);
1401         client->in_shutdown = true;
1402         spin_unlock_irqrestore(&client->lock, flags);
1403
1404         idr_for_each(&client->resource_idr, shutdown_resource, client);
1405         idr_remove_all(&client->resource_idr);
1406         idr_destroy(&client->resource_idr);
1407
1408         list_for_each_entry_safe(e, next_e, &client->event_list, link)
1409                 kfree(e);
1410
1411         client_put(client);
1412
1413         return 0;
1414 }
1415
1416 static unsigned int fw_device_op_poll(struct file *file, poll_table * pt)
1417 {
1418         struct client *client = file->private_data;
1419         unsigned int mask = 0;
1420
1421         poll_wait(file, &client->wait, pt);
1422
1423         if (fw_device_is_shutdown(client->device))
1424                 mask |= POLLHUP | POLLERR;
1425         if (!list_empty(&client->event_list))
1426                 mask |= POLLIN | POLLRDNORM;
1427
1428         return mask;
1429 }
1430
1431 const struct file_operations fw_device_ops = {
1432         .owner          = THIS_MODULE,
1433         .open           = fw_device_op_open,
1434         .read           = fw_device_op_read,
1435         .unlocked_ioctl = fw_device_op_ioctl,
1436         .poll           = fw_device_op_poll,
1437         .release        = fw_device_op_release,
1438         .mmap           = fw_device_op_mmap,
1439
1440 #ifdef CONFIG_COMPAT
1441         .compat_ioctl   = fw_device_op_compat_ioctl,
1442 #endif
1443 };