[deliverable/binutils-gdb.git] / bfd / compress.c

/* Compressed section support (intended for debug sections).
   Copyright (C) 2008-2014 Free Software Foundation, Inc.

   This file is part of BFD, the Binary File Descriptor library.

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston,
   MA 02110-1301, USA.  */

#include "sysdep.h"
#include "bfd.h"
#include "libbfd.h"
#ifdef HAVE_ZLIB_H
#include <zlib.h>
#endif

#ifdef HAVE_ZLIB_H
static bfd_boolean
decompress_contents (bfd_byte *compressed_buffer,
		     bfd_size_type compressed_size,
		     bfd_byte *uncompressed_buffer,
		     bfd_size_type uncompressed_size)
{
  z_stream strm;
  int rc;

  /* It is possible the section consists of several compressed
     buffers concatenated together, so we uncompress in a loop.  */
  strm.zalloc = NULL;
  strm.zfree = NULL;
  strm.opaque = NULL;
  strm.avail_in = compressed_size - 12;
  strm.next_in = (Bytef*) compressed_buffer + 12;
  strm.avail_out = uncompressed_size;

  BFD_ASSERT (Z_OK == 0);
  rc = inflateInit (&strm);
  while (strm.avail_in > 0 && strm.avail_out > 0)
    {
      if (rc != Z_OK)
	break;
      strm.next_out = ((Bytef*) uncompressed_buffer
                       + (uncompressed_size - strm.avail_out));
      rc = inflate (&strm, Z_FINISH);
      if (rc != Z_STREAM_END)
	break;
      rc = inflateReset (&strm);
    }
  rc |= inflateEnd (&strm);
  return rc == Z_OK && strm.avail_out == 0;
}
#endif

/*
FUNCTION
	bfd_compress_section_contents

SYNOPSIS
	bfd_boolean bfd_compress_section_contents
	  (bfd *abfd, asection *section, bfd_byte *uncompressed_buffer,
	   bfd_size_type uncompressed_size);

DESCRIPTION

	Compress data of the size specified in @var{uncompressed_size}
	and pointed to by @var{uncompressed_buffer} using zlib and store
	as the contents field.  This function assumes the contents
	field was allocated using bfd_malloc() or equivalent.  If zlib
	is not installed on this machine, the input is unmodified.

	Return @code{TRUE} if the full section contents is compressed
	successfully.
*/

bfd_boolean
bfd_compress_section_contents (bfd *abfd ATTRIBUTE_UNUSED,
			       sec_ptr sec ATTRIBUTE_UNUSED,
			       bfd_byte *uncompressed_buffer ATTRIBUTE_UNUSED,
			       bfd_size_type uncompressed_size ATTRIBUTE_UNUSED)
{
#ifndef HAVE_ZLIB_H
  bfd_set_error (bfd_error_invalid_operation);
  return FALSE;
#else
  uLong compressed_size;
  bfd_byte *compressed_buffer;

  compressed_size = compressBound (uncompressed_size) + 12;
  compressed_buffer = (bfd_byte *) bfd_malloc (compressed_size);

  if (compressed_buffer == NULL)
    return FALSE;

  if (compress ((Bytef*) compressed_buffer + 12,
		&compressed_size,
		(const Bytef*) uncompressed_buffer,
		uncompressed_size) != Z_OK)
    {
      free (compressed_buffer);
      bfd_set_error (bfd_error_bad_value);
      return FALSE;
    }

  /* Write the zlib header.  In this case, it should be "ZLIB" followed
     by the uncompressed section size, 8 bytes in big-endian order.  */
  memcpy (compressed_buffer, "ZLIB", 4);
  compressed_buffer[11] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[10] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[9] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[8] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[7] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[6] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[5] = uncompressed_size; uncompressed_size >>= 8;
  compressed_buffer[4] = uncompressed_size;
  compressed_size += 12;

  /* Free the uncompressed contents if we compress in place.  */
  if (uncompressed_buffer == sec->contents)
    free (uncompressed_buffer);

  sec->contents = compressed_buffer;
  sec->size = compressed_size;
  sec->compress_status = COMPRESS_SECTION_DONE;

  return TRUE;
#endif  /* HAVE_ZLIB_H */
}

