io_trapped.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297
  1. /*
  2. * Trapped io support
  3. *
  4. * Copyright (C) 2008 Magnus Damm
  5. *
  6. * Intercept io operations by trapping.
  7. *
  8. * This file is subject to the terms and conditions of the GNU General Public
  9. * License. See the file "COPYING" in the main directory of this archive
  10. * for more details.
  11. */
  12. #include <linux/kernel.h>
  13. #include <linux/mm.h>
  14. #include <linux/bitops.h>
  15. #include <linux/vmalloc.h>
  16. #include <linux/module.h>
  17. #include <linux/init.h>
  18. #include <asm/system.h>
  19. #include <asm/mmu_context.h>
  20. #include <asm/uaccess.h>
  21. #include <asm/io.h>
  22. #include <asm/io_trapped.h>
  23. #define TRAPPED_PAGES_MAX 16
  24. #ifdef CONFIG_HAS_IOPORT
  25. LIST_HEAD(trapped_io);
  26. EXPORT_SYMBOL_GPL(trapped_io);
  27. #endif
  28. #ifdef CONFIG_HAS_IOMEM
  29. LIST_HEAD(trapped_mem);
  30. EXPORT_SYMBOL_GPL(trapped_mem);
  31. #endif
  32. static DEFINE_SPINLOCK(trapped_lock);
  33. static int trapped_io_disable __read_mostly;
  34. static int __init trapped_io_setup(char *__unused)
  35. {
  36. trapped_io_disable = 1;
  37. return 1;
  38. }
  39. __setup("noiotrap", trapped_io_setup);
  40. int register_trapped_io(struct trapped_io *tiop)
  41. {
  42. struct resource *res;
  43. unsigned long len = 0, flags = 0;
  44. struct page *pages[TRAPPED_PAGES_MAX];
  45. int k, n;
  46. if (unlikely(trapped_io_disable))
  47. return 0;
  48. /* structure must be page aligned */
  49. if ((unsigned long)tiop & (PAGE_SIZE - 1))
  50. goto bad;
  51. for (k = 0; k < tiop->num_resources; k++) {
  52. res = tiop->resource + k;
  53. len += roundup((res->end - res->start) + 1, PAGE_SIZE);
  54. flags |= res->flags;
  55. }
  56. /* support IORESOURCE_IO _or_ MEM, not both */
  57. if (hweight_long(flags) != 1)
  58. goto bad;
  59. n = len >> PAGE_SHIFT;
  60. if (n >= TRAPPED_PAGES_MAX)
  61. goto bad;
  62. for (k = 0; k < n; k++)
  63. pages[k] = virt_to_page(tiop);
  64. tiop->virt_base = vmap(pages, n, VM_MAP, PAGE_NONE);
  65. if (!tiop->virt_base)
  66. goto bad;
  67. len = 0;
  68. for (k = 0; k < tiop->num_resources; k++) {
  69. res = tiop->resource + k;
  70. pr_info("trapped io 0x%08lx overrides %s 0x%08lx\n",
  71. (unsigned long)(tiop->virt_base + len),
  72. res->flags & IORESOURCE_IO ? "io" : "mmio",
  73. (unsigned long)res->start);
  74. len += roundup((res->end - res->start) + 1, PAGE_SIZE);
  75. }
  76. tiop->magic = IO_TRAPPED_MAGIC;
  77. INIT_LIST_HEAD(&tiop->list);
  78. spin_lock_irq(&trapped_lock);
  79. #ifdef CONFIG_HAS_IOPORT
  80. if (flags & IORESOURCE_IO)
  81. list_add(&tiop->list, &trapped_io);
  82. #endif
  83. #ifdef CONFIG_HAS_IOMEM
  84. if (flags & IORESOURCE_MEM)
  85. list_add(&tiop->list, &trapped_mem);
  86. #endif
  87. spin_unlock_irq(&trapped_lock);
  88. return 0;
  89. bad:
  90. pr_warning("unable to install trapped io filter\n");
  91. return -1;
  92. }
  93. EXPORT_SYMBOL_GPL(register_trapped_io);
  94. void __iomem *match_trapped_io_handler(struct list_head *list,
  95. unsigned long offset,
  96. unsigned long size)
  97. {
  98. unsigned long voffs;
  99. struct trapped_io *tiop;
  100. struct resource *res;
  101. int k, len;
  102. unsigned long flags;
  103. spin_lock_irqsave(&trapped_lock, flags);
  104. list_for_each_entry(tiop, list, list) {
  105. voffs = 0;
  106. for (k = 0; k < tiop->num_resources; k++) {
  107. res = tiop->resource + k;
  108. if (res->start == offset) {
  109. spin_unlock_irqrestore(&trapped_lock, flags);
  110. return tiop->virt_base + voffs;
  111. }
  112. len = (res->end - res->start) + 1;
  113. voffs += roundup(len, PAGE_SIZE);
  114. }
  115. }
  116. spin_unlock_irqrestore(&trapped_lock, flags);
  117. return NULL;
  118. }
  119. EXPORT_SYMBOL_GPL(match_trapped_io_handler);
  120. static struct trapped_io *lookup_tiop(unsigned long address)
  121. {
  122. pgd_t *pgd_k;
  123. pud_t *pud_k;
  124. pmd_t *pmd_k;
  125. pte_t *pte_k;
  126. pte_t entry;
  127. pgd_k = swapper_pg_dir + pgd_index(address);
  128. if (!pgd_present(*pgd_k))
  129. return NULL;
  130. pud_k = pud_offset(pgd_k, address);
  131. if (!pud_present(*pud_k))
  132. return NULL;
  133. pmd_k = pmd_offset(pud_k, address);
  134. if (!pmd_present(*pmd_k))
  135. return NULL;
  136. pte_k = pte_offset_kernel(pmd_k, address);
  137. entry = *pte_k;
  138. return pfn_to_kaddr(pte_pfn(entry));
  139. }
  140. static unsigned long lookup_address(struct trapped_io *tiop,
  141. unsigned long address)
  142. {
  143. struct resource *res;
  144. unsigned long vaddr = (unsigned long)tiop->virt_base;
  145. unsigned long len;
  146. int k;
  147. for (k = 0; k < tiop->num_resources; k++) {
  148. res = tiop->resource + k;
  149. len = roundup((res->end - res->start) + 1, PAGE_SIZE);
  150. if (address < (vaddr + len))
  151. return res->start + (address - vaddr);
  152. vaddr += len;
  153. }
  154. return 0;
  155. }
  156. static unsigned long long copy_word(unsigned long src_addr, int src_len,
  157. unsigned long dst_addr, int dst_len)
  158. {
  159. unsigned long long tmp = 0;
  160. switch (src_len) {
  161. case 1:
  162. tmp = __raw_readb(src_addr);
  163. break;
  164. case 2:
  165. tmp = __raw_readw(src_addr);
  166. break;
  167. case 4:
  168. tmp = __raw_readl(src_addr);
  169. break;
  170. case 8:
  171. tmp = __raw_readq(src_addr);
  172. break;
  173. }
  174. switch (dst_len) {
  175. case 1:
  176. __raw_writeb(tmp, dst_addr);
  177. break;
  178. case 2:
  179. __raw_writew(tmp, dst_addr);
  180. break;
  181. case 4:
  182. __raw_writel(tmp, dst_addr);
  183. break;
  184. case 8:
  185. __raw_writeq(tmp, dst_addr);
  186. break;
  187. }
  188. return tmp;
  189. }
  190. static unsigned long from_device(void *dst, const void *src, unsigned long cnt)
  191. {
  192. struct trapped_io *tiop;
  193. unsigned long src_addr = (unsigned long)src;
  194. unsigned long long tmp;
  195. pr_debug("trapped io read 0x%08lx (%ld)\n", src_addr, cnt);
  196. tiop = lookup_tiop(src_addr);
  197. WARN_ON(!tiop || (tiop->magic != IO_TRAPPED_MAGIC));
  198. src_addr = lookup_address(tiop, src_addr);
  199. if (!src_addr)
  200. return cnt;
  201. tmp = copy_word(src_addr,
  202. max_t(unsigned long, cnt,
  203. (tiop->minimum_bus_width / 8)),
  204. (unsigned long)dst, cnt);
  205. pr_debug("trapped io read 0x%08lx -> 0x%08llx\n", src_addr, tmp);
  206. return 0;
  207. }
  208. static unsigned long to_device(void *dst, const void *src, unsigned long cnt)
  209. {
  210. struct trapped_io *tiop;
  211. unsigned long dst_addr = (unsigned long)dst;
  212. unsigned long long tmp;
  213. pr_debug("trapped io write 0x%08lx (%ld)\n", dst_addr, cnt);
  214. tiop = lookup_tiop(dst_addr);
  215. WARN_ON(!tiop || (tiop->magic != IO_TRAPPED_MAGIC));
  216. dst_addr = lookup_address(tiop, dst_addr);
  217. if (!dst_addr)
  218. return cnt;
  219. tmp = copy_word((unsigned long)src, cnt,
  220. dst_addr, max_t(unsigned long, cnt,
  221. (tiop->minimum_bus_width / 8)));
  222. pr_debug("trapped io write 0x%08lx -> 0x%08llx\n", dst_addr, tmp);
  223. return 0;
  224. }
  225. static struct mem_access trapped_io_access = {
  226. from_device,
  227. to_device,
  228. };
  229. int handle_trapped_io(struct pt_regs *regs, unsigned long address)
  230. {
  231. mm_segment_t oldfs;
  232. insn_size_t instruction;
  233. int tmp;
  234. if (trapped_io_disable)
  235. return 0;
  236. if (!lookup_tiop(address))
  237. return 0;
  238. WARN_ON(user_mode(regs));
  239. oldfs = get_fs();
  240. set_fs(KERNEL_DS);
  241. if (copy_from_user(&instruction, (void *)(regs->pc),
  242. sizeof(instruction))) {
  243. set_fs(oldfs);
  244. return 0;
  245. }
  246. tmp = handle_unaligned_access(instruction, regs,
  247. &trapped_io_access, 1);
  248. set_fs(oldfs);
  249. return tmp == 0;
  250. }