grive2/libgrive/src/http/Download.cc

79 lines
1.9 KiB
C++

/*
grive: an GPL program to sync a local directory with Google Drive
Copyright (C) 2012 Wan Wai Ho
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 version 2
of the License.
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 "Download.hh"
// #include "util/SignalHandler.hh"
#include "Error.hh"
#include "util/Crypt.hh"
// boost headers
#include <boost/throw_exception.hpp>
#include <boost/exception/errinfo_api_function.hpp>
#include <boost/exception/errinfo_at_line.hpp>
#include <boost/exception/errinfo_errno.hpp>
#include <boost/exception/errinfo_file_handle.hpp>
#include <boost/exception/errinfo_file_name.hpp>
#include <boost/exception/errinfo_file_open_mode.hpp>
#include <boost/exception/info.hpp>
#include <cassert>
#include <new>
#include <signal.h>
namespace gr { namespace http {
Download::Download( const std::string& filename ) :
m_file( filename, 0600 ),
m_crypt( new crypt::MD5 )
{
}
Download::Download( const std::string& filename, NoChecksum ) :
m_file( filename, 0600 )
{
}
Download::~Download()
{
}
void Download::Clear()
{
// no need to do anything
}
std::string Download::Finish() const
{
return m_crypt.get() != 0 ? m_crypt->Get() : "" ;
}
std::size_t Download::OnData( void *data, std::size_t count )
{
assert( data != 0 ) ;
if ( m_crypt.get() != 0 )
m_crypt->Write( data, count ) ;
return m_file.Write( static_cast<char*>(data), count ) ;
}
} } // end of namespace