/*
FUNCTION
	bfd_get_full_section_contents

SYNOPSIS
	bfd_boolean bfd_get_full_section_contents
	  (bfd *abfd, asection *section, bfd_byte **ptr);

DESCRIPTION
	Read all data from @var{section} in BFD @var{abfd}, decompress
	if needed, and store in @var{*ptr}.  If @var{*ptr} is NULL,
	return @var{*ptr} with memory malloc'd by this function.

	Return @code{TRUE} if the full section contents is retrieved
	successfully.
*/

bfd_boolean
bfd_get_full_section_contents (bfd *abfd, sec_ptr sec, bfd_byte **ptr)
{
  bfd_size_type sz;
  bfd_byte *p = *ptr;
#ifdef HAVE_ZLIB_H
  bfd_boolean ret;
  bfd_size_type save_size;
  bfd_size_type save_rawsize;
  bfd_byte *compressed_buffer;
#endif

  if (abfd->direction != write_direction && sec->rawsize != 0)
    sz = sec->rawsize;
  else
    sz = sec->size;
  if (sz == 0)
    return TRUE;

  switch (sec->compress_status)
    {
    case COMPRESS_SECTION_NONE:
      if (p == NULL)
	{
	  p = (bfd_byte *) bfd_malloc (sz);
	  if (p == NULL)
	    return FALSE;
	}
      if (!bfd_get_section_contents (abfd, sec, p, 0, sz))
	{
	  if (*ptr != p)
	    free (p);
	  return FALSE;
	}
      *ptr = p;
      return TRUE;

    case DECOMPRESS_SECTION_SIZED:
#ifndef HAVE_ZLIB_H
      bfd_set_error (bfd_error_invalid_operation);
      return FALSE;
#else
      /* Read in the full compressed section contents.  */
      compressed_buffer = (bfd_byte *) bfd_malloc (sec->compressed_size);
      if (compressed_buffer == NULL)
	return FALSE;
      save_rawsize = sec->rawsize;
      save_size = sec->size;
      /* Clear rawsize, set size to compressed size and set compress_status
	 to COMPRESS_SECTION_NONE.  If the compressed size is bigger than
	 the uncompressed size, bfd_get_section_contents will fail.  */
      sec->rawsize = 0;
      sec->size = sec->compressed_size;
      sec->compress_status = COMPRESS_SECTION_NONE;
      ret = bfd_get_section_contents (abfd, sec, compressed_buffer,
				      0, sec->compressed_size);
      /* Restore rawsize and size.  */
      sec->rawsize = save_rawsize;
      sec->size = save_size;
      sec->compress_status = DECOMPRESS_SECTION_SIZED;
      if (!ret)
	goto fail_compressed;

      if (p == NULL)
	p = (bfd_byte *) bfd_malloc (sz);
      if (p == NULL)
	goto fail_compressed;

      if (!decompress_contents (compressed_buffer, sec->compressed_size, p, sz))
	{
	  bfd_set_error (bfd_error_bad_value);
	  if (p != *ptr)
	    free (p);
	fail_compressed:
	  free (compressed_buffer);
	  return FALSE;
	}

      free (compressed_buffer);
      *ptr = p;
      return TRUE;
#endif

    case COMPRESS_SECTION_DONE:
      if (p == NULL)
	{
	  p = (bfd_byte *) bfd_malloc (sz);
	  if (p == NULL)
	    return FALSE;
	  *ptr = p;
	}
      memcpy (p, sec->contents, sz);
      return TRUE;

    default:
      abort ();
    }
}

/*
FUNCTION
	bfd_cache_section_contents

SYNOPSIS
	void bfd_cache_section_contents
	  (asection *sec, void *contents);

DESCRIPTION
	Stash @var(contents) so any following reads of @var(sec) do
	not need to decompress again.
*/

