1 /* 2 * ELF payload decoding support. 3 * 4 * Copyright (C) 2022, 2023 Paul Boddie <paul@boddie.org.uk> 5 * 6 * This program is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU General Public License as 8 * published by the Free Software Foundation; either version 2 of 9 * the License, or (at your option) any later version. 10 * 11 * This program is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 * GNU General Public License for more details. 15 * 16 * You should have received a copy of the GNU General Public License 17 * along with this program; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, 19 * Boston, MA 02110-1301, USA 20 */ 21 22 #include <l4/util/elf.h> 23 24 #include "elf.h" 25 26 27 28 /* Program segment construction for 32- or 64-bit variants. */ 29 30 template <typename PROGRAM_HEADER> 31 ProgramSegmentVariant<PROGRAM_HEADER>::ProgramSegmentVariant(PROGRAM_HEADER *header) 32 : _header(header) 33 { 34 init(); 35 } 36 37 template <typename PROGRAM_HEADER> 38 bool ProgramSegmentVariant<PROGRAM_HEADER>::loadable() 39 { 40 return _header->p_type == PT_LOAD; 41 } 42 43 template <typename PROGRAM_HEADER> 44 offset_t ProgramSegmentVariant<PROGRAM_HEADER>::file_contents() 45 { 46 return _header->p_filesz; 47 } 48 49 template <typename PROGRAM_HEADER> 50 offset_t ProgramSegmentVariant<PROGRAM_HEADER>::file_offset() 51 { 52 return _header->p_offset; 53 } 54 55 template <typename PROGRAM_HEADER> 56 l4_addr_t ProgramSegmentVariant<PROGRAM_HEADER>::region_address() 57 { 58 return _header->p_vaddr; 59 } 60 61 template <typename PROGRAM_HEADER> 62 offset_t ProgramSegmentVariant<PROGRAM_HEADER>::region_size() 63 { 64 return _header->p_memsz; 65 } 66 67 template <typename PROGRAM_HEADER> 68 l4re_rm_flags_t ProgramSegmentVariant<PROGRAM_HEADER>::region_flags() 69 { 70 l4re_rm_flags_t flags = 0; 71 72 if (_header->p_flags & PF_R) 73 flags |= L4RE_RM_F_R; 74 if (_header->p_flags & PF_W) 75 flags |= L4RE_RM_F_W; 76 if (_header->p_flags & PF_X) 77 flags |= L4RE_RM_F_X; 78 79 return flags; 80 } 81 82 83 84 /* Generic payload destruction. */ 85 86 Payload::~Payload() 87 { 88 } 89 90 unsigned int Payload::segments() 91 { 92 return 0; 93 } 94 95 96 97 /* ELF payload construction for 32- or 64-bit variants. */ 98 99 template <typename HEADER, typename PROGRAM_HEADER> 100 PayloadVariant<HEADER, PROGRAM_HEADER>::PayloadVariant(HEADER *header) 101 : _header(header) 102 { 103 } 104 105 /* Specific payload destruction. */ 106 107 template <typename HEADER, typename PROGRAM_HEADER> 108 PayloadVariant<HEADER, PROGRAM_HEADER>::~PayloadVariant() 109 { 110 if (_segments != NULL) 111 { 112 for (unsigned int i = 0; i < segments(); i++) 113 delete _segments[i]; 114 } 115 } 116 117 template <typename HEADER, typename PROGRAM_HEADER> 118 l4_addr_t PayloadVariant<HEADER, PROGRAM_HEADER>::entry_point() 119 { 120 return _header->e_entry; 121 } 122 123 template <typename HEADER, typename PROGRAM_HEADER> 124 offset_t PayloadVariant<HEADER, PROGRAM_HEADER>::header_extent() 125 { 126 return sizeof(HEADER); 127 } 128 129 template <typename HEADER, typename PROGRAM_HEADER> 130 offset_t PayloadVariant<HEADER, PROGRAM_HEADER>::program_header_extent() 131 { 132 return _header->e_phoff + _header->e_phentsize * _header->e_phnum; 133 } 134 135 template <typename HEADER, typename PROGRAM_HEADER> 136 unsigned int PayloadVariant<HEADER, PROGRAM_HEADER>::segments() 137 { 138 return _header->e_phnum; 139 } 140 141 template <typename HEADER, typename PROGRAM_HEADER> 142 Segment *PayloadVariant<HEADER, PROGRAM_HEADER>::segment(unsigned int i) 143 { 144 if (i >= segments()) 145 return NULL; 146 147 if (_segments == NULL) 148 { 149 _segments = new Segment *[segments()]; 150 151 for (unsigned int j = 0; j < segments(); j++) 152 _segments[j] = NULL; 153 } 154 155 Segment *segment = _segments[i]; 156 157 if (segment == NULL) 158 { 159 segment = new ProgramSegmentVariant<PROGRAM_HEADER>( 160 (PROGRAM_HEADER *) ((char *) _header + _header->e_phoff + 161 _header->e_phentsize * i)); 162 _segments[i] = segment; 163 } 164 165 return segment; 166 } 167 168 169 170 /* Return the appropriate payload object. */ 171 172 Payload *exec_open_payload(char *buf) 173 { 174 switch (buf[EI_CLASS]) 175 { 176 case (char) 1: 177 return new PayloadVariant<Elf32_Ehdr, Elf32_Phdr>((Elf32_Ehdr *) buf); 178 case (char) 2: 179 return new PayloadVariant<Elf64_Ehdr, Elf64_Phdr>((Elf64_Ehdr *) buf); 180 default: 181 return NULL; 182 } 183 } 184 185 /* vim: tabstop=2 expandtab shiftwidth=2 186 */