SAFE public projects git trees. - safe/jmp/linux-2.6/blob - 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 0;
 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                         header_length = u.packet.header_length;
 964                 } else {
 965                         /*
 966                          * We require that header_length is a multiple of
 967                          * the fixed header size, ctx->header_size.
 968                          */
 969                         if (ctx->header_size == 0) {
 970                                 if (u.packet.header_length > 0)
 971                                         return -EINVAL;
 972                         } else if (u.packet.header_length % ctx->header_size != 0) {
 973                                 return -EINVAL;
 974                         }
 975                         header_length = 0;
 976                 }
 977
 978                 next = (struct fw_cdev_iso_packet __user *)
 979                         &p->header[header_length / 4];
 980                 if (next > end)
 981                         return -EINVAL;
 982                 if (__copy_from_user
 983                     (u.packet.header, p->header, header_length))
 984                         return -EFAULT;
 985                 if (u.packet.skip && ctx->type == FW_ISO_CONTEXT_TRANSMIT &&
 986                     u.packet.header_length + u.packet.payload_length > 0)
 987                         return -EINVAL;
 988                 if (payload + u.packet.payload_length > buffer_end)
 989                         return -EINVAL;
 990
 991                 if (fw_iso_context_queue(ctx, &u.packet,
 992                                          &client->buffer, payload))
 993                         break;
 994
 995                 p = next;
 996                 payload += u.packet.payload_length;
 997                 count++;
 998         }
 999