void
bfd_cache_section_contents (asection *sec, void *contents)
{
  if (sec->compress_status == DECOMPRESS_SECTION_SIZED)
    sec->compress_status = COMPRESS_SECTION_DONE;
  sec->contents = contents;
  sec->flags |= SEC_IN_MEMORY;
}


/*
FUNCTION
	bfd_is_section_compressed

SYNOPSIS
	bfd_boolean bfd_is_section_compressed
	  (bfd *abfd, asection *section);

DESCRIPTION
	Return @code{TRUE} if @var{section} is compressed.
*/

bfd_boolean
bfd_is_section_compressed (bfd *abfd, sec_ptr sec)
{
  bfd_byte compressed_buffer [12];
  unsigned int saved = sec->compress_status;
  bfd_boolean compressed;

  /* Don't decompress the section.  */
  sec->compress_status = COMPRESS_SECTION_NONE;

  /* Read the zlib header.  In this case, it should be "ZLIB" followed
     by the uncompressed section size, 8 bytes in big-endian order.  */
  compressed = (bfd_get_section_contents (abfd, sec, compressed_buffer, 0, 12)
		&& CONST_STRNEQ ((char*) compressed_buffer, "ZLIB"));

  /* Restore compress_status.  */
  sec->compress_status = saved;
  return compressed;
}

/*
FUNCTION
	bfd_init_section_decompress_status

SYNOPSIS
	bfd_boolean bfd_init_section_decompress_status
	  (bfd *abfd, asection *section);

DESCRIPTION
	Record compressed section size, update section size with
	decompressed size and set compress_status to
	DECOMPRESS_SECTION_SIZED.

	Return @code{FALSE} if the section is not a valid compressed
	section or zlib is not installed on this machine.  Otherwise,
	return @code{TRUE}.
*/

bfd_boolean
bfd_init_section_decompress_status (bfd *abfd ATTRIBUTE_UNUSED,
				    sec_ptr sec ATTRIBUTE_UNUSED)
{
#ifndef HAVE_ZLIB_H
  bfd_set_error (bfd_error_invalid_operation);
  return FALSE;
#else
  bfd_byte compressed_buffer [12];
  bfd_size_type uncompressed_size;

  if (sec->rawsize != 0
      || sec->contents != NULL
      || sec->compress_status != COMPRESS_SECTION_NONE
      || !bfd_get_section_contents (abfd, sec, compressed_buffer, 0, 12))
    {
      bfd_set_error (bfd_error_invalid_operation);
      return FALSE;
    }

  /* Read the zlib header.  In this case, it should be "ZLIB" followed
     by the uncompressed section size, 8 bytes in big-endian order.  */
  if (! CONST_STRNEQ ((char*) compressed_buffer, "ZLIB"))
    {
      bfd_set_error (bfd_error_wrong_format);
      return FALSE;
    }

  uncompressed_size = compressed_buffer[4]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[5]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[6]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[7]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[8]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[9]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[10]; uncompressed_size <<= 8;
  uncompressed_size += compressed_buffer[11];

  sec->compressed_size = sec->size;
  sec->size = uncompressed_size;
  sec->compress_status = DECOMPRESS_SECTION_SIZED;

  return TRUE;
#endif
}

/*
FUNCTION
	bfd_init_section_compress_status

SYNOPSIS
	bfd_boolean bfd_init_section_compress_status
	  (bfd *abfd, asection *section);

DESCRIPTION
	If open for read, compress section, update section size with
	compressed size and set compress_status to COMPRESS_SECTION_DONE.

	Return @code{FALSE} if the section is not a valid compressed
	section or zlib is not installed on this machine.  Otherwise,
	return @code{TRUE}.
*/

