1 #include <linux/bootmem.h>
3 #include <linux/export.h>
4 #include <linux/rwlock.h>
5 #include <linux/slab.h>
6 #include <linux/types.h>
7 #include <linux/dma-mapping.h>
8 #include <linux/vmalloc.h>
9 #include <linux/swiotlb.h>
12 #include <xen/interface/memory.h>
13 #include <xen/swiotlb-xen.h>
15 #include <asm/cacheflush.h>
16 #include <asm/xen/page.h>
17 #include <asm/xen/hypercall.h>
18 #include <asm/xen/interface.h>
20 struct xen_p2m_entry {
23 unsigned long nr_pages;
24 struct rb_node rbnode_mach;
25 struct rb_node rbnode_phys;
28 static rwlock_t p2m_lock;
29 struct rb_root phys_to_mach = RB_ROOT;
30 EXPORT_SYMBOL_GPL(phys_to_mach);
31 static struct rb_root mach_to_phys = RB_ROOT;
33 static int xen_add_phys_to_mach_entry(struct xen_p2m_entry *new)
35 struct rb_node **link = &phys_to_mach.rb_node;
36 struct rb_node *parent = NULL;
37 struct xen_p2m_entry *entry;
42 entry = rb_entry(parent, struct xen_p2m_entry, rbnode_phys);
44 if (new->mfn == entry->mfn)
46 if (new->pfn == entry->pfn)
49 if (new->pfn < entry->pfn)
50 link = &(*link)->rb_left;
52 link = &(*link)->rb_right;
54 rb_link_node(&new->rbnode_phys, parent, link);
55 rb_insert_color(&new->rbnode_phys, &phys_to_mach);
60 pr_warn("%s: cannot add pfn=%pa -> mfn=%pa: pfn=%pa -> mfn=%pa already exists\n",
61 __func__, &new->pfn, &new->mfn, &entry->pfn, &entry->mfn);
66 unsigned long __pfn_to_mfn(unsigned long pfn)
68 struct rb_node *n = phys_to_mach.rb_node;
69 struct xen_p2m_entry *entry;
70 unsigned long irqflags;
72 read_lock_irqsave(&p2m_lock, irqflags);
74 entry = rb_entry(n, struct xen_p2m_entry, rbnode_phys);
75 if (entry->pfn <= pfn &&
76 entry->pfn + entry->nr_pages > pfn) {
77 read_unlock_irqrestore(&p2m_lock, irqflags);
78 return entry->mfn + (pfn - entry->pfn);
85 read_unlock_irqrestore(&p2m_lock, irqflags);
87 return INVALID_P2M_ENTRY;
89 EXPORT_SYMBOL_GPL(__pfn_to_mfn);
91 static int xen_add_mach_to_phys_entry(struct xen_p2m_entry *new)
93 struct rb_node **link = &mach_to_phys.rb_node;
94 struct rb_node *parent = NULL;
95 struct xen_p2m_entry *entry;
100 entry = rb_entry(parent, struct xen_p2m_entry, rbnode_mach);
102 if (new->mfn == entry->mfn)
104 if (new->pfn == entry->pfn)
107 if (new->mfn < entry->mfn)
108 link = &(*link)->rb_left;
110 link = &(*link)->rb_right;
112 rb_link_node(&new->rbnode_mach, parent, link);
113 rb_insert_color(&new->rbnode_mach, &mach_to_phys);
118 pr_warn("%s: cannot add pfn=%pa -> mfn=%pa: pfn=%pa -> mfn=%pa already exists\n",
119 __func__, &new->pfn, &new->mfn, &entry->pfn, &entry->mfn);
124 unsigned long __mfn_to_pfn(unsigned long mfn)
126 struct rb_node *n = mach_to_phys.rb_node;
127 struct xen_p2m_entry *entry;
128 unsigned long irqflags;
130 read_lock_irqsave(&p2m_lock, irqflags);
132 entry = rb_entry(n, struct xen_p2m_entry, rbnode_mach);
133 if (entry->mfn <= mfn &&
134 entry->mfn + entry->nr_pages > mfn) {
135 read_unlock_irqrestore(&p2m_lock, irqflags);
136 return entry->pfn + (mfn - entry->mfn);
138 if (mfn < entry->mfn)
143 read_unlock_irqrestore(&p2m_lock, irqflags);
145 return INVALID_P2M_ENTRY;
147 EXPORT_SYMBOL_GPL(__mfn_to_pfn);
149 bool __set_phys_to_machine_multi(unsigned long pfn,
150 unsigned long mfn, unsigned long nr_pages)
153 unsigned long irqflags;
154 struct xen_p2m_entry *p2m_entry;
155 struct rb_node *n = phys_to_mach.rb_node;
157 if (mfn == INVALID_P2M_ENTRY) {
158 write_lock_irqsave(&p2m_lock, irqflags);
160 p2m_entry = rb_entry(n, struct xen_p2m_entry, rbnode_phys);
161 if (p2m_entry->pfn <= pfn &&
162 p2m_entry->pfn + p2m_entry->nr_pages > pfn) {
163 rb_erase(&p2m_entry->rbnode_mach, &mach_to_phys);
164 rb_erase(&p2m_entry->rbnode_phys, &phys_to_mach);
165 write_unlock_irqrestore(&p2m_lock, irqflags);
169 if (pfn < p2m_entry->pfn)
174 write_unlock_irqrestore(&p2m_lock, irqflags);
178 p2m_entry = kzalloc(sizeof(struct xen_p2m_entry), GFP_NOWAIT);
180 pr_warn("cannot allocate xen_p2m_entry\n");
183 p2m_entry->pfn = pfn;
184 p2m_entry->nr_pages = nr_pages;
185 p2m_entry->mfn = mfn;
187 write_lock_irqsave(&p2m_lock, irqflags);
188 if ((rc = xen_add_phys_to_mach_entry(p2m_entry) < 0) ||
189 (rc = xen_add_mach_to_phys_entry(p2m_entry) < 0)) {
190 write_unlock_irqrestore(&p2m_lock, irqflags);
193 write_unlock_irqrestore(&p2m_lock, irqflags);
196 EXPORT_SYMBOL_GPL(__set_phys_to_machine_multi);
198 bool __set_phys_to_machine(unsigned long pfn, unsigned long mfn)
200 return __set_phys_to_machine_multi(pfn, mfn, 1);
202 EXPORT_SYMBOL_GPL(__set_phys_to_machine);
204 static int p2m_init(void)
206 rwlock_init(&p2m_lock);
209 arch_initcall(p2m_init);