drivers/usb/typec/tcpm/tcpm.c

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