| 1 | //----------------------------------------------------------------------------- |
| 2 | // Jonathan Westhues, Mar 2006 |
| 3 | // |
| 4 | // This code is licensed to you under the terms of the GNU GPL, version 2 or, |
| 5 | // at your option, any later version. See the LICENSE.txt file for the text of |
| 6 | // the license. |
| 7 | //----------------------------------------------------------------------------- |
| 8 | // Just vector to AppMain(). This is in its own file so that I can place it |
| 9 | // with the linker script. |
| 10 | //----------------------------------------------------------------------------- |
| 11 | |
| 12 | #include "proxmark3.h" |
| 13 | #include "apps.h" |
| 14 | #include "zlib.h" |
| 15 | #include "BigBuf.h" |
| 16 | |
| 17 | static uint8_t *next_free_memory; |
| 18 | extern struct common_area common_area; |
| 19 | extern char __data_src_start__, __data_start__, __data_end__, __bss_start__, __bss_end__; |
| 20 | |
| 21 | |
| 22 | static voidpf inflate_malloc(voidpf opaque, uInt items, uInt size) |
| 23 | { |
| 24 | uint8_t *allocated_memory; |
| 25 | |
| 26 | allocated_memory = next_free_memory; |
| 27 | next_free_memory += items*size; |
| 28 | return allocated_memory; |
| 29 | } |
| 30 | |
| 31 | |
| 32 | static void inflate_free(voidpf opaque, voidpf address) |
| 33 | { |
| 34 | // nothing to do |
| 35 | |
| 36 | } |
| 37 | |
| 38 | static void uncompress_data_section(void) |
| 39 | { |
| 40 | z_stream data_section; |
| 41 | |
| 42 | next_free_memory = BigBuf_get_addr(); |
| 43 | |
| 44 | // initialize zstream structure |
| 45 | data_section.next_in = (uint8_t *) &__data_src_start__; |
| 46 | data_section.avail_in = &__data_end__ - &__data_start__; // uncompressed size. Wrong but doesn't matter. |
| 47 | data_section.next_out = (uint8_t *) &__data_start__; |
| 48 | data_section.avail_out = &__data_end__ - &__data_start__; // uncompressed size. Correct. |
| 49 | data_section.zalloc = &inflate_malloc; |
| 50 | data_section.zfree = &inflate_free; |
| 51 | data_section.opaque = NULL; |
| 52 | |
| 53 | // initialize zlib for inflate |
| 54 | inflateInit2(&data_section, 15); |
| 55 | |
| 56 | // uncompress data segment to RAM |
| 57 | inflate(&data_section, Z_FINISH); |
| 58 | |
| 59 | // save the size of the compressed data section |
| 60 | common_area.arg1 = data_section.total_in; |
| 61 | } |
| 62 | |
| 63 | |
| 64 | void __attribute__((section(".startos"))) Vector(void) |
| 65 | { |
| 66 | /* Stack should have been set up by the bootloader */ |
| 67 | // char *src; |
| 68 | char *dst, *end; |
| 69 | |
| 70 | uncompress_data_section(); |
| 71 | |
| 72 | /* Set up (that is: clear) BSS. */ |
| 73 | dst = &__bss_start__; |
| 74 | end = &__bss_end__; |
| 75 | while(dst < end) *dst++ = 0; |
| 76 | |
| 77 | // Set up data segment: Copy from flash to ram |
| 78 | // src = &__data_src_start__; |
| 79 | // dst = &__data_start__; |
| 80 | // end = &__data_end__; |
| 81 | // while(dst < end) *dst++ = *src++; |
| 82 | |
| 83 | |
| 84 | AppMain(); |
| 85 | } |