drivers/usb/typec/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/jiffies.h>
  12 #include <linux/kernel.h>
  13 #include <linux/module.h>
  14 #include <linux/mutex.h>
  15 #include <linux/power_supply.h>
  16 #include <linux/proc_fs.h>
  17 #include <linux/property.h>
  18 #include <linux/sched/clock.h>
  19 #include <linux/seq_file.h>
  20 #include <linux/slab.h>
  21 #include <linux/spinlock.h>
  22 #include <linux/usb/pd.h>
  23 #include <linux/usb/pd_ado.h>
  24 #include <linux/usb/pd_bdo.h>
  25 #include <linux/usb/pd_ext_sdb.h>
  26 #include <linux/usb/pd_vdo.h>
  27 #include <linux/usb/role.h>
  28 #include <linux/usb/tcpm.h>
  29 #include <linux/usb/typec.h>
  30 #include <linux/workqueue.h>
  31
  32 #define FOREACH_STATE(S)                        \
  33         S(INVALID_STATE),                       \
  34         S(DRP_TOGGLING),                        \
  35         S(SRC_UNATTACHED),                      \
  36         S(SRC_ATTACH_WAIT),                     \
  37         S(SRC_ATTACHED),                        \
  38         S(SRC_STARTUP),                         \
  39         S(SRC_SEND_CAPABILITIES),               \
  40         S(SRC_NEGOTIATE_CAPABILITIES),          \
  41         S(SRC_TRANSITION_SUPPLY),               \
  42         S(SRC_READY),                           \
  43         S(SRC_WAIT_NEW_CAPABILITIES),           \
  44                                                 \
  45         S(SNK_UNATTACHED),                      \
  46         S(SNK_ATTACH_WAIT),                     \
  47         S(SNK_DEBOUNCED),                       \
  48         S(SNK_ATTACHED),                        \
  49         S(SNK_STARTUP),                         \
  50         S(SNK_DISCOVERY),                       \
  51         S(SNK_DISCOVERY_DEBOUNCE),              \
  52         S(SNK_DISCOVERY_DEBOUNCE_DONE),         \
  53         S(SNK_WAIT_CAPABILITIES),               \
  54         S(SNK_NEGOTIATE_CAPABILITIES),          \
  55         S(SNK_NEGOTIATE_PPS_CAPABILITIES),      \
  56         S(SNK_TRANSITION_SINK),                 \
  57         S(SNK_TRANSITION_SINK_VBUS),            \
  58         S(SNK_READY),                           \
  59                                                 \
  60         S(ACC_UNATTACHED),                      \
  61         S(DEBUG_ACC_ATTACHED),                  \
  62         S(AUDIO_ACC_ATTACHED),                  \
  63         S(AUDIO_ACC_DEBOUNCE),                  \
  64                                                 \
  65         S(HARD_RESET_SEND),                     \
  66         S(HARD_RESET_START),                    \
  67         S(SRC_HARD_RESET_VBUS_OFF),             \
  68         S(SRC_HARD_RESET_VBUS_ON),              \
  69         S(SNK_HARD_RESET_SINK_OFF),             \
  70         S(SNK_HARD_RESET_WAIT_VBUS),            \
  71         S(SNK_HARD_RESET_SINK_ON),              \
  72                                                 \
  73         S(SOFT_RESET),                          \
  74         S(SOFT_RESET_SEND),                     \
  75                                                 \
  76         S(DR_SWAP_ACCEPT),                      \
  77         S(DR_SWAP_SEND),                        \
  78         S(DR_SWAP_SEND_TIMEOUT),                \
  79         S(DR_SWAP_CANCEL),                      \
  80         S(DR_SWAP_CHANGE_DR),                   \
  81                                                 \
  82         S(PR_SWAP_ACCEPT),                      \
  83         S(PR_SWAP_SEND),                        \
  84         S(PR_SWAP_SEND_TIMEOUT),                \
  85         S(PR_SWAP_CANCEL),                      \
  86         S(PR_SWAP_START),                       \
  87         S(PR_SWAP_SRC_SNK_TRANSITION_OFF),      \
  88         S(PR_SWAP_SRC_SNK_SOURCE_OFF),          \
  89         S(PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED), \
  90         S(PR_SWAP_SRC_SNK_SINK_ON),             \
  91         S(PR_SWAP_SNK_SRC_SINK_OFF),            \
  92         S(PR_SWAP_SNK_SRC_SOURCE_ON),           \
  93         S(PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP),    \
  94                                                 \
  95         S(VCONN_SWAP_ACCEPT),                   \
  96         S(VCONN_SWAP_SEND),                     \
  97         S(VCONN_SWAP_SEND_TIMEOUT),             \
  98         S(VCONN_SWAP_CANCEL),                   \
  99         S(VCONN_SWAP_START),                    \
 100         S(VCONN_SWAP_WAIT_FOR_VCONN),           \
 101         S(VCONN_SWAP_TURN_ON_VCONN),            \
 102         S(VCONN_SWAP_TURN_OFF_VCONN),           \
 103                                                 \
 104         S(SNK_TRY),                             \
 105         S(SNK_TRY_WAIT),                        \
 106         S(SNK_TRY_WAIT_DEBOUNCE),               \
 107         S(SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS),    \
 108         S(SRC_TRYWAIT),                         \
 109         S(SRC_TRYWAIT_DEBOUNCE),                \
 110         S(SRC_TRYWAIT_UNATTACHED),              \
 111                                                 \
 112         S(SRC_TRY),                             \
 113         S(SRC_TRY_WAIT),                        \
 114         S(SRC_TRY_DEBOUNCE),                    \
 115         S(SNK_TRYWAIT),                         \
 116         S(SNK_TRYWAIT_DEBOUNCE),                \
 117         S(SNK_TRYWAIT_VBUS),                    \
 118         S(BIST_RX),                             \
 119                                                 \
 120         S(GET_STATUS_SEND),                     \
 121         S(GET_STATUS_SEND_TIMEOUT),             \
 122         S(GET_PPS_STATUS_SEND),                 \
 123         S(GET_PPS_STATUS_SEND_TIMEOUT),         \
 124                                                 \
 125         S(ERROR_RECOVERY),                      \
 126         S(PORT_RESET),                          \
 127         S(PORT_RESET_WAIT_OFF)
 128
 129 #define GENERATE_ENUM(e)        e
 130 #define GENERATE_STRING(s)      #s
 131
 132 enum tcpm_state {
 133         FOREACH_STATE(GENERATE_ENUM)
 134 };
 135
 136 static const char * const tcpm_states[] = {
 137         FOREACH_STATE(GENERATE_STRING)
 138 };
 139
 140 enum vdm_states {
 141         VDM_STATE_ERR_BUSY = -3,
 142         VDM_STATE_ERR_SEND = -2,
 143         VDM_STATE_ERR_TMOUT = -1,
 144         VDM_STATE_DONE = 0,
 145         /* Anything >0 represents an active state */
 146         VDM_STATE_READY = 1,
 147         VDM_STATE_BUSY = 2,
 148         VDM_STATE_WAIT_RSP_BUSY = 3,
 149 };
 150
 151 enum pd_msg_request {
 152         PD_MSG_NONE = 0,
 153         PD_MSG_CTRL_REJECT,
 154         PD_MSG_CTRL_WAIT,
 155         PD_MSG_CTRL_NOT_SUPP,
 156         PD_MSG_DATA_SINK_CAP,
 157         PD_MSG_DATA_SOURCE_CAP,
 158 };
 159
 160 /* Events from low level driver */
 161
 162 #define TCPM_CC_EVENT           BIT(0)
 163 #define TCPM_VBUS_EVENT         BIT(1)
 164 #define TCPM_RESET_EVENT        BIT(2)
 165
 166 #define LOG_BUFFER_ENTRIES      1024
 167 #define LOG_BUFFER_ENTRY_SIZE   128
 168
 169 /* Alternate mode support */
 170
 171 #define SVID_DISCOVERY_MAX      16
 172
 173 struct pd_mode_data {
 174         int svid_index;         /* current SVID index           */
 175         int nsvids;
 176         u16 svids[SVID_DISCOVERY_MAX];
 177         int altmodes;           /* number of alternate modes    */
 178         struct typec_altmode_desc altmode_desc[SVID_DISCOVERY_MAX];
 179 };
 180
 181 struct pd_pps_data {
 182         u32 min_volt;
 183         u32 max_volt;
 184         u32 max_curr;
 185         u32 out_volt;
 186         u32 op_curr;
 187         bool supported;
 188         bool active;
 189 };
 190
 191 struct tcpm_port {
 192         struct device *dev;
 193
 194         struct mutex lock;              /* tcpm state machine lock */
 195         struct workqueue_struct *wq;
 196
 197         struct typec_capability typec_caps;
 198         struct typec_port *typec_port;
 199
 200         struct tcpc_dev *tcpc;
 201         struct usb_role_switch *role_sw;
 202
 203         enum typec_role vconn_role;
 204         enum typec_role pwr_role;
 205         enum typec_data_role data_role;
 206         enum typec_pwr_opmode pwr_opmode;
 207
 208         struct usb_pd_identity partner_ident;
 209         struct typec_partner_desc partner_desc;
 210         struct typec_partner *partner;
 211
 212         enum typec_cc_status cc_req;
 213
 214         enum typec_cc_status cc1;
 215         enum typec_cc_status cc2;
 216         enum typec_cc_polarity polarity;
 217
 218         bool attached;
 219         bool connected;
 220         enum typec_port_type port_type;
 221         bool vbus_present;
 222         bool vbus_never_low;
 223         bool vbus_source;
 224         bool vbus_charge;
 225
 226         bool send_discover;
 227         bool op_vsafe5v;
 228
 229         int try_role;
 230         int try_snk_count;
 231         int try_src_count;
 232
 233         enum pd_msg_request queued_message;
 234
 235         enum tcpm_state enter_state;
 236         enum tcpm_state prev_state;
 237         enum tcpm_state state;
 238         enum tcpm_state delayed_state;
 239         unsigned long delayed_runtime;
 240         unsigned long delay_ms;
 241
 242         spinlock_t pd_event_lock;
 243         u32 pd_events;
 244
 245         struct work_struct event_work;
 246         struct delayed_work state_machine;
 247         struct delayed_work vdm_state_machine;
 248         bool state_machine_running;
 249
 250         struct completion tx_complete;
 251         enum tcpm_transmit_status tx_status;
 252
 253         struct mutex swap_lock;         /* swap command lock */
 254         bool swap_pending;
 255         bool non_pd_role_swap;
 256         struct completion swap_complete;
 257         int swap_status;
 258
 259         unsigned int negotiated_rev;
 260         unsigned int message_id;
 261         unsigned int caps_count;
 262         unsigned int hard_reset_count;
 263         bool pd_capable;
 264         bool explicit_contract;
 265         unsigned int rx_msgid;
 266
 267         /* Partner capabilities/requests */
 268         u32 sink_request;
 269         u32 source_caps[PDO_MAX_OBJECTS];
 270         unsigned int nr_source_caps;
 271         u32 sink_caps[PDO_MAX_OBJECTS];
 272         unsigned int nr_sink_caps;
 273
 274         /* Local capabilities */
 275         u32 src_pdo[PDO_MAX_OBJECTS];
 276         unsigned int nr_src_pdo;
 277         u32 snk_pdo[PDO_MAX_OBJECTS];
 278         unsigned int nr_snk_pdo;
 279         u32 snk_vdo[VDO_MAX_OBJECTS];
 280         unsigned int nr_snk_vdo;
 281
 282         unsigned int operating_snk_mw;
 283         bool update_sink_caps;
 284
 285         /* Requested current / voltage */
 286         u32 current_limit;
 287         u32 supply_voltage;
 288
 289         /* Used to export TA voltage and current */
 290         struct power_supply *psy;
 291         struct power_supply_desc psy_desc;
 292         enum power_supply_usb_type usb_type;
 293
 294         u32 bist_request;
 295
 296         /* PD state for Vendor Defined Messages */
 297         enum vdm_states vdm_state;
 298         u32 vdm_retries;
 299         /* next Vendor Defined Message to send */
 300         u32 vdo_data[VDO_MAX_SIZE];
 301         u8 vdo_count;
 302         /* VDO to retry if UFP responder replied busy */
 303         u32 vdo_retry;
 304
 305         /* PPS */
 306         struct pd_pps_data pps_data;
 307         struct completion pps_complete;
 308         bool pps_pending;
 309         int pps_status;
 310
 311         /* Alternate mode data */
 312         struct pd_mode_data mode_data;
 313         struct typec_altmode *partner_altmode[SVID_DISCOVERY_MAX * 6];
 314         struct typec_altmode *port_altmode[SVID_DISCOVERY_MAX * 6];
 315
 316         /* Deadline in jiffies to exit src_try_wait state */
 317         unsigned long max_wait;
 318
 319 #ifdef CONFIG_DEBUG_FS
 320         struct dentry *dentry;
 321         struct mutex logbuffer_lock;    /* log buffer access lock */
 322         int logbuffer_head;
 323         int logbuffer_tail;
 324         u8 *logbuffer[LOG_BUFFER_ENTRIES];
 325 #endif
 326 };
 327
 328 struct pd_rx_event {
 329         struct work_struct work;
 330         struct tcpm_port *port;
 331         struct pd_message msg;
 332 };
 333
 334 #define tcpm_cc_is_sink(cc) \
 335         ((cc) == TYPEC_CC_RP_DEF || (cc) == TYPEC_CC_RP_1_5 || \
 336          (cc) == TYPEC_CC_RP_3_0)
 337
 338 #define tcpm_port_is_sink(port) \
 339         ((tcpm_cc_is_sink((port)->cc1) && !tcpm_cc_is_sink((port)->cc2)) || \
 340          (tcpm_cc_is_sink((port)->cc2) && !tcpm_cc_is_sink((port)->cc1)))
 341
 342 #define tcpm_cc_is_source(cc) ((cc) == TYPEC_CC_RD)
 343 #define tcpm_cc_is_audio(cc) ((cc) == TYPEC_CC_RA)
 344 #define tcpm_cc_is_open(cc) ((cc) == TYPEC_CC_OPEN)
 345
 346 #define tcpm_port_is_source(port) \
 347         ((tcpm_cc_is_source((port)->cc1) && \
 348          !tcpm_cc_is_source((port)->cc2)) || \
 349          (tcpm_cc_is_source((port)->cc2) && \
 350           !tcpm_cc_is_source((port)->cc1)))
 351
 352 #define tcpm_port_is_debug(port) \
 353         (tcpm_cc_is_source((port)->cc1) && tcpm_cc_is_source((port)->cc2))
 354
 355 #define tcpm_port_is_audio(port) \
 356         (tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_audio((port)->cc2))
 357
 358 #define tcpm_port_is_audio_detached(port) \
 359         ((tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_open((port)->cc2)) || \
 360          (tcpm_cc_is_audio((port)->cc2) && tcpm_cc_is_open((port)->cc1)))
 361
 362 #define tcpm_try_snk(port) \
 363         ((port)->try_snk_count == 0 && (port)->try_role == TYPEC_SINK && \
 364         (port)->port_type == TYPEC_PORT_DRP)
 365
 366 #define tcpm_try_src(port) \
 367         ((port)->try_src_count == 0 && (port)->try_role == TYPEC_SOURCE && \
 368         (port)->port_type == TYPEC_PORT_DRP)
 369
 370 static enum tcpm_state tcpm_default_state(struct tcpm_port *port)
 371 {
 372         if (port->port_type == TYPEC_PORT_DRP) {
 373                 if (port->try_role == TYPEC_SINK)
 374                         return SNK_UNATTACHED;
 375                 else if (port->try_role == TYPEC_SOURCE)
 376                         return SRC_UNATTACHED;
 377                 else if (port->tcpc->config->default_role == TYPEC_SINK)
 378                         return SNK_UNATTACHED;
 379                 /* Fall through to return SRC_UNATTACHED */
 380         } else if (port->port_type == TYPEC_PORT_SNK) {
 381                 return SNK_UNATTACHED;
 382         }
 383         return SRC_UNATTACHED;
 384 }
 385
 386 static inline
 387 struct tcpm_port *typec_cap_to_tcpm(const struct typec_capability *cap)
 388 {
 389         return container_of(cap, struct tcpm_port, typec_caps);
 390 }
 391
 392 static bool tcpm_port_is_disconnected(struct tcpm_port *port)
 393 {
 394         return (!port->attached && port->cc1 == TYPEC_CC_OPEN &&
 395                 port->cc2 == TYPEC_CC_OPEN) ||
 396                (port->attached && ((port->polarity == TYPEC_POLARITY_CC1 &&
 397                                     port->cc1 == TYPEC_CC_OPEN) ||
 398                                    (port->polarity == TYPEC_POLARITY_CC2 &&
 399                                     port->cc2 == TYPEC_CC_OPEN)));
 400 }
 401
 402 /*
 403  * Logging
 404  */
 405
 406 #ifdef CONFIG_DEBUG_FS
 407
 408 static bool tcpm_log_full(struct tcpm_port *port)
 409 {
 410         return port->logbuffer_tail ==
 411                 (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 412 }
 413
 414 __printf(2, 0)
 415 static void _tcpm_log(struct tcpm_port *port, const char *fmt, va_list args)
 416 {
 417         char tmpbuffer[LOG_BUFFER_ENTRY_SIZE];
 418         u64 ts_nsec = local_clock();
 419         unsigned long rem_nsec;
 420
 421         mutex_lock(&port->logbuffer_lock);
 422         if (!port->logbuffer[port->logbuffer_head]) {
 423                 port->logbuffer[port->logbuffer_head] =
 424                                 kzalloc(LOG_BUFFER_ENTRY_SIZE, GFP_KERNEL);
 425                 if (!port->logbuffer[port->logbuffer_head]) {
 426                         mutex_unlock(&port->logbuffer_lock);
 427                         return;
 428                 }
 429         }
 430
 431         vsnprintf(tmpbuffer, sizeof(tmpbuffer), fmt, args);
 432
 433         if (tcpm_log_full(port)) {
 434                 port->logbuffer_head = max(port->logbuffer_head - 1, 0);
 435                 strcpy(tmpbuffer, "overflow");
 436         }
 437
 438         if (port->logbuffer_head < 0 ||
 439             port->logbuffer_head >= LOG_BUFFER_ENTRIES) {
 440                 dev_warn(port->dev,
 441                          "Bad log buffer index %d\n", port->logbuffer_head);
 442                 goto abort;
 443         }
 444
 445         if (!port->logbuffer[port->logbuffer_head]) {
 446                 dev_warn(port->dev,
 447                          "Log buffer index %d is NULL\n", port->logbuffer_head);
 448                 goto abort;
 449         }
 450
 451         rem_nsec = do_div(ts_nsec, 1000000000);
 452         scnprintf(port->logbuffer[port->logbuffer_head],
 453                   LOG_BUFFER_ENTRY_SIZE, "[%5lu.%06lu] %s",
 454                   (unsigned long)ts_nsec, rem_nsec / 1000,
 455                   tmpbuffer);
 456         port->logbuffer_head = (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
 457
 458 abort:
 459         mutex_unlock(&port->logbuffer_lock);
 460 }
 461
 462 __printf(2, 3)
 463 static void tcpm_log(struct tcpm_port *port, const char *fmt, ...)
 464 {
 465         va_list args;
 466
 467         /* Do not log while disconnected and unattached */
 468         if (tcpm_port_is_disconnected(port) &&
 469             (port->state == SRC_UNATTACHED || port->state == SNK_UNATTACHED ||
 470              port->state == DRP_TOGGLING))
 471                 return;
 472
 473         va_start(args, fmt);
 474         _tcpm_log(port, fmt, args);
 475         va_end(args);
 476 }
 477
 478 __printf(2, 3)
 479 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...)
 480 {
 481         va_list args;
 482
 483         va_start(args, fmt);
 484         _tcpm_log(port, fmt, args);
 485         va_end(args);
 486 }
 487
 488 static void tcpm_log_source_caps(struct tcpm_port *port)
 489 {
 490         int i;
 491
 492         for (i = 0; i < port->nr_source_caps; i++) {
 493                 u32 pdo = port->source_caps[i];
 494                 enum pd_pdo_type type = pdo_type(pdo);
 495                 char msg[64];
 496
 497                 switch (type) {
 498                 case PDO_TYPE_FIXED:
 499                         scnprintf(msg, sizeof(msg),
 500                                   "%u mV, %u mA [%s%s%s%s%s%s]",
 501                                   pdo_fixed_voltage(pdo),
 502                                   pdo_max_current(pdo),
 503                                   (pdo & PDO_FIXED_DUAL_ROLE) ?
 504                                                         "R" : "",
 505                                   (pdo & PDO_FIXED_SUSPEND) ?
 506                                                         "S" : "",
 507                                   (pdo & PDO_FIXED_HIGHER_CAP) ?
 508                                                         "H" : "",
 509                                   (pdo & PDO_FIXED_USB_COMM) ?
 510                                                         "U" : "",
 511                                   (pdo & PDO_FIXED_DATA_SWAP) ?
 512                                                         "D" : "",
 513                                   (pdo & PDO_FIXED_EXTPOWER) ?
 514                                                         "E" : "");
 515                         break;
 516                 case PDO_TYPE_VAR:
 517                         scnprintf(msg, sizeof(msg),
 518                                   "%u-%u mV, %u mA",
 519                                   pdo_min_voltage(pdo),
 520                                   pdo_max_voltage(pdo),
 521                                   pdo_max_current(pdo));
 522                         break;
 523                 case PDO_TYPE_BATT:
 524                         scnprintf(msg, sizeof(msg),
 525                                   "%u-%u mV, %u mW",
 526                                   pdo_min_voltage(pdo),
 527                                   pdo_max_voltage(pdo),
 528                                   pdo_max_power(pdo));
 529                         break;
 530                 case PDO_TYPE_APDO:
 531                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS)
 532                                 scnprintf(msg, sizeof(msg),
 533                                           "%u-%u mV, %u mA",
 534                                           pdo_pps_apdo_min_voltage(pdo),
 535                                           pdo_pps_apdo_max_voltage(pdo),
 536                                           pdo_pps_apdo_max_current(pdo));
 537                         else
 538                                 strcpy(msg, "undefined APDO");
 539                         break;
 540                 default:
 541                         strcpy(msg, "undefined");
 542                         break;
 543                 }
 544                 tcpm_log(port, " PDO %d: type %d, %s",
 545                          i, type, msg);
 546         }
 547 }
 548
 549 static int tcpm_debug_show(struct seq_file *s, void *v)
 550 {
 551         struct tcpm_port *port = (struct tcpm_port *)s->private;
 552         int tail;
 553
 554         mutex_lock(&port->logbuffer_lock);
 555         tail = port->logbuffer_tail;
 556         while (tail != port->logbuffer_head) {
 557                 seq_printf(s, "%s\n", port->logbuffer[tail]);
 558                 tail = (tail + 1) % LOG_BUFFER_ENTRIES;
 559         }
 560         if (!seq_has_overflowed(s))
 561                 port->logbuffer_tail = tail;
 562         mutex_unlock(&port->logbuffer_lock);
 563
 564         return 0;
 565 }
 566 DEFINE_SHOW_ATTRIBUTE(tcpm_debug);
 567
 568 static struct dentry *rootdir;
 569
 570 static void tcpm_debugfs_init(struct tcpm_port *port)
 571 {
 572         mutex_init(&port->logbuffer_lock);
 573         /* /sys/kernel/debug/tcpm/usbcX */
 574         if (!rootdir)
 575                 rootdir = debugfs_create_dir("tcpm", NULL);
 576
 577         port->dentry = debugfs_create_file(dev_name(port->dev),
 578                                            S_IFREG | 0444, rootdir,
 579                                            port, &tcpm_debug_fops);
 580 }
 581
 582 static void tcpm_debugfs_exit(struct tcpm_port *port)
 583 {
 584         debugfs_remove(port->dentry);
 585 }
 586
 587 #else
 588
 589 __printf(2, 3)
 590 static void tcpm_log(const struct tcpm_port *port, const char *fmt, ...) { }
 591 __printf(2, 3)
 592 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...) { }
 593 static void tcpm_log_source_caps(struct tcpm_port *port) { }
 594 static void tcpm_debugfs_init(const struct tcpm_port *port) { }
 595 static void tcpm_debugfs_exit(const struct tcpm_port *port) { }
 596
 597 #endif
 598
 599 static int tcpm_pd_transmit(struct tcpm_port *port,
 600                             enum tcpm_transmit_type type,
 601                             const struct pd_message *msg)
 602 {
 603         unsigned long timeout;
 604         int ret;
 605
 606         if (msg)
 607                 tcpm_log(port, "PD TX, header: %#x", le16_to_cpu(msg->header));
 608         else
 609                 tcpm_log(port, "PD TX, type: %#x", type);
 610
 611         reinit_completion(&port->tx_complete);
 612         ret = port->tcpc->pd_transmit(port->tcpc, type, msg);
 613         if (ret < 0)
 614                 return ret;
 615
 616         mutex_unlock(&port->lock);
 617         timeout = wait_for_completion_timeout(&port->tx_complete,
 618                                 msecs_to_jiffies(PD_T_TCPC_TX_TIMEOUT));
 619         mutex_lock(&port->lock);
 620         if (!timeout)
 621                 return -ETIMEDOUT;
 622
 623         switch (port->tx_status) {
 624         case TCPC_TX_SUCCESS:
 625                 port->message_id = (port->message_id + 1) & PD_HEADER_ID_MASK;
 626                 return 0;
 627         case TCPC_TX_DISCARDED:
 628                 return -EAGAIN;
 629         case TCPC_TX_FAILED:
 630         default:
 631                 return -EIO;
 632         }
 633 }
 634
 635 void tcpm_pd_transmit_complete(struct tcpm_port *port,
 636                                enum tcpm_transmit_status status)
 637 {
 638         tcpm_log(port, "PD TX complete, status: %u", status);
 639         port->tx_status = status;
 640         complete(&port->tx_complete);
 641 }
 642 EXPORT_SYMBOL_GPL(tcpm_pd_transmit_complete);
 643
 644 static int tcpm_mux_set(struct tcpm_port *port, enum tcpc_mux_mode mode,
 645                         enum usb_role usb_role,
 646                         enum typec_orientation orientation)
 647 {
 648         int ret;
 649
 650         tcpm_log(port, "Requesting mux mode %d, usb-role %d, orientation %d",
 651                  mode, usb_role, orientation);
 652
 653         ret = typec_set_orientation(port->typec_port, orientation);
 654         if (ret)
 655                 return ret;
 656
 657         if (port->role_sw) {
 658                 ret = usb_role_switch_set_role(port->role_sw, usb_role);
 659                 if (ret)
 660                         return ret;
 661         }
 662
 663         return typec_set_mode(port->typec_port, mode);
 664 }
 665
 666 static int tcpm_set_polarity(struct tcpm_port *port,
 667                              enum typec_cc_polarity polarity)
 668 {
 669         int ret;
 670
 671         tcpm_log(port, "polarity %d", polarity);
 672
 673         ret = port->tcpc->set_polarity(port->tcpc, polarity);
 674         if (ret < 0)
 675                 return ret;
 676
 677         port->polarity = polarity;
 678
 679         return 0;
 680 }
 681
 682 static int tcpm_set_vconn(struct tcpm_port *port, bool enable)
 683 {
 684         int ret;
 685
 686         tcpm_log(port, "vconn:=%d", enable);
 687
 688         ret = port->tcpc->set_vconn(port->tcpc, enable);
 689         if (!ret) {
 690                 port->vconn_role = enable ? TYPEC_SOURCE : TYPEC_SINK;
 691                 typec_set_vconn_role(port->typec_port, port->vconn_role);
 692         }
 693
 694         return ret;
 695 }
 696
 697 static u32 tcpm_get_current_limit(struct tcpm_port *port)
 698 {
 699         enum typec_cc_status cc;
 700         u32 limit;
 701
 702         cc = port->polarity ? port->cc2 : port->cc1;
 703         switch (cc) {
 704         case TYPEC_CC_RP_1_5:
 705                 limit = 1500;
 706                 break;
 707         case TYPEC_CC_RP_3_0:
 708                 limit = 3000;
 709                 break;
 710         case TYPEC_CC_RP_DEF:
 711         default:
 712                 if (port->tcpc->get_current_limit)
 713                         limit = port->tcpc->get_current_limit(port->tcpc);
 714                 else
 715                         limit = 0;
 716                 break;
 717         }
 718
 719         return limit;
 720 }
 721
 722 static int tcpm_set_current_limit(struct tcpm_port *port, u32 max_ma, u32 mv)
 723 {
 724         int ret = -EOPNOTSUPP;
 725
 726         tcpm_log(port, "Setting voltage/current limit %u mV %u mA", mv, max_ma);
 727
 728         if (port->tcpc->set_current_limit)
 729                 ret = port->tcpc->set_current_limit(port->tcpc, max_ma, mv);
 730
 731         return ret;
 732 }
 733
 734 /*
 735  * Determine RP value to set based on maximum current supported
 736  * by a port if configured as source.
 737  * Returns CC value to report to link partner.
 738  */
 739 static enum typec_cc_status tcpm_rp_cc(struct tcpm_port *port)
 740 {
 741         const u32 *src_pdo = port->src_pdo;
 742         int nr_pdo = port->nr_src_pdo;
 743         int i;
 744
 745         /*
 746          * Search for first entry with matching voltage.
 747          * It should report the maximum supported current.
 748          */
 749         for (i = 0; i < nr_pdo; i++) {
 750                 const u32 pdo = src_pdo[i];
 751
 752                 if (pdo_type(pdo) == PDO_TYPE_FIXED &&
 753                     pdo_fixed_voltage(pdo) == 5000) {
 754                         unsigned int curr = pdo_max_current(pdo);
 755
 756                         if (curr >= 3000)
 757                                 return TYPEC_CC_RP_3_0;
 758                         else if (curr >= 1500)
 759                                 return TYPEC_CC_RP_1_5;
 760                         return TYPEC_CC_RP_DEF;
 761                 }
 762         }
 763
 764         return TYPEC_CC_RP_DEF;
 765 }
 766
 767 static int tcpm_set_attached_state(struct tcpm_port *port, bool attached)
 768 {
 769         return port->tcpc->set_roles(port->tcpc, attached, port->pwr_role,
 770                                      port->data_role);
 771 }
 772
 773 static int tcpm_set_roles(struct tcpm_port *port, bool attached,
 774                           enum typec_role role, enum typec_data_role data)
 775 {
 776         enum typec_orientation orientation;
 777         enum usb_role usb_role;
 778         int ret;
 779
 780         if (port->polarity == TYPEC_POLARITY_CC1)
 781                 orientation = TYPEC_ORIENTATION_NORMAL;
 782         else
 783                 orientation = TYPEC_ORIENTATION_REVERSE;
 784
 785         if (data == TYPEC_HOST)
 786                 usb_role = USB_ROLE_HOST;
 787         else
 788                 usb_role = USB_ROLE_DEVICE;
 789
 790         ret = tcpm_mux_set(port, TYPEC_MUX_USB, usb_role, orientation);
 791         if (ret < 0)
 792                 return ret;
 793
 794         ret = port->tcpc->set_roles(port->tcpc, attached, role, data);
 795         if (ret < 0)
 796                 return ret;
 797
 798         port->pwr_role = role;
 799         port->data_role = data;
 800         typec_set_data_role(port->typec_port, data);
 801         typec_set_pwr_role(port->typec_port, role);
 802
 803         return 0;
 804 }
 805
 806 static int tcpm_set_pwr_role(struct tcpm_port *port, enum typec_role role)
 807 {
 808         int ret;
 809
 810         ret = port->tcpc->set_roles(port->tcpc, true, role,
 811                                     port->data_role);
 812         if (ret < 0)
 813                 return ret;
 814
 815         port->pwr_role = role;
 816         typec_set_pwr_role(port->typec_port, role);
 817
 818         return 0;
 819 }
 820
 821 static int tcpm_pd_send_source_caps(struct tcpm_port *port)
 822 {
 823         struct pd_message msg;
 824         int i;
 825
 826         memset(&msg, 0, sizeof(msg));
 827         if (!port->nr_src_pdo) {
 828                 /* No source capabilities defined, sink only */
 829                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
 830                                           port->pwr_role,
 831                                           port->data_role,
 832                                           port->negotiated_rev,
 833                                           port->message_id, 0);
 834         } else {
 835                 msg.header = PD_HEADER_LE(PD_DATA_SOURCE_CAP,
 836                                           port->pwr_role,
 837                                           port->data_role,
 838                                           port->negotiated_rev,
 839                                           port->message_id,
 840                                           port->nr_src_pdo);
 841         }
 842         for (i = 0; i < port->nr_src_pdo; i++)
 843                 msg.payload[i] = cpu_to_le32(port->src_pdo[i]);
 844
 845         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
 846 }
 847
 848 static int tcpm_pd_send_sink_caps(struct tcpm_port *port)
 849 {
 850         struct pd_message msg;
 851         int i;
 852
 853         memset(&msg, 0, sizeof(msg));
 854         if (!port->nr_snk_pdo) {
 855                 /* No sink capabilities defined, source only */
 856                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
 857                                           port->pwr_role,
 858                                           port->data_role,
 859                                           port->negotiated_rev,
 860                                           port->message_id, 0);
 861         } else {
 862                 msg.header = PD_HEADER_LE(PD_DATA_SINK_CAP,
 863                                           port->pwr_role,
 864                                           port->data_role,
 865                                           port->negotiated_rev,
 866                                           port->message_id,
 867                                           port->nr_snk_pdo);
 868         }
 869         for (i = 0; i < port->nr_snk_pdo; i++)
 870                 msg.payload[i] = cpu_to_le32(port->snk_pdo[i]);
 871
 872         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
 873 }
 874
 875 static void tcpm_set_state(struct tcpm_port *port, enum tcpm_state state,
 876                            unsigned int delay_ms)
 877 {
 878         if (delay_ms) {
 879                 tcpm_log(port, "pending state change %s -> %s @ %u ms",
 880                          tcpm_states[port->state], tcpm_states[state],
 881                          delay_ms);
 882                 port->delayed_state = state;
 883                 mod_delayed_work(port->wq, &port->state_machine,
 884                                  msecs_to_jiffies(delay_ms));
 885                 port->delayed_runtime = jiffies + msecs_to_jiffies(delay_ms);
 886                 port->delay_ms = delay_ms;
 887         } else {
 888                 tcpm_log(port, "state change %s -> %s",
 889                          tcpm_states[port->state], tcpm_states[state]);
 890                 port->delayed_state = INVALID_STATE;
 891                 port->prev_state = port->state;
 892                 port->state = state;
 893                 /*
 894                  * Don't re-queue the state machine work item if we're currently
 895                  * in the state machine and we're immediately changing states.
 896                  * tcpm_state_machine_work() will continue running the state
 897                  * machine.
 898                  */
 899                 if (!port->state_machine_running)
 900                         mod_delayed_work(port->wq, &port->state_machine, 0);
 901         }
 902 }
 903
 904 static void tcpm_set_state_cond(struct tcpm_port *port, enum tcpm_state state,
 905                                 unsigned int delay_ms)
 906 {
 907         if (port->enter_state == port->state)
 908                 tcpm_set_state(port, state, delay_ms);
 909         else
 910                 tcpm_log(port,
 911                          "skipped %sstate change %s -> %s [%u ms], context state %s",
 912                          delay_ms ? "delayed " : "",
 913                          tcpm_states[port->state], tcpm_states[state],
 914                          delay_ms, tcpm_states[port->enter_state]);
 915 }
 916
 917 static void tcpm_queue_message(struct tcpm_port *port,
 918                                enum pd_msg_request message)
 919 {
 920         port->queued_message = message;
 921         mod_delayed_work(port->wq, &port->state_machine, 0);
 922 }
 923
 924 /*
 925  * VDM/VDO handling functions
 926  */
 927 static void tcpm_queue_vdm(struct tcpm_port *port, const u32 header,
 928                            const u32 *data, int cnt)
 929 {
 930         port->vdo_count = cnt + 1;
 931         port->vdo_data[0] = header;
 932         memcpy(&port->vdo_data[1], data, sizeof(u32) * cnt);
 933         /* Set ready, vdm state machine will actually send */
 934         port->vdm_retries = 0;
 935         port->vdm_state = VDM_STATE_READY;
 936 }
 937
 938 static void svdm_consume_identity(struct tcpm_port *port, const __le32 *payload,
 939                                   int cnt)
 940 {
 941         u32 vdo = le32_to_cpu(payload[VDO_INDEX_IDH]);
 942         u32 product = le32_to_cpu(payload[VDO_INDEX_PRODUCT]);
 943
 944         memset(&port->mode_data, 0, sizeof(port->mode_data));
 945
 946         port->partner_ident.id_header = vdo;
 947         port->partner_ident.cert_stat = le32_to_cpu(payload[VDO_INDEX_CSTAT]);
 948         port->partner_ident.product = product;
 949
 950         typec_partner_set_identity(port->partner);
 951
 952         tcpm_log(port, "Identity: %04x:%04x.%04x",
 953                  PD_IDH_VID(vdo),
 954                  PD_PRODUCT_PID(product), product & 0xffff);
 955 }
 956
 957 static bool svdm_consume_svids(struct tcpm_port *port, const __le32 *payload,
 958                                int cnt)
 959 {
 960         struct pd_mode_data *pmdata = &port->mode_data;
 961         int i;
 962
 963         for (i = 1; i < cnt; i++) {
 964                 u32 p = le32_to_cpu(payload[i]);
 965                 u16 svid;
 966
 967                 svid = (p >> 16) & 0xffff;
 968                 if (!svid)
 969                         return false;
 970
 971                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
 972                         goto abort;
 973
 974                 pmdata->svids[pmdata->nsvids++] = svid;
 975                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
 976
 977                 svid = p & 0xffff;
 978                 if (!svid)
 979                         return false;
 980
 981                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
 982                         goto abort;
 983
 984                 pmdata->svids[pmdata->nsvids++] = svid;
 985                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
 986         }
 987         return true;
 988 abort:
 989         tcpm_log(port, "SVID_DISCOVERY_MAX(%d) too low!", SVID_DISCOVERY_MAX);
 990         return false;
 991 }
 992
 993 static void svdm_consume_modes(struct tcpm_port *port, const __le32 *payload,
 994                                int cnt)
 995 {
 996         struct pd_mode_data *pmdata = &port->mode_data;
 997         struct typec_altmode_desc *paltmode;
 998         int i;
 999
1000         if (pmdata->altmodes >= ARRAY_SIZE(port->partner_altmode)) {
1001                 /* Already logged in svdm_consume_svids() */
1002                 return;
1003         }
1004
1005         for (i = 1; i < cnt; i++) {
1006                 paltmode = &pmdata->altmode_desc[pmdata->altmodes];
1007                 memset(paltmode, 0, sizeof(*paltmode));
1008
1009                 paltmode->svid = pmdata->svids[pmdata->svid_index];
1010                 paltmode->mode = i;
1011                 paltmode->vdo = le32_to_cpu(payload[i]);
1012
1013                 tcpm_log(port, " Alternate mode %d: SVID 0x%04x, VDO %d: 0x%08x",
1014                          pmdata->altmodes, paltmode->svid,
1015                          paltmode->mode, paltmode->vdo);
1016
1017                 port->partner_altmode[pmdata->altmodes] =
1018                         typec_partner_register_altmode(port->partner, paltmode);
1019                 if (!port->partner_altmode[pmdata->altmodes]) {
1020                         tcpm_log(port,
1021                                  "Failed to register modes for SVID 0x%04x",
1022                                  paltmode->svid);
1023                         return;
1024                 }
1025                 pmdata->altmodes++;
1026         }
1027 }
1028
1029 #define supports_modal(port)    PD_IDH_MODAL_SUPP((port)->partner_ident.id_header)
1030
1031 static int tcpm_pd_svdm(struct tcpm_port *port, const __le32 *payload, int cnt,
1032                         u32 *response)
1033 {
1034         u32 p0 = le32_to_cpu(payload[0]);
1035         int cmd_type = PD_VDO_CMDT(p0);
1036         int cmd = PD_VDO_CMD(p0);
1037         struct pd_mode_data *modep;
1038         int rlen = 0;
1039         u16 svid;
1040         int i;
1041
1042         tcpm_log(port, "Rx VDM cmd 0x%x type %d cmd %d len %d",
1043                  p0, cmd_type, cmd, cnt);
1044
1045         modep = &port->mode_data;
1046
1047         switch (cmd_type) {
1048         case CMDT_INIT:
1049                 switch (cmd) {
1050                 case CMD_DISCOVER_IDENT:
1051                         /* 6.4.4.3.1: Only respond as UFP (device) */
1052                         if (port->data_role == TYPEC_DEVICE &&
1053                             port->nr_snk_vdo) {
1054                                 for (i = 0; i <  port->nr_snk_vdo; i++)
1055                                         response[i + 1] = port->snk_vdo[i];
1056                                 rlen = port->nr_snk_vdo + 1;
1057                         }
1058                         break;
1059                 case CMD_DISCOVER_SVID:
1060                         break;
1061                 case CMD_DISCOVER_MODES:
1062                         break;
1063                 case CMD_ENTER_MODE:
1064                         break;
1065                 case CMD_EXIT_MODE:
1066                         break;
1067                 case CMD_ATTENTION:
1068                         break;
1069                 default:
1070                         break;
1071                 }
1072                 if (rlen >= 1) {
1073                         response[0] = p0 | VDO_CMDT(CMDT_RSP_ACK);
1074                 } else if (rlen == 0) {
1075                         response[0] = p0 | VDO_CMDT(CMDT_RSP_NAK);
1076                         rlen = 1;
1077                 } else {
1078                         response[0] = p0 | VDO_CMDT(CMDT_RSP_BUSY);
1079                         rlen = 1;
1080                 }
1081                 break;
1082         case CMDT_RSP_ACK:
1083                 /* silently drop message if we are not connected */
1084                 if (IS_ERR_OR_NULL(port->partner))
1085                         break;
1086
1087                 switch (cmd) {
1088                 case CMD_DISCOVER_IDENT:
1089                         /* 6.4.4.3.1 */
1090                         svdm_consume_identity(port, payload, cnt);
1091                         response[0] = VDO(USB_SID_PD, 1, CMD_DISCOVER_SVID);
1092                         rlen = 1;
1093                         break;
1094                 case CMD_DISCOVER_SVID:
1095                         /* 6.4.4.3.2 */
1096                         if (svdm_consume_svids(port, payload, cnt)) {
1097                                 response[0] = VDO(USB_SID_PD, 1,
1098                                                   CMD_DISCOVER_SVID);
1099                                 rlen = 1;
1100                         } else if (modep->nsvids && supports_modal(port)) {
1101                                 response[0] = VDO(modep->svids[0], 1,
1102                                                   CMD_DISCOVER_MODES);
1103                                 rlen = 1;
1104                         }
1105                         break;
1106                 case CMD_DISCOVER_MODES:
1107                         /* 6.4.4.3.3 */
1108                         svdm_consume_modes(port, payload, cnt);
1109                         modep->svid_index++;
1110                         if (modep->svid_index < modep->nsvids) {
1111                                 svid = modep->svids[modep->svid_index];
1112                                 response[0] = VDO(svid, 1, CMD_DISCOVER_MODES);
1113                                 rlen = 1;
1114                         } else {
1115                                 /* enter alternate mode if/when implemented */
1116                         }
1117                         break;
1118                 case CMD_ENTER_MODE:
1119                         break;
1120                 default:
1121                         break;
1122                 }
1123                 break;
1124         default:
1125                 break;
1126         }
1127
1128         return rlen;
1129 }
1130
1131 static void tcpm_handle_vdm_request(struct tcpm_port *port,
1132                                     const __le32 *payload, int cnt)
1133 {
1134         int rlen = 0;
1135         u32 response[8] = { };
1136         u32 p0 = le32_to_cpu(payload[0]);
1137
1138         if (port->vdm_state == VDM_STATE_BUSY) {
1139                 /* If UFP responded busy retry after timeout */
1140                 if (PD_VDO_CMDT(p0) == CMDT_RSP_BUSY) {
1141                         port->vdm_state = VDM_STATE_WAIT_RSP_BUSY;
1142                         port->vdo_retry = (p0 & ~VDO_CMDT_MASK) |
1143                                 CMDT_INIT;
1144                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1145                                          msecs_to_jiffies(PD_T_VDM_BUSY));
1146                         return;
1147                 }
1148                 port->vdm_state = VDM_STATE_DONE;
1149         }
1150
1151         if (PD_VDO_SVDM(p0))
1152                 rlen = tcpm_pd_svdm(port, payload, cnt, response);
1153
1154         if (rlen > 0) {
1155                 tcpm_queue_vdm(port, response[0], &response[1], rlen - 1);
1156                 mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1157         }
1158 }
1159
1160 static void tcpm_send_vdm(struct tcpm_port *port, u32 vid, int cmd,
1161                           const u32 *data, int count)
1162 {
1163         u32 header;
1164
1165         if (WARN_ON(count > VDO_MAX_SIZE - 1))
1166                 count = VDO_MAX_SIZE - 1;
1167
1168         /* set VDM header with VID & CMD */
1169         header = VDO(vid, ((vid & USB_SID_PD) == USB_SID_PD) ?
1170                         1 : (PD_VDO_CMD(cmd) <= CMD_ATTENTION), cmd);
1171         tcpm_queue_vdm(port, header, data, count);
1172
1173         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1174 }
1175
1176 static unsigned int vdm_ready_timeout(u32 vdm_hdr)
1177 {
1178         unsigned int timeout;
1179         int cmd = PD_VDO_CMD(vdm_hdr);
1180
1181         /* its not a structured VDM command */
1182         if (!PD_VDO_SVDM(vdm_hdr))
1183                 return PD_T_VDM_UNSTRUCTURED;
1184
1185         switch (PD_VDO_CMDT(vdm_hdr)) {
1186         case CMDT_INIT:
1187                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1188                         timeout = PD_T_VDM_WAIT_MODE_E;
1189                 else
1190                         timeout = PD_T_VDM_SNDR_RSP;
1191                 break;
1192         default:
1193                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1194                         timeout = PD_T_VDM_E_MODE;
1195                 else
1196                         timeout = PD_T_VDM_RCVR_RSP;
1197                 break;
1198         }
1199         return timeout;
1200 }
1201
1202 static void vdm_run_state_machine(struct tcpm_port *port)
1203 {
1204         struct pd_message msg;
1205         int i, res;
1206
1207         switch (port->vdm_state) {
1208         case VDM_STATE_READY:
1209                 /* Only transmit VDM if attached */
1210                 if (!port->attached) {
1211                         port->vdm_state = VDM_STATE_ERR_BUSY;
1212                         break;
1213                 }
1214
1215                 /*
1216                  * if there's traffic or we're not in PDO ready state don't send
1217                  * a VDM.
1218                  */
1219                 if (port->state != SRC_READY && port->state != SNK_READY)
1220                         break;
1221
1222                 /* Prepare and send VDM */
1223                 memset(&msg, 0, sizeof(msg));
1224                 msg.header = PD_HEADER_LE(PD_DATA_VENDOR_DEF,
1225                                           port->pwr_role,
1226                                           port->data_role,
1227                                           port->negotiated_rev,
1228                                           port->message_id, port->vdo_count);
1229                 for (i = 0; i < port->vdo_count; i++)
1230                         msg.payload[i] = cpu_to_le32(port->vdo_data[i]);
1231                 res = tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1232                 if (res < 0) {
1233                         port->vdm_state = VDM_STATE_ERR_SEND;
1234                 } else {
1235                         unsigned long timeout;
1236
1237                         port->vdm_retries = 0;
1238                         port->vdm_state = VDM_STATE_BUSY;
1239                         timeout = vdm_ready_timeout(port->vdo_data[0]);
1240                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1241                                          timeout);
1242                 }
1243                 break;
1244         case VDM_STATE_WAIT_RSP_BUSY:
1245                 port->vdo_data[0] = port->vdo_retry;
1246                 port->vdo_count = 1;
1247                 port->vdm_state = VDM_STATE_READY;
1248                 break;
1249         case VDM_STATE_BUSY:
1250                 port->vdm_state = VDM_STATE_ERR_TMOUT;
1251                 break;
1252         case VDM_STATE_ERR_SEND:
1253                 /*
1254                  * A partner which does not support USB PD will not reply,
1255                  * so this is not a fatal error. At the same time, some
1256                  * devices may not return GoodCRC under some circumstances,
1257                  * so we need to retry.
1258                  */
1259                 if (port->vdm_retries < 3) {
1260                         tcpm_log(port, "VDM Tx error, retry");
1261                         port->vdm_retries++;
1262                         port->vdm_state = VDM_STATE_READY;
1263                 }
1264                 break;
1265         default:
1266                 break;
1267         }
1268 }
1269
1270 static void vdm_state_machine_work(struct work_struct *work)
1271 {
1272         struct tcpm_port *port = container_of(work, struct tcpm_port,
1273                                               vdm_state_machine.work);
1274         enum vdm_states prev_state;
1275
1276         mutex_lock(&port->lock);
1277
1278         /*
1279          * Continue running as long as the port is not busy and there was
1280          * a state change.
1281          */
1282         do {
1283                 prev_state = port->vdm_state;
1284                 vdm_run_state_machine(port);
1285         } while (port->vdm_state != prev_state &&
1286                  port->vdm_state != VDM_STATE_BUSY);
1287
1288         mutex_unlock(&port->lock);
1289 }
1290
1291 enum pdo_err {
1292         PDO_NO_ERR,
1293         PDO_ERR_NO_VSAFE5V,
1294         PDO_ERR_VSAFE5V_NOT_FIRST,
1295         PDO_ERR_PDO_TYPE_NOT_IN_ORDER,
1296         PDO_ERR_FIXED_NOT_SORTED,
1297         PDO_ERR_VARIABLE_BATT_NOT_SORTED,
1298         PDO_ERR_DUPE_PDO,
1299         PDO_ERR_PPS_APDO_NOT_SORTED,
1300         PDO_ERR_DUPE_PPS_APDO,
1301 };
1302
1303 static const char * const pdo_err_msg[] = {
1304         [PDO_ERR_NO_VSAFE5V] =
1305         " err: source/sink caps should atleast have vSafe5V",
1306         [PDO_ERR_VSAFE5V_NOT_FIRST] =
1307         " err: vSafe5V Fixed Supply Object Shall always be the first object",
1308         [PDO_ERR_PDO_TYPE_NOT_IN_ORDER] =
1309         " err: PDOs should be in the following order: Fixed; Battery; Variable",
1310         [PDO_ERR_FIXED_NOT_SORTED] =
1311         " err: Fixed supply pdos should be in increasing order of their fixed voltage",
1312         [PDO_ERR_VARIABLE_BATT_NOT_SORTED] =
1313         " err: Variable/Battery supply pdos should be in increasing order of their minimum voltage",
1314         [PDO_ERR_DUPE_PDO] =
1315         " err: Variable/Batt supply pdos cannot have same min/max voltage",
1316         [PDO_ERR_PPS_APDO_NOT_SORTED] =
1317         " err: Programmable power supply apdos should be in increasing order of their maximum voltage",
1318         [PDO_ERR_DUPE_PPS_APDO] =
1319         " err: Programmable power supply apdos cannot have same min/max voltage and max current",
1320 };
1321
1322 static enum pdo_err tcpm_caps_err(struct tcpm_port *port, const u32 *pdo,
1323                                   unsigned int nr_pdo)
1324 {
1325         unsigned int i;
1326
1327         /* Should at least contain vSafe5v */
1328         if (nr_pdo < 1)
1329                 return PDO_ERR_NO_VSAFE5V;
1330
1331         /* The vSafe5V Fixed Supply Object Shall always be the first object */
1332         if (pdo_type(pdo[0]) != PDO_TYPE_FIXED ||
1333             pdo_fixed_voltage(pdo[0]) != VSAFE5V)
1334                 return PDO_ERR_VSAFE5V_NOT_FIRST;
1335
1336         for (i = 1; i < nr_pdo; i++) {
1337                 if (pdo_type(pdo[i]) < pdo_type(pdo[i - 1])) {
1338                         return PDO_ERR_PDO_TYPE_NOT_IN_ORDER;
1339                 } else if (pdo_type(pdo[i]) == pdo_type(pdo[i - 1])) {
1340                         enum pd_pdo_type type = pdo_type(pdo[i]);
1341
1342                         switch (type) {
1343                         /*
1344                          * The remaining Fixed Supply Objects, if
1345                          * present, shall be sent in voltage order;
1346                          * lowest to highest.
1347                          */
1348                         case PDO_TYPE_FIXED:
1349                                 if (pdo_fixed_voltage(pdo[i]) <=
1350                                     pdo_fixed_voltage(pdo[i - 1]))
1351                                         return PDO_ERR_FIXED_NOT_SORTED;
1352                                 break;
1353                         /*
1354                          * The Battery Supply Objects and Variable
1355                          * supply, if present shall be sent in Minimum
1356                          * Voltage order; lowest to highest.
1357                          */
1358                         case PDO_TYPE_VAR:
1359                         case PDO_TYPE_BATT:
1360                                 if (pdo_min_voltage(pdo[i]) <
1361                                     pdo_min_voltage(pdo[i - 1]))
1362                                         return PDO_ERR_VARIABLE_BATT_NOT_SORTED;
1363                                 else if ((pdo_min_voltage(pdo[i]) ==
1364                                           pdo_min_voltage(pdo[i - 1])) &&
1365                                          (pdo_max_voltage(pdo[i]) ==
1366                                           pdo_min_voltage(pdo[i - 1])))
1367                                         return PDO_ERR_DUPE_PDO;
1368                                 break;
1369                         /*
1370                          * The Programmable Power Supply APDOs, if present,
1371                          * shall be sent in Maximum Voltage order;
1372                          * lowest to highest.
1373                          */
1374                         case PDO_TYPE_APDO:
1375                                 if (pdo_apdo_type(pdo[i]) != APDO_TYPE_PPS)
1376                                         break;
1377
1378                                 if (pdo_pps_apdo_max_current(pdo[i]) <
1379                                     pdo_pps_apdo_max_current(pdo[i - 1]))
1380                                         return PDO_ERR_PPS_APDO_NOT_SORTED;
1381                                 else if (pdo_pps_apdo_min_voltage(pdo[i]) ==
1382                                           pdo_pps_apdo_min_voltage(pdo[i - 1]) &&
1383                                          pdo_pps_apdo_max_voltage(pdo[i]) ==
1384                                           pdo_pps_apdo_max_voltage(pdo[i - 1]) &&
1385                                          pdo_pps_apdo_max_current(pdo[i]) ==
1386                                           pdo_pps_apdo_max_current(pdo[i - 1]))
1387                                         return PDO_ERR_DUPE_PPS_APDO;
1388                                 break;
1389                         default:
1390                                 tcpm_log_force(port, " Unknown pdo type");
1391                         }
1392                 }
1393         }
1394
1395         return PDO_NO_ERR;
1396 }
1397
1398 static int tcpm_validate_caps(struct tcpm_port *port, const u32 *pdo,
1399                               unsigned int nr_pdo)
1400 {
1401         enum pdo_err err_index = tcpm_caps_err(port, pdo, nr_pdo);
1402
1403         if (err_index != PDO_NO_ERR) {
1404                 tcpm_log_force(port, " %s", pdo_err_msg[err_index]);
1405                 return -EINVAL;
1406         }
1407
1408         return 0;
1409 }
1410
1411 /*
1412  * PD (data, control) command handling functions
1413  */
1414 static inline enum tcpm_state ready_state(struct tcpm_port *port)
1415 {
1416         if (port->pwr_role == TYPEC_SOURCE)
1417                 return SRC_READY;
1418         else
1419                 return SNK_READY;
1420 }
1421
1422 static int tcpm_pd_send_control(struct tcpm_port *port,
1423                                 enum pd_ctrl_msg_type type);
1424
1425 static void tcpm_handle_alert(struct tcpm_port *port, const __le32 *payload,
1426                               int cnt)
1427 {
1428         u32 p0 = le32_to_cpu(payload[0]);
1429         unsigned int type = usb_pd_ado_type(p0);
1430
1431         if (!type) {
1432                 tcpm_log(port, "Alert message received with no type");
1433                 return;
1434         }
1435
1436         /* Just handling non-battery alerts for now */
1437         if (!(type & USB_PD_ADO_TYPE_BATT_STATUS_CHANGE)) {
1438                 switch (port->state) {
1439                 case SRC_READY:
1440                 case SNK_READY:
1441                         tcpm_set_state(port, GET_STATUS_SEND, 0);
1442                         break;
1443                 default:
1444                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1445                         break;
1446                 }
1447         }
1448 }
1449
1450 static void tcpm_pd_data_request(struct tcpm_port *port,
1451                                  const struct pd_message *msg)
1452 {
1453         enum pd_data_msg_type type = pd_header_type_le(msg->header);
1454         unsigned int cnt = pd_header_cnt_le(msg->header);
1455         unsigned int rev = pd_header_rev_le(msg->header);
1456         unsigned int i;
1457
1458         switch (type) {
1459         case PD_DATA_SOURCE_CAP:
1460                 if (port->pwr_role != TYPEC_SINK)
1461                         break;
1462
1463                 for (i = 0; i < cnt; i++)
1464                         port->source_caps[i] = le32_to_cpu(msg->payload[i]);
1465
1466                 port->nr_source_caps = cnt;
1467
1468                 tcpm_log_source_caps(port);
1469
1470                 tcpm_validate_caps(port, port->source_caps,
1471                                    port->nr_source_caps);
1472
1473                 /*
1474                  * Adjust revision in subsequent message headers, as required,
1475                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1476                  * support Rev 1.0 so just do nothing in that scenario.
1477                  */
1478                 if (rev == PD_REV10)
1479                         break;
1480
1481                 if (rev < PD_MAX_REV)
1482                         port->negotiated_rev = rev;
1483
1484                 /*
1485                  * This message may be received even if VBUS is not
1486                  * present. This is quite unexpected; see USB PD
1487                  * specification, sections 8.3.3.6.3.1 and 8.3.3.6.3.2.
1488                  * However, at the same time, we must be ready to
1489                  * receive this message and respond to it 15ms after
1490                  * receiving PS_RDY during power swap operations, no matter
1491                  * if VBUS is available or not (USB PD specification,
1492                  * section 6.5.9.2).
1493                  * So we need to accept the message either way,
1494                  * but be prepared to keep waiting for VBUS after it was
1495                  * handled.
1496                  */
1497                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
1498                 break;
1499         case PD_DATA_REQUEST:
1500                 if (port->pwr_role != TYPEC_SOURCE ||
1501                     cnt != 1) {
1502                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1503                         break;
1504                 }
1505
1506                 /*
1507                  * Adjust revision in subsequent message headers, as required,
1508                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1509                  * support Rev 1.0 so just reject in that scenario.
1510                  */
1511                 if (rev == PD_REV10) {
1512                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1513                         break;
1514                 }
1515
1516                 if (rev < PD_MAX_REV)
1517                         port->negotiated_rev = rev;
1518
1519                 port->sink_request = le32_to_cpu(msg->payload[0]);
1520                 tcpm_set_state(port, SRC_NEGOTIATE_CAPABILITIES, 0);
1521                 break;
1522         case PD_DATA_SINK_CAP:
1523                 /* We don't do anything with this at the moment... */
1524                 for (i = 0; i < cnt; i++)
1525                         port->sink_caps[i] = le32_to_cpu(msg->payload[i]);
1526                 port->nr_sink_caps = cnt;
1527                 break;
1528         case PD_DATA_VENDOR_DEF:
1529                 tcpm_handle_vdm_request(port, msg->payload, cnt);
1530                 break;
1531         case PD_DATA_BIST:
1532                 if (port->state == SRC_READY || port->state == SNK_READY) {
1533                         port->bist_request = le32_to_cpu(msg->payload[0]);
1534                         tcpm_set_state(port, BIST_RX, 0);
1535                 }
1536                 break;
1537         case PD_DATA_ALERT:
1538                 tcpm_handle_alert(port, msg->payload, cnt);
1539                 break;
1540         case PD_DATA_BATT_STATUS:
1541         case PD_DATA_GET_COUNTRY_INFO:
1542                 /* Currently unsupported */
1543                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1544                 break;
1545         default:
1546                 tcpm_log(port, "Unhandled data message type %#x", type);
1547                 break;
1548         }
1549 }
1550
1551 static void tcpm_pps_complete(struct tcpm_port *port, int result)
1552 {
1553         if (port->pps_pending) {
1554                 port->pps_status = result;
1555                 port->pps_pending = false;
1556                 complete(&port->pps_complete);
1557         }
1558 }
1559
1560 static void tcpm_pd_ctrl_request(struct tcpm_port *port,
1561                                  const struct pd_message *msg)
1562 {
1563         enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1564         enum tcpm_state next_state;
1565
1566         switch (type) {
1567         case PD_CTRL_GOOD_CRC:
1568         case PD_CTRL_PING:
1569                 break;
1570         case PD_CTRL_GET_SOURCE_CAP:
1571                 switch (port->state) {
1572                 case SRC_READY:
1573                 case SNK_READY:
1574                         tcpm_queue_message(port, PD_MSG_DATA_SOURCE_CAP);
1575                         break;
1576                 default:
1577                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1578                         break;
1579                 }
1580                 break;
1581         case PD_CTRL_GET_SINK_CAP:
1582                 switch (port->state) {
1583                 case SRC_READY:
1584                 case SNK_READY:
1585                         tcpm_queue_message(port, PD_MSG_DATA_SINK_CAP);
1586                         break;
1587                 default:
1588                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1589                         break;
1590                 }
1591                 break;
1592         case PD_CTRL_GOTO_MIN:
1593                 break;
1594         case PD_CTRL_PS_RDY:
1595                 switch (port->state) {
1596                 case SNK_TRANSITION_SINK:
1597                         if (port->vbus_present) {
1598                                 tcpm_set_current_limit(port,
1599                                                        port->current_limit,
1600                                                        port->supply_voltage);
1601                                 port->explicit_contract = true;
1602                                 tcpm_set_state(port, SNK_READY, 0);
1603                         } else {
1604                                 /*
1605                                  * Seen after power swap. Keep waiting for VBUS
1606                                  * in a transitional state.
1607                                  */
1608                                 tcpm_set_state(port,
1609                                                SNK_TRANSITION_SINK_VBUS, 0);
1610                         }
1611                         break;
1612                 case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
1613                         tcpm_set_state(port, PR_SWAP_SRC_SNK_SINK_ON, 0);
1614                         break;
1615                 case PR_SWAP_SNK_SRC_SINK_OFF:
1616                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON, 0);
1617                         break;
1618                 case VCONN_SWAP_WAIT_FOR_VCONN:
1619                         tcpm_set_state(port, VCONN_SWAP_TURN_OFF_VCONN, 0);
1620                         break;
1621                 default:
1622                         break;
1623                 }
1624                 break;
1625         case PD_CTRL_REJECT:
1626         case PD_CTRL_WAIT:
1627         case PD_CTRL_NOT_SUPP:
1628                 switch (port->state) {
1629                 case SNK_NEGOTIATE_CAPABILITIES:
1630                         /* USB PD specification, Figure 8-43 */
1631                         if (port->explicit_contract)
1632                                 next_state = SNK_READY;
1633                         else
1634                                 next_state = SNK_WAIT_CAPABILITIES;
1635                         tcpm_set_state(port, next_state, 0);
1636                         break;
1637                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1638                         /* Revert data back from any requested PPS updates */
1639                         port->pps_data.out_volt = port->supply_voltage;
1640                         port->pps_data.op_curr = port->current_limit;
1641                         port->pps_status = (type == PD_CTRL_WAIT ?
1642                                             -EAGAIN : -EOPNOTSUPP);
1643                         tcpm_set_state(port, SNK_READY, 0);
1644                         break;
1645                 case DR_SWAP_SEND:
1646                         port->swap_status = (type == PD_CTRL_WAIT ?
1647                                              -EAGAIN : -EOPNOTSUPP);
1648                         tcpm_set_state(port, DR_SWAP_CANCEL, 0);
1649                         break;
1650                 case PR_SWAP_SEND:
1651                         port->swap_status = (type == PD_CTRL_WAIT ?
1652                                              -EAGAIN : -EOPNOTSUPP);
1653                         tcpm_set_state(port, PR_SWAP_CANCEL, 0);
1654                         break;
1655                 case VCONN_SWAP_SEND:
1656                         port->swap_status = (type == PD_CTRL_WAIT ?
1657                                              -EAGAIN : -EOPNOTSUPP);
1658                         tcpm_set_state(port, VCONN_SWAP_CANCEL, 0);
1659                         break;
1660                 default:
1661                         break;
1662                 }
1663                 break;
1664         case PD_CTRL_ACCEPT:
1665                 switch (port->state) {
1666                 case SNK_NEGOTIATE_CAPABILITIES:
1667                         port->pps_data.active = false;
1668                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1669                         break;
1670                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1671                         port->pps_data.active = true;
1672                         port->supply_voltage = port->pps_data.out_volt;
1673                         port->current_limit = port->pps_data.op_curr;
1674                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1675                         break;
1676                 case SOFT_RESET_SEND:
1677                         port->message_id = 0;
1678                         port->rx_msgid = -1;
1679                         if (port->pwr_role == TYPEC_SOURCE)
1680                                 next_state = SRC_SEND_CAPABILITIES;
1681                         else
1682                                 next_state = SNK_WAIT_CAPABILITIES;
1683                         tcpm_set_state(port, next_state, 0);
1684                         break;
1685                 case DR_SWAP_SEND:
1686                         tcpm_set_state(port, DR_SWAP_CHANGE_DR, 0);
1687                         break;
1688                 case PR_SWAP_SEND:
1689                         tcpm_set_state(port, PR_SWAP_START, 0);
1690                         break;
1691                 case VCONN_SWAP_SEND:
1692                         tcpm_set_state(port, VCONN_SWAP_START, 0);
1693                         break;
1694                 default:
1695                         break;
1696                 }
1697                 break;
1698         case PD_CTRL_SOFT_RESET:
1699                 tcpm_set_state(port, SOFT_RESET, 0);
1700                 break;
1701         case PD_CTRL_DR_SWAP:
1702                 if (port->port_type != TYPEC_PORT_DRP) {
1703                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1704                         break;
1705                 }
1706                 /*
1707                  * XXX
1708                  * 6.3.9: If an alternate mode is active, a request to swap
1709                  * alternate modes shall trigger a port reset.
1710                  */
1711                 switch (port->state) {
1712                 case SRC_READY:
1713                 case SNK_READY:
1714                         tcpm_set_state(port, DR_SWAP_ACCEPT, 0);
1715                         break;
1716                 default:
1717                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1718                         break;
1719                 }
1720                 break;
1721         case PD_CTRL_PR_SWAP:
1722                 if (port->port_type != TYPEC_PORT_DRP) {
1723                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1724                         break;
1725                 }
1726                 switch (port->state) {
1727                 case SRC_READY:
1728                 case SNK_READY:
1729                         tcpm_set_state(port, PR_SWAP_ACCEPT, 0);
1730                         break;
1731                 default:
1732                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1733                         break;
1734                 }
1735                 break;
1736         case PD_CTRL_VCONN_SWAP:
1737                 switch (port->state) {
1738                 case SRC_READY:
1739                 case SNK_READY:
1740                         tcpm_set_state(port, VCONN_SWAP_ACCEPT, 0);
1741                         break;
1742                 default:
1743                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1744                         break;
1745                 }
1746                 break;
1747         case PD_CTRL_GET_SOURCE_CAP_EXT:
1748         case PD_CTRL_GET_STATUS:
1749         case PD_CTRL_FR_SWAP:
1750         case PD_CTRL_GET_PPS_STATUS:
1751         case PD_CTRL_GET_COUNTRY_CODES:
1752                 /* Currently not supported */
1753                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1754                 break;
1755         default:
1756                 tcpm_log(port, "Unhandled ctrl message type %#x", type);
1757                 break;
1758         }
1759 }
1760
1761 static void tcpm_pd_ext_msg_request(struct tcpm_port *port,
1762                                     const struct pd_message *msg)
1763 {
1764         enum pd_ext_msg_type type = pd_header_type_le(msg->header);
1765         unsigned int data_size = pd_ext_header_data_size_le(msg->ext_msg.header);
1766
1767         if (!(msg->ext_msg.header & PD_EXT_HDR_CHUNKED)) {
1768                 tcpm_log(port, "Unchunked extended messages unsupported");
1769                 return;
1770         }
1771
1772         if (data_size > PD_EXT_MAX_CHUNK_DATA) {
1773                 tcpm_log(port, "Chunk handling not yet supported");
1774                 return;
1775         }
1776
1777         switch (type) {
1778         case PD_EXT_STATUS:
1779                 /*
1780                  * If PPS related events raised then get PPS status to clear
1781                  * (see USB PD 3.0 Spec, 6.5.2.4)
1782                  */
1783                 if (msg->ext_msg.data[USB_PD_EXT_SDB_EVENT_FLAGS] &
1784                     USB_PD_EXT_SDB_PPS_EVENTS)
1785                         tcpm_set_state(port, GET_PPS_STATUS_SEND, 0);
1786                 else
1787                         tcpm_set_state(port, ready_state(port), 0);
1788                 break;
1789         case PD_EXT_PPS_STATUS:
1790                 /*
1791                  * For now the PPS status message is used to clear events
1792                  * and nothing more.
1793                  */
1794                 tcpm_set_state(port, ready_state(port), 0);
1795                 break;
1796         case PD_EXT_SOURCE_CAP_EXT:
1797         case PD_EXT_GET_BATT_CAP:
1798         case PD_EXT_GET_BATT_STATUS:
1799         case PD_EXT_BATT_CAP:
1800         case PD_EXT_GET_MANUFACTURER_INFO:
1801         case PD_EXT_MANUFACTURER_INFO:
1802         case PD_EXT_SECURITY_REQUEST:
1803         case PD_EXT_SECURITY_RESPONSE:
1804         case PD_EXT_FW_UPDATE_REQUEST:
1805         case PD_EXT_FW_UPDATE_RESPONSE:
1806         case PD_EXT_COUNTRY_INFO:
1807         case PD_EXT_COUNTRY_CODES:
1808                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1809                 break;
1810         default:
1811                 tcpm_log(port, "Unhandled extended message type %#x", type);
1812                 break;
1813         }
1814 }
1815
1816 static void tcpm_pd_rx_handler(struct work_struct *work)
1817 {
1818         struct pd_rx_event *event = container_of(work,
1819                                                  struct pd_rx_event, work);
1820         const struct pd_message *msg = &event->msg;
1821         unsigned int cnt = pd_header_cnt_le(msg->header);
1822         struct tcpm_port *port = event->port;
1823
1824         mutex_lock(&port->lock);
1825
1826         tcpm_log(port, "PD RX, header: %#x [%d]", le16_to_cpu(msg->header),
1827                  port->attached);
1828
1829         if (port->attached) {
1830                 enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1831                 unsigned int msgid = pd_header_msgid_le(msg->header);
1832
1833                 /*
1834                  * USB PD standard, 6.6.1.2:
1835                  * "... if MessageID value in a received Message is the
1836                  * same as the stored value, the receiver shall return a
1837                  * GoodCRC Message with that MessageID value and drop
1838                  * the Message (this is a retry of an already received
1839                  * Message). Note: this shall not apply to the Soft_Reset
1840                  * Message which always has a MessageID value of zero."
1841                  */
1842                 if (msgid == port->rx_msgid && type != PD_CTRL_SOFT_RESET)
1843                         goto done;
1844                 port->rx_msgid = msgid;
1845
1846                 /*
1847                  * If both ends believe to be DFP/host, we have a data role
1848                  * mismatch.
1849                  */
1850                 if (!!(le16_to_cpu(msg->header) & PD_HEADER_DATA_ROLE) ==
1851                     (port->data_role == TYPEC_HOST)) {
1852                         tcpm_log(port,
1853                                  "Data role mismatch, initiating error recovery");
1854                         tcpm_set_state(port, ERROR_RECOVERY, 0);
1855                 } else {
1856                         if (msg->header & PD_HEADER_EXT_HDR)
1857                                 tcpm_pd_ext_msg_request(port, msg);
1858                         else if (cnt)
1859                                 tcpm_pd_data_request(port, msg);
1860                         else
1861                                 tcpm_pd_ctrl_request(port, msg);
1862                 }
1863         }
1864
1865 done:
1866         mutex_unlock(&port->lock);
1867         kfree(event);
1868 }
1869
1870 void tcpm_pd_receive(struct tcpm_port *port, const struct pd_message *msg)
1871 {
1872         struct pd_rx_event *event;
1873
1874         event = kzalloc(sizeof(*event), GFP_ATOMIC);
1875         if (!event)
1876                 return;
1877
1878         INIT_WORK(&event->work, tcpm_pd_rx_handler);
1879         event->port = port;
1880         memcpy(&event->msg, msg, sizeof(*msg));
1881         queue_work(port->wq, &event->work);
1882 }
1883 EXPORT_SYMBOL_GPL(tcpm_pd_receive);
1884
1885 static int tcpm_pd_send_control(struct tcpm_port *port,
1886                                 enum pd_ctrl_msg_type type)
1887 {
1888         struct pd_message msg;
1889
1890         memset(&msg, 0, sizeof(msg));
1891         msg.header = PD_HEADER_LE(type, port->pwr_role,
1892                                   port->data_role,
1893                                   port->negotiated_rev,
1894                                   port->message_id, 0);
1895
1896         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1897 }
1898
1899 /*
1900  * Send queued message without affecting state.
1901  * Return true if state machine should go back to sleep,
1902  * false otherwise.
1903  */
1904 static bool tcpm_send_queued_message(struct tcpm_port *port)
1905 {
1906         enum pd_msg_request queued_message;
1907
1908         do {
1909                 queued_message = port->queued_message;
1910                 port->queued_message = PD_MSG_NONE;
1911
1912                 switch (queued_message) {
1913                 case PD_MSG_CTRL_WAIT:
1914                         tcpm_pd_send_control(port, PD_CTRL_WAIT);
1915                         break;
1916                 case PD_MSG_CTRL_REJECT:
1917                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
1918                         break;
1919                 case PD_MSG_CTRL_NOT_SUPP:
1920                         tcpm_pd_send_control(port, PD_CTRL_NOT_SUPP);
1921                         break;
1922                 case PD_MSG_DATA_SINK_CAP:
1923                         tcpm_pd_send_sink_caps(port);
1924                         break;
1925                 case PD_MSG_DATA_SOURCE_CAP:
1926                         tcpm_pd_send_source_caps(port);
1927                         break;
1928                 default:
1929                         break;
1930                 }
1931         } while (port->queued_message != PD_MSG_NONE);
1932
1933         if (port->delayed_state != INVALID_STATE) {
1934                 if (time_is_after_jiffies(port->delayed_runtime)) {
1935                         mod_delayed_work(port->wq, &port->state_machine,
1936                                          port->delayed_runtime - jiffies);
1937                         return true;
1938                 }
1939                 port->delayed_state = INVALID_STATE;
1940         }
1941         return false;
1942 }
1943
1944 static int tcpm_pd_check_request(struct tcpm_port *port)
1945 {
1946         u32 pdo, rdo = port->sink_request;
1947         unsigned int max, op, pdo_max, index;
1948         enum pd_pdo_type type;
1949
1950         index = rdo_index(rdo);
1951         if (!index || index > port->nr_src_pdo)
1952                 return -EINVAL;
1953
1954         pdo = port->src_pdo[index - 1];
1955         type = pdo_type(pdo);
1956         switch (type) {
1957         case PDO_TYPE_FIXED:
1958         case PDO_TYPE_VAR:
1959                 max = rdo_max_current(rdo);
1960                 op = rdo_op_current(rdo);
1961                 pdo_max = pdo_max_current(pdo);
1962
1963                 if (op > pdo_max)
1964                         return -EINVAL;
1965                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1966                         return -EINVAL;
1967
1968                 if (type == PDO_TYPE_FIXED)
1969                         tcpm_log(port,
1970                                  "Requested %u mV, %u mA for %u / %u mA",
1971                                  pdo_fixed_voltage(pdo), pdo_max, op, max);
1972                 else
1973                         tcpm_log(port,
1974                                  "Requested %u -> %u mV, %u mA for %u / %u mA",
1975                                  pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1976                                  pdo_max, op, max);
1977                 break;
1978         case PDO_TYPE_BATT:
1979                 max = rdo_max_power(rdo);
1980                 op = rdo_op_power(rdo);
1981                 pdo_max = pdo_max_power(pdo);
1982
1983                 if (op > pdo_max)
1984                         return -EINVAL;
1985                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1986                         return -EINVAL;
1987                 tcpm_log(port,
1988                          "Requested %u -> %u mV, %u mW for %u / %u mW",
1989                          pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1990                          pdo_max, op, max);
1991                 break;
1992         default:
1993                 return -EINVAL;
1994         }
1995
1996         port->op_vsafe5v = index == 1;
1997
1998         return 0;
1999 }
2000
2001 #define min_power(x, y) min(pdo_max_power(x), pdo_max_power(y))
2002 #define min_current(x, y) min(pdo_max_current(x), pdo_max_current(y))
2003
2004 static int tcpm_pd_select_pdo(struct tcpm_port *port, int *sink_pdo,
2005                               int *src_pdo)
2006 {
2007         unsigned int i, j, max_src_mv = 0, min_src_mv = 0, max_mw = 0,
2008                      max_mv = 0, src_mw = 0, src_ma = 0, max_snk_mv = 0,
2009                      min_snk_mv = 0;
2010         int ret = -EINVAL;
2011
2012         port->pps_data.supported = false;
2013         port->usb_type = POWER_SUPPLY_USB_TYPE_PD;
2014
2015         /*
2016          * Select the source PDO providing the most power which has a
2017          * matchig sink cap.
2018          */
2019         for (i = 0; i < port->nr_source_caps; i++) {
2020                 u32 pdo = port->source_caps[i];
2021                 enum pd_pdo_type type = pdo_type(pdo);
2022
2023                 switch (type) {
2024                 case PDO_TYPE_FIXED:
2025                         max_src_mv = pdo_fixed_voltage(pdo);
2026                         min_src_mv = max_src_mv;
2027                         break;
2028                 case PDO_TYPE_BATT:
2029                 case PDO_TYPE_VAR:
2030                         max_src_mv = pdo_max_voltage(pdo);
2031                         min_src_mv = pdo_min_voltage(pdo);
2032                         break;
2033                 case PDO_TYPE_APDO:
2034                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS) {
2035                                 port->pps_data.supported = true;
2036                                 port->usb_type =
2037                                         POWER_SUPPLY_USB_TYPE_PD_PPS;
2038                         }
2039                         continue;
2040                 default:
2041                         tcpm_log(port, "Invalid source PDO type, ignoring");
2042                         continue;
2043                 }
2044
2045                 switch (type) {
2046                 case PDO_TYPE_FIXED:
2047                 case PDO_TYPE_VAR:
2048                         src_ma = pdo_max_current(pdo);
2049                         src_mw = src_ma * min_src_mv / 1000;
2050                         break;
2051                 case PDO_TYPE_BATT:
2052                         src_mw = pdo_max_power(pdo);
2053                         break;
2054                 case PDO_TYPE_APDO:
2055                         continue;
2056                 default:
2057                         tcpm_log(port, "Invalid source PDO type, ignoring");
2058                         continue;
2059                 }
2060
2061                 for (j = 0; j < port->nr_snk_pdo; j++) {
2062                         pdo = port->snk_pdo[j];
2063
2064                         switch (pdo_type(pdo)) {
2065                         case PDO_TYPE_FIXED:
2066                                 max_snk_mv = pdo_fixed_voltage(pdo);
2067                                 min_snk_mv = max_snk_mv;
2068                                 break;
2069                         case PDO_TYPE_BATT:
2070                         case PDO_TYPE_VAR:
2071                                 max_snk_mv = pdo_max_voltage(pdo);
2072                                 min_snk_mv = pdo_min_voltage(pdo);
2073                                 break;
2074                         case PDO_TYPE_APDO:
2075                                 continue;
2076                         default:
2077                                 tcpm_log(port, "Invalid sink PDO type, ignoring");
2078                                 continue;
2079                         }
2080
2081                         if (max_src_mv <= max_snk_mv &&
2082                                 min_src_mv >= min_snk_mv) {
2083                                 /* Prefer higher voltages if available */
2084                                 if ((src_mw == max_mw && min_src_mv > max_mv) ||
2085                                                         src_mw > max_mw) {
2086                                         *src_pdo = i;
2087                                         *sink_pdo = j;
2088                                         max_mw = src_mw;
2089                                         max_mv = min_src_mv;
2090                                         ret = 0;
2091                                 }
2092                         }
2093                 }
2094         }
2095
2096         return ret;
2097 }
2098
2099 #define min_pps_apdo_current(x, y)      \
2100         min(pdo_pps_apdo_max_current(x), pdo_pps_apdo_max_current(y))
2101
2102 static unsigned int tcpm_pd_select_pps_apdo(struct tcpm_port *port)
2103 {
2104         unsigned int i, j, max_mw = 0, max_mv = 0;
2105         unsigned int min_src_mv, max_src_mv, src_ma, src_mw;
2106         unsigned int min_snk_mv, max_snk_mv, snk_ma;
2107         u32 pdo;
2108         unsigned int src_pdo = 0, snk_pdo = 0;
2109
2110         /*
2111          * Select the source PPS APDO providing the most power while staying
2112          * within the board's limits. We skip the first PDO as this is always
2113          * 5V 3A.
2114          */
2115         for (i = 1; i < port->nr_source_caps; ++i) {
2116                 pdo = port->source_caps[i];
2117
2118                 switch (pdo_type(pdo)) {
2119                 case PDO_TYPE_APDO:
2120                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2121                                 tcpm_log(port, "Not PPS APDO (source), ignoring");
2122                                 continue;
2123                         }
2124
2125                         min_src_mv = pdo_pps_apdo_min_voltage(pdo);
2126                         max_src_mv = pdo_pps_apdo_max_voltage(pdo);
2127                         src_ma = pdo_pps_apdo_max_current(pdo);
2128                         src_mw = (src_ma * max_src_mv) / 1000;
2129
2130                         /*
2131                          * Now search through the sink PDOs to find a matching
2132                          * PPS APDO. Again skip the first sink PDO as this will
2133                          * always be 5V 3A.
2134                          */
2135                         for (j = i; j < port->nr_snk_pdo; j++) {
2136                                 pdo = port->snk_pdo[j];
2137
2138                                 switch (pdo_type(pdo)) {
2139                                 case PDO_TYPE_APDO:
2140                                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2141                                                 tcpm_log(port,
2142                                                          "Not PPS APDO (sink), ignoring");
2143                                                 continue;
2144                                         }
2145
2146                                         min_snk_mv =
2147                                                 pdo_pps_apdo_min_voltage(pdo);
2148                                         max_snk_mv =
2149                                                 pdo_pps_apdo_max_voltage(pdo);
2150                                         snk_ma =
2151                                                 pdo_pps_apdo_max_current(pdo);
2152                                         break;
2153                                 default:
2154                                         tcpm_log(port,
2155                                                  "Not APDO type (sink), ignoring");
2156                                         continue;
2157                                 }
2158
2159                                 if (max_src_mv <= max_snk_mv &&
2160                                     min_src_mv >= min_snk_mv) {
2161                                         /* Prefer higher voltages if available */
2162                                         if ((src_mw == max_mw &&
2163                                              min_src_mv > max_mv) ||
2164                                             src_mw > max_mw) {
2165                                                 src_pdo = i;
2166                                                 snk_pdo = j;
2167                                                 max_mw = src_mw;
2168                                                 max_mv = max_src_mv;
2169                                         }
2170                                 }
2171                         }
2172
2173                         break;
2174                 default:
2175                         tcpm_log(port, "Not APDO type (source), ignoring");
2176                         continue;
2177                 }
2178         }
2179
2180         if (src_pdo) {
2181                 pdo = port->source_caps[src_pdo];
2182
2183                 port->pps_data.min_volt = pdo_pps_apdo_min_voltage(pdo);
2184                 port->pps_data.max_volt = pdo_pps_apdo_max_voltage(pdo);
2185                 port->pps_data.max_curr =
2186                         min_pps_apdo_current(pdo, port->snk_pdo[snk_pdo]);
2187                 port->pps_data.out_volt =
2188                         min(pdo_pps_apdo_max_voltage(pdo), port->pps_data.out_volt);
2189                 port->pps_data.op_curr =
2190                         min(port->pps_data.max_curr, port->pps_data.op_curr);
2191         }
2192
2193         return src_pdo;
2194 }
2195
2196 static int tcpm_pd_build_request(struct tcpm_port *port, u32 *rdo)
2197 {
2198         unsigned int mv, ma, mw, flags;
2199         unsigned int max_ma, max_mw;
2200         enum pd_pdo_type type;
2201         u32 pdo, matching_snk_pdo;
2202         int src_pdo_index = 0;
2203         int snk_pdo_index = 0;
2204         int ret;
2205
2206         ret = tcpm_pd_select_pdo(port, &snk_pdo_index, &src_pdo_index);
2207         if (ret < 0)
2208                 return ret;
2209
2210         pdo = port->source_caps[src_pdo_index];
2211         matching_snk_pdo = port->snk_pdo[snk_pdo_index];
2212         type = pdo_type(pdo);
2213
2214         switch (type) {
2215         case PDO_TYPE_FIXED:
2216                 mv = pdo_fixed_voltage(pdo);
2217                 break;
2218         case PDO_TYPE_BATT:
2219         case PDO_TYPE_VAR:
2220                 mv = pdo_min_voltage(pdo);
2221                 break;
2222         default:
2223                 tcpm_log(port, "Invalid PDO selected!");
2224                 return -EINVAL;
2225         }
2226
2227         /* Select maximum available current within the sink pdo's limit */
2228         if (type == PDO_TYPE_BATT) {
2229                 mw = min_power(pdo, matching_snk_pdo);
2230                 ma = 1000 * mw / mv;
2231         } else {
2232                 ma = min_current(pdo, matching_snk_pdo);
2233                 mw = ma * mv / 1000;
2234         }
2235
2236         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2237
2238         /* Set mismatch bit if offered power is less than operating power */
2239         max_ma = ma;
2240         max_mw = mw;
2241         if (mw < port->operating_snk_mw) {
2242                 flags |= RDO_CAP_MISMATCH;
2243                 if (type == PDO_TYPE_BATT &&
2244                     (pdo_max_power(matching_snk_pdo) > pdo_max_power(pdo)))
2245                         max_mw = pdo_max_power(matching_snk_pdo);
2246                 else if (pdo_max_current(matching_snk_pdo) >
2247                          pdo_max_current(pdo))
2248                         max_ma = pdo_max_current(matching_snk_pdo);
2249         }
2250
2251         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2252                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2253                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2254                  port->polarity);
2255
2256         if (type == PDO_TYPE_BATT) {
2257                 *rdo = RDO_BATT(src_pdo_index + 1, mw, max_mw, flags);
2258
2259                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mW%s",
2260                          src_pdo_index, mv, mw,
2261                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2262         } else {
2263                 *rdo = RDO_FIXED(src_pdo_index + 1, ma, max_ma, flags);
2264
2265                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mA%s",
2266                          src_pdo_index, mv, ma,
2267                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2268         }
2269
2270         port->current_limit = ma;
2271         port->supply_voltage = mv;
2272
2273         return 0;
2274 }
2275
2276 static int tcpm_pd_send_request(struct tcpm_port *port)
2277 {
2278         struct pd_message msg;
2279         int ret;
2280         u32 rdo;
2281
2282         ret = tcpm_pd_build_request(port, &rdo);
2283         if (ret < 0)
2284                 return ret;
2285
2286         memset(&msg, 0, sizeof(msg));
2287         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2288                                   port->pwr_role,
2289                                   port->data_role,
2290                                   port->negotiated_rev,
2291                                   port->message_id, 1);
2292         msg.payload[0] = cpu_to_le32(rdo);
2293
2294         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2295 }
2296
2297 static int tcpm_pd_build_pps_request(struct tcpm_port *port, u32 *rdo)
2298 {
2299         unsigned int out_mv, op_ma, op_mw, min_mv, max_mv, max_ma, flags;
2300         enum pd_pdo_type type;
2301         unsigned int src_pdo_index;
2302         u32 pdo;
2303
2304         src_pdo_index = tcpm_pd_select_pps_apdo(port);
2305         if (!src_pdo_index)
2306                 return -EOPNOTSUPP;
2307
2308         pdo = port->source_caps[src_pdo_index];
2309         type = pdo_type(pdo);
2310
2311         switch (type) {
2312         case PDO_TYPE_APDO:
2313                 if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2314                         tcpm_log(port, "Invalid APDO selected!");
2315                         return -EINVAL;
2316                 }
2317                 min_mv = port->pps_data.min_volt;
2318                 max_mv = port->pps_data.max_volt;
2319                 max_ma = port->pps_data.max_curr;
2320                 out_mv = port->pps_data.out_volt;
2321                 op_ma = port->pps_data.op_curr;
2322                 break;
2323         default:
2324                 tcpm_log(port, "Invalid PDO selected!");
2325                 return -EINVAL;
2326         }
2327
2328         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2329
2330         op_mw = (op_ma * out_mv) / 1000;
2331         if (op_mw < port->operating_snk_mw) {
2332                 /*
2333                  * Try raising current to meet power needs. If that's not enough
2334                  * then try upping the voltage. If that's still not enough
2335                  * then we've obviously chosen a PPS APDO which really isn't
2336                  * suitable so abandon ship.
2337                  */
2338                 op_ma = (port->operating_snk_mw * 1000) / out_mv;
2339                 if ((port->operating_snk_mw * 1000) % out_mv)
2340                         ++op_ma;
2341                 op_ma += RDO_PROG_CURR_MA_STEP - (op_ma % RDO_PROG_CURR_MA_STEP);
2342
2343                 if (op_ma > max_ma) {
2344                         op_ma = max_ma;
2345                         out_mv = (port->operating_snk_mw * 1000) / op_ma;
2346                         if ((port->operating_snk_mw * 1000) % op_ma)
2347                                 ++out_mv;
2348                         out_mv += RDO_PROG_VOLT_MV_STEP -
2349                                   (out_mv % RDO_PROG_VOLT_MV_STEP);
2350
2351                         if (out_mv > max_mv) {
2352                                 tcpm_log(port, "Invalid PPS APDO selected!");
2353                                 return -EINVAL;
2354                         }
2355                 }
2356         }
2357
2358         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2359                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2360                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2361                  port->polarity);
2362
2363         *rdo = RDO_PROG(src_pdo_index + 1, out_mv, op_ma, flags);
2364
2365         tcpm_log(port, "Requesting APDO %d: %u mV, %u mA",
2366                  src_pdo_index, out_mv, op_ma);
2367
2368         port->pps_data.op_curr = op_ma;
2369         port->pps_data.out_volt = out_mv;
2370
2371         return 0;
2372 }
2373
2374 static int tcpm_pd_send_pps_request(struct tcpm_port *port)
2375 {
2376         struct pd_message msg;
2377         int ret;
2378         u32 rdo;
2379
2380         ret = tcpm_pd_build_pps_request(port, &rdo);
2381         if (ret < 0)
2382                 return ret;
2383
2384         memset(&msg, 0, sizeof(msg));
2385         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2386                                   port->pwr_role,
2387                                   port->data_role,
2388                                   port->negotiated_rev,
2389                                   port->message_id, 1);
2390         msg.payload[0] = cpu_to_le32(rdo);
2391
2392         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2393 }
2394
2395 static int tcpm_set_vbus(struct tcpm_port *port, bool enable)
2396 {
2397         int ret;
2398
2399         if (enable && port->vbus_charge)
2400                 return -EINVAL;
2401
2402         tcpm_log(port, "vbus:=%d charge=%d", enable, port->vbus_charge);
2403
2404         ret = port->tcpc->set_vbus(port->tcpc, enable, port->vbus_charge);
2405         if (ret < 0)
2406                 return ret;
2407
2408         port->vbus_source = enable;
2409         return 0;
2410 }
2411
2412 static int tcpm_set_charge(struct tcpm_port *port, bool charge)
2413 {
2414         int ret;
2415
2416         if (charge && port->vbus_source)
2417                 return -EINVAL;
2418
2419         if (charge != port->vbus_charge) {
2420                 tcpm_log(port, "vbus=%d charge:=%d", port->vbus_source, charge);
2421                 ret = port->tcpc->set_vbus(port->tcpc, port->vbus_source,
2422                                            charge);
2423                 if (ret < 0)
2424                         return ret;
2425         }
2426         port->vbus_charge = charge;
2427         return 0;
2428 }
2429
2430 static bool tcpm_start_drp_toggling(struct tcpm_port *port,
2431                                     enum typec_cc_status cc)
2432 {
2433         int ret;
2434
2435         if (port->tcpc->start_drp_toggling &&
2436             port->port_type == TYPEC_PORT_DRP) {
2437                 tcpm_log_force(port, "Start DRP toggling");
2438                 ret = port->tcpc->start_drp_toggling(port->tcpc, cc);
2439                 if (!ret)
2440                         return true;
2441         }
2442
2443         return false;
2444 }
2445
2446 static void tcpm_set_cc(struct tcpm_port *port, enum typec_cc_status cc)
2447 {
2448         tcpm_log(port, "cc:=%d", cc);
2449         port->cc_req = cc;
2450         port->tcpc->set_cc(port->tcpc, cc);
2451 }
2452
2453 static int tcpm_init_vbus(struct tcpm_port *port)
2454 {
2455         int ret;
2456
2457         ret = port->tcpc->set_vbus(port->tcpc, false, false);
2458         port->vbus_source = false;
2459         port->vbus_charge = false;
2460         return ret;
2461 }
2462
2463 static int tcpm_init_vconn(struct tcpm_port *port)
2464 {
2465         int ret;
2466
2467         ret = port->tcpc->set_vconn(port->tcpc, false);
2468         port->vconn_role = TYPEC_SINK;
2469         return ret;
2470 }
2471
2472 static void tcpm_typec_connect(struct tcpm_port *port)
2473 {
2474         if (!port->connected) {
2475                 /* Make sure we don't report stale identity information */
2476                 memset(&port->partner_ident, 0, sizeof(port->partner_ident));
2477                 port->partner_desc.usb_pd = port->pd_capable;
2478                 if (tcpm_port_is_debug(port))
2479                         port->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
2480                 else if (tcpm_port_is_audio(port))
2481                         port->partner_desc.accessory = TYPEC_ACCESSORY_AUDIO;
2482                 else
2483                         port->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
2484                 port->partner = typec_register_partner(port->typec_port,
2485                                                        &port->partner_desc);
2486                 port->connected = true;
2487         }
2488 }
2489
2490 static int tcpm_src_attach(struct tcpm_port *port)
2491 {
2492         enum typec_cc_polarity polarity =
2493                                 port->cc2 == TYPEC_CC_RD ? TYPEC_POLARITY_CC2
2494                                                          : TYPEC_POLARITY_CC1;
2495         int ret;
2496
2497         if (port->attached)
2498                 return 0;
2499
2500         ret = tcpm_set_polarity(port, polarity);
2501         if (ret < 0)
2502                 return ret;
2503
2504         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2505         if (ret < 0)
2506                 return ret;
2507
2508         ret = port->tcpc->set_pd_rx(port->tcpc, true);
2509         if (ret < 0)
2510                 goto out_disable_mux;
2511
2512         /*
2513          * USB Type-C specification, version 1.2,
2514          * chapter 4.5.2.2.8.1 (Attached.SRC Requirements)
2515          * Enable VCONN only if the non-RD port is set to RA.
2516          */
2517         if ((polarity == TYPEC_POLARITY_CC1 && port->cc2 == TYPEC_CC_RA) ||
2518             (polarity == TYPEC_POLARITY_CC2 && port->cc1 == TYPEC_CC_RA)) {
2519                 ret = tcpm_set_vconn(port, true);
2520                 if (ret < 0)
2521                         goto out_disable_pd;
2522         }
2523
2524         ret = tcpm_set_vbus(port, true);
2525         if (ret < 0)
2526                 goto out_disable_vconn;
2527
2528         port->pd_capable = false;
2529
2530         port->partner = NULL;
2531
2532         port->attached = true;
2533         port->send_discover = true;
2534
2535         return 0;
2536
2537 out_disable_vconn:
2538         tcpm_set_vconn(port, false);
2539 out_disable_pd:
2540         port->tcpc->set_pd_rx(port->tcpc, false);
2541 out_disable_mux:
2542         tcpm_mux_set(port, TYPEC_MUX_NONE, USB_ROLE_NONE,
2543                      TYPEC_ORIENTATION_NONE);
2544         return ret;
2545 }
2546
2547 static void tcpm_typec_disconnect(struct tcpm_port *port)
2548 {
2549         if (port->connected) {
2550                 typec_unregister_partner(port->partner);
2551                 port->partner = NULL;
2552                 port->connected = false;
2553         }
2554 }
2555
2556 static void tcpm_unregister_altmodes(struct tcpm_port *port)
2557 {
2558         struct pd_mode_data *modep = &port->mode_data;
2559         int i;
2560
2561         for (i = 0; i < modep->altmodes; i++) {
2562                 typec_unregister_altmode(port->partner_altmode[i]);
2563                 port->partner_altmode[i] = NULL;
2564         }
2565
2566         memset(modep, 0, sizeof(*modep));
2567 }
2568
2569 static void tcpm_reset_port(struct tcpm_port *port)
2570 {
2571         tcpm_unregister_altmodes(port);
2572         tcpm_typec_disconnect(port);
2573         port->attached = false;
2574         port->pd_capable = false;
2575         port->pps_data.supported = false;
2576
2577         /*
2578          * First Rx ID should be 0; set this to a sentinel of -1 so that
2579          * we can check tcpm_pd_rx_handler() if we had seen it before.
2580          */
2581         port->rx_msgid = -1;
2582
2583         port->tcpc->set_pd_rx(port->tcpc, false);
2584         tcpm_init_vbus(port);   /* also disables charging */
2585         tcpm_init_vconn(port);
2586         tcpm_set_current_limit(port, 0, 0);
2587         tcpm_set_polarity(port, TYPEC_POLARITY_CC1);
2588         tcpm_mux_set(port, TYPEC_MUX_NONE, USB_ROLE_NONE,
2589                      TYPEC_ORIENTATION_NONE);
2590         tcpm_set_attached_state(port, false);
2591         port->try_src_count = 0;
2592         port->try_snk_count = 0;
2593         port->supply_voltage = 0;
2594         port->current_limit = 0;
2595         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
2596
2597         power_supply_changed(port->psy);
2598 }
2599
2600 static void tcpm_detach(struct tcpm_port *port)
2601 {
2602         if (!port->attached)
2603                 return;
2604
2605         if (tcpm_port_is_disconnected(port))
2606                 port->hard_reset_count = 0;
2607
2608         tcpm_reset_port(port);
2609 }
2610
2611 static void tcpm_src_detach(struct tcpm_port *port)
2612 {
2613         tcpm_detach(port);
2614 }
2615
2616 static int tcpm_snk_attach(struct tcpm_port *port)
2617 {
2618         int ret;
2619
2620         if (port->attached)
2621                 return 0;
2622
2623         ret = tcpm_set_polarity(port, port->cc2 != TYPEC_CC_OPEN ?
2624                                 TYPEC_POLARITY_CC2 : TYPEC_POLARITY_CC1);
2625         if (ret < 0)
2626                 return ret;
2627
2628         ret = tcpm_set_roles(port, true, TYPEC_SINK, TYPEC_DEVICE);
2629         if (ret < 0)
2630                 return ret;
2631
2632         port->pd_capable = false;
2633
2634         port->partner = NULL;
2635
2636         port->attached = true;
2637         port->send_discover = true;
2638
2639         return 0;
2640 }
2641
2642 static void tcpm_snk_detach(struct tcpm_port *port)
2643 {
2644         tcpm_detach(port);
2645 }
2646
2647 static int tcpm_acc_attach(struct tcpm_port *port)
2648 {
2649         int ret;
2650
2651         if (port->attached)
2652                 return 0;
2653
2654         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2655         if (ret < 0)
2656                 return ret;
2657
2658         port->partner = NULL;
2659
2660         tcpm_typec_connect(port);
2661
2662         port->attached = true;
2663
2664         return 0;
2665 }
2666
2667 static void tcpm_acc_detach(struct tcpm_port *port)
2668 {
2669         tcpm_detach(port);
2670 }
2671
2672 static inline enum tcpm_state hard_reset_state(struct tcpm_port *port)
2673 {
2674         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
2675                 return HARD_RESET_SEND;
2676         if (port->pd_capable)
2677                 return ERROR_RECOVERY;
2678         if (port->pwr_role == TYPEC_SOURCE)
2679                 return SRC_UNATTACHED;
2680         if (port->state == SNK_WAIT_CAPABILITIES)
2681                 return SNK_READY;
2682         return SNK_UNATTACHED;
2683 }
2684
2685 static inline enum tcpm_state unattached_state(struct tcpm_port *port)
2686 {
2687         if (port->port_type == TYPEC_PORT_DRP) {
2688                 if (port->pwr_role == TYPEC_SOURCE)
2689                         return SRC_UNATTACHED;
2690                 else
2691                         return SNK_UNATTACHED;
2692         } else if (port->port_type == TYPEC_PORT_SRC) {
2693                 return SRC_UNATTACHED;
2694         }
2695
2696         return SNK_UNATTACHED;
2697 }
2698
2699 static void tcpm_check_send_discover(struct tcpm_port *port)
2700 {
2701         if (port->data_role == TYPEC_HOST && port->send_discover &&
2702             port->pd_capable) {
2703                 tcpm_send_vdm(port, USB_SID_PD, CMD_DISCOVER_IDENT, NULL, 0);
2704                 port->send_discover = false;
2705         }
2706 }
2707
2708 static void tcpm_swap_complete(struct tcpm_port *port, int result)
2709 {
2710         if (port->swap_pending) {
2711                 port->swap_status = result;
2712                 port->swap_pending = false;
2713                 port->non_pd_role_swap = false;
2714                 complete(&port->swap_complete);
2715         }
2716 }
2717
2718 static enum typec_pwr_opmode tcpm_get_pwr_opmode(enum typec_cc_status cc)
2719 {
2720         switch (cc) {
2721         case TYPEC_CC_RP_1_5:
2722                 return TYPEC_PWR_MODE_1_5A;
2723         case TYPEC_CC_RP_3_0:
2724                 return TYPEC_PWR_MODE_3_0A;
2725         case TYPEC_CC_RP_DEF:
2726         default:
2727                 return TYPEC_PWR_MODE_USB;
2728         }
2729 }
2730
2731 static void run_state_machine(struct tcpm_port *port)
2732 {
2733         int ret;
2734         enum typec_pwr_opmode opmode;
2735         unsigned int msecs;
2736
2737         port->enter_state = port->state;
2738         switch (port->state) {
2739         case DRP_TOGGLING:
2740                 break;
2741         /* SRC states */
2742         case SRC_UNATTACHED:
2743                 if (!port->non_pd_role_swap)
2744                         tcpm_swap_complete(port, -ENOTCONN);
2745                 tcpm_src_detach(port);
2746                 if (tcpm_start_drp_toggling(port, tcpm_rp_cc(port))) {
2747                         tcpm_set_state(port, DRP_TOGGLING, 0);
2748                         break;
2749                 }
2750                 tcpm_set_cc(port, tcpm_rp_cc(port));
2751                 if (port->port_type == TYPEC_PORT_DRP)
2752                         tcpm_set_state(port, SNK_UNATTACHED, PD_T_DRP_SNK);
2753                 break;
2754         case SRC_ATTACH_WAIT:
2755                 if (tcpm_port_is_debug(port))
2756                         tcpm_set_state(port, DEBUG_ACC_ATTACHED,
2757                                        PD_T_CC_DEBOUNCE);
2758                 else if (tcpm_port_is_audio(port))
2759                         tcpm_set_state(port, AUDIO_ACC_ATTACHED,
2760                                        PD_T_CC_DEBOUNCE);
2761                 else if (tcpm_port_is_source(port))
2762                         tcpm_set_state(port,
2763                                        tcpm_try_snk(port) ? SNK_TRY
2764                                                           : SRC_ATTACHED,
2765                                        PD_T_CC_DEBOUNCE);
2766                 break;
2767
2768         case SNK_TRY:
2769                 port->try_snk_count++;
2770                 /*
2771                  * Requirements:
2772                  * - Do not drive vconn or vbus
2773                  * - Terminate CC pins (both) to Rd
2774                  * Action:
2775                  * - Wait for tDRPTry (PD_T_DRP_TRY).
2776                  *   Until then, ignore any state changes.
2777                  */
2778                 tcpm_set_cc(port, TYPEC_CC_RD);
2779                 tcpm_set_state(port, SNK_TRY_WAIT, PD_T_DRP_TRY);
2780                 break;
2781         case SNK_TRY_WAIT:
2782                 if (tcpm_port_is_sink(port)) {
2783                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE, 0);
2784                 } else {
2785                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2786                         port->max_wait = 0;
2787                 }
2788                 break;
2789         case SNK_TRY_WAIT_DEBOUNCE:
2790                 tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS,
2791                                PD_T_PD_DEBOUNCE);
2792                 break;
2793         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
2794                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2795                         tcpm_set_state(port, SNK_ATTACHED, 0);
2796                 } else {
2797                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2798                         port->max_wait = 0;
2799                 }
2800                 break;
2801         case SRC_TRYWAIT:
2802                 tcpm_set_cc(port, tcpm_rp_cc(port));
2803                 if (port->max_wait == 0) {
2804                         port->max_wait = jiffies +
2805                                          msecs_to_jiffies(PD_T_DRP_TRY);
2806                         tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2807                                        PD_T_DRP_TRY);
2808                 } else {
2809                         if (time_is_after_jiffies(port->max_wait))
2810                                 tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2811                                                jiffies_to_msecs(port->max_wait -
2812                                                                 jiffies));
2813                         else
2814                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2815                 }
2816                 break;
2817         case SRC_TRYWAIT_DEBOUNCE:
2818                 tcpm_set_state(port, SRC_ATTACHED, PD_T_CC_DEBOUNCE);
2819                 break;
2820         case SRC_TRYWAIT_UNATTACHED:
2821                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2822                 break;
2823
2824         case SRC_ATTACHED:
2825                 ret = tcpm_src_attach(port);
2826                 tcpm_set_state(port, SRC_UNATTACHED,
2827                                ret < 0 ? 0 : PD_T_PS_SOURCE_ON);
2828                 break;
2829         case SRC_STARTUP:
2830                 opmode =  tcpm_get_pwr_opmode(tcpm_rp_cc(port));
2831                 typec_set_pwr_opmode(port->typec_port, opmode);
2832                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
2833                 port->caps_count = 0;
2834                 port->negotiated_rev = PD_MAX_REV;
2835                 port->message_id = 0;
2836                 port->rx_msgid = -1;
2837                 port->explicit_contract = false;
2838                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
2839                 break;
2840         case SRC_SEND_CAPABILITIES:
2841                 port->caps_count++;
2842                 if (port->caps_count > PD_N_CAPS_COUNT) {
2843                         tcpm_set_state(port, SRC_READY, 0);
2844                         break;
2845                 }
2846                 ret = tcpm_pd_send_source_caps(port);
2847                 if (ret < 0) {
2848                         tcpm_set_state(port, SRC_SEND_CAPABILITIES,
2849                                        PD_T_SEND_SOURCE_CAP);
2850                 } else {
2851                         /*
2852                          * Per standard, we should clear the reset counter here.
2853                          * However, that can result in state machine hang-ups.
2854                          * Reset it only in READY state to improve stability.
2855                          */
2856                         /* port->hard_reset_count = 0; */
2857                         port->caps_count = 0;
2858                         port->pd_capable = true;
2859                         tcpm_set_state_cond(port, hard_reset_state(port),
2860                                             PD_T_SEND_SOURCE_CAP);
2861                 }
2862                 break;
2863         case SRC_NEGOTIATE_CAPABILITIES:
2864                 ret = tcpm_pd_check_request(port);
2865                 if (ret < 0) {
2866                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
2867                         if (!port->explicit_contract) {
2868                                 tcpm_set_state(port,
2869                                                SRC_WAIT_NEW_CAPABILITIES, 0);
2870                         } else {
2871                                 tcpm_set_state(port, SRC_READY, 0);
2872                         }
2873                 } else {
2874                         tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2875                         tcpm_set_state(port, SRC_TRANSITION_SUPPLY,
2876                                        PD_T_SRC_TRANSITION);
2877                 }
2878                 break;
2879         case SRC_TRANSITION_SUPPLY:
2880                 /* XXX: regulator_set_voltage(vbus, ...) */
2881                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
2882                 port->explicit_contract = true;
2883                 typec_set_pwr_opmode(port->typec_port, TYPEC_PWR_MODE_PD);
2884                 port->pwr_opmode = TYPEC_PWR_MODE_PD;
2885                 tcpm_set_state_cond(port, SRC_READY, 0);
2886                 break;
2887         case SRC_READY:
2888 #if 1
2889                 port->hard_reset_count = 0;
2890 #endif
2891                 port->try_src_count = 0;
2892
2893                 tcpm_swap_complete(port, 0);
2894                 tcpm_typec_connect(port);
2895
2896                 tcpm_check_send_discover(port);
2897                 /*
2898                  * 6.3.5
2899                  * Sending ping messages is not necessary if
2900                  * - the source operates at vSafe5V
2901                  * or
2902                  * - The system is not operating in PD mode
2903                  * or
2904                  * - Both partners are connected using a Type-C connector
2905                  *
2906                  * There is no actual need to send PD messages since the local
2907                  * port type-c and the spec does not clearly say whether PD is
2908                  * possible when type-c is connected to Type-A/B
2909                  */
2910                 break;
2911         case SRC_WAIT_NEW_CAPABILITIES:
2912                 /* Nothing to do... */
2913                 break;
2914
2915         /* SNK states */
2916         case SNK_UNATTACHED:
2917                 if (!port->non_pd_role_swap)
2918                         tcpm_swap_complete(port, -ENOTCONN);
2919                 tcpm_pps_complete(port, -ENOTCONN);
2920                 tcpm_snk_detach(port);
2921                 if (tcpm_start_drp_toggling(port, TYPEC_CC_RD)) {
2922                         tcpm_set_state(port, DRP_TOGGLING, 0);
2923                         break;
2924                 }
2925                 tcpm_set_cc(port, TYPEC_CC_RD);
2926                 if (port->port_type == TYPEC_PORT_DRP)
2927                         tcpm_set_state(port, SRC_UNATTACHED, PD_T_DRP_SRC);
2928                 break;
2929         case SNK_ATTACH_WAIT:
2930                 if ((port->cc1 == TYPEC_CC_OPEN &&
2931                      port->cc2 != TYPEC_CC_OPEN) ||
2932                     (port->cc1 != TYPEC_CC_OPEN &&
2933                      port->cc2 == TYPEC_CC_OPEN))
2934                         tcpm_set_state(port, SNK_DEBOUNCED,
2935                                        PD_T_CC_DEBOUNCE);
2936                 else if (tcpm_port_is_disconnected(port))
2937                         tcpm_set_state(port, SNK_UNATTACHED,
2938                                        PD_T_PD_DEBOUNCE);
2939                 break;
2940         case SNK_DEBOUNCED:
2941                 if (tcpm_port_is_disconnected(port))
2942                         tcpm_set_state(port, SNK_UNATTACHED,
2943                                        PD_T_PD_DEBOUNCE);
2944                 else if (port->vbus_present)
2945                         tcpm_set_state(port,
2946                                        tcpm_try_src(port) ? SRC_TRY
2947                                                           : SNK_ATTACHED,
2948                                        0);
2949                 else
2950                         /* Wait for VBUS, but not forever */
2951                         tcpm_set_state(port, PORT_RESET, PD_T_PS_SOURCE_ON);
2952                 break;
2953
2954         case SRC_TRY:
2955                 port->try_src_count++;
2956                 tcpm_set_cc(port, tcpm_rp_cc(port));
2957                 port->max_wait = 0;
2958                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
2959                 break;
2960         case SRC_TRY_WAIT:
2961                 if (port->max_wait == 0) {
2962                         port->max_wait = jiffies +
2963                                          msecs_to_jiffies(PD_T_DRP_TRY);
2964                         msecs = PD_T_DRP_TRY;
2965                 } else {
2966                         if (time_is_after_jiffies(port->max_wait))
2967                                 msecs = jiffies_to_msecs(port->max_wait -
2968                                                          jiffies);
2969                         else
2970                                 msecs = 0;
2971                 }
2972                 tcpm_set_state(port, SNK_TRYWAIT, msecs);
2973                 break;
2974         case SRC_TRY_DEBOUNCE:
2975                 tcpm_set_state(port, SRC_ATTACHED, PD_T_PD_DEBOUNCE);
2976                 break;
2977         case SNK_TRYWAIT:
2978                 tcpm_set_cc(port, TYPEC_CC_RD);
2979                 tcpm_set_state(port, SNK_TRYWAIT_VBUS, PD_T_CC_DEBOUNCE);
2980                 break;
2981         case SNK_TRYWAIT_VBUS:
2982                 /*
2983                  * TCPM stays in this state indefinitely until VBUS
2984                  * is detected as long as Rp is not detected for
2985                  * more than a time period of tPDDebounce.
2986                  */
2987                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2988                         tcpm_set_state(port, SNK_ATTACHED, 0);
2989                         break;
2990                 }
2991                 if (!tcpm_port_is_sink(port))
2992                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
2993                 break;
2994         case SNK_TRYWAIT_DEBOUNCE:
2995                 tcpm_set_state(port, SNK_UNATTACHED, PD_T_PD_DEBOUNCE);
2996                 break;
2997         case SNK_ATTACHED:
2998                 ret = tcpm_snk_attach(port);
2999                 if (ret < 0)
3000                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3001                 else
3002                         tcpm_set_state(port, SNK_STARTUP, 0);
3003                 break;
3004         case SNK_STARTUP:
3005                 opmode =  tcpm_get_pwr_opmode(port->polarity ?
3006                                               port->cc2 : port->cc1);
3007                 typec_set_pwr_opmode(port->typec_port, opmode);
3008                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
3009                 port->negotiated_rev = PD_MAX_REV;
3010                 port->message_id = 0;
3011                 port->rx_msgid = -1;
3012                 port->explicit_contract = false;
3013                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3014                 break;
3015         case SNK_DISCOVERY:
3016                 if (port->vbus_present) {
3017                         tcpm_set_current_limit(port,
3018                                                tcpm_get_current_limit(port),
3019                                                5000);
3020                         tcpm_set_charge(port, true);
3021                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3022                         break;
3023                 }
3024                 /*
3025                  * For DRP, timeouts differ. Also, handling is supposed to be
3026                  * different and much more complex (dead battery detection;
3027                  * see USB power delivery specification, section 8.3.3.6.1.5.1).
3028                  */
3029                 tcpm_set_state(port, hard_reset_state(port),
3030                                port->port_type == TYPEC_PORT_DRP ?
3031                                         PD_T_DB_DETECT : PD_T_NO_RESPONSE);
3032                 break;
3033         case SNK_DISCOVERY_DEBOUNCE:
3034                 tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE_DONE,
3035                                PD_T_CC_DEBOUNCE);
3036                 break;
3037         case SNK_DISCOVERY_DEBOUNCE_DONE:
3038                 if (!tcpm_port_is_disconnected(port) &&
3039                     tcpm_port_is_sink(port) &&
3040                     time_is_after_jiffies(port->delayed_runtime)) {
3041                         tcpm_set_state(port, SNK_DISCOVERY,
3042                                        jiffies_to_msecs(port->delayed_runtime -
3043                                                         jiffies));
3044                         break;
3045                 }
3046                 tcpm_set_state(port, unattached_state(port), 0);
3047                 break;
3048         case SNK_WAIT_CAPABILITIES:
3049                 ret = port->tcpc->set_pd_rx(port->tcpc, true);
3050                 if (ret < 0) {
3051                         tcpm_set_state(port, SNK_READY, 0);
3052                         break;
3053                 }
3054                 /*
3055                  * If VBUS has never been low, and we time out waiting
3056                  * for source cap, try a soft reset first, in case we
3057                  * were already in a stable contract before this boot.
3058                  * Do this only once.
3059                  */
3060                 if (port->vbus_never_low) {
3061                         port->vbus_never_low = false;
3062                         tcpm_set_state(port, SOFT_RESET_SEND,
3063                                        PD_T_SINK_WAIT_CAP);
3064                 } else {
3065                         tcpm_set_state(port, hard_reset_state(port),
3066                                        PD_T_SINK_WAIT_CAP);
3067                 }
3068                 break;
3069         case SNK_NEGOTIATE_CAPABILITIES:
3070                 port->pd_capable = true;
3071                 port->hard_reset_count = 0;
3072                 ret = tcpm_pd_send_request(port);
3073                 if (ret < 0) {
3074                         /* Let the Source send capabilities again. */
3075                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3076                 } else {
3077                         tcpm_set_state_cond(port, hard_reset_state(port),
3078                                             PD_T_SENDER_RESPONSE);
3079                 }
3080                 break;
3081         case SNK_NEGOTIATE_PPS_CAPABILITIES:
3082                 ret = tcpm_pd_send_pps_request(port);
3083                 if (ret < 0) {
3084                         port->pps_status = ret;
3085                         /*
3086                          * If this was called due to updates to sink
3087                          * capabilities, and pps is no longer valid, we should
3088                          * safely fall back to a standard PDO.
3089                          */
3090                         if (port->update_sink_caps)
3091                                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
3092                         else
3093                                 tcpm_set_state(port, SNK_READY, 0);
3094                 } else {
3095                         tcpm_set_state_cond(port, hard_reset_state(port),
3096                                             PD_T_SENDER_RESPONSE);
3097                 }
3098                 break;
3099         case SNK_TRANSITION_SINK:
3100         case SNK_TRANSITION_SINK_VBUS:
3101                 tcpm_set_state(port, hard_reset_state(port),
3102                                PD_T_PS_TRANSITION);
3103                 break;
3104         case SNK_READY:
3105                 port->try_snk_count = 0;
3106                 port->update_sink_caps = false;
3107                 if (port->explicit_contract) {
3108                         typec_set_pwr_opmode(port->typec_port,
3109                                              TYPEC_PWR_MODE_PD);
3110                         port->pwr_opmode = TYPEC_PWR_MODE_PD;
3111                 }
3112
3113                 tcpm_swap_complete(port, 0);
3114                 tcpm_typec_connect(port);
3115                 tcpm_check_send_discover(port);
3116                 tcpm_pps_complete(port, port->pps_status);
3117
3118                 power_supply_changed(port->psy);
3119
3120                 break;
3121
3122         /* Accessory states */
3123         case ACC_UNATTACHED:
3124                 tcpm_acc_detach(port);
3125                 tcpm_set_state(port, SRC_UNATTACHED, 0);
3126                 break;
3127         case DEBUG_ACC_ATTACHED:
3128         case AUDIO_ACC_ATTACHED:
3129                 ret = tcpm_acc_attach(port);
3130                 if (ret < 0)
3131                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3132                 break;
3133         case AUDIO_ACC_DEBOUNCE:
3134                 tcpm_set_state(port, ACC_UNATTACHED, PD_T_CC_DEBOUNCE);
3135                 break;
3136
3137         /* Hard_Reset states */
3138         case HARD_RESET_SEND:
3139                 tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
3140                 tcpm_set_state(port, HARD_RESET_START, 0);
3141                 break;
3142         case HARD_RESET_START:
3143                 port->hard_reset_count++;
3144                 port->tcpc->set_pd_rx(port->tcpc, false);
3145                 tcpm_unregister_altmodes(port);
3146                 port->send_discover = true;
3147                 if (port->pwr_role == TYPEC_SOURCE)
3148                         tcpm_set_state(port, SRC_HARD_RESET_VBUS_OFF,
3149                                        PD_T_PS_HARD_RESET);
3150                 else
3151                         tcpm_set_state(port, SNK_HARD_RESET_SINK_OFF, 0);
3152                 break;
3153         case SRC_HARD_RESET_VBUS_OFF:
3154                 tcpm_set_vconn(port, true);
3155                 tcpm_set_vbus(port, false);
3156                 tcpm_set_roles(port, false, TYPEC_SOURCE, TYPEC_HOST);
3157                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SRC_RECOVER);
3158                 break;
3159         case SRC_HARD_RESET_VBUS_ON:
3160                 tcpm_set_vbus(port, true);
3161                 port->tcpc->set_pd_rx(port->tcpc, true);
3162                 tcpm_set_attached_state(port, true);
3163                 tcpm_set_state(port, SRC_UNATTACHED, PD_T_PS_SOURCE_ON);
3164                 break;
3165         case SNK_HARD_RESET_SINK_OFF:
3166                 memset(&port->pps_data, 0, sizeof(port->pps_data));
3167                 tcpm_set_vconn(port, false);
3168                 tcpm_set_charge(port, false);
3169                 tcpm_set_roles(port, false, TYPEC_SINK, TYPEC_DEVICE);
3170                 /*
3171                  * VBUS may or may not toggle, depending on the adapter.
3172                  * If it doesn't toggle, transition to SNK_HARD_RESET_SINK_ON
3173                  * directly after timeout.
3174                  */
3175                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, PD_T_SAFE_0V);
3176                 break;
3177         case SNK_HARD_RESET_WAIT_VBUS:
3178                 /* Assume we're disconnected if VBUS doesn't come back. */
3179                 tcpm_set_state(port, SNK_UNATTACHED,
3180                                PD_T_SRC_RECOVER_MAX + PD_T_SRC_TURN_ON);
3181                 break;
3182         case SNK_HARD_RESET_SINK_ON:
3183                 /* Note: There is no guarantee that VBUS is on in this state */
3184                 /*
3185                  * XXX:
3186                  * The specification suggests that dual mode ports in sink
3187                  * mode should transition to state PE_SRC_Transition_to_default.
3188                  * See USB power delivery specification chapter 8.3.3.6.1.3.
3189                  * This would mean to to
3190                  * - turn off VCONN, reset power supply
3191                  * - request hardware reset
3192                  * - turn on VCONN
3193                  * - Transition to state PE_Src_Startup
3194                  * SNK only ports shall transition to state Snk_Startup
3195                  * (see chapter 8.3.3.3.8).
3196                  * Similar, dual-mode ports in source mode should transition
3197                  * to PE_SNK_Transition_to_default.
3198                  */
3199                 tcpm_set_attached_state(port, true);
3200                 tcpm_set_state(port, SNK_STARTUP, 0);
3201                 break;
3202
3203         /* Soft_Reset states */
3204         case SOFT_RESET:
3205                 port->message_id = 0;
3206                 port->rx_msgid = -1;
3207                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3208                 if (port->pwr_role == TYPEC_SOURCE)
3209                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3210                 else
3211                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3212                 break;
3213         case SOFT_RESET_SEND:
3214                 port->message_id = 0;
3215                 port->rx_msgid = -1;
3216                 if (tcpm_pd_send_control(port, PD_CTRL_SOFT_RESET))
3217                         tcpm_set_state_cond(port, hard_reset_state(port), 0);
3218                 else
3219                         tcpm_set_state_cond(port, hard_reset_state(port),
3220                                             PD_T_SENDER_RESPONSE);
3221                 break;
3222
3223         /* DR_Swap states */
3224         case DR_SWAP_SEND:
3225                 tcpm_pd_send_control(port, PD_CTRL_DR_SWAP);
3226                 tcpm_set_state_cond(port, DR_SWAP_SEND_TIMEOUT,
3227                                     PD_T_SENDER_RESPONSE);
3228                 break;
3229         case DR_SWAP_ACCEPT:
3230                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3231                 tcpm_set_state_cond(port, DR_SWAP_CHANGE_DR, 0);
3232                 break;
3233         case DR_SWAP_SEND_TIMEOUT:
3234                 tcpm_swap_complete(port, -ETIMEDOUT);
3235                 tcpm_set_state(port, ready_state(port), 0);
3236                 break;
3237         case DR_SWAP_CHANGE_DR:
3238                 if (port->data_role == TYPEC_HOST) {
3239                         tcpm_unregister_altmodes(port);
3240                         tcpm_set_roles(port, true, port->pwr_role,
3241                                        TYPEC_DEVICE);
3242                 } else {
3243                         tcpm_set_roles(port, true, port->pwr_role,
3244                                        TYPEC_HOST);
3245                         port->send_discover = true;
3246                 }
3247                 tcpm_set_state(port, ready_state(port), 0);
3248                 break;
3249
3250         /* PR_Swap states */
3251         case PR_SWAP_ACCEPT:
3252                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3253                 tcpm_set_state(port, PR_SWAP_START, 0);
3254                 break;
3255         case PR_SWAP_SEND:
3256                 tcpm_pd_send_control(port, PD_CTRL_PR_SWAP);
3257                 tcpm_set_state_cond(port, PR_SWAP_SEND_TIMEOUT,
3258                                     PD_T_SENDER_RESPONSE);
3259                 break;
3260         case PR_SWAP_SEND_TIMEOUT:
3261                 tcpm_swap_complete(port, -ETIMEDOUT);
3262                 tcpm_set_state(port, ready_state(port), 0);
3263                 break;
3264         case PR_SWAP_START:
3265                 if (port->pwr_role == TYPEC_SOURCE)
3266                         tcpm_set_state(port, PR_SWAP_SRC_SNK_TRANSITION_OFF,
3267                                        PD_T_SRC_TRANSITION);
3268                 else
3269                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SINK_OFF, 0);
3270                 break;
3271         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3272                 tcpm_set_vbus(port, false);
3273                 port->explicit_contract = false;
3274                 /* allow time for Vbus discharge, must be < tSrcSwapStdby */
3275                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF,
3276                                PD_T_SRCSWAPSTDBY);
3277                 break;
3278         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3279                 tcpm_set_cc(port, TYPEC_CC_RD);
3280                 /* allow CC debounce */
3281                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED,
3282                                PD_T_CC_DEBOUNCE);
3283                 break;
3284         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3285                 /*
3286                  * USB-PD standard, 6.2.1.4, Port Power Role:
3287                  * "During the Power Role Swap Sequence, for the initial Source
3288                  * Port, the Port Power Role field shall be set to Sink in the
3289                  * PS_RDY Message indicating that the initial Source’s power
3290                  * supply is turned off"
3291                  */
3292                 tcpm_set_pwr_role(port, TYPEC_SINK);
3293                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
3294                         tcpm_set_state(port, ERROR_RECOVERY, 0);
3295                         break;
3296                 }
3297                 tcpm_set_state_cond(port, SNK_UNATTACHED, PD_T_PS_SOURCE_ON);
3298                 break;
3299         case PR_SWAP_SRC_SNK_SINK_ON:
3300                 tcpm_set_state(port, SNK_STARTUP, 0);
3301                 break;
3302         case PR_SWAP_SNK_SRC_SINK_OFF:
3303                 tcpm_set_charge(port, false);
3304                 tcpm_set_state(port, hard_reset_state(port),
3305                                PD_T_PS_SOURCE_OFF);
3306                 break;
3307         case PR_SWAP_SNK_SRC_SOURCE_ON:
3308                 tcpm_set_cc(port, tcpm_rp_cc(port));
3309                 tcpm_set_vbus(port, true);
3310                 /*
3311                  * allow time VBUS ramp-up, must be < tNewSrc
3312                  * Also, this window overlaps with CC debounce as well.
3313                  * So, Wait for the max of two which is PD_T_NEWSRC
3314                  */
3315                 tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP,
3316                                PD_T_NEWSRC);
3317                 break;
3318         case PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP:
3319                 /*
3320                  * USB PD standard, 6.2.1.4:
3321                  * "Subsequent Messages initiated by the Policy Engine,
3322                  * such as the PS_RDY Message sent to indicate that Vbus
3323                  * is ready, will have the Port Power Role field set to
3324                  * Source."
3325                  */
3326                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
3327                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3328                 tcpm_set_state(port, SRC_STARTUP, 0);
3329                 break;
3330
3331         case VCONN_SWAP_ACCEPT:
3332                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3333                 tcpm_set_state(port, VCONN_SWAP_START, 0);
3334                 break;
3335         case VCONN_SWAP_SEND:
3336                 tcpm_pd_send_control(port, PD_CTRL_VCONN_SWAP);
3337                 tcpm_set_state(port, VCONN_SWAP_SEND_TIMEOUT,
3338                                PD_T_SENDER_RESPONSE);
3339                 break;
3340         case VCONN_SWAP_SEND_TIMEOUT:
3341                 tcpm_swap_complete(port, -ETIMEDOUT);
3342                 tcpm_set_state(port, ready_state(port), 0);
3343                 break;
3344         case VCONN_SWAP_START:
3345                 if (port->vconn_role == TYPEC_SOURCE)
3346                         tcpm_set_state(port, VCONN_SWAP_WAIT_FOR_VCONN, 0);
3347                 else
3348                         tcpm_set_state(port, VCONN_SWAP_TURN_ON_VCONN, 0);
3349                 break;
3350         case VCONN_SWAP_WAIT_FOR_VCONN:
3351                 tcpm_set_state(port, hard_reset_state(port),
3352                                PD_T_VCONN_SOURCE_ON);
3353                 break;
3354         case VCONN_SWAP_TURN_ON_VCONN:
3355                 tcpm_set_vconn(port, true);
3356                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3357                 tcpm_set_state(port, ready_state(port), 0);
3358                 break;
3359         case VCONN_SWAP_TURN_OFF_VCONN:
3360                 tcpm_set_vconn(port, false);
3361                 tcpm_set_state(port, ready_state(port), 0);
3362                 break;
3363
3364         case DR_SWAP_CANCEL:
3365         case PR_SWAP_CANCEL:
3366         case VCONN_SWAP_CANCEL:
3367                 tcpm_swap_complete(port, port->swap_status);
3368                 if (port->pwr_role == TYPEC_SOURCE)
3369                         tcpm_set_state(port, SRC_READY, 0);
3370                 else
3371                         tcpm_set_state(port, SNK_READY, 0);
3372                 break;
3373
3374         case BIST_RX:
3375                 switch (BDO_MODE_MASK(port->bist_request)) {
3376                 case BDO_MODE_CARRIER2:
3377                         tcpm_pd_transmit(port, TCPC_TX_BIST_MODE_2, NULL);
3378                         break;
3379                 default:
3380                         break;
3381                 }
3382                 /* Always switch to unattached state */
3383                 tcpm_set_state(port, unattached_state(port), 0);
3384                 break;
3385         case GET_STATUS_SEND:
3386                 tcpm_pd_send_control(port, PD_CTRL_GET_STATUS);
3387                 tcpm_set_state(port, GET_STATUS_SEND_TIMEOUT,
3388                                PD_T_SENDER_RESPONSE);
3389                 break;
3390         case GET_STATUS_SEND_TIMEOUT:
3391                 tcpm_set_state(port, ready_state(port), 0);
3392                 break;
3393         case GET_PPS_STATUS_SEND:
3394                 tcpm_pd_send_control(port, PD_CTRL_GET_PPS_STATUS);
3395                 tcpm_set_state(port, GET_PPS_STATUS_SEND_TIMEOUT,
3396                                PD_T_SENDER_RESPONSE);
3397                 break;
3398         case GET_PPS_STATUS_SEND_TIMEOUT:
3399                 tcpm_set_state(port, ready_state(port), 0);
3400                 break;
3401         case ERROR_RECOVERY:
3402                 tcpm_swap_complete(port, -EPROTO);
3403                 tcpm_pps_complete(port, -EPROTO);
3404                 tcpm_set_state(port, PORT_RESET, 0);
3405                 break;
3406         case PORT_RESET:
3407                 tcpm_reset_port(port);
3408                 tcpm_set_cc(port, TYPEC_CC_OPEN);
3409                 tcpm_set_state(port, PORT_RESET_WAIT_OFF,
3410                                PD_T_ERROR_RECOVERY);
3411                 break;
3412         case PORT_RESET_WAIT_OFF:
3413                 tcpm_set_state(port,
3414                                tcpm_default_state(port),
3415                                port->vbus_present ? PD_T_PS_SOURCE_OFF : 0);
3416                 break;
3417         default:
3418                 WARN(1, "Unexpected port state %d\n", port->state);
3419                 break;
3420         }
3421 }
3422
3423 static void tcpm_state_machine_work(struct work_struct *work)
3424 {
3425         struct tcpm_port *port = container_of(work, struct tcpm_port,
3426                                               state_machine.work);
3427         enum tcpm_state prev_state;
3428
3429         mutex_lock(&port->lock);
3430         port->state_machine_running = true;
3431
3432         if (port->queued_message && tcpm_send_queued_message(port))
3433                 goto done;
3434
3435         /* If we were queued due to a delayed state change, update it now */
3436         if (port->delayed_state) {
3437                 tcpm_log(port, "state change %s -> %s [delayed %ld ms]",
3438                          tcpm_states[port->state],
3439                          tcpm_states[port->delayed_state], port->delay_ms);
3440                 port->prev_state = port->state;
3441                 port->state = port->delayed_state;
3442                 port->delayed_state = INVALID_STATE;
3443         }
3444
3445         /*
3446          * Continue running as long as we have (non-delayed) state changes
3447          * to make.
3448          */
3449         do {
3450                 prev_state = port->state;
3451                 run_state_machine(port);
3452                 if (port->queued_message)
3453                         tcpm_send_queued_message(port);
3454         } while (port->state != prev_state && !port->delayed_state);
3455
3456 done:
3457         port->state_machine_running = false;
3458         mutex_unlock(&port->lock);
3459 }
3460
3461 static void _tcpm_cc_change(struct tcpm_port *port, enum typec_cc_status cc1,
3462                             enum typec_cc_status cc2)
3463 {
3464         enum typec_cc_status old_cc1, old_cc2;
3465         enum tcpm_state new_state;
3466
3467         old_cc1 = port->cc1;
3468         old_cc2 = port->cc2;
3469         port->cc1 = cc1;
3470         port->cc2 = cc2;
3471
3472         tcpm_log_force(port,
3473                        "CC1: %u -> %u, CC2: %u -> %u [state %s, polarity %d, %s]",
3474                        old_cc1, cc1, old_cc2, cc2, tcpm_states[port->state],
3475                        port->polarity,
3476                        tcpm_port_is_disconnected(port) ? "disconnected"
3477                                                        : "connected");
3478
3479         switch (port->state) {
3480         case DRP_TOGGLING:
3481                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3482                     tcpm_port_is_source(port))
3483                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3484                 else if (tcpm_port_is_sink(port))
3485                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3486                 break;
3487         case SRC_UNATTACHED:
3488         case ACC_UNATTACHED:
3489                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3490                     tcpm_port_is_source(port))
3491                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3492                 break;
3493         case SRC_ATTACH_WAIT:
3494                 if (tcpm_port_is_disconnected(port) ||
3495                     tcpm_port_is_audio_detached(port))
3496                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3497                 else if (cc1 != old_cc1 || cc2 != old_cc2)
3498                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3499                 break;
3500         case SRC_ATTACHED:
3501         case SRC_SEND_CAPABILITIES:
3502         case SRC_READY:
3503                 if (tcpm_port_is_disconnected(port) ||
3504                     !tcpm_port_is_source(port))
3505                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3506                 break;
3507         case SNK_UNATTACHED:
3508                 if (tcpm_port_is_sink(port))
3509                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3510                 break;
3511         case SNK_ATTACH_WAIT:
3512                 if ((port->cc1 == TYPEC_CC_OPEN &&
3513                      port->cc2 != TYPEC_CC_OPEN) ||
3514                     (port->cc1 != TYPEC_CC_OPEN &&
3515                      port->cc2 == TYPEC_CC_OPEN))
3516                         new_state = SNK_DEBOUNCED;
3517                 else if (tcpm_port_is_disconnected(port))
3518                         new_state = SNK_UNATTACHED;
3519                 else
3520                         break;
3521                 if (new_state != port->delayed_state)
3522                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3523                 break;
3524         case SNK_DEBOUNCED:
3525                 if (tcpm_port_is_disconnected(port))
3526                         new_state = SNK_UNATTACHED;
3527                 else if (port->vbus_present)
3528                         new_state = tcpm_try_src(port) ? SRC_TRY : SNK_ATTACHED;
3529                 else
3530                         new_state = SNK_UNATTACHED;
3531                 if (new_state != port->delayed_state)
3532                         tcpm_set_state(port, SNK_DEBOUNCED, 0);
3533                 break;
3534         case SNK_READY:
3535                 if (tcpm_port_is_disconnected(port))
3536                         tcpm_set_state(port, unattached_state(port), 0);
3537                 else if (!port->pd_capable &&
3538                          (cc1 != old_cc1 || cc2 != old_cc2))
3539                         tcpm_set_current_limit(port,
3540                                                tcpm_get_current_limit(port),
3541                                                5000);
3542                 break;
3543
3544         case AUDIO_ACC_ATTACHED:
3545                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3546                         tcpm_set_state(port, AUDIO_ACC_DEBOUNCE, 0);
3547                 break;
3548         case AUDIO_ACC_DEBOUNCE:
3549                 if (tcpm_port_is_audio(port))
3550                         tcpm_set_state(port, AUDIO_ACC_ATTACHED, 0);
3551                 break;
3552
3553         case DEBUG_ACC_ATTACHED:
3554                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3555                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3556                 break;
3557
3558         case SNK_TRY:
3559                 /* Do nothing, waiting for timeout */
3560                 break;
3561
3562         case SNK_DISCOVERY:
3563                 /* CC line is unstable, wait for debounce */
3564                 if (tcpm_port_is_disconnected(port))
3565                         tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE, 0);
3566                 break;
3567         case SNK_DISCOVERY_DEBOUNCE:
3568                 break;
3569
3570         case SRC_TRYWAIT:
3571                 /* Hand over to state machine if needed */
3572                 if (!port->vbus_present && tcpm_port_is_source(port))
3573                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3574                 break;
3575         case SRC_TRYWAIT_DEBOUNCE:
3576                 if (port->vbus_present || !tcpm_port_is_source(port))
3577                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3578                 break;
3579         case SNK_TRY_WAIT_DEBOUNCE:
3580                 if (!tcpm_port_is_sink(port)) {
3581                         port->max_wait = 0;
3582                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3583                 }
3584                 break;
3585         case SRC_TRY_WAIT:
3586                 if (tcpm_port_is_source(port))
3587                         tcpm_set_state(port, SRC_TRY_DEBOUNCE, 0);
3588                 break;
3589         case SRC_TRY_DEBOUNCE:
3590                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
3591                 break;
3592         case SNK_TRYWAIT_DEBOUNCE:
3593                 if (tcpm_port_is_sink(port))
3594                         tcpm_set_state(port, SNK_TRYWAIT_VBUS, 0);
3595                 break;
3596         case SNK_TRYWAIT_VBUS:
3597                 if (!tcpm_port_is_sink(port))
3598                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
3599                 break;
3600         case SNK_TRYWAIT:
3601                 /* Do nothing, waiting for tCCDebounce */
3602                 break;
3603         case PR_SWAP_SNK_SRC_SINK_OFF:
3604         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3605         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3606         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3607         case PR_SWAP_SNK_SRC_SOURCE_ON:
3608                 /*
3609                  * CC state change is expected in PR_SWAP
3610                  * Ignore it.
3611                  */
3612                 break;
3613
3614         default:
3615                 if (tcpm_port_is_disconnected(port))
3616                         tcpm_set_state(port, unattached_state(port), 0);
3617                 break;
3618         }
3619 }
3620
3621 static void _tcpm_pd_vbus_on(struct tcpm_port *port)
3622 {
3623         tcpm_log_force(port, "VBUS on");
3624         port->vbus_present = true;
3625         switch (port->state) {
3626         case SNK_TRANSITION_SINK_VBUS:
3627                 port->explicit_contract = true;
3628                 tcpm_set_state(port, SNK_READY, 0);
3629                 break;
3630         case SNK_DISCOVERY:
3631                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3632                 break;
3633
3634         case SNK_DEBOUNCED:
3635                 tcpm_set_state(port, tcpm_try_src(port) ? SRC_TRY
3636                                                         : SNK_ATTACHED,
3637                                        0);
3638                 break;
3639         case SNK_HARD_RESET_WAIT_VBUS:
3640                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, 0);
3641                 break;
3642         case SRC_ATTACHED:
3643                 tcpm_set_state(port, SRC_STARTUP, 0);
3644                 break;
3645         case SRC_HARD_RESET_VBUS_ON:
3646                 tcpm_set_state(port, SRC_STARTUP, 0);
3647                 break;
3648
3649         case SNK_TRY:
3650                 /* Do nothing, waiting for timeout */
3651                 break;
3652         case SRC_TRYWAIT:
3653                 /* Do nothing, Waiting for Rd to be detected */
3654                 break;
3655         case SRC_TRYWAIT_DEBOUNCE:
3656                 tcpm_set_state(port, SRC_TRYWAIT, 0);
3657                 break;
3658         case SNK_TRY_WAIT_DEBOUNCE:
3659                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3660                 break;
3661         case SNK_TRYWAIT:
3662                 /* Do nothing, waiting for tCCDebounce */
3663                 break;
3664         case SNK_TRYWAIT_VBUS:
3665                 if (tcpm_port_is_sink(port))
3666                         tcpm_set_state(port, SNK_ATTACHED, 0);
3667                 break;
3668         case SNK_TRYWAIT_DEBOUNCE:
3669                 /* Do nothing, waiting for Rp */
3670                 break;
3671         case SRC_TRY_WAIT:
3672         case SRC_TRY_DEBOUNCE:
3673                 /* Do nothing, waiting for sink detection */
3674                 break;
3675         default:
3676                 break;
3677         }
3678 }
3679
3680 static void _tcpm_pd_vbus_off(struct tcpm_port *port)
3681 {
3682         tcpm_log_force(port, "VBUS off");
3683         port->vbus_present = false;
3684         port->vbus_never_low = false;
3685         switch (port->state) {
3686         case SNK_HARD_RESET_SINK_OFF:
3687                 tcpm_set_state(port, SNK_HARD_RESET_WAIT_VBUS, 0);
3688                 break;
3689         case SRC_HARD_RESET_VBUS_OFF:
3690                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, 0);
3691                 break;
3692         case HARD_RESET_SEND:
3693                 break;
3694
3695         case SNK_TRY:
3696                 /* Do nothing, waiting for timeout */
3697                 break;
3698         case SRC_TRYWAIT:
3699                 /* Hand over to state machine if needed */
3700                 if (tcpm_port_is_source(port))
3701                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3702                 break;
3703         case SNK_TRY_WAIT_DEBOUNCE:
3704                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3705                 break;
3706         case SNK_TRYWAIT:
3707         case SNK_TRYWAIT_VBUS:
3708         case SNK_TRYWAIT_DEBOUNCE:
3709                 break;
3710         case SNK_ATTACH_WAIT:
3711                 tcpm_set_state(port, SNK_UNATTACHED, 0);
3712                 break;
3713
3714         case SNK_NEGOTIATE_CAPABILITIES:
3715                 break;
3716
3717         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3718                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF, 0);
3719                 break;
3720
3721         case PR_SWAP_SNK_SRC_SINK_OFF:
3722                 /* Do nothing, expected */
3723                 break;
3724
3725         case PORT_RESET_WAIT_OFF:
3726                 tcpm_set_state(port, tcpm_default_state(port), 0);
3727                 break;
3728         case SRC_TRY_WAIT:
3729         case SRC_TRY_DEBOUNCE:
3730                 /* Do nothing, waiting for sink detection */
3731                 break;
3732         default:
3733                 if (port->pwr_role == TYPEC_SINK &&
3734                     port->attached)
3735                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3736                 break;
3737         }
3738 }
3739
3740 static void _tcpm_pd_hard_reset(struct tcpm_port *port)
3741 {
3742         tcpm_log_force(port, "Received hard reset");
3743         /*
3744          * If we keep receiving hard reset requests, executing the hard reset
3745          * must have failed. Revert to error recovery if that happens.
3746          */
3747         tcpm_set_state(port,
3748                        port->hard_reset_count < PD_N_HARD_RESET_COUNT ?
3749                                 HARD_RESET_START : ERROR_RECOVERY,
3750                        0);
3751 }
3752
3753 static void tcpm_pd_event_handler(struct work_struct *work)
3754 {
3755         struct tcpm_port *port = container_of(work, struct tcpm_port,
3756                                               event_work);
3757         u32 events;
3758
3759         mutex_lock(&port->lock);
3760
3761         spin_lock(&port->pd_event_lock);
3762         while (port->pd_events) {
3763                 events = port->pd_events;
3764                 port->pd_events = 0;
3765                 spin_unlock(&port->pd_event_lock);
3766                 if (events & TCPM_RESET_EVENT)
3767                         _tcpm_pd_hard_reset(port);
3768                 if (events & TCPM_VBUS_EVENT) {
3769                         bool vbus;
3770
3771                         vbus = port->tcpc->get_vbus(port->tcpc);
3772                         if (vbus)
3773                                 _tcpm_pd_vbus_on(port);
3774                         else
3775                                 _tcpm_pd_vbus_off(port);
3776                 }
3777                 if (events & TCPM_CC_EVENT) {
3778                         enum typec_cc_status cc1, cc2;
3779
3780                         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
3781                                 _tcpm_cc_change(port, cc1, cc2);
3782                 }
3783                 spin_lock(&port->pd_event_lock);
3784         }
3785         spin_unlock(&port->pd_event_lock);
3786         mutex_unlock(&port->lock);
3787 }
3788
3789 void tcpm_cc_change(struct tcpm_port *port)
3790 {
3791         spin_lock(&port->pd_event_lock);
3792         port->pd_events |= TCPM_CC_EVENT;
3793         spin_unlock(&port->pd_event_lock);
3794         queue_work(port->wq, &port->event_work);
3795 }
3796 EXPORT_SYMBOL_GPL(tcpm_cc_change);
3797
3798 void tcpm_vbus_change(struct tcpm_port *port)
3799 {
3800         spin_lock(&port->pd_event_lock);
3801         port->pd_events |= TCPM_VBUS_EVENT;
3802         spin_unlock(&port->pd_event_lock);
3803         queue_work(port->wq, &port->event_work);
3804 }
3805 EXPORT_SYMBOL_GPL(tcpm_vbus_change);
3806
3807 void tcpm_pd_hard_reset(struct tcpm_port *port)
3808 {
3809         spin_lock(&port->pd_event_lock);
3810         port->pd_events = TCPM_RESET_EVENT;
3811         spin_unlock(&port->pd_event_lock);
3812         queue_work(port->wq, &port->event_work);
3813 }
3814 EXPORT_SYMBOL_GPL(tcpm_pd_hard_reset);
3815
3816 static int tcpm_dr_set(const struct typec_capability *cap,
3817                        enum typec_data_role data)
3818 {
3819         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3820         int ret;
3821
3822         mutex_lock(&port->swap_lock);
3823         mutex_lock(&port->lock);
3824
3825         if (port->port_type != TYPEC_PORT_DRP) {
3826                 ret = -EINVAL;
3827                 goto port_unlock;
3828         }
3829         if (port->state != SRC_READY && port->state != SNK_READY) {
3830                 ret = -EAGAIN;
3831                 goto port_unlock;
3832         }
3833
3834         if (port->data_role == data) {
3835                 ret = 0;
3836                 goto port_unlock;
3837         }
3838
3839         /*
3840          * XXX
3841          * 6.3.9: If an alternate mode is active, a request to swap
3842          * alternate modes shall trigger a port reset.
3843          * Reject data role swap request in this case.
3844          */
3845
3846         if (!port->pd_capable) {
3847                 /*
3848                  * If the partner is not PD capable, reset the port to
3849                  * trigger a role change. This can only work if a preferred
3850                  * role is configured, and if it matches the requested role.
3851                  */
3852                 if (port->try_role == TYPEC_NO_PREFERRED_ROLE ||
3853                     port->try_role == port->pwr_role) {
3854                         ret = -EINVAL;
3855                         goto port_unlock;
3856                 }
3857                 port->non_pd_role_swap = true;
3858                 tcpm_set_state(port, PORT_RESET, 0);
3859         } else {
3860                 tcpm_set_state(port, DR_SWAP_SEND, 0);
3861         }
3862
3863         port->swap_status = 0;
3864         port->swap_pending = true;
3865         reinit_completion(&port->swap_complete);
3866         mutex_unlock(&port->lock);
3867
3868         if (!wait_for_completion_timeout(&port->swap_complete,
3869                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3870                 ret = -ETIMEDOUT;
3871         else
3872                 ret = port->swap_status;
3873
3874         port->non_pd_role_swap = false;
3875         goto swap_unlock;
3876
3877 port_unlock:
3878         mutex_unlock(&port->lock);
3879 swap_unlock:
3880         mutex_unlock(&port->swap_lock);
3881         return ret;
3882 }
3883
3884 static int tcpm_pr_set(const struct typec_capability *cap,
3885                        enum typec_role role)
3886 {
3887         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3888         int ret;
3889
3890         mutex_lock(&port->swap_lock);
3891         mutex_lock(&port->lock);
3892
3893         if (port->port_type != TYPEC_PORT_DRP) {
3894                 ret = -EINVAL;
3895                 goto port_unlock;
3896         }
3897         if (port->state != SRC_READY && port->state != SNK_READY) {
3898                 ret = -EAGAIN;
3899                 goto port_unlock;
3900         }
3901
3902         if (role == port->pwr_role) {
3903                 ret = 0;
3904                 goto port_unlock;
3905         }
3906
3907         port->swap_status = 0;
3908         port->swap_pending = true;
3909         reinit_completion(&port->swap_complete);
3910         tcpm_set_state(port, PR_SWAP_SEND, 0);
3911         mutex_unlock(&port->lock);
3912
3913         if (!wait_for_completion_timeout(&port->swap_complete,
3914                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3915                 ret = -ETIMEDOUT;
3916         else
3917                 ret = port->swap_status;
3918
3919         goto swap_unlock;
3920
3921 port_unlock:
3922         mutex_unlock(&port->lock);
3923 swap_unlock:
3924         mutex_unlock(&port->swap_lock);
3925         return ret;
3926 }
3927
3928 static int tcpm_vconn_set(const struct typec_capability *cap,
3929                           enum typec_role role)
3930 {
3931         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3932         int ret;
3933
3934         mutex_lock(&port->swap_lock);
3935         mutex_lock(&port->lock);
3936
3937         if (port->state != SRC_READY && port->state != SNK_READY) {
3938                 ret = -EAGAIN;
3939                 goto port_unlock;
3940         }
3941
3942         if (role == port->vconn_role) {
3943                 ret = 0;
3944                 goto port_unlock;
3945         }
3946
3947         port->swap_status = 0;
3948         port->swap_pending = true;
3949         reinit_completion(&port->swap_complete);
3950         tcpm_set_state(port, VCONN_SWAP_SEND, 0);
3951         mutex_unlock(&port->lock);
3952
3953         if (!wait_for_completion_timeout(&port->swap_complete,
3954                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3955                 ret = -ETIMEDOUT;
3956         else
3957                 ret = port->swap_status;
3958
3959         goto swap_unlock;
3960
3961 port_unlock:
3962         mutex_unlock(&port->lock);
3963 swap_unlock:
3964         mutex_unlock(&port->swap_lock);
3965         return ret;
3966 }
3967
3968 static int tcpm_try_role(const struct typec_capability *cap, int role)
3969 {
3970         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3971         struct tcpc_dev *tcpc = port->tcpc;
3972         int ret = 0;
3973
3974         mutex_lock(&port->lock);
3975         if (tcpc->try_role)
3976                 ret = tcpc->try_role(tcpc, role);
3977         if (!ret && !tcpc->config->try_role_hw)
3978                 port->try_role = role;
3979         port->try_src_count = 0;
3980         port->try_snk_count = 0;
3981         mutex_unlock(&port->lock);
3982
3983         return ret;
3984 }
3985
3986 static int tcpm_pps_set_op_curr(struct tcpm_port *port, u16 op_curr)
3987 {
3988         unsigned int target_mw;
3989         int ret;
3990
3991         mutex_lock(&port->swap_lock);
3992         mutex_lock(&port->lock);
3993
3994         if (!port->pps_data.active) {
3995                 ret = -EOPNOTSUPP;
3996                 goto port_unlock;
3997         }
3998
3999         if (port->state != SNK_READY) {
4000                 ret = -EAGAIN;
4001                 goto port_unlock;
4002         }
4003
4004         if (op_curr > port->pps_data.max_curr) {
4005                 ret = -EINVAL;
4006                 goto port_unlock;
4007         }
4008
4009         target_mw = (op_curr * port->pps_data.out_volt) / 1000;
4010         if (target_mw < port->operating_snk_mw) {
4011                 ret = -EINVAL;
4012                 goto port_unlock;
4013         }
4014
4015         reinit_completion(&port->pps_complete);
4016         port->pps_data.op_curr = op_curr;
4017         port->pps_status = 0;
4018         port->pps_pending = true;
4019         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4020         mutex_unlock(&port->lock);
4021
4022         if (!wait_for_completion_timeout(&port->pps_complete,
4023                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4024                 ret = -ETIMEDOUT;
4025         else
4026                 ret = port->pps_status;
4027
4028         goto swap_unlock;
4029
4030 port_unlock:
4031         mutex_unlock(&port->lock);
4032 swap_unlock:
4033         mutex_unlock(&port->swap_lock);
4034
4035         return ret;
4036 }
4037
4038 static int tcpm_pps_set_out_volt(struct tcpm_port *port, u16 out_volt)
4039 {
4040         unsigned int target_mw;
4041         int ret;
4042
4043         mutex_lock(&port->swap_lock);
4044         mutex_lock(&port->lock);
4045
4046         if (!port->pps_data.active) {
4047                 ret = -EOPNOTSUPP;
4048                 goto port_unlock;
4049         }
4050
4051         if (port->state != SNK_READY) {
4052                 ret = -EAGAIN;
4053                 goto port_unlock;
4054         }
4055
4056         if (out_volt < port->pps_data.min_volt ||
4057             out_volt > port->pps_data.max_volt) {
4058                 ret = -EINVAL;
4059                 goto port_unlock;
4060         }
4061
4062         target_mw = (port->pps_data.op_curr * out_volt) / 1000;
4063         if (target_mw < port->operating_snk_mw) {
4064                 ret = -EINVAL;
4065                 goto port_unlock;
4066         }
4067
4068         reinit_completion(&port->pps_complete);
4069         port->pps_data.out_volt = out_volt;
4070         port->pps_status = 0;
4071         port->pps_pending = true;
4072         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4073         mutex_unlock(&port->lock);
4074
4075         if (!wait_for_completion_timeout(&port->pps_complete,
4076                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4077                 ret = -ETIMEDOUT;
4078         else
4079                 ret = port->pps_status;
4080
4081         goto swap_unlock;
4082
4083 port_unlock:
4084         mutex_unlock(&port->lock);
4085 swap_unlock:
4086         mutex_unlock(&port->swap_lock);
4087
4088         return ret;
4089 }
4090
4091 static int tcpm_pps_activate(struct tcpm_port *port, bool activate)
4092 {
4093         int ret = 0;
4094
4095         mutex_lock(&port->swap_lock);
4096         mutex_lock(&port->lock);
4097
4098         if (!port->pps_data.supported) {
4099                 ret = -EOPNOTSUPP;
4100                 goto port_unlock;
4101         }
4102
4103         /* Trying to deactivate PPS when already deactivated so just bail */
4104         if (!port->pps_data.active && !activate)
4105                 goto port_unlock;
4106
4107         if (port->state != SNK_READY) {
4108                 ret = -EAGAIN;
4109                 goto port_unlock;
4110         }
4111
4112         reinit_completion(&port->pps_complete);
4113         port->pps_status = 0;
4114         port->pps_pending = true;
4115
4116         /* Trigger PPS request or move back to standard PDO contract */
4117         if (activate) {
4118                 port->pps_data.out_volt = port->supply_voltage;
4119                 port->pps_data.op_curr = port->current_limit;
4120                 tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4121         } else {
4122                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4123         }
4124         mutex_unlock(&port->lock);
4125
4126         if (!wait_for_completion_timeout(&port->pps_complete,
4127                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4128                 ret = -ETIMEDOUT;
4129         else
4130                 ret = port->pps_status;
4131
4132         goto swap_unlock;
4133
4134 port_unlock:
4135         mutex_unlock(&port->lock);
4136 swap_unlock:
4137         mutex_unlock(&port->swap_lock);
4138
4139         return ret;
4140 }
4141
4142 static void tcpm_init(struct tcpm_port *port)
4143 {
4144         enum typec_cc_status cc1, cc2;
4145
4146         port->tcpc->init(port->tcpc);
4147
4148         tcpm_reset_port(port);
4149
4150         /*
4151          * XXX
4152          * Should possibly wait for VBUS to settle if it was enabled locally
4153          * since tcpm_reset_port() will disable VBUS.
4154          */
4155         port->vbus_present = port->tcpc->get_vbus(port->tcpc);
4156         if (port->vbus_present)
4157                 port->vbus_never_low = true;
4158
4159         tcpm_set_state(port, tcpm_default_state(port), 0);
4160
4161         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
4162                 _tcpm_cc_change(port, cc1, cc2);
4163
4164         /*
4165          * Some adapters need a clean slate at startup, and won't recover
4166          * otherwise. So do not try to be fancy and force a clean disconnect.
4167          */
4168         tcpm_set_state(port, PORT_RESET, 0);
4169 }
4170
4171 static int tcpm_port_type_set(const struct typec_capability *cap,
4172                               enum typec_port_type type)
4173 {
4174         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4175
4176         mutex_lock(&port->lock);
4177         if (type == port->port_type)
4178                 goto port_unlock;
4179
4180         port->port_type = type;
4181
4182         if (!port->connected) {
4183                 tcpm_set_state(port, PORT_RESET, 0);
4184         } else if (type == TYPEC_PORT_SNK) {
4185                 if (!(port->pwr_role == TYPEC_SINK &&
4186                       port->data_role == TYPEC_DEVICE))
4187                         tcpm_set_state(port, PORT_RESET, 0);
4188         } else if (type == TYPEC_PORT_SRC) {
4189                 if (!(port->pwr_role == TYPEC_SOURCE &&
4190                       port->data_role == TYPEC_HOST))
4191                         tcpm_set_state(port, PORT_RESET, 0);
4192         }
4193
4194 port_unlock:
4195         mutex_unlock(&port->lock);
4196         return 0;
4197 }
4198
4199 void tcpm_tcpc_reset(struct tcpm_port *port)
4200 {
4201         mutex_lock(&port->lock);
4202         /* XXX: Maintain PD connection if possible? */
4203         tcpm_init(port);
4204         mutex_unlock(&port->lock);
4205 }
4206 EXPORT_SYMBOL_GPL(tcpm_tcpc_reset);
4207
4208 static int tcpm_copy_pdos(u32 *dest_pdo, const u32 *src_pdo,
4209                           unsigned int nr_pdo)
4210 {
4211         unsigned int i;
4212
4213         if (nr_pdo > PDO_MAX_OBJECTS)
4214                 nr_pdo = PDO_MAX_OBJECTS;
4215
4216         for (i = 0; i < nr_pdo; i++)
4217                 dest_pdo[i] = src_pdo[i];
4218
4219         return nr_pdo;
4220 }
4221
4222 static int tcpm_copy_vdos(u32 *dest_vdo, const u32 *src_vdo,
4223                           unsigned int nr_vdo)
4224 {
4225         unsigned int i;
4226
4227         if (nr_vdo > VDO_MAX_OBJECTS)
4228                 nr_vdo = VDO_MAX_OBJECTS;
4229
4230         for (i = 0; i < nr_vdo; i++)
4231                 dest_vdo[i] = src_vdo[i];
4232
4233         return nr_vdo;
4234 }
4235
4236 static int tcpm_fw_get_caps(struct tcpm_port *port,
4237                             struct fwnode_handle *fwnode)
4238 {
4239         const char *cap_str;
4240         int ret;
4241         u32 mw;
4242
4243         if (!fwnode)
4244                 return -EINVAL;
4245
4246         /* USB data support is optional */
4247         ret = fwnode_property_read_string(fwnode, "data-role", &cap_str);
4248         if (ret == 0) {
4249                 port->typec_caps.data = typec_find_port_data_role(cap_str);
4250                 if (port->typec_caps.data < 0)
4251                         return -EINVAL;
4252         }
4253
4254         ret = fwnode_property_read_string(fwnode, "power-role", &cap_str);
4255         if (ret < 0)
4256                 return ret;
4257
4258         port->typec_caps.type = typec_find_port_power_role(cap_str);
4259         if (port->typec_caps.type < 0)
4260                 return -EINVAL;
4261         port->port_type = port->typec_caps.type;
4262
4263         if (port->port_type == TYPEC_PORT_SNK)
4264                 goto sink;
4265
4266         /* Get source pdos */
4267         ret = fwnode_property_read_u32_array(fwnode, "source-pdos",
4268                                              NULL, 0);
4269         if (ret <= 0)
4270                 return -EINVAL;
4271
4272         port->nr_src_pdo = min(ret, PDO_MAX_OBJECTS);
4273         ret = fwnode_property_read_u32_array(fwnode, "source-pdos",
4274                                              port->src_pdo, port->nr_src_pdo);
4275         if ((ret < 0) || tcpm_validate_caps(port, port->src_pdo,
4276                                             port->nr_src_pdo))
4277                 return -EINVAL;
4278
4279         if (port->port_type == TYPEC_PORT_SRC)
4280                 return 0;
4281
4282         /* Get the preferred power role for DRP */
4283         ret = fwnode_property_read_string(fwnode, "try-power-role", &cap_str);
4284         if (ret < 0)
4285                 return ret;
4286
4287         port->typec_caps.prefer_role = typec_find_power_role(cap_str);
4288         if (port->typec_caps.prefer_role < 0)
4289                 return -EINVAL;
4290 sink:
4291         /* Get sink pdos */
4292         ret = fwnode_property_read_u32_array(fwnode, "sink-pdos",
4293                                              NULL, 0);
4294         if (ret <= 0)
4295                 return -EINVAL;
4296
4297         port->nr_snk_pdo = min(ret, PDO_MAX_OBJECTS);
4298         ret = fwnode_property_read_u32_array(fwnode, "sink-pdos",
4299                                              port->snk_pdo, port->nr_snk_pdo);
4300         if ((ret < 0) || tcpm_validate_caps(port, port->snk_pdo,
4301                                             port->nr_snk_pdo))
4302                 return -EINVAL;
4303
4304         if (fwnode_property_read_u32(fwnode, "op-sink-microwatt", &mw) < 0)
4305                 return -EINVAL;
4306         port->operating_snk_mw = mw / 1000;
4307
4308         return 0;
4309 }
4310
4311 int tcpm_update_source_capabilities(struct tcpm_port *port, const u32 *pdo,
4312                                     unsigned int nr_pdo)
4313 {
4314         if (tcpm_validate_caps(port, pdo, nr_pdo))
4315                 return -EINVAL;
4316
4317         mutex_lock(&port->lock);
4318         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, pdo, nr_pdo);
4319         switch (port->state) {
4320         case SRC_UNATTACHED:
4321         case SRC_ATTACH_WAIT:
4322         case SRC_TRYWAIT:
4323                 tcpm_set_cc(port, tcpm_rp_cc(port));
4324                 break;
4325         case SRC_SEND_CAPABILITIES:
4326         case SRC_NEGOTIATE_CAPABILITIES:
4327         case SRC_READY:
4328         case SRC_WAIT_NEW_CAPABILITIES:
4329                 tcpm_set_cc(port, tcpm_rp_cc(port));
4330                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
4331                 break;
4332         default:
4333                 break;
4334         }
4335         mutex_unlock(&port->lock);
4336         return 0;
4337 }
4338 EXPORT_SYMBOL_GPL(tcpm_update_source_capabilities);
4339
4340 int tcpm_update_sink_capabilities(struct tcpm_port *port, const u32 *pdo,
4341                                   unsigned int nr_pdo,
4342                                   unsigned int operating_snk_mw)
4343 {
4344         if (tcpm_validate_caps(port, pdo, nr_pdo))
4345                 return -EINVAL;
4346
4347         mutex_lock(&port->lock);
4348         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, pdo, nr_pdo);
4349         port->operating_snk_mw = operating_snk_mw;
4350         port->update_sink_caps = true;
4351
4352         switch (port->state) {
4353         case SNK_NEGOTIATE_CAPABILITIES:
4354         case SNK_NEGOTIATE_PPS_CAPABILITIES:
4355         case SNK_READY:
4356         case SNK_TRANSITION_SINK:
4357         case SNK_TRANSITION_SINK_VBUS:
4358                 if (port->pps_data.active)
4359                         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4360                 else
4361                         tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4362                 break;
4363         default:
4364                 break;
4365         }
4366         mutex_unlock(&port->lock);
4367         return 0;
4368 }
4369 EXPORT_SYMBOL_GPL(tcpm_update_sink_capabilities);
4370
4371 /* Power Supply access to expose source power information */
4372 enum tcpm_psy_online_states {
4373         TCPM_PSY_OFFLINE = 0,
4374         TCPM_PSY_FIXED_ONLINE,
4375         TCPM_PSY_PROG_ONLINE,
4376 };
4377
4378 static enum power_supply_property tcpm_psy_props[] = {
4379         POWER_SUPPLY_PROP_USB_TYPE,
4380         POWER_SUPPLY_PROP_ONLINE,
4381         POWER_SUPPLY_PROP_VOLTAGE_MIN,
4382         POWER_SUPPLY_PROP_VOLTAGE_MAX,
4383         POWER_SUPPLY_PROP_VOLTAGE_NOW,
4384         POWER_SUPPLY_PROP_CURRENT_MAX,
4385         POWER_SUPPLY_PROP_CURRENT_NOW,
4386 };
4387
4388 static int tcpm_psy_get_online(struct tcpm_port *port,
4389                                union power_supply_propval *val)
4390 {
4391         if (port->vbus_charge) {
4392                 if (port->pps_data.active)
4393                         val->intval = TCPM_PSY_PROG_ONLINE;
4394                 else
4395                         val->intval = TCPM_PSY_FIXED_ONLINE;
4396         } else {
4397                 val->intval = TCPM_PSY_OFFLINE;
4398         }
4399
4400         return 0;
4401 }
4402
4403 static int tcpm_psy_get_voltage_min(struct tcpm_port *port,
4404                                     union power_supply_propval *val)
4405 {
4406         if (port->pps_data.active)
4407                 val->intval = port->pps_data.min_volt * 1000;
4408         else
4409                 val->intval = port->supply_voltage * 1000;
4410
4411         return 0;
4412 }
4413
4414 static int tcpm_psy_get_voltage_max(struct tcpm_port *port,
4415                                     union power_supply_propval *val)
4416 {
4417         if (port->pps_data.active)
4418                 val->intval = port->pps_data.max_volt * 1000;
4419         else
4420                 val->intval = port->supply_voltage * 1000;
4421
4422         return 0;
4423 }
4424
4425 static int tcpm_psy_get_voltage_now(struct tcpm_port *port,
4426                                     union power_supply_propval *val)
4427 {
4428         val->intval = port->supply_voltage * 1000;
4429
4430         return 0;
4431 }
4432
4433 static int tcpm_psy_get_current_max(struct tcpm_port *port,
4434                                     union power_supply_propval *val)
4435 {
4436         if (port->pps_data.active)
4437                 val->intval = port->pps_data.max_curr * 1000;
4438         else
4439                 val->intval = port->current_limit * 1000;
4440
4441         return 0;
4442 }
4443
4444 static int tcpm_psy_get_current_now(struct tcpm_port *port,
4445                                     union power_supply_propval *val)
4446 {
4447         val->intval = port->current_limit * 1000;
4448
4449         return 0;
4450 }
4451
4452 static int tcpm_psy_get_prop(struct power_supply *psy,
4453                              enum power_supply_property psp,
4454                              union power_supply_propval *val)
4455 {
4456         struct tcpm_port *port = power_supply_get_drvdata(psy);
4457         int ret = 0;
4458
4459         switch (psp) {
4460         case POWER_SUPPLY_PROP_USB_TYPE:
4461                 val->intval = port->usb_type;
4462                 break;
4463         case POWER_SUPPLY_PROP_ONLINE:
4464                 ret = tcpm_psy_get_online(port, val);
4465                 break;
4466         case POWER_SUPPLY_PROP_VOLTAGE_MIN:
4467                 ret = tcpm_psy_get_voltage_min(port, val);
4468                 break;
4469         case POWER_SUPPLY_PROP_VOLTAGE_MAX:
4470                 ret = tcpm_psy_get_voltage_max(port, val);
4471                 break;
4472         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4473                 ret = tcpm_psy_get_voltage_now(port, val);
4474                 break;
4475         case POWER_SUPPLY_PROP_CURRENT_MAX:
4476                 ret = tcpm_psy_get_current_max(port, val);
4477                 break;
4478         case POWER_SUPPLY_PROP_CURRENT_NOW:
4479                 ret = tcpm_psy_get_current_now(port, val);
4480                 break;
4481         default:
4482                 ret = -EINVAL;
4483                 break;
4484         }
4485
4486         return ret;
4487 }
4488
4489 static int tcpm_psy_set_online(struct tcpm_port *port,
4490                                const union power_supply_propval *val)
4491 {
4492         int ret;
4493
4494         switch (val->intval) {
4495         case TCPM_PSY_FIXED_ONLINE:
4496                 ret = tcpm_pps_activate(port, false);
4497                 break;
4498         case TCPM_PSY_PROG_ONLINE:
4499                 ret = tcpm_pps_activate(port, true);
4500                 break;
4501         default:
4502                 ret = -EINVAL;
4503                 break;
4504         }
4505
4506         return ret;
4507 }
4508
4509 static int tcpm_psy_set_prop(struct power_supply *psy,
4510                              enum power_supply_property psp,
4511                              const union power_supply_propval *val)
4512 {
4513         struct tcpm_port *port = power_supply_get_drvdata(psy);
4514         int ret;
4515
4516         switch (psp) {
4517         case POWER_SUPPLY_PROP_ONLINE:
4518                 ret = tcpm_psy_set_online(port, val);
4519                 break;
4520         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4521                 if (val->intval < port->pps_data.min_volt * 1000 ||
4522                     val->intval > port->pps_data.max_volt * 1000)
4523                         ret = -EINVAL;
4524                 else
4525                         ret = tcpm_pps_set_out_volt(port, val->intval / 1000);
4526                 break;
4527         case POWER_SUPPLY_PROP_CURRENT_NOW:
4528                 if (val->intval > port->pps_data.max_curr * 1000)
4529                         ret = -EINVAL;
4530                 else
4531                         ret = tcpm_pps_set_op_curr(port, val->intval / 1000);
4532                 break;
4533         default:
4534                 ret = -EINVAL;
4535                 break;
4536         }
4537
4538         return ret;
4539 }
4540
4541 static int tcpm_psy_prop_writeable(struct power_supply *psy,
4542                                    enum power_supply_property psp)
4543 {
4544         switch (psp) {
4545         case POWER_SUPPLY_PROP_ONLINE:
4546         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4547         case POWER_SUPPLY_PROP_CURRENT_NOW:
4548                 return 1;
4549         default:
4550                 return 0;
4551         }
4552 }
4553
4554 static enum power_supply_usb_type tcpm_psy_usb_types[] = {
4555         POWER_SUPPLY_USB_TYPE_C,
4556         POWER_SUPPLY_USB_TYPE_PD,
4557         POWER_SUPPLY_USB_TYPE_PD_PPS,
4558 };
4559
4560 static const char *tcpm_psy_name_prefix = "tcpm-source-psy-";
4561
4562 static int devm_tcpm_psy_register(struct tcpm_port *port)
4563 {
4564         struct power_supply_config psy_cfg = {};
4565         const char *port_dev_name = dev_name(port->dev);
4566         size_t psy_name_len = strlen(tcpm_psy_name_prefix) +
4567                                      strlen(port_dev_name) + 1;
4568         char *psy_name;
4569
4570         psy_cfg.drv_data = port;
4571         psy_cfg.fwnode = dev_fwnode(port->dev);
4572         psy_name = devm_kzalloc(port->dev, psy_name_len, GFP_KERNEL);
4573         if (!psy_name)
4574                 return -ENOMEM;
4575
4576         snprintf(psy_name, psy_name_len, "%s%s", tcpm_psy_name_prefix,
4577                  port_dev_name);
4578         port->psy_desc.name = psy_name;
4579         port->psy_desc.type = POWER_SUPPLY_TYPE_USB,
4580         port->psy_desc.usb_types = tcpm_psy_usb_types;
4581         port->psy_desc.num_usb_types = ARRAY_SIZE(tcpm_psy_usb_types);
4582         port->psy_desc.properties = tcpm_psy_props,
4583         port->psy_desc.num_properties = ARRAY_SIZE(tcpm_psy_props),
4584         port->psy_desc.get_property = tcpm_psy_get_prop,
4585         port->psy_desc.set_property = tcpm_psy_set_prop,
4586         port->psy_desc.property_is_writeable = tcpm_psy_prop_writeable,
4587
4588         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
4589
4590         port->psy = devm_power_supply_register(port->dev, &port->psy_desc,
4591                                                &psy_cfg);
4592
4593         return PTR_ERR_OR_ZERO(port->psy);
4594 }
4595
4596 static int tcpm_copy_caps(struct tcpm_port *port,
4597                           const struct tcpc_config *tcfg)
4598 {
4599         if (tcpm_validate_caps(port, tcfg->src_pdo, tcfg->nr_src_pdo) ||
4600             tcpm_validate_caps(port, tcfg->snk_pdo, tcfg->nr_snk_pdo))
4601                 return -EINVAL;
4602
4603         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, tcfg->src_pdo,
4604                                           tcfg->nr_src_pdo);
4605         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, tcfg->snk_pdo,
4606                                           tcfg->nr_snk_pdo);
4607
4608         port->nr_snk_vdo = tcpm_copy_vdos(port->snk_vdo, tcfg->snk_vdo,
4609                                           tcfg->nr_snk_vdo);
4610
4611         port->operating_snk_mw = tcfg->operating_snk_mw;
4612
4613         port->typec_caps.prefer_role = tcfg->default_role;
4614         port->typec_caps.type = tcfg->type;
4615         port->typec_caps.data = tcfg->data;
4616
4617         return 0;
4618 }
4619
4620 struct tcpm_port *tcpm_register_port(struct device *dev, struct tcpc_dev *tcpc)
4621 {
4622         struct tcpm_port *port;
4623         int i, err;
4624
4625         if (!dev || !tcpc ||
4626             !tcpc->get_vbus || !tcpc->set_cc || !tcpc->get_cc ||
4627             !tcpc->set_polarity || !tcpc->set_vconn || !tcpc->set_vbus ||
4628             !tcpc->set_pd_rx || !tcpc->set_roles || !tcpc->pd_transmit)
4629                 return ERR_PTR(-EINVAL);
4630
4631         port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
4632         if (!port)
4633                 return ERR_PTR(-ENOMEM);
4634
4635         port->dev = dev;
4636         port->tcpc = tcpc;
4637
4638         mutex_init(&port->lock);
4639         mutex_init(&port->swap_lock);
4640
4641         port->wq = create_singlethread_workqueue(dev_name(dev));
4642         if (!port->wq)
4643                 return ERR_PTR(-ENOMEM);
4644         INIT_DELAYED_WORK(&port->state_machine, tcpm_state_machine_work);
4645         INIT_DELAYED_WORK(&port->vdm_state_machine, vdm_state_machine_work);
4646         INIT_WORK(&port->event_work, tcpm_pd_event_handler);
4647
4648         spin_lock_init(&port->pd_event_lock);
4649
4650         init_completion(&port->tx_complete);
4651         init_completion(&port->swap_complete);
4652         init_completion(&port->pps_complete);
4653         tcpm_debugfs_init(port);
4654
4655         err = tcpm_fw_get_caps(port, tcpc->fwnode);
4656         if ((err < 0) && tcpc->config)
4657                 err = tcpm_copy_caps(port, tcpc->config);
4658         if (err < 0)
4659                 goto out_destroy_wq;
4660
4661         if (!tcpc->config || !tcpc->config->try_role_hw)
4662                 port->try_role = port->typec_caps.prefer_role;
4663         else
4664                 port->try_role = TYPEC_NO_PREFERRED_ROLE;
4665
4666         port->typec_caps.fwnode = tcpc->fwnode;
4667         port->typec_caps.revision = 0x0120;     /* Type-C spec release 1.2 */
4668         port->typec_caps.pd_revision = 0x0300;  /* USB-PD spec release 3.0 */
4669         port->typec_caps.dr_set = tcpm_dr_set;
4670         port->typec_caps.pr_set = tcpm_pr_set;
4671         port->typec_caps.vconn_set = tcpm_vconn_set;
4672         port->typec_caps.try_role = tcpm_try_role;
4673         port->typec_caps.port_type_set = tcpm_port_type_set;
4674
4675         port->partner_desc.identity = &port->partner_ident;
4676         port->port_type = port->typec_caps.type;
4677
4678         port->role_sw = usb_role_switch_get(port->dev);
4679         if (IS_ERR(port->role_sw)) {
4680                 err = PTR_ERR(port->role_sw);
4681                 goto out_destroy_wq;
4682         }
4683
4684         err = devm_tcpm_psy_register(port);
4685         if (err)
4686                 goto out_destroy_wq;
4687
4688         port->typec_port = typec_register_port(port->dev, &port->typec_caps);
4689         if (IS_ERR(port->typec_port)) {
4690                 err = PTR_ERR(port->typec_port);
4691                 goto out_destroy_wq;
4692         }
4693
4694         if (tcpc->config && tcpc->config->alt_modes) {
4695                 const struct typec_altmode_desc *paltmode = tcpc->config->alt_modes;
4696
4697                 i = 0;
4698                 while (paltmode->svid && i < ARRAY_SIZE(port->port_altmode)) {
4699                         struct typec_altmode *alt;
4700
4701                         alt = typec_port_register_altmode(port->typec_port,
4702                                                           paltmode);
4703                         if (IS_ERR(alt)) {
4704                                 tcpm_log(port,
4705                                          "%s: failed to register port alternate mode 0x%x",
4706                                          dev_name(dev), paltmode->svid);
4707                                 break;
4708                         }
4709                         port->port_altmode[i] = alt;
4710                         i++;
4711                         paltmode++;
4712                 }
4713         }
4714
4715         mutex_lock(&port->lock);
4716         tcpm_init(port);
4717         mutex_unlock(&port->lock);
4718
4719         tcpm_log(port, "%s: registered", dev_name(dev));
4720         return port;
4721
4722 out_destroy_wq:
4723         usb_role_switch_put(port->role_sw);
4724         destroy_workqueue(port->wq);
4725         return ERR_PTR(err);
4726 }
4727 EXPORT_SYMBOL_GPL(tcpm_register_port);
4728
4729 void tcpm_unregister_port(struct tcpm_port *port)
4730 {
4731         int i;
4732
4733         tcpm_reset_port(port);
4734         for (i = 0; i < ARRAY_SIZE(port->port_altmode); i++)
4735                 typec_unregister_altmode(port->port_altmode[i]);
4736         typec_unregister_port(port->typec_port);
4737         usb_role_switch_put(port->role_sw);
4738         tcpm_debugfs_exit(port);
4739         destroy_workqueue(port->wq);
4740 }
4741 EXPORT_SYMBOL_GPL(tcpm_unregister_port);
4742
4743 MODULE_AUTHOR("Guenter Roeck <groeck@chromium.org>");
4744 MODULE_DESCRIPTION("USB Type-C Port Manager");
4745 MODULE_LICENSE("GPL");