arch/x86/boot/compressed/mkpiggy.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *  Copyright (C) 2009 Intel Corporation. All rights reserved.
   4  *
   5  *  This program is free software; you can redistribute it and/or
   6  *  modify it under the terms of the GNU General Public License version
   7  *  2 as published by the Free Software Foundation.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License
  15  *  along with this program; if not, write to the Free Software
  16  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  17  *  02110-1301, USA.
  18  *
  19  *  H. Peter Anvin <hpa@linux.intel.com>
  20  *
  21  * ----------------------------------------------------------------------- */
  22
  23 /*
  24  * Compute the desired load offset from a compressed program; outputs
  25  * a small assembly wrapper with the appropriate symbols defined.
  26  */
  27
  28 #include <stdlib.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <inttypes.h>
  32 #include <tools/le_byteshift.h>
  33
  34 int main(int argc, char *argv[])
  35 {
  36         uint32_t olen;
  37         long ilen;
  38         unsigned long offs;
  39         unsigned long run_size;
  40         FILE *f = NULL;
  41         int retval = 1;
  42
  43         if (argc < 3) {
  44                 fprintf(stderr, "Usage: %s compressed_file run_size\n",
  45                                 argv[0]);
  46                 goto bail;
  47         }
  48
  49         /* Get the information for the compressed kernel image first */
  50
  51         f = fopen(argv[1], "r");
  52         if (!f) {
  53                 perror(argv[1]);
  54                 goto bail;
  55         }
  56
  57
  58         if (fseek(f, -4L, SEEK_END)) {
  59                 perror(argv[1]);
  60         }
  61
  62         if (fread(&olen, sizeof(olen), 1, f) != 1) {
  63                 perror(argv[1]);
  64                 goto bail;
  65         }
  66
  67         ilen = ftell(f);
  68         olen = get_unaligned_le32(&olen);
  69
  70         /*
  71          * Now we have the input (compressed) and output (uncompressed)
  72          * sizes, compute the necessary decompression offset...
  73          */
  74
  75         offs = (olen > ilen) ? olen - ilen : 0;
  76         offs += olen >> 12;     /* Add 8 bytes for each 32K block */
  77         offs += 64*1024 + 128;  /* Add 64K + 128 bytes slack */
  78         offs = (offs+4095) & ~4095; /* Round to a 4K boundary */
  79         run_size = atoi(argv[2]);
  80
  81         printf(".section \".rodata..compressed\",\"a\",@progbits\n");
  82         printf(".globl z_input_len\n");
  83         printf("z_input_len = %lu\n", ilen);
  84         printf(".globl z_output_len\n");
  85         printf("z_output_len = %lu\n", (unsigned long)olen);
  86         printf(".globl z_extract_offset\n");
  87         printf("z_extract_offset = 0x%lx\n", offs);
  88         /* z_extract_offset_negative allows simplification of head_32.S */
  89         printf(".globl z_extract_offset_negative\n");
  90         printf("z_extract_offset_negative = -0x%lx\n", offs);
  91         printf(".globl z_run_size\n");
  92         printf("z_run_size = %lu\n", run_size);
  93
  94         printf(".globl input_data, input_data_end\n");
  95         printf("input_data:\n");
  96         printf(".incbin \"%s\"\n", argv[1]);
  97         printf("input_data_end:\n");
  98
  99         retval = 0;
 100 bail:
 101         if (f)
 102                 fclose(f);
 103         return retval;
 104 }