2 * Copyright (c) 2016 Kai Wang
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
27 #include <sys/param.h>
35 ELFTC_VCSID("$Id: libpe_buffer.c 3312 2016-01-10 09:23:51Z kaiwang27 $");
38 libpe_alloc_buffer(PE_Scn *ps, size_t sz)
42 if ((sb = malloc(sizeof(PE_SecBuf))) == NULL) {
49 sb->sb_pb.pb_align = 1;
51 sb->sb_pb.pb_size = sz;
53 if ((sb->sb_pb.pb_buf = malloc(sz)) == NULL) {
58 sb->sb_flags |= LIBPE_F_BUFFER_MALLOCED;
60 sb->sb_pb.pb_buf = NULL;
62 STAILQ_INSERT_TAIL(&ps->ps_b, sb, sb_next);
68 libpe_release_buffer(PE_SecBuf *sb)
76 STAILQ_REMOVE(&ps->ps_b, sb, _PE_SecBuf, sb_next);
78 if (sb->sb_flags & LIBPE_F_BUFFER_MALLOCED)
79 free(sb->sb_pb.pb_buf);
85 cmp_sb(PE_SecBuf *a, PE_SecBuf *b)
88 if (a->sb_pb.pb_off < b->sb_pb.pb_off)
90 else if (a->sb_pb.pb_off == b->sb_pb.pb_off)
97 sort_buffers(PE_Scn *ps)
100 if (STAILQ_EMPTY(&ps->ps_b))
103 STAILQ_SORT(&ps->ps_b, _PE_SecBuf, sb_next, cmp_sb);
107 libpe_resync_buffers(PE_Scn *ps)
113 assert(ps->ps_flags & LIBPE_F_LOAD_SECTION);
118 STAILQ_FOREACH(sb, &ps->ps_b, sb_next) {
119 if (ps->ps_flags & PE_F_DIRTY)
120 sb->sb_flags |= PE_F_DIRTY;
122 pb = (PE_Buffer *) sb;
123 if (pb->pb_align > ps->ps_falign)
124 pb->pb_align = ps->ps_falign;
125 if (pb->pb_buf == NULL || pb->pb_size == 0)
128 sz = roundup(sz, pb->pb_align);
130 if (pb->pb_off != (off_t) sz) {
132 sb->sb_flags |= PE_F_DIRTY;
141 libpe_write_buffers(PE_Scn *ps)
148 assert(ps->ps_flags & LIBPE_F_LOAD_SECTION);
153 STAILQ_FOREACH(sb, &ps->ps_b, sb_next) {
155 if (pb->pb_buf == NULL || pb->pb_size == 0)
158 if ((sb->sb_flags & PE_F_DIRTY) == 0) {
159 assert((pe->pe_flags & LIBPE_F_SPECIAL_FILE) == 0);
160 if (lseek(pe->pe_fd, (off_t) pb->pb_size, SEEK_CUR) <
168 if (pb->pb_off > off) {
169 if (libpe_pad(pe, pb->pb_off - off) < 0)
174 if (write(pe->pe_fd, pb->pb_buf, pb->pb_size) !=
175 (ssize_t) pb->pb_size) {