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