mirror of https://github.com/vitalif/grive2
182 lines
4.0 KiB
C++
182 lines
4.0 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 "AuthAgent.hh"
|
|
|
|
#include "http/Error.hh"
|
|
#include "http/Header.hh"
|
|
#include "util/log/Log.hh"
|
|
#include "util/OS.hh"
|
|
|
|
#include <cassert>
|
|
|
|
namespace gr {
|
|
|
|
using namespace http ;
|
|
|
|
AuthAgent::AuthAgent( const OAuth2& auth, std::auto_ptr<Agent> real_agent ) :
|
|
m_auth ( auth ),
|
|
m_agent ( real_agent )
|
|
{
|
|
assert( m_agent.get() != 0 ) ;
|
|
}
|
|
|
|
Header AuthAgent::AppendHeader( const Header& hdr ) const
|
|
{
|
|
Header h(hdr) ;
|
|
h.Add( "Authorization: Bearer " + m_auth.AccessToken() ) ;
|
|
h.Add( "GData-Version: 3.0" ) ;
|
|
return h ;
|
|
}
|
|
|
|
long AuthAgent::Put(
|
|
const std::string& url,
|
|
const std::string& data,
|
|
DataStream *dest,
|
|
const Header& hdr )
|
|
{
|
|
Header auth = AppendHeader(hdr) ;
|
|
|
|
long response ;
|
|
while ( CheckRetry(
|
|
response = m_agent->Put(url, data, dest, auth) ) ) ;
|
|
|
|
return CheckHttpResponse(response, url, auth) ;
|
|
}
|
|
|
|
long AuthAgent::Put(
|
|
const std::string& url,
|
|
File *file,
|
|
DataStream *dest,
|
|
const Header& hdr )
|
|
{
|
|
Header auth = AppendHeader(hdr) ;
|
|
|
|
long response ;
|
|
while ( CheckRetry(
|
|
response = m_agent->Put( url, file, dest, AppendHeader(hdr) ) ) ) ;
|
|
|
|
return CheckHttpResponse(response, url, auth) ;
|
|
}
|
|
|
|
long AuthAgent::Get(
|
|
const std::string& url,
|
|
DataStream *dest,
|
|
const Header& hdr )
|
|
{
|
|
Header auth = AppendHeader(hdr) ;
|
|
|
|
long response ;
|
|
while ( CheckRetry(
|
|
response = m_agent->Get( url, dest, AppendHeader(hdr) ) ) ) ;
|
|
|
|
return CheckHttpResponse(response, url, auth) ;
|
|
}
|
|
|
|
long AuthAgent::Post(
|
|
const std::string& url,
|
|
const std::string& data,
|
|
DataStream *dest,
|
|
const Header& hdr )
|
|
{
|
|
Header auth = AppendHeader(hdr) ;
|
|
|
|
long response ;
|
|
while ( CheckRetry(
|
|
response = m_agent->Post( url, data, dest, AppendHeader(hdr) ) ) ) ;
|
|
|
|
return CheckHttpResponse(response, url, auth) ;
|
|
}
|
|
|
|
long AuthAgent::Custom(
|
|
const std::string& method,
|
|
const std::string& url,
|
|
DataStream *dest,
|
|
const Header& hdr )
|
|
{
|
|
Header auth = AppendHeader(hdr) ;
|
|
|
|
long response ;
|
|
while ( CheckRetry(
|
|
response = m_agent->Custom( method, url, dest, AppendHeader(hdr) ) ) ) ;
|
|
|
|
return CheckHttpResponse(response, url, auth) ;
|
|
}
|
|
|
|
std::string AuthAgent::RedirLocation() const
|
|
{
|
|
return m_agent->RedirLocation() ;
|
|
}
|
|
|
|
std::string AuthAgent::Escape( const std::string& str )
|
|
{
|
|
return m_agent->Escape( str ) ;
|
|
}
|
|
|
|
std::string AuthAgent::Unescape( const std::string& str )
|
|
{
|
|
return m_agent->Unescape( str ) ;
|
|
}
|
|
|
|
bool AuthAgent::CheckRetry( long response )
|
|
{
|
|
// HTTP 500 and 503 should be temperory. just wait a bit and retry
|
|
if ( response == 500 || response == 503 )
|
|
{
|
|
Log( "resquest failed due to temperory error: %1%. retrying in 5 seconds",
|
|
response, log::warning ) ;
|
|
|
|
os::Sleep( 5 ) ;
|
|
return true ;
|
|
}
|
|
|
|
// HTTP 401 Unauthorized. the auth token has been expired. refresh it
|
|
else if ( response == 401 )
|
|
{
|
|
Log( "resquest failed due to auth token expired: %1%. refreshing token",
|
|
response, log::warning ) ;
|
|
|
|
os::Sleep( 5 ) ;
|
|
m_auth.Refresh() ;
|
|
return true ;
|
|
}
|
|
else
|
|
return false ;
|
|
}
|
|
|
|
long AuthAgent::CheckHttpResponse(
|
|
long response,
|
|
const std::string& url,
|
|
const http::Header& hdr )
|
|
{
|
|
// throw for other HTTP errors
|
|
if ( response >= 400 && response < 500 )
|
|
{
|
|
BOOST_THROW_EXCEPTION(
|
|
Error()
|
|
<< HttpResponse( response )
|
|
<< Url( url )
|
|
<< HttpHeader( hdr ) ) ;
|
|
}
|
|
|
|
return response ;
|
|
}
|
|
|
|
} // end of namespace
|