Skip to content
Fetching contributors…
Cannot retrieve contributors at this time
153 lines (123 sloc) 3.29 KB
/*
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/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,
Receivable *dest,
const Header& hdr )
{
long response ;
while ( CheckRetry(
response = m_agent->Put(url, data, dest, AppendHeader(hdr)) ) ) ;
return response ;
}
long AuthAgent::Put(
const std::string& url,
StdioFile& file,
Receivable *dest,
const Header& hdr )
{
long response ;
while ( CheckRetry(
response = m_agent->Put( url, file, dest, AppendHeader(hdr) ) ) ) ;
return response ;
}
long AuthAgent::Get(
const std::string& url,
Receivable *dest,
const Header& hdr )
{
long response ;
while ( CheckRetry(
response = m_agent->Get( url, dest, AppendHeader(hdr) ) ) ) ;
return response ;
}
long AuthAgent::Post(
const std::string& url,
const std::string& data,
Receivable *dest,
const Header& hdr )
{
long response ;
while ( CheckRetry(
response = m_agent->Post( url, data, dest, AppendHeader(hdr) ) ) ) ;
return response ;
}
long AuthAgent::Custom(
const std::string& method,
const std::string& url,
Receivable *dest,
const Header& hdr )
{
long response ;
while ( CheckRetry(
response = m_agent->Custom( method, url, dest, AppendHeader(hdr) ) ) ) ;
return response ;
}
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 ) ;
m_auth.Refresh() ;
return true ;
}
else
return false ;
}
} // end of namespace
Jump to Line
Something went wrong with that request. Please try again.