Blame tests/arextract.c

Packit Service 97d2fb
/* Copyright (C) 1999, 2000, 2002 Red Hat, Inc.
Packit Service 97d2fb
   This file is part of elfutils.
Packit Service 97d2fb
   Written by Ulrich Drepper <drepper@redhat.com>, 1999.
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 <fcntl.h>
Packit Service 97d2fb
#include <gelf.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
#include <system.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 fd;
Packit Service 97d2fb
  Elf *elf;
Packit Service 97d2fb
  Elf *subelf;
Packit Service 97d2fb
  Elf_Cmd cmd;
Packit Service 97d2fb
  off_t offset;
Packit Service 97d2fb
  size_t todo;
Packit Service 97d2fb
Packit Service 97d2fb
  if (argc < 4)
Packit Service 97d2fb
    exit (1);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Open the archive.  */
Packit Service 97d2fb
  fd = open (argv[1], O_RDONLY);
Packit Service 97d2fb
  if (fd == -1)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("Cannot open input file: %m");
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Set the ELF version.  */
Packit Service 97d2fb
  elf_version (EV_CURRENT);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Create an ELF descriptor.  */
Packit Service 97d2fb
  cmd = ELF_C_READ;
Packit Service 97d2fb
  elf = elf_begin (fd, cmd, NULL);
Packit Service 97d2fb
  if (elf == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("Cannot create ELF descriptor: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* If it is no archive punt.  */
Packit Service 97d2fb
  if (elf_kind (elf) != ELF_K_AR)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("`%s' is no archive\n", argv[1]);
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Get the elements of the archive one after the other.  */
Packit Service 97d2fb
  while ((subelf = elf_begin (fd, cmd, elf)) != NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      /* The the header for this element.  */
Packit Service 97d2fb
      Elf_Arhdr *arhdr = elf_getarhdr (subelf);
Packit Service 97d2fb
Packit Service 97d2fb
      if (arhdr == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot get arhdr: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (strcmp (arhdr->ar_name, argv[2]) == 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  int outfd;
Packit Service 97d2fb
Packit Service 97d2fb
	  /* Get the offset of the file in the archive.  */
Packit Service 97d2fb
	  offset = elf_getbase (subelf);
Packit Service 97d2fb
	  if (offset == -1)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      printf ("\
Packit Service 97d2fb
Failed to get base address for the archive element: %s\n",
Packit Service 97d2fb
		      elf_errmsg (-1));
Packit Service 97d2fb
	      exit (1);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
Packit Service 97d2fb
	  /* Open the output file.  */
Packit Service 97d2fb
	  outfd = open (argv[3], O_CREAT | O_TRUNC | O_RDWR, 0666);
Packit Service 97d2fb
	  if (outfd == -1)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      printf ("cannot open output file: %m");
Packit Service 97d2fb
	      exit (1);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
Packit Service 97d2fb
	  /* Now write out the data.  */
Packit Service 97d2fb
	  todo = arhdr->ar_size;
Packit Service 97d2fb
	  while (todo > 0)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      char buf[1024];
Packit Service 97d2fb
	      ssize_t n = pread (fd, buf, MIN (sizeof buf, todo), offset);
Packit Service 97d2fb
	      if (n == 0)
Packit Service 97d2fb
		break;
Packit Service 97d2fb
Packit Service 97d2fb
	      if (write (outfd, buf, n) != n)
Packit Service 97d2fb
		{
Packit Service 97d2fb
		  puts ("Writing output failed");
Packit Service 97d2fb
		  exit (1);
Packit Service 97d2fb
		}
Packit Service 97d2fb
Packit Service 97d2fb
	      offset += n;
Packit Service 97d2fb
	      todo -= n;
Packit Service 97d2fb
	    }
Packit Service 97d2fb
Packit Service 97d2fb
	  /* Check whether all the date was read and written out.  */
Packit Service 97d2fb
	  if (todo != 0)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      puts ("Reading archive member failed.");
Packit Service 97d2fb
	      exit (1);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
Packit Service 97d2fb
	  /* Close the descriptors.  */
Packit Service 97d2fb
	  if (elf_end (subelf) != 0 || elf_end (elf) != 0)
Packit Service 97d2fb
	    {
Packit Service 97d2fb
	      printf ("Freeing ELF descriptors failed: %s", elf_errmsg (-1));
Packit Service 97d2fb
	      exit (1);
Packit Service 97d2fb
	    }
Packit Service 97d2fb
Packit Service 97d2fb
	  close (outfd);
Packit Service 97d2fb
	  close (fd);
Packit Service 97d2fb
Packit Service 97d2fb
	  /* All went well.  */
Packit Service 97d2fb
	  exit (0);
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      /* Get next archive element.  */
Packit Service 97d2fb
      cmd = elf_next (subelf);
Packit Service 97d2fb
      if (elf_end (subelf) != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("error while freeing sub-ELF descriptor: %s\n",
Packit Service 97d2fb
		  elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* When we reach this point we haven't found the given file in the
Packit Service 97d2fb
     archive.  */
Packit Service 97d2fb
  printf ("File `%s' not found in archive\n", argv[2]);
Packit Service 97d2fb
  exit (1);
Packit Service 97d2fb
}