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