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