drivers/usb/typec/tcpm/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/hrtimer.h>
  12 #include <linux/jiffies.h>
  13 #include <linux/kernel.h>
  14 #include <linux/kthread.h>
  15 #include <linux/module.h>
  16 #include <linux/mutex.h>
  17 #include <linux/power_supply.h>
  18 #include <linux/proc_fs.h>
  19 #include <linux/property.h>
  20 #include <linux/sched/clock.h>
  21 #include <linux/seq_file.h>
  22 #include <linux/slab.h>
  23 #include <linux/spinlock.h>
  24 #include <linux/usb.h>
  25 #include <linux/usb/pd.h>
  26 #include <linux/usb/pd_ado.h>
  27 #include <linux/usb/pd_bdo.h>
  28 #include <linux/usb/pd_ext_sdb.h>
  29 #include <linux/usb/pd_vdo.h>
  30 #include <linux/usb/role.h>
  31 #include <linux/usb/tcpm.h>
  32 #include <linux/usb/typec_altmode.h>
  33
  34 #include <uapi/linux/sched/types.h>
  35
  36 #define FOREACH_STATE(S)                        \
  37         S(INVALID_STATE),                       \
  38         S(TOGGLING),                    \
  39         S(SRC_UNATTACHED),                      \
  40         S(SRC_ATTACH_WAIT),                     \
  41         S(SRC_ATTACHED),                        \
  42         S(SRC_STARTUP),                         \
  43         S(SRC_SEND_CAPABILITIES),               \
  44         S(SRC_SEND_CAPABILITIES_TIMEOUT),       \
  45         S(SRC_NEGOTIATE_CAPABILITIES),          \
  46         S(SRC_TRANSITION_SUPPLY),               \
  47         S(SRC_READY),                           \
  48         S(SRC_WAIT_NEW_CAPABILITIES),           \
  49                                                 \
  50         S(SNK_UNATTACHED),                      \
  51         S(SNK_ATTACH_WAIT),                     \
  52         S(SNK_DEBOUNCED),                       \
  53         S(SNK_ATTACHED),                        \
  54         S(SNK_STARTUP),                         \
  55         S(SNK_DISCOVERY),                       \
  56         S(SNK_DISCOVERY_DEBOUNCE),              \
  57         S(SNK_DISCOVERY_DEBOUNCE_DONE),         \
  58         S(SNK_WAIT_CAPABILITIES),               \
  59         S(SNK_NEGOTIATE_CAPABILITIES),          \
  60         S(SNK_NEGOTIATE_PPS_CAPABILITIES),      \
  61         S(SNK_TRANSITION_SINK),                 \
  62         S(SNK_TRANSITION_SINK_VBUS),            \
  63         S(SNK_READY),                           \
  64                                                 \
  65         S(ACC_UNATTACHED),                      \
  66         S(DEBUG_ACC_ATTACHED),                  \
  67         S(AUDIO_ACC_ATTACHED),                  \
  68         S(AUDIO_ACC_DEBOUNCE),                  \
  69                                                 \
  70         S(HARD_RESET_SEND),                     \
  71         S(HARD_RESET_START),                    \
  72         S(SRC_HARD_RESET_VBUS_OFF),             \
  73         S(SRC_HARD_RESET_VBUS_ON),              \
  74         S(SNK_HARD_RESET_SINK_OFF),             \
  75         S(SNK_HARD_RESET_WAIT_VBUS),            \
  76         S(SNK_HARD_RESET_SINK_ON),              \
  77                                                 \
  78         S(SOFT_RESET),                          \
  79         S(SRC_SOFT_RESET_WAIT_SNK_TX),          \
  80         S(SNK_SOFT_RESET),                      \
  81         S(SOFT_RESET_SEND),                     \
  82                                                 \
  83         S(DR_SWAP_ACCEPT),                      \
  84         S(DR_SWAP_SEND),                        \
  85         S(DR_SWAP_SEND_TIMEOUT),                \
  86         S(DR_SWAP_CANCEL),                      \
  87         S(DR_SWAP_CHANGE_DR),                   \
  88                                                 \
  89         S(PR_SWAP_ACCEPT),                      \
  90         S(PR_SWAP_SEND),                        \
  91         S(PR_SWAP_SEND_TIMEOUT),                \
  92         S(PR_SWAP_CANCEL),                      \
  93         S(PR_SWAP_START),                       \
  94         S(PR_SWAP_SRC_SNK_TRANSITION_OFF),      \
  95         S(PR_SWAP_SRC_SNK_SOURCE_OFF),          \
  96         S(PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED), \
  97         S(PR_SWAP_SRC_SNK_SINK_ON),             \
  98         S(PR_SWAP_SNK_SRC_SINK_OFF),            \
  99         S(PR_SWAP_SNK_SRC_SOURCE_ON),           \
 100         S(PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP),    \
 101                                                 \
 102         S(VCONN_SWAP_ACCEPT),                   \
 103         S(VCONN_SWAP_SEND),                     \
 104         S(VCONN_SWAP_SEND_TIMEOUT),             \
 105         S(VCONN_SWAP_CANCEL),                   \
 106         S(VCONN_SWAP_START),                    \
 107         S(VCONN_SWAP_WAIT_FOR_VCONN),           \
 108         S(VCONN_SWAP_TURN_ON_VCONN),            \
 109         S(VCONN_SWAP_TURN_OFF_VCONN),           \
 110                                                 \
 111         S(FR_SWAP_SEND),                        \
 112         S(FR_SWAP_SEND_TIMEOUT),                \
 113         S(FR_SWAP_SNK_SRC_TRANSITION_TO_OFF),                   \
 114         S(FR_SWAP_SNK_SRC_NEW_SINK_READY),              \
 115         S(FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED), \
 116         S(FR_SWAP_CANCEL),                      \
 117                                                 \
 118         S(SNK_TRY),                             \
 119         S(SNK_TRY_WAIT),                        \
 120         S(SNK_TRY_WAIT_DEBOUNCE),               \
 121         S(SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS),    \
 122         S(SRC_TRYWAIT),                         \
 123         S(SRC_TRYWAIT_DEBOUNCE),                \
 124         S(SRC_TRYWAIT_UNATTACHED),              \
 125                                                 \
 126         S(SRC_TRY),                             \
 127         S(SRC_TRY_WAIT),                        \
 128         S(SRC_TRY_DEBOUNCE),                    \
 129         S(SNK_TRYWAIT),                         \
 130         S(SNK_TRYWAIT_DEBOUNCE),                \
 131         S(SNK_TRYWAIT_VBUS),                    \
 132         S(BIST_RX),                             \
 133                                                 \
 134         S(GET_STATUS_SEND),                     \
 135         S(GET_STATUS_SEND_TIMEOUT),             \
 136         S(GET_PPS_STATUS_SEND),                 \
 137         S(GET_PPS_STATUS_SEND_TIMEOUT),         \
 138                                                 \
 139         S(GET_SINK_CAP),                        \
 140         S(GET_SINK_CAP_TIMEOUT),                \
 141                                                 \
 142         S(ERROR_RECOVERY),                      \
 143         S(PORT_RESET),                          \
 144         S(PORT_RESET_WAIT_OFF),                 \
 145                                                 \
 146         S(AMS_START),                           \
 147         S(CHUNK_NOT_SUPP)
 148
 149 #define FOREACH_AMS(S)                          \
 150         S(NONE_AMS),                            \
 151         S(POWER_NEGOTIATION),                   \
 152         S(GOTOMIN),                             \
 153         S(SOFT_RESET_AMS),                      \
 154         S(HARD_RESET),                          \
 155         S(CABLE_RESET),                         \
 156         S(GET_SOURCE_CAPABILITIES),             \
 157         S(GET_SINK_CAPABILITIES),               \
 158         S(POWER_ROLE_SWAP),                     \
 159         S(FAST_ROLE_SWAP),                      \
 160         S(DATA_ROLE_SWAP),                      \
 161         S(VCONN_SWAP),                          \
 162         S(SOURCE_ALERT),                        \
 163         S(GETTING_SOURCE_EXTENDED_CAPABILITIES),\
 164         S(GETTING_SOURCE_SINK_STATUS),          \
 165         S(GETTING_BATTERY_CAPABILITIES),        \
 166         S(GETTING_BATTERY_STATUS),              \
 167         S(GETTING_MANUFACTURER_INFORMATION),    \
 168         S(SECURITY),                            \
 169         S(FIRMWARE_UPDATE),                     \
 170         S(DISCOVER_IDENTITY),                   \
 171         S(SOURCE_STARTUP_CABLE_PLUG_DISCOVER_IDENTITY), \
 172         S(DISCOVER_SVIDS),                      \
 173         S(DISCOVER_MODES),                      \
 174         S(DFP_TO_UFP_ENTER_MODE),               \
 175         S(DFP_TO_UFP_EXIT_MODE),                \
 176         S(DFP_TO_CABLE_PLUG_ENTER_MODE),        \
 177         S(DFP_TO_CABLE_PLUG_EXIT_MODE),         \
 178         S(ATTENTION),                           \
 179         S(BIST),                                \
 180         S(UNSTRUCTURED_VDMS),                   \
 181         S(STRUCTURED_VDMS),                     \
 182         S(COUNTRY_INFO),                        \
 183         S(COUNTRY_CODES)
 184
 185 #define GENERATE_ENUM(e)        e
 186 #define GENERATE_STRING(s)      #s
 187
 188 enum tcpm_state {
 189         FOREACH_STATE(GENERATE_ENUM)
 190 };
 191
 192 static const char * const tcpm_states[] = {
 193         FOREACH_STATE(GENERATE_STRING)
 194 };
 195
 196 enum tcpm_ams {
 197         FOREACH_AMS(GENERATE_ENUM)
 198 };
 199
 200 static const char * const tcpm_ams_str[] = {
 201         FOREACH_AMS(GENERATE_STRING)
 202 };
 203
 204 enum vdm_states {
 205         VDM_STATE_ERR_BUSY = -3,
 206         VDM_STATE_ERR_SEND = -2,
 207         VDM_STATE_ERR_TMOUT = -1,
 208         VDM_STATE_DONE = 0,
 209         /* Anything >0 represents an active state */
 210         VDM_STATE_READY = 1,
 211         VDM_STATE_BUSY = 2,
 212         VDM_STATE_WAIT_RSP_BUSY = 3,
 213         VDM_STATE_SEND_MESSAGE = 4,
 214 };
 215
 216 enum pd_msg_request {
 217         PD_MSG_NONE = 0,
 218         PD_MSG_CTRL_REJECT,
 219         PD_MSG_CTRL_WAIT,
 220         PD_MSG_CTRL_NOT_SUPP,
 221         PD_MSG_DATA_SINK_CAP,
 222         PD_MSG_DATA_SOURCE_CAP,
 223 };
 224
 225 enum adev_actions {
 226         ADEV_NONE = 0,
 227         ADEV_NOTIFY_USB_AND_QUEUE_VDM,
 228         ADEV_QUEUE_VDM,
 229         ADEV_QUEUE_VDM_SEND_EXIT_MODE_ON_FAIL,
 230         ADEV_ATTENTION,
 231 };
 232
 233 /*
 234  * Initial current capability of the new source when vSafe5V is applied during PD3.0 Fast Role Swap.
 235  * Based on "Table 6-14 Fixed Supply PDO - Sink" of "USB Power Delivery Specification Revision 3.0,
 236  * Version 1.2"
 237  */
 238 enum frs_typec_current {
 239         FRS_NOT_SUPPORTED,
 240         FRS_DEFAULT_POWER,
 241         FRS_5V_1P5A,
 242         FRS_5V_3A,
 243 };
 244
 245 /* Events from low level driver */
 246
 247 #define TCPM_CC_EVENT           BIT(0)
 248 #define TCPM_VBUS_EVENT         BIT(1)
 249 #define TCPM_RESET_EVENT        BIT(2)
 250 #define TCPM_FRS_EVENT          BIT(3)
 251 #define TCPM_SOURCING_VBUS      BIT(4)
 252
 253 #define LOG_BUFFER_ENTRIES      1024
 254 #define LOG_BUFFER_ENTRY_SIZE   128
 255
 256 /* Alternate mode support */
 257
 258 #define SVID_DISCOVERY_MAX      16
 259 #define ALTMODE_DISCOVERY_MAX   (SVID_DISCOVERY_MAX * MODE_DISCOVERY_MAX)
 260
 261 #define GET_SINK_CAP_RETRY_MS   100
 262
 263 struct pd_mode_data {
 264         int svid_index;         /* current SVID index           */
 265         int nsvids;
 266         u16 svids[SVID_DISCOVERY_MAX];
 267         int altmodes;           /* number of alternate modes    */
 268         struct typec_altmode_desc altmode_desc[ALTMODE_DISCOVERY_MAX];
 269 };
 270
 271 struct pd_pps_data {
 272         u32 min_volt;
 273         u32 max_volt;
 274         u32 max_curr;
 275         u32 out_volt;
 276         u32 op_curr;
 277         bool supported;
 278         bool active;
 279 };
 280
 281 struct tcpm_port {
 282         struct device *dev;
 283
 284         struct mutex lock;              /* tcpm state machine lock */
 285         struct kthread_worker *wq;
 286
 287         struct typec_capability typec_caps;
 288         struct typec_port *typec_port;
 289
 290         struct tcpc_dev *tcpc;
 291         struct usb_role_switch *role_sw;
 292
 293         enum typec_role vconn_role;
 294         enum typec_role pwr_role;
 295         enum typec_data_role data_role;
 296         enum typec_pwr_opmode pwr_opmode;
 297
 298         struct usb_pd_identity partner_ident;
 299         struct typec_partner_desc partner_desc;
 300         struct typec_partner *partner;
 301
 302         enum typec_cc_status cc_req;
 303
 304         enum typec_cc_status cc1;
 305         enum typec_cc_status cc2;
 306         enum typec_cc_polarity polarity;
 307
 308         bool attached;
 309         bool connected;
 310         enum typec_port_type port_type;
 311
 312         /*
 313          * Set to true when vbus is greater than VSAFE5V min.
 314          * Set to false when vbus falls below vSinkDisconnect max threshold.
 315          */
 316         bool vbus_present;
 317
 318         /*
 319          * Set to true when vbus is less than VSAFE0V max.
 320          * Set to false when vbus is greater than VSAFE0V max.
 321          */
 322         bool vbus_vsafe0v;
 323
 324         bool vbus_never_low;
 325         bool vbus_source;
 326         bool vbus_charge;
 327
 328         bool send_discover;
 329         bool op_vsafe5v;
 330
 331         int try_role;
 332         int try_snk_count;
 333         int try_src_count;
 334
 335         enum pd_msg_request queued_message;
 336
 337         enum tcpm_state enter_state;
 338         enum tcpm_state prev_state;
 339         enum tcpm_state state;
 340         enum tcpm_state delayed_state;
 341         ktime_t delayed_runtime;
 342         unsigned long delay_ms;
 343
 344         spinlock_t pd_event_lock;
 345         u32 pd_events;
 346
 347         struct kthread_work event_work;
 348         struct hrtimer state_machine_timer;
 349         struct kthread_work state_machine;
 350         struct hrtimer vdm_state_machine_timer;
 351         struct kthread_work vdm_state_machine;
 352         struct hrtimer enable_frs_timer;
 353         struct kthread_work enable_frs;
 354         bool state_machine_running;
 355         bool vdm_sm_running;
 356
 357         struct completion tx_complete;
 358         enum tcpm_transmit_status tx_status;
 359
 360         struct mutex swap_lock;         /* swap command lock */
 361         bool swap_pending;
 362         bool non_pd_role_swap;
 363         struct completion swap_complete;
 364         int swap_status;
 365
 366         unsigned int negotiated_rev;
 367         unsigned int message_id;
 368         unsigned int caps_count;
 369         unsigned int hard_reset_count;
 370         bool pd_capable;
 371         bool explicit_contract;
 372         unsigned int rx_msgid;
 373
 374         /* Partner capabilities/requests */
 375         u32 sink_request;
 376         u32 source_caps[PDO_MAX_OBJECTS];
 377         unsigned int nr_source_caps;
 378         u32 sink_caps[PDO_MAX_OBJECTS];
 379         unsigned int nr_sink_caps;
 380
 381         /* Local capabilities */
 382         u32 src_pdo[PDO_MAX_OBJECTS];
 383         unsigned int nr_src_pdo;
 384         u32 snk_pdo[PDO_MAX_OBJECTS];
 385         unsigned int nr_snk_pdo;
 386         u32 snk_vdo[VDO_MAX_OBJECTS];
 387         unsigned int nr_snk_vdo;
 388
 389         unsigned int operating_snk_mw;
 390         bool update_sink_caps;
 391
 392         /* Requested current / voltage */
 393         u32 current_limit;
 394         u32 supply_voltage;
 395
 396         /* Used to export TA voltage and current */
 397         struct power_supply *psy;
 398         struct power_supply_desc psy_desc;
 399         enum power_supply_usb_type usb_type;
 400
 401         u32 bist_request;
 402
 403         /* PD state for Vendor Defined Messages */
 404         enum vdm_states vdm_state;
 405         u32 vdm_retries;
 406         /* next Vendor Defined Message to send */
 407         u32 vdo_data[VDO_MAX_SIZE];
 408         u8 vdo_count;
 409         /* VDO to retry if UFP responder replied busy */
 410         u32 vdo_retry;
 411
 412         /* PPS */
 413         struct pd_pps_data pps_data;
 414         struct completion pps_complete;
 415         bool pps_pending;
 416         int pps_status;
 417
 418         /* Alternate mode data */
 419         struct pd_mode_data mode_data;
 420         struct typec_altmode *partner_altmode[ALTMODE_DISCOVERY_MAX];
 421         struct typec_altmode *port_altmode[ALTMODE_DISCOVERY_MAX];
 422
 423         /* Deadline in jiffies to exit src_try_wait state */
 424         unsigned long max_wait;
 425
 426         /* port belongs to a self powered device */
 427         bool self_powered;
 428
 429         /* Sink FRS */
 430         enum frs_typec_current new_source_frs_current;
 431
 432         /* Sink caps have been queried */
 433         bool sink_cap_done;
 434
 435         /* Collision Avoidance and Atomic Message Sequence */
 436         enum tcpm_state upcoming_state;
 437         enum tcpm_ams ams;
 438         enum tcpm_ams next_ams;
 439         bool in_ams;
 440
 441         /* Auto vbus discharge status */
 442         bool auto_vbus_discharge_enabled;
 443
 444 #ifdef CONFIG_DEBUG_FS
 445         struct dentry *dentry;
 446         struct mutex logbuffer_lock;    /* log buffer access lock */
 447         int logbuffer_head;
 448         int logbuffer_tail;
 449         u8 *logbuffer[LOG_BUFFER_ENTRIES];
 450 #endif
 451 };
 452
 453 struct pd_rx_event {
 454         struct kthread_work work;
 455         struct tcpm_port *port;
 456         struct pd_message msg;
 457 };
 458
 459 static const char * const pd_rev[] = {
 460         [PD_REV10]              = "rev1",
 461         [PD_REV20]              = "rev2",
 462         [PD_REV30]              = "rev3",
 463 };
 464
 465 #define tcpm_cc_is_sink(cc) \
 466         ((cc) == TYPEC_CC_RP_DEF || (cc) == TYPEC_CC_RP_1_5 || \
 467          (cc) == TYPEC_CC_RP_3_0)
 468
 469 #define tcpm_port_is_sink(port) \
 470         ((tcpm_cc_is_sink((port)->cc1) && !tcpm_cc_is_sink((port)->cc2)) || \
 471          (tcpm_cc_is_sink((port)->cc2) && !tcpm_cc_is_sink((port)->cc1)))
 472
 473 #define tcpm_cc_is_source(cc) ((cc) == TYPEC_CC_RD)
 474 #define tcpm_cc_is_audio(cc) ((cc) == TYPEC_CC_RA)
 475 #define tcpm_cc_is_open(cc) ((cc) == TYPEC_CC_OPEN)
 476
 477 #define tcpm_port_is_source(port) \
 478         ((tcpm_cc_is_source((port)->cc1) && \
 479          !tcpm_cc_is_source((port)->cc2)) || \
 480          (tcpm_cc_is_source((port)->cc2) && \
 481           !tcpm_cc_is_source((port)->cc1)))
 482
 483 #define tcpm_port_is_debug(port) \
 484         (tcpm_cc_is_source((port)->cc1) && tcpm_cc_is_source((port)->cc2))
 485
 486 #define tcpm_port_is_audio(port) \
 487         (tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_audio((port)->cc2))
 488
 489 #define tcpm_port_is_audio_detached(port) \
 490         ((tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_open((port)->cc2)) || \
 491          (tcpm_cc_is_audio((port)->cc2) && tcpm_cc_is_open((port)->cc1)))
 492
 493 #define tcpm_try_snk(port) \
 494         ((port)->try_snk_count == 0 && (port)->try_role == TYPEC_SINK && \
 495         (port)->port_type == TYPEC_PORT_DRP)
 496
 497 #define tcpm_try_src(port) \
 498         ((port)->try_src_count == 0 && (port)->try_role == TYPEC_SOURCE && \
 499         (port)->port_type == TYPEC_PORT_DRP)
 500
 501 #define tcpm_data_role_for_source(port) \
 502         ((port)->typec_caps.data == TYPEC_PORT_UFP ? \
 503         TYPEC_DEVICE : TYPEC_HOST)
 504
 505 #define tcpm_data_role_for_sink(port) \
 506         ((port)->typec_caps.data == TYPEC_PORT_DFP ? \
 507         TYPEC_HOST : TYPEC_DEVICE)
 508
 509 #define tcpm_sink_tx_ok(port) \
 510         (tcpm_port_is_sink(port) && \
 511         ((port)->cc1 == TYPEC_CC_RP_3_0 || (port)->cc2 == TYPEC_CC_RP_3_0))
 512
 513 #define tcpm_wait_for_discharge(port) \
 514         (((port)->auto_vbus_discharge_enabled && !(port)->vbus_vsafe0v) ? PD_T_SAFE_0V : 0)
 515
 516 static enum tcpm_state tcpm_default_state(struct tcpm_port *port)
 517 {
 518         if (port->port_type == TYPEC_PORT_DRP) {
 519                 if (port->try_role == TYPEC_SINK)
 520                         return SNK_UNATTACHED;
 521                 else if (port->try_role == TYPEC_SOURCE)
 522                         return SRC_UNATTACHED;
 523                 /* Fall through to return SRC_UNATTACHED */
 524         } else if (port->port_type == TYPEC_PORT_SNK) {
 525                 return SNK_UNATTACHED;
 526         }
 527         return SRC_UNATTACHED;
 528 }
 529
 530 static bool tcpm_port_is_disconnected(struct tcpm_port *port)
 531 {
 532         return (!port->attached && port->cc1 == TYPEC_CC_OPEN &&
 533                 port->cc2 == TYPEC_CC_OPEN) ||
 534                (port->attached && ((port->polarity == TYPEC_POLARITY_CC1 &&
 535                                     port->cc1 == TYPEC_CC_OPEN) ||
 536                                    (port->polarity == TYPEC_POLARITY_CC2 &&
 537                                     port->cc2 == TYPEC_CC_OPEN)));
 538 }
 539
 540 /*
 541  * Logging
 542  */
 543
 544 #ifdef CONFIG_DEBUG_FS
 545
 546 static bool tcpm_log_full(struct tcpm_port *port)
 547 {
 548         return port->logbuffer_tail ==
 549                 (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 550 }
 551
 552 __printf(2, 0)
 553 static void _tcpm_log(struct tcpm_port *port, const char *fmt, va_list args)
 554 {
 555         char tmpbuffer[LOG_BUFFER_ENTRY_SIZE];
 556         u64 ts_nsec = local_clock();
 557         unsigned long rem_nsec;
 558
 559         mutex_lock(&port->logbuffer_lock);
 560         if (!port->logbuffer[port->logbuffer_head]) {
 561                 port->logbuffer[port->logbuffer_head] =
 562                                 kzalloc(LOG_BUFFER_ENTRY_SIZE, GFP_KERNEL);
 563                 if (!port->logbuffer[port->logbuffer_head]) {
 564                         mutex_unlock(&port->logbuffer_lock);
 565                         return;
 566                 }
 567         }
 568
 569         vsnprintf(tmpbuffer, sizeof(tmpbuffer), fmt, args);
 570
 571         if (tcpm_log_full(port)) {
 572                 port->logbuffer_head = max(port->logbuffer_head - 1, 0);
 573                 strcpy(tmpbuffer, "overflow");
 574         }
 575
 576         if (port->logbuffer_head < 0 ||
 577             port->logbuffer_head >= LOG_BUFFER_ENTRIES) {
 578                 dev_warn(port->dev,
 579                          "Bad log buffer index %d\n", port->logbuffer_head);
 580                 goto abort;
 581         }
 582
 583         if (!port->logbuffer[port->logbuffer_head]) {
 584                 dev_warn(port->dev,
 585                          "Log buffer index %d is NULL\n", port->logbuffer_head);
 586                 goto abort;
 587         }
 588
 589         rem_nsec = do_div(ts_nsec, 1000000000);
 590         scnprintf(port->logbuffer[port->logbuffer_head],
 591                   LOG_BUFFER_ENTRY_SIZE, "[%5lu.%06lu] %s",
 592                   (unsigned long)ts_nsec, rem_nsec / 1000,
 593                   tmpbuffer);
 594         port->logbuffer_head = (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 595
 596 abort:
 597         mutex_unlock(&port->logbuffer_lock);
 598 }
 599
 600 __printf(2, 3)
 601 static void tcpm_log(struct tcpm_port *port, const char *fmt, ...)
 602 {
 603         va_list args;
 604
 605         /* Do not log while disconnected and unattached */
 606         if (tcpm_port_is_disconnected(port) &&
 607             (port->state == SRC_UNATTACHED || port->state == SNK_UNATTACHED ||
 608              port->state == TOGGLING))
 609                 return;
 610
 611         va_start(args, fmt);
 612         _tcpm_log(port, fmt, args);
 613         va_end(args);
 614 }
 615
 616 __printf(2, 3)
 617 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...)
 618 {
 619         va_list args;
 620
 621         va_start(args, fmt);
 622         _tcpm_log(port, fmt, args);
 623         va_end(args);
 624 }
 625
 626 static void tcpm_log_source_caps(struct tcpm_port *port)
 627 {
 628         int i;
 629
 630         for (i = 0; i < port->nr_source_caps; i++) {
 631                 u32 pdo = port->source_caps[i];
 632                 enum pd_pdo_type type = pdo_type(pdo);
 633                 char msg[64];
 634
 635                 switch (type) {
 636                 case PDO_TYPE_FIXED:
 637                         scnprintf(msg, sizeof(msg),
 638                                   "%u mV, %u mA [%s%s%s%s%s%s]",
 639                                   pdo_fixed_voltage(pdo),
 640                                   pdo_max_current(pdo),
 641                                   (pdo & PDO_FIXED_DUAL_ROLE) ?
 642                                                         "R" : "",
 643                                   (pdo & PDO_FIXED_SUSPEND) ?
 644                                                         "S" : "",
 645                                   (pdo & PDO_FIXED_HIGHER_CAP) ?
 646                                                         "H" : "",
 647                                   (pdo & PDO_FIXED_USB_COMM) ?
 648                                                         "U" : "",
 649                                   (pdo & PDO_FIXED_DATA_SWAP) ?
 650                                                         "D" : "",
 651                                   (pdo & PDO_FIXED_EXTPOWER) ?
 652                                                         "E" : "");
 653                         break;
 654                 case PDO_TYPE_VAR:
 655                         scnprintf(msg, sizeof(msg),
 656                                   "%u-%u mV, %u mA",
 657                                   pdo_min_voltage(pdo),
 658                                   pdo_max_voltage(pdo),
 659                                   pdo_max_current(pdo));
 660                         break;
 661                 case PDO_TYPE_BATT:
 662                         scnprintf(msg, sizeof(msg),
 663                                   "%u-%u mV, %u mW",
 664                                   pdo_min_voltage(pdo),
 665                                   pdo_max_voltage(pdo),
 666                                   pdo_max_power(pdo));
 667                         break;
 668                 case PDO_TYPE_APDO:
 669                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS)
 670                                 scnprintf(msg, sizeof(msg),
 671                                           "%u-%u mV, %u mA",
 672                                           pdo_pps_apdo_min_voltage(pdo),
 673                                           pdo_pps_apdo_max_voltage(pdo),
 674                                           pdo_pps_apdo_max_current(pdo));
 675                         else
 676                                 strcpy(msg, "undefined APDO");
 677                         break;
 678                 default:
 679                         strcpy(msg, "undefined");
 680                         break;
 681                 }
 682                 tcpm_log(port, " PDO %d: type %d, %s",
 683                          i, type, msg);
 684         }
 685 }
 686
 687 static int tcpm_debug_show(struct seq_file *s, void *v)
 688 {
 689         struct tcpm_port *port = (struct tcpm_port *)s->private;
 690         int tail;
 691
 692         mutex_lock(&port->logbuffer_lock);
 693         tail = port->logbuffer_tail;
 694         while (tail != port->logbuffer_head) {
 695                 seq_printf(s, "%s\n", port->logbuffer[tail]);
 696                 tail = (tail + 1) % LOG_BUFFER_ENTRIES;
 697         }
 698         if (!seq_has_overflowed(s))
 699                 port->logbuffer_tail = tail;
 700         mutex_unlock(&port->logbuffer_lock);
 701
 702         return 0;
 703 }
 704 DEFINE_SHOW_ATTRIBUTE(tcpm_debug);
 705
 706 static void tcpm_debugfs_init(struct tcpm_port *port)
 707 {
 708         char name[NAME_MAX];
 709
 710         mutex_init(&port->logbuffer_lock);
 711         snprintf(name, NAME_MAX, "tcpm-%s", dev_name(port->dev));
 712         port->dentry = debugfs_create_dir(name, usb_debug_root);
 713         debugfs_create_file("log", S_IFREG | 0444, port->dentry, port,
 714                             &tcpm_debug_fops);
 715 }
 716
 717 static void tcpm_debugfs_exit(struct tcpm_port *port)
 718 {
 719         int i;
 720
 721         mutex_lock(&port->logbuffer_lock);
 722         for (i = 0; i < LOG_BUFFER_ENTRIES; i++) {
 723                 kfree(port->logbuffer[i]);
 724                 port->logbuffer[i] = NULL;
 725         }
 726         mutex_unlock(&port->logbuffer_lock);
 727
 728         debugfs_remove(port->dentry);
 729 }
 730
 731 #else
 732
 733 __printf(2, 3)
 734 static void tcpm_log(const struct tcpm_port *port, const char *fmt, ...) { }
 735 __printf(2, 3)
 736 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...) { }
 737 static void tcpm_log_source_caps(struct tcpm_port *port) { }
 738 static void tcpm_debugfs_init(const struct tcpm_port *port) { }
 739 static void tcpm_debugfs_exit(const struct tcpm_port *port) { }
 740
 741 #endif
 742
 743 static void tcpm_set_cc(struct tcpm_port *port, enum typec_cc_status cc)
 744 {
 745         tcpm_log(port, "cc:=%d", cc);
 746         port->cc_req = cc;
 747         port->tcpc->set_cc(port->tcpc, cc);
 748 }
 749
 750 /*
 751  * Determine RP value to set based on maximum current supported
 752  * by a port if configured as source.
 753  * Returns CC value to report to link partner.
 754  */
 755 static enum typec_cc_status tcpm_rp_cc(struct tcpm_port *port)
 756 {
 757         const u32 *src_pdo = port->src_pdo;
 758         int nr_pdo = port->nr_src_pdo;
 759         int i;
 760
 761         /*
 762          * Search for first entry with matching voltage.
 763          * It should report the maximum supported current.
 764          */
 765         for (i = 0; i < nr_pdo; i++) {
 766                 const u32 pdo = src_pdo[i];
 767
 768                 if (pdo_type(pdo) == PDO_TYPE_FIXED &&
 769                     pdo_fixed_voltage(pdo) == 5000) {
 770                         unsigned int curr = pdo_max_current(pdo);
 771
 772                         if (curr >= 3000)
 773                                 return TYPEC_CC_RP_3_0;
 774                         else if (curr >= 1500)
 775                                 return TYPEC_CC_RP_1_5;
 776                         return TYPEC_CC_RP_DEF;
 777                 }
 778         }
 779
 780         return TYPEC_CC_RP_DEF;
 781 }
 782
 783 static void tcpm_ams_finish(struct tcpm_port *port)
 784 {
 785         tcpm_log(port, "AMS %s finished", tcpm_ams_str[port->ams]);
 786
 787         if (port->pd_capable && port->pwr_role == TYPEC_SOURCE) {
 788                 if (port->negotiated_rev >= PD_REV30)
 789                         tcpm_set_cc(port, SINK_TX_OK);
 790                 else
 791                         tcpm_set_cc(port, SINK_TX_NG);
 792         } else if (port->pwr_role == TYPEC_SOURCE) {
 793                 tcpm_set_cc(port, tcpm_rp_cc(port));
 794         }
 795
 796         port->in_ams = false;
 797         port->ams = NONE_AMS;
 798 }
 799
 800 static int tcpm_pd_transmit(struct tcpm_port *port,
 801                             enum tcpm_transmit_type type,
 802                             const struct pd_message *msg)
 803 {
 804         unsigned long timeout;
 805         int ret;
 806
 807         if (msg)
 808                 tcpm_log(port, "PD TX, header: %#x", le16_to_cpu(msg->header));
 809         else
 810                 tcpm_log(port, "PD TX, type: %#x", type);
 811
 812         reinit_completion(&port->tx_complete);
 813         ret = port->tcpc->pd_transmit(port->tcpc, type, msg, port->negotiated_rev);
 814         if (ret < 0)
 815                 return ret;
 816
 817         mutex_unlock(&port->lock);
 818         timeout = wait_for_completion_timeout(&port->tx_complete,
 819                                 msecs_to_jiffies(PD_T_TCPC_TX_TIMEOUT));
 820         mutex_lock(&port->lock);
 821         if (!timeout)
 822                 return -ETIMEDOUT;
 823
 824         switch (port->tx_status) {
 825         case TCPC_TX_SUCCESS:
 826                 port->message_id = (port->message_id + 1) & PD_HEADER_ID_MASK;
 827                 /*
 828                  * USB PD rev 2.0, 8.3.2.2.1:
 829                  * USB PD rev 3.0, 8.3.2.1.3:
 830                  * "... Note that every AMS is Interruptible until the first
 831                  * Message in the sequence has been successfully sent (GoodCRC
 832                  * Message received)."
 833                  */
 834                 if (port->ams != NONE_AMS)
 835                         port->in_ams = true;
 836                 break;
 837         case TCPC_TX_DISCARDED:
 838                 ret = -EAGAIN;
 839                 break;
 840         case TCPC_TX_FAILED:
 841         default:
 842                 ret = -EIO;
 843                 break;
 844         }
 845
 846         /* Some AMS don't expect responses. Finish them here. */
 847         if (port->ams == ATTENTION || port->ams == SOURCE_ALERT)
 848                 tcpm_ams_finish(port);
 849
 850         return ret;
 851 }
 852
 853 void tcpm_pd_transmit_complete(struct tcpm_port *port,
 854                                enum tcpm_transmit_status status)
 855 {
 856         tcpm_log(port, "PD TX complete, status: %u", status);
 857         port->tx_status = status;
 858         complete(&port->tx_complete);
 859 }
 860 EXPORT_SYMBOL_GPL(tcpm_pd_transmit_complete);
 861
 862 static int tcpm_mux_set(struct tcpm_port *port, int state,
 863                         enum usb_role usb_role,
 864                         enum typec_orientation orientation)
 865 {
 866         int ret;
 867
 868         tcpm_log(port, "Requesting mux state %d, usb-role %d, orientation %d",
 869                  state, usb_role, orientation);
 870
 871         ret = typec_set_orientation(port->typec_port, orientation);
 872         if (ret)
 873                 return ret;
 874
 875         if (port->role_sw) {
 876                 ret = usb_role_switch_set_role(port->role_sw, usb_role);
 877                 if (ret)
 878                         return ret;
 879         }
 880
 881         return typec_set_mode(port->typec_port, state);
 882 }
 883
 884 static int tcpm_set_polarity(struct tcpm_port *port,
 885                              enum typec_cc_polarity polarity)
 886 {
 887         int ret;
 888
 889         tcpm_log(port, "polarity %d", polarity);
 890
 891         ret = port->tcpc->set_polarity(port->tcpc, polarity);
 892         if (ret < 0)
 893                 return ret;
 894
 895         port->polarity = polarity;
 896
 897         return 0;
 898 }
 899
 900 static int tcpm_set_vconn(struct tcpm_port *port, bool enable)
 901 {
 902         int ret;
 903
 904         tcpm_log(port, "vconn:=%d", enable);
 905
 906         ret = port->tcpc->set_vconn(port->tcpc, enable);
 907         if (!ret) {
 908                 port->vconn_role = enable ? TYPEC_SOURCE : TYPEC_SINK;
 909                 typec_set_vconn_role(port->typec_port, port->vconn_role);
 910         }
 911
 912         return ret;
 913 }
 914
 915 static u32 tcpm_get_current_limit(struct tcpm_port *port)
 916 {
 917         enum typec_cc_status cc;
 918         u32 limit;
 919
 920         cc = port->polarity ? port->cc2 : port->cc1;
 921         switch (cc) {
 922         case TYPEC_CC_RP_1_5:
 923                 limit = 1500;
 924                 break;
 925         case TYPEC_CC_RP_3_0:
 926                 limit = 3000;
 927                 break;
 928         case TYPEC_CC_RP_DEF:
 929         default:
 930                 if (port->tcpc->get_current_limit)
 931                         limit = port->tcpc->get_current_limit(port->tcpc);
 932                 else
 933                         limit = 0;
 934                 break;
 935         }
 936
 937         return limit;
 938 }
 939
 940 static int tcpm_set_current_limit(struct tcpm_port *port, u32 max_ma, u32 mv)
 941 {
 942         int ret = -EOPNOTSUPP;
 943
 944         tcpm_log(port, "Setting voltage/current limit %u mV %u mA", mv, max_ma);
 945
 946         port->supply_voltage = mv;
 947         port->current_limit = max_ma;
 948         power_supply_changed(port->psy);
 949
 950         if (port->tcpc->set_current_limit)
 951                 ret = port->tcpc->set_current_limit(port->tcpc, max_ma, mv);
 952
 953         return ret;
 954 }
 955
 956 static int tcpm_set_attached_state(struct tcpm_port *port, bool attached)
 957 {
 958         return port->tcpc->set_roles(port->tcpc, attached, port->pwr_role,
 959                                      port->data_role);
 960 }
 961
 962 static int tcpm_set_roles(struct tcpm_port *port, bool attached,
 963                           enum typec_role role, enum typec_data_role data)
 964 {
 965         enum typec_orientation orientation;
 966         enum usb_role usb_role;
 967         int ret;
 968
 969         if (port->polarity == TYPEC_POLARITY_CC1)
 970                 orientation = TYPEC_ORIENTATION_NORMAL;
 971         else
 972                 orientation = TYPEC_ORIENTATION_REVERSE;
 973
 974         if (port->typec_caps.data == TYPEC_PORT_DRD) {
 975                 if (data == TYPEC_HOST)
 976                         usb_role = USB_ROLE_HOST;
 977                 else
 978                         usb_role = USB_ROLE_DEVICE;
 979         } else if (port->typec_caps.data == TYPEC_PORT_DFP) {
 980                 if (data == TYPEC_HOST) {
 981                         if (role == TYPEC_SOURCE)
 982                                 usb_role = USB_ROLE_HOST;
 983                         else
 984                                 usb_role = USB_ROLE_NONE;
 985                 } else {
 986                         return -ENOTSUPP;
 987                 }
 988         } else {
 989                 if (data == TYPEC_DEVICE) {
 990                         if (role == TYPEC_SINK)
 991                                 usb_role = USB_ROLE_DEVICE;
 992                         else
 993                                 usb_role = USB_ROLE_NONE;
 994                 } else {
 995                         return -ENOTSUPP;
 996                 }
 997         }
 998
 999         ret = tcpm_mux_set(port, TYPEC_STATE_USB, usb_role, orientation);
1000         if (ret < 0)
1001                 return ret;
1002
1003         ret = port->tcpc->set_roles(port->tcpc, attached, role, data);
1004         if (ret < 0)
1005                 return ret;
1006
1007         port->pwr_role = role;
1008         port->data_role = data;
1009         typec_set_data_role(port->typec_port, data);
1010         typec_set_pwr_role(port->typec_port, role);
1011
1012         return 0;
1013 }
1014
1015 static int tcpm_set_pwr_role(struct tcpm_port *port, enum typec_role role)
1016 {
1017         int ret;
1018
1019         ret = port->tcpc->set_roles(port->tcpc, true, role,
1020                                     port->data_role);
1021         if (ret < 0)
1022                 return ret;
1023
1024         port->pwr_role = role;
1025         typec_set_pwr_role(port->typec_port, role);
1026
1027         return 0;
1028 }
1029
1030 /*
1031  * Transform the PDO to be compliant to PD rev2.0.
1032  * Return 0 if the PDO type is not defined in PD rev2.0.
1033  * Otherwise, return the converted PDO.
1034  */
1035 static u32 tcpm_forge_legacy_pdo(struct tcpm_port *port, u32 pdo, enum typec_role role)
1036 {
1037         switch (pdo_type(pdo)) {
1038         case PDO_TYPE_FIXED:
1039                 if (role == TYPEC_SINK)
1040                         return pdo & ~PDO_FIXED_FRS_CURR_MASK;
1041                 else
1042                         return pdo & ~PDO_FIXED_UNCHUNK_EXT;
1043         case PDO_TYPE_VAR:
1044         case PDO_TYPE_BATT:
1045                 return pdo;
1046         case PDO_TYPE_APDO:
1047         default:
1048                 return 0;
1049         }
1050 }
1051
1052 static int tcpm_pd_send_source_caps(struct tcpm_port *port)
1053 {
1054         struct pd_message msg;
1055         u32 pdo;
1056         unsigned int i, nr_pdo = 0;
1057
1058         memset(&msg, 0, sizeof(msg));
1059
1060         for (i = 0; i < port->nr_src_pdo; i++) {
1061                 if (port->negotiated_rev >= PD_REV30) {
1062                         msg.payload[nr_pdo++] = cpu_to_le32(port->src_pdo[i]);
1063                 } else {
1064                         pdo = tcpm_forge_legacy_pdo(port, port->src_pdo[i], TYPEC_SOURCE);
1065                         if (pdo)
1066                                 msg.payload[nr_pdo++] = cpu_to_le32(pdo);
1067                 }
1068         }
1069
1070         if (!nr_pdo) {
1071                 /* No source capabilities defined, sink only */
1072                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
1073                                           port->pwr_role,
1074                                           port->data_role,
1075                                           port->negotiated_rev,
1076                                           port->message_id, 0);
1077         } else {
1078                 msg.header = PD_HEADER_LE(PD_DATA_SOURCE_CAP,
1079                                           port->pwr_role,
1080                                           port->data_role,
1081                                           port->negotiated_rev,
1082                                           port->message_id,
1083                                           nr_pdo);
1084         }
1085
1086         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1087 }
1088
1089 static int tcpm_pd_send_sink_caps(struct tcpm_port *port)
1090 {
1091         struct pd_message msg;
1092         u32 pdo;
1093         unsigned int i, nr_pdo = 0;
1094
1095         memset(&msg, 0, sizeof(msg));
1096
1097         for (i = 0; i < port->nr_snk_pdo; i++) {
1098                 if (port->negotiated_rev >= PD_REV30) {
1099                         msg.payload[nr_pdo++] = cpu_to_le32(port->snk_pdo[i]);
1100                 } else {
1101                         pdo = tcpm_forge_legacy_pdo(port, port->snk_pdo[i], TYPEC_SINK);
1102                         if (pdo)
1103                                 msg.payload[nr_pdo++] = cpu_to_le32(pdo);
1104                 }
1105         }
1106
1107         if (!nr_pdo) {
1108                 /* No sink capabilities defined, source only */
1109                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
1110                                           port->pwr_role,
1111                                           port->data_role,
1112                                           port->negotiated_rev,
1113                                           port->message_id, 0);
1114         } else {
1115                 msg.header = PD_HEADER_LE(PD_DATA_SINK_CAP,
1116                                           port->pwr_role,
1117                                           port->data_role,
1118                                           port->negotiated_rev,
1119                                           port->message_id,
1120                                           nr_pdo);
1121         }
1122
1123         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1124 }
1125
1126 static void mod_tcpm_delayed_work(struct tcpm_port *port, unsigned int delay_ms)
1127 {
1128         if (delay_ms) {
1129                 hrtimer_start(&port->state_machine_timer, ms_to_ktime(delay_ms), HRTIMER_MODE_REL);
1130         } else {
1131                 hrtimer_cancel(&port->state_machine_timer);
1132                 kthread_queue_work(port->wq, &port->state_machine);
1133         }
1134 }
1135
1136 static void mod_vdm_delayed_work(struct tcpm_port *port, unsigned int delay_ms)
1137 {
1138         if (delay_ms) {
1139                 hrtimer_start(&port->vdm_state_machine_timer, ms_to_ktime(delay_ms),
1140                               HRTIMER_MODE_REL);
1141         } else {
1142                 hrtimer_cancel(&port->vdm_state_machine_timer);
1143                 kthread_queue_work(port->wq, &port->vdm_state_machine);
1144         }
1145 }
1146
1147 static void mod_enable_frs_delayed_work(struct tcpm_port *port, unsigned int delay_ms)
1148 {
1149         if (delay_ms) {
1150                 hrtimer_start(&port->enable_frs_timer, ms_to_ktime(delay_ms), HRTIMER_MODE_REL);
1151         } else {
1152                 hrtimer_cancel(&port->enable_frs_timer);
1153                 kthread_queue_work(port->wq, &port->enable_frs);
1154         }
1155 }
1156
1157 static void tcpm_set_state(struct tcpm_port *port, enum tcpm_state state,
1158                            unsigned int delay_ms)
1159 {
1160         if (delay_ms) {
1161                 tcpm_log(port, "pending state change %s -> %s @ %u ms [%s %s]",
1162                          tcpm_states[port->state], tcpm_states[state], delay_ms,
1163                          pd_rev[port->negotiated_rev], tcpm_ams_str[port->ams]);
1164                 port->delayed_state = state;
1165                 mod_tcpm_delayed_work(port, delay_ms);
1166                 port->delayed_runtime = ktime_add(ktime_get(), ms_to_ktime(delay_ms));
1167                 port->delay_ms = delay_ms;
1168         } else {
1169                 tcpm_log(port, "state change %s -> %s [%s %s]",
1170                          tcpm_states[port->state], tcpm_states[state],
1171                          pd_rev[port->negotiated_rev], tcpm_ams_str[port->ams]);
1172                 port->delayed_state = INVALID_STATE;
1173                 port->prev_state = port->state;
1174                 port->state = state;
1175                 /*
1176                  * Don't re-queue the state machine work item if we're currently
1177                  * in the state machine and we're immediately changing states.
1178                  * tcpm_state_machine_work() will continue running the state
1179                  * machine.
1180                  */
1181                 if (!port->state_machine_running)
1182                         mod_tcpm_delayed_work(port, 0);
1183         }
1184 }
1185
1186 static void tcpm_set_state_cond(struct tcpm_port *port, enum tcpm_state state,
1187                                 unsigned int delay_ms)
1188 {
1189         if (port->enter_state == port->state)
1190                 tcpm_set_state(port, state, delay_ms);
1191         else
1192                 tcpm_log(port,
1193                          "skipped %sstate change %s -> %s [%u ms], context state %s [%s %s]",
1194                          delay_ms ? "delayed " : "",
1195                          tcpm_states[port->state], tcpm_states[state],
1196                          delay_ms, tcpm_states[port->enter_state],
1197                          pd_rev[port->negotiated_rev], tcpm_ams_str[port->ams]);
1198 }
1199
1200 static void tcpm_queue_message(struct tcpm_port *port,
1201                                enum pd_msg_request message)
1202 {
1203         port->queued_message = message;
1204         mod_tcpm_delayed_work(port, 0);
1205 }
1206
1207 static bool tcpm_vdm_ams(struct tcpm_port *port)
1208 {
1209         switch (port->ams) {
1210         case DISCOVER_IDENTITY:
1211         case SOURCE_STARTUP_CABLE_PLUG_DISCOVER_IDENTITY:
1212         case DISCOVER_SVIDS:
1213         case DISCOVER_MODES:
1214         case DFP_TO_UFP_ENTER_MODE:
1215         case DFP_TO_UFP_EXIT_MODE:
1216         case DFP_TO_CABLE_PLUG_ENTER_MODE:
1217         case DFP_TO_CABLE_PLUG_EXIT_MODE:
1218         case ATTENTION:
1219         case UNSTRUCTURED_VDMS:
1220         case STRUCTURED_VDMS:
1221                 break;
1222         default:
1223                 return false;
1224         }
1225
1226         return true;
1227 }
1228
1229 static bool tcpm_ams_interruptible(struct tcpm_port *port)
1230 {
1231         switch (port->ams) {
1232         /* Interruptible AMS */
1233         case NONE_AMS:
1234         case SECURITY:
1235         case FIRMWARE_UPDATE:
1236         case DISCOVER_IDENTITY:
1237         case SOURCE_STARTUP_CABLE_PLUG_DISCOVER_IDENTITY:
1238         case DISCOVER_SVIDS:
1239         case DISCOVER_MODES:
1240         case DFP_TO_UFP_ENTER_MODE:
1241         case DFP_TO_UFP_EXIT_MODE:
1242         case DFP_TO_CABLE_PLUG_ENTER_MODE:
1243         case DFP_TO_CABLE_PLUG_EXIT_MODE:
1244         case UNSTRUCTURED_VDMS:
1245         case STRUCTURED_VDMS:
1246         case COUNTRY_INFO:
1247         case COUNTRY_CODES:
1248                 break;
1249         /* Non-Interruptible AMS */
1250         default:
1251                 if (port->in_ams)
1252                         return false;
1253                 break;
1254         }
1255
1256         return true;
1257 }
1258
1259 static int tcpm_ams_start(struct tcpm_port *port, enum tcpm_ams ams)
1260 {
1261         int ret = 0;
1262
1263         tcpm_log(port, "AMS %s start", tcpm_ams_str[ams]);
1264
1265         if (!tcpm_ams_interruptible(port) &&
1266             !(ams == HARD_RESET || ams == SOFT_RESET_AMS)) {
1267                 port->upcoming_state = INVALID_STATE;
1268                 tcpm_log(port, "AMS %s not interruptible, aborting",
1269                          tcpm_ams_str[port->ams]);
1270                 return -EAGAIN;
1271         }
1272
1273         if (port->pwr_role == TYPEC_SOURCE) {
1274                 enum typec_cc_status cc_req = port->cc_req;
1275
1276                 port->ams = ams;
1277
1278                 if (ams == HARD_RESET) {
1279                         tcpm_set_cc(port, tcpm_rp_cc(port));
1280                         tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
1281                         tcpm_set_state(port, HARD_RESET_START, 0);
1282                         return ret;
1283                 } else if (ams == SOFT_RESET_AMS) {
1284                         if (!port->explicit_contract)
1285                                 tcpm_set_cc(port, tcpm_rp_cc(port));
1286                         tcpm_set_state(port, SOFT_RESET_SEND, 0);
1287                         return ret;
1288                 } else if (tcpm_vdm_ams(port)) {
1289                         /* tSinkTx is enforced in vdm_run_state_machine */
1290                         if (port->negotiated_rev >= PD_REV30)
1291                                 tcpm_set_cc(port, SINK_TX_NG);
1292                         return ret;
1293                 }
1294
1295                 if (port->negotiated_rev >= PD_REV30)
1296                         tcpm_set_cc(port, SINK_TX_NG);
1297
1298                 switch (port->state) {
1299                 case SRC_READY:
1300                 case SRC_STARTUP:
1301                 case SRC_SOFT_RESET_WAIT_SNK_TX:
1302                 case SOFT_RESET:
1303                 case SOFT_RESET_SEND:
1304                         if (port->negotiated_rev >= PD_REV30)
1305                                 tcpm_set_state(port, AMS_START,
1306                                                cc_req == SINK_TX_OK ?
1307                                                PD_T_SINK_TX : 0);
1308                         else
1309                                 tcpm_set_state(port, AMS_START, 0);
1310                         break;
1311                 default:
1312                         if (port->negotiated_rev >= PD_REV30)
1313                                 tcpm_set_state(port, SRC_READY,
1314                                                cc_req == SINK_TX_OK ?
1315                                                PD_T_SINK_TX : 0);
1316                         else
1317                                 tcpm_set_state(port, SRC_READY, 0);
1318                         break;
1319                 }
1320         } else {
1321                 if (port->negotiated_rev >= PD_REV30 &&
1322                     !tcpm_sink_tx_ok(port) &&
1323                     ams != SOFT_RESET_AMS &&
1324                     ams != HARD_RESET) {
1325                         port->upcoming_state = INVALID_STATE;
1326                         tcpm_log(port, "Sink TX No Go");
1327                         return -EAGAIN;
1328                 }
1329
1330                 port->ams = ams;
1331
1332                 if (ams == HARD_RESET) {
1333                         tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
1334                         tcpm_set_state(port, HARD_RESET_START, 0);
1335                         return ret;
1336                 } else if (tcpm_vdm_ams(port)) {
1337                         return ret;
1338                 }
1339
1340                 if (port->state == SNK_READY ||
1341                     port->state == SNK_SOFT_RESET)
1342                         tcpm_set_state(port, AMS_START, 0);
1343                 else
1344                         tcpm_set_state(port, SNK_READY, 0);
1345         }
1346
1347         return ret;
1348 }
1349
1350 /*
1351  * VDM/VDO handling functions
1352  */
1353 static void tcpm_queue_vdm(struct tcpm_port *port, const u32 header,
1354                            const u32 *data, int cnt)
1355 {
1356         WARN_ON(!mutex_is_locked(&port->lock));
1357
1358         /* Make sure we are not still processing a previous VDM packet */
1359         WARN_ON(port->vdm_state > VDM_STATE_DONE);
1360
1361         port->vdo_count = cnt + 1;
1362         port->vdo_data[0] = header;
1363         memcpy(&port->vdo_data[1], data, sizeof(u32) * cnt);
1364         /* Set ready, vdm state machine will actually send */
1365         port->vdm_retries = 0;
1366         port->vdm_state = VDM_STATE_READY;
1367
1368         mod_vdm_delayed_work(port, 0);
1369 }
1370
1371 static void tcpm_queue_vdm_unlocked(struct tcpm_port *port, const u32 header,
1372                                     const u32 *data, int cnt)
1373 {
1374         mutex_lock(&port->lock);
1375         tcpm_queue_vdm(port, header, data, cnt);
1376         mutex_unlock(&port->lock);
1377 }
1378
1379 static void svdm_consume_identity(struct tcpm_port *port, const u32 *p, int cnt)
1380 {
1381         u32 vdo = p[VDO_INDEX_IDH];
1382         u32 product = p[VDO_INDEX_PRODUCT];
1383
1384         memset(&port->mode_data, 0, sizeof(port->mode_data));
1385
1386         port->partner_ident.id_header = vdo;
1387         port->partner_ident.cert_stat = p[VDO_INDEX_CSTAT];
1388         port->partner_ident.product = product;
1389
1390         typec_partner_set_identity(port->partner);
1391
1392         tcpm_log(port, "Identity: %04x:%04x.%04x",
1393                  PD_IDH_VID(vdo),
1394                  PD_PRODUCT_PID(product), product & 0xffff);
1395 }
1396
1397 static bool svdm_consume_svids(struct tcpm_port *port, const u32 *p, int cnt)
1398 {
1399         struct pd_mode_data *pmdata = &port->mode_data;
1400         int i;
1401
1402         for (i = 1; i < cnt; i++) {
1403                 u16 svid;
1404
1405                 svid = (p[i] >> 16) & 0xffff;
1406                 if (!svid)
1407                         return false;
1408
1409                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
1410                         goto abort;
1411
1412                 pmdata->svids[pmdata->nsvids++] = svid;
1413                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
1414
1415                 svid = p[i] & 0xffff;
1416                 if (!svid)
1417                         return false;
1418
1419                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
1420                         goto abort;
1421
1422                 pmdata->svids[pmdata->nsvids++] = svid;
1423                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
1424         }
1425         return true;
1426 abort:
1427         tcpm_log(port, "SVID_DISCOVERY_MAX(%d) too low!", SVID_DISCOVERY_MAX);
1428         return false;
1429 }
1430
1431 static void svdm_consume_modes(struct tcpm_port *port, const u32 *p, int cnt)
1432 {
1433         struct pd_mode_data *pmdata = &port->mode_data;
1434         struct typec_altmode_desc *paltmode;
1435         int i;
1436
1437         if (pmdata->altmodes >= ARRAY_SIZE(port->partner_altmode)) {
1438                 /* Already logged in svdm_consume_svids() */
1439                 return;
1440         }
1441
1442         for (i = 1; i < cnt; i++) {
1443                 paltmode = &pmdata->altmode_desc[pmdata->altmodes];
1444                 memset(paltmode, 0, sizeof(*paltmode));
1445
1446                 paltmode->svid = pmdata->svids[pmdata->svid_index];
1447                 paltmode->mode = i;
1448                 paltmode->vdo = p[i];
1449
1450                 tcpm_log(port, " Alternate mode %d: SVID 0x%04x, VDO %d: 0x%08x",
1451                          pmdata->altmodes, paltmode->svid,
1452                          paltmode->mode, paltmode->vdo);
1453
1454                 pmdata->altmodes++;
1455         }
1456 }
1457
1458 static void tcpm_register_partner_altmodes(struct tcpm_port *port)
1459 {
1460         struct pd_mode_data *modep = &port->mode_data;
1461         struct typec_altmode *altmode;
1462         int i;
1463
1464         for (i = 0; i < modep->altmodes; i++) {
1465                 altmode = typec_partner_register_altmode(port->partner,
1466                                                 &modep->altmode_desc[i]);
1467                 if (IS_ERR(altmode)) {
1468                         tcpm_log(port, "Failed to register partner SVID 0x%04x",
1469                                  modep->altmode_desc[i].svid);
1470                         altmode = NULL;
1471                 }
1472                 port->partner_altmode[i] = altmode;
1473         }
1474 }
1475
1476 #define supports_modal(port)    PD_IDH_MODAL_SUPP((port)->partner_ident.id_header)
1477
1478 static int tcpm_pd_svdm(struct tcpm_port *port, struct typec_altmode *adev,
1479                         const u32 *p, int cnt, u32 *response,
1480                         enum adev_actions *adev_action)
1481 {
1482         struct typec_port *typec = port->typec_port;
1483         struct typec_altmode *pdev;
1484         struct pd_mode_data *modep;
1485         int svdm_version;
1486         int rlen = 0;
1487         int cmd_type;
1488         int cmd;
1489         int i;
1490
1491         cmd_type = PD_VDO_CMDT(p[0]);
1492         cmd = PD_VDO_CMD(p[0]);
1493
1494         tcpm_log(port, "Rx VDM cmd 0x%x type %d cmd %d len %d",
1495                  p[0], cmd_type, cmd, cnt);
1496
1497         modep = &port->mode_data;
1498
1499         pdev = typec_match_altmode(port->partner_altmode, ALTMODE_DISCOVERY_MAX,
1500                                    PD_VDO_VID(p[0]), PD_VDO_OPOS(p[0]));
1501
1502         svdm_version = typec_get_negotiated_svdm_version(typec);
1503         if (svdm_version < 0)
1504                 return 0;
1505
1506         switch (cmd_type) {
1507         case CMDT_INIT:
1508                 switch (cmd) {
1509                 case CMD_DISCOVER_IDENT:
1510                         if (PD_VDO_VID(p[0]) != USB_SID_PD)
1511                                 break;
1512
1513                         if (PD_VDO_SVDM_VER(p[0]) < svdm_version)
1514                                 typec_partner_set_svdm_version(port->partner,
1515                                                                PD_VDO_SVDM_VER(p[0]));
1516                         /* 6.4.4.3.1: Only respond as UFP (device) */
1517                         if (port->data_role == TYPEC_DEVICE &&
1518                             port->nr_snk_vdo) {
1519                                 /*
1520                                  * Product Type DFP and Connector Type are not defined in SVDM
1521                                  * version 1.0 and shall be set to zero.
1522                                  */
1523                                 if (typec_get_negotiated_svdm_version(typec) < SVDM_VER_2_0)
1524                                         response[1] = port->snk_vdo[0] & ~IDH_DFP_MASK
1525                                                       & ~IDH_CONN_MASK;
1526                                 else
1527                                         response[1] = port->snk_vdo[0];
1528                                 for (i = 1; i <  port->nr_snk_vdo; i++)
1529                                         response[i + 1] = port->snk_vdo[i];
1530                                 rlen = port->nr_snk_vdo + 1;
1531                         }
1532                         break;
1533                 case CMD_DISCOVER_SVID:
1534                         break;
1535                 case CMD_DISCOVER_MODES:
1536                         break;
1537                 case CMD_ENTER_MODE:
1538                         break;
1539                 case CMD_EXIT_MODE:
1540                         break;
1541                 case CMD_ATTENTION:
1542                         /* Attention command does not have response */
1543                         *adev_action = ADEV_ATTENTION;
1544                         return 0;
1545                 default:
1546                         break;
1547                 }
1548                 if (rlen >= 1) {
1549                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_ACK);
1550                 } else if (rlen == 0) {
1551                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_NAK);
1552                         rlen = 1;
1553                 } else {
1554                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_BUSY);
1555                         rlen = 1;
1556                 }
1557                 response[0] = (response[0] & ~VDO_SVDM_VERS_MASK) |
1558                               (VDO_SVDM_VERS(typec_get_negotiated_svdm_version(typec)));
1559                 break;
1560         case CMDT_RSP_ACK:
1561                 /* silently drop message if we are not connected */
1562                 if (IS_ERR_OR_NULL(port->partner))
1563                         break;
1564
1565                 tcpm_ams_finish(port);
1566
1567                 switch (cmd) {
1568                 case CMD_DISCOVER_IDENT:
1569                         if (PD_VDO_SVDM_VER(p[0]) < svdm_version)
1570                                 typec_partner_set_svdm_version(port->partner,
1571                                                                PD_VDO_SVDM_VER(p[0]));
1572                         /* 6.4.4.3.1 */
1573                         svdm_consume_identity(port, p, cnt);
1574                         response[0] = VDO(USB_SID_PD, 1, typec_get_negotiated_svdm_version(typec),
1575                                           CMD_DISCOVER_SVID);
1576                         rlen = 1;
1577                         break;
1578                 case CMD_DISCOVER_SVID:
1579                         /* 6.4.4.3.2 */
1580                         if (svdm_consume_svids(port, p, cnt)) {
1581                                 response[0] = VDO(USB_SID_PD, 1, svdm_version, CMD_DISCOVER_SVID);
1582                                 rlen = 1;
1583                         } else if (modep->nsvids && supports_modal(port)) {
1584                                 response[0] = VDO(modep->svids[0], 1, svdm_version,
1585                                                   CMD_DISCOVER_MODES);
1586                                 rlen = 1;
1587                         }
1588                         break;
1589                 case CMD_DISCOVER_MODES:
1590                         /* 6.4.4.3.3 */
1591                         svdm_consume_modes(port, p, cnt);
1592                         modep->svid_index++;
1593                         if (modep->svid_index < modep->nsvids) {
1594                                 u16 svid = modep->svids[modep->svid_index];
1595                                 response[0] = VDO(svid, 1, svdm_version, CMD_DISCOVER_MODES);
1596                                 rlen = 1;
1597                         } else {
1598                                 tcpm_register_partner_altmodes(port);
1599                                 port->vdm_sm_running = false;
1600                         }
1601                         break;
1602                 case CMD_ENTER_MODE:
1603                         if (adev && pdev) {
1604                                 typec_altmode_update_active(pdev, true);
1605                                 *adev_action = ADEV_QUEUE_VDM_SEND_EXIT_MODE_ON_FAIL;
1606                         }
1607                         return 0;
1608                 case CMD_EXIT_MODE:
1609                         if (adev && pdev) {
1610                                 typec_altmode_update_active(pdev, false);
1611                                 /* Back to USB Operation */
1612                                 *adev_action = ADEV_NOTIFY_USB_AND_QUEUE_VDM;
1613                                 return 0;
1614                         }
1615                         break;
1616                 case VDO_CMD_VENDOR(0) ... VDO_CMD_VENDOR(15):
1617                         break;
1618                 default:
1619                         /* Unrecognized SVDM */
1620                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_NAK);
1621                         rlen = 1;
1622                         response[0] = (response[0] & ~VDO_SVDM_VERS_MASK) |
1623                                       (VDO_SVDM_VERS(svdm_version));
1624                         break;
1625                 }
1626                 break;
1627         case CMDT_RSP_NAK:
1628                 tcpm_ams_finish(port);
1629                 switch (cmd) {
1630                 case CMD_DISCOVER_IDENT:
1631                 case CMD_DISCOVER_SVID:
1632                 case CMD_DISCOVER_MODES:
1633                 case VDO_CMD_VENDOR(0) ... VDO_CMD_VENDOR(15):
1634                         break;
1635                 case CMD_ENTER_MODE:
1636                         /* Back to USB Operation */
1637                         *adev_action = ADEV_NOTIFY_USB_AND_QUEUE_VDM;
1638                         return 0;
1639                 default:
1640                         /* Unrecognized SVDM */
1641                         response[0] = p[0] | VDO_CMDT(CMDT_RSP_NAK);
1642                         rlen = 1;
1643                         response[0] = (response[0] & ~VDO_SVDM_VERS_MASK) |
1644                                       (VDO_SVDM_VERS(svdm_version));
1645                         break;
1646                 }
1647                 port->vdm_sm_running = false;
1648                 break;
1649         default:
1650                 response[0] = p[0] | VDO_CMDT(CMDT_RSP_NAK);
1651                 rlen = 1;
1652                 response[0] = (response[0] & ~VDO_SVDM_VERS_MASK) |
1653                               (VDO_SVDM_VERS(svdm_version));
1654                 port->vdm_sm_running = false;
1655                 break;
1656         }
1657
1658         /* Informing the alternate mode drivers about everything */
1659         *adev_action = ADEV_QUEUE_VDM;
1660         return rlen;
1661 }
1662
1663 static void tcpm_handle_vdm_request(struct tcpm_port *port,
1664                                     const __le32 *payload, int cnt)
1665 {
1666         enum adev_actions adev_action = ADEV_NONE;
1667         struct typec_altmode *adev;
1668         u32 p[PD_MAX_PAYLOAD];
1669         u32 response[8] = { };
1670         int i, rlen = 0;
1671
1672         for (i = 0; i < cnt; i++)
1673                 p[i] = le32_to_cpu(payload[i]);
1674
1675         adev = typec_match_altmode(port->port_altmode, ALTMODE_DISCOVERY_MAX,
1676                                    PD_VDO_VID(p[0]), PD_VDO_OPOS(p[0]));
1677
1678         if (port->vdm_state == VDM_STATE_BUSY) {
1679                 /* If UFP responded busy retry after timeout */
1680                 if (PD_VDO_CMDT(p[0]) == CMDT_RSP_BUSY) {
1681                         port->vdm_state = VDM_STATE_WAIT_RSP_BUSY;
1682                         port->vdo_retry = (p[0] & ~VDO_CMDT_MASK) |
1683                                 CMDT_INIT;
1684                         mod_vdm_delayed_work(port, PD_T_VDM_BUSY);
1685                         return;
1686                 }
1687                 port->vdm_state = VDM_STATE_DONE;
1688         }
1689
1690         if (PD_VDO_SVDM(p[0])) {
1691                 rlen = tcpm_pd_svdm(port, adev, p, cnt, response, &adev_action);
1692         } else {
1693                 if (port->negotiated_rev >= PD_REV30)
1694                         tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1695         }
1696
1697         /*
1698          * We are done with any state stored in the port struct now, except
1699          * for any port struct changes done by the tcpm_queue_vdm() call
1700          * below, which is a separate operation.
1701          *
1702          * So we can safely release the lock here; and we MUST release the
1703          * lock here to avoid an AB BA lock inversion:
1704          *
1705          * If we keep the lock here then the lock ordering in this path is:
1706          * 1. tcpm_pd_rx_handler take the tcpm port lock
1707          * 2. One of the typec_altmode_* calls below takes the alt-mode's lock
1708          *
1709          * And we also have this ordering:
1710          * 1. alt-mode driver takes the alt-mode's lock
1711          * 2. alt-mode driver calls tcpm_altmode_enter which takes the
1712          *    tcpm port lock
1713          *
1714          * Dropping our lock here avoids this.
1715          */
1716         mutex_unlock(&port->lock);
1717
1718         if (adev) {
1719                 switch (adev_action) {
1720                 case ADEV_NONE:
1721                         break;
1722                 case ADEV_NOTIFY_USB_AND_QUEUE_VDM:
1723                         WARN_ON(typec_altmode_notify(adev, TYPEC_STATE_USB, NULL));
1724                         typec_altmode_vdm(adev, p[0], &p[1], cnt);
1725                         break;
1726                 case ADEV_QUEUE_VDM:
1727                         typec_altmode_vdm(adev, p[0], &p[1], cnt);
1728                         break;
1729                 case ADEV_QUEUE_VDM_SEND_EXIT_MODE_ON_FAIL:
1730                         if (typec_altmode_vdm(adev, p[0], &p[1], cnt)) {
1731                                 int svdm_version = typec_get_negotiated_svdm_version(
1732                                                                         port->typec_port);
1733                                 if (svdm_version < 0)
1734                                         break;
1735
1736                                 response[0] = VDO(adev->svid, 1, svdm_version,
1737                                                   CMD_EXIT_MODE);
1738                                 response[0] |= VDO_OPOS(adev->mode);
1739                                 rlen = 1;
1740                         }
1741                         break;
1742                 case ADEV_ATTENTION:
1743                         typec_altmode_attention(adev, p[1]);
1744                         break;
1745                 }
1746         }
1747
1748         /*
1749          * We must re-take the lock here to balance the unlock in
1750          * tcpm_pd_rx_handler, note that no changes, other then the
1751          * tcpm_queue_vdm call, are made while the lock is held again.
1752          * All that is done after the call is unwinding the call stack until
1753          * we return to tcpm_pd_rx_handler and do the unlock there.
1754          */
1755         mutex_lock(&port->lock);
1756
1757         if (rlen > 0)
1758                 tcpm_queue_vdm(port, response[0], &response[1], rlen - 1);
1759 }
1760
1761 static void tcpm_send_vdm(struct tcpm_port *port, u32 vid, int cmd,
1762                           const u32 *data, int count)
1763 {
1764         int svdm_version = typec_get_negotiated_svdm_version(port->typec_port);
1765         u32 header;
1766
1767         if (svdm_version < 0)
1768                 return;
1769
1770         if (WARN_ON(count > VDO_MAX_SIZE - 1))
1771                 count = VDO_MAX_SIZE - 1;
1772
1773         /* set VDM header with VID & CMD */
1774         header = VDO(vid, ((vid & USB_SID_PD) == USB_SID_PD) ?
1775                         1 : (PD_VDO_CMD(cmd) <= CMD_ATTENTION),
1776                         svdm_version, cmd);
1777         tcpm_queue_vdm(port, header, data, count);
1778 }
1779
1780 static unsigned int vdm_ready_timeout(u32 vdm_hdr)
1781 {
1782         unsigned int timeout;
1783         int cmd = PD_VDO_CMD(vdm_hdr);
1784
1785         /* its not a structured VDM command */
1786         if (!PD_VDO_SVDM(vdm_hdr))
1787                 return PD_T_VDM_UNSTRUCTURED;
1788
1789         switch (PD_VDO_CMDT(vdm_hdr)) {
1790         case CMDT_INIT:
1791                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1792                         timeout = PD_T_VDM_WAIT_MODE_E;
1793                 else
1794                         timeout = PD_T_VDM_SNDR_RSP;
1795                 break;
1796         default:
1797                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1798                         timeout = PD_T_VDM_E_MODE;
1799                 else
1800                         timeout = PD_T_VDM_RCVR_RSP;
1801                 break;
1802         }
1803         return timeout;
1804 }
1805
1806 static void vdm_run_state_machine(struct tcpm_port *port)
1807 {
1808         struct pd_message msg;
1809         int i, res = 0;
1810         u32 vdo_hdr = port->vdo_data[0];
1811
1812         switch (port->vdm_state) {
1813         case VDM_STATE_READY:
1814                 /* Only transmit VDM if attached */
1815                 if (!port->attached) {
1816                         port->vdm_state = VDM_STATE_ERR_BUSY;
1817                         break;
1818                 }
1819
1820                 /*
1821                  * if there's traffic or we're not in PDO ready state don't send
1822                  * a VDM.
1823                  */
1824                 if (port->state != SRC_READY && port->state != SNK_READY)
1825                         break;
1826
1827                 /* TODO: AMS operation for Unstructured VDM */
1828                 if (PD_VDO_SVDM(vdo_hdr) && PD_VDO_CMDT(vdo_hdr) == CMDT_INIT) {
1829                         switch (PD_VDO_CMD(vdo_hdr)) {
1830                         case CMD_DISCOVER_IDENT:
1831                                 res = tcpm_ams_start(port, DISCOVER_IDENTITY);
1832                                 if (res == 0)
1833                                         port->send_discover = false;
1834                                 break;
1835                         case CMD_DISCOVER_SVID:
1836                                 res = tcpm_ams_start(port, DISCOVER_SVIDS);
1837                                 break;
1838                         case CMD_DISCOVER_MODES:
1839                                 res = tcpm_ams_start(port, DISCOVER_MODES);
1840                                 break;
1841                         case CMD_ENTER_MODE:
1842                                 res = tcpm_ams_start(port, DFP_TO_UFP_ENTER_MODE);
1843                                 break;
1844                         case CMD_EXIT_MODE:
1845                                 res = tcpm_ams_start(port, DFP_TO_UFP_EXIT_MODE);
1846                                 break;
1847                         case CMD_ATTENTION:
1848                                 res = tcpm_ams_start(port, ATTENTION);
1849                                 break;
1850                         case VDO_CMD_VENDOR(0) ... VDO_CMD_VENDOR(15):
1851                                 res = tcpm_ams_start(port, STRUCTURED_VDMS);
1852                                 break;
1853                         default:
1854                                 res = -EOPNOTSUPP;
1855                                 break;
1856                         }
1857
1858                         if (res < 0) {
1859                                 port->vdm_sm_running = false;
1860                                 return;
1861                         }
1862                 }
1863
1864                 port->vdm_state = VDM_STATE_SEND_MESSAGE;
1865                 mod_vdm_delayed_work(port, (port->negotiated_rev >= PD_REV30 &&
1866                                             port->pwr_role == TYPEC_SOURCE &&
1867                                             PD_VDO_SVDM(vdo_hdr) &&
1868                                             PD_VDO_CMDT(vdo_hdr) == CMDT_INIT) ?
1869                                            PD_T_SINK_TX : 0);
1870                 break;
1871         case VDM_STATE_WAIT_RSP_BUSY:
1872                 port->vdo_data[0] = port->vdo_retry;
1873                 port->vdo_count = 1;
1874                 port->vdm_state = VDM_STATE_READY;
1875                 break;
1876         case VDM_STATE_BUSY:
1877                 port->vdm_state = VDM_STATE_ERR_TMOUT;
1878                 if (port->ams != NONE_AMS)
1879                         tcpm_ams_finish(port);
1880                 break;
1881         case VDM_STATE_ERR_SEND:
1882                 /*
1883                  * A partner which does not support USB PD will not reply,
1884                  * so this is not a fatal error. At the same time, some
1885                  * devices may not return GoodCRC under some circumstances,
1886                  * so we need to retry.
1887                  */
1888                 if (port->vdm_retries < 3) {
1889                         tcpm_log(port, "VDM Tx error, retry");
1890                         port->vdm_retries++;
1891                         port->vdm_state = VDM_STATE_READY;
1892                         tcpm_ams_finish(port);
1893                 }
1894                 break;
1895         case VDM_STATE_SEND_MESSAGE:
1896                 /* Prepare and send VDM */
1897                 memset(&msg, 0, sizeof(msg));
1898                 msg.header = PD_HEADER_LE(PD_DATA_VENDOR_DEF,
1899                                           port->pwr_role,
1900                                           port->data_role,
1901                                           port->negotiated_rev,
1902                                           port->message_id, port->vdo_count);
1903                 for (i = 0; i < port->vdo_count; i++)
1904                         msg.payload[i] = cpu_to_le32(port->vdo_data[i]);
1905                 res = tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1906                 if (res < 0) {
1907                         port->vdm_state = VDM_STATE_ERR_SEND;
1908                 } else {
1909                         unsigned long timeout;
1910
1911                         port->vdm_retries = 0;
1912                         port->vdm_state = VDM_STATE_BUSY;
1913                         timeout = vdm_ready_timeout(vdo_hdr);
1914                         mod_vdm_delayed_work(port, timeout);
1915                 }
1916                 break;
1917         default:
1918                 break;
1919         }
1920 }
1921
1922 static void vdm_state_machine_work(struct kthread_work *work)
1923 {
1924         struct tcpm_port *port = container_of(work, struct tcpm_port, vdm_state_machine);
1925         enum vdm_states prev_state;
1926
1927         mutex_lock(&port->lock);
1928
1929         /*
1930          * Continue running as long as the port is not busy and there was
1931          * a state change.
1932          */
1933         do {
1934                 prev_state = port->vdm_state;
1935                 vdm_run_state_machine(port);
1936         } while (port->vdm_state != prev_state &&
1937                  port->vdm_state != VDM_STATE_BUSY &&
1938                  port->vdm_state != VDM_STATE_SEND_MESSAGE);
1939
1940         if (port->vdm_state == VDM_STATE_ERR_TMOUT)
1941                 port->vdm_sm_running = false;
1942
1943         mutex_unlock(&port->lock);
1944 }
1945
1946 enum pdo_err {
1947         PDO_NO_ERR,
1948         PDO_ERR_NO_VSAFE5V,
1949         PDO_ERR_VSAFE5V_NOT_FIRST,
1950         PDO_ERR_PDO_TYPE_NOT_IN_ORDER,
1951         PDO_ERR_FIXED_NOT_SORTED,
1952         PDO_ERR_VARIABLE_BATT_NOT_SORTED,
1953         PDO_ERR_DUPE_PDO,
1954         PDO_ERR_PPS_APDO_NOT_SORTED,
1955         PDO_ERR_DUPE_PPS_APDO,
1956 };
1957
1958 static const char * const pdo_err_msg[] = {
1959         [PDO_ERR_NO_VSAFE5V] =
1960         " err: source/sink caps should atleast have vSafe5V",
1961         [PDO_ERR_VSAFE5V_NOT_FIRST] =
1962         " err: vSafe5V Fixed Supply Object Shall always be the first object",
1963         [PDO_ERR_PDO_TYPE_NOT_IN_ORDER] =
1964         " err: PDOs should be in the following order: Fixed; Battery; Variable",
1965         [PDO_ERR_FIXED_NOT_SORTED] =
1966         " err: Fixed supply pdos should be in increasing order of their fixed voltage",
1967         [PDO_ERR_VARIABLE_BATT_NOT_SORTED] =
1968         " err: Variable/Battery supply pdos should be in increasing order of their minimum voltage",
1969         [PDO_ERR_DUPE_PDO] =
1970         " err: Variable/Batt supply pdos cannot have same min/max voltage",
1971         [PDO_ERR_PPS_APDO_NOT_SORTED] =
1972         " err: Programmable power supply apdos should be in increasing order of their maximum voltage",
1973         [PDO_ERR_DUPE_PPS_APDO] =
1974         " err: Programmable power supply apdos cannot have same min/max voltage and max current",
1975 };
1976
1977 static enum pdo_err tcpm_caps_err(struct tcpm_port *port, const u32 *pdo,
1978                                   unsigned int nr_pdo)
1979 {
1980         unsigned int i;
1981
1982         /* Should at least contain vSafe5v */
1983         if (nr_pdo < 1)
1984                 return PDO_ERR_NO_VSAFE5V;
1985
1986         /* The vSafe5V Fixed Supply Object Shall always be the first object */
1987         if (pdo_type(pdo[0]) != PDO_TYPE_FIXED ||
1988             pdo_fixed_voltage(pdo[0]) != VSAFE5V)
1989                 return PDO_ERR_VSAFE5V_NOT_FIRST;
1990
1991         for (i = 1; i < nr_pdo; i++) {
1992                 if (pdo_type(pdo[i]) < pdo_type(pdo[i - 1])) {
1993                         return PDO_ERR_PDO_TYPE_NOT_IN_ORDER;
1994                 } else if (pdo_type(pdo[i]) == pdo_type(pdo[i - 1])) {
1995                         enum pd_pdo_type type = pdo_type(pdo[i]);
1996
1997                         switch (type) {
1998                         /*
1999                          * The remaining Fixed Supply Objects, if
2000                          * present, shall be sent in voltage order;
2001                          * lowest to highest.
2002                          */
2003                         case PDO_TYPE_FIXED:
2004                                 if (pdo_fixed_voltage(pdo[i]) <=
2005                                     pdo_fixed_voltage(pdo[i - 1]))
2006                                         return PDO_ERR_FIXED_NOT_SORTED;
2007                                 break;
2008                         /*
2009                          * The Battery Supply Objects and Variable
2010                          * supply, if present shall be sent in Minimum
2011                          * Voltage order; lowest to highest.
2012                          */
2013                         case PDO_TYPE_VAR:
2014                         case PDO_TYPE_BATT:
2015                                 if (pdo_min_voltage(pdo[i]) <
2016                                     pdo_min_voltage(pdo[i - 1]))
2017                                         return PDO_ERR_VARIABLE_BATT_NOT_SORTED;
2018                                 else if ((pdo_min_voltage(pdo[i]) ==
2019                                           pdo_min_voltage(pdo[i - 1])) &&
2020                                          (pdo_max_voltage(pdo[i]) ==
2021                                           pdo_max_voltage(pdo[i - 1])))
2022                                         return PDO_ERR_DUPE_PDO;
2023                                 break;
2024                         /*
2025                          * The Programmable Power Supply APDOs, if present,
2026                          * shall be sent in Maximum Voltage order;
2027                          * lowest to highest.
2028                          */
2029                         case PDO_TYPE_APDO:
2030                                 if (pdo_apdo_type(pdo[i]) != APDO_TYPE_PPS)
2031                                         break;
2032
2033                                 if (pdo_pps_apdo_max_voltage(pdo[i]) <
2034                                     pdo_pps_apdo_max_voltage(pdo[i - 1]))
2035                                         return PDO_ERR_PPS_APDO_NOT_SORTED;
2036                                 else if (pdo_pps_apdo_min_voltage(pdo[i]) ==
2037                                           pdo_pps_apdo_min_voltage(pdo[i - 1]) &&
2038                                          pdo_pps_apdo_max_voltage(pdo[i]) ==
2039                                           pdo_pps_apdo_max_voltage(pdo[i - 1]) &&
2040                                          pdo_pps_apdo_max_current(pdo[i]) ==
2041                                           pdo_pps_apdo_max_current(pdo[i - 1]))
2042                                         return PDO_ERR_DUPE_PPS_APDO;
2043                                 break;
2044                         default:
2045                                 tcpm_log_force(port, " Unknown pdo type");
2046                         }
2047                 }
2048         }
2049
2050         return PDO_NO_ERR;
2051 }
2052
2053 static int tcpm_validate_caps(struct tcpm_port *port, const u32 *pdo,
2054                               unsigned int nr_pdo)
2055 {
2056         enum pdo_err err_index = tcpm_caps_err(port, pdo, nr_pdo);
2057
2058         if (err_index != PDO_NO_ERR) {
2059                 tcpm_log_force(port, " %s", pdo_err_msg[err_index]);
2060                 return -EINVAL;
2061         }
2062
2063         return 0;
2064 }
2065
2066 static int tcpm_altmode_enter(struct typec_altmode *altmode, u32 *vdo)
2067 {
2068         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
2069         int svdm_version;
2070         u32 header;
2071
2072         svdm_version = typec_get_negotiated_svdm_version(port->typec_port);
2073         if (svdm_version < 0)
2074                 return svdm_version;
2075
2076         header = VDO(altmode->svid, vdo ? 2 : 1, svdm_version, CMD_ENTER_MODE);
2077         header |= VDO_OPOS(altmode->mode);
2078
2079         tcpm_queue_vdm_unlocked(port, header, vdo, vdo ? 1 : 0);
2080         return 0;
2081 }
2082
2083 static int tcpm_altmode_exit(struct typec_altmode *altmode)
2084 {
2085         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
2086         int svdm_version;
2087         u32 header;
2088
2089         svdm_version = typec_get_negotiated_svdm_version(port->typec_port);
2090         if (svdm_version < 0)
2091                 return svdm_version;
2092
2093         header = VDO(altmode->svid, 1, svdm_version, CMD_EXIT_MODE);
2094         header |= VDO_OPOS(altmode->mode);
2095
2096         tcpm_queue_vdm_unlocked(port, header, NULL, 0);
2097         return 0;
2098 }
2099
2100 static int tcpm_altmode_vdm(struct typec_altmode *altmode,
2101                             u32 header, const u32 *data, int count)
2102 {
2103         struct tcpm_port *port = typec_altmode_get_drvdata(altmode);
2104
2105         tcpm_queue_vdm_unlocked(port, header, data, count - 1);
2106
2107         return 0;
2108 }
2109
2110 static const struct typec_altmode_ops tcpm_altmode_ops = {
2111         .enter = tcpm_altmode_enter,
2112         .exit = tcpm_altmode_exit,
2113         .vdm = tcpm_altmode_vdm,
2114 };
2115
2116 /*
2117  * PD (data, control) command handling functions
2118  */
2119 static inline enum tcpm_state ready_state(struct tcpm_port *port)
2120 {
2121         if (port->pwr_role == TYPEC_SOURCE)
2122                 return SRC_READY;
2123         else
2124                 return SNK_READY;
2125 }
2126
2127 static int tcpm_pd_send_control(struct tcpm_port *port,
2128                                 enum pd_ctrl_msg_type type);
2129
2130 static void tcpm_handle_alert(struct tcpm_port *port, const __le32 *payload,
2131                               int cnt)
2132 {
2133         u32 p0 = le32_to_cpu(payload[0]);
2134         unsigned int type = usb_pd_ado_type(p0);
2135
2136         if (!type) {
2137                 tcpm_log(port, "Alert message received with no type");
2138                 return;
2139         }
2140
2141         /* Just handling non-battery alerts for now */
2142         if (!(type & USB_PD_ADO_TYPE_BATT_STATUS_CHANGE)) {
2143                 switch (port->state) {
2144                 case SRC_READY:
2145                 case SNK_READY:
2146                         tcpm_set_state(port, GET_STATUS_SEND, 0);
2147                         break;
2148                 default:
2149                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
2150                         break;
2151                 }
2152         }
2153 }
2154
2155 static int tcpm_set_auto_vbus_discharge_threshold(struct tcpm_port *port,
2156                                                   enum typec_pwr_opmode mode, bool pps_active,
2157                                                   u32 requested_vbus_voltage)
2158 {
2159         int ret;
2160
2161         if (!port->tcpc->set_auto_vbus_discharge_threshold)
2162                 return 0;
2163
2164         ret = port->tcpc->set_auto_vbus_discharge_threshold(port->tcpc, mode, pps_active,
2165                                                             requested_vbus_voltage);
2166         tcpm_log_force(port,
2167                        "set_auto_vbus_discharge_threshold mode:%d pps_active:%c vbus:%u ret:%d",
2168                        mode, pps_active ? 'y' : 'n', requested_vbus_voltage, ret);
2169
2170         return ret;
2171 }
2172
2173 static void tcpm_pd_handle_state(struct tcpm_port *port,
2174                                  enum tcpm_state state,
2175                                  enum tcpm_ams ams,
2176                                  unsigned int delay_ms)
2177 {
2178         switch (port->state) {
2179         case SRC_READY:
2180         case SNK_READY:
2181                 port->ams = ams;
2182                 tcpm_set_state(port, state, delay_ms);
2183                 break;
2184         /* 8.3.3.4.1.1 and 6.8.1 power transitioning */
2185         case SNK_TRANSITION_SINK:
2186         case SNK_TRANSITION_SINK_VBUS:
2187         case SRC_TRANSITION_SUPPLY:
2188                 tcpm_set_state(port, HARD_RESET_SEND, 0);
2189                 break;
2190         default:
2191                 if (!tcpm_ams_interruptible(port)) {
2192                         tcpm_set_state(port, port->pwr_role == TYPEC_SOURCE ?
2193                                        SRC_SOFT_RESET_WAIT_SNK_TX :
2194                                        SNK_SOFT_RESET,
2195                                        0);
2196                 } else {
2197                         /* process the Message 6.8.1 */
2198                         port->upcoming_state = state;
2199                         port->next_ams = ams;
2200                         tcpm_set_state(port, ready_state(port), delay_ms);
2201                 }
2202                 break;
2203         }
2204 }
2205
2206 static void tcpm_pd_handle_msg(struct tcpm_port *port,
2207                                enum pd_msg_request message,
2208                                enum tcpm_ams ams)
2209 {
2210         switch (port->state) {
2211         case SRC_READY:
2212         case SNK_READY:
2213                 port->ams = ams;
2214                 tcpm_queue_message(port, message);
2215                 break;
2216         /* PD 3.0 Spec 8.3.3.4.1.1 and 6.8.1 */
2217         case SNK_TRANSITION_SINK:
2218         case SNK_TRANSITION_SINK_VBUS:
2219         case SRC_TRANSITION_SUPPLY:
2220                 tcpm_set_state(port, HARD_RESET_SEND, 0);
2221                 break;
2222         default:
2223                 if (!tcpm_ams_interruptible(port)) {
2224                         tcpm_set_state(port, port->pwr_role == TYPEC_SOURCE ?
2225                                        SRC_SOFT_RESET_WAIT_SNK_TX :
2226                                        SNK_SOFT_RESET,
2227                                        0);
2228                 } else {
2229                         port->next_ams = ams;
2230                         tcpm_set_state(port, ready_state(port), 0);
2231                         /* 6.8.1 process the Message */
2232                         tcpm_queue_message(port, message);
2233                 }
2234                 break;
2235         }
2236 }
2237
2238 static void tcpm_pd_data_request(struct tcpm_port *port,
2239                                  const struct pd_message *msg)
2240 {
2241         enum pd_data_msg_type type = pd_header_type_le(msg->header);
2242         unsigned int cnt = pd_header_cnt_le(msg->header);
2243         unsigned int rev = pd_header_rev_le(msg->header);
2244         unsigned int i;
2245         enum frs_typec_current partner_frs_current;
2246         bool frs_enable;
2247         int ret;
2248
2249         switch (type) {
2250         case PD_DATA_SOURCE_CAP:
2251                 for (i = 0; i < cnt; i++)
2252                         port->source_caps[i] = le32_to_cpu(msg->payload[i]);
2253
2254                 port->nr_source_caps = cnt;
2255
2256                 tcpm_log_source_caps(port);
2257
2258                 tcpm_validate_caps(port, port->source_caps,
2259                                    port->nr_source_caps);
2260
2261                 /*
2262                  * Adjust revision in subsequent message headers, as required,
2263                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
2264                  * support Rev 1.0 so just do nothing in that scenario.
2265                  */
2266                 if (rev == PD_REV10) {
2267                         if (port->ams == GET_SOURCE_CAPABILITIES)
2268                                 tcpm_ams_finish(port);
2269                         break;
2270                 }
2271
2272                 if (rev < PD_MAX_REV)
2273                         port->negotiated_rev = rev;
2274
2275                 if (port->pwr_role == TYPEC_SOURCE) {
2276                         if (port->ams == GET_SOURCE_CAPABILITIES)
2277                                 tcpm_pd_handle_state(port, SRC_READY, NONE_AMS, 0);
2278                         /* Unexpected Source Capabilities */
2279                         else
2280                                 tcpm_pd_handle_msg(port,
2281                                                    port->negotiated_rev < PD_REV30 ?
2282                                                    PD_MSG_CTRL_REJECT :
2283                                                    PD_MSG_CTRL_NOT_SUPP,
2284                                                    NONE_AMS);
2285                 } else if (port->state == SNK_WAIT_CAPABILITIES) {
2286                 /*
2287                  * This message may be received even if VBUS is not
2288                  * present. This is quite unexpected; see USB PD
2289                  * specification, sections 8.3.3.6.3.1 and 8.3.3.6.3.2.
2290                  * However, at the same time, we must be ready to
2291                  * receive this message and respond to it 15ms after
2292                  * receiving PS_RDY during power swap operations, no matter
2293                  * if VBUS is available or not (USB PD specification,
2294                  * section 6.5.9.2).
2295                  * So we need to accept the message either way,
2296                  * but be prepared to keep waiting for VBUS after it was
2297                  * handled.
2298                  */
2299                         port->ams = POWER_NEGOTIATION;
2300                         port->in_ams = true;
2301                         tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
2302                 } else {
2303                         if (port->ams == GET_SOURCE_CAPABILITIES)
2304                                 tcpm_ams_finish(port);
2305                         tcpm_pd_handle_state(port, SNK_NEGOTIATE_CAPABILITIES,
2306                                              POWER_NEGOTIATION, 0);
2307                 }
2308                 break;
2309         case PD_DATA_REQUEST:
2310                 /*
2311                  * Adjust revision in subsequent message headers, as required,
2312                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
2313                  * support Rev 1.0 so just reject in that scenario.
2314                  */
2315                 if (rev == PD_REV10) {
2316                         tcpm_pd_handle_msg(port,
2317                                            port->negotiated_rev < PD_REV30 ?
2318                                            PD_MSG_CTRL_REJECT :
2319                                            PD_MSG_CTRL_NOT_SUPP,
2320                                            NONE_AMS);
2321                         break;
2322                 }
2323
2324                 if (rev < PD_MAX_REV)
2325                         port->negotiated_rev = rev;
2326
2327                 if (port->pwr_role != TYPEC_SOURCE || cnt != 1) {
2328                         tcpm_pd_handle_msg(port,
2329                                            port->negotiated_rev < PD_REV30 ?
2330                                            PD_MSG_CTRL_REJECT :
2331                                            PD_MSG_CTRL_NOT_SUPP,
2332                                            NONE_AMS);
2333                         break;
2334                 }
2335
2336                 port->sink_request = le32_to_cpu(msg->payload[0]);
2337
2338                 if (port->vdm_sm_running && port->explicit_contract) {
2339                         tcpm_pd_handle_msg(port, PD_MSG_CTRL_WAIT, port->ams);
2340                         break;
2341                 }
2342
2343                 if (port->state == SRC_SEND_CAPABILITIES)
2344                         tcpm_set_state(port, SRC_NEGOTIATE_CAPABILITIES, 0);
2345                 else
2346                         tcpm_pd_handle_state(port, SRC_NEGOTIATE_CAPABILITIES,
2347                                              POWER_NEGOTIATION, 0);
2348                 break;
2349         case PD_DATA_SINK_CAP:
2350                 /* We don't do anything with this at the moment... */
2351                 for (i = 0; i < cnt; i++)
2352                         port->sink_caps[i] = le32_to_cpu(msg->payload[i]);
2353
2354                 partner_frs_current = (port->sink_caps[0] & PDO_FIXED_FRS_CURR_MASK) >>
2355                         PDO_FIXED_FRS_CURR_SHIFT;
2356                 frs_enable = partner_frs_current && (partner_frs_current <=
2357                                                      port->new_source_frs_current);
2358                 tcpm_log(port,
2359                          "Port partner FRS capable partner_frs_current:%u port_frs_current:%u enable:%c",
2360                          partner_frs_current, port->new_source_frs_current, frs_enable ? 'y' : 'n');
2361                 if (frs_enable) {
2362                         ret  = port->tcpc->enable_frs(port->tcpc, true);
2363                         tcpm_log(port, "Enable FRS %s, ret:%d\n", ret ? "fail" : "success", ret);
2364                 }
2365
2366                 port->nr_sink_caps = cnt;
2367                 port->sink_cap_done = true;
2368                 if (port->ams == GET_SINK_CAPABILITIES)
2369                         tcpm_pd_handle_state(port, ready_state(port), NONE_AMS, 0);
2370                 /* Unexpected Sink Capabilities */
2371                 else
2372                         tcpm_pd_handle_msg(port,
2373                                            port->negotiated_rev < PD_REV30 ?
2374                                            PD_MSG_CTRL_REJECT :
2375                                            PD_MSG_CTRL_NOT_SUPP,
2376                                            NONE_AMS);
2377                 break;
2378         case PD_DATA_VENDOR_DEF:
2379                 tcpm_handle_vdm_request(port, msg->payload, cnt);
2380                 break;
2381         case PD_DATA_BIST:
2382                 port->bist_request = le32_to_cpu(msg->payload[0]);
2383                 tcpm_pd_handle_state(port, BIST_RX, BIST, 0);
2384                 break;
2385         case PD_DATA_ALERT:
2386                 tcpm_handle_alert(port, msg->payload, cnt);
2387                 break;
2388         case PD_DATA_BATT_STATUS:
2389         case PD_DATA_GET_COUNTRY_INFO:
2390                 /* Currently unsupported */
2391                 tcpm_pd_handle_msg(port, port->negotiated_rev < PD_REV30 ?
2392                                    PD_MSG_CTRL_REJECT :
2393                                    PD_MSG_CTRL_NOT_SUPP,
2394                                    NONE_AMS);
2395                 break;
2396         default:
2397                 tcpm_pd_handle_msg(port, port->negotiated_rev < PD_REV30 ?
2398                                    PD_MSG_CTRL_REJECT :
2399                                    PD_MSG_CTRL_NOT_SUPP,
2400                                    NONE_AMS);
2401                 tcpm_log(port, "Unrecognized data message type %#x", type);
2402                 break;
2403         }
2404 }
2405
2406 static void tcpm_pps_complete(struct tcpm_port *port, int result)
2407 {
2408         if (port->pps_pending) {
2409                 port->pps_status = result;
2410                 port->pps_pending = false;
2411                 complete(&port->pps_complete);
2412         }
2413 }
2414
2415 static void tcpm_pd_ctrl_request(struct tcpm_port *port,
2416                                  const struct pd_message *msg)
2417 {
2418         enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
2419         enum tcpm_state next_state;
2420
2421         switch (type) {
2422         case PD_CTRL_GOOD_CRC:
2423         case PD_CTRL_PING:
2424                 break;
2425         case PD_CTRL_GET_SOURCE_CAP:
2426                 tcpm_pd_handle_msg(port, PD_MSG_DATA_SOURCE_CAP, GET_SOURCE_CAPABILITIES);
2427                 break;
2428         case PD_CTRL_GET_SINK_CAP:
2429                 tcpm_pd_handle_msg(port, PD_MSG_DATA_SINK_CAP, GET_SINK_CAPABILITIES);
2430                 break;
2431         case PD_CTRL_GOTO_MIN:
2432                 break;
2433         case PD_CTRL_PS_RDY:
2434                 switch (port->state) {
2435                 case SNK_TRANSITION_SINK:
2436                         if (port->vbus_present) {
2437                                 tcpm_set_current_limit(port,
2438                                                        port->current_limit,
2439                                                        port->supply_voltage);
2440                                 port->explicit_contract = true;
2441                                 tcpm_set_auto_vbus_discharge_threshold(port,
2442                                                                        TYPEC_PWR_MODE_PD,
2443                                                                        port->pps_data.active,
2444                                                                        port->supply_voltage);
2445                                 /* Set VDM running flag ASAP */
2446                                 if (port->data_role == TYPEC_HOST &&
2447                                     port->send_discover)
2448                                         port->vdm_sm_running = true;
2449                                 tcpm_set_state(port, SNK_READY, 0);
2450                         } else {
2451                                 /*
2452                                  * Seen after power swap. Keep waiting for VBUS
2453                                  * in a transitional state.
2454                                  */
2455                                 tcpm_set_state(port,
2456                                                SNK_TRANSITION_SINK_VBUS, 0);
2457                         }
2458                         break;
2459                 case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
2460                         tcpm_set_state(port, PR_SWAP_SRC_SNK_SINK_ON, 0);
2461                         break;
2462                 case PR_SWAP_SNK_SRC_SINK_OFF:
2463                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON, 0);
2464                         break;
2465                 case VCONN_SWAP_WAIT_FOR_VCONN:
2466                         tcpm_set_state(port, VCONN_SWAP_TURN_OFF_VCONN, 0);
2467                         break;
2468                 case FR_SWAP_SNK_SRC_TRANSITION_TO_OFF:
2469                         tcpm_set_state(port, FR_SWAP_SNK_SRC_NEW_SINK_READY, 0);
2470                         break;
2471                 default:
2472                         tcpm_pd_handle_state(port,
2473                                              port->pwr_role == TYPEC_SOURCE ?
2474                                              SRC_SOFT_RESET_WAIT_SNK_TX :
2475                                              SNK_SOFT_RESET,
2476                                              NONE_AMS, 0);
2477                         break;
2478                 }
2479                 break;
2480         case PD_CTRL_REJECT:
2481         case PD_CTRL_WAIT:
2482         case PD_CTRL_NOT_SUPP:
2483                 switch (port->state) {
2484                 case SNK_NEGOTIATE_CAPABILITIES:
2485                         /* USB PD specification, Figure 8-43 */
2486                         if (port->explicit_contract) {
2487                                 next_state = SNK_READY;
2488                                 if (port->data_role == TYPEC_HOST &&
2489                                     port->send_discover)
2490                                         port->vdm_sm_running = true;
2491                         } else {
2492                                 next_state = SNK_WAIT_CAPABILITIES;
2493                         }
2494                         tcpm_set_state(port, next_state, 0);
2495                         break;
2496                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
2497                         /* Revert data back from any requested PPS updates */
2498                         port->pps_data.out_volt = port->supply_voltage;
2499                         port->pps_data.op_curr = port->current_limit;
2500                         port->pps_status = (type == PD_CTRL_WAIT ?
2501                                             -EAGAIN : -EOPNOTSUPP);
2502
2503                         if (port->data_role == TYPEC_HOST &&
2504                             port->send_discover)
2505                                 port->vdm_sm_running = true;
2506
2507                         tcpm_set_state(port, SNK_READY, 0);
2508                         break;
2509                 case DR_SWAP_SEND:
2510                         port->swap_status = (type == PD_CTRL_WAIT ?
2511                                              -EAGAIN : -EOPNOTSUPP);
2512                         tcpm_set_state(port, DR_SWAP_CANCEL, 0);
2513                         break;
2514                 case PR_SWAP_SEND:
2515                         port->swap_status = (type == PD_CTRL_WAIT ?
2516                                              -EAGAIN : -EOPNOTSUPP);
2517                         tcpm_set_state(port, PR_SWAP_CANCEL, 0);
2518                         break;
2519                 case VCONN_SWAP_SEND:
2520                         port->swap_status = (type == PD_CTRL_WAIT ?
2521                                              -EAGAIN : -EOPNOTSUPP);
2522                         tcpm_set_state(port, VCONN_SWAP_CANCEL, 0);
2523                         break;
2524                 case FR_SWAP_SEND:
2525                         tcpm_set_state(port, FR_SWAP_CANCEL, 0);
2526                         break;
2527                 case GET_SINK_CAP:
2528                         port->sink_cap_done = true;
2529                         tcpm_set_state(port, ready_state(port), 0);
2530                         break;
2531                 default:
2532                         tcpm_pd_handle_state(port,
2533                                              port->pwr_role == TYPEC_SOURCE ?
2534                                              SRC_SOFT_RESET_WAIT_SNK_TX :
2535                                              SNK_SOFT_RESET,
2536                                              NONE_AMS, 0);
2537                         break;
2538                 }
2539                 break;
2540         case PD_CTRL_ACCEPT:
2541                 switch (port->state) {
2542                 case SNK_NEGOTIATE_CAPABILITIES:
2543                         port->pps_data.active = false;
2544                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
2545                         break;
2546                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
2547                         port->pps_data.active = true;
2548                         port->supply_voltage = port->pps_data.out_volt;
2549                         port->current_limit = port->pps_data.op_curr;
2550                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
2551                         break;
2552                 case SOFT_RESET_SEND:
2553                         if (port->ams == SOFT_RESET_AMS)
2554                                 tcpm_ams_finish(port);
2555                         if (port->pwr_role == TYPEC_SOURCE) {
2556                                 port->upcoming_state = SRC_SEND_CAPABILITIES;
2557                                 tcpm_ams_start(port, POWER_NEGOTIATION);
2558                         } else {
2559                                 tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
2560                         }
2561                         break;
2562                 case DR_SWAP_SEND:
2563                         if (port->data_role == TYPEC_DEVICE &&
2564                             port->send_discover)
2565                                 port->vdm_sm_running = true;
2566
2567                         tcpm_set_state(port, DR_SWAP_CHANGE_DR, 0);
2568                         break;
2569                 case PR_SWAP_SEND:
2570                         tcpm_set_state(port, PR_SWAP_START, 0);
2571                         break;
2572                 case VCONN_SWAP_SEND:
2573                         tcpm_set_state(port, VCONN_SWAP_START, 0);
2574                         break;
2575                 case FR_SWAP_SEND:
2576                         tcpm_set_state(port, FR_SWAP_SNK_SRC_TRANSITION_TO_OFF, 0);
2577                         break;
2578                 default:
2579                         tcpm_pd_handle_state(port,
2580                                              port->pwr_role == TYPEC_SOURCE ?
2581                                              SRC_SOFT_RESET_WAIT_SNK_TX :
2582                                              SNK_SOFT_RESET,
2583                                              NONE_AMS, 0);
2584                         break;
2585                 }
2586                 break;
2587         case PD_CTRL_SOFT_RESET:
2588                 port->ams = SOFT_RESET_AMS;
2589                 tcpm_set_state(port, SOFT_RESET, 0);
2590                 break;
2591         case PD_CTRL_DR_SWAP:
2592                 /*
2593                  * XXX
2594                  * 6.3.9: If an alternate mode is active, a request to swap
2595                  * alternate modes shall trigger a port reset.
2596                  */
2597                 if (port->typec_caps.data != TYPEC_PORT_DRD) {
2598                         tcpm_pd_handle_msg(port,
2599                                            port->negotiated_rev < PD_REV30 ?
2600                                            PD_MSG_CTRL_REJECT :
2601                                            PD_MSG_CTRL_NOT_SUPP,
2602                                            NONE_AMS);
2603                 } else {
2604                         if (port->vdm_sm_running) {
2605                                 tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
2606                                 break;
2607                         }
2608
2609                         tcpm_pd_handle_state(port, DR_SWAP_ACCEPT, DATA_ROLE_SWAP, 0);
2610                 }
2611                 break;
2612         case PD_CTRL_PR_SWAP:
2613                 if (port->port_type != TYPEC_PORT_DRP) {
2614                         tcpm_pd_handle_msg(port,
2615                                            port->negotiated_rev < PD_REV30 ?
2616                                            PD_MSG_CTRL_REJECT :
2617                                            PD_MSG_CTRL_NOT_SUPP,
2618                                            NONE_AMS);
2619                 } else {
2620                         if (port->vdm_sm_running) {
2621                                 tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
2622                                 break;
2623                         }
2624
2625                         tcpm_pd_handle_state(port, PR_SWAP_ACCEPT, POWER_ROLE_SWAP, 0);
2626                 }
2627                 break;
2628         case PD_CTRL_VCONN_SWAP:
2629                 if (port->vdm_sm_running) {
2630                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
2631                         break;
2632                 }
2633
2634                 tcpm_pd_handle_state(port, VCONN_SWAP_ACCEPT, VCONN_SWAP, 0);
2635                 break;
2636         case PD_CTRL_GET_SOURCE_CAP_EXT:
2637         case PD_CTRL_GET_STATUS:
2638         case PD_CTRL_FR_SWAP:
2639         case PD_CTRL_GET_PPS_STATUS:
2640         case PD_CTRL_GET_COUNTRY_CODES:
2641                 /* Currently not supported */
2642                 tcpm_pd_handle_msg(port,
2643                                    port->negotiated_rev < PD_REV30 ?
2644                                    PD_MSG_CTRL_REJECT :
2645                                    PD_MSG_CTRL_NOT_SUPP,
2646                                    NONE_AMS);
2647                 break;
2648         default:
2649                 tcpm_pd_handle_msg(port,
2650                                    port->negotiated_rev < PD_REV30 ?
2651                                    PD_MSG_CTRL_REJECT :
2652                                    PD_MSG_CTRL_NOT_SUPP,
2653                                    NONE_AMS);
2654                 tcpm_log(port, "Unrecognized ctrl message type %#x", type);
2655                 break;
2656         }
2657 }
2658
2659 static void tcpm_pd_ext_msg_request(struct tcpm_port *port,
2660                                     const struct pd_message *msg)
2661 {
2662         enum pd_ext_msg_type type = pd_header_type_le(msg->header);
2663         unsigned int data_size = pd_ext_header_data_size_le(msg->ext_msg.header);
2664
2665         if (!(msg->ext_msg.header & PD_EXT_HDR_CHUNKED)) {
2666                 tcpm_pd_handle_msg(port, PD_MSG_CTRL_NOT_SUPP, NONE_AMS);
2667                 tcpm_log(port, "Unchunked extended messages unsupported");
2668                 return;
2669         }
2670
2671         if (data_size > PD_EXT_MAX_CHUNK_DATA) {
2672                 tcpm_pd_handle_state(port, CHUNK_NOT_SUPP, NONE_AMS, PD_T_CHUNK_NOT_SUPP);
2673                 tcpm_log(port, "Chunk handling not yet supported");
2674                 return;
2675         }
2676
2677         switch (type) {
2678         case PD_EXT_STATUS:
2679                 /*
2680                  * If PPS related events raised then get PPS status to clear
2681                  * (see USB PD 3.0 Spec, 6.5.2.4)
2682                  */
2683                 if (msg->ext_msg.data[USB_PD_EXT_SDB_EVENT_FLAGS] &
2684                     USB_PD_EXT_SDB_PPS_EVENTS)
2685                         tcpm_pd_handle_state(port, GET_PPS_STATUS_SEND,
2686                                              GETTING_SOURCE_SINK_STATUS, 0);
2687
2688                 else
2689                         tcpm_pd_handle_state(port, ready_state(port), NONE_AMS, 0);
2690                 break;
2691         case PD_EXT_PPS_STATUS:
2692                 /*
2693                  * For now the PPS status message is used to clear events
2694                  * and nothing more.
2695                  */
2696                 tcpm_pd_handle_state(port, ready_state(port), NONE_AMS, 0);
2697                 break;
2698         case PD_EXT_SOURCE_CAP_EXT:
2699         case PD_EXT_GET_BATT_CAP:
2700         case PD_EXT_GET_BATT_STATUS:
2701         case PD_EXT_BATT_CAP:
2702         case PD_EXT_GET_MANUFACTURER_INFO:
2703         case PD_EXT_MANUFACTURER_INFO:
2704         case PD_EXT_SECURITY_REQUEST:
2705         case PD_EXT_SECURITY_RESPONSE:
2706         case PD_EXT_FW_UPDATE_REQUEST:
2707         case PD_EXT_FW_UPDATE_RESPONSE:
2708         case PD_EXT_COUNTRY_INFO:
2709         case PD_EXT_COUNTRY_CODES:
2710                 tcpm_pd_handle_msg(port, PD_MSG_CTRL_NOT_SUPP, NONE_AMS);
2711                 break;
2712         default:
2713                 tcpm_pd_handle_msg(port, PD_MSG_CTRL_NOT_SUPP, NONE_AMS);
2714                 tcpm_log(port, "Unrecognized extended message type %#x", type);
2715                 break;
2716         }
2717 }
2718
2719 static void tcpm_pd_rx_handler(struct kthread_work *work)
2720 {
2721         struct pd_rx_event *event = container_of(work,
2722                                                  struct pd_rx_event, work);
2723         const struct pd_message *msg = &event->msg;
2724         unsigned int cnt = pd_header_cnt_le(msg->header);
2725         struct tcpm_port *port = event->port;
2726
2727         mutex_lock(&port->lock);
2728
2729         tcpm_log(port, "PD RX, header: %#x [%d]", le16_to_cpu(msg->header),
2730                  port->attached);
2731
2732         if (port->attached) {
2733                 enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
2734                 unsigned int msgid = pd_header_msgid_le(msg->header);
2735
2736                 /*
2737                  * USB PD standard, 6.6.1.2:
2738                  * "... if MessageID value in a received Message is the
2739                  * same as the stored value, the receiver shall return a
2740                  * GoodCRC Message with that MessageID value and drop
2741                  * the Message (this is a retry of an already received
2742                  * Message). Note: this shall not apply to the Soft_Reset
2743                  * Message which always has a MessageID value of zero."
2744                  */
2745                 if (msgid == port->rx_msgid && type != PD_CTRL_SOFT_RESET)
2746                         goto done;
2747                 port->rx_msgid = msgid;
2748
2749                 /*
2750                  * If both ends believe to be DFP/host, we have a data role
2751                  * mismatch.
2752                  */
2753                 if (!!(le16_to_cpu(msg->header) & PD_HEADER_DATA_ROLE) ==
2754                     (port->data_role == TYPEC_HOST)) {
2755                         tcpm_log(port,
2756                                  "Data role mismatch, initiating error recovery");
2757                         tcpm_set_state(port, ERROR_RECOVERY, 0);
2758                 } else {
2759                         if (msg->header & PD_HEADER_EXT_HDR)
2760                                 tcpm_pd_ext_msg_request(port, msg);
2761                         else if (cnt)
2762                                 tcpm_pd_data_request(port, msg);
2763                         else
2764                                 tcpm_pd_ctrl_request(port, msg);
2765                 }
2766         }
2767
2768 done:
2769         mutex_unlock(&port->lock);
2770         kfree(event);
2771 }
2772
2773 void tcpm_pd_receive(struct tcpm_port *port, const struct pd_message *msg)
2774 {
2775         struct pd_rx_event *event;
2776
2777         event = kzalloc(sizeof(*event), GFP_ATOMIC);
2778         if (!event)
2779                 return;
2780
2781         kthread_init_work(&event->work, tcpm_pd_rx_handler);
2782         event->port = port;
2783         memcpy(&event->msg, msg, sizeof(*msg));
2784         kthread_queue_work(port->wq, &event->work);
2785 }
2786 EXPORT_SYMBOL_GPL(tcpm_pd_receive);
2787
2788 static int tcpm_pd_send_control(struct tcpm_port *port,
2789                                 enum pd_ctrl_msg_type type)
2790 {
2791         struct pd_message msg;
2792
2793         memset(&msg, 0, sizeof(msg));
2794         msg.header = PD_HEADER_LE(type, port->pwr_role,
2795                                   port->data_role,
2796                                   port->negotiated_rev,
2797                                   port->message_id, 0);
2798
2799         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2800 }
2801
2802 /*
2803  * Send queued message without affecting state.
2804  * Return true if state machine should go back to sleep,
2805  * false otherwise.
2806  */
2807 static bool tcpm_send_queued_message(struct tcpm_port *port)
2808 {
2809         enum pd_msg_request queued_message;
2810         int ret;
2811
2812         do {
2813                 queued_message = port->queued_message;
2814                 port->queued_message = PD_MSG_NONE;
2815
2816                 switch (queued_message) {
2817                 case PD_MSG_CTRL_WAIT:
2818                         tcpm_pd_send_control(port, PD_CTRL_WAIT);
2819                         break;
2820                 case PD_MSG_CTRL_REJECT:
2821                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
2822                         break;
2823                 case PD_MSG_CTRL_NOT_SUPP:
2824                         tcpm_pd_send_control(port, PD_CTRL_NOT_SUPP);
2825                         break;
2826                 case PD_MSG_DATA_SINK_CAP:
2827                         ret = tcpm_pd_send_sink_caps(port);
2828                         if (ret < 0) {
2829                                 tcpm_log(port, "Unable to send snk caps, ret=%d", ret);
2830                                 tcpm_set_state(port, SNK_SOFT_RESET, 0);
2831                         }
2832                         tcpm_ams_finish(port);
2833                         break;
2834                 case PD_MSG_DATA_SOURCE_CAP:
2835                         ret = tcpm_pd_send_source_caps(port);
2836                         if (ret < 0) {
2837                                 tcpm_log(port,
2838                                          "Unable to send src caps, ret=%d",
2839                                          ret);
2840                                 tcpm_set_state(port, SOFT_RESET_SEND, 0);
2841                         } else if (port->pwr_role == TYPEC_SOURCE) {
2842                                 tcpm_ams_finish(port);
2843                                 tcpm_set_state(port, HARD_RESET_SEND,
2844                                                PD_T_SENDER_RESPONSE);
2845                         } else {
2846                                 tcpm_ams_finish(port);
2847                         }
2848                         break;
2849                 default:
2850                         break;
2851                 }
2852         } while (port->queued_message != PD_MSG_NONE);
2853
2854         if (port->delayed_state != INVALID_STATE) {
2855                 if (ktime_after(port->delayed_runtime, ktime_get())) {
2856                         mod_tcpm_delayed_work(port, ktime_to_ms(ktime_sub(port->delayed_runtime,
2857                                                                           ktime_get())));
2858                         return true;
2859                 }
2860                 port->delayed_state = INVALID_STATE;
2861         }
2862         return false;
2863 }
2864
2865 static int tcpm_pd_check_request(struct tcpm_port *port)
2866 {
2867         u32 pdo, rdo = port->sink_request;
2868         unsigned int max, op, pdo_max, index;
2869         enum pd_pdo_type type;
2870
2871         index = rdo_index(rdo);
2872         if (!index || index > port->nr_src_pdo)
2873                 return -EINVAL;
2874
2875         pdo = port->src_pdo[index - 1];
2876         type = pdo_type(pdo);
2877         switch (type) {
2878         case PDO_TYPE_FIXED:
2879         case PDO_TYPE_VAR:
2880                 max = rdo_max_current(rdo);
2881                 op = rdo_op_current(rdo);
2882                 pdo_max = pdo_max_current(pdo);
2883
2884                 if (op > pdo_max)
2885                         return -EINVAL;
2886                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
2887                         return -EINVAL;
2888
2889                 if (type == PDO_TYPE_FIXED)
2890                         tcpm_log(port,
2891                                  "Requested %u mV, %u mA for %u / %u mA",
2892                                  pdo_fixed_voltage(pdo), pdo_max, op, max);
2893                 else
2894                         tcpm_log(port,
2895                                  "Requested %u -> %u mV, %u mA for %u / %u mA",
2896                                  pdo_min_voltage(pdo), pdo_max_voltage(pdo),
2897                                  pdo_max, op, max);
2898                 break;
2899         case PDO_TYPE_BATT:
2900                 max = rdo_max_power(rdo);
2901                 op = rdo_op_power(rdo);
2902                 pdo_max = pdo_max_power(pdo);
2903
2904                 if (op > pdo_max)
2905                         return -EINVAL;
2906                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
2907                         return -EINVAL;
2908                 tcpm_log(port,
2909                          "Requested %u -> %u mV, %u mW for %u / %u mW",
2910                          pdo_min_voltage(pdo), pdo_max_voltage(pdo),
2911                          pdo_max, op, max);
2912                 break;
2913         default:
2914                 return -EINVAL;
2915         }
2916
2917         port->op_vsafe5v = index == 1;
2918
2919         return 0;
2920 }
2921
2922 #define min_power(x, y) min(pdo_max_power(x), pdo_max_power(y))
2923 #define min_current(x, y) min(pdo_max_current(x), pdo_max_current(y))
2924
2925 static int tcpm_pd_select_pdo(struct tcpm_port *port, int *sink_pdo,
2926                               int *src_pdo)
2927 {
2928         unsigned int i, j, max_src_mv = 0, min_src_mv = 0, max_mw = 0,
2929                      max_mv = 0, src_mw = 0, src_ma = 0, max_snk_mv = 0,
2930                      min_snk_mv = 0;
2931         int ret = -EINVAL;
2932
2933         port->pps_data.supported = false;
2934         port->usb_type = POWER_SUPPLY_USB_TYPE_PD;
2935         power_supply_changed(port->psy);
2936
2937         /*
2938          * Select the source PDO providing the most power which has a
2939          * matchig sink cap.
2940          */
2941         for (i = 0; i < port->nr_source_caps; i++) {
2942                 u32 pdo = port->source_caps[i];
2943                 enum pd_pdo_type type = pdo_type(pdo);
2944
2945                 switch (type) {
2946                 case PDO_TYPE_FIXED:
2947                         max_src_mv = pdo_fixed_voltage(pdo);
2948                         min_src_mv = max_src_mv;
2949                         break;
2950                 case PDO_TYPE_BATT:
2951                 case PDO_TYPE_VAR:
2952                         max_src_mv = pdo_max_voltage(pdo);
2953                         min_src_mv = pdo_min_voltage(pdo);
2954                         break;
2955                 case PDO_TYPE_APDO:
2956                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS) {
2957                                 port->pps_data.supported = true;
2958                                 port->usb_type =
2959                                         POWER_SUPPLY_USB_TYPE_PD_PPS;
2960                                 power_supply_changed(port->psy);
2961                         }
2962                         continue;
2963                 default:
2964                         tcpm_log(port, "Invalid source PDO type, ignoring");
2965                         continue;
2966                 }
2967
2968                 switch (type) {
2969                 case PDO_TYPE_FIXED:
2970                 case PDO_TYPE_VAR:
2971                         src_ma = pdo_max_current(pdo);
2972                         src_mw = src_ma * min_src_mv / 1000;
2973                         break;
2974                 case PDO_TYPE_BATT:
2975                         src_mw = pdo_max_power(pdo);
2976                         break;
2977                 case PDO_TYPE_APDO:
2978                         continue;
2979                 default:
2980                         tcpm_log(port, "Invalid source PDO type, ignoring");
2981                         continue;
2982                 }
2983
2984                 for (j = 0; j < port->nr_snk_pdo; j++) {
2985                         pdo = port->snk_pdo[j];
2986
2987                         switch (pdo_type(pdo)) {
2988                         case PDO_TYPE_FIXED:
2989                                 max_snk_mv = pdo_fixed_voltage(pdo);
2990                                 min_snk_mv = max_snk_mv;
2991                                 break;
2992                         case PDO_TYPE_BATT:
2993                         case PDO_TYPE_VAR:
2994                                 max_snk_mv = pdo_max_voltage(pdo);
2995                                 min_snk_mv = pdo_min_voltage(pdo);
2996                                 break;
2997                         case PDO_TYPE_APDO:
2998                                 continue;
2999                         default:
3000                                 tcpm_log(port, "Invalid sink PDO type, ignoring");
3001                                 continue;
3002                         }
3003
3004                         if (max_src_mv <= max_snk_mv &&
3005                                 min_src_mv >= min_snk_mv) {
3006                                 /* Prefer higher voltages if available */
3007                                 if ((src_mw == max_mw && min_src_mv > max_mv) ||
3008                                                         src_mw > max_mw) {
3009                                         *src_pdo = i;
3010                                         *sink_pdo = j;
3011                                         max_mw = src_mw;
3012                                         max_mv = min_src_mv;
3013                                         ret = 0;
3014                                 }
3015                         }
3016                 }
3017         }
3018
3019         return ret;
3020 }
3021
3022 #define min_pps_apdo_current(x, y)      \
3023         min(pdo_pps_apdo_max_current(x), pdo_pps_apdo_max_current(y))
3024
3025 static unsigned int tcpm_pd_select_pps_apdo(struct tcpm_port *port)
3026 {
3027         unsigned int i, j, max_mw = 0, max_mv = 0;
3028         unsigned int min_src_mv, max_src_mv, src_ma, src_mw;
3029         unsigned int min_snk_mv, max_snk_mv;
3030         unsigned int max_op_mv;
3031         u32 pdo, src, snk;
3032         unsigned int src_pdo = 0, snk_pdo = 0;
3033
3034         /*
3035          * Select the source PPS APDO providing the most power while staying
3036          * within the board's limits. We skip the first PDO as this is always
3037          * 5V 3A.
3038          */
3039         for (i = 1; i < port->nr_source_caps; ++i) {
3040                 pdo = port->source_caps[i];
3041
3042                 switch (pdo_type(pdo)) {
3043                 case PDO_TYPE_APDO:
3044                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
3045                                 tcpm_log(port, "Not PPS APDO (source), ignoring");
3046                                 continue;
3047                         }
3048
3049                         min_src_mv = pdo_pps_apdo_min_voltage(pdo);
3050                         max_src_mv = pdo_pps_apdo_max_voltage(pdo);
3051                         src_ma = pdo_pps_apdo_max_current(pdo);
3052                         src_mw = (src_ma * max_src_mv) / 1000;
3053
3054                         /*
3055                          * Now search through the sink PDOs to find a matching
3056                          * PPS APDO. Again skip the first sink PDO as this will
3057                          * always be 5V 3A.
3058                          */
3059                         for (j = 1; j < port->nr_snk_pdo; j++) {
3060                                 pdo = port->snk_pdo[j];
3061
3062                                 switch (pdo_type(pdo)) {
3063                                 case PDO_TYPE_APDO:
3064                                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
3065                                                 tcpm_log(port,
3066                                                          "Not PPS APDO (sink), ignoring");
3067                                                 continue;
3068                                         }
3069
3070                                         min_snk_mv =
3071                                                 pdo_pps_apdo_min_voltage(pdo);
3072                                         max_snk_mv =
3073                                                 pdo_pps_apdo_max_voltage(pdo);
3074                                         break;
3075                                 default:
3076                                         tcpm_log(port,
3077                                                  "Not APDO type (sink), ignoring");
3078                                         continue;
3079                                 }
3080
3081                                 if (min_src_mv <= max_snk_mv &&
3082                                     max_src_mv >= min_snk_mv) {
3083                                         max_op_mv = min(max_src_mv, max_snk_mv);
3084                                         src_mw = (max_op_mv * src_ma) / 1000;
3085                                         /* Prefer higher voltages if available */
3086                                         if ((src_mw == max_mw &&
3087                                              max_op_mv > max_mv) ||
3088                                             src_mw > max_mw) {
3089                                                 src_pdo = i;
3090                                                 snk_pdo = j;
3091                                                 max_mw = src_mw;
3092                                                 max_mv = max_op_mv;
3093                                         }
3094                                 }
3095                         }
3096
3097                         break;
3098                 default:
3099                         tcpm_log(port, "Not APDO type (source), ignoring");
3100                         continue;
3101                 }
3102         }
3103
3104         if (src_pdo) {
3105                 src = port->source_caps[src_pdo];
3106                 snk = port->snk_pdo[snk_pdo];
3107
3108                 port->pps_data.min_volt = max(pdo_pps_apdo_min_voltage(src),
3109                                               pdo_pps_apdo_min_voltage(snk));
3110                 port->pps_data.max_volt = min(pdo_pps_apdo_max_voltage(src),
3111                                               pdo_pps_apdo_max_voltage(snk));
3112                 port->pps_data.max_curr = min_pps_apdo_current(src, snk);
3113                 port->pps_data.out_volt = min(port->pps_data.max_volt,
3114                                               max(port->pps_data.min_volt,
3115                                                   port->pps_data.out_volt));
3116                 port->pps_data.op_curr = min(port->pps_data.max_curr,
3117                                              port->pps_data.op_curr);
3118                 power_supply_changed(port->psy);
3119         }
3120
3121         return src_pdo;
3122 }
3123
3124 static int tcpm_pd_build_request(struct tcpm_port *port, u32 *rdo)
3125 {
3126         unsigned int mv, ma, mw, flags;
3127         unsigned int max_ma, max_mw;
3128         enum pd_pdo_type type;
3129         u32 pdo, matching_snk_pdo;
3130         int src_pdo_index = 0;
3131         int snk_pdo_index = 0;
3132         int ret;
3133
3134         ret = tcpm_pd_select_pdo(port, &snk_pdo_index, &src_pdo_index);
3135         if (ret < 0)
3136                 return ret;
3137
3138         pdo = port->source_caps[src_pdo_index];
3139         matching_snk_pdo = port->snk_pdo[snk_pdo_index];
3140         type = pdo_type(pdo);
3141
3142         switch (type) {
3143         case PDO_TYPE_FIXED:
3144                 mv = pdo_fixed_voltage(pdo);
3145                 break;
3146         case PDO_TYPE_BATT:
3147         case PDO_TYPE_VAR:
3148                 mv = pdo_min_voltage(pdo);
3149                 break;
3150         default:
3151                 tcpm_log(port, "Invalid PDO selected!");
3152                 return -EINVAL;
3153         }
3154
3155         /* Select maximum available current within the sink pdo's limit */
3156         if (type == PDO_TYPE_BATT) {
3157                 mw = min_power(pdo, matching_snk_pdo);
3158                 ma = 1000 * mw / mv;
3159         } else {
3160                 ma = min_current(pdo, matching_snk_pdo);
3161                 mw = ma * mv / 1000;
3162         }
3163
3164         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
3165
3166         /* Set mismatch bit if offered power is less than operating power */
3167         max_ma = ma;
3168         max_mw = mw;
3169         if (mw < port->operating_snk_mw) {
3170                 flags |= RDO_CAP_MISMATCH;
3171                 if (type == PDO_TYPE_BATT &&
3172                     (pdo_max_power(matching_snk_pdo) > pdo_max_power(pdo)))
3173                         max_mw = pdo_max_power(matching_snk_pdo);
3174                 else if (pdo_max_current(matching_snk_pdo) >
3175                          pdo_max_current(pdo))
3176                         max_ma = pdo_max_current(matching_snk_pdo);
3177         }
3178
3179         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
3180                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
3181                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
3182                  port->polarity);
3183
3184         if (type == PDO_TYPE_BATT) {
3185                 *rdo = RDO_BATT(src_pdo_index + 1, mw, max_mw, flags);
3186
3187                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mW%s",
3188                          src_pdo_index, mv, mw,
3189                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
3190         } else {
3191                 *rdo = RDO_FIXED(src_pdo_index + 1, ma, max_ma, flags);
3192
3193                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mA%s",
3194                          src_pdo_index, mv, ma,
3195                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
3196         }
3197
3198         port->current_limit = ma;
3199         port->supply_voltage = mv;
3200
3201         return 0;
3202 }
3203
3204 static int tcpm_pd_send_request(struct tcpm_port *port)
3205 {
3206         struct pd_message msg;
3207         int ret;
3208         u32 rdo;
3209
3210         ret = tcpm_pd_build_request(port, &rdo);
3211         if (ret < 0)
3212                 return ret;
3213
3214         memset(&msg, 0, sizeof(msg));
3215         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
3216                                   port->pwr_role,
3217                                   port->data_role,
3218                                   port->negotiated_rev,
3219                                   port->message_id, 1);
3220         msg.payload[0] = cpu_to_le32(rdo);
3221
3222         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
3223 }
3224
3225 static int tcpm_pd_build_pps_request(struct tcpm_port *port, u32 *rdo)
3226 {
3227         unsigned int out_mv, op_ma, op_mw, max_mv, max_ma, flags;
3228         enum pd_pdo_type type;
3229         unsigned int src_pdo_index;
3230         u32 pdo;
3231
3232         src_pdo_index = tcpm_pd_select_pps_apdo(port);
3233         if (!src_pdo_index)
3234                 return -EOPNOTSUPP;
3235
3236         pdo = port->source_caps[src_pdo_index];
3237         type = pdo_type(pdo);
3238
3239         switch (type) {
3240         case PDO_TYPE_APDO:
3241                 if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
3242                         tcpm_log(port, "Invalid APDO selected!");
3243                         return -EINVAL;
3244                 }
3245                 max_mv = port->pps_data.max_volt;
3246                 max_ma = port->pps_data.max_curr;
3247                 out_mv = port->pps_data.out_volt;
3248                 op_ma = port->pps_data.op_curr;
3249                 break;
3250         default:
3251                 tcpm_log(port, "Invalid PDO selected!");
3252                 return -EINVAL;
3253         }
3254
3255         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
3256
3257         op_mw = (op_ma * out_mv) / 1000;
3258         if (op_mw < port->operating_snk_mw) {
3259                 /*
3260                  * Try raising current to meet power needs. If that's not enough
3261                  * then try upping the voltage. If that's still not enough
3262                  * then we've obviously chosen a PPS APDO which really isn't
3263                  * suitable so abandon ship.
3264                  */
3265                 op_ma = (port->operating_snk_mw * 1000) / out_mv;
3266                 if ((port->operating_snk_mw * 1000) % out_mv)
3267                         ++op_ma;
3268                 op_ma += RDO_PROG_CURR_MA_STEP - (op_ma % RDO_PROG_CURR_MA_STEP);
3269
3270                 if (op_ma > max_ma) {
3271                         op_ma = max_ma;
3272                         out_mv = (port->operating_snk_mw * 1000) / op_ma;
3273                         if ((port->operating_snk_mw * 1000) % op_ma)
3274                                 ++out_mv;
3275                         out_mv += RDO_PROG_VOLT_MV_STEP -
3276                                   (out_mv % RDO_PROG_VOLT_MV_STEP);
3277
3278                         if (out_mv > max_mv) {
3279                                 tcpm_log(port, "Invalid PPS APDO selected!");
3280                                 return -EINVAL;
3281                         }
3282                 }
3283         }
3284
3285         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
3286                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
3287                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
3288                  port->polarity);
3289
3290         *rdo = RDO_PROG(src_pdo_index + 1, out_mv, op_ma, flags);
3291
3292         tcpm_log(port, "Requesting APDO %d: %u mV, %u mA",
3293                  src_pdo_index, out_mv, op_ma);
3294
3295         port->pps_data.op_curr = op_ma;
3296         port->pps_data.out_volt = out_mv;
3297
3298         return 0;
3299 }
3300
3301 static int tcpm_pd_send_pps_request(struct tcpm_port *port)
3302 {
3303         struct pd_message msg;
3304         int ret;
3305         u32 rdo;
3306
3307         ret = tcpm_pd_build_pps_request(port, &rdo);
3308         if (ret < 0)
3309                 return ret;
3310
3311         memset(&msg, 0, sizeof(msg));
3312         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
3313                                   port->pwr_role,
3314                                   port->data_role,
3315                                   port->negotiated_rev,
3316                                   port->message_id, 1);
3317         msg.payload[0] = cpu_to_le32(rdo);
3318
3319         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
3320 }
3321
3322 static int tcpm_set_vbus(struct tcpm_port *port, bool enable)
3323 {
3324         int ret;
3325
3326         if (enable && port->vbus_charge)
3327                 return -EINVAL;
3328
3329         tcpm_log(port, "vbus:=%d charge=%d", enable, port->vbus_charge);
3330
3331         ret = port->tcpc->set_vbus(port->tcpc, enable, port->vbus_charge);
3332         if (ret < 0)
3333                 return ret;
3334
3335         port->vbus_source = enable;
3336         return 0;
3337 }
3338
3339 static int tcpm_set_charge(struct tcpm_port *port, bool charge)
3340 {
3341         int ret;
3342
3343         if (charge && port->vbus_source)
3344                 return -EINVAL;
3345
3346         if (charge != port->vbus_charge) {
3347                 tcpm_log(port, "vbus=%d charge:=%d", port->vbus_source, charge);
3348                 ret = port->tcpc->set_vbus(port->tcpc, port->vbus_source,
3349                                            charge);
3350                 if (ret < 0)
3351                         return ret;
3352         }
3353         port->vbus_charge = charge;
3354         power_supply_changed(port->psy);
3355         return 0;
3356 }
3357
3358 static bool tcpm_start_toggling(struct tcpm_port *port, enum typec_cc_status cc)
3359 {
3360         int ret;
3361
3362         if (!port->tcpc->start_toggling)
3363                 return false;
3364
3365         tcpm_log_force(port, "Start toggling");
3366         ret = port->tcpc->start_toggling(port->tcpc, port->port_type, cc);
3367         return ret == 0;
3368 }
3369
3370 static int tcpm_init_vbus(struct tcpm_port *port)
3371 {
3372         int ret;
3373
3374         ret = port->tcpc->set_vbus(port->tcpc, false, false);
3375         port->vbus_source = false;
3376         port->vbus_charge = false;
3377         return ret;
3378 }
3379
3380 static int tcpm_init_vconn(struct tcpm_port *port)
3381 {
3382         int ret;
3383
3384         ret = port->tcpc->set_vconn(port->tcpc, false);
3385         port->vconn_role = TYPEC_SINK;
3386         return ret;
3387 }
3388
3389 static void tcpm_typec_connect(struct tcpm_port *port)
3390 {
3391         if (!port->connected) {
3392                 /* Make sure we don't report stale identity information */
3393                 memset(&port->partner_ident, 0, sizeof(port->partner_ident));
3394                 port->partner_desc.usb_pd = port->pd_capable;
3395                 if (tcpm_port_is_debug(port))
3396                         port->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
3397                 else if (tcpm_port_is_audio(port))
3398                         port->partner_desc.accessory = TYPEC_ACCESSORY_AUDIO;
3399                 else
3400                         port->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
3401                 port->partner = typec_register_partner(port->typec_port,
3402                                                        &port->partner_desc);
3403                 port->connected = true;
3404         }
3405 }
3406
3407 static int tcpm_src_attach(struct tcpm_port *port)
3408 {
3409         enum typec_cc_polarity polarity =
3410                                 port->cc2 == TYPEC_CC_RD ? TYPEC_POLARITY_CC2
3411                                                          : TYPEC_POLARITY_CC1;
3412         int ret;
3413
3414         if (port->attached)
3415                 return 0;
3416
3417         ret = tcpm_set_polarity(port, polarity);
3418         if (ret < 0)
3419                 return ret;
3420
3421         if (port->tcpc->enable_auto_vbus_discharge) {
3422                 ret = port->tcpc->enable_auto_vbus_discharge(port->tcpc, true);
3423                 tcpm_log_force(port, "enable vbus discharge ret:%d", ret);
3424                 if (!ret)
3425                         port->auto_vbus_discharge_enabled = true;
3426         }
3427
3428         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, tcpm_data_role_for_source(port));
3429         if (ret < 0)
3430                 return ret;
3431
3432         ret = port->tcpc->set_pd_rx(port->tcpc, true);
3433         if (ret < 0)
3434                 goto out_disable_mux;
3435
3436         /*
3437          * USB Type-C specification, version 1.2,
3438          * chapter 4.5.2.2.8.1 (Attached.SRC Requirements)
3439          * Enable VCONN only if the non-RD port is set to RA.
3440          */
3441         if ((polarity == TYPEC_POLARITY_CC1 && port->cc2 == TYPEC_CC_RA) ||
3442             (polarity == TYPEC_POLARITY_CC2 && port->cc1 == TYPEC_CC_RA)) {
3443                 ret = tcpm_set_vconn(port, true);
3444                 if (ret < 0)
3445                         goto out_disable_pd;
3446         }
3447
3448         ret = tcpm_set_vbus(port, true);
3449         if (ret < 0)
3450                 goto out_disable_vconn;
3451
3452         port->pd_capable = false;
3453
3454         port->partner = NULL;
3455
3456         port->attached = true;
3457         port->send_discover = true;
3458
3459         return 0;
3460
3461 out_disable_vconn:
3462         tcpm_set_vconn(port, false);
3463 out_disable_pd:
3464         port->tcpc->set_pd_rx(port->tcpc, false);
3465 out_disable_mux:
3466         tcpm_mux_set(port, TYPEC_STATE_SAFE, USB_ROLE_NONE,
3467                      TYPEC_ORIENTATION_NONE);
3468         return ret;
3469 }
3470
3471 static void tcpm_typec_disconnect(struct tcpm_port *port)
3472 {
3473         if (port->connected) {
3474                 typec_unregister_partner(port->partner);
3475                 port->partner = NULL;
3476                 port->connected = false;
3477         }
3478 }
3479
3480 static void tcpm_unregister_altmodes(struct tcpm_port *port)
3481 {
3482         struct pd_mode_data *modep = &port->mode_data;
3483         int i;
3484
3485         for (i = 0; i < modep->altmodes; i++) {
3486                 typec_unregister_altmode(port->partner_altmode[i]);
3487                 port->partner_altmode[i] = NULL;
3488         }
3489
3490         memset(modep, 0, sizeof(*modep));
3491 }
3492
3493 static void tcpm_set_partner_usb_comm_capable(struct tcpm_port *port, bool capable)
3494 {
3495         tcpm_log(port, "Setting usb_comm capable %s", capable ? "true" : "false");
3496
3497         if (port->tcpc->set_partner_usb_comm_capable)
3498                 port->tcpc->set_partner_usb_comm_capable(port->tcpc, capable);
3499 }
3500
3501 static void tcpm_reset_port(struct tcpm_port *port)
3502 {
3503         int ret;
3504
3505         if (port->tcpc->enable_auto_vbus_discharge) {
3506                 ret = port->tcpc->enable_auto_vbus_discharge(port->tcpc, false);
3507                 tcpm_log_force(port, "Disable vbus discharge ret:%d", ret);
3508                 if (!ret)
3509                         port->auto_vbus_discharge_enabled = false;
3510         }
3511         port->in_ams = false;
3512         port->ams = NONE_AMS;
3513         port->vdm_sm_running = false;
3514         tcpm_unregister_altmodes(port);
3515         tcpm_typec_disconnect(port);
3516         port->attached = false;
3517         port->pd_capable = false;
3518         port->pps_data.supported = false;
3519         tcpm_set_partner_usb_comm_capable(port, false);
3520
3521         /*
3522          * First Rx ID should be 0; set this to a sentinel of -1 so that
3523          * we can check tcpm_pd_rx_handler() if we had seen it before.
3524          */
3525         port->rx_msgid = -1;
3526
3527         port->tcpc->set_pd_rx(port->tcpc, false);
3528         tcpm_init_vbus(port);   /* also disables charging */
3529         tcpm_init_vconn(port);
3530         tcpm_set_current_limit(port, 0, 0);
3531         tcpm_set_polarity(port, TYPEC_POLARITY_CC1);
3532         tcpm_mux_set(port, TYPEC_STATE_SAFE, USB_ROLE_NONE,
3533                      TYPEC_ORIENTATION_NONE);
3534         tcpm_set_attached_state(port, false);
3535         port->try_src_count = 0;
3536         port->try_snk_count = 0;
3537         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
3538         power_supply_changed(port->psy);
3539         port->nr_sink_caps = 0;
3540         port->sink_cap_done = false;
3541         if (port->tcpc->enable_frs)
3542                 port->tcpc->enable_frs(port->tcpc, false);
3543
3544         power_supply_changed(port->psy);
3545 }
3546
3547 static void tcpm_detach(struct tcpm_port *port)
3548 {
3549         if (tcpm_port_is_disconnected(port))
3550                 port->hard_reset_count = 0;
3551
3552         if (!port->attached)
3553                 return;
3554
3555         if (port->tcpc->set_bist_data) {
3556                 tcpm_log(port, "disable BIST MODE TESTDATA");
3557                 port->tcpc->set_bist_data(port->tcpc, false);
3558         }
3559
3560         tcpm_reset_port(port);
3561 }
3562
3563 static void tcpm_src_detach(struct tcpm_port *port)
3564 {
3565         tcpm_detach(port);
3566 }
3567
3568 static int tcpm_snk_attach(struct tcpm_port *port)
3569 {
3570         int ret;
3571
3572         if (port->attached)
3573                 return 0;
3574
3575         ret = tcpm_set_polarity(port, port->cc2 != TYPEC_CC_OPEN ?
3576                                 TYPEC_POLARITY_CC2 : TYPEC_POLARITY_CC1);
3577         if (ret < 0)
3578                 return ret;
3579
3580         if (port->tcpc->enable_auto_vbus_discharge) {
3581                 tcpm_set_auto_vbus_discharge_threshold(port, TYPEC_PWR_MODE_USB, false, VSAFE5V);
3582                 ret = port->tcpc->enable_auto_vbus_discharge(port->tcpc, true);
3583                 tcpm_log_force(port, "enable vbus discharge ret:%d", ret);
3584                 if (!ret)
3585                         port->auto_vbus_discharge_enabled = true;
3586         }
3587
3588         ret = tcpm_set_roles(port, true, TYPEC_SINK, tcpm_data_role_for_sink(port));
3589         if (ret < 0)
3590                 return ret;
3591
3592         port->pd_capable = false;
3593
3594         port->partner = NULL;
3595
3596         port->attached = true;
3597         port->send_discover = true;
3598
3599         return 0;
3600 }
3601
3602 static void tcpm_snk_detach(struct tcpm_port *port)
3603 {
3604         tcpm_detach(port);
3605 }
3606
3607 static int tcpm_acc_attach(struct tcpm_port *port)
3608 {
3609         int ret;
3610
3611         if (port->attached)
3612                 return 0;
3613
3614         ret = tcpm_set_roles(port, true, TYPEC_SOURCE,
3615                              tcpm_data_role_for_source(port));
3616         if (ret < 0)
3617                 return ret;
3618
3619         port->partner = NULL;
3620
3621         tcpm_typec_connect(port);
3622
3623         port->attached = true;
3624
3625         return 0;
3626 }
3627
3628 static void tcpm_acc_detach(struct tcpm_port *port)
3629 {
3630         tcpm_detach(port);
3631 }
3632
3633 static inline enum tcpm_state hard_reset_state(struct tcpm_port *port)
3634 {
3635         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
3636                 return HARD_RESET_SEND;
3637         if (port->pd_capable)
3638                 return ERROR_RECOVERY;
3639         if (port->pwr_role == TYPEC_SOURCE)
3640                 return SRC_UNATTACHED;
3641         if (port->state == SNK_WAIT_CAPABILITIES)
3642                 return SNK_READY;
3643         return SNK_UNATTACHED;
3644 }
3645
3646 static inline enum tcpm_state unattached_state(struct tcpm_port *port)
3647 {
3648         if (port->port_type == TYPEC_PORT_DRP) {
3649                 if (port->pwr_role == TYPEC_SOURCE)
3650                         return SRC_UNATTACHED;
3651                 else
3652                         return SNK_UNATTACHED;
3653         } else if (port->port_type == TYPEC_PORT_SRC) {
3654                 return SRC_UNATTACHED;
3655         }
3656
3657         return SNK_UNATTACHED;
3658 }
3659
3660 static void tcpm_check_send_discover(struct tcpm_port *port)
3661 {
3662         if ((port->data_role == TYPEC_HOST || port->negotiated_rev > PD_REV20) &&
3663             port->send_discover && port->pd_capable)
3664                 tcpm_send_vdm(port, USB_SID_PD, CMD_DISCOVER_IDENT, NULL, 0);
3665         port->send_discover = false;
3666 }
3667
3668 static void tcpm_swap_complete(struct tcpm_port *port, int result)
3669 {
3670         if (port->swap_pending) {
3671                 port->swap_status = result;
3672                 port->swap_pending = false;
3673                 port->non_pd_role_swap = false;
3674                 complete(&port->swap_complete);
3675         }
3676 }
3677
3678 static enum typec_pwr_opmode tcpm_get_pwr_opmode(enum typec_cc_status cc)
3679 {
3680         switch (cc) {
3681         case TYPEC_CC_RP_1_5:
3682                 return TYPEC_PWR_MODE_1_5A;
3683         case TYPEC_CC_RP_3_0:
3684                 return TYPEC_PWR_MODE_3_0A;
3685         case TYPEC_CC_RP_DEF:
3686         default:
3687                 return TYPEC_PWR_MODE_USB;
3688         }
3689 }
3690
3691 static void run_state_machine(struct tcpm_port *port)
3692 {
3693         int ret;
3694         enum typec_pwr_opmode opmode;
3695         unsigned int msecs;
3696         enum tcpm_state upcoming_state;
3697
3698         port->enter_state = port->state;
3699         switch (port->state) {
3700         case TOGGLING:
3701                 break;
3702         /* SRC states */
3703         case SRC_UNATTACHED:
3704                 if (!port->non_pd_role_swap)
3705                         tcpm_swap_complete(port, -ENOTCONN);
3706                 tcpm_src_detach(port);
3707                 if (tcpm_start_toggling(port, tcpm_rp_cc(port))) {
3708                         tcpm_set_state(port, TOGGLING, 0);
3709                         break;
3710                 }
3711                 tcpm_set_cc(port, tcpm_rp_cc(port));
3712                 if (port->port_type == TYPEC_PORT_DRP)
3713                         tcpm_set_state(port, SNK_UNATTACHED, PD_T_DRP_SNK);
3714                 break;
3715         case SRC_ATTACH_WAIT:
3716                 if (tcpm_port_is_debug(port))
3717                         tcpm_set_state(port, DEBUG_ACC_ATTACHED,
3718                                        PD_T_CC_DEBOUNCE);
3719                 else if (tcpm_port_is_audio(port))
3720                         tcpm_set_state(port, AUDIO_ACC_ATTACHED,
3721                                        PD_T_CC_DEBOUNCE);
3722                 else if (tcpm_port_is_source(port) && port->vbus_vsafe0v)
3723                         tcpm_set_state(port,
3724                                        tcpm_try_snk(port) ? SNK_TRY
3725                                                           : SRC_ATTACHED,
3726                                        PD_T_CC_DEBOUNCE);
3727                 break;
3728
3729         case SNK_TRY:
3730                 port->try_snk_count++;
3731                 /*
3732                  * Requirements:
3733                  * - Do not drive vconn or vbus
3734                  * - Terminate CC pins (both) to Rd
3735                  * Action:
3736                  * - Wait for tDRPTry (PD_T_DRP_TRY).
3737                  *   Until then, ignore any state changes.
3738                  */
3739                 tcpm_set_cc(port, TYPEC_CC_RD);
3740                 tcpm_set_state(port, SNK_TRY_WAIT, PD_T_DRP_TRY);
3741                 break;
3742         case SNK_TRY_WAIT:
3743                 if (tcpm_port_is_sink(port)) {
3744                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE, 0);
3745                 } else {
3746                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3747                         port->max_wait = 0;
3748                 }
3749                 break;
3750         case SNK_TRY_WAIT_DEBOUNCE:
3751                 tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS,
3752                                PD_T_TRY_CC_DEBOUNCE);
3753                 break;
3754         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
3755                 if (port->vbus_present && tcpm_port_is_sink(port))
3756                         tcpm_set_state(port, SNK_ATTACHED, 0);
3757                 else
3758                         port->max_wait = 0;
3759                 break;
3760         case SRC_TRYWAIT:
3761                 tcpm_set_cc(port, tcpm_rp_cc(port));
3762                 if (port->max_wait == 0) {
3763                         port->max_wait = jiffies +
3764                                          msecs_to_jiffies(PD_T_DRP_TRY);
3765                         tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
3766                                        PD_T_DRP_TRY);
3767                 } else {
3768                         if (time_is_after_jiffies(port->max_wait))
3769                                 tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
3770                                                jiffies_to_msecs(port->max_wait -
3771                                                                 jiffies));
3772                         else
3773                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
3774                 }
3775                 break;
3776         case SRC_TRYWAIT_DEBOUNCE:
3777                 tcpm_set_state(port, SRC_ATTACHED, PD_T_CC_DEBOUNCE);
3778                 break;
3779         case SRC_TRYWAIT_UNATTACHED:
3780                 tcpm_set_state(port, SNK_UNATTACHED, 0);
3781                 break;
3782
3783         case SRC_ATTACHED:
3784                 ret = tcpm_src_attach(port);
3785                 tcpm_set_state(port, SRC_UNATTACHED,
3786                                ret < 0 ? 0 : PD_T_PS_SOURCE_ON);
3787                 break;
3788         case SRC_STARTUP:
3789                 opmode =  tcpm_get_pwr_opmode(tcpm_rp_cc(port));
3790                 typec_set_pwr_opmode(port->typec_port, opmode);
3791                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
3792                 port->caps_count = 0;
3793                 port->negotiated_rev = PD_MAX_REV;
3794                 port->message_id = 0;
3795                 port->rx_msgid = -1;
3796                 port->explicit_contract = false;
3797                 /* SNK -> SRC POWER/FAST_ROLE_SWAP finished */
3798                 if (port->ams == POWER_ROLE_SWAP ||
3799                     port->ams == FAST_ROLE_SWAP)
3800                         tcpm_ams_finish(port);
3801                 port->upcoming_state = SRC_SEND_CAPABILITIES;
3802                 tcpm_ams_start(port, POWER_NEGOTIATION);
3803                 break;
3804         case SRC_SEND_CAPABILITIES:
3805                 port->caps_count++;
3806                 if (port->caps_count > PD_N_CAPS_COUNT) {
3807                         tcpm_set_state(port, SRC_READY, 0);
3808                         break;
3809                 }
3810                 ret = tcpm_pd_send_source_caps(port);
3811                 if (ret < 0) {
3812                         tcpm_set_state(port, SRC_SEND_CAPABILITIES,
3813                                        PD_T_SEND_SOURCE_CAP);
3814                 } else {
3815                         /*
3816                          * Per standard, we should clear the reset counter here.
3817                          * However, that can result in state machine hang-ups.
3818                          * Reset it only in READY state to improve stability.
3819                          */
3820                         /* port->hard_reset_count = 0; */
3821                         port->caps_count = 0;
3822                         port->pd_capable = true;
3823                         tcpm_set_state_cond(port, SRC_SEND_CAPABILITIES_TIMEOUT,
3824                                             PD_T_SEND_SOURCE_CAP);
3825                 }
3826                 break;
3827         case SRC_SEND_CAPABILITIES_TIMEOUT:
3828                 /*
3829                  * Error recovery for a PD_DATA_SOURCE_CAP reply timeout.
3830                  *
3831                  * PD 2.0 sinks are supposed to accept src-capabilities with a
3832                  * 3.0 header and simply ignore any src PDOs which the sink does
3833                  * not understand such as PPS but some 2.0 sinks instead ignore
3834                  * the entire PD_DATA_SOURCE_CAP message, causing contract
3835                  * negotiation to fail.
3836                  *
3837                  * After PD_N_HARD_RESET_COUNT hard-reset attempts, we try
3838                  * sending src-capabilities with a lower PD revision to
3839                  * make these broken sinks work.
3840                  */
3841                 if (port->hard_reset_count < PD_N_HARD_RESET_COUNT) {
3842                         tcpm_set_state(port, HARD_RESET_SEND, 0);
3843                 } else if (port->negotiated_rev > PD_REV20) {
3844                         port->negotiated_rev--;
3845                         port->hard_reset_count = 0;
3846                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3847                 } else {
3848                         tcpm_set_state(port, hard_reset_state(port), 0);
3849                 }
3850                 break;
3851         case SRC_NEGOTIATE_CAPABILITIES:
3852                 ret = tcpm_pd_check_request(port);
3853                 if (ret < 0) {
3854                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
3855                         if (!port->explicit_contract) {
3856                                 tcpm_set_state(port,
3857                                                SRC_WAIT_NEW_CAPABILITIES, 0);
3858                         } else {
3859                                 tcpm_set_state(port, SRC_READY, 0);
3860                         }
3861                 } else {
3862                         tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3863                         tcpm_set_partner_usb_comm_capable(port,
3864                                                           !!(port->sink_request & RDO_USB_COMM));
3865                         tcpm_set_state(port, SRC_TRANSITION_SUPPLY,
3866                                        PD_T_SRC_TRANSITION);
3867                 }
3868                 break;
3869         case SRC_TRANSITION_SUPPLY:
3870                 /* XXX: regulator_set_voltage(vbus, ...) */
3871                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3872                 port->explicit_contract = true;
3873                 typec_set_pwr_opmode(port->typec_port, TYPEC_PWR_MODE_PD);
3874                 port->pwr_opmode = TYPEC_PWR_MODE_PD;
3875                 tcpm_set_state_cond(port, SRC_READY, 0);
3876                 break;
3877         case SRC_READY:
3878 #if 1
3879                 port->hard_reset_count = 0;
3880 #endif
3881                 port->try_src_count = 0;
3882
3883                 tcpm_swap_complete(port, 0);
3884                 tcpm_typec_connect(port);
3885
3886                 if (port->ams != NONE_AMS)
3887                         tcpm_ams_finish(port);
3888                 if (port->next_ams != NONE_AMS) {
3889                         port->ams = port->next_ams;
3890                         port->next_ams = NONE_AMS;
3891                 }
3892
3893                 /*
3894                  * If previous AMS is interrupted, switch to the upcoming
3895                  * state.
3896                  */
3897                 if (port->upcoming_state != INVALID_STATE) {
3898                         upcoming_state = port->upcoming_state;
3899                         port->upcoming_state = INVALID_STATE;
3900                         tcpm_set_state(port, upcoming_state, 0);
3901                         break;
3902                 }
3903
3904                 tcpm_check_send_discover(port);
3905                 /*
3906                  * 6.3.5
3907                  * Sending ping messages is not necessary if
3908                  * - the source operates at vSafe5V
3909                  * or
3910                  * - The system is not operating in PD mode
3911                  * or
3912                  * - Both partners are connected using a Type-C connector
3913                  *
3914                  * There is no actual need to send PD messages since the local
3915                  * port type-c and the spec does not clearly say whether PD is
3916                  * possible when type-c is connected to Type-A/B
3917                  */
3918                 break;
3919         case SRC_WAIT_NEW_CAPABILITIES:
3920                 /* Nothing to do... */
3921                 break;
3922
3923         /* SNK states */
3924         case SNK_UNATTACHED:
3925                 if (!port->non_pd_role_swap)
3926                         tcpm_swap_complete(port, -ENOTCONN);
3927                 tcpm_pps_complete(port, -ENOTCONN);
3928                 tcpm_snk_detach(port);
3929                 if (tcpm_start_toggling(port, TYPEC_CC_RD)) {
3930                         tcpm_set_state(port, TOGGLING, 0);
3931                         break;
3932                 }
3933                 tcpm_set_cc(port, TYPEC_CC_RD);
3934                 if (port->port_type == TYPEC_PORT_DRP)
3935                         tcpm_set_state(port, SRC_UNATTACHED, PD_T_DRP_SRC);
3936                 break;
3937         case SNK_ATTACH_WAIT:
3938                 if ((port->cc1 == TYPEC_CC_OPEN &&
3939                      port->cc2 != TYPEC_CC_OPEN) ||
3940                     (port->cc1 != TYPEC_CC_OPEN &&
3941                      port->cc2 == TYPEC_CC_OPEN))
3942                         tcpm_set_state(port, SNK_DEBOUNCED,
3943                                        PD_T_CC_DEBOUNCE);
3944                 else if (tcpm_port_is_disconnected(port))
3945                         tcpm_set_state(port, SNK_UNATTACHED,
3946                                        PD_T_PD_DEBOUNCE);
3947                 break;
3948         case SNK_DEBOUNCED:
3949                 if (tcpm_port_is_disconnected(port))
3950                         tcpm_set_state(port, SNK_UNATTACHED,
3951                                        PD_T_PD_DEBOUNCE);
3952                 else if (port->vbus_present)
3953                         tcpm_set_state(port,
3954                                        tcpm_try_src(port) ? SRC_TRY
3955                                                           : SNK_ATTACHED,
3956                                        0);
3957                 else
3958                         /* Wait for VBUS, but not forever */
3959                         tcpm_set_state(port, PORT_RESET, PD_T_PS_SOURCE_ON);
3960                 break;
3961
3962         case SRC_TRY:
3963                 port->try_src_count++;
3964                 tcpm_set_cc(port, tcpm_rp_cc(port));
3965                 port->max_wait = 0;
3966                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
3967                 break;
3968         case SRC_TRY_WAIT:
3969                 if (port->max_wait == 0) {
3970                         port->max_wait = jiffies +
3971                                          msecs_to_jiffies(PD_T_DRP_TRY);
3972                         msecs = PD_T_DRP_TRY;
3973                 } else {
3974                         if (time_is_after_jiffies(port->max_wait))
3975                                 msecs = jiffies_to_msecs(port->max_wait -
3976                                                          jiffies);
3977                         else
3978                                 msecs = 0;
3979                 }
3980                 tcpm_set_state(port, SNK_TRYWAIT, msecs);
3981                 break;
3982         case SRC_TRY_DEBOUNCE:
3983                 tcpm_set_state(port, SRC_ATTACHED, PD_T_PD_DEBOUNCE);
3984                 break;
3985         case SNK_TRYWAIT:
3986                 tcpm_set_cc(port, TYPEC_CC_RD);
3987                 tcpm_set_state(port, SNK_TRYWAIT_VBUS, PD_T_CC_DEBOUNCE);
3988                 break;
3989         case SNK_TRYWAIT_VBUS:
3990                 /*
3991                  * TCPM stays in this state indefinitely until VBUS
3992                  * is detected as long as Rp is not detected for
3993                  * more than a time period of tPDDebounce.
3994                  */
3995                 if (port->vbus_present && tcpm_port_is_sink(port)) {
3996                         tcpm_set_state(port, SNK_ATTACHED, 0);
3997                         break;
3998                 }
3999                 if (!tcpm_port_is_sink(port))
4000                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
4001                 break;
4002         case SNK_TRYWAIT_DEBOUNCE:
4003                 tcpm_set_state(port, SNK_UNATTACHED, PD_T_PD_DEBOUNCE);
4004                 break;
4005         case SNK_ATTACHED:
4006                 ret = tcpm_snk_attach(port);
4007                 if (ret < 0)
4008                         tcpm_set_state(port, SNK_UNATTACHED, 0);
4009                 else
4010                         tcpm_set_state(port, SNK_STARTUP, 0);
4011                 break;
4012         case SNK_STARTUP:
4013                 opmode =  tcpm_get_pwr_opmode(port->polarity ?
4014                                               port->cc2 : port->cc1);
4015                 typec_set_pwr_opmode(port->typec_port, opmode);
4016                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
4017                 port->negotiated_rev = PD_MAX_REV;
4018                 port->message_id = 0;
4019                 port->rx_msgid = -1;
4020                 port->explicit_contract = false;
4021
4022                 if (port->ams == POWER_ROLE_SWAP ||
4023                     port->ams == FAST_ROLE_SWAP)
4024                         /* SRC -> SNK POWER/FAST_ROLE_SWAP finished */
4025                         tcpm_ams_finish(port);
4026
4027                 tcpm_set_state(port, SNK_DISCOVERY, 0);
4028                 break;
4029         case SNK_DISCOVERY:
4030                 if (port->vbus_present) {
4031                         tcpm_set_current_limit(port,
4032                                                tcpm_get_current_limit(port),
4033                                                5000);
4034                         tcpm_set_charge(port, true);
4035                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
4036                         break;
4037                 }
4038                 /*
4039                  * For DRP, timeouts differ. Also, handling is supposed to be
4040                  * different and much more complex (dead battery detection;
4041                  * see USB power delivery specification, section 8.3.3.6.1.5.1).
4042                  */
4043                 tcpm_set_state(port, hard_reset_state(port),
4044                                port->port_type == TYPEC_PORT_DRP ?
4045                                         PD_T_DB_DETECT : PD_T_NO_RESPONSE);
4046                 break;
4047         case SNK_DISCOVERY_DEBOUNCE:
4048                 tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE_DONE,
4049                                PD_T_CC_DEBOUNCE);
4050                 break;
4051         case SNK_DISCOVERY_DEBOUNCE_DONE:
4052                 if (!tcpm_port_is_disconnected(port) &&
4053                     tcpm_port_is_sink(port) &&
4054                     ktime_after(port->delayed_runtime, ktime_get())) {
4055                         tcpm_set_state(port, SNK_DISCOVERY,
4056                                        ktime_to_ms(ktime_sub(port->delayed_runtime, ktime_get())));
4057                         break;
4058                 }
4059                 tcpm_set_state(port, unattached_state(port), 0);
4060                 break;
4061         case SNK_WAIT_CAPABILITIES:
4062                 ret = port->tcpc->set_pd_rx(port->tcpc, true);
4063                 if (ret < 0) {
4064                         tcpm_set_state(port, SNK_READY, 0);
4065                         break;
4066                 }
4067                 /*
4068                  * If VBUS has never been low, and we time out waiting
4069                  * for source cap, try a soft reset first, in case we
4070                  * were already in a stable contract before this boot.
4071                  * Do this only once.
4072                  */
4073                 if (port->vbus_never_low) {
4074                         port->vbus_never_low = false;
4075                         tcpm_set_state(port, SNK_SOFT_RESET,
4076                                        PD_T_SINK_WAIT_CAP);
4077                 } else {
4078                         tcpm_set_state(port, hard_reset_state(port),
4079                                        PD_T_SINK_WAIT_CAP);
4080                 }
4081                 break;
4082         case SNK_NEGOTIATE_CAPABILITIES:
4083                 port->pd_capable = true;
4084                 tcpm_set_partner_usb_comm_capable(port,
4085                                                   !!(port->source_caps[0] & PDO_FIXED_USB_COMM));
4086                 port->hard_reset_count = 0;
4087                 ret = tcpm_pd_send_request(port);
4088                 if (ret < 0) {
4089                         /* Let the Source send capabilities again. */
4090                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
4091                 } else {
4092                         tcpm_set_state_cond(port, hard_reset_state(port),
4093                                             PD_T_SENDER_RESPONSE);
4094                 }
4095                 break;
4096         case SNK_NEGOTIATE_PPS_CAPABILITIES:
4097                 ret = tcpm_pd_send_pps_request(port);
4098                 if (ret < 0) {
4099                         port->pps_status = ret;
4100                         /*
4101                          * If this was called due to updates to sink
4102                          * capabilities, and pps is no longer valid, we should
4103                          * safely fall back to a standard PDO.
4104                          */
4105                         if (port->update_sink_caps)
4106                                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4107                         else
4108                                 tcpm_set_state(port, SNK_READY, 0);
4109                 } else {
4110                         tcpm_set_state_cond(port, hard_reset_state(port),
4111                                             PD_T_SENDER_RESPONSE);
4112                 }
4113                 break;
4114         case SNK_TRANSITION_SINK:
4115         case SNK_TRANSITION_SINK_VBUS:
4116                 tcpm_set_state(port, hard_reset_state(port),
4117                                PD_T_PS_TRANSITION);
4118                 break;
4119         case SNK_READY:
4120                 port->try_snk_count = 0;
4121                 port->update_sink_caps = false;
4122                 if (port->explicit_contract) {
4123                         typec_set_pwr_opmode(port->typec_port,
4124                                              TYPEC_PWR_MODE_PD);
4125                         port->pwr_opmode = TYPEC_PWR_MODE_PD;
4126                 }
4127
4128                 tcpm_swap_complete(port, 0);
4129                 tcpm_typec_connect(port);
4130                 mod_enable_frs_delayed_work(port, 0);
4131                 tcpm_pps_complete(port, port->pps_status);
4132
4133                 if (port->ams != NONE_AMS)
4134                         tcpm_ams_finish(port);
4135                 if (port->next_ams != NONE_AMS) {
4136                         port->ams = port->next_ams;
4137                         port->next_ams = NONE_AMS;
4138                 }
4139
4140                 /*
4141                  * If previous AMS is interrupted, switch to the upcoming
4142                  * state.
4143                  */
4144                 if (port->upcoming_state != INVALID_STATE) {
4145                         upcoming_state = port->upcoming_state;
4146                         port->upcoming_state = INVALID_STATE;
4147                         tcpm_set_state(port, upcoming_state, 0);
4148                         break;
4149                 }
4150
4151                 tcpm_check_send_discover(port);
4152                 power_supply_changed(port->psy);
4153                 break;
4154
4155         /* Accessory states */
4156         case ACC_UNATTACHED:
4157                 tcpm_acc_detach(port);
4158                 tcpm_set_state(port, SRC_UNATTACHED, 0);
4159                 break;
4160         case DEBUG_ACC_ATTACHED:
4161         case AUDIO_ACC_ATTACHED:
4162                 ret = tcpm_acc_attach(port);
4163                 if (ret < 0)
4164                         tcpm_set_state(port, ACC_UNATTACHED, 0);
4165                 break;
4166         case AUDIO_ACC_DEBOUNCE:
4167                 tcpm_set_state(port, ACC_UNATTACHED, PD_T_CC_DEBOUNCE);
4168                 break;
4169
4170         /* Hard_Reset states */
4171         case HARD_RESET_SEND:
4172                 if (port->ams != NONE_AMS)
4173                         tcpm_ams_finish(port);
4174                 /*
4175                  * State machine will be directed to HARD_RESET_START,
4176                  * thus set upcoming_state to INVALID_STATE.
4177                  */
4178                 port->upcoming_state = INVALID_STATE;
4179                 tcpm_ams_start(port, HARD_RESET);
4180                 break;
4181         case HARD_RESET_START:
4182                 port->sink_cap_done = false;
4183                 if (port->tcpc->enable_frs)
4184                         port->tcpc->enable_frs(port->tcpc, false);
4185                 port->hard_reset_count++;
4186                 port->tcpc->set_pd_rx(port->tcpc, false);
4187                 tcpm_unregister_altmodes(port);
4188                 port->nr_sink_caps = 0;
4189                 port->send_discover = true;
4190                 if (port->pwr_role == TYPEC_SOURCE)
4191                         tcpm_set_state(port, SRC_HARD_RESET_VBUS_OFF,
4192                                        PD_T_PS_HARD_RESET);
4193                 else
4194                         tcpm_set_state(port, SNK_HARD_RESET_SINK_OFF, 0);
4195                 break;
4196         case SRC_HARD_RESET_VBUS_OFF:
4197                 /*
4198                  * 7.1.5 Response to Hard Resets
4199                  * Hard Reset Signaling indicates a communication failure has occurred and the
4200                  * Source Shall stop driving VCONN, Shall remove Rp from the VCONN pin and Shall
4201                  * drive VBUS to vSafe0V as shown in Figure 7-9.
4202                  */
4203                 tcpm_set_vconn(port, false);
4204                 tcpm_set_vbus(port, false);
4205                 tcpm_set_roles(port, port->self_powered, TYPEC_SOURCE,
4206                                tcpm_data_role_for_source(port));
4207                 /*
4208                  * If tcpc fails to notify vbus off, TCPM will wait for PD_T_SAFE_0V +
4209                  * PD_T_SRC_RECOVER before turning vbus back on.
4210                  * From Table 7-12 Sequence Description for a Source Initiated Hard Reset:
4211                  * 4. Policy Engine waits tPSHardReset after sending Hard Reset Signaling and then
4212                  * tells the Device Policy Manager to instruct the power supply to perform a
4213                  * Hard Reset. The transition to vSafe0V Shall occur within tSafe0V (t2).
4214                  * 5. After tSrcRecover the Source applies power to VBUS in an attempt to
4215                  * re-establish communication with the Sink and resume USB Default Operation.
4216                  * The transition to vSafe5V Shall occur within tSrcTurnOn(t4).
4217                  */
4218                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SAFE_0V + PD_T_SRC_RECOVER);
4219                 break;
4220         case SRC_HARD_RESET_VBUS_ON:
4221                 tcpm_set_vconn(port, true);
4222                 tcpm_set_vbus(port, true);
4223                 if (port->ams == HARD_RESET)
4224                         tcpm_ams_finish(port);
4225                 port->tcpc->set_pd_rx(port->tcpc, true);
4226                 tcpm_set_attached_state(port, true);
4227                 tcpm_set_state(port, SRC_UNATTACHED, PD_T_PS_SOURCE_ON);
4228                 break;
4229         case SNK_HARD_RESET_SINK_OFF:
4230                 /* Do not discharge/disconnect during hard reseet */
4231                 tcpm_set_auto_vbus_discharge_threshold(port, TYPEC_PWR_MODE_USB, false, 0);
4232                 memset(&port->pps_data, 0, sizeof(port->pps_data));
4233                 tcpm_set_vconn(port, false);
4234                 if (port->pd_capable)
4235                         tcpm_set_charge(port, false);
4236                 tcpm_set_roles(port, port->self_powered, TYPEC_SINK,
4237                                tcpm_data_role_for_sink(port));
4238                 /*
4239                  * VBUS may or may not toggle, depending on the adapter.
4240                  * If it doesn't toggle, transition to SNK_HARD_RESET_SINK_ON
4241                  * directly after timeout.
4242                  */
4243                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, PD_T_SAFE_0V);
4244                 break;
4245         case SNK_HARD_RESET_WAIT_VBUS:
4246                 if (port->ams == HARD_RESET)
4247                         tcpm_ams_finish(port);
4248                 /* Assume we're disconnected if VBUS doesn't come back. */
4249                 tcpm_set_state(port, SNK_UNATTACHED,
4250                                PD_T_SRC_RECOVER_MAX + PD_T_SRC_TURN_ON);
4251                 break;
4252         case SNK_HARD_RESET_SINK_ON:
4253                 /* Note: There is no guarantee that VBUS is on in this state */
4254                 /*
4255                  * XXX:
4256                  * The specification suggests that dual mode ports in sink
4257                  * mode should transition to state PE_SRC_Transition_to_default.
4258                  * See USB power delivery specification chapter 8.3.3.6.1.3.
4259                  * This would mean to to
4260                  * - turn off VCONN, reset power supply
4261                  * - request hardware reset
4262                  * - turn on VCONN
4263                  * - Transition to state PE_Src_Startup
4264                  * SNK only ports shall transition to state Snk_Startup
4265                  * (see chapter 8.3.3.3.8).
4266                  * Similar, dual-mode ports in source mode should transition
4267                  * to PE_SNK_Transition_to_default.
4268                  */
4269                 if (port->pd_capable) {
4270                         tcpm_set_current_limit(port,
4271                                                tcpm_get_current_limit(port),
4272                                                5000);
4273                         tcpm_set_charge(port, true);
4274                 }
4275                 if (port->ams == HARD_RESET)
4276                         tcpm_ams_finish(port);
4277                 tcpm_set_attached_state(port, true);
4278                 tcpm_set_auto_vbus_discharge_threshold(port, TYPEC_PWR_MODE_USB, false, VSAFE5V);
4279                 tcpm_set_state(port, SNK_STARTUP, 0);
4280                 break;
4281
4282         /* Soft_Reset states */
4283         case SOFT_RESET:
4284                 port->message_id = 0;
4285                 port->rx_msgid = -1;
4286                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
4287                 tcpm_ams_finish(port);
4288                 if (port->pwr_role == TYPEC_SOURCE) {
4289                         port->upcoming_state = SRC_SEND_CAPABILITIES;
4290                         tcpm_ams_start(port, POWER_NEGOTIATION);
4291                 } else {
4292                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
4293                 }
4294                 break;
4295         case SRC_SOFT_RESET_WAIT_SNK_TX:
4296         case SNK_SOFT_RESET:
4297                 if (port->ams != NONE_AMS)
4298                         tcpm_ams_finish(port);
4299                 port->upcoming_state = SOFT_RESET_SEND;
4300                 tcpm_ams_start(port, SOFT_RESET_AMS);
4301                 break;
4302         case SOFT_RESET_SEND:
4303                 port->message_id = 0;
4304                 port->rx_msgid = -1;
4305                 if (tcpm_pd_send_control(port, PD_CTRL_SOFT_RESET))
4306                         tcpm_set_state_cond(port, hard_reset_state(port), 0);
4307                 else
4308                         tcpm_set_state_cond(port, hard_reset_state(port),
4309                                             PD_T_SENDER_RESPONSE);
4310                 break;
4311
4312         /* DR_Swap states */
4313         case DR_SWAP_SEND:
4314                 tcpm_pd_send_control(port, PD_CTRL_DR_SWAP);
4315                 tcpm_set_state_cond(port, DR_SWAP_SEND_TIMEOUT,
4316                                     PD_T_SENDER_RESPONSE);
4317                 break;
4318         case DR_SWAP_ACCEPT:
4319                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
4320                 /* Set VDM state machine running flag ASAP */
4321                 if (port->data_role == TYPEC_DEVICE && port->send_discover)
4322                         port->vdm_sm_running = true;
4323                 tcpm_set_state_cond(port, DR_SWAP_CHANGE_DR, 0);
4324                 break;
4325         case DR_SWAP_SEND_TIMEOUT:
4326                 tcpm_swap_complete(port, -ETIMEDOUT);
4327                 tcpm_ams_finish(port);
4328                 tcpm_set_state(port, ready_state(port), 0);
4329                 break;
4330         case DR_SWAP_CHANGE_DR:
4331                 if (port->data_role == TYPEC_HOST) {
4332                         tcpm_unregister_altmodes(port);
4333                         tcpm_set_roles(port, true, port->pwr_role,
4334                                        TYPEC_DEVICE);
4335                 } else {
4336                         tcpm_set_roles(port, true, port->pwr_role,
4337                                        TYPEC_HOST);
4338                         port->send_discover = true;
4339                 }
4340                 tcpm_ams_finish(port);
4341                 tcpm_set_state(port, ready_state(port), 0);
4342                 break;
4343
4344         case FR_SWAP_SEND:
4345                 if (tcpm_pd_send_control(port, PD_CTRL_FR_SWAP)) {
4346                         tcpm_set_state(port, ERROR_RECOVERY, 0);
4347                         break;
4348                 }
4349                 tcpm_set_state_cond(port, FR_SWAP_SEND_TIMEOUT, PD_T_SENDER_RESPONSE);
4350                 break;
4351         case FR_SWAP_SEND_TIMEOUT:
4352                 tcpm_set_state(port, ERROR_RECOVERY, 0);
4353                 break;
4354         case FR_SWAP_SNK_SRC_TRANSITION_TO_OFF:
4355                 tcpm_set_state(port, ERROR_RECOVERY, PD_T_PS_SOURCE_OFF);
4356                 break;
4357         case FR_SWAP_SNK_SRC_NEW_SINK_READY:
4358                 if (port->vbus_source)
4359                         tcpm_set_state(port, FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED, 0);
4360                 else
4361                         tcpm_set_state(port, ERROR_RECOVERY, PD_T_RECEIVER_RESPONSE);
4362                 break;
4363         case FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED:
4364                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
4365                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
4366                         tcpm_set_state(port, ERROR_RECOVERY, 0);
4367                         break;
4368                 }
4369                 tcpm_set_cc(port, tcpm_rp_cc(port));
4370                 tcpm_set_state(port, SRC_STARTUP, PD_T_SWAP_SRC_START);
4371                 break;
4372
4373         /* PR_Swap states */
4374         case PR_SWAP_ACCEPT:
4375                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
4376                 tcpm_set_state(port, PR_SWAP_START, 0);
4377                 break;
4378         case PR_SWAP_SEND:
4379                 tcpm_pd_send_control(port, PD_CTRL_PR_SWAP);
4380                 tcpm_set_state_cond(port, PR_SWAP_SEND_TIMEOUT,
4381                                     PD_T_SENDER_RESPONSE);
4382                 break;
4383         case PR_SWAP_SEND_TIMEOUT:
4384                 tcpm_swap_complete(port, -ETIMEDOUT);
4385                 tcpm_set_state(port, ready_state(port), 0);
4386                 break;
4387         case PR_SWAP_START:
4388                 if (port->pwr_role == TYPEC_SOURCE)
4389                         tcpm_set_state(port, PR_SWAP_SRC_SNK_TRANSITION_OFF,
4390                                        PD_T_SRC_TRANSITION);
4391                 else
4392                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SINK_OFF, 0);
4393                 break;
4394         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
4395                 /*
4396                  * Prevent vbus discharge circuit from turning on during PR_SWAP
4397                  * as this is not a disconnect.
4398                  */
4399                 tcpm_set_vbus(port, false);
4400                 port->explicit_contract = false;
4401                 /* allow time for Vbus discharge, must be < tSrcSwapStdby */
4402                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF,
4403                                PD_T_SRCSWAPSTDBY);
4404                 break;
4405         case PR_SWAP_SRC_SNK_SOURCE_OFF:
4406                 tcpm_set_cc(port, TYPEC_CC_RD);
4407                 /* allow CC debounce */
4408                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED,
4409                                PD_T_CC_DEBOUNCE);
4410                 break;
4411         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
4412                 /*
4413                  * USB-PD standard, 6.2.1.4, Port Power Role:
4414                  * "During the Power Role Swap Sequence, for the initial Source
4415                  * Port, the Port Power Role field shall be set to Sink in the
4416                  * PS_RDY Message indicating that the initial Source’s power
4417                  * supply is turned off"
4418                  */
4419                 tcpm_set_pwr_role(port, TYPEC_SINK);
4420                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
4421                         tcpm_set_state(port, ERROR_RECOVERY, 0);
4422                         break;
4423                 }
4424                 tcpm_set_state(port, ERROR_RECOVERY, PD_T_PS_SOURCE_ON_PRS);
4425                 break;
4426         case PR_SWAP_SRC_SNK_SINK_ON:
4427                 /* Set the vbus disconnect threshold for implicit contract */
4428                 tcpm_set_auto_vbus_discharge_threshold(port, TYPEC_PWR_MODE_USB, false, VSAFE5V);
4429                 tcpm_set_state(port, SNK_STARTUP, 0);
4430                 break;
4431         case PR_SWAP_SNK_SRC_SINK_OFF:
4432                 /*
4433                  * Prevent vbus discharge circuit from turning on during PR_SWAP
4434                  * as this is not a disconnect.
4435                  */
4436                 tcpm_set_auto_vbus_discharge_threshold(port, TYPEC_PWR_MODE_USB,
4437                                                        port->pps_data.active, 0);
4438                 tcpm_set_charge(port, false);
4439                 tcpm_set_state(port, hard_reset_state(port),
4440                                PD_T_PS_SOURCE_OFF);
4441                 break;
4442         case PR_SWAP_SNK_SRC_SOURCE_ON:
4443                 tcpm_set_cc(port, tcpm_rp_cc(port));
4444                 tcpm_set_vbus(port, true);
4445                 /*
4446                  * allow time VBUS ramp-up, must be < tNewSrc
4447                  * Also, this window overlaps with CC debounce as well.
4448                  * So, Wait for the max of two which is PD_T_NEWSRC
4449                  */
4450                 tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP,
4451                                PD_T_NEWSRC);
4452                 break;
4453         case PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP:
4454                 /*
4455                  * USB PD standard, 6.2.1.4:
4456                  * "Subsequent Messages initiated by the Policy Engine,
4457                  * such as the PS_RDY Message sent to indicate that Vbus
4458                  * is ready, will have the Port Power Role field set to
4459                  * Source."
4460                  */
4461                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
4462                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
4463                 tcpm_set_state(port, SRC_STARTUP, PD_T_SWAP_SRC_START);
4464                 break;
4465
4466         case VCONN_SWAP_ACCEPT:
4467                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
4468                 tcpm_ams_finish(port);
4469                 tcpm_set_state(port, VCONN_SWAP_START, 0);
4470                 break;
4471         case VCONN_SWAP_SEND:
4472                 tcpm_pd_send_control(port, PD_CTRL_VCONN_SWAP);
4473                 tcpm_set_state(port, VCONN_SWAP_SEND_TIMEOUT,
4474                                PD_T_SENDER_RESPONSE);
4475                 break;
4476         case VCONN_SWAP_SEND_TIMEOUT:
4477                 tcpm_swap_complete(port, -ETIMEDOUT);
4478                 if (port->data_role == TYPEC_HOST && port->send_discover)
4479                         port->vdm_sm_running = true;
4480                 tcpm_set_state(port, ready_state(port), 0);
4481                 break;
4482         case VCONN_SWAP_START:
4483                 if (port->vconn_role == TYPEC_SOURCE)
4484                         tcpm_set_state(port, VCONN_SWAP_WAIT_FOR_VCONN, 0);
4485                 else
4486                         tcpm_set_state(port, VCONN_SWAP_TURN_ON_VCONN, 0);
4487                 break;
4488         case VCONN_SWAP_WAIT_FOR_VCONN:
4489                 tcpm_set_state(port, hard_reset_state(port),
4490                                PD_T_VCONN_SOURCE_ON);
4491                 break;
4492         case VCONN_SWAP_TURN_ON_VCONN:
4493                 tcpm_set_vconn(port, true);
4494                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
4495                 if (port->data_role == TYPEC_HOST && port->send_discover)
4496                         port->vdm_sm_running = true;
4497                 tcpm_set_state(port, ready_state(port), 0);
4498                 break;
4499         case VCONN_SWAP_TURN_OFF_VCONN:
4500                 tcpm_set_vconn(port, false);
4501                 if (port->data_role == TYPEC_HOST && port->send_discover)
4502                         port->vdm_sm_running = true;
4503                 tcpm_set_state(port, ready_state(port), 0);
4504                 break;
4505
4506         case DR_SWAP_CANCEL:
4507         case PR_SWAP_CANCEL:
4508         case VCONN_SWAP_CANCEL:
4509                 tcpm_swap_complete(port, port->swap_status);
4510                 if (port->data_role == TYPEC_HOST && port->send_discover)
4511                         port->vdm_sm_running = true;
4512                 if (port->pwr_role == TYPEC_SOURCE)
4513                         tcpm_set_state(port, SRC_READY, 0);
4514                 else
4515                         tcpm_set_state(port, SNK_READY, 0);
4516                 break;
4517         case FR_SWAP_CANCEL:
4518                 if (port->pwr_role == TYPEC_SOURCE)
4519                         tcpm_set_state(port, SRC_READY, 0);
4520                 else
4521                         tcpm_set_state(port, SNK_READY, 0);
4522                 break;
4523
4524         case BIST_RX:
4525                 switch (BDO_MODE_MASK(port->bist_request)) {
4526                 case BDO_MODE_CARRIER2:
4527                         tcpm_pd_transmit(port, TCPC_TX_BIST_MODE_2, NULL);
4528                         tcpm_set_state(port, unattached_state(port),
4529                                        PD_T_BIST_CONT_MODE);
4530                         break;
4531                 case BDO_MODE_TESTDATA:
4532                         if (port->tcpc->set_bist_data) {
4533                                 tcpm_log(port, "Enable BIST MODE TESTDATA");
4534                                 port->tcpc->set_bist_data(port->tcpc, true);
4535                         }
4536                         break;
4537                 default:
4538                         break;
4539                 }
4540                 break;
4541         case GET_STATUS_SEND:
4542                 tcpm_pd_send_control(port, PD_CTRL_GET_STATUS);
4543                 tcpm_set_state(port, GET_STATUS_SEND_TIMEOUT,
4544                                PD_T_SENDER_RESPONSE);
4545                 break;
4546         case GET_STATUS_SEND_TIMEOUT:
4547                 tcpm_set_state(port, ready_state(port), 0);
4548                 break;
4549         case GET_PPS_STATUS_SEND:
4550                 tcpm_pd_send_control(port, PD_CTRL_GET_PPS_STATUS);
4551                 tcpm_set_state(port, GET_PPS_STATUS_SEND_TIMEOUT,
4552                                PD_T_SENDER_RESPONSE);
4553                 break;
4554         case GET_PPS_STATUS_SEND_TIMEOUT:
4555                 tcpm_set_state(port, ready_state(port), 0);
4556                 break;
4557         case GET_SINK_CAP:
4558                 tcpm_pd_send_control(port, PD_CTRL_GET_SINK_CAP);
4559                 tcpm_set_state(port, GET_SINK_CAP_TIMEOUT, PD_T_SENDER_RESPONSE);
4560                 break;
4561         case GET_SINK_CAP_TIMEOUT:
4562                 port->sink_cap_done = true;
4563                 tcpm_set_state(port, ready_state(port), 0);
4564                 break;
4565         case ERROR_RECOVERY:
4566                 tcpm_swap_complete(port, -EPROTO);
4567                 tcpm_pps_complete(port, -EPROTO);
4568                 tcpm_set_state(port, PORT_RESET, 0);
4569                 break;
4570         case PORT_RESET:
4571                 tcpm_reset_port(port);
4572                 tcpm_set_cc(port, TYPEC_CC_OPEN);
4573                 tcpm_set_state(port, PORT_RESET_WAIT_OFF,
4574                                PD_T_ERROR_RECOVERY);
4575                 break;
4576         case PORT_RESET_WAIT_OFF:
4577                 tcpm_set_state(port,
4578                                tcpm_default_state(port),
4579                                port->vbus_present ? PD_T_PS_SOURCE_OFF : 0);
4580                 break;
4581
4582         /* AMS intermediate state */
4583         case AMS_START:
4584                 if (port->upcoming_state == INVALID_STATE) {
4585                         tcpm_set_state(port, port->pwr_role == TYPEC_SOURCE ?
4586                                        SRC_READY : SNK_READY, 0);
4587                         break;
4588                 }
4589
4590                 upcoming_state = port->upcoming_state;
4591                 port->upcoming_state = INVALID_STATE;
4592                 tcpm_set_state(port, upcoming_state, 0);
4593                 break;
4594
4595         /* Chunk state */
4596         case CHUNK_NOT_SUPP:
4597                 tcpm_pd_send_control(port, PD_CTRL_NOT_SUPP);
4598                 tcpm_set_state(port, port->pwr_role == TYPEC_SOURCE ? SRC_READY : SNK_READY, 0);
4599                 break;
4600         default:
4601                 WARN(1, "Unexpected port state %d\n", port->state);
4602                 break;
4603         }
4604 }
4605
4606 static void tcpm_state_machine_work(struct kthread_work *work)
4607 {
4608         struct tcpm_port *port = container_of(work, struct tcpm_port, state_machine);
4609         enum tcpm_state prev_state;
4610
4611         mutex_lock(&port->lock);
4612         port->state_machine_running = true;
4613
4614         if (port->queued_message && tcpm_send_queued_message(port))
4615                 goto done;
4616
4617         /* If we were queued due to a delayed state change, update it now */
4618         if (port->delayed_state) {
4619                 tcpm_log(port, "state change %s -> %s [delayed %ld ms]",
4620                          tcpm_states[port->state],
4621                          tcpm_states[port->delayed_state], port->delay_ms);
4622                 port->prev_state = port->state;
4623                 port->state = port->delayed_state;
4624                 port->delayed_state = INVALID_STATE;
4625         }
4626
4627         /*
4628          * Continue running as long as we have (non-delayed) state changes
4629          * to make.
4630          */
4631         do {
4632                 prev_state = port->state;
4633                 run_state_machine(port);
4634                 if (port->queued_message)
4635                         tcpm_send_queued_message(port);
4636         } while (port->state != prev_state && !port->delayed_state);
4637
4638 done:
4639         port->state_machine_running = false;
4640         mutex_unlock(&port->lock);
4641 }
4642
4643 static void _tcpm_cc_change(struct tcpm_port *port, enum typec_cc_status cc1,
4644                             enum typec_cc_status cc2)
4645 {
4646         enum typec_cc_status old_cc1, old_cc2;
4647         enum tcpm_state new_state;
4648
4649         old_cc1 = port->cc1;
4650         old_cc2 = port->cc2;
4651         port->cc1 = cc1;
4652         port->cc2 = cc2;
4653
4654         tcpm_log_force(port,
4655                        "CC1: %u -> %u, CC2: %u -> %u [state %s, polarity %d, %s]",
4656                        old_cc1, cc1, old_cc2, cc2, tcpm_states[port->state],
4657                        port->polarity,
4658                        tcpm_port_is_disconnected(port) ? "disconnected"
4659                                                        : "connected");
4660
4661         switch (port->state) {
4662         case TOGGLING:
4663                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
4664                     tcpm_port_is_source(port))
4665                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
4666                 else if (tcpm_port_is_sink(port))
4667                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
4668                 break;
4669         case SRC_UNATTACHED:
4670         case ACC_UNATTACHED:
4671                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
4672                     tcpm_port_is_source(port))
4673                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
4674                 break;
4675         case SRC_ATTACH_WAIT:
4676                 if (tcpm_port_is_disconnected(port) ||
4677                     tcpm_port_is_audio_detached(port))
4678                         tcpm_set_state(port, SRC_UNATTACHED, 0);
4679                 else if (cc1 != old_cc1 || cc2 != old_cc2)
4680                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
4681                 break;
4682         case SRC_ATTACHED:
4683         case SRC_SEND_CAPABILITIES:
4684         case SRC_READY:
4685                 if (tcpm_port_is_disconnected(port) ||
4686                     !tcpm_port_is_source(port)) {
4687                         if (port->port_type == TYPEC_PORT_SRC)
4688                                 tcpm_set_state(port, SRC_UNATTACHED, tcpm_wait_for_discharge(port));
4689                         else
4690                                 tcpm_set_state(port, SNK_UNATTACHED, tcpm_wait_for_discharge(port));
4691                 }
4692                 break;
4693         case SNK_UNATTACHED:
4694                 if (tcpm_port_is_sink(port))
4695                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
4696                 break;
4697         case SNK_ATTACH_WAIT:
4698                 if ((port->cc1 == TYPEC_CC_OPEN &&
4699                      port->cc2 != TYPEC_CC_OPEN) ||
4700                     (port->cc1 != TYPEC_CC_OPEN &&
4701                      port->cc2 == TYPEC_CC_OPEN))
4702                         new_state = SNK_DEBOUNCED;
4703                 else if (tcpm_port_is_disconnected(port))
4704                         new_state = SNK_UNATTACHED;
4705                 else
4706                         break;
4707                 if (new_state != port->delayed_state)
4708                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
4709                 break;
4710         case SNK_DEBOUNCED:
4711                 if (tcpm_port_is_disconnected(port))
4712                         new_state = SNK_UNATTACHED;
4713                 else if (port->vbus_present)
4714                         new_state = tcpm_try_src(port) ? SRC_TRY : SNK_ATTACHED;
4715                 else
4716                         new_state = SNK_UNATTACHED;
4717                 if (new_state != port->delayed_state)
4718                         tcpm_set_state(port, SNK_DEBOUNCED, 0);
4719                 break;
4720         case SNK_READY:
4721                 /*
4722                  * EXIT condition is based primarily on vbus disconnect and CC is secondary.
4723                  * "A port that has entered into USB PD communications with the Source and
4724                  * has seen the CC voltage exceed vRd-USB may monitor the CC pin to detect
4725                  * cable disconnect in addition to monitoring VBUS.
4726                  *
4727                  * A port that is monitoring the CC voltage for disconnect (but is not in
4728                  * the process of a USB PD PR_Swap or USB PD FR_Swap) shall transition to
4729                  * Unattached.SNK within tSinkDisconnect after the CC voltage remains below
4730                  * vRd-USB for tPDDebounce."
4731                  *
4732                  * When set_auto_vbus_discharge_threshold is enabled, CC pins go
4733                  * away before vbus decays to disconnect threshold. Allow
4734                  * disconnect to be driven by vbus disconnect when auto vbus
4735                  * discharge is enabled.
4736                  */
4737                 if (!port->auto_vbus_discharge_enabled && tcpm_port_is_disconnected(port))
4738                         tcpm_set_state(port, unattached_state(port), 0);
4739                 else if (!port->pd_capable &&
4740                          (cc1 != old_cc1 || cc2 != old_cc2))
4741                         tcpm_set_current_limit(port,
4742                                                tcpm_get_current_limit(port),
4743                                                5000);
4744                 break;
4745
4746         case AUDIO_ACC_ATTACHED:
4747                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
4748                         tcpm_set_state(port, AUDIO_ACC_DEBOUNCE, 0);
4749                 break;
4750         case AUDIO_ACC_DEBOUNCE:
4751                 if (tcpm_port_is_audio(port))
4752                         tcpm_set_state(port, AUDIO_ACC_ATTACHED, 0);
4753                 break;
4754
4755         case DEBUG_ACC_ATTACHED:
4756                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
4757                         tcpm_set_state(port, ACC_UNATTACHED, 0);
4758                 break;
4759
4760         case SNK_TRY:
4761                 /* Do nothing, waiting for timeout */
4762                 break;
4763
4764         case SNK_DISCOVERY:
4765                 /* CC line is unstable, wait for debounce */
4766                 if (tcpm_port_is_disconnected(port))
4767                         tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE, 0);
4768                 break;
4769         case SNK_DISCOVERY_DEBOUNCE:
4770                 break;
4771
4772         case SRC_TRYWAIT:
4773                 /* Hand over to state machine if needed */
4774                 if (!port->vbus_present && tcpm_port_is_source(port))
4775                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
4776                 break;
4777         case SRC_TRYWAIT_DEBOUNCE:
4778                 if (port->vbus_present || !tcpm_port_is_source(port))
4779                         tcpm_set_state(port, SRC_TRYWAIT, 0);
4780                 break;
4781         case SNK_TRY_WAIT_DEBOUNCE:
4782                 if (!tcpm_port_is_sink(port)) {
4783                         port->max_wait = 0;
4784                         tcpm_set_state(port, SRC_TRYWAIT, 0);
4785                 }
4786                 break;
4787         case SRC_TRY_WAIT:
4788                 if (tcpm_port_is_source(port))
4789                         tcpm_set_state(port, SRC_TRY_DEBOUNCE, 0);
4790                 break;
4791         case SRC_TRY_DEBOUNCE:
4792                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
4793                 break;
4794         case SNK_TRYWAIT_DEBOUNCE:
4795                 if (tcpm_port_is_sink(port))
4796                         tcpm_set_state(port, SNK_TRYWAIT_VBUS, 0);
4797                 break;
4798         case SNK_TRYWAIT_VBUS:
4799                 if (!tcpm_port_is_sink(port))
4800                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
4801                 break;
4802         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
4803                 if (!tcpm_port_is_sink(port))
4804                         tcpm_set_state(port, SRC_TRYWAIT, PD_T_TRY_CC_DEBOUNCE);
4805                 else
4806                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS, 0);
4807                 break;
4808         case SNK_TRYWAIT:
4809                 /* Do nothing, waiting for tCCDebounce */
4810                 break;
4811         case PR_SWAP_SNK_SRC_SINK_OFF:
4812         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
4813         case PR_SWAP_SRC_SNK_SOURCE_OFF:
4814         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
4815         case PR_SWAP_SNK_SRC_SOURCE_ON:
4816                 /*
4817                  * CC state change is expected in PR_SWAP
4818                  * Ignore it.
4819                  */
4820                 break;
4821         case FR_SWAP_SEND:
4822         case FR_SWAP_SEND_TIMEOUT:
4823         case FR_SWAP_SNK_SRC_TRANSITION_TO_OFF:
4824         case FR_SWAP_SNK_SRC_NEW_SINK_READY:
4825         case FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED:
4826                 /* Do nothing, CC change expected */
4827                 break;
4828
4829         case PORT_RESET:
4830         case PORT_RESET_WAIT_OFF:
4831                 /*
4832                  * State set back to default mode once the timer completes.
4833                  * Ignore CC changes here.
4834                  */
4835                 break;
4836         default:
4837                 /*
4838                  * While acting as sink and auto vbus discharge is enabled, Allow disconnect
4839                  * to be driven by vbus disconnect.
4840                  */
4841                 if (tcpm_port_is_disconnected(port) && !(port->pwr_role == TYPEC_SINK &&
4842                                                          port->auto_vbus_discharge_enabled))
4843                         tcpm_set_state(port, unattached_state(port), 0);
4844                 break;
4845         }
4846 }
4847
4848 static void _tcpm_pd_vbus_on(struct tcpm_port *port)
4849 {
4850         tcpm_log_force(port, "VBUS on");
4851         port->vbus_present = true;
4852         /*
4853          * When vbus_present is true i.e. Voltage at VBUS is greater than VSAFE5V implicitly
4854          * states that vbus is not at VSAFE0V, hence clear the vbus_vsafe0v flag here.
4855          */
4856         port->vbus_vsafe0v = false;
4857
4858         switch (port->state) {
4859         case SNK_TRANSITION_SINK_VBUS:
4860                 port->explicit_contract = true;
4861                 /* Set the VDM flag ASAP */
4862                 if (port->data_role == TYPEC_HOST && port->send_discover)
4863                         port->vdm_sm_running = true;
4864                 tcpm_set_state(port, SNK_READY, 0);
4865                 break;
4866         case SNK_DISCOVERY:
4867                 tcpm_set_state(port, SNK_DISCOVERY, 0);
4868                 break;
4869
4870         case SNK_DEBOUNCED:
4871                 tcpm_set_state(port, tcpm_try_src(port) ? SRC_TRY
4872                                                         : SNK_ATTACHED,
4873                                        0);
4874                 break;
4875         case SNK_HARD_RESET_WAIT_VBUS:
4876                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, 0);
4877                 break;
4878         case SRC_ATTACHED:
4879                 tcpm_set_state(port, SRC_STARTUP, 0);
4880                 break;
4881         case SRC_HARD_RESET_VBUS_ON:
4882                 tcpm_set_state(port, SRC_STARTUP, 0);
4883                 break;
4884
4885         case SNK_TRY:
4886                 /* Do nothing, waiting for timeout */
4887                 break;
4888         case SRC_TRYWAIT:
4889                 /* Do nothing, Waiting for Rd to be detected */
4890                 break;
4891         case SRC_TRYWAIT_DEBOUNCE:
4892                 tcpm_set_state(port, SRC_TRYWAIT, 0);
4893                 break;
4894         case SNK_TRY_WAIT_DEBOUNCE:
4895                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
4896                 break;
4897         case SNK_TRYWAIT:
4898                 /* Do nothing, waiting for tCCDebounce */
4899                 break;
4900         case SNK_TRYWAIT_VBUS:
4901                 if (tcpm_port_is_sink(port))
4902                         tcpm_set_state(port, SNK_ATTACHED, 0);
4903                 break;
4904         case SNK_TRYWAIT_DEBOUNCE:
4905                 /* Do nothing, waiting for Rp */
4906                 break;
4907         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
4908                 if (port->vbus_present && tcpm_port_is_sink(port))
4909                         tcpm_set_state(port, SNK_ATTACHED, 0);
4910                 break;
4911         case SRC_TRY_WAIT:
4912         case SRC_TRY_DEBOUNCE:
4913                 /* Do nothing, waiting for sink detection */
4914                 break;
4915         case FR_SWAP_SEND:
4916         case FR_SWAP_SEND_TIMEOUT:
4917         case FR_SWAP_SNK_SRC_TRANSITION_TO_OFF:
4918         case FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED:
4919                 if (port->tcpc->frs_sourcing_vbus)
4920                         port->tcpc->frs_sourcing_vbus(port->tcpc);
4921                 break;
4922         case FR_SWAP_SNK_SRC_NEW_SINK_READY:
4923                 if (port->tcpc->frs_sourcing_vbus)
4924                         port->tcpc->frs_sourcing_vbus(port->tcpc);
4925                 tcpm_set_state(port, FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED, 0);
4926                 break;
4927
4928         case PORT_RESET:
4929         case PORT_RESET_WAIT_OFF:
4930                 /*
4931                  * State set back to default mode once the timer completes.
4932                  * Ignore vbus changes here.
4933                  */
4934                 break;
4935
4936         default:
4937                 break;
4938         }
4939 }
4940
4941 static void _tcpm_pd_vbus_off(struct tcpm_port *port)
4942 {
4943         tcpm_log_force(port, "VBUS off");
4944         port->vbus_present = false;
4945         port->vbus_never_low = false;
4946         switch (port->state) {
4947         case SNK_HARD_RESET_SINK_OFF:
4948                 tcpm_set_state(port, SNK_HARD_RESET_WAIT_VBUS, 0);
4949                 break;
4950         case HARD_RESET_SEND:
4951                 break;
4952         case SNK_TRY:
4953                 /* Do nothing, waiting for timeout */
4954                 break;
4955         case SRC_TRYWAIT:
4956                 /* Hand over to state machine if needed */
4957                 if (tcpm_port_is_source(port))
4958                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
4959                 break;
4960         case SNK_TRY_WAIT_DEBOUNCE:
4961                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
4962                 break;
4963         case SNK_TRYWAIT:
4964         case SNK_TRYWAIT_VBUS:
4965         case SNK_TRYWAIT_DEBOUNCE:
4966                 break;
4967         case SNK_ATTACH_WAIT:
4968                 tcpm_set_state(port, SNK_UNATTACHED, 0);
4969                 break;
4970
4971         case SNK_NEGOTIATE_CAPABILITIES:
4972                 break;
4973
4974         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
4975                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF, 0);
4976                 break;
4977
4978         case PR_SWAP_SNK_SRC_SINK_OFF:
4979                 /* Do nothing, expected */
4980                 break;
4981
4982         case PR_SWAP_SNK_SRC_SOURCE_ON:
4983                 /*
4984                  * Do nothing when vbus off notification is received.
4985                  * TCPM can wait for PD_T_NEWSRC in PR_SWAP_SNK_SRC_SOURCE_ON
4986                  * for the vbus source to ramp up.
4987                  */
4988                 break;
4989
4990         case PORT_RESET_WAIT_OFF:
4991                 tcpm_set_state(port, tcpm_default_state(port), 0);
4992                 break;
4993
4994         case SRC_TRY_WAIT:
4995         case SRC_TRY_DEBOUNCE:
4996                 /* Do nothing, waiting for sink detection */
4997                 break;
4998
4999         case SRC_STARTUP:
5000         case SRC_SEND_CAPABILITIES:
5001         case SRC_SEND_CAPABILITIES_TIMEOUT:
5002         case SRC_NEGOTIATE_CAPABILITIES:
5003         case SRC_TRANSITION_SUPPLY:
5004         case SRC_READY:
5005         case SRC_WAIT_NEW_CAPABILITIES:
5006                 /*
5007                  * Force to unattached state to re-initiate connection.
5008                  * DRP port should move to Unattached.SNK instead of Unattached.SRC if
5009                  * sink removed. Although sink removal here is due to source's vbus collapse,
5010                  * treat it the same way for consistency.
5011                  */
5012                 if (port->port_type == TYPEC_PORT_SRC)
5013                         tcpm_set_state(port, SRC_UNATTACHED, tcpm_wait_for_discharge(port));
5014                 else
5015                         tcpm_set_state(port, SNK_UNATTACHED, tcpm_wait_for_discharge(port));
5016                 break;
5017
5018         case PORT_RESET:
5019                 /*
5020                  * State set back to default mode once the timer completes.
5021                  * Ignore vbus changes here.
5022                  */
5023                 break;
5024
5025         case FR_SWAP_SEND:
5026         case FR_SWAP_SEND_TIMEOUT:
5027         case FR_SWAP_SNK_SRC_TRANSITION_TO_OFF:
5028         case FR_SWAP_SNK_SRC_NEW_SINK_READY:
5029         case FR_SWAP_SNK_SRC_SOURCE_VBUS_APPLIED:
5030                 /* Do nothing, vbus drop expected */
5031                 break;
5032
5033         default:
5034                 if (port->pwr_role == TYPEC_SINK && port->attached)
5035                         tcpm_set_state(port, SNK_UNATTACHED, tcpm_wait_for_discharge(port));
5036                 break;
5037         }
5038 }
5039
5040 static void _tcpm_pd_vbus_vsafe0v(struct tcpm_port *port)
5041 {
5042         tcpm_log_force(port, "VBUS VSAFE0V");
5043         port->vbus_vsafe0v = true;
5044         switch (port->state) {
5045         case SRC_HARD_RESET_VBUS_OFF:
5046                 /*
5047                  * After establishing the vSafe0V voltage condition on VBUS, the Source Shall wait
5048                  * tSrcRecover before re-applying VCONN and restoring VBUS to vSafe5V.
5049                  */
5050                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SRC_RECOVER);
5051                 break;
5052         case SRC_ATTACH_WAIT:
5053                 if (tcpm_port_is_source(port))
5054                         tcpm_set_state(port, tcpm_try_snk(port) ? SNK_TRY : SRC_ATTACHED,
5055                                        PD_T_CC_DEBOUNCE);
5056                 break;
5057         case SRC_STARTUP:
5058         case SRC_SEND_CAPABILITIES:
5059         case SRC_SEND_CAPABILITIES_TIMEOUT:
5060         case SRC_NEGOTIATE_CAPABILITIES:
5061         case SRC_TRANSITION_SUPPLY:
5062         case SRC_READY:
5063         case SRC_WAIT_NEW_CAPABILITIES:
5064                 if (port->auto_vbus_discharge_enabled) {
5065                         if (port->port_type == TYPEC_PORT_SRC)
5066                                 tcpm_set_state(port, SRC_UNATTACHED, 0);
5067                         else
5068                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
5069                 }
5070                 break;
5071         default:
5072                 if (port->pwr_role == TYPEC_SINK && port->auto_vbus_discharge_enabled)
5073                         tcpm_set_state(port, SNK_UNATTACHED, 0);
5074                 break;
5075         }
5076 }
5077
5078 static void _tcpm_pd_hard_reset(struct tcpm_port *port)
5079 {
5080         tcpm_log_force(port, "Received hard reset");
5081         if (port->bist_request == BDO_MODE_TESTDATA && port->tcpc->set_bist_data)
5082                 port->tcpc->set_bist_data(port->tcpc, false);
5083
5084         if (port->ams != NONE_AMS)
5085                 port->ams = NONE_AMS;
5086         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
5087                 port->ams = HARD_RESET;
5088         /*
5089          * If we keep receiving hard reset requests, executing the hard reset
5090          * must have failed. Revert to error recovery if that happens.
5091          */
5092         tcpm_set_state(port,
5093                        port->hard_reset_count < PD_N_HARD_RESET_COUNT ?
5094                                 HARD_RESET_START : ERROR_RECOVERY,
5095                        0);
5096 }
5097
5098 static void tcpm_pd_event_handler(struct kthread_work *work)
5099 {
5100         struct tcpm_port *port = container_of(work, struct tcpm_port,
5101                                               event_work);
5102         u32 events;
5103
5104         mutex_lock(&port->lock);
5105
5106         spin_lock(&port->pd_event_lock);
5107         while (port->pd_events) {
5108                 events = port->pd_events;
5109                 port->pd_events = 0;
5110                 spin_unlock(&port->pd_event_lock);
5111                 if (events & TCPM_RESET_EVENT)
5112                         _tcpm_pd_hard_reset(port);
5113                 if (events & TCPM_VBUS_EVENT) {
5114                         bool vbus;
5115
5116                         vbus = port->tcpc->get_vbus(port->tcpc);
5117                         if (vbus) {
5118                                 _tcpm_pd_vbus_on(port);
5119                         } else {
5120                                 _tcpm_pd_vbus_off(port);
5121                                 /*
5122                                  * When TCPC does not support detecting vsafe0v voltage level,
5123                                  * treat vbus absent as vsafe0v. Else invoke is_vbus_vsafe0v
5124                                  * to see if vbus has discharge to VSAFE0V.
5125                                  */
5126                                 if (!port->tcpc->is_vbus_vsafe0v ||
5127                                     port->tcpc->is_vbus_vsafe0v(port->tcpc))
5128                                         _tcpm_pd_vbus_vsafe0v(port);
5129                         }
5130                 }
5131                 if (events & TCPM_CC_EVENT) {
5132                         enum typec_cc_status cc1, cc2;
5133
5134                         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
5135                                 _tcpm_cc_change(port, cc1, cc2);
5136                 }
5137                 if (events & TCPM_FRS_EVENT) {
5138                         if (port->state == SNK_READY) {
5139                                 int ret;
5140
5141                                 port->upcoming_state = FR_SWAP_SEND;
5142                                 ret = tcpm_ams_start(port, FAST_ROLE_SWAP);
5143                                 if (ret == -EAGAIN)
5144                                         port->upcoming_state = INVALID_STATE;
5145                         } else {
5146                                 tcpm_log(port, "Discarding FRS_SIGNAL! Not in sink ready");
5147                         }
5148                 }
5149                 if (events & TCPM_SOURCING_VBUS) {
5150                         tcpm_log(port, "sourcing vbus");
5151                         /*
5152                          * In fast role swap case TCPC autonomously sources vbus. Set vbus_source
5153                          * true as TCPM wouldn't have called tcpm_set_vbus.
5154                          *
5155                          * When vbus is sourced on the command on TCPM i.e. TCPM called
5156                          * tcpm_set_vbus to source vbus, vbus_source would already be true.
5157                          */
5158                         port->vbus_source = true;
5159                         _tcpm_pd_vbus_on(port);
5160                 }
5161
5162                 spin_lock(&port->pd_event_lock);
5163         }
5164         spin_unlock(&port->pd_event_lock);
5165         mutex_unlock(&port->lock);
5166 }
5167
5168 void tcpm_cc_change(struct tcpm_port *port)
5169 {
5170         spin_lock(&port->pd_event_lock);
5171         port->pd_events |= TCPM_CC_EVENT;
5172         spin_unlock(&port->pd_event_lock);
5173         kthread_queue_work(port->wq, &port->event_work);
5174 }
5175 EXPORT_SYMBOL_GPL(tcpm_cc_change);
5176
5177 void tcpm_vbus_change(struct tcpm_port *port)
5178 {
5179         spin_lock(&port->pd_event_lock);
5180         port->pd_events |= TCPM_VBUS_EVENT;
5181         spin_unlock(&port->pd_event_lock);
5182         kthread_queue_work(port->wq, &port->event_work);
5183 }
5184 EXPORT_SYMBOL_GPL(tcpm_vbus_change);
5185
5186 void tcpm_pd_hard_reset(struct tcpm_port *port)
5187 {
5188         spin_lock(&port->pd_event_lock);
5189         port->pd_events = TCPM_RESET_EVENT;
5190         spin_unlock(&port->pd_event_lock);
5191         kthread_queue_work(port->wq, &port->event_work);
5192 }
5193 EXPORT_SYMBOL_GPL(tcpm_pd_hard_reset);
5194
5195 void tcpm_sink_frs(struct tcpm_port *port)
5196 {
5197         spin_lock(&port->pd_event_lock);
5198         port->pd_events = TCPM_FRS_EVENT;
5199         spin_unlock(&port->pd_event_lock);
5200         kthread_queue_work(port->wq, &port->event_work);
5201 }
5202 EXPORT_SYMBOL_GPL(tcpm_sink_frs);
5203
5204 void tcpm_sourcing_vbus(struct tcpm_port *port)
5205 {
5206         spin_lock(&port->pd_event_lock);
5207         port->pd_events = TCPM_SOURCING_VBUS;
5208         spin_unlock(&port->pd_event_lock);
5209         kthread_queue_work(port->wq, &port->event_work);
5210 }
5211 EXPORT_SYMBOL_GPL(tcpm_sourcing_vbus);
5212
5213 static void tcpm_enable_frs_work(struct kthread_work *work)
5214 {
5215         struct tcpm_port *port = container_of(work, struct tcpm_port, enable_frs);
5216         int ret;
5217
5218         mutex_lock(&port->lock);
5219         /* Not FRS capable */
5220         if (!port->connected || port->port_type != TYPEC_PORT_DRP ||
5221             port->pwr_opmode != TYPEC_PWR_MODE_PD ||
5222             !port->tcpc->enable_frs ||
5223             /* Sink caps queried */
5224             port->sink_cap_done || port->negotiated_rev < PD_REV30)
5225                 goto unlock;
5226
5227         /* Send when the state machine is idle */
5228         if (port->state != SNK_READY || port->vdm_sm_running || port->send_discover)
5229                 goto resched;
5230
5231         port->upcoming_state = GET_SINK_CAP;
5232         ret = tcpm_ams_start(port, GET_SINK_CAPABILITIES);
5233         if (ret == -EAGAIN) {
5234                 port->upcoming_state = INVALID_STATE;
5235         } else {
5236                 port->sink_cap_done = true;
5237                 goto unlock;
5238         }
5239 resched:
5240         mod_enable_frs_delayed_work(port, GET_SINK_CAP_RETRY_MS);
5241 unlock:
5242         mutex_unlock(&port->lock);
5243 }
5244
5245 static int tcpm_dr_set(struct typec_port *p, enum typec_data_role data)
5246 {
5247         struct tcpm_port *port = typec_get_drvdata(p);
5248         int ret;
5249
5250         mutex_lock(&port->swap_lock);
5251         mutex_lock(&port->lock);
5252
5253         if (port->typec_caps.data != TYPEC_PORT_DRD) {
5254                 ret = -EINVAL;
5255                 goto port_unlock;
5256         }
5257         if (port->state != SRC_READY && port->state != SNK_READY) {
5258                 ret = -EAGAIN;
5259                 goto port_unlock;
5260         }
5261
5262         if (port->data_role == data) {
5263                 ret = 0;
5264                 goto port_unlock;
5265         }
5266
5267         /*
5268          * XXX
5269          * 6.3.9: If an alternate mode is active, a request to swap
5270          * alternate modes shall trigger a port reset.
5271          * Reject data role swap request in this case.
5272          */
5273
5274         if (!port->pd_capable) {
5275                 /*
5276                  * If the partner is not PD capable, reset the port to
5277                  * trigger a role change. This can only work if a preferred
5278                  * role is configured, and if it matches the requested role.
5279                  */
5280                 if (port->try_role == TYPEC_NO_PREFERRED_ROLE ||
5281                     port->try_role == port->pwr_role) {
5282                         ret = -EINVAL;
5283                         goto port_unlock;
5284                 }
5285                 port->non_pd_role_swap = true;
5286                 tcpm_set_state(port, PORT_RESET, 0);
5287         } else {
5288                 port->upcoming_state = DR_SWAP_SEND;
5289                 ret = tcpm_ams_start(port, DATA_ROLE_SWAP);
5290                 if (ret == -EAGAIN) {
5291                         port->upcoming_state = INVALID_STATE;
5292                         goto port_unlock;
5293                 }
5294         }
5295
5296         port->swap_status = 0;
5297         port->swap_pending = true;
5298         reinit_completion(&port->swap_complete);
5299         mutex_unlock(&port->lock);
5300
5301         if (!wait_for_completion_timeout(&port->swap_complete,
5302                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
5303                 ret = -ETIMEDOUT;
5304         else
5305                 ret = port->swap_status;
5306
5307         port->non_pd_role_swap = false;
5308         goto swap_unlock;
5309
5310 port_unlock:
5311         mutex_unlock(&port->lock);
5312 swap_unlock:
5313         mutex_unlock(&port->swap_lock);
5314         return ret;
5315 }
5316
5317 static int tcpm_pr_set(struct typec_port *p, enum typec_role role)
5318 {
5319         struct tcpm_port *port = typec_get_drvdata(p);
5320         int ret;
5321
5322         mutex_lock(&port->swap_lock);
5323         mutex_lock(&port->lock);
5324
5325         if (port->port_type != TYPEC_PORT_DRP) {
5326                 ret = -EINVAL;
5327                 goto port_unlock;
5328         }
5329         if (port->state != SRC_READY && port->state != SNK_READY) {
5330                 ret = -EAGAIN;
5331                 goto port_unlock;
5332         }
5333
5334         if (role == port->pwr_role) {
5335                 ret = 0;
5336                 goto port_unlock;
5337         }
5338
5339         port->upcoming_state = PR_SWAP_SEND;
5340         ret = tcpm_ams_start(port, POWER_ROLE_SWAP);
5341         if (ret == -EAGAIN) {
5342                 port->upcoming_state = INVALID_STATE;
5343                 goto port_unlock;
5344         }
5345
5346         port->swap_status = 0;
5347         port->swap_pending = true;
5348         reinit_completion(&port->swap_complete);
5349         mutex_unlock(&port->lock);
5350
5351         if (!wait_for_completion_timeout(&port->swap_complete,
5352                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
5353                 ret = -ETIMEDOUT;
5354         else
5355                 ret = port->swap_status;
5356
5357         goto swap_unlock;
5358
5359 port_unlock:
5360         mutex_unlock(&port->lock);
5361 swap_unlock:
5362         mutex_unlock(&port->swap_lock);
5363         return ret;
5364 }
5365
5366 static int tcpm_vconn_set(struct typec_port *p, enum typec_role role)
5367 {
5368         struct tcpm_port *port = typec_get_drvdata(p);
5369         int ret;
5370
5371         mutex_lock(&port->swap_lock);
5372         mutex_lock(&port->lock);
5373
5374         if (port->state != SRC_READY && port->state != SNK_READY) {
5375                 ret = -EAGAIN;
5376                 goto port_unlock;
5377         }
5378
5379         if (role == port->vconn_role) {
5380                 ret = 0;
5381                 goto port_unlock;
5382         }
5383
5384         port->upcoming_state = VCONN_SWAP_SEND;
5385         ret = tcpm_ams_start(port, VCONN_SWAP);
5386         if (ret == -EAGAIN) {
5387                 port->upcoming_state = INVALID_STATE;
5388                 goto port_unlock;
5389         }
5390
5391         port->swap_status = 0;
5392         port->swap_pending = true;
5393         reinit_completion(&port->swap_complete);
5394         mutex_unlock(&port->lock);
5395
5396         if (!wait_for_completion_timeout(&port->swap_complete,
5397                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
5398                 ret = -ETIMEDOUT;
5399         else
5400                 ret = port->swap_status;
5401
5402         goto swap_unlock;
5403
5404 port_unlock:
5405         mutex_unlock(&port->lock);
5406 swap_unlock:
5407         mutex_unlock(&port->swap_lock);
5408         return ret;
5409 }
5410
5411 static int tcpm_try_role(struct typec_port *p, int role)
5412 {
5413         struct tcpm_port *port = typec_get_drvdata(p);
5414         struct tcpc_dev *tcpc = port->tcpc;
5415         int ret = 0;
5416
5417         mutex_lock(&port->lock);
5418         if (tcpc->try_role)
5419                 ret = tcpc->try_role(tcpc, role);
5420         if (!ret)
5421                 port->try_role = role;
5422         port->try_src_count = 0;
5423         port->try_snk_count = 0;
5424         mutex_unlock(&port->lock);
5425
5426         return ret;
5427 }
5428
5429 static int tcpm_pps_set_op_curr(struct tcpm_port *port, u16 op_curr)
5430 {
5431         unsigned int target_mw;
5432         int ret;
5433
5434         mutex_lock(&port->swap_lock);
5435         mutex_lock(&port->lock);
5436
5437         if (!port->pps_data.active) {
5438                 ret = -EOPNOTSUPP;
5439                 goto port_unlock;
5440         }
5441
5442         if (port->state != SNK_READY) {
5443                 ret = -EAGAIN;
5444                 goto port_unlock;
5445         }
5446
5447         if (op_curr > port->pps_data.max_curr) {
5448                 ret = -EINVAL;
5449                 goto port_unlock;
5450         }
5451
5452         target_mw = (op_curr * port->pps_data.out_volt) / 1000;
5453         if (target_mw < port->operating_snk_mw) {
5454                 ret = -EINVAL;
5455                 goto port_unlock;
5456         }
5457
5458         port->upcoming_state = SNK_NEGOTIATE_PPS_CAPABILITIES;
5459         ret = tcpm_ams_start(port, POWER_NEGOTIATION);
5460         if (ret == -EAGAIN) {
5461                 port->upcoming_state = INVALID_STATE;
5462                 goto port_unlock;
5463         }
5464
5465         /* Round down operating current to align with PPS valid steps */
5466         op_curr = op_curr - (op_curr % RDO_PROG_CURR_MA_STEP);
5467
5468         reinit_completion(&port->pps_complete);
5469         port->pps_data.op_curr = op_curr;
5470         port->pps_status = 0;
5471         port->pps_pending = true;
5472         mutex_unlock(&port->lock);
5473
5474         if (!wait_for_completion_timeout(&port->pps_complete,
5475                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
5476                 ret = -ETIMEDOUT;
5477         else
5478                 ret = port->pps_status;
5479
5480         goto swap_unlock;
5481
5482 port_unlock:
5483         mutex_unlock(&port->lock);
5484 swap_unlock:
5485         mutex_unlock(&port->swap_lock);
5486
5487         return ret;
5488 }
5489
5490 static int tcpm_pps_set_out_volt(struct tcpm_port *port, u16 out_volt)
5491 {
5492         unsigned int target_mw;
5493         int ret;
5494
5495         mutex_lock(&port->swap_lock);
5496         mutex_lock(&port->lock);
5497
5498         if (!port->pps_data.active) {
5499                 ret = -EOPNOTSUPP;
5500                 goto port_unlock;
5501         }
5502
5503         if (port->state != SNK_READY) {
5504                 ret = -EAGAIN;
5505                 goto port_unlock;
5506         }
5507
5508         if (out_volt < port->pps_data.min_volt ||
5509             out_volt > port->pps_data.max_volt) {
5510                 ret = -EINVAL;
5511                 goto port_unlock;
5512         }
5513
5514         target_mw = (port->pps_data.op_curr * out_volt) / 1000;
5515         if (target_mw < port->operating_snk_mw) {
5516                 ret = -EINVAL;
5517                 goto port_unlock;
5518         }
5519
5520         port->upcoming_state = SNK_NEGOTIATE_PPS_CAPABILITIES;
5521         ret = tcpm_ams_start(port, POWER_NEGOTIATION);
5522         if (ret == -EAGAIN) {
5523                 port->upcoming_state = INVALID_STATE;
5524                 goto port_unlock;
5525         }
5526
5527         /* Round down output voltage to align with PPS valid steps */
5528         out_volt = out_volt - (out_volt % RDO_PROG_VOLT_MV_STEP);
5529
5530         reinit_completion(&port->pps_complete);
5531         port->pps_data.out_volt = out_volt;
5532         port->pps_status = 0;
5533         port->pps_pending = true;
5534         mutex_unlock(&port->lock);
5535
5536         if (!wait_for_completion_timeout(&port->pps_complete,
5537                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
5538                 ret = -ETIMEDOUT;
5539         else
5540                 ret = port->pps_status;
5541
5542         goto swap_unlock;
5543
5544 port_unlock:
5545         mutex_unlock(&port->lock);
5546 swap_unlock:
5547         mutex_unlock(&port->swap_lock);
5548
5549         return ret;
5550 }
5551
5552 static int tcpm_pps_activate(struct tcpm_port *port, bool activate)
5553 {
5554         int ret = 0;
5555
5556         mutex_lock(&port->swap_lock);
5557         mutex_lock(&port->lock);
5558
5559         if (!port->pps_data.supported) {
5560                 ret = -EOPNOTSUPP;
5561                 goto port_unlock;
5562         }
5563
5564         /* Trying to deactivate PPS when already deactivated so just bail */
5565         if (!port->pps_data.active && !activate)
5566                 goto port_unlock;
5567
5568         if (port->state != SNK_READY) {
5569                 ret = -EAGAIN;
5570                 goto port_unlock;
5571         }
5572
5573         if (activate)
5574                 port->upcoming_state = SNK_NEGOTIATE_PPS_CAPABILITIES;
5575         else
5576                 port->upcoming_state = SNK_NEGOTIATE_CAPABILITIES;
5577         ret = tcpm_ams_start(port, POWER_NEGOTIATION);
5578         if (ret == -EAGAIN) {
5579                 port->upcoming_state = INVALID_STATE;
5580                 goto port_unlock;
5581         }
5582
5583         reinit_completion(&port->pps_complete);
5584         port->pps_status = 0;
5585         port->pps_pending = true;
5586
5587         /* Trigger PPS request or move back to standard PDO contract */
5588         if (activate) {
5589                 port->pps_data.out_volt = port->supply_voltage;
5590                 port->pps_data.op_curr = port->current_limit;
5591         }
5592         mutex_unlock(&port->lock);
5593
5594         if (!wait_for_completion_timeout(&port->pps_complete,
5595                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
5596                 ret = -ETIMEDOUT;
5597         else
5598                 ret = port->pps_status;
5599
5600         goto swap_unlock;
5601
5602 port_unlock:
5603         mutex_unlock(&port->lock);
5604 swap_unlock:
5605         mutex_unlock(&port->swap_lock);
5606
5607         return ret;
5608 }
5609
5610 static void tcpm_init(struct tcpm_port *port)
5611 {
5612         enum typec_cc_status cc1, cc2;
5613
5614         port->tcpc->init(port->tcpc);
5615
5616         tcpm_reset_port(port);
5617
5618         /*
5619          * XXX
5620          * Should possibly wait for VBUS to settle if it was enabled locally
5621          * since tcpm_reset_port() will disable VBUS.
5622          */
5623         port->vbus_present = port->tcpc->get_vbus(port->tcpc);
5624         if (port->vbus_present)
5625                 port->vbus_never_low = true;
5626
5627         /*
5628          * 1. When vbus_present is true, voltage on VBUS is already at VSAFE5V.
5629          * So implicitly vbus_vsafe0v = false.
5630          *
5631          * 2. When vbus_present is false and TCPC does NOT support querying
5632          * vsafe0v status, then, it's best to assume vbus is at VSAFE0V i.e.
5633          * vbus_vsafe0v is true.
5634          *
5635          * 3. When vbus_present is false and TCPC does support querying vsafe0v,
5636          * then, query tcpc for vsafe0v status.
5637          */
5638         if (port->vbus_present)
5639                 port->vbus_vsafe0v = false;
5640         else if (!port->tcpc->is_vbus_vsafe0v)
5641                 port->vbus_vsafe0v = true;
5642         else
5643                 port->vbus_vsafe0v = port->tcpc->is_vbus_vsafe0v(port->tcpc);
5644
5645         tcpm_set_state(port, tcpm_default_state(port), 0);
5646
5647         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
5648                 _tcpm_cc_change(port, cc1, cc2);
5649
5650         /*
5651          * Some adapters need a clean slate at startup, and won't recover
5652          * otherwise. So do not try to be fancy and force a clean disconnect.
5653          */
5654         tcpm_set_state(port, PORT_RESET, 0);
5655 }
5656
5657 static int tcpm_port_type_set(struct typec_port *p, enum typec_port_type type)
5658 {
5659         struct tcpm_port *port = typec_get_drvdata(p);
5660
5661         mutex_lock(&port->lock);
5662         if (type == port->port_type)
5663                 goto port_unlock;
5664
5665         port->port_type = type;
5666
5667         if (!port->connected) {
5668                 tcpm_set_state(port, PORT_RESET, 0);
5669         } else if (type == TYPEC_PORT_SNK) {
5670                 if (!(port->pwr_role == TYPEC_SINK &&
5671                       port->data_role == TYPEC_DEVICE))
5672                         tcpm_set_state(port, PORT_RESET, 0);
5673         } else if (type == TYPEC_PORT_SRC) {
5674                 if (!(port->pwr_role == TYPEC_SOURCE &&
5675                       port->data_role == TYPEC_HOST))
5676                         tcpm_set_state(port, PORT_RESET, 0);
5677         }
5678
5679 port_unlock:
5680         mutex_unlock(&port->lock);
5681         return 0;
5682 }
5683
5684 static const struct typec_operations tcpm_ops = {
5685         .try_role = tcpm_try_role,
5686         .dr_set = tcpm_dr_set,
5687         .pr_set = tcpm_pr_set,
5688         .vconn_set = tcpm_vconn_set,
5689         .port_type_set = tcpm_port_type_set
5690 };
5691
5692 void tcpm_tcpc_reset(struct tcpm_port *port)
5693 {
5694         mutex_lock(&port->lock);
5695         /* XXX: Maintain PD connection if possible? */
5696         tcpm_init(port);
5697         mutex_unlock(&port->lock);
5698 }
5699 EXPORT_SYMBOL_GPL(tcpm_tcpc_reset);
5700
5701 static int tcpm_fw_get_caps(struct tcpm_port *port,
5702                             struct fwnode_handle *fwnode)
5703 {
5704         const char *cap_str;
5705         int ret;
5706         u32 mw, frs_current;
5707
5708         if (!fwnode)
5709                 return -EINVAL;
5710
5711         /* USB data support is optional */
5712         ret = fwnode_property_read_string(fwnode, "data-role", &cap_str);
5713         if (ret == 0) {
5714                 ret = typec_find_port_data_role(cap_str);
5715                 if (ret < 0)
5716                         return ret;
5717                 port->typec_caps.data = ret;
5718         }
5719
5720         ret = fwnode_property_read_string(fwnode, "power-role", &cap_str);
5721         if (ret < 0)
5722                 return ret;
5723
5724         ret = typec_find_port_power_role(cap_str);
5725         if (ret < 0)
5726                 return ret;
5727         port->typec_caps.type = ret;
5728         port->port_type = port->typec_caps.type;
5729
5730         if (port->port_type == TYPEC_PORT_SNK)
5731                 goto sink;
5732
5733         /* Get source pdos */
5734         ret = fwnode_property_count_u32(fwnode, "source-pdos");
5735         if (ret <= 0)
5736                 return -EINVAL;
5737
5738         port->nr_src_pdo = min(ret, PDO_MAX_OBJECTS);
5739         ret = fwnode_property_read_u32_array(fwnode, "source-pdos",
5740                                              port->src_pdo, port->nr_src_pdo);
5741         if ((ret < 0) || tcpm_validate_caps(port, port->src_pdo,
5742                                             port->nr_src_pdo))
5743                 return -EINVAL;
5744
5745         if (port->port_type == TYPEC_PORT_SRC)
5746                 return 0;
5747
5748         /* Get the preferred power role for DRP */
5749         ret = fwnode_property_read_string(fwnode, "try-power-role", &cap_str);
5750         if (ret < 0)
5751                 return ret;
5752
5753         port->typec_caps.prefer_role = typec_find_power_role(cap_str);
5754         if (port->typec_caps.prefer_role < 0)
5755                 return -EINVAL;
5756 sink:
5757         /* Get sink pdos */
5758         ret = fwnode_property_count_u32(fwnode, "sink-pdos");
5759         if (ret <= 0)
5760                 return -EINVAL;
5761
5762         port->nr_snk_pdo = min(ret, PDO_MAX_OBJECTS);
5763         ret = fwnode_property_read_u32_array(fwnode, "sink-pdos",
5764                                              port->snk_pdo, port->nr_snk_pdo);
5765         if ((ret < 0) || tcpm_validate_caps(port, port->snk_pdo,
5766                                             port->nr_snk_pdo))
5767                 return -EINVAL;
5768
5769         if (fwnode_property_read_u32(fwnode, "op-sink-microwatt", &mw) < 0)
5770                 return -EINVAL;
5771         port->operating_snk_mw = mw / 1000;
5772
5773         port->self_powered = fwnode_property_read_bool(fwnode, "self-powered");
5774
5775         /* FRS can only be supported byb DRP ports */
5776         if (port->port_type == TYPEC_PORT_DRP) {
5777                 ret = fwnode_property_read_u32(fwnode, "new-source-frs-typec-current",
5778                                                &frs_current);
5779                 if (ret >= 0 && frs_current <= FRS_5V_3A)
5780                         port->new_source_frs_current = frs_current;
5781         }
5782
5783         /* sink-vdos is optional */
5784         ret = fwnode_property_count_u32(fwnode, "sink-vdos");
5785         if (ret < 0)
5786                 ret = 0;
5787
5788         port->nr_snk_vdo = min(ret, VDO_MAX_OBJECTS);
5789         if (port->nr_snk_vdo) {
5790                 ret = fwnode_property_read_u32_array(fwnode, "sink-vdos",
5791                                                      port->snk_vdo,
5792                                                      port->nr_snk_vdo);
5793                 if (ret < 0)
5794                         return ret;
5795         }
5796
5797         return 0;
5798 }
5799
5800 /* Power Supply access to expose source power information */
5801 enum tcpm_psy_online_states {
5802         TCPM_PSY_OFFLINE = 0,
5803         TCPM_PSY_FIXED_ONLINE,
5804         TCPM_PSY_PROG_ONLINE,
5805 };
5806
5807 static enum power_supply_property tcpm_psy_props[] = {
5808         POWER_SUPPLY_PROP_USB_TYPE,
5809         POWER_SUPPLY_PROP_ONLINE,
5810         POWER_SUPPLY_PROP_VOLTAGE_MIN,
5811         POWER_SUPPLY_PROP_VOLTAGE_MAX,
5812         POWER_SUPPLY_PROP_VOLTAGE_NOW,
5813         POWER_SUPPLY_PROP_CURRENT_MAX,
5814         POWER_SUPPLY_PROP_CURRENT_NOW,
5815 };
5816
5817 static int tcpm_psy_get_online(struct tcpm_port *port,
5818                                union power_supply_propval *val)
5819 {
5820         if (port->vbus_charge) {
5821                 if (port->pps_data.active)
5822                         val->intval = TCPM_PSY_PROG_ONLINE;
5823                 else
5824                         val->intval = TCPM_PSY_FIXED_ONLINE;
5825         } else {
5826                 val->intval = TCPM_PSY_OFFLINE;
5827         }
5828
5829         return 0;
5830 }
5831
5832 static int tcpm_psy_get_voltage_min(struct tcpm_port *port,
5833                                     union power_supply_propval *val)
5834 {
5835         if (port->pps_data.active)
5836                 val->intval = port->pps_data.min_volt * 1000;
5837         else
5838                 val->intval = port->supply_voltage * 1000;
5839
5840         return 0;
5841 }
5842
5843 static int tcpm_psy_get_voltage_max(struct tcpm_port *port,
5844                                     union power_supply_propval *val)
5845 {
5846         if (port->pps_data.active)
5847                 val->intval = port->pps_data.max_volt * 1000;
5848         else
5849                 val->intval = port->supply_voltage * 1000;
5850
5851         return 0;
5852 }
5853
5854 static int tcpm_psy_get_voltage_now(struct tcpm_port *port,
5855                                     union power_supply_propval *val)
5856 {
5857         val->intval = port->supply_voltage * 1000;
5858
5859         return 0;
5860 }
5861
5862 static int tcpm_psy_get_current_max(struct tcpm_port *port,
5863                                     union power_supply_propval *val)
5864 {
5865         if (port->pps_data.active)
5866                 val->intval = port->pps_data.max_curr * 1000;
5867         else
5868                 val->intval = port->current_limit * 1000;
5869
5870         return 0;
5871 }
5872
5873 static int tcpm_psy_get_current_now(struct tcpm_port *port,
5874                                     union power_supply_propval *val)
5875 {
5876         val->intval = port->current_limit * 1000;
5877
5878         return 0;
5879 }
5880
5881 static int tcpm_psy_get_prop(struct power_supply *psy,
5882                              enum power_supply_property psp,
5883                              union power_supply_propval *val)
5884 {
5885         struct tcpm_port *port = power_supply_get_drvdata(psy);
5886         int ret = 0;
5887
5888         switch (psp) {
5889         case POWER_SUPPLY_PROP_USB_TYPE:
5890                 val->intval = port->usb_type;
5891                 break;
5892         case POWER_SUPPLY_PROP_ONLINE:
5893                 ret = tcpm_psy_get_online(port, val);
5894                 break;
5895         case POWER_SUPPLY_PROP_VOLTAGE_MIN:
5896                 ret = tcpm_psy_get_voltage_min(port, val);
5897                 break;
5898         case POWER_SUPPLY_PROP_VOLTAGE_MAX:
5899                 ret = tcpm_psy_get_voltage_max(port, val);
5900                 break;
5901         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
5902                 ret = tcpm_psy_get_voltage_now(port, val);
5903                 break;
5904         case POWER_SUPPLY_PROP_CURRENT_MAX:
5905                 ret = tcpm_psy_get_current_max(port, val);
5906                 break;
5907         case POWER_SUPPLY_PROP_CURRENT_NOW:
5908                 ret = tcpm_psy_get_current_now(port, val);
5909                 break;
5910         default:
5911                 ret = -EINVAL;
5912                 break;
5913         }
5914
5915         return ret;
5916 }
5917
5918 static int tcpm_psy_set_online(struct tcpm_port *port,
5919                                const union power_supply_propval *val)
5920 {
5921         int ret;
5922
5923         switch (val->intval) {
5924         case TCPM_PSY_FIXED_ONLINE:
5925                 ret = tcpm_pps_activate(port, false);
5926                 break;
5927         case TCPM_PSY_PROG_ONLINE:
5928                 ret = tcpm_pps_activate(port, true);
5929                 break;
5930         default:
5931                 ret = -EINVAL;
5932                 break;
5933         }
5934
5935         return ret;
5936 }
5937
5938 static int tcpm_psy_set_prop(struct power_supply *psy,
5939                              enum power_supply_property psp,
5940                              const union power_supply_propval *val)
5941 {
5942         struct tcpm_port *port = power_supply_get_drvdata(psy);
5943         int ret;
5944
5945         switch (psp) {
5946         case POWER_SUPPLY_PROP_ONLINE:
5947                 ret = tcpm_psy_set_online(port, val);
5948                 break;
5949         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
5950                 if (val->intval < port->pps_data.min_volt * 1000 ||
5951                     val->intval > port->pps_data.max_volt * 1000)
5952                         ret = -EINVAL;
5953                 else
5954                         ret = tcpm_pps_set_out_volt(port, val->intval / 1000);
5955                 break;
5956         case POWER_SUPPLY_PROP_CURRENT_NOW:
5957                 if (val->intval > port->pps_data.max_curr * 1000)
5958                         ret = -EINVAL;
5959                 else
5960                         ret = tcpm_pps_set_op_curr(port, val->intval / 1000);
5961                 break;
5962         default:
5963                 ret = -EINVAL;
5964                 break;
5965         }
5966         power_supply_changed(port->psy);
5967         return ret;
5968 }
5969
5970 static int tcpm_psy_prop_writeable(struct power_supply *psy,
5971                                    enum power_supply_property psp)
5972 {
5973         switch (psp) {
5974         case POWER_SUPPLY_PROP_ONLINE:
5975         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
5976         case POWER_SUPPLY_PROP_CURRENT_NOW:
5977                 return 1;
5978         default:
5979                 return 0;
5980         }
5981 }
5982
5983 static enum power_supply_usb_type tcpm_psy_usb_types[] = {
5984         POWER_SUPPLY_USB_TYPE_C,
5985         POWER_SUPPLY_USB_TYPE_PD,
5986         POWER_SUPPLY_USB_TYPE_PD_PPS,
5987 };
5988
5989 static const char *tcpm_psy_name_prefix = "tcpm-source-psy-";
5990
5991 static int devm_tcpm_psy_register(struct tcpm_port *port)
5992 {
5993         struct power_supply_config psy_cfg = {};
5994         const char *port_dev_name = dev_name(port->dev);
5995         size_t psy_name_len = strlen(tcpm_psy_name_prefix) +
5996                                      strlen(port_dev_name) + 1;
5997         char *psy_name;
5998
5999         psy_cfg.drv_data = port;
6000         psy_cfg.fwnode = dev_fwnode(port->dev);
6001         psy_name = devm_kzalloc(port->dev, psy_name_len, GFP_KERNEL);
6002         if (!psy_name)
6003                 return -ENOMEM;
6004
6005         snprintf(psy_name, psy_name_len, "%s%s", tcpm_psy_name_prefix,
6006                  port_dev_name);
6007         port->psy_desc.name = psy_name;
6008         port->psy_desc.type = POWER_SUPPLY_TYPE_USB;
6009         port->psy_desc.usb_types = tcpm_psy_usb_types;
6010         port->psy_desc.num_usb_types = ARRAY_SIZE(tcpm_psy_usb_types);
6011         port->psy_desc.properties = tcpm_psy_props;
6012         port->psy_desc.num_properties = ARRAY_SIZE(tcpm_psy_props);
6013         port->psy_desc.get_property = tcpm_psy_get_prop;
6014         port->psy_desc.set_property = tcpm_psy_set_prop;
6015         port->psy_desc.property_is_writeable = tcpm_psy_prop_writeable;
6016
6017         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
6018
6019         port->psy = devm_power_supply_register(port->dev, &port->psy_desc,
6020                                                &psy_cfg);
6021
6022         return PTR_ERR_OR_ZERO(port->psy);
6023 }
6024
6025 static enum hrtimer_restart state_machine_timer_handler(struct hrtimer *timer)
6026 {
6027         struct tcpm_port *port = container_of(timer, struct tcpm_port, state_machine_timer);
6028
6029         kthread_queue_work(port->wq, &port->state_machine);
6030         return HRTIMER_NORESTART;
6031 }
6032
6033 static enum hrtimer_restart vdm_state_machine_timer_handler(struct hrtimer *timer)
6034 {
6035         struct tcpm_port *port = container_of(timer, struct tcpm_port, vdm_state_machine_timer);
6036
6037         kthread_queue_work(port->wq, &port->vdm_state_machine);
6038         return HRTIMER_NORESTART;
6039 }
6040
6041 static enum hrtimer_restart enable_frs_timer_handler(struct hrtimer *timer)
6042 {
6043         struct tcpm_port *port = container_of(timer, struct tcpm_port, enable_frs_timer);
6044
6045         kthread_queue_work(port->wq, &port->enable_frs);
6046         return HRTIMER_NORESTART;
6047 }
6048
6049 struct tcpm_port *tcpm_register_port(struct device *dev, struct tcpc_dev *tcpc)
6050 {
6051         struct tcpm_port *port;
6052         int err;
6053
6054         if (!dev || !tcpc ||
6055             !tcpc->get_vbus || !tcpc->set_cc || !tcpc->get_cc ||
6056             !tcpc->set_polarity || !tcpc->set_vconn || !tcpc->set_vbus ||
6057             !tcpc->set_pd_rx || !tcpc->set_roles || !tcpc->pd_transmit)
6058                 return ERR_PTR(-EINVAL);
6059
6060         port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
6061         if (!port)
6062                 return ERR_PTR(-ENOMEM);
6063
6064         port->dev = dev;
6065         port->tcpc = tcpc;
6066
6067         mutex_init(&port->lock);
6068         mutex_init(&port->swap_lock);
6069
6070         port->wq = kthread_create_worker(0, dev_name(dev));
6071         if (IS_ERR(port->wq))
6072                 return ERR_CAST(port->wq);
6073         sched_set_fifo(port->wq->task);
6074
6075         kthread_init_work(&port->state_machine, tcpm_state_machine_work);
6076         kthread_init_work(&port->vdm_state_machine, vdm_state_machine_work);
6077         kthread_init_work(&port->event_work, tcpm_pd_event_handler);
6078         kthread_init_work(&port->enable_frs, tcpm_enable_frs_work);
6079         hrtimer_init(&port->state_machine_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
6080         port->state_machine_timer.function = state_machine_timer_handler;
6081         hrtimer_init(&port->vdm_state_machine_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
6082         port->vdm_state_machine_timer.function = vdm_state_machine_timer_handler;
6083         hrtimer_init(&port->enable_frs_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
6084         port->enable_frs_timer.function = enable_frs_timer_handler;
6085
6086         spin_lock_init(&port->pd_event_lock);
6087
6088         init_completion(&port->tx_complete);
6089         init_completion(&port->swap_complete);
6090         init_completion(&port->pps_complete);
6091         tcpm_debugfs_init(port);
6092
6093         err = tcpm_fw_get_caps(port, tcpc->fwnode);
6094         if (err < 0)
6095                 goto out_destroy_wq;
6096
6097         port->try_role = port->typec_caps.prefer_role;
6098
6099         port->typec_caps.fwnode = tcpc->fwnode;
6100         port->typec_caps.revision = 0x0120;     /* Type-C spec release 1.2 */
6101         port->typec_caps.pd_revision = 0x0300;  /* USB-PD spec release 3.0 */
6102         port->typec_caps.svdm_version = SVDM_VER_2_0;
6103         port->typec_caps.driver_data = port;
6104         port->typec_caps.ops = &tcpm_ops;
6105         port->typec_caps.orientation_aware = 1;
6106
6107         port->partner_desc.identity = &port->partner_ident;
6108         port->port_type = port->typec_caps.type;
6109
6110         port->role_sw = usb_role_switch_get(port->dev);
6111         if (IS_ERR(port->role_sw)) {
6112                 err = PTR_ERR(port->role_sw);
6113                 goto out_destroy_wq;
6114         }
6115
6116         err = devm_tcpm_psy_register(port);
6117         if (err)
6118                 goto out_role_sw_put;
6119         power_supply_changed(port->psy);
6120
6121         port->typec_port = typec_register_port(port->dev, &port->typec_caps);
6122         if (IS_ERR(port->typec_port)) {
6123                 err = PTR_ERR(port->typec_port);
6124                 goto out_role_sw_put;
6125         }
6126
6127         mutex_lock(&port->lock);
6128         tcpm_init(port);
6129         mutex_unlock(&port->lock);
6130
6131         tcpm_log(port, "%s: registered", dev_name(dev));
6132         return port;
6133
6134 out_role_sw_put:
6135         usb_role_switch_put(port->role_sw);
6136 out_destroy_wq:
6137         tcpm_debugfs_exit(port);
6138         kthread_destroy_worker(port->wq);
6139         return ERR_PTR(err);
6140 }
6141 EXPORT_SYMBOL_GPL(tcpm_register_port);
6142
6143 void tcpm_unregister_port(struct tcpm_port *port)
6144 {
6145         int i;
6146
6147         tcpm_reset_port(port);
6148         for (i = 0; i < ARRAY_SIZE(port->port_altmode); i++)
6149                 typec_unregister_altmode(port->port_altmode[i]);
6150         typec_unregister_port(port->typec_port);
6151         usb_role_switch_put(port->role_sw);
6152         tcpm_debugfs_exit(port);
6153         kthread_destroy_worker(port->wq);
6154 }
6155 EXPORT_SYMBOL_GPL(tcpm_unregister_port);
6156
6157 MODULE_AUTHOR("Guenter Roeck <groeck@chromium.org>");
6158 MODULE_DESCRIPTION("USB Type-C Port Manager");
6159 MODULE_LICENSE("GPL");