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/spinlock.h>
  38 #include <linux/string.h>
  39 #include <linux/time.h>
  40 #include <linux/uaccess.h>
  41 #include <linux/vmalloc.h>
  42 #include <linux/wait.h>
  43 #include <linux/workqueue.h>
  44
  45 #include <asm/system.h>
  46
  47 #include "core.h"
  48
  49 struct client {
  50         u32 version;
  51         struct fw_device *device;
  52
  53         spinlock_t lock;
  54         bool in_shutdown;
  55         struct idr resource_idr;
  56         struct list_head event_list;
  57         wait_queue_head_t wait;
  58         u64 bus_reset_closure;
  59
  60         struct fw_iso_context *iso_context;
  61         u64 iso_closure;
  62         struct fw_iso_buffer buffer;
  63         unsigned long vm_start;
  64
  65         struct list_head link;
  66         struct kref kref;
  67 };
  68
  69 static inline void client_get(struct client *client)
  70 {
  71         kref_get(&client->kref);
  72 }
  73
  74 static void client_release(struct kref *kref)
  75 {
  76         struct client *client = container_of(kref, struct client, kref);
  77
  78         fw_device_put(client->device);
  79         kfree(client);
  80 }
  81
  82 static void client_put(struct client *client)
  83 {
  84         kref_put(&client->kref, client_release);
  85 }
  86
  87 struct client_resource;
  88 typedef void (*client_resource_release_fn_t)(struct client *,
  89                                              struct client_resource *);
  90 struct client_resource {
  91         client_resource_release_fn_t release;
  92         int handle;
  93 };
  94
  95 struct address_handler_resource {
  96         struct client_resource resource;
  97         struct fw_address_handler handler;
  98         __u64 closure;
  99         struct client *client;
 100 };
 101
 102 struct outbound_transaction_resource {
 103         struct client_resource resource;
 104         struct fw_transaction transaction;
 105 };
 106
 107 struct inbound_transaction_resource {
 108         struct client_resource resource;
 109         struct fw_card *card;
 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(r->card, r->request, RCODE_CONFLICT_ERROR);
 630         kfree(r);
 631 }
 632
 633 static void handle_request(struct fw_card *card, struct fw_request *request,
 634                            int tcode, int destination, int source,
 635                            int generation, unsigned long long offset,
 636                            void *payload, size_t length, void *callback_data)
 637 {
 638         struct address_handler_resource *handler = callback_data;
 639         struct inbound_transaction_resource *r;
 640         struct inbound_transaction_event *e;
 641         void *fcp_frame = NULL;
 642         int ret;
 643
 644         r = kmalloc(sizeof(*r), GFP_ATOMIC);
 645         e = kmalloc(sizeof(*e), GFP_ATOMIC);
 646         if (r == NULL || e == NULL)
 647                 goto failed;
 648
 649         r->card    = card;
 650         r->request = request;
 651         r->data    = payload;
 652         r->length  = length;
 653
 654         if (is_fcp_request(request)) {
 655                 /*
 656                  * FIXME: Let core-transaction.c manage a
 657                  * single reference-counted copy?
 658                  */
 659                 fcp_frame = kmemdup(payload, length, GFP_ATOMIC);
 660                 if (fcp_frame == NULL)
 661                         goto failed;
 662
 663                 r->data = fcp_frame;
 664         }
 665
 666         r->resource.release = release_request;
 667         ret = add_client_resource(handler->client, &r->resource, GFP_ATOMIC);
 668         if (ret < 0)
 669                 goto failed;
 670
 671         e->request.type    = FW_CDEV_EVENT_REQUEST;
 672         e->request.tcode   = tcode;
 673         e->request.offset  = offset;
 674         e->request.length  = length;
 675         e->request.handle  = r->resource.handle;
 676         e->request.closure = handler->closure;
 677
 678         queue_event(handler->client, &e->event,
 679                     &e->request, sizeof(e->request), r->data, length);
 680         return;
 681
 682  failed:
 683         kfree(r);
 684         kfree(e);
 685         kfree(fcp_frame);
 686
 687         if (!is_fcp_request(request))
 688                 fw_send_response(card, request, RCODE_CONFLICT_ERROR);
 689 }
 690
 691 static void release_address_handler(struct client *client,
 692                                     struct client_resource *resource)
 693 {
 694         struct address_handler_resource *r =
 695             container_of(resource, struct address_handler_resource, resource);
 696
 697         fw_core_remove_address_handler(&r->handler);
 698         kfree(r);
 699 }
 700
 701 static int ioctl_allocate(struct client *client, union ioctl_arg *arg)
 702 {
 703         struct fw_cdev_allocate *a = &arg->allocate;
 704         struct address_handler_resource *r;
 705         struct fw_address_region region;
 706         int ret;
 707
 708         r = kmalloc(sizeof(*r), GFP_KERNEL);
 709         if (r == NULL)
 710                 return -ENOMEM;
 711
 712         region.start = a->offset;
 713         region.end   = a->offset + a->length;
 714         r->handler.length           = a->length;
 715         r->handler.address_callback = handle_request;
 716         r->handler.callback_data    = r;
 717         r->closure   = a->closure;
 718         r->client    = client;
 719
 720         ret = fw_core_add_address_handler(&r->handler, &region);
 721         if (ret < 0) {
 722                 kfree(r);
 723                 return ret;
 724         }
 725
 726         r->resource.release = release_address_handler;
 727         ret = add_client_resource(client, &r->resource, GFP_KERNEL);
 728         if (ret < 0) {
 729                 release_address_handler(client, &r->resource);
 730                 return ret;
 731         }
 732         a->handle = r->resource.handle;
 733
 734         return 0;
 735 }
 736
 737 static int ioctl_deallocate(struct client *client, union ioctl_arg *arg)
 738 {
 739         return release_client_resource(client, arg->deallocate.handle,
 740                                        release_address_handler, NULL);
 741 }
 742
 743 static int ioctl_send_response(struct client *client, union ioctl_arg *arg)
 744 {
 745         struct fw_cdev_send_response *a = &arg->send_response;
 746         struct client_resource *resource;
 747         struct inbound_transaction_resource *r;
 748         int ret = 0;
 749
 750         if (release_client_resource(client, a->handle,
 751                                     release_request, &resource) < 0)
 752                 return -EINVAL;
 753
 754         r = container_of(resource, struct inbound_transaction_resource,
 755                          resource);
 756         if (is_fcp_request(r->request))
 757                 goto out;
 758
 759         if (a->length != fw_get_response_length(r->request)) {
 760                 ret = -EINVAL;
 761                 kfree(r->request);
 762                 goto out;
 763         }
 764         if (copy_from_user(r->data, u64_to_uptr(a->data), a->length)) {
 765                 ret = -EFAULT;
 766                 kfree(r->request);
 767                 goto out;
 768         }
 769         fw_send_response(r->card, r->request, a->rcode);
 770  out:
 771         kfree(r);
 772
 773         return ret;
 774 }
 775
 776 static int ioctl_initiate_bus_reset(struct client *client, union ioctl_arg *arg)
 777 {
 778         return fw_core_initiate_bus_reset(client->device->card,
 779                         arg->initiate_bus_reset.type == FW_CDEV_SHORT_RESET);
 780 }
 781
 782 static void release_descriptor(struct client *client,
 783                                struct client_resource *resource)
 784 {
 785         struct descriptor_resource *r =
 786                 container_of(resource, struct descriptor_resource, resource);
 787
 788         fw_core_remove_descriptor(&r->descriptor);
 789         kfree(r);
 790 }
 791
 792 static int ioctl_add_descriptor(struct client *client, union ioctl_arg *arg)
 793 {
 794         struct fw_cdev_add_descriptor *a = &arg->add_descriptor;
 795         struct descriptor_resource *r;
 796         int ret;
 797
 798         /* Access policy: Allow this ioctl only on local nodes' device files. */
 799         if (!client->device->is_local)
 800                 return -ENOSYS;
 801
 802         if (a->length > 256)
 803                 return -EINVAL;
 804
 805         r = kmalloc(sizeof(*r) + a->length * 4, GFP_KERNEL);
 806         if (r == NULL)
 807                 return -ENOMEM;
 808
 809         if (copy_from_user(r->data, u64_to_uptr(a->data), a->length * 4)) {
 810                 ret = -EFAULT;
 811                 goto failed;
 812         }
 813
 814         r->descriptor.length    = a->length;
 815         r->descriptor.immediate = a->immediate;
 816         r->descriptor.key       = a->key;
 817         r->descriptor.data      = r->data;
 818
 819         ret = fw_core_add_descriptor(&r->descriptor);
 820         if (ret < 0)
 821                 goto failed;
 822
 823         r->resource.release = release_descriptor;
 824         ret = add_client_resource(client, &r->resource, GFP_KERNEL);
 825         if (ret < 0) {
 826                 fw_core_remove_descriptor(&r->descriptor);
 827                 goto failed;
 828         }
 829         a->handle = r->resource.handle;
 830
 831         return 0;
 832  failed:
 833         kfree(r);
 834
 835         return ret;
 836 }
 837
 838 static int ioctl_remove_descriptor(struct client *client, union ioctl_arg *arg)
 839 {
 840         return release_client_resource(client, arg->remove_descriptor.handle,
 841                                        release_descriptor, NULL);
 842 }
 843
 844 static void iso_callback(struct fw_iso_context *context, u32 cycle,
 845                          size_t header_length, void *header, void *data)
 846 {
 847         struct client *client = data;
 848         struct iso_interrupt_event *e;
 849
 850         e = kmalloc(sizeof(*e) + header_length, GFP_ATOMIC);
 851         if (e == NULL)
 852                 return;
 853
 854         e->interrupt.type      = FW_CDEV_EVENT_ISO_INTERRUPT;
 855         e->interrupt.closure   = client->iso_closure;
 856         e->interrupt.cycle     = cycle;
 857         e->interrupt.header_length = header_length;
 858         memcpy(e->interrupt.header, header, header_length);
 859         queue_event(client, &e->event, &e->interrupt,
 860                     sizeof(e->interrupt) + header_length, NULL, 0);
 861 }
 862
 863 static int ioctl_create_iso_context(struct client *client, union ioctl_arg *arg)
 864 {
 865         struct fw_cdev_create_iso_context *a = &arg->create_iso_context;
 866         struct fw_iso_context *context;
 867
 868         if (a->channel > 63)
 869                 return -EINVAL;
 870
 871         switch (a->type) {
 872         case FW_ISO_CONTEXT_RECEIVE:
 873                 if (a->header_size < 4 || (a->header_size & 3))
 874                         return -EINVAL;
 875                 break;
 876
 877         case FW_ISO_CONTEXT_TRANSMIT:
 878                 if (a->speed > SCODE_3200)
 879                         return -EINVAL;
 880                 break;
 881
 882         default:
 883                 return -EINVAL;
 884         }
 885
 886         context = fw_iso_context_create(client->device->card, a->type,
 887                                         a->channel, a->speed, a->header_size,
 888                                         iso_callback, client);
 889         if (IS_ERR(context))
 890                 return PTR_ERR(context);
 891
 892         /* We only support one context at this time. */
 893         spin_lock_irq(&client->lock);
 894         if (client->iso_context != NULL) {
 895                 spin_unlock_irq(&client->lock);
 896                 fw_iso_context_destroy(context);
 897                 return -EBUSY;
 898         }
 899         client->iso_closure = a->closure;
 900         client->iso_context = context;
 901         spin_unlock_irq(&client->lock);
 902
 903         a->handle = 0;
 904
 905         return 0;
 906 }
 907
 908 /* Macros for decoding the iso packet control header. */
 909 #define GET_PAYLOAD_LENGTH(v)   ((v) & 0xffff)
 910 #define GET_INTERRUPT(v)        (((v) >> 16) & 0x01)
 911 #define GET_SKIP(v)             (((v) >> 17) & 0x01)
 912 #define GET_TAG(v)              (((v) >> 18) & 0x03)
 913 #define GET_SY(v)               (((v) >> 20) & 0x0f)
 914 #define GET_HEADER_LENGTH(v)    (((v) >> 24) & 0xff)
 915
 916 static int ioctl_queue_iso(struct client *client, union ioctl_arg *arg)
 917 {
 918         struct fw_cdev_queue_iso *a = &arg->queue_iso;
 919         struct fw_cdev_iso_packet __user *p, *end, *next;
 920         struct fw_iso_context *ctx = client->iso_context;
 921         unsigned long payload, buffer_end, header_length;
 922         u32 control;
 923         int count;
 924         struct {
 925                 struct fw_iso_packet packet;
 926                 u8 header[256];
 927         } u;
 928
 929         if (ctx == NULL || a->handle != 0)
 930                 return -EINVAL;
 931
 932         /*
 933          * If the user passes a non-NULL data pointer, has mmap()'ed
 934          * the iso buffer, and the pointer points inside the buffer,
 935          * we setup the payload pointers accordingly.  Otherwise we
 936          * set them both to 0, which will still let packets with
 937          * payload_length == 0 through.  In other words, if no packets
 938          * use the indirect payload, the iso buffer need not be mapped
 939          * and the a->data pointer is ignored.
 940          */
 941
 942         payload = (unsigned long)a->data - client->vm_start;
 943         buffer_end = client->buffer.page_count << PAGE_SHIFT;
 944         if (a->data == 0 || client->buffer.pages == NULL ||
 945             payload >= buffer_end) {
 946                 payload = 0;
 947                 buffer_end = 0;
 948         }
 949
 950         p = (struct fw_cdev_iso_packet __user *)u64_to_uptr(a->packets);
 951
 952         if (!access_ok(VERIFY_READ, p, a->size))
 953                 return -EFAULT;
 954
 955         end = (void __user *)p + a->size;
 956         count = 0;
 957         while (p < end) {
 958                 if (get_user(control, &p->control))
 959                         return -EFAULT;
 960                 u.packet.payload_length = GET_PAYLOAD_LENGTH(control);
 961                 u.packet.interrupt = GET_INTERRUPT(control);
 962                 u.packet.skip = GET_SKIP(control);
 963                 u.packet.tag = GET_TAG(control);
 964                 u.packet.sy = GET_SY(control);
 965                 u.packet.header_length = GET_HEADER_LENGTH(control);
 966
 967                 if (ctx->type == FW_ISO_CONTEXT_TRANSMIT) {
 968                         if (u.packet.header_length % 4 != 0)
 969                                 return -EINVAL;
 970                         header_length = u.packet.header_length;
 971                 } else {
 972                         /*
 973                          * We require that header_length is a multiple of
 974                          * the fixed header size, ctx->header_size.
 975                          */
 976                         if (ctx->header_size == 0) {
 977                                 if (u.packet.header_length > 0)
 978                                         return -EINVAL;
 979                         } else if (u.packet.header_length == 0 ||
 980                                    u.packet.header_length % ctx->header_size != 0) {
 981                                 return -EINVAL;
 982                         }
 983                         header_length = 0;
 984                 }
 985
 986                 next = (struct fw_cdev_iso_packet __user *)
 987                         &p->header[header_length / 4];
 988                 if (next > end)
 989                         return -EINVAL;
 990                 if (__copy_from_user
 991                     (u.packet.header, p->header, header_length))
 992                         return -EFAULT;
 993                 if (u.packet.skip && ctx->type == FW_ISO_CONTEXT_TRANSMIT &&
 994                     u.packet.header_length + u.packet.payload_length > 0)
 995                         return -EINVAL;
 996                 if (payload + u.packet.payload_length > buffer_end)
 997                         return -EINVAL;
 998
 999                 if (fw_iso_context_queue(ctx, &u.packet,
1000                                          &client->buffer, payload))
1001                         break;
1002
1003                 p = next;
1004                 payload += u.packet.payload_length;
1005                 count++;
1006         }
1007
1008         a->size    -= uptr_to_u64(p) - a->packets;
1009         a->packets  = uptr_to_u64(p);
1010         a->data     = client->vm_start + payload;
1011
1012         return count;
1013 }
1014
1015 static int ioctl_start_iso(struct client *client, union ioctl_arg *arg)
1016 {
1017         struct fw_cdev_start_iso *a = &arg->start_iso;
1018
1019         if (client->iso_context == NULL || a->handle != 0)
1020                 return -EINVAL;
1021
1022         if (client->iso_context->type == FW_ISO_CONTEXT_RECEIVE &&
1023             (a->tags == 0 || a->tags > 15 || a->sync > 15))
1024                 return -EINVAL;
1025
1026         return fw_iso_context_start(client->iso_context,
1027                                     a->cycle, a->sync, a->tags);
1028 }
1029
1030 static int ioctl_stop_iso(struct client *client, union ioctl_arg *arg)
1031 {
1032         struct fw_cdev_stop_iso *a = &arg->stop_iso;
1033
1034         if (client->iso_context == NULL || a->handle != 0)
1035                 return -EINVAL;
1036
1037         return fw_iso_context_stop(client->iso_context);
1038 }
1039
1040 static int ioctl_get_cycle_timer2(struct client *client, union ioctl_arg *arg)
1041 {
1042         struct fw_cdev_get_cycle_timer2 *a = &arg->get_cycle_timer2;
1043         struct fw_card *card = client->device->card;
1044         struct timespec ts = {0, 0};
1045         u32 cycle_time;
1046         int ret = 0;
1047
1048         local_irq_disable();
1049
1050         cycle_time = card->driver->read_csr(card, CSR_CYCLE_TIME);
1051
1052         switch (a->clk_id) {
1053         case CLOCK_REALTIME:      getnstimeofday(&ts);                   break;
1054         case CLOCK_MONOTONIC:     do_posix_clock_monotonic_gettime(&ts); break;
1055         case CLOCK_MONOTONIC_RAW: getrawmonotonic(&ts);                  break;
1056         default:
1057                 ret = -EINVAL;
1058         }
1059
1060         local_irq_enable();
1061
1062         a->tv_sec      = ts.tv_sec;
1063         a->tv_nsec     = ts.tv_nsec;
1064         a->cycle_timer = cycle_time;
1065
1066         return ret;
1067 }
1068
1069 static int ioctl_get_cycle_timer(struct client *client, union ioctl_arg *arg)
1070 {
1071         struct fw_cdev_get_cycle_timer *a = &arg->get_cycle_timer;
1072         struct fw_cdev_get_cycle_timer2 ct2;
1073
1074         ct2.clk_id = CLOCK_REALTIME;
1075         ioctl_get_cycle_timer2(client, (union ioctl_arg *)&ct2);
1076
1077         a->local_time = ct2.tv_sec * USEC_PER_SEC + ct2.tv_nsec / NSEC_PER_USEC;
1078         a->cycle_timer = ct2.cycle_timer;
1079
1080         return 0;
1081 }
1082
1083 static void iso_resource_work(struct work_struct *work)
1084 {
1085         struct iso_resource_event *e;
1086         struct iso_resource *r =
1087                         container_of(work, struct iso_resource, work.work);
1088         struct client *client = r->client;
1089         int generation, channel, bandwidth, todo;
1090         bool skip, free, success;
1091
1092         spin_lock_irq(&client->lock);
1093         generation = client->device->generation;
1094         todo = r->todo;
1095         /* Allow 1000ms grace period for other reallocations. */
1096         if (todo == ISO_RES_ALLOC &&
1097             time_is_after_jiffies(client->device->card->reset_jiffies + HZ)) {
1098                 schedule_iso_resource(r, DIV_ROUND_UP(HZ, 3));
1099                 skip = true;
1100         } else {
1101                 /* We could be called twice within the same generation. */
1102                 skip = todo == ISO_RES_REALLOC &&
1103                        r->generation == generation;
1104         }
1105         free = todo == ISO_RES_DEALLOC ||
1106                todo == ISO_RES_ALLOC_ONCE ||
1107                todo == ISO_RES_DEALLOC_ONCE;
1108         r->generation = generation;
1109         spin_unlock_irq(&client->lock);
1110
1111         if (skip)
1112                 goto out;
1113
1114         bandwidth = r->bandwidth;
1115
1116         fw_iso_resource_manage(client->device->card, generation,
1117                         r->channels, &channel, &bandwidth,
1118                         todo == ISO_RES_ALLOC ||
1119                         todo == ISO_RES_REALLOC ||
1120                         todo == ISO_RES_ALLOC_ONCE,
1121                         r->transaction_data);
1122         /*
1123          * Is this generation outdated already?  As long as this resource sticks
1124          * in the idr, it will be scheduled again for a newer generation or at
1125          * shutdown.
1126          */
1127         if (channel == -EAGAIN &&
1128             (todo == ISO_RES_ALLOC || todo == ISO_RES_REALLOC))
1129                 goto out;
1130
1131         success = channel >= 0 || bandwidth > 0;
1132
1133         spin_lock_irq(&client->lock);
1134         /*
1135          * Transit from allocation to reallocation, except if the client
1136          * requested deallocation in the meantime.
1137          */
1138         if (r->todo == ISO_RES_ALLOC)
1139                 r->todo = ISO_RES_REALLOC;
1140         /*
1141          * Allocation or reallocation failure?  Pull this resource out of the
1142          * idr and prepare for deletion, unless the client is shutting down.
1143          */
1144         if (r->todo == ISO_RES_REALLOC && !success &&
1145             !client->in_shutdown &&
1146             idr_find(&client->resource_idr, r->resource.handle)) {
1147                 idr_remove(&client->resource_idr, r->resource.handle);
1148                 client_put(client);
1149                 free = true;
1150         }
1151         spin_unlock_irq(&client->lock);
1152
1153         if (todo == ISO_RES_ALLOC && channel >= 0)
1154                 r->channels = 1ULL << channel;
1155
1156         if (todo == ISO_RES_REALLOC && success)
1157                 goto out;
1158
1159         if (todo == ISO_RES_ALLOC || todo == ISO_RES_ALLOC_ONCE) {
1160                 e = r->e_alloc;
1161                 r->e_alloc = NULL;
1162         } else {
1163                 e = r->e_dealloc;
1164                 r->e_dealloc = NULL;
1165         }
1166         e->iso_resource.handle    = r->resource.handle;
1167         e->iso_resource.channel   = channel;
1168         e->iso_resource.bandwidth = bandwidth;
1169
1170         queue_event(client, &e->event,
1171                     &e->iso_resource, sizeof(e->iso_resource), NULL, 0);
1172
1173         if (free) {
1174                 cancel_delayed_work(&r->work);
1175                 kfree(r->e_alloc);
1176                 kfree(r->e_dealloc);
1177                 kfree(r);
1178         }
1179  out:
1180         client_put(client);
1181 }
1182
1183 static void release_iso_resource(struct client *client,
1184                                  struct client_resource *resource)
1185 {
1186         struct iso_resource *r =
1187                 container_of(resource, struct iso_resource, resource);
1188
1189         spin_lock_irq(&client->lock);
1190         r->todo = ISO_RES_DEALLOC;
1191         schedule_iso_resource(r, 0);
1192         spin_unlock_irq(&client->lock);
1193 }
1194
1195 static int init_iso_resource(struct client *client,
1196                 struct fw_cdev_allocate_iso_resource *request, int todo)
1197 {
1198         struct iso_resource_event *e1, *e2;
1199         struct iso_resource *r;
1200         int ret;
1201
1202         if ((request->channels == 0 && request->bandwidth == 0) ||
1203             request->bandwidth > BANDWIDTH_AVAILABLE_INITIAL ||
1204             request->bandwidth < 0)
1205                 return -EINVAL;
1206
1207         r  = kmalloc(sizeof(*r), GFP_KERNEL);
1208         e1 = kmalloc(sizeof(*e1), GFP_KERNEL);
1209         e2 = kmalloc(sizeof(*e2), GFP_KERNEL);
1210         if (r == NULL || e1 == NULL || e2 == NULL) {
1211                 ret = -ENOMEM;
1212                 goto fail;
1213         }
1214
1215         INIT_DELAYED_WORK(&r->work, iso_resource_work);
1216         r->client       = client;
1217         r->todo         = todo;
1218         r->generation   = -1;
1219         r->channels     = request->channels;
1220         r->bandwidth    = request->bandwidth;
1221         r->e_alloc      = e1;
1222         r->e_dealloc    = e2;
1223
1224         e1->iso_resource.closure = request->closure;
1225         e1->iso_resource.type    = FW_CDEV_EVENT_ISO_RESOURCE_ALLOCATED;
1226         e2->iso_resource.closure = request->closure;
1227         e2->iso_resource.type    = FW_CDEV_EVENT_ISO_RESOURCE_DEALLOCATED;
1228
1229         if (todo == ISO_RES_ALLOC) {
1230                 r->resource.release = release_iso_resource;
1231                 ret = add_client_resource(client, &r->resource, GFP_KERNEL);
1232                 if (ret < 0)
1233                         goto fail;
1234         } else {
1235                 r->resource.release = NULL;
1236                 r->resource.handle = -1;
1237                 schedule_iso_resource(r, 0);
1238         }
1239         request->handle = r->resource.handle;
1240
1241         return 0;
1242  fail:
1243         kfree(r);
1244         kfree(e1);
1245         kfree(e2);
1246
1247         return ret;
1248 }
1249
1250 static int ioctl_allocate_iso_resource(struct client *client,
1251                                        union ioctl_arg *arg)
1252 {
1253         return init_iso_resource(client,
1254                         &arg->allocate_iso_resource, ISO_RES_ALLOC);
1255 }
1256
1257 static int ioctl_deallocate_iso_resource(struct client *client,
1258                                          union ioctl_arg *arg)
1259 {
1260         return release_client_resource(client,
1261                         arg->deallocate.handle, release_iso_resource, NULL);
1262 }
1263
1264 static int ioctl_allocate_iso_resource_once(struct client *client,
1265                                             union ioctl_arg *arg)
1266 {
1267         return init_iso_resource(client,
1268                         &arg->allocate_iso_resource, ISO_RES_ALLOC_ONCE);
1269 }
1270
1271 static int ioctl_deallocate_iso_resource_once(struct client *client,
1272                                               union ioctl_arg *arg)
1273 {
1274         return init_iso_resource(client,
1275                         &arg->allocate_iso_resource, ISO_RES_DEALLOC_ONCE);
1276 }
1277
1278 /*
1279  * Returns a speed code:  Maximum speed to or from this device,
1280  * limited by the device's link speed, the local node's link speed,
1281  * and all PHY port speeds between the two links.
1282  */
1283 static int ioctl_get_speed(struct client *client, union ioctl_arg *arg)
1284 {
1285         return client->device->max_speed;
1286 }
1287
1288 static int ioctl_send_broadcast_request(struct client *client,
1289                                         union ioctl_arg *arg)
1290 {
1291         struct fw_cdev_send_request *a = &arg->send_request;
1292
1293         switch (a->tcode) {
1294         case TCODE_WRITE_QUADLET_REQUEST:
1295         case TCODE_WRITE_BLOCK_REQUEST:
1296                 break;
1297         default:
1298                 return -EINVAL;
1299         }
1300
1301         /* Security policy: Only allow accesses to Units Space. */
1302         if (a->offset < CSR_REGISTER_BASE + CSR_CONFIG_ROM_END)
1303                 return -EACCES;
1304
1305         return init_request(client, a, LOCAL_BUS | 0x3f, SCODE_100);
1306 }
1307
1308 static int ioctl_send_stream_packet(struct client *client, union ioctl_arg *arg)
1309 {
1310         struct fw_cdev_send_stream_packet *a = &arg->send_stream_packet;
1311         struct fw_cdev_send_request request;
1312         int dest;
1313
1314         if (a->speed > client->device->card->link_speed ||
1315             a->length > 1024 << a->speed)
1316                 return -EIO;
1317
1318         if (a->tag > 3 || a->channel > 63 || a->sy > 15)
1319                 return -EINVAL;
1320
1321         dest = fw_stream_packet_destination_id(a->tag, a->channel, a->sy);
1322         request.tcode           = TCODE_STREAM_DATA;
1323         request.length          = a->length;
1324         request.closure         = a->closure;
1325         request.data            = a->data;
1326         request.generation      = a->generation;
1327
1328         return init_request(client, &request, dest, a->speed);
1329 }
1330
1331 static int (* const ioctl_handlers[])(struct client *, union ioctl_arg *) = {
1332         ioctl_get_info,
1333         ioctl_send_request,
1334         ioctl_allocate,
1335         ioctl_deallocate,
1336         ioctl_send_response,
1337         ioctl_initiate_bus_reset,
1338         ioctl_add_descriptor,
1339         ioctl_remove_descriptor,
1340         ioctl_create_iso_context,
1341         ioctl_queue_iso,
1342         ioctl_start_iso,
1343         ioctl_stop_iso,
1344         ioctl_get_cycle_timer,
1345         ioctl_allocate_iso_resource,
1346         ioctl_deallocate_iso_resource,
1347         ioctl_allocate_iso_resource_once,
1348         ioctl_deallocate_iso_resource_once,
1349         ioctl_get_speed,
1350         ioctl_send_broadcast_request,
1351         ioctl_send_stream_packet,
1352         ioctl_get_cycle_timer2,
1353 };
1354
1355 static int dispatch_ioctl(struct client *client,
1356                           unsigned int cmd, void __user *arg)
1357 {
1358         union ioctl_arg buffer;
1359         int ret;
1360
1361         if (fw_device_is_shutdown(client->device))
1362                 return -ENODEV;
1363
1364         if (_IOC_TYPE(cmd) != '#' ||
1365             _IOC_NR(cmd) >= ARRAY_SIZE(ioctl_handlers) ||
1366             _IOC_SIZE(cmd) > sizeof(buffer))
1367                 return -EINVAL;
1368
1369         if (_IOC_DIR(cmd) == _IOC_READ)
1370                 memset(&buffer, 0, _IOC_SIZE(cmd));
1371
1372         if (_IOC_DIR(cmd) & _IOC_WRITE)
1373                 if (copy_from_user(&buffer, arg, _IOC_SIZE(cmd)))
1374                         return -EFAULT;
1375
1376         ret = ioctl_handlers[_IOC_NR(cmd)](client, &buffer);
1377         if (ret < 0)
1378                 return ret;
1379
1380         if (_IOC_DIR(cmd) & _IOC_READ)
1381                 if (copy_to_user(arg, &buffer, _IOC_SIZE(cmd)))
1382                         return -EFAULT;
1383
1384         return ret;
1385 }
1386
1387 static long fw_device_op_ioctl(struct file *file,
1388                                unsigned int cmd, unsigned long arg)
1389 {
1390         return dispatch_ioctl(file->private_data, cmd, (void __user *)arg);
1391 }
1392
1393 #ifdef CONFIG_COMPAT
1394 static long fw_device_op_compat_ioctl(struct file *file,
1395                                       unsigned int cmd, unsigned long arg)
1396 {
1397         return dispatch_ioctl(file->private_data, cmd, compat_ptr(arg));
1398 }
1399 #endif
1400
1401 static int fw_device_op_mmap(struct file *file, struct vm_area_struct *vma)
1402 {
1403         struct client *client = file->private_data;
1404         enum dma_data_direction direction;
1405         unsigned long size;
1406         int page_count, ret;
1407
1408         if (fw_device_is_shutdown(client->device))
1409                 return -ENODEV;
1410
1411         /* FIXME: We could support multiple buffers, but we don't. */
1412         if (client->buffer.pages != NULL)
1413                 return -EBUSY;
1414
1415         if (!(vma->vm_flags & VM_SHARED))
1416                 return -EINVAL;
1417
1418         if (vma->vm_start & ~PAGE_MASK)
1419                 return -EINVAL;
1420
1421         client->vm_start = vma->vm_start;
1422         size = vma->vm_end - vma->vm_start;
1423         page_count = size >> PAGE_SHIFT;
1424         if (size & ~PAGE_MASK)
1425                 return -EINVAL;
1426
1427         if (vma->vm_flags & VM_WRITE)
1428                 direction = DMA_TO_DEVICE;
1429         else
1430                 direction = DMA_FROM_DEVICE;
1431
1432         ret = fw_iso_buffer_init(&client->buffer, client->device->card,
1433                                  page_count, direction);
1434         if (ret < 0)
1435                 return ret;
1436
1437         ret = fw_iso_buffer_map(&client->buffer, vma);
1438         if (ret < 0)
1439                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1440
1441         return ret;
1442 }
1443
1444 static int shutdown_resource(int id, void *p, void *data)
1445 {
1446         struct client_resource *resource = p;
1447         struct client *client = data;
1448
1449         resource->release(client, resource);
1450         client_put(client);
1451
1452         return 0;
1453 }
1454
1455 static int fw_device_op_release(struct inode *inode, struct file *file)
1456 {
1457         struct client *client = file->private_data;
1458         struct event *event, *next_event;
1459
1460         mutex_lock(&client->device->client_list_mutex);
1461         list_del(&client->link);
1462         mutex_unlock(&client->device->client_list_mutex);
1463
1464         if (client->iso_context)
1465                 fw_iso_context_destroy(client->iso_context);
1466
1467         if (client->buffer.pages)
1468                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1469
1470         /* Freeze client->resource_idr and client->event_list */
1471         spin_lock_irq(&client->lock);
1472         client->in_shutdown = true;
1473         spin_unlock_irq(&client->lock);
1474
1475         idr_for_each(&client->resource_idr, shutdown_resource, client);
1476         idr_remove_all(&client->resource_idr);
1477         idr_destroy(&client->resource_idr);
1478
1479         list_for_each_entry_safe(event, next_event, &client->event_list, link)
1480                 kfree(event);
1481
1482         client_put(client);
1483
1484         return 0;
1485 }
1486
1487 static unsigned int fw_device_op_poll(struct file *file, poll_table * pt)
1488 {
1489         struct client *client = file->private_data;
1490         unsigned int mask = 0;
1491
1492         poll_wait(file, &client->wait, pt);
1493
1494         if (fw_device_is_shutdown(client->device))
1495                 mask |= POLLHUP | POLLERR;
1496         if (!list_empty(&client->event_list))
1497                 mask |= POLLIN | POLLRDNORM;
1498
1499         return mask;
1500 }
1501
1502 const struct file_operations fw_device_ops = {
1503         .owner          = THIS_MODULE,
1504         .llseek         = no_llseek,
1505         .open           = fw_device_op_open,
1506         .read           = fw_device_op_read,
1507         .unlocked_ioctl = fw_device_op_ioctl,
1508         .mmap           = fw_device_op_mmap,
1509         .release        = fw_device_op_release,
1510         .poll           = fw_device_op_poll,
1511 #ifdef CONFIG_COMPAT
1512         .compat_ioctl   = fw_device_op_compat_ioctl,
1513 #endif
1514 };