mirror of https://github.com/vitalif/grive2
Compare commits
1 Commits
b788284020
...
a588cbfb7b
Author | SHA1 | Date |
---|---|---|
Kilian von Pflugk | a588cbfb7b |
|
@ -4,7 +4,7 @@ project(grive2)
|
|||
include(GNUInstallDirs)
|
||||
|
||||
# Grive version. remember to update it for every new release!
|
||||
set( GRIVE_VERSION "0.5.3" CACHE STRING "Grive version" )
|
||||
set( GRIVE_VERSION "0.5.2-dev" CACHE STRING "Grive version" )
|
||||
message(WARNING "Version to build: ${GRIVE_VERSION}")
|
||||
|
||||
# common compile options
|
||||
|
|
15
README.md
15
README.md
|
@ -1,6 +1,6 @@
|
|||
# Grive2 0.5.3
|
||||
# Grive2 0.5.2-dev
|
||||
|
||||
09 Nov 2022, Vitaliy Filippov
|
||||
13 Nov 2019, Vitaliy Filippov
|
||||
|
||||
http://yourcmc.ru/wiki/Grive2
|
||||
|
||||
|
@ -39,10 +39,10 @@ grive -a
|
|||
|
||||
A URL should be printed. Go to the link. You will need to login to your Google
|
||||
account if you haven't done so. After granting the permission to Grive, the
|
||||
authorization code will be forwarded to the Grive application and you will be
|
||||
redirected to a localhost web page confirming the authorization.
|
||||
browser will show you an authenication code. Copy-and-paste that to the
|
||||
standard input of Grive.
|
||||
|
||||
If everything works fine, Grive will create .grive and .grive\_state files in your
|
||||
If everything works fine, Grive will create .grive and .grive_state files in your
|
||||
current directory. It will also start downloading files from your Google Drive to
|
||||
your current directory.
|
||||
|
||||
|
@ -203,10 +203,7 @@ Alternativly you can define your own client_id and client_secret during build
|
|||
|
||||
## Version History
|
||||
|
||||
### Grive2 v0.5.3
|
||||
|
||||
- Implement Google OAuth loopback IP redirect flow
|
||||
- Various small fixes
|
||||
### Grive2 v0.5.2-dev
|
||||
|
||||
### Grive2 v0.5.1
|
||||
|
||||
|
|
|
@ -1,10 +1,3 @@
|
|||
grive2 (0.5.3) unstable; urgency=medium
|
||||
|
||||
* Implement Google OAuth loopback IP redirect flow
|
||||
* Various small fixes
|
||||
|
||||
-- Vitaliy Filippov <vitalif@yourcmc.ru> Wed, 09 Nov 2022 12:42:28 +0300
|
||||
|
||||
grive2 (0.5.2+git20210315) unstable; urgency=medium
|
||||
|
||||
* Newer dev version
|
||||
|
|
|
@ -194,15 +194,17 @@ int Main( int argc, char **argv )
|
|||
|
||||
std::cout
|
||||
<< "-----------------------\n"
|
||||
<< "Please open this URL in your browser to authenticate Grive2:\n\n"
|
||||
<< "Please go to this URL and get an authentication code:\n\n"
|
||||
<< token.MakeAuthURL()
|
||||
<< std::endl ;
|
||||
|
||||
if ( !token.GetCode() )
|
||||
{
|
||||
std::cout << "Authentication failed\n";
|
||||
return -1;
|
||||
}
|
||||
std::cout
|
||||
<< "\n-----------------------\n"
|
||||
<< "Please input the authentication code here: " << std::endl ;
|
||||
std::string code ;
|
||||
std::cin >> code ;
|
||||
|
||||
token.Auth( code ) ;
|
||||
|
||||
// save to config
|
||||
config.Set( "id", Val( id ) ) ;
|
||||
|
|
|
@ -21,7 +21,6 @@
|
|||
|
||||
#include "util/FileSystem.hh"
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <iosfwd>
|
||||
|
|
|
@ -25,13 +25,6 @@
|
|||
#include "http/Header.hh"
|
||||
#include "util/log/Log.hh"
|
||||
|
||||
#include <netinet/in.h>
|
||||
#include <sys/socket.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <poll.h>
|
||||
|
||||
// for debugging
|
||||
#include <iostream>
|
||||
|
||||
|
@ -57,29 +50,18 @@ OAuth2::OAuth2(
|
|||
const std::string& client_id,
|
||||
const std::string& client_secret ) :
|
||||
m_agent( agent ),
|
||||
m_port( 0 ),
|
||||
m_socket( -1 ),
|
||||
m_client_id( client_id ),
|
||||
m_client_secret( client_secret )
|
||||
{
|
||||
}
|
||||
|
||||
OAuth2::~OAuth2()
|
||||
{
|
||||
if ( m_socket >= 0 )
|
||||
{
|
||||
close( m_socket );
|
||||
m_socket = -1;
|
||||
}
|
||||
}
|
||||
|
||||
bool OAuth2::Auth( const std::string& auth_code )
|
||||
void OAuth2::Auth( const std::string& auth_code )
|
||||
{
|
||||
std::string post =
|
||||
"code=" + auth_code +
|
||||
"&client_id=" + m_client_id +
|
||||
"&client_secret=" + m_client_secret +
|
||||
"&redirect_uri=http%3A%2F%2Flocalhost:" + std::to_string( m_port ) + "%2Fauth" +
|
||||
"&redirect_uri=" + "urn:ietf:wg:oauth:2.0:oob" +
|
||||
"&grant_type=authorization_code" ;
|
||||
|
||||
http::ValResponse resp ;
|
||||
|
@ -95,120 +77,19 @@ bool OAuth2::Auth( const std::string& auth_code )
|
|||
{
|
||||
Log( "Failed to obtain auth token: HTTP %1%, body: %2%",
|
||||
code, m_agent->LastError(), log::error ) ;
|
||||
return false;
|
||||
BOOST_THROW_EXCEPTION( AuthFailed() );
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
std::string OAuth2::MakeAuthURL()
|
||||
{
|
||||
if ( !m_port )
|
||||
{
|
||||
sockaddr_storage addr = { 0 };
|
||||
addr.ss_family = AF_INET;
|
||||
m_socket = socket( AF_INET, SOCK_STREAM, 0 );
|
||||
if ( m_socket < 0 )
|
||||
throw std::runtime_error( std::string("socket: ") + strerror(errno) );
|
||||
if ( bind( m_socket, (sockaddr*)&addr, sizeof( addr ) ) < 0 )
|
||||
{
|
||||
close( m_socket );
|
||||
m_socket = -1;
|
||||
throw std::runtime_error( std::string("bind: ") + strerror(errno) );
|
||||
}
|
||||
socklen_t len = sizeof( addr );
|
||||
if ( getsockname( m_socket, (sockaddr *)&addr, &len ) == -1 )
|
||||
{
|
||||
close( m_socket );
|
||||
m_socket = -1;
|
||||
throw std::runtime_error( std::string("getsockname: ") + strerror(errno) );
|
||||
}
|
||||
m_port = ntohs(((sockaddr_in*)&addr)->sin_port);
|
||||
if ( listen( m_socket, 128 ) < 0 )
|
||||
{
|
||||
close( m_socket );
|
||||
m_socket = -1;
|
||||
m_port = 0;
|
||||
throw std::runtime_error( std::string("listen: ") + strerror(errno) );
|
||||
}
|
||||
}
|
||||
return "https://accounts.google.com/o/oauth2/auth"
|
||||
"?scope=" + m_agent->Escape( "https://www.googleapis.com/auth/drive" ) +
|
||||
"&redirect_uri=http%3A%2F%2Flocalhost:" + std::to_string( m_port ) + "%2Fauth" +
|
||||
"&redirect_uri=urn:ietf:wg:oauth:2.0:oob"
|
||||
"&response_type=code"
|
||||
"&client_id=" + m_client_id ;
|
||||
}
|
||||
|
||||
bool OAuth2::GetCode( )
|
||||
{
|
||||
sockaddr_storage addr = { 0 };
|
||||
int peer_fd = -1;
|
||||
while ( peer_fd < 0 )
|
||||
{
|
||||
socklen_t peer_addr_size = sizeof( addr );
|
||||
peer_fd = accept( m_socket, (sockaddr*)&addr, &peer_addr_size );
|
||||
if ( peer_fd == -1 && errno != EAGAIN && errno != EINTR )
|
||||
throw std::runtime_error( std::string("accept: ") + strerror(errno) );
|
||||
}
|
||||
fcntl( peer_fd, F_SETFL, fcntl( peer_fd, F_GETFL, 0 ) | O_NONBLOCK );
|
||||
struct pollfd pfd = (struct pollfd){
|
||||
.fd = peer_fd,
|
||||
.events = POLLIN|POLLRDHUP,
|
||||
};
|
||||
char buf[4096];
|
||||
std::string request;
|
||||
while ( true )
|
||||
{
|
||||
pfd.revents = 0;
|
||||
poll( &pfd, 1, -1 );
|
||||
if ( pfd.revents & POLLRDHUP )
|
||||
break;
|
||||
int r = 1;
|
||||
while ( r > 0 )
|
||||
{
|
||||
r = read( peer_fd, buf, sizeof( buf ) );
|
||||
if ( r > 0 )
|
||||
request += std::string( buf, r );
|
||||
else if ( r == 0 )
|
||||
break;
|
||||
else if ( errno != EAGAIN && errno != EINTR )
|
||||
throw std::runtime_error( std::string("read: ") + strerror(errno) );
|
||||
}
|
||||
if ( r == 0 || ( r < 0 && request.find( "\n" ) > 0 ) ) // GET ... HTTP/1.1\r\n
|
||||
break;
|
||||
}
|
||||
bool ok = false;
|
||||
if ( request.substr( 0, 10 ) == "GET /auth?" )
|
||||
{
|
||||
std::string line = request;
|
||||
int p = line.find( "\n" );
|
||||
if ( p > 0 )
|
||||
line = line.substr( 0, p );
|
||||
p = line.rfind( " " );
|
||||
if ( p > 0 )
|
||||
line = line.substr( 0, p );
|
||||
p = line.find( "code=" );
|
||||
if ( p > 0 )
|
||||
line = line.substr( p+5 );
|
||||
p = line.find( "&" );
|
||||
if ( p > 0 )
|
||||
line = line.substr( 0, p );
|
||||
ok = Auth( line );
|
||||
}
|
||||
std::string response = ( ok
|
||||
? "Authenticated successfully. Please close the page"
|
||||
: "Authentication error. Please try again" );
|
||||
response = "HTTP/1.1 200 OK\r\n"
|
||||
"Content-Type: text/html; charset=utf-8\r\n"
|
||||
"Connection: close\r\n"
|
||||
"\r\n"+
|
||||
response+
|
||||
"\r\n";
|
||||
write( peer_fd, response.c_str(), response.size() );
|
||||
close( peer_fd );
|
||||
return ok;
|
||||
}
|
||||
|
||||
void OAuth2::Refresh( )
|
||||
{
|
||||
std::string post =
|
||||
|
|
|
@ -41,15 +41,13 @@ public :
|
|||
const std::string& refresh_code,
|
||||
const std::string& client_id,
|
||||
const std::string& client_secret ) ;
|
||||
~OAuth2( ) ;
|
||||
|
||||
std::string Str() const ;
|
||||
|
||||
std::string MakeAuthURL() ;
|
||||
|
||||
bool Auth( const std::string& auth_code ) ;
|
||||
void Auth( const std::string& auth_code ) ;
|
||||
void Refresh( ) ;
|
||||
bool GetCode( ) ;
|
||||
|
||||
std::string RefreshToken( ) const ;
|
||||
std::string AccessToken( ) const ;
|
||||
|
@ -61,8 +59,6 @@ private :
|
|||
std::string m_access ;
|
||||
std::string m_refresh ;
|
||||
http::Agent* m_agent ;
|
||||
int m_port ;
|
||||
int m_socket ;
|
||||
|
||||
const std::string m_client_id ;
|
||||
const std::string m_client_secret ;
|
||||
|
|
|
@ -84,7 +84,7 @@ void Config::Save( )
|
|||
|
||||
void Config::Set( const std::string& key, const Val& value )
|
||||
{
|
||||
m_file.Set( key, value ) ;
|
||||
m_file.Add( key, value ) ;
|
||||
}
|
||||
|
||||
Val Config::Get( const std::string& key ) const
|
||||
|
|
Loading…
Reference in New Issue