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