← Back to branch summary

~ubuntu-branches/debian/lenny/elfutils/lenny

« back to all changes in this revision

Viewing changes to tests/arextract.c

  • Committer: Bazaar Package Importer
  • Author(s): Kurt Roeckx
  • Date: 2006-08-27 15:48:23 UTC
  • Revision ID: james.westby@ubuntu.com-20060827154823-mjwd7ydlbxgwqn4u
Tags: upstream-0.123
Import upstream version 0.123

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/arextract.c
 
1
/* Copyright (C) 1999, 2000, 2002 Red Hat, Inc.
 
2
   This file is part of Red Hat elfutils.
 
3
   Written by Ulrich Drepper <drepper@redhat.com>, 1999.
 
4
 
 
5
   Red Hat elfutils is free software; you can redistribute it and/or modify
 
6
   it under the terms of the GNU General Public License as published by the
 
7
   Free Software Foundation; version 2 of the License.
 
8
 
 
9
   Red Hat elfutils is distributed in the hope that it will be useful, but
 
10
   WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
12
   General Public License for more details.
 
13
 
 
14
   You should have received a copy of the GNU General Public License along
 
15
   with Red Hat elfutils; if not, write to the Free Software Foundation,
 
16
   Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301 USA.
 
17
 
 
18
   Red Hat elfutils is an included package of the Open Invention Network.
 
19
   An included package of the Open Invention Network is a package for which
 
20
   Open Invention Network licensees cross-license their patents.  No patent
 
21
   license is granted, either expressly or impliedly, by designation as an
 
22
   included package.  Should you wish to participate in the Open Invention
 
23
   Network licensing program, please visit www.openinventionnetwork.com
 
24
   <http://www.openinventionnetwork.com>.  */
 
25
 
 
26
#ifdef HAVE_CONFIG_H
 
27
# include <config.h>
 
28
#endif
 
29
 
 
30
#include <fcntl.h>
 
31
#include <gelf.h>
 
32
#include <stdio.h>
 
33
#include <stdlib.h>
 
34
#include <string.h>
 
35
#include <unistd.h>
 
36
#include <sys/param.h>
 
37
 
 
38
 
 
39
int
 
40
main (int argc, char *argv[])
 
41
{
 
42
  int fd;
 
43
  Elf *elf;
 
44
  Elf *subelf;
 
45
  Elf_Cmd cmd;
 
46
  off_t offset;
 
47
  size_t todo;
 
48
 
 
49
  if (argc < 4)
 
50
    exit (1);
 
51
 
 
52
  /* Open the archive.  */
 
53
  fd = open (argv[1], O_RDONLY);
 
54
  if (fd == -1)
 
55
    {
 
56
      printf ("Cannot open input file: %m");
 
57
      exit (1);
 
58
    }
 
59
 
 
60
  /* Set the ELF version.  */
 
61
  elf_version (EV_CURRENT);
 
62
 
 
63
  /* Create an ELF descriptor.  */
 
64
  cmd = ELF_C_READ;
 
65
  elf = elf_begin (fd, cmd, NULL);
 
66
  if (elf == NULL)
 
67
    {
 
68
      printf ("Cannot create ELF descriptor: %s\n", elf_errmsg (-1));
 
69
      exit (1);
 
70
    }
 
71
 
 
72
  /* If it is no archive punt.  */
 
73
  if (elf_kind (elf) != ELF_K_AR)
 
74
    {
 
75
      printf ("`%s' is no archive\n", argv[1]);
 
76
      exit (1);
 
77
    }
 
78
 
 
79
  /* Get the elements of the archive one after the other.  */
 
80
  while ((subelf = elf_begin (fd, cmd, elf)) != NULL)
 
81
    {
 
82
      /* The the header for this element.  */
 
83
      Elf_Arhdr *arhdr = elf_getarhdr (subelf);
 
84
 
 
85
      if (arhdr == NULL)
 
86
        {
 
87
          printf ("cannot get arhdr: %s\n", elf_errmsg (-1));
 
88
          exit (1);
 
89
        }
 
90
 
 
91
      if (strcmp (arhdr->ar_name, argv[2]) == 0)
 
92
        {
 
93
          int outfd;
 
94
 
 
95
          /* Get the offset of the file in the archive.  */
 
96
          offset = elf_getbase (subelf);
 
97
          if (offset == -1)
 
98
            {
 
99
              printf ("\
 
100
Failed to get base address for the archive element: %s\n",
 
101
                      elf_errmsg (-1));
 
102
              exit (1);
 
103
            }
 
104
 
 
105
          /* Open the output file.  */
 
106
          outfd = open (argv[3], O_CREAT | O_TRUNC | O_RDWR, 0666);
 
107
          if (outfd == -1)
 
108
            {
 
109
              printf ("cannot open output file: %m");
 
110
              exit (1);
 
111
            }
 
112
 
 
113
          /* Now write out the data.  */
 
114
          todo = arhdr->ar_size;
 
115
          while (todo > 0)
 
116
            {
 
117
              char buf[1024];
 
118
              ssize_t n = pread (fd, buf, MIN (sizeof buf, todo), offset);
 
119
              if (n == 0)
 
120
                break;
 
121
 
 
122
              if (write (outfd, buf, n) != n)
 
123
                {
 
124
                  puts ("Writing output failed");
 
125
                  exit (1);
 
126
                }
 
127
 
 
128
              offset += n;
 
129
              todo -= n;
 
130
            }
 
131
 
 
132
          /* Check whether all the date was read and written out.  */
 
133
          if (todo != 0)
 
134
            {
 
135
              puts ("Reading archive member failed.");
 
136
              exit (1);
 
137
            }
 
138
 
 
139
          /* Close the descriptors.  */
 
140
          if (elf_end (subelf) != 0 || elf_end (elf) != 0)
 
141
            {
 
142
              printf ("Freeing ELF descriptors failed: %s", elf_errmsg (-1));
 
143
              exit (1);
 
144
            }
 
145
 
 
146
          close (outfd);
 
147
          close (fd);
 
148
 
 
149
          /* All went well.  */
 
150
          exit (0);
 
151
        }
 
152
 
 
153
      /* Get next archive element.  */
 
154
      cmd = elf_next (subelf);
 
155
      if (elf_end (subelf) != 0)
 
156
        {
 
157
          printf ("error while freeing sub-ELF descriptor: %s\n",
 
158
                  elf_errmsg (-1));
 
159
          exit (1);
 
160
        }
 
161
    }
 
162
 
 
163
  /* When we reach this point we haven't found the given file in the
 
164
     archive.  */
 
165
  printf ("File `%s' not found in archive\n", argv[2]);
 
166
  exit (1);
 
167
}