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