GNU Linux-libre 5.19-rc6-gnu
[releases.git] / drivers / usb / host / xhci-mtk-sch.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2015 MediaTek Inc.
4  * Author:
5  *  Zhigang.Wei <zhigang.wei@mediatek.com>
6  *  Chunfeng.Yun <chunfeng.yun@mediatek.com>
7  */
8
9 #include <linux/kernel.h>
10 #include <linux/module.h>
11 #include <linux/slab.h>
12
13 #include "xhci.h"
14 #include "xhci-mtk.h"
15
16 #define SSP_BW_BOUNDARY 130000
17 #define SS_BW_BOUNDARY  51000
18 /* table 5-5. High-speed Isoc Transaction Limits in usb_20 spec */
19 #define HS_BW_BOUNDARY  6144
20 /* usb2 spec section11.18.1: at most 188 FS bytes per microframe */
21 #define FS_PAYLOAD_MAX 188
22
23 #define DBG_BUF_EN      64
24
25 /* schedule error type */
26 #define ESCH_SS_Y6              1001
27 #define ESCH_SS_OVERLAP         1002
28 #define ESCH_CS_OVERFLOW        1003
29 #define ESCH_BW_OVERFLOW        1004
30 #define ESCH_FIXME              1005
31
32 /* mtk scheduler bitmasks */
33 #define EP_BPKTS(p)     ((p) & 0x7f)
34 #define EP_BCSCOUNT(p)  (((p) & 0x7) << 8)
35 #define EP_BBM(p)       ((p) << 11)
36 #define EP_BOFFSET(p)   ((p) & 0x3fff)
37 #define EP_BREPEAT(p)   (((p) & 0x7fff) << 16)
38
39 static char *sch_error_string(int err_num)
40 {
41         switch (err_num) {
42         case ESCH_SS_Y6:
43                 return "Can't schedule Start-Split in Y6";
44         case ESCH_SS_OVERLAP:
45                 return "Can't find a suitable Start-Split location";
46         case ESCH_CS_OVERFLOW:
47                 return "The last Complete-Split is greater than 7";
48         case ESCH_BW_OVERFLOW:
49                 return "Bandwidth exceeds the maximum limit";
50         case ESCH_FIXME:
51                 return "FIXME, to be resolved";
52         default:
53                 return "Unknown";
54         }
55 }
56
57 static int is_fs_or_ls(enum usb_device_speed speed)
58 {
59         return speed == USB_SPEED_FULL || speed == USB_SPEED_LOW;
60 }
61
62 static const char *
63 decode_ep(struct usb_host_endpoint *ep, enum usb_device_speed speed)
64 {
65         static char buf[DBG_BUF_EN];
66         struct usb_endpoint_descriptor *epd = &ep->desc;
67         unsigned int interval;
68         const char *unit;
69
70         interval = usb_decode_interval(epd, speed);
71         if (interval % 1000) {
72                 unit = "us";
73         } else {
74                 unit = "ms";
75                 interval /= 1000;
76         }
77
78         snprintf(buf, DBG_BUF_EN, "%s ep%d%s %s, mpkt:%d, interval:%d/%d%s",
79                  usb_speed_string(speed), usb_endpoint_num(epd),
80                  usb_endpoint_dir_in(epd) ? "in" : "out",
81                  usb_ep_type_string(usb_endpoint_type(epd)),
82                  usb_endpoint_maxp(epd), epd->bInterval, interval, unit);
83
84         return buf;
85 }
86
87 static u32 get_bw_boundary(enum usb_device_speed speed)
88 {
89         u32 boundary;
90
91         switch (speed) {
92         case USB_SPEED_SUPER_PLUS:
93                 boundary = SSP_BW_BOUNDARY;
94                 break;
95         case USB_SPEED_SUPER:
96                 boundary = SS_BW_BOUNDARY;
97                 break;
98         default:
99                 boundary = HS_BW_BOUNDARY;
100                 break;
101         }
102
103         return boundary;
104 }
105
106 /*
107 * get the bandwidth domain which @ep belongs to.
108 *
109 * the bandwidth domain array is saved to @sch_array of struct xhci_hcd_mtk,
110 * each HS root port is treated as a single bandwidth domain,
111 * but each SS root port is treated as two bandwidth domains, one for IN eps,
112 * one for OUT eps.
113 * @real_port value is defined as follow according to xHCI spec:
114 * 1 for SSport0, ..., N+1 for SSportN, N+2 for HSport0, N+3 for HSport1, etc
115 * so the bandwidth domain array is organized as follow for simplification:
116 * SSport0-OUT, SSport0-IN, ..., SSportX-OUT, SSportX-IN, HSport0, ..., HSportY
117 */
118 static struct mu3h_sch_bw_info *
119 get_bw_info(struct xhci_hcd_mtk *mtk, struct usb_device *udev,
120             struct usb_host_endpoint *ep)
121 {
122         struct xhci_hcd *xhci = hcd_to_xhci(mtk->hcd);
123         struct xhci_virt_device *virt_dev;
124         int bw_index;
125
126         virt_dev = xhci->devs[udev->slot_id];
127         if (!virt_dev->real_port) {
128                 WARN_ONCE(1, "%s invalid real_port\n", dev_name(&udev->dev));
129                 return NULL;
130         }
131
132         if (udev->speed >= USB_SPEED_SUPER) {
133                 if (usb_endpoint_dir_out(&ep->desc))
134                         bw_index = (virt_dev->real_port - 1) * 2;
135                 else
136                         bw_index = (virt_dev->real_port - 1) * 2 + 1;
137         } else {
138                 /* add one more for each SS port */
139                 bw_index = virt_dev->real_port + xhci->usb3_rhub.num_ports - 1;
140         }
141
142         return &mtk->sch_array[bw_index];
143 }
144
145 static u32 get_esit(struct xhci_ep_ctx *ep_ctx)
146 {
147         u32 esit;
148
149         esit = 1 << CTX_TO_EP_INTERVAL(le32_to_cpu(ep_ctx->ep_info));
150         if (esit > XHCI_MTK_MAX_ESIT)
151                 esit = XHCI_MTK_MAX_ESIT;
152
153         return esit;
154 }
155
156 static struct mu3h_sch_tt *find_tt(struct usb_device *udev)
157 {
158         struct usb_tt *utt = udev->tt;
159         struct mu3h_sch_tt *tt, **tt_index, **ptt;
160         bool allocated_index = false;
161
162         if (!utt)
163                 return NULL;    /* Not below a TT */
164
165         /*
166          * Find/create our data structure.
167          * For hubs with a single TT, we get it directly.
168          * For hubs with multiple TTs, there's an extra level of pointers.
169          */
170         tt_index = NULL;
171         if (utt->multi) {
172                 tt_index = utt->hcpriv;
173                 if (!tt_index) {        /* Create the index array */
174                         tt_index = kcalloc(utt->hub->maxchild,
175                                         sizeof(*tt_index), GFP_KERNEL);
176                         if (!tt_index)
177                                 return ERR_PTR(-ENOMEM);
178                         utt->hcpriv = tt_index;
179                         allocated_index = true;
180                 }
181                 ptt = &tt_index[udev->ttport - 1];
182         } else {
183                 ptt = (struct mu3h_sch_tt **) &utt->hcpriv;
184         }
185
186         tt = *ptt;
187         if (!tt) {      /* Create the mu3h_sch_tt */
188                 tt = kzalloc(sizeof(*tt), GFP_KERNEL);
189                 if (!tt) {
190                         if (allocated_index) {
191                                 utt->hcpriv = NULL;
192                                 kfree(tt_index);
193                         }
194                         return ERR_PTR(-ENOMEM);
195                 }
196                 INIT_LIST_HEAD(&tt->ep_list);
197                 *ptt = tt;
198         }
199
200         return tt;
201 }
202
203 /* Release the TT above udev, if it's not in use */
204 static void drop_tt(struct usb_device *udev)
205 {
206         struct usb_tt *utt = udev->tt;
207         struct mu3h_sch_tt *tt, **tt_index, **ptt;
208         int i, cnt;
209
210         if (!utt || !utt->hcpriv)
211                 return;         /* Not below a TT, or never allocated */
212
213         cnt = 0;
214         if (utt->multi) {
215                 tt_index = utt->hcpriv;
216                 ptt = &tt_index[udev->ttport - 1];
217                 /*  How many entries are left in tt_index? */
218                 for (i = 0; i < utt->hub->maxchild; ++i)
219                         cnt += !!tt_index[i];
220         } else {
221                 tt_index = NULL;
222                 ptt = (struct mu3h_sch_tt **)&utt->hcpriv;
223         }
224
225         tt = *ptt;
226         if (!tt || !list_empty(&tt->ep_list))
227                 return;         /* never allocated , or still in use*/
228
229         *ptt = NULL;
230         kfree(tt);
231
232         if (cnt == 1) {
233                 utt->hcpriv = NULL;
234                 kfree(tt_index);
235         }
236 }
237
238 static struct mu3h_sch_ep_info *
239 create_sch_ep(struct xhci_hcd_mtk *mtk, struct usb_device *udev,
240               struct usb_host_endpoint *ep)
241 {
242         struct mu3h_sch_ep_info *sch_ep;
243         struct mu3h_sch_bw_info *bw_info;
244         struct mu3h_sch_tt *tt = NULL;
245
246         bw_info = get_bw_info(mtk, udev, ep);
247         if (!bw_info)
248                 return ERR_PTR(-ENODEV);
249
250         sch_ep = kzalloc(sizeof(*sch_ep), GFP_KERNEL);
251         if (!sch_ep)
252                 return ERR_PTR(-ENOMEM);
253
254         if (is_fs_or_ls(udev->speed)) {
255                 tt = find_tt(udev);
256                 if (IS_ERR(tt)) {
257                         kfree(sch_ep);
258                         return ERR_PTR(-ENOMEM);
259                 }
260         }
261
262         sch_ep->bw_info = bw_info;
263         sch_ep->sch_tt = tt;
264         sch_ep->ep = ep;
265         sch_ep->speed = udev->speed;
266         INIT_LIST_HEAD(&sch_ep->endpoint);
267         INIT_LIST_HEAD(&sch_ep->tt_endpoint);
268         INIT_HLIST_NODE(&sch_ep->hentry);
269
270         return sch_ep;
271 }
272
273 static void setup_sch_info(struct xhci_ep_ctx *ep_ctx,
274                            struct mu3h_sch_ep_info *sch_ep)
275 {
276         u32 ep_type;
277         u32 maxpkt;
278         u32 max_burst;
279         u32 mult;
280         u32 esit_pkts;
281         u32 max_esit_payload;
282
283         ep_type = CTX_TO_EP_TYPE(le32_to_cpu(ep_ctx->ep_info2));
284         maxpkt = MAX_PACKET_DECODED(le32_to_cpu(ep_ctx->ep_info2));
285         max_burst = CTX_TO_MAX_BURST(le32_to_cpu(ep_ctx->ep_info2));
286         mult = CTX_TO_EP_MULT(le32_to_cpu(ep_ctx->ep_info));
287         max_esit_payload =
288                 (CTX_TO_MAX_ESIT_PAYLOAD_HI(
289                         le32_to_cpu(ep_ctx->ep_info)) << 16) |
290                  CTX_TO_MAX_ESIT_PAYLOAD(le32_to_cpu(ep_ctx->tx_info));
291
292         sch_ep->esit = get_esit(ep_ctx);
293         sch_ep->num_esit = XHCI_MTK_MAX_ESIT / sch_ep->esit;
294         sch_ep->ep_type = ep_type;
295         sch_ep->maxpkt = maxpkt;
296         sch_ep->offset = 0;
297         sch_ep->burst_mode = 0;
298         sch_ep->repeat = 0;
299
300         if (sch_ep->speed == USB_SPEED_HIGH) {
301                 sch_ep->cs_count = 0;
302
303                 /*
304                  * usb_20 spec section5.9
305                  * a single microframe is enough for HS synchromous endpoints
306                  * in a interval
307                  */
308                 sch_ep->num_budget_microframes = 1;
309
310                 /*
311                  * xHCI spec section6.2.3.4
312                  * @max_burst is the number of additional transactions
313                  * opportunities per microframe
314                  */
315                 sch_ep->pkts = max_burst + 1;
316                 sch_ep->bw_cost_per_microframe = maxpkt * sch_ep->pkts;
317         } else if (sch_ep->speed >= USB_SPEED_SUPER) {
318                 /* usb3_r1 spec section4.4.7 & 4.4.8 */
319                 sch_ep->cs_count = 0;
320                 sch_ep->burst_mode = 1;
321                 /*
322                  * some device's (d)wBytesPerInterval is set as 0,
323                  * then max_esit_payload is 0, so evaluate esit_pkts from
324                  * mult and burst
325                  */
326                 esit_pkts = DIV_ROUND_UP(max_esit_payload, maxpkt);
327                 if (esit_pkts == 0)
328                         esit_pkts = (mult + 1) * (max_burst + 1);
329
330                 if (ep_type == INT_IN_EP || ep_type == INT_OUT_EP) {
331                         sch_ep->pkts = esit_pkts;
332                         sch_ep->num_budget_microframes = 1;
333                 }
334
335                 if (ep_type == ISOC_IN_EP || ep_type == ISOC_OUT_EP) {
336
337                         if (sch_ep->esit == 1)
338                                 sch_ep->pkts = esit_pkts;
339                         else if (esit_pkts <= sch_ep->esit)
340                                 sch_ep->pkts = 1;
341                         else
342                                 sch_ep->pkts = roundup_pow_of_two(esit_pkts)
343                                         / sch_ep->esit;
344
345                         sch_ep->num_budget_microframes =
346                                 DIV_ROUND_UP(esit_pkts, sch_ep->pkts);
347
348                         sch_ep->repeat = !!(sch_ep->num_budget_microframes > 1);
349                 }
350                 sch_ep->bw_cost_per_microframe = maxpkt * sch_ep->pkts;
351         } else if (is_fs_or_ls(sch_ep->speed)) {
352                 sch_ep->pkts = 1; /* at most one packet for each microframe */
353
354                 /*
355                  * num_budget_microframes and cs_count will be updated when
356                  * check TT for INT_OUT_EP, ISOC/INT_IN_EP type
357                  */
358                 sch_ep->cs_count = DIV_ROUND_UP(maxpkt, FS_PAYLOAD_MAX);
359                 sch_ep->num_budget_microframes = sch_ep->cs_count;
360                 sch_ep->bw_cost_per_microframe = min_t(u32, maxpkt, FS_PAYLOAD_MAX);
361         }
362 }
363
364 /* Get maximum bandwidth when we schedule at offset slot. */
365 static u32 get_max_bw(struct mu3h_sch_bw_info *sch_bw,
366         struct mu3h_sch_ep_info *sch_ep, u32 offset)
367 {
368         u32 max_bw = 0;
369         u32 bw;
370         int i, j, k;
371
372         for (i = 0; i < sch_ep->num_esit; i++) {
373                 u32 base = offset + i * sch_ep->esit;
374
375                 for (j = 0; j < sch_ep->num_budget_microframes; j++) {
376                         k = XHCI_MTK_BW_INDEX(base + j);
377                         bw = sch_bw->bus_bw[k] + sch_ep->bw_cost_per_microframe;
378                         if (bw > max_bw)
379                                 max_bw = bw;
380                 }
381         }
382         return max_bw;
383 }
384
385 static void update_bus_bw(struct mu3h_sch_bw_info *sch_bw,
386         struct mu3h_sch_ep_info *sch_ep, bool used)
387 {
388         int bw_updated;
389         u32 base;
390         int i, j;
391
392         bw_updated = sch_ep->bw_cost_per_microframe * (used ? 1 : -1);
393
394         for (i = 0; i < sch_ep->num_esit; i++) {
395                 base = sch_ep->offset + i * sch_ep->esit;
396                 for (j = 0; j < sch_ep->num_budget_microframes; j++)
397                         sch_bw->bus_bw[XHCI_MTK_BW_INDEX(base + j)] += bw_updated;
398         }
399 }
400
401 static int check_fs_bus_bw(struct mu3h_sch_ep_info *sch_ep, int offset)
402 {
403         struct mu3h_sch_tt *tt = sch_ep->sch_tt;
404         u32 tmp;
405         int base;
406         int i, j, k;
407
408         for (i = 0; i < sch_ep->num_esit; i++) {
409                 base = offset + i * sch_ep->esit;
410
411                 /*
412                  * Compared with hs bus, no matter what ep type,
413                  * the hub will always delay one uframe to send data
414                  */
415                 for (j = 0; j < sch_ep->num_budget_microframes; j++) {
416                         k = XHCI_MTK_BW_INDEX(base + j);
417                         tmp = tt->fs_bus_bw[k] + sch_ep->bw_cost_per_microframe;
418                         if (tmp > FS_PAYLOAD_MAX)
419                                 return -ESCH_BW_OVERFLOW;
420                 }
421         }
422
423         return 0;
424 }
425
426 static int check_sch_tt(struct mu3h_sch_ep_info *sch_ep, u32 offset)
427 {
428         u32 extra_cs_count;
429         u32 start_ss, last_ss;
430         u32 start_cs, last_cs;
431
432         if (!sch_ep->sch_tt)
433                 return 0;
434
435         start_ss = offset % 8;
436
437         if (sch_ep->ep_type == ISOC_OUT_EP) {
438                 last_ss = start_ss + sch_ep->cs_count - 1;
439
440                 /*
441                  * usb_20 spec section11.18:
442                  * must never schedule Start-Split in Y6
443                  */
444                 if (!(start_ss == 7 || last_ss < 6))
445                         return -ESCH_SS_Y6;
446
447         } else {
448                 u32 cs_count = DIV_ROUND_UP(sch_ep->maxpkt, FS_PAYLOAD_MAX);
449
450                 /*
451                  * usb_20 spec section11.18:
452                  * must never schedule Start-Split in Y6
453                  */
454                 if (start_ss == 6)
455                         return -ESCH_SS_Y6;
456
457                 /* one uframe for ss + one uframe for idle */
458                 start_cs = (start_ss + 2) % 8;
459                 last_cs = start_cs + cs_count - 1;
460
461                 if (last_cs > 7)
462                         return -ESCH_CS_OVERFLOW;
463
464                 if (sch_ep->ep_type == ISOC_IN_EP)
465                         extra_cs_count = (last_cs == 7) ? 1 : 2;
466                 else /*  ep_type : INTR IN / INTR OUT */
467                         extra_cs_count = 1;
468
469                 cs_count += extra_cs_count;
470                 if (cs_count > 7)
471                         cs_count = 7; /* HW limit */
472
473                 sch_ep->cs_count = cs_count;
474                 /* one for ss, the other for idle */
475                 sch_ep->num_budget_microframes = cs_count + 2;
476
477                 /*
478                  * if interval=1, maxp >752, num_budge_micoframe is larger
479                  * than sch_ep->esit, will overstep boundary
480                  */
481                 if (sch_ep->num_budget_microframes > sch_ep->esit)
482                         sch_ep->num_budget_microframes = sch_ep->esit;
483         }
484
485         return check_fs_bus_bw(sch_ep, offset);
486 }
487
488 static void update_sch_tt(struct mu3h_sch_ep_info *sch_ep, bool used)
489 {
490         struct mu3h_sch_tt *tt = sch_ep->sch_tt;
491         int bw_updated;
492         u32 base;
493         int i, j;
494
495         bw_updated = sch_ep->bw_cost_per_microframe * (used ? 1 : -1);
496
497         for (i = 0; i < sch_ep->num_esit; i++) {
498                 base = sch_ep->offset + i * sch_ep->esit;
499
500                 for (j = 0; j < sch_ep->num_budget_microframes; j++)
501                         tt->fs_bus_bw[XHCI_MTK_BW_INDEX(base + j)] += bw_updated;
502         }
503
504         if (used)
505                 list_add_tail(&sch_ep->tt_endpoint, &tt->ep_list);
506         else
507                 list_del(&sch_ep->tt_endpoint);
508 }
509
510 static int load_ep_bw(struct mu3h_sch_bw_info *sch_bw,
511                       struct mu3h_sch_ep_info *sch_ep, bool loaded)
512 {
513         if (sch_ep->sch_tt)
514                 update_sch_tt(sch_ep, loaded);
515
516         /* update bus bandwidth info */
517         update_bus_bw(sch_bw, sch_ep, loaded);
518         sch_ep->allocated = loaded;
519
520         return 0;
521 }
522
523 static int check_sch_bw(struct mu3h_sch_ep_info *sch_ep)
524 {
525         struct mu3h_sch_bw_info *sch_bw = sch_ep->bw_info;
526         const u32 bw_boundary = get_bw_boundary(sch_ep->speed);
527         u32 offset;
528         u32 worst_bw;
529         u32 min_bw = ~0;
530         int min_index = -1;
531         int ret = 0;
532
533         /*
534          * Search through all possible schedule microframes.
535          * and find a microframe where its worst bandwidth is minimum.
536          */
537         for (offset = 0; offset < sch_ep->esit; offset++) {
538                 ret = check_sch_tt(sch_ep, offset);
539                 if (ret)
540                         continue;
541
542                 worst_bw = get_max_bw(sch_bw, sch_ep, offset);
543                 if (worst_bw > bw_boundary)
544                         continue;
545
546                 if (min_bw > worst_bw) {
547                         min_bw = worst_bw;
548                         min_index = offset;
549                 }
550
551                 /* use first-fit for LS/FS */
552                 if (sch_ep->sch_tt && min_index >= 0)
553                         break;
554
555                 if (min_bw == 0)
556                         break;
557         }
558
559         if (min_index < 0)
560                 return ret ? ret : -ESCH_BW_OVERFLOW;
561
562         sch_ep->offset = min_index;
563
564         return load_ep_bw(sch_bw, sch_ep, true);
565 }
566
567 static void destroy_sch_ep(struct xhci_hcd_mtk *mtk, struct usb_device *udev,
568                            struct mu3h_sch_ep_info *sch_ep)
569 {
570         /* only release ep bw check passed by check_sch_bw() */
571         if (sch_ep->allocated)
572                 load_ep_bw(sch_ep->bw_info, sch_ep, false);
573
574         if (sch_ep->sch_tt)
575                 drop_tt(udev);
576
577         list_del(&sch_ep->endpoint);
578         hlist_del(&sch_ep->hentry);
579         kfree(sch_ep);
580 }
581
582 static bool need_bw_sch(struct usb_device *udev,
583                         struct usb_host_endpoint *ep)
584 {
585         bool has_tt = udev->tt && udev->tt->hub->parent;
586
587         /* only for periodic endpoints */
588         if (usb_endpoint_xfer_control(&ep->desc)
589                 || usb_endpoint_xfer_bulk(&ep->desc))
590                 return false;
591
592         /*
593          * for LS & FS periodic endpoints which its device is not behind
594          * a TT are also ignored, root-hub will schedule them directly,
595          * but need set @bpkts field of endpoint context to 1.
596          */
597         if (is_fs_or_ls(udev->speed) && !has_tt)
598                 return false;
599
600         /* skip endpoint with zero maxpkt */
601         if (usb_endpoint_maxp(&ep->desc) == 0)
602                 return false;
603
604         return true;
605 }
606
607 int xhci_mtk_sch_init(struct xhci_hcd_mtk *mtk)
608 {
609         struct xhci_hcd *xhci = hcd_to_xhci(mtk->hcd);
610         struct mu3h_sch_bw_info *sch_array;
611         int num_usb_bus;
612
613         /* ss IN and OUT are separated */
614         num_usb_bus = xhci->usb3_rhub.num_ports * 2 + xhci->usb2_rhub.num_ports;
615
616         sch_array = kcalloc(num_usb_bus, sizeof(*sch_array), GFP_KERNEL);
617         if (sch_array == NULL)
618                 return -ENOMEM;
619
620         mtk->sch_array = sch_array;
621
622         INIT_LIST_HEAD(&mtk->bw_ep_chk_list);
623         hash_init(mtk->sch_ep_hash);
624
625         return 0;
626 }
627
628 void xhci_mtk_sch_exit(struct xhci_hcd_mtk *mtk)
629 {
630         kfree(mtk->sch_array);
631 }
632
633 static int add_ep_quirk(struct usb_hcd *hcd, struct usb_device *udev,
634                         struct usb_host_endpoint *ep)
635 {
636         struct xhci_hcd_mtk *mtk = hcd_to_mtk(hcd);
637         struct xhci_hcd *xhci = hcd_to_xhci(hcd);
638         struct xhci_ep_ctx *ep_ctx;
639         struct xhci_virt_device *virt_dev;
640         struct mu3h_sch_ep_info *sch_ep;
641         unsigned int ep_index;
642
643         virt_dev = xhci->devs[udev->slot_id];
644         ep_index = xhci_get_endpoint_index(&ep->desc);
645         ep_ctx = xhci_get_ep_ctx(xhci, virt_dev->in_ctx, ep_index);
646
647         if (!need_bw_sch(udev, ep)) {
648                 /*
649                  * set @bpkts to 1 if it is LS or FS periodic endpoint, and its
650                  * device does not connected through an external HS hub
651                  */
652                 if (usb_endpoint_xfer_int(&ep->desc)
653                         || usb_endpoint_xfer_isoc(&ep->desc))
654                         ep_ctx->reserved[0] = cpu_to_le32(EP_BPKTS(1));
655
656                 return 0;
657         }
658
659         xhci_dbg(xhci, "%s %s\n", __func__, decode_ep(ep, udev->speed));
660
661         sch_ep = create_sch_ep(mtk, udev, ep);
662         if (IS_ERR_OR_NULL(sch_ep))
663                 return -ENOMEM;
664
665         setup_sch_info(ep_ctx, sch_ep);
666
667         list_add_tail(&sch_ep->endpoint, &mtk->bw_ep_chk_list);
668         hash_add(mtk->sch_ep_hash, &sch_ep->hentry, (unsigned long)ep);
669
670         return 0;
671 }
672
673 static void drop_ep_quirk(struct usb_hcd *hcd, struct usb_device *udev,
674                           struct usb_host_endpoint *ep)
675 {
676         struct xhci_hcd_mtk *mtk = hcd_to_mtk(hcd);
677         struct xhci_hcd *xhci = hcd_to_xhci(hcd);
678         struct mu3h_sch_ep_info *sch_ep;
679         struct hlist_node *hn;
680
681         if (!need_bw_sch(udev, ep))
682                 return;
683
684         xhci_dbg(xhci, "%s %s\n", __func__, decode_ep(ep, udev->speed));
685
686         hash_for_each_possible_safe(mtk->sch_ep_hash, sch_ep,
687                                     hn, hentry, (unsigned long)ep) {
688                 if (sch_ep->ep == ep) {
689                         destroy_sch_ep(mtk, udev, sch_ep);
690                         break;
691                 }
692         }
693 }
694
695 int xhci_mtk_check_bandwidth(struct usb_hcd *hcd, struct usb_device *udev)
696 {
697         struct xhci_hcd_mtk *mtk = hcd_to_mtk(hcd);
698         struct xhci_hcd *xhci = hcd_to_xhci(hcd);
699         struct xhci_virt_device *virt_dev = xhci->devs[udev->slot_id];
700         struct mu3h_sch_ep_info *sch_ep;
701         int ret;
702
703         xhci_dbg(xhci, "%s() udev %s\n", __func__, dev_name(&udev->dev));
704
705         list_for_each_entry(sch_ep, &mtk->bw_ep_chk_list, endpoint) {
706                 struct xhci_ep_ctx *ep_ctx;
707                 struct usb_host_endpoint *ep = sch_ep->ep;
708                 unsigned int ep_index = xhci_get_endpoint_index(&ep->desc);
709
710                 ret = check_sch_bw(sch_ep);
711                 if (ret) {
712                         xhci_err(xhci, "Not enough bandwidth! (%s)\n",
713                                  sch_error_string(-ret));
714                         return -ENOSPC;
715                 }
716
717                 ep_ctx = xhci_get_ep_ctx(xhci, virt_dev->in_ctx, ep_index);
718                 ep_ctx->reserved[0] = cpu_to_le32(EP_BPKTS(sch_ep->pkts)
719                         | EP_BCSCOUNT(sch_ep->cs_count)
720                         | EP_BBM(sch_ep->burst_mode));
721                 ep_ctx->reserved[1] = cpu_to_le32(EP_BOFFSET(sch_ep->offset)
722                         | EP_BREPEAT(sch_ep->repeat));
723
724                 xhci_dbg(xhci, " PKTS:%x, CSCOUNT:%x, BM:%x, OFFSET:%x, REPEAT:%x\n",
725                         sch_ep->pkts, sch_ep->cs_count, sch_ep->burst_mode,
726                         sch_ep->offset, sch_ep->repeat);
727         }
728
729         ret = xhci_check_bandwidth(hcd, udev);
730         if (!ret)
731                 list_del_init(&mtk->bw_ep_chk_list);
732
733         return ret;
734 }
735
736 void xhci_mtk_reset_bandwidth(struct usb_hcd *hcd, struct usb_device *udev)
737 {
738         struct xhci_hcd_mtk *mtk = hcd_to_mtk(hcd);
739         struct xhci_hcd *xhci = hcd_to_xhci(hcd);
740         struct mu3h_sch_ep_info *sch_ep, *tmp;
741
742         xhci_dbg(xhci, "%s() udev %s\n", __func__, dev_name(&udev->dev));
743
744         list_for_each_entry_safe(sch_ep, tmp, &mtk->bw_ep_chk_list, endpoint)
745                 destroy_sch_ep(mtk, udev, sch_ep);
746
747         xhci_reset_bandwidth(hcd, udev);
748 }
749
750 int xhci_mtk_add_ep(struct usb_hcd *hcd, struct usb_device *udev,
751                     struct usb_host_endpoint *ep)
752 {
753         int ret;
754
755         ret = xhci_add_endpoint(hcd, udev, ep);
756         if (ret)
757                 return ret;
758
759         if (ep->hcpriv)
760                 ret = add_ep_quirk(hcd, udev, ep);
761
762         return ret;
763 }
764
765 int xhci_mtk_drop_ep(struct usb_hcd *hcd, struct usb_device *udev,
766                      struct usb_host_endpoint *ep)
767 {
768         int ret;
769
770         ret = xhci_drop_endpoint(hcd, udev, ep);
771         if (ret)
772                 return ret;
773
774         if (ep->hcpriv)
775                 drop_ep_quirk(hcd, udev, ep);
776
777         return 0;
778 }