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];
 314         struct typec_altmode *port_altmode[SVID_DISCOVERY_MAX];
 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         struct typec_mode_desc *pmode;
 999         int i;
1000
1001         if (pmdata->altmodes >= ARRAY_SIZE(port->partner_altmode)) {
1002                 /* Already logged in svdm_consume_svids() */
1003                 return;
1004         }
1005
1006         paltmode = &pmdata->altmode_desc[pmdata->altmodes];
1007         memset(paltmode, 0, sizeof(*paltmode));
1008
1009         paltmode->svid = pmdata->svids[pmdata->svid_index];
1010
1011         tcpm_log(port, " Alternate mode %d: SVID 0x%04x",
1012                  pmdata->altmodes, paltmode->svid);
1013
1014         for (i = 1; i < cnt && paltmode->n_modes < ALTMODE_MAX_MODES; i++) {
1015                 pmode = &paltmode->modes[paltmode->n_modes];
1016                 memset(pmode, 0, sizeof(*pmode));
1017                 pmode->vdo = le32_to_cpu(payload[i]);
1018                 pmode->index = i - 1;
1019                 paltmode->n_modes++;
1020                 tcpm_log(port, "  VDO %d: 0x%08x",
1021                          pmode->index, pmode->vdo);
1022         }
1023         port->partner_altmode[pmdata->altmodes] =
1024                 typec_partner_register_altmode(port->partner, paltmode);
1025         if (!port->partner_altmode[pmdata->altmodes]) {
1026                 tcpm_log(port,
1027                          "Failed to register alternate modes for SVID 0x%04x",
1028                          paltmode->svid);
1029                 return;
1030         }
1031         pmdata->altmodes++;
1032 }
1033
1034 #define supports_modal(port)    PD_IDH_MODAL_SUPP((port)->partner_ident.id_header)
1035
1036 static int tcpm_pd_svdm(struct tcpm_port *port, const __le32 *payload, int cnt,
1037                         u32 *response)
1038 {
1039         u32 p0 = le32_to_cpu(payload[0]);
1040         int cmd_type = PD_VDO_CMDT(p0);
1041         int cmd = PD_VDO_CMD(p0);
1042         struct pd_mode_data *modep;
1043         int rlen = 0;
1044         u16 svid;
1045         int i;
1046
1047         tcpm_log(port, "Rx VDM cmd 0x%x type %d cmd %d len %d",
1048                  p0, cmd_type, cmd, cnt);
1049
1050         modep = &port->mode_data;
1051
1052         switch (cmd_type) {
1053         case CMDT_INIT:
1054                 switch (cmd) {
1055                 case CMD_DISCOVER_IDENT:
1056                         /* 6.4.4.3.1: Only respond as UFP (device) */
1057                         if (port->data_role == TYPEC_DEVICE &&
1058                             port->nr_snk_vdo) {
1059                                 for (i = 0; i <  port->nr_snk_vdo; i++)
1060                                         response[i + 1] = port->snk_vdo[i];
1061                                 rlen = port->nr_snk_vdo + 1;
1062                         }
1063                         break;
1064                 case CMD_DISCOVER_SVID:
1065                         break;
1066                 case CMD_DISCOVER_MODES:
1067                         break;
1068                 case CMD_ENTER_MODE:
1069                         break;
1070                 case CMD_EXIT_MODE:
1071                         break;
1072                 case CMD_ATTENTION:
1073                         break;
1074                 default:
1075                         break;
1076                 }
1077                 if (rlen >= 1) {
1078                         response[0] = p0 | VDO_CMDT(CMDT_RSP_ACK);
1079                 } else if (rlen == 0) {
1080                         response[0] = p0 | VDO_CMDT(CMDT_RSP_NAK);
1081                         rlen = 1;
1082                 } else {
1083                         response[0] = p0 | VDO_CMDT(CMDT_RSP_BUSY);
1084                         rlen = 1;
1085                 }
1086                 break;
1087         case CMDT_RSP_ACK:
1088                 /* silently drop message if we are not connected */
1089                 if (IS_ERR_OR_NULL(port->partner))
1090                         break;
1091
1092                 switch (cmd) {
1093                 case CMD_DISCOVER_IDENT:
1094                         /* 6.4.4.3.1 */
1095                         svdm_consume_identity(port, payload, cnt);
1096                         response[0] = VDO(USB_SID_PD, 1, CMD_DISCOVER_SVID);
1097                         rlen = 1;
1098                         break;
1099                 case CMD_DISCOVER_SVID:
1100                         /* 6.4.4.3.2 */
1101                         if (svdm_consume_svids(port, payload, cnt)) {
1102                                 response[0] = VDO(USB_SID_PD, 1,
1103                                                   CMD_DISCOVER_SVID);
1104                                 rlen = 1;
1105                         } else if (modep->nsvids && supports_modal(port)) {
1106                                 response[0] = VDO(modep->svids[0], 1,
1107                                                   CMD_DISCOVER_MODES);
1108                                 rlen = 1;
1109                         }
1110                         break;
1111                 case CMD_DISCOVER_MODES:
1112                         /* 6.4.4.3.3 */
1113                         svdm_consume_modes(port, payload, cnt);
1114                         modep->svid_index++;
1115                         if (modep->svid_index < modep->nsvids) {
1116                                 svid = modep->svids[modep->svid_index];
1117                                 response[0] = VDO(svid, 1, CMD_DISCOVER_MODES);
1118                                 rlen = 1;
1119                         } else {
1120                                 /* enter alternate mode if/when implemented */
1121                         }
1122                         break;
1123                 case CMD_ENTER_MODE:
1124                         break;
1125                 default:
1126                         break;
1127                 }
1128                 break;
1129         default:
1130                 break;
1131         }
1132
1133         return rlen;
1134 }
1135
1136 static void tcpm_handle_vdm_request(struct tcpm_port *port,
1137                                     const __le32 *payload, int cnt)
1138 {
1139         int rlen = 0;
1140         u32 response[8] = { };
1141         u32 p0 = le32_to_cpu(payload[0]);
1142
1143         if (port->vdm_state == VDM_STATE_BUSY) {
1144                 /* If UFP responded busy retry after timeout */
1145                 if (PD_VDO_CMDT(p0) == CMDT_RSP_BUSY) {
1146                         port->vdm_state = VDM_STATE_WAIT_RSP_BUSY;
1147                         port->vdo_retry = (p0 & ~VDO_CMDT_MASK) |
1148                                 CMDT_INIT;
1149                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1150                                          msecs_to_jiffies(PD_T_VDM_BUSY));
1151                         return;
1152                 }
1153                 port->vdm_state = VDM_STATE_DONE;
1154         }
1155
1156         if (PD_VDO_SVDM(p0))
1157                 rlen = tcpm_pd_svdm(port, payload, cnt, response);
1158
1159         if (rlen > 0) {
1160                 tcpm_queue_vdm(port, response[0], &response[1], rlen - 1);
1161                 mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1162         }
1163 }
1164
1165 static void tcpm_send_vdm(struct tcpm_port *port, u32 vid, int cmd,
1166                           const u32 *data, int count)
1167 {
1168         u32 header;
1169
1170         if (WARN_ON(count > VDO_MAX_SIZE - 1))
1171                 count = VDO_MAX_SIZE - 1;
1172
1173         /* set VDM header with VID & CMD */
1174         header = VDO(vid, ((vid & USB_SID_PD) == USB_SID_PD) ?
1175                         1 : (PD_VDO_CMD(cmd) <= CMD_ATTENTION), cmd);
1176         tcpm_queue_vdm(port, header, data, count);
1177
1178         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1179 }
1180
1181 static unsigned int vdm_ready_timeout(u32 vdm_hdr)
1182 {
1183         unsigned int timeout;
1184         int cmd = PD_VDO_CMD(vdm_hdr);
1185
1186         /* its not a structured VDM command */
1187         if (!PD_VDO_SVDM(vdm_hdr))
1188                 return PD_T_VDM_UNSTRUCTURED;
1189
1190         switch (PD_VDO_CMDT(vdm_hdr)) {
1191         case CMDT_INIT:
1192                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1193                         timeout = PD_T_VDM_WAIT_MODE_E;
1194                 else
1195                         timeout = PD_T_VDM_SNDR_RSP;
1196                 break;
1197         default:
1198                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1199                         timeout = PD_T_VDM_E_MODE;
1200                 else
1201                         timeout = PD_T_VDM_RCVR_RSP;
1202                 break;
1203         }
1204         return timeout;
1205 }
1206
1207 static void vdm_run_state_machine(struct tcpm_port *port)
1208 {
1209         struct pd_message msg;
1210         int i, res;
1211
1212         switch (port->vdm_state) {
1213         case VDM_STATE_READY:
1214                 /* Only transmit VDM if attached */
1215                 if (!port->attached) {
1216                         port->vdm_state = VDM_STATE_ERR_BUSY;
1217                         break;
1218                 }
1219
1220                 /*
1221                  * if there's traffic or we're not in PDO ready state don't send
1222                  * a VDM.
1223                  */
1224                 if (port->state != SRC_READY && port->state != SNK_READY)
1225                         break;
1226
1227                 /* Prepare and send VDM */
1228                 memset(&msg, 0, sizeof(msg));
1229                 msg.header = PD_HEADER_LE(PD_DATA_VENDOR_DEF,
1230                                           port->pwr_role,
1231                                           port->data_role,
1232                                           port->negotiated_rev,
1233                                           port->message_id, port->vdo_count);
1234                 for (i = 0; i < port->vdo_count; i++)
1235                         msg.payload[i] = cpu_to_le32(port->vdo_data[i]);
1236                 res = tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1237                 if (res < 0) {
1238                         port->vdm_state = VDM_STATE_ERR_SEND;
1239                 } else {
1240                         unsigned long timeout;
1241
1242                         port->vdm_retries = 0;
1243                         port->vdm_state = VDM_STATE_BUSY;
1244                         timeout = vdm_ready_timeout(port->vdo_data[0]);
1245                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1246                                          timeout);
1247                 }
1248                 break;
1249         case VDM_STATE_WAIT_RSP_BUSY:
1250                 port->vdo_data[0] = port->vdo_retry;
1251                 port->vdo_count = 1;
1252                 port->vdm_state = VDM_STATE_READY;
1253                 break;
1254         case VDM_STATE_BUSY:
1255                 port->vdm_state = VDM_STATE_ERR_TMOUT;
1256                 break;
1257         case VDM_STATE_ERR_SEND:
1258                 /*
1259                  * A partner which does not support USB PD will not reply,
1260                  * so this is not a fatal error. At the same time, some
1261                  * devices may not return GoodCRC under some circumstances,
1262                  * so we need to retry.
1263                  */
1264                 if (port->vdm_retries < 3) {
1265                         tcpm_log(port, "VDM Tx error, retry");
1266                         port->vdm_retries++;
1267                         port->vdm_state = VDM_STATE_READY;
1268                 }
1269                 break;
1270         default:
1271                 break;
1272         }
1273 }
1274
1275 static void vdm_state_machine_work(struct work_struct *work)
1276 {
1277         struct tcpm_port *port = container_of(work, struct tcpm_port,
1278                                               vdm_state_machine.work);
1279         enum vdm_states prev_state;
1280
1281         mutex_lock(&port->lock);
1282
1283         /*
1284          * Continue running as long as the port is not busy and there was
1285          * a state change.
1286          */
1287         do {
1288                 prev_state = port->vdm_state;
1289                 vdm_run_state_machine(port);
1290         } while (port->vdm_state != prev_state &&
1291                  port->vdm_state != VDM_STATE_BUSY);
1292
1293         mutex_unlock(&port->lock);
1294 }
1295
1296 enum pdo_err {
1297         PDO_NO_ERR,
1298         PDO_ERR_NO_VSAFE5V,
1299         PDO_ERR_VSAFE5V_NOT_FIRST,
1300         PDO_ERR_PDO_TYPE_NOT_IN_ORDER,
1301         PDO_ERR_FIXED_NOT_SORTED,
1302         PDO_ERR_VARIABLE_BATT_NOT_SORTED,
1303         PDO_ERR_DUPE_PDO,
1304         PDO_ERR_PPS_APDO_NOT_SORTED,
1305         PDO_ERR_DUPE_PPS_APDO,
1306 };
1307
1308 static const char * const pdo_err_msg[] = {
1309         [PDO_ERR_NO_VSAFE5V] =
1310         " err: source/sink caps should atleast have vSafe5V",
1311         [PDO_ERR_VSAFE5V_NOT_FIRST] =
1312         " err: vSafe5V Fixed Supply Object Shall always be the first object",
1313         [PDO_ERR_PDO_TYPE_NOT_IN_ORDER] =
1314         " err: PDOs should be in the following order: Fixed; Battery; Variable",
1315         [PDO_ERR_FIXED_NOT_SORTED] =
1316         " err: Fixed supply pdos should be in increasing order of their fixed voltage",
1317         [PDO_ERR_VARIABLE_BATT_NOT_SORTED] =
1318         " err: Variable/Battery supply pdos should be in increasing order of their minimum voltage",
1319         [PDO_ERR_DUPE_PDO] =
1320         " err: Variable/Batt supply pdos cannot have same min/max voltage",
1321         [PDO_ERR_PPS_APDO_NOT_SORTED] =
1322         " err: Programmable power supply apdos should be in increasing order of their maximum voltage",
1323         [PDO_ERR_DUPE_PPS_APDO] =
1324         " err: Programmable power supply apdos cannot have same min/max voltage and max current",
1325 };
1326
1327 static enum pdo_err tcpm_caps_err(struct tcpm_port *port, const u32 *pdo,
1328                                   unsigned int nr_pdo)
1329 {
1330         unsigned int i;
1331
1332         /* Should at least contain vSafe5v */
1333         if (nr_pdo < 1)
1334                 return PDO_ERR_NO_VSAFE5V;
1335
1336         /* The vSafe5V Fixed Supply Object Shall always be the first object */
1337         if (pdo_type(pdo[0]) != PDO_TYPE_FIXED ||
1338             pdo_fixed_voltage(pdo[0]) != VSAFE5V)
1339                 return PDO_ERR_VSAFE5V_NOT_FIRST;
1340
1341         for (i = 1; i < nr_pdo; i++) {
1342                 if (pdo_type(pdo[i]) < pdo_type(pdo[i - 1])) {
1343                         return PDO_ERR_PDO_TYPE_NOT_IN_ORDER;
1344                 } else if (pdo_type(pdo[i]) == pdo_type(pdo[i - 1])) {
1345                         enum pd_pdo_type type = pdo_type(pdo[i]);
1346
1347                         switch (type) {
1348                         /*
1349                          * The remaining Fixed Supply Objects, if
1350                          * present, shall be sent in voltage order;
1351                          * lowest to highest.
1352                          */
1353                         case PDO_TYPE_FIXED:
1354                                 if (pdo_fixed_voltage(pdo[i]) <=
1355                                     pdo_fixed_voltage(pdo[i - 1]))
1356                                         return PDO_ERR_FIXED_NOT_SORTED;
1357                                 break;
1358                         /*
1359                          * The Battery Supply Objects and Variable
1360                          * supply, if present shall be sent in Minimum
1361                          * Voltage order; lowest to highest.
1362                          */
1363                         case PDO_TYPE_VAR:
1364                         case PDO_TYPE_BATT:
1365                                 if (pdo_min_voltage(pdo[i]) <
1366                                     pdo_min_voltage(pdo[i - 1]))
1367                                         return PDO_ERR_VARIABLE_BATT_NOT_SORTED;
1368                                 else if ((pdo_min_voltage(pdo[i]) ==
1369                                           pdo_min_voltage(pdo[i - 1])) &&
1370                                          (pdo_max_voltage(pdo[i]) ==
1371                                           pdo_min_voltage(pdo[i - 1])))
1372                                         return PDO_ERR_DUPE_PDO;
1373                                 break;
1374                         /*
1375                          * The Programmable Power Supply APDOs, if present,
1376                          * shall be sent in Maximum Voltage order;
1377                          * lowest to highest.
1378                          */
1379                         case PDO_TYPE_APDO:
1380                                 if (pdo_apdo_type(pdo[i]) != APDO_TYPE_PPS)
1381                                         break;
1382
1383                                 if (pdo_pps_apdo_max_current(pdo[i]) <
1384                                     pdo_pps_apdo_max_current(pdo[i - 1]))
1385                                         return PDO_ERR_PPS_APDO_NOT_SORTED;
1386                                 else if (pdo_pps_apdo_min_voltage(pdo[i]) ==
1387                                           pdo_pps_apdo_min_voltage(pdo[i - 1]) &&
1388                                          pdo_pps_apdo_max_voltage(pdo[i]) ==
1389                                           pdo_pps_apdo_max_voltage(pdo[i - 1]) &&
1390                                          pdo_pps_apdo_max_current(pdo[i]) ==
1391                                           pdo_pps_apdo_max_current(pdo[i - 1]))
1392                                         return PDO_ERR_DUPE_PPS_APDO;
1393                                 break;
1394                         default:
1395                                 tcpm_log_force(port, " Unknown pdo type");
1396                         }
1397                 }
1398         }
1399
1400         return PDO_NO_ERR;
1401 }
1402
1403 static int tcpm_validate_caps(struct tcpm_port *port, const u32 *pdo,
1404                               unsigned int nr_pdo)
1405 {
1406         enum pdo_err err_index = tcpm_caps_err(port, pdo, nr_pdo);
1407
1408         if (err_index != PDO_NO_ERR) {
1409                 tcpm_log_force(port, " %s", pdo_err_msg[err_index]);
1410                 return -EINVAL;
1411         }
1412
1413         return 0;
1414 }
1415
1416 /*
1417  * PD (data, control) command handling functions
1418  */
1419 static inline enum tcpm_state ready_state(struct tcpm_port *port)
1420 {
1421         if (port->pwr_role == TYPEC_SOURCE)
1422                 return SRC_READY;
1423         else
1424                 return SNK_READY;
1425 }
1426
1427 static int tcpm_pd_send_control(struct tcpm_port *port,
1428                                 enum pd_ctrl_msg_type type);
1429
1430 static void tcpm_handle_alert(struct tcpm_port *port, const __le32 *payload,
1431                               int cnt)
1432 {
1433         u32 p0 = le32_to_cpu(payload[0]);
1434         unsigned int type = usb_pd_ado_type(p0);
1435
1436         if (!type) {
1437                 tcpm_log(port, "Alert message received with no type");
1438                 return;
1439         }
1440
1441         /* Just handling non-battery alerts for now */
1442         if (!(type & USB_PD_ADO_TYPE_BATT_STATUS_CHANGE)) {
1443                 switch (port->state) {
1444                 case SRC_READY:
1445                 case SNK_READY:
1446                         tcpm_set_state(port, GET_STATUS_SEND, 0);
1447                         break;
1448                 default:
1449                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1450                         break;
1451                 }
1452         }
1453 }
1454
1455 static void tcpm_pd_data_request(struct tcpm_port *port,
1456                                  const struct pd_message *msg)
1457 {
1458         enum pd_data_msg_type type = pd_header_type_le(msg->header);
1459         unsigned int cnt = pd_header_cnt_le(msg->header);
1460         unsigned int rev = pd_header_rev_le(msg->header);
1461         unsigned int i;
1462
1463         switch (type) {
1464         case PD_DATA_SOURCE_CAP:
1465                 if (port->pwr_role != TYPEC_SINK)
1466                         break;
1467
1468                 for (i = 0; i < cnt; i++)
1469                         port->source_caps[i] = le32_to_cpu(msg->payload[i]);
1470
1471                 port->nr_source_caps = cnt;
1472
1473                 tcpm_log_source_caps(port);
1474
1475                 tcpm_validate_caps(port, port->source_caps,
1476                                    port->nr_source_caps);
1477
1478                 /*
1479                  * Adjust revision in subsequent message headers, as required,
1480                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1481                  * support Rev 1.0 so just do nothing in that scenario.
1482                  */
1483                 if (rev == PD_REV10)
1484                         break;
1485
1486                 if (rev < PD_MAX_REV)
1487                         port->negotiated_rev = rev;
1488
1489                 /*
1490                  * This message may be received even if VBUS is not
1491                  * present. This is quite unexpected; see USB PD
1492                  * specification, sections 8.3.3.6.3.1 and 8.3.3.6.3.2.
1493                  * However, at the same time, we must be ready to
1494                  * receive this message and respond to it 15ms after
1495                  * receiving PS_RDY during power swap operations, no matter
1496                  * if VBUS is available or not (USB PD specification,
1497                  * section 6.5.9.2).
1498                  * So we need to accept the message either way,
1499                  * but be prepared to keep waiting for VBUS after it was
1500                  * handled.
1501                  */
1502                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
1503                 break;
1504         case PD_DATA_REQUEST:
1505                 if (port->pwr_role != TYPEC_SOURCE ||
1506                     cnt != 1) {
1507                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1508                         break;
1509                 }
1510
1511                 /*
1512                  * Adjust revision in subsequent message headers, as required,
1513                  * to comply with 6.2.1.1.5 of the USB PD 3.0 spec. We don't
1514                  * support Rev 1.0 so just reject in that scenario.
1515                  */
1516                 if (rev == PD_REV10) {
1517                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1518                         break;
1519                 }
1520
1521                 if (rev < PD_MAX_REV)
1522                         port->negotiated_rev = rev;
1523
1524                 port->sink_request = le32_to_cpu(msg->payload[0]);
1525                 tcpm_set_state(port, SRC_NEGOTIATE_CAPABILITIES, 0);
1526                 break;
1527         case PD_DATA_SINK_CAP:
1528                 /* We don't do anything with this at the moment... */
1529                 for (i = 0; i < cnt; i++)
1530                         port->sink_caps[i] = le32_to_cpu(msg->payload[i]);
1531                 port->nr_sink_caps = cnt;
1532                 break;
1533         case PD_DATA_VENDOR_DEF:
1534                 tcpm_handle_vdm_request(port, msg->payload, cnt);
1535                 break;
1536         case PD_DATA_BIST:
1537                 if (port->state == SRC_READY || port->state == SNK_READY) {
1538                         port->bist_request = le32_to_cpu(msg->payload[0]);
1539                         tcpm_set_state(port, BIST_RX, 0);
1540                 }
1541                 break;
1542         case PD_DATA_ALERT:
1543                 tcpm_handle_alert(port, msg->payload, cnt);
1544                 break;
1545         case PD_DATA_BATT_STATUS:
1546         case PD_DATA_GET_COUNTRY_INFO:
1547                 /* Currently unsupported */
1548                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1549                 break;
1550         default:
1551                 tcpm_log(port, "Unhandled data message type %#x", type);
1552                 break;
1553         }
1554 }
1555
1556 static void tcpm_pps_complete(struct tcpm_port *port, int result)
1557 {
1558         if (port->pps_pending) {
1559                 port->pps_status = result;
1560                 port->pps_pending = false;
1561                 complete(&port->pps_complete);
1562         }
1563 }
1564
1565 static void tcpm_pd_ctrl_request(struct tcpm_port *port,
1566                                  const struct pd_message *msg)
1567 {
1568         enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1569         enum tcpm_state next_state;
1570
1571         switch (type) {
1572         case PD_CTRL_GOOD_CRC:
1573         case PD_CTRL_PING:
1574                 break;
1575         case PD_CTRL_GET_SOURCE_CAP:
1576                 switch (port->state) {
1577                 case SRC_READY:
1578                 case SNK_READY:
1579                         tcpm_queue_message(port, PD_MSG_DATA_SOURCE_CAP);
1580                         break;
1581                 default:
1582                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1583                         break;
1584                 }
1585                 break;
1586         case PD_CTRL_GET_SINK_CAP:
1587                 switch (port->state) {
1588                 case SRC_READY:
1589                 case SNK_READY:
1590                         tcpm_queue_message(port, PD_MSG_DATA_SINK_CAP);
1591                         break;
1592                 default:
1593                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1594                         break;
1595                 }
1596                 break;
1597         case PD_CTRL_GOTO_MIN:
1598                 break;
1599         case PD_CTRL_PS_RDY:
1600                 switch (port->state) {
1601                 case SNK_TRANSITION_SINK:
1602                         if (port->vbus_present) {
1603                                 tcpm_set_current_limit(port,
1604                                                        port->current_limit,
1605                                                        port->supply_voltage);
1606                                 port->explicit_contract = true;
1607                                 tcpm_set_state(port, SNK_READY, 0);
1608                         } else {
1609                                 /*
1610                                  * Seen after power swap. Keep waiting for VBUS
1611                                  * in a transitional state.
1612                                  */
1613                                 tcpm_set_state(port,
1614                                                SNK_TRANSITION_SINK_VBUS, 0);
1615                         }
1616                         break;
1617                 case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
1618                         tcpm_set_state(port, PR_SWAP_SRC_SNK_SINK_ON, 0);
1619                         break;
1620                 case PR_SWAP_SNK_SRC_SINK_OFF:
1621                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON, 0);
1622                         break;
1623                 case VCONN_SWAP_WAIT_FOR_VCONN:
1624                         tcpm_set_state(port, VCONN_SWAP_TURN_OFF_VCONN, 0);
1625                         break;
1626                 default:
1627                         break;
1628                 }
1629                 break;
1630         case PD_CTRL_REJECT:
1631         case PD_CTRL_WAIT:
1632         case PD_CTRL_NOT_SUPP:
1633                 switch (port->state) {
1634                 case SNK_NEGOTIATE_CAPABILITIES:
1635                         /* USB PD specification, Figure 8-43 */
1636                         if (port->explicit_contract)
1637                                 next_state = SNK_READY;
1638                         else
1639                                 next_state = SNK_WAIT_CAPABILITIES;
1640                         tcpm_set_state(port, next_state, 0);
1641                         break;
1642                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1643                         /* Revert data back from any requested PPS updates */
1644                         port->pps_data.out_volt = port->supply_voltage;
1645                         port->pps_data.op_curr = port->current_limit;
1646                         port->pps_status = (type == PD_CTRL_WAIT ?
1647                                             -EAGAIN : -EOPNOTSUPP);
1648                         tcpm_set_state(port, SNK_READY, 0);
1649                         break;
1650                 case DR_SWAP_SEND:
1651                         port->swap_status = (type == PD_CTRL_WAIT ?
1652                                              -EAGAIN : -EOPNOTSUPP);
1653                         tcpm_set_state(port, DR_SWAP_CANCEL, 0);
1654                         break;
1655                 case PR_SWAP_SEND:
1656                         port->swap_status = (type == PD_CTRL_WAIT ?
1657                                              -EAGAIN : -EOPNOTSUPP);
1658                         tcpm_set_state(port, PR_SWAP_CANCEL, 0);
1659                         break;
1660                 case VCONN_SWAP_SEND:
1661                         port->swap_status = (type == PD_CTRL_WAIT ?
1662                                              -EAGAIN : -EOPNOTSUPP);
1663                         tcpm_set_state(port, VCONN_SWAP_CANCEL, 0);
1664                         break;
1665                 default:
1666                         break;
1667                 }
1668                 break;
1669         case PD_CTRL_ACCEPT:
1670                 switch (port->state) {
1671                 case SNK_NEGOTIATE_CAPABILITIES:
1672                         port->pps_data.active = false;
1673                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1674                         break;
1675                 case SNK_NEGOTIATE_PPS_CAPABILITIES:
1676                         port->pps_data.active = true;
1677                         port->supply_voltage = port->pps_data.out_volt;
1678                         port->current_limit = port->pps_data.op_curr;
1679                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1680                         break;
1681                 case SOFT_RESET_SEND:
1682                         port->message_id = 0;
1683                         port->rx_msgid = -1;
1684                         if (port->pwr_role == TYPEC_SOURCE)
1685                                 next_state = SRC_SEND_CAPABILITIES;
1686                         else
1687                                 next_state = SNK_WAIT_CAPABILITIES;
1688                         tcpm_set_state(port, next_state, 0);
1689                         break;
1690                 case DR_SWAP_SEND:
1691                         tcpm_set_state(port, DR_SWAP_CHANGE_DR, 0);
1692                         break;
1693                 case PR_SWAP_SEND:
1694                         tcpm_set_state(port, PR_SWAP_START, 0);
1695                         break;
1696                 case VCONN_SWAP_SEND:
1697                         tcpm_set_state(port, VCONN_SWAP_START, 0);
1698                         break;
1699                 default:
1700                         break;
1701                 }
1702                 break;
1703         case PD_CTRL_SOFT_RESET:
1704                 tcpm_set_state(port, SOFT_RESET, 0);
1705                 break;
1706         case PD_CTRL_DR_SWAP:
1707                 if (port->port_type != TYPEC_PORT_DRP) {
1708                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1709                         break;
1710                 }
1711                 /*
1712                  * XXX
1713                  * 6.3.9: If an alternate mode is active, a request to swap
1714                  * alternate modes shall trigger a port reset.
1715                  */
1716                 switch (port->state) {
1717                 case SRC_READY:
1718                 case SNK_READY:
1719                         tcpm_set_state(port, DR_SWAP_ACCEPT, 0);
1720                         break;
1721                 default:
1722                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1723                         break;
1724                 }
1725                 break;
1726         case PD_CTRL_PR_SWAP:
1727                 if (port->port_type != TYPEC_PORT_DRP) {
1728                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1729                         break;
1730                 }
1731                 switch (port->state) {
1732                 case SRC_READY:
1733                 case SNK_READY:
1734                         tcpm_set_state(port, PR_SWAP_ACCEPT, 0);
1735                         break;
1736                 default:
1737                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1738                         break;
1739                 }
1740                 break;
1741         case PD_CTRL_VCONN_SWAP:
1742                 switch (port->state) {
1743                 case SRC_READY:
1744                 case SNK_READY:
1745                         tcpm_set_state(port, VCONN_SWAP_ACCEPT, 0);
1746                         break;
1747                 default:
1748                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1749                         break;
1750                 }
1751                 break;
1752         case PD_CTRL_GET_SOURCE_CAP_EXT:
1753         case PD_CTRL_GET_STATUS:
1754         case PD_CTRL_FR_SWAP:
1755         case PD_CTRL_GET_PPS_STATUS:
1756         case PD_CTRL_GET_COUNTRY_CODES:
1757                 /* Currently not supported */
1758                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1759                 break;
1760         default:
1761                 tcpm_log(port, "Unhandled ctrl message type %#x", type);
1762                 break;
1763         }
1764 }
1765
1766 static void tcpm_pd_ext_msg_request(struct tcpm_port *port,
1767                                     const struct pd_message *msg)
1768 {
1769         enum pd_ext_msg_type type = pd_header_type_le(msg->header);
1770         unsigned int data_size = pd_ext_header_data_size_le(msg->ext_msg.header);
1771
1772         if (!(msg->ext_msg.header & PD_EXT_HDR_CHUNKED)) {
1773                 tcpm_log(port, "Unchunked extended messages unsupported");
1774                 return;
1775         }
1776
1777         if (data_size > PD_EXT_MAX_CHUNK_DATA) {
1778                 tcpm_log(port, "Chunk handling not yet supported");
1779                 return;
1780         }
1781
1782         switch (type) {
1783         case PD_EXT_STATUS:
1784                 /*
1785                  * If PPS related events raised then get PPS status to clear
1786                  * (see USB PD 3.0 Spec, 6.5.2.4)
1787                  */
1788                 if (msg->ext_msg.data[USB_PD_EXT_SDB_EVENT_FLAGS] &
1789                     USB_PD_EXT_SDB_PPS_EVENTS)
1790                         tcpm_set_state(port, GET_PPS_STATUS_SEND, 0);
1791                 else
1792                         tcpm_set_state(port, ready_state(port), 0);
1793                 break;
1794         case PD_EXT_PPS_STATUS:
1795                 /*
1796                  * For now the PPS status message is used to clear events
1797                  * and nothing more.
1798                  */
1799                 tcpm_set_state(port, ready_state(port), 0);
1800                 break;
1801         case PD_EXT_SOURCE_CAP_EXT:
1802         case PD_EXT_GET_BATT_CAP:
1803         case PD_EXT_GET_BATT_STATUS:
1804         case PD_EXT_BATT_CAP:
1805         case PD_EXT_GET_MANUFACTURER_INFO:
1806         case PD_EXT_MANUFACTURER_INFO:
1807         case PD_EXT_SECURITY_REQUEST:
1808         case PD_EXT_SECURITY_RESPONSE:
1809         case PD_EXT_FW_UPDATE_REQUEST:
1810         case PD_EXT_FW_UPDATE_RESPONSE:
1811         case PD_EXT_COUNTRY_INFO:
1812         case PD_EXT_COUNTRY_CODES:
1813                 tcpm_queue_message(port, PD_MSG_CTRL_NOT_SUPP);
1814                 break;
1815         default:
1816                 tcpm_log(port, "Unhandled extended message type %#x", type);
1817                 break;
1818         }
1819 }
1820
1821 static void tcpm_pd_rx_handler(struct work_struct *work)
1822 {
1823         struct pd_rx_event *event = container_of(work,
1824                                                  struct pd_rx_event, work);
1825         const struct pd_message *msg = &event->msg;
1826         unsigned int cnt = pd_header_cnt_le(msg->header);
1827         struct tcpm_port *port = event->port;
1828
1829         mutex_lock(&port->lock);
1830
1831         tcpm_log(port, "PD RX, header: %#x [%d]", le16_to_cpu(msg->header),
1832                  port->attached);
1833
1834         if (port->attached) {
1835                 enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1836                 unsigned int msgid = pd_header_msgid_le(msg->header);
1837
1838                 /*
1839                  * USB PD standard, 6.6.1.2:
1840                  * "... if MessageID value in a received Message is the
1841                  * same as the stored value, the receiver shall return a
1842                  * GoodCRC Message with that MessageID value and drop
1843                  * the Message (this is a retry of an already received
1844                  * Message). Note: this shall not apply to the Soft_Reset
1845                  * Message which always has a MessageID value of zero."
1846                  */
1847                 if (msgid == port->rx_msgid && type != PD_CTRL_SOFT_RESET)
1848                         goto done;
1849                 port->rx_msgid = msgid;
1850
1851                 /*
1852                  * If both ends believe to be DFP/host, we have a data role
1853                  * mismatch.
1854                  */
1855                 if (!!(le16_to_cpu(msg->header) & PD_HEADER_DATA_ROLE) ==
1856                     (port->data_role == TYPEC_HOST)) {
1857                         tcpm_log(port,
1858                                  "Data role mismatch, initiating error recovery");
1859                         tcpm_set_state(port, ERROR_RECOVERY, 0);
1860                 } else {
1861                         if (msg->header & PD_HEADER_EXT_HDR)
1862                                 tcpm_pd_ext_msg_request(port, msg);
1863                         else if (cnt)
1864                                 tcpm_pd_data_request(port, msg);
1865                         else
1866                                 tcpm_pd_ctrl_request(port, msg);
1867                 }
1868         }
1869
1870 done:
1871         mutex_unlock(&port->lock);
1872         kfree(event);
1873 }
1874
1875 void tcpm_pd_receive(struct tcpm_port *port, const struct pd_message *msg)
1876 {
1877         struct pd_rx_event *event;
1878
1879         event = kzalloc(sizeof(*event), GFP_ATOMIC);
1880         if (!event)
1881                 return;
1882
1883         INIT_WORK(&event->work, tcpm_pd_rx_handler);
1884         event->port = port;
1885         memcpy(&event->msg, msg, sizeof(*msg));
1886         queue_work(port->wq, &event->work);
1887 }
1888 EXPORT_SYMBOL_GPL(tcpm_pd_receive);
1889
1890 static int tcpm_pd_send_control(struct tcpm_port *port,
1891                                 enum pd_ctrl_msg_type type)
1892 {
1893         struct pd_message msg;
1894
1895         memset(&msg, 0, sizeof(msg));
1896         msg.header = PD_HEADER_LE(type, port->pwr_role,
1897                                   port->data_role,
1898                                   port->negotiated_rev,
1899                                   port->message_id, 0);
1900
1901         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1902 }
1903
1904 /*
1905  * Send queued message without affecting state.
1906  * Return true if state machine should go back to sleep,
1907  * false otherwise.
1908  */
1909 static bool tcpm_send_queued_message(struct tcpm_port *port)
1910 {
1911         enum pd_msg_request queued_message;
1912
1913         do {
1914                 queued_message = port->queued_message;
1915                 port->queued_message = PD_MSG_NONE;
1916
1917                 switch (queued_message) {
1918                 case PD_MSG_CTRL_WAIT:
1919                         tcpm_pd_send_control(port, PD_CTRL_WAIT);
1920                         break;
1921                 case PD_MSG_CTRL_REJECT:
1922                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
1923                         break;
1924                 case PD_MSG_CTRL_NOT_SUPP:
1925                         tcpm_pd_send_control(port, PD_CTRL_NOT_SUPP);
1926                         break;
1927                 case PD_MSG_DATA_SINK_CAP:
1928                         tcpm_pd_send_sink_caps(port);
1929                         break;
1930                 case PD_MSG_DATA_SOURCE_CAP:
1931                         tcpm_pd_send_source_caps(port);
1932                         break;
1933                 default:
1934                         break;
1935                 }
1936         } while (port->queued_message != PD_MSG_NONE);
1937
1938         if (port->delayed_state != INVALID_STATE) {
1939                 if (time_is_after_jiffies(port->delayed_runtime)) {
1940                         mod_delayed_work(port->wq, &port->state_machine,
1941                                          port->delayed_runtime - jiffies);
1942                         return true;
1943                 }
1944                 port->delayed_state = INVALID_STATE;
1945         }
1946         return false;
1947 }
1948
1949 static int tcpm_pd_check_request(struct tcpm_port *port)
1950 {
1951         u32 pdo, rdo = port->sink_request;
1952         unsigned int max, op, pdo_max, index;
1953         enum pd_pdo_type type;
1954
1955         index = rdo_index(rdo);
1956         if (!index || index > port->nr_src_pdo)
1957                 return -EINVAL;
1958
1959         pdo = port->src_pdo[index - 1];
1960         type = pdo_type(pdo);
1961         switch (type) {
1962         case PDO_TYPE_FIXED:
1963         case PDO_TYPE_VAR:
1964                 max = rdo_max_current(rdo);
1965                 op = rdo_op_current(rdo);
1966                 pdo_max = pdo_max_current(pdo);
1967
1968                 if (op > pdo_max)
1969                         return -EINVAL;
1970                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1971                         return -EINVAL;
1972
1973                 if (type == PDO_TYPE_FIXED)
1974                         tcpm_log(port,
1975                                  "Requested %u mV, %u mA for %u / %u mA",
1976                                  pdo_fixed_voltage(pdo), pdo_max, op, max);
1977                 else
1978                         tcpm_log(port,
1979                                  "Requested %u -> %u mV, %u mA for %u / %u mA",
1980                                  pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1981                                  pdo_max, op, max);
1982                 break;
1983         case PDO_TYPE_BATT:
1984                 max = rdo_max_power(rdo);
1985                 op = rdo_op_power(rdo);
1986                 pdo_max = pdo_max_power(pdo);
1987
1988                 if (op > pdo_max)
1989                         return -EINVAL;
1990                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1991                         return -EINVAL;
1992                 tcpm_log(port,
1993                          "Requested %u -> %u mV, %u mW for %u / %u mW",
1994                          pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1995                          pdo_max, op, max);
1996                 break;
1997         default:
1998                 return -EINVAL;
1999         }
2000
2001         port->op_vsafe5v = index == 1;
2002
2003         return 0;
2004 }
2005
2006 #define min_power(x, y) min(pdo_max_power(x), pdo_max_power(y))
2007 #define min_current(x, y) min(pdo_max_current(x), pdo_max_current(y))
2008
2009 static int tcpm_pd_select_pdo(struct tcpm_port *port, int *sink_pdo,
2010                               int *src_pdo)
2011 {
2012         unsigned int i, j, max_src_mv = 0, min_src_mv = 0, max_mw = 0,
2013                      max_mv = 0, src_mw = 0, src_ma = 0, max_snk_mv = 0,
2014                      min_snk_mv = 0;
2015         int ret = -EINVAL;
2016
2017         port->pps_data.supported = false;
2018         port->usb_type = POWER_SUPPLY_USB_TYPE_PD;
2019
2020         /*
2021          * Select the source PDO providing the most power which has a
2022          * matchig sink cap.
2023          */
2024         for (i = 0; i < port->nr_source_caps; i++) {
2025                 u32 pdo = port->source_caps[i];
2026                 enum pd_pdo_type type = pdo_type(pdo);
2027
2028                 switch (type) {
2029                 case PDO_TYPE_FIXED:
2030                         max_src_mv = pdo_fixed_voltage(pdo);
2031                         min_src_mv = max_src_mv;
2032                         break;
2033                 case PDO_TYPE_BATT:
2034                 case PDO_TYPE_VAR:
2035                         max_src_mv = pdo_max_voltage(pdo);
2036                         min_src_mv = pdo_min_voltage(pdo);
2037                         break;
2038                 case PDO_TYPE_APDO:
2039                         if (pdo_apdo_type(pdo) == APDO_TYPE_PPS) {
2040                                 port->pps_data.supported = true;
2041                                 port->usb_type =
2042                                         POWER_SUPPLY_USB_TYPE_PD_PPS;
2043                         }
2044                         continue;
2045                 default:
2046                         tcpm_log(port, "Invalid source PDO type, ignoring");
2047                         continue;
2048                 }
2049
2050                 switch (type) {
2051                 case PDO_TYPE_FIXED:
2052                 case PDO_TYPE_VAR:
2053                         src_ma = pdo_max_current(pdo);
2054                         src_mw = src_ma * min_src_mv / 1000;
2055                         break;
2056                 case PDO_TYPE_BATT:
2057                         src_mw = pdo_max_power(pdo);
2058                         break;
2059                 case PDO_TYPE_APDO:
2060                         continue;
2061                 default:
2062                         tcpm_log(port, "Invalid source PDO type, ignoring");
2063                         continue;
2064                 }
2065
2066                 for (j = 0; j < port->nr_snk_pdo; j++) {
2067                         pdo = port->snk_pdo[j];
2068
2069                         switch (pdo_type(pdo)) {
2070                         case PDO_TYPE_FIXED:
2071                                 max_snk_mv = pdo_fixed_voltage(pdo);
2072                                 min_snk_mv = max_snk_mv;
2073                                 break;
2074                         case PDO_TYPE_BATT:
2075                         case PDO_TYPE_VAR:
2076                                 max_snk_mv = pdo_max_voltage(pdo);
2077                                 min_snk_mv = pdo_min_voltage(pdo);
2078                                 break;
2079                         case PDO_TYPE_APDO:
2080                                 continue;
2081                         default:
2082                                 tcpm_log(port, "Invalid sink PDO type, ignoring");
2083                                 continue;
2084                         }
2085
2086                         if (max_src_mv <= max_snk_mv &&
2087                                 min_src_mv >= min_snk_mv) {
2088                                 /* Prefer higher voltages if available */
2089                                 if ((src_mw == max_mw && min_src_mv > max_mv) ||
2090                                                         src_mw > max_mw) {
2091                                         *src_pdo = i;
2092                                         *sink_pdo = j;
2093                                         max_mw = src_mw;
2094                                         max_mv = min_src_mv;
2095                                         ret = 0;
2096                                 }
2097                         }
2098                 }
2099         }
2100
2101         return ret;
2102 }
2103
2104 #define min_pps_apdo_current(x, y)      \
2105         min(pdo_pps_apdo_max_current(x), pdo_pps_apdo_max_current(y))
2106
2107 static unsigned int tcpm_pd_select_pps_apdo(struct tcpm_port *port)
2108 {
2109         unsigned int i, j, max_mw = 0, max_mv = 0;
2110         unsigned int min_src_mv, max_src_mv, src_ma, src_mw;
2111         unsigned int min_snk_mv, max_snk_mv, snk_ma;
2112         u32 pdo;
2113         unsigned int src_pdo = 0, snk_pdo = 0;
2114
2115         /*
2116          * Select the source PPS APDO providing the most power while staying
2117          * within the board's limits. We skip the first PDO as this is always
2118          * 5V 3A.
2119          */
2120         for (i = 1; i < port->nr_source_caps; ++i) {
2121                 pdo = port->source_caps[i];
2122
2123                 switch (pdo_type(pdo)) {
2124                 case PDO_TYPE_APDO:
2125                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2126                                 tcpm_log(port, "Not PPS APDO (source), ignoring");
2127                                 continue;
2128                         }
2129
2130                         min_src_mv = pdo_pps_apdo_min_voltage(pdo);
2131                         max_src_mv = pdo_pps_apdo_max_voltage(pdo);
2132                         src_ma = pdo_pps_apdo_max_current(pdo);
2133                         src_mw = (src_ma * max_src_mv) / 1000;
2134
2135                         /*
2136                          * Now search through the sink PDOs to find a matching
2137                          * PPS APDO. Again skip the first sink PDO as this will
2138                          * always be 5V 3A.
2139                          */
2140                         for (j = i; j < port->nr_snk_pdo; j++) {
2141                                 pdo = port->snk_pdo[j];
2142
2143                                 switch (pdo_type(pdo)) {
2144                                 case PDO_TYPE_APDO:
2145                                         if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2146                                                 tcpm_log(port,
2147                                                          "Not PPS APDO (sink), ignoring");
2148                                                 continue;
2149                                         }
2150
2151                                         min_snk_mv =
2152                                                 pdo_pps_apdo_min_voltage(pdo);
2153                                         max_snk_mv =
2154                                                 pdo_pps_apdo_max_voltage(pdo);
2155                                         snk_ma =
2156                                                 pdo_pps_apdo_max_current(pdo);
2157                                         break;
2158                                 default:
2159                                         tcpm_log(port,
2160                                                  "Not APDO type (sink), ignoring");
2161                                         continue;
2162                                 }
2163
2164                                 if (max_src_mv <= max_snk_mv &&
2165                                     min_src_mv >= min_snk_mv) {
2166                                         /* Prefer higher voltages if available */
2167                                         if ((src_mw == max_mw &&
2168                                              min_src_mv > max_mv) ||
2169                                             src_mw > max_mw) {
2170                                                 src_pdo = i;
2171                                                 snk_pdo = j;
2172                                                 max_mw = src_mw;
2173                                                 max_mv = max_src_mv;
2174                                         }
2175                                 }
2176                         }
2177
2178                         break;
2179                 default:
2180                         tcpm_log(port, "Not APDO type (source), ignoring");
2181                         continue;
2182                 }
2183         }
2184
2185         if (src_pdo) {
2186                 pdo = port->source_caps[src_pdo];
2187
2188                 port->pps_data.min_volt = pdo_pps_apdo_min_voltage(pdo);
2189                 port->pps_data.max_volt = pdo_pps_apdo_max_voltage(pdo);
2190                 port->pps_data.max_curr =
2191                         min_pps_apdo_current(pdo, port->snk_pdo[snk_pdo]);
2192                 port->pps_data.out_volt =
2193                         min(pdo_pps_apdo_max_voltage(pdo), port->pps_data.out_volt);
2194                 port->pps_data.op_curr =
2195                         min(port->pps_data.max_curr, port->pps_data.op_curr);
2196         }
2197
2198         return src_pdo;
2199 }
2200
2201 static int tcpm_pd_build_request(struct tcpm_port *port, u32 *rdo)
2202 {
2203         unsigned int mv, ma, mw, flags;
2204         unsigned int max_ma, max_mw;
2205         enum pd_pdo_type type;
2206         u32 pdo, matching_snk_pdo;
2207         int src_pdo_index = 0;
2208         int snk_pdo_index = 0;
2209         int ret;
2210
2211         ret = tcpm_pd_select_pdo(port, &snk_pdo_index, &src_pdo_index);
2212         if (ret < 0)
2213                 return ret;
2214
2215         pdo = port->source_caps[src_pdo_index];
2216         matching_snk_pdo = port->snk_pdo[snk_pdo_index];
2217         type = pdo_type(pdo);
2218
2219         switch (type) {
2220         case PDO_TYPE_FIXED:
2221                 mv = pdo_fixed_voltage(pdo);
2222                 break;
2223         case PDO_TYPE_BATT:
2224         case PDO_TYPE_VAR:
2225                 mv = pdo_min_voltage(pdo);
2226                 break;
2227         default:
2228                 tcpm_log(port, "Invalid PDO selected!");
2229                 return -EINVAL;
2230         }
2231
2232         /* Select maximum available current within the sink pdo's limit */
2233         if (type == PDO_TYPE_BATT) {
2234                 mw = min_power(pdo, matching_snk_pdo);
2235                 ma = 1000 * mw / mv;
2236         } else {
2237                 ma = min_current(pdo, matching_snk_pdo);
2238                 mw = ma * mv / 1000;
2239         }
2240
2241         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2242
2243         /* Set mismatch bit if offered power is less than operating power */
2244         max_ma = ma;
2245         max_mw = mw;
2246         if (mw < port->operating_snk_mw) {
2247                 flags |= RDO_CAP_MISMATCH;
2248                 if (type == PDO_TYPE_BATT &&
2249                     (pdo_max_power(matching_snk_pdo) > pdo_max_power(pdo)))
2250                         max_mw = pdo_max_power(matching_snk_pdo);
2251                 else if (pdo_max_current(matching_snk_pdo) >
2252                          pdo_max_current(pdo))
2253                         max_ma = pdo_max_current(matching_snk_pdo);
2254         }
2255
2256         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2257                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2258                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2259                  port->polarity);
2260
2261         if (type == PDO_TYPE_BATT) {
2262                 *rdo = RDO_BATT(src_pdo_index + 1, mw, max_mw, flags);
2263
2264                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mW%s",
2265                          src_pdo_index, mv, mw,
2266                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2267         } else {
2268                 *rdo = RDO_FIXED(src_pdo_index + 1, ma, max_ma, flags);
2269
2270                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mA%s",
2271                          src_pdo_index, mv, ma,
2272                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
2273         }
2274
2275         port->current_limit = ma;
2276         port->supply_voltage = mv;
2277
2278         return 0;
2279 }
2280
2281 static int tcpm_pd_send_request(struct tcpm_port *port)
2282 {
2283         struct pd_message msg;
2284         int ret;
2285         u32 rdo;
2286
2287         ret = tcpm_pd_build_request(port, &rdo);
2288         if (ret < 0)
2289                 return ret;
2290
2291         memset(&msg, 0, sizeof(msg));
2292         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2293                                   port->pwr_role,
2294                                   port->data_role,
2295                                   port->negotiated_rev,
2296                                   port->message_id, 1);
2297         msg.payload[0] = cpu_to_le32(rdo);
2298
2299         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2300 }
2301
2302 static int tcpm_pd_build_pps_request(struct tcpm_port *port, u32 *rdo)
2303 {
2304         unsigned int out_mv, op_ma, op_mw, min_mv, max_mv, max_ma, flags;
2305         enum pd_pdo_type type;
2306         unsigned int src_pdo_index;
2307         u32 pdo;
2308
2309         src_pdo_index = tcpm_pd_select_pps_apdo(port);
2310         if (!src_pdo_index)
2311                 return -EOPNOTSUPP;
2312
2313         pdo = port->source_caps[src_pdo_index];
2314         type = pdo_type(pdo);
2315
2316         switch (type) {
2317         case PDO_TYPE_APDO:
2318                 if (pdo_apdo_type(pdo) != APDO_TYPE_PPS) {
2319                         tcpm_log(port, "Invalid APDO selected!");
2320                         return -EINVAL;
2321                 }
2322                 min_mv = port->pps_data.min_volt;
2323                 max_mv = port->pps_data.max_volt;
2324                 max_ma = port->pps_data.max_curr;
2325                 out_mv = port->pps_data.out_volt;
2326                 op_ma = port->pps_data.op_curr;
2327                 break;
2328         default:
2329                 tcpm_log(port, "Invalid PDO selected!");
2330                 return -EINVAL;
2331         }
2332
2333         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
2334
2335         op_mw = (op_ma * out_mv) / 1000;
2336         if (op_mw < port->operating_snk_mw) {
2337                 /*
2338                  * Try raising current to meet power needs. If that's not enough
2339                  * then try upping the voltage. If that's still not enough
2340                  * then we've obviously chosen a PPS APDO which really isn't
2341                  * suitable so abandon ship.
2342                  */
2343                 op_ma = (port->operating_snk_mw * 1000) / out_mv;
2344                 if ((port->operating_snk_mw * 1000) % out_mv)
2345                         ++op_ma;
2346                 op_ma += RDO_PROG_CURR_MA_STEP - (op_ma % RDO_PROG_CURR_MA_STEP);
2347
2348                 if (op_ma > max_ma) {
2349                         op_ma = max_ma;
2350                         out_mv = (port->operating_snk_mw * 1000) / op_ma;
2351                         if ((port->operating_snk_mw * 1000) % op_ma)
2352                                 ++out_mv;
2353                         out_mv += RDO_PROG_VOLT_MV_STEP -
2354                                   (out_mv % RDO_PROG_VOLT_MV_STEP);
2355
2356                         if (out_mv > max_mv) {
2357                                 tcpm_log(port, "Invalid PPS APDO selected!");
2358                                 return -EINVAL;
2359                         }
2360                 }
2361         }
2362
2363         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
2364                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
2365                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
2366                  port->polarity);
2367
2368         *rdo = RDO_PROG(src_pdo_index + 1, out_mv, op_ma, flags);
2369
2370         tcpm_log(port, "Requesting APDO %d: %u mV, %u mA",
2371                  src_pdo_index, out_mv, op_ma);
2372
2373         port->pps_data.op_curr = op_ma;
2374         port->pps_data.out_volt = out_mv;
2375
2376         return 0;
2377 }
2378
2379 static int tcpm_pd_send_pps_request(struct tcpm_port *port)
2380 {
2381         struct pd_message msg;
2382         int ret;
2383         u32 rdo;
2384
2385         ret = tcpm_pd_build_pps_request(port, &rdo);
2386         if (ret < 0)
2387                 return ret;
2388
2389         memset(&msg, 0, sizeof(msg));
2390         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
2391                                   port->pwr_role,
2392                                   port->data_role,
2393                                   port->negotiated_rev,
2394                                   port->message_id, 1);
2395         msg.payload[0] = cpu_to_le32(rdo);
2396
2397         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
2398 }
2399
2400 static int tcpm_set_vbus(struct tcpm_port *port, bool enable)
2401 {
2402         int ret;
2403
2404         if (enable && port->vbus_charge)
2405                 return -EINVAL;
2406
2407         tcpm_log(port, "vbus:=%d charge=%d", enable, port->vbus_charge);
2408
2409         ret = port->tcpc->set_vbus(port->tcpc, enable, port->vbus_charge);
2410         if (ret < 0)
2411                 return ret;
2412
2413         port->vbus_source = enable;
2414         return 0;
2415 }
2416
2417 static int tcpm_set_charge(struct tcpm_port *port, bool charge)
2418 {
2419         int ret;
2420
2421         if (charge && port->vbus_source)
2422                 return -EINVAL;
2423
2424         if (charge != port->vbus_charge) {
2425                 tcpm_log(port, "vbus=%d charge:=%d", port->vbus_source, charge);
2426                 ret = port->tcpc->set_vbus(port->tcpc, port->vbus_source,
2427                                            charge);
2428                 if (ret < 0)
2429                         return ret;
2430         }
2431         port->vbus_charge = charge;
2432         return 0;
2433 }
2434
2435 static bool tcpm_start_drp_toggling(struct tcpm_port *port)
2436 {
2437         int ret;
2438
2439         if (port->tcpc->start_drp_toggling &&
2440             port->port_type == TYPEC_PORT_DRP) {
2441                 tcpm_log_force(port, "Start DRP toggling");
2442                 ret = port->tcpc->start_drp_toggling(port->tcpc,
2443                                                      tcpm_rp_cc(port));
2444                 if (!ret)
2445                         return true;
2446         }
2447
2448         return false;
2449 }
2450
2451 static void tcpm_set_cc(struct tcpm_port *port, enum typec_cc_status cc)
2452 {
2453         tcpm_log(port, "cc:=%d", cc);
2454         port->cc_req = cc;
2455         port->tcpc->set_cc(port->tcpc, cc);
2456 }
2457
2458 static int tcpm_init_vbus(struct tcpm_port *port)
2459 {
2460         int ret;
2461
2462         ret = port->tcpc->set_vbus(port->tcpc, false, false);
2463         port->vbus_source = false;
2464         port->vbus_charge = false;
2465         return ret;
2466 }
2467
2468 static int tcpm_init_vconn(struct tcpm_port *port)
2469 {
2470         int ret;
2471
2472         ret = port->tcpc->set_vconn(port->tcpc, false);
2473         port->vconn_role = TYPEC_SINK;
2474         return ret;
2475 }
2476
2477 static void tcpm_typec_connect(struct tcpm_port *port)
2478 {
2479         if (!port->connected) {
2480                 /* Make sure we don't report stale identity information */
2481                 memset(&port->partner_ident, 0, sizeof(port->partner_ident));
2482                 port->partner_desc.usb_pd = port->pd_capable;
2483                 if (tcpm_port_is_debug(port))
2484                         port->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
2485                 else if (tcpm_port_is_audio(port))
2486                         port->partner_desc.accessory = TYPEC_ACCESSORY_AUDIO;
2487                 else
2488                         port->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
2489                 port->partner = typec_register_partner(port->typec_port,
2490                                                        &port->partner_desc);
2491                 port->connected = true;
2492         }
2493 }
2494
2495 static int tcpm_src_attach(struct tcpm_port *port)
2496 {
2497         enum typec_cc_polarity polarity =
2498                                 port->cc2 == TYPEC_CC_RD ? TYPEC_POLARITY_CC2
2499                                                          : TYPEC_POLARITY_CC1;
2500         int ret;
2501
2502         if (port->attached)
2503                 return 0;
2504
2505         ret = tcpm_set_polarity(port, polarity);
2506         if (ret < 0)
2507                 return ret;
2508
2509         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2510         if (ret < 0)
2511                 return ret;
2512
2513         ret = port->tcpc->set_pd_rx(port->tcpc, true);
2514         if (ret < 0)
2515                 goto out_disable_mux;
2516
2517         /*
2518          * USB Type-C specification, version 1.2,
2519          * chapter 4.5.2.2.8.1 (Attached.SRC Requirements)
2520          * Enable VCONN only if the non-RD port is set to RA.
2521          */
2522         if ((polarity == TYPEC_POLARITY_CC1 && port->cc2 == TYPEC_CC_RA) ||
2523             (polarity == TYPEC_POLARITY_CC2 && port->cc1 == TYPEC_CC_RA)) {
2524                 ret = tcpm_set_vconn(port, true);
2525                 if (ret < 0)
2526                         goto out_disable_pd;
2527         }
2528
2529         ret = tcpm_set_vbus(port, true);
2530         if (ret < 0)
2531                 goto out_disable_vconn;
2532
2533         port->pd_capable = false;
2534
2535         port->partner = NULL;
2536
2537         port->attached = true;
2538         port->send_discover = true;
2539
2540         return 0;
2541
2542 out_disable_vconn:
2543         tcpm_set_vconn(port, false);
2544 out_disable_pd:
2545         port->tcpc->set_pd_rx(port->tcpc, false);
2546 out_disable_mux:
2547         tcpm_mux_set(port, TYPEC_MUX_NONE, USB_ROLE_NONE,
2548                      TYPEC_ORIENTATION_NONE);
2549         return ret;
2550 }
2551
2552 static void tcpm_typec_disconnect(struct tcpm_port *port)
2553 {
2554         if (port->connected) {
2555                 typec_unregister_partner(port->partner);
2556                 port->partner = NULL;
2557                 port->connected = false;
2558         }
2559 }
2560
2561 static void tcpm_unregister_altmodes(struct tcpm_port *port)
2562 {
2563         struct pd_mode_data *modep = &port->mode_data;
2564         int i;
2565
2566         for (i = 0; i < modep->altmodes; i++) {
2567                 typec_unregister_altmode(port->partner_altmode[i]);
2568                 port->partner_altmode[i] = NULL;
2569         }
2570
2571         memset(modep, 0, sizeof(*modep));
2572 }
2573
2574 static void tcpm_reset_port(struct tcpm_port *port)
2575 {
2576         tcpm_unregister_altmodes(port);
2577         tcpm_typec_disconnect(port);
2578         port->attached = false;
2579         port->pd_capable = false;
2580         port->pps_data.supported = false;
2581
2582         /*
2583          * First Rx ID should be 0; set this to a sentinel of -1 so that
2584          * we can check tcpm_pd_rx_handler() if we had seen it before.
2585          */
2586         port->rx_msgid = -1;
2587
2588         port->tcpc->set_pd_rx(port->tcpc, false);
2589         tcpm_init_vbus(port);   /* also disables charging */
2590         tcpm_init_vconn(port);
2591         tcpm_set_current_limit(port, 0, 0);
2592         tcpm_set_polarity(port, TYPEC_POLARITY_CC1);
2593         tcpm_mux_set(port, TYPEC_MUX_NONE, USB_ROLE_NONE,
2594                      TYPEC_ORIENTATION_NONE);
2595         tcpm_set_attached_state(port, false);
2596         port->try_src_count = 0;
2597         port->try_snk_count = 0;
2598         port->supply_voltage = 0;
2599         port->current_limit = 0;
2600         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
2601
2602         power_supply_changed(port->psy);
2603 }
2604
2605 static void tcpm_detach(struct tcpm_port *port)
2606 {
2607         if (!port->attached)
2608                 return;
2609
2610         if (tcpm_port_is_disconnected(port))
2611                 port->hard_reset_count = 0;
2612
2613         tcpm_reset_port(port);
2614 }
2615
2616 static void tcpm_src_detach(struct tcpm_port *port)
2617 {
2618         tcpm_detach(port);
2619 }
2620
2621 static int tcpm_snk_attach(struct tcpm_port *port)
2622 {
2623         int ret;
2624
2625         if (port->attached)
2626                 return 0;
2627
2628         ret = tcpm_set_polarity(port, port->cc2 != TYPEC_CC_OPEN ?
2629                                 TYPEC_POLARITY_CC2 : TYPEC_POLARITY_CC1);
2630         if (ret < 0)
2631                 return ret;
2632
2633         ret = tcpm_set_roles(port, true, TYPEC_SINK, TYPEC_DEVICE);
2634         if (ret < 0)
2635                 return ret;
2636
2637         port->pd_capable = false;
2638
2639         port->partner = NULL;
2640
2641         port->attached = true;
2642         port->send_discover = true;
2643
2644         return 0;
2645 }
2646
2647 static void tcpm_snk_detach(struct tcpm_port *port)
2648 {
2649         tcpm_detach(port);
2650 }
2651
2652 static int tcpm_acc_attach(struct tcpm_port *port)
2653 {
2654         int ret;
2655
2656         if (port->attached)
2657                 return 0;
2658
2659         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2660         if (ret < 0)
2661                 return ret;
2662
2663         port->partner = NULL;
2664
2665         tcpm_typec_connect(port);
2666
2667         port->attached = true;
2668
2669         return 0;
2670 }
2671
2672 static void tcpm_acc_detach(struct tcpm_port *port)
2673 {
2674         tcpm_detach(port);
2675 }
2676
2677 static inline enum tcpm_state hard_reset_state(struct tcpm_port *port)
2678 {
2679         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
2680                 return HARD_RESET_SEND;
2681         if (port->pd_capable)
2682                 return ERROR_RECOVERY;
2683         if (port->pwr_role == TYPEC_SOURCE)
2684                 return SRC_UNATTACHED;
2685         if (port->state == SNK_WAIT_CAPABILITIES)
2686                 return SNK_READY;
2687         return SNK_UNATTACHED;
2688 }
2689
2690 static inline enum tcpm_state unattached_state(struct tcpm_port *port)
2691 {
2692         if (port->port_type == TYPEC_PORT_DRP) {
2693                 if (port->pwr_role == TYPEC_SOURCE)
2694                         return SRC_UNATTACHED;
2695                 else
2696                         return SNK_UNATTACHED;
2697         } else if (port->port_type == TYPEC_PORT_SRC) {
2698                 return SRC_UNATTACHED;
2699         }
2700
2701         return SNK_UNATTACHED;
2702 }
2703
2704 static void tcpm_check_send_discover(struct tcpm_port *port)
2705 {
2706         if (port->data_role == TYPEC_HOST && port->send_discover &&
2707             port->pd_capable) {
2708                 tcpm_send_vdm(port, USB_SID_PD, CMD_DISCOVER_IDENT, NULL, 0);
2709                 port->send_discover = false;
2710         }
2711 }
2712
2713 static void tcpm_swap_complete(struct tcpm_port *port, int result)
2714 {
2715         if (port->swap_pending) {
2716                 port->swap_status = result;
2717                 port->swap_pending = false;
2718                 port->non_pd_role_swap = false;
2719                 complete(&port->swap_complete);
2720         }
2721 }
2722
2723 static enum typec_pwr_opmode tcpm_get_pwr_opmode(enum typec_cc_status cc)
2724 {
2725         switch (cc) {
2726         case TYPEC_CC_RP_1_5:
2727                 return TYPEC_PWR_MODE_1_5A;
2728         case TYPEC_CC_RP_3_0:
2729                 return TYPEC_PWR_MODE_3_0A;
2730         case TYPEC_CC_RP_DEF:
2731         default:
2732                 return TYPEC_PWR_MODE_USB;
2733         }
2734 }
2735
2736 static void run_state_machine(struct tcpm_port *port)
2737 {
2738         int ret;
2739         enum typec_pwr_opmode opmode;
2740         unsigned int msecs;
2741
2742         port->enter_state = port->state;
2743         switch (port->state) {
2744         case DRP_TOGGLING:
2745                 break;
2746         /* SRC states */
2747         case SRC_UNATTACHED:
2748                 if (!port->non_pd_role_swap)
2749                         tcpm_swap_complete(port, -ENOTCONN);
2750                 tcpm_src_detach(port);
2751                 if (tcpm_start_drp_toggling(port)) {
2752                         tcpm_set_state(port, DRP_TOGGLING, 0);
2753                         break;
2754                 }
2755                 tcpm_set_cc(port, tcpm_rp_cc(port));
2756                 if (port->port_type == TYPEC_PORT_DRP)
2757                         tcpm_set_state(port, SNK_UNATTACHED, PD_T_DRP_SNK);
2758                 break;
2759         case SRC_ATTACH_WAIT:
2760                 if (tcpm_port_is_debug(port))
2761                         tcpm_set_state(port, DEBUG_ACC_ATTACHED,
2762                                        PD_T_CC_DEBOUNCE);
2763                 else if (tcpm_port_is_audio(port))
2764                         tcpm_set_state(port, AUDIO_ACC_ATTACHED,
2765                                        PD_T_CC_DEBOUNCE);
2766                 else if (tcpm_port_is_source(port))
2767                         tcpm_set_state(port,
2768                                        tcpm_try_snk(port) ? SNK_TRY
2769                                                           : SRC_ATTACHED,
2770                                        PD_T_CC_DEBOUNCE);
2771                 break;
2772
2773         case SNK_TRY:
2774                 port->try_snk_count++;
2775                 /*
2776                  * Requirements:
2777                  * - Do not drive vconn or vbus
2778                  * - Terminate CC pins (both) to Rd
2779                  * Action:
2780                  * - Wait for tDRPTry (PD_T_DRP_TRY).
2781                  *   Until then, ignore any state changes.
2782                  */
2783                 tcpm_set_cc(port, TYPEC_CC_RD);
2784                 tcpm_set_state(port, SNK_TRY_WAIT, PD_T_DRP_TRY);
2785                 break;
2786         case SNK_TRY_WAIT:
2787                 if (tcpm_port_is_sink(port)) {
2788                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE, 0);
2789                 } else {
2790                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2791                         port->max_wait = 0;
2792                 }
2793                 break;
2794         case SNK_TRY_WAIT_DEBOUNCE:
2795                 tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS,
2796                                PD_T_PD_DEBOUNCE);
2797                 break;
2798         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
2799                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2800                         tcpm_set_state(port, SNK_ATTACHED, 0);
2801                 } else {
2802                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2803                         port->max_wait = 0;
2804                 }
2805                 break;
2806         case SRC_TRYWAIT:
2807                 tcpm_set_cc(port, tcpm_rp_cc(port));
2808                 if (port->max_wait == 0) {
2809                         port->max_wait = jiffies +
2810                                          msecs_to_jiffies(PD_T_DRP_TRY);
2811                         tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2812                                        PD_T_DRP_TRY);
2813                 } else {
2814                         if (time_is_after_jiffies(port->max_wait))
2815                                 tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2816                                                jiffies_to_msecs(port->max_wait -
2817                                                                 jiffies));
2818                         else
2819                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2820                 }
2821                 break;
2822         case SRC_TRYWAIT_DEBOUNCE:
2823                 tcpm_set_state(port, SRC_ATTACHED, PD_T_CC_DEBOUNCE);
2824                 break;
2825         case SRC_TRYWAIT_UNATTACHED:
2826                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2827                 break;
2828
2829         case SRC_ATTACHED:
2830                 ret = tcpm_src_attach(port);
2831                 tcpm_set_state(port, SRC_UNATTACHED,
2832                                ret < 0 ? 0 : PD_T_PS_SOURCE_ON);
2833                 break;
2834         case SRC_STARTUP:
2835                 opmode =  tcpm_get_pwr_opmode(tcpm_rp_cc(port));
2836                 typec_set_pwr_opmode(port->typec_port, opmode);
2837                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
2838                 port->caps_count = 0;
2839                 port->negotiated_rev = PD_MAX_REV;
2840                 port->message_id = 0;
2841                 port->rx_msgid = -1;
2842                 port->explicit_contract = false;
2843                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
2844                 break;
2845         case SRC_SEND_CAPABILITIES:
2846                 port->caps_count++;
2847                 if (port->caps_count > PD_N_CAPS_COUNT) {
2848                         tcpm_set_state(port, SRC_READY, 0);
2849                         break;
2850                 }
2851                 ret = tcpm_pd_send_source_caps(port);
2852                 if (ret < 0) {
2853                         tcpm_set_state(port, SRC_SEND_CAPABILITIES,
2854                                        PD_T_SEND_SOURCE_CAP);
2855                 } else {
2856                         /*
2857                          * Per standard, we should clear the reset counter here.
2858                          * However, that can result in state machine hang-ups.
2859                          * Reset it only in READY state to improve stability.
2860                          */
2861                         /* port->hard_reset_count = 0; */
2862                         port->caps_count = 0;
2863                         port->pd_capable = true;
2864                         tcpm_set_state_cond(port, hard_reset_state(port),
2865                                             PD_T_SEND_SOURCE_CAP);
2866                 }
2867                 break;
2868         case SRC_NEGOTIATE_CAPABILITIES:
2869                 ret = tcpm_pd_check_request(port);
2870                 if (ret < 0) {
2871                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
2872                         if (!port->explicit_contract) {
2873                                 tcpm_set_state(port,
2874                                                SRC_WAIT_NEW_CAPABILITIES, 0);
2875                         } else {
2876                                 tcpm_set_state(port, SRC_READY, 0);
2877                         }
2878                 } else {
2879                         tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2880                         tcpm_set_state(port, SRC_TRANSITION_SUPPLY,
2881                                        PD_T_SRC_TRANSITION);
2882                 }
2883                 break;
2884         case SRC_TRANSITION_SUPPLY:
2885                 /* XXX: regulator_set_voltage(vbus, ...) */
2886                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
2887                 port->explicit_contract = true;
2888                 typec_set_pwr_opmode(port->typec_port, TYPEC_PWR_MODE_PD);
2889                 port->pwr_opmode = TYPEC_PWR_MODE_PD;
2890                 tcpm_set_state_cond(port, SRC_READY, 0);
2891                 break;
2892         case SRC_READY:
2893 #if 1
2894                 port->hard_reset_count = 0;
2895 #endif
2896                 port->try_src_count = 0;
2897
2898                 tcpm_swap_complete(port, 0);
2899                 tcpm_typec_connect(port);
2900
2901                 tcpm_check_send_discover(port);
2902                 /*
2903                  * 6.3.5
2904                  * Sending ping messages is not necessary if
2905                  * - the source operates at vSafe5V
2906                  * or
2907                  * - The system is not operating in PD mode
2908                  * or
2909                  * - Both partners are connected using a Type-C connector
2910                  *
2911                  * There is no actual need to send PD messages since the local
2912                  * port type-c and the spec does not clearly say whether PD is
2913                  * possible when type-c is connected to Type-A/B
2914                  */
2915                 break;
2916         case SRC_WAIT_NEW_CAPABILITIES:
2917                 /* Nothing to do... */
2918                 break;
2919
2920         /* SNK states */
2921         case SNK_UNATTACHED:
2922                 if (!port->non_pd_role_swap)
2923                         tcpm_swap_complete(port, -ENOTCONN);
2924                 tcpm_pps_complete(port, -ENOTCONN);
2925                 tcpm_snk_detach(port);
2926                 if (tcpm_start_drp_toggling(port)) {
2927                         tcpm_set_state(port, DRP_TOGGLING, 0);
2928                         break;
2929                 }
2930                 tcpm_set_cc(port, TYPEC_CC_RD);
2931                 if (port->port_type == TYPEC_PORT_DRP)
2932                         tcpm_set_state(port, SRC_UNATTACHED, PD_T_DRP_SRC);
2933                 break;
2934         case SNK_ATTACH_WAIT:
2935                 if ((port->cc1 == TYPEC_CC_OPEN &&
2936                      port->cc2 != TYPEC_CC_OPEN) ||
2937                     (port->cc1 != TYPEC_CC_OPEN &&
2938                      port->cc2 == TYPEC_CC_OPEN))
2939                         tcpm_set_state(port, SNK_DEBOUNCED,
2940                                        PD_T_CC_DEBOUNCE);
2941                 else if (tcpm_port_is_disconnected(port))
2942                         tcpm_set_state(port, SNK_UNATTACHED,
2943                                        PD_T_PD_DEBOUNCE);
2944                 break;
2945         case SNK_DEBOUNCED:
2946                 if (tcpm_port_is_disconnected(port))
2947                         tcpm_set_state(port, SNK_UNATTACHED,
2948                                        PD_T_PD_DEBOUNCE);
2949                 else if (port->vbus_present)
2950                         tcpm_set_state(port,
2951                                        tcpm_try_src(port) ? SRC_TRY
2952                                                           : SNK_ATTACHED,
2953                                        0);
2954                 else
2955                         /* Wait for VBUS, but not forever */
2956                         tcpm_set_state(port, PORT_RESET, PD_T_PS_SOURCE_ON);
2957                 break;
2958
2959         case SRC_TRY:
2960                 port->try_src_count++;
2961                 tcpm_set_cc(port, tcpm_rp_cc(port));
2962                 port->max_wait = 0;
2963                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
2964                 break;
2965         case SRC_TRY_WAIT:
2966                 if (port->max_wait == 0) {
2967                         port->max_wait = jiffies +
2968                                          msecs_to_jiffies(PD_T_DRP_TRY);
2969                         msecs = PD_T_DRP_TRY;
2970                 } else {
2971                         if (time_is_after_jiffies(port->max_wait))
2972                                 msecs = jiffies_to_msecs(port->max_wait -
2973                                                          jiffies);
2974                         else
2975                                 msecs = 0;
2976                 }
2977                 tcpm_set_state(port, SNK_TRYWAIT, msecs);
2978                 break;
2979         case SRC_TRY_DEBOUNCE:
2980                 tcpm_set_state(port, SRC_ATTACHED, PD_T_PD_DEBOUNCE);
2981                 break;
2982         case SNK_TRYWAIT:
2983                 tcpm_set_cc(port, TYPEC_CC_RD);
2984                 tcpm_set_state(port, SNK_TRYWAIT_VBUS, PD_T_CC_DEBOUNCE);
2985                 break;
2986         case SNK_TRYWAIT_VBUS:
2987                 /*
2988                  * TCPM stays in this state indefinitely until VBUS
2989                  * is detected as long as Rp is not detected for
2990                  * more than a time period of tPDDebounce.
2991                  */
2992                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2993                         tcpm_set_state(port, SNK_ATTACHED, 0);
2994                         break;
2995                 }
2996                 if (!tcpm_port_is_sink(port))
2997                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
2998                 break;
2999         case SNK_TRYWAIT_DEBOUNCE:
3000                 tcpm_set_state(port, SNK_UNATTACHED, PD_T_PD_DEBOUNCE);
3001                 break;
3002         case SNK_ATTACHED:
3003                 ret = tcpm_snk_attach(port);
3004                 if (ret < 0)
3005                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3006                 else
3007                         tcpm_set_state(port, SNK_STARTUP, 0);
3008                 break;
3009         case SNK_STARTUP:
3010                 opmode =  tcpm_get_pwr_opmode(port->polarity ?
3011                                               port->cc2 : port->cc1);
3012                 typec_set_pwr_opmode(port->typec_port, opmode);
3013                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
3014                 port->negotiated_rev = PD_MAX_REV;
3015                 port->message_id = 0;
3016                 port->rx_msgid = -1;
3017                 port->explicit_contract = false;
3018                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3019                 break;
3020         case SNK_DISCOVERY:
3021                 if (port->vbus_present) {
3022                         tcpm_set_current_limit(port,
3023                                                tcpm_get_current_limit(port),
3024                                                5000);
3025                         tcpm_set_charge(port, true);
3026                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3027                         break;
3028                 }
3029                 /*
3030                  * For DRP, timeouts differ. Also, handling is supposed to be
3031                  * different and much more complex (dead battery detection;
3032                  * see USB power delivery specification, section 8.3.3.6.1.5.1).
3033                  */
3034                 tcpm_set_state(port, hard_reset_state(port),
3035                                port->port_type == TYPEC_PORT_DRP ?
3036                                         PD_T_DB_DETECT : PD_T_NO_RESPONSE);
3037                 break;
3038         case SNK_DISCOVERY_DEBOUNCE:
3039                 tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE_DONE,
3040                                PD_T_CC_DEBOUNCE);
3041                 break;
3042         case SNK_DISCOVERY_DEBOUNCE_DONE:
3043                 if (!tcpm_port_is_disconnected(port) &&
3044                     tcpm_port_is_sink(port) &&
3045                     time_is_after_jiffies(port->delayed_runtime)) {
3046                         tcpm_set_state(port, SNK_DISCOVERY,
3047                                        jiffies_to_msecs(port->delayed_runtime -
3048                                                         jiffies));
3049                         break;
3050                 }
3051                 tcpm_set_state(port, unattached_state(port), 0);
3052                 break;
3053         case SNK_WAIT_CAPABILITIES:
3054                 ret = port->tcpc->set_pd_rx(port->tcpc, true);
3055                 if (ret < 0) {
3056                         tcpm_set_state(port, SNK_READY, 0);
3057                         break;
3058                 }
3059                 /*
3060                  * If VBUS has never been low, and we time out waiting
3061                  * for source cap, try a soft reset first, in case we
3062                  * were already in a stable contract before this boot.
3063                  * Do this only once.
3064                  */
3065                 if (port->vbus_never_low) {
3066                         port->vbus_never_low = false;
3067                         tcpm_set_state(port, SOFT_RESET_SEND,
3068                                        PD_T_SINK_WAIT_CAP);
3069                 } else {
3070                         tcpm_set_state(port, hard_reset_state(port),
3071                                        PD_T_SINK_WAIT_CAP);
3072                 }
3073                 break;
3074         case SNK_NEGOTIATE_CAPABILITIES:
3075                 port->pd_capable = true;
3076                 port->hard_reset_count = 0;
3077                 ret = tcpm_pd_send_request(port);
3078                 if (ret < 0) {
3079                         /* Let the Source send capabilities again. */
3080                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3081                 } else {
3082                         tcpm_set_state_cond(port, hard_reset_state(port),
3083                                             PD_T_SENDER_RESPONSE);
3084                 }
3085                 break;
3086         case SNK_NEGOTIATE_PPS_CAPABILITIES:
3087                 ret = tcpm_pd_send_pps_request(port);
3088                 if (ret < 0) {
3089                         port->pps_status = ret;
3090                         /*
3091                          * If this was called due to updates to sink
3092                          * capabilities, and pps is no longer valid, we should
3093                          * safely fall back to a standard PDO.
3094                          */
3095                         if (port->update_sink_caps)
3096                                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
3097                         else
3098                                 tcpm_set_state(port, SNK_READY, 0);
3099                 } else {
3100                         tcpm_set_state_cond(port, hard_reset_state(port),
3101                                             PD_T_SENDER_RESPONSE);
3102                 }
3103                 break;
3104         case SNK_TRANSITION_SINK:
3105         case SNK_TRANSITION_SINK_VBUS:
3106                 tcpm_set_state(port, hard_reset_state(port),
3107                                PD_T_PS_TRANSITION);
3108                 break;
3109         case SNK_READY:
3110                 port->try_snk_count = 0;
3111                 port->update_sink_caps = false;
3112                 if (port->explicit_contract) {
3113                         typec_set_pwr_opmode(port->typec_port,
3114                                              TYPEC_PWR_MODE_PD);
3115                         port->pwr_opmode = TYPEC_PWR_MODE_PD;
3116                 }
3117
3118                 tcpm_swap_complete(port, 0);
3119                 tcpm_typec_connect(port);
3120                 tcpm_check_send_discover(port);
3121                 tcpm_pps_complete(port, port->pps_status);
3122
3123                 power_supply_changed(port->psy);
3124
3125                 break;
3126
3127         /* Accessory states */
3128         case ACC_UNATTACHED:
3129                 tcpm_acc_detach(port);
3130                 tcpm_set_state(port, SRC_UNATTACHED, 0);
3131                 break;
3132         case DEBUG_ACC_ATTACHED:
3133         case AUDIO_ACC_ATTACHED:
3134                 ret = tcpm_acc_attach(port);
3135                 if (ret < 0)
3136                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3137                 break;
3138         case AUDIO_ACC_DEBOUNCE:
3139                 tcpm_set_state(port, ACC_UNATTACHED, PD_T_CC_DEBOUNCE);
3140                 break;
3141
3142         /* Hard_Reset states */
3143         case HARD_RESET_SEND:
3144                 tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
3145                 tcpm_set_state(port, HARD_RESET_START, 0);
3146                 break;
3147         case HARD_RESET_START:
3148                 port->hard_reset_count++;
3149                 port->tcpc->set_pd_rx(port->tcpc, false);
3150                 tcpm_unregister_altmodes(port);
3151                 port->send_discover = true;
3152                 if (port->pwr_role == TYPEC_SOURCE)
3153                         tcpm_set_state(port, SRC_HARD_RESET_VBUS_OFF,
3154                                        PD_T_PS_HARD_RESET);
3155                 else
3156                         tcpm_set_state(port, SNK_HARD_RESET_SINK_OFF, 0);
3157                 break;
3158         case SRC_HARD_RESET_VBUS_OFF:
3159                 tcpm_set_vconn(port, true);
3160                 tcpm_set_vbus(port, false);
3161                 tcpm_set_roles(port, false, TYPEC_SOURCE, TYPEC_HOST);
3162                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SRC_RECOVER);
3163                 break;
3164         case SRC_HARD_RESET_VBUS_ON:
3165                 tcpm_set_vbus(port, true);
3166                 port->tcpc->set_pd_rx(port->tcpc, true);
3167                 tcpm_set_attached_state(port, true);
3168                 tcpm_set_state(port, SRC_UNATTACHED, PD_T_PS_SOURCE_ON);
3169                 break;
3170         case SNK_HARD_RESET_SINK_OFF:
3171                 memset(&port->pps_data, 0, sizeof(port->pps_data));
3172                 tcpm_set_vconn(port, false);
3173                 tcpm_set_charge(port, false);
3174                 tcpm_set_roles(port, false, TYPEC_SINK, TYPEC_DEVICE);
3175                 /*
3176                  * VBUS may or may not toggle, depending on the adapter.
3177                  * If it doesn't toggle, transition to SNK_HARD_RESET_SINK_ON
3178                  * directly after timeout.
3179                  */
3180                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, PD_T_SAFE_0V);
3181                 break;
3182         case SNK_HARD_RESET_WAIT_VBUS:
3183                 /* Assume we're disconnected if VBUS doesn't come back. */
3184                 tcpm_set_state(port, SNK_UNATTACHED,
3185                                PD_T_SRC_RECOVER_MAX + PD_T_SRC_TURN_ON);
3186                 break;
3187         case SNK_HARD_RESET_SINK_ON:
3188                 /* Note: There is no guarantee that VBUS is on in this state */
3189                 /*
3190                  * XXX:
3191                  * The specification suggests that dual mode ports in sink
3192                  * mode should transition to state PE_SRC_Transition_to_default.
3193                  * See USB power delivery specification chapter 8.3.3.6.1.3.
3194                  * This would mean to to
3195                  * - turn off VCONN, reset power supply
3196                  * - request hardware reset
3197                  * - turn on VCONN
3198                  * - Transition to state PE_Src_Startup
3199                  * SNK only ports shall transition to state Snk_Startup
3200                  * (see chapter 8.3.3.3.8).
3201                  * Similar, dual-mode ports in source mode should transition
3202                  * to PE_SNK_Transition_to_default.
3203                  */
3204                 tcpm_set_attached_state(port, true);
3205                 tcpm_set_state(port, SNK_STARTUP, 0);
3206                 break;
3207
3208         /* Soft_Reset states */
3209         case SOFT_RESET:
3210                 port->message_id = 0;
3211                 port->rx_msgid = -1;
3212                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3213                 if (port->pwr_role == TYPEC_SOURCE)
3214                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3215                 else
3216                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
3217                 break;
3218         case SOFT_RESET_SEND:
3219                 port->message_id = 0;
3220                 port->rx_msgid = -1;
3221                 if (tcpm_pd_send_control(port, PD_CTRL_SOFT_RESET))
3222                         tcpm_set_state_cond(port, hard_reset_state(port), 0);
3223                 else
3224                         tcpm_set_state_cond(port, hard_reset_state(port),
3225                                             PD_T_SENDER_RESPONSE);
3226                 break;
3227
3228         /* DR_Swap states */
3229         case DR_SWAP_SEND:
3230                 tcpm_pd_send_control(port, PD_CTRL_DR_SWAP);
3231                 tcpm_set_state_cond(port, DR_SWAP_SEND_TIMEOUT,
3232                                     PD_T_SENDER_RESPONSE);
3233                 break;
3234         case DR_SWAP_ACCEPT:
3235                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3236                 tcpm_set_state_cond(port, DR_SWAP_CHANGE_DR, 0);
3237                 break;
3238         case DR_SWAP_SEND_TIMEOUT:
3239                 tcpm_swap_complete(port, -ETIMEDOUT);
3240                 tcpm_set_state(port, ready_state(port), 0);
3241                 break;
3242         case DR_SWAP_CHANGE_DR:
3243                 if (port->data_role == TYPEC_HOST) {
3244                         tcpm_unregister_altmodes(port);
3245                         tcpm_set_roles(port, true, port->pwr_role,
3246                                        TYPEC_DEVICE);
3247                 } else {
3248                         tcpm_set_roles(port, true, port->pwr_role,
3249                                        TYPEC_HOST);
3250                         port->send_discover = true;
3251                 }
3252                 tcpm_set_state(port, ready_state(port), 0);
3253                 break;
3254
3255         /* PR_Swap states */
3256         case PR_SWAP_ACCEPT:
3257                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3258                 tcpm_set_state(port, PR_SWAP_START, 0);
3259                 break;
3260         case PR_SWAP_SEND:
3261                 tcpm_pd_send_control(port, PD_CTRL_PR_SWAP);
3262                 tcpm_set_state_cond(port, PR_SWAP_SEND_TIMEOUT,
3263                                     PD_T_SENDER_RESPONSE);
3264                 break;
3265         case PR_SWAP_SEND_TIMEOUT:
3266                 tcpm_swap_complete(port, -ETIMEDOUT);
3267                 tcpm_set_state(port, ready_state(port), 0);
3268                 break;
3269         case PR_SWAP_START:
3270                 if (port->pwr_role == TYPEC_SOURCE)
3271                         tcpm_set_state(port, PR_SWAP_SRC_SNK_TRANSITION_OFF,
3272                                        PD_T_SRC_TRANSITION);
3273                 else
3274                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SINK_OFF, 0);
3275                 break;
3276         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3277                 tcpm_set_vbus(port, false);
3278                 port->explicit_contract = false;
3279                 /* allow time for Vbus discharge, must be < tSrcSwapStdby */
3280                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF,
3281                                PD_T_SRCSWAPSTDBY);
3282                 break;
3283         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3284                 tcpm_set_cc(port, TYPEC_CC_RD);
3285                 /* allow CC debounce */
3286                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED,
3287                                PD_T_CC_DEBOUNCE);
3288                 break;
3289         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3290                 /*
3291                  * USB-PD standard, 6.2.1.4, Port Power Role:
3292                  * "During the Power Role Swap Sequence, for the initial Source
3293                  * Port, the Port Power Role field shall be set to Sink in the
3294                  * PS_RDY Message indicating that the initial Source’s power
3295                  * supply is turned off"
3296                  */
3297                 tcpm_set_pwr_role(port, TYPEC_SINK);
3298                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
3299                         tcpm_set_state(port, ERROR_RECOVERY, 0);
3300                         break;
3301                 }
3302                 tcpm_set_state_cond(port, SNK_UNATTACHED, PD_T_PS_SOURCE_ON);
3303                 break;
3304         case PR_SWAP_SRC_SNK_SINK_ON:
3305                 tcpm_set_state(port, SNK_STARTUP, 0);
3306                 break;
3307         case PR_SWAP_SNK_SRC_SINK_OFF:
3308                 tcpm_set_charge(port, false);
3309                 tcpm_set_state(port, hard_reset_state(port),
3310                                PD_T_PS_SOURCE_OFF);
3311                 break;
3312         case PR_SWAP_SNK_SRC_SOURCE_ON:
3313                 tcpm_set_cc(port, tcpm_rp_cc(port));
3314                 tcpm_set_vbus(port, true);
3315                 /*
3316                  * allow time VBUS ramp-up, must be < tNewSrc
3317                  * Also, this window overlaps with CC debounce as well.
3318                  * So, Wait for the max of two which is PD_T_NEWSRC
3319                  */
3320                 tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP,
3321                                PD_T_NEWSRC);
3322                 break;
3323         case PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP:
3324                 /*
3325                  * USB PD standard, 6.2.1.4:
3326                  * "Subsequent Messages initiated by the Policy Engine,
3327                  * such as the PS_RDY Message sent to indicate that Vbus
3328                  * is ready, will have the Port Power Role field set to
3329                  * Source."
3330                  */
3331                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
3332                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3333                 tcpm_set_state(port, SRC_STARTUP, 0);
3334                 break;
3335
3336         case VCONN_SWAP_ACCEPT:
3337                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
3338                 tcpm_set_state(port, VCONN_SWAP_START, 0);
3339                 break;
3340         case VCONN_SWAP_SEND:
3341                 tcpm_pd_send_control(port, PD_CTRL_VCONN_SWAP);
3342                 tcpm_set_state(port, VCONN_SWAP_SEND_TIMEOUT,
3343                                PD_T_SENDER_RESPONSE);
3344                 break;
3345         case VCONN_SWAP_SEND_TIMEOUT:
3346                 tcpm_swap_complete(port, -ETIMEDOUT);
3347                 tcpm_set_state(port, ready_state(port), 0);
3348                 break;
3349         case VCONN_SWAP_START:
3350                 if (port->vconn_role == TYPEC_SOURCE)
3351                         tcpm_set_state(port, VCONN_SWAP_WAIT_FOR_VCONN, 0);
3352                 else
3353                         tcpm_set_state(port, VCONN_SWAP_TURN_ON_VCONN, 0);
3354                 break;
3355         case VCONN_SWAP_WAIT_FOR_VCONN:
3356                 tcpm_set_state(port, hard_reset_state(port),
3357                                PD_T_VCONN_SOURCE_ON);
3358                 break;
3359         case VCONN_SWAP_TURN_ON_VCONN:
3360                 tcpm_set_vconn(port, true);
3361                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
3362                 tcpm_set_state(port, ready_state(port), 0);
3363                 break;
3364         case VCONN_SWAP_TURN_OFF_VCONN:
3365                 tcpm_set_vconn(port, false);
3366                 tcpm_set_state(port, ready_state(port), 0);
3367                 break;
3368
3369         case DR_SWAP_CANCEL:
3370         case PR_SWAP_CANCEL:
3371         case VCONN_SWAP_CANCEL:
3372                 tcpm_swap_complete(port, port->swap_status);
3373                 if (port->pwr_role == TYPEC_SOURCE)
3374                         tcpm_set_state(port, SRC_READY, 0);
3375                 else
3376                         tcpm_set_state(port, SNK_READY, 0);
3377                 break;
3378
3379         case BIST_RX:
3380                 switch (BDO_MODE_MASK(port->bist_request)) {
3381                 case BDO_MODE_CARRIER2:
3382                         tcpm_pd_transmit(port, TCPC_TX_BIST_MODE_2, NULL);
3383                         break;
3384                 default:
3385                         break;
3386                 }
3387                 /* Always switch to unattached state */
3388                 tcpm_set_state(port, unattached_state(port), 0);
3389                 break;
3390         case GET_STATUS_SEND:
3391                 tcpm_pd_send_control(port, PD_CTRL_GET_STATUS);
3392                 tcpm_set_state(port, GET_STATUS_SEND_TIMEOUT,
3393                                PD_T_SENDER_RESPONSE);
3394                 break;
3395         case GET_STATUS_SEND_TIMEOUT:
3396                 tcpm_set_state(port, ready_state(port), 0);
3397                 break;
3398         case GET_PPS_STATUS_SEND:
3399                 tcpm_pd_send_control(port, PD_CTRL_GET_PPS_STATUS);
3400                 tcpm_set_state(port, GET_PPS_STATUS_SEND_TIMEOUT,
3401                                PD_T_SENDER_RESPONSE);
3402                 break;
3403         case GET_PPS_STATUS_SEND_TIMEOUT:
3404                 tcpm_set_state(port, ready_state(port), 0);
3405                 break;
3406         case ERROR_RECOVERY:
3407                 tcpm_swap_complete(port, -EPROTO);
3408                 tcpm_pps_complete(port, -EPROTO);
3409                 tcpm_set_state(port, PORT_RESET, 0);
3410                 break;
3411         case PORT_RESET:
3412                 tcpm_reset_port(port);
3413                 tcpm_set_cc(port, TYPEC_CC_OPEN);
3414                 tcpm_set_state(port, PORT_RESET_WAIT_OFF,
3415                                PD_T_ERROR_RECOVERY);
3416                 break;
3417         case PORT_RESET_WAIT_OFF:
3418                 tcpm_set_state(port,
3419                                tcpm_default_state(port),
3420                                port->vbus_present ? PD_T_PS_SOURCE_OFF : 0);
3421                 break;
3422         default:
3423                 WARN(1, "Unexpected port state %d\n", port->state);
3424                 break;
3425         }
3426 }
3427
3428 static void tcpm_state_machine_work(struct work_struct *work)
3429 {
3430         struct tcpm_port *port = container_of(work, struct tcpm_port,
3431                                               state_machine.work);
3432         enum tcpm_state prev_state;
3433
3434         mutex_lock(&port->lock);
3435         port->state_machine_running = true;
3436
3437         if (port->queued_message && tcpm_send_queued_message(port))
3438                 goto done;
3439
3440         /* If we were queued due to a delayed state change, update it now */
3441         if (port->delayed_state) {
3442                 tcpm_log(port, "state change %s -> %s [delayed %ld ms]",
3443                          tcpm_states[port->state],
3444                          tcpm_states[port->delayed_state], port->delay_ms);
3445                 port->prev_state = port->state;
3446                 port->state = port->delayed_state;
3447                 port->delayed_state = INVALID_STATE;
3448         }
3449
3450         /*
3451          * Continue running as long as we have (non-delayed) state changes
3452          * to make.
3453          */
3454         do {
3455                 prev_state = port->state;
3456                 run_state_machine(port);
3457                 if (port->queued_message)
3458                         tcpm_send_queued_message(port);
3459         } while (port->state != prev_state && !port->delayed_state);
3460
3461 done:
3462         port->state_machine_running = false;
3463         mutex_unlock(&port->lock);
3464 }
3465
3466 static void _tcpm_cc_change(struct tcpm_port *port, enum typec_cc_status cc1,
3467                             enum typec_cc_status cc2)
3468 {
3469         enum typec_cc_status old_cc1, old_cc2;
3470         enum tcpm_state new_state;
3471
3472         old_cc1 = port->cc1;
3473         old_cc2 = port->cc2;
3474         port->cc1 = cc1;
3475         port->cc2 = cc2;
3476
3477         tcpm_log_force(port,
3478                        "CC1: %u -> %u, CC2: %u -> %u [state %s, polarity %d, %s]",
3479                        old_cc1, cc1, old_cc2, cc2, tcpm_states[port->state],
3480                        port->polarity,
3481                        tcpm_port_is_disconnected(port) ? "disconnected"
3482                                                        : "connected");
3483
3484         switch (port->state) {
3485         case DRP_TOGGLING:
3486                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3487                     tcpm_port_is_source(port))
3488                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3489                 else if (tcpm_port_is_sink(port))
3490                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3491                 break;
3492         case SRC_UNATTACHED:
3493         case ACC_UNATTACHED:
3494                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
3495                     tcpm_port_is_source(port))
3496                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3497                 break;
3498         case SRC_ATTACH_WAIT:
3499                 if (tcpm_port_is_disconnected(port) ||
3500                     tcpm_port_is_audio_detached(port))
3501                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3502                 else if (cc1 != old_cc1 || cc2 != old_cc2)
3503                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
3504                 break;
3505         case SRC_ATTACHED:
3506         case SRC_SEND_CAPABILITIES:
3507         case SRC_READY:
3508                 if (tcpm_port_is_disconnected(port) ||
3509                     !tcpm_port_is_source(port))
3510                         tcpm_set_state(port, SRC_UNATTACHED, 0);
3511                 break;
3512         case SNK_UNATTACHED:
3513                 if (tcpm_port_is_sink(port))
3514                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3515                 break;
3516         case SNK_ATTACH_WAIT:
3517                 if ((port->cc1 == TYPEC_CC_OPEN &&
3518                      port->cc2 != TYPEC_CC_OPEN) ||
3519                     (port->cc1 != TYPEC_CC_OPEN &&
3520                      port->cc2 == TYPEC_CC_OPEN))
3521                         new_state = SNK_DEBOUNCED;
3522                 else if (tcpm_port_is_disconnected(port))
3523                         new_state = SNK_UNATTACHED;
3524                 else
3525                         break;
3526                 if (new_state != port->delayed_state)
3527                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
3528                 break;
3529         case SNK_DEBOUNCED:
3530                 if (tcpm_port_is_disconnected(port))
3531                         new_state = SNK_UNATTACHED;
3532                 else if (port->vbus_present)
3533                         new_state = tcpm_try_src(port) ? SRC_TRY : SNK_ATTACHED;
3534                 else
3535                         new_state = SNK_UNATTACHED;
3536                 if (new_state != port->delayed_state)
3537                         tcpm_set_state(port, SNK_DEBOUNCED, 0);
3538                 break;
3539         case SNK_READY:
3540                 if (tcpm_port_is_disconnected(port))
3541                         tcpm_set_state(port, unattached_state(port), 0);
3542                 else if (!port->pd_capable &&
3543                          (cc1 != old_cc1 || cc2 != old_cc2))
3544                         tcpm_set_current_limit(port,
3545                                                tcpm_get_current_limit(port),
3546                                                5000);
3547                 break;
3548
3549         case AUDIO_ACC_ATTACHED:
3550                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3551                         tcpm_set_state(port, AUDIO_ACC_DEBOUNCE, 0);
3552                 break;
3553         case AUDIO_ACC_DEBOUNCE:
3554                 if (tcpm_port_is_audio(port))
3555                         tcpm_set_state(port, AUDIO_ACC_ATTACHED, 0);
3556                 break;
3557
3558         case DEBUG_ACC_ATTACHED:
3559                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
3560                         tcpm_set_state(port, ACC_UNATTACHED, 0);
3561                 break;
3562
3563         case SNK_TRY:
3564                 /* Do nothing, waiting for timeout */
3565                 break;
3566
3567         case SNK_DISCOVERY:
3568                 /* CC line is unstable, wait for debounce */
3569                 if (tcpm_port_is_disconnected(port))
3570                         tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE, 0);
3571                 break;
3572         case SNK_DISCOVERY_DEBOUNCE:
3573                 break;
3574
3575         case SRC_TRYWAIT:
3576                 /* Hand over to state machine if needed */
3577                 if (!port->vbus_present && tcpm_port_is_source(port))
3578                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3579                 break;
3580         case SRC_TRYWAIT_DEBOUNCE:
3581                 if (port->vbus_present || !tcpm_port_is_source(port))
3582                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3583                 break;
3584         case SNK_TRY_WAIT_DEBOUNCE:
3585                 if (!tcpm_port_is_sink(port)) {
3586                         port->max_wait = 0;
3587                         tcpm_set_state(port, SRC_TRYWAIT, 0);
3588                 }
3589                 break;
3590         case SRC_TRY_WAIT:
3591                 if (tcpm_port_is_source(port))
3592                         tcpm_set_state(port, SRC_TRY_DEBOUNCE, 0);
3593                 break;
3594         case SRC_TRY_DEBOUNCE:
3595                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
3596                 break;
3597         case SNK_TRYWAIT_DEBOUNCE:
3598                 if (tcpm_port_is_sink(port))
3599                         tcpm_set_state(port, SNK_TRYWAIT_VBUS, 0);
3600                 break;
3601         case SNK_TRYWAIT_VBUS:
3602                 if (!tcpm_port_is_sink(port))
3603                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
3604                 break;
3605         case SNK_TRYWAIT:
3606                 /* Do nothing, waiting for tCCDebounce */
3607                 break;
3608         case PR_SWAP_SNK_SRC_SINK_OFF:
3609         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3610         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3611         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3612         case PR_SWAP_SNK_SRC_SOURCE_ON:
3613                 /*
3614                  * CC state change is expected in PR_SWAP
3615                  * Ignore it.
3616                  */
3617                 break;
3618
3619         default:
3620                 if (tcpm_port_is_disconnected(port))
3621                         tcpm_set_state(port, unattached_state(port), 0);
3622                 break;
3623         }
3624 }
3625
3626 static void _tcpm_pd_vbus_on(struct tcpm_port *port)
3627 {
3628         tcpm_log_force(port, "VBUS on");
3629         port->vbus_present = true;
3630         switch (port->state) {
3631         case SNK_TRANSITION_SINK_VBUS:
3632                 port->explicit_contract = true;
3633                 tcpm_set_state(port, SNK_READY, 0);
3634                 break;
3635         case SNK_DISCOVERY:
3636                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3637                 break;
3638
3639         case SNK_DEBOUNCED:
3640                 tcpm_set_state(port, tcpm_try_src(port) ? SRC_TRY
3641                                                         : SNK_ATTACHED,
3642                                        0);
3643                 break;
3644         case SNK_HARD_RESET_WAIT_VBUS:
3645                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, 0);
3646                 break;
3647         case SRC_ATTACHED:
3648                 tcpm_set_state(port, SRC_STARTUP, 0);
3649                 break;
3650         case SRC_HARD_RESET_VBUS_ON:
3651                 tcpm_set_state(port, SRC_STARTUP, 0);
3652                 break;
3653
3654         case SNK_TRY:
3655                 /* Do nothing, waiting for timeout */
3656                 break;
3657         case SRC_TRYWAIT:
3658                 /* Do nothing, Waiting for Rd to be detected */
3659                 break;
3660         case SRC_TRYWAIT_DEBOUNCE:
3661                 tcpm_set_state(port, SRC_TRYWAIT, 0);
3662                 break;
3663         case SNK_TRY_WAIT_DEBOUNCE:
3664                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3665                 break;
3666         case SNK_TRYWAIT:
3667                 /* Do nothing, waiting for tCCDebounce */
3668                 break;
3669         case SNK_TRYWAIT_VBUS:
3670                 if (tcpm_port_is_sink(port))
3671                         tcpm_set_state(port, SNK_ATTACHED, 0);
3672                 break;
3673         case SNK_TRYWAIT_DEBOUNCE:
3674                 /* Do nothing, waiting for Rp */
3675                 break;
3676         case SRC_TRY_WAIT:
3677         case SRC_TRY_DEBOUNCE:
3678                 /* Do nothing, waiting for sink detection */
3679                 break;
3680         default:
3681                 break;
3682         }
3683 }
3684
3685 static void _tcpm_pd_vbus_off(struct tcpm_port *port)
3686 {
3687         tcpm_log_force(port, "VBUS off");
3688         port->vbus_present = false;
3689         port->vbus_never_low = false;
3690         switch (port->state) {
3691         case SNK_HARD_RESET_SINK_OFF:
3692                 tcpm_set_state(port, SNK_HARD_RESET_WAIT_VBUS, 0);
3693                 break;
3694         case SRC_HARD_RESET_VBUS_OFF:
3695                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, 0);
3696                 break;
3697         case HARD_RESET_SEND:
3698                 break;
3699
3700         case SNK_TRY:
3701                 /* Do nothing, waiting for timeout */
3702                 break;
3703         case SRC_TRYWAIT:
3704                 /* Hand over to state machine if needed */
3705                 if (tcpm_port_is_source(port))
3706                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3707                 break;
3708         case SNK_TRY_WAIT_DEBOUNCE:
3709                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3710                 break;
3711         case SNK_TRYWAIT:
3712         case SNK_TRYWAIT_VBUS:
3713         case SNK_TRYWAIT_DEBOUNCE:
3714                 break;
3715         case SNK_ATTACH_WAIT:
3716                 tcpm_set_state(port, SNK_UNATTACHED, 0);
3717                 break;
3718
3719         case SNK_NEGOTIATE_CAPABILITIES:
3720                 break;
3721
3722         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3723                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF, 0);
3724                 break;
3725
3726         case PR_SWAP_SNK_SRC_SINK_OFF:
3727                 /* Do nothing, expected */
3728                 break;
3729
3730         case PORT_RESET_WAIT_OFF:
3731                 tcpm_set_state(port, tcpm_default_state(port), 0);
3732                 break;
3733         case SRC_TRY_WAIT:
3734         case SRC_TRY_DEBOUNCE:
3735                 /* Do nothing, waiting for sink detection */
3736                 break;
3737         default:
3738                 if (port->pwr_role == TYPEC_SINK &&
3739                     port->attached)
3740                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3741                 break;
3742         }
3743 }
3744
3745 static void _tcpm_pd_hard_reset(struct tcpm_port *port)
3746 {
3747         tcpm_log_force(port, "Received hard reset");
3748         /*
3749          * If we keep receiving hard reset requests, executing the hard reset
3750          * must have failed. Revert to error recovery if that happens.
3751          */
3752         tcpm_set_state(port,
3753                        port->hard_reset_count < PD_N_HARD_RESET_COUNT ?
3754                                 HARD_RESET_START : ERROR_RECOVERY,
3755                        0);
3756 }
3757
3758 static void tcpm_pd_event_handler(struct work_struct *work)
3759 {
3760         struct tcpm_port *port = container_of(work, struct tcpm_port,
3761                                               event_work);
3762         u32 events;
3763
3764         mutex_lock(&port->lock);
3765
3766         spin_lock(&port->pd_event_lock);
3767         while (port->pd_events) {
3768                 events = port->pd_events;
3769                 port->pd_events = 0;
3770                 spin_unlock(&port->pd_event_lock);
3771                 if (events & TCPM_RESET_EVENT)
3772                         _tcpm_pd_hard_reset(port);
3773                 if (events & TCPM_VBUS_EVENT) {
3774                         bool vbus;
3775
3776                         vbus = port->tcpc->get_vbus(port->tcpc);
3777                         if (vbus)
3778                                 _tcpm_pd_vbus_on(port);
3779                         else
3780                                 _tcpm_pd_vbus_off(port);
3781                 }
3782                 if (events & TCPM_CC_EVENT) {
3783                         enum typec_cc_status cc1, cc2;
3784
3785                         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
3786                                 _tcpm_cc_change(port, cc1, cc2);
3787                 }
3788                 spin_lock(&port->pd_event_lock);
3789         }
3790         spin_unlock(&port->pd_event_lock);
3791         mutex_unlock(&port->lock);
3792 }
3793
3794 void tcpm_cc_change(struct tcpm_port *port)
3795 {
3796         spin_lock(&port->pd_event_lock);
3797         port->pd_events |= TCPM_CC_EVENT;
3798         spin_unlock(&port->pd_event_lock);
3799         queue_work(port->wq, &port->event_work);
3800 }
3801 EXPORT_SYMBOL_GPL(tcpm_cc_change);
3802
3803 void tcpm_vbus_change(struct tcpm_port *port)
3804 {
3805         spin_lock(&port->pd_event_lock);
3806         port->pd_events |= TCPM_VBUS_EVENT;
3807         spin_unlock(&port->pd_event_lock);
3808         queue_work(port->wq, &port->event_work);
3809 }
3810 EXPORT_SYMBOL_GPL(tcpm_vbus_change);
3811
3812 void tcpm_pd_hard_reset(struct tcpm_port *port)
3813 {
3814         spin_lock(&port->pd_event_lock);
3815         port->pd_events = TCPM_RESET_EVENT;
3816         spin_unlock(&port->pd_event_lock);
3817         queue_work(port->wq, &port->event_work);
3818 }
3819 EXPORT_SYMBOL_GPL(tcpm_pd_hard_reset);
3820
3821 static int tcpm_dr_set(const struct typec_capability *cap,
3822                        enum typec_data_role data)
3823 {
3824         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3825         int ret;
3826
3827         mutex_lock(&port->swap_lock);
3828         mutex_lock(&port->lock);
3829
3830         if (port->port_type != TYPEC_PORT_DRP) {
3831                 ret = -EINVAL;
3832                 goto port_unlock;
3833         }
3834         if (port->state != SRC_READY && port->state != SNK_READY) {
3835                 ret = -EAGAIN;
3836                 goto port_unlock;
3837         }
3838
3839         if (port->data_role == data) {
3840                 ret = 0;
3841                 goto port_unlock;
3842         }
3843
3844         /*
3845          * XXX
3846          * 6.3.9: If an alternate mode is active, a request to swap
3847          * alternate modes shall trigger a port reset.
3848          * Reject data role swap request in this case.
3849          */
3850
3851         if (!port->pd_capable) {
3852                 /*
3853                  * If the partner is not PD capable, reset the port to
3854                  * trigger a role change. This can only work if a preferred
3855                  * role is configured, and if it matches the requested role.
3856                  */
3857                 if (port->try_role == TYPEC_NO_PREFERRED_ROLE ||
3858                     port->try_role == port->pwr_role) {
3859                         ret = -EINVAL;
3860                         goto port_unlock;
3861                 }
3862                 port->non_pd_role_swap = true;
3863                 tcpm_set_state(port, PORT_RESET, 0);
3864         } else {
3865                 tcpm_set_state(port, DR_SWAP_SEND, 0);
3866         }
3867
3868         port->swap_status = 0;
3869         port->swap_pending = true;
3870         reinit_completion(&port->swap_complete);
3871         mutex_unlock(&port->lock);
3872
3873         if (!wait_for_completion_timeout(&port->swap_complete,
3874                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3875                 ret = -ETIMEDOUT;
3876         else
3877                 ret = port->swap_status;
3878
3879         port->non_pd_role_swap = false;
3880         goto swap_unlock;
3881
3882 port_unlock:
3883         mutex_unlock(&port->lock);
3884 swap_unlock:
3885         mutex_unlock(&port->swap_lock);
3886         return ret;
3887 }
3888
3889 static int tcpm_pr_set(const struct typec_capability *cap,
3890                        enum typec_role role)
3891 {
3892         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3893         int ret;
3894
3895         mutex_lock(&port->swap_lock);
3896         mutex_lock(&port->lock);
3897
3898         if (port->port_type != TYPEC_PORT_DRP) {
3899                 ret = -EINVAL;
3900                 goto port_unlock;
3901         }
3902         if (port->state != SRC_READY && port->state != SNK_READY) {
3903                 ret = -EAGAIN;
3904                 goto port_unlock;
3905         }
3906
3907         if (role == port->pwr_role) {
3908                 ret = 0;
3909                 goto port_unlock;
3910         }
3911
3912         port->swap_status = 0;
3913         port->swap_pending = true;
3914         reinit_completion(&port->swap_complete);
3915         tcpm_set_state(port, PR_SWAP_SEND, 0);
3916         mutex_unlock(&port->lock);
3917
3918         if (!wait_for_completion_timeout(&port->swap_complete,
3919                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3920                 ret = -ETIMEDOUT;
3921         else
3922                 ret = port->swap_status;
3923
3924         goto swap_unlock;
3925
3926 port_unlock:
3927         mutex_unlock(&port->lock);
3928 swap_unlock:
3929         mutex_unlock(&port->swap_lock);
3930         return ret;
3931 }
3932
3933 static int tcpm_vconn_set(const struct typec_capability *cap,
3934                           enum typec_role role)
3935 {
3936         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3937         int ret;
3938
3939         mutex_lock(&port->swap_lock);
3940         mutex_lock(&port->lock);
3941
3942         if (port->state != SRC_READY && port->state != SNK_READY) {
3943                 ret = -EAGAIN;
3944                 goto port_unlock;
3945         }
3946
3947         if (role == port->vconn_role) {
3948                 ret = 0;
3949                 goto port_unlock;
3950         }
3951
3952         port->swap_status = 0;
3953         port->swap_pending = true;
3954         reinit_completion(&port->swap_complete);
3955         tcpm_set_state(port, VCONN_SWAP_SEND, 0);
3956         mutex_unlock(&port->lock);
3957
3958         if (!wait_for_completion_timeout(&port->swap_complete,
3959                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3960                 ret = -ETIMEDOUT;
3961         else
3962                 ret = port->swap_status;
3963
3964         goto swap_unlock;
3965
3966 port_unlock:
3967         mutex_unlock(&port->lock);
3968 swap_unlock:
3969         mutex_unlock(&port->swap_lock);
3970         return ret;
3971 }
3972
3973 static int tcpm_try_role(const struct typec_capability *cap, int role)
3974 {
3975         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3976         struct tcpc_dev *tcpc = port->tcpc;
3977         int ret = 0;
3978
3979         mutex_lock(&port->lock);
3980         if (tcpc->try_role)
3981                 ret = tcpc->try_role(tcpc, role);
3982         if (!ret && !tcpc->config->try_role_hw)
3983                 port->try_role = role;
3984         port->try_src_count = 0;
3985         port->try_snk_count = 0;
3986         mutex_unlock(&port->lock);
3987
3988         return ret;
3989 }
3990
3991 static int tcpm_pps_set_op_curr(struct tcpm_port *port, u16 op_curr)
3992 {
3993         unsigned int target_mw;
3994         int ret;
3995
3996         mutex_lock(&port->swap_lock);
3997         mutex_lock(&port->lock);
3998
3999         if (!port->pps_data.active) {
4000                 ret = -EOPNOTSUPP;
4001                 goto port_unlock;
4002         }
4003
4004         if (port->state != SNK_READY) {
4005                 ret = -EAGAIN;
4006                 goto port_unlock;
4007         }
4008
4009         if (op_curr > port->pps_data.max_curr) {
4010                 ret = -EINVAL;
4011                 goto port_unlock;
4012         }
4013
4014         target_mw = (op_curr * port->pps_data.out_volt) / 1000;
4015         if (target_mw < port->operating_snk_mw) {
4016                 ret = -EINVAL;
4017                 goto port_unlock;
4018         }
4019
4020         reinit_completion(&port->pps_complete);
4021         port->pps_data.op_curr = op_curr;
4022         port->pps_status = 0;
4023         port->pps_pending = true;
4024         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4025         mutex_unlock(&port->lock);
4026
4027         if (!wait_for_completion_timeout(&port->pps_complete,
4028                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4029                 ret = -ETIMEDOUT;
4030         else
4031                 ret = port->pps_status;
4032
4033         goto swap_unlock;
4034
4035 port_unlock:
4036         mutex_unlock(&port->lock);
4037 swap_unlock:
4038         mutex_unlock(&port->swap_lock);
4039
4040         return ret;
4041 }
4042
4043 static int tcpm_pps_set_out_volt(struct tcpm_port *port, u16 out_volt)
4044 {
4045         unsigned int target_mw;
4046         int ret;
4047
4048         mutex_lock(&port->swap_lock);
4049         mutex_lock(&port->lock);
4050
4051         if (!port->pps_data.active) {
4052                 ret = -EOPNOTSUPP;
4053                 goto port_unlock;
4054         }
4055
4056         if (port->state != SNK_READY) {
4057                 ret = -EAGAIN;
4058                 goto port_unlock;
4059         }
4060
4061         if (out_volt < port->pps_data.min_volt ||
4062             out_volt > port->pps_data.max_volt) {
4063                 ret = -EINVAL;
4064                 goto port_unlock;
4065         }
4066
4067         target_mw = (port->pps_data.op_curr * out_volt) / 1000;
4068         if (target_mw < port->operating_snk_mw) {
4069                 ret = -EINVAL;
4070                 goto port_unlock;
4071         }
4072
4073         reinit_completion(&port->pps_complete);
4074         port->pps_data.out_volt = out_volt;
4075         port->pps_status = 0;
4076         port->pps_pending = true;
4077         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4078         mutex_unlock(&port->lock);
4079
4080         if (!wait_for_completion_timeout(&port->pps_complete,
4081                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4082                 ret = -ETIMEDOUT;
4083         else
4084                 ret = port->pps_status;
4085
4086         goto swap_unlock;
4087
4088 port_unlock:
4089         mutex_unlock(&port->lock);
4090 swap_unlock:
4091         mutex_unlock(&port->swap_lock);
4092
4093         return ret;
4094 }
4095
4096 static int tcpm_pps_activate(struct tcpm_port *port, bool activate)
4097 {
4098         int ret = 0;
4099
4100         mutex_lock(&port->swap_lock);
4101         mutex_lock(&port->lock);
4102
4103         if (!port->pps_data.supported) {
4104                 ret = -EOPNOTSUPP;
4105                 goto port_unlock;
4106         }
4107
4108         /* Trying to deactivate PPS when already deactivated so just bail */
4109         if (!port->pps_data.active && !activate)
4110                 goto port_unlock;
4111
4112         if (port->state != SNK_READY) {
4113                 ret = -EAGAIN;
4114                 goto port_unlock;
4115         }
4116
4117         reinit_completion(&port->pps_complete);
4118         port->pps_status = 0;
4119         port->pps_pending = true;
4120
4121         /* Trigger PPS request or move back to standard PDO contract */
4122         if (activate) {
4123                 port->pps_data.out_volt = port->supply_voltage;
4124                 port->pps_data.op_curr = port->current_limit;
4125                 tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4126         } else {
4127                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4128         }
4129         mutex_unlock(&port->lock);
4130
4131         if (!wait_for_completion_timeout(&port->pps_complete,
4132                                 msecs_to_jiffies(PD_PPS_CTRL_TIMEOUT)))
4133                 ret = -ETIMEDOUT;
4134         else
4135                 ret = port->pps_status;
4136
4137         goto swap_unlock;
4138
4139 port_unlock:
4140         mutex_unlock(&port->lock);
4141 swap_unlock:
4142         mutex_unlock(&port->swap_lock);
4143
4144         return ret;
4145 }
4146
4147 static void tcpm_init(struct tcpm_port *port)
4148 {
4149         enum typec_cc_status cc1, cc2;
4150
4151         port->tcpc->init(port->tcpc);
4152
4153         tcpm_reset_port(port);
4154
4155         /*
4156          * XXX
4157          * Should possibly wait for VBUS to settle if it was enabled locally
4158          * since tcpm_reset_port() will disable VBUS.
4159          */
4160         port->vbus_present = port->tcpc->get_vbus(port->tcpc);
4161         if (port->vbus_present)
4162                 port->vbus_never_low = true;
4163
4164         tcpm_set_state(port, tcpm_default_state(port), 0);
4165
4166         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
4167                 _tcpm_cc_change(port, cc1, cc2);
4168
4169         /*
4170          * Some adapters need a clean slate at startup, and won't recover
4171          * otherwise. So do not try to be fancy and force a clean disconnect.
4172          */
4173         tcpm_set_state(port, PORT_RESET, 0);
4174 }
4175
4176 static int tcpm_port_type_set(const struct typec_capability *cap,
4177                               enum typec_port_type type)
4178 {
4179         struct tcpm_port *port = typec_cap_to_tcpm(cap);
4180
4181         mutex_lock(&port->lock);
4182         if (type == port->port_type)
4183                 goto port_unlock;
4184
4185         port->port_type = type;
4186
4187         if (!port->connected) {
4188                 tcpm_set_state(port, PORT_RESET, 0);
4189         } else if (type == TYPEC_PORT_SNK) {
4190                 if (!(port->pwr_role == TYPEC_SINK &&
4191                       port->data_role == TYPEC_DEVICE))
4192                         tcpm_set_state(port, PORT_RESET, 0);
4193         } else if (type == TYPEC_PORT_SRC) {
4194                 if (!(port->pwr_role == TYPEC_SOURCE &&
4195                       port->data_role == TYPEC_HOST))
4196                         tcpm_set_state(port, PORT_RESET, 0);
4197         }
4198
4199 port_unlock:
4200         mutex_unlock(&port->lock);
4201         return 0;
4202 }
4203
4204 void tcpm_tcpc_reset(struct tcpm_port *port)
4205 {
4206         mutex_lock(&port->lock);
4207         /* XXX: Maintain PD connection if possible? */
4208         tcpm_init(port);
4209         mutex_unlock(&port->lock);
4210 }
4211 EXPORT_SYMBOL_GPL(tcpm_tcpc_reset);
4212
4213 static int tcpm_copy_pdos(u32 *dest_pdo, const u32 *src_pdo,
4214                           unsigned int nr_pdo)
4215 {
4216         unsigned int i;
4217
4218         if (nr_pdo > PDO_MAX_OBJECTS)
4219                 nr_pdo = PDO_MAX_OBJECTS;
4220
4221         for (i = 0; i < nr_pdo; i++)
4222                 dest_pdo[i] = src_pdo[i];
4223
4224         return nr_pdo;
4225 }
4226
4227 static int tcpm_copy_vdos(u32 *dest_vdo, const u32 *src_vdo,
4228                           unsigned int nr_vdo)
4229 {
4230         unsigned int i;
4231
4232         if (nr_vdo > VDO_MAX_OBJECTS)
4233                 nr_vdo = VDO_MAX_OBJECTS;
4234
4235         for (i = 0; i < nr_vdo; i++)
4236                 dest_vdo[i] = src_vdo[i];
4237
4238         return nr_vdo;
4239 }
4240
4241 int tcpm_update_source_capabilities(struct tcpm_port *port, const u32 *pdo,
4242                                     unsigned int nr_pdo)
4243 {
4244         if (tcpm_validate_caps(port, pdo, nr_pdo))
4245                 return -EINVAL;
4246
4247         mutex_lock(&port->lock);
4248         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, pdo, nr_pdo);
4249         switch (port->state) {
4250         case SRC_UNATTACHED:
4251         case SRC_ATTACH_WAIT:
4252         case SRC_TRYWAIT:
4253                 tcpm_set_cc(port, tcpm_rp_cc(port));
4254                 break;
4255         case SRC_SEND_CAPABILITIES:
4256         case SRC_NEGOTIATE_CAPABILITIES:
4257         case SRC_READY:
4258         case SRC_WAIT_NEW_CAPABILITIES:
4259                 tcpm_set_cc(port, tcpm_rp_cc(port));
4260                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
4261                 break;
4262         default:
4263                 break;
4264         }
4265         mutex_unlock(&port->lock);
4266         return 0;
4267 }
4268 EXPORT_SYMBOL_GPL(tcpm_update_source_capabilities);
4269
4270 int tcpm_update_sink_capabilities(struct tcpm_port *port, const u32 *pdo,
4271                                   unsigned int nr_pdo,
4272                                   unsigned int operating_snk_mw)
4273 {
4274         if (tcpm_validate_caps(port, pdo, nr_pdo))
4275                 return -EINVAL;
4276
4277         mutex_lock(&port->lock);
4278         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, pdo, nr_pdo);
4279         port->operating_snk_mw = operating_snk_mw;
4280         port->update_sink_caps = true;
4281
4282         switch (port->state) {
4283         case SNK_NEGOTIATE_CAPABILITIES:
4284         case SNK_NEGOTIATE_PPS_CAPABILITIES:
4285         case SNK_READY:
4286         case SNK_TRANSITION_SINK:
4287         case SNK_TRANSITION_SINK_VBUS:
4288                 if (port->pps_data.active)
4289                         tcpm_set_state(port, SNK_NEGOTIATE_PPS_CAPABILITIES, 0);
4290                 else
4291                         tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
4292                 break;
4293         default:
4294                 break;
4295         }
4296         mutex_unlock(&port->lock);
4297         return 0;
4298 }
4299 EXPORT_SYMBOL_GPL(tcpm_update_sink_capabilities);
4300
4301 /* Power Supply access to expose source power information */
4302 enum tcpm_psy_online_states {
4303         TCPM_PSY_OFFLINE = 0,
4304         TCPM_PSY_FIXED_ONLINE,
4305         TCPM_PSY_PROG_ONLINE,
4306 };
4307
4308 static enum power_supply_property tcpm_psy_props[] = {
4309         POWER_SUPPLY_PROP_USB_TYPE,
4310         POWER_SUPPLY_PROP_ONLINE,
4311         POWER_SUPPLY_PROP_VOLTAGE_MIN,
4312         POWER_SUPPLY_PROP_VOLTAGE_MAX,
4313         POWER_SUPPLY_PROP_VOLTAGE_NOW,
4314         POWER_SUPPLY_PROP_CURRENT_MAX,
4315         POWER_SUPPLY_PROP_CURRENT_NOW,
4316 };
4317
4318 static int tcpm_psy_get_online(struct tcpm_port *port,
4319                                union power_supply_propval *val)
4320 {
4321         if (port->vbus_charge) {
4322                 if (port->pps_data.active)
4323                         val->intval = TCPM_PSY_PROG_ONLINE;
4324                 else
4325                         val->intval = TCPM_PSY_FIXED_ONLINE;
4326         } else {
4327                 val->intval = TCPM_PSY_OFFLINE;
4328         }
4329
4330         return 0;
4331 }
4332
4333 static int tcpm_psy_get_voltage_min(struct tcpm_port *port,
4334                                     union power_supply_propval *val)
4335 {
4336         if (port->pps_data.active)
4337                 val->intval = port->pps_data.min_volt * 1000;
4338         else
4339                 val->intval = port->supply_voltage * 1000;
4340
4341         return 0;
4342 }
4343
4344 static int tcpm_psy_get_voltage_max(struct tcpm_port *port,
4345                                     union power_supply_propval *val)
4346 {
4347         if (port->pps_data.active)
4348                 val->intval = port->pps_data.max_volt * 1000;
4349         else
4350                 val->intval = port->supply_voltage * 1000;
4351
4352         return 0;
4353 }
4354
4355 static int tcpm_psy_get_voltage_now(struct tcpm_port *port,
4356                                     union power_supply_propval *val)
4357 {
4358         val->intval = port->supply_voltage * 1000;
4359
4360         return 0;
4361 }
4362
4363 static int tcpm_psy_get_current_max(struct tcpm_port *port,
4364                                     union power_supply_propval *val)
4365 {
4366         if (port->pps_data.active)
4367                 val->intval = port->pps_data.max_curr * 1000;
4368         else
4369                 val->intval = port->current_limit * 1000;
4370
4371         return 0;
4372 }
4373
4374 static int tcpm_psy_get_current_now(struct tcpm_port *port,
4375                                     union power_supply_propval *val)
4376 {
4377         val->intval = port->current_limit * 1000;
4378
4379         return 0;
4380 }
4381
4382 static int tcpm_psy_get_prop(struct power_supply *psy,
4383                              enum power_supply_property psp,
4384                              union power_supply_propval *val)
4385 {
4386         struct tcpm_port *port = power_supply_get_drvdata(psy);
4387         int ret = 0;
4388
4389         switch (psp) {
4390         case POWER_SUPPLY_PROP_USB_TYPE:
4391                 val->intval = port->usb_type;
4392                 break;
4393         case POWER_SUPPLY_PROP_ONLINE:
4394                 ret = tcpm_psy_get_online(port, val);
4395                 break;
4396         case POWER_SUPPLY_PROP_VOLTAGE_MIN:
4397                 ret = tcpm_psy_get_voltage_min(port, val);
4398                 break;
4399         case POWER_SUPPLY_PROP_VOLTAGE_MAX:
4400                 ret = tcpm_psy_get_voltage_max(port, val);
4401                 break;
4402         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4403                 ret = tcpm_psy_get_voltage_now(port, val);
4404                 break;
4405         case POWER_SUPPLY_PROP_CURRENT_MAX:
4406                 ret = tcpm_psy_get_current_max(port, val);
4407                 break;
4408         case POWER_SUPPLY_PROP_CURRENT_NOW:
4409                 ret = tcpm_psy_get_current_now(port, val);
4410                 break;
4411         default:
4412                 ret = -EINVAL;
4413                 break;
4414         }
4415
4416         return ret;
4417 }
4418
4419 static int tcpm_psy_set_online(struct tcpm_port *port,
4420                                const union power_supply_propval *val)
4421 {
4422         int ret;
4423
4424         switch (val->intval) {
4425         case TCPM_PSY_FIXED_ONLINE:
4426                 ret = tcpm_pps_activate(port, false);
4427                 break;
4428         case TCPM_PSY_PROG_ONLINE:
4429                 ret = tcpm_pps_activate(port, true);
4430                 break;
4431         default:
4432                 ret = -EINVAL;
4433                 break;
4434         }
4435
4436         return ret;
4437 }
4438
4439 static int tcpm_psy_set_prop(struct power_supply *psy,
4440                              enum power_supply_property psp,
4441                              const union power_supply_propval *val)
4442 {
4443         struct tcpm_port *port = power_supply_get_drvdata(psy);
4444         int ret;
4445
4446         switch (psp) {
4447         case POWER_SUPPLY_PROP_ONLINE:
4448                 ret = tcpm_psy_set_online(port, val);
4449                 break;
4450         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4451                 if (val->intval < port->pps_data.min_volt * 1000 ||
4452                     val->intval > port->pps_data.max_volt * 1000)
4453                         ret = -EINVAL;
4454                 else
4455                         ret = tcpm_pps_set_out_volt(port, val->intval / 1000);
4456                 break;
4457         case POWER_SUPPLY_PROP_CURRENT_NOW:
4458                 if (val->intval > port->pps_data.max_curr * 1000)
4459                         ret = -EINVAL;
4460                 else
4461                         ret = tcpm_pps_set_op_curr(port, val->intval / 1000);
4462                 break;
4463         default:
4464                 ret = -EINVAL;
4465                 break;
4466         }
4467
4468         return ret;
4469 }
4470
4471 static int tcpm_psy_prop_writeable(struct power_supply *psy,
4472                                    enum power_supply_property psp)
4473 {
4474         switch (psp) {
4475         case POWER_SUPPLY_PROP_ONLINE:
4476         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
4477         case POWER_SUPPLY_PROP_CURRENT_NOW:
4478                 return 1;
4479         default:
4480                 return 0;
4481         }
4482 }
4483
4484 static enum power_supply_usb_type tcpm_psy_usb_types[] = {
4485         POWER_SUPPLY_USB_TYPE_C,
4486         POWER_SUPPLY_USB_TYPE_PD,
4487         POWER_SUPPLY_USB_TYPE_PD_PPS,
4488 };
4489
4490 static const char *tcpm_psy_name_prefix = "tcpm-source-psy-";
4491
4492 static int devm_tcpm_psy_register(struct tcpm_port *port)
4493 {
4494         struct power_supply_config psy_cfg = {};
4495         const char *port_dev_name = dev_name(port->dev);
4496         size_t psy_name_len = strlen(tcpm_psy_name_prefix) +
4497                                      strlen(port_dev_name) + 1;
4498         char *psy_name;
4499
4500         psy_cfg.drv_data = port;
4501         psy_cfg.fwnode = dev_fwnode(port->dev);
4502         psy_name = devm_kzalloc(port->dev, psy_name_len, GFP_KERNEL);
4503         if (!psy_name)
4504                 return -ENOMEM;
4505
4506         snprintf(psy_name, psy_name_len, "%s%s", tcpm_psy_name_prefix,
4507                  port_dev_name);
4508         port->psy_desc.name = psy_name;
4509         port->psy_desc.type = POWER_SUPPLY_TYPE_USB,
4510         port->psy_desc.usb_types = tcpm_psy_usb_types;
4511         port->psy_desc.num_usb_types = ARRAY_SIZE(tcpm_psy_usb_types);
4512         port->psy_desc.properties = tcpm_psy_props,
4513         port->psy_desc.num_properties = ARRAY_SIZE(tcpm_psy_props),
4514         port->psy_desc.get_property = tcpm_psy_get_prop,
4515         port->psy_desc.set_property = tcpm_psy_set_prop,
4516         port->psy_desc.property_is_writeable = tcpm_psy_prop_writeable,
4517
4518         port->usb_type = POWER_SUPPLY_USB_TYPE_C;
4519
4520         port->psy = devm_power_supply_register(port->dev, &port->psy_desc,
4521                                                &psy_cfg);
4522
4523         return PTR_ERR_OR_ZERO(port->psy);
4524 }
4525
4526 struct tcpm_port *tcpm_register_port(struct device *dev, struct tcpc_dev *tcpc)
4527 {
4528         struct tcpm_port *port;
4529         int i, err;
4530
4531         if (!dev || !tcpc || !tcpc->config ||
4532             !tcpc->get_vbus || !tcpc->set_cc || !tcpc->get_cc ||
4533             !tcpc->set_polarity || !tcpc->set_vconn || !tcpc->set_vbus ||
4534             !tcpc->set_pd_rx || !tcpc->set_roles || !tcpc->pd_transmit)
4535                 return ERR_PTR(-EINVAL);
4536
4537         port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
4538         if (!port)
4539                 return ERR_PTR(-ENOMEM);
4540
4541         port->dev = dev;
4542         port->tcpc = tcpc;
4543
4544         mutex_init(&port->lock);
4545         mutex_init(&port->swap_lock);
4546
4547         port->wq = create_singlethread_workqueue(dev_name(dev));
4548         if (!port->wq)
4549                 return ERR_PTR(-ENOMEM);
4550         INIT_DELAYED_WORK(&port->state_machine, tcpm_state_machine_work);
4551         INIT_DELAYED_WORK(&port->vdm_state_machine, vdm_state_machine_work);
4552         INIT_WORK(&port->event_work, tcpm_pd_event_handler);
4553
4554         spin_lock_init(&port->pd_event_lock);
4555
4556         init_completion(&port->tx_complete);
4557         init_completion(&port->swap_complete);
4558         init_completion(&port->pps_complete);
4559         tcpm_debugfs_init(port);
4560
4561         if (tcpm_validate_caps(port, tcpc->config->src_pdo,
4562                                tcpc->config->nr_src_pdo) ||
4563             tcpm_validate_caps(port, tcpc->config->snk_pdo,
4564                                tcpc->config->nr_snk_pdo)) {
4565                 err = -EINVAL;
4566                 goto out_destroy_wq;
4567         }
4568         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, tcpc->config->src_pdo,
4569                                           tcpc->config->nr_src_pdo);
4570         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, tcpc->config->snk_pdo,
4571                                           tcpc->config->nr_snk_pdo);
4572         port->nr_snk_vdo = tcpm_copy_vdos(port->snk_vdo, tcpc->config->snk_vdo,
4573                                           tcpc->config->nr_snk_vdo);
4574
4575         port->operating_snk_mw = tcpc->config->operating_snk_mw;
4576         if (!tcpc->config->try_role_hw)
4577                 port->try_role = tcpc->config->default_role;
4578         else
4579                 port->try_role = TYPEC_NO_PREFERRED_ROLE;
4580
4581         port->typec_caps.prefer_role = tcpc->config->default_role;
4582         port->typec_caps.type = tcpc->config->type;
4583         port->typec_caps.data = tcpc->config->data;
4584         port->typec_caps.revision = 0x0120;     /* Type-C spec release 1.2 */
4585         port->typec_caps.pd_revision = 0x0300;  /* USB-PD spec release 3.0 */
4586         port->typec_caps.dr_set = tcpm_dr_set;
4587         port->typec_caps.pr_set = tcpm_pr_set;
4588         port->typec_caps.vconn_set = tcpm_vconn_set;
4589         port->typec_caps.try_role = tcpm_try_role;
4590         port->typec_caps.port_type_set = tcpm_port_type_set;
4591
4592         port->partner_desc.identity = &port->partner_ident;
4593         port->port_type = tcpc->config->type;
4594
4595         port->role_sw = usb_role_switch_get(port->dev);
4596         if (IS_ERR(port->role_sw)) {
4597                 err = PTR_ERR(port->role_sw);
4598                 goto out_destroy_wq;
4599         }
4600
4601         err = devm_tcpm_psy_register(port);
4602         if (err)
4603                 goto out_destroy_wq;
4604
4605         port->typec_port = typec_register_port(port->dev, &port->typec_caps);
4606         if (IS_ERR(port->typec_port)) {
4607                 err = PTR_ERR(port->typec_port);
4608                 goto out_destroy_wq;
4609         }
4610
4611         if (tcpc->config->alt_modes) {
4612                 const struct typec_altmode_desc *paltmode = tcpc->config->alt_modes;
4613
4614                 i = 0;
4615                 while (paltmode->svid && i < ARRAY_SIZE(port->port_altmode)) {
4616                         struct typec_altmode *alt;
4617
4618                         alt = typec_port_register_altmode(port->typec_port,
4619                                                           paltmode);
4620                         if (IS_ERR(alt)) {
4621                                 tcpm_log(port,
4622                                          "%s: failed to register port alternate mode 0x%x",
4623                                          dev_name(dev), paltmode->svid);
4624                                 break;
4625                         }
4626                         port->port_altmode[i] = alt;
4627                         i++;
4628                         paltmode++;
4629                 }
4630         }
4631
4632         mutex_lock(&port->lock);
4633         tcpm_init(port);
4634         mutex_unlock(&port->lock);
4635
4636         tcpm_log(port, "%s: registered", dev_name(dev));
4637         return port;
4638
4639 out_destroy_wq:
4640         usb_role_switch_put(port->role_sw);
4641         destroy_workqueue(port->wq);
4642         return ERR_PTR(err);
4643 }
4644 EXPORT_SYMBOL_GPL(tcpm_register_port);
4645
4646 void tcpm_unregister_port(struct tcpm_port *port)
4647 {
4648         int i;
4649
4650         tcpm_reset_port(port);
4651         for (i = 0; i < ARRAY_SIZE(port->port_altmode); i++)
4652                 typec_unregister_altmode(port->port_altmode[i]);
4653         typec_unregister_port(port->typec_port);
4654         usb_role_switch_put(port->role_sw);
4655         tcpm_debugfs_exit(port);
4656         destroy_workqueue(port->wq);
4657 }
4658 EXPORT_SYMBOL_GPL(tcpm_unregister_port);
4659
4660 MODULE_AUTHOR("Guenter Roeck <[email protected]>");
4661 MODULE_DESCRIPTION("USB Type-C Port Manager");
4662 MODULE_LICENSE("GPL");