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
;
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
);
206 rwlock_init(&p2m_lock
);
209 arch_initcall(p2m_init
);