1000         a->size    -= uptr_to_u64(p) - a->packets;
1001         a->packets  = uptr_to_u64(p);
1002         a->data     = client->vm_start + payload;
1003
1004         return count;
1005 }
1006
1007 static int ioctl_start_iso(struct client *client, union ioctl_arg *arg)
1008 {
1009         struct fw_cdev_start_iso *a = &arg->start_iso;
1010
1011         if (client->iso_context == NULL || a->handle != 0)
1012                 return -EINVAL;
1013
1014         if (client->iso_context->type == FW_ISO_CONTEXT_RECEIVE &&
1015             (a->tags == 0 || a->tags > 15 || a->sync > 15))
1016                 return -EINVAL;
1017
1018         return fw_iso_context_start(client->iso_context,
1019                                     a->cycle, a->sync, a->tags);
1020 }
1021
1022 static int ioctl_stop_iso(struct client *client, union ioctl_arg *arg)
1023 {
1024         struct fw_cdev_stop_iso *a = &arg->stop_iso;
1025
1026         if (client->iso_context == NULL || a->handle != 0)
1027                 return -EINVAL;
1028
1029         return fw_iso_context_stop(client->iso_context);
1030 }
1031
1032 static int ioctl_get_cycle_timer2(struct client *client, union ioctl_arg *arg)
1033 {
1034         struct fw_cdev_get_cycle_timer2 *a = &arg->get_cycle_timer2;
1035         struct fw_card *card = client->device->card;
1036         struct timespec ts = {0, 0};
1037         u32 cycle_time;
1038         int ret = 0;
1039
1040         local_irq_disable();
1041
1042         cycle_time = card->driver->get_cycle_time(card);
1043
1044         switch (a->clk_id) {
1045         case CLOCK_REALTIME:      getnstimeofday(&ts);                   break;
1046         case CLOCK_MONOTONIC:     do_posix_clock_monotonic_gettime(&ts); break;
1047         case CLOCK_MONOTONIC_RAW: getrawmonotonic(&ts);                  break;
1048         default:
1049                 ret = -EINVAL;
1050         }
1051
1052         local_irq_enable();
1053
1054         a->tv_sec      = ts.tv_sec;
1055         a->tv_nsec     = ts.tv_nsec;
1056         a->cycle_timer = cycle_time;
1057
1058         return ret;
1059 }
1060
1061 static int ioctl_get_cycle_timer(struct client *client, union ioctl_arg *arg)
1062 {
1063         struct fw_cdev_get_cycle_timer *a = &arg->get_cycle_timer;
1064         struct fw_cdev_get_cycle_timer2 ct2;
1065
1066         ct2.clk_id = CLOCK_REALTIME;
1067         ioctl_get_cycle_timer2(client, (union ioctl_arg *)&ct2);
1068
1069         a->local_time = ct2.tv_sec * USEC_PER_SEC + ct2.tv_nsec / NSEC_PER_USEC;
1070         a->cycle_timer = ct2.cycle_timer;
1071
1072         return 0;
1073 }
1074
1075 static void iso_resource_work(struct work_struct *work)
1076 {
1077         struct iso_resource_event *e;
1078         struct iso_resource *r =
1079                         container_of(work, struct iso_resource, work.work);
1080         struct client *client = r->client;
1081         int generation, channel, bandwidth, todo;
1082         bool skip, free, success;
1083
1084         spin_lock_irq(&client->lock);
1085         generation = client->device->generation;
1086         todo = r->todo;
1087         /* Allow 1000ms grace period for other reallocations. */
1088         if (todo == ISO_RES_ALLOC &&
1089             time_is_after_jiffies(client->device->card->reset_jiffies + HZ)) {
1090                 schedule_iso_resource(r, DIV_ROUND_UP(HZ, 3));
1091                 skip = true;
1092         } else {
1093                 /* We could be called twice within the same generation. */
1094                 skip = todo == ISO_RES_REALLOC &&
1095                        r->generation == generation;
1096         }
1097         free = todo == ISO_RES_DEALLOC ||
1098                todo == ISO_RES_ALLOC_ONCE ||
1099                todo == ISO_RES_DEALLOC_ONCE;
1100         r->generation = generation;
1101         spin_unlock_irq(&client->lock);
1102
1103         if (skip)
1104                 goto out;
1105
1106         bandwidth = r->bandwidth;
1107
1108         fw_iso_resource_manage(client->device->card, generation,
1109                         r->channels, &channel, &bandwidth,
1110                         todo == ISO_RES_ALLOC ||
1111                         todo == ISO_RES_REALLOC ||
1112                         todo == ISO_RES_ALLOC_ONCE,
1113                         r->transaction_data);
1114         /*
1115          * Is this generation outdated already?  As long as this resource sticks
1116          * in the idr, it will be scheduled again for a newer generation or at
1117          * shutdown.
1118          */
1119         if (channel == -EAGAIN &&
1120             (todo == ISO_RES_ALLOC || todo == ISO_RES_REALLOC))
1121                 goto out;
1122
1123         success = channel >= 0 || bandwidth > 0;
1124
1125         spin_lock_irq(&client->lock);
1126         /*
1127          * Transit from allocation to reallocation, except if the client
1128          * requested deallocation in the meantime.
1129          */
1130         if (r->todo == ISO_RES_ALLOC)
1131                 r->todo = ISO_RES_REALLOC;
1132         /*
1133          * Allocation or reallocation failure?  Pull this resource out of the
1134          * idr and prepare for deletion, unless the client is shutting down.
1135          */
1136         if (r->todo == ISO_RES_REALLOC && !success &&
1137             !client->in_shutdown &&
1138             idr_find(&client->resource_idr, r->resource.handle)) {
1139                 idr_remove(&client->resource_idr, r->resource.handle);
1140                 client_put(client);
1141                 free = true;
1142         }
1143         spin_unlock_irq(&client->lock);
1144
1145         if (todo == ISO_RES_ALLOC && channel >= 0)
1146                 r->channels = 1ULL << channel;
1147
1148         if (todo == ISO_RES_REALLOC && success)
1149                 goto out;
1150
1151         if (todo == ISO_RES_ALLOC || todo == ISO_RES_ALLOC_ONCE) {
1152                 e = r->e_alloc;
1153                 r->e_alloc = NULL;
1154         } else {
1155                 e = r->e_dealloc;
1156                 r->e_dealloc = NULL;
1157         }
1158         e->iso_resource.handle    = r->resource.handle;
1159         e->iso_resource.channel   = channel;
1160         e->iso_resource.bandwidth = bandwidth;
1161
1162         queue_event(client, &e->event,
1163                     &e->iso_resource, sizeof(e->iso_resource), NULL, 0);
1164
1165         if (free) {
1166                 cancel_delayed_work(&r->work);
1167                 kfree(r->e_alloc);
1168                 kfree(r->e_dealloc);
1169                 kfree(r);
1170         }
1171  out:
1172         client_put(client);
1173 }
1174
1175 static void release_iso_resource(struct client *client,
1176                                  struct client_resource *resource)
1177 {
1178         struct iso_resource *r =
1179                 container_of(resource, struct iso_resource, resource);
1180
1181         spin_lock_irq(&client->lock);
1182         r->todo = ISO_RES_DEALLOC;
1183         schedule_iso_resource(r, 0);
1184         spin_unlock_irq(&client->lock);
1185 }
1186
1187 static int init_iso_resource(struct client *client,
1188                 struct fw_cdev_allocate_iso_resource *request, int todo)
1189 {
1190         struct iso_resource_event *e1, *e2;
1191         struct iso_resource *r;
1192         int ret;
1193
1194         if ((request->channels == 0 && request->bandwidth == 0) ||
1195             request->bandwidth > BANDWIDTH_AVAILABLE_INITIAL ||
1196             request->bandwidth < 0)
1197                 return -EINVAL;
1198
1199         r  = kmalloc(sizeof(*r), GFP_KERNEL);
1200         e1 = kmalloc(sizeof(*e1), GFP_KERNEL);
1201         e2 = kmalloc(sizeof(*e2), GFP_KERNEL);
1202         if (r == NULL || e1 == NULL || e2 == NULL) {
1203                 ret = -ENOMEM;
1204                 goto fail;
1205         }
1206
1207         INIT_DELAYED_WORK(&r->work, iso_resource_work);
1208         r->client       = client;
1209         r->todo         = todo;
1210         r->generation   = -1;
1211         r->channels     = request->channels;
1212         r->bandwidth    = request->bandwidth;
1213         r->e_alloc      = e1;
1214         r->e_dealloc    = e2;
1215
1216         e1->iso_resource.closure = request->closure;
1217         e1->iso_resource.type    = FW_CDEV_EVENT_ISO_RESOURCE_ALLOCATED;
1218         e2->iso_resource.closure = request->closure;
1219         e2->iso_resource.type    = FW_CDEV_EVENT_ISO_RESOURCE_DEALLOCATED;
1220
1221         if (todo == ISO_RES_ALLOC) {
1222                 r->resource.release = release_iso_resource;
1223                 ret = add_client_resource(client, &r->resource, GFP_KERNEL);
1224                 if (ret < 0)
1225                         goto fail;
1226         } else {
1227                 r->resource.release = NULL;
1228                 r->resource.handle = -1;
1229                 schedule_iso_resource(r, 0);
1230         }
1231         request->handle = r->resource.handle;
1232
1233         return 0;
1234  fail:
1235         kfree(r);
1236         kfree(e1);
1237         kfree(e2);
1238
1239         return ret;
1240 }
1241
1242 static int ioctl_allocate_iso_resource(struct client *client,
1243                                        union ioctl_arg *arg)
1244 {
1245         return init_iso_resource(client,
1246                         &arg->allocate_iso_resource, ISO_RES_ALLOC);
1247 }
1248
1249 static int ioctl_deallocate_iso_resource(struct client *client,
1250                                          union ioctl_arg *arg)
1251 {
1252         return release_client_resource(client,
1253                         arg->deallocate.handle, release_iso_resource, NULL);
1254 }
1255
1256 static int ioctl_allocate_iso_resource_once(struct client *client,
1257                                             union ioctl_arg *arg)
1258 {
1259         return init_iso_resource(client,
1260                         &arg->allocate_iso_resource, ISO_RES_ALLOC_ONCE);
1261 }
1262
1263 static int ioctl_deallocate_iso_resource_once(struct client *client,
1264                                               union ioctl_arg *arg)
1265 {
1266         return init_iso_resource(client,
1267                         &arg->allocate_iso_resource, ISO_RES_DEALLOC_ONCE);
1268 }
1269
1270 /*
1271  * Returns a speed code:  Maximum speed to or from this device,
1272  * limited by the device's link speed, the local node's link speed,
1273  * and all PHY port speeds between the two links.
1274  */
1275 static int ioctl_get_speed(struct client *client, union ioctl_arg *arg)
1276 {
1277         return client->device->max_speed;
1278 }
1279
1280 static int ioctl_send_broadcast_request(struct client *client,
1281                                         union ioctl_arg *arg)
1282 {
1283         struct fw_cdev_send_request *a = &arg->send_request;
1284
1285         switch (a->tcode) {
1286         case TCODE_WRITE_QUADLET_REQUEST:
1287         case TCODE_WRITE_BLOCK_REQUEST:
1288                 break;
1289         default:
1290                 return -EINVAL;
1291         }
1292
1293         /* Security policy: Only allow accesses to Units Space. */
1294         if (a->offset < CSR_REGISTER_BASE + CSR_CONFIG_ROM_END)
1295                 return -EACCES;
1296
1297         return init_request(client, a, LOCAL_BUS | 0x3f, SCODE_100);
1298 }
1299
1300 static int ioctl_send_stream_packet(struct client *client, union ioctl_arg *arg)
1301 {
1302         struct fw_cdev_send_stream_packet *a = &arg->send_stream_packet;
1303         struct fw_cdev_send_request request;
1304         int dest;
1305
1306         if (a->speed > client->device->card->link_speed ||
1307             a->length > 1024 << a->speed)
1308                 return -EIO;
1309
1310         if (a->tag > 3 || a->channel > 63 || a->sy > 15)
1311                 return -EINVAL;
1312
1313         dest = fw_stream_packet_destination_id(a->tag, a->channel, a->sy);
1314         request.tcode           = TCODE_STREAM_DATA;
1315         request.length          = a->length;
1316         request.closure         = a->closure;
1317         request.data            = a->data;
1318         request.generation      = a->generation;
1319
1320         return init_request(client, &request, dest, a->speed);
1321 }
1322
1323 static int (* const ioctl_handlers[])(struct client *, union ioctl_arg *) = {
1324         ioctl_get_info,
1325         ioctl_send_request,
1326         ioctl_allocate,
1327         ioctl_deallocate,
1328         ioctl_send_response,
1329         ioctl_initiate_bus_reset,
1330         ioctl_add_descriptor,
1331         ioctl_remove_descriptor,
1332         ioctl_create_iso_context,
1333         ioctl_queue_iso,
1334         ioctl_start_iso,
1335         ioctl_stop_iso,
1336         ioctl_get_cycle_timer,
1337         ioctl_allocate_iso_resource,
1338         ioctl_deallocate_iso_resource,
1339         ioctl_allocate_iso_resource_once,
1340         ioctl_deallocate_iso_resource_once,
1341         ioctl_get_speed,
1342         ioctl_send_broadcast_request,
1343         ioctl_send_stream_packet,
1344         ioctl_get_cycle_timer2,
1345 };
1346
1347 static int dispatch_ioctl(struct client *client,
1348                           unsigned int cmd, void __user *arg)
1349 {
1350         union ioctl_arg buffer;
1351         int ret;
1352
1353         if (fw_device_is_shutdown(client->device))
1354                 return -ENODEV;
1355
1356         if (_IOC_TYPE(cmd) != '#' ||
1357             _IOC_NR(cmd) >= ARRAY_SIZE(ioctl_handlers))
1358                 return -EINVAL;
1359
1360         if (_IOC_DIR(cmd) & _IOC_WRITE) {
1361                 if (_IOC_SIZE(cmd) > sizeof(buffer) ||
1362                     copy_from_user(&buffer, arg, _IOC_SIZE(cmd)))
1363                         return -EFAULT;
1364         }
1365
1366         ret = ioctl_handlers[_IOC_NR(cmd)](client, &buffer);
1367         if (ret < 0)
1368                 return ret;
1369
1370         if (_IOC_DIR(cmd) & _IOC_READ) {
1371                 if (_IOC_SIZE(cmd) > sizeof(buffer) ||
1372                     copy_to_user(arg, &buffer, _IOC_SIZE(cmd)))
1373                         return -EFAULT;
1374         }
1375
1376         return ret;
1377 }
1378
1379 static long fw_device_op_ioctl(struct file *file,
1380                                unsigned int cmd, unsigned long arg)
1381 {
1382         return dispatch_ioctl(file->private_data, cmd, (void __user *)arg);
1383 }
1384
1385 #ifdef CONFIG_COMPAT
1386 static long fw_device_op_compat_ioctl(struct file *file,
1387                                       unsigned int cmd, unsigned long arg)
1388 {
1389         return dispatch_ioctl(file->private_data, cmd, compat_ptr(arg));
1390 }
1391 #endif
1392
1393 static int fw_device_op_mmap(struct file *file, struct vm_area_struct *vma)
1394 {
1395         struct client *client = file->private_data;
1396         enum dma_data_direction direction;
1397         unsigned long size;
1398         int page_count, ret;
1399
1400         if (fw_device_is_shutdown(client->device))
1401                 return -ENODEV;
1402
1403         /* FIXME: We could support multiple buffers, but we don't. */
1404         if (client->buffer.pages != NULL)
1405                 return -EBUSY;
1406
1407         if (!(vma->vm_flags & VM_SHARED))
1408                 return -EINVAL;
1409
1410         if (vma->vm_start & ~PAGE_MASK)
1411                 return -EINVAL;
1412
1413         client->vm_start = vma->vm_start;
1414         size = vma->vm_end - vma->vm_start;
1415         page_count = size >> PAGE_SHIFT;
1416         if (size & ~PAGE_MASK)
1417                 return -EINVAL;
1418
1419         if (vma->vm_flags & VM_WRITE)
1420                 direction = DMA_TO_DEVICE;
1421         else
1422                 direction = DMA_FROM_DEVICE;
1423
1424         ret = fw_iso_buffer_init(&client->buffer, client->device->card,
1425                                  page_count, direction);
1426         if (ret < 0)
1427                 return ret;
1428
1429         ret = fw_iso_buffer_map(&client->buffer, vma);
1430         if (ret < 0)
1431                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1432
1433         return ret;
1434 }
1435
1436 static int shutdown_resource(int id, void *p, void *data)
1437 {
1438         struct client_resource *resource = p;
1439         struct client *client = data;
1440
1441         resource->release(client, resource);
1442         client_put(client);
1443
1444         return 0;
1445 }
1446
1447 static int fw_device_op_release(struct inode *inode, struct file *file)
1448 {
1449         struct client *client = file->private_data;
1450         struct event *event, *next_event;
1451
1452         mutex_lock(&client->device->client_list_mutex);
1453         list_del(&client->link);
1454         mutex_unlock(&client->device->client_list_mutex);
1455
1456         if (client->iso_context)
1457                 fw_iso_context_destroy(client->iso_context);
1458
1459         if (client->buffer.pages)
1460                 fw_iso_buffer_destroy(&client->buffer, client->device->card);
1461
1462         /* Freeze client->resource_idr and client->event_list */
1463         spin_lock_irq(&client->lock);
1464         client->in_shutdown = true;
1465         spin_unlock_irq(&client->lock);
1466
1467         idr_for_each(&client->resource_idr, shutdown_resource, client);
1468         idr_remove_all(&client->resource_idr);
1469         idr_destroy(&client->resource_idr);
1470
1471         list_for_each_entry_safe(event, next_event, &client->event_list, link)
1472                 kfree(event);
1473
1474         client_put(client);
1475
1476         return 0;
1477 }
1478
1479 static unsigned int fw_device_op_poll(struct file *file, poll_table * pt)
1480 {
1481         struct client *client = file->private_data;
1482         unsigned int mask = 0;
1483
1484         poll_wait(file, &client->wait, pt);
1485
1486         if (fw_device_is_shutdown(client->device))
1487                 mask |= POLLHUP | POLLERR;
1488         if (!list_empty(&client->event_list))
1489                 mask |= POLLIN | POLLRDNORM;
1490
1491         return mask;
1492 }
1493
1494 const struct file_operations fw_device_ops = {
1495         .owner          = THIS_MODULE,
1496         .open           = fw_device_op_open,
1497         .read           = fw_device_op_read,
1498         .unlocked_ioctl = fw_device_op_ioctl,
1499         .poll           = fw_device_op_poll,
1500         .release        = fw_device_op_release,
1501         .mmap           = fw_device_op_mmap,
1502
1503 #ifdef CONFIG_COMPAT
1504         .compat_ioctl   = fw_device_op_compat_ioctl,
1505 #endif
1506 };