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