Blame tests/elfputzdata.c

Packit Service 97d2fb
/* Copyright (C) 2015 Red Hat, Inc.
Packit Service 97d2fb
   This file is part of elfutils.
Packit Service 97d2fb
Packit Service 97d2fb
   This file is free software; you can redistribute it and/or modify
Packit Service 97d2fb
   it under the terms of the GNU General Public License as published by
Packit Service 97d2fb
   the Free Software Foundation; either version 3 of the License, or
Packit Service 97d2fb
   (at your option) any later version.
Packit Service 97d2fb
Packit Service 97d2fb
   elfutils is distributed in the hope that it will be useful, but
Packit Service 97d2fb
   WITHOUT ANY WARRANTY; without even the implied warranty of
Packit Service 97d2fb
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit Service 97d2fb
   GNU General Public License for more details.
Packit Service 97d2fb
Packit Service 97d2fb
   You should have received a copy of the GNU General Public License
Packit Service 97d2fb
   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
Packit Service 97d2fb
Packit Service 97d2fb
#ifdef HAVE_CONFIG_H
Packit Service 97d2fb
# include <config.h>
Packit Service 97d2fb
#endif
Packit Service 97d2fb
Packit Service 97d2fb
#include <sys/types.h>
Packit Service 97d2fb
#include <sys/stat.h>
Packit Service 97d2fb
#include <fcntl.h>
Packit Service 97d2fb
#include <inttypes.h>
Packit Service 97d2fb
#include <libelf.h>
Packit Service 97d2fb
#include <gelf.h>
Packit Service 97d2fb
#include <stdbool.h>
Packit Service 97d2fb
#include <stdio.h>
Packit Service 97d2fb
#include <stdlib.h>
Packit Service 97d2fb
#include <string.h>
Packit Service 97d2fb
#include <unistd.h>
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
int
Packit Service 97d2fb
main (int argc, char *argv[])
Packit Service 97d2fb
{
Packit Service 97d2fb
  int result = 0;
Packit Service 97d2fb
  int cnt;
Packit Service 97d2fb
Packit Service 97d2fb
  if (argc < 3
Packit Service 97d2fb
      || (strcmp (argv[1], "elf") != 0
Packit Service 97d2fb
	  && strcmp (argv[1], "gnu") != 0))
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("Usage: (elf|gnu) files...\n");
Packit Service 97d2fb
      return -1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  int gnu;
Packit Service 97d2fb
  if (strcmp (argv[1], "gnu") == 0)
Packit Service 97d2fb
    gnu = 1;
Packit Service 97d2fb
  else
Packit Service 97d2fb
    gnu = 0;
Packit Service 97d2fb
Packit Service 97d2fb
  elf_version (EV_CURRENT);
Packit Service 97d2fb
Packit Service 97d2fb
  for (cnt = 2; cnt < argc; ++cnt)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      int fd = open (argv[cnt], O_RDONLY);
Packit Service 97d2fb
Packit Service 97d2fb
      Elf *elf = elf_begin (fd, ELF_C_READ, NULL);
Packit Service 97d2fb
      if (elf == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("%s not usable %s\n", argv[cnt], elf_errmsg (-1));
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	  close (fd);
Packit Service 97d2fb
	  continue;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      /* To get the section names.  */
Packit Service 97d2fb
      size_t strndx;
Packit Service 97d2fb
      elf_getshdrstrndx (elf, &strndx);
Packit Service 97d2fb
Packit Service 97d2fb
      Elf_Scn *scn = NULL;
