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