drivers/usb/typec/tcpm.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright 2015-2017 Google, Inc
   4  *
   5  * USB Power Delivery protocol stack.
   6  */
   7
   8 #include <linux/completion.h>
   9 #include <linux/debugfs.h>
  10 #include <linux/device.h>
  11 #include <linux/jiffies.h>
  12 #include <linux/kernel.h>
  13 #include <linux/module.h>
  14 #include <linux/mutex.h>
  15 #include <linux/power_supply.h>
  16 #include <linux/proc_fs.h>
  17 #include <linux/property.h>
  18 #include <linux/sched/clock.h>
  19 #include <linux/seq_file.h>
  20 #include <linux/slab.h>
  21 #include <linux/spinlock.h>
  22 #include <linux/usb/pd.h>
  23 #include <linux/usb/pd_ado.h>
  24 #include <linux/usb/pd_bdo.h>
  25 #include <linux/usb/pd_ext_sdb.h>
  26 #include <linux/usb/pd_vdo.h>
  27 #include <linux/usb/role.h>
  28 #include <linux/usb/tcpm.h>
  29 #include <linux/usb/typec_altmode.h>
  30 #include <linux/workqueue.h>
  31
  32 #define FOREACH_STATE(S)                        \
  33         S(INVALID_STATE),                       \
  34         S(TOGGLING),                    \
  35         S(SRC_UNATTACHED),                      \
  36         S(SRC_ATTACH_WAIT),                     \
  37         S(SRC_ATTACHED),                        \
  38         S(SRC_STARTUP),                         \
  39         S(SRC_SEND_CAPABILITIES),               \
  40         S(SRC_SEND_CAPABILITIES_TIMEOUT),       \
  41         S(SRC_NEGOTIATE_CAPABILITIES),          \
  42         S(SRC_TRANSITION_SUPPLY),               \
  43         S(SRC_READY),                           \
  44         S(SRC_WAIT_NEW_CAPABILITIES),           \
  45                                                 \
  46         S(SNK_UNATTACHED),                      \
  47         S(SNK_ATTACH_WAIT),                     \
  48         S(SNK_DEBOUNCED),                       \
  49         S(SNK_ATTACHED),                        \
  50         S(SNK_STARTUP),                         \
  51         S(SNK_DISCOVERY),                       \
  52         S(SNK_DISCOVERY_DEBOUNCE),              \
  53         S(SNK_DISCOVERY_DEBOUNCE_DONE),         \
  54         S(SNK_WAIT_CAPABILITIES),               \
  55         S(SNK_NEGOTIATE_CAPABILITIES),          \
  56         S(SNK_NEGOTIATE_PPS_CAPABILITIES),      \
  57         S(SNK_TRANSITION_SINK),                 \
  58         S(SNK_TRANSITION_SINK_VBUS),            \
  59         S(SNK_READY),                           \
  60                                                 \
  61         S(ACC_UNATTACHED),                      \
  62         S(DEBUG_ACC_ATTACHED),                  \
  63         S(AUDIO_ACC_ATTACHED),                  \
  64         S(AUDIO_ACC_DEBOUNCE),                  \
  65                                                 \
  66         S(HARD_RESET_SEND),                     \
  67         S(HARD_RESET_START),                    \
  68         S(SRC_HARD_RESET_VBUS_OFF),             \
  69         S(SRC_HARD_RESET_VBUS_ON),              \
  70         S(SNK_HARD_RESET_SINK_OFF),             \
  71         S(SNK_HARD_RESET_WAIT_VBUS),            \
  72         S(SNK_HARD_RESET_SINK_ON),              \
  73                                                 \
  74         S(SOFT_RESET),                          \
  75         S(SOFT_RESET_SEND),                     \
  76                                                 \
  77         S(DR_SWAP_ACCEPT),                      \
  78         S(DR_SWAP_SEND),                        \
  79         S(DR_SWAP_SEND_TIMEOUT),                \
  80         S(DR_SWAP_CANCEL),                      \
  81         S(DR_SWAP_CHANGE_DR),                   \
  82                                                 \
  83         S(PR_SWAP_ACCEPT),                      \
  84         S(PR_SWAP_SEND),                        \
  85         S(PR_SWAP_SEND_TIMEOUT),                \
  86         S(PR_SWAP_CANCEL),                      \
  87         S(PR_SWAP_START),                       \
  88         S(PR_SWAP_SRC_SNK_TRANSITION_OFF),      \
  89         S(PR_SWAP_SRC_SNK_SOURCE_OFF),          \
  90         S(PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED), \
  91         S(PR_SWAP_SRC_SNK_SINK_ON),             \
  92         S(PR_SWAP_SNK_SRC_SINK_OFF),            \
  93         S(PR_SWAP_SNK_SRC_SOURCE_ON),           \
  94         S(PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP),    \
  95                                                 \
  96         S(VCONN_SWAP_ACCEPT),                   \
  97         S(VCONN_SWAP_SEND),                     \
  98         S(VCONN_SWAP_SEND_TIMEOUT),             \
  99         S(VCONN_SWAP_CANCEL),                   \
 100         S(VCONN_SWAP_START),                    \
 101         S(VCONN_SWAP_WAIT_FOR_VCONN),           \
 102         S(VCONN_SWAP_TURN_ON_VCONN),            \
 103         S(VCONN_SWAP_TURN_OFF_VCONN),           \
 104                                                 \
 105         S(SNK_TRY),                             \
 106         S(SNK_TRY_WAIT),                        \
 107         S(SNK_TRY_WAIT_DEBOUNCE),               \
 108         S(SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS),    \
 109         S(SRC_TRYWAIT),                         \
 110         S(SRC_TRYWAIT_DEBOUNCE),                \
 111         S(SRC_TRYWAIT_UNATTACHED),              \
 112                                                 \
 113         S(SRC_TRY),                             \
 114         S(SRC_TRY_WAIT),                        \
 115         S(SRC_TRY_DEBOUNCE),                    \
 116         S(SNK_TRYWAIT),                         \
 117         S(SNK_TRYWAIT_DEBOUNCE),                \
 118         S(SNK_TRYWAIT_VBUS),                    \
 119         S(BIST_RX),                             \
 120                                                 \
 121         S(GET_STATUS_SEND),                     \
 122         S(GET_STATUS_SEND_TIMEOUT),             \
 123         S(GET_PPS_STATUS_SEND),                 \
 124         S(GET_PPS_STATUS_SEND_TIMEOUT),         \
 125                                                 \
 126         S(ERROR_RECOVERY),                      \
 127         S(PORT_RESET),                          \
 128         S(PORT_RESET_WAIT_OFF)
 129
 130 #define GENERATE_ENUM(e)        e
 131 #define GENERATE_STRING(s)      #s
 132
 133 enum tcpm_state {
 134         FOREACH_STATE(GENERATE_ENUM)
 135 };
 136
 137 static const char * const tcpm_states[] = {
 138         FOREACH_STATE(GENERATE_STRING)
 139 };
 140
 141 enum vdm_states {
 142         VDM_STATE_ERR_BUSY = -3,
 143         VDM_STATE_ERR_SEND = -2,
 144         VDM_STATE_ERR_TMOUT = -1,
 145         VDM_STATE_DONE = 0,
 146         /* Anything >0 represents an active state */
 147         VDM_STATE_READY = 1,
 148         VDM_STATE_BUSY = 2,
 149         VDM_STATE_WAIT_RSP_BUSY = 3,
 150 };
 151
 152 enum pd_msg_request {
 153         PD_MSG_NONE = 0,
 154         PD_MSG_CTRL_REJECT,
 155         PD_MSG_CTRL_WAIT,
 156         PD_MSG_CTRL_NOT_SUPP,
 157         PD_MSG_DATA_SINK_CAP,
 158         PD_MSG_DATA_SOURCE_CAP,
 159 };
 160
 161 /* Events from low level driver */
 162
 163 #define TCPM_CC_EVENT           BIT(0)
 164 #define TCPM_VBUS_EVENT         BIT(1)
 165 #define TCPM_RESET_EVENT        BIT(2)
 166
 167 #define LOG_BUFFER_ENTRIES      1024
 168 #define LOG_BUFFER_ENTRY_SIZE   128
 169
 170 /* Alternate mode support */
 171
 172 #define SVID_DISCOVERY_MAX      16
 173 #define ALTMODE_DISCOVERY_MAX   (SVID_DISCOVERY_MAX * MODE_DISCOVERY_MAX)
 174
 175 struct pd_mode_data {
 176         int svid_index;         /* current SVID index           */
 177         int nsvids;
 178         u16 svids[SVID_DISCOVERY_MAX];
 179         int altmodes;           /* number of alternate modes    */
 180         struct typec_altmode_desc altmode_desc[ALTMODE_DISCOVERY_MAX];
 181 };
 182
 183 struct pd_pps_data {
 184         u32 min_volt;
 185         u32 max_volt;
 186         u32 max_curr;
 187         u32 out_volt;
 188         u32 op_curr;
 189         bool supported;
 190         bool active;
 191 };
 192
 193 struct tcpm_port {
 194         struct device *dev;
 195
 196         struct mutex lock;              /* tcpm state machine lock */
 197         struct workqueue_struct *wq;
 198
 199         struct typec_capability typec_caps;
 200         struct typec_port *typec_port;
 201
 202         struct tcpc_dev *tcpc;
 203         struct usb_role_switch *role_sw;
 204
 205         enum typec_role vconn_role;
 206         enum typec_role pwr_role;
 207         enum typec_data_role data_role;
 208         enum typec_pwr_opmode pwr_opmode;
 209
 210         struct usb_pd_identity partner_ident;
 211         struct typec_partner_desc partner_desc;
 212         struct typec_partner *partner;
 213
 214         enum typec_cc_status cc_req;
 215
 216         enum typec_cc_status cc1;
 217         enum typec_cc_status cc2;
 218         enum typec_cc_polarity polarity;
 219
 220         bool attached;
 221         bool connected;
 222         enum typec_port_type port_type;
 223         bool vbus_present;
 224         bool vbus_never_low;
 225         bool vbus_source;
 226         bool vbus_charge;
 227
 228         bool send_discover;
 229         bool op_vsafe5v;
 230
 231         int try_role;
 232         int try_snk_count;
 233         int try_src_count;
 234
 235         enum pd_msg_request queued_message;
 236
 237         enum tcpm_state enter_state;
 238         enum tcpm_state prev_state;
 239         enum tcpm_state state;
 240         enum tcpm_state delayed_state;
 241         unsigned long delayed_runtime;
 242         unsigned long delay_ms;
 243
 244         spinlock_t pd_event_lock;
 245         u32 pd_events;
 246
 247         struct work_struct event_work;
 248         struct delayed_work state_machine;
 249         struct delayed_work vdm_state_machine;
 250         bool state_machine_running;
 251
 252         struct completion tx_complete;
 253         enum tcpm_transmit_status tx_status;
 254
 255         struct mutex swap_lock;         /* swap command lock */
 256         bool swap_pending;
 257         bool non_pd_role_swap;
 258         struct completion swap_complete;
 259         int swap_status;
 260
 261         unsigned int negotiated_rev;
 262         unsigned int message_id;
 263         unsigned int caps_count;
 264         unsigned int hard_reset_count;
 265         bool pd_capable;
 266         bool explicit_contract;
 267         unsigned int rx_msgid;
 268
 269         /* Partner capabilities/requests */
 270         u32 sink_request;
 271         u32 source_caps[PDO_MAX_OBJECTS];
 272         unsigned int nr_source_caps;
 273         u32 sink_caps[PDO_MAX_OBJECTS];
 274         unsigned int nr_sink_caps;
 275
 276         /* Local capabilities */
 277         u32 src_pdo[PDO_MAX_OBJECTS];
 278         unsigned int nr_src_pdo;
 279         u32 snk_pdo[PDO_MAX_OBJECTS];
 280         unsigned int nr_snk_pdo;
 281         u32 snk_vdo[VDO_MAX_OBJECTS];
 282         unsigned int nr_snk_vdo;
 283
 284         unsigned int operating_snk_mw;
 285         bool update_sink_caps;
 286
 287         /* Requested current / voltage */
 288         u32 current_limit;
 289         u32 supply_voltage;
 290
 291         /* Used to export TA voltage and current */
 292         struct power_supply *psy;
 293         struct power_supply_desc psy_desc;
 294         enum power_supply_usb_type usb_type;
 295
 296         u32 bist_request;
 297
 298         /* PD state for Vendor Defined Messages */
 299         enum vdm_states vdm_state;
 300         u32 vdm_retries;
 301         /* next Vendor Defined Message to send */
 302         u32 vdo_data[VDO_MAX_SIZE];
 303         u8 vdo_count;
 304         /* VDO to retry if UFP responder replied busy */
 305         u32 vdo_retry;
 306
 307         /* PPS */
 308         struct pd_pps_data pps_data;
 309         struct completion pps_complete;
 310         bool pps_pending;
 311         int pps_status;
 312
 313         /* Alternate mode data */
 314         struct pd_mode_data mode_data;
 315         struct typec_altmode *partner_altmode[ALTMODE_DISCOVERY_MAX];
 316         struct typec_altmode *port_altmode[ALTMODE_DISCOVERY_MAX];
 317
 318         /* Deadline in jiffies to exit src_try_wait state */
 319         unsigned long max_wait;
 320
 321         /* port belongs to a self powered device */
 322         bool self_powered;
 323
 324 #ifdef CONFIG_DEBUG_FS
 325         struct dentry *dentry;
 326         struct mutex logbuffer_lock;    /* log buffer access lock */
 327         int logbuffer_head;
 328         int logbuffer_tail;
 329         u8 *logbuffer[LOG_BUFFER_ENTRIES];
 330 #endif
 331 };
 332
 333 struct pd_rx_event {
 334         struct work_struct work;
 335         struct tcpm_port *port;
 336         struct pd_message msg;
 337 };
 338
 339 #define tcpm_cc_is_sink(cc) \
 340         ((cc) == TYPEC_CC_RP_DEF || (cc) == TYPEC_CC_RP_1_5 || \
 341          (cc) == TYPEC_CC_RP_3_0)
 342
 343 #define tcpm_port_is_sink(port) \
 344         ((tcpm_cc_is_sink((port)->cc1) && !tcpm_cc_is_sink((port)->cc2)) || \
 345          (tcpm_cc_is_sink((port)->cc2) && !tcpm_cc_is_sink((port)->cc1)))
 346
 347 #define tcpm_cc_is_source(cc) ((cc) == TYPEC_CC_RD)
 348 #define tcpm_cc_is_audio(cc) ((cc) == TYPEC_CC_RA)
 349 #define tcpm_cc_is_open(cc) ((cc) == TYPEC_CC_OPEN)
 350
 351 #define tcpm_port_is_source(port) \
 352         ((tcpm_cc_is_source((port)->cc1) && \
 353          !tcpm_cc_is_source((port)->cc2)) || \
 354          (tcpm_cc_is_source((port)->cc2) && \
 355           !tcpm_cc_is_source((port)->cc1)))
 356
 357 #define tcpm_port_is_debug(port) \
 358         (tcpm_cc_is_source((port)->cc1) && tcpm_cc_is_source((port)->cc2))
 359
 360 #define tcpm_port_is_audio(port) \
 361         (tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_audio((port)->cc2))
 362
 363 #define tcpm_port_is_audio_detached(port) \
 364         ((tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_open((port)->cc2)) || \
 365          (tcpm_cc_is_audio((port)->cc2) && tcpm_cc_is_open((port)->cc1)))
 366
 367 #define tcpm_try_snk(port) \
 368         ((port)->try_snk_count == 0 && (port)->try_role == TYPEC_SINK && \
 369         (port)->port_type == TYPEC_PORT_DRP)
 370
 371 #define tcpm_try_src(port) \
 372         ((port)->try_src_count == 0 && (port)->try_role == TYPEC_SOURCE && \
 373         (port)->port_type == TYPEC_PORT_DRP)
 374
 375 static enum tcpm_state tcpm_default_state(struct tcpm_port *port)
 376 {
 377         if (port->port_type == TYPEC_PORT_DRP) {
 378                 if (port->try_role == TYPEC_SINK)
 379                         return SNK_UNATTACHED;
 380                 else if (port->try_role == TYPEC_SOURCE)
 381                         return SRC_UNATTACHED;
 382                 else if (port->tcpc->config &&
 383                          port->tcpc->config->default_role == TYPEC_SINK)
 384                         return SNK_UNATTACHED;
 385                 /* Fall through to return SRC_UNATTACHED */
 386         } else if (port->port_type == TYPEC_PORT_SNK) {
 387                 return SNK_UNATTACHED;
 388         }
 389         return SRC_UNATTACHED;
 390 }
 391
 392 static inline
 393 struct tcpm_port *typec_cap_to_tcpm(const struct typec_capability *cap)
 394 {
 395         return container_of(cap, struct tcpm_port, typec_caps);
 396 }
 397
 398 static bool tcpm_port_is_disconnected(struct tcpm_port *port)
 399 {
 400         return (!port->attached && port->cc1 == TYPEC_CC_OPEN &&
 401                 port->cc2 == TYPEC_CC_OPEN) ||
 402                (port->attached && ((port->polarity == TYPEC_POLARITY_CC1 &&
 403                                     port->cc1 == TYPEC_CC_OPEN) ||
 404                                    (port->polarity == TYPEC_POLARITY_CC2 &&
 405                                     port->cc2 == TYPEC_CC_OPEN)));
 406 }
 407
 408 /*
 409  * Logging
 410  */
 411
 412 #ifdef CONFIG_DEBUG_FS
 413
 414 static bool tcpm_log_full(struct tcpm_port *port)
 415 {
 416         return port->logbuffer_tail ==
 417                 (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 418 }
 419
 420 __printf(2, 0)
 421 static void _tcpm_log(struct tcpm_port *port, const char *fmt, va_list args)
 422 {
 423         char tmpbuffer[LOG_BUFFER_ENTRY_SIZE];
 424         u64 ts_nsec = local_clock();
 425         unsigned long rem_nsec;
 426
 427         mutex_lock(&port->logbuffer_lock);
 428         if (!port->logbuffer[port->logbuffer_head]) {
 429                 port->logbuffer[port->logbuffer_head] =
 430                                 kzalloc(LOG_BUFFER_ENTRY_SIZE, GFP_KERNEL);
 431                 if (!port->logbuffer[port->logbuffer_head]) {
 432                         mutex_unlock(&port->logbuffer_lock);
 433                         return;
 434                 }
 435         }
 436
 437         vsnprintf(tmpbuffer, sizeof(tmpbuffer), fmt, args);
 438
 439         if (tcpm_log_full(port)) {
 440                 port->logbuffer_head = max(port->logbuffer_head - 1, 0);
 441                 strcpy(tmpbuffer, "overflow");
 442         }
 443
 444         if (port->logbuffer_head < 0 ||
 445             port->logbuffer_head >= LOG_BUFFER_ENTRIES) {
 446                 dev_warn(port->dev,
 447                          "Bad log buffer index %d\n", port->logbuffer_head);
 448                 goto abort;
 449         }
 450
 451         if (!port->logbuffer[port->logbuffer_head]) {
 452                 dev_warn(port->dev,
 453                          "Log buffer index %d is NULL\n", port->logbuffer_head);
 454                 goto abort;
 455         }
 456
 457         rem_nsec = do_div(ts_nsec, 1000000000);
 458         scnprintf(port->logbuffer[port->logbuffer_head],
 459                   LOG_BUFFER_ENTRY_SIZE, "[%5lu.%06lu] %s",
 460                   (unsigned long)ts_nsec, rem_nsec / 1000,
 461                   tmpbuffer);
 462         port->logbuffer_head = (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 463
 464 abort:
 465         mutex_unlock(&port->logbuffer_lock);
 466 }
 467
 468 __printf(2, 3)
 469 static void tcpm_log(struct tcpm_port *port, const char *fmt, ...)
 470 {
 471         va_list args;
 472
 473         /* Do not log while disconnected and unattached */
 474         if (tcpm_port_is_disconnected(port) &&
 475             (port->state == SRC_UNATTACHED || port->state == SNK_UNATTACHED ||
 476              port->state == TOGGLING))
 477                 return;
 478
 479         va_start(args, fmt);
 480         _tcpm_log(port, fmt, args);
 481         va_end(args);
 482 }
 483
 484 __printf(2, 3)
 485 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...)
 486 {
 487         va_list args;
 488
 489         va_start(args, fmt);
 490         _tcpm_log(port, fmt, args);
 491         va_end(args);
 492 }
 493
 494 static void tcpm_log_source_caps(struct tcpm_port *port)
 495 {
 496         int i;
 497
 498         for (i = 0; i < port->nr_source_caps; i++) {
 499                 u32 pdo = port->source_caps[i];
 500                 enum pd_pdo_type type = pdo_type(pdo);
 501                 char msg[64];
 502
 503                 switch (type) {
 504                 case PDO_TYPE_FIXED:
 505                         scnprintf(msg, sizeof(msg),
 506                                   "%u mV, %u mA [%s%s%s%s%s%s]",
 507                                   pdo_fixed_voltage(pdo),
 508                                   pdo_max_current(pdo),
 509                                   (pdo & PDO_FIXED_DUAL_ROLE) ?
 510                                                         "R" : "",
 511                                   (pdo & PDO_FIXED_SUSPEND) ?
 512                                                         "S" : "",
 513                                   (pdo & PDO_FIXED_HIGHER_CAP) ?
 514                                                         "H" : "",
 515                                   (pdo & PDO_FIXED_USB_COMM) ?
 516                                                         "U" : "",
 517                                   (pdo & PDO_FIXED_DATA_SWAP) ?
 518                                                         "D" : "",
 519                                   (pdo & PDO_FIXED_EXTPOWER) ?
 520                                                         "E" : "");
 521                         break;
 522                 case PDO_TYPE_VAR:
 523                         scnprintf(msg, sizeof(msg),
 524                                   "%u-%u mV, %u mA",
 525                                   pdo_min_voltage(pdo),
 526                                   pdo_max_voltage(pdo),
 527                                   pdo_max_current(pdo));
 528                         break;
 529                 case PDO_TYPE_BATT:
 530                         scnprintf(msg, sizeof(msg),
 531                                   "%u-%u mV, %u mW",
 532                                   pdo_min_voltage(pdo),
 533                                   pdo_max_voltage(pdo),
 534                                   pdo_max_power(pdo));
 535                         break;
 536                 case PDO_TYPE_APDO:
 537                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS)
 538                                 scnprintf(msg, sizeof(msg),
 539                                           "%u-%u mV, %u mA",
 540                                           pdo_pps_apdo_min_voltage(pdo),
 541                                           pdo_pps_apdo_max_voltage(pdo),
 542                                           pdo_pps_apdo_max_current(pdo));
 543                         else
 544                                 strcpy(msg, "undefined APDO");
 545                         break;
 546                 default:
 547                         strcpy(msg, "undefined");
 548                         break;
 549                 }
 550                 tcpm_log(port, " PDO %d: type %d, %s",
 551                          i, type, msg);
 552         }
 553 }
 554
 555 static int tcpm_debug_show(struct seq_file *s, void *v)
 556 {
 557         struct tcpm_port *port = (struct tcpm_port *)s->private;
 558         int tail;
 559
 560         mutex_lock(&port->logbuffer_lock);
 561         tail = port->logbuffer_tail;
 562         while (tail != port->logbuffer_head) {
 563                 seq_printf(s, "%s\n", port->logbuffer[tail]);
 564                 tail = (tail + 1) % LOG_BUFFER_ENTRIES;
 565         }
 566         if (!seq_has_overflowed(s))
 567                 port->logbuffer_tail = tail;
 568         mutex_unlock(&port->logbuffer_lock);
 569
 570         return 0;
 571 }
 572 DEFINE_SHOW_ATTRIBUTE(tcpm_debug);
 573
 574 static struct dentry *rootdir;
 575
 576 static void tcpm_debugfs_init(struct tcpm_port *port)
 577 {
 578         mutex_init(&port->logbuffer_lock);
 579         /* /sys/kernel/debug/tcpm/usbcX */
 580         if (!rootdir)
 581                 rootdir = debugfs_create_dir("tcpm", NULL);
 582
 583         port->dentry = debugfs_create_file(dev_name(port->dev),
 584                                            S_IFREG | 0444, rootdir,
 585                                            port, &tcpm_debug_fops);
 586 }
 587
 588 static void tcpm_debugfs_exit(struct tcpm_port *port)
 589 {
 590         int i;
 591
 592         mutex_lock(&port->logbuffer_lock);
 593         for (i = 0; i < LOG_BUFFER_ENTRIES; i++) {
 594                 kfree(port->logbuffer[i]);
 595                 port->logbuffer[i] = NULL;
 596         }
 597         mutex_unlock(&port->logbuffer_lock);
 598
 599         debugfs_remove(port->dentry);
 600         if (list_empty(&rootdir->d_subdirs)) {
 601                 debugfs_remove(rootdir);
 602                 rootdir = NULL;
 603         }
 604 }
 605
 606 #else
 607
 608 __printf(2, 3)
 609 static void tcpm_log(const struct tcpm_port *port, const char *fmt, ...) { }
 610 __printf(2, 3)
 611 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...) { }
 612 static void tcpm_log_source_caps(struct tcpm_port *port) { }
 613 static void tcpm_debugfs_init(const struct tcpm_port *port) { }
 614 static void tcpm_debugfs_exit(const struct tcpm_port *port) { }
 615
 616 #endif
 617
 618 static int tcpm_pd_transmit(struct tcpm_port *port,
 619                             enum tcpm_transmit_type type,
 620                             const struct pd_message *msg)
 621 {
 622         unsigned long timeout;
 623         int ret;
 624
 625         if (msg)
 626                 tcpm_log(port, "PD TX, header: %#x", le16_to_cpu(msg->header));
 627         else
 628                 tcpm_log(port, "PD TX, type: %#x", type);
 629
 630         reinit_completion(&port->tx_complete);
 631         ret = port->tcpc->pd_transmit(port->tcpc, type, msg);
 632         if (ret < 0)
 633                 return ret;
 634
 635         mutex_unlock(&port->lock);
 636         timeout = wait_for_completion_timeout(&port->tx_complete,
 637                                 msecs_to_jiffies(PD_T_TCPC_TX_TIMEOUT));
 638         mutex_lock(&port->lock);
 639         if (!timeout)
 640                 return -ETIMEDOUT;
 641
 642         switch (port->tx_status) {
 643         case TCPC_TX_SUCCESS:
 644                 port->message_id = (port->message_id + 1) & PD_HEADER_ID_MASK;
 645                 return 0;
 646         case TCPC_TX_DISCARDED:
 647                 return -EAGAIN;
 648         case TCPC_TX_FAILED:
 649         default:
 650                 return -EIO;
 651         }
 652 }
 653
 654 void tcpm_pd_transmit_complete(struct tcpm_port *port,
 655                                enum tcpm_transmit_status status)
 656 {
 657         tcpm_log(port, "PD TX complete, status: %u", status);
 658         port->tx_status = status;
 659         complete(&port->tx_complete);
 660 }
 661 EXPORT_SYMBOL_GPL(tcpm_pd_transmit_complete);
 662
 663 static int tcpm_mux_set(struct tcpm_port *port, int state,
 664                         enum usb_role usb_role,
 665                         enum typec_orientation orientation)
 666 {
 667         int ret;
 668
 669         tcpm_log(port, "Requesting mux state %d, usb-role %d, orientation %d",
 670                  state, usb_role, orientation);
 671
 672         ret = typec_set_orientation(port->typec_port, orientation);
 673         if (ret)
 674                 return ret;
 675
 676         if (port->role_sw) {
 677                 ret = usb_role_switch_set_role(port->role_sw, usb_role);
 678                 if (ret)
 679                         return ret;
 680         }
 681
 682         return typec_set_mode(port->typec_port, state);
 683 }
 684
 685 static int tcpm_set_polarity(struct tcpm_port *port,
 686                              enum typec_cc_polarity polarity)
 687 {
 688         int ret;
 689
 690         tcpm_log(port, "polarity %d", polarity);
 691
 692         ret = port->tcpc->set_polarity(port->tcpc, polarity);
 693         if (ret < 0)
 694                 return ret;
 695
 696         port->polarity = polarity;
 697
 698         return 0;
 699 }
 700
 701 static int tcpm_set_vconn(struct tcpm_port *port, bool enable)
 702 {
 703         int ret;
 704
 705         tcpm_log(port, "vconn:=%d", enable);
 706
 707         ret = port->tcpc->set_vconn(port->tcpc, enable);
 708         if (!ret) {
 709                 port->vconn_role = enable ? TYPEC_SOURCE : TYPEC_SINK;
 710                 typec_set_vconn_role(port->typec_port, port->vconn_role);
 711         }
 712
 713         return ret;
 714 }
 715
 716 static u32 tcpm_get_current_limit(struct tcpm_port *port)
 717 {
 718         enum typec_cc_status cc;
 719         u32 limit;
 720
 721         cc = port->polarity ? port->cc2 : port->cc1;
 722         switch (cc) {
 723         case TYPEC_CC_RP_1_5:
 724                 limit = 1500;
 725                 break;
 726         case TYPEC_CC_RP_3_0:
 727                 limit = 3000;
 728                 break;
 729         case TYPEC_CC_RP_DEF:
 730         default:
 731                 if (port->tcpc->get_current_limit)
 732                         limit = port->tcpc->get_current_limit(port->tcpc);
 733                 else
 734                         limit = 0;
 735                 break;
 736         }
 737
 738         return limit;
 739 }
 740
 741 static int tcpm_set_current_limit(struct tcpm_port *port, u32 max_ma, u32 mv)
 742 {
 743         int ret = -EOPNOTSUPP;
 744
 745         tcpm_log(port, "Setting voltage/current limit %u mV %u mA", mv, max_ma);
 746
 747         port->supply_voltage = mv;
 748         port->current_limit = max_ma;
 749
 750         if (port->tcpc->set_current_limit)
 751                 ret = port->tcpc->set_current_limit(port->tcpc, max_ma, mv);
 752
 753         return ret;
 754 }
 755
 756 /*
 757  * Determine RP value to set based on maximum current supported
 758  * by a port if configured as source.
 759  * Returns CC value to report to link partner.
 760  */
 761 static enum typec_cc_status tcpm_rp_cc(struct tcpm_port *port)
 762 {
 763         const u32 *src_pdo = port->src_pdo;
 764         int nr_pdo = port->nr_src_pdo;
 765         int i;
 766
 767         /*
 768          * Search for first entry with matching voltage.
 769          * It should report the maximum supported current.
 770          */
 771         for (i = 0; i < nr_pdo; i++) {
 772                 const u32 pdo = src_pdo[i];
 773
 774                 if (pdo_type(pdo) == PDO_TYPE_FIXED &&
 775                     pdo_fixed_voltage(pdo) == 5000) {
 776                         unsigned int curr = pdo_max_current(pdo);
 777
 778                         if (curr >= 3000)
 779                                 return TYPEC_CC_RP_3_0;
 780                         else if (curr >= 1500)
 781                                 return TYPEC_CC_RP_1_5;
 782                         return TYPEC_CC_RP_DEF;
 783                 }
 784         }
 785
 786         return TYPEC_CC_RP_DEF;
 787 }
 788
 789 static int tcpm_set_attached_state(struct tcpm_port *port, bool attached)
 790 {
 791         return port->tcpc->set_roles(port->tcpc, attached, port->pwr_role,
 792                                      port->data_role);
 793 }
 794
 795 static int tcpm_set_roles(struct tcpm_port *port, bool attached,
 796                           enum typec_role role, enum typec_data_role data)
 797 {
 798         enum typec_orientation orientation;
 799         enum usb_role usb_role;
 800         int ret;
 801
 802         if (port->polarity == TYPEC_POLARITY_CC1)
 803                 orientation = TYPEC_ORIENTATION_NORMAL;
 804         else
 805                 orientation = TYPEC_ORIENTATION_REVERSE;
 806
 807         if (data == TYPEC_HOST)
 808                 usb_role = USB_ROLE_HOST;
 809         else
 810                 usb_role = USB_ROLE_DEVICE;
 811
 812         ret = tcpm_mux_set(port, TYPEC_STATE_USB, usb_role, orientation);
 813         if (ret < 0)
 814                 return ret;
 815
 816         ret = port->tcpc->set_roles(port->tcpc, attached, role, data);
 817         if (ret < 0)
 818                 return ret;
 819
 820         port->pwr_role = role;
 821         port->data_role = data;
 822         typec_set_data_role(port->typec_port, data);
 823         typec_set_pwr_role(port->typec_port, role);
 824
 825         return 0;
 826 }
 827
 828 static int tcpm_set_pwr_role(struct tcpm_port *port, enum typec_role role)
 829 {
 830         int ret;
 831
 832         ret = port->tcpc->set_roles(port->tcpc, true, role,
 833                                     port->data_role);
 834         if (ret < 0)
 835                 return ret;
 836
 837         port->pwr_role = role;
 838         typec_set_pwr_role(port->typec_port, role);
 839
 840         return 0;
 841 }
 842
 843 static int tcpm_pd_send_source_caps(struct tcpm_port *port)
 844 {
 845         struct pd_message msg;
 846         int i;
 847
 848         memset(&msg, 0, sizeof(msg));
 849         if (!port->nr_src_pdo) {
 850                 /* No source capabilities defined, sink only */
 851                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
 852                                           port->pwr_role,
 853                                           port->data_role,
 854                                           port->negotiated_rev,
 855                                           port->message_id, 0);
 856         } else {
 857                 msg.header = PD_HEADER_LE(PD_DATA_SOURCE_CAP,
 858                                           port->pwr_role,
 859                                           port->data_role,
 860                                           port->negotiated_rev,
 861                                           port->message_id,
 862                                           port->nr_src_pdo);
 863         }
 864         for (i = 0; i < port->nr_src_pdo; i++)
 865                 msg.payload[i] = cpu_to_le32(port->src_pdo[i]);
 866
 867         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
 868 }
 869
 870 static int tcpm_pd_send_sink_caps(struct tcpm_port *port)
 871 {
 872         struct pd_message msg;
 873         int i;
 874
 875         memset(&msg, 0, sizeof(msg));
 876         if (!port->nr_snk_pdo) {
 877                 /* No sink capabilities defined, source only */
 878                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
 879                                           port->pwr_role,
 880                                           port->data_role,
 881                                           port->negotiated_rev,
 882                                           port->message_id, 0);
 883         } else {
 884                 msg.header = PD_HEADER_LE(PD_DATA_SINK_CAP,
 885                                           port->pwr_role,
 886                                           port->data_role,
 887                                           port->negotiated_rev,
 888                                           port->message_id,
 889                                           port->nr_snk_pdo);
 890         }
 891         for (i = 0; i < port->nr_snk_pdo; i++)
 892                 msg.payload[i] = cpu_to_le32(port->snk_pdo[i]);
 893
 894         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
 895 }
 896
 897 static void tcpm_set_state(struct tcpm_port *port, enum tcpm_state state,
 898                            unsigned int delay_ms)
 899 {
 900         if (delay_ms) {
 901                 tcpm_log(port, "pending state change %s -> %s @ %u ms",
 902                          tcpm_states[port->state], tcpm_states[state],
 903                          delay_ms);
 904                 port->delayed_state = state;
 905                 mod_delayed_work(port->wq, &port->state_machine,
 906                                  msecs_to_jiffies(delay_ms));
 907                 port->delayed_runtime = jiffies + msecs_to_jiffies(delay_ms);
 908                 port->delay_ms = delay_ms;
 909         } else {
 910                 tcpm_log(port, "state change %s -> %s",
 911                          tcpm_states[port->state], tcpm_states[state]);
 912                 port->delayed_state = INVALID_STATE;
 913                 port->prev_state = port->state;
 914                 port->state = state;
 915                 /*
 916                  * Don't re-queue the state machine work item if we're currently
 917                  * in the state machine and we're immediately changing states.
 918                  * tcpm_state_machine_work() will continue running the state
 919                  * machine.
 920                  */
 921                 if (!port->state_machine_running)
 922                         mod_delayed_work(port->wq, &port->state_machine, 0);
 923         }
 924 }
 925
 926 static void tcpm_set_state_cond(struct tcpm_port *port, enum tcpm_state state,
 927                                 unsigned int delay_ms)
 928 {
 929         if (port->enter_state == port->state)
 930                 tcpm_set_state(port, state, delay_ms);
 931         else
 932                 tcpm_log(port,
 933                          "skipped %sstate change %s -> %s [%u ms], context state %s",
 934                          delay_ms ? "delayed " : "",
 935                          tcpm_states[port->state], tcpm_states[state],
 936                          delay_ms, tcpm_states[port->enter_state]);
 937 }
 938
 939 static void tcpm_queue_message(struct tcpm_port *port,
 940                                enum pd_msg_request message)
 941 {
 942         port->queued_message = message;
 943         mod_delayed_work(port->wq, &port->state_machine, 0);
 944 }
 945
 946 /*
 947  * VDM/VDO handling functions
 948  */
 949 static void tcpm_queue_vdm(struct tcpm_port *port, const u32 header,
 950                            const u32 *data, int cnt)
 951 {
 952         port->vdo_count = cnt + 1;
 953         port->vdo_data[0] = header;
 954         memcpy(&port->vdo_data[1], data, sizeof(u32) * cnt);
 955         /* Set ready, vdm state machine will actually send */
 956         port->vdm_retries = 0;
 957         port->vdm_state = VDM_STATE_READY;
 958 }
 959
 960 static void svdm_consume_identity(struct tcpm_port *port, const __le32 *payload,
 961                                   int cnt)
 962 {
 963         u32 vdo = le32_to_cpu(payload[VDO_INDEX_IDH]);
 964         u32 product = le32_to_cpu(payload[VDO_INDEX_PRODUCT]);
 965
 966         memset(&port->mode_data, 0, sizeof(port->mode_data));
 967
 968         port->partner_ident.id_header = vdo;
 969         port->partner_ident.cert_stat = le32_to_cpu(payload[VDO_INDEX_CSTAT]);
 970         port->partner_ident.product = product;
 971
 972         typec_partner_set_identity(port->partner);
 973
 974         tcpm_log(port, "Identity: %04x:%04x.%04x",
 975                  PD_IDH_VID(vdo),
 976                  PD_PRODUCT_PID(product), product & 0xffff);
 977 }
 978
 979 static bool svdm_consume_svids(struct tcpm_port *port, const __le32 *payload,
 980                                int cnt)
 981 {
 982         struct pd_mode_data *pmdata = &port->mode_data;
 983         int i;
 984
 985         for (i = 1; i < cnt; i++) {
 986                 u32 p = le32_to_cpu(payload[i]);
 987                 u16 svid;
 988
 989                 svid = (p >> 16) & 0xffff;
 990                 if (!svid)
 991                         return false;
 992
 993                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
 994                         goto abort;
 995
 996                 pmdata->svids[pmdata->nsvids++] = svid;
 997                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
 998
 999                 svid = p & 0xffff;
1000                 if (!svid)
1001                         return false;
1002
1003                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
1004                         goto abort;
1005
1006                 pmdata->svids[pmdata->nsvids++] = svid;
1007                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
1008         }
1009         return true;
1010 abort:
1011         tcpm_log(port, "SVID_DISCOVERY_MAX(%d) too low!", SVID_DISCOVERY_MAX);
1012         return false;
1013 }
1014
1015 static void svdm_consume_modes(struct tcpm_port *port, const __le32 *payload,
1016                                int cnt)
1017 {
1018         struct pd_mode_data *pmdata = &port->mode_data;
1019         struct typec_altmode_desc *paltmode;
1020         int i;
1021
1022         if (pmdata->altmodes >= ARRAY_SIZE(port->partner_altmode)) {
1023                 /* Already logged in svdm_consume_svids() */
1024                 return;
1025         }
1026
1027         for (i = 1; i < cnt; i++) {
1028                 paltmode = &pmdata->altmode_desc[pmdata->altmodes];
1029                 memset(paltmode, 0, sizeof(*paltmode));
1030
1031                 paltmode->svid = pmdata->svids[pmdata->svid_index];
1032                 paltmode->mode = i;
1033                 paltmode->vdo = le32_to_cpu(payload[i]);
1034
1035                 tcpm_log(port, " Alternate mode %d: SVID 0x%04x, VDO %d: 0x%08x",
1036                          pmdata->altmodes, paltmode->svid,
1037                          paltmode->mode, paltmode->vdo);
1038
1039                 pmdata->altmodes++;
1040         }
1041 }
1042
1043 static void tcpm_register_partner_altmodes(struct tcpm_port *port)
1044 {
1045         struct pd_mode_data *modep = &port->mode_data;
1046         struct typec_altmode *altmode;
1047         int i;
1048
1049         for (i = 0; i < modep->altmodes; i++) {
1050                 altmode = typec_partner_register_altmode(port->partner,
1051                                                 &modep->altmode_desc[i]);
1052                 if (!altmode)
1053                         tcpm_log(port, "Failed to register partner SVID 0x%04x",
1054                                  modep->altmode_desc[i].svid);
1055                 port->partner_altmode[i] = altmode;
1056         }
1057 }
1058
1059 #define supports_modal(port)    PD_IDH_MODAL_SUPP((port)->partner_ident.id_header)
1060
1061 static int tcpm_pd_svdm(struct tcpm_port *port, const __le32 *payload, int cnt,
1062                         u32 *response)
1063 {
1064         struct typec_altmode *adev;
1065         struct typec_altmode *pdev;
1066         struct pd_mode_data *modep;
1067         u32 p[PD_MAX_PAYLOAD];
1068         int rlen = 0;
1069         int cmd_type;
1070         int cmd;
1071         int i;
1072
1073         for (i = 0; i < cnt; i++)
1074                 p[i] = le32_to_cpu(payload[i]);
1075
1076         cmd_type = PD_VDO_CMDT(p[0]);
1077         cmd = PD_VDO_CMD(p[0]);
1078
1079         tcpm_log(port, "Rx VDM cmd 0x%x type %d cmd %d len %d",
1080                  p[0], cmd_type, cmd, cnt);
1081
1082         modep = &port->mode_data;
1083
1084         adev = typec_match_altmode(port->port_altmode, ALTMODE_DISCOVERY_MAX,
1085                                    PD_VDO_VID(p[0]), PD_VDO_OPOS(p[0]));
1086
1087         pdev = typec_match_altmode(port->partner_altmode, ALTMODE_DISCOVERY_MAX,
1088                                    PD_VDO_VID(p[0]), PD_VDO_OPOS(p[0]));
1089
1090         switch (cmd_type) {
1091         case CMDT_INIT:
1092                 switch (cmd) {
1093                 case CMD_DISCOVER_IDENT:
1094                         /* 6.4.4.3.1: Only respond as UFP (device) */
1095                         if (port->data_role == TYPEC_DEVICE &&
1096                             port->nr_snk_vdo) {
1097                                 for (i = 0; i <  port->nr_snk_vdo; i++)
1098                                         response[i + 1] = port->snk_vdo[i];
1099                                 rlen = port->nr_snk_vdo + 1;
1100                         }
1101                         break;
1102                 case CMD_DISCOVER_SVID:
1103                         break;
1104                 case CMD_DISCOVER_MODES:
1105                         break;
1106                 case CMD_ENTER_MODE:
1107                         break;
1108                 case CMD_EXIT_MODE:
1109                         break;
1110                 case CMD_ATTENTION:
1111                         /* Attention command does not have response */
1112                         if (adev)
1113                                 typec_altmode_attention(adev, p[1]);
1114                         return 0;
1115                 default:
1116                         break;
1117                 }
1118                 if (rlen >= 1) {
1119                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_ACK);
1120                 } else if (rlen == 0) {
1121                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_NAK);
1122                         rlen = 1;
1123                 } else {
1124                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_BUSY);
1125                         rlen = 1;
1126                 }
1127                 break;
1128         case CMDT_RSP_ACK:
1129                 /* silently drop message if we are not connected */
1130                 if (IS_ERR_OR_NULL(port->partner))
1131                         break;
1132
1133                 switch (cmd) {
1134                 case CMD_DISCOVER_IDENT:
1135                         /* 6.4.4.3.1 */
1136                         svdm_consume_identity(port, payload, cnt);
1137                         response[0] = VDO(USB_SID_PD, 1, CMD_DISCOVER_SVID);
1138                         rlen = 1;
1139                         break;
1140                 case CMD_DISCOVER_SVID:
1141                         /* 6.4.4.3.2 */
1142                         if (svdm_consume_svids(port, payload, cnt)) {
1143                                 response[0] = VDO(USB_SID_PD, 1,
1144                                                   CMD_DISCOVER_SVID);
1145                                 rlen = 1;
1146                         } else if (modep->nsvids && supports_modal(port)) {
1147                                 response[0] = VDO(modep->svids[0], 1,
1148                                                   CMD_DISCOVER_MODES);
1149                                 rlen = 1;
1150                         }
1151                         break;
1152                 case CMD_DISCOVER_MODES:
1153                         /* 6.4.4.3.3 */
1154                         svdm_consume_modes(port, payload, cnt);
1155                         modep->svid_index++;
1156                         if (modep->svid_index < modep->nsvids) {
1157                                 u16 svid = modep->svids[modep->svid_index];
1158                                 response[0] = VDO(svid, 1, CMD_DISCOVER_MODES);
1159                                 rlen = 1;
1160                         } else {
1161                                 tcpm_register_partner_altmodes(port);
1162                         }
1163                         break;
1164                 case CMD_ENTER_MODE:
1165                         if (adev && pdev) {
1166                                 typec_altmode_update_active(pdev, true);
1167
1168                                 if (typec_altmode_vdm(adev, p[0], &p[1], cnt)) {
1169                                         response[0] = VDO(adev->svid, 1,
1170                                                           CMD_EXIT_MODE);
1171                                         response[0] |= VDO_OPOS(adev->mode);
1172                                         return 1;
1173                                 }
1174                         }
1175                         return 0;
1176                 case CMD_EXIT_MODE:
1177                         if (adev && pdev) {
1178                                 typec_altmode_update_active(pdev, false);
1179
1180                                 /* Back to USB Operation */
1181                                 WARN_ON(typec_altmode_notify(adev,
1182                                                              TYPEC_STATE_USB,
1183                                                              NULL));
1184                         }
1185                         break;
1186                 default:
1187                         break;
1188                 }
1189                 break;
1190         case CMDT_RSP_NAK:
1191                 switch (cmd) {
1192                 case CMD_ENTER_MODE:
1193                         /* Back to USB Operation */
1194                         if (adev)
1195                                 WARN_ON(typec_altmode_notify(adev,
1196                                                              TYPEC_STATE_USB,
1197                                                              NULL));
1198                         break;
1199                 default:
1200                         break;
1201                 }
1202                 break;
1203         default:
1204                 break;
1205         }
1206
1207         /* Informing the alternate mode drivers about everything */
1208         if (adev)
1209                 typec_altmode_vdm(adev, p[0], &p[1], cnt);
1210
1211         return rlen;
1212 }
1213
1214 static void tcpm_handle_vdm_request(struct tcpm_port *port,
1215                                     const __le32 *payload, int cnt)
1216 {
1217         int rlen = 0;
1218         u32 response[8] = { };
1219         u32 p0 = le32_to_cpu(payload[0]);
1220
1221         if (port->vdm_state == VDM_STATE_BUSY) {
1222                 /* If UFP responded busy retry after timeout */
1223                 if (PD_VDO_CMDT(p0) == CMDT_RSP_BUSY) {
1224                         port->vdm_state = VDM_STATE_WAIT_RSP_BUSY;
1225                         port->vdo_retry = (p0 & ~VDO_CMDT_MASK) |
1226                                 CMDT_INIT;
1227                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1228                                          msecs_to_jiffies(PD_T_VDM_BUSY));
1229                         return;
1230                 }
1231                 port->vdm_state = VDM_STATE_DONE;
1232         }
1233
1234         if (PD_VDO_SVDM(p0))
1235                 rlen = tcpm_pd_svdm(port, payload, cnt, response);
1236
1237         if (rlen > 0) {
1238                 tcpm_queue_vdm(port, response[0], &response[1], rlen - 1);
1239                 mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1240         }
1241 }
1242
1243 static void tcpm_send_vdm(struct tcpm_port *port, u32 vid, int cmd,
1244                           const u32 *data, int count)
1245 {
1246         u32 header;
1247
1248         if (WARN_ON(count > VDO_MAX_SIZE - 1))
1249                 count = VDO_MAX_SIZE - 1;
1250
1251         /* set VDM header with VID & CMD */
1252         header = VDO(vid, ((vid & USB_SID_PD) == USB_SID_PD) ?
1253                         1 : (PD_VDO_CMD(cmd) <= CMD_ATTENTION), cmd);
1254         tcpm_queue_vdm(port, header, data, count);
1255
1256         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1257 }
1258
1259 static unsigned int vdm_ready_timeout(u32 vdm_hdr)
1260 {
1261         unsigned int timeout;
1262         int cmd = PD_VDO_CMD(vdm_hdr);
1263
1264         /* its not a structured VDM command */
1265         if (!PD_VDO_SVDM(vdm_hdr))
1266                 return PD_T_VDM_UNSTRUCTURED;
1267
1268         switch (PD_VDO_CMDT(vdm_hdr)) {
1269         case CMDT_INIT:
1270                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1271                         timeout = PD_T_VDM_WAIT_MODE_E;
1272                 else
1273                         timeout = PD_T_VDM_SNDR_RSP;
1274                 break;
1275         default:
1276                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1277                         timeout = PD_T_VDM_E_MODE;
1278                 else
1279                         timeout = PD_T_VDM_RCVR_RSP;
1280                 break;
1281         }
1282         return timeout;
1283 }
1284
1285 static void vdm_run_state_machine(struct tcpm_port *port)
1286 {
1287         struct pd_message msg;
1288         int i, res;
1289
1290         switch (port->vdm_state) {
1291         case VDM_STATE_READY:
1292                 /* Only transmit VDM if attached */
1293                 if (!port->attached) {
1294                         port->vdm_state = VDM_STATE_ERR_BUSY;
1295                         break;
1296                 }
1297
1298                 /*
1299                  * if there's traffic or we're not in PDO ready state don't send
1300                  * a VDM.
1301                  */
1302                 if (port->state != SRC_READY && port->state != SNK_READY)
1303                         break;
1304
1305                 /* Prepare and send VDM */
1306                 memset(&msg, 0, sizeof(msg));
1307                 msg.header = PD_HEADER_LE(PD_DATA_VENDOR_DEF,
1308                                           port->pwr_role,
1309                                           port->data_role,
1310                                           port->negotiated_rev,
1311                                           port->message_id, port->vdo_count);
1312                 for (i = 0; i < port->vdo_count; i++)
1313                         msg.payload[i] = cpu_to_le32(port->vdo_data[i]);
1314                 res = tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1315                 if (res < 0) {
1316                         port->vdm_state = VDM_STATE_ERR_SEND;
1317                 } else {
1318                         unsigned long timeout;
1319
1320                         port->vdm_retries = 0;
1321                         port->vdm_state = VDM_STATE_BUSY;
1322                         timeout = vdm_ready_timeout(port->vdo_data[0]);
1323                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1324                                          timeout);
1325                 }
1326                 break;
1327         case VDM_STATE_WAIT_RSP_BUSY:
1328                 port->vdo_data[0] = port->vdo_retry;
1329                 port->vdo_count = 1;
1330                 port->vdm_state = VDM_STATE_READY;
1331                 break;
1332         case VDM_STATE_BUSY:
1333                 port->vdm_state = VDM_STATE_ERR_TMOUT;
1334                 break;
1335         case VDM_STATE_ERR_SEND:
1336                 /*
1337                  * A partner which does not support USB PD will not reply,
1338                  * so this is not a fatal error. At the same time, some
1339                  * devices may not return GoodCRC under some circumstances,
1340                  * so we need to retry.
1341                  */
1342                 if (port->vdm_retries < 3) {
1343                         tcpm_log(port, "VDM Tx error, retry");
1344                         port->vdm_retries++;
1345                         port->vdm_state = VDM_STATE_READY;
1346                 }
1347                 break;
1348         default:
1349                 break;
1350         }
1351 }
1352
1353 static void vdm_state_machine_work(struct work_struct *work)
1354 {
1355         struct tcpm_port *port = container_of(work, struct tcpm_port,
1356                                               vdm_state_machine.work);
1357         enum vdm_states prev_state;
1358
1359         mutex_lock(&port->lock);
1360
1361         /*
1362          * Continue running as long as the port is not busy and there was
1363          * a state change.
1364          */
1365         do {
1366                 prev_state = port->vdm_state;
1367                 vdm_run_state_machine(port);
1368         } while (port->vdm_state != prev_state &&
1369                  port->vdm_state != VDM_STATE_BUSY);
1370
1371         mutex_unlock(&port->lock);
1372 }
1373
1374 enum pdo_err {
1375         PDO_NO_ERR,
1376         PDO_ERR_NO_VSAFE5V,
1377         PDO_ERR_VSAFE5V_NOT_FIRST,
1378         PDO_ERR_PDO_TYPE_NOT_IN_ORDER,
1379         PDO_ERR_FIXED_NOT_SORTED,
1380         PDO_ERR_VARIABLE_BATT_NOT_SORTED,
1381         PDO_ERR_DUPE_PDO,
1382         PDO_ERR_PPS_APDO_NOT_SORTED,
1383         PDO_ERR_DUPE_PPS_APDO,
1384 };
1385
1386 static const char * const pdo_err_msg[] = {
1387         [PDO_ERR_NO_VSAFE5V] =
1388         " err: source/sink caps should atleast have vSafe5V",
1389         [PDO_ERR_VSAFE5V_NOT_FIRST] =
1390         " err: vSafe5V Fixed Supply Object Shall always be the first object",
1391         [PDO_ERR_PDO_TYPE_NOT_IN_ORDER] =
1392         " err: PDOs should be in the following order: Fixed; Battery; Variable",
1393         [PDO_ERR_FIXED_NOT_SORTED] =
1394         " err: Fixed supply pdos should be in increasing order of their fixed voltage",
1395         [PDO_ERR_VARIABLE_BATT_NOT_SORTED] =
1396         " err: Variable/Battery supply pdos should be in increasing order of their minimum voltage",
1397         [PDO_ERR_DUPE_PDO] =
1398         " err: Variable/Batt supply pdos cannot have same min/max voltage",
1399         [PDO_ERR_PPS_APDO_NOT_SORTED] =
1400         " err: Programmable power supply apdos should be in increasing order of their maximum voltage",
1401         [PDO_ERR_DUPE_PPS_APDO] =
1402         " err: Programmable power supply apdos cannot have same min/max voltage and max current",
1403 };
1404
1405 static enum pdo_err tcpm_caps_err(struct tcpm_port *port, const u32 *pdo,
1406                                   unsigned int nr_pdo)
1407 {
1408         unsigned int i;
1409
1410         /* Should at least contain vSafe5v */
1411         if (nr_pdo < 1)
1412                 return PDO_ERR_NO_VSAFE5V;
1413
1414         /* The vSafe5V Fixed Supply Object Shall always be the first object */
1415         if (pdo_type(pdo[0]) != PDO_TYPE_FIXED ||
1416             pdo_fixed_voltage(pdo[0]) != VSAFE5V)
1417                 return PDO_ERR_VSAFE5V_NOT_FIRST;
1418
1419         for (i = 1; i < nr_pdo; i++) {
1420                 if (pdo_type(pdo[i]) < pdo_type(pdo[i - 1])) {
1421                         return PDO_ERR_PDO_TYPE_NOT_IN_ORDER;
1422                 } else if (pdo_type(pdo[i]) == pdo_type(pdo[i - 1])) {
1423                         enum pd_pdo_type type = pdo_type(pdo[i]);
1424
1425                         switch (type) {
1426                         /*
1427                          * The remaining Fixed Supply Objects, if
1428                          * present, shall be sent in voltage order;
1429                          * lowest to highest.
1430                          */
1431                         case PDO_TYPE_FIXED:
1432                                 if (pdo_fixed_voltage(pdo[i]) <=
1433                                     pdo_fixed_voltage(pdo[i - 1]))
1434                                         return PDO_ERR_FIXED_NOT_SORTED;
1435                                 break;
1436                         /*
1437                          * The Battery Supply Objects and Variable
1438                          * supply, if present shall be sent in Minimum
1439                          * Voltage order; lowest to highest.
1440                          */
1441                         case PDO_TYPE_VAR:
1442                         case PDO_TYPE_BATT:
1443                                 if (pdo_min_voltage(pdo[i]) <
1444                                     pdo_min_voltage(pdo[i - 1]))
1445                                         return PDO_ERR_VARIABLE_BATT_NOT_SORTED;
1446                                 else if ((pdo_min_voltage(pdo[i]) ==
1447                                           pdo_min_voltage(pdo[i - 1])) &&
1448                                          (pdo_max_voltage(pdo[i]) ==
1449                                           pdo_max_voltage(pdo[i - 1])))
1450                                         return PDO_ERR_DUPE_PDO;
1451                                 break;
1452                         /*
1453                          * The Programmable Power Supply APDOs, if present,
1454                          * shall be sent in Maximum Voltage order;
1455                          * lowest to highest.
1456                          */
1457                         case PDO_TYPE_APDO:
1458                                 if (pdo_apdo_type(pdo[i]) != APDO_TYPE_PPS)
1459                                         break;
1460
1461                                 if (pdo_pps_apdo_max_voltage(pdo[i]) <
1462                                     pdo_pps_apdo_max_voltage(pdo[i - 1]))
1463                                         return PDO_ERR_PPS_APDO_NOT_SORTED;
1464                                 else if (pdo_pps_apdo_min_voltage(pdo[i]) ==
1465                                           pdo_pps_apdo_min_voltage(pdo[i - 1]) &&
1466                                          pdo_pps_apdo_max_voltage(pdo[i]) ==
1467                                           pdo_pps_apdo_max_voltage(pdo[i - 1]) &&
1468                                          pdo_pps_apdo_max_current(pdo[i]) ==
1469                                           pdo_pps_apdo_max_current(pdo[i - 1]))
1470                                         return PDO_ERR_DUPE_PPS_APDO;
1471                                 break;
1472                         default:
1473                                 tcpm_log_force(port, " Unknown pdo type");
1474                         }
1475                 }
1476         }
1477
1478         return PDO_NO_ERR;
1479 }
1480
1481 static int tcpm_validate_caps(struct tcpm_port *port, const u32 *pdo,
1482                               unsigned int nr_pdo)
1483 {
1484         enum pdo_err err_index = tcpm_caps_err(port, pdo, nr_pdo);
1485
1486         if (err_index != PDO_NO_ERR) {
1487                 tcpm_log_force(port, " %s", pdo_err_msg[err_index]);
1488                 return -EINVAL;
1489         }
1490
1491         return 0;
1492 }
1493
1494 static int tcpm_altmode_enter(struct typec_altmode *altmode)
1495 {
1496         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
1497         u32 header;
1498
1499         mutex_lock(&port->lock);
1500         header = VDO(altmode->svid, 1, CMD_ENTER_MODE);
1501         header |= VDO_OPOS(altmode->mode);
1502
1503         tcpm_queue_vdm(port, header, NULL, 0);
1504         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1505         mutex_unlock(&port->lock);
1506
1507         return 0;
1508 }
1509
1510 static int tcpm_altmode_exit(struct typec_altmode *altmode)
1511 {
1512         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
1513         u32 header;
1514
1515         mutex_lock(&port->lock);
1516         header = VDO(altmode->svid, 1, CMD_EXIT_MODE);
1517         header |= VDO_OPOS(altmode->mode);
1518
1519         tcpm_queue_vdm(port, header, NULL, 0);
1520         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1521         mutex_unlock(&port->lock);
1522
1523         return 0;
1524 }
1525
1526 static int tcpm_altmode_vdm(struct typec_altmode *altmode,
1527                             u32 header, const u32 *data, int count)
1528 {
1529         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
1530
1531         mutex_lock(&port->lock);
1532         tcpm_queue_vdm(port, header, data, count - 1);
1533         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1534         mutex_unlock(&port->lock);
1535
1536         return 0;
1537 }
1538
1539 static const struct typec_altmode_ops tcpm_altmode_ops = {
1540         .enter = tcpm_altmode_enter,
1541         .exit = tcpm_altmode_exit,
1542         .vdm = tcpm_altmode_vdm,
1543 };
1544
1545 /*
1546  * PD (data, control) command handling functions
1547  */
1548 static inline enum tcpm_state ready_state(struct tcpm_port *port)
1549 {
1550         if (port->pwr_role == TYPEC_SOURCE)
1551                 return SRC_READY;
1552         else
1553                 return SNK_READY;
1554 }
1555
1556 static int tcpm_pd_send_control(struct tcpm_port *port,
1557                                 enum pd_ctrl_msg_type type);
1558
1559 static void tcpm_handle_alert(struct tcpm_port *port, const __le32 *payload,
1560                               int cnt)
1561 {
1562         u32 p0 = le32_to_cpu(payload[0]);
1563         unsigned int type = usb_pd_ado_type(p0);
1564
1565         if (!type) {
1566                 tcpm_log(port, "Alert message received with no type");
1567                 return;
1568         }
1569
1570         /* Just handling non-battery alerts for now */
1571         if (!(type & USB_PD_ADO_TYPE_BATT_STATUS_CHANGE)) {
1572                 switch (port->state) {
1573                 case SRC_READY:
1574                 case SNK_READY:
1575                         tcpm_set_state(port, GET_STATUS_SEND, 0);
1576                         break;
1577                 default:
1578                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1579                         break;
1580                 }
1581         }
1582 }
1583
1584 static void tcpm_pd_data_request(struct tcpm_port *port,
1585                                  const struct pd_message *msg)
1586 {
1587         enum pd_data_msg_type type = pd_header_type_le(msg->header);
1588         unsigned int cnt = pd_header_cnt_le(msg->header);
1589         unsigned int rev = pd_header_rev_le(msg->header);
1590         unsigned int i;
1591
1592         switch (type) {
1593         case PD_DATA_SOURCE_CAP:
1594                 if (port->pwr_role != TYPEC_SINK)
1595                         break;
1596
1597                 for (i = 0; i < cnt; i++)
1598                         port->source_caps[i] = le32_to_cpu(msg->payload[i]);
1599
1600                 port->nr_source_caps = cnt;
1601
1602                 tcpm_log_source_caps(port);
1603
1604                 tcpm_validate_caps(port, port->source_caps,
1605                                    port->nr_source_caps);
1606
1607                 /*
1608                  * Adjust revision in subsequent message headers, as required,
1609                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1610                  * support Rev 1.0 so just do nothing in that scenario.
1611                  */
1612                 if (rev == PD_REV10)
1613                         break;
1614
1615                 if (rev < PD_MAX_REV)
1616                         port->negotiated_rev = rev;
1617
1618                 /*
1619                  * This message may be received even if VBUS is not
1620                  * present. This is quite unexpected; see USB PD
1621                  * specification, sections 8.3.3.6.3.1 and 8.3.3.6.3.2.
1622                  * However, at the same time, we must be ready to
1623                  * receive this message and respond to it 15ms after
1624                  * receiving PS_RDY during power swap operations, no matter
1625                  * if VBUS is available or not (USB PD specification,
1626                  * section 6.5.9.2).
1627                  * So we need to accept the message either way,
1628                  * but be prepared to keep waiting for VBUS after it was
1629                  * handled.
1630                  */
1631                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
1632                 break;
1633         case PD_DATA_REQUEST:
1634                 if (port->pwr_role != TYPEC_SOURCE ||
1635                     cnt != 1) {
1636                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1637                         break;
1638                 }
1639
1640                 /*
1641                  * Adjust revision in subsequent message headers, as required,
1642                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1643                  * support Rev 1.0 so just reject in that scenario.
1644                  */
1645                 if (rev == PD_REV10) {
1646                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1647                         break;
1648                 }
1649
1650                 if (rev < PD_MAX_REV)
1651                         port->negotiated_rev = rev;
1652
1653                 port->sink_request = le32_to_cpu(msg->payload[0]);
1654                 tcpm_set_state(port, SRC_NEGOTIATE_CAPABILITIES, 0);
1655                 break;
1656         case PD_DATA_SINK_CAP:
1657                 /* We don't do anything with this at the moment... */
1658                 for (i = 0; i < cnt; i++)
1659                         port->sink_caps[i] = le32_to_cpu(msg->payload[i]);
1660                 port->nr_sink_caps = cnt;
1661                 break;
1662         case PD_DATA_VENDOR_DEF:
1663                 tcpm_handle_vdm_request(port, msg->payload, cnt);
1664                 break;
1665         case PD_DATA_BIST:
1666                 if (port->state == SRC_READY || port->state == SNK_READY) {
1667                         port->bist_request = le32_to_cpu(msg->payload[0]);
1668                         tcpm_set_state(port, BIST_RX, 0);
1669                 }
1670                 break;
1671         case PD_DATA_ALERT:
1672                 tcpm_handle_alert(port, msg->payload, cnt);
1673                 break;
1674         case PD_DATA_BATT_STATUS:
1675         case PD_DATA_GET_COUNTRY_INFO:
1676                 /* Currently unsupported */
1677                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1678                 break;
1679         default:
1680                 tcpm_log(port, "Unhandled data message type %#x", type);
1681                 break;
1682         }
1683 }
1684
1685 static void tcpm_pps_complete(struct tcpm_port *port, int result)
1686 {
1687         if (port->pps_pending) {
1688                 port->pps_status = result;
1689                 port->pps_pending = false;
1690                 complete(&port->pps_complete);
1691         }
1692 }
1693
1694 static void tcpm_pd_ctrl_request(struct tcpm_port *port,
1695                                  const struct pd_message *msg)
1696 {
1697         enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1698         enum tcpm_state next_state;
1699
1700         switch (type) {
1701         case PD_CTRL_GOOD_CRC:
1702         case PD_CTRL_PING:
1703                 break;
1704         case PD_CTRL_GET_SOURCE_CAP:
1705                 switch (port->state) {
1706                 case SRC_READY:
1707                 case SNK_READY:
1708                         tcpm_queue_message(port, PD_MSG_DATA_SOURCE_CAP);
1709                         break;
1710                 default:
1711                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1712                         break;
1713                 }
1714                 break;
1715         case PD_CTRL_GET_SINK_CAP:
1716                 switch (port->state) {
1717                 case SRC_READY:
1718                 case SNK_READY:
1719                         tcpm_queue_message(port, PD_MSG_DATA_SINK_CAP);
1720                         break;
1721                 default:
1722                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1723                         break;
1724                 }
1725                 break;
1726         case PD_CTRL_GOTO_MIN:
1727                 break;
1728         case PD_CTRL_PS_RDY:
1729                 switch (port->state) {
1730                 case SNK_TRANSITION_SINK:
1731                         if (port->vbus_present) {
1732                                 tcpm_set_current_limit(port,
1733                                                        port->current_limit,
1734                                                        port->supply_voltage);
1735                                 port->explicit_contract = true;
1736                                 tcpm_set_state(port, SNK_READY, 0);
1737                         } else {
1738                                 /*
1739                                  * Seen after power swap. Keep waiting for VBUS
1740                                  * in a transitional state.
1741                                  */
1742                                 tcpm_set_state(port,
1743                                                SNK_TRANSITION_SINK_VBUS, 0);
1744                         }
1745                         break;
1746                 case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
1747                         tcpm_set_state(port, PR_SWAP_SRC_SNK_SINK_ON, 0);
1748                         break;
1749                 case PR_SWAP_SNK_SRC_SINK_OFF:
1750                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON, 0);
1751                         break;
1752                 case VCONN_SWAP_WAIT_FOR_VCONN:
1753                         tcpm_set_state(port, VCONN_SWAP_TURN_OFF_VCONN, 0);
1754                         break;
1755                 default:
1756                         break;
1757                 }
1758                 break;
1759         case PD_CTRL_REJECT:
1760         case PD_CTRL_WAIT:
1761         case PD_CTRL_NOT_SUPP:
1762                 switch (port->state) {
1763                 case SNK_NEGOTIATE_CAPABILITIES:
1764                         /* USB PD specification, Figure 8-43 */
1765                         if (port->explicit_contract)
1766                                 next_state = SNK_READY;
1767                         else
1768                                 next_state = SNK_WAIT_CAPABILITIES;
1769                         tcpm_set_state(port, next_state, 0);
1770                         break;
1771                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1772                         /* Revert data back from any requested PPS updates */
1773                         port->pps_data.out_volt = port->supply_voltage;
1774                         port->pps_data.op_curr = port->current_limit;
1775                         port->pps_status = (type == PD_CTRL_WAIT ?
1776                                             -EAGAIN : -EOPNOTSUPP);
1777                         tcpm_set_state(port, SNK_READY, 0);
1778                         break;
1779                 case DR_SWAP_SEND:
1780                         port->swap_status = (type == PD_CTRL_WAIT ?
1781                                              -EAGAIN : -EOPNOTSUPP);
1782                         tcpm_set_state(port, DR_SWAP_CANCEL, 0);
1783                         break;
1784                 case PR_SWAP_SEND:
1785                         port->swap_status = (type == PD_CTRL_WAIT ?
1786                                              -EAGAIN : -EOPNOTSUPP);
1787                         tcpm_set_state(port, PR_SWAP_CANCEL, 0);
1788                         break;
1789                 case VCONN_SWAP_SEND:
1790                         port->swap_status = (type == PD_CTRL_WAIT ?
1791                                              -EAGAIN : -EOPNOTSUPP);
1792                         tcpm_set_state(port, VCONN_SWAP_CANCEL, 0);
1793                         break;
1794                 default:
1795                         break;
1796                 }
1797                 break;
1798         case PD_CTRL_ACCEPT:
1799                 switch (port->state) {
1800                 case SNK_NEGOTIATE_CAPABILITIES:
1801                         port->pps_data.active = false;
1802                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1803                         break;
1804                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1805                         port->pps_data.active = true;
1806                         port->supply_voltage = port->pps_data.out_volt;
1807                         port->current_limit = port->pps_data.op_curr;
1808                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1809                         break;
1810                 case SOFT_RESET_SEND:
1811                         port->message_id = 0;
1812                         port->rx_msgid = -1;
1813                         if (port->pwr_role == TYPEC_SOURCE)
1814                                 next_state = SRC_SEND_CAPABILITIES;
1815                         else
1816                                 next_state = SNK_WAIT_CAPABILITIES;
1817                         tcpm_set_state(port, next_state, 0);
1818                         break;
1819                 case DR_SWAP_SEND:
1820                         tcpm_set_state(port, DR_SWAP_CHANGE_DR, 0);
1821                         break;
1822                 case PR_SWAP_SEND:
1823                         tcpm_set_state(port, PR_SWAP_START, 0);
1824                         break;
1825                 case VCONN_SWAP_SEND:
1826                         tcpm_set_state(port, VCONN_SWAP_START, 0);
1827                         break;
1828                 default:
1829                         break;
1830                 }
1831                 break;
1832         case PD_CTRL_SOFT_RESET:
1833                 tcpm_set_state(port, SOFT_RESET, 0);
1834                 break;
1835         case PD_CTRL_DR_SWAP:
1836                 if (port->port_type != TYPEC_PORT_DRP) {
1837                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1838                         break;
1839                 }
1840                 /*
1841                  * XXX
1842                  * 6.3.9: If an alternate mode is active, a request to swap
1843                  * alternate modes shall trigger a port reset.
1844                  */
1845                 switch (port->state) {
1846                 case SRC_READY:
1847                 case SNK_READY:
1848                         tcpm_set_state(port, DR_SWAP_ACCEPT, 0);
1849                         break;
1850                 default:
1851                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1852                         break;
1853                 }
1854                 break;
1855         case PD_CTRL_PR_SWAP:
1856                 if (port->port_type != TYPEC_PORT_DRP) {
1857                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1858                         break;
1859                 }
1860                 switch (port->state) {
1861                 case SRC_READY:
1862                 case SNK_READY:
1863                         tcpm_set_state(port, PR_SWAP_ACCEPT, 0);
1864                         break;
1865                 default:
1866                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1867                         break;
1868                 }
1869                 break;
1870         case PD_CTRL_VCONN_SWAP:
1871                 switch (port->state) {
1872                 case SRC_READY:
1873                 case SNK_READY:
1874                         tcpm_set_state(port, VCONN_SWAP_ACCEPT, 0);
1875                         break;
1876                 default:
1877                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1878                         break;
1879                 }
1880                 break;
1881         case PD_CTRL_GET_SOURCE_CAP_EXT:
1882         case PD_CTRL_GET_STATUS:
1883         case PD_CTRL_FR_SWAP:
1884         case PD_CTRL_GET_PPS_STATUS:
1885         case PD_CTRL_GET_COUNTRY_CODES:
1886                 /* Currently not supported */
1887                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1888                 break;
1889         default:
1890                 tcpm_log(port, "Unhandled ctrl message type %#x", type);
1891                 break;
1892         }
1893 }
1894
1895 static void tcpm_pd_ext_msg_request(struct tcpm_port *port,
1896                                     const struct pd_message *msg)
1897 {
1898         enum pd_ext_msg_type type = pd_header_type_le(msg->header);
1899         unsigned int data_size = pd_ext_header_data_size_le(msg->ext_msg.header);
1900
1901         if (!(msg->ext_msg.header & PD_EXT_HDR_CHUNKED)) {
1902                 tcpm_log(port, "Unchunked extended messages unsupported");
1903                 return;
1904         }
1905
1906         if (data_size > PD_EXT_MAX_CHUNK_DATA) {
1907                 tcpm_log(port, "Chunk handling not yet supported");
1908                 return;
1909         }
1910
1911         switch (type) {
1912         case PD_EXT_STATUS:
1913                 /*
1914                  * If PPS related events raised then get PPS status to clear
1915                  * (see USB PD 3.0 Spec, 6.5.2.4)
1916                  */
1917                 if (msg->ext_msg.data[USB_PD_EXT_SDB_EVENT_FLAGS] &
1918                     USB_PD_EXT_SDB_PPS_EVENTS)
1919                         tcpm_set_state(port, GET_PPS_STATUS_SEND, 0);
1920                 else
1921                         tcpm_set_state(port, ready_state(port), 0);
1922                 break;
1923         case PD_EXT_PPS_STATUS:
1924                 /*
1925                  * For now the PPS status message is used to clear events
1926                  * and nothing more.
1927                  */
1928                 tcpm_set_state(port, ready_state(port), 0);
1929                 break;
1930         case PD_EXT_SOURCE_CAP_EXT:
1931         case PD_EXT_GET_BATT_CAP:
1932         case PD_EXT_GET_BATT_STATUS:
1933         case PD_EXT_BATT_CAP:
1934         case PD_EXT_GET_MANUFACTURER_INFO:
1935         case PD_EXT_MANUFACTURER_INFO:
1936         case PD_EXT_SECURITY_REQUEST:
1937         case PD_EXT_SECURITY_RESPONSE:
1938         case PD_EXT_FW_UPDATE_REQUEST:
1939         case PD_EXT_FW_UPDATE_RESPONSE:
1940         case PD_EXT_COUNTRY_INFO:
1941         case PD_EXT_COUNTRY_CODES:
1942                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1943                 break;
1944         default:
1945                 tcpm_log(port, "Unhandled extended message type %#x", type);
1946                 break;
1947         }
1948 }
1949
1950 static void tcpm_pd_rx_handler(struct work_struct *work)
1951 {
1952         struct pd_rx_event *event = container_of(work,
1953                                                  struct pd_rx_event, work);
1954         const struct pd_message *msg = &event->msg;
1955         unsigned int cnt = pd_header_cnt_le(msg->header);
1956         struct tcpm_port *port = event->port;
1957
1958         mutex_lock(&port->lock);
1959
1960         tcpm_log(port, "PD RX, header: %#x [%d]", le16_to_cpu(msg->header),
1961                  port->attached);
1962
1963         if (port->attached) {
1964                 enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1965                 unsigned int msgid = pd_header_msgid_le(msg->header);
1966
1967                 /*
1968                  * USB PD standard, 6.6.1.2:
1969                  * "... if MessageID value in a received Message is the
1970                  * same as the stored value, the receiver shall return a
1971                  * GoodCRC Message with that MessageID value and drop
1972                  * the Message (this is a retry of an already received
1973                  * Message). Note: this shall not apply to the Soft_Reset
1974                  * Message which always has a MessageID value of zero."
1975                  */
1976                 if (msgid == port->rx_msgid && type != PD_CTRL_SOFT_RESET)
1977                         goto done;
1978                 port->rx_msgid = msgid;
1979
1980                 /*
1981                  * If both ends believe to be DFP/host, we have a data role
1982                  * mismatch.
1983                  */
1984                 if (!!(le16_to_cpu(msg->header) & PD_HEADER_DATA_ROLE) ==
1985                     (port->data_role == TYPEC_HOST)) {
1986                         tcpm_log(port,
1987                                  "Data role mismatch, initiating error recovery");
1988                         tcpm_set_state(port, ERROR_RECOVERY, 0);
1989                 } else {
1990                         if (msg->header & PD_HEADER_EXT_HDR)
1991                                 tcpm_pd_ext_msg_request(port, msg);
1992                         else if (cnt)
1993                                 tcpm_pd_data_request(port, msg);
1994                         else
1995                                 tcpm_pd_ctrl_request(port, msg);
1996                 }
1997         }
1998
1999 done:
2000         mutex_unlock(&port->lock);
2001         kfree(event);
2002 }
2003
2004 void tcpm_pd_receive(struct tcpm_port *port, const struct pd_message *msg)
2005 {
2006         struct pd_rx_event *event;
2007
2008         event = kzalloc(sizeof(*event), GFP_ATOMIC);
2009         if (!event)
2010                 return;
2011
2012         INIT_WORK(&event->work, tcpm_pd_rx_handler);
2013         event->port = port;
2014         memcpy(&event->msg, msg, sizeof(*msg));
2015         queue_work(port->wq, &event->work);
2016 }
2017 EXPORT_SYMBOL_GPL(tcpm_pd_receive);
2018
2019 static int tcpm_pd_send_control(struct tcpm_port *port,
2020                                 enum pd_ctrl_msg_type type)
2021 {
2022         struct pd_message msg;
2023
2024         memset(&msg, 0, sizeof(msg));
2025         msg.header = PD_HEADER_LE(type, port->pwr_role,
2026                                   port->data_role,
2027                                   port->negotiated_rev,
2028                                   port->message_id, 0);
2029
2030         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2031 }
2032
2033 /*
2034  * Send queued message without affecting state.
2035  * Return true if state machine should go back to sleep,
2036  * false otherwise.
2037  */
2038 static bool tcpm_send_queued_message(struct tcpm_port *port)
2039 {
2040         enum pd_msg_request queued_message;
2041
2042         do {
2043                 queued_message = port->queued_message;
2044                 port->queued_message = PD_MSG_NONE;
2045
2046                 switch (queued_message) {
2047                 case PD_MSG_CTRL_WAIT:
2048                         tcpm_pd_send_control(port, PD_CTRL_WAIT);
2049                         break;
2050                 case PD_MSG_CTRL_REJECT:
2051                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
2052                         break;
2053                 case PD_MSG_CTRL_NOT_SUPP:
2054                         tcpm_pd_send_control(port, PD_CTRL_NOT_SUPP);
2055                         break;
2056                 case PD_MSG_DATA_SINK_CAP:
2057                         tcpm_pd_send_sink_caps(port);
2058                         break;
2059                 case PD_MSG_DATA_SOURCE_CAP:
2060                         tcpm_pd_send_source_caps(port);
2061                         break;
2062                 default:
2063                         break;
2064                 }
2065         } while (port->queued_message != PD_MSG_NONE);
2066
2067         if (port->delayed_state != INVALID_STATE) {
2068                 if (time_is_after_jiffies(port->delayed_runtime)) {
2069                         mod_delayed_work(port->wq, &port->state_machine,
2070                                          port->delayed_runtime - jiffies);
2071                         return true;
2072                 }
2073                 port->delayed_state = INVALID_STATE;
2074         }
2075         return false;
2076 }
2077
2078 static int tcpm_pd_check_request(struct tcpm_port *port)
2079 {
2080         u32 pdo, rdo = port->sink_request;
2081         unsigned int max, op, pdo_max, index;
2082         enum pd_pdo_type type;
2083
2084         index = rdo_index(rdo);
2085         if (!index || index > port->nr_src_pdo)
2086                 return -EINVAL;
2087
2088         pdo = port->src_pdo[index - 1];
2089         type = pdo_type(pdo);
2090         switch (type) {
2091         case PDO_TYPE_FIXED:
2092         case PDO_TYPE_VAR:
2093                 max = rdo_max_current(rdo);
2094                 op = rdo_op_current(rdo);
2095                 pdo_max = pdo_max_current(pdo);
2096
2097                 if (op > pdo_max)
2098                         return -EINVAL;
2099                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
2100                         return -EINVAL;
2101
2102                 if (type == PDO_TYPE_FIXED)
2103                         tcpm_log(port,
2104                                  "Requested %u mV, %u mA for %u / %u mA",
2105                                  pdo_fixed_voltage(pdo), pdo_max, op, max);
2106                 else
2107                         tcpm_log(port,
2108                                  "Requested %u -> %u mV, %u mA for %u / %u mA",
2109                                  pdo_min_voltage(pdo), pdo_max_voltage(pdo),
2110                                  pdo_max, op, max);
2111                 break;
2112         case PDO_TYPE_BATT:
2113                 max = rdo_max_power(rdo);
2114                 op = rdo_op_power(rdo);
2115                 pdo_max = pdo_max_power(pdo);
2116
2117                 if (op > pdo_max)
2118                         return -EINVAL;
2119                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
2120                         return -EINVAL;
2121                 tcpm_log(port,
2122                          "Requested %u -> %u mV, %u mW for %u / %u mW",
2123                          pdo_min_voltage(pdo), pdo_max_voltage(pdo),
2124                          pdo_max, op, max);
2125                 break;
2126         default:
2127                 return -EINVAL;
2128         }
2129
2130         port->op_vsafe5v = index == 1;
2131
2132         return 0;
2133 }
2134
2135 #define min_power(x, y) min(pdo_max_power(x), pdo_max_power(y))
2136 #define min_current(x, y) min(pdo_max_current(x), pdo_max_current(y))
2137
2138 static int tcpm_pd_select_pdo(struct tcpm_port *port, int *sink_pdo,
2139                               int *src_pdo)
2140 {
2141         unsigned int i, j, max_src_mv = 0, min_src_mv = 0, max_mw = 0,
2142                      max_mv = 0, src_mw = 0, src_ma = 0, max_snk_mv = 0,
2143                      min_snk_mv = 0;
2144         int ret = -EINVAL;
2145
2146         port->pps_data.supported = false;
2147         port->usb_type = POWER_SUPPLY_USB_TYPE_PD;
2148
2149         /*
2150          * Select the source PDO providing the most power which has a
2151          * matchig sink cap.
2152          */
2153         for (i = 0; i < port->nr_source_caps; i++) {
2154                 u32 pdo = port->source_caps[i];
2155                 enum pd_pdo_type type = pdo_type(pdo);
2156
2157                 switch (type) {
2158                 case PDO_TYPE_FIXED:
2159                         max_src_mv = pdo_fixed_voltage(pdo);
2160                         min_src_mv = max_src_mv;
2161                         break;
2162                 case PDO_TYPE_BATT:
2163                 case PDO_TYPE_VAR:
2164                         max_src_mv = pdo_max_voltage(pdo);
2165                         min_src_mv = pdo_min_voltage(pdo);
2166                         break;
2167                 case PDO_TYPE_APDO:
2168                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS) {
2169                                 port->pps_data.supported = true;
2170                                 port->usb_type =
2171                                         POWER_SUPPLY_USB_TYPE_PD_PPS;
2172                         }
2173                         continue;
2174                 default:
2175                         tcpm_log(port, "Invalid source PDO type, ignoring");
2176                         continue;
2177                 }
2178
2179                 switch (type) {
2180                 case PDO_TYPE_FIXED:
2181                 case PDO_TYPE_VAR:
2182                         src_ma = pdo_max_current(pdo);
2183                         src_mw = src_ma * min_src_mv / 1000;
2184                         break;
2185                 case PDO_TYPE_BATT:
2186                         src_mw = pdo_max_power(pdo);
2187                         break;
2188                 case PDO_TYPE_APDO:
2189                         continue;
2190                 default:
2191                         tcpm_log(port, "Invalid source PDO type, ignoring");
2192                         continue;
2193                 }
2194
2195                 for (j = 0; j < port->nr_snk_pdo; j++) {
2196                         pdo = port->snk_pdo[j];
2197
2198                         switch (pdo_type(pdo)) {
2199                         case PDO_TYPE_FIXED:
2200                                 max_snk_mv = pdo_fixed_voltage(pdo);
2201                                 min_snk_mv = max_snk_mv;
2202                                 break;
2203                         case PDO_TYPE_BATT:
2204                         case PDO_TYPE_VAR:
2205                                 max_snk_mv = pdo_max_voltage(pdo);
2206                                 min_snk_mv = pdo_min_voltage(pdo);
2207                                 break;
2208                         case PDO_TYPE_APDO:
2209                                 continue;
2210                         default:
2211                                 tcpm_log(port, "Invalid sink PDO type, ignoring");
2212                                 continue;
2213                         }
2214
2215                         if (max_src_mv <= max_snk_mv &&
2216                                 min_src_mv >= min_snk_mv) {
2217                                 /* Prefer higher voltages if available */
2218                                 if ((src_mw == max_mw && min_src_mv > max_mv) ||
2219                                                         src_mw > max_mw) {
2220                                         *src_pdo = i;
2221                                         *sink_pdo = j;
2222                                         max_mw = src_mw;
2223                                         max_mv = min_src_mv;
2224                                         ret = 0;
2225                                 }
2226                         }
2227                 }
2228         }
2229
2230         return ret;
2231 }
2232
2233 #define min_pps_apdo_current(x, y)      \
2234         min(pdo_pps_apdo_max_current(x), pdo_pps_apdo_max_current(y))
2235
2236 static unsigned int tcpm_pd_select_pps_apdo(struct tcpm_port *port)
2237 {
2238         unsigned int i, j, max_mw = 0, max_mv = 0;
2239         unsigned int min_src_mv, max_src_mv, src_ma, src_mw;
2240         unsigned int min_snk_mv, max_snk_mv, snk_ma;
2241         u32 pdo;
2242         unsigned int src_pdo = 0, snk_pdo = 0;
2243
2244         /*
2245          * Select the source PPS APDO providing the most power while staying
2246          * within the board's limits. We skip the first PDO as this is always
2247          * 5V 3A.
2248          */
2249         for (i = 1; i < port->nr_source_caps; ++i) {
2250                 pdo = port->source_caps[i];
2251
2252                 switch (pdo_type(pdo)) {
2253                 case PDO_TYPE_APDO:
2254                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2255                                 tcpm_log(port, "Not PPS APDO (source), ignoring");
2256                                 continue;
2257                         }
2258
2259                         min_src_mv = pdo_pps_apdo_min_voltage(pdo);
2260                         max_src_mv = pdo_pps_apdo_max_voltage(pdo);
2261                         src_ma = pdo_pps_apdo_max_current(pdo);
2262                         src_mw = (src_ma * max_src_mv) / 1000;
2263
2264                         /*
2265                          * Now search through the sink PDOs to find a matching
2266                          * PPS APDO. Again skip the first sink PDO as this will
2267                          * always be 5V 3A.
2268                          */
2269                         for (j = 1; j < port->nr_snk_pdo; j++) {
2270                                 pdo = port->snk_pdo[j];
2271
2272                                 switch (pdo_type(pdo)) {
2273                                 case PDO_TYPE_APDO:
2274                                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2275                                                 tcpm_log(port,
2276                                                          "Not PPS APDO (sink), ignoring");
2277                                                 continue;
2278                                         }
2279
2280                                         min_snk_mv =
2281                                                 pdo_pps_apdo_min_voltage(pdo);
2282                                         max_snk_mv =
2283                                                 pdo_pps_apdo_max_voltage(pdo);
2284                                         snk_ma =
2285                                                 pdo_pps_apdo_max_current(pdo);
2286                                         break;
2287                                 default:
2288                                         tcpm_log(port,
2289                                                  "Not APDO type (sink), ignoring");
2290                                         continue;
2291                                 }
2292
2293                                 if (max_src_mv <= max_snk_mv &&
2294                                     min_src_mv >= min_snk_mv) {
2295                                         /* Prefer higher voltages if available */
2296                                         if ((src_mw == max_mw &&
2297                                              min_src_mv > max_mv) ||
2298                                             src_mw > max_mw) {
2299                                                 src_pdo = i;
2300                                                 snk_pdo = j;
2301                                                 max_mw = src_mw;
2302                                                 max_mv = max_src_mv;
2303                                         }
2304                                 }
2305                         }
2306
2307                         break;
2308                 default:
2309                         tcpm_log(port, "Not APDO type (source), ignoring");
2310                         continue;
2311                 }
2312         }
2313
2314         if (src_pdo) {
2315                 pdo = port->source_caps[src_pdo];
2316
2317                 port->pps_data.min_volt = pdo_pps_apdo_min_voltage(pdo);
2318                 port->pps_data.max_volt = pdo_pps_apdo_max_voltage(pdo);
2319                 port->pps_data.max_curr =
2320                         min_pps_apdo_current(pdo, port->snk_pdo[snk_pdo]);
2321                 port->pps_data.out_volt =
2322                         min(pdo_pps_apdo_max_voltage(pdo), port->pps_data.out_volt);
2323                 port->pps_data.op_curr =
2324                         min(port->pps_data.max_curr, port->pps_data.op_curr);
2325         }
2326
2327         return src_pdo;
2328 }
2329
2330 static int tcpm_pd_build_request(struct tcpm_port *port, u32 *rdo)
2331 {
2332         unsigned int mv, ma, mw, flags;
2333         unsigned int max_ma, max_mw;
2334         enum pd_pdo_type type;
2335         u32 pdo, matching_snk_pdo;
2336         int src_pdo_index = 0;
2337         int snk_pdo_index = 0;
2338         int ret;
2339
2340         ret = tcpm_pd_select_pdo(port, &snk_pdo_index, &src_pdo_index);
2341         if (ret < 0)
2342                 return ret;
2343
2344         pdo = port->source_caps[src_pdo_index];
2345         matching_snk_pdo = port->snk_pdo[snk_pdo_index];
2346         type = pdo_type(pdo);
2347
2348         switch (type) {
2349         case PDO_TYPE_FIXED:
2350                 mv = pdo_fixed_voltage(pdo);
2351                 break;
2352         case PDO_TYPE_BATT:
2353         case PDO_TYPE_VAR:
2354                 mv = pdo_min_voltage(pdo);
2355                 break;
2356         default:
2357                 tcpm_log(port, "Invalid PDO selected!");
2358                 return -EINVAL;
2359         }
2360
2361         /* Select maximum available current within the sink pdo's limit */
2362         if (type == PDO_TYPE_BATT) {
2363                 mw = min_power(pdo, matching_snk_pdo);
2364                 ma = 1000 * mw / mv;
2365         } else {
2366                 ma = min_current(pdo, matching_snk_pdo);
2367                 mw = ma * mv / 1000;
2368         }
2369
2370         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2371
2372         /* Set mismatch bit if offered power is less than operating power */
2373         max_ma = ma;
2374         max_mw = mw;
2375         if (mw < port->operating_snk_mw) {
2376                 flags |= RDO_CAP_MISMATCH;
2377                 if (type == PDO_TYPE_BATT &&
2378                     (pdo_max_power(matching_snk_pdo) > pdo_max_power(pdo)))
2379                         max_mw = pdo_max_power(matching_snk_pdo);
2380                 else if (pdo_max_current(matching_snk_pdo) >
2381                          pdo_max_current(pdo))
2382                         max_ma = pdo_max_current(matching_snk_pdo);
2383         }
2384
2385         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2386                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2387                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2388                  port->polarity);
2389
2390         if (type == PDO_TYPE_BATT) {
2391                 *rdo = RDO_BATT(src_pdo_index + 1, mw, max_mw, flags);
2392
2393                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mW%s",
2394                          src_pdo_index, mv, mw,
2395                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2396         } else {
2397                 *rdo = RDO_FIXED(src_pdo_index + 1, ma, max_ma, flags);
2398
2399                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mA%s",
2400                          src_pdo_index, mv, ma,
2401                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2402         }
2403
2404         port->current_limit = ma;
2405         port->supply_voltage = mv;
2406
2407         return 0;
2408 }
2409
2410 static int tcpm_pd_send_request(struct tcpm_port *port)
2411 {
2412         struct pd_message msg;
2413         int ret;
2414         u32 rdo;
2415
2416         ret = tcpm_pd_build_request(port, &rdo);
2417         if (ret < 0)
2418                 return ret;
2419
2420         memset(&msg, 0, sizeof(msg));
2421         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2422                                   port->pwr_role,
2423                                   port->data_role,
2424                                   port->negotiated_rev,
2425                                   port->message_id, 1);
2426         msg.payload[0] = cpu_to_le32(rdo);
2427
2428         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2429 }
2430
2431 static int tcpm_pd_build_pps_request(struct tcpm_port *port, u32 *rdo)
2432 {
2433         unsigned int out_mv, op_ma, op_mw, min_mv, max_mv, max_ma, flags;
2434         enum pd_pdo_type type;
2435         unsigned int src_pdo_index;
2436         u32 pdo;
2437
2438         src_pdo_index = tcpm_pd_select_pps_apdo(port);
2439         if (!src_pdo_index)
2440                 return -EOPNOTSUPP;
2441
2442         pdo = port->source_caps[src_pdo_index];
2443         type = pdo_type(pdo);
2444
2445         switch (type) {
2446         case PDO_TYPE_APDO:
2447                 if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2448                         tcpm_log(port, "Invalid APDO selected!");
2449                         return -EINVAL;
2450                 }
2451                 min_mv = port->pps_data.min_volt;
2452                 max_mv = port->pps_data.max_volt;
2453                 max_ma = port->pps_data.max_curr;
2454                 out_mv = port->pps_data.out_volt;
2455                 op_ma = port->pps_data.op_curr;
2456                 break;
2457         default:
2458                 tcpm_log(port, "Invalid PDO selected!");
2459                 return -EINVAL;
2460         }
2461
2462         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2463
2464         op_mw = (op_ma * out_mv) / 1000;
2465         if (op_mw < port->operating_snk_mw) {
2466                 /*
2467                  * Try raising current to meet power needs. If that's not enough
2468                  * then try upping the voltage. If that's still not enough
2469                  * then we've obviously chosen a PPS APDO which really isn't
2470                  * suitable so abandon ship.
2471                  */
2472                 op_ma = (port->operating_snk_mw * 1000) / out_mv;
2473                 if ((port->operating_snk_mw * 1000) % out_mv)
2474                         ++op_ma;
2475                 op_ma += RDO_PROG_CURR_MA_STEP - (op_ma % RDO_PROG_CURR_MA_STEP);
2476
2477                 if (op_ma > max_ma) {
2478                         op_ma = max_ma;
2479                         out_mv = (port->operating_snk_mw * 1000) / op_ma;
2480                         if ((port->operating_snk_mw * 1000) % op_ma)
2481                                 ++out_mv;
2482                         out_mv += RDO_PROG_VOLT_MV_STEP -
2483                                   (out_mv % RDO_PROG_VOLT_MV_STEP);
2484
2485                         if (out_mv > max_mv) {
2486                                 tcpm_log(port, "Invalid PPS APDO selected!");
2487                                 return -EINVAL;
2488                         }
2489                 }
2490         }
2491
2492         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2493                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2494                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2495                  port->polarity);
2496
2497         *rdo = RDO_PROG(src_pdo_index + 1, out_mv, op_ma, flags);
2498
2499         tcpm_log(port, "Requesting APDO %d: %u mV, %u mA",
2500                  src_pdo_index, out_mv, op_ma);
2501
2502         port->pps_data.op_curr = op_ma;
2503         port->pps_data.out_volt = out_mv;
2504
2505         return 0;
2506 }
2507
2508 static int tcpm_pd_send_pps_request(struct tcpm_port *port)
2509 {
2510         struct pd_message msg;
2511         int ret;
2512         u32 rdo;
2513
2514         ret = tcpm_pd_build_pps_request(port, &rdo);
2515         if (ret < 0)
2516                 return ret;
2517
2518         memset(&msg, 0, sizeof(msg));
2519         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2520                                   port->pwr_role,
2521                                   port->data_role,
2522                                   port->negotiated_rev,
2523                                   port->message_id, 1);
2524         msg.payload[0] = cpu_to_le32(rdo);
2525
2526         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2527 }
2528
2529 static int tcpm_set_vbus(struct tcpm_port *port, bool enable)
2530 {
2531         int ret;
2532
2533         if (enable && port->vbus_charge)
2534                 return -EINVAL;
2535
2536         tcpm_log(port, "vbus:=%d charge=%d", enable, port->vbus_charge);
2537
2538         ret = port->tcpc->set_vbus(port->tcpc, enable, port->vbus_charge);
2539         if (ret < 0)
2540                 return ret;
2541
2542         port->vbus_source = enable;
2543         return 0;
2544 }
2545
2546 static int tcpm_set_charge(struct tcpm_port *port, bool charge)
2547 {
2548         int ret;
2549
2550         if (charge && port->vbus_source)
2551                 return -EINVAL;
2552
2553         if (charge != port->vbus_charge) {
2554                 tcpm_log(port, "vbus=%d charge:=%d", port->vbus_source, charge);
2555                 ret = port->tcpc->set_vbus(port->tcpc, port->vbus_source,
2556                                            charge);
2557                 if (ret < 0)
2558                         return ret;
2559         }
2560         port->vbus_charge = charge;
2561         return 0;
2562 }
2563
2564 static bool tcpm_start_toggling(struct tcpm_port *port, enum typec_cc_status cc)
2565 {
2566         int ret;
2567
2568         if (!port->tcpc->start_toggling)
2569                 return false;
2570
2571         tcpm_log_force(port, "Start toggling");
2572         ret = port->tcpc->start_toggling(port->tcpc, port->port_type, cc);
2573         return ret == 0;
2574 }
2575
2576 static void tcpm_set_cc(struct tcpm_port *port, enum typec_cc_status cc)
2577 {
2578         tcpm_log(port, "cc:=%d", cc);
2579         port->cc_req = cc;
2580         port->tcpc->set_cc(port->tcpc, cc);
2581 }
2582
2583 static int tcpm_init_vbus(struct tcpm_port *port)
2584 {
2585         int ret;
2586
2587         ret = port->tcpc->set_vbus(port->tcpc, false, false);
2588         port->vbus_source = false;
2589         port->vbus_charge = false;
2590         return ret;
2591 }
2592
2593 static int tcpm_init_vconn(struct tcpm_port *port)
2594 {
2595         int ret;
2596
2597         ret = port->tcpc->set_vconn(port->tcpc, false);
2598         port->vconn_role = TYPEC_SINK;
2599         return ret;
2600 }
2601
2602 static void tcpm_typec_connect(struct tcpm_port *port)
2603 {
2604         if (!port->connected) {
2605                 /* Make sure we don't report stale identity information */
2606                 memset(&port->partner_ident, 0, sizeof(port->partner_ident));
2607                 port->partner_desc.usb_pd = port->pd_capable;
2608                 if (tcpm_port_is_debug(port))
2609                         port->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
2610                 else if (tcpm_port_is_audio(port))
2611                         port->partner_desc.accessory = TYPEC_ACCESSORY_AUDIO;
2612                 else
2613                         port->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
2614                 port->partner = typec_register_partner(port->typec_port,
2615                                                        &port->partner_desc);
2616                 port->connected = true;
2617         }
2618 }
2619
2620 static int tcpm_src_attach(struct tcpm_port *port)
2621 {
2622         enum typec_cc_polarity polarity =
2623                                 port->cc2 == TYPEC_CC_RD ? TYPEC_POLARITY_CC2
2624                                                          : TYPEC_POLARITY_CC1;
2625         int ret;
2626
2627         if (port->attached)
2628                 return 0;
2629
2630         ret = tcpm_set_polarity(port, polarity);
2631         if (ret < 0)
2632                 return ret;
2633
2634         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2635         if (ret < 0)
2636                 return ret;
2637
2638         ret = port->tcpc->set_pd_rx(port->tcpc, true);
2639         if (ret < 0)
2640                 goto out_disable_mux;
2641
2642         /*
2643          * USB Type-C specification, version 1.2,
2644          * chapter 4.5.2.2.8.1 (Attached.SRC Requirements)
2645          * Enable VCONN only if the non-RD port is set to RA.
2646          */
2647         if ((polarity == TYPEC_POLARITY_CC1 && port->cc2 == TYPEC_CC_RA) ||
2648             (polarity == TYPEC_POLARITY_CC2 && port->cc1 == TYPEC_CC_RA)) {
2649                 ret = tcpm_set_vconn(port, true);
2650                 if (ret < 0)
2651                         goto out_disable_pd;
2652         }
2653
2654         ret = tcpm_set_vbus(port, true);
2655         if (ret < 0)
2656                 goto out_disable_vconn;
2657
2658         port->pd_capable = false;
2659
2660         port->partner = NULL;
2661
2662         port->attached = true;
2663         port->send_discover = true;
2664
2665         return 0;
2666
2667 out_disable_vconn:
2668         tcpm_set_vconn(port, false);
2669 out_disable_pd:
2670         port->tcpc->set_pd_rx(port->tcpc, false);
2671 out_disable_mux:
2672         tcpm_mux_set(port, TYPEC_STATE_SAFE, USB_ROLE_NONE,
2673                      TYPEC_ORIENTATION_NONE);
2674         return ret;
2675 }
2676
2677 static void tcpm_typec_disconnect(struct tcpm_port *port)
2678 {
2679         if (port->connected) {
2680                 typec_unregister_partner(port->partner);
2681                 port->partner = NULL;
2682                 port->connected = false;
2683         }
2684 }
2685
2686 static void tcpm_unregister_altmodes(struct tcpm_port *port)
2687 {
2688         struct pd_mode_data *modep = &port->mode_data;
2689         int i;
2690
2691         for (i = 0; i < modep->altmodes; i++) {
2692                 typec_unregister_altmode(port->partner_altmode[i]);
2693                 port->partner_altmode[i] = NULL;
2694         }
2695
2696         memset(modep, 0, sizeof(*modep));
2697 }
2698
2699 static void tcpm_reset_port(struct tcpm_port *port)
2700 {
2701         tcpm_unregister_altmodes(port);
2702         tcpm_typec_disconnect(port);
2703         port->attached = false;
2704         port->pd_capable = false;
2705         port->pps_data.supported = false;
2706
2707         /*
2708          * First Rx ID should be 0; set this to a sentinel of -1 so that
2709          * we can check tcpm_pd_rx_handler() if we had seen it before.
2710          */
2711         port->rx_msgid = -1;
2712
2713         port->tcpc->set_pd_rx(port->tcpc, false);
2714         tcpm_init_vbus(port);   /* also disables charging */
2715         tcpm_init_vconn(port);
2716         tcpm_set_current_limit(port, 0, 0);
2717         tcpm_set_polarity(port, TYPEC_POLARITY_CC1);
2718         tcpm_mux_set(port, TYPEC_STATE_SAFE, USB_ROLE_NONE,
2719                      TYPEC_ORIENTATION_NONE);
2720         tcpm_set_attached_state(port, false);
2721         port->try_src_count = 0;
2722         port->try_snk_count = 0;
2723         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
2724
2725         power_supply_changed(port->psy);
2726 }
2727
2728 static void tcpm_detach(struct tcpm_port *port)
2729 {
2730         if (tcpm_port_is_disconnected(port))
2731                 port->hard_reset_count = 0;
2732
2733         if (!port->attached)
2734                 return;
2735
2736         tcpm_reset_port(port);
2737 }
2738
2739 static void tcpm_src_detach(struct tcpm_port *port)
2740 {
2741         tcpm_detach(port);
2742 }
2743
2744 static int tcpm_snk_attach(struct tcpm_port *port)
2745 {
2746         int ret;
2747
2748         if (port->attached)
2749                 return 0;
2750
2751         ret = tcpm_set_polarity(port, port->cc2 != TYPEC_CC_OPEN ?
2752                                 TYPEC_POLARITY_CC2 : TYPEC_POLARITY_CC1);
2753         if (ret < 0)
2754                 return ret;
2755
2756         ret = tcpm_set_roles(port, true, TYPEC_SINK, TYPEC_DEVICE);
2757         if (ret < 0)
2758                 return ret;
2759
2760         port->pd_capable = false;
2761
2762         port->partner = NULL;
2763
2764         port->attached = true;
2765         port->send_discover = true;
2766
2767         return 0;
2768 }
2769
2770 static void tcpm_snk_detach(struct tcpm_port *port)
2771 {
2772         tcpm_detach(port);
2773 }
2774
2775 static int tcpm_acc_attach(struct tcpm_port *port)
2776 {
2777         int ret;
2778
2779         if (port->attached)
2780                 return 0;
2781
2782         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2783         if (ret < 0)
2784                 return ret;
2785
2786         port->partner = NULL;
2787
2788         tcpm_typec_connect(port);
2789
2790         port->attached = true;
2791
2792         return 0;
2793 }
2794
2795 static void tcpm_acc_detach(struct tcpm_port *port)
2796 {
2797         tcpm_detach(port);
2798 }
2799
2800 static inline enum tcpm_state hard_reset_state(struct tcpm_port *port)
2801 {
2802         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
2803                 return HARD_RESET_SEND;
2804         if (port->pd_capable)
2805                 return ERROR_RECOVERY;
2806         if (port->pwr_role == TYPEC_SOURCE)
2807                 return SRC_UNATTACHED;
2808         if (port->state == SNK_WAIT_CAPABILITIES)
2809                 return SNK_READY;
2810         return SNK_UNATTACHED;
2811 }
2812
2813 static inline enum tcpm_state unattached_state(struct tcpm_port *port)
2814 {
2815         if (port->port_type == TYPEC_PORT_DRP) {
2816                 if (port->pwr_role == TYPEC_SOURCE)
2817                         return SRC_UNATTACHED;
2818                 else
2819                         return SNK_UNATTACHED;
2820         } else if (port->port_type == TYPEC_PORT_SRC) {
2821                 return SRC_UNATTACHED;
2822         }
2823
2824         return SNK_UNATTACHED;
2825 }
2826
2827 static void tcpm_check_send_discover(struct tcpm_port *port)
2828 {
2829         if (port->data_role == TYPEC_HOST && port->send_discover &&
2830             port->pd_capable) {
2831                 tcpm_send_vdm(port, USB_SID_PD, CMD_DISCOVER_IDENT, NULL, 0);
2832                 port->send_discover = false;
2833         }
2834 }
2835
2836 static void tcpm_swap_complete(struct tcpm_port *port, int result)
2837 {
2838         if (port->swap_pending) {
2839                 port->swap_status = result;
2840                 port->swap_pending = false;
2841                 port->non_pd_role_swap = false;
2842                 complete(&port->swap_complete);
2843         }
2844 }
2845
2846 static enum typec_pwr_opmode tcpm_get_pwr_opmode(enum typec_cc_status cc)
2847 {
2848         switch (cc) {
2849         case TYPEC_CC_RP_1_5:
2850                 return TYPEC_PWR_MODE_1_5A;
2851         case TYPEC_CC_RP_3_0:
2852                 return TYPEC_PWR_MODE_3_0A;
2853         case TYPEC_CC_RP_DEF:
2854         default:
2855                 return TYPEC_PWR_MODE_USB;
2856         }
2857 }
2858
2859 static void run_state_machine(struct tcpm_port *port)
2860 {
2861         int ret;
2862         enum typec_pwr_opmode opmode;
2863         unsigned int msecs;
2864
2865         port->enter_state = port->state;
2866         switch (port->state) {
2867         case TOGGLING:
2868                 break;
2869         /* SRC states */
2870         case SRC_UNATTACHED:
2871                 if (!port->non_pd_role_swap)
2872                         tcpm_swap_complete(port, -ENOTCONN);
2873                 tcpm_src_detach(port);
2874                 if (tcpm_start_toggling(port, tcpm_rp_cc(port))) {
2875                         tcpm_set_state(port, TOGGLING, 0);
2876                         break;
2877                 }
2878                 tcpm_set_cc(port, tcpm_rp_cc(port));
2879                 if (port->port_type == TYPEC_PORT_DRP)
2880                         tcpm_set_state(port, SNK_UNATTACHED, PD_T_DRP_SNK);
2881                 break;
2882         case SRC_ATTACH_WAIT:
2883                 if (tcpm_port_is_debug(port))
2884                         tcpm_set_state(port, DEBUG_ACC_ATTACHED,
2885                                        PD_T_CC_DEBOUNCE);
2886                 else if (tcpm_port_is_audio(port))
2887                         tcpm_set_state(port, AUDIO_ACC_ATTACHED,
2888                                        PD_T_CC_DEBOUNCE);
2889                 else if (tcpm_port_is_source(port))
2890                         tcpm_set_state(port,
2891                                        tcpm_try_snk(port) ? SNK_TRY
2892                                                           : SRC_ATTACHED,
2893                                        PD_T_CC_DEBOUNCE);
2894                 break;
2895
2896         case SNK_TRY:
2897                 port->try_snk_count++;
2898                 /*
2899                  * Requirements:
2900                  * - Do not drive vconn or vbus
2901                  * - Terminate CC pins (both) to Rd
2902                  * Action:
2903                  * - Wait for tDRPTry (PD_T_DRP_TRY).
2904                  *   Until then, ignore any state changes.
2905                  */
2906                 tcpm_set_cc(port, TYPEC_CC_RD);
2907                 tcpm_set_state(port, SNK_TRY_WAIT, PD_T_DRP_TRY);
2908                 break;
2909         case SNK_TRY_WAIT:
2910                 if (tcpm_port_is_sink(port)) {
2911                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE, 0);
2912                 } else {
2913                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2914                         port->max_wait = 0;
2915                 }
2916                 break;
2917         case SNK_TRY_WAIT_DEBOUNCE:
2918                 tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS,
2919                                PD_T_PD_DEBOUNCE);
2920                 break;
2921         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
2922                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2923                         tcpm_set_state(port, SNK_ATTACHED, 0);
2924                 } else {
2925                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2926                         port->max_wait = 0;
2927                 }
2928                 break;
2929         case SRC_TRYWAIT:
2930                 tcpm_set_cc(port, tcpm_rp_cc(port));
2931                 if (port->max_wait == 0) {
2932                         port->max_wait = jiffies +
2933                                          msecs_to_jiffies(PD_T_DRP_TRY);
2934                         tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2935                                        PD_T_DRP_TRY);
2936                 } else {
2937                         if (time_is_after_jiffies(port->max_wait))
2938                                 tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2939                                                jiffies_to_msecs(port->max_wait -
2940                                                                 jiffies));
2941                         else
2942                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2943                 }
2944                 break;
2945         case SRC_TRYWAIT_DEBOUNCE:
2946                 tcpm_set_state(port, SRC_ATTACHED, PD_T_CC_DEBOUNCE);
2947                 break;
2948         case SRC_TRYWAIT_UNATTACHED:
2949                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2950                 break;
2951
2952         case SRC_ATTACHED:
2953                 ret = tcpm_src_attach(port);
2954                 tcpm_set_state(port, SRC_UNATTACHED,
2955                                ret < 0 ? 0 : PD_T_PS_SOURCE_ON);
2956                 break;
2957         case SRC_STARTUP:
2958                 opmode =  tcpm_get_pwr_opmode(tcpm_rp_cc(port));
2959                 typec_set_pwr_opmode(port->typec_port, opmode);
2960                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
2961                 port->caps_count = 0;
2962                 port->negotiated_rev = PD_MAX_REV;
2963                 port->message_id = 0;
2964                 port->rx_msgid = -1;
2965                 port->explicit_contract = false;
2966                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
2967                 break;
2968         case SRC_SEND_CAPABILITIES:
2969                 port->caps_count++;
2970                 if (port->caps_count > PD_N_CAPS_COUNT) {
2971                         tcpm_set_state(port, SRC_READY, 0);
2972                         break;
2973                 }
2974                 ret = tcpm_pd_send_source_caps(port);
2975                 if (ret < 0) {
2976                         tcpm_set_state(port, SRC_SEND_CAPABILITIES,
2977                                        PD_T_SEND_SOURCE_CAP);
2978                 } else {
2979                         /*
2980                          * Per standard, we should clear the reset counter here.
2981                          * However, that can result in state machine hang-ups.
2982                          * Reset it only in READY state to improve stability.
2983                          */
2984                         /* port->hard_reset_count = 0; */
2985                         port->caps_count = 0;
2986                         port->pd_capable = true;
2987                         tcpm_set_state_cond(port, SRC_SEND_CAPABILITIES_TIMEOUT,
2988                                             PD_T_SEND_SOURCE_CAP);
2989                 }
2990                 break;
2991         case SRC_SEND_CAPABILITIES_TIMEOUT:
2992                 /*
2993                  * Error recovery for a PD_DATA_SOURCE_CAP reply timeout.
2994                  *
2995                  * PD 2.0 sinks are supposed to accept src-capabilities with a
2996                  * 3.0 header and simply ignore any src PDOs which the sink does
2997                  * not understand such as PPS but some 2.0 sinks instead ignore
2998                  * the entire PD_DATA_SOURCE_CAP message, causing contract
2999                  * negotiation to fail.
3000                  *
3001                  * After PD_N_HARD_RESET_COUNT hard-reset attempts, we try
3002                  * sending src-capabilities with a lower PD revision to
3003                  * make these broken sinks work.
3004                  */
3005                 if (port->hard_reset_count < PD_N_HARD_RESET_COUNT) {
3006                         tcpm_set_state(port, HARD_RESET_SEND, 0);
3007                 } else if (port->negotiated_rev > PD_REV20) {
3008                         port->negotiated_rev--;
3009                         port->hard_reset_count = 0;
3010                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3011                 } else {
3012                         tcpm_set_state(port, hard_reset_state(port), 0);
3013                 }
3014                 break;
3015         case SRC_NEGOTIATE_CAPABILITIES:
3016                 ret = tcpm_pd_check_request(port);
3017                 if (ret < 0) {
3018                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
3019                         if (!port->explicit_contract) {
3020                                 tcpm_set_state(port,
3021                                                SRC_WAIT_NEW_CAPABILITIES, 0);
3022                         } else {
3023                                 tcpm_set_state(port, SRC_READY, 0);
3024                         }
3025                 } else {
3026                         tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3027                         tcpm_set_state(port, SRC_TRANSITION_SUPPLY,
3028                                        PD_T_SRC_TRANSITION);
3029                 }
3030                 break;
3031         case SRC_TRANSITION_SUPPLY:
3032                 /* XXX: regulator_set_voltage(vbus, ...) */
3033                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3034                 port->explicit_contract = true;
3035                 typec_set_pwr_opmode(port->typec_port, TYPEC_PWR_MODE_PD);
3036                 port->pwr_opmode = TYPEC_PWR_MODE_PD;
3037                 tcpm_set_state_cond(port, SRC_READY, 0);
3038                 break;
3039         case SRC_READY:
3040 #if 1
3041                 port->hard_reset_count = 0;
3042 #endif
3043                 port->try_src_count = 0;
3044
3045                 tcpm_swap_complete(port, 0);
3046                 tcpm_typec_connect(port);
3047
3048                 tcpm_check_send_discover(port);
3049                 /*
3050                  * 6.3.5
3051                  * Sending ping messages is not necessary if
3052                  * - the source operates at vSafe5V
3053                  * or
3054                  * - The system is not operating in PD mode
3055                  * or
3056                  * - Both partners are connected using a Type-C connector
3057                  *
3058                  * There is no actual need to send PD messages since the local
3059                  * port type-c and the spec does not clearly say whether PD is
3060                  * possible when type-c is connected to Type-A/B
3061                  */
3062                 break;
3063         case SRC_WAIT_NEW_CAPABILITIES:
3064                 /* Nothing to do... */
3065                 break;
3066
3067         /* SNK states */
3068         case SNK_UNATTACHED:
3069                 if (!port->non_pd_role_swap)
3070                         tcpm_swap_complete(port, -ENOTCONN);
3071                 tcpm_pps_complete(port, -ENOTCONN);
3072                 tcpm_snk_detach(port);
3073                 if (tcpm_start_toggling(port, TYPEC_CC_RD)) {
3074                         tcpm_set_state(port, TOGGLING, 0);
3075                         break;
3076                 }
3077                 tcpm_set_cc(port, TYPEC_CC_RD);
3078                 if (port->port_type == TYPEC_PORT_DRP)
3079                         tcpm_set_state(port, SRC_UNATTACHED, PD_T_DRP_SRC);
3080                 break;
3081         case SNK_ATTACH_WAIT:
3082                 if ((port->cc1 == TYPEC_CC_OPEN &&
3083                      port->cc2 != TYPEC_CC_OPEN) ||
3084                     (port->cc1 != TYPEC_CC_OPEN &&
3085                      port->cc2 == TYPEC_CC_OPEN))
3086                         tcpm_set_state(port, SNK_DEBOUNCED,
3087                                        PD_T_CC_DEBOUNCE);
3088                 else if (tcpm_port_is_disconnected(port))
3089                         tcpm_set_state(port, SNK_UNATTACHED,
3090                                        PD_T_PD_DEBOUNCE);
3091                 break;
3092         case SNK_DEBOUNCED:
3093                 if (tcpm_port_is_disconnected(port))
3094                         tcpm_set_state(port, SNK_UNATTACHED,
3095                                        PD_T_PD_DEBOUNCE);
3096                 else if (port->vbus_present)
3097                         tcpm_set_state(port,
3098                                        tcpm_try_src(port) ? SRC_TRY
3099                                                           : SNK_ATTACHED,
3100                                        0);
3101                 break;
3102         case SRC_TRY:
3103                 port->try_src_count++;
3104                 tcpm_set_cc(port, tcpm_rp_cc(port));
3105                 port->max_wait = 0;
3106                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
3107                 break;
3108         case SRC_TRY_WAIT:
3109                 if (port->max_wait == 0) {
3110                         port->max_wait = jiffies +
3111                                          msecs_to_jiffies(PD_T_DRP_TRY);
3112                         msecs = PD_T_DRP_TRY;
3113                 } else {
3114                         if (time_is_after_jiffies(port->max_wait))
3115                                 msecs = jiffies_to_msecs(port->max_wait -
3116                                                          jiffies);
3117                         else
3118                                 msecs = 0;
3119                 }
3120                 tcpm_set_state(port, SNK_TRYWAIT, msecs);
3121                 break;
3122         case SRC_TRY_DEBOUNCE:
3123                 tcpm_set_state(port, SRC_ATTACHED, PD_T_PD_DEBOUNCE);
3124                 break;
3125         case SNK_TRYWAIT:
3126                 tcpm_set_cc(port, TYPEC_CC_RD);
3127                 tcpm_set_state(port, SNK_TRYWAIT_VBUS, PD_T_CC_DEBOUNCE);
3128                 break;
3129         case SNK_TRYWAIT_VBUS:
3130                 /*
3131                  * TCPM stays in this state indefinitely until VBUS
3132                  * is detected as long as Rp is not detected for
3133                  * more than a time period of tPDDebounce.
3134                  */
3135                 if (port->vbus_present && tcpm_port_is_sink(port)) {
3136                         tcpm_set_state(port, SNK_ATTACHED, 0);
3137                         break;
3138                 }
3139                 if (!tcpm_port_is_sink(port))
3140                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
3141                 break;
3142         case SNK_TRYWAIT_DEBOUNCE:
3143                 tcpm_set_state(port, SNK_UNATTACHED, PD_T_PD_DEBOUNCE);
3144                 break;
3145         case SNK_ATTACHED:
3146                 ret = tcpm_snk_attach(port);
3147                 if (ret < 0)
3148                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3149                 else
3150                         tcpm_set_state(port, SNK_STARTUP, 0);
3151                 break;
3152         case SNK_STARTUP:
3153                 opmode =  tcpm_get_pwr_opmode(port->polarity ?
3154                                               port->cc2 : port->cc1);
3155                 typec_set_pwr_opmode(port->typec_port, opmode);
3156                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
3157                 port->negotiated_rev = PD_MAX_REV;
3158                 port->message_id = 0;
3159                 port->rx_msgid = -1;
3160                 port->explicit_contract = false;
3161                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3162                 break;
3163         case SNK_DISCOVERY:
3164                 if (port->vbus_present) {
3165                         tcpm_set_current_limit(port,
3166                                                tcpm_get_current_limit(port),
3167                                                5000);
3168                         tcpm_set_charge(port, true);
3169                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3170                         break;
3171                 }
3172                 /*
3173                  * For DRP, timeouts differ. Also, handling is supposed to be
3174                  * different and much more complex (dead battery detection;
3175                  * see USB power delivery specification, section 8.3.3.6.1.5.1).
3176                  */
3177                 tcpm_set_state(port, hard_reset_state(port),
3178                                port->port_type == TYPEC_PORT_DRP ?
3179                                         PD_T_DB_DETECT : PD_T_NO_RESPONSE);
3180                 break;
3181         case SNK_DISCOVERY_DEBOUNCE:
3182                 tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE_DONE,
3183                                PD_T_CC_DEBOUNCE);
3184                 break;
3185         case SNK_DISCOVERY_DEBOUNCE_DONE:
3186                 if (!tcpm_port_is_disconnected(port) &&
3187                     tcpm_port_is_sink(port) &&
3188                     time_is_after_jiffies(port->delayed_runtime)) {
3189                         tcpm_set_state(port, SNK_DISCOVERY,
3190                                        jiffies_to_msecs(port->delayed_runtime -
3191                                                         jiffies));
3192                         break;
3193                 }
3194                 tcpm_set_state(port, unattached_state(port), 0);
3195                 break;
3196         case SNK_WAIT_CAPABILITIES:
3197                 ret = port->tcpc->set_pd_rx(port->tcpc, true);
3198                 if (ret < 0) {
3199                         tcpm_set_state(port, SNK_READY, 0);
3200                         break;
3201                 }
3202                 /*
3203                  * If VBUS has never been low, and we time out waiting
3204                  * for source cap, try a soft reset first, in case we
3205                  * were already in a stable contract before this boot.
3206                  * Do this only once.
3207                  */
3208                 if (port->vbus_never_low) {
3209                         port->vbus_never_low = false;
3210                         tcpm_set_state(port, SOFT_RESET_SEND,
3211                                        PD_T_SINK_WAIT_CAP);
3212                 } else {
3213                         tcpm_set_state(port, hard_reset_state(port),
3214                                        PD_T_SINK_WAIT_CAP);
3215                 }
3216                 break;
3217         case SNK_NEGOTIATE_CAPABILITIES:
3218                 port->pd_capable = true;
3219                 port->hard_reset_count = 0;
3220                 ret = tcpm_pd_send_request(port);
3221                 if (ret < 0) {
3222                         /* Let the Source send capabilities again. */
3223                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3224                 } else {
3225                         tcpm_set_state_cond(port, hard_reset_state(port),
3226                                             PD_T_SENDER_RESPONSE);
3227                 }
3228                 break;
3229         case SNK_NEGOTIATE_PPS_CAPABILITIES:
3230                 ret = tcpm_pd_send_pps_request(port);
3231                 if (ret < 0) {
3232                         port->pps_status = ret;
3233                         /*
3234                          * If this was called due to updates to sink
3235                          * capabilities, and pps is no longer valid, we should
3236                          * safely fall back to a standard PDO.
3237                          */
3238                         if (port->update_sink_caps)
3239                                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
3240                         else
3241                                 tcpm_set_state(port, SNK_READY, 0);
3242                 } else {
3243                         tcpm_set_state_cond(port, hard_reset_state(port),
3244                                             PD_T_SENDER_RESPONSE);
3245                 }
3246                 break;
3247         case SNK_TRANSITION_SINK:
3248         case SNK_TRANSITION_SINK_VBUS:
3249                 tcpm_set_state(port, hard_reset_state(port),
3250                                PD_T_PS_TRANSITION);
3251                 break;
3252         case SNK_READY:
3253                 port->try_snk_count = 0;
3254                 port->update_sink_caps = false;
3255                 if (port->explicit_contract) {
3256                         typec_set_pwr_opmode(port->typec_port,
3257                                              TYPEC_PWR_MODE_PD);
3258                         port->pwr_opmode = TYPEC_PWR_MODE_PD;
3259                 }
3260
3261                 tcpm_swap_complete(port, 0);
3262                 tcpm_typec_connect(port);
3263                 tcpm_check_send_discover(port);
3264                 tcpm_pps_complete(port, port->pps_status);
3265
3266                 power_supply_changed(port->psy);
3267
3268                 break;
3269
3270         /* Accessory states */
3271         case ACC_UNATTACHED:
3272                 tcpm_acc_detach(port);
3273                 tcpm_set_state(port, SRC_UNATTACHED, 0);
3274                 break;
3275         case DEBUG_ACC_ATTACHED:
3276         case AUDIO_ACC_ATTACHED:
3277                 ret = tcpm_acc_attach(port);
3278                 if (ret < 0)
3279                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3280                 break;
3281         case AUDIO_ACC_DEBOUNCE:
3282                 tcpm_set_state(port, ACC_UNATTACHED, PD_T_CC_DEBOUNCE);
3283                 break;
3284
3285         /* Hard_Reset states */
3286         case HARD_RESET_SEND:
3287                 tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
3288                 tcpm_set_state(port, HARD_RESET_START, 0);
3289                 break;
3290         case HARD_RESET_START:
3291                 port->hard_reset_count++;
3292                 port->tcpc->set_pd_rx(port->tcpc, false);
3293                 tcpm_unregister_altmodes(port);
3294                 port->send_discover = true;
3295                 if (port->pwr_role == TYPEC_SOURCE)
3296                         tcpm_set_state(port, SRC_HARD_RESET_VBUS_OFF,
3297                                        PD_T_PS_HARD_RESET);
3298                 else
3299                         tcpm_set_state(port, SNK_HARD_RESET_SINK_OFF, 0);
3300                 break;
3301         case SRC_HARD_RESET_VBUS_OFF:
3302                 tcpm_set_vconn(port, true);
3303                 tcpm_set_vbus(port, false);
3304                 tcpm_set_roles(port, port->self_powered, TYPEC_SOURCE,
3305                                TYPEC_HOST);
3306                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SRC_RECOVER);
3307                 break;
3308         case SRC_HARD_RESET_VBUS_ON:
3309                 tcpm_set_vbus(port, true);
3310                 port->tcpc->set_pd_rx(port->tcpc, true);
3311                 tcpm_set_attached_state(port, true);
3312                 tcpm_set_state(port, SRC_UNATTACHED, PD_T_PS_SOURCE_ON);
3313                 break;
3314         case SNK_HARD_RESET_SINK_OFF:
3315                 memset(&port->pps_data, 0, sizeof(port->pps_data));
3316                 tcpm_set_vconn(port, false);
3317                 if (port->pd_capable)
3318                         tcpm_set_charge(port, false);
3319                 tcpm_set_roles(port, port->self_powered, TYPEC_SINK,
3320                                TYPEC_DEVICE);
3321                 /*
3322                  * VBUS may or may not toggle, depending on the adapter.
3323                  * If it doesn't toggle, transition to SNK_HARD_RESET_SINK_ON
3324                  * directly after timeout.
3325                  */
3326                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, PD_T_SAFE_0V);
3327                 break;
3328         case SNK_HARD_RESET_WAIT_VBUS:
3329                 /* Assume we're disconnected if VBUS doesn't come back. */
3330                 tcpm_set_state(port, SNK_UNATTACHED,
3331                                PD_T_SRC_RECOVER_MAX + PD_T_SRC_TURN_ON);
3332                 break;
3333         case SNK_HARD_RESET_SINK_ON:
3334                 /* Note: There is no guarantee that VBUS is on in this state */
3335                 /*
3336                  * XXX:
3337                  * The specification suggests that dual mode ports in sink
3338                  * mode should transition to state PE_SRC_Transition_to_default.
3339                  * See USB power delivery specification chapter 8.3.3.6.1.3.
3340                  * This would mean to to
3341                  * - turn off VCONN, reset power supply
3342                  * - request hardware reset
3343                  * - turn on VCONN
3344                  * - Transition to state PE_Src_Startup
3345                  * SNK only ports shall transition to state Snk_Startup
3346                  * (see chapter 8.3.3.3.8).
3347                  * Similar, dual-mode ports in source mode should transition
3348                  * to PE_SNK_Transition_to_default.
3349                  */
3350                 if (port->pd_capable) {
3351                         tcpm_set_current_limit(port,
3352                                                tcpm_get_current_limit(port),
3353                                                5000);
3354                         tcpm_set_charge(port, true);
3355                 }
3356                 tcpm_set_attached_state(port, true);
3357                 tcpm_set_state(port, SNK_STARTUP, 0);
3358                 break;
3359
3360         /* Soft_Reset states */
3361         case SOFT_RESET:
3362                 port->message_id = 0;
3363                 port->rx_msgid = -1;
3364                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3365                 if (port->pwr_role == TYPEC_SOURCE)
3366                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3367                 else
3368                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3369                 break;
3370         case SOFT_RESET_SEND:
3371                 port->message_id = 0;
3372                 port->rx_msgid = -1;
3373                 if (tcpm_pd_send_control(port, PD_CTRL_SOFT_RESET))
3374                         tcpm_set_state_cond(port, hard_reset_state(port), 0);
3375                 else
3376                         tcpm_set_state_cond(port, hard_reset_state(port),
3377                                             PD_T_SENDER_RESPONSE);
3378                 break;
3379
3380         /* DR_Swap states */
3381         case DR_SWAP_SEND:
3382                 tcpm_pd_send_control(port, PD_CTRL_DR_SWAP);
3383                 tcpm_set_state_cond(port, DR_SWAP_SEND_TIMEOUT,
3384                                     PD_T_SENDER_RESPONSE);
3385                 break;
3386         case DR_SWAP_ACCEPT:
3387                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3388                 tcpm_set_state_cond(port, DR_SWAP_CHANGE_DR, 0);
3389                 break;
3390         case DR_SWAP_SEND_TIMEOUT:
3391                 tcpm_swap_complete(port, -ETIMEDOUT);
3392                 tcpm_set_state(port, ready_state(port), 0);
3393                 break;
3394         case DR_SWAP_CHANGE_DR:
3395                 if (port->data_role == TYPEC_HOST) {
3396                         tcpm_unregister_altmodes(port);
3397                         tcpm_set_roles(port, true, port->pwr_role,
3398                                        TYPEC_DEVICE);
3399                 } else {
3400                         tcpm_set_roles(port, true, port->pwr_role,
3401                                        TYPEC_HOST);
3402                         port->send_discover = true;
3403                 }
3404                 tcpm_set_state(port, ready_state(port), 0);
3405                 break;
3406
3407         /* PR_Swap states */
3408         case PR_SWAP_ACCEPT:
3409                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3410                 tcpm_set_state(port, PR_SWAP_START, 0);
3411                 break;
3412         case PR_SWAP_SEND:
3413                 tcpm_pd_send_control(port, PD_CTRL_PR_SWAP);
3414                 tcpm_set_state_cond(port, PR_SWAP_SEND_TIMEOUT,
3415                                     PD_T_SENDER_RESPONSE);
3416                 break;
3417         case PR_SWAP_SEND_TIMEOUT:
3418                 tcpm_swap_complete(port, -ETIMEDOUT);
3419                 tcpm_set_state(port, ready_state(port), 0);
3420                 break;
3421         case PR_SWAP_START:
3422                 if (port->pwr_role == TYPEC_SOURCE)
3423                         tcpm_set_state(port, PR_SWAP_SRC_SNK_TRANSITION_OFF,
3424                                        PD_T_SRC_TRANSITION);
3425                 else
3426                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SINK_OFF, 0);
3427                 break;
3428         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3429                 tcpm_set_vbus(port, false);
3430                 port->explicit_contract = false;
3431                 /* allow time for Vbus discharge, must be < tSrcSwapStdby */
3432                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF,
3433                                PD_T_SRCSWAPSTDBY);
3434                 break;
3435         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3436                 tcpm_set_cc(port, TYPEC_CC_RD);
3437                 /* allow CC debounce */
3438                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED,
3439                                PD_T_CC_DEBOUNCE);
3440                 break;
3441         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3442                 /*
3443                  * USB-PD standard, 6.2.1.4, Port Power Role:
3444                  * "During the Power Role Swap Sequence, for the initial Source
3445                  * Port, the Port Power Role field shall be set to Sink in the
3446                  * PS_RDY Message indicating that the initial Source’s power
3447                  * supply is turned off"
3448                  */
3449                 tcpm_set_pwr_role(port, TYPEC_SINK);
3450                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
3451                         tcpm_set_state(port, ERROR_RECOVERY, 0);
3452                         break;
3453                 }
3454                 tcpm_set_state_cond(port, SNK_UNATTACHED, PD_T_PS_SOURCE_ON);
3455                 break;
3456         case PR_SWAP_SRC_SNK_SINK_ON:
3457                 tcpm_set_state(port, SNK_STARTUP, 0);
3458                 break;
3459         case PR_SWAP_SNK_SRC_SINK_OFF:
3460                 tcpm_set_charge(port, false);
3461                 tcpm_set_state(port, hard_reset_state(port),
3462                                PD_T_PS_SOURCE_OFF);
3463                 break;
3464         case PR_SWAP_SNK_SRC_SOURCE_ON:
3465                 tcpm_set_cc(port, tcpm_rp_cc(port));
3466                 tcpm_set_vbus(port, true);
3467                 /*
3468                  * allow time VBUS ramp-up, must be < tNewSrc
3469                  * Also, this window overlaps with CC debounce as well.
3470                  * So, Wait for the max of two which is PD_T_NEWSRC
3471                  */
3472                 tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP,
3473                                PD_T_NEWSRC);
3474                 break;
3475         case PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP:
3476                 /*
3477                  * USB PD standard, 6.2.1.4:
3478                  * "Subsequent Messages initiated by the Policy Engine,
3479                  * such as the PS_RDY Message sent to indicate that Vbus
3480                  * is ready, will have the Port Power Role field set to
3481                  * Source."
3482                  */
3483                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
3484                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3485                 tcpm_set_state(port, SRC_STARTUP, PD_T_SWAP_SRC_START);
3486                 break;
3487
3488         case VCONN_SWAP_ACCEPT:
3489                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3490                 tcpm_set_state(port, VCONN_SWAP_START, 0);
3491                 break;
3492         case VCONN_SWAP_SEND:
3493                 tcpm_pd_send_control(port, PD_CTRL_VCONN_SWAP);
3494                 tcpm_set_state(port, VCONN_SWAP_SEND_TIMEOUT,
3495                                PD_T_SENDER_RESPONSE);
3496                 break;
3497         case VCONN_SWAP_SEND_TIMEOUT:
3498                 tcpm_swap_complete(port, -ETIMEDOUT);
3499                 tcpm_set_state(port, ready_state(port), 0);
3500                 break;
3501         case VCONN_SWAP_START:
3502                 if (port->vconn_role == TYPEC_SOURCE)
3503                         tcpm_set_state(port, VCONN_SWAP_WAIT_FOR_VCONN, 0);
3504                 else
3505                         tcpm_set_state(port, VCONN_SWAP_TURN_ON_VCONN, 0);
3506                 break;
3507         case VCONN_SWAP_WAIT_FOR_VCONN:
3508                 tcpm_set_state(port, hard_reset_state(port),
3509                                PD_T_VCONN_SOURCE_ON);
3510                 break;
3511         case VCONN_SWAP_TURN_ON_VCONN:
3512                 tcpm_set_vconn(port, true);
3513                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3514                 tcpm_set_state(port, ready_state(port), 0);
3515                 break;
3516         case VCONN_SWAP_TURN_OFF_VCONN:
3517                 tcpm_set_vconn(port, false);
3518                 tcpm_set_state(port, ready_state(port), 0);
3519                 break;
3520
3521         case DR_SWAP_CANCEL:
3522         case PR_SWAP_CANCEL:
3523         case VCONN_SWAP_CANCEL:
3524                 tcpm_swap_complete(port, port->swap_status);
3525                 if (port->pwr_role == TYPEC_SOURCE)
3526                         tcpm_set_state(port, SRC_READY, 0);
3527                 else
3528                         tcpm_set_state(port, SNK_READY, 0);
3529                 break;
3530
3531         case BIST_RX:
3532                 switch (BDO_MODE_MASK(port->bist_request)) {
3533                 case BDO_MODE_CARRIER2:
3534                         tcpm_pd_transmit(port, TCPC_TX_BIST_MODE_2, NULL);
3535                         break;
3536                 default:
3537                         break;
3538                 }
3539                 /* Always switch to unattached state */
3540                 tcpm_set_state(port, unattached_state(port), 0);
3541                 break;
3542         case GET_STATUS_SEND:
3543                 tcpm_pd_send_control(port, PD_CTRL_GET_STATUS);
3544                 tcpm_set_state(port, GET_STATUS_SEND_TIMEOUT,
3545                                PD_T_SENDER_RESPONSE);
3546                 break;
3547         case GET_STATUS_SEND_TIMEOUT:
3548                 tcpm_set_state(port, ready_state(port), 0);
3549                 break;
3550         case GET_PPS_STATUS_SEND:
3551                 tcpm_pd_send_control(port, PD_CTRL_GET_PPS_STATUS);
3552                 tcpm_set_state(port, GET_PPS_STATUS_SEND_TIMEOUT,
3553                                PD_T_SENDER_RESPONSE);
3554                 break;
3555         case GET_PPS_STATUS_SEND_TIMEOUT:
3556                 tcpm_set_state(port, ready_state(port), 0);
3557                 break;
3558         case ERROR_RECOVERY:
3559                 tcpm_swap_complete(port, -EPROTO);
3560                 tcpm_pps_complete(port, -EPROTO);
3561                 tcpm_set_state(port, PORT_RESET, 0);
3562                 break;
3563         case PORT_RESET:
3564                 tcpm_reset_port(port);
3565                 tcpm_set_cc(port, TYPEC_CC_OPEN);
3566                 tcpm_set_state(port, PORT_RESET_WAIT_OFF,
3567                                PD_T_ERROR_RECOVERY);
3568                 break;
3569         case PORT_RESET_WAIT_OFF:
3570                 tcpm_set_state(port,
3571                                tcpm_default_state(port),
3572                                port->vbus_present ? PD_T_PS_SOURCE_OFF : 0);
3573                 break;
3574         default:
3575                 WARN(1, "Unexpected port state %d\n", port->state);
3576                 break;
3577         }
3578 }
3579
3580 static void tcpm_state_machine_work(struct work_struct *work)
3581 {
3582         struct tcpm_port *port = container_of(work, struct tcpm_port,
3583                                               state_machine.work);
3584         enum tcpm_state prev_state;
3585
3586         mutex_lock(&port->lock);
3587         port->state_machine_running = true;
3588
3589         if (port->queued_message && tcpm_send_queued_message(port))
3590                 goto done;
3591
3592         /* If we were queued due to a delayed state change, update it now */
3593         if (port->delayed_state) {
3594                 tcpm_log(port, "state change %s -> %s [delayed %ld ms]",
3595                          tcpm_states[port->state],
3596                          tcpm_states[port->delayed_state], port->delay_ms);
3597                 port->prev_state = port->state;
3598                 port->state = port->delayed_state;
3599                 port->delayed_state = INVALID_STATE;
3600         }
3601
3602         /*
3603          * Continue running as long as we have (non-delayed) state changes
3604          * to make.
3605          */
3606         do {
3607                 prev_state = port->state;
3608                 run_state_machine(port);
3609                 if (port->queued_message)
3610                         tcpm_send_queued_message(port);
3611         } while (port->state != prev_state && !port->delayed_state);
3612
3613 done:
3614         port->state_machine_running = false;
3615         mutex_unlock(&port->lock);
3616 }
3617
3618 static void _tcpm_cc_change(struct tcpm_port *port, enum typec_cc_status cc1,
3619                             enum typec_cc_status cc2)
3620 {
3621         enum typec_cc_status old_cc1, old_cc2;
3622         enum tcpm_state new_state;
3623
3624         old_cc1 = port->cc1;
3625         old_cc2 = port->cc2;
3626         port->cc1 = cc1;
3627         port->cc2 = cc2;
3628
3629         tcpm_log_force(port,
3630                        "CC1: %u -> %u, CC2: %u -> %u [state %s, polarity %d, %s]",
3631                        old_cc1, cc1, old_cc2, cc2, tcpm_states[port->state],
3632                        port->polarity,
3633                        tcpm_port_is_disconnected(port) ? "disconnected"
3634                                                        : "connected");
3635
3636         switch (port->state) {
3637         case TOGGLING:
3638                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3639                     tcpm_port_is_source(port))
3640                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3641                 else if (tcpm_port_is_sink(port))
3642                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3643                 break;
3644         case SRC_UNATTACHED:
3645         case ACC_UNATTACHED:
3646                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3647                     tcpm_port_is_source(port))
3648                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3649                 break;
3650         case SRC_ATTACH_WAIT:
3651                 if (tcpm_port_is_disconnected(port) ||
3652                     tcpm_port_is_audio_detached(port))
3653                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3654                 else if (cc1 != old_cc1 || cc2 != old_cc2)
3655                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3656                 break;
3657         case SRC_ATTACHED:
3658         case SRC_SEND_CAPABILITIES:
3659         case SRC_READY:
3660                 if (tcpm_port_is_disconnected(port) ||
3661                     !tcpm_port_is_source(port))
3662                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3663                 break;
3664         case SNK_UNATTACHED:
3665                 if (tcpm_port_is_sink(port))
3666                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3667                 break;
3668         case SNK_ATTACH_WAIT:
3669                 if ((port->cc1 == TYPEC_CC_OPEN &&
3670                      port->cc2 != TYPEC_CC_OPEN) ||
3671                     (port->cc1 != TYPEC_CC_OPEN &&
3672                      port->cc2 == TYPEC_CC_OPEN))
3673                         new_state = SNK_DEBOUNCED;
3674                 else if (tcpm_port_is_disconnected(port))
3675                         new_state = SNK_UNATTACHED;
3676                 else
3677                         break;
3678                 if (new_state != port->delayed_state)
3679                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3680                 break;
3681         case SNK_DEBOUNCED:
3682                 if (tcpm_port_is_disconnected(port))
3683                         new_state = SNK_UNATTACHED;
3684                 else if (port->vbus_present)
3685                         new_state = tcpm_try_src(port) ? SRC_TRY : SNK_ATTACHED;
3686                 else
3687                         new_state = SNK_UNATTACHED;
3688                 if (new_state != port->delayed_state)
3689                         tcpm_set_state(port, SNK_DEBOUNCED, 0);
3690                 break;
3691         case SNK_READY:
3692                 if (tcpm_port_is_disconnected(port))
3693                         tcpm_set_state(port, unattached_state(port), 0);
3694                 else if (!port->pd_capable &&
3695                          (cc1 != old_cc1 || cc2 != old_cc2))
3696                         tcpm_set_current_limit(port,
3697                                                tcpm_get_current_limit(port),
3698                                                5000);
3699                 break;
3700
3701         case AUDIO_ACC_ATTACHED:
3702                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3703                         tcpm_set_state(port, AUDIO_ACC_DEBOUNCE, 0);
3704                 break;
3705         case AUDIO_ACC_DEBOUNCE:
3706                 if (tcpm_port_is_audio(port))
3707                         tcpm_set_state(port, AUDIO_ACC_ATTACHED, 0);
3708                 break;
3709
3710         case DEBUG_ACC_ATTACHED:
3711                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3712                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3713                 break;
3714
3715         case SNK_TRY:
3716                 /* Do nothing, waiting for timeout */
3717                 break;
3718
3719         case SNK_DISCOVERY:
3720                 /* CC line is unstable, wait for debounce */
3721                 if (tcpm_port_is_disconnected(port))
3722                         tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE, 0);
3723                 break;
3724         case SNK_DISCOVERY_DEBOUNCE:
3725                 break;
3726
3727         case SRC_TRYWAIT:
3728                 /* Hand over to state machine if needed */
3729                 if (!port->vbus_present && tcpm_port_is_source(port))
3730                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3731                 break;
3732         case SRC_TRYWAIT_DEBOUNCE:
3733                 if (port->vbus_present || !tcpm_port_is_source(port))
3734                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3735                 break;
3736         case SNK_TRY_WAIT_DEBOUNCE:
3737                 if (!tcpm_port_is_sink(port)) {
3738                         port->max_wait = 0;
3739                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3740                 }
3741                 break;
3742         case SRC_TRY_WAIT:
3743                 if (tcpm_port_is_source(port))
3744                         tcpm_set_state(port, SRC_TRY_DEBOUNCE, 0);
3745                 break;
3746         case SRC_TRY_DEBOUNCE:
3747                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
3748                 break;
3749         case SNK_TRYWAIT_DEBOUNCE:
3750                 if (tcpm_port_is_sink(port))
3751                         tcpm_set_state(port, SNK_TRYWAIT_VBUS, 0);
3752                 break;
3753         case SNK_TRYWAIT_VBUS:
3754                 if (!tcpm_port_is_sink(port))
3755                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
3756                 break;
3757         case SNK_TRYWAIT:
3758                 /* Do nothing, waiting for tCCDebounce */
3759                 break;
3760         case PR_SWAP_SNK_SRC_SINK_OFF:
3761         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3762         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3763         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3764         case PR_SWAP_SNK_SRC_SOURCE_ON:
3765                 /*
3766                  * CC state change is expected in PR_SWAP
3767                  * Ignore it.
3768                  */
3769                 break;
3770
3771         default:
3772                 if (tcpm_port_is_disconnected(port))
3773                         tcpm_set_state(port, unattached_state(port), 0);
3774                 break;
3775         }
3776 }
3777
3778 static void _tcpm_pd_vbus_on(struct tcpm_port *port)
3779 {
3780         tcpm_log_force(port, "VBUS on");
3781         port->vbus_present = true;
3782         switch (port->state) {
3783         case SNK_TRANSITION_SINK_VBUS:
3784                 port->explicit_contract = true;
3785                 tcpm_set_state(port, SNK_READY, 0);
3786                 break;
3787         case SNK_DISCOVERY:
3788                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3789                 break;
3790
3791         case SNK_DEBOUNCED:
3792                 tcpm_set_state(port, tcpm_try_src(port) ? SRC_TRY
3793                                                         : SNK_ATTACHED,
3794                                        0);
3795                 break;
3796         case SNK_HARD_RESET_WAIT_VBUS:
3797                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, 0);
3798                 break;
3799         case SRC_ATTACHED:
3800                 tcpm_set_state(port, SRC_STARTUP, 0);
3801                 break;
3802         case SRC_HARD_RESET_VBUS_ON:
3803                 tcpm_set_state(port, SRC_STARTUP, 0);
3804                 break;
3805
3806         case SNK_TRY:
3807                 /* Do nothing, waiting for timeout */
3808                 break;
3809         case SRC_TRYWAIT:
3810                 /* Do nothing, Waiting for Rd to be detected */
3811                 break;
3812         case SRC_TRYWAIT_DEBOUNCE:
3813                 tcpm_set_state(port, SRC_TRYWAIT, 0);
3814                 break;
3815         case SNK_TRY_WAIT_DEBOUNCE:
3816                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3817                 break;
3818         case SNK_TRYWAIT:
3819                 /* Do nothing, waiting for tCCDebounce */
3820                 break;
3821         case SNK_TRYWAIT_VBUS:
3822                 if (tcpm_port_is_sink(port))
3823                         tcpm_set_state(port, SNK_ATTACHED, 0);
3824                 break;
3825         case SNK_TRYWAIT_DEBOUNCE:
3826                 /* Do nothing, waiting for Rp */
3827                 break;
3828         case SRC_TRY_WAIT:
3829         case SRC_TRY_DEBOUNCE:
3830                 /* Do nothing, waiting for sink detection */
3831                 break;
3832         default:
3833                 break;
3834         }
3835 }
3836
3837 static void _tcpm_pd_vbus_off(struct tcpm_port *port)
3838 {
3839         tcpm_log_force(port, "VBUS off");
3840         port->vbus_present = false;
3841         port->vbus_never_low = false;
3842         switch (port->state) {
3843         case SNK_HARD_RESET_SINK_OFF:
3844                 tcpm_set_state(port, SNK_HARD_RESET_WAIT_VBUS, 0);
3845                 break;
3846         case SRC_HARD_RESET_VBUS_OFF:
3847                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, 0);
3848                 break;
3849         case HARD_RESET_SEND:
3850                 break;
3851
3852         case SNK_TRY:
3853                 /* Do nothing, waiting for timeout */
3854                 break;
3855         case SRC_TRYWAIT:
3856                 /* Hand over to state machine if needed */
3857                 if (tcpm_port_is_source(port))
3858                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3859                 break;
3860         case SNK_TRY_WAIT_DEBOUNCE:
3861                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3862                 break;
3863         case SNK_TRYWAIT:
3864         case SNK_TRYWAIT_VBUS:
3865         case SNK_TRYWAIT_DEBOUNCE:
3866                 break;
3867         case SNK_ATTACH_WAIT:
3868         case SNK_DEBOUNCED:
3869                 /* Do nothing, as TCPM is still waiting for vbus to reaach VSAFE5V to connect */
3870                 break;
3871
3872         case SNK_NEGOTIATE_CAPABILITIES:
3873                 break;
3874
3875         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3876                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF, 0);
3877                 break;
3878
3879         case PR_SWAP_SNK_SRC_SINK_OFF:
3880                 /* Do nothing, expected */
3881                 break;
3882
3883         case PORT_RESET_WAIT_OFF:
3884                 tcpm_set_state(port, tcpm_default_state(port), 0);
3885                 break;
3886         case SRC_TRY_WAIT:
3887         case SRC_TRY_DEBOUNCE:
3888                 /* Do nothing, waiting for sink detection */
3889                 break;
3890         default:
3891                 if (port->pwr_role == TYPEC_SINK &&
3892                     port->attached)
3893                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3894                 break;
3895         }
3896 }
3897
3898 static void _tcpm_pd_hard_reset(struct tcpm_port *port)
3899 {
3900         tcpm_log_force(port, "Received hard reset");
3901         /*
3902          * If we keep receiving hard reset requests, executing the hard reset
3903          * must have failed. Revert to error recovery if that happens.
3904          */
3905         tcpm_set_state(port,
3906                        port->hard_reset_count < PD_N_HARD_RESET_COUNT ?
3907                                 HARD_RESET_START : ERROR_RECOVERY,
3908                        0);
3909 }
3910
3911 static void tcpm_pd_event_handler(struct work_struct *work)
3912 {
3913         struct tcpm_port *port = container_of(work, struct tcpm_port,
3914                                               event_work);
3915         u32 events;
3916
3917         mutex_lock(&port->lock);
3918
3919         spin_lock(&port->pd_event_lock);
3920         while (port->pd_events) {
3921                 events = port->pd_events;
3922                 port->pd_events = 0;
3923                 spin_unlock(&port->pd_event_lock);
3924                 if (events & TCPM_RESET_EVENT)
3925                         _tcpm_pd_hard_reset(port);
3926                 if (events & TCPM_VBUS_EVENT) {
3927                         bool vbus;
3928
3929                         vbus = port->tcpc->get_vbus(port->tcpc);
3930                         if (vbus)
3931                                 _tcpm_pd_vbus_on(port);
3932                         else
3933                                 _tcpm_pd_vbus_off(port);
3934                 }
3935                 if (events & TCPM_CC_EVENT) {
3936                         enum typec_cc_status cc1, cc2;
3937
3938                         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
3939                                 _tcpm_cc_change(port, cc1, cc2);
3940                 }
3941                 spin_lock(&port->pd_event_lock);
3942         }
3943         spin_unlock(&port->pd_event_lock);
3944         mutex_unlock(&port->lock);
3945 }
3946
3947 void tcpm_cc_change(struct tcpm_port *port)
3948 {
3949         spin_lock(&port->pd_event_lock);
3950         port->pd_events |= TCPM_CC_EVENT;
3951         spin_unlock(&port->pd_event_lock);
3952         queue_work(port->wq, &port->event_work);
3953 }
3954 EXPORT_SYMBOL_GPL(tcpm_cc_change);
3955
3956 void tcpm_vbus_change(struct tcpm_port *port)
3957 {
3958         spin_lock(&port->pd_event_lock);
3959         port->pd_events |= TCPM_VBUS_EVENT;
3960         spin_unlock(&port->pd_event_lock);
3961         queue_work(port->wq, &port->event_work);
3962 }
3963 EXPORT_SYMBOL_GPL(tcpm_vbus_change);
3964
3965 void tcpm_pd_hard_reset(struct tcpm_port *port)
3966 {
3967         spin_lock(&port->pd_event_lock);
3968         port->pd_events = TCPM_RESET_EVENT;
3969         spin_unlock(&port->pd_event_lock);
3970         queue_work(port->wq, &port->event_work);
3971 }
3972 EXPORT_SYMBOL_GPL(tcpm_pd_hard_reset);
3973
3974 static int tcpm_dr_set(const struct typec_capability *cap,
3975                        enum typec_data_role data)
3976 {
3977         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3978         int ret;
3979
3980         mutex_lock(&port->swap_lock);
3981         mutex_lock(&port->lock);
3982
3983         if (port->port_type != TYPEC_PORT_DRP) {
3984                 ret = -EINVAL;
3985                 goto port_unlock;
3986         }
3987         if (port->state != SRC_READY && port->state != SNK_READY) {
3988                 ret = -EAGAIN;
3989                 goto port_unlock;
3990         }
3991
3992         if (port->data_role == data) {
3993                 ret = 0;
3994                 goto port_unlock;
3995         }
3996
3997         /*
3998          * XXX
3999          * 6.3.9: If an alternate mode is active, a request to swap
4000          * alternate modes shall trigger a port reset.
4001          * Reject data role swap request in this case.
4002          */
4003
4004         if (!port->pd_capable) {
4005                 /*
4006                  * If the partner is not PD capable, reset the port to
4007                  * trigger a role change. This can only work if a preferred
4008                  * role is configured, and if it matches the requested role.
4009                  */
4010                 if (port->try_role == TYPEC_NO_PREFERRED_ROLE ||
4011                     port->try_role == port->pwr_role) {
4012                         ret = -EINVAL;
4013                         goto port_unlock;
4014                 }
4015                 port->non_pd_role_swap = true;
4016                 tcpm_set_state(port, PORT_RESET, 0);
4017         } else {
4018                 tcpm_set_state(port, DR_SWAP_SEND, 0);
4019         }
4020
4021         port->swap_status = 0;
4022         port->swap_pending = true;
4023         reinit_completion(&port->swap_complete);
4024         mutex_unlock(&port->lock);
4025
4026         if (!wait_for_completion_timeout(&port->swap_complete,
4027                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
4028                 ret = -ETIMEDOUT;
4029         else
4030                 ret = port->swap_status;
4031
4032         port->non_pd_role_swap = false;
4033         goto swap_unlock;
4034
4035 port_unlock:
4036         mutex_unlock(&port->lock);
4037 swap_unlock:
4038         mutex_unlock(&port->swap_lock);
4039         return ret;
4040 }
4041
4042 static int tcpm_pr_set(const struct typec_capability *cap,
4043                        enum typec_role role)
4044 {
4045         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4046         int ret;
4047
4048         mutex_lock(&port->swap_lock);
4049         mutex_lock(&port->lock);
4050
4051         if (port->port_type != TYPEC_PORT_DRP) {
4052                 ret = -EINVAL;
4053                 goto port_unlock;
4054         }
4055         if (port->state != SRC_READY && port->state != SNK_READY) {
4056                 ret = -EAGAIN;
4057                 goto port_unlock;
4058         }
4059
4060         if (role == port->pwr_role) {
4061                 ret = 0;
4062                 goto port_unlock;
4063         }
4064
4065         port->swap_status = 0;
4066         port->swap_pending = true;
4067         reinit_completion(&port->swap_complete);
4068         tcpm_set_state(port, PR_SWAP_SEND, 0);
4069         mutex_unlock(&port->lock);
4070
4071         if (!wait_for_completion_timeout(&port->swap_complete,
4072                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
4073                 ret = -ETIMEDOUT;
4074         else
4075                 ret = port->swap_status;
4076
4077         goto swap_unlock;
4078
4079 port_unlock:
4080         mutex_unlock(&port->lock);
4081 swap_unlock:
4082         mutex_unlock(&port->swap_lock);
4083         return ret;
4084 }
4085
4086 static int tcpm_vconn_set(const struct typec_capability *cap,
4087                           enum typec_role role)
4088 {
4089         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4090         int ret;
4091
4092         mutex_lock(&port->swap_lock);
4093         mutex_lock(&port->lock);
4094
4095         if (port->state != SRC_READY && port->state != SNK_READY) {
4096                 ret = -EAGAIN;
4097                 goto port_unlock;
4098         }
4099
4100         if (role == port->vconn_role) {
4101                 ret = 0;
4102                 goto port_unlock;
4103         }
4104
4105         port->swap_status = 0;
4106         port->swap_pending = true;
4107         reinit_completion(&port->swap_complete);
4108         tcpm_set_state(port, VCONN_SWAP_SEND, 0);
4109         mutex_unlock(&port->lock);
4110
4111         if (!wait_for_completion_timeout(&port->swap_complete,
4112                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
4113                 ret = -ETIMEDOUT;
4114         else
4115                 ret = port->swap_status;
4116
4117         goto swap_unlock;
4118
4119 port_unlock:
4120         mutex_unlock(&port->lock);
4121 swap_unlock:
4122         mutex_unlock(&port->swap_lock);
4123         return ret;
4124 }
4125
4126 static int tcpm_try_role(const struct typec_capability *cap, int role)
4127 {
4128         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4129         struct tcpc_dev *tcpc = port->tcpc;
4130         int ret = 0;
4131
4132         mutex_lock(&port->lock);
4133         if (tcpc->try_role)
4134                 ret = tcpc->try_role(tcpc, role);
4135         if (!ret && (!tcpc->config || !tcpc->config->try_role_hw))
4136                 port->try_role = role;
4137         port->try_src_count = 0;
4138         port->try_snk_count = 0;
4139         mutex_unlock(&port->lock);
4140
4141         return ret;
4142 }
4143
4144 static int tcpm_pps_set_op_curr(struct tcpm_port *port, u16 op_curr)
4145 {
4146         unsigned int target_mw;
4147         int ret;
4148
4149         mutex_lock(&port->swap_lock);
4150         mutex_lock(&port->lock);
4151
4152         if (!port->pps_data.active) {
4153                 ret = -EOPNOTSUPP;
4154                 goto port_unlock;
4155         }
4156
4157         if (port->state != SNK_READY) {
4158                 ret = -EAGAIN;
4159                 goto port_unlock;
4160         }
4161
4162         if (op_curr > port->pps_data.max_curr) {
4163                 ret = -EINVAL;
4164                 goto port_unlock;
4165         }
4166
4167         target_mw = (op_curr * port->pps_data.out_volt) / 1000;
4168         if (target_mw < port->operating_snk_mw) {
4169                 ret = -EINVAL;
4170                 goto port_unlock;
4171         }
4172
4173         /* Round down operating current to align with PPS valid steps */
4174         op_curr = op_curr - (op_curr % RDO_PROG_CURR_MA_STEP);
4175
4176         reinit_completion(&port->pps_complete);
4177         port->pps_data.op_curr = op_curr;
4178         port->pps_status = 0;
4179         port->pps_pending = true;
4180         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4181         mutex_unlock(&port->lock);
4182
4183         if (!wait_for_completion_timeout(&port->pps_complete,
4184                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4185                 ret = -ETIMEDOUT;
4186         else
4187                 ret = port->pps_status;
4188
4189         goto swap_unlock;
4190
4191 port_unlock:
4192         mutex_unlock(&port->lock);
4193 swap_unlock:
4194         mutex_unlock(&port->swap_lock);
4195
4196         return ret;
4197 }
4198
4199 static int tcpm_pps_set_out_volt(struct tcpm_port *port, u16 out_volt)
4200 {
4201         unsigned int target_mw;
4202         int ret;
4203
4204         mutex_lock(&port->swap_lock);
4205         mutex_lock(&port->lock);
4206
4207         if (!port->pps_data.active) {
4208                 ret = -EOPNOTSUPP;
4209                 goto port_unlock;
4210         }
4211
4212         if (port->state != SNK_READY) {
4213                 ret = -EAGAIN;
4214                 goto port_unlock;
4215         }
4216
4217         if (out_volt < port->pps_data.min_volt ||
4218             out_volt > port->pps_data.max_volt) {
4219                 ret = -EINVAL;
4220                 goto port_unlock;
4221         }
4222
4223         target_mw = (port->pps_data.op_curr * out_volt) / 1000;
4224         if (target_mw < port->operating_snk_mw) {
4225                 ret = -EINVAL;
4226                 goto port_unlock;
4227         }
4228
4229         /* Round down output voltage to align with PPS valid steps */
4230         out_volt = out_volt - (out_volt % RDO_PROG_VOLT_MV_STEP);
4231
4232         reinit_completion(&port->pps_complete);
4233         port->pps_data.out_volt = out_volt;
4234         port->pps_status = 0;
4235         port->pps_pending = true;
4236         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4237         mutex_unlock(&port->lock);
4238
4239         if (!wait_for_completion_timeout(&port->pps_complete,
4240                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4241                 ret = -ETIMEDOUT;
4242         else
4243                 ret = port->pps_status;
4244
4245         goto swap_unlock;
4246
4247 port_unlock:
4248         mutex_unlock(&port->lock);
4249 swap_unlock:
4250         mutex_unlock(&port->swap_lock);
4251
4252         return ret;
4253 }
4254
4255 static int tcpm_pps_activate(struct tcpm_port *port, bool activate)
4256 {
4257         int ret = 0;
4258
4259         mutex_lock(&port->swap_lock);
4260         mutex_lock(&port->lock);
4261
4262         if (!port->pps_data.supported) {
4263                 ret = -EOPNOTSUPP;
4264                 goto port_unlock;
4265         }
4266
4267         /* Trying to deactivate PPS when already deactivated so just bail */
4268         if (!port->pps_data.active && !activate)
4269                 goto port_unlock;
4270
4271         if (port->state != SNK_READY) {
4272                 ret = -EAGAIN;
4273                 goto port_unlock;
4274         }
4275
4276         reinit_completion(&port->pps_complete);
4277         port->pps_status = 0;
4278         port->pps_pending = true;
4279
4280         /* Trigger PPS request or move back to standard PDO contract */
4281         if (activate) {
4282                 port->pps_data.out_volt = port->supply_voltage;
4283                 port->pps_data.op_curr = port->current_limit;
4284                 tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4285         } else {
4286                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4287         }
4288         mutex_unlock(&port->lock);
4289
4290         if (!wait_for_completion_timeout(&port->pps_complete,
4291                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4292                 ret = -ETIMEDOUT;
4293         else
4294                 ret = port->pps_status;
4295
4296         goto swap_unlock;
4297
4298 port_unlock:
4299         mutex_unlock(&port->lock);
4300 swap_unlock:
4301         mutex_unlock(&port->swap_lock);
4302
4303         return ret;
4304 }
4305
4306 static void tcpm_init(struct tcpm_port *port)
4307 {
4308         enum typec_cc_status cc1, cc2;
4309
4310         port->tcpc->init(port->tcpc);
4311
4312         tcpm_reset_port(port);
4313
4314         /*
4315          * XXX
4316          * Should possibly wait for VBUS to settle if it was enabled locally
4317          * since tcpm_reset_port() will disable VBUS.
4318          */
4319         port->vbus_present = port->tcpc->get_vbus(port->tcpc);
4320         if (port->vbus_present)
4321                 port->vbus_never_low = true;
4322
4323         tcpm_set_state(port, tcpm_default_state(port), 0);
4324
4325         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
4326                 _tcpm_cc_change(port, cc1, cc2);
4327
4328         /*
4329          * Some adapters need a clean slate at startup, and won't recover
4330          * otherwise. So do not try to be fancy and force a clean disconnect.
4331          */
4332         tcpm_set_state(port, PORT_RESET, 0);
4333 }
4334
4335 static int tcpm_port_type_set(const struct typec_capability *cap,
4336                               enum typec_port_type type)
4337 {
4338         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4339
4340         mutex_lock(&port->lock);
4341         if (type == port->port_type)
4342                 goto port_unlock;
4343
4344         port->port_type = type;
4345
4346         if (!port->connected) {
4347                 tcpm_set_state(port, PORT_RESET, 0);
4348         } else if (type == TYPEC_PORT_SNK) {
4349                 if (!(port->pwr_role == TYPEC_SINK &&
4350                       port->data_role == TYPEC_DEVICE))
4351                         tcpm_set_state(port, PORT_RESET, 0);
4352         } else if (type == TYPEC_PORT_SRC) {
4353                 if (!(port->pwr_role == TYPEC_SOURCE &&
4354                       port->data_role == TYPEC_HOST))
4355                         tcpm_set_state(port, PORT_RESET, 0);
4356         }
4357
4358 port_unlock:
4359         mutex_unlock(&port->lock);
4360         return 0;
4361 }
4362
4363 void tcpm_tcpc_reset(struct tcpm_port *port)
4364 {
4365         mutex_lock(&port->lock);
4366         /* XXX: Maintain PD connection if possible? */
4367         tcpm_init(port);
4368         mutex_unlock(&port->lock);
4369 }
4370 EXPORT_SYMBOL_GPL(tcpm_tcpc_reset);
4371
4372 static int tcpm_copy_pdos(u32 *dest_pdo, const u32 *src_pdo,
4373                           unsigned int nr_pdo)
4374 {
4375         unsigned int i;
4376
4377         if (nr_pdo > PDO_MAX_OBJECTS)
4378                 nr_pdo = PDO_MAX_OBJECTS;
4379
4380         for (i = 0; i < nr_pdo; i++)
4381                 dest_pdo[i] = src_pdo[i];
4382
4383         return nr_pdo;
4384 }
4385
4386 static int tcpm_copy_vdos(u32 *dest_vdo, const u32 *src_vdo,
4387                           unsigned int nr_vdo)
4388 {
4389         unsigned int i;
4390
4391         if (nr_vdo > VDO_MAX_OBJECTS)
4392                 nr_vdo = VDO_MAX_OBJECTS;
4393
4394         for (i = 0; i < nr_vdo; i++)
4395                 dest_vdo[i] = src_vdo[i];
4396
4397         return nr_vdo;
4398 }
4399
4400 static int tcpm_fw_get_caps(struct tcpm_port *port,
4401                             struct fwnode_handle *fwnode)
4402 {
4403         const char *cap_str;
4404         int ret;
4405         u32 mw;
4406
4407         if (!fwnode)
4408                 return -EINVAL;
4409
4410         /* USB data support is optional */
4411         ret = fwnode_property_read_string(fwnode, "data-role", &cap_str);
4412         if (ret == 0) {
4413                 port->typec_caps.data = typec_find_port_data_role(cap_str);
4414                 if (port->typec_caps.data < 0)
4415                         return -EINVAL;
4416         }
4417
4418         ret = fwnode_property_read_string(fwnode, "power-role", &cap_str);
4419         if (ret < 0)
4420                 return ret;
4421
4422         port->typec_caps.type = typec_find_port_power_role(cap_str);
4423         if (port->typec_caps.type < 0)
4424                 return -EINVAL;
4425         port->port_type = port->typec_caps.type;
4426
4427         if (port->port_type == TYPEC_PORT_SNK)
4428                 goto sink;
4429
4430         /* Get source pdos */
4431         ret = fwnode_property_read_u32_array(fwnode, "source-pdos",
4432                                              NULL, 0);
4433         if (ret <= 0)
4434                 return -EINVAL;
4435
4436         port->nr_src_pdo = min(ret, PDO_MAX_OBJECTS);
4437         ret = fwnode_property_read_u32_array(fwnode, "source-pdos",
4438                                              port->src_pdo, port->nr_src_pdo);
4439         if ((ret < 0) || tcpm_validate_caps(port, port->src_pdo,
4440                                             port->nr_src_pdo))
4441                 return -EINVAL;
4442
4443         if (port->port_type == TYPEC_PORT_SRC)
4444                 return 0;
4445
4446         /* Get the preferred power role for DRP */
4447         ret = fwnode_property_read_string(fwnode, "try-power-role", &cap_str);
4448         if (ret < 0)
4449                 return ret;
4450
4451         port->typec_caps.prefer_role = typec_find_power_role(cap_str);
4452         if (port->typec_caps.prefer_role < 0)
4453                 return -EINVAL;
4454 sink:
4455         /* Get sink pdos */
4456         ret = fwnode_property_read_u32_array(fwnode, "sink-pdos",
4457                                              NULL, 0);
4458         if (ret <= 0)
4459                 return -EINVAL;
4460
4461         port->nr_snk_pdo = min(ret, PDO_MAX_OBJECTS);
4462         ret = fwnode_property_read_u32_array(fwnode, "sink-pdos",
4463                                              port->snk_pdo, port->nr_snk_pdo);
4464         if ((ret < 0) || tcpm_validate_caps(port, port->snk_pdo,
4465                                             port->nr_snk_pdo))
4466                 return -EINVAL;
4467
4468         if (fwnode_property_read_u32(fwnode, "op-sink-microwatt", &mw) < 0)
4469                 return -EINVAL;
4470         port->operating_snk_mw = mw / 1000;
4471
4472         port->self_powered = fwnode_property_read_bool(fwnode, "self-powered");
4473
4474         return 0;
4475 }
4476
4477 int tcpm_update_source_capabilities(struct tcpm_port *port, const u32 *pdo,
4478                                     unsigned int nr_pdo)
4479 {
4480         if (tcpm_validate_caps(port, pdo, nr_pdo))
4481                 return -EINVAL;
4482
4483         mutex_lock(&port->lock);
4484         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, pdo, nr_pdo);
4485         switch (port->state) {
4486         case SRC_UNATTACHED:
4487         case SRC_ATTACH_WAIT:
4488         case SRC_TRYWAIT:
4489                 tcpm_set_cc(port, tcpm_rp_cc(port));
4490                 break;
4491         case SRC_SEND_CAPABILITIES:
4492         case SRC_NEGOTIATE_CAPABILITIES:
4493         case SRC_READY:
4494         case SRC_WAIT_NEW_CAPABILITIES:
4495                 tcpm_set_cc(port, tcpm_rp_cc(port));
4496                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
4497                 break;
4498         default:
4499                 break;
4500         }
4501         mutex_unlock(&port->lock);
4502         return 0;
4503 }
4504 EXPORT_SYMBOL_GPL(tcpm_update_source_capabilities);
4505
4506 int tcpm_update_sink_capabilities(struct tcpm_port *port, const u32 *pdo,
4507                                   unsigned int nr_pdo,
4508                                   unsigned int operating_snk_mw)
4509 {
4510         if (tcpm_validate_caps(port, pdo, nr_pdo))
4511                 return -EINVAL;
4512
4513         mutex_lock(&port->lock);
4514         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, pdo, nr_pdo);
4515         port->operating_snk_mw = operating_snk_mw;
4516         port->update_sink_caps = true;
4517
4518         switch (port->state) {
4519         case SNK_NEGOTIATE_CAPABILITIES:
4520         case SNK_NEGOTIATE_PPS_CAPABILITIES:
4521         case SNK_READY:
4522         case SNK_TRANSITION_SINK:
4523         case SNK_TRANSITION_SINK_VBUS:
4524                 if (port->pps_data.active)
4525                         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4526                 else
4527                         tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4528                 break;
4529         default:
4530                 break;
4531         }
4532         mutex_unlock(&port->lock);
4533         return 0;
4534 }
4535 EXPORT_SYMBOL_GPL(tcpm_update_sink_capabilities);
4536
4537 /* Power Supply access to expose source power information */
4538 enum tcpm_psy_online_states {
4539         TCPM_PSY_OFFLINE = 0,
4540         TCPM_PSY_FIXED_ONLINE,
4541         TCPM_PSY_PROG_ONLINE,
4542 };
4543
4544 static enum power_supply_property tcpm_psy_props[] = {
4545         POWER_SUPPLY_PROP_USB_TYPE,
4546         POWER_SUPPLY_PROP_ONLINE,
4547         POWER_SUPPLY_PROP_VOLTAGE_MIN,
4548         POWER_SUPPLY_PROP_VOLTAGE_MAX,
4549         POWER_SUPPLY_PROP_VOLTAGE_NOW,
4550         POWER_SUPPLY_PROP_CURRENT_MAX,
4551         POWER_SUPPLY_PROP_CURRENT_NOW,
4552 };
4553
4554 static int tcpm_psy_get_online(struct tcpm_port *port,
4555                                union power_supply_propval *val)
4556 {
4557         if (port->vbus_charge) {
4558                 if (port->pps_data.active)
4559                         val->intval = TCPM_PSY_PROG_ONLINE;
4560                 else
4561                         val->intval = TCPM_PSY_FIXED_ONLINE;
4562         } else {
4563                 val->intval = TCPM_PSY_OFFLINE;
4564         }
4565
4566         return 0;
4567 }
4568
4569 static int tcpm_psy_get_voltage_min(struct tcpm_port *port,
4570                                     union power_supply_propval *val)
4571 {
4572         if (port->pps_data.active)
4573                 val->intval = port->pps_data.min_volt * 1000;
4574         else
4575                 val->intval = port->supply_voltage * 1000;
4576
4577         return 0;
4578 }
4579
4580 static int tcpm_psy_get_voltage_max(struct tcpm_port *port,
4581                                     union power_supply_propval *val)
4582 {
4583         if (port->pps_data.active)
4584                 val->intval = port->pps_data.max_volt * 1000;
4585         else
4586                 val->intval = port->supply_voltage * 1000;
4587
4588         return 0;
4589 }
4590
4591 static int tcpm_psy_get_voltage_now(struct tcpm_port *port,
4592                                     union power_supply_propval *val)
4593 {
4594         val->intval = port->supply_voltage * 1000;
4595
4596         return 0;
4597 }
4598
4599 static int tcpm_psy_get_current_max(struct tcpm_port *port,
4600                                     union power_supply_propval *val)
4601 {
4602         if (port->pps_data.active)
4603                 val->intval = port->pps_data.max_curr * 1000;
4604         else
4605                 val->intval = port->current_limit * 1000;
4606
4607         return 0;
4608 }
4609
4610 static int tcpm_psy_get_current_now(struct tcpm_port *port,
4611                                     union power_supply_propval *val)
4612 {
4613         val->intval = port->current_limit * 1000;
4614
4615         return 0;
4616 }
4617
4618 static int tcpm_psy_get_prop(struct power_supply *psy,
4619                              enum power_supply_property psp,
4620                              union power_supply_propval *val)
4621 {
4622         struct tcpm_port *port = power_supply_get_drvdata(psy);
4623         int ret = 0;
4624
4625         switch (psp) {
4626         case POWER_SUPPLY_PROP_USB_TYPE:
4627                 val->intval = port->usb_type;
4628                 break;
4629         case POWER_SUPPLY_PROP_ONLINE:
4630                 ret = tcpm_psy_get_online(port, val);
4631                 break;
4632         case POWER_SUPPLY_PROP_VOLTAGE_MIN:
4633                 ret = tcpm_psy_get_voltage_min(port, val);
4634                 break;
4635         case POWER_SUPPLY_PROP_VOLTAGE_MAX:
4636                 ret = tcpm_psy_get_voltage_max(port, val);
4637                 break;
4638         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4639                 ret = tcpm_psy_get_voltage_now(port, val);
4640                 break;
4641         case POWER_SUPPLY_PROP_CURRENT_MAX:
4642                 ret = tcpm_psy_get_current_max(port, val);
4643                 break;
4644         case POWER_SUPPLY_PROP_CURRENT_NOW:
4645                 ret = tcpm_psy_get_current_now(port, val);
4646                 break;
4647         default:
4648                 ret = -EINVAL;
4649                 break;
4650         }
4651
4652         return ret;
4653 }
4654
4655 static int tcpm_psy_set_online(struct tcpm_port *port,
4656                                const union power_supply_propval *val)
4657 {
4658         int ret;
4659
4660         switch (val->intval) {
4661         case TCPM_PSY_FIXED_ONLINE:
4662                 ret = tcpm_pps_activate(port, false);
4663                 break;
4664         case TCPM_PSY_PROG_ONLINE:
4665                 ret = tcpm_pps_activate(port, true);
4666                 break;
4667         default:
4668                 ret = -EINVAL;
4669                 break;
4670         }
4671
4672         return ret;
4673 }
4674
4675 static int tcpm_psy_set_prop(struct power_supply *psy,
4676                              enum power_supply_property psp,
4677                              const union power_supply_propval *val)
4678 {
4679         struct tcpm_port *port = power_supply_get_drvdata(psy);
4680         int ret;
4681
4682         switch (psp) {
4683         case POWER_SUPPLY_PROP_ONLINE:
4684                 ret = tcpm_psy_set_online(port, val);
4685                 break;
4686         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4687                 if (val->intval < port->pps_data.min_volt * 1000 ||
4688                     val->intval > port->pps_data.max_volt * 1000)
4689                         ret = -EINVAL;
4690                 else
4691                         ret = tcpm_pps_set_out_volt(port, val->intval / 1000);
4692                 break;
4693         case POWER_SUPPLY_PROP_CURRENT_NOW:
4694                 if (val->intval > port->pps_data.max_curr * 1000)
4695                         ret = -EINVAL;
4696                 else
4697                         ret = tcpm_pps_set_op_curr(port, val->intval / 1000);
4698                 break;
4699         default:
4700                 ret = -EINVAL;
4701                 break;
4702         }
4703
4704         return ret;
4705 }
4706
4707 static int tcpm_psy_prop_writeable(struct power_supply *psy,
4708                                    enum power_supply_property psp)
4709 {
4710         switch (psp) {
4711         case POWER_SUPPLY_PROP_ONLINE:
4712         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4713         case POWER_SUPPLY_PROP_CURRENT_NOW:
4714                 return 1;
4715         default:
4716                 return 0;
4717         }
4718 }
4719
4720 static enum power_supply_usb_type tcpm_psy_usb_types[] = {
4721         POWER_SUPPLY_USB_TYPE_C,
4722         POWER_SUPPLY_USB_TYPE_PD,
4723         POWER_SUPPLY_USB_TYPE_PD_PPS,
4724 };
4725
4726 static const char *tcpm_psy_name_prefix = "tcpm-source-psy-";
4727
4728 static int devm_tcpm_psy_register(struct tcpm_port *port)
4729 {
4730         struct power_supply_config psy_cfg = {};
4731         const char *port_dev_name = dev_name(port->dev);
4732         size_t psy_name_len = strlen(tcpm_psy_name_prefix) +
4733                                      strlen(port_dev_name) + 1;
4734         char *psy_name;
4735
4736         psy_cfg.drv_data = port;
4737         psy_cfg.fwnode = dev_fwnode(port->dev);
4738         psy_name = devm_kzalloc(port->dev, psy_name_len, GFP_KERNEL);
4739         if (!psy_name)
4740                 return -ENOMEM;
4741
4742         snprintf(psy_name, psy_name_len, "%s%s", tcpm_psy_name_prefix,
4743                  port_dev_name);
4744         port->psy_desc.name = psy_name;
4745         port->psy_desc.type = POWER_SUPPLY_TYPE_USB,
4746         port->psy_desc.usb_types = tcpm_psy_usb_types;
4747         port->psy_desc.num_usb_types = ARRAY_SIZE(tcpm_psy_usb_types);
4748         port->psy_desc.properties = tcpm_psy_props,
4749         port->psy_desc.num_properties = ARRAY_SIZE(tcpm_psy_props),
4750         port->psy_desc.get_property = tcpm_psy_get_prop,
4751         port->psy_desc.set_property = tcpm_psy_set_prop,
4752         port->psy_desc.property_is_writeable = tcpm_psy_prop_writeable,
4753
4754         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
4755
4756         port->psy = devm_power_supply_register(port->dev, &port->psy_desc,
4757                                                &psy_cfg);
4758
4759         return PTR_ERR_OR_ZERO(port->psy);
4760 }
4761
4762 static int tcpm_copy_caps(struct tcpm_port *port,
4763                           const struct tcpc_config *tcfg)
4764 {
4765         if (tcpm_validate_caps(port, tcfg->src_pdo, tcfg->nr_src_pdo) ||
4766             tcpm_validate_caps(port, tcfg->snk_pdo, tcfg->nr_snk_pdo))
4767                 return -EINVAL;
4768
4769         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, tcfg->src_pdo,
4770                                           tcfg->nr_src_pdo);
4771         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, tcfg->snk_pdo,
4772                                           tcfg->nr_snk_pdo);
4773
4774         port->nr_snk_vdo = tcpm_copy_vdos(port->snk_vdo, tcfg->snk_vdo,
4775                                           tcfg->nr_snk_vdo);
4776
4777         port->operating_snk_mw = tcfg->operating_snk_mw;
4778
4779         port->typec_caps.prefer_role = tcfg->default_role;
4780         port->typec_caps.type = tcfg->type;
4781         port->typec_caps.data = tcfg->data;
4782         port->self_powered = tcfg->self_powered;
4783
4784         return 0;
4785 }
4786
4787 struct tcpm_port *tcpm_register_port(struct device *dev, struct tcpc_dev *tcpc)
4788 {
4789         struct tcpm_port *port;
4790         int i, err;
4791
4792         if (!dev || !tcpc ||
4793             !tcpc->get_vbus || !tcpc->set_cc || !tcpc->get_cc ||
4794             !tcpc->set_polarity || !tcpc->set_vconn || !tcpc->set_vbus ||
4795             !tcpc->set_pd_rx || !tcpc->set_roles || !tcpc->pd_transmit)
4796                 return ERR_PTR(-EINVAL);
4797
4798         port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
4799         if (!port)
4800                 return ERR_PTR(-ENOMEM);
4801
4802         port->dev = dev;
4803         port->tcpc = tcpc;
4804
4805         mutex_init(&port->lock);
4806         mutex_init(&port->swap_lock);
4807
4808         port->wq = create_singlethread_workqueue(dev_name(dev));
4809         if (!port->wq)
4810                 return ERR_PTR(-ENOMEM);
4811         INIT_DELAYED_WORK(&port->state_machine, tcpm_state_machine_work);
4812         INIT_DELAYED_WORK(&port->vdm_state_machine, vdm_state_machine_work);
4813         INIT_WORK(&port->event_work, tcpm_pd_event_handler);
4814
4815         spin_lock_init(&port->pd_event_lock);
4816
4817         init_completion(&port->tx_complete);
4818         init_completion(&port->swap_complete);
4819         init_completion(&port->pps_complete);
4820         tcpm_debugfs_init(port);
4821
4822         err = tcpm_fw_get_caps(port, tcpc->fwnode);
4823         if ((err < 0) && tcpc->config)
4824                 err = tcpm_copy_caps(port, tcpc->config);
4825         if (err < 0)
4826                 goto out_destroy_wq;
4827
4828         if (!tcpc->config || !tcpc->config->try_role_hw)
4829                 port->try_role = port->typec_caps.prefer_role;
4830         else
4831                 port->try_role = TYPEC_NO_PREFERRED_ROLE;
4832
4833         port->typec_caps.fwnode = tcpc->fwnode;
4834         port->typec_caps.revision = 0x0120;     /* Type-C spec release 1.2 */
4835         port->typec_caps.pd_revision = 0x0300;  /* USB-PD spec release 3.0 */
4836         port->typec_caps.dr_set = tcpm_dr_set;
4837         port->typec_caps.pr_set = tcpm_pr_set;
4838         port->typec_caps.vconn_set = tcpm_vconn_set;
4839         port->typec_caps.try_role = tcpm_try_role;
4840         port->typec_caps.port_type_set = tcpm_port_type_set;
4841
4842         port->partner_desc.identity = &port->partner_ident;
4843         port->port_type = port->typec_caps.type;
4844
4845         port->role_sw = usb_role_switch_get(port->dev);
4846         if (IS_ERR(port->role_sw)) {
4847                 err = PTR_ERR(port->role_sw);
4848                 goto out_destroy_wq;
4849         }
4850
4851         err = devm_tcpm_psy_register(port);
4852         if (err)
4853                 goto out_destroy_wq;
4854
4855         port->typec_port = typec_register_port(port->dev, &port->typec_caps);
4856         if (IS_ERR(port->typec_port)) {
4857                 err = PTR_ERR(port->typec_port);
4858                 goto out_destroy_wq;
4859         }
4860
4861         if (tcpc->config && tcpc->config->alt_modes) {
4862                 const struct typec_altmode_desc *paltmode = tcpc->config->alt_modes;
4863
4864                 i = 0;
4865                 while (paltmode->svid && i < ARRAY_SIZE(port->port_altmode)) {
4866                         struct typec_altmode *alt;
4867
4868                         alt = typec_port_register_altmode(port->typec_port,
4869                                                           paltmode);
4870                         if (IS_ERR(alt)) {
4871                                 tcpm_log(port,
4872                                          "%s: failed to register port alternate mode 0x%x",
4873                                          dev_name(dev), paltmode->svid);
4874                                 break;
4875                         }
4876                         typec_altmode_set_drvdata(alt, port);
4877                         alt->ops = &tcpm_altmode_ops;
4878                         port->port_altmode[i] = alt;
4879                         i++;
4880                         paltmode++;
4881                 }
4882         }
4883
4884         mutex_lock(&port->lock);
4885         tcpm_init(port);
4886         mutex_unlock(&port->lock);
4887
4888         tcpm_log(port, "%s: registered", dev_name(dev));
4889         return port;
4890
4891 out_destroy_wq:
4892         usb_role_switch_put(port->role_sw);
4893         destroy_workqueue(port->wq);
4894         return ERR_PTR(err);
4895 }
4896 EXPORT_SYMBOL_GPL(tcpm_register_port);
4897
4898 void tcpm_unregister_port(struct tcpm_port *port)
4899 {
4900         int i;
4901
4902         tcpm_reset_port(port);
4903         for (i = 0; i < ARRAY_SIZE(port->port_altmode); i++)
4904                 typec_unregister_altmode(port->port_altmode[i]);
4905         typec_unregister_port(port->typec_port);
4906         usb_role_switch_put(port->role_sw);
4907         tcpm_debugfs_exit(port);
4908         destroy_workqueue(port->wq);
4909 }
4910 EXPORT_SYMBOL_GPL(tcpm_unregister_port);
4911
4912 MODULE_AUTHOR("Guenter Roeck <groeck@chromium.org>");
4913 MODULE_DESCRIPTION("USB Type-C Port Manager");
4914 MODULE_LICENSE("GPL");