bfd_boolean
bfd_init_section_compress_status (bfd *abfd ATTRIBUTE_UNUSED,
				  sec_ptr sec ATTRIBUTE_UNUSED)
{
#ifndef HAVE_ZLIB_H
  bfd_set_error (bfd_error_invalid_operation);
  return FALSE;
#else
  bfd_size_type uncompressed_size;
  bfd_byte *uncompressed_buffer;
  bfd_boolean ret;

  /* Error if not opened for read.  */
  if (abfd->direction != read_direction
      || sec->size == 0
      || sec->rawsize != 0
      || sec->contents != NULL
      || sec->compress_status != COMPRESS_SECTION_NONE)
    {
      bfd_set_error (bfd_error_invalid_operation);
      return FALSE;
    }

  /* Read in the full section contents and compress it.  */
  uncompressed_size = sec->size;
  uncompressed_buffer = (bfd_byte *) bfd_malloc (uncompressed_size);
  if (!bfd_get_section_contents (abfd, sec, uncompressed_buffer,
				 0, uncompressed_size))
    ret = FALSE;
  else
    ret = bfd_compress_section_contents (abfd, sec,
					 uncompressed_buffer,
					 uncompressed_size);

  free (uncompressed_buffer);
  return ret;
#endif
}
Commit	Line	Data
0acf065b	1	/* Compressed section support (intended for debug sections).
4b95cf5c	2	Copyright (C) 2008-2014 Free Software Foundation, Inc.
1b315056 CS	3
	4	This file is part of BFD, the Binary File Descriptor library.
	5
	6	This program is free software; you can redistribute it and/or modify
	7	it under the terms of the GNU General Public License as published by
	8	the Free Software Foundation; either version 3 of the License, or
	9	(at your option) any later version.
	10
	11	This program is distributed in the hope that it will be useful,
	12	but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	GNU General Public License for more details.
	15
	16	You should have received a copy of the GNU General Public License
	17	along with this program; if not, write to the Free Software
	18	Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston,
	19	MA 02110-1301, USA. */
	20
1b315056 CS	21	#include "sysdep.h"
	22	#include "bfd.h"
	23	#include "libbfd.h"
	24	#ifdef HAVE_ZLIB_H
	25	#include <zlib.h>
	26	#endif
	27
4a114e3e L	28	#ifdef HAVE_ZLIB_H
	29	static bfd_boolean
	30	decompress_contents (bfd_byte *compressed_buffer,
	31	bfd_size_type compressed_size,
	32	bfd_byte *uncompressed_buffer,
	33	bfd_size_type uncompressed_size)
	34	{
	35	z_stream strm;
	36	int rc;
	37
	38	/* It is possible the section consists of several compressed
	39	buffers concatenated together, so we uncompress in a loop. */
	40	strm.zalloc = NULL;
	41	strm.zfree = NULL;
	42	strm.opaque = NULL;
	43	strm.avail_in = compressed_size - 12;
	44	strm.next_in = (Bytef*) compressed_buffer + 12;
	45	strm.avail_out = uncompressed_size;
	46
a253d456	47	BFD_ASSERT (Z_OK == 0);
4a114e3e	48	rc = inflateInit (&strm);
a29a8af8	49	while (strm.avail_in > 0 && strm.avail_out > 0)
4a114e3e L	50	{
4a114e3e L	51	if (rc != Z_OK)
a253d456	52	break;
4a114e3e L	53	strm.next_out = ((Bytef*) uncompressed_buffer
	54	+ (uncompressed_size - strm.avail_out));
	55	rc = inflate (&strm, Z_FINISH);
	56	if (rc != Z_STREAM_END)
a253d456	57	break;
4a114e3e L	58	rc = inflateReset (&strm);
4a114e3e L	59	}
a253d456	60	rc \|= inflateEnd (&strm);
82c6068a	61	return rc == Z_OK && strm.avail_out == 0;
4a114e3e L	62	}
	63	#endif
	64
