libs/EDocBFD/Section.cpp

Go to the documentation of this file.

/*******************************************************************************

   Copyright (C) 2007 Brendon Costa

   This code 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 2, or (at your option)
   any later version.

   This code 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 code; see the file COPYING.  If not, write to
   the Free Software Foundation, 59 Temple Place - Suite 330,
   Boston, MA 02111-1307, USA.

*******************************************************************************/
#include "config.h"
#include "EDocBFD/Section.h"

#include "EDocBFD/exceptions.h"
#include <bfd.h>


#ifdef HAVE_STRING_H
   #include <string.h>
#endif


namespace EDocBFD
{
   //===========================================================================
   Section::Section(bfd* abfd_in, asection* asect_in) :
      abfd(abfd_in),
      asect(asect_in),
      file_pos(0)
   {
      // @@@Brendon seems this is just used by get section size.
      //sectsize = bfd_get_section_size_before_reloc(asect);
      sectsize = bfd_section_size(bfd, asect);
   }
   //===========================================================================
   ssize_t Section::Write(unsigned char data)
   {
      return Write(&data, 1);
   }
   //===========================================================================
   ssize_t Section::Write(const void* data, size_t len)
   {
      if (len + file_pos > sectsize)
      {
         len = sectsize - file_pos;
      }

      if (len == 0)
      {
         EDOC_THROW_EXCEPTION(EDoc::EOFException, "End of section.", "");
      }

      if (!bfd_set_section_contents(abfd, asect, data, file_pos, len))
      {
         bfd_error_type error_code = bfd_get_error();
         EDOC_THROW_EXCEPTION(BFDException, bfd_errmsg(error_code), 
            "section name: " << asect->name 
            << ", file_pos: " << file_pos
            << ", len: " << len
            << ", sectsize: " << sectsize);
      }
      return len;
   }
   //===========================================================================
   std::string Section::Read(size_t max_size)
   {
      if (max_size == 0)
      {
         return "";
      }

      if (max_size == 0xFFFFFFFF)
      {
         max_size = sectsize;
      }

      if ((max_size + file_pos) > sectsize)
      {
         max_size = sectsize - file_pos;
      }

      char* sectbuf = new char[max_size + 1];
      memset(sectbuf, 0, max_size + 1);

      try
      {
         ssize_t result = Read(sectbuf, max_size);

         std::string ret(sectbuf, result);
         delete [] sectbuf;
         sectbuf = NULL;
         return ret;
      }
      catch(...)
      {
         delete [] sectbuf;
         sectbuf = NULL;
         throw;
      }
   }
   //===========================================================================
   ssize_t Section::Read(void* sectbuf, size_t max_size)
   {
      if (max_size == 0)
      {
         return 0;
      }

      if ((max_size + file_pos) > sectsize)
      {
         max_size = sectsize - file_pos;
      }

      if (max_size == 0)
      {
         EDOC_THROW_EXCEPTION(EDoc::EOFException, "End of section.", 
            "file_pos: " << file_pos 
            << ", sectsize: " << sectsize 
            << ", name: " << asect->name);
      }

      memset(sectbuf, 0, max_size + 1);
      if (!bfd_get_section_contents(abfd, asect, sectbuf, file_pos, max_size))
      {
         bfd_error_type error_code = bfd_get_error();
         EDOC_THROW_EXCEPTION(BFDException, bfd_errmsg(error_code), 
            "section name: " << asect->name 
            << ", file_pos: " << file_pos
            << ", max_size: " << max_size
            << ", sectsize: " << sectsize);
      }

      file_pos += max_size;
      return max_size;
   }
   //===========================================================================
   std::string Section::ReadEx(size_t max_size)
   {
      std::string ret = Read(max_size);
      if ((max_size != 0xFFFFFFFF) && (ret.size() != max_size))
      {
         EDOC_THROW_EXCEPTION(BFDException, "Insufficient data.", 
            "section name: " << asect->name 
            << ", file_pos: " << file_pos
            << ", max_size: " << max_size
            << ", sectsize: " << sectsize
            << ", data size: " << ret.size()
            << ", data: " << ret);
      }
      return ret;
   }
   //===========================================================================
   ssize_t Section::ReadEx(void* sectbuf, size_t max_size)
   {
      ssize_t ret = Read(sectbuf, max_size);
      if ((max_size != 0xFFFFFFFFUL) && (ret != (ssize_t)max_size))
      {
         EDOC_THROW_EXCEPTION(BFDException, "Insufficient data.", 
            "section name: " << asect->name 
            << ", file_pos: " << file_pos
            << ", max_size: " << max_size
            << ", sectsize: " << sectsize
            << ", data size: " << ret);
      }
      return ret;
   }
   //===========================================================================
}

---