GNU Linux-libre 4.14.251-gnu1
[releases.git] / drivers / staging / typec / tcpm.c
1 /*
2  * Copyright 2015-2017 Google, Inc
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * USB Power Delivery protocol stack.
15  */
16
17 #include <linux/completion.h>
18 #include <linux/debugfs.h>
19 #include <linux/device.h>
20 #include <linux/jiffies.h>
21 #include <linux/kernel.h>
22 #include <linux/module.h>
23 #include <linux/mutex.h>
24 #include <linux/proc_fs.h>
25 #include <linux/sched/clock.h>
26 #include <linux/seq_file.h>
27 #include <linux/slab.h>
28 #include <linux/spinlock.h>
29 #include <linux/usb/typec.h>
30 #include <linux/workqueue.h>
31
32 #include "pd.h"
33 #include "pd_vdo.h"
34 #include "pd_bdo.h"
35 #include "tcpm.h"
36
37 #define FOREACH_STATE(S)                        \
38         S(INVALID_STATE),                       \
39         S(DRP_TOGGLING),                        \
40         S(SRC_UNATTACHED),                      \
41         S(SRC_ATTACH_WAIT),                     \
42         S(SRC_ATTACHED),                        \
43         S(SRC_STARTUP),                         \
44         S(SRC_SEND_CAPABILITIES),               \
45         S(SRC_NEGOTIATE_CAPABILITIES),          \
46         S(SRC_TRANSITION_SUPPLY),               \
47         S(SRC_READY),                           \
48         S(SRC_WAIT_NEW_CAPABILITIES),           \
49                                                 \
50         S(SNK_UNATTACHED),                      \
51         S(SNK_ATTACH_WAIT),                     \
52         S(SNK_DEBOUNCED),                       \
53         S(SNK_ATTACHED),                        \
54         S(SNK_STARTUP),                         \
55         S(SNK_DISCOVERY),                       \
56         S(SNK_DISCOVERY_DEBOUNCE),              \
57         S(SNK_DISCOVERY_DEBOUNCE_DONE),         \
58         S(SNK_WAIT_CAPABILITIES),               \
59         S(SNK_NEGOTIATE_CAPABILITIES),          \
60         S(SNK_TRANSITION_SINK),                 \
61         S(SNK_TRANSITION_SINK_VBUS),            \
62         S(SNK_READY),                           \
63                                                 \
64         S(ACC_UNATTACHED),                      \
65         S(DEBUG_ACC_ATTACHED),                  \
66         S(AUDIO_ACC_ATTACHED),                  \
67         S(AUDIO_ACC_DEBOUNCE),                  \
68                                                 \
69         S(HARD_RESET_SEND),                     \
70         S(HARD_RESET_START),                    \
71         S(SRC_HARD_RESET_VBUS_OFF),             \
72         S(SRC_HARD_RESET_VBUS_ON),              \
73         S(SNK_HARD_RESET_SINK_OFF),             \
74         S(SNK_HARD_RESET_WAIT_VBUS),            \
75         S(SNK_HARD_RESET_SINK_ON),              \
76                                                 \
77         S(SOFT_RESET),                          \
78         S(SOFT_RESET_SEND),                     \
79                                                 \
80         S(DR_SWAP_ACCEPT),                      \
81         S(DR_SWAP_SEND),                        \
82         S(DR_SWAP_SEND_TIMEOUT),                \
83         S(DR_SWAP_CANCEL),                      \
84         S(DR_SWAP_CHANGE_DR),                   \
85                                                 \
86         S(PR_SWAP_ACCEPT),                      \
87         S(PR_SWAP_SEND),                        \
88         S(PR_SWAP_SEND_TIMEOUT),                \
89         S(PR_SWAP_CANCEL),                      \
90         S(PR_SWAP_START),                       \
91         S(PR_SWAP_SRC_SNK_TRANSITION_OFF),      \
92         S(PR_SWAP_SRC_SNK_SOURCE_OFF),          \
93         S(PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED), \
94         S(PR_SWAP_SRC_SNK_SINK_ON),             \
95         S(PR_SWAP_SNK_SRC_SINK_OFF),            \
96         S(PR_SWAP_SNK_SRC_SOURCE_ON),           \
97         S(PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP),    \
98                                                 \
99         S(VCONN_SWAP_ACCEPT),                   \
100         S(VCONN_SWAP_SEND),                     \
101         S(VCONN_SWAP_SEND_TIMEOUT),             \
102         S(VCONN_SWAP_CANCEL),                   \
103         S(VCONN_SWAP_START),                    \
104         S(VCONN_SWAP_WAIT_FOR_VCONN),           \
105         S(VCONN_SWAP_TURN_ON_VCONN),            \
106         S(VCONN_SWAP_TURN_OFF_VCONN),           \
107                                                 \
108         S(SNK_TRY),                             \
109         S(SNK_TRY_WAIT),                        \
110         S(SNK_TRY_WAIT_DEBOUNCE),               \
111         S(SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS),    \
112         S(SRC_TRYWAIT),                         \
113         S(SRC_TRYWAIT_DEBOUNCE),                \
114         S(SRC_TRYWAIT_UNATTACHED),              \
115                                                 \
116         S(SRC_TRY),                             \
117         S(SRC_TRY_WAIT),                        \
118         S(SRC_TRY_DEBOUNCE),                    \
119         S(SNK_TRYWAIT),                         \
120         S(SNK_TRYWAIT_DEBOUNCE),                \
121         S(SNK_TRYWAIT_VBUS),                    \
122         S(BIST_RX),                             \
123                                                 \
124         S(ERROR_RECOVERY),                      \
125         S(PORT_RESET),                          \
126         S(PORT_RESET_WAIT_OFF)
127
128 #define GENERATE_ENUM(e)        e
129 #define GENERATE_STRING(s)      #s
130
131 enum tcpm_state {
132         FOREACH_STATE(GENERATE_ENUM)
133 };
134
135 static const char * const tcpm_states[] = {
136         FOREACH_STATE(GENERATE_STRING)
137 };
138
139 enum vdm_states {
140         VDM_STATE_ERR_BUSY = -3,
141         VDM_STATE_ERR_SEND = -2,
142         VDM_STATE_ERR_TMOUT = -1,
143         VDM_STATE_DONE = 0,
144         /* Anything >0 represents an active state */
145         VDM_STATE_READY = 1,
146         VDM_STATE_BUSY = 2,
147         VDM_STATE_WAIT_RSP_BUSY = 3,
148 };
149
150 enum pd_msg_request {
151         PD_MSG_NONE = 0,
152         PD_MSG_CTRL_REJECT,
153         PD_MSG_CTRL_WAIT,
154         PD_MSG_DATA_SINK_CAP,
155         PD_MSG_DATA_SOURCE_CAP,
156 };
157
158 /* Events from low level driver */
159
160 #define TCPM_CC_EVENT           BIT(0)
161 #define TCPM_VBUS_EVENT         BIT(1)
162 #define TCPM_RESET_EVENT        BIT(2)
163
164 #define LOG_BUFFER_ENTRIES      1024
165 #define LOG_BUFFER_ENTRY_SIZE   128
166
167 /* Alternate mode support */
168
169 #define SVID_DISCOVERY_MAX      16
170
171 struct pd_mode_data {
172         int svid_index;         /* current SVID index           */
173         int nsvids;
174         u16 svids[SVID_DISCOVERY_MAX];
175         int altmodes;           /* number of alternate modes    */
176         struct typec_altmode_desc altmode_desc[SVID_DISCOVERY_MAX];
177 };
178
179 struct tcpm_port {
180         struct device *dev;
181
182         struct mutex lock;              /* tcpm state machine lock */
183         struct workqueue_struct *wq;
184
185         struct typec_capability typec_caps;
186         struct typec_port *typec_port;
187
188         struct tcpc_dev *tcpc;
189
190         enum typec_role vconn_role;
191         enum typec_role pwr_role;
192         enum typec_data_role data_role;
193         enum typec_pwr_opmode pwr_opmode;
194
195         struct usb_pd_identity partner_ident;
196         struct typec_partner_desc partner_desc;
197         struct typec_partner *partner;
198
199         enum typec_cc_status cc_req;
200
201         enum typec_cc_status cc1;
202         enum typec_cc_status cc2;
203         enum typec_cc_polarity polarity;
204
205         bool attached;
206         bool connected;
207         enum typec_port_type port_type;
208         bool vbus_present;
209         bool vbus_never_low;
210         bool vbus_source;
211         bool vbus_charge;
212
213         bool send_discover;
214         bool op_vsafe5v;
215
216         int try_role;
217         int try_snk_count;
218         int try_src_count;
219
220         enum pd_msg_request queued_message;
221
222         enum tcpm_state enter_state;
223         enum tcpm_state prev_state;
224         enum tcpm_state state;
225         enum tcpm_state delayed_state;
226         unsigned long delayed_runtime;
227         unsigned long delay_ms;
228
229         spinlock_t pd_event_lock;
230         u32 pd_events;
231
232         struct work_struct event_work;
233         struct delayed_work state_machine;
234         struct delayed_work vdm_state_machine;
235         bool state_machine_running;
236
237         struct completion tx_complete;
238         enum tcpm_transmit_status tx_status;
239
240         struct mutex swap_lock;         /* swap command lock */
241         bool swap_pending;
242         bool non_pd_role_swap;
243         struct completion swap_complete;
244         int swap_status;
245
246         unsigned int message_id;
247         unsigned int caps_count;
248         unsigned int hard_reset_count;
249         bool pd_capable;
250         bool explicit_contract;
251         unsigned int rx_msgid;
252
253         /* Partner capabilities/requests */
254         u32 sink_request;
255         u32 source_caps[PDO_MAX_OBJECTS];
256         unsigned int nr_source_caps;
257         u32 sink_caps[PDO_MAX_OBJECTS];
258         unsigned int nr_sink_caps;
259
260         /* Local capabilities */
261         u32 src_pdo[PDO_MAX_OBJECTS];
262         unsigned int nr_src_pdo;
263         u32 snk_pdo[PDO_MAX_OBJECTS];
264         unsigned int nr_snk_pdo;
265         u32 snk_vdo[VDO_MAX_OBJECTS];
266         unsigned int nr_snk_vdo;
267
268         unsigned int max_snk_mv;
269         unsigned int max_snk_ma;
270         unsigned int max_snk_mw;
271         unsigned int operating_snk_mw;
272
273         /* Requested current / voltage */
274         u32 current_limit;
275         u32 supply_voltage;
276
277         u32 bist_request;
278
279         /* PD state for Vendor Defined Messages */
280         enum vdm_states vdm_state;
281         u32 vdm_retries;
282         /* next Vendor Defined Message to send */
283         u32 vdo_data[VDO_MAX_SIZE];
284         u8 vdo_count;
285         /* VDO to retry if UFP responder replied busy */
286         u32 vdo_retry;
287
288         /* Alternate mode data */
289
290         struct pd_mode_data mode_data;
291         struct typec_altmode *partner_altmode[SVID_DISCOVERY_MAX];
292         struct typec_altmode *port_altmode[SVID_DISCOVERY_MAX];
293
294         /* Deadline in jiffies to exit src_try_wait state */
295         unsigned long max_wait;
296
297 #ifdef CONFIG_DEBUG_FS
298         struct dentry *dentry;
299         struct mutex logbuffer_lock;    /* log buffer access lock */
300         int logbuffer_head;
301         int logbuffer_tail;
302         u8 *logbuffer[LOG_BUFFER_ENTRIES];
303 #endif
304 };
305
306 struct pd_rx_event {
307         struct work_struct work;
308         struct tcpm_port *port;
309         struct pd_message msg;
310 };
311
312 #define tcpm_cc_is_sink(cc) \
313         ((cc) == TYPEC_CC_RP_DEF || (cc) == TYPEC_CC_RP_1_5 || \
314          (cc) == TYPEC_CC_RP_3_0)
315
316 #define tcpm_port_is_sink(port) \
317         ((tcpm_cc_is_sink((port)->cc1) && !tcpm_cc_is_sink((port)->cc2)) || \
318          (tcpm_cc_is_sink((port)->cc2) && !tcpm_cc_is_sink((port)->cc1)))
319
320 #define tcpm_cc_is_source(cc) ((cc) == TYPEC_CC_RD)
321 #define tcpm_cc_is_audio(cc) ((cc) == TYPEC_CC_RA)
322 #define tcpm_cc_is_open(cc) ((cc) == TYPEC_CC_OPEN)
323
324 #define tcpm_port_is_source(port) \
325         ((tcpm_cc_is_source((port)->cc1) && \
326          !tcpm_cc_is_source((port)->cc2)) || \
327          (tcpm_cc_is_source((port)->cc2) && \
328           !tcpm_cc_is_source((port)->cc1)))
329
330 #define tcpm_port_is_debug(port) \
331         (tcpm_cc_is_source((port)->cc1) && tcpm_cc_is_source((port)->cc2))
332
333 #define tcpm_port_is_audio(port) \
334         (tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_audio((port)->cc2))
335
336 #define tcpm_port_is_audio_detached(port) \
337         ((tcpm_cc_is_audio((port)->cc1) && tcpm_cc_is_open((port)->cc2)) || \
338          (tcpm_cc_is_audio((port)->cc2) && tcpm_cc_is_open((port)->cc1)))
339
340 #define tcpm_try_snk(port) \
341         ((port)->try_snk_count == 0 && (port)->try_role == TYPEC_SINK && \
342         (port)->port_type == TYPEC_PORT_DRP)
343
344 #define tcpm_try_src(port) \
345         ((port)->try_src_count == 0 && (port)->try_role == TYPEC_SOURCE && \
346         (port)->port_type == TYPEC_PORT_DRP)
347
348 static enum tcpm_state tcpm_default_state(struct tcpm_port *port)
349 {
350         if (port->port_type == TYPEC_PORT_DRP) {
351                 if (port->try_role == TYPEC_SINK)
352                         return SNK_UNATTACHED;
353                 else if (port->try_role == TYPEC_SOURCE)
354                         return SRC_UNATTACHED;
355                 else if (port->tcpc->config->default_role == TYPEC_SINK)
356                         return SNK_UNATTACHED;
357                 /* Fall through to return SRC_UNATTACHED */
358         } else if (port->port_type == TYPEC_PORT_UFP) {
359                 return SNK_UNATTACHED;
360         }
361         return SRC_UNATTACHED;
362 }
363
364 static inline
365 struct tcpm_port *typec_cap_to_tcpm(const struct typec_capability *cap)
366 {
367         return container_of(cap, struct tcpm_port, typec_caps);
368 }
369
370 static bool tcpm_port_is_disconnected(struct tcpm_port *port)
371 {
372         return (!port->attached && port->cc1 == TYPEC_CC_OPEN &&
373                 port->cc2 == TYPEC_CC_OPEN) ||
374                (port->attached && ((port->polarity == TYPEC_POLARITY_CC1 &&
375                                     port->cc1 == TYPEC_CC_OPEN) ||
376                                    (port->polarity == TYPEC_POLARITY_CC2 &&
377                                     port->cc2 == TYPEC_CC_OPEN)));
378 }
379
380 /*
381  * Logging
382  */
383
384 #ifdef CONFIG_DEBUG_FS
385
386 static bool tcpm_log_full(struct tcpm_port *port)
387 {
388         return port->logbuffer_tail ==
389                 (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
390 }
391
392 __printf(2, 0)
393 static void _tcpm_log(struct tcpm_port *port, const char *fmt, va_list args)
394 {
395         char tmpbuffer[LOG_BUFFER_ENTRY_SIZE];
396         u64 ts_nsec = local_clock();
397         unsigned long rem_nsec;
398
399         if (!port->logbuffer[port->logbuffer_head]) {
400                 port->logbuffer[port->logbuffer_head] =
401                                 kzalloc(LOG_BUFFER_ENTRY_SIZE, GFP_KERNEL);
402                 if (!port->logbuffer[port->logbuffer_head])
403                         return;
404         }
405
406         vsnprintf(tmpbuffer, sizeof(tmpbuffer), fmt, args);
407
408         mutex_lock(&port->logbuffer_lock);
409
410         if (tcpm_log_full(port)) {
411                 port->logbuffer_head = max(port->logbuffer_head - 1, 0);
412                 strcpy(tmpbuffer, "overflow");
413         }
414
415         if (port->logbuffer_head < 0 ||
416             port->logbuffer_head >= LOG_BUFFER_ENTRIES) {
417                 dev_warn(port->dev,
418                          "Bad log buffer index %d\n", port->logbuffer_head);
419                 goto abort;
420         }
421
422         if (!port->logbuffer[port->logbuffer_head]) {
423                 dev_warn(port->dev,
424                          "Log buffer index %d is NULL\n", port->logbuffer_head);
425                 goto abort;
426         }
427
428         rem_nsec = do_div(ts_nsec, 1000000000);
429         scnprintf(port->logbuffer[port->logbuffer_head],
430                   LOG_BUFFER_ENTRY_SIZE, "[%5lu.%06lu] %s",
431                   (unsigned long)ts_nsec, rem_nsec / 1000,
432                   tmpbuffer);
433         port->logbuffer_head = (port->logbuffer_head + 1) % LOG_BUFFER_ENTRIES;
434
435 abort:
436         mutex_unlock(&port->logbuffer_lock);
437 }
438
439 __printf(2, 3)
440 static void tcpm_log(struct tcpm_port *port, const char *fmt, ...)
441 {
442         va_list args;
443
444         /* Do not log while disconnected and unattached */
445         if (tcpm_port_is_disconnected(port) &&
446             (port->state == SRC_UNATTACHED || port->state == SNK_UNATTACHED ||
447              port->state == DRP_TOGGLING))
448                 return;
449
450         va_start(args, fmt);
451         _tcpm_log(port, fmt, args);
452         va_end(args);
453 }
454
455 __printf(2, 3)
456 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...)
457 {
458         va_list args;
459
460         va_start(args, fmt);
461         _tcpm_log(port, fmt, args);
462         va_end(args);
463 }
464
465 static void tcpm_log_source_caps(struct tcpm_port *port)
466 {
467         int i;
468
469         for (i = 0; i < port->nr_source_caps; i++) {
470                 u32 pdo = port->source_caps[i];
471                 enum pd_pdo_type type = pdo_type(pdo);
472                 char msg[64];
473
474                 switch (type) {
475                 case PDO_TYPE_FIXED:
476                         scnprintf(msg, sizeof(msg),
477                                   "%u mV, %u mA [%s%s%s%s%s%s]",
478                                   pdo_fixed_voltage(pdo),
479                                   pdo_max_current(pdo),
480                                   (pdo & PDO_FIXED_DUAL_ROLE) ?
481                                                         "R" : "",
482                                   (pdo & PDO_FIXED_SUSPEND) ?
483                                                         "S" : "",
484                                   (pdo & PDO_FIXED_HIGHER_CAP) ?
485                                                         "H" : "",
486                                   (pdo & PDO_FIXED_USB_COMM) ?
487                                                         "U" : "",
488                                   (pdo & PDO_FIXED_DATA_SWAP) ?
489                                                         "D" : "",
490                                   (pdo & PDO_FIXED_EXTPOWER) ?
491                                                         "E" : "");
492                         break;
493                 case PDO_TYPE_VAR:
494                         scnprintf(msg, sizeof(msg),
495                                   "%u-%u mV, %u mA",
496                                   pdo_min_voltage(pdo),
497                                   pdo_max_voltage(pdo),
498                                   pdo_max_current(pdo));
499                         break;
500                 case PDO_TYPE_BATT:
501                         scnprintf(msg, sizeof(msg),
502                                   "%u-%u mV, %u mW",
503                                   pdo_min_voltage(pdo),
504                                   pdo_max_voltage(pdo),
505                                   pdo_max_power(pdo));
506                         break;
507                 default:
508                         strcpy(msg, "undefined");
509                         break;
510                 }
511                 tcpm_log(port, " PDO %d: type %d, %s",
512                          i, type, msg);
513         }
514 }
515
516 static int tcpm_seq_show(struct seq_file *s, void *v)
517 {
518         struct tcpm_port *port = (struct tcpm_port *)s->private;
519         int tail;
520
521         mutex_lock(&port->logbuffer_lock);
522         tail = port->logbuffer_tail;
523         while (tail != port->logbuffer_head) {
524                 seq_printf(s, "%s\n", port->logbuffer[tail]);
525                 tail = (tail + 1) % LOG_BUFFER_ENTRIES;
526         }
527         if (!seq_has_overflowed(s))
528                 port->logbuffer_tail = tail;
529         mutex_unlock(&port->logbuffer_lock);
530
531         return 0;
532 }
533
534 static int tcpm_debug_open(struct inode *inode, struct file *file)
535 {
536         return single_open(file, tcpm_seq_show, inode->i_private);
537 }
538
539 static const struct file_operations tcpm_debug_operations = {
540         .open           = tcpm_debug_open,
541         .llseek         = seq_lseek,
542         .read           = seq_read,
543         .release        = single_release,
544 };
545
546 static struct dentry *rootdir;
547
548 static int tcpm_debugfs_init(struct tcpm_port *port)
549 {
550         mutex_init(&port->logbuffer_lock);
551         /* /sys/kernel/debug/tcpm/usbcX */
552         if (!rootdir) {
553                 rootdir = debugfs_create_dir("tcpm", NULL);
554                 if (!rootdir)
555                         return -ENOMEM;
556         }
557
558         port->dentry = debugfs_create_file(dev_name(port->dev),
559                                            S_IFREG | 0444, rootdir,
560                                            port, &tcpm_debug_operations);
561
562         return 0;
563 }
564
565 static void tcpm_debugfs_exit(struct tcpm_port *port)
566 {
567         debugfs_remove(port->dentry);
568 }
569
570 #else
571
572 __printf(2, 3)
573 static void tcpm_log(const struct tcpm_port *port, const char *fmt, ...) { }
574 __printf(2, 3)
575 static void tcpm_log_force(struct tcpm_port *port, const char *fmt, ...) { }
576 static void tcpm_log_source_caps(struct tcpm_port *port) { }
577 static int tcpm_debugfs_init(const struct tcpm_port *port) { return 0; }
578 static void tcpm_debugfs_exit(const struct tcpm_port *port) { }
579
580 #endif
581
582 static int tcpm_pd_transmit(struct tcpm_port *port,
583                             enum tcpm_transmit_type type,
584                             const struct pd_message *msg)
585 {
586         unsigned long timeout;
587         int ret;
588
589         if (msg)
590                 tcpm_log(port, "PD TX, header: %#x", le16_to_cpu(msg->header));
591         else
592                 tcpm_log(port, "PD TX, type: %#x", type);
593
594         reinit_completion(&port->tx_complete);
595         ret = port->tcpc->pd_transmit(port->tcpc, type, msg);
596         if (ret < 0)
597                 return ret;
598
599         mutex_unlock(&port->lock);
600         timeout = wait_for_completion_timeout(&port->tx_complete,
601                                 msecs_to_jiffies(PD_T_TCPC_TX_TIMEOUT));
602         mutex_lock(&port->lock);
603         if (!timeout)
604                 return -ETIMEDOUT;
605
606         switch (port->tx_status) {
607         case TCPC_TX_SUCCESS:
608                 port->message_id = (port->message_id + 1) & PD_HEADER_ID_MASK;
609                 return 0;
610         case TCPC_TX_DISCARDED:
611                 return -EAGAIN;
612         case TCPC_TX_FAILED:
613         default:
614                 return -EIO;
615         }
616 }
617
618 void tcpm_pd_transmit_complete(struct tcpm_port *port,
619                                enum tcpm_transmit_status status)
620 {
621         tcpm_log(port, "PD TX complete, status: %u", status);
622         port->tx_status = status;
623         complete(&port->tx_complete);
624 }
625 EXPORT_SYMBOL_GPL(tcpm_pd_transmit_complete);
626
627 static int tcpm_mux_set(struct tcpm_port *port, enum tcpc_mux_mode mode,
628                         enum tcpc_usb_switch config)
629 {
630         int ret = 0;
631
632         tcpm_log(port, "Requesting mux mode %d, config %d, polarity %d",
633                  mode, config, port->polarity);
634
635         if (port->tcpc->mux)
636                 ret = port->tcpc->mux->set(port->tcpc->mux, mode, config,
637                                            port->polarity);
638
639         return ret;
640 }
641
642 static int tcpm_set_polarity(struct tcpm_port *port,
643                              enum typec_cc_polarity polarity)
644 {
645         int ret;
646
647         tcpm_log(port, "polarity %d", polarity);
648
649         ret = port->tcpc->set_polarity(port->tcpc, polarity);
650         if (ret < 0)
651                 return ret;
652
653         port->polarity = polarity;
654
655         return 0;
656 }
657
658 static int tcpm_set_vconn(struct tcpm_port *port, bool enable)
659 {
660         int ret;
661
662         tcpm_log(port, "vconn:=%d", enable);
663
664         ret = port->tcpc->set_vconn(port->tcpc, enable);
665         if (!ret) {
666                 port->vconn_role = enable ? TYPEC_SOURCE : TYPEC_SINK;
667                 typec_set_vconn_role(port->typec_port, port->vconn_role);
668         }
669
670         return ret;
671 }
672
673 static u32 tcpm_get_current_limit(struct tcpm_port *port)
674 {
675         enum typec_cc_status cc;
676         u32 limit;
677
678         cc = port->polarity ? port->cc2 : port->cc1;
679         switch (cc) {
680         case TYPEC_CC_RP_1_5:
681                 limit = 1500;
682                 break;
683         case TYPEC_CC_RP_3_0:
684                 limit = 3000;
685                 break;
686         case TYPEC_CC_RP_DEF:
687         default:
688                 if (port->tcpc->get_current_limit)
689                         limit = port->tcpc->get_current_limit(port->tcpc);
690                 else
691                         limit = 0;
692                 break;
693         }
694
695         return limit;
696 }
697
698 static int tcpm_set_current_limit(struct tcpm_port *port, u32 max_ma, u32 mv)
699 {
700         int ret = -EOPNOTSUPP;
701
702         tcpm_log(port, "Setting voltage/current limit %u mV %u mA", mv, max_ma);
703
704         if (port->tcpc->set_current_limit)
705                 ret = port->tcpc->set_current_limit(port->tcpc, max_ma, mv);
706
707         return ret;
708 }
709
710 /*
711  * Determine RP value to set based on maximum current supported
712  * by a port if configured as source.
713  * Returns CC value to report to link partner.
714  */
715 static enum typec_cc_status tcpm_rp_cc(struct tcpm_port *port)
716 {
717         const u32 *src_pdo = port->src_pdo;
718         int nr_pdo = port->nr_src_pdo;
719         int i;
720
721         /*
722          * Search for first entry with matching voltage.
723          * It should report the maximum supported current.
724          */
725         for (i = 0; i < nr_pdo; i++) {
726                 const u32 pdo = src_pdo[i];
727
728                 if (pdo_type(pdo) == PDO_TYPE_FIXED &&
729                     pdo_fixed_voltage(pdo) == 5000) {
730                         unsigned int curr = pdo_max_current(pdo);
731
732                         if (curr >= 3000)
733                                 return TYPEC_CC_RP_3_0;
734                         else if (curr >= 1500)
735                                 return TYPEC_CC_RP_1_5;
736                         return TYPEC_CC_RP_DEF;
737                 }
738         }
739
740         return TYPEC_CC_RP_DEF;
741 }
742
743 static int tcpm_set_attached_state(struct tcpm_port *port, bool attached)
744 {
745         return port->tcpc->set_roles(port->tcpc, attached, port->pwr_role,
746                                      port->data_role);
747 }
748
749 static int tcpm_set_roles(struct tcpm_port *port, bool attached,
750                           enum typec_role role, enum typec_data_role data)
751 {
752         int ret;
753
754         if (data == TYPEC_HOST)
755                 ret = tcpm_mux_set(port, TYPEC_MUX_USB,
756                                    TCPC_USB_SWITCH_CONNECT);
757         else
758                 ret = tcpm_mux_set(port, TYPEC_MUX_NONE,
759                                    TCPC_USB_SWITCH_DISCONNECT);
760         if (ret < 0)
761                 return ret;
762
763         ret = port->tcpc->set_roles(port->tcpc, attached, role, data);
764         if (ret < 0)
765                 return ret;
766
767         port->pwr_role = role;
768         port->data_role = data;
769         typec_set_data_role(port->typec_port, data);
770         typec_set_pwr_role(port->typec_port, role);
771
772         return 0;
773 }
774
775 static int tcpm_set_pwr_role(struct tcpm_port *port, enum typec_role role)
776 {
777         int ret;
778
779         ret = port->tcpc->set_roles(port->tcpc, true, role,
780                                     port->data_role);
781         if (ret < 0)
782                 return ret;
783
784         port->pwr_role = role;
785         typec_set_pwr_role(port->typec_port, role);
786
787         return 0;
788 }
789
790 static int tcpm_pd_send_source_caps(struct tcpm_port *port)
791 {
792         struct pd_message msg;
793         int i;
794
795         memset(&msg, 0, sizeof(msg));
796         if (!port->nr_src_pdo) {
797                 /* No source capabilities defined, sink only */
798                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
799                                           port->pwr_role,
800                                           port->data_role,
801                                           port->message_id, 0);
802         } else {
803                 msg.header = PD_HEADER_LE(PD_DATA_SOURCE_CAP,
804                                           port->pwr_role,
805                                           port->data_role,
806                                           port->message_id,
807                                           port->nr_src_pdo);
808         }
809         for (i = 0; i < port->nr_src_pdo; i++)
810                 msg.payload[i] = cpu_to_le32(port->src_pdo[i]);
811
812         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
813 }
814
815 static int tcpm_pd_send_sink_caps(struct tcpm_port *port)
816 {
817         struct pd_message msg;
818         int i;
819
820         memset(&msg, 0, sizeof(msg));
821         if (!port->nr_snk_pdo) {
822                 /* No sink capabilities defined, source only */
823                 msg.header = PD_HEADER_LE(PD_CTRL_REJECT,
824                                           port->pwr_role,
825                                           port->data_role,
826                                           port->message_id, 0);
827         } else {
828                 msg.header = PD_HEADER_LE(PD_DATA_SINK_CAP,
829                                           port->pwr_role,
830                                           port->data_role,
831                                           port->message_id,
832                                           port->nr_snk_pdo);
833         }
834         for (i = 0; i < port->nr_snk_pdo; i++)
835                 msg.payload[i] = cpu_to_le32(port->snk_pdo[i]);
836
837         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
838 }
839
840 static void tcpm_set_state(struct tcpm_port *port, enum tcpm_state state,
841                            unsigned int delay_ms)
842 {
843         if (delay_ms) {
844                 tcpm_log(port, "pending state change %s -> %s @ %u ms",
845                          tcpm_states[port->state], tcpm_states[state],
846                          delay_ms);
847                 port->delayed_state = state;
848                 mod_delayed_work(port->wq, &port->state_machine,
849                                  msecs_to_jiffies(delay_ms));
850                 port->delayed_runtime = jiffies + msecs_to_jiffies(delay_ms);
851                 port->delay_ms = delay_ms;
852         } else {
853                 tcpm_log(port, "state change %s -> %s",
854                          tcpm_states[port->state], tcpm_states[state]);
855                 port->delayed_state = INVALID_STATE;
856                 port->prev_state = port->state;
857                 port->state = state;
858                 /*
859                  * Don't re-queue the state machine work item if we're currently
860                  * in the state machine and we're immediately changing states.
861                  * tcpm_state_machine_work() will continue running the state
862                  * machine.
863                  */
864                 if (!port->state_machine_running)
865                         mod_delayed_work(port->wq, &port->state_machine, 0);
866         }
867 }
868
869 static void tcpm_set_state_cond(struct tcpm_port *port, enum tcpm_state state,
870                                 unsigned int delay_ms)
871 {
872         if (port->enter_state == port->state)
873                 tcpm_set_state(port, state, delay_ms);
874         else
875                 tcpm_log(port,
876                          "skipped %sstate change %s -> %s [%u ms], context state %s",
877                          delay_ms ? "delayed " : "",
878                          tcpm_states[port->state], tcpm_states[state],
879                          delay_ms, tcpm_states[port->enter_state]);
880 }
881
882 static void tcpm_queue_message(struct tcpm_port *port,
883                                enum pd_msg_request message)
884 {
885         port->queued_message = message;
886         mod_delayed_work(port->wq, &port->state_machine, 0);
887 }
888
889 /*
890  * VDM/VDO handling functions
891  */
892 static void tcpm_queue_vdm(struct tcpm_port *port, const u32 header,
893                            const u32 *data, int cnt)
894 {
895         port->vdo_count = cnt + 1;
896         port->vdo_data[0] = header;
897         memcpy(&port->vdo_data[1], data, sizeof(u32) * cnt);
898         /* Set ready, vdm state machine will actually send */
899         port->vdm_retries = 0;
900         port->vdm_state = VDM_STATE_READY;
901 }
902
903 static void svdm_consume_identity(struct tcpm_port *port, const __le32 *payload,
904                                   int cnt)
905 {
906         u32 vdo = le32_to_cpu(payload[VDO_INDEX_IDH]);
907         u32 product = le32_to_cpu(payload[VDO_INDEX_PRODUCT]);
908
909         memset(&port->mode_data, 0, sizeof(port->mode_data));
910
911 #if 0 /* Not really a match */
912         switch (PD_IDH_PTYPE(vdo)) {
913         case IDH_PTYPE_UNDEF:
914                 port->partner.type = TYPEC_PARTNER_NONE; /* no longer exists */
915                 break;
916         case IDH_PTYPE_HUB:
917                 break;
918         case IDH_PTYPE_PERIPH:
919                 break;
920         case IDH_PTYPE_PCABLE:
921                 break;
922         case IDH_PTYPE_ACABLE:
923                 break;
924         case IDH_PTYPE_AMA:
925                 port->partner.type = TYPEC_PARTNER_ALTMODE;
926                 break;
927         default:
928                 break;
929         }
930 #endif
931
932         port->partner_ident.id_header = vdo;
933         port->partner_ident.cert_stat = le32_to_cpu(payload[VDO_INDEX_CSTAT]);
934         port->partner_ident.product = product;
935
936         typec_partner_set_identity(port->partner);
937
938         tcpm_log(port, "Identity: %04x:%04x.%04x",
939                  PD_IDH_VID(vdo),
940                  PD_PRODUCT_PID(product), product & 0xffff);
941 }
942
943 static bool svdm_consume_svids(struct tcpm_port *port, const __le32 *payload,
944                                int cnt)
945 {
946         struct pd_mode_data *pmdata = &port->mode_data;
947         int i;
948
949         for (i = 1; i < cnt; i++) {
950                 u32 p = le32_to_cpu(payload[i]);
951                 u16 svid;
952
953                 svid = (p >> 16) & 0xffff;
954                 if (!svid)
955                         return false;
956
957                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
958                         goto abort;
959
960                 pmdata->svids[pmdata->nsvids++] = svid;
961                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
962
963                 svid = p & 0xffff;
964                 if (!svid)
965                         return false;
966
967                 if (pmdata->nsvids >= SVID_DISCOVERY_MAX)
968                         goto abort;
969
970                 pmdata->svids[pmdata->nsvids++] = svid;
971                 tcpm_log(port, "SVID %d: 0x%x", pmdata->nsvids, svid);
972         }
973         return true;
974 abort:
975         tcpm_log(port, "SVID_DISCOVERY_MAX(%d) too low!", SVID_DISCOVERY_MAX);
976         return false;
977 }
978
979 static void svdm_consume_modes(struct tcpm_port *port, const __le32 *payload,
980                                int cnt)
981 {
982         struct pd_mode_data *pmdata = &port->mode_data;
983         struct typec_altmode_desc *paltmode;
984         struct typec_mode_desc *pmode;
985         int i;
986
987         if (pmdata->altmodes >= ARRAY_SIZE(port->partner_altmode)) {
988                 /* Already logged in svdm_consume_svids() */
989                 return;
990         }
991
992         paltmode = &pmdata->altmode_desc[pmdata->altmodes];
993         memset(paltmode, 0, sizeof(*paltmode));
994
995         paltmode->svid = pmdata->svids[pmdata->svid_index];
996
997         tcpm_log(port, " Alternate mode %d: SVID 0x%04x",
998                  pmdata->altmodes, paltmode->svid);
999
1000         for (i = 1; i < cnt && paltmode->n_modes < ALTMODE_MAX_MODES; i++) {
1001                 pmode = &paltmode->modes[paltmode->n_modes];
1002                 memset(pmode, 0, sizeof(*pmode));
1003                 pmode->vdo = le32_to_cpu(payload[i]);
1004                 pmode->index = i - 1;
1005                 paltmode->n_modes++;
1006                 tcpm_log(port, "  VDO %d: 0x%08x",
1007                          pmode->index, pmode->vdo);
1008         }
1009         port->partner_altmode[pmdata->altmodes] =
1010                 typec_partner_register_altmode(port->partner, paltmode);
1011         if (port->partner_altmode[pmdata->altmodes] == NULL) {
1012                 tcpm_log(port,
1013                          "Failed to register alternate modes for SVID 0x%04x",
1014                          paltmode->svid);
1015                 return;
1016         }
1017         pmdata->altmodes++;
1018 }
1019
1020 #define supports_modal(port)    PD_IDH_MODAL_SUPP((port)->partner_ident.id_header)
1021
1022 static int tcpm_pd_svdm(struct tcpm_port *port, const __le32 *payload, int cnt,
1023                         u32 *response)
1024 {
1025         u32 p0 = le32_to_cpu(payload[0]);
1026         int cmd_type = PD_VDO_CMDT(p0);
1027         int cmd = PD_VDO_CMD(p0);
1028         struct pd_mode_data *modep;
1029         int rlen = 0;
1030         u16 svid;
1031         int i;
1032
1033         tcpm_log(port, "Rx VDM cmd 0x%x type %d cmd %d len %d",
1034                  p0, cmd_type, cmd, cnt);
1035
1036         modep = &port->mode_data;
1037
1038         switch (cmd_type) {
1039         case CMDT_INIT:
1040                 switch (cmd) {
1041                 case CMD_DISCOVER_IDENT:
1042                         /* 6.4.4.3.1: Only respond as UFP (device) */
1043                         if (port->data_role == TYPEC_DEVICE &&
1044                             port->nr_snk_vdo) {
1045                                 for (i = 0; i <  port->nr_snk_vdo; i++)
1046                                         response[i + 1] = port->snk_vdo[i];
1047                                 rlen = port->nr_snk_vdo + 1;
1048                         }
1049                         break;
1050                 case CMD_DISCOVER_SVID:
1051                         break;
1052                 case CMD_DISCOVER_MODES:
1053                         break;
1054                 case CMD_ENTER_MODE:
1055                         break;
1056                 case CMD_EXIT_MODE:
1057                         break;
1058                 case CMD_ATTENTION:
1059                         break;
1060                 default:
1061                         break;
1062                 }
1063                 if (rlen >= 1) {
1064                         response[0] = p0 | VDO_CMDT(CMDT_RSP_ACK);
1065                 } else if (rlen == 0) {
1066                         response[0] = p0 | VDO_CMDT(CMDT_RSP_NAK);
1067                         rlen = 1;
1068                 } else {
1069                         response[0] = p0 | VDO_CMDT(CMDT_RSP_BUSY);
1070                         rlen = 1;
1071                 }
1072                 break;
1073         case CMDT_RSP_ACK:
1074                 /* silently drop message if we are not connected */
1075                 if (!port->partner)
1076                         break;
1077
1078                 switch (cmd) {
1079                 case CMD_DISCOVER_IDENT:
1080                         /* 6.4.4.3.1 */
1081                         svdm_consume_identity(port, payload, cnt);
1082                         response[0] = VDO(USB_SID_PD, 1, CMD_DISCOVER_SVID);
1083                         rlen = 1;
1084                         break;
1085                 case CMD_DISCOVER_SVID:
1086                         /* 6.4.4.3.2 */
1087                         if (svdm_consume_svids(port, payload, cnt)) {
1088                                 response[0] = VDO(USB_SID_PD, 1,
1089                                                   CMD_DISCOVER_SVID);
1090                                 rlen = 1;
1091                         } else if (modep->nsvids && supports_modal(port)) {
1092                                 response[0] = VDO(modep->svids[0], 1,
1093                                                   CMD_DISCOVER_MODES);
1094                                 rlen = 1;
1095                         }
1096                         break;
1097                 case CMD_DISCOVER_MODES:
1098                         /* 6.4.4.3.3 */
1099                         svdm_consume_modes(port, payload, cnt);
1100                         modep->svid_index++;
1101                         if (modep->svid_index < modep->nsvids) {
1102                                 svid = modep->svids[modep->svid_index];
1103                                 response[0] = VDO(svid, 1, CMD_DISCOVER_MODES);
1104                                 rlen = 1;
1105                         } else {
1106 #if 0
1107                                 response[0] = pd_dfp_enter_mode(port, 0, 0);
1108                                 if (response[0])
1109                                         rlen = 1;
1110 #endif
1111                         }
1112                         break;
1113                 case CMD_ENTER_MODE:
1114                         break;
1115                 default:
1116                         break;
1117                 }
1118                 break;
1119         default:
1120                 break;
1121         }
1122
1123         return rlen;
1124 }
1125
1126 static void tcpm_handle_vdm_request(struct tcpm_port *port,
1127                                     const __le32 *payload, int cnt)
1128 {
1129         int rlen = 0;
1130         u32 response[8] = { };
1131         u32 p0 = le32_to_cpu(payload[0]);
1132
1133         if (port->vdm_state == VDM_STATE_BUSY) {
1134                 /* If UFP responded busy retry after timeout */
1135                 if (PD_VDO_CMDT(p0) == CMDT_RSP_BUSY) {
1136                         port->vdm_state = VDM_STATE_WAIT_RSP_BUSY;
1137                         port->vdo_retry = (p0 & ~VDO_CMDT_MASK) |
1138                                 CMDT_INIT;
1139                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1140                                          msecs_to_jiffies(PD_T_VDM_BUSY));
1141                         return;
1142                 }
1143                 port->vdm_state = VDM_STATE_DONE;
1144         }
1145
1146         if (PD_VDO_SVDM(p0))
1147                 rlen = tcpm_pd_svdm(port, payload, cnt, response);
1148 #if 0
1149         else
1150                 rlen = tcpm_pd_custom_vdm(port, cnt, payload, response);
1151 #endif
1152
1153         if (rlen > 0) {
1154                 tcpm_queue_vdm(port, response[0], &response[1], rlen - 1);
1155                 mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1156         }
1157 }
1158
1159 static void tcpm_send_vdm(struct tcpm_port *port, u32 vid, int cmd,
1160                           const u32 *data, int count)
1161 {
1162         u32 header;
1163
1164         if (WARN_ON(count > VDO_MAX_SIZE - 1))
1165                 count = VDO_MAX_SIZE - 1;
1166
1167         /* set VDM header with VID & CMD */
1168         header = VDO(vid, ((vid & USB_SID_PD) == USB_SID_PD) ?
1169                         1 : (PD_VDO_CMD(cmd) <= CMD_ATTENTION), cmd);
1170         tcpm_queue_vdm(port, header, data, count);
1171
1172         mod_delayed_work(port->wq, &port->vdm_state_machine, 0);
1173 }
1174
1175 static unsigned int vdm_ready_timeout(u32 vdm_hdr)
1176 {
1177         unsigned int timeout;
1178         int cmd = PD_VDO_CMD(vdm_hdr);
1179
1180         /* its not a structured VDM command */
1181         if (!PD_VDO_SVDM(vdm_hdr))
1182                 return PD_T_VDM_UNSTRUCTURED;
1183
1184         switch (PD_VDO_CMDT(vdm_hdr)) {
1185         case CMDT_INIT:
1186                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1187                         timeout = PD_T_VDM_WAIT_MODE_E;
1188                 else
1189                         timeout = PD_T_VDM_SNDR_RSP;
1190                 break;
1191         default:
1192                 if (cmd == CMD_ENTER_MODE || cmd == CMD_EXIT_MODE)
1193                         timeout = PD_T_VDM_E_MODE;
1194                 else
1195                         timeout = PD_T_VDM_RCVR_RSP;
1196                 break;
1197         }
1198         return timeout;
1199 }
1200
1201 static void vdm_run_state_machine(struct tcpm_port *port)
1202 {
1203         struct pd_message msg;
1204         int i, res;
1205
1206         switch (port->vdm_state) {
1207         case VDM_STATE_READY:
1208                 /* Only transmit VDM if attached */
1209                 if (!port->attached) {
1210                         port->vdm_state = VDM_STATE_ERR_BUSY;
1211                         break;
1212                 }
1213
1214                 /*
1215                  * if there's traffic or we're not in PDO ready state don't send
1216                  * a VDM.
1217                  */
1218                 if (port->state != SRC_READY && port->state != SNK_READY)
1219                         break;
1220
1221                 /* Prepare and send VDM */
1222                 memset(&msg, 0, sizeof(msg));
1223                 msg.header = PD_HEADER_LE(PD_DATA_VENDOR_DEF,
1224                                           port->pwr_role,
1225                                           port->data_role,
1226                                           port->message_id, port->vdo_count);
1227                 for (i = 0; i < port->vdo_count; i++)
1228                         msg.payload[i] = cpu_to_le32(port->vdo_data[i]);
1229                 res = tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1230                 if (res < 0) {
1231                         port->vdm_state = VDM_STATE_ERR_SEND;
1232                 } else {
1233                         unsigned long timeout;
1234
1235                         port->vdm_retries = 0;
1236                         port->vdm_state = VDM_STATE_BUSY;
1237                         timeout = vdm_ready_timeout(port->vdo_data[0]);
1238                         mod_delayed_work(port->wq, &port->vdm_state_machine,
1239                                          timeout);
1240                 }
1241                 break;
1242         case VDM_STATE_WAIT_RSP_BUSY:
1243                 port->vdo_data[0] = port->vdo_retry;
1244                 port->vdo_count = 1;
1245                 port->vdm_state = VDM_STATE_READY;
1246                 break;
1247         case VDM_STATE_BUSY:
1248                 port->vdm_state = VDM_STATE_ERR_TMOUT;
1249                 break;
1250         case VDM_STATE_ERR_SEND:
1251                 /*
1252                  * A partner which does not support USB PD will not reply,
1253                  * so this is not a fatal error. At the same time, some
1254                  * devices may not return GoodCRC under some circumstances,
1255                  * so we need to retry.
1256                  */
1257                 if (port->vdm_retries < 3) {
1258                         tcpm_log(port, "VDM Tx error, retry");
1259                         port->vdm_retries++;
1260                         port->vdm_state = VDM_STATE_READY;
1261                 }
1262                 break;
1263         default:
1264                 break;
1265         }
1266 }
1267
1268 static void vdm_state_machine_work(struct work_struct *work)
1269 {
1270         struct tcpm_port *port = container_of(work, struct tcpm_port,
1271                                               vdm_state_machine.work);
1272         enum vdm_states prev_state;
1273
1274         mutex_lock(&port->lock);
1275
1276         /*
1277          * Continue running as long as the port is not busy and there was
1278          * a state change.
1279          */
1280         do {
1281                 prev_state = port->vdm_state;
1282                 vdm_run_state_machine(port);
1283         } while (port->vdm_state != prev_state &&
1284                  port->vdm_state != VDM_STATE_BUSY);
1285
1286         mutex_unlock(&port->lock);
1287 }
1288
1289 /*
1290  * PD (data, control) command handling functions
1291  */
1292 static void tcpm_pd_data_request(struct tcpm_port *port,
1293                                  const struct pd_message *msg)
1294 {
1295         enum pd_data_msg_type type = pd_header_type_le(msg->header);
1296         unsigned int cnt = pd_header_cnt_le(msg->header);
1297         unsigned int i;
1298
1299         switch (type) {
1300         case PD_DATA_SOURCE_CAP:
1301                 if (port->pwr_role != TYPEC_SINK)
1302                         break;
1303
1304                 for (i = 0; i < cnt; i++)
1305                         port->source_caps[i] = le32_to_cpu(msg->payload[i]);
1306
1307                 port->nr_source_caps = cnt;
1308
1309                 tcpm_log_source_caps(port);
1310
1311                 /*
1312                  * This message may be received even if VBUS is not
1313                  * present. This is quite unexpected; see USB PD
1314                  * specification, sections 8.3.3.6.3.1 and 8.3.3.6.3.2.
1315                  * However, at the same time, we must be ready to
1316                  * receive this message and respond to it 15ms after
1317                  * receiving PS_RDY during power swap operations, no matter
1318                  * if VBUS is available or not (USB PD specification,
1319                  * section 6.5.9.2).
1320                  * So we need to accept the message either way,
1321                  * but be prepared to keep waiting for VBUS after it was
1322                  * handled.
1323                  */
1324                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
1325                 break;
1326         case PD_DATA_REQUEST:
1327                 if (port->pwr_role != TYPEC_SOURCE ||
1328                     cnt != 1) {
1329                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1330                         break;
1331                 }
1332                 port->sink_request = le32_to_cpu(msg->payload[0]);
1333                 tcpm_set_state(port, SRC_NEGOTIATE_CAPABILITIES, 0);
1334                 break;
1335         case PD_DATA_SINK_CAP:
1336                 /* We don't do anything with this at the moment... */
1337                 for (i = 0; i < cnt; i++)
1338                         port->sink_caps[i] = le32_to_cpu(msg->payload[i]);
1339                 port->nr_sink_caps = cnt;
1340                 break;
1341         case PD_DATA_VENDOR_DEF:
1342                 tcpm_handle_vdm_request(port, msg->payload, cnt);
1343                 break;
1344         case PD_DATA_BIST:
1345                 if (port->state == SRC_READY || port->state == SNK_READY) {
1346                         port->bist_request = le32_to_cpu(msg->payload[0]);
1347                         tcpm_set_state(port, BIST_RX, 0);
1348                 }
1349                 break;
1350         default:
1351                 tcpm_log(port, "Unhandled data message type %#x", type);
1352                 break;
1353         }
1354 }
1355
1356 static void tcpm_pd_ctrl_request(struct tcpm_port *port,
1357                                  const struct pd_message *msg)
1358 {
1359         enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1360         enum tcpm_state next_state;
1361
1362         switch (type) {
1363         case PD_CTRL_GOOD_CRC:
1364         case PD_CTRL_PING:
1365                 break;
1366         case PD_CTRL_GET_SOURCE_CAP:
1367                 switch (port->state) {
1368                 case SRC_READY:
1369                 case SNK_READY:
1370                         tcpm_queue_message(port, PD_MSG_DATA_SOURCE_CAP);
1371                         break;
1372                 default:
1373                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1374                         break;
1375                 }
1376                 break;
1377         case PD_CTRL_GET_SINK_CAP:
1378                 switch (port->state) {
1379                 case SRC_READY:
1380                 case SNK_READY:
1381                         tcpm_queue_message(port, PD_MSG_DATA_SINK_CAP);
1382                         break;
1383                 default:
1384                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1385                         break;
1386                 }
1387                 break;
1388         case PD_CTRL_GOTO_MIN:
1389                 break;
1390         case PD_CTRL_PS_RDY:
1391                 switch (port->state) {
1392                 case SNK_TRANSITION_SINK:
1393                         if (port->vbus_present) {
1394                                 tcpm_set_current_limit(port,
1395                                                        port->current_limit,
1396                                                        port->supply_voltage);
1397                                 port->explicit_contract = true;
1398                                 tcpm_set_state(port, SNK_READY, 0);
1399                         } else {
1400                                 /*
1401                                  * Seen after power swap. Keep waiting for VBUS
1402                                  * in a transitional state.
1403                                  */
1404                                 tcpm_set_state(port,
1405                                                SNK_TRANSITION_SINK_VBUS, 0);
1406                         }
1407                         break;
1408                 case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
1409                         tcpm_set_state(port, PR_SWAP_SRC_SNK_SINK_ON, 0);
1410                         break;
1411                 case PR_SWAP_SNK_SRC_SINK_OFF:
1412                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON, 0);
1413                         break;
1414                 case VCONN_SWAP_WAIT_FOR_VCONN:
1415                         tcpm_set_state(port, VCONN_SWAP_TURN_OFF_VCONN, 0);
1416                         break;
1417                 default:
1418                         break;
1419                 }
1420                 break;
1421         case PD_CTRL_REJECT:
1422         case PD_CTRL_WAIT:
1423                 switch (port->state) {
1424                 case SNK_NEGOTIATE_CAPABILITIES:
1425                         /* USB PD specification, Figure 8-43 */
1426                         if (port->explicit_contract)
1427                                 next_state = SNK_READY;
1428                         else
1429                                 next_state = SNK_WAIT_CAPABILITIES;
1430                         tcpm_set_state(port, next_state, 0);
1431                         break;
1432                 case DR_SWAP_SEND:
1433                         port->swap_status = (type == PD_CTRL_WAIT ?
1434                                              -EAGAIN : -EOPNOTSUPP);
1435                         tcpm_set_state(port, DR_SWAP_CANCEL, 0);
1436                         break;
1437                 case PR_SWAP_SEND:
1438                         port->swap_status = (type == PD_CTRL_WAIT ?
1439                                              -EAGAIN : -EOPNOTSUPP);
1440                         tcpm_set_state(port, PR_SWAP_CANCEL, 0);
1441                         break;
1442                 case VCONN_SWAP_SEND:
1443                         port->swap_status = (type == PD_CTRL_WAIT ?
1444                                              -EAGAIN : -EOPNOTSUPP);
1445                         tcpm_set_state(port, VCONN_SWAP_CANCEL, 0);
1446                         break;
1447                 default:
1448                         break;
1449                 }
1450                 break;
1451         case PD_CTRL_ACCEPT:
1452                 switch (port->state) {
1453                 case SNK_NEGOTIATE_CAPABILITIES:
1454                         tcpm_set_state(port, SNK_TRANSITION_SINK, 0);
1455                         break;
1456                 case SOFT_RESET_SEND:
1457                         port->message_id = 0;
1458                         port->rx_msgid = -1;
1459                         if (port->pwr_role == TYPEC_SOURCE)
1460                                 next_state = SRC_SEND_CAPABILITIES;
1461                         else
1462                                 next_state = SNK_WAIT_CAPABILITIES;
1463                         tcpm_set_state(port, next_state, 0);
1464                         break;
1465                 case DR_SWAP_SEND:
1466                         tcpm_set_state(port, DR_SWAP_CHANGE_DR, 0);
1467                         break;
1468                 case PR_SWAP_SEND:
1469                         tcpm_set_state(port, PR_SWAP_START, 0);
1470                         break;
1471                 case VCONN_SWAP_SEND:
1472                         tcpm_set_state(port, VCONN_SWAP_START, 0);
1473                         break;
1474                 default:
1475                         break;
1476                 }
1477                 break;
1478         case PD_CTRL_SOFT_RESET:
1479                 tcpm_set_state(port, SOFT_RESET, 0);
1480                 break;
1481         case PD_CTRL_DR_SWAP:
1482                 if (port->port_type != TYPEC_PORT_DRP) {
1483                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1484                         break;
1485                 }
1486                 /*
1487                  * XXX
1488                  * 6.3.9: If an alternate mode is active, a request to swap
1489                  * alternate modes shall trigger a port reset.
1490                  */
1491                 switch (port->state) {
1492                 case SRC_READY:
1493                 case SNK_READY:
1494                         tcpm_set_state(port, DR_SWAP_ACCEPT, 0);
1495                         break;
1496                 default:
1497                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1498                         break;
1499                 }
1500                 break;
1501         case PD_CTRL_PR_SWAP:
1502                 if (port->port_type != TYPEC_PORT_DRP) {
1503                         tcpm_queue_message(port, PD_MSG_CTRL_REJECT);
1504                         break;
1505                 }
1506                 switch (port->state) {
1507                 case SRC_READY:
1508                 case SNK_READY:
1509                         tcpm_set_state(port, PR_SWAP_ACCEPT, 0);
1510                         break;
1511                 default:
1512                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1513                         break;
1514                 }
1515                 break;
1516         case PD_CTRL_VCONN_SWAP:
1517                 switch (port->state) {
1518                 case SRC_READY:
1519                 case SNK_READY:
1520                         tcpm_set_state(port, VCONN_SWAP_ACCEPT, 0);
1521                         break;
1522                 default:
1523                         tcpm_queue_message(port, PD_MSG_CTRL_WAIT);
1524                         break;
1525                 }
1526                 break;
1527         default:
1528                 tcpm_log(port, "Unhandled ctrl message type %#x", type);
1529                 break;
1530         }
1531 }
1532
1533 static void tcpm_pd_rx_handler(struct work_struct *work)
1534 {
1535         struct pd_rx_event *event = container_of(work,
1536                                                  struct pd_rx_event, work);
1537         const struct pd_message *msg = &event->msg;
1538         unsigned int cnt = pd_header_cnt_le(msg->header);
1539         struct tcpm_port *port = event->port;
1540
1541         mutex_lock(&port->lock);
1542
1543         tcpm_log(port, "PD RX, header: %#x [%d]", le16_to_cpu(msg->header),
1544                  port->attached);
1545
1546         if (port->attached) {
1547                 enum pd_ctrl_msg_type type = pd_header_type_le(msg->header);
1548                 unsigned int msgid = pd_header_msgid_le(msg->header);
1549
1550                 /*
1551                  * USB PD standard, 6.6.1.2:
1552                  * "... if MessageID value in a received Message is the
1553                  * same as the stored value, the receiver shall return a
1554                  * GoodCRC Message with that MessageID value and drop
1555                  * the Message (this is a retry of an already received
1556                  * Message). Note: this shall not apply to the Soft_Reset
1557                  * Message which always has a MessageID value of zero."
1558                  */
1559                 if (msgid == port->rx_msgid && type != PD_CTRL_SOFT_RESET)
1560                         goto done;
1561                 port->rx_msgid = msgid;
1562
1563                 /*
1564                  * If both ends believe to be DFP/host, we have a data role
1565                  * mismatch.
1566                  */
1567                 if (!!(le16_to_cpu(msg->header) & PD_HEADER_DATA_ROLE) ==
1568                     (port->data_role == TYPEC_HOST)) {
1569                         tcpm_log(port,
1570                                  "Data role mismatch, initiating error recovery");
1571                         tcpm_set_state(port, ERROR_RECOVERY, 0);
1572                 } else {
1573                         if (cnt)
1574                                 tcpm_pd_data_request(port, msg);
1575                         else
1576                                 tcpm_pd_ctrl_request(port, msg);
1577                 }
1578         }
1579
1580 done:
1581         mutex_unlock(&port->lock);
1582         kfree(event);
1583 }
1584
1585 void tcpm_pd_receive(struct tcpm_port *port, const struct pd_message *msg)
1586 {
1587         struct pd_rx_event *event;
1588
1589         event = kzalloc(sizeof(*event), GFP_ATOMIC);
1590         if (!event)
1591                 return;
1592
1593         INIT_WORK(&event->work, tcpm_pd_rx_handler);
1594         event->port = port;
1595         memcpy(&event->msg, msg, sizeof(*msg));
1596         queue_work(port->wq, &event->work);
1597 }
1598 EXPORT_SYMBOL_GPL(tcpm_pd_receive);
1599
1600 static int tcpm_pd_send_control(struct tcpm_port *port,
1601                                 enum pd_ctrl_msg_type type)
1602 {
1603         struct pd_message msg;
1604
1605         memset(&msg, 0, sizeof(msg));
1606         msg.header = PD_HEADER_LE(type, port->pwr_role,
1607                                   port->data_role,
1608                                   port->message_id, 0);
1609
1610         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1611 }
1612
1613 /*
1614  * Send queued message without affecting state.
1615  * Return true if state machine should go back to sleep,
1616  * false otherwise.
1617  */
1618 static bool tcpm_send_queued_message(struct tcpm_port *port)
1619 {
1620         enum pd_msg_request queued_message;
1621
1622         do {
1623                 queued_message = port->queued_message;
1624                 port->queued_message = PD_MSG_NONE;
1625
1626                 switch (queued_message) {
1627                 case PD_MSG_CTRL_WAIT:
1628                         tcpm_pd_send_control(port, PD_CTRL_WAIT);
1629                         break;
1630                 case PD_MSG_CTRL_REJECT:
1631                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
1632                         break;
1633                 case PD_MSG_DATA_SINK_CAP:
1634                         tcpm_pd_send_sink_caps(port);
1635                         break;
1636                 case PD_MSG_DATA_SOURCE_CAP:
1637                         tcpm_pd_send_source_caps(port);
1638                         break;
1639                 default:
1640                         break;
1641                 }
1642         } while (port->queued_message != PD_MSG_NONE);
1643
1644         if (port->delayed_state != INVALID_STATE) {
1645                 if (time_is_after_jiffies(port->delayed_runtime)) {
1646                         mod_delayed_work(port->wq, &port->state_machine,
1647                                          port->delayed_runtime - jiffies);
1648                         return true;
1649                 }
1650                 port->delayed_state = INVALID_STATE;
1651         }
1652         return false;
1653 }
1654
1655 static int tcpm_pd_check_request(struct tcpm_port *port)
1656 {
1657         u32 pdo, rdo = port->sink_request;
1658         unsigned int max, op, pdo_max, index;
1659         enum pd_pdo_type type;
1660
1661         index = rdo_index(rdo);
1662         if (!index || index > port->nr_src_pdo)
1663                 return -EINVAL;
1664
1665         pdo = port->src_pdo[index - 1];
1666         type = pdo_type(pdo);
1667         switch (type) {
1668         case PDO_TYPE_FIXED:
1669         case PDO_TYPE_VAR:
1670                 max = rdo_max_current(rdo);
1671                 op = rdo_op_current(rdo);
1672                 pdo_max = pdo_max_current(pdo);
1673
1674                 if (op > pdo_max)
1675                         return -EINVAL;
1676                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1677                         return -EINVAL;
1678
1679                 if (type == PDO_TYPE_FIXED)
1680                         tcpm_log(port,
1681                                  "Requested %u mV, %u mA for %u / %u mA",
1682                                  pdo_fixed_voltage(pdo), pdo_max, op, max);
1683                 else
1684                         tcpm_log(port,
1685                                  "Requested %u -> %u mV, %u mA for %u / %u mA",
1686                                  pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1687                                  pdo_max, op, max);
1688                 break;
1689         case PDO_TYPE_BATT:
1690                 max = rdo_max_power(rdo);
1691                 op = rdo_op_power(rdo);
1692                 pdo_max = pdo_max_power(pdo);
1693
1694                 if (op > pdo_max)
1695                         return -EINVAL;
1696                 if (max > pdo_max && !(rdo & RDO_CAP_MISMATCH))
1697                         return -EINVAL;
1698                 tcpm_log(port,
1699                          "Requested %u -> %u mV, %u mW for %u / %u mW",
1700                          pdo_min_voltage(pdo), pdo_max_voltage(pdo),
1701                          pdo_max, op, max);
1702                 break;
1703         default:
1704                 return -EINVAL;
1705         }
1706
1707         port->op_vsafe5v = index == 1;
1708
1709         return 0;
1710 }
1711
1712 static int tcpm_pd_select_pdo(struct tcpm_port *port)
1713 {
1714         unsigned int i, max_mw = 0, max_mv = 0;
1715         int ret = -EINVAL;
1716
1717         /*
1718          * Select the source PDO providing the most power while staying within
1719          * the board's voltage limits. Prefer PDO providing exp
1720          */
1721         for (i = 0; i < port->nr_source_caps; i++) {
1722                 u32 pdo = port->source_caps[i];
1723                 enum pd_pdo_type type = pdo_type(pdo);
1724                 unsigned int mv, ma, mw;
1725
1726                 if (type == PDO_TYPE_FIXED)
1727                         mv = pdo_fixed_voltage(pdo);
1728                 else
1729                         mv = pdo_min_voltage(pdo);
1730
1731                 if (type == PDO_TYPE_BATT) {
1732                         mw = pdo_max_power(pdo);
1733                 } else {
1734                         ma = min(pdo_max_current(pdo),
1735                                  port->max_snk_ma);
1736                         mw = ma * mv / 1000;
1737                 }
1738
1739                 /* Perfer higher voltages if available */
1740                 if ((mw > max_mw || (mw == max_mw && mv > max_mv)) &&
1741                     mv <= port->max_snk_mv) {
1742                         ret = i;
1743                         max_mw = mw;
1744                         max_mv = mv;
1745                 }
1746         }
1747
1748         return ret;
1749 }
1750
1751 static int tcpm_pd_build_request(struct tcpm_port *port, u32 *rdo)
1752 {
1753         unsigned int mv, ma, mw, flags;
1754         unsigned int max_ma, max_mw;
1755         enum pd_pdo_type type;
1756         int index;
1757         u32 pdo;
1758
1759         index = tcpm_pd_select_pdo(port);
1760         if (index < 0)
1761                 return -EINVAL;
1762         pdo = port->source_caps[index];
1763         type = pdo_type(pdo);
1764
1765         if (type == PDO_TYPE_FIXED)
1766                 mv = pdo_fixed_voltage(pdo);
1767         else
1768                 mv = pdo_min_voltage(pdo);
1769
1770         /* Select maximum available current within the board's power limit */
1771         if (type == PDO_TYPE_BATT) {
1772                 mw = pdo_max_power(pdo);
1773                 ma = 1000 * min(mw, port->max_snk_mw) / mv;
1774         } else {
1775                 ma = min(pdo_max_current(pdo),
1776                          1000 * port->max_snk_mw / mv);
1777         }
1778         ma = min(ma, port->max_snk_ma);
1779
1780         flags = RDO_USB_COMM | RDO_NO_SUSPEND;
1781
1782         /* Set mismatch bit if offered power is less than operating power */
1783         mw = ma * mv / 1000;
1784         max_ma = ma;
1785         max_mw = mw;
1786         if (mw < port->operating_snk_mw) {
1787                 flags |= RDO_CAP_MISMATCH;
1788                 max_mw = port->operating_snk_mw;
1789                 max_ma = max_mw * 1000 / mv;
1790         }
1791
1792         tcpm_log(port, "cc=%d cc1=%d cc2=%d vbus=%d vconn=%s polarity=%d",
1793                  port->cc_req, port->cc1, port->cc2, port->vbus_source,
1794                  port->vconn_role == TYPEC_SOURCE ? "source" : "sink",
1795                  port->polarity);
1796
1797         if (type == PDO_TYPE_BATT) {
1798                 *rdo = RDO_BATT(index + 1, mw, max_mw, flags);
1799
1800                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mW%s",
1801                          index, mv, mw,
1802                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
1803         } else {
1804                 *rdo = RDO_FIXED(index + 1, ma, max_ma, flags);
1805
1806                 tcpm_log(port, "Requesting PDO %d: %u mV, %u mA%s",
1807                          index, mv, ma,
1808                          flags & RDO_CAP_MISMATCH ? " [mismatch]" : "");
1809         }
1810
1811         port->current_limit = ma;
1812         port->supply_voltage = mv;
1813
1814         return 0;
1815 }
1816
1817 static int tcpm_pd_send_request(struct tcpm_port *port)
1818 {
1819         struct pd_message msg;
1820         int ret;
1821         u32 rdo;
1822
1823         ret = tcpm_pd_build_request(port, &rdo);
1824         if (ret < 0)
1825                 return ret;
1826
1827         memset(&msg, 0, sizeof(msg));
1828         msg.header = PD_HEADER_LE(PD_DATA_REQUEST,
1829                                   port->pwr_role,
1830                                   port->data_role,
1831                                   port->message_id, 1);
1832         msg.payload[0] = cpu_to_le32(rdo);
1833
1834         return tcpm_pd_transmit(port, TCPC_TX_SOP, &msg);
1835 }
1836
1837 static int tcpm_set_vbus(struct tcpm_port *port, bool enable)
1838 {
1839         int ret;
1840
1841         if (enable && port->vbus_charge)
1842                 return -EINVAL;
1843
1844         tcpm_log(port, "vbus:=%d charge=%d", enable, port->vbus_charge);
1845
1846         ret = port->tcpc->set_vbus(port->tcpc, enable, port->vbus_charge);
1847         if (ret < 0)
1848                 return ret;
1849
1850         port->vbus_source = enable;
1851         return 0;
1852 }
1853
1854 static int tcpm_set_charge(struct tcpm_port *port, bool charge)
1855 {
1856         int ret;
1857
1858         if (charge && port->vbus_source)
1859                 return -EINVAL;
1860
1861         if (charge != port->vbus_charge) {
1862                 tcpm_log(port, "vbus=%d charge:=%d", port->vbus_source, charge);
1863                 ret = port->tcpc->set_vbus(port->tcpc, port->vbus_source,
1864                                            charge);
1865                 if (ret < 0)
1866                         return ret;
1867         }
1868         port->vbus_charge = charge;
1869         return 0;
1870 }
1871
1872 static bool tcpm_start_drp_toggling(struct tcpm_port *port)
1873 {
1874         int ret;
1875
1876         if (port->tcpc->start_drp_toggling &&
1877             port->port_type == TYPEC_PORT_DRP) {
1878                 tcpm_log_force(port, "Start DRP toggling");
1879                 ret = port->tcpc->start_drp_toggling(port->tcpc,
1880                                                      tcpm_rp_cc(port));
1881                 if (!ret)
1882                         return true;
1883         }
1884
1885         return false;
1886 }
1887
1888 static void tcpm_set_cc(struct tcpm_port *port, enum typec_cc_status cc)
1889 {
1890         tcpm_log(port, "cc:=%d", cc);
1891         port->cc_req = cc;
1892         port->tcpc->set_cc(port->tcpc, cc);
1893 }
1894
1895 static int tcpm_init_vbus(struct tcpm_port *port)
1896 {
1897         int ret;
1898
1899         ret = port->tcpc->set_vbus(port->tcpc, false, false);
1900         port->vbus_source = false;
1901         port->vbus_charge = false;
1902         return ret;
1903 }
1904
1905 static int tcpm_init_vconn(struct tcpm_port *port)
1906 {
1907         int ret;
1908
1909         ret = port->tcpc->set_vconn(port->tcpc, false);
1910         port->vconn_role = TYPEC_SINK;
1911         return ret;
1912 }
1913
1914 static void tcpm_typec_connect(struct tcpm_port *port)
1915 {
1916         if (!port->connected) {
1917                 /* Make sure we don't report stale identity information */
1918                 memset(&port->partner_ident, 0, sizeof(port->partner_ident));
1919                 port->partner_desc.usb_pd = port->pd_capable;
1920                 if (tcpm_port_is_debug(port))
1921                         port->partner_desc.accessory = TYPEC_ACCESSORY_DEBUG;
1922                 else if (tcpm_port_is_audio(port))
1923                         port->partner_desc.accessory = TYPEC_ACCESSORY_AUDIO;
1924                 else
1925                         port->partner_desc.accessory = TYPEC_ACCESSORY_NONE;
1926                 port->partner = typec_register_partner(port->typec_port,
1927                                                        &port->partner_desc);
1928                 port->connected = true;
1929         }
1930 }
1931
1932 static int tcpm_src_attach(struct tcpm_port *port)
1933 {
1934         enum typec_cc_polarity polarity =
1935                                 port->cc2 == TYPEC_CC_RD ? TYPEC_POLARITY_CC2
1936                                                          : TYPEC_POLARITY_CC1;
1937         int ret;
1938
1939         if (port->attached)
1940                 return 0;
1941
1942         ret = tcpm_set_polarity(port, polarity);
1943         if (ret < 0)
1944                 return ret;
1945
1946         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
1947         if (ret < 0)
1948                 return ret;
1949
1950         ret = port->tcpc->set_pd_rx(port->tcpc, true);
1951         if (ret < 0)
1952                 goto out_disable_mux;
1953
1954         /*
1955          * USB Type-C specification, version 1.2,
1956          * chapter 4.5.2.2.8.1 (Attached.SRC Requirements)
1957          * Enable VCONN only if the non-RD port is set to RA.
1958          */
1959         if ((polarity == TYPEC_POLARITY_CC1 && port->cc2 == TYPEC_CC_RA) ||
1960             (polarity == TYPEC_POLARITY_CC2 && port->cc1 == TYPEC_CC_RA)) {
1961                 ret = tcpm_set_vconn(port, true);
1962                 if (ret < 0)
1963                         goto out_disable_pd;
1964         }
1965
1966         ret = tcpm_set_vbus(port, true);
1967         if (ret < 0)
1968                 goto out_disable_vconn;
1969
1970         port->pd_capable = false;
1971
1972         port->partner = NULL;
1973
1974         port->attached = true;
1975         port->send_discover = true;
1976
1977         return 0;
1978
1979 out_disable_vconn:
1980         tcpm_set_vconn(port, false);
1981 out_disable_pd:
1982         port->tcpc->set_pd_rx(port->tcpc, false);
1983 out_disable_mux:
1984         tcpm_mux_set(port, TYPEC_MUX_NONE, TCPC_USB_SWITCH_DISCONNECT);
1985         return ret;
1986 }
1987
1988 static void tcpm_typec_disconnect(struct tcpm_port *port)
1989 {
1990         if (port->connected) {
1991                 typec_unregister_partner(port->partner);
1992                 port->partner = NULL;
1993                 port->connected = false;
1994         }
1995 }
1996
1997 static void tcpm_unregister_altmodes(struct tcpm_port *port)
1998 {
1999         struct pd_mode_data *modep = &port->mode_data;
2000         int i;
2001
2002         for (i = 0; i < modep->altmodes; i++) {
2003                 typec_unregister_altmode(port->partner_altmode[i]);
2004                 port->partner_altmode[i] = NULL;
2005         }
2006
2007         memset(modep, 0, sizeof(*modep));
2008 }
2009
2010 static void tcpm_reset_port(struct tcpm_port *port)
2011 {
2012         tcpm_unregister_altmodes(port);
2013         tcpm_typec_disconnect(port);
2014         port->attached = false;
2015         port->pd_capable = false;
2016
2017         /*
2018          * First Rx ID should be 0; set this to a sentinel of -1 so that
2019          * we can check tcpm_pd_rx_handler() if we had seen it before.
2020          */
2021         port->rx_msgid = -1;
2022
2023         port->tcpc->set_pd_rx(port->tcpc, false);
2024         tcpm_init_vbus(port);   /* also disables charging */
2025         tcpm_init_vconn(port);
2026         tcpm_set_current_limit(port, 0, 0);
2027         tcpm_set_polarity(port, TYPEC_POLARITY_CC1);
2028         tcpm_set_attached_state(port, false);
2029         port->try_src_count = 0;
2030         port->try_snk_count = 0;
2031 }
2032
2033 static void tcpm_detach(struct tcpm_port *port)
2034 {
2035         if (!port->attached)
2036                 return;
2037
2038         if (tcpm_port_is_disconnected(port))
2039                 port->hard_reset_count = 0;
2040
2041         tcpm_reset_port(port);
2042 }
2043
2044 static void tcpm_src_detach(struct tcpm_port *port)
2045 {
2046         tcpm_detach(port);
2047 }
2048
2049 static int tcpm_snk_attach(struct tcpm_port *port)
2050 {
2051         int ret;
2052
2053         if (port->attached)
2054                 return 0;
2055
2056         ret = tcpm_set_polarity(port, port->cc2 != TYPEC_CC_OPEN ?
2057                                 TYPEC_POLARITY_CC2 : TYPEC_POLARITY_CC1);
2058         if (ret < 0)
2059                 return ret;
2060
2061         ret = tcpm_set_roles(port, true, TYPEC_SINK, TYPEC_DEVICE);
2062         if (ret < 0)
2063                 return ret;
2064
2065         port->pd_capable = false;
2066
2067         port->partner = NULL;
2068
2069         port->attached = true;
2070         port->send_discover = true;
2071
2072         return 0;
2073 }
2074
2075 static void tcpm_snk_detach(struct tcpm_port *port)
2076 {
2077         tcpm_detach(port);
2078
2079         /* XXX: (Dis)connect SuperSpeed mux? */
2080 }
2081
2082 static int tcpm_acc_attach(struct tcpm_port *port)
2083 {
2084         int ret;
2085
2086         if (port->attached)
2087                 return 0;
2088
2089         ret = tcpm_set_roles(port, true, TYPEC_SOURCE, TYPEC_HOST);
2090         if (ret < 0)
2091                 return ret;
2092
2093         port->partner = NULL;
2094
2095         tcpm_typec_connect(port);
2096
2097         port->attached = true;
2098
2099         return 0;
2100 }
2101
2102 static void tcpm_acc_detach(struct tcpm_port *port)
2103 {
2104         tcpm_detach(port);
2105 }
2106
2107 static inline enum tcpm_state hard_reset_state(struct tcpm_port *port)
2108 {
2109         if (port->hard_reset_count < PD_N_HARD_RESET_COUNT)
2110                 return HARD_RESET_SEND;
2111         if (port->pd_capable)
2112                 return ERROR_RECOVERY;
2113         if (port->pwr_role == TYPEC_SOURCE)
2114                 return SRC_UNATTACHED;
2115         if (port->state == SNK_WAIT_CAPABILITIES)
2116                 return SNK_READY;
2117         return SNK_UNATTACHED;
2118 }
2119
2120 static inline enum tcpm_state ready_state(struct tcpm_port *port)
2121 {
2122         if (port->pwr_role == TYPEC_SOURCE)
2123                 return SRC_READY;
2124         else
2125                 return SNK_READY;
2126 }
2127
2128 static inline enum tcpm_state unattached_state(struct tcpm_port *port)
2129 {
2130         if (port->port_type == TYPEC_PORT_DRP) {
2131                 if (port->pwr_role == TYPEC_SOURCE)
2132                         return SRC_UNATTACHED;
2133                 else
2134                         return SNK_UNATTACHED;
2135         } else if (port->port_type == TYPEC_PORT_DFP) {
2136                 return SRC_UNATTACHED;
2137         }
2138
2139         return SNK_UNATTACHED;
2140 }
2141
2142 static void tcpm_check_send_discover(struct tcpm_port *port)
2143 {
2144         if (port->data_role == TYPEC_HOST && port->send_discover &&
2145             port->pd_capable) {
2146                 tcpm_send_vdm(port, USB_SID_PD, CMD_DISCOVER_IDENT, NULL, 0);
2147                 port->send_discover = false;
2148         }
2149 }
2150
2151 static void tcpm_swap_complete(struct tcpm_port *port, int result)
2152 {
2153         if (port->swap_pending) {
2154                 port->swap_status = result;
2155                 port->swap_pending = false;
2156                 port->non_pd_role_swap = false;
2157                 complete(&port->swap_complete);
2158         }
2159 }
2160
2161 static enum typec_pwr_opmode tcpm_get_pwr_opmode(enum typec_cc_status cc)
2162 {
2163         switch (cc) {
2164         case TYPEC_CC_RP_1_5:
2165                 return TYPEC_PWR_MODE_1_5A;
2166         case TYPEC_CC_RP_3_0:
2167                 return TYPEC_PWR_MODE_3_0A;
2168         case TYPEC_CC_RP_DEF:
2169         default:
2170                 return TYPEC_PWR_MODE_USB;
2171         }
2172 }
2173
2174 static void run_state_machine(struct tcpm_port *port)
2175 {
2176         int ret;
2177         enum typec_pwr_opmode opmode;
2178         unsigned int msecs;
2179
2180         port->enter_state = port->state;
2181         switch (port->state) {
2182         case DRP_TOGGLING:
2183                 break;
2184         /* SRC states */
2185         case SRC_UNATTACHED:
2186                 if (!port->non_pd_role_swap)
2187                         tcpm_swap_complete(port, -ENOTCONN);
2188                 tcpm_src_detach(port);
2189                 if (tcpm_start_drp_toggling(port)) {
2190                         tcpm_set_state(port, DRP_TOGGLING, 0);
2191                         break;
2192                 }
2193                 tcpm_set_cc(port, tcpm_rp_cc(port));
2194                 if (port->port_type == TYPEC_PORT_DRP)
2195                         tcpm_set_state(port, SNK_UNATTACHED, PD_T_DRP_SNK);
2196                 break;
2197         case SRC_ATTACH_WAIT:
2198                 if (tcpm_port_is_debug(port))
2199                         tcpm_set_state(port, DEBUG_ACC_ATTACHED,
2200                                        PD_T_CC_DEBOUNCE);
2201                 else if (tcpm_port_is_audio(port))
2202                         tcpm_set_state(port, AUDIO_ACC_ATTACHED,
2203                                        PD_T_CC_DEBOUNCE);
2204                 else if (tcpm_port_is_source(port))
2205                         tcpm_set_state(port,
2206                                        tcpm_try_snk(port) ? SNK_TRY
2207                                                           : SRC_ATTACHED,
2208                                        PD_T_CC_DEBOUNCE);
2209                 break;
2210
2211         case SNK_TRY:
2212                 port->try_snk_count++;
2213                 /*
2214                  * Requirements:
2215                  * - Do not drive vconn or vbus
2216                  * - Terminate CC pins (both) to Rd
2217                  * Action:
2218                  * - Wait for tDRPTry (PD_T_DRP_TRY).
2219                  *   Until then, ignore any state changes.
2220                  */
2221                 tcpm_set_cc(port, TYPEC_CC_RD);
2222                 tcpm_set_state(port, SNK_TRY_WAIT, PD_T_DRP_TRY);
2223                 break;
2224         case SNK_TRY_WAIT:
2225                 if (tcpm_port_is_sink(port)) {
2226                         tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE, 0);
2227                 } else {
2228                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2229                         port->max_wait = 0;
2230                 }
2231                 break;
2232         case SNK_TRY_WAIT_DEBOUNCE:
2233                 tcpm_set_state(port, SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS,
2234                                PD_T_PD_DEBOUNCE);
2235                 break;
2236         case SNK_TRY_WAIT_DEBOUNCE_CHECK_VBUS:
2237                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2238                         tcpm_set_state(port, SNK_ATTACHED, 0);
2239                 } else {
2240                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2241                         port->max_wait = 0;
2242                 }
2243                 break;
2244         case SRC_TRYWAIT:
2245                 tcpm_set_cc(port, tcpm_rp_cc(port));
2246                 if (port->max_wait == 0) {
2247                         port->max_wait = jiffies +
2248                                          msecs_to_jiffies(PD_T_DRP_TRY);
2249                         tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2250                                        PD_T_DRP_TRY);
2251                 } else {
2252                         if (time_is_after_jiffies(port->max_wait))
2253                                 tcpm_set_state(port, SRC_TRYWAIT_UNATTACHED,
2254                                                jiffies_to_msecs(port->max_wait -
2255                                                                 jiffies));
2256                         else
2257                                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2258                 }
2259                 break;
2260         case SRC_TRYWAIT_DEBOUNCE:
2261                 tcpm_set_state(port, SRC_ATTACHED, PD_T_CC_DEBOUNCE);
2262                 break;
2263         case SRC_TRYWAIT_UNATTACHED:
2264                 tcpm_set_state(port, SNK_UNATTACHED, 0);
2265                 break;
2266
2267         case SRC_ATTACHED:
2268                 ret = tcpm_src_attach(port);
2269                 tcpm_set_state(port, SRC_UNATTACHED,
2270                                ret < 0 ? 0 : PD_T_PS_SOURCE_ON);
2271                 break;
2272         case SRC_STARTUP:
2273                 opmode =  tcpm_get_pwr_opmode(tcpm_rp_cc(port));
2274                 typec_set_pwr_opmode(port->typec_port, opmode);
2275                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
2276                 port->caps_count = 0;
2277                 port->message_id = 0;
2278                 port->rx_msgid = -1;
2279                 port->explicit_contract = false;
2280                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
2281                 break;
2282         case SRC_SEND_CAPABILITIES:
2283                 port->caps_count++;
2284                 if (port->caps_count > PD_N_CAPS_COUNT) {
2285                         tcpm_set_state(port, SRC_READY, 0);
2286                         break;
2287                 }
2288                 ret = tcpm_pd_send_source_caps(port);
2289                 if (ret < 0) {
2290                         tcpm_set_state(port, SRC_SEND_CAPABILITIES,
2291                                        PD_T_SEND_SOURCE_CAP);
2292                 } else {
2293                         /*
2294                          * Per standard, we should clear the reset counter here.
2295                          * However, that can result in state machine hang-ups.
2296                          * Reset it only in READY state to improve stability.
2297                          */
2298                         /* port->hard_reset_count = 0; */
2299                         port->caps_count = 0;
2300                         port->pd_capable = true;
2301                         tcpm_set_state_cond(port, hard_reset_state(port),
2302                                             PD_T_SEND_SOURCE_CAP);
2303                 }
2304                 break;
2305         case SRC_NEGOTIATE_CAPABILITIES:
2306                 ret = tcpm_pd_check_request(port);
2307                 if (ret < 0) {
2308                         tcpm_pd_send_control(port, PD_CTRL_REJECT);
2309                         if (!port->explicit_contract) {
2310                                 tcpm_set_state(port,
2311                                                SRC_WAIT_NEW_CAPABILITIES, 0);
2312                         } else {
2313                                 tcpm_set_state(port, SRC_READY, 0);
2314                         }
2315                 } else {
2316                         tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2317                         tcpm_set_state(port, SRC_TRANSITION_SUPPLY,
2318                                        PD_T_SRC_TRANSITION);
2319                 }
2320                 break;
2321         case SRC_TRANSITION_SUPPLY:
2322                 /* XXX: regulator_set_voltage(vbus, ...) */
2323                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
2324                 port->explicit_contract = true;
2325                 typec_set_pwr_opmode(port->typec_port, TYPEC_PWR_MODE_PD);
2326                 port->pwr_opmode = TYPEC_PWR_MODE_PD;
2327                 tcpm_set_state_cond(port, SRC_READY, 0);
2328                 break;
2329         case SRC_READY:
2330 #if 1
2331                 port->hard_reset_count = 0;
2332 #endif
2333                 port->try_src_count = 0;
2334
2335                 tcpm_swap_complete(port, 0);
2336                 tcpm_typec_connect(port);
2337                 tcpm_check_send_discover(port);
2338                 /*
2339                  * 6.3.5
2340                  * Sending ping messages is not necessary if
2341                  * - the source operates at vSafe5V
2342                  * or
2343                  * - The system is not operating in PD mode
2344                  * or
2345                  * - Both partners are connected using a Type-C connector
2346                  *
2347                  * There is no actual need to send PD messages since the local
2348                  * port type-c and the spec does not clearly say whether PD is
2349                  * possible when type-c is connected to Type-A/B
2350                  */
2351                 break;
2352         case SRC_WAIT_NEW_CAPABILITIES:
2353                 /* Nothing to do... */
2354                 break;
2355
2356         /* SNK states */
2357         case SNK_UNATTACHED:
2358                 if (!port->non_pd_role_swap)
2359                         tcpm_swap_complete(port, -ENOTCONN);
2360                 tcpm_snk_detach(port);
2361                 if (tcpm_start_drp_toggling(port)) {
2362                         tcpm_set_state(port, DRP_TOGGLING, 0);
2363                         break;
2364                 }
2365                 tcpm_set_cc(port, TYPEC_CC_RD);
2366                 if (port->port_type == TYPEC_PORT_DRP)
2367                         tcpm_set_state(port, SRC_UNATTACHED, PD_T_DRP_SRC);
2368                 break;
2369         case SNK_ATTACH_WAIT:
2370                 if ((port->cc1 == TYPEC_CC_OPEN &&
2371                      port->cc2 != TYPEC_CC_OPEN) ||
2372                     (port->cc1 != TYPEC_CC_OPEN &&
2373                      port->cc2 == TYPEC_CC_OPEN))
2374                         tcpm_set_state(port, SNK_DEBOUNCED,
2375                                        PD_T_CC_DEBOUNCE);
2376                 else if (tcpm_port_is_disconnected(port))
2377                         tcpm_set_state(port, SNK_UNATTACHED,
2378                                        PD_T_PD_DEBOUNCE);
2379                 break;
2380         case SNK_DEBOUNCED:
2381                 if (tcpm_port_is_disconnected(port))
2382                         tcpm_set_state(port, SNK_UNATTACHED,
2383                                        PD_T_PD_DEBOUNCE);
2384                 else if (port->vbus_present)
2385                         tcpm_set_state(port,
2386                                        tcpm_try_src(port) ? SRC_TRY
2387                                                           : SNK_ATTACHED,
2388                                        0);
2389                 else
2390                         /* Wait for VBUS, but not forever */
2391                         tcpm_set_state(port, PORT_RESET, PD_T_PS_SOURCE_ON);
2392                 break;
2393
2394         case SRC_TRY:
2395                 port->try_src_count++;
2396                 tcpm_set_cc(port, tcpm_rp_cc(port));
2397                 port->max_wait = 0;
2398                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
2399                 break;
2400         case SRC_TRY_WAIT:
2401                 if (port->max_wait == 0) {
2402                         port->max_wait = jiffies +
2403                                          msecs_to_jiffies(PD_T_DRP_TRY);
2404                         msecs = PD_T_DRP_TRY;
2405                 } else {
2406                         if (time_is_after_jiffies(port->max_wait))
2407                                 msecs = jiffies_to_msecs(port->max_wait -
2408                                                          jiffies);
2409                         else
2410                                 msecs = 0;
2411                 }
2412                 tcpm_set_state(port, SNK_TRYWAIT, msecs);
2413                 break;
2414         case SRC_TRY_DEBOUNCE:
2415                 tcpm_set_state(port, SRC_ATTACHED, PD_T_PD_DEBOUNCE);
2416                 break;
2417         case SNK_TRYWAIT:
2418                 tcpm_set_cc(port, TYPEC_CC_RD);
2419                 tcpm_set_state(port, SNK_TRYWAIT_VBUS, PD_T_CC_DEBOUNCE);
2420                 break;
2421         case SNK_TRYWAIT_VBUS:
2422                 /*
2423                  * TCPM stays in this state indefinitely until VBUS
2424                  * is detected as long as Rp is not detected for
2425                  * more than a time period of tPDDebounce.
2426                  */
2427                 if (port->vbus_present && tcpm_port_is_sink(port)) {
2428                         tcpm_set_state(port, SNK_ATTACHED, 0);
2429                         break;
2430                 }
2431                 if (!tcpm_port_is_sink(port))
2432                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
2433                 break;
2434         case SNK_TRYWAIT_DEBOUNCE:
2435                 tcpm_set_state(port, SNK_UNATTACHED, PD_T_PD_DEBOUNCE);
2436                 break;
2437         case SNK_ATTACHED:
2438                 ret = tcpm_snk_attach(port);
2439                 if (ret < 0)
2440                         tcpm_set_state(port, SNK_UNATTACHED, 0);
2441                 else
2442                         tcpm_set_state(port, SNK_STARTUP, 0);
2443                 break;
2444         case SNK_STARTUP:
2445                 /* XXX: callback into infrastructure */
2446                 opmode =  tcpm_get_pwr_opmode(port->polarity ?
2447                                               port->cc2 : port->cc1);
2448                 typec_set_pwr_opmode(port->typec_port, opmode);
2449                 port->pwr_opmode = TYPEC_PWR_MODE_USB;
2450                 port->message_id = 0;
2451                 port->rx_msgid = -1;
2452                 port->explicit_contract = false;
2453                 tcpm_set_state(port, SNK_DISCOVERY, 0);
2454                 break;
2455         case SNK_DISCOVERY:
2456                 if (port->vbus_present) {
2457                         tcpm_set_current_limit(port,
2458                                                tcpm_get_current_limit(port),
2459                                                5000);
2460                         tcpm_set_charge(port, true);
2461                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
2462                         break;
2463                 }
2464                 /*
2465                  * For DRP, timeouts differ. Also, handling is supposed to be
2466                  * different and much more complex (dead battery detection;
2467                  * see USB power delivery specification, section 8.3.3.6.1.5.1).
2468                  */
2469                 tcpm_set_state(port, hard_reset_state(port),
2470                                port->port_type == TYPEC_PORT_DRP ?
2471                                         PD_T_DB_DETECT : PD_T_NO_RESPONSE);
2472                 break;
2473         case SNK_DISCOVERY_DEBOUNCE:
2474                 tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE_DONE,
2475                                PD_T_CC_DEBOUNCE);
2476                 break;
2477         case SNK_DISCOVERY_DEBOUNCE_DONE:
2478                 if (!tcpm_port_is_disconnected(port) &&
2479                     tcpm_port_is_sink(port) &&
2480                     time_is_after_jiffies(port->delayed_runtime)) {
2481                         tcpm_set_state(port, SNK_DISCOVERY,
2482                                        jiffies_to_msecs(port->delayed_runtime -
2483                                                         jiffies));
2484                         break;
2485                 }
2486                 tcpm_set_state(port, unattached_state(port), 0);
2487                 break;
2488         case SNK_WAIT_CAPABILITIES:
2489                 ret = port->tcpc->set_pd_rx(port->tcpc, true);
2490                 if (ret < 0) {
2491                         tcpm_set_state(port, SNK_READY, 0);
2492                         break;
2493                 }
2494                 /*
2495                  * If VBUS has never been low, and we time out waiting
2496                  * for source cap, try a soft reset first, in case we
2497                  * were already in a stable contract before this boot.
2498                  * Do this only once.
2499                  */
2500                 if (port->vbus_never_low) {
2501                         port->vbus_never_low = false;
2502                         tcpm_set_state(port, SOFT_RESET_SEND,
2503                                        PD_T_SINK_WAIT_CAP);
2504                 } else {
2505                         tcpm_set_state(port, hard_reset_state(port),
2506                                        PD_T_SINK_WAIT_CAP);
2507                 }
2508                 break;
2509         case SNK_NEGOTIATE_CAPABILITIES:
2510                 port->pd_capable = true;
2511                 port->hard_reset_count = 0;
2512                 ret = tcpm_pd_send_request(port);
2513                 if (ret < 0) {
2514                         /* Let the Source send capabilities again. */
2515                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
2516                 } else {
2517                         tcpm_set_state_cond(port, hard_reset_state(port),
2518                                             PD_T_SENDER_RESPONSE);
2519                 }
2520                 break;
2521         case SNK_TRANSITION_SINK:
2522         case SNK_TRANSITION_SINK_VBUS:
2523                 tcpm_set_state(port, hard_reset_state(port),
2524                                PD_T_PS_TRANSITION);
2525                 break;
2526         case SNK_READY:
2527                 port->try_snk_count = 0;
2528                 if (port->explicit_contract) {
2529                         typec_set_pwr_opmode(port->typec_port,
2530                                              TYPEC_PWR_MODE_PD);
2531                         port->pwr_opmode = TYPEC_PWR_MODE_PD;
2532                 }
2533
2534                 tcpm_swap_complete(port, 0);
2535                 tcpm_typec_connect(port);
2536                 tcpm_check_send_discover(port);
2537                 break;
2538
2539         /* Accessory states */
2540         case ACC_UNATTACHED:
2541                 tcpm_acc_detach(port);
2542                 tcpm_set_state(port, SRC_UNATTACHED, 0);
2543                 break;
2544         case DEBUG_ACC_ATTACHED:
2545         case AUDIO_ACC_ATTACHED:
2546                 ret = tcpm_acc_attach(port);
2547                 if (ret < 0)
2548                         tcpm_set_state(port, ACC_UNATTACHED, 0);
2549                 break;
2550         case AUDIO_ACC_DEBOUNCE:
2551                 tcpm_set_state(port, ACC_UNATTACHED, PD_T_CC_DEBOUNCE);
2552                 break;
2553
2554         /* Hard_Reset states */
2555         case HARD_RESET_SEND:
2556                 tcpm_pd_transmit(port, TCPC_TX_HARD_RESET, NULL);
2557                 tcpm_set_state(port, HARD_RESET_START, 0);
2558                 break;
2559         case HARD_RESET_START:
2560                 port->hard_reset_count++;
2561                 port->tcpc->set_pd_rx(port->tcpc, false);
2562                 tcpm_unregister_altmodes(port);
2563                 port->send_discover = true;
2564                 if (port->pwr_role == TYPEC_SOURCE)
2565                         tcpm_set_state(port, SRC_HARD_RESET_VBUS_OFF,
2566                                        PD_T_PS_HARD_RESET);
2567                 else
2568                         tcpm_set_state(port, SNK_HARD_RESET_SINK_OFF, 0);
2569                 break;
2570         case SRC_HARD_RESET_VBUS_OFF:
2571                 tcpm_set_vconn(port, true);
2572                 tcpm_set_vbus(port, false);
2573                 tcpm_set_roles(port, false, TYPEC_SOURCE, TYPEC_HOST);
2574                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, PD_T_SRC_RECOVER);
2575                 break;
2576         case SRC_HARD_RESET_VBUS_ON:
2577                 tcpm_set_vbus(port, true);
2578                 port->tcpc->set_pd_rx(port->tcpc, true);
2579                 tcpm_set_attached_state(port, true);
2580                 tcpm_set_state(port, SRC_UNATTACHED, PD_T_PS_SOURCE_ON);
2581                 break;
2582         case SNK_HARD_RESET_SINK_OFF:
2583                 tcpm_set_vconn(port, false);
2584                 tcpm_set_charge(port, false);
2585                 tcpm_set_roles(port, false, TYPEC_SINK, TYPEC_DEVICE);
2586                 /*
2587                  * VBUS may or may not toggle, depending on the adapter.
2588                  * If it doesn't toggle, transition to SNK_HARD_RESET_SINK_ON
2589                  * directly after timeout.
2590                  */
2591                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, PD_T_SAFE_0V);
2592                 break;
2593         case SNK_HARD_RESET_WAIT_VBUS:
2594                 /* Assume we're disconnected if VBUS doesn't come back. */
2595                 tcpm_set_state(port, SNK_UNATTACHED,
2596                                PD_T_SRC_RECOVER_MAX + PD_T_SRC_TURN_ON);
2597                 break;
2598         case SNK_HARD_RESET_SINK_ON:
2599                 /* Note: There is no guarantee that VBUS is on in this state */
2600                 /*
2601                  * XXX:
2602                  * The specification suggests that dual mode ports in sink
2603                  * mode should transition to state PE_SRC_Transition_to_default.
2604                  * See USB power delivery specification chapter 8.3.3.6.1.3.
2605                  * This would mean to to
2606                  * - turn off VCONN, reset power supply
2607                  * - request hardware reset
2608                  * - turn on VCONN
2609                  * - Transition to state PE_Src_Startup
2610                  * SNK only ports shall transition to state Snk_Startup
2611                  * (see chapter 8.3.3.3.8).
2612                  * Similar, dual-mode ports in source mode should transition
2613                  * to PE_SNK_Transition_to_default.
2614                  */
2615                 tcpm_set_attached_state(port, true);
2616                 tcpm_set_state(port, SNK_STARTUP, 0);
2617                 break;
2618
2619         /* Soft_Reset states */
2620         case SOFT_RESET:
2621                 port->message_id = 0;
2622                 port->rx_msgid = -1;
2623                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2624                 if (port->pwr_role == TYPEC_SOURCE)
2625                         tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
2626                 else
2627                         tcpm_set_state(port, SNK_WAIT_CAPABILITIES, 0);
2628                 break;
2629         case SOFT_RESET_SEND:
2630                 port->message_id = 0;
2631                 port->rx_msgid = -1;
2632                 if (tcpm_pd_send_control(port, PD_CTRL_SOFT_RESET))
2633                         tcpm_set_state_cond(port, hard_reset_state(port), 0);
2634                 else
2635                         tcpm_set_state_cond(port, hard_reset_state(port),
2636                                             PD_T_SENDER_RESPONSE);
2637                 break;
2638
2639         /* DR_Swap states */
2640         case DR_SWAP_SEND:
2641                 tcpm_pd_send_control(port, PD_CTRL_DR_SWAP);
2642                 tcpm_set_state_cond(port, DR_SWAP_SEND_TIMEOUT,
2643                                     PD_T_SENDER_RESPONSE);
2644                 break;
2645         case DR_SWAP_ACCEPT:
2646                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2647                 tcpm_set_state_cond(port, DR_SWAP_CHANGE_DR, 0);
2648                 break;
2649         case DR_SWAP_SEND_TIMEOUT:
2650                 tcpm_swap_complete(port, -ETIMEDOUT);
2651                 tcpm_set_state(port, ready_state(port), 0);
2652                 break;
2653         case DR_SWAP_CHANGE_DR:
2654                 if (port->data_role == TYPEC_HOST) {
2655                         tcpm_unregister_altmodes(port);
2656                         tcpm_set_roles(port, true, port->pwr_role,
2657                                        TYPEC_DEVICE);
2658                 } else {
2659                         tcpm_set_roles(port, true, port->pwr_role,
2660                                        TYPEC_HOST);
2661                         port->send_discover = true;
2662                 }
2663                 tcpm_set_state(port, ready_state(port), 0);
2664                 break;
2665
2666         /* PR_Swap states */
2667         case PR_SWAP_ACCEPT:
2668                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2669                 tcpm_set_state(port, PR_SWAP_START, 0);
2670                 break;
2671         case PR_SWAP_SEND:
2672                 tcpm_pd_send_control(port, PD_CTRL_PR_SWAP);
2673                 tcpm_set_state_cond(port, PR_SWAP_SEND_TIMEOUT,
2674                                     PD_T_SENDER_RESPONSE);
2675                 break;
2676         case PR_SWAP_SEND_TIMEOUT:
2677                 tcpm_swap_complete(port, -ETIMEDOUT);
2678                 tcpm_set_state(port, ready_state(port), 0);
2679                 break;
2680         case PR_SWAP_START:
2681                 if (port->pwr_role == TYPEC_SOURCE)
2682                         tcpm_set_state(port, PR_SWAP_SRC_SNK_TRANSITION_OFF,
2683                                        PD_T_SRC_TRANSITION);
2684                 else
2685                         tcpm_set_state(port, PR_SWAP_SNK_SRC_SINK_OFF, 0);
2686                 break;
2687         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
2688                 tcpm_set_vbus(port, false);
2689                 port->explicit_contract = false;
2690                 /* allow time for Vbus discharge, must be < tSrcSwapStdby */
2691                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF,
2692                                PD_T_SRCSWAPSTDBY);
2693                 break;
2694         case PR_SWAP_SRC_SNK_SOURCE_OFF:
2695                 tcpm_set_cc(port, TYPEC_CC_RD);
2696                 /* allow CC debounce */
2697                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED,
2698                                PD_T_CC_DEBOUNCE);
2699                 break;
2700         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
2701                 /*
2702                  * USB-PD standard, 6.2.1.4, Port Power Role:
2703                  * "During the Power Role Swap Sequence, for the initial Source
2704                  * Port, the Port Power Role field shall be set to Sink in the
2705                  * PS_RDY Message indicating that the initial Source’s power
2706                  * supply is turned off"
2707                  */
2708                 tcpm_set_pwr_role(port, TYPEC_SINK);
2709                 if (tcpm_pd_send_control(port, PD_CTRL_PS_RDY)) {
2710                         tcpm_set_state(port, ERROR_RECOVERY, 0);
2711                         break;
2712                 }
2713                 tcpm_set_state_cond(port, SNK_UNATTACHED, PD_T_PS_SOURCE_ON);
2714                 break;
2715         case PR_SWAP_SRC_SNK_SINK_ON:
2716                 tcpm_set_state(port, SNK_STARTUP, 0);
2717                 break;
2718         case PR_SWAP_SNK_SRC_SINK_OFF:
2719                 tcpm_set_charge(port, false);
2720                 tcpm_set_state(port, hard_reset_state(port),
2721                                PD_T_PS_SOURCE_OFF);
2722                 break;
2723         case PR_SWAP_SNK_SRC_SOURCE_ON:
2724                 tcpm_set_cc(port, tcpm_rp_cc(port));
2725                 tcpm_set_vbus(port, true);
2726                 /*
2727                  * allow time VBUS ramp-up, must be < tNewSrc
2728                  * Also, this window overlaps with CC debounce as well.
2729                  * So, Wait for the max of two which is PD_T_NEWSRC
2730                  */
2731                 tcpm_set_state(port, PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP,
2732                                PD_T_NEWSRC);
2733                 break;
2734         case PR_SWAP_SNK_SRC_SOURCE_ON_VBUS_RAMPED_UP:
2735                 /*
2736                  * USB PD standard, 6.2.1.4:
2737                  * "Subsequent Messages initiated by the Policy Engine,
2738                  * such as the PS_RDY Message sent to indicate that Vbus
2739                  * is ready, will have the Port Power Role field set to
2740                  * Source."
2741                  */
2742                 tcpm_set_pwr_role(port, TYPEC_SOURCE);
2743                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
2744                 tcpm_set_state(port, SRC_STARTUP, PD_T_SWAP_SRC_START);
2745                 break;
2746
2747         case VCONN_SWAP_ACCEPT:
2748                 tcpm_pd_send_control(port, PD_CTRL_ACCEPT);
2749                 tcpm_set_state(port, VCONN_SWAP_START, 0);
2750                 break;
2751         case VCONN_SWAP_SEND:
2752                 tcpm_pd_send_control(port, PD_CTRL_VCONN_SWAP);
2753                 tcpm_set_state(port, VCONN_SWAP_SEND_TIMEOUT,
2754                                PD_T_SENDER_RESPONSE);
2755                 break;
2756         case VCONN_SWAP_SEND_TIMEOUT:
2757                 tcpm_swap_complete(port, -ETIMEDOUT);
2758                 tcpm_set_state(port, ready_state(port), 0);
2759                 break;
2760         case VCONN_SWAP_START:
2761                 if (port->vconn_role == TYPEC_SOURCE)
2762                         tcpm_set_state(port, VCONN_SWAP_WAIT_FOR_VCONN, 0);
2763                 else
2764                         tcpm_set_state(port, VCONN_SWAP_TURN_ON_VCONN, 0);
2765                 break;
2766         case VCONN_SWAP_WAIT_FOR_VCONN:
2767                 tcpm_set_state(port, hard_reset_state(port),
2768                                PD_T_VCONN_SOURCE_ON);
2769                 break;
2770         case VCONN_SWAP_TURN_ON_VCONN:
2771                 tcpm_set_vconn(port, true);
2772                 tcpm_pd_send_control(port, PD_CTRL_PS_RDY);
2773                 tcpm_set_state(port, ready_state(port), 0);
2774                 break;
2775         case VCONN_SWAP_TURN_OFF_VCONN:
2776                 tcpm_set_vconn(port, false);
2777                 tcpm_set_state(port, ready_state(port), 0);
2778                 break;
2779
2780         case DR_SWAP_CANCEL:
2781         case PR_SWAP_CANCEL:
2782         case VCONN_SWAP_CANCEL:
2783                 tcpm_swap_complete(port, port->swap_status);
2784                 if (port->pwr_role == TYPEC_SOURCE)
2785                         tcpm_set_state(port, SRC_READY, 0);
2786                 else
2787                         tcpm_set_state(port, SNK_READY, 0);
2788                 break;
2789
2790         case BIST_RX:
2791                 switch (BDO_MODE_MASK(port->bist_request)) {
2792                 case BDO_MODE_CARRIER2:
2793                         tcpm_pd_transmit(port, TCPC_TX_BIST_MODE_2, NULL);
2794                         break;
2795                 default:
2796                         break;
2797                 }
2798                 /* Always switch to unattached state */
2799                 tcpm_set_state(port, unattached_state(port), 0);
2800                 break;
2801         case ERROR_RECOVERY:
2802                 tcpm_swap_complete(port, -EPROTO);
2803                 tcpm_set_state(port, PORT_RESET, 0);
2804                 break;
2805         case PORT_RESET:
2806                 tcpm_reset_port(port);
2807                 tcpm_set_cc(port, TYPEC_CC_OPEN);
2808                 tcpm_set_state(port, PORT_RESET_WAIT_OFF,
2809                                PD_T_ERROR_RECOVERY);
2810                 break;
2811         case PORT_RESET_WAIT_OFF:
2812                 tcpm_set_state(port,
2813                                tcpm_default_state(port),
2814                                port->vbus_present ? PD_T_PS_SOURCE_OFF : 0);
2815                 break;
2816         default:
2817                 WARN(1, "Unexpected port state %d\n", port->state);
2818                 break;
2819         }
2820 }
2821
2822 static void tcpm_state_machine_work(struct work_struct *work)
2823 {
2824         struct tcpm_port *port = container_of(work, struct tcpm_port,
2825                                               state_machine.work);
2826         enum tcpm_state prev_state;
2827
2828         mutex_lock(&port->lock);
2829         port->state_machine_running = true;
2830
2831         if (port->queued_message && tcpm_send_queued_message(port))
2832                 goto done;
2833
2834         /* If we were queued due to a delayed state change, update it now */
2835         if (port->delayed_state) {
2836                 tcpm_log(port, "state change %s -> %s [delayed %ld ms]",
2837                          tcpm_states[port->state],
2838                          tcpm_states[port->delayed_state], port->delay_ms);
2839                 port->prev_state = port->state;
2840                 port->state = port->delayed_state;
2841                 port->delayed_state = INVALID_STATE;
2842         }
2843
2844         /*
2845          * Continue running as long as we have (non-delayed) state changes
2846          * to make.
2847          */
2848         do {
2849                 prev_state = port->state;
2850                 run_state_machine(port);
2851                 if (port->queued_message)
2852                         tcpm_send_queued_message(port);
2853         } while (port->state != prev_state && !port->delayed_state);
2854
2855 done:
2856         port->state_machine_running = false;
2857         mutex_unlock(&port->lock);
2858 }
2859
2860 static void _tcpm_cc_change(struct tcpm_port *port, enum typec_cc_status cc1,
2861                             enum typec_cc_status cc2)
2862 {
2863         enum typec_cc_status old_cc1, old_cc2;
2864         enum tcpm_state new_state;
2865
2866         old_cc1 = port->cc1;
2867         old_cc2 = port->cc2;
2868         port->cc1 = cc1;
2869         port->cc2 = cc2;
2870
2871         tcpm_log_force(port,
2872                        "CC1: %u -> %u, CC2: %u -> %u [state %s, polarity %d, %s]",
2873                        old_cc1, cc1, old_cc2, cc2, tcpm_states[port->state],
2874                        port->polarity,
2875                        tcpm_port_is_disconnected(port) ? "disconnected"
2876                                                        : "connected");
2877
2878         switch (port->state) {
2879         case DRP_TOGGLING:
2880                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
2881                     tcpm_port_is_source(port))
2882                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
2883                 else if (tcpm_port_is_sink(port))
2884                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
2885                 break;
2886         case SRC_UNATTACHED:
2887         case ACC_UNATTACHED:
2888                 if (tcpm_port_is_debug(port) || tcpm_port_is_audio(port) ||
2889                     tcpm_port_is_source(port))
2890                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
2891                 break;
2892         case SRC_ATTACH_WAIT:
2893                 if (tcpm_port_is_disconnected(port) ||
2894                     tcpm_port_is_audio_detached(port))
2895                         tcpm_set_state(port, SRC_UNATTACHED, 0);
2896                 else if (cc1 != old_cc1 || cc2 != old_cc2)
2897                         tcpm_set_state(port, SRC_ATTACH_WAIT, 0);
2898                 break;
2899         case SRC_ATTACHED:
2900         case SRC_SEND_CAPABILITIES:
2901         case SRC_READY:
2902                 if (tcpm_port_is_disconnected(port) ||
2903                     !tcpm_port_is_source(port))
2904                         tcpm_set_state(port, SRC_UNATTACHED, 0);
2905                 break;
2906         case SNK_UNATTACHED:
2907                 if (tcpm_port_is_sink(port))
2908                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
2909                 break;
2910         case SNK_ATTACH_WAIT:
2911                 if ((port->cc1 == TYPEC_CC_OPEN &&
2912                      port->cc2 != TYPEC_CC_OPEN) ||
2913                     (port->cc1 != TYPEC_CC_OPEN &&
2914                      port->cc2 == TYPEC_CC_OPEN))
2915                         new_state = SNK_DEBOUNCED;
2916                 else if (tcpm_port_is_disconnected(port))
2917                         new_state = SNK_UNATTACHED;
2918                 else
2919                         break;
2920                 if (new_state != port->delayed_state)
2921                         tcpm_set_state(port, SNK_ATTACH_WAIT, 0);
2922                 break;
2923         case SNK_DEBOUNCED:
2924                 if (tcpm_port_is_disconnected(port))
2925                         new_state = SNK_UNATTACHED;
2926                 else if (port->vbus_present)
2927                         new_state = tcpm_try_src(port) ? SRC_TRY : SNK_ATTACHED;
2928                 else
2929                         new_state = SNK_UNATTACHED;
2930                 if (new_state != port->delayed_state)
2931                         tcpm_set_state(port, SNK_DEBOUNCED, 0);
2932                 break;
2933         case SNK_READY:
2934                 if (tcpm_port_is_disconnected(port))
2935                         tcpm_set_state(port, unattached_state(port), 0);
2936                 else if (!port->pd_capable &&
2937                          (cc1 != old_cc1 || cc2 != old_cc2))
2938                         tcpm_set_current_limit(port,
2939                                                tcpm_get_current_limit(port),
2940                                                5000);
2941                 break;
2942
2943         case AUDIO_ACC_ATTACHED:
2944                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
2945                         tcpm_set_state(port, AUDIO_ACC_DEBOUNCE, 0);
2946                 break;
2947         case AUDIO_ACC_DEBOUNCE:
2948                 if (tcpm_port_is_audio(port))
2949                         tcpm_set_state(port, AUDIO_ACC_ATTACHED, 0);
2950                 break;
2951
2952         case DEBUG_ACC_ATTACHED:
2953                 if (cc1 == TYPEC_CC_OPEN || cc2 == TYPEC_CC_OPEN)
2954                         tcpm_set_state(port, ACC_UNATTACHED, 0);
2955                 break;
2956
2957         case SNK_TRY:
2958                 /* Do nothing, waiting for timeout */
2959                 break;
2960
2961         case SNK_DISCOVERY:
2962                 /* CC line is unstable, wait for debounce */
2963                 if (tcpm_port_is_disconnected(port))
2964                         tcpm_set_state(port, SNK_DISCOVERY_DEBOUNCE, 0);
2965                 break;
2966         case SNK_DISCOVERY_DEBOUNCE:
2967                 break;
2968
2969         case SRC_TRYWAIT:
2970                 /* Hand over to state machine if needed */
2971                 if (!port->vbus_present && tcpm_port_is_source(port))
2972                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
2973                 break;
2974         case SRC_TRYWAIT_DEBOUNCE:
2975                 if (port->vbus_present || !tcpm_port_is_source(port))
2976                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2977                 break;
2978         case SNK_TRY_WAIT_DEBOUNCE:
2979                 if (!tcpm_port_is_sink(port)) {
2980                         port->max_wait = 0;
2981                         tcpm_set_state(port, SRC_TRYWAIT, 0);
2982                 }
2983                 break;
2984         case SRC_TRY_WAIT:
2985                 if (tcpm_port_is_source(port))
2986                         tcpm_set_state(port, SRC_TRY_DEBOUNCE, 0);
2987                 break;
2988         case SRC_TRY_DEBOUNCE:
2989                 tcpm_set_state(port, SRC_TRY_WAIT, 0);
2990                 break;
2991         case SNK_TRYWAIT_DEBOUNCE:
2992                 if (tcpm_port_is_sink(port))
2993                         tcpm_set_state(port, SNK_TRYWAIT_VBUS, 0);
2994                 break;
2995         case SNK_TRYWAIT_VBUS:
2996                 if (!tcpm_port_is_sink(port))
2997                         tcpm_set_state(port, SNK_TRYWAIT_DEBOUNCE, 0);
2998                 break;
2999         case SNK_TRYWAIT:
3000                 /* Do nothing, waiting for tCCDebounce */
3001                 break;
3002         case PR_SWAP_SNK_SRC_SINK_OFF:
3003         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3004         case PR_SWAP_SRC_SNK_SOURCE_OFF:
3005         case PR_SWAP_SRC_SNK_SOURCE_OFF_CC_DEBOUNCED:
3006         case PR_SWAP_SNK_SRC_SOURCE_ON:
3007                 /*
3008                  * CC state change is expected in PR_SWAP
3009                  * Ignore it.
3010                  */
3011                 break;
3012
3013         default:
3014                 if (tcpm_port_is_disconnected(port))
3015                         tcpm_set_state(port, unattached_state(port), 0);
3016                 break;
3017         }
3018 }
3019
3020 static void _tcpm_pd_vbus_on(struct tcpm_port *port)
3021 {
3022         tcpm_log_force(port, "VBUS on");
3023         port->vbus_present = true;
3024         switch (port->state) {
3025         case SNK_TRANSITION_SINK_VBUS:
3026                 port->explicit_contract = true;
3027                 tcpm_set_state(port, SNK_READY, 0);
3028                 break;
3029         case SNK_DISCOVERY:
3030                 tcpm_set_state(port, SNK_DISCOVERY, 0);
3031                 break;
3032
3033         case SNK_DEBOUNCED:
3034                 tcpm_set_state(port, tcpm_try_src(port) ? SRC_TRY
3035                                                         : SNK_ATTACHED,
3036                                        0);
3037                 break;
3038         case SNK_HARD_RESET_WAIT_VBUS:
3039                 tcpm_set_state(port, SNK_HARD_RESET_SINK_ON, 0);
3040                 break;
3041         case SRC_ATTACHED:
3042                 tcpm_set_state(port, SRC_STARTUP, 0);
3043                 break;
3044         case SRC_HARD_RESET_VBUS_ON:
3045                 tcpm_set_state(port, SRC_STARTUP, 0);
3046                 break;
3047
3048         case SNK_TRY:
3049                 /* Do nothing, waiting for timeout */
3050                 break;
3051         case SRC_TRYWAIT:
3052                 /* Do nothing, Waiting for Rd to be detected */
3053                 break;
3054         case SRC_TRYWAIT_DEBOUNCE:
3055                 tcpm_set_state(port, SRC_TRYWAIT, 0);
3056                 break;
3057         case SNK_TRY_WAIT_DEBOUNCE:
3058                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3059                 break;
3060         case SNK_TRYWAIT:
3061                 /* Do nothing, waiting for tCCDebounce */
3062                 break;
3063         case SNK_TRYWAIT_VBUS:
3064                 if (tcpm_port_is_sink(port))
3065                         tcpm_set_state(port, SNK_ATTACHED, 0);
3066                 break;
3067         case SNK_TRYWAIT_DEBOUNCE:
3068                 /* Do nothing, waiting for Rp */
3069                 break;
3070         case SRC_TRY_WAIT:
3071         case SRC_TRY_DEBOUNCE:
3072                 /* Do nothing, waiting for sink detection */
3073                 break;
3074         default:
3075                 break;
3076         }
3077 }
3078
3079 static void _tcpm_pd_vbus_off(struct tcpm_port *port)
3080 {
3081         tcpm_log_force(port, "VBUS off");
3082         port->vbus_present = false;
3083         port->vbus_never_low = false;
3084         switch (port->state) {
3085         case SNK_HARD_RESET_SINK_OFF:
3086                 tcpm_set_state(port, SNK_HARD_RESET_WAIT_VBUS, 0);
3087                 break;
3088         case SRC_HARD_RESET_VBUS_OFF:
3089                 tcpm_set_state(port, SRC_HARD_RESET_VBUS_ON, 0);
3090                 break;
3091         case HARD_RESET_SEND:
3092                 break;
3093
3094         case SNK_TRY:
3095                 /* Do nothing, waiting for timeout */
3096                 break;
3097         case SRC_TRYWAIT:
3098                 /* Hand over to state machine if needed */
3099                 if (tcpm_port_is_source(port))
3100                         tcpm_set_state(port, SRC_TRYWAIT_DEBOUNCE, 0);
3101                 break;
3102         case SNK_TRY_WAIT_DEBOUNCE:
3103                 /* Do nothing, waiting for PD_DEBOUNCE to do be done */
3104                 break;
3105         case SNK_TRYWAIT:
3106         case SNK_TRYWAIT_VBUS:
3107         case SNK_TRYWAIT_DEBOUNCE:
3108                 break;
3109         case SNK_ATTACH_WAIT:
3110                 tcpm_set_state(port, SNK_UNATTACHED, 0);
3111                 break;
3112
3113         case SNK_NEGOTIATE_CAPABILITIES:
3114                 break;
3115
3116         case PR_SWAP_SRC_SNK_TRANSITION_OFF:
3117                 tcpm_set_state(port, PR_SWAP_SRC_SNK_SOURCE_OFF, 0);
3118                 break;
3119
3120         case PR_SWAP_SNK_SRC_SINK_OFF:
3121                 /* Do nothing, expected */
3122                 break;
3123
3124         case PORT_RESET_WAIT_OFF:
3125                 tcpm_set_state(port, tcpm_default_state(port), 0);
3126                 break;
3127         case SRC_TRY_WAIT:
3128         case SRC_TRY_DEBOUNCE:
3129                 /* Do nothing, waiting for sink detection */
3130                 break;
3131         default:
3132                 if (port->pwr_role == TYPEC_SINK &&
3133                     port->attached)
3134                         tcpm_set_state(port, SNK_UNATTACHED, 0);
3135                 break;
3136         }
3137 }
3138
3139 static void _tcpm_pd_hard_reset(struct tcpm_port *port)
3140 {
3141         tcpm_log_force(port, "Received hard reset");
3142         /*
3143          * If we keep receiving hard reset requests, executing the hard reset
3144          * must have failed. Revert to error recovery if that happens.
3145          */
3146         tcpm_set_state(port,
3147                        port->hard_reset_count < PD_N_HARD_RESET_COUNT ?
3148                                 HARD_RESET_START : ERROR_RECOVERY,
3149                        0);
3150 }
3151
3152 static void tcpm_pd_event_handler(struct work_struct *work)
3153 {
3154         struct tcpm_port *port = container_of(work, struct tcpm_port,
3155                                               event_work);
3156         u32 events;
3157
3158         mutex_lock(&port->lock);
3159
3160         spin_lock(&port->pd_event_lock);
3161         while (port->pd_events) {
3162                 events = port->pd_events;
3163                 port->pd_events = 0;
3164                 spin_unlock(&port->pd_event_lock);
3165                 if (events & TCPM_RESET_EVENT)
3166                         _tcpm_pd_hard_reset(port);
3167                 if (events & TCPM_VBUS_EVENT) {
3168                         bool vbus;
3169
3170                         vbus = port->tcpc->get_vbus(port->tcpc);
3171                         if (vbus)
3172                                 _tcpm_pd_vbus_on(port);
3173                         else
3174                                 _tcpm_pd_vbus_off(port);
3175                 }
3176                 if (events & TCPM_CC_EVENT) {
3177                         enum typec_cc_status cc1, cc2;
3178
3179                         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
3180                                 _tcpm_cc_change(port, cc1, cc2);
3181                 }
3182                 spin_lock(&port->pd_event_lock);
3183         }
3184         spin_unlock(&port->pd_event_lock);
3185         mutex_unlock(&port->lock);
3186 }
3187
3188 void tcpm_cc_change(struct tcpm_port *port)
3189 {
3190         spin_lock(&port->pd_event_lock);
3191         port->pd_events |= TCPM_CC_EVENT;
3192         spin_unlock(&port->pd_event_lock);
3193         queue_work(port->wq, &port->event_work);
3194 }
3195 EXPORT_SYMBOL_GPL(tcpm_cc_change);
3196
3197 void tcpm_vbus_change(struct tcpm_port *port)
3198 {
3199         spin_lock(&port->pd_event_lock);
3200         port->pd_events |= TCPM_VBUS_EVENT;
3201         spin_unlock(&port->pd_event_lock);
3202         queue_work(port->wq, &port->event_work);
3203 }
3204 EXPORT_SYMBOL_GPL(tcpm_vbus_change);
3205
3206 void tcpm_pd_hard_reset(struct tcpm_port *port)
3207 {
3208         spin_lock(&port->pd_event_lock);
3209         port->pd_events = TCPM_RESET_EVENT;
3210         spin_unlock(&port->pd_event_lock);
3211         queue_work(port->wq, &port->event_work);
3212 }
3213 EXPORT_SYMBOL_GPL(tcpm_pd_hard_reset);
3214
3215 static int tcpm_dr_set(const struct typec_capability *cap,
3216                        enum typec_data_role data)
3217 {
3218         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3219         int ret;
3220
3221         mutex_lock(&port->swap_lock);
3222         mutex_lock(&port->lock);
3223
3224         if (port->port_type != TYPEC_PORT_DRP) {
3225                 ret = -EINVAL;
3226                 goto port_unlock;
3227         }
3228         if (port->state != SRC_READY && port->state != SNK_READY) {
3229                 ret = -EAGAIN;
3230                 goto port_unlock;
3231         }
3232
3233         if (port->data_role == data) {
3234                 ret = 0;
3235                 goto port_unlock;
3236         }
3237
3238         /*
3239          * XXX
3240          * 6.3.9: If an alternate mode is active, a request to swap
3241          * alternate modes shall trigger a port reset.
3242          * Reject data role swap request in this case.
3243          */
3244
3245         if (!port->pd_capable) {
3246                 /*
3247                  * If the partner is not PD capable, reset the port to
3248                  * trigger a role change. This can only work if a preferred
3249                  * role is configured, and if it matches the requested role.
3250                  */
3251                 if (port->try_role == TYPEC_NO_PREFERRED_ROLE ||
3252                     port->try_role == port->pwr_role) {
3253                         ret = -EINVAL;
3254                         goto port_unlock;
3255                 }
3256                 port->non_pd_role_swap = true;
3257                 tcpm_set_state(port, PORT_RESET, 0);
3258         } else {
3259                 tcpm_set_state(port, DR_SWAP_SEND, 0);
3260         }
3261
3262         port->swap_status = 0;
3263         port->swap_pending = true;
3264         reinit_completion(&port->swap_complete);
3265         mutex_unlock(&port->lock);
3266
3267         if (!wait_for_completion_timeout(&port->swap_complete,
3268                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3269                 ret = -ETIMEDOUT;
3270         else
3271                 ret = port->swap_status;
3272
3273         port->non_pd_role_swap = false;
3274         goto swap_unlock;
3275
3276 port_unlock:
3277         mutex_unlock(&port->lock);
3278 swap_unlock:
3279         mutex_unlock(&port->swap_lock);
3280         return ret;
3281 }
3282
3283 static int tcpm_pr_set(const struct typec_capability *cap,
3284                        enum typec_role role)
3285 {
3286         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3287         int ret;
3288
3289         mutex_lock(&port->swap_lock);
3290         mutex_lock(&port->lock);
3291
3292         if (port->port_type != TYPEC_PORT_DRP) {
3293                 ret = -EINVAL;
3294                 goto port_unlock;
3295         }
3296         if (port->state != SRC_READY && port->state != SNK_READY) {
3297                 ret = -EAGAIN;
3298                 goto port_unlock;
3299         }
3300
3301         if (role == port->pwr_role) {
3302                 ret = 0;
3303                 goto port_unlock;
3304         }
3305
3306         port->swap_status = 0;
3307         port->swap_pending = true;
3308         reinit_completion(&port->swap_complete);
3309         tcpm_set_state(port, PR_SWAP_SEND, 0);
3310         mutex_unlock(&port->lock);
3311
3312         if (!wait_for_completion_timeout(&port->swap_complete,
3313                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3314                 ret = -ETIMEDOUT;
3315         else
3316                 ret = port->swap_status;
3317
3318         goto swap_unlock;
3319
3320 port_unlock:
3321         mutex_unlock(&port->lock);
3322 swap_unlock:
3323         mutex_unlock(&port->swap_lock);
3324         return ret;
3325 }
3326
3327 static int tcpm_vconn_set(const struct typec_capability *cap,
3328                           enum typec_role role)
3329 {
3330         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3331         int ret;
3332
3333         mutex_lock(&port->swap_lock);
3334         mutex_lock(&port->lock);
3335
3336         if (port->state != SRC_READY && port->state != SNK_READY) {
3337                 ret = -EAGAIN;
3338                 goto port_unlock;
3339         }
3340
3341         if (role == port->vconn_role) {
3342                 ret = 0;
3343                 goto port_unlock;
3344         }
3345
3346         port->swap_status = 0;
3347         port->swap_pending = true;
3348         reinit_completion(&port->swap_complete);
3349         tcpm_set_state(port, VCONN_SWAP_SEND, 0);
3350         mutex_unlock(&port->lock);
3351
3352         if (!wait_for_completion_timeout(&port->swap_complete,
3353                                 msecs_to_jiffies(PD_ROLE_SWAP_TIMEOUT)))
3354                 ret = -ETIMEDOUT;
3355         else
3356                 ret = port->swap_status;
3357
3358         goto swap_unlock;
3359
3360 port_unlock:
3361         mutex_unlock(&port->lock);
3362 swap_unlock:
3363         mutex_unlock(&port->swap_lock);
3364         return ret;
3365 }
3366
3367 static int tcpm_try_role(const struct typec_capability *cap, int role)
3368 {
3369         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3370         struct tcpc_dev *tcpc = port->tcpc;
3371         int ret = 0;
3372
3373         mutex_lock(&port->lock);
3374         if (tcpc->try_role)
3375                 ret = tcpc->try_role(tcpc, role);
3376         if (!ret && !tcpc->config->try_role_hw)
3377                 port->try_role = role;
3378         port->try_src_count = 0;
3379         port->try_snk_count = 0;
3380         mutex_unlock(&port->lock);
3381
3382         return ret;
3383 }
3384
3385 static void tcpm_init(struct tcpm_port *port)
3386 {
3387         enum typec_cc_status cc1, cc2;
3388
3389         port->tcpc->init(port->tcpc);
3390
3391         tcpm_reset_port(port);
3392
3393         /*
3394          * XXX
3395          * Should possibly wait for VBUS to settle if it was enabled locally
3396          * since tcpm_reset_port() will disable VBUS.
3397          */
3398         port->vbus_present = port->tcpc->get_vbus(port->tcpc);
3399         if (port->vbus_present)
3400                 port->vbus_never_low = true;
3401
3402         tcpm_set_state(port, tcpm_default_state(port), 0);
3403
3404         if (port->tcpc->get_cc(port->tcpc, &cc1, &cc2) == 0)
3405                 _tcpm_cc_change(port, cc1, cc2);
3406
3407         /*
3408          * Some adapters need a clean slate at startup, and won't recover
3409          * otherwise. So do not try to be fancy and force a clean disconnect.
3410          */
3411         tcpm_set_state(port, PORT_RESET, 0);
3412 }
3413
3414 static int tcpm_port_type_set(const struct typec_capability *cap,
3415                               enum typec_port_type type)
3416 {
3417         struct tcpm_port *port = typec_cap_to_tcpm(cap);
3418
3419         mutex_lock(&port->lock);
3420         if (type == port->port_type)
3421                 goto port_unlock;
3422
3423         port->port_type = type;
3424
3425         if (!port->connected) {
3426                 tcpm_set_state(port, PORT_RESET, 0);
3427         } else if (type == TYPEC_PORT_UFP) {
3428                 if (!(port->pwr_role == TYPEC_SINK &&
3429                       port->data_role == TYPEC_DEVICE))
3430                         tcpm_set_state(port, PORT_RESET, 0);
3431         } else if (type == TYPEC_PORT_DFP) {
3432                 if (!(port->pwr_role == TYPEC_SOURCE &&
3433                       port->data_role == TYPEC_HOST))
3434                         tcpm_set_state(port, PORT_RESET, 0);
3435         }
3436
3437 port_unlock:
3438         mutex_unlock(&port->lock);
3439         return 0;
3440 }
3441
3442 void tcpm_tcpc_reset(struct tcpm_port *port)
3443 {
3444         mutex_lock(&port->lock);
3445         /* XXX: Maintain PD connection if possible? */
3446         tcpm_init(port);
3447         mutex_unlock(&port->lock);
3448 }
3449 EXPORT_SYMBOL_GPL(tcpm_tcpc_reset);
3450
3451 static int tcpm_copy_pdos(u32 *dest_pdo, const u32 *src_pdo,
3452                           unsigned int nr_pdo)
3453 {
3454         unsigned int i;
3455
3456         if (nr_pdo > PDO_MAX_OBJECTS)
3457                 nr_pdo = PDO_MAX_OBJECTS;
3458
3459         for (i = 0; i < nr_pdo; i++)
3460                 dest_pdo[i] = src_pdo[i];
3461
3462         return nr_pdo;
3463 }
3464
3465 static int tcpm_copy_vdos(u32 *dest_vdo, const u32 *src_vdo,
3466                           unsigned int nr_vdo)
3467 {
3468         unsigned int i;
3469
3470         if (nr_vdo > VDO_MAX_OBJECTS)
3471                 nr_vdo = VDO_MAX_OBJECTS;
3472
3473         for (i = 0; i < nr_vdo; i++)
3474                 dest_vdo[i] = src_vdo[i];
3475
3476         return nr_vdo;
3477 }
3478
3479 void tcpm_update_source_capabilities(struct tcpm_port *port, const u32 *pdo,
3480                                      unsigned int nr_pdo)
3481 {
3482         mutex_lock(&port->lock);
3483         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, pdo, nr_pdo);
3484         switch (port->state) {
3485         case SRC_UNATTACHED:
3486         case SRC_ATTACH_WAIT:
3487         case SRC_TRYWAIT:
3488                 tcpm_set_cc(port, tcpm_rp_cc(port));
3489                 break;
3490         case SRC_SEND_CAPABILITIES:
3491         case SRC_NEGOTIATE_CAPABILITIES:
3492         case SRC_READY:
3493         case SRC_WAIT_NEW_CAPABILITIES:
3494                 tcpm_set_cc(port, tcpm_rp_cc(port));
3495                 tcpm_set_state(port, SRC_SEND_CAPABILITIES, 0);
3496                 break;
3497         default:
3498                 break;
3499         }
3500         mutex_unlock(&port->lock);
3501 }
3502 EXPORT_SYMBOL_GPL(tcpm_update_source_capabilities);
3503
3504 void tcpm_update_sink_capabilities(struct tcpm_port *port, const u32 *pdo,
3505                                    unsigned int nr_pdo,
3506                                    unsigned int max_snk_mv,
3507                                    unsigned int max_snk_ma,
3508                                    unsigned int max_snk_mw,
3509                                    unsigned int operating_snk_mw)
3510 {
3511         mutex_lock(&port->lock);
3512         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, pdo, nr_pdo);
3513         port->max_snk_mv = max_snk_mv;
3514         port->max_snk_ma = max_snk_ma;
3515         port->max_snk_mw = max_snk_mw;
3516         port->operating_snk_mw = operating_snk_mw;
3517
3518         switch (port->state) {
3519         case SNK_NEGOTIATE_CAPABILITIES:
3520         case SNK_READY:
3521         case SNK_TRANSITION_SINK:
3522         case SNK_TRANSITION_SINK_VBUS:
3523                 tcpm_set_state(port, SNK_NEGOTIATE_CAPABILITIES, 0);
3524                 break;
3525         default:
3526                 break;
3527         }
3528         mutex_unlock(&port->lock);
3529 }
3530 EXPORT_SYMBOL_GPL(tcpm_update_sink_capabilities);
3531
3532 struct tcpm_port *tcpm_register_port(struct device *dev, struct tcpc_dev *tcpc)
3533 {
3534         struct tcpm_port *port;
3535         int i, err;
3536
3537         if (!dev || !tcpc || !tcpc->config ||
3538             !tcpc->get_vbus || !tcpc->set_cc || !tcpc->get_cc ||
3539             !tcpc->set_polarity || !tcpc->set_vconn || !tcpc->set_vbus ||
3540             !tcpc->set_pd_rx || !tcpc->set_roles || !tcpc->pd_transmit)
3541                 return ERR_PTR(-EINVAL);
3542
3543         port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
3544         if (!port)
3545                 return ERR_PTR(-ENOMEM);
3546
3547         port->dev = dev;
3548         port->tcpc = tcpc;
3549
3550         mutex_init(&port->lock);
3551         mutex_init(&port->swap_lock);
3552
3553         port->wq = create_singlethread_workqueue(dev_name(dev));
3554         if (!port->wq)
3555                 return ERR_PTR(-ENOMEM);
3556         INIT_DELAYED_WORK(&port->state_machine, tcpm_state_machine_work);
3557         INIT_DELAYED_WORK(&port->vdm_state_machine, vdm_state_machine_work);
3558         INIT_WORK(&port->event_work, tcpm_pd_event_handler);
3559
3560         spin_lock_init(&port->pd_event_lock);
3561
3562         init_completion(&port->tx_complete);
3563         init_completion(&port->swap_complete);
3564
3565         port->nr_src_pdo = tcpm_copy_pdos(port->src_pdo, tcpc->config->src_pdo,
3566                                           tcpc->config->nr_src_pdo);
3567         port->nr_snk_pdo = tcpm_copy_pdos(port->snk_pdo, tcpc->config->snk_pdo,
3568                                           tcpc->config->nr_snk_pdo);
3569         port->nr_snk_vdo = tcpm_copy_vdos(port->snk_vdo, tcpc->config->snk_vdo,
3570                                           tcpc->config->nr_snk_vdo);
3571
3572         port->max_snk_mv = tcpc->config->max_snk_mv;
3573         port->max_snk_ma = tcpc->config->max_snk_ma;
3574         port->max_snk_mw = tcpc->config->max_snk_mw;
3575         port->operating_snk_mw = tcpc->config->operating_snk_mw;
3576         if (!tcpc->config->try_role_hw)
3577                 port->try_role = tcpc->config->default_role;
3578         else
3579                 port->try_role = TYPEC_NO_PREFERRED_ROLE;
3580
3581         port->typec_caps.prefer_role = tcpc->config->default_role;
3582         port->typec_caps.type = tcpc->config->type;
3583         port->typec_caps.revision = 0x0120;     /* Type-C spec release 1.2 */
3584         port->typec_caps.pd_revision = 0x0200;  /* USB-PD spec release 2.0 */
3585         port->typec_caps.dr_set = tcpm_dr_set;
3586         port->typec_caps.pr_set = tcpm_pr_set;
3587         port->typec_caps.vconn_set = tcpm_vconn_set;
3588         port->typec_caps.try_role = tcpm_try_role;
3589         port->typec_caps.port_type_set = tcpm_port_type_set;
3590
3591         port->partner_desc.identity = &port->partner_ident;
3592         port->port_type = tcpc->config->type;
3593         /*
3594          * TODO:
3595          *  - alt_modes, set_alt_mode
3596          *  - {debug,audio}_accessory
3597          */
3598
3599         port->typec_port = typec_register_port(port->dev, &port->typec_caps);
3600         if (!port->typec_port) {
3601                 err = -ENOMEM;
3602                 goto out_destroy_wq;
3603         }
3604
3605         if (tcpc->config->alt_modes) {
3606                 const struct typec_altmode_desc *paltmode = tcpc->config->alt_modes;
3607
3608                 i = 0;
3609                 while (paltmode->svid && i < ARRAY_SIZE(port->port_altmode)) {
3610                         port->port_altmode[i] =
3611                           typec_port_register_altmode(port->typec_port,
3612                                                       paltmode);
3613                         if (!port->port_altmode[i]) {
3614                                 tcpm_log(port,
3615                                          "%s: failed to register port alternate mode 0x%x",
3616                                          dev_name(dev), paltmode->svid);
3617                                 break;
3618                         }
3619                         i++;
3620                         paltmode++;
3621                 }
3622         }
3623
3624         tcpm_debugfs_init(port);
3625         mutex_lock(&port->lock);
3626         tcpm_init(port);
3627         mutex_unlock(&port->lock);
3628
3629         tcpm_log(port, "%s: registered", dev_name(dev));
3630         return port;
3631
3632 out_destroy_wq:
3633         destroy_workqueue(port->wq);
3634         return ERR_PTR(err);
3635 }
3636 EXPORT_SYMBOL_GPL(tcpm_register_port);
3637
3638 void tcpm_unregister_port(struct tcpm_port *port)
3639 {
3640         int i;
3641
3642         for (i = 0; i < ARRAY_SIZE(port->port_altmode); i++)
3643                 typec_unregister_altmode(port->port_altmode[i]);
3644         typec_unregister_port(port->typec_port);
3645         tcpm_debugfs_exit(port);
3646         destroy_workqueue(port->wq);
3647 }
3648 EXPORT_SYMBOL_GPL(tcpm_unregister_port);
3649
3650 MODULE_AUTHOR("Guenter Roeck <groeck@chromium.org>");
3651 MODULE_DESCRIPTION("USB Type-C Port Manager");
3652 MODULE_LICENSE("GPL");