1b315056 CS	65	/*
1b315056 CS	66	FUNCTION
4a114e3e	67	bfd_compress_section_contents
1b315056 CS	68
1b315056 CS	69	SYNOPSIS
4a114e3e L	70	bfd_boolean bfd_compress_section_contents
	71	(bfd abfd, asection section, bfd_byte *uncompressed_buffer,
	72	bfd_size_type uncompressed_size);
1b315056 CS	73
	74	DESCRIPTION
	75
4a114e3e L	76	Compress data of the size specified in @var{uncompressed_size}
	77	and pointed to by @var{uncompressed_buffer} using zlib and store
	78	as the contents field. This function assumes the contents
	79	field was allocated using bfd_malloc() or equivalent. If zlib
	80	is not installed on this machine, the input is unmodified.
	81
68ffbac6	82	Return @code{TRUE} if the full section contents is compressed
4a114e3e	83	successfully.
1b315056 CS	84	*/
	85
	86	bfd_boolean
4a114e3e L	87	bfd_compress_section_contents (bfd *abfd ATTRIBUTE_UNUSED,
	88	sec_ptr sec ATTRIBUTE_UNUSED,
	89	bfd_byte *uncompressed_buffer ATTRIBUTE_UNUSED,
	90	bfd_size_type uncompressed_size ATTRIBUTE_UNUSED)
