git.joshuawise.com / netwatch.git / blame
| Commit | Line | Data |
|---|---|---|
| 05c5b136 JP |
1 | #include <stdint.h> |
| 2 | #include "packet.h" | |
| 3 | ||
| 4 | #define PTE_FOR(x) (((unsigned int)(x) >> 12) & 0x3FF) | |
| 5 | #define PDE_FOR(x) ((unsigned int)(x) >> 22) | |
| 6 | #define ADDR_12_MASK(x) ((unsigned int)(x) & ~((1 << 12) - 1)) | |
| 7 | #define ADDR_22_MASK(x) ((unsigned int)(x) & ~((1 << 22) - 1)) | |
| 8 | #define LOWER_12(x) ((unsigned int)(x) & ((1 << 12) - 1)) | |
| 9 | #define LOWER_22(x) ((unsigned int)(x) & ((1 << 22) - 1)) | |
| 10 | ||
| 11 | #define PDE_4M_ADDR_SHIFT 22 | |
| 12 | #define PTE_4K_ADDR_SHIFT 12 | |
| 13 | #define PDE_TABLE_ADDR_SHIFT 12 | |
| 14 | #define PTE_FRAME_ADDR_MASK (~((1 << PTE_4K_ADDR_SHIFT) - 1)) | |
| 15 | #define PDE_TABLE_ADDR_MASK (~((1 << PDE_TABLE_ADDR_SHIFT) - 1)) | |
| 16 | ||
| 17 | #define PDE_ATTRIB_INDEX (1 << 12) | |
| 18 | #define PDE_GLOBAL (1 << 8) | |
| 19 | #define PDE_PAGE_SIZE (1 << 7) | |
| 20 | #define PDE_DIRTY (1 << 6) | |
| 21 | #define PDE_ACCESSED (1 << 5) | |
| 22 | #define PDE_NO_CACHE (1 << 4) | |
| 23 | #define PDE_WRITE_THROUGH (1 << 3) | |
| 24 | #define PDE_USER (1 << 2) | |
| 25 | #define PDE_READ_WRITE (1 << 1) | |
| 26 | #define PDE_PRESENT (1 << 0) | |
| 27 | ||
| 28 | #define PTE_GLOBAL (1 << 8) | |
| 29 | #define PTE_ATTRIB_INDEX (1 << 7) | |
| 30 | #define PTE_DIRTY (1 << 6) | |
| 31 | #define PTE_ACCESSED (1 << 5) | |
| 32 | #define PTE_NO_CACHE (1 << 4) | |
| 33 | #define PTE_WRITE_THROUGH (1 << 3) | |
| 34 | #define PTE_USER (1 << 2) | |
| 35 | #define PTE_READ_WRITE (1 << 1) | |
| 36 | #define PTE_PRESENT (1 << 0) | |
| 37 | ||
| 38 | asm ("get_cr3: mov %cr3, %eax \n ret"); | |
| 39 | uint32_t * get_cr3(); | |
| 40 | ||
| 41 | void * l2p (void * addr) { | |
| 42 | ||
| 43 | uint32_t * cr3 = get_cr3(); | |
| 44 | uint32_t * page_table; | |
| 45 | uint32_t pde, pte; | |
| 46 | ||
| 47 | pde = cr3[PDE_FOR(addr)]; | |
| 48 | ||
| 49 | if (!(pde & PDE_PRESENT)) { | |
| 50 | return 0; | |
| 51 | } | |
| 52 | ||
| 53 | if (pde & PDE_PAGE_SIZE) { | |
| 54 | return (void *)(ADDR_22_MASK(pde) | LOWER_22(addr)); | |
| 55 | } else { | |
| 56 | page_table = (uint32_t *)(ADDR_12_MASK(pde)); | |
| 57 | pte = page_table[PTE_FOR(addr)]; | |
| 58 | ||
| 59 | if (!pte & PTE_PRESENT) { | |
| 60 | return 0; | |
| 61 | } | |
| 62 | return (void *)(ADDR_12_MASK(pte) | LOWER_12(addr)); | |
| 63 | } | |
| 64 | } | |
| 65 | ||
| 66 | packet_t * check_packet (uint32_t logical_base) { | |
| 67 | packet_t * physical_base; | |
| 68 | ||
| 69 | if (LOWER_12(logical_base) != 0) return 0; | |
| 70 | ||
| 71 | physical_base = l2p((void *)logical_base); | |
| 72 | ||
| 73 | if (!physical_base) return 0; | |
| 74 | if (physical_base->signature != 0x1BADD00D) return 0; | |
| 75 | ||
| 76 | return physical_base; | |
| 77 | } |