libs/EDocBFD/BFDArchive.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/BFDArchive.h"

#include "EDocBFD/bfd_utils.h"
#include "EDocBFD/exceptions.h"
#include "EDoc/utils.h"
#include "EDoc/persistence_data.h"

#ifdef HAVE_FCNTL_H
   #include <fcntl.h>
#endif

#ifdef HAVE_UNISTD_H
   #include <unistd.h>
#endif

#ifdef HAVE_ERRNO_H
   #include <errno.h>
#endif

namespace EDocBFD
{

   //===========================================================================
   ExtractedBFDFile::ExtractedBFDFile(std::string temp_file_name_in) :
      sync(false),
      temp_file_name(temp_file_name_in)
   {
      EDOC_Fine("Create new BFD Read tempfile: " << temp_file_name);
      EDoc::Persistence::AddTempFile(temp_file_name);
   }
   //===========================================================================
   ExtractedBFDFile::ExtractedBFDFile(std::string bfd_file_name_in, 
         std::string temp_file_name_in,
         std::string embedded_file_name_in) :
      
      sync(true),
      bfd_file_name(bfd_file_name_in),
      temp_file_name(temp_file_name_in),
      embedded_file_name(embedded_file_name_in)
   {
      EDOC_Fine("Create new BFD Write tempfile: " << temp_file_name);
      EDoc::Persistence::AddTempFile(temp_file_name);
   }
   //===========================================================================
   ExtractedBFDFile::~ExtractedBFDFile()
   {
      if (sync)
      {
         EDOC_Fine("Syncing write tempfile: " << temp_file_name);
         EmbedFileInBinary(temp_file_name, 
            embedded_file_name,
            bfd_file_name, 
            EDOC_SECTION_NAME);
      }
      
      EDOC_Fine("Erasing tempfile: " << temp_file_name);
      unlink(temp_file_name.c_str());
      EDoc::Persistence::RemoveTempFile(temp_file_name);
   }

   //===========================================================================
   //===========================================================================
   BFDArchive::BFDArchive()
   {
      InitialiseBFD();
   }
   //===========================================================================
   bool BFDArchive::HandlesType(const EDoc::ManagedFile& filename)
   {
      try
      {
         struct bfd* abfd = OpenBFDFile(filename.GetWorkFileName());
         bfd_close(abfd);
         return abfd;
      }
      catch (BFDException& e)
      {
         return false;
      }
   }
   //===========================================================================
   std::list<EDoc::ManagedFile> BFDArchive::ReadExtractAll(
      const EDoc::ManagedFile& filename)
   {
      std::list<EDoc::ManagedFile> files;
      struct bfd* abfd = OpenBFDFile(filename.GetWorkFileName());
      EDOC_Debug("Opened BFD file: " << filename.GetFileNameRep());
      try
      {
         ProcessReadBFD(files, abfd, filename);
         bfd_close(abfd);
      }
      catch(...)
      {
         bfd_close(abfd);
         throw;
      }
      
      return files;
   }
   //===========================================================================
   EDoc::ManagedFile BFDArchive::WriteOpen(const EDoc::ManagedFile& arch, 
      std::string item_filename)
   {
      std::string outfile = EDoc::CreateTempFilename();
      int fd = open(outfile.c_str(), O_WRONLY, O_TRUNC);
      if (fd == -1)
      {
         EDOC_THROW_EXCEPTION(BFDException, "", "");
      }
      close(fd);
   
      std::list<std::string> id = arch.GetIdentifier();
      id.push_back(item_filename);
      
      return EDoc::ManagedFile(id, outfile, new ExtractedBFDFile(
         arch.GetWorkFileName(), outfile, item_filename));
   }
   //===========================================================================
}

---