151 static inline unsigned int index_from_irqs(struct vmd_dev *vmd,  in index_from_irqs()  argument
154 	return irqs - vmd->irqs;  in index_from_irqs()
169 	struct vmd_dev *vmd = irq_data_get_irq_handler_data(data);  in vmd_compose_msi_msg()  local
174 	msg->arch_addr_lo.destid_0_7 = index_from_irqs(vmd, irq);  in vmd_compose_msi_msg()
226 static struct vmd_irq_list *vmd_next_irq(struct vmd_dev *vmd, struct msi_desc *desc)  in vmd_next_irq()  argument
231 	if (vmd->msix_count == 1 + vmd->first_vec)  in vmd_next_irq()
232 		return &vmd->irqs[vmd->first_vec];  in vmd_next_irq()
242 		return &vmd->irqs[vmd->first_vec];  in vmd_next_irq()
246 	best = vmd->first_vec + 1;  in vmd_next_irq()
247 	for (i = best; i < vmd->msix_count; i++)  in vmd_next_irq()
248 		if (vmd->irqs[i].count < vmd->irqs[best].count)  in vmd_next_irq()
250 	vmd->irqs[best].count++;  in vmd_next_irq()
253 	return &vmd->irqs[best];  in vmd_next_irq()
261 	struct vmd_dev *vmd = vmd_from_bus(msi_desc_to_pci_dev(desc)->bus);  in vmd_msi_init()  local
268 	vmdirq->irq = vmd_next_irq(vmd, desc);  in vmd_msi_init()
272 			    handle_untracked_irq, vmd, NULL);  in vmd_msi_init()
296 	struct vmd_dev *vmd = vmd_from_bus(pdev->bus);  in vmd_msi_prepare()  local
298 	if (nvec > vmd->msix_count)  in vmd_msi_prepare()
299 		return vmd->msix_count;  in vmd_msi_prepare()
325 static void vmd_set_msi_remapping(struct vmd_dev *vmd, bool enable)  in vmd_set_msi_remapping()  argument
329 	pci_read_config_word(vmd->dev, PCI_REG_VMCONFIG, &reg);  in vmd_set_msi_remapping()
332 	pci_write_config_word(vmd->dev, PCI_REG_VMCONFIG, reg);  in vmd_set_msi_remapping()
335 static int vmd_create_irq_domain(struct vmd_dev *vmd)  in vmd_create_irq_domain()  argument
339 	fn = irq_domain_alloc_named_id_fwnode("VMD-MSI", vmd->sysdata.domain);  in vmd_create_irq_domain()
343 	vmd->irq_domain = pci_msi_create_irq_domain(fn, &vmd_msi_domain_info, NULL);  in vmd_create_irq_domain()
344 	if (!vmd->irq_domain) {  in vmd_create_irq_domain()
352 static void vmd_remove_irq_domain(struct vmd_dev *vmd)  in vmd_remove_irq_domain()  argument
358 	if (!vmd->msix_count)  in vmd_remove_irq_domain()
359 		vmd_set_msi_remapping(vmd, true);  in vmd_remove_irq_domain()
361 	if (vmd->irq_domain) {  in vmd_remove_irq_domain()
362 		struct fwnode_handle *fn = vmd->irq_domain->fwnode;  in vmd_remove_irq_domain()
364 		irq_domain_remove(vmd->irq_domain);  in vmd_remove_irq_domain()
369 static void __iomem *vmd_cfg_addr(struct vmd_dev *vmd, struct pci_bus *bus,  in vmd_cfg_addr()  argument
372 	unsigned int busnr_ecam = bus->number - vmd->busn_start;  in vmd_cfg_addr()
375 	if (offset + len >= resource_size(&vmd->dev->resource[VMD_CFGBAR]))  in vmd_cfg_addr()
378 	return vmd->cfgbar + offset;  in vmd_cfg_addr()
388 	struct vmd_dev *vmd = vmd_from_bus(bus);  in vmd_pci_read()  local
389 	void __iomem *addr = vmd_cfg_addr(vmd, bus, devfn, reg, len);  in vmd_pci_read()
396 	raw_spin_lock_irqsave(&vmd->cfg_lock, flags);  in vmd_pci_read()
411 	raw_spin_unlock_irqrestore(&vmd->cfg_lock, flags);  in vmd_pci_read()
423 	struct vmd_dev *vmd = vmd_from_bus(bus);  in vmd_pci_write()  local
424 	void __iomem *addr = vmd_cfg_addr(vmd, bus, devfn, reg, len);  in vmd_pci_write()
431 	raw_spin_lock_irqsave(&vmd->cfg_lock, flags);  in vmd_pci_write()
449 	raw_spin_unlock_irqrestore(&vmd->cfg_lock, flags);  in vmd_pci_write()
508 static void vmd_domain_reset(struct vmd_dev *vmd)  in vmd_domain_reset()  argument
510 	u16 bus, max_buses = resource_size(&vmd->resources[0]);  in vmd_domain_reset()
516 			base = vmd->cfgbar + PCIE_ECAM_OFFSET(bus,  in vmd_domain_reset()
523 				base = vmd->cfgbar + PCIE_ECAM_OFFSET(bus,  in vmd_domain_reset()
556 static void vmd_attach_resources(struct vmd_dev *vmd)  in vmd_attach_resources()  argument
558 	vmd->dev->resource[VMD_MEMBAR1].child = &vmd->resources[1];  in vmd_attach_resources()
559 	vmd->dev->resource[VMD_MEMBAR2].child = &vmd->resources[2];  in vmd_attach_resources()
562 static void vmd_detach_resources(struct vmd_dev *vmd)  in vmd_detach_resources()  argument
564 	vmd->dev->resource[VMD_MEMBAR1].child = NULL;  in vmd_detach_resources()
565 	vmd->dev->resource[VMD_MEMBAR2].child = NULL;  in vmd_detach_resources()
584 static int vmd_get_phys_offsets(struct vmd_dev *vmd, bool native_hint,  in vmd_get_phys_offsets()  argument
588 	struct pci_dev *dev = vmd->dev;  in vmd_get_phys_offsets()
638 static int vmd_get_bus_number_start(struct vmd_dev *vmd)  in vmd_get_bus_number_start()  argument
640 	struct pci_dev *dev = vmd->dev;  in vmd_get_bus_number_start()
649 			vmd->busn_start = 0;  in vmd_get_bus_number_start()
652 			vmd->busn_start = 128;  in vmd_get_bus_number_start()
655 			vmd->busn_start = 224;  in vmd_get_bus_number_start()
681 static int vmd_alloc_irqs(struct vmd_dev *vmd)  in vmd_alloc_irqs()  argument
683 	struct pci_dev *dev = vmd->dev;  in vmd_alloc_irqs()
686 	vmd->msix_count = pci_msix_vec_count(dev);  in vmd_alloc_irqs()
687 	if (vmd->msix_count < 0)  in vmd_alloc_irqs()
690 	vmd->msix_count = pci_alloc_irq_vectors(dev, vmd->first_vec + 1,  in vmd_alloc_irqs()
691 						vmd->msix_count, PCI_IRQ_MSIX);  in vmd_alloc_irqs()
692 	if (vmd->msix_count < 0)  in vmd_alloc_irqs()
693 		return vmd->msix_count;  in vmd_alloc_irqs()
695 	vmd->irqs = devm_kcalloc(&dev->dev, vmd->msix_count, sizeof(*vmd->irqs),  in vmd_alloc_irqs()
697 	if (!vmd->irqs)  in vmd_alloc_irqs()
700 	for (i = 0; i < vmd->msix_count; i++) {  in vmd_alloc_irqs()
701 		err = init_srcu_struct(&vmd->irqs[i].srcu);  in vmd_alloc_irqs()
705 		INIT_LIST_HEAD(&vmd->irqs[i].irq_list);  in vmd_alloc_irqs()
706 		vmd->irqs[i].virq = pci_irq_vector(dev, i);  in vmd_alloc_irqs()
707 		err = devm_request_irq(&dev->dev, vmd->irqs[i].virq,  in vmd_alloc_irqs()
709 				       vmd->name, &vmd->irqs[i]);  in vmd_alloc_irqs()
777 static int vmd_enable_domain(struct vmd_dev *vmd, unsigned long features)  in vmd_enable_domain()  argument
779 	struct pci_sysdata *sd = &vmd->sysdata;  in vmd_enable_domain()
798 		ret = vmd_get_phys_offsets(vmd, true, &offset[0], &offset[1]);  in vmd_enable_domain()
802 		ret = vmd_get_phys_offsets(vmd, false, &offset[0], &offset[1]);  in vmd_enable_domain()
812 		ret = vmd_get_bus_number_start(vmd);  in vmd_enable_domain()
817 	res = &vmd->dev->resource[VMD_CFGBAR];  in vmd_enable_domain()
818 	vmd->resources[0] = (struct resource) {  in vmd_enable_domain()
820 		.start = vmd->busn_start,  in vmd_enable_domain()
821 		.end   = vmd->busn_start + (resource_size(res) >> 20) - 1,  in vmd_enable_domain()
842 	res = &vmd->dev->resource[VMD_MEMBAR1];  in vmd_enable_domain()
847 	vmd->resources[1] = (struct resource) {  in vmd_enable_domain()
855 	res = &vmd->dev->resource[VMD_MEMBAR2];  in vmd_enable_domain()
860 	vmd->resources[2] = (struct resource) {  in vmd_enable_domain()
868 	sd->vmd_dev = vmd->dev;  in vmd_enable_domain()
873 	sd->node = pcibus_to_node(vmd->dev->bus);  in vmd_enable_domain()
883 		ret = vmd_alloc_irqs(vmd);  in vmd_enable_domain()
887 		vmd_set_msi_remapping(vmd, true);  in vmd_enable_domain()
889 		ret = vmd_create_irq_domain(vmd);  in vmd_enable_domain()
897 		irq_domain_update_bus_token(vmd->irq_domain, DOMAIN_BUS_VMD_MSI);  in vmd_enable_domain()
899 		vmd_set_msi_remapping(vmd, false);  in vmd_enable_domain()
902 	pci_add_resource(&resources, &vmd->resources[0]);  in vmd_enable_domain()
903 	pci_add_resource_offset(&resources, &vmd->resources[1], offset[0]);  in vmd_enable_domain()
904 	pci_add_resource_offset(&resources, &vmd->resources[2], offset[1]);  in vmd_enable_domain()
906 	vmd->bus = pci_create_root_bus(&vmd->dev->dev, vmd->busn_start,  in vmd_enable_domain()
908 	if (!vmd->bus) {  in vmd_enable_domain()
910 		vmd_remove_irq_domain(vmd);  in vmd_enable_domain()
914 	vmd_copy_host_bridge_flags(pci_find_host_bridge(vmd->dev->bus),  in vmd_enable_domain()
915 				   to_pci_host_bridge(vmd->bus->bridge));  in vmd_enable_domain()
917 	vmd_attach_resources(vmd);  in vmd_enable_domain()
918 	if (vmd->irq_domain)  in vmd_enable_domain()
919 		dev_set_msi_domain(&vmd->bus->dev, vmd->irq_domain);  in vmd_enable_domain()
921 		dev_set_msi_domain(&vmd->bus->dev,  in vmd_enable_domain()
922 				   dev_get_msi_domain(&vmd->dev->dev));  in vmd_enable_domain()
924 	WARN(sysfs_create_link(&vmd->dev->dev.kobj, &vmd->bus->dev.kobj,  in vmd_enable_domain()
929 	pci_scan_child_bus(vmd->bus);  in vmd_enable_domain()
930 	vmd_domain_reset(vmd);  in vmd_enable_domain()
939 	list_for_each_entry(child, &vmd->bus->children, node) {  in vmd_enable_domain()
951 	pci_assign_unassigned_bus_resources(vmd->bus);  in vmd_enable_domain()
953 	pci_walk_bus(vmd->bus, vmd_pm_enable_quirk, &features);  in vmd_enable_domain()
960 	list_for_each_entry(child, &vmd->bus->children, node)  in vmd_enable_domain()
963 	pci_bus_add_devices(vmd->bus);  in vmd_enable_domain()
972 	struct vmd_dev *vmd;  in vmd_probe()  local
996 	vmd = devm_kzalloc(&dev->dev, sizeof(*vmd), GFP_KERNEL);  in vmd_probe()
997 	if (!vmd)  in vmd_probe()
1000 	vmd->dev = dev;  in vmd_probe()
1001 	vmd->instance = ida_alloc(&vmd_instance_ida, GFP_KERNEL);  in vmd_probe()
1002 	if (vmd->instance < 0)  in vmd_probe()
1003 		return vmd->instance;  in vmd_probe()
1005 	vmd->name = devm_kasprintf(&dev->dev, GFP_KERNEL, "vmd%d",  in vmd_probe()
1006 				   vmd->instance);  in vmd_probe()
1007 	if (!vmd->name) {  in vmd_probe()
1016 	vmd->cfgbar = pcim_iomap(dev, VMD_CFGBAR, 0);  in vmd_probe()
1017 	if (!vmd->cfgbar) {  in vmd_probe()
1030 		vmd->first_vec = 1;  in vmd_probe()
1032 	raw_spin_lock_init(&vmd->cfg_lock);  in vmd_probe()
1033 	pci_set_drvdata(dev, vmd);  in vmd_probe()
1034 	err = vmd_enable_domain(vmd, features);  in vmd_probe()
1038 	dev_info(&vmd->dev->dev, "Bound to PCI domain %04x\n",  in vmd_probe()
1039 		 vmd->sysdata.domain);  in vmd_probe()
1043 	ida_free(&vmd_instance_ida, vmd->instance);  in vmd_probe()
1047 static void vmd_cleanup_srcu(struct vmd_dev *vmd)  in vmd_cleanup_srcu()  argument
1051 	for (i = 0; i < vmd->msix_count; i++)  in vmd_cleanup_srcu()
1052 		cleanup_srcu_struct(&vmd->irqs[i].srcu);  in vmd_cleanup_srcu()
1057 	struct vmd_dev *vmd = pci_get_drvdata(dev);  in vmd_remove()  local
1059 	pci_stop_root_bus(vmd->bus);  in vmd_remove()
1060 	sysfs_remove_link(&vmd->dev->dev.kobj, "domain");  in vmd_remove()
1061 	pci_remove_root_bus(vmd->bus);  in vmd_remove()
1062 	vmd_cleanup_srcu(vmd);  in vmd_remove()
1063 	vmd_detach_resources(vmd);  in vmd_remove()
1064 	vmd_remove_irq_domain(vmd);  in vmd_remove()
1065 	ida_free(&vmd_instance_ida, vmd->instance);  in vmd_remove()
1070 	struct vmd_dev *vmd = pci_get_drvdata(dev);  in vmd_shutdown()  local
1072 	vmd_remove_irq_domain(vmd);  in vmd_shutdown()
1079 	struct vmd_dev *vmd = pci_get_drvdata(pdev);  in vmd_suspend()  local
1082 	for (i = 0; i < vmd->msix_count; i++)  in vmd_suspend()
1083 		devm_free_irq(dev, vmd->irqs[i].virq, &vmd->irqs[i]);  in vmd_suspend()
1091 	struct vmd_dev *vmd = pci_get_drvdata(pdev);  in vmd_resume()  local
1094 	vmd_set_msi_remapping(vmd, !!vmd->irq_domain);  in vmd_resume()
1096 	for (i = 0; i < vmd->msix_count; i++) {  in vmd_resume()
1097 		err = devm_request_irq(dev, vmd->irqs[i].virq,  in vmd_resume()
1099 				       vmd->name, &vmd->irqs[i]);  in vmd_resume()