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