drivers/usb/typec/ucsi/ucsi.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * USB Type-C Connector System Software Interface driver
   4  *
   5  * Copyright (C) 2017, Intel Corporation
   6  * Author: Heikki Krogerus <heikki.krogerus@linux.intel.com>
   7  */
   8
   9 #include <linux/completion.h>
  10 #include <linux/property.h>
  11 #include <linux/device.h>
  12 #include <linux/module.h>
  13 #include <linux/delay.h>
  14 #include <linux/slab.h>
  15 #include <linux/usb/typec_dp.h>
  16
  17 #include "ucsi.h"
  18 #include "trace.h"
  19
  20 /*
  21  * UCSI_TIMEOUT_MS - PPM communication timeout
  22  *
  23  * Ideally we could use MIN_TIME_TO_RESPOND_WITH_BUSY (which is defined in UCSI
  24  * specification) here as reference, but unfortunately we can't. It is very
  25  * difficult to estimate the time it takes for the system to process the command
  26  * before it is actually passed to the PPM.
  27  */
  28 #define UCSI_TIMEOUT_MS         5000
  29
  30 /*
  31  * UCSI_SWAP_TIMEOUT_MS - Timeout for role swap requests
  32  *
  33  * 5 seconds is close to the time it takes for CapsCounter to reach 0, so even
  34  * if the PPM does not generate Connector Change events before that with
  35  * partners that do not support USB Power Delivery, this should still work.
  36  */
  37 #define UCSI_SWAP_TIMEOUT_MS    5000
  38
  39 static int ucsi_acknowledge_command(struct ucsi *ucsi)
  40 {
  41         u64 ctrl;
  42
  43         ctrl = UCSI_ACK_CC_CI;
  44         ctrl |= UCSI_ACK_COMMAND_COMPLETE;
  45
  46         return ucsi->ops->sync_write(ucsi, UCSI_CONTROL, &ctrl, sizeof(ctrl));
  47 }
  48
  49 static int ucsi_acknowledge_connector_change(struct ucsi *ucsi)
  50 {
  51         u64 ctrl;
  52
  53         ctrl = UCSI_ACK_CC_CI;
  54         ctrl |= UCSI_ACK_CONNECTOR_CHANGE;
  55
  56         return ucsi->ops->sync_write(ucsi, UCSI_CONTROL, &ctrl, sizeof(ctrl));
  57 }
  58
  59 static int ucsi_exec_command(struct ucsi *ucsi, u64 command);
  60
  61 static int ucsi_read_error(struct ucsi *ucsi)
  62 {
  63         u16 error;
  64         int ret;
  65
  66         /* Acknowledge the command that failed */
  67         ret = ucsi_acknowledge_command(ucsi);
  68         if (ret)
  69                 return ret;
  70
  71         ret = ucsi_exec_command(ucsi, UCSI_GET_ERROR_STATUS);
  72         if (ret < 0)
  73                 return ret;
  74
  75         ret = ucsi->ops->read(ucsi, UCSI_MESSAGE_IN, &error, sizeof(error));
  76         if (ret)
  77                 return ret;
  78
  79         switch (error) {
  80         case UCSI_ERROR_INCOMPATIBLE_PARTNER:
  81                 return -EOPNOTSUPP;
  82         case UCSI_ERROR_CC_COMMUNICATION_ERR:
  83                 return -ECOMM;
  84         case UCSI_ERROR_CONTRACT_NEGOTIATION_FAIL:
  85                 return -EPROTO;
  86         case UCSI_ERROR_DEAD_BATTERY:
  87                 dev_warn(ucsi->dev, "Dead battery condition!\n");
  88                 return -EPERM;
  89         case UCSI_ERROR_INVALID_CON_NUM:
  90         case UCSI_ERROR_UNREGONIZED_CMD:
  91         case UCSI_ERROR_INVALID_CMD_ARGUMENT:
  92                 dev_err(ucsi->dev, "possible UCSI driver bug %u\n", error);
  93                 return -EINVAL;
  94         case UCSI_ERROR_OVERCURRENT:
  95                 dev_warn(ucsi->dev, "Overcurrent condition\n");
  96                 break;
  97         case UCSI_ERROR_PARTNER_REJECTED_SWAP:
  98                 dev_warn(ucsi->dev, "Partner rejected swap\n");
  99                 break;
 100         case UCSI_ERROR_HARD_RESET:
 101                 dev_warn(ucsi->dev, "Hard reset occurred\n");
 102                 break;
 103         case UCSI_ERROR_PPM_POLICY_CONFLICT:
 104                 dev_warn(ucsi->dev, "PPM Policy conflict\n");
 105                 break;
 106         case UCSI_ERROR_SWAP_REJECTED:
 107                 dev_warn(ucsi->dev, "Swap rejected\n");
 108                 break;
 109         case UCSI_ERROR_UNDEFINED:
 110         default:
 111                 dev_err(ucsi->dev, "unknown error %u\n", error);
 112                 break;
 113         }
 114
 115         return -EIO;
 116 }
 117
 118 static int ucsi_exec_command(struct ucsi *ucsi, u64 cmd)
 119 {
 120         u32 cci;
 121         int ret;
 122
 123         ret = ucsi->ops->sync_write(ucsi, UCSI_CONTROL, &cmd, sizeof(cmd));
 124         if (ret)
 125                 return ret;
 126
 127         ret = ucsi->ops->read(ucsi, UCSI_CCI, &cci, sizeof(cci));
 128         if (ret)
 129                 return ret;
 130
 131         if (cci & UCSI_CCI_BUSY) {
 132                 ucsi->ops->async_write(ucsi, UCSI_CANCEL, NULL, 0);
 133                 return -EBUSY;
 134         }
 135
 136         if (!(cci & UCSI_CCI_COMMAND_COMPLETE))
 137                 return -EIO;
 138
 139         if (cci & UCSI_CCI_NOT_SUPPORTED)
 140                 return -EOPNOTSUPP;
 141
 142         if (cci & UCSI_CCI_ERROR) {
 143                 if (cmd == UCSI_GET_ERROR_STATUS)
 144                         return -EIO;
 145                 return ucsi_read_error(ucsi);
 146         }
 147
 148         return UCSI_CCI_LENGTH(cci);
 149 }
 150
 151 int ucsi_send_command(struct ucsi *ucsi, u64 command,
 152                       void *data, size_t size)
 153 {
 154         u8 length;
 155         int ret;
 156
 157         mutex_lock(&ucsi->ppm_lock);
 158
 159         ret = ucsi_exec_command(ucsi, command);
 160         if (ret < 0)
 161                 goto out;
 162
 163         length = ret;
 164
 165         if (data) {
 166                 ret = ucsi->ops->read(ucsi, UCSI_MESSAGE_IN, data, size);
 167                 if (ret)
 168                         goto out;
 169         }
 170
 171         ret = ucsi_acknowledge_command(ucsi);
 172         if (ret)
 173                 goto out;
 174
 175         ret = length;
 176 out:
 177         mutex_unlock(&ucsi->ppm_lock);
 178         return ret;
 179 }
 180 EXPORT_SYMBOL_GPL(ucsi_send_command);
 181
 182 int ucsi_resume(struct ucsi *ucsi)
 183 {
 184         u64 command;
 185
 186         /* Restore UCSI notification enable mask after system resume */
 187         command = UCSI_SET_NOTIFICATION_ENABLE | ucsi->ntfy;
 188
 189         return ucsi_send_command(ucsi, command, NULL, 0);
 190 }
 191 EXPORT_SYMBOL_GPL(ucsi_resume);
 192 /* -------------------------------------------------------------------------- */
 193
 194 struct ucsi_work {
 195         struct delayed_work work;
 196         unsigned long delay;
 197         unsigned int count;
 198         struct ucsi_connector *con;
 199         int (*cb)(struct ucsi_connector *);
 200 };
 201
 202 static void ucsi_poll_worker(struct work_struct *work)
 203 {
 204         struct ucsi_work *uwork = container_of(work, struct ucsi_work, work.work);
 205         struct ucsi_connector *con = uwork->con;
 206         int ret;
 207
 208         mutex_lock(&con->lock);
 209
 210         if (!con->partner) {
 211                 mutex_unlock(&con->lock);
 212                 kfree(uwork);
 213                 return;
 214         }
 215
 216         ret = uwork->cb(con);
 217
 218         if (uwork->count-- && (ret == -EBUSY || ret == -ETIMEDOUT))
 219                 queue_delayed_work(con->wq, &uwork->work, uwork->delay);
 220         else
 221                 kfree(uwork);
 222
 223         mutex_unlock(&con->lock);
 224 }
 225
 226 static int ucsi_partner_task(struct ucsi_connector *con,
 227                              int (*cb)(struct ucsi_connector *),
 228                              int retries, unsigned long delay)
 229 {
 230         struct ucsi_work *uwork;
 231
 232         if (!con->partner)
 233                 return 0;
 234
 235         uwork = kzalloc(sizeof(*uwork), GFP_KERNEL);
 236         if (!uwork)
 237                 return -ENOMEM;
 238
 239         INIT_DELAYED_WORK(&uwork->work, ucsi_poll_worker);
 240         uwork->count = retries;
 241         uwork->delay = delay;
 242         uwork->con = con;
 243         uwork->cb = cb;
 244
 245         queue_delayed_work(con->wq, &uwork->work, delay);
 246
 247         return 0;
 248 }
 249
 250 /* -------------------------------------------------------------------------- */
 251
 252 void ucsi_altmode_update_active(struct ucsi_connector *con)
 253 {
 254         const struct typec_altmode *altmode = NULL;
 255         u64 command;
 256         int ret;
 257         u8 cur;
 258         int i;
 259
 260         command = UCSI_GET_CURRENT_CAM | UCSI_CONNECTOR_NUMBER(con->num);
 261         ret = ucsi_send_command(con->ucsi, command, &cur, sizeof(cur));
 262         if (ret < 0) {
 263                 if (con->ucsi->version > 0x0100) {
 264                         dev_err(con->ucsi->dev,
 265                                 "GET_CURRENT_CAM command failed\n");
 266                         return;
 267                 }
 268                 cur = 0xff;
 269         }
 270
 271         if (cur < UCSI_MAX_ALTMODES)
 272                 altmode = typec_altmode_get_partner(con->port_altmode[cur]);
 273
 274         for (i = 0; con->partner_altmode[i]; i++)
 275                 typec_altmode_update_active(con->partner_altmode[i],
 276                                             con->partner_altmode[i] == altmode);
 277 }
 278
 279 static int ucsi_altmode_next_mode(struct typec_altmode **alt, u16 svid)
 280 {
 281         u8 mode = 1;
 282         int i;
 283
 284         for (i = 0; alt[i]; i++) {
 285                 if (i > MODE_DISCOVERY_MAX)
 286                         return -ERANGE;
 287
 288                 if (alt[i]->svid == svid)
 289                         mode++;
 290         }
 291
 292         return mode;
 293 }
 294
 295 static int ucsi_next_altmode(struct typec_altmode **alt)
 296 {
 297         int i = 0;
 298
 299         for (i = 0; i < UCSI_MAX_ALTMODES; i++)
 300                 if (!alt[i])
 301                         return i;
 302
 303         return -ENOENT;
 304 }
 305
 306 static int ucsi_register_altmode(struct ucsi_connector *con,
 307                                  struct typec_altmode_desc *desc,
 308                                  u8 recipient)
 309 {
 310         struct typec_altmode *alt;
 311         bool override;
 312         int ret;
 313         int i;
 314
 315         override = !!(con->ucsi->cap.features & UCSI_CAP_ALT_MODE_OVERRIDE);
 316
 317         switch (recipient) {
 318         case UCSI_RECIPIENT_CON:
 319                 i = ucsi_next_altmode(con->port_altmode);
 320                 if (i < 0) {
 321                         ret = i;
 322                         goto err;
 323                 }
 324
 325                 ret = ucsi_altmode_next_mode(con->port_altmode, desc->svid);
 326                 if (ret < 0)
 327                         return ret;
 328
 329                 desc->mode = ret;
 330
 331                 switch (desc->svid) {
 332                 case USB_TYPEC_DP_SID:
 333                         alt = ucsi_register_displayport(con, override, i, desc);
 334                         break;
 335                 case USB_TYPEC_NVIDIA_VLINK_SID:
 336                         if (desc->vdo == USB_TYPEC_NVIDIA_VLINK_DBG_VDO)
 337                                 alt = typec_port_register_altmode(con->port,
 338                                                                   desc);
 339                         else
 340                                 alt = ucsi_register_displayport(con, override,
 341                                                                 i, desc);
 342                         break;
 343                 default:
 344                         alt = typec_port_register_altmode(con->port, desc);
 345                         break;
 346                 }
 347
 348                 if (IS_ERR(alt)) {
 349                         ret = PTR_ERR(alt);
 350                         goto err;
 351                 }
 352
 353                 con->port_altmode[i] = alt;
 354                 break;
 355         case UCSI_RECIPIENT_SOP:
 356                 i = ucsi_next_altmode(con->partner_altmode);
 357                 if (i < 0) {
 358                         ret = i;
 359                         goto err;
 360                 }
 361
 362                 ret = ucsi_altmode_next_mode(con->partner_altmode, desc->svid);
 363                 if (ret < 0)
 364                         return ret;
 365
 366                 desc->mode = ret;
 367
 368                 alt = typec_partner_register_altmode(con->partner, desc);
 369                 if (IS_ERR(alt)) {
 370                         ret = PTR_ERR(alt);
 371                         goto err;
 372                 }
 373
 374                 con->partner_altmode[i] = alt;
 375                 break;
 376         default:
 377                 return -EINVAL;
 378         }
 379
 380         trace_ucsi_register_altmode(recipient, alt);
 381
 382         return 0;
 383
 384 err:
 385         dev_err(con->ucsi->dev, "failed to registers svid 0x%04x mode %d\n",
 386                 desc->svid, desc->mode);
 387
 388         return ret;
 389 }
 390
 391 static int
 392 ucsi_register_altmodes_nvidia(struct ucsi_connector *con, u8 recipient)
 393 {
 394         int max_altmodes = UCSI_MAX_ALTMODES;
 395         struct typec_altmode_desc desc;
 396         struct ucsi_altmode alt;
 397         struct ucsi_altmode orig[UCSI_MAX_ALTMODES];
 398         struct ucsi_altmode updated[UCSI_MAX_ALTMODES];
 399         struct ucsi *ucsi = con->ucsi;
 400         bool multi_dp = false;
 401         u64 command;
 402         int ret;
 403         int len;
 404         int i;
 405         int k = 0;
 406
 407         if (recipient == UCSI_RECIPIENT_CON)
 408                 max_altmodes = con->ucsi->cap.num_alt_modes;
 409
 410         memset(orig, 0, sizeof(orig));
 411         memset(updated, 0, sizeof(updated));
 412
 413         /* First get all the alternate modes */
 414         for (i = 0; i < max_altmodes; i++) {
 415                 memset(&alt, 0, sizeof(alt));
 416                 command = UCSI_GET_ALTERNATE_MODES;
 417                 command |= UCSI_GET_ALTMODE_RECIPIENT(recipient);
 418                 command |= UCSI_GET_ALTMODE_CONNECTOR_NUMBER(con->num);
 419                 command |= UCSI_GET_ALTMODE_OFFSET(i);
 420                 len = ucsi_send_command(con->ucsi, command, &alt, sizeof(alt));
 421                 /*
 422                  * We are collecting all altmodes first and then registering.
 423                  * Some type-C device will return zero length data beyond last
 424                  * alternate modes. We should not return if length is zero.
 425                  */
 426                 if (len < 0)
 427                         return len;
 428
 429                 /* We got all altmodes, now break out and register them */
 430                 if (!len || !alt.svid)
 431                         break;
 432
 433                 orig[k].mid = alt.mid;
 434                 orig[k].svid = alt.svid;
 435                 k++;
 436         }
 437         /*
 438          * Update the original altmode table as some ppms may report
 439          * multiple DP altmodes.
 440          */
 441         if (recipient == UCSI_RECIPIENT_CON)
 442                 multi_dp = ucsi->ops->update_altmodes(ucsi, orig, updated);
 443
 444         /* now register altmodes */
 445         for (i = 0; i < max_altmodes; i++) {
 446                 memset(&desc, 0, sizeof(desc));
 447                 if (multi_dp && recipient == UCSI_RECIPIENT_CON) {
 448                         desc.svid = updated[i].svid;
 449                         desc.vdo = updated[i].mid;
 450                 } else {
 451                         desc.svid = orig[i].svid;
 452                         desc.vdo = orig[i].mid;
 453                 }
 454                 desc.roles = TYPEC_PORT_DRD;
 455
 456                 if (!desc.svid)
 457                         return 0;
 458
 459                 ret = ucsi_register_altmode(con, &desc, recipient);
 460                 if (ret)
 461                         return ret;
 462         }
 463
 464         return 0;
 465 }
 466
 467 static int ucsi_register_altmodes(struct ucsi_connector *con, u8 recipient)
 468 {
 469         int max_altmodes = UCSI_MAX_ALTMODES;
 470         struct typec_altmode_desc desc;
 471         struct ucsi_altmode alt[2];
 472         u64 command;
 473         int num;
 474         int ret;
 475         int len;
 476         int j;
 477         int i;
 478
 479         if (!(con->ucsi->cap.features & UCSI_CAP_ALT_MODE_DETAILS))
 480                 return 0;
 481
 482         if (recipient == UCSI_RECIPIENT_SOP && con->partner_altmode[0])
 483                 return 0;
 484
 485         if (con->ucsi->ops->update_altmodes)
 486                 return ucsi_register_altmodes_nvidia(con, recipient);
 487
 488         if (recipient == UCSI_RECIPIENT_CON)
 489                 max_altmodes = con->ucsi->cap.num_alt_modes;
 490
 491         for (i = 0; i < max_altmodes;) {
 492                 memset(alt, 0, sizeof(alt));
 493                 command = UCSI_GET_ALTERNATE_MODES;
 494                 command |= UCSI_GET_ALTMODE_RECIPIENT(recipient);
 495                 command |= UCSI_GET_ALTMODE_CONNECTOR_NUMBER(con->num);
 496                 command |= UCSI_GET_ALTMODE_OFFSET(i);
 497                 len = ucsi_send_command(con->ucsi, command, alt, sizeof(alt));
 498                 if (len == -EBUSY)
 499                         continue;
 500                 if (len <= 0)
 501                         return len;
 502
 503                 /*
 504                  * This code is requesting one alt mode at a time, but some PPMs
 505                  * may still return two. If that happens both alt modes need be
 506                  * registered and the offset for the next alt mode has to be
 507                  * incremented.
 508                  */
 509                 num = len / sizeof(alt[0]);
 510                 i += num;
 511
 512                 for (j = 0; j < num; j++) {
 513                         if (!alt[j].svid)
 514                                 return 0;
 515
 516                         memset(&desc, 0, sizeof(desc));
 517                         desc.vdo = alt[j].mid;
 518                         desc.svid = alt[j].svid;
 519                         desc.roles = TYPEC_PORT_DRD;
 520
 521                         ret = ucsi_register_altmode(con, &desc, recipient);
 522                         if (ret)
 523                                 return ret;
 524                 }
 525         }
 526
 527         return 0;
 528 }
 529
 530 static void ucsi_unregister_altmodes(struct ucsi_connector *con, u8 recipient)
 531 {
 532         const struct typec_altmode *pdev;
 533         struct typec_altmode **adev;
 534         int i = 0;
 535
 536         switch (recipient) {
 537         case UCSI_RECIPIENT_CON:
 538                 adev = con->port_altmode;
 539                 break;
 540         case UCSI_RECIPIENT_SOP:
 541                 adev = con->partner_altmode;
 542                 break;
 543         default:
 544                 return;
 545         }
 546
 547         while (adev[i]) {
 548                 if (recipient == UCSI_RECIPIENT_SOP &&
 549                     (adev[i]->svid == USB_TYPEC_DP_SID ||
 550                         (adev[i]->svid == USB_TYPEC_NVIDIA_VLINK_SID &&
 551                         adev[i]->vdo != USB_TYPEC_NVIDIA_VLINK_DBG_VDO))) {
 552                         pdev = typec_altmode_get_partner(adev[i]);
 553                         ucsi_displayport_remove_partner((void *)pdev);
 554                 }
 555                 typec_unregister_altmode(adev[i]);
 556                 adev[i++] = NULL;
 557         }
 558 }
 559
 560 static int ucsi_get_pdos(struct ucsi_connector *con, int is_partner,
 561                          u32 *pdos, int offset, int num_pdos)
 562 {
 563         struct ucsi *ucsi = con->ucsi;
 564         u64 command;
 565         int ret;
 566
 567         command = UCSI_COMMAND(UCSI_GET_PDOS) | UCSI_CONNECTOR_NUMBER(con->num);
 568         command |= UCSI_GET_PDOS_PARTNER_PDO(is_partner);
 569         command |= UCSI_GET_PDOS_PDO_OFFSET(offset);
 570         command |= UCSI_GET_PDOS_NUM_PDOS(num_pdos - 1);
 571         command |= UCSI_GET_PDOS_SRC_PDOS;
 572         ret = ucsi_send_command(ucsi, command, pdos + offset,
 573                                 num_pdos * sizeof(u32));
 574         if (ret < 0)
 575                 dev_err(ucsi->dev, "UCSI_GET_PDOS failed (%d)\n", ret);
 576         if (ret == 0 && offset == 0)
 577                 dev_warn(ucsi->dev, "UCSI_GET_PDOS returned 0 bytes\n");
 578
 579         return ret;
 580 }
 581
 582 static void ucsi_get_src_pdos(struct ucsi_connector *con, int is_partner)
 583 {
 584         int ret;
 585
 586         /* UCSI max payload means only getting at most 4 PDOs at a time */
 587         ret = ucsi_get_pdos(con, 1, con->src_pdos, 0, UCSI_MAX_PDOS);
 588         if (ret < 0)
 589                 return;
 590
 591         con->num_pdos = ret / sizeof(u32); /* number of bytes to 32-bit PDOs */
 592         if (con->num_pdos < UCSI_MAX_PDOS)
 593                 return;
 594
 595         /* get the remaining PDOs, if any */
 596         ret = ucsi_get_pdos(con, 1, con->src_pdos, UCSI_MAX_PDOS,
 597                             PDO_MAX_OBJECTS - UCSI_MAX_PDOS);
 598         if (ret < 0)
 599                 return;
 600
 601         con->num_pdos += ret / sizeof(u32);
 602 }
 603
 604 static int ucsi_check_altmodes(struct ucsi_connector *con)
 605 {
 606         int ret;
 607
 608         ret = ucsi_register_altmodes(con, UCSI_RECIPIENT_SOP);
 609         if (ret && ret != -ETIMEDOUT)
 610                 dev_err(con->ucsi->dev,
 611                         "con%d: failed to register partner alt modes (%d)\n",
 612                         con->num, ret);
 613
 614         /* Ignoring the errors in this case. */
 615         if (con->partner_altmode[0]) {
 616                 ucsi_altmode_update_active(con);
 617                 return 0;
 618         }
 619
 620         return ret;
 621 }
 622
 623 static void ucsi_pwr_opmode_change(struct ucsi_connector *con)
 624 {
 625         switch (UCSI_CONSTAT_PWR_OPMODE(con->status.flags)) {
 626         case UCSI_CONSTAT_PWR_OPMODE_PD:
 627                 con->rdo = con->status.request_data_obj;
 628                 typec_set_pwr_opmode(con->port, TYPEC_PWR_MODE_PD);
 629                 ucsi_get_src_pdos(con, 1);
 630                 break;
 631         case UCSI_CONSTAT_PWR_OPMODE_TYPEC1_5:
 632                 con->rdo = 0;
 633                 typec_set_pwr_opmode(con->port, TYPEC_PWR_MODE_1_5A);
 634                 break;
 635         case UCSI_CONSTAT_PWR_OPMODE_TYPEC3_0:
 636                 con->rdo = 0;
 637                 typec_set_pwr_opmode(con->port, TYPEC_PWR_MODE_3_0A);
 638                 break;
 639         default:
 640                 con->rdo = 0;
 641                 typec_set_pwr_opmode(con->port, TYPEC_PWR_MODE_USB);
 642                 break;
 643         }
 644 }
 645
 646 static int ucsi_register_partner(struct ucsi_connector *con)
 647 {
 648         u8 pwr_opmode = UCSI_CONSTAT_PWR_OPMODE(con->status.flags);
 649         struct typec_partner_desc desc;
 650         struct typec_partner *partner;
 651
 652         if (con->partner)
 653                 return 0;
 654
 655         memset(&desc, 0, sizeof(desc));
 656
 657         switch (UCSI_CONSTAT_PARTNER_TYPE(con->status.flags)) {
 658         case UCSI_CONSTAT_PARTNER_TYPE_DEBUG:
 659                 desc.accessory = TYPEC_ACCESSORY_DEBUG;
 660                 break;
 661         case UCSI_CONSTAT_PARTNER_TYPE_AUDIO:
 662                 desc.accessory = TYPEC_ACCESSORY_AUDIO;
 663                 break;
 664         default:
 665                 break;
 666         }
 667
 668         desc.usb_pd = pwr_opmode == UCSI_CONSTAT_PWR_OPMODE_PD;
 669
 670         partner = typec_register_partner(con->port, &desc);
 671         if (IS_ERR(partner)) {
 672                 dev_err(con->ucsi->dev,
 673                         "con%d: failed to register partner (%ld)\n", con->num,
 674                         PTR_ERR(partner));
 675                 return PTR_ERR(partner);
 676         }
 677
 678         con->partner = partner;
 679
 680         return 0;
 681 }
 682
 683 static void ucsi_unregister_partner(struct ucsi_connector *con)
 684 {
 685         if (!con->partner)
 686                 return;
 687
 688         ucsi_unregister_altmodes(con, UCSI_RECIPIENT_SOP);
 689         typec_unregister_partner(con->partner);
 690         con->partner = NULL;
 691 }
 692
 693 static void ucsi_partner_change(struct ucsi_connector *con)
 694 {
 695         enum usb_role u_role = USB_ROLE_NONE;
 696         int ret;
 697
 698         if (!con->partner)
 699                 return;
 700
 701         switch (UCSI_CONSTAT_PARTNER_TYPE(con->status.flags)) {
 702         case UCSI_CONSTAT_PARTNER_TYPE_UFP:
 703         case UCSI_CONSTAT_PARTNER_TYPE_CABLE_AND_UFP:
 704                 u_role = USB_ROLE_HOST;
 705                 fallthrough;
 706         case UCSI_CONSTAT_PARTNER_TYPE_CABLE:
 707                 typec_set_data_role(con->port, TYPEC_HOST);
 708                 break;
 709         case UCSI_CONSTAT_PARTNER_TYPE_DFP:
 710                 u_role = USB_ROLE_DEVICE;
 711                 typec_set_data_role(con->port, TYPEC_DEVICE);
 712                 break;
 713         default:
 714                 break;
 715         }
 716
 717         /* Complete pending data role swap */
 718         if (!completion_done(&con->complete))
 719                 complete(&con->complete);
 720
 721         /* Only notify USB controller if partner supports USB data */
 722         if (!(UCSI_CONSTAT_PARTNER_FLAGS(con->status.flags) & UCSI_CONSTAT_PARTNER_FLAG_USB))
 723                 u_role = USB_ROLE_NONE;
 724
 725         ret = usb_role_switch_set_role(con->usb_role_sw, u_role);
 726         if (ret)
 727                 dev_err(con->ucsi->dev, "con:%d: failed to set usb role:%d\n",
 728                         con->num, u_role);
 729
 730         ucsi_partner_task(con, ucsi_check_altmodes, 30, 0);
 731 }
 732
 733 static void ucsi_handle_connector_change(struct work_struct *work)
 734 {
 735         struct ucsi_connector *con = container_of(work, struct ucsi_connector,
 736                                                   work);
 737         struct ucsi *ucsi = con->ucsi;
 738         struct ucsi_connector_status pre_ack_status;
 739         struct ucsi_connector_status post_ack_status;
 740         enum typec_role role;
 741         enum usb_role u_role = USB_ROLE_NONE;
 742         u16 inferred_changes;
 743         u16 changed_flags;
 744         u64 command;
 745         int ret;
 746
 747         mutex_lock(&con->lock);
 748
 749         /*
 750          * Some/many PPMs have an issue where all fields in the change bitfield
 751          * are cleared when an ACK is send. This will causes any change
 752          * between GET_CONNECTOR_STATUS and ACK to be lost.
 753          *
 754          * We work around this by re-fetching the connector status afterwards.
 755          * We then infer any changes that we see have happened but that may not
 756          * be represented in the change bitfield.
 757          *
 758          * Also, even though we don't need to know the currently supported alt
 759          * modes, we run the GET_CAM_SUPPORTED command to ensure the PPM does
 760          * not get stuck in case it assumes we do.
 761          * Always do this, rather than relying on UCSI_CONSTAT_CAM_CHANGE to be
 762          * set in the change bitfield.
 763          *
 764          * We end up with the following actions:
 765          *  1. UCSI_GET_CONNECTOR_STATUS, store result, update unprocessed_changes
 766          *  2. UCSI_GET_CAM_SUPPORTED, discard result
 767          *  3. ACK connector change
 768          *  4. UCSI_GET_CONNECTOR_STATUS, store result
 769          *  5. Infere lost changes by comparing UCSI_GET_CONNECTOR_STATUS results
 770          *  6. If PPM reported a new change, then restart in order to ACK
 771          *  7. Process everything as usual.
 772          *
 773          * We may end up seeing a change twice, but we can only miss extremely
 774          * short transitional changes.
 775          */
 776
 777         /* 1. First UCSI_GET_CONNECTOR_STATUS */
 778         command = UCSI_GET_CONNECTOR_STATUS | UCSI_CONNECTOR_NUMBER(con->num);
 779         ret = ucsi_send_command(ucsi, command, &pre_ack_status,
 780                                 sizeof(pre_ack_status));
 781         if (ret < 0) {
 782                 dev_err(ucsi->dev, "%s: GET_CONNECTOR_STATUS failed (%d)\n",
 783                         __func__, ret);
 784                 goto out_unlock;
 785         }
 786         con->unprocessed_changes |= pre_ack_status.change;
 787
 788         /* 2. Run UCSI_GET_CAM_SUPPORTED and discard the result. */
 789         command = UCSI_GET_CAM_SUPPORTED;
 790         command |= UCSI_CONNECTOR_NUMBER(con->num);
 791         ucsi_send_command(con->ucsi, command, NULL, 0);
 792
 793         /* 3. ACK connector change */
 794         ret = ucsi_acknowledge_connector_change(ucsi);
 795         clear_bit(EVENT_PENDING, &ucsi->flags);
 796         if (ret) {
 797                 dev_err(ucsi->dev, "%s: ACK failed (%d)", __func__, ret);
 798                 goto out_unlock;
 799         }
 800
 801         /* 4. Second UCSI_GET_CONNECTOR_STATUS */
 802         command = UCSI_GET_CONNECTOR_STATUS | UCSI_CONNECTOR_NUMBER(con->num);
 803         ret = ucsi_send_command(ucsi, command, &post_ack_status,
 804                                 sizeof(post_ack_status));
 805         if (ret < 0) {
 806                 dev_err(ucsi->dev, "%s: GET_CONNECTOR_STATUS failed (%d)\n",
 807                         __func__, ret);
 808                 goto out_unlock;
 809         }
 810
 811         /* 5. Inferre any missing changes */
 812         changed_flags = pre_ack_status.flags ^ post_ack_status.flags;
 813         inferred_changes = 0;
 814         if (UCSI_CONSTAT_PWR_OPMODE(changed_flags) != 0)
 815                 inferred_changes |= UCSI_CONSTAT_POWER_OPMODE_CHANGE;
 816
 817         if (changed_flags & UCSI_CONSTAT_CONNECTED)
 818                 inferred_changes |= UCSI_CONSTAT_CONNECT_CHANGE;
 819
 820         if (changed_flags & UCSI_CONSTAT_PWR_DIR)
 821                 inferred_changes |= UCSI_CONSTAT_POWER_DIR_CHANGE;
 822
 823         if (UCSI_CONSTAT_PARTNER_FLAGS(changed_flags) != 0)
 824                 inferred_changes |= UCSI_CONSTAT_PARTNER_CHANGE;
 825
 826         if (UCSI_CONSTAT_PARTNER_TYPE(changed_flags) != 0)
 827                 inferred_changes |= UCSI_CONSTAT_PARTNER_CHANGE;
 828
 829         /* Mask out anything that was correctly notified in the later call. */
 830         inferred_changes &= ~post_ack_status.change;
 831         if (inferred_changes)
 832                 dev_dbg(ucsi->dev, "%s: Inferred changes that would have been lost: 0x%04x\n",
 833                         __func__, inferred_changes);
 834
 835         con->unprocessed_changes |= inferred_changes;
 836
 837         /* 6. If PPM reported a new change, then restart in order to ACK */
 838         if (post_ack_status.change)
 839                 goto out_unlock;
 840
 841         /* 7. Continue as if nothing happened */
 842         con->status = post_ack_status;
 843         con->status.change = con->unprocessed_changes;
 844         con->unprocessed_changes = 0;
 845
 846         role = !!(con->status.flags & UCSI_CONSTAT_PWR_DIR);
 847
 848         if (con->status.change & UCSI_CONSTAT_POWER_OPMODE_CHANGE ||
 849             con->status.change & UCSI_CONSTAT_POWER_LEVEL_CHANGE) {
 850                 ucsi_pwr_opmode_change(con);
 851                 ucsi_port_psy_changed(con);
 852         }
 853
 854         if (con->status.change & UCSI_CONSTAT_POWER_DIR_CHANGE) {
 855                 typec_set_pwr_role(con->port, role);
 856
 857                 /* Complete pending power role swap */
 858                 if (!completion_done(&con->complete))
 859                         complete(&con->complete);
 860         }
 861
 862         if (con->status.change & UCSI_CONSTAT_CONNECT_CHANGE) {
 863                 typec_set_pwr_role(con->port, role);
 864
 865                 switch (UCSI_CONSTAT_PARTNER_TYPE(con->status.flags)) {
 866                 case UCSI_CONSTAT_PARTNER_TYPE_UFP:
 867                 case UCSI_CONSTAT_PARTNER_TYPE_CABLE_AND_UFP:
 868                         u_role = USB_ROLE_HOST;
 869                         fallthrough;
 870                 case UCSI_CONSTAT_PARTNER_TYPE_CABLE:
 871                         typec_set_data_role(con->port, TYPEC_HOST);
 872                         break;
 873                 case UCSI_CONSTAT_PARTNER_TYPE_DFP:
 874                         u_role = USB_ROLE_DEVICE;
 875                         typec_set_data_role(con->port, TYPEC_DEVICE);
 876                         break;
 877                 default:
 878                         break;
 879                 }
 880
 881                 if (con->status.flags & UCSI_CONSTAT_CONNECTED)
 882                         ucsi_register_partner(con);
 883                 else
 884                         ucsi_unregister_partner(con);
 885
 886                 ucsi_port_psy_changed(con);
 887
 888                 /* Only notify USB controller if partner supports USB data */
 889                 if (!(UCSI_CONSTAT_PARTNER_FLAGS(con->status.flags) &
 890                                 UCSI_CONSTAT_PARTNER_FLAG_USB))
 891                         u_role = USB_ROLE_NONE;
 892
 893                 ret = usb_role_switch_set_role(con->usb_role_sw, u_role);
 894                 if (ret)
 895                         dev_err(ucsi->dev, "con:%d: failed to set usb role:%d\n",
 896                                 con->num, u_role);
 897         }
 898
 899         if (con->status.change & UCSI_CONSTAT_PARTNER_CHANGE)
 900                 ucsi_partner_change(con);
 901
 902         trace_ucsi_connector_change(con->num, &con->status);
 903
 904 out_unlock:
 905         if (test_and_clear_bit(EVENT_PENDING, &ucsi->flags)) {
 906                 schedule_work(&con->work);
 907                 mutex_unlock(&con->lock);
 908                 return;
 909         }
 910
 911         clear_bit(EVENT_PROCESSING, &ucsi->flags);
 912         mutex_unlock(&con->lock);
 913 }
 914
 915 /**
 916  * ucsi_connector_change - Process Connector Change Event
 917  * @ucsi: UCSI Interface
 918  * @num: Connector number
 919  */
 920 void ucsi_connector_change(struct ucsi *ucsi, u8 num)
 921 {
 922         struct ucsi_connector *con = &ucsi->connector[num - 1];
 923
 924         if (!(ucsi->ntfy & UCSI_ENABLE_NTFY_CONNECTOR_CHANGE)) {
 925                 dev_dbg(ucsi->dev, "Bogus connector change event\n");
 926                 return;
 927         }
 928
 929         set_bit(EVENT_PENDING, &ucsi->flags);
 930
 931         if (!test_and_set_bit(EVENT_PROCESSING, &ucsi->flags))
 932                 schedule_work(&con->work);
 933 }
 934 EXPORT_SYMBOL_GPL(ucsi_connector_change);
 935
 936 /* -------------------------------------------------------------------------- */
 937
 938 static int ucsi_reset_connector(struct ucsi_connector *con, bool hard)
 939 {
 940         u64 command;
 941
 942         command = UCSI_CONNECTOR_RESET | UCSI_CONNECTOR_NUMBER(con->num);
 943         command |= hard ? UCSI_CONNECTOR_RESET_HARD : 0;
 944
 945         return ucsi_send_command(con->ucsi, command, NULL, 0);
 946 }
 947
 948 static int ucsi_reset_ppm(struct ucsi *ucsi)
 949 {
 950         u64 command = UCSI_PPM_RESET;
 951         unsigned long tmo;
 952         u32 cci;
 953         int ret;
 954
 955         mutex_lock(&ucsi->ppm_lock);
 956
 957         ret = ucsi->ops->async_write(ucsi, UCSI_CONTROL, &command,
 958                                      sizeof(command));
 959         if (ret < 0)
 960                 goto out;
 961
 962         tmo = jiffies + msecs_to_jiffies(UCSI_TIMEOUT_MS);
 963
 964         do {
 965                 if (time_is_before_jiffies(tmo)) {
 966                         ret = -ETIMEDOUT;
 967                         goto out;
 968                 }
 969
 970                 ret = ucsi->ops->read(ucsi, UCSI_CCI, &cci, sizeof(cci));
 971                 if (ret)
 972                         goto out;
 973
 974                 /* If the PPM is still doing something else, reset it again. */
 975                 if (cci & ~UCSI_CCI_RESET_COMPLETE) {
 976                         ret = ucsi->ops->async_write(ucsi, UCSI_CONTROL,
 977                                                      &command,
 978                                                      sizeof(command));
 979                         if (ret < 0)
 980                                 goto out;
 981                 }
 982
 983                 msleep(20);
 984         } while (!(cci & UCSI_CCI_RESET_COMPLETE));
 985
 986 out:
 987         mutex_unlock(&ucsi->ppm_lock);
 988         return ret;
 989 }
 990
 991 static int ucsi_role_cmd(struct ucsi_connector *con, u64 command)
 992 {
 993         int ret;
 994
 995         ret = ucsi_send_command(con->ucsi, command, NULL, 0);
 996         if (ret == -ETIMEDOUT) {
 997                 u64 c;
 998
 999                 /* PPM most likely stopped responding. Resetting everything. */
1000                 ucsi_reset_ppm(con->ucsi);
1001
1002                 c = UCSI_SET_NOTIFICATION_ENABLE | con->ucsi->ntfy;
1003                 ucsi_send_command(con->ucsi, c, NULL, 0);
1004
1005                 ucsi_reset_connector(con, true);
1006         }
1007
1008         return ret;
1009 }
1010
1011 static int ucsi_dr_swap(struct typec_port *port, enum typec_data_role role)
1012 {
1013         struct ucsi_connector *con = typec_get_drvdata(port);
1014         u8 partner_type;
1015         u64 command;
1016         int ret = 0;
1017
1018         mutex_lock(&con->lock);
1019
1020         if (!con->partner) {
1021                 ret = -ENOTCONN;
1022                 goto out_unlock;
1023         }
1024
1025         partner_type = UCSI_CONSTAT_PARTNER_TYPE(con->status.flags);
1026         if ((partner_type == UCSI_CONSTAT_PARTNER_TYPE_DFP &&
1027              role == TYPEC_DEVICE) ||
1028             (partner_type == UCSI_CONSTAT_PARTNER_TYPE_UFP &&
1029              role == TYPEC_HOST))
1030                 goto out_unlock;
1031
1032         command = UCSI_SET_UOR | UCSI_CONNECTOR_NUMBER(con->num);
1033         command |= UCSI_SET_UOR_ROLE(role);
1034         command |= UCSI_SET_UOR_ACCEPT_ROLE_SWAPS;
1035         ret = ucsi_role_cmd(con, command);
1036         if (ret < 0)
1037                 goto out_unlock;
1038
1039         if (!wait_for_completion_timeout(&con->complete,
1040                                         msecs_to_jiffies(UCSI_SWAP_TIMEOUT_MS)))
1041                 ret = -ETIMEDOUT;
1042
1043 out_unlock:
1044         mutex_unlock(&con->lock);
1045
1046         return ret < 0 ? ret : 0;
1047 }
1048
1049 static int ucsi_pr_swap(struct typec_port *port, enum typec_role role)
1050 {
1051         struct ucsi_connector *con = typec_get_drvdata(port);
1052         enum typec_role cur_role;
1053         u64 command;
1054         int ret = 0;
1055
1056         mutex_lock(&con->lock);
1057
1058         if (!con->partner) {
1059                 ret = -ENOTCONN;
1060                 goto out_unlock;
1061         }
1062
1063         cur_role = !!(con->status.flags & UCSI_CONSTAT_PWR_DIR);
1064
1065         if (cur_role == role)
1066                 goto out_unlock;
1067
1068         command = UCSI_SET_PDR | UCSI_CONNECTOR_NUMBER(con->num);
1069         command |= UCSI_SET_PDR_ROLE(role);
1070         command |= UCSI_SET_PDR_ACCEPT_ROLE_SWAPS;
1071         ret = ucsi_role_cmd(con, command);
1072         if (ret < 0)
1073                 goto out_unlock;
1074
1075         if (!wait_for_completion_timeout(&con->complete,
1076                                 msecs_to_jiffies(UCSI_SWAP_TIMEOUT_MS))) {
1077                 ret = -ETIMEDOUT;
1078                 goto out_unlock;
1079         }
1080
1081         /* Something has gone wrong while swapping the role */
1082         if (UCSI_CONSTAT_PWR_OPMODE(con->status.flags) !=
1083             UCSI_CONSTAT_PWR_OPMODE_PD) {
1084                 ucsi_reset_connector(con, true);
1085                 ret = -EPROTO;
1086         }
1087
1088 out_unlock:
1089         mutex_unlock(&con->lock);
1090
1091         return ret;
1092 }
1093
1094 static const struct typec_operations ucsi_ops = {
1095         .dr_set = ucsi_dr_swap,
1096         .pr_set = ucsi_pr_swap
1097 };
1098
1099 /* Caller must call fwnode_handle_put() after use */
1100 static struct fwnode_handle *ucsi_find_fwnode(struct ucsi_connector *con)
1101 {
1102         struct fwnode_handle *fwnode;
1103         int i = 1;
1104
1105         device_for_each_child_node(con->ucsi->dev, fwnode)
1106                 if (i++ == con->num)
1107                         return fwnode;
1108         return NULL;
1109 }
1110
1111 static int ucsi_register_port(struct ucsi *ucsi, int index)
1112 {
1113         struct ucsi_connector *con = &ucsi->connector[index];
1114         struct typec_capability *cap = &con->typec_cap;
1115         enum typec_accessory *accessory = cap->accessory;
1116         enum usb_role u_role = USB_ROLE_NONE;
1117         u64 command;
1118         char *name;
1119         int ret;
1120
1121         name = kasprintf(GFP_KERNEL, "%s-con%d", dev_name(ucsi->dev), con->num);
1122         if (!name)
1123                 return -ENOMEM;
1124
1125         con->wq = create_singlethread_workqueue(name);
1126         kfree(name);
1127         if (!con->wq)
1128                 return -ENOMEM;
1129
1130         INIT_WORK(&con->work, ucsi_handle_connector_change);
1131         init_completion(&con->complete);
1132         mutex_init(&con->lock);
1133         con->num = index + 1;
1134         con->ucsi = ucsi;
1135
1136         /* Delay other interactions with the con until registration is complete */
1137         mutex_lock(&con->lock);
1138
1139         /* Get connector capability */
1140         command = UCSI_GET_CONNECTOR_CAPABILITY;
1141         command |= UCSI_CONNECTOR_NUMBER(con->num);
1142         ret = ucsi_send_command(ucsi, command, &con->cap, sizeof(con->cap));
1143         if (ret < 0)
1144                 goto out_unlock;
1145
1146         if (con->cap.op_mode & UCSI_CONCAP_OPMODE_DRP)
1147                 cap->data = TYPEC_PORT_DRD;
1148         else if (con->cap.op_mode & UCSI_CONCAP_OPMODE_DFP)
1149                 cap->data = TYPEC_PORT_DFP;
1150         else if (con->cap.op_mode & UCSI_CONCAP_OPMODE_UFP)
1151                 cap->data = TYPEC_PORT_UFP;
1152
1153         if ((con->cap.flags & UCSI_CONCAP_FLAG_PROVIDER) &&
1154             (con->cap.flags & UCSI_CONCAP_FLAG_CONSUMER))
1155                 cap->type = TYPEC_PORT_DRP;
1156         else if (con->cap.flags & UCSI_CONCAP_FLAG_PROVIDER)
1157                 cap->type = TYPEC_PORT_SRC;
1158         else if (con->cap.flags & UCSI_CONCAP_FLAG_CONSUMER)
1159                 cap->type = TYPEC_PORT_SNK;
1160
1161         cap->revision = ucsi->cap.typec_version;
1162         cap->pd_revision = ucsi->cap.pd_version;
1163         cap->svdm_version = SVDM_VER_2_0;
1164         cap->prefer_role = TYPEC_NO_PREFERRED_ROLE;
1165
1166         if (con->cap.op_mode & UCSI_CONCAP_OPMODE_AUDIO_ACCESSORY)
1167                 *accessory++ = TYPEC_ACCESSORY_AUDIO;
1168         if (con->cap.op_mode & UCSI_CONCAP_OPMODE_DEBUG_ACCESSORY)
1169                 *accessory = TYPEC_ACCESSORY_DEBUG;
1170
1171         cap->fwnode = ucsi_find_fwnode(con);
1172         cap->driver_data = con;
1173         cap->ops = &ucsi_ops;
1174
1175         ret = ucsi_register_port_psy(con);
1176         if (ret)
1177                 goto out;
1178
1179         /* Register the connector */
1180         con->port = typec_register_port(ucsi->dev, cap);
1181         if (IS_ERR(con->port)) {
1182                 ret = PTR_ERR(con->port);
1183                 goto out;
1184         }
1185
1186         /* Alternate modes */
1187         ret = ucsi_register_altmodes(con, UCSI_RECIPIENT_CON);
1188         if (ret) {
1189                 dev_err(ucsi->dev, "con%d: failed to register alt modes\n",
1190                         con->num);
1191                 goto out;
1192         }
1193
1194         /* Get the status */
1195         command = UCSI_GET_CONNECTOR_STATUS | UCSI_CONNECTOR_NUMBER(con->num);
1196         ret = ucsi_send_command(ucsi, command, &con->status, sizeof(con->status));
1197         if (ret < 0) {
1198                 dev_err(ucsi->dev, "con%d: failed to get status\n", con->num);
1199                 ret = 0;
1200                 goto out;
1201         }
1202         ret = 0; /* ucsi_send_command() returns length on success */
1203
1204         switch (UCSI_CONSTAT_PARTNER_TYPE(con->status.flags)) {
1205         case UCSI_CONSTAT_PARTNER_TYPE_UFP:
1206         case UCSI_CONSTAT_PARTNER_TYPE_CABLE_AND_UFP:
1207                 u_role = USB_ROLE_HOST;
1208                 fallthrough;
1209         case UCSI_CONSTAT_PARTNER_TYPE_CABLE:
1210                 typec_set_data_role(con->port, TYPEC_HOST);
1211                 break;
1212         case UCSI_CONSTAT_PARTNER_TYPE_DFP:
1213                 u_role = USB_ROLE_DEVICE;
1214                 typec_set_data_role(con->port, TYPEC_DEVICE);
1215                 break;
1216         default:
1217                 break;
1218         }
1219
1220         /* Check if there is already something connected */
1221         if (con->status.flags & UCSI_CONSTAT_CONNECTED) {
1222                 typec_set_pwr_role(con->port,
1223                                   !!(con->status.flags & UCSI_CONSTAT_PWR_DIR));
1224                 ucsi_pwr_opmode_change(con);
1225                 ucsi_register_partner(con);
1226                 ucsi_port_psy_changed(con);
1227         }
1228
1229         con->usb_role_sw = fwnode_usb_role_switch_get(cap->fwnode);
1230         if (IS_ERR(con->usb_role_sw)) {
1231                 dev_err(ucsi->dev, "con%d: failed to get usb role switch\n",
1232                         con->num);
1233                 con->usb_role_sw = NULL;
1234         }
1235
1236         /* Only notify USB controller if partner supports USB data */
1237         if (!(UCSI_CONSTAT_PARTNER_FLAGS(con->status.flags) & UCSI_CONSTAT_PARTNER_FLAG_USB))
1238                 u_role = USB_ROLE_NONE;
1239
1240         ret = usb_role_switch_set_role(con->usb_role_sw, u_role);
1241         if (ret) {
1242                 dev_err(ucsi->dev, "con:%d: failed to set usb role:%d\n",
1243                         con->num, u_role);
1244                 ret = 0;
1245         }
1246
1247         if (con->partner)
1248                 ucsi_check_altmodes(con);
1249
1250         trace_ucsi_register_port(con->num, &con->status);
1251
1252 out:
1253         fwnode_handle_put(cap->fwnode);
1254 out_unlock:
1255         mutex_unlock(&con->lock);
1256
1257         if (ret && con->wq) {
1258                 destroy_workqueue(con->wq);
1259                 con->wq = NULL;
1260         }
1261
1262         return ret;
1263 }
1264
1265 /**
1266  * ucsi_init - Initialize UCSI interface
1267  * @ucsi: UCSI to be initialized
1268  *
1269  * Registers all ports @ucsi has and enables all notification events.
1270  */
1271 static int ucsi_init(struct ucsi *ucsi)
1272 {
1273         struct ucsi_connector *con;
1274         u64 command;
1275         int ret;
1276         int i;
1277
1278         /* Reset the PPM */
1279         ret = ucsi_reset_ppm(ucsi);
1280         if (ret) {
1281                 dev_err(ucsi->dev, "failed to reset PPM!\n");
1282                 goto err;
1283         }
1284
1285         /* Enable basic notifications */
1286         ucsi->ntfy = UCSI_ENABLE_NTFY_CMD_COMPLETE | UCSI_ENABLE_NTFY_ERROR;
1287         command = UCSI_SET_NOTIFICATION_ENABLE | ucsi->ntfy;
1288         ret = ucsi_send_command(ucsi, command, NULL, 0);
1289         if (ret < 0)
1290                 goto err_reset;
1291
1292         /* Get PPM capabilities */
1293         command = UCSI_GET_CAPABILITY;
1294         ret = ucsi_send_command(ucsi, command, &ucsi->cap, sizeof(ucsi->cap));
1295         if (ret < 0)
1296                 goto err_reset;
1297
1298         if (!ucsi->cap.num_connectors) {
1299                 ret = -ENODEV;
1300                 goto err_reset;
1301         }
1302
1303         /* Allocate the connectors. Released in ucsi_unregister() */
1304         ucsi->connector = kcalloc(ucsi->cap.num_connectors + 1,
1305                                   sizeof(*ucsi->connector), GFP_KERNEL);
1306         if (!ucsi->connector) {
1307                 ret = -ENOMEM;
1308                 goto err_reset;
1309         }
1310
1311         /* Register all connectors */
1312         for (i = 0; i < ucsi->cap.num_connectors; i++) {
1313                 ret = ucsi_register_port(ucsi, i);
1314                 if (ret)
1315                         goto err_unregister;
1316         }
1317
1318         /* Enable all notifications */
1319         ucsi->ntfy = UCSI_ENABLE_NTFY_ALL;
1320         command = UCSI_SET_NOTIFICATION_ENABLE | ucsi->ntfy;
1321         ret = ucsi_send_command(ucsi, command, NULL, 0);
1322         if (ret < 0)
1323                 goto err_unregister;
1324
1325         return 0;
1326
1327 err_unregister:
1328         for (con = ucsi->connector; con->port; con++) {
1329                 ucsi_unregister_partner(con);
1330                 ucsi_unregister_altmodes(con, UCSI_RECIPIENT_CON);
1331                 ucsi_unregister_port_psy(con);
1332                 if (con->wq)
1333                         destroy_workqueue(con->wq);
1334                 typec_unregister_port(con->port);
1335                 con->port = NULL;
1336         }
1337
1338 err_reset:
1339         memset(&ucsi->cap, 0, sizeof(ucsi->cap));
1340         ucsi_reset_ppm(ucsi);
1341 err:
1342         return ret;
1343 }
1344
1345 static void ucsi_init_work(struct work_struct *work)
1346 {
1347         struct ucsi *ucsi = container_of(work, struct ucsi, work);
1348         int ret;
1349
1350         ret = ucsi_init(ucsi);
1351         if (ret)
1352                 dev_err(ucsi->dev, "PPM init failed (%d)\n", ret);
1353 }
1354
1355 /**
1356  * ucsi_get_drvdata - Return private driver data pointer
1357  * @ucsi: UCSI interface
1358  */
1359 void *ucsi_get_drvdata(struct ucsi *ucsi)
1360 {
1361         return ucsi->driver_data;
1362 }
1363 EXPORT_SYMBOL_GPL(ucsi_get_drvdata);
1364
1365 /**
1366  * ucsi_set_drvdata - Assign private driver data pointer
1367  * @ucsi: UCSI interface
1368  * @data: Private data pointer
1369  */
1370 void ucsi_set_drvdata(struct ucsi *ucsi, void *data)
1371 {
1372         ucsi->driver_data = data;
1373 }
1374 EXPORT_SYMBOL_GPL(ucsi_set_drvdata);
1375
1376 /**
1377  * ucsi_create - Allocate UCSI instance
1378  * @dev: Device interface to the PPM (Platform Policy Manager)
1379  * @ops: I/O routines
1380  */
1381 struct ucsi *ucsi_create(struct device *dev, const struct ucsi_operations *ops)
1382 {
1383         struct ucsi *ucsi;
1384
1385         if (!ops || !ops->read || !ops->sync_write || !ops->async_write)
1386                 return ERR_PTR(-EINVAL);
1387
1388         ucsi = kzalloc(sizeof(*ucsi), GFP_KERNEL);
1389         if (!ucsi)
1390                 return ERR_PTR(-ENOMEM);
1391
1392         INIT_WORK(&ucsi->work, ucsi_init_work);
1393         mutex_init(&ucsi->ppm_lock);
1394         ucsi->dev = dev;
1395         ucsi->ops = ops;
1396
1397         return ucsi;
1398 }
1399 EXPORT_SYMBOL_GPL(ucsi_create);
1400
1401 /**
1402  * ucsi_destroy - Free UCSI instance
1403  * @ucsi: UCSI instance to be freed
1404  */
1405 void ucsi_destroy(struct ucsi *ucsi)
1406 {
1407         kfree(ucsi);
1408 }
1409 EXPORT_SYMBOL_GPL(ucsi_destroy);
1410
1411 /**
1412  * ucsi_register - Register UCSI interface
1413  * @ucsi: UCSI instance
1414  */
1415 int ucsi_register(struct ucsi *ucsi)
1416 {
1417         int ret;
1418
1419         ret = ucsi->ops->read(ucsi, UCSI_VERSION, &ucsi->version,
1420                               sizeof(ucsi->version));
1421         if (ret)
1422                 return ret;
1423
1424         if (!ucsi->version)
1425                 return -ENODEV;
1426
1427         queue_work(system_long_wq, &ucsi->work);
1428
1429         return 0;
1430 }
1431 EXPORT_SYMBOL_GPL(ucsi_register);
1432
1433 /**
1434  * ucsi_unregister - Unregister UCSI interface
1435  * @ucsi: UCSI interface to be unregistered
1436  *
1437  * Unregister UCSI interface that was created with ucsi_register().
1438  */
1439 void ucsi_unregister(struct ucsi *ucsi)
1440 {
1441         u64 cmd = UCSI_SET_NOTIFICATION_ENABLE;
1442         int i;
1443
1444         /* Make sure that we are not in the middle of driver initialization */
1445         cancel_work_sync(&ucsi->work);
1446
1447         /* Disable notifications */
1448         ucsi->ops->async_write(ucsi, UCSI_CONTROL, &cmd, sizeof(cmd));
1449
1450         for (i = 0; i < ucsi->cap.num_connectors; i++) {
1451                 cancel_work_sync(&ucsi->connector[i].work);
1452                 ucsi_unregister_partner(&ucsi->connector[i]);
1453                 ucsi_unregister_altmodes(&ucsi->connector[i],
1454                                          UCSI_RECIPIENT_CON);
1455                 ucsi_unregister_port_psy(&ucsi->connector[i]);
1456                 if (ucsi->connector[i].wq)
1457                         destroy_workqueue(ucsi->connector[i].wq);
1458                 typec_unregister_port(ucsi->connector[i].port);
1459         }
1460
1461         kfree(ucsi->connector);
1462 }
1463 EXPORT_SYMBOL_GPL(ucsi_unregister);
1464
1465 MODULE_AUTHOR("Heikki Krogerus <heikki.krogerus@linux.intel.com>");
1466 MODULE_LICENSE("GPL v2");
1467 MODULE_DESCRIPTION("USB Type-C Connector System Software Interface driver");