Packit Service 97d2fb
      while ((scn = elf_nextscn (elf, scn)) != NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  size_t idx = elf_ndxscn (scn);
Packit Service 97d2fb
	  GElf_Shdr mem;
Packit Service 97d2fb
	  GElf_Shdr *shdr = gelf_getshdr (scn, &mem;;
Packit Service 97d2fb
	  const char *name = elf_strptr (elf, strndx, shdr->sh_name);
Packit Service 97d2fb
	  if (shdr->sh_type == SHT_NOBITS
Packit Service 97d2fb
	      || (shdr->sh_flags & SHF_ALLOC) != 0)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      printf ("Cannot compress %zd %s\n", idx, name);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
	  else if ((shdr->sh_flags & SHF_COMPRESSED) != 0
Packit Service 97d2fb
		   || strncmp (name, ".zdebug", strlen (".zdebug")) == 0)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      printf ("Already compressed %zd %s\n", idx, name);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
	  else
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      size_t orig_size = shdr->sh_size;
Packit Service 97d2fb
	      printf ("Lets compress %zd %s, size: %" PRId64 "\n",
Packit Service 97d2fb
		      idx, name, shdr->sh_size);
Packit Service 97d2fb
	      Elf_Data *d = elf_getdata (scn, NULL);
Packit Service 97d2fb
	      if (d == NULL)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("Couldn't get orig data for section %zd\n", idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      /* Make a copy so we can compare after
Packit Service 97d2fb
		 compression/decompression.  */
Packit Service 97d2fb
	      if (d->d_size != orig_size)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("Unexpected data size for orig section %zd\n", idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      char *orig_buf = NULL;
Packit Service 97d2fb
	      if (orig_size > 0)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  orig_buf = malloc (d->d_size);
Packit Service 97d2fb
		  if (orig_buf == NULL)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      printf ("No memory to copy section %zd data\n", idx);
Packit Service 97d2fb
		      return -1;
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		  memcpy (orig_buf, d->d_buf, orig_size);
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      bool forced = false;
Packit Service 97d2fb
	      if (gnu)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  int res = elf_compress_gnu (scn, 1, 0);
Packit Service 97d2fb
		  if (res == 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      forced = true;
Packit Service 97d2fb
		      res = elf_compress_gnu (scn, 1, ELF_CHF_FORCE);
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		  if (res < 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      printf ("elf_compress_gnu%sfailed for section %zd: %s\n",
Packit Service 97d2fb
			      forced ? " (forced) " : " ",
Packit Service 97d2fb
			      idx, elf_errmsg (-1));
Packit Service 97d2fb
		      return -1;
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      else
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  int res = elf_compress (scn, ELFCOMPRESS_ZLIB, 0);
Packit Service 97d2fb
		  if (res == 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      forced = true;
Packit Service 97d2fb
		      res = elf_compress (scn, ELFCOMPRESS_ZLIB, ELF_CHF_FORCE);
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		  if (res < 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      printf ("elf_compress%sfailed for section %zd: %s\n",
Packit Service 97d2fb
			      forced ? " (forced) " : " ",
Packit Service 97d2fb
			      idx, elf_errmsg (-1));
Packit Service 97d2fb
		      return -1;
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      GElf_Shdr newmem;
Packit Service 97d2fb
	      GElf_Shdr *newshdr = gelf_getshdr (scn, &newmem);
Packit Service 97d2fb
	      size_t new_size = newshdr->sh_size;
Packit Service 97d2fb
	      d = elf_getdata (scn, NULL);
Packit Service 97d2fb
	      // Don't check this, might depend on zlib implementation.
Packit Service 97d2fb
	      // fprintf (stderr, "  new_size: %zd\n", new_size);
Packit Service 97d2fb
	      if (d->d_size != new_size)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("Unexpected data size for compressed section %zd\n",
Packit Service 97d2fb
			  idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      if (forced && new_size < orig_size)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("section %zd forced to compress, but size smaller\n",
Packit Service 97d2fb
			  idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      if (! forced && new_size >= orig_size)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("section %zd compressed to bigger size\n",
Packit Service 97d2fb
			  idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      if (new_size == orig_size
Packit Service 97d2fb
		  && (orig_buf == NULL
Packit Service 97d2fb
		      || memcmp (orig_buf, d->d_buf, orig_size) == 0))
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("section %zd didn't compress\n", idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      if (gnu)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  if (elf_compress_gnu (scn, 0, 0) < 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      printf ("elf_[un]compress_gnu failed for section %zd: %s\n",
Packit Service 97d2fb
			      idx, elf_errmsg (-1));
Packit Service 97d2fb
		      return -1;
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      else
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  if (elf_compress (scn, 0, 0) < 0)
Packit Service 97d2fb
		    {
Packit Service 97d2fb
		      printf ("elf_[un]compress failed for section %zd: %s\n",
Packit Service 97d2fb
			      idx, elf_errmsg (-1));
Packit Service 97d2fb
		      return -1;
Packit Service 97d2fb
		    }
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      GElf_Shdr newermem;
Packit Service 97d2fb
	      GElf_Shdr *newershdr = gelf_getshdr (scn, &newermem);
Packit Service 97d2fb
	      size_t newer_size = newershdr->sh_size;
Packit Service 97d2fb
	      d = elf_getdata (scn, NULL);
Packit Service 97d2fb
	      // fprintf (stderr, "  newer_size: %zd\n", newer_size);
Packit Service 97d2fb
	      if (d->d_size != newer_size)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("Unexpected data size for compressed section %zd\n",
Packit Service 97d2fb
			  idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      if (newer_size != orig_size
Packit Service 97d2fb
		  && (orig_buf == NULL
Packit Service 97d2fb
		      || memcmp (orig_buf, d->d_buf, orig_size) != 0))
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("section %zd didn't correctly uncompress\n", idx);
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
	      free (orig_buf);
Packit Service 97d2fb
	      // Recompress the string table, just to make sure
Packit Service 97d2fb
	      // everything keeps working. See elf_strptr above.
Packit Service 97d2fb
	      if (! gnu && idx == strndx
Packit Service 97d2fb
		  && elf_compress (scn, ELFCOMPRESS_ZLIB, 0) < 0)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  printf ("couldn't recompress section header strings: %s\n",
Packit Service 97d2fb
			  elf_errmsg (-1));
Packit Service 97d2fb
		  return -1;
Packit Service 97d2fb
		}
Packit Service 97d2fb
	    }
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      elf_end (elf);
Packit Service 97d2fb
      close (fd);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  return result;
Packit Service 97d2fb
}