1/* size.c -- report size of various sections of an executable file.
2   Copyright 1991, 1992, 1993, 1994, 1995, 1996, 1997, 1998, 1999, 2000, 2001,
3   2002, 2003 Free Software Foundation, Inc.
4
5   This file is part of GNU Binutils.
6
7   This program is free software; you can redistribute it and/or modify
8   it under the terms of the GNU General Public License as published by
9   the Free Software Foundation; either version 2 of the License, or
10   (at your option) any later version.
11
12   This program is distributed in the hope that it will be useful,
13   but WITHOUT ANY WARRANTY; without even the implied warranty of
14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15   GNU General Public License for more details.
16
17   You should have received a copy of the GNU General Public License
18   along with this program; if not, write to the Free Software
19   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
20
21/* Extensions/incompatibilities:
22   o - BSD output has filenames at the end.
23   o - BSD output can appear in different radicies.
24   o - SysV output has less redundant whitespace.  Filename comes at end.
25   o - SysV output doesn't show VMA which is always the same as the PMA.
26   o - We also handle core files.
27   o - We also handle archives.
28   If you write shell scripts which manipulate this info then you may be
29   out of luck; there's no --compatibility or --pedantic option.  */
30
31#include "bfd.h"
32#include "bucomm.h"
33#include "libiberty.h"
34#include "getopt.h"
35
36#ifndef BSD_DEFAULT
37#define BSD_DEFAULT 1
38#endif
39
40/* Program options.  */
41
42enum
43  {
44    decimal, octal, hex
45  }
46radix = decimal;
47
48int berkeley_format = BSD_DEFAULT;	/* 0 means use AT&T-style output.  */
49int show_version = 0;
50int show_help = 0;
51int show_totals = 0;
52
53static bfd_size_type total_bsssize;
54static bfd_size_type total_datasize;
55static bfd_size_type total_textsize;
56
57/* Program exit status.  */
58int return_code = 0;
59
60static char *target = NULL;
61
62/* Static declarations.  */
63
64static void usage (FILE *, int);
65static void display_file (char *);
66static void display_bfd (bfd *);
67static void display_archive (bfd *);
68static int size_number (bfd_size_type);
69#if 0
70static void lprint_number (int, bfd_size_type);
71#endif
72static void rprint_number (int, bfd_size_type);
73static void print_berkeley_format (bfd *);
74static void sysv_internal_sizer (bfd *, asection *, void *);
75static void sysv_internal_printer (bfd *, asection *, void *);
76static void print_sysv_format (bfd *);
77static void print_sizes (bfd * file);
78static void berkeley_sum (bfd *, sec_ptr, void *);
79
80static void
81usage (FILE *stream, int status)
82{
83  fprintf (stream, _("Usage: %s [option(s)] [file(s)]\n"), program_name);
84  fprintf (stream, _(" Displays the sizes of sections inside binary files\n"));
85  fprintf (stream, _(" If no input file(s) are specified, a.out is assumed\n"));
86  fprintf (stream, _(" The options are:\n\
87  -A|-B     --format={sysv|berkeley}  Select output style (default is %s)\n\
88  -o|-d|-x  --radix={8|10|16}         Display numbers in octal, decimal or hex\n\
89  -t        --totals                  Display the total sizes (Berkeley only)\n\
90            --target=<bfdname>        Set the binary file format\n\
91  -h        --help                    Display this information\n\
92  -v        --version                 Display the program's version\n\
93\n"),
94#if BSD_DEFAULT
95  "berkeley"
96#else
97  "sysv"
98#endif
99);
100  list_supported_targets (program_name, stream);
101  if (status == 0)
102    fprintf (stream, _("Report bugs to %s\n"), REPORT_BUGS_TO);
103  exit (status);
104}
105
106struct option long_options[] =
107{
108  {"format", required_argument, 0, 200},
109  {"radix", required_argument, 0, 201},
110  {"target", required_argument, 0, 202},
111  {"totals", no_argument, &show_totals, 1},
112  {"version", no_argument, &show_version, 1},
113  {"help", no_argument, &show_help, 1},
114  {0, no_argument, 0, 0}
115};
116
117int main (int, char **);
118
119int
120main (int argc, char **argv)
121{
122  int temp;
123  int c;
124
125#if defined (HAVE_SETLOCALE) && defined (HAVE_LC_MESSAGES)
126  setlocale (LC_MESSAGES, "");
127#endif
128#if defined (HAVE_SETLOCALE)
129  setlocale (LC_CTYPE, "");
130#endif
131  bindtextdomain (PACKAGE, LOCALEDIR);
132  textdomain (PACKAGE);
133
134  program_name = *argv;
135  xmalloc_set_program_name (program_name);
136
137  expandargv (&argc, &argv);
138
139  bfd_init ();
140  set_default_bfd_target ();
141
142  while ((c = getopt_long (argc, argv, "ABHhVvdfotx", long_options,
143			   (int *) 0)) != EOF)
144    switch (c)
145      {
146      case 200:		/* --format */
147	switch (*optarg)
148	  {
149	  case 'B':
150	  case 'b':
151	    berkeley_format = 1;
152	    break;
153	  case 'S':
154	  case 's':
155	    berkeley_format = 0;
156	    break;
157	  default:
158	    non_fatal (_("invalid argument to --format: %s"), optarg);
159	    usage (stderr, 1);
160	  }
161	break;
162
163      case 202:		/* --target */
164	target = optarg;
165	break;
166
167      case 201:		/* --radix */
168#ifdef ANSI_LIBRARIES
169	temp = strtol (optarg, NULL, 10);
170#else
171	temp = atol (optarg);
172#endif
173	switch (temp)
174	  {
175	  case 10:
176	    radix = decimal;
177	    break;
178	  case 8:
179	    radix = octal;
180	    break;
181	  case 16:
182	    radix = hex;
183	    break;
184	  default:
185	    non_fatal (_("Invalid radix: %s\n"), optarg);
186	    usage (stderr, 1);
187	  }
188	break;
189
190      case 'A':
191	berkeley_format = 0;
192	break;
193      case 'B':
194	berkeley_format = 1;
195	break;
196      case 'v':
197      case 'V':
198	show_version = 1;
199	break;
200      case 'd':
201	radix = decimal;
202	break;
203      case 'x':
204	radix = hex;
205	break;
206      case 'o':
207	radix = octal;
208	break;
209      case 't':
210	show_totals = 1;
211	break;
212      case 'f': /* FIXME : For sysv68, `-f' means `full format', i.e.
213		   `[fname:] M(.text) + N(.data) + O(.bss) + P(.comment) = Q'
214		   where `fname: ' appears only if there are >= 2 input files,
215		   and M, N, O, P, Q are expressed in decimal by default,
216		   hexa or octal if requested by `-x' or `-o'.
217		   Just to make things interesting, Solaris also accepts -f,
218		   which prints out the size of each allocatable section, the
219		   name of the section, and the total of the section sizes.  */
220		/* For the moment, accept `-f' silently, and ignore it.  */
221	break;
222      case 0:
223	break;
224      case 'h':
225      case 'H':
226      case '?':
227	usage (stderr, 1);
228      }
229
230  if (show_version)
231    print_version ("size");
232  if (show_help)
233    usage (stdout, 0);
234
235  if (optind == argc)
236    display_file ("a.out");
237  else
238    for (; optind < argc;)
239      display_file (argv[optind++]);
240
241  if (show_totals && berkeley_format)
242    {
243      bfd_size_type total = total_textsize + total_datasize + total_bsssize;
244
245      rprint_number (7, total_textsize);
246      putchar('\t');
247      rprint_number (7, total_datasize);
248      putchar('\t');
249      rprint_number (7, total_bsssize);
250      printf (((radix == octal) ? "\t%7lo\t%7lx\t" : "\t%7lu\t%7lx\t"),
251	      (unsigned long) total, (unsigned long) total);
252      fputs ("(TOTALS)\n", stdout);
253    }
254
255  return return_code;
256}
257
258/* Display stats on file or archive member ABFD.  */
259
260static void
261display_bfd (bfd *abfd)
262{
263  char **matching;
264
265  if (bfd_check_format (abfd, bfd_archive))
266    /* An archive within an archive.  */
267    return;
268
269  if (bfd_check_format_matches (abfd, bfd_object, &matching))
270    {
271      print_sizes (abfd);
272      printf ("\n");
273      return;
274    }
275
276  if (bfd_get_error () == bfd_error_file_ambiguously_recognized)
277    {
278      bfd_nonfatal (bfd_get_filename (abfd));
279      list_matching_formats (matching);
280      free (matching);
281      return_code = 3;
282      return;
283    }
284
285  if (bfd_check_format_matches (abfd, bfd_core, &matching))
286    {
287      const char *core_cmd;
288
289      print_sizes (abfd);
290      fputs (" (core file", stdout);
291
292      core_cmd = bfd_core_file_failing_command (abfd);
293      if (core_cmd)
294	printf (" invoked as %s", core_cmd);
295
296      puts (")\n");
297      return;
298    }
299
300  bfd_nonfatal (bfd_get_filename (abfd));
301
302  if (bfd_get_error () == bfd_error_file_ambiguously_recognized)
303    {
304      list_matching_formats (matching);
305      free (matching);
306    }
307
308  return_code = 3;
309}
310
311static void
312display_archive (bfd *file)
313{
314  bfd *arfile = (bfd *) NULL;
315  bfd *last_arfile = (bfd *) NULL;
316
317  for (;;)
318    {
319      bfd_set_error (bfd_error_no_error);
320
321      arfile = bfd_openr_next_archived_file (file, arfile);
322      if (arfile == NULL)
323	{
324	  if (bfd_get_error () != bfd_error_no_more_archived_files)
325	    {
326	      bfd_nonfatal (bfd_get_filename (file));
327	      return_code = 2;
328	    }
329	  break;
330	}
331
332      display_bfd (arfile);
333
334      if (last_arfile != NULL)
335	bfd_close (last_arfile);
336      last_arfile = arfile;
337    }
338
339  if (last_arfile != NULL)
340    bfd_close (last_arfile);
341}
342
343static void
344display_file (char *filename)
345{
346  bfd *file;
347
348  if (get_file_size (filename) < 1)
349    return;
350
351  file = bfd_openr (filename, target);
352  if (file == NULL)
353    {
354      bfd_nonfatal (filename);
355      return_code = 1;
356      return;
357    }
358
359  if (bfd_check_format (file, bfd_archive))
360    display_archive (file);
361  else
362    display_bfd (file);
363
364  if (!bfd_close (file))
365    {
366      bfd_nonfatal (filename);
367      return_code = 1;
368      return;
369    }
370}
371
372/* This is what lexical functions are for.  */
373
374static int
375size_number (bfd_size_type num)
376{
377  char buffer[40];
378
379  sprintf (buffer,
380	   (radix == decimal ? "%lu" :
381	   ((radix == octal) ? "0%lo" : "0x%lx")),
382	   (unsigned long) num);
383
384  return strlen (buffer);
385}
386
387#if 0
388
389/* This is not used.  */
390
391static void
392lprint_number (int width, bfd_size_type num)
393{
394  char buffer[40];
395
396  sprintf (buffer,
397	   (radix == decimal ? "%lu" :
398	   ((radix == octal) ? "0%lo" : "0x%lx")),
399	   (unsigned long) num);
400
401  printf ("%-*s", width, buffer);
402}
403
404#endif
405
406static void
407rprint_number (int width, bfd_size_type num)
408{
409  char buffer[40];
410
411  sprintf (buffer,
412	   (radix == decimal ? "%lu" :
413	   ((radix == octal) ? "0%lo" : "0x%lx")),
414	   (unsigned long) num);
415
416  printf ("%*s", width, buffer);
417}
418
419static bfd_size_type bsssize;
420static bfd_size_type datasize;
421static bfd_size_type textsize;
422
423static void
424berkeley_sum (bfd *abfd ATTRIBUTE_UNUSED, sec_ptr sec,
425	      void *ignore ATTRIBUTE_UNUSED)
426{
427  flagword flags;
428  bfd_size_type size;
429
430  flags = bfd_get_section_flags (abfd, sec);
431  if ((flags & SEC_ALLOC) == 0)
432    return;
433
434  size = bfd_get_section_size_before_reloc (sec);
435  if ((flags & SEC_CODE) != 0 || (flags & SEC_READONLY) != 0)
436    textsize += size;
437  else if ((flags & SEC_HAS_CONTENTS) != 0)
438    datasize += size;
439  else
440    bsssize += size;
441}
442
443static void
444print_berkeley_format (bfd *abfd)
445{
446  static int files_seen = 0;
447  bfd_size_type total;
448
449  bsssize = 0;
450  datasize = 0;
451  textsize = 0;
452
453  bfd_map_over_sections (abfd, berkeley_sum, NULL);
454
455  if (files_seen++ == 0)
456#if 0
457    /* Intel doesn't like bss/stk because they don't have core files.  */
458    puts ((radix == octal) ? "   text\t   data\tbss/stk\t    oct\t    hex\tfilename" :
459	  "   text\t   data\tbss/stk\t    dec\t    hex\tfilename");
460#else
461    puts ((radix == octal) ? "   text\t   data\t    bss\t    oct\t    hex\tfilename" :
462	  "   text\t   data\t    bss\t    dec\t    hex\tfilename");
463#endif
464
465  total = textsize + datasize + bsssize;
466
467  if (show_totals)
468    {
469      total_textsize += textsize;
470      total_datasize += datasize;
471      total_bsssize  += bsssize;
472    }
473
474  rprint_number (7, textsize);
475  putchar ('\t');
476  rprint_number (7, datasize);
477  putchar ('\t');
478  rprint_number (7, bsssize);
479  printf (((radix == octal) ? "\t%7lo\t%7lx\t" : "\t%7lu\t%7lx\t"),
480	  (unsigned long) total, (unsigned long) total);
481
482  fputs (bfd_get_filename (abfd), stdout);
483
484  if (bfd_my_archive (abfd))
485    printf (" (ex %s)", bfd_get_filename (bfd_my_archive (abfd)));
486}
487
488/* I REALLY miss lexical functions! */
489bfd_size_type svi_total = 0;
490bfd_vma svi_maxvma = 0;
491int svi_namelen = 0;
492int svi_vmalen = 0;
493int svi_sizelen = 0;
494
495static void
496sysv_internal_sizer (bfd *file ATTRIBUTE_UNUSED, sec_ptr sec,
497		     void *ignore ATTRIBUTE_UNUSED)
498{
499  bfd_size_type size = bfd_section_size (file, sec);
500
501  if (   ! bfd_is_abs_section (sec)
502      && ! bfd_is_com_section (sec)
503      && ! bfd_is_und_section (sec))
504    {
505      int namelen = strlen (bfd_section_name (file, sec));
506
507      if (namelen > svi_namelen)
508	svi_namelen = namelen;
509
510      svi_total += size;
511
512      if (bfd_section_vma (file, sec) > svi_maxvma)
513	svi_maxvma = bfd_section_vma (file, sec);
514    }
515}
516
517static void
518sysv_internal_printer (bfd *file ATTRIBUTE_UNUSED, sec_ptr sec,
519		       void *ignore ATTRIBUTE_UNUSED)
520{
521  bfd_size_type size = bfd_section_size (file, sec);
522
523  if (   ! bfd_is_abs_section (sec)
524      && ! bfd_is_com_section (sec)
525      && ! bfd_is_und_section (sec))
526    {
527      svi_total += size;
528
529      printf ("%-*s   ", svi_namelen, bfd_section_name (file, sec));
530      rprint_number (svi_sizelen, size);
531      printf ("   ");
532      rprint_number (svi_vmalen, bfd_section_vma (file, sec));
533      printf ("\n");
534    }
535}
536
537static void
538print_sysv_format (bfd *file)
539{
540  /* Size all of the columns.  */
541  svi_total = 0;
542  svi_maxvma = 0;
543  svi_namelen = 0;
544  bfd_map_over_sections (file, sysv_internal_sizer, NULL);
545  svi_vmalen = size_number ((bfd_size_type)svi_maxvma);
546
547  if ((size_t) svi_vmalen < sizeof ("addr") - 1)
548    svi_vmalen = sizeof ("addr")-1;
549
550  svi_sizelen = size_number (svi_total);
551  if ((size_t) svi_sizelen < sizeof ("size") - 1)
552    svi_sizelen = sizeof ("size")-1;
553
554  svi_total = 0;
555  printf ("%s  ", bfd_get_filename (file));
556
557  if (bfd_my_archive (file))
558    printf (" (ex %s)", bfd_get_filename (bfd_my_archive (file)));
559
560  printf (":\n%-*s   %*s   %*s\n", svi_namelen, "section",
561	  svi_sizelen, "size", svi_vmalen, "addr");
562
563  bfd_map_over_sections (file, sysv_internal_printer, NULL);
564
565  printf ("%-*s   ", svi_namelen, "Total");
566  rprint_number (svi_sizelen, svi_total);
567  printf ("\n\n");
568}
569
570static void
571print_sizes (bfd *file)
572{
573  if (berkeley_format)
574    print_berkeley_format (file);
575  else
576    print_sysv_format (file);
577}
578