contrib/elftoolchain/libpe/pe_buffer.c

   1 /*-
   2  * Copyright (c) 2016 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
  27 #include <errno.h>
  28
  29 #include "_libpe.h"
  30
  31 ELFTC_VCSID("$Id: pe_buffer.c 3312 2016-01-10 09:23:51Z kaiwang27 $");
  32
  33 PE_Buffer *
  34 pe_getbuffer(PE_Scn *ps, PE_Buffer *pb)
  35 {
  36         PE *pe;
  37         PE_SecBuf *sb;
  38
  39         if (ps == NULL) {
  40                 errno = EINVAL;
  41                 return (NULL);
  42         }
  43
  44         pe = ps->ps_pe;
  45
  46         if ((ps->ps_flags & LIBPE_F_LOAD_SECTION) == 0) {
  47                 if (pe->pe_flags & LIBPE_F_FD_DONE) {
  48                         errno = EACCES;
  49                         return (NULL);
  50                 }
  51                 if (pe->pe_flags & LIBPE_F_SPECIAL_FILE) {
  52                         if (libpe_load_all_sections(pe) < 0)
  53                                 return (NULL);
  54                 } else {
  55                         if (libpe_load_section(pe, ps) < 0)
  56                                 return (NULL);
  57                 }
  58         }
  59
  60         sb = (PE_SecBuf *) pb;
  61
  62         if (sb == NULL)
  63                 sb = STAILQ_FIRST(&ps->ps_b);
  64         else
  65                 sb = STAILQ_NEXT(sb, sb_next);
  66
  67         return ((PE_Buffer *) sb);
  68 }
  69
  70 PE_Buffer *
  71 pe_newbuffer(PE_Scn *ps)
  72 {
  73         PE *pe;
  74         PE_SecBuf *sb;
  75
  76         if (ps == NULL) {
  77                 errno = EINVAL;
  78                 return (NULL);
  79         }
  80
  81         pe = ps->ps_pe;
  82
  83         if (pe->pe_flags & LIBPE_F_FD_DONE) {
  84                 errno = EACCES;
  85                 return (NULL);
  86         }
  87
  88         if ((ps->ps_flags & LIBPE_F_LOAD_SECTION) == 0) {
  89                 if (libpe_load_section(pe, ps) < 0)
  90                         return (NULL);
  91         }
  92
  93         if ((sb = libpe_alloc_buffer(ps, 0)) == NULL)
  94                 return (NULL);
  95
  96         sb->sb_flags |= PE_F_DIRTY;
  97         ps->ps_flags |= PE_F_DIRTY;
  98
  99         return ((PE_Buffer *) sb);
 100 }