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