mirror of
https://github.com/rclone/rclone.git
synced 2024-11-29 03:48:27 +08:00
oauthutil: read a fresh token config file before using the refresh token.
This means that rclone will pick up tokens from concurrently running rclones. This helps for Box which only allows each refresh token to be used once. Without this fix, rclone caches the refresh token at the start of the run, then when the token expires the refresh token may have been used already by a concurrently running rclone. This also will retry the oauth up to 5 times at 1 second intervals. See: https://forum.rclone.org/t/box-token-refresh-timing/8175
This commit is contained in:
parent
b8280521a5
commit
2d01a65e36
|
@ -575,6 +575,17 @@ func SetValueAndSave(name, key, value string) (err error) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// FileGetFresh reads the config key under section return the value or
|
||||||
|
// an error if the config file was not found or that value couldn't be
|
||||||
|
// read.
|
||||||
|
func FileGetFresh(section, key string) (value string, err error) {
|
||||||
|
reloadedConfigFile, err := loadConfigFile()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return reloadedConfigFile.GetValue(section, key)
|
||||||
|
}
|
||||||
|
|
||||||
// ShowRemotes shows an overview of the config file
|
// ShowRemotes shows an overview of the config file
|
||||||
func ShowRemotes() {
|
func ShowRemotes() {
|
||||||
remotes := getConfigData().GetSectionList()
|
remotes := getConfigData().GetSectionList()
|
||||||
|
|
|
@ -153,6 +153,30 @@ type TokenSource struct {
|
||||||
expiryTimer *time.Timer // signals whenever the token expires
|
expiryTimer *time.Timer // signals whenever the token expires
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// If token has expired then first try re-reading it from the config
|
||||||
|
// file in case a concurrently runnng rclone has updated it already
|
||||||
|
func (ts *TokenSource) reReadToken() bool {
|
||||||
|
tokenString, err := config.FileGetFresh(ts.name, config.ConfigToken)
|
||||||
|
if err != nil {
|
||||||
|
fs.Debugf(ts.name, "Failed to read token out of config file: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
newToken := new(oauth2.Token)
|
||||||
|
err = json.Unmarshal([]byte(tokenString), newToken)
|
||||||
|
if err != nil {
|
||||||
|
fs.Debugf(ts.name, "Failed to parse token out of config file: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if !newToken.Valid() {
|
||||||
|
fs.Debugf(ts.name, "Loaded invalid token from config file - ignoring")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
fs.Debugf(ts.name, "Loaded fresh token from config file")
|
||||||
|
ts.token = newToken
|
||||||
|
ts.tokenSource = nil // invalidate since we changed the token
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
// Token returns a token or an error.
|
// Token returns a token or an error.
|
||||||
// Token must be safe for concurrent use by multiple goroutines.
|
// Token must be safe for concurrent use by multiple goroutines.
|
||||||
// The returned Token must not be modified.
|
// The returned Token must not be modified.
|
||||||
|
@ -161,17 +185,39 @@ type TokenSource struct {
|
||||||
func (ts *TokenSource) Token() (*oauth2.Token, error) {
|
func (ts *TokenSource) Token() (*oauth2.Token, error) {
|
||||||
ts.mu.Lock()
|
ts.mu.Lock()
|
||||||
defer ts.mu.Unlock()
|
defer ts.mu.Unlock()
|
||||||
|
var (
|
||||||
|
token *oauth2.Token
|
||||||
|
err error
|
||||||
|
changed = false
|
||||||
|
)
|
||||||
|
const maxTries = 5
|
||||||
|
|
||||||
// Make a new token source if required
|
// Try getting the token a few times
|
||||||
if ts.tokenSource == nil {
|
for i := 1; i <= maxTries; i++ {
|
||||||
ts.tokenSource = ts.config.TokenSource(ts.ctx, ts.token)
|
// Try reading the token from the config file in case it has
|
||||||
|
// been updated by a concurrent rclone process
|
||||||
|
if !ts.token.Valid() {
|
||||||
|
if ts.reReadToken() {
|
||||||
|
changed = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make a new token source if required
|
||||||
|
if ts.tokenSource == nil {
|
||||||
|
ts.tokenSource = ts.config.TokenSource(ts.ctx, ts.token)
|
||||||
|
}
|
||||||
|
|
||||||
|
token, err = ts.tokenSource.Token()
|
||||||
|
if err == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
fs.Debugf(ts.name, "Token refresh failed try %d/%d: %v", i, maxTries, err)
|
||||||
|
time.Sleep(1 * time.Second)
|
||||||
}
|
}
|
||||||
|
|
||||||
token, err := ts.tokenSource.Token()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
changed := *token != *ts.token
|
changed = changed || (*token != *ts.token)
|
||||||
ts.token = token
|
ts.token = token
|
||||||
if changed {
|
if changed {
|
||||||
// Bump on the expiry timer if it is set
|
// Bump on the expiry timer if it is set
|
||||||
|
|
Loading…
Reference in New Issue
Block a user