Merge tag 'selinux-pr-20181115' of git://git.kernel.org/pub/scm/linux/kernel/git...
[linux-2.6-microblaze.git] / drivers / thunderbolt / domain.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Thunderbolt bus support
4  *
5  * Copyright (C) 2017, Intel Corporation
6  * Author: Mika Westerberg <mika.westerberg@linux.intel.com>
7  */
8
9 #include <linux/device.h>
10 #include <linux/idr.h>
11 #include <linux/module.h>
12 #include <linux/pm_runtime.h>
13 #include <linux/slab.h>
14 #include <linux/random.h>
15 #include <crypto/hash.h>
16
17 #include "tb.h"
18
19 static DEFINE_IDA(tb_domain_ida);
20
21 static bool match_service_id(const struct tb_service_id *id,
22                              const struct tb_service *svc)
23 {
24         if (id->match_flags & TBSVC_MATCH_PROTOCOL_KEY) {
25                 if (strcmp(id->protocol_key, svc->key))
26                         return false;
27         }
28
29         if (id->match_flags & TBSVC_MATCH_PROTOCOL_ID) {
30                 if (id->protocol_id != svc->prtcid)
31                         return false;
32         }
33
34         if (id->match_flags & TBSVC_MATCH_PROTOCOL_VERSION) {
35                 if (id->protocol_version != svc->prtcvers)
36                         return false;
37         }
38
39         if (id->match_flags & TBSVC_MATCH_PROTOCOL_VERSION) {
40                 if (id->protocol_revision != svc->prtcrevs)
41                         return false;
42         }
43
44         return true;
45 }
46
47 static const struct tb_service_id *__tb_service_match(struct device *dev,
48                                                       struct device_driver *drv)
49 {
50         struct tb_service_driver *driver;
51         const struct tb_service_id *ids;
52         struct tb_service *svc;
53
54         svc = tb_to_service(dev);
55         if (!svc)
56                 return NULL;
57
58         driver = container_of(drv, struct tb_service_driver, driver);
59         if (!driver->id_table)
60                 return NULL;
61
62         for (ids = driver->id_table; ids->match_flags != 0; ids++) {
63                 if (match_service_id(ids, svc))
64                         return ids;
65         }
66
67         return NULL;
68 }
69
70 static int tb_service_match(struct device *dev, struct device_driver *drv)
71 {
72         return !!__tb_service_match(dev, drv);
73 }
74
75 static int tb_service_probe(struct device *dev)
76 {
77         struct tb_service *svc = tb_to_service(dev);
78         struct tb_service_driver *driver;
79         const struct tb_service_id *id;
80
81         driver = container_of(dev->driver, struct tb_service_driver, driver);
82         id = __tb_service_match(dev, &driver->driver);
83
84         return driver->probe(svc, id);
85 }
86
87 static int tb_service_remove(struct device *dev)
88 {
89         struct tb_service *svc = tb_to_service(dev);
90         struct tb_service_driver *driver;
91
92         driver = container_of(dev->driver, struct tb_service_driver, driver);
93         if (driver->remove)
94                 driver->remove(svc);
95
96         return 0;
97 }
98
99 static void tb_service_shutdown(struct device *dev)
100 {
101         struct tb_service_driver *driver;
102         struct tb_service *svc;
103
104         svc = tb_to_service(dev);
105         if (!svc || !dev->driver)
106                 return;
107
108         driver = container_of(dev->driver, struct tb_service_driver, driver);
109         if (driver->shutdown)
110                 driver->shutdown(svc);
111 }
112
113 static const char * const tb_security_names[] = {
114         [TB_SECURITY_NONE] = "none",
115         [TB_SECURITY_USER] = "user",
116         [TB_SECURITY_SECURE] = "secure",
117         [TB_SECURITY_DPONLY] = "dponly",
118         [TB_SECURITY_USBONLY] = "usbonly",
119 };
120
121 static ssize_t boot_acl_show(struct device *dev, struct device_attribute *attr,
122                              char *buf)
123 {
124         struct tb *tb = container_of(dev, struct tb, dev);
125         uuid_t *uuids;
126         ssize_t ret;
127         int i;
128
129         uuids = kcalloc(tb->nboot_acl, sizeof(uuid_t), GFP_KERNEL);
130         if (!uuids)
131                 return -ENOMEM;
132
133         pm_runtime_get_sync(&tb->dev);
134
135         if (mutex_lock_interruptible(&tb->lock)) {
136                 ret = -ERESTARTSYS;
137                 goto out;
138         }
139         ret = tb->cm_ops->get_boot_acl(tb, uuids, tb->nboot_acl);
140         if (ret) {
141                 mutex_unlock(&tb->lock);
142                 goto out;
143         }
144         mutex_unlock(&tb->lock);
145
146         for (ret = 0, i = 0; i < tb->nboot_acl; i++) {
147                 if (!uuid_is_null(&uuids[i]))
148                         ret += snprintf(buf + ret, PAGE_SIZE - ret, "%pUb",
149                                         &uuids[i]);
150
151                 ret += snprintf(buf + ret, PAGE_SIZE - ret, "%s",
152                                i < tb->nboot_acl - 1 ? "," : "\n");
153         }
154
155 out:
156         pm_runtime_mark_last_busy(&tb->dev);
157         pm_runtime_put_autosuspend(&tb->dev);
158         kfree(uuids);
159
160         return ret;
161 }
162
163 static ssize_t boot_acl_store(struct device *dev, struct device_attribute *attr,
164                               const char *buf, size_t count)
165 {
166         struct tb *tb = container_of(dev, struct tb, dev);
167         char *str, *s, *uuid_str;
168         ssize_t ret = 0;
169         uuid_t *acl;
170         int i = 0;
171
172         /*
173          * Make sure the value is not bigger than tb->nboot_acl * UUID
174          * length + commas and optional "\n". Also the smallest allowable
175          * string is tb->nboot_acl * ",".
176          */
177         if (count > (UUID_STRING_LEN + 1) * tb->nboot_acl + 1)
178                 return -EINVAL;
179         if (count < tb->nboot_acl - 1)
180                 return -EINVAL;
181
182         str = kstrdup(buf, GFP_KERNEL);
183         if (!str)
184                 return -ENOMEM;
185
186         acl = kcalloc(tb->nboot_acl, sizeof(uuid_t), GFP_KERNEL);
187         if (!acl) {
188                 ret = -ENOMEM;
189                 goto err_free_str;
190         }
191
192         uuid_str = strim(str);
193         while ((s = strsep(&uuid_str, ",")) != NULL && i < tb->nboot_acl) {
194                 size_t len = strlen(s);
195
196                 if (len) {
197                         if (len != UUID_STRING_LEN) {
198                                 ret = -EINVAL;
199                                 goto err_free_acl;
200                         }
201                         ret = uuid_parse(s, &acl[i]);
202                         if (ret)
203                                 goto err_free_acl;
204                 }
205
206                 i++;
207         }
208
209         if (s || i < tb->nboot_acl) {
210                 ret = -EINVAL;
211                 goto err_free_acl;
212         }
213
214         pm_runtime_get_sync(&tb->dev);
215
216         if (mutex_lock_interruptible(&tb->lock)) {
217                 ret = -ERESTARTSYS;
218                 goto err_rpm_put;
219         }
220         ret = tb->cm_ops->set_boot_acl(tb, acl, tb->nboot_acl);
221         if (!ret) {
222                 /* Notify userspace about the change */
223                 kobject_uevent(&tb->dev.kobj, KOBJ_CHANGE);
224         }
225         mutex_unlock(&tb->lock);
226
227 err_rpm_put:
228         pm_runtime_mark_last_busy(&tb->dev);
229         pm_runtime_put_autosuspend(&tb->dev);
230 err_free_acl:
231         kfree(acl);
232 err_free_str:
233         kfree(str);
234
235         return ret ?: count;
236 }
237 static DEVICE_ATTR_RW(boot_acl);
238
239 static ssize_t security_show(struct device *dev, struct device_attribute *attr,
240                              char *buf)
241 {
242         struct tb *tb = container_of(dev, struct tb, dev);
243         const char *name = "unknown";
244
245         if (tb->security_level < ARRAY_SIZE(tb_security_names))
246                 name = tb_security_names[tb->security_level];
247
248         return sprintf(buf, "%s\n", name);
249 }
250 static DEVICE_ATTR_RO(security);
251
252 static struct attribute *domain_attrs[] = {
253         &dev_attr_boot_acl.attr,
254         &dev_attr_security.attr,
255         NULL,
256 };
257
258 static umode_t domain_attr_is_visible(struct kobject *kobj,
259                                       struct attribute *attr, int n)
260 {
261         struct device *dev = container_of(kobj, struct device, kobj);
262         struct tb *tb = container_of(dev, struct tb, dev);
263
264         if (attr == &dev_attr_boot_acl.attr) {
265                 if (tb->nboot_acl &&
266                     tb->cm_ops->get_boot_acl &&
267                     tb->cm_ops->set_boot_acl)
268                         return attr->mode;
269                 return 0;
270         }
271
272         return attr->mode;
273 }
274
275 static struct attribute_group domain_attr_group = {
276         .is_visible = domain_attr_is_visible,
277         .attrs = domain_attrs,
278 };
279
280 static const struct attribute_group *domain_attr_groups[] = {
281         &domain_attr_group,
282         NULL,
283 };
284
285 struct bus_type tb_bus_type = {
286         .name = "thunderbolt",
287         .match = tb_service_match,
288         .probe = tb_service_probe,
289         .remove = tb_service_remove,
290         .shutdown = tb_service_shutdown,
291 };
292
293 static void tb_domain_release(struct device *dev)
294 {
295         struct tb *tb = container_of(dev, struct tb, dev);
296
297         tb_ctl_free(tb->ctl);
298         destroy_workqueue(tb->wq);
299         ida_simple_remove(&tb_domain_ida, tb->index);
300         mutex_destroy(&tb->lock);
301         kfree(tb);
302 }
303
304 struct device_type tb_domain_type = {
305         .name = "thunderbolt_domain",
306         .release = tb_domain_release,
307 };
308
309 /**
310  * tb_domain_alloc() - Allocate a domain
311  * @nhi: Pointer to the host controller
312  * @privsize: Size of the connection manager private data
313  *
314  * Allocates and initializes a new Thunderbolt domain. Connection
315  * managers are expected to call this and then fill in @cm_ops
316  * accordingly.
317  *
318  * Call tb_domain_put() to release the domain before it has been added
319  * to the system.
320  *
321  * Return: allocated domain structure on %NULL in case of error
322  */
323 struct tb *tb_domain_alloc(struct tb_nhi *nhi, size_t privsize)
324 {
325         struct tb *tb;
326
327         /*
328          * Make sure the structure sizes map with that the hardware
329          * expects because bit-fields are being used.
330          */
331         BUILD_BUG_ON(sizeof(struct tb_regs_switch_header) != 5 * 4);
332         BUILD_BUG_ON(sizeof(struct tb_regs_port_header) != 8 * 4);
333         BUILD_BUG_ON(sizeof(struct tb_regs_hop) != 2 * 4);
334
335         tb = kzalloc(sizeof(*tb) + privsize, GFP_KERNEL);
336         if (!tb)
337                 return NULL;
338
339         tb->nhi = nhi;
340         mutex_init(&tb->lock);
341
342         tb->index = ida_simple_get(&tb_domain_ida, 0, 0, GFP_KERNEL);
343         if (tb->index < 0)
344                 goto err_free;
345
346         tb->wq = alloc_ordered_workqueue("thunderbolt%d", 0, tb->index);
347         if (!tb->wq)
348                 goto err_remove_ida;
349
350         tb->dev.parent = &nhi->pdev->dev;
351         tb->dev.bus = &tb_bus_type;
352         tb->dev.type = &tb_domain_type;
353         tb->dev.groups = domain_attr_groups;
354         dev_set_name(&tb->dev, "domain%d", tb->index);
355         device_initialize(&tb->dev);
356
357         return tb;
358
359 err_remove_ida:
360         ida_simple_remove(&tb_domain_ida, tb->index);
361 err_free:
362         kfree(tb);
363
364         return NULL;
365 }
366
367 static bool tb_domain_event_cb(void *data, enum tb_cfg_pkg_type type,
368                                const void *buf, size_t size)
369 {
370         struct tb *tb = data;
371
372         if (!tb->cm_ops->handle_event) {
373                 tb_warn(tb, "domain does not have event handler\n");
374                 return true;
375         }
376
377         switch (type) {
378         case TB_CFG_PKG_XDOMAIN_REQ:
379         case TB_CFG_PKG_XDOMAIN_RESP:
380                 return tb_xdomain_handle_request(tb, type, buf, size);
381
382         default:
383                 tb->cm_ops->handle_event(tb, type, buf, size);
384         }
385
386         return true;
387 }
388
389 /**
390  * tb_domain_add() - Add domain to the system
391  * @tb: Domain to add
392  *
393  * Starts the domain and adds it to the system. Hotplugging devices will
394  * work after this has been returned successfully. In order to remove
395  * and release the domain after this function has been called, call
396  * tb_domain_remove().
397  *
398  * Return: %0 in case of success and negative errno in case of error
399  */
400 int tb_domain_add(struct tb *tb)
401 {
402         int ret;
403
404         if (WARN_ON(!tb->cm_ops))
405                 return -EINVAL;
406
407         mutex_lock(&tb->lock);
408
409         tb->ctl = tb_ctl_alloc(tb->nhi, tb_domain_event_cb, tb);
410         if (!tb->ctl) {
411                 ret = -ENOMEM;
412                 goto err_unlock;
413         }
414
415         /*
416          * tb_schedule_hotplug_handler may be called as soon as the config
417          * channel is started. Thats why we have to hold the lock here.
418          */
419         tb_ctl_start(tb->ctl);
420
421         if (tb->cm_ops->driver_ready) {
422                 ret = tb->cm_ops->driver_ready(tb);
423                 if (ret)
424                         goto err_ctl_stop;
425         }
426
427         ret = device_add(&tb->dev);
428         if (ret)
429                 goto err_ctl_stop;
430
431         /* Start the domain */
432         if (tb->cm_ops->start) {
433                 ret = tb->cm_ops->start(tb);
434                 if (ret)
435                         goto err_domain_del;
436         }
437
438         /* This starts event processing */
439         mutex_unlock(&tb->lock);
440
441         pm_runtime_no_callbacks(&tb->dev);
442         pm_runtime_set_active(&tb->dev);
443         pm_runtime_enable(&tb->dev);
444         pm_runtime_set_autosuspend_delay(&tb->dev, TB_AUTOSUSPEND_DELAY);
445         pm_runtime_mark_last_busy(&tb->dev);
446         pm_runtime_use_autosuspend(&tb->dev);
447
448         return 0;
449
450 err_domain_del:
451         device_del(&tb->dev);
452 err_ctl_stop:
453         tb_ctl_stop(tb->ctl);
454 err_unlock:
455         mutex_unlock(&tb->lock);
456
457         return ret;
458 }
459
460 /**
461  * tb_domain_remove() - Removes and releases a domain
462  * @tb: Domain to remove
463  *
464  * Stops the domain, removes it from the system and releases all
465  * resources once the last reference has been released.
466  */
467 void tb_domain_remove(struct tb *tb)
468 {
469         mutex_lock(&tb->lock);
470         if (tb->cm_ops->stop)
471                 tb->cm_ops->stop(tb);
472         /* Stop the domain control traffic */
473         tb_ctl_stop(tb->ctl);
474         mutex_unlock(&tb->lock);
475
476         flush_workqueue(tb->wq);
477         device_unregister(&tb->dev);
478 }
479
480 /**
481  * tb_domain_suspend_noirq() - Suspend a domain
482  * @tb: Domain to suspend
483  *
484  * Suspends all devices in the domain and stops the control channel.
485  */
486 int tb_domain_suspend_noirq(struct tb *tb)
487 {
488         int ret = 0;
489
490         /*
491          * The control channel interrupt is left enabled during suspend
492          * and taking the lock here prevents any events happening before
493          * we actually have stopped the domain and the control channel.
494          */
495         mutex_lock(&tb->lock);
496         if (tb->cm_ops->suspend_noirq)
497                 ret = tb->cm_ops->suspend_noirq(tb);
498         if (!ret)
499                 tb_ctl_stop(tb->ctl);
500         mutex_unlock(&tb->lock);
501
502         return ret;
503 }
504
505 /**
506  * tb_domain_resume_noirq() - Resume a domain
507  * @tb: Domain to resume
508  *
509  * Re-starts the control channel, and resumes all devices connected to
510  * the domain.
511  */
512 int tb_domain_resume_noirq(struct tb *tb)
513 {
514         int ret = 0;
515
516         mutex_lock(&tb->lock);
517         tb_ctl_start(tb->ctl);
518         if (tb->cm_ops->resume_noirq)
519                 ret = tb->cm_ops->resume_noirq(tb);
520         mutex_unlock(&tb->lock);
521
522         return ret;
523 }
524
525 int tb_domain_suspend(struct tb *tb)
526 {
527         return tb->cm_ops->suspend ? tb->cm_ops->suspend(tb) : 0;
528 }
529
530 void tb_domain_complete(struct tb *tb)
531 {
532         if (tb->cm_ops->complete)
533                 tb->cm_ops->complete(tb);
534 }
535
536 int tb_domain_runtime_suspend(struct tb *tb)
537 {
538         if (tb->cm_ops->runtime_suspend) {
539                 int ret = tb->cm_ops->runtime_suspend(tb);
540                 if (ret)
541                         return ret;
542         }
543         tb_ctl_stop(tb->ctl);
544         return 0;
545 }
546
547 int tb_domain_runtime_resume(struct tb *tb)
548 {
549         tb_ctl_start(tb->ctl);
550         if (tb->cm_ops->runtime_resume) {
551                 int ret = tb->cm_ops->runtime_resume(tb);
552                 if (ret)
553                         return ret;
554         }
555         return 0;
556 }
557
558 /**
559  * tb_domain_approve_switch() - Approve switch
560  * @tb: Domain the switch belongs to
561  * @sw: Switch to approve
562  *
563  * This will approve switch by connection manager specific means. In
564  * case of success the connection manager will create tunnels for all
565  * supported protocols.
566  */
567 int tb_domain_approve_switch(struct tb *tb, struct tb_switch *sw)
568 {
569         struct tb_switch *parent_sw;
570
571         if (!tb->cm_ops->approve_switch)
572                 return -EPERM;
573
574         /* The parent switch must be authorized before this one */
575         parent_sw = tb_to_switch(sw->dev.parent);
576         if (!parent_sw || !parent_sw->authorized)
577                 return -EINVAL;
578
579         return tb->cm_ops->approve_switch(tb, sw);
580 }
581
582 /**
583  * tb_domain_approve_switch_key() - Approve switch and add key
584  * @tb: Domain the switch belongs to
585  * @sw: Switch to approve
586  *
587  * For switches that support secure connect, this function first adds
588  * key to the switch NVM using connection manager specific means. If
589  * adding the key is successful, the switch is approved and connected.
590  *
591  * Return: %0 on success and negative errno in case of failure.
592  */
593 int tb_domain_approve_switch_key(struct tb *tb, struct tb_switch *sw)
594 {
595         struct tb_switch *parent_sw;
596         int ret;
597
598         if (!tb->cm_ops->approve_switch || !tb->cm_ops->add_switch_key)
599                 return -EPERM;
600
601         /* The parent switch must be authorized before this one */
602         parent_sw = tb_to_switch(sw->dev.parent);
603         if (!parent_sw || !parent_sw->authorized)
604                 return -EINVAL;
605
606         ret = tb->cm_ops->add_switch_key(tb, sw);
607         if (ret)
608                 return ret;
609
610         return tb->cm_ops->approve_switch(tb, sw);
611 }
612
613 /**
614  * tb_domain_challenge_switch_key() - Challenge and approve switch
615  * @tb: Domain the switch belongs to
616  * @sw: Switch to approve
617  *
618  * For switches that support secure connect, this function generates
619  * random challenge and sends it to the switch. The switch responds to
620  * this and if the response matches our random challenge, the switch is
621  * approved and connected.
622  *
623  * Return: %0 on success and negative errno in case of failure.
624  */
625 int tb_domain_challenge_switch_key(struct tb *tb, struct tb_switch *sw)
626 {
627         u8 challenge[TB_SWITCH_KEY_SIZE];
628         u8 response[TB_SWITCH_KEY_SIZE];
629         u8 hmac[TB_SWITCH_KEY_SIZE];
630         struct tb_switch *parent_sw;
631         struct crypto_shash *tfm;
632         struct shash_desc *shash;
633         int ret;
634
635         if (!tb->cm_ops->approve_switch || !tb->cm_ops->challenge_switch_key)
636                 return -EPERM;
637
638         /* The parent switch must be authorized before this one */
639         parent_sw = tb_to_switch(sw->dev.parent);
640         if (!parent_sw || !parent_sw->authorized)
641                 return -EINVAL;
642
643         get_random_bytes(challenge, sizeof(challenge));
644         ret = tb->cm_ops->challenge_switch_key(tb, sw, challenge, response);
645         if (ret)
646                 return ret;
647
648         tfm = crypto_alloc_shash("hmac(sha256)", 0, 0);
649         if (IS_ERR(tfm))
650                 return PTR_ERR(tfm);
651
652         ret = crypto_shash_setkey(tfm, sw->key, TB_SWITCH_KEY_SIZE);
653         if (ret)
654                 goto err_free_tfm;
655
656         shash = kzalloc(sizeof(*shash) + crypto_shash_descsize(tfm),
657                         GFP_KERNEL);
658         if (!shash) {
659                 ret = -ENOMEM;
660                 goto err_free_tfm;
661         }
662
663         shash->tfm = tfm;
664         shash->flags = CRYPTO_TFM_REQ_MAY_SLEEP;
665
666         memset(hmac, 0, sizeof(hmac));
667         ret = crypto_shash_digest(shash, challenge, sizeof(hmac), hmac);
668         if (ret)
669                 goto err_free_shash;
670
671         /* The returned HMAC must match the one we calculated */
672         if (memcmp(response, hmac, sizeof(hmac))) {
673                 ret = -EKEYREJECTED;
674                 goto err_free_shash;
675         }
676
677         crypto_free_shash(tfm);
678         kfree(shash);
679
680         return tb->cm_ops->approve_switch(tb, sw);
681
682 err_free_shash:
683         kfree(shash);
684 err_free_tfm:
685         crypto_free_shash(tfm);
686
687         return ret;
688 }
689
690 /**
691  * tb_domain_disconnect_pcie_paths() - Disconnect all PCIe paths
692  * @tb: Domain whose PCIe paths to disconnect
693  *
694  * This needs to be called in preparation for NVM upgrade of the host
695  * controller. Makes sure all PCIe paths are disconnected.
696  *
697  * Return %0 on success and negative errno in case of error.
698  */
699 int tb_domain_disconnect_pcie_paths(struct tb *tb)
700 {
701         if (!tb->cm_ops->disconnect_pcie_paths)
702                 return -EPERM;
703
704         return tb->cm_ops->disconnect_pcie_paths(tb);
705 }
706
707 /**
708  * tb_domain_approve_xdomain_paths() - Enable DMA paths for XDomain
709  * @tb: Domain enabling the DMA paths
710  * @xd: XDomain DMA paths are created to
711  *
712  * Calls connection manager specific method to enable DMA paths to the
713  * XDomain in question.
714  *
715  * Return: 0% in case of success and negative errno otherwise. In
716  * particular returns %-ENOTSUPP if the connection manager
717  * implementation does not support XDomains.
718  */
719 int tb_domain_approve_xdomain_paths(struct tb *tb, struct tb_xdomain *xd)
720 {
721         if (!tb->cm_ops->approve_xdomain_paths)
722                 return -ENOTSUPP;
723
724         return tb->cm_ops->approve_xdomain_paths(tb, xd);
725 }
726
727 /**
728  * tb_domain_disconnect_xdomain_paths() - Disable DMA paths for XDomain
729  * @tb: Domain disabling the DMA paths
730  * @xd: XDomain whose DMA paths are disconnected
731  *
732  * Calls connection manager specific method to disconnect DMA paths to
733  * the XDomain in question.
734  *
735  * Return: 0% in case of success and negative errno otherwise. In
736  * particular returns %-ENOTSUPP if the connection manager
737  * implementation does not support XDomains.
738  */
739 int tb_domain_disconnect_xdomain_paths(struct tb *tb, struct tb_xdomain *xd)
740 {
741         if (!tb->cm_ops->disconnect_xdomain_paths)
742                 return -ENOTSUPP;
743
744         return tb->cm_ops->disconnect_xdomain_paths(tb, xd);
745 }
746
747 static int disconnect_xdomain(struct device *dev, void *data)
748 {
749         struct tb_xdomain *xd;
750         struct tb *tb = data;
751         int ret = 0;
752
753         xd = tb_to_xdomain(dev);
754         if (xd && xd->tb == tb)
755                 ret = tb_xdomain_disable_paths(xd);
756
757         return ret;
758 }
759
760 /**
761  * tb_domain_disconnect_all_paths() - Disconnect all paths for the domain
762  * @tb: Domain whose paths are disconnected
763  *
764  * This function can be used to disconnect all paths (PCIe, XDomain) for
765  * example in preparation for host NVM firmware upgrade. After this is
766  * called the paths cannot be established without resetting the switch.
767  *
768  * Return: %0 in case of success and negative errno otherwise.
769  */
770 int tb_domain_disconnect_all_paths(struct tb *tb)
771 {
772         int ret;
773
774         ret = tb_domain_disconnect_pcie_paths(tb);
775         if (ret)
776                 return ret;
777
778         return bus_for_each_dev(&tb_bus_type, NULL, tb, disconnect_xdomain);
779 }
780
781 int tb_domain_init(void)
782 {
783         int ret;
784
785         ret = tb_xdomain_init();
786         if (ret)
787                 return ret;
788         ret = bus_register(&tb_bus_type);
789         if (ret)
790                 tb_xdomain_exit();
791
792         return ret;
793 }
794
795 void tb_domain_exit(void)
796 {
797         bus_unregister(&tb_bus_type);
798         ida_destroy(&tb_domain_ida);
799         tb_switch_exit();
800         tb_xdomain_exit();
801 }