1b315056 CS	91	{
1b315056 CS	92	#ifndef HAVE_ZLIB_H
4a114e3e	93	bfd_set_error (bfd_error_invalid_operation);
1b315056 CS	94	return FALSE;
1b315056 CS	95	#else
ae6a0217	96	uLong compressed_size;
4a114e3e L	97	bfd_byte *compressed_buffer;
	98
	99	compressed_size = compressBound (uncompressed_size) + 12;
	100	compressed_buffer = (bfd_byte *) bfd_malloc (compressed_size);
	101
4281caad MS	102	if (compressed_buffer == NULL)
	103	return FALSE;
	104
4a114e3e L	105	if (compress ((Bytef*) compressed_buffer + 12,
	106	&compressed_size,
	107	(const Bytef*) uncompressed_buffer,
	108	uncompressed_size) != Z_OK)
	109	{
	110	free (compressed_buffer);
	111	bfd_set_error (bfd_error_bad_value);
	112	return FALSE;
	113	}
	114
	115	/* Write the zlib header. In this case, it should be "ZLIB" followed
	116	by the uncompressed section size, 8 bytes in big-endian order. */
	117	memcpy (compressed_buffer, "ZLIB", 4);
	118	compressed_buffer[11] = uncompressed_size; uncompressed_size >>= 8;
	119	compressed_buffer[10] = uncompressed_size; uncompressed_size >>= 8;
	120	compressed_buffer[9] = uncompressed_size; uncompressed_size >>= 8;
	121	compressed_buffer[8] = uncompressed_size; uncompressed_size >>= 8;
	122	compressed_buffer[7] = uncompressed_size; uncompressed_size >>= 8;
	123	compressed_buffer[6] = uncompressed_size; uncompressed_size >>= 8;
	124	compressed_buffer[5] = uncompressed_size; uncompressed_size >>= 8;
	125	compressed_buffer[4] = uncompressed_size;
	126	compressed_size += 12;
	127
	128	/* Free the uncompressed contents if we compress in place. */
	129	if (uncompressed_buffer == sec->contents)
	130	free (uncompressed_buffer);
	131
	132	sec->contents = compressed_buffer;
	133	sec->size = compressed_size;
	134	sec->compress_status = COMPRESS_SECTION_DONE;
	135
	136	return TRUE;
	137	#endif /* HAVE_ZLIB_H */
	138	}
	139
	140	/*
	141	FUNCTION
	142	bfd_get_full_section_contents
	143
	144	SYNOPSIS
	145	bfd_boolean bfd_get_full_section_contents
	146	(bfd abfd, asection section, bfd_byte **ptr);
	147
	148	DESCRIPTION
	149	Read all data from @var{section} in BFD @var{abfd}, decompress
	150	if needed, and store in @var{ptr}. If @var{ptr} is NULL,
68ffbac6	151	return @var{*ptr} with memory malloc'd by this function.
4a114e3e L	152
	153	Return @code{TRUE} if the full section contents is retrieved
	154	successfully.
	155	*/
	156
	157	bfd_boolean
	158	bfd_get_full_section_contents (bfd abfd, sec_ptr sec, bfd_byte *ptr)
	159	{
e57278ef	160	bfd_size_type sz;
4a114e3e	161	bfd_byte p = ptr;
4a114e3e	162	#ifdef HAVE_ZLIB_H
82c6068a	163	bfd_boolean ret;
38b774d2 AM	164	bfd_size_type save_size;
38b774d2 AM	165	bfd_size_type save_rawsize;
4a114e3e	166	bfd_byte *compressed_buffer;
4a114e3e L	167	#endif
4a114e3e L	168
e57278ef AM	169	if (abfd->direction != write_direction && sec->rawsize != 0)
	170	sz = sec->rawsize;
	171	else
	172	sz = sec->size;
4a114e3e L	173	if (sz == 0)
	174	return TRUE;
	175
	176	switch (sec->compress_status)
	177	{
	178	case COMPRESS_SECTION_NONE:
	179	if (p == NULL)
	180	{
0d13c96b	181	p = (bfd_byte *) bfd_malloc (sz);
4a114e3e L	182	if (p == NULL)
4a114e3e L	183	return FALSE;
4a114e3e	184	}
82c6068a AM	185	if (!bfd_get_section_contents (abfd, sec, p, 0, sz))
	186	{
	187	if (*ptr != p)
	188	free (p);
	189	return FALSE;
	190	}
	191	*ptr = p;
4a114e3e L	192	return TRUE;
	193
	194	case DECOMPRESS_SECTION_SIZED:
4a114e3e	195	#ifndef HAVE_ZLIB_H
82c6068a AM	196	bfd_set_error (bfd_error_invalid_operation);
82c6068a AM	197	return FALSE;
4a114e3e	198	#else
82c6068a	199	/* Read in the full compressed section contents. */
38b774d2	200	compressed_buffer = (bfd_byte *) bfd_malloc (sec->compressed_size);
82c6068a AM	201	if (compressed_buffer == NULL)
82c6068a AM	202	return FALSE;
38b774d2 AM	203	save_rawsize = sec->rawsize;
38b774d2 AM	204	save_size = sec->size;
82c6068a AM	205	/* Clear rawsize, set size to compressed size and set compress_status
	206	to COMPRESS_SECTION_NONE. If the compressed size is bigger than
	207	the uncompressed size, bfd_get_section_contents will fail. */
	208	sec->rawsize = 0;
38b774d2	209	sec->size = sec->compressed_size;
82c6068a AM	210	sec->compress_status = COMPRESS_SECTION_NONE;
82c6068a AM	211	ret = bfd_get_section_contents (abfd, sec, compressed_buffer,
38b774d2	212	0, sec->compressed_size);
82c6068a	213	/* Restore rawsize and size. */
38b774d2 AM	214	sec->rawsize = save_rawsize;
38b774d2 AM	215	sec->size = save_size;
4a114e3e	216	sec->compress_status = DECOMPRESS_SECTION_SIZED;
82c6068a AM	217	if (!ret)
82c6068a AM	218	goto fail_compressed;
4a114e3e	219
38b774d2 AM	220	if (p == NULL)
	221	p = (bfd_byte *) bfd_malloc (sz);
	222	if (p == NULL)
4a114e3e	223	goto fail_compressed;
4a114e3e	224
38b774d2	225	if (!decompress_contents (compressed_buffer, sec->compressed_size, p, sz))
82c6068a AM	226	{
82c6068a AM	227	bfd_set_error (bfd_error_bad_value);
38b774d2 AM	228	if (p != *ptr)
38b774d2 AM	229	free (p);
82c6068a AM	230	fail_compressed:
	231	free (compressed_buffer);
	232	return FALSE;
	233	}
4a114e3e	234
82c6068a	235	free (compressed_buffer);
38b774d2 AM	236	*ptr = p;
38b774d2 AM	237	return TRUE;
82c6068a	238	#endif
4a114e3e	239
82c6068a AM	240	case COMPRESS_SECTION_DONE:
	241	if (p == NULL)
	242	{
	243	p = (bfd_byte *) bfd_malloc (sz);
	244	if (p == NULL)
	245	return FALSE;
	246	*ptr = p;
	247	}
	248	memcpy (p, sec->contents, sz);
	249	return TRUE;
4a114e3e	250
82c6068a AM	251	default:
	252	abort ();
	253	}
4a114e3e L	254	}
4a114e3e L	255
8a72cc6e AM	256	/*
	257	FUNCTION
	258	bfd_cache_section_contents
	259
	260	SYNOPSIS
	261	void bfd_cache_section_contents
	262	(asection sec, void contents);
	263
	264	DESCRIPTION
	265	Stash @var(contents) so any following reads of @var(sec) do
	266	not need to decompress again.
	267	*/
	268
	269	void
	270	bfd_cache_section_contents (asection sec, void contents)
	271	{
	272	if (sec->compress_status == DECOMPRESS_SECTION_SIZED)
	273	sec->compress_status = COMPRESS_SECTION_DONE;
	274	sec->contents = contents;
	275	sec->flags \|= SEC_IN_MEMORY;
	276	}
	277
	278
