1 /*-
2 * Copyright (c) 2015 Kai Wang
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 * $Id: _libpe.h 3312 2016-01-10 09:23:51Z kaiwang27 $
27 */
28
29 #ifndef __LIBPE_H_
30 #define __LIBPE_H_
31
32 #include <sys/types.h>
33 #include <sys/queue.h>
34
35 #include "libpe.h"
36
37 #include "_elftc.h"
38
39 typedef struct _PE_SecBuf {
40 PE_Buffer sb_pb; /* application buffer */
41 PE_Scn *sb_ps; /* PE_Scn pointer */
42 unsigned int sb_flags; /* buffer flags */
43 STAILQ_ENTRY(_PE_SecBuf) sb_next;
44 } PE_SecBuf;
45
46 struct _PE_Scn {
47 PE *ps_pe; /* PE descriptor */
48 PE_SecHdr ps_sh; /* section header */
49 unsigned int ps_ndx; /* 1-based section index */
50 unsigned int ps_flags; /* section flags */
51 unsigned int ps_falign; /* section file alignment */
52 STAILQ_HEAD(, _PE_SecBuf) ps_b; /* buffer list */
53 STAILQ_ENTRY(_PE_Scn) ps_next;
54 };
55
56 struct _PE {
57 int pe_fd; /* file descriptor */
58 PE_Cmd pe_cmd; /* open mode */
59 PE_Object pe_obj; /* PE32/PE32+/COFF */
60 size_t pe_fsize; /* file size */
61 unsigned int pe_flags; /* library flags */
62 PE_DosHdr *pe_dh; /* MS-DOS header */
63 char *pe_stub; /* MS-DOS stub */
64 size_t pe_stub_ex; /* MS-DOS stub len (exclude hdr) */
65 char *pe_stub_app; /* MS-DOS stub (app supplied) */
66 size_t pe_stub_app_sz; /* MS-DOS stub len (app supplied) */
67 PE_RichHdr *pe_rh; /* rich header */
68 char *pe_rh_start; /* pointer to rich header */
69 PE_CoffHdr *pe_ch; /* COFF header */
70 PE_OptHdr *pe_oh; /* optional header */
71 PE_DataDir *pe_dd; /* data directories */
72 unsigned int pe_nscn; /* num. of sections */
73 char *pe_symtab; /* COFF symbol table */
74 size_t pe_symbtab_sz; /* size of symbol table */
75 unsigned int pe_nsym; /* num. of symbols */
76 unsigned int pe_rvamax; /* maximum RVA */
77 STAILQ_HEAD(, _PE_Scn) pe_scn; /* section list */
78 };
79
80 /* Library internal flags */
81 #define LIBPE_F_API_MASK 0x000FFFU
82 #define LIBPE_F_SPECIAL_FILE 0x001000U
83 #define LIBPE_F_BAD_DOS_HEADER 0x002000U
84 #define LIBPE_F_BAD_PE_HEADER 0x004000U
85 #define LIBPE_F_BAD_COFF_HEADER 0x008000U
86 #define LIBPE_F_BAD_OPT_HEADER 0x010000U
87 #define LIBPE_F_BAD_SEC_HEADER 0x020000U
88 #define LIBPE_F_LOAD_DOS_STUB 0x040000U
89 #define LIBPE_F_FD_DONE 0x080000U
90 #define LIBPE_F_DIRTY_DOS_HEADER 0x100000U
91 #define LIBPE_F_DIRTY_COFF_HEADER 0x200000U
92 #define LIBPE_F_DIRTY_OPT_HEADER 0x400000U
93 #define LIBPE_F_DIRTY_SEC_HEADER 0x800000U
94
95 /* Internal section flags */
96 #define LIBPE_F_LOAD_SECTION 0x1000U
97 #define LIBPE_F_STRIP_SECTION 0x2000U
98
99 /* Internal buffer flags */
100 #define LIBPE_F_BUFFER_MALLOCED 0x1000U
101
102 /* Library internal defines */
103 #define PE_DOS_MAGIC 0x5a4dU
104 #define PE_RICH_TEXT "Rich"
105 #define PE_RICH_HIDDEN 0x536e6144U /* DanS */
106 #define PE_SIGNATURE 0x4550U /* PE\0\0 */
107 #define PE_COFF_OPT_SIZE_32 224
108 #define PE_COFF_OPT_SIZE_32P 240
109 #define PE_SYM_ENTRY_SIZE 18
110
111 /* Encode/Decode macros */
112 #if defined(ELFTC_NEED_BYTEORDER_EXTENSIONS)
113 static __inline uint16_t
le16dec(const void * pp)114 le16dec(const void *pp)
115 {
116 unsigned char const *p = (unsigned char const *)pp;
117
118 return ((p[1] << 8) | p[0]);
119 }
120
121 static __inline uint32_t
le32dec(const void * pp)122 le32dec(const void *pp)
123 {
124 unsigned char const *p = (unsigned char const *)pp;
125
126 return ((p[3] << 24) | (p[2] << 16) | (p[1] << 8) | p[0]);
127 }
128
129 static __inline uint64_t
le64dec(const void * pp)130 le64dec(const void *pp)
131 {
132 unsigned char const *p = (unsigned char const *)pp;
133
134 return (((uint64_t)le32dec(p + 4) << 32) | le32dec(p));
135 }
136
137 static __inline void
le16enc(void * pp,uint16_t u)138 le16enc(void *pp, uint16_t u)
139 {
140 unsigned char *p = (unsigned char *)pp;
141
142 p[0] = u & 0xff;
143 p[1] = (u >> 8) & 0xff;
144 }
145
146 static __inline void
le32enc(void * pp,uint32_t u)147 le32enc(void *pp, uint32_t u)
148 {
149 unsigned char *p = (unsigned char *)pp;
150
151 p[0] = u & 0xff;
152 p[1] = (u >> 8) & 0xff;
153 p[2] = (u >> 16) & 0xff;
154 p[3] = (u >> 24) & 0xff;
155 }
156
157 static __inline void
le64enc(void * pp,uint64_t u)158 le64enc(void *pp, uint64_t u)
159 {
160 unsigned char *p = (unsigned char *)pp;
161
162 le32enc(p, (uint32_t)(u & 0xffffffffU));
163 le32enc(p + 4, (uint32_t)(u >> 32));
164 }
165 #endif /* ELFTC_NEED_BYTEORDER_EXTENSIONS */
166
167 #define PE_READ16(p,v) do { \
168 (v) = le16dec((p)); \
169 (p) += 2; \
170 } while(0)
171
172 #define PE_READ32(p,v) do { \
173 (v) = le32dec((p)); \
174 (p) += 4; \
175 } while(0)
176
177 #define PE_WRITE16(p,v) do { \
178 le16enc((p), (v)); \
179 (p) += 2; \
180 } while(0)
181
182 #define PE_WRITE32(p,v) do { \
183 le32enc((p), (v)); \
184 (p) += 4; \
185 } while(0)
186
187
188 /* Internal function declarations */
189 off_t libpe_align(PE *, off_t, size_t);
190 PE_SecBuf *libpe_alloc_buffer(PE_Scn *, size_t);
191 PE_Scn *libpe_alloc_scn(PE *);
192 int libpe_load_all_sections(PE *);
193 int libpe_load_section(PE *, PE_Scn *);
194 int libpe_open_object(PE *);
195 int libpe_pad(PE *, size_t);
196 int libpe_parse_msdos_header(PE *, char *);
197 int libpe_parse_coff_header(PE *, char *);
198 int libpe_parse_rich_header(PE *);
199 int libpe_parse_section_headers(PE *);
200 int libpe_read_msdos_stub(PE *);
201 void libpe_release_buffer(PE_SecBuf *);
202 void libpe_release_object(PE *);
203 void libpe_release_scn(PE_Scn *);
204 size_t libpe_resync_buffers(PE_Scn *);
205 int libpe_resync_sections(PE *, off_t);
206 int libpe_write_buffers(PE_Scn *);
207 off_t libpe_write_coff_header(PE *, off_t);
208 off_t libpe_write_msdos_stub(PE *, off_t);
209 off_t libpe_write_pe_header(PE *, off_t);
210 off_t libpe_write_sections(PE *, off_t);
211 off_t libpe_write_section_headers(PE *, off_t);
212
213 #endif /* !__LIBPE_H_ */
214