contrib/elftoolchain/libpe/pe_coff.c

   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
  27 #include <errno.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30
  31 #include "_libpe.h"
  32
  33 ELFTC_VCSID("$Id: pe_coff.c 3312 2016-01-10 09:23:51Z kaiwang27 $");
  34
  35 PE_CoffHdr *
  36 pe_coff_header(PE *pe)
  37 {
  38
  39         if (pe->pe_ch == NULL) {
  40                 errno = ENOENT;
  41                 return (NULL);
  42         }
  43
  44         return (pe->pe_ch);
  45 }
  46
  47 PE_OptHdr *
  48 pe_opt_header(PE *pe)
  49 {
  50
  51         if (pe->pe_oh == NULL) {
  52                 errno = ENOENT;
  53                 return (NULL);
  54         }
  55
  56         return (pe->pe_oh);
  57 }
  58
  59 PE_DataDir *
  60 pe_data_dir(PE *pe)
  61 {
  62
  63         if (pe->pe_dd == NULL) {
  64                 errno = ENOENT;
  65                 return (NULL);
  66         }
  67
  68         return (pe->pe_dd);
  69 }
  70
  71 int
  72 pe_update_coff_header(PE *pe, PE_CoffHdr *ch)
  73 {
  74
  75         if (pe == NULL || ch == NULL) {
  76                 errno = EINVAL;
  77                 return (-1);
  78         }
  79
  80         if (pe->pe_cmd == PE_C_READ || pe->pe_flags & LIBPE_F_FD_DONE) {
  81                 errno = EACCES;
  82                 return (-1);
  83         }
  84
  85         if (pe->pe_ch == NULL) {
  86                 if ((pe->pe_ch = malloc(sizeof(PE_CoffHdr))) == NULL) {
  87                         errno = ENOMEM;
  88                         return (-1);
  89                 }
  90         } else {
  91                 /* Rewrite optional header if `optsize' field changed. */
  92                 if (pe->pe_ch->ch_optsize != ch->ch_optsize)
  93                         pe->pe_flags |= LIBPE_F_DIRTY_OPT_HEADER;
  94         }
  95
  96         *pe->pe_ch = *ch;
  97
  98         pe->pe_flags |= LIBPE_F_DIRTY_COFF_HEADER;
  99
 100         return (0);
 101 }
 102
 103 int
 104 pe_update_opt_header(PE *pe, PE_OptHdr *oh)
 105 {
 106
 107         if (pe == NULL || oh == NULL) {
 108                 errno = EINVAL;
 109                 return (-1);
 110         }
 111
 112         if (pe->pe_cmd == PE_C_READ || pe->pe_flags & LIBPE_F_FD_DONE) {
 113                 errno = EACCES;
 114                 return (-1);
 115         }
 116
 117         if (pe->pe_oh == NULL) {
 118                 if ((pe->pe_oh = malloc(sizeof(PE_OptHdr))) == NULL) {
 119                         errno = ENOMEM;
 120                         return (-1);
 121                 }
 122         }
 123
 124         *pe->pe_oh = *oh;
 125
 126         pe->pe_flags |= LIBPE_F_DIRTY_OPT_HEADER;
 127
 128         return (0);
 129 }
 130
 131 int
 132 pe_update_data_dir(PE *pe, PE_DataDir *dd)
 133 {
 134
 135         if (pe == NULL || dd == NULL) {
 136                 errno = EINVAL;
 137                 return (-1);
 138         }
 139
 140         if (pe->pe_cmd == PE_C_READ || pe->pe_flags & LIBPE_F_FD_DONE) {
 141                 errno = EACCES;
 142                 return (-1);
 143         }
 144
 145         if (pe->pe_dd == NULL) {
 146                 if ((pe->pe_dd = malloc(sizeof(PE_DataDir))) == NULL) {
 147                         errno = ENOMEM;
 148                         return (-1);
 149                 }
 150         }
 151
 152         *pe->pe_dd = *dd;
 153
 154         pe->pe_flags |= LIBPE_F_DIRTY_OPT_HEADER;
 155
 156         return (0);
 157 }