4a114e3e L	279	/*
	280	FUNCTION
	281	bfd_is_section_compressed
	282
	283	SYNOPSIS
	284	bfd_boolean bfd_is_section_compressed
	285	(bfd abfd, asection section);
	286
	287	DESCRIPTION
	288	Return @code{TRUE} if @var{section} is compressed.
	289	*/
	290
	291	bfd_boolean
	292	bfd_is_section_compressed (bfd *abfd, sec_ptr sec)
	293	{
	294	bfd_byte compressed_buffer [12];
64f40162 L	295	unsigned int saved = sec->compress_status;
	296	bfd_boolean compressed;
	297
	298	/* Don't decompress the section. */
	299	sec->compress_status = COMPRESS_SECTION_NONE;
4a114e3e L	300
	301	/* Read the zlib header. In this case, it should be "ZLIB" followed
	302	by the uncompressed section size, 8 bytes in big-endian order. */
64f40162 L	303	compressed = (bfd_get_section_contents (abfd, sec, compressed_buffer, 0, 12)
	304	&& CONST_STRNEQ ((char*) compressed_buffer, "ZLIB"));
	305
	306	/* Restore compress_status. */
	307	sec->compress_status = saved;
	308	return compressed;
4a114e3e L	309	}
	310
	311	/*
	312	FUNCTION
	313	bfd_init_section_decompress_status
	314
	315	SYNOPSIS
	316	bfd_boolean bfd_init_section_decompress_status
	317	(bfd abfd, asection section);
	318
	319	DESCRIPTION
	320	Record compressed section size, update section size with
	321	decompressed size and set compress_status to
	322	DECOMPRESS_SECTION_SIZED.
	323
	324	Return @code{FALSE} if the section is not a valid compressed
	325	section or zlib is not installed on this machine. Otherwise,
	326	return @code{TRUE}.
	327	*/
	328
	329	bfd_boolean
	330	bfd_init_section_decompress_status (bfd *abfd ATTRIBUTE_UNUSED,
	331	sec_ptr sec ATTRIBUTE_UNUSED)
	332	{
	333	#ifndef HAVE_ZLIB_H
	334	bfd_set_error (bfd_error_invalid_operation);
	335	return FALSE;
	336	#else
	337	bfd_byte compressed_buffer [12];
	338	bfd_size_type uncompressed_size;
	339
	340	if (sec->rawsize != 0
	341	\|\| sec->contents != NULL
	342	\|\| sec->compress_status != COMPRESS_SECTION_NONE
	343	\|\| !bfd_get_section_contents (abfd, sec, compressed_buffer, 0, 12))
	344	{
	345	bfd_set_error (bfd_error_invalid_operation);
	346	return FALSE;
	347	}
1b315056 CS	348
	349	/* Read the zlib header. In this case, it should be "ZLIB" followed
	350	by the uncompressed section size, 8 bytes in big-endian order. */
4a114e3e L	351	if (! CONST_STRNEQ ((char*) compressed_buffer, "ZLIB"))
	352	{
	353	bfd_set_error (bfd_error_wrong_format);
	354	return FALSE;
	355	}
	356
1b315056 CS	357	uncompressed_size = compressed_buffer[4]; uncompressed_size <<= 8;
	358	uncompressed_size += compressed_buffer[5]; uncompressed_size <<= 8;
	359	uncompressed_size += compressed_buffer[6]; uncompressed_size <<= 8;
	360	uncompressed_size += compressed_buffer[7]; uncompressed_size <<= 8;
	361	uncompressed_size += compressed_buffer[8]; uncompressed_size <<= 8;
	362	uncompressed_size += compressed_buffer[9]; uncompressed_size <<= 8;
	363	uncompressed_size += compressed_buffer[10]; uncompressed_size <<= 8;
	364	uncompressed_size += compressed_buffer[11];
	365
4a114e3e L	366	sec->compressed_size = sec->size;
	367	sec->size = uncompressed_size;
	368	sec->compress_status = DECOMPRESS_SECTION_SIZED;
1b315056	369
4a114e3e L	370	return TRUE;
	371	#endif
	372	}
	373
	374	/*
	375	FUNCTION
	376	bfd_init_section_compress_status
	377
	378	SYNOPSIS
	379	bfd_boolean bfd_init_section_compress_status
	380	(bfd abfd, asection section);
	381
	382	DESCRIPTION
	383	If open for read, compress section, update section size with
	384	compressed size and set compress_status to COMPRESS_SECTION_DONE.
	385
	386	Return @code{FALSE} if the section is not a valid compressed
	387	section or zlib is not installed on this machine. Otherwise,
	388	return @code{TRUE}.
	389	*/
	390
	391	bfd_boolean
	392	bfd_init_section_compress_status (bfd *abfd ATTRIBUTE_UNUSED,
	393	sec_ptr sec ATTRIBUTE_UNUSED)
	394	{
	395	#ifndef HAVE_ZLIB_H
	396	bfd_set_error (bfd_error_invalid_operation);
	397	return FALSE;
	398	#else
	399	bfd_size_type uncompressed_size;
	400	bfd_byte *uncompressed_buffer;
	401	bfd_boolean ret;
	402
	403	/* Error if not opened for read. */
	404	if (abfd->direction != read_direction
	405	\|\| sec->size == 0
	406	\|\| sec->rawsize != 0
	407	\|\| sec->contents != NULL
	408	\|\| sec->compress_status != COMPRESS_SECTION_NONE)
1b315056	409	{
4a114e3e L	410	bfd_set_error (bfd_error_invalid_operation);
4a114e3e L	411	return FALSE;
1b315056	412	}
1b315056	413
4a114e3e L	414	/* Read in the full section contents and compress it. */
	415	uncompressed_size = sec->size;
	416	uncompressed_buffer = (bfd_byte *) bfd_malloc (uncompressed_size);
	417	if (!bfd_get_section_contents (abfd, sec, uncompressed_buffer,
	418	0, uncompressed_size))
	419	ret = FALSE;
	420	else
	421	ret = bfd_compress_section_contents (abfd, sec,
	422	uncompressed_buffer,
	423	uncompressed_size);
1b315056	424
1b315056	425	free (uncompressed_buffer);
4a114e3e L	426	return ret;
4a114e3e L	427	#endif
1b315056	428	}