drivers/infiniband/hw/ehca/ehca_irq.c

   1 /*
   2  *  IBM eServer eHCA Infiniband device driver for Linux on POWER
   3  *
   4  *  Functions for EQs, NEQs and interrupts
   5  *
   6  *  Authors: Heiko J Schick <schickhj@de.ibm.com>
   7  *           Khadija Souissi <souissi@de.ibm.com>
   8  *
   9  *  Copyright (c) 2005 IBM Corporation
  10  *
  11  *  All rights reserved.
  12  *
  13  *  This source code is distributed under a dual license of GPL v2.0 and OpenIB
  14  *  BSD.
  15  *
  16  * OpenIB BSD License
  17  *
  18  * Redistribution and use in source and binary forms, with or without
  19  * modification, are permitted provided that the following conditions are met:
  20  *
  21  * Redistributions of source code must retain the above copyright notice, this
  22  * list of conditions and the following disclaimer.
  23  *
  24  * Redistributions in binary form must reproduce the above copyright notice,
  25  * this list of conditions and the following disclaimer in the documentation
  26  * and/or other materials
  27  * provided with the distribution.
  28  *
  29  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  30  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  31  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  32  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  33  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  34  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  35  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
  36  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  37  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  38  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  39  * POSSIBILITY OF SUCH DAMAGE.
  40  */
  41
  42 #include "ehca_classes.h"
  43 #include "ehca_irq.h"
  44 #include "ehca_iverbs.h"
  45 #include "ehca_tools.h"
  46 #include "hcp_if.h"
  47 #include "hipz_fns.h"
  48 #include "ipz_pt_fn.h"
  49
  50 #define EQE_COMPLETION_EVENT   EHCA_BMASK_IBM(1,1)
  51 #define EQE_CQ_QP_NUMBER       EHCA_BMASK_IBM(8,31)
  52 #define EQE_EE_IDENTIFIER      EHCA_BMASK_IBM(2,7)
  53 #define EQE_CQ_NUMBER          EHCA_BMASK_IBM(8,31)
  54 #define EQE_QP_NUMBER          EHCA_BMASK_IBM(8,31)
  55 #define EQE_QP_TOKEN           EHCA_BMASK_IBM(32,63)
  56 #define EQE_CQ_TOKEN           EHCA_BMASK_IBM(32,63)
  57
  58 #define NEQE_COMPLETION_EVENT  EHCA_BMASK_IBM(1,1)
  59 #define NEQE_EVENT_CODE        EHCA_BMASK_IBM(2,7)
  60 #define NEQE_PORT_NUMBER       EHCA_BMASK_IBM(8,15)
  61 #define NEQE_PORT_AVAILABILITY EHCA_BMASK_IBM(16,16)
  62
  63 #define ERROR_DATA_LENGTH      EHCA_BMASK_IBM(52,63)
  64 #define ERROR_DATA_TYPE        EHCA_BMASK_IBM(0,7)
  65
  66 static void queue_comp_task(struct ehca_cq *__cq);
  67
  68 static struct ehca_comp_pool* pool;
  69 static struct notifier_block comp_pool_callback_nb;
  70
  71 static inline void comp_event_callback(struct ehca_cq *cq)
  72 {
  73         if (!cq->ib_cq.comp_handler)
  74                 return;
  75
  76         spin_lock(&cq->cb_lock);
  77         cq->ib_cq.comp_handler(&cq->ib_cq, cq->ib_cq.cq_context);
  78         spin_unlock(&cq->cb_lock);
  79
  80         return;
  81 }
  82
  83 static void print_error_data(struct ehca_shca * shca, void* data,
  84                              u64* rblock, int length)
  85 {
  86         u64 type = EHCA_BMASK_GET(ERROR_DATA_TYPE, rblock[2]);
  87         u64 resource = rblock[1];
  88
  89         switch (type) {
  90         case 0x1: /* Queue Pair */
  91         {
  92                 struct ehca_qp *qp = (struct ehca_qp*)data;
  93
  94                 /* only print error data if AER is set */
  95                 if (rblock[6] == 0)
  96                         return;
  97
  98                 ehca_err(&shca->ib_device,
  99                          "QP 0x%x (resource=%lx) has errors.",
 100                          qp->ib_qp.qp_num, resource);
 101                 break;
 102         }
 103         case 0x4: /* Completion Queue */
 104         {
 105                 struct ehca_cq *cq = (struct ehca_cq*)data;
 106
 107                 ehca_err(&shca->ib_device,
 108                          "CQ 0x%x (resource=%lx) has errors.",
 109                          cq->cq_number, resource);
 110                 break;
 111         }
 112         default:
 113                 ehca_err(&shca->ib_device,
 114                          "Unknown errror type: %lx on %s.",
 115                          type, shca->ib_device.name);
 116                 break;
 117         }
 118
 119         ehca_err(&shca->ib_device, "Error data is available: %lx.", resource);
 120         ehca_err(&shca->ib_device, "EHCA ----- error data begin "
 121                  "---------------------------------------------------");
 122         ehca_dmp(rblock, length, "resource=%lx", resource);
 123         ehca_err(&shca->ib_device, "EHCA ----- error data end "
 124                  "----------------------------------------------------");
 125
 126         return;
 127 }
 128
 129 int ehca_error_data(struct ehca_shca *shca, void *data,
 130                     u64 resource)
 131 {
 132
 133         unsigned long ret;
 134         u64 *rblock;
 135         unsigned long block_count;
 136
 137         rblock = ehca_alloc_fw_ctrlblock(GFP_ATOMIC);
 138         if (!rblock) {
 139                 ehca_err(&shca->ib_device, "Cannot allocate rblock memory.");
 140                 ret = -ENOMEM;
 141                 goto error_data1;
 142         }
 143
 144         /* rblock must be 4K aligned and should be 4K large */
 145         ret = hipz_h_error_data(shca->ipz_hca_handle,
 146                                 resource,
 147                                 rblock,
 148                                 &block_count);
 149
 150         if (ret == H_R_STATE)
 151                 ehca_err(&shca->ib_device,
 152                          "No error data is available: %lx.", resource);
 153         else if (ret == H_SUCCESS) {
 154                 int length;
 155
 156                 length = EHCA_BMASK_GET(ERROR_DATA_LENGTH, rblock[0]);
 157
 158                 if (length > EHCA_PAGESIZE)
 159                         length = EHCA_PAGESIZE;
 160
 161                 print_error_data(shca, data, rblock, length);
 162         } else
 163                 ehca_err(&shca->ib_device,
 164                          "Error data could not be fetched: %lx", resource);
 165
 166         ehca_free_fw_ctrlblock(rblock);
 167
 168 error_data1:
 169         return ret;
 170
 171 }
 172
 173 static void qp_event_callback(struct ehca_shca *shca,
 174                               u64 eqe,
 175                               enum ib_event_type event_type)
 176 {
 177         struct ib_event event;
 178         struct ehca_qp *qp;
 179         unsigned long flags;
 180         u32 token = EHCA_BMASK_GET(EQE_QP_TOKEN, eqe);
 181
 182         spin_lock_irqsave(&ehca_qp_idr_lock, flags);
 183         qp = idr_find(&ehca_qp_idr, token);
 184         spin_unlock_irqrestore(&ehca_qp_idr_lock, flags);
 185
 186
 187         if (!qp)
 188                 return;
 189
 190         ehca_error_data(shca, qp, qp->ipz_qp_handle.handle);
 191
 192         if (!qp->ib_qp.event_handler)
 193                 return;
 194
 195         event.device     = &shca->ib_device;
 196         event.event      = event_type;
 197         event.element.qp = &qp->ib_qp;
 198
 199         qp->ib_qp.event_handler(&event, qp->ib_qp.qp_context);
 200
 201         return;
 202 }
 203
 204 static void cq_event_callback(struct ehca_shca *shca,
 205                               u64 eqe)
 206 {
 207         struct ehca_cq *cq;
 208         unsigned long flags;
 209         u32 token = EHCA_BMASK_GET(EQE_CQ_TOKEN, eqe);
 210
 211         spin_lock_irqsave(&ehca_cq_idr_lock, flags);
 212         cq = idr_find(&ehca_cq_idr, token);
 213         spin_unlock_irqrestore(&ehca_cq_idr_lock, flags);
 214
 215         if (!cq)
 216                 return;
 217
 218         ehca_error_data(shca, cq, cq->ipz_cq_handle.handle);
 219
 220         return;
 221 }
 222
 223 static void parse_identifier(struct ehca_shca *shca, u64 eqe)
 224 {
 225         u8 identifier = EHCA_BMASK_GET(EQE_EE_IDENTIFIER, eqe);
 226
 227         switch (identifier) {
 228         case 0x02: /* path migrated */
 229                 qp_event_callback(shca, eqe, IB_EVENT_PATH_MIG);
 230                 break;
 231         case 0x03: /* communication established */
 232                 qp_event_callback(shca, eqe, IB_EVENT_COMM_EST);
 233                 break;
 234         case 0x04: /* send queue drained */
 235                 qp_event_callback(shca, eqe, IB_EVENT_SQ_DRAINED);
 236                 break;
 237         case 0x05: /* QP error */
 238         case 0x06: /* QP error */
 239                 qp_event_callback(shca, eqe, IB_EVENT_QP_FATAL);
 240                 break;
 241         case 0x07: /* CQ error */
 242         case 0x08: /* CQ error */
 243                 cq_event_callback(shca, eqe);
 244                 break;
 245         case 0x09: /* MRMWPTE error */
 246                 ehca_err(&shca->ib_device, "MRMWPTE error.");
 247                 break;
 248         case 0x0A: /* port event */
 249                 ehca_err(&shca->ib_device, "Port event.");
 250                 break;
 251         case 0x0B: /* MR access error */
 252                 ehca_err(&shca->ib_device, "MR access error.");
 253                 break;
 254         case 0x0C: /* EQ error */
 255                 ehca_err(&shca->ib_device, "EQ error.");
 256                 break;
 257         case 0x0D: /* P/Q_Key mismatch */
 258                 ehca_err(&shca->ib_device, "P/Q_Key mismatch.");
 259                 break;
 260         case 0x10: /* sampling complete */
 261                 ehca_err(&shca->ib_device, "Sampling complete.");
 262                 break;
 263         case 0x11: /* unaffiliated access error */
 264                 ehca_err(&shca->ib_device, "Unaffiliated access error.");
 265                 break;
 266         case 0x12: /* path migrating error */
 267                 ehca_err(&shca->ib_device, "Path migration error.");
 268                 break;
 269         case 0x13: /* interface trace stopped */
 270                 ehca_err(&shca->ib_device, "Interface trace stopped.");
 271                 break;
 272         case 0x14: /* first error capture info available */
 273         default:
 274                 ehca_err(&shca->ib_device, "Unknown identifier: %x on %s.",
 275                          identifier, shca->ib_device.name);
 276                 break;
 277         }
 278
 279         return;
 280 }
 281
 282 static void parse_ec(struct ehca_shca *shca, u64 eqe)
 283 {
 284         struct ib_event event;
 285         u8 ec   = EHCA_BMASK_GET(NEQE_EVENT_CODE, eqe);
 286         u8 port = EHCA_BMASK_GET(NEQE_PORT_NUMBER, eqe);
 287
 288         switch (ec) {
 289         case 0x30: /* port availability change */
 290                 if (EHCA_BMASK_GET(NEQE_PORT_AVAILABILITY, eqe)) {
 291                         ehca_info(&shca->ib_device,
 292                                   "port %x is active.", port);
 293                         event.device = &shca->ib_device;
 294                         event.event = IB_EVENT_PORT_ACTIVE;
 295                         event.element.port_num = port;
 296                         shca->sport[port - 1].port_state = IB_PORT_ACTIVE;
 297                         ib_dispatch_event(&event);
 298                 } else {
 299                         ehca_info(&shca->ib_device,
 300                                   "port %x is inactive.", port);
 301                         event.device = &shca->ib_device;
 302                         event.event = IB_EVENT_PORT_ERR;
 303                         event.element.port_num = port;
 304                         shca->sport[port - 1].port_state = IB_PORT_DOWN;
 305                         ib_dispatch_event(&event);
 306                 }
 307                 break;
 308         case 0x31:
 309                 /* port configuration change
 310                  * disruptive change is caused by
 311                  * LID, PKEY or SM change
 312                  */
 313                 ehca_warn(&shca->ib_device,
 314                           "disruptive port %x configuration change", port);
 315
 316                 ehca_info(&shca->ib_device,
 317                           "port %x is inactive.", port);
 318                 event.device = &shca->ib_device;
 319                 event.event = IB_EVENT_PORT_ERR;
 320                 event.element.port_num = port;
 321                 shca->sport[port - 1].port_state = IB_PORT_DOWN;
 322                 ib_dispatch_event(&event);
 323
 324                 ehca_info(&shca->ib_device,
 325                           "port %x is active.", port);
 326                 event.device = &shca->ib_device;
 327                 event.event = IB_EVENT_PORT_ACTIVE;
 328                 event.element.port_num = port;
 329                 shca->sport[port - 1].port_state = IB_PORT_ACTIVE;
 330                 ib_dispatch_event(&event);
 331                 break;
 332         case 0x32: /* adapter malfunction */
 333                 ehca_err(&shca->ib_device, "Adapter malfunction.");
 334                 break;
 335         case 0x33:  /* trace stopped */
 336                 ehca_err(&shca->ib_device, "Traced stopped.");
 337                 break;
 338         default:
 339                 ehca_err(&shca->ib_device, "Unknown event code: %x on %s.",
 340                          ec, shca->ib_device.name);
 341                 break;
 342         }
 343
 344         return;
 345 }
 346
 347 static inline void reset_eq_pending(struct ehca_cq *cq)
 348 {
 349         u64 CQx_EP;
 350         struct h_galpa gal = cq->galpas.kernel;
 351
 352         hipz_galpa_store_cq(gal, cqx_ep, 0x0);
 353         CQx_EP = hipz_galpa_load(gal, CQTEMM_OFFSET(cqx_ep));
 354
 355         return;
 356 }
 357
 358 irqreturn_t ehca_interrupt_neq(int irq, void *dev_id)
 359 {
 360         struct ehca_shca *shca = (struct ehca_shca*)dev_id;
 361
 362         tasklet_hi_schedule(&shca->neq.interrupt_task);
 363
 364         return IRQ_HANDLED;
 365 }
 366
 367 void ehca_tasklet_neq(unsigned long data)
 368 {
 369         struct ehca_shca *shca = (struct ehca_shca*)data;
 370         struct ehca_eqe *eqe;
 371         u64 ret;
 372
 373         eqe = (struct ehca_eqe *)ehca_poll_eq(shca, &shca->neq);
 374
 375         while (eqe) {
 376                 if (!EHCA_BMASK_GET(NEQE_COMPLETION_EVENT, eqe->entry))
 377                         parse_ec(shca, eqe->entry);
 378
 379                 eqe = (struct ehca_eqe *)ehca_poll_eq(shca, &shca->neq);
 380         }
 381
 382         ret = hipz_h_reset_event(shca->ipz_hca_handle,
 383                                  shca->neq.ipz_eq_handle, 0xFFFFFFFFFFFFFFFFL);
 384
 385         if (ret != H_SUCCESS)
 386                 ehca_err(&shca->ib_device, "Can't clear notification events.");
 387
 388         return;
 389 }
 390
 391 irqreturn_t ehca_interrupt_eq(int irq, void *dev_id)
 392 {
 393         struct ehca_shca *shca = (struct ehca_shca*)dev_id;
 394
 395         tasklet_hi_schedule(&shca->eq.interrupt_task);
 396
 397         return IRQ_HANDLED;
 398 }
 399
 400
 401 static inline void process_eqe(struct ehca_shca *shca, struct ehca_eqe *eqe)
 402 {
 403         u64 eqe_value;
 404         u32 token;
 405         unsigned long flags;
 406         struct ehca_cq *cq;
 407         eqe_value = eqe->entry;
 408         ehca_dbg(&shca->ib_device, "eqe_value=%lx", eqe_value);
 409         if (EHCA_BMASK_GET(EQE_COMPLETION_EVENT, eqe_value)) {
 410                 ehca_dbg(&shca->ib_device, "... completion event");
 411                 token = EHCA_BMASK_GET(EQE_CQ_TOKEN, eqe_value);
 412                 spin_lock_irqsave(&ehca_cq_idr_lock, flags);
 413                 cq = idr_find(&ehca_cq_idr, token);
 414                 if (cq == NULL) {
 415                         spin_unlock_irqrestore(&ehca_cq_idr_lock, flags);
 416                         ehca_err(&shca->ib_device,
 417                                  "Invalid eqe for non-existing cq token=%x",
 418                                  token);
 419                         return;
 420                 }
 421                 reset_eq_pending(cq);
 422                 if (ehca_scaling_code) {
 423                         queue_comp_task(cq);
 424                         spin_unlock_irqrestore(&ehca_cq_idr_lock, flags);
 425                 } else {
 426                         spin_unlock_irqrestore(&ehca_cq_idr_lock, flags);
 427                         comp_event_callback(cq);
 428                 }
 429         } else {
 430                 ehca_dbg(&shca->ib_device,
 431                          "Got non completion event");
 432                 parse_identifier(shca, eqe_value);
 433         }
 434 }
 435
 436 void ehca_process_eq(struct ehca_shca *shca, int is_irq)
 437 {
 438         struct ehca_eq *eq = &shca->eq;
 439         struct ehca_eqe_cache_entry *eqe_cache = eq->eqe_cache;
 440         u64 eqe_value;
 441         unsigned long flags;
 442         int eqe_cnt, i;
 443         int eq_empty = 0;
 444
 445         spin_lock_irqsave(&eq->irq_spinlock, flags);
 446         if (is_irq) {
 447                 const int max_query_cnt = 100;
 448                 int query_cnt = 0;
 449                 int int_state = 1;
 450                 do {
 451                         int_state = hipz_h_query_int_state(
 452                                 shca->ipz_hca_handle, eq->ist);
 453                         query_cnt++;
 454                         iosync();
 455                 } while (int_state && query_cnt < max_query_cnt);
 456                 if (unlikely((query_cnt == max_query_cnt)))
 457                         ehca_dbg(&shca->ib_device, "int_state=%x query_cnt=%x",
 458                                  int_state, query_cnt);
 459         }
 460
 461         /* read out all eqes */
 462         eqe_cnt = 0;
 463         do {
 464                 u32 token;
 465                 eqe_cache[eqe_cnt].eqe =
 466                         (struct ehca_eqe *)ehca_poll_eq(shca, eq);
 467                 if (!eqe_cache[eqe_cnt].eqe)
 468                         break;
 469                 eqe_value = eqe_cache[eqe_cnt].eqe->entry;
 470                 if (EHCA_BMASK_GET(EQE_COMPLETION_EVENT, eqe_value)) {
 471                         token = EHCA_BMASK_GET(EQE_CQ_TOKEN, eqe_value);
 472                         spin_lock(&ehca_cq_idr_lock);
 473                         eqe_cache[eqe_cnt].cq = idr_find(&ehca_cq_idr, token);
 474                         if (!eqe_cache[eqe_cnt].cq) {
 475                                 spin_unlock(&ehca_cq_idr_lock);
 476                                 ehca_err(&shca->ib_device,
 477                                          "Invalid eqe for non-existing cq "
 478                                          "token=%x", token);
 479                                 continue;
 480                         }
 481                         spin_unlock(&ehca_cq_idr_lock);
 482                 } else
 483                         eqe_cache[eqe_cnt].cq = NULL;
 484                 eqe_cnt++;
 485         } while (eqe_cnt < EHCA_EQE_CACHE_SIZE);
 486         if (!eqe_cnt) {
 487                 if (is_irq)
 488                         ehca_dbg(&shca->ib_device,
 489                                  "No eqe found for irq event");
 490                 goto unlock_irq_spinlock;
 491         } else if (!is_irq)
 492                 ehca_dbg(&shca->ib_device, "deadman found %x eqe", eqe_cnt);
 493         if (unlikely(eqe_cnt == EHCA_EQE_CACHE_SIZE))
 494                 ehca_dbg(&shca->ib_device, "too many eqes for one irq event");
 495         /* enable irq for new packets */
 496         for (i = 0; i < eqe_cnt; i++) {
 497                 if (eq->eqe_cache[i].cq)
 498                         reset_eq_pending(eq->eqe_cache[i].cq);
 499         }
 500         /* check eq */
 501         spin_lock(&eq->spinlock);
 502         eq_empty = (!ipz_eqit_eq_peek_valid(&shca->eq.ipz_queue));
 503         spin_unlock(&eq->spinlock);
 504         /* call completion handler for cached eqes */
 505         for (i = 0; i < eqe_cnt; i++)
 506                 if (eq->eqe_cache[i].cq) {
 507                         if (ehca_scaling_code) {
 508                                 spin_lock(&ehca_cq_idr_lock);
 509                                 queue_comp_task(eq->eqe_cache[i].cq);
 510                                 spin_unlock(&ehca_cq_idr_lock);
 511                         } else
 512                                 comp_event_callback(eq->eqe_cache[i].cq);
 513                 } else {
 514                         ehca_dbg(&shca->ib_device, "Got non completion event");
 515                         parse_identifier(shca, eq->eqe_cache[i].eqe->entry);
 516                 }
 517         /* poll eq if not empty */
 518         if (eq_empty)
 519                 goto unlock_irq_spinlock;
 520         do {
 521                 struct ehca_eqe *eqe;
 522                 eqe = (struct ehca_eqe *)ehca_poll_eq(shca, &shca->eq);
 523                 if (!eqe)
 524                         break;
 525                 process_eqe(shca, eqe);
 526                 eqe_cnt++;
 527         } while (1);
 528
 529 unlock_irq_spinlock:
 530         spin_unlock_irqrestore(&eq->irq_spinlock, flags);
 531 }
 532
 533 void ehca_tasklet_eq(unsigned long data)
 534 {
 535         ehca_process_eq((struct ehca_shca*)data, 1);
 536 }
 537
 538 static inline int find_next_online_cpu(struct ehca_comp_pool* pool)
 539 {
 540         int cpu;
 541         unsigned long flags;
 542
 543         WARN_ON_ONCE(!in_interrupt());
 544         if (ehca_debug_level)
 545                 ehca_dmp(&cpu_online_map, sizeof(cpumask_t), "");
 546
 547         spin_lock_irqsave(&pool->last_cpu_lock, flags);
 548         cpu = next_cpu(pool->last_cpu, cpu_online_map);
 549         if (cpu == NR_CPUS)
 550                 cpu = first_cpu(cpu_online_map);
 551         pool->last_cpu = cpu;
 552         spin_unlock_irqrestore(&pool->last_cpu_lock, flags);
 553
 554         return cpu;
 555 }
 556
 557 static void __queue_comp_task(struct ehca_cq *__cq,
 558                               struct ehca_cpu_comp_task *cct)
 559 {
 560         unsigned long flags;
 561
 562         spin_lock_irqsave(&cct->task_lock, flags);
 563         spin_lock(&__cq->task_lock);
 564
 565         if (__cq->nr_callbacks == 0) {
 566                 __cq->nr_callbacks++;
 567                 list_add_tail(&__cq->entry, &cct->cq_list);
 568                 cct->cq_jobs++;
 569                 wake_up(&cct->wait_queue);
 570         }
 571         else
 572                 __cq->nr_callbacks++;
 573
 574         spin_unlock(&__cq->task_lock);
 575         spin_unlock_irqrestore(&cct->task_lock, flags);
 576 }
 577
 578 static void queue_comp_task(struct ehca_cq *__cq)
 579 {
 580         int cpu;
 581         int cpu_id;
 582         struct ehca_cpu_comp_task *cct;
 583
 584         cpu = get_cpu();
 585         cpu_id = find_next_online_cpu(pool);
 586         BUG_ON(!cpu_online(cpu_id));
 587
 588         cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu_id);
 589         BUG_ON(!cct);
 590
 591         if (cct->cq_jobs > 0) {
 592                 cpu_id = find_next_online_cpu(pool);
 593                 cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu_id);
 594                 BUG_ON(!cct);
 595         }
 596
 597         __queue_comp_task(__cq, cct);
 598 }
 599
 600 static void run_comp_task(struct ehca_cpu_comp_task* cct)
 601 {
 602         struct ehca_cq *cq;
 603         unsigned long flags;
 604
 605         spin_lock_irqsave(&cct->task_lock, flags);
 606
 607         while (!list_empty(&cct->cq_list)) {
 608                 cq = list_entry(cct->cq_list.next, struct ehca_cq, entry);
 609                 spin_unlock_irqrestore(&cct->task_lock, flags);
 610                 comp_event_callback(cq);
 611                 spin_lock_irqsave(&cct->task_lock, flags);
 612
 613                 spin_lock(&cq->task_lock);
 614                 cq->nr_callbacks--;
 615                 if (cq->nr_callbacks == 0) {
 616                         list_del_init(cct->cq_list.next);
 617                         cct->cq_jobs--;
 618                 }
 619                 spin_unlock(&cq->task_lock);
 620         }
 621
 622         spin_unlock_irqrestore(&cct->task_lock, flags);
 623 }
 624
 625 static int comp_task(void *__cct)
 626 {
 627         struct ehca_cpu_comp_task* cct = __cct;
 628         int cql_empty;
 629         DECLARE_WAITQUEUE(wait, current);
 630
 631         set_current_state(TASK_INTERRUPTIBLE);
 632         while(!kthread_should_stop()) {
 633                 add_wait_queue(&cct->wait_queue, &wait);
 634
 635                 spin_lock_irq(&cct->task_lock);
 636                 cql_empty = list_empty(&cct->cq_list);
 637                 spin_unlock_irq(&cct->task_lock);
 638                 if (cql_empty)
 639                         schedule();
 640                 else
 641                         __set_current_state(TASK_RUNNING);
 642
 643                 remove_wait_queue(&cct->wait_queue, &wait);
 644
 645                 spin_lock_irq(&cct->task_lock);
 646                 cql_empty = list_empty(&cct->cq_list);
 647                 spin_unlock_irq(&cct->task_lock);
 648                 if (!cql_empty)
 649                         run_comp_task(__cct);
 650
 651                 set_current_state(TASK_INTERRUPTIBLE);
 652         }
 653         __set_current_state(TASK_RUNNING);
 654
 655         return 0;
 656 }
 657
 658 static struct task_struct *create_comp_task(struct ehca_comp_pool *pool,
 659                                             int cpu)
 660 {
 661         struct ehca_cpu_comp_task *cct;
 662
 663         cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu);
 664         spin_lock_init(&cct->task_lock);
 665         INIT_LIST_HEAD(&cct->cq_list);
 666         init_waitqueue_head(&cct->wait_queue);
 667         cct->task = kthread_create(comp_task, cct, "ehca_comp/%d", cpu);
 668
 669         return cct->task;
 670 }
 671
 672 static void destroy_comp_task(struct ehca_comp_pool *pool,
 673                               int cpu)
 674 {
 675         struct ehca_cpu_comp_task *cct;
 676         struct task_struct *task;
 677         unsigned long flags_cct;
 678
 679         cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu);
 680
 681         spin_lock_irqsave(&cct->task_lock, flags_cct);
 682
 683         task = cct->task;
 684         cct->task = NULL;
 685         cct->cq_jobs = 0;
 686
 687         spin_unlock_irqrestore(&cct->task_lock, flags_cct);
 688
 689         if (task)
 690                 kthread_stop(task);
 691 }
 692
 693 static void take_over_work(struct ehca_comp_pool *pool,
 694                            int cpu)
 695 {
 696         struct ehca_cpu_comp_task *cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu);
 697         LIST_HEAD(list);
 698         struct ehca_cq *cq;
 699         unsigned long flags_cct;
 700
 701         spin_lock_irqsave(&cct->task_lock, flags_cct);
 702
 703         list_splice_init(&cct->cq_list, &list);
 704
 705         while(!list_empty(&list)) {
 706                 cq = list_entry(cct->cq_list.next, struct ehca_cq, entry);
 707
 708                 list_del(&cq->entry);
 709                 __queue_comp_task(cq, per_cpu_ptr(pool->cpu_comp_tasks,
 710                                                   smp_processor_id()));
 711         }
 712
 713         spin_unlock_irqrestore(&cct->task_lock, flags_cct);
 714
 715 }
 716
 717 static int comp_pool_callback(struct notifier_block *nfb,
 718                               unsigned long action,
 719                               void *hcpu)
 720 {
 721         unsigned int cpu = (unsigned long)hcpu;
 722         struct ehca_cpu_comp_task *cct;
 723
 724         switch (action) {
 725         case CPU_UP_PREPARE:
 726                 ehca_gen_dbg("CPU: %x (CPU_PREPARE)", cpu);
 727                 if(!create_comp_task(pool, cpu)) {
 728                         ehca_gen_err("Can't create comp_task for cpu: %x", cpu);
 729                         return NOTIFY_BAD;
 730                 }
 731                 break;
 732         case CPU_UP_CANCELED:
 733                 ehca_gen_dbg("CPU: %x (CPU_CANCELED)", cpu);
 734                 cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu);
 735                 kthread_bind(cct->task, any_online_cpu(cpu_online_map));
 736                 destroy_comp_task(pool, cpu);
 737                 break;
 738         case CPU_ONLINE:
 739                 ehca_gen_dbg("CPU: %x (CPU_ONLINE)", cpu);
 740                 cct = per_cpu_ptr(pool->cpu_comp_tasks, cpu);
 741                 kthread_bind(cct->task, cpu);
 742                 wake_up_process(cct->task);
 743                 break;
 744         case CPU_DOWN_PREPARE:
 745                 ehca_gen_dbg("CPU: %x (CPU_DOWN_PREPARE)", cpu);
 746                 break;
 747         case CPU_DOWN_FAILED:
 748                 ehca_gen_dbg("CPU: %x (CPU_DOWN_FAILED)", cpu);
 749                 break;
 750         case CPU_DEAD:
 751                 ehca_gen_dbg("CPU: %x (CPU_DEAD)", cpu);
 752                 destroy_comp_task(pool, cpu);
 753                 take_over_work(pool, cpu);
 754                 break;
 755         }
 756
 757         return NOTIFY_OK;
 758 }
 759
 760 int ehca_create_comp_pool(void)
 761 {
 762         int cpu;
 763         struct task_struct *task;
 764
 765         if (!ehca_scaling_code)
 766                 return 0;
 767
 768         pool = kzalloc(sizeof(struct ehca_comp_pool), GFP_KERNEL);
 769         if (pool == NULL)
 770                 return -ENOMEM;
 771
 772         spin_lock_init(&pool->last_cpu_lock);
 773         pool->last_cpu = any_online_cpu(cpu_online_map);
 774
 775         pool->cpu_comp_tasks = alloc_percpu(struct ehca_cpu_comp_task);
 776         if (pool->cpu_comp_tasks == NULL) {
 777                 kfree(pool);
 778                 return -EINVAL;
 779         }
 780
 781         for_each_online_cpu(cpu) {
 782                 task = create_comp_task(pool, cpu);
 783                 if (task) {
 784                         kthread_bind(task, cpu);
 785                         wake_up_process(task);
 786                 }
 787         }
 788
 789         comp_pool_callback_nb.notifier_call = comp_pool_callback;
 790         comp_pool_callback_nb.priority =0;
 791         register_cpu_notifier(&comp_pool_callback_nb);
 792
 793         printk(KERN_INFO "eHCA scaling code enabled\n");
 794
 795         return 0;
 796 }
 797
 798 void ehca_destroy_comp_pool(void)
 799 {
 800         int i;
 801
 802         if (!ehca_scaling_code)
 803                 return;
 804
 805         unregister_cpu_notifier(&comp_pool_callback_nb);
 806
 807         for (i = 0; i < NR_CPUS; i++) {
 808                 if (cpu_online(i))
 809                         destroy_comp_task(pool, i);
 810         }
 811         free_percpu(pool->cpu_comp_tasks);
 812         kfree(pool);
 813 }