From 22ac80e83a19d452817fb9fe368ed5acb940b604 Mon Sep 17 00:00:00 2001 From: Henning Surmeier Date: Mon, 17 Sep 2018 22:49:02 +0200 Subject: [PATCH] webdav/sharepoint: renew cookies after 12hrs --- backend/webdav/odrvcookie/renew.go | 32 ++++++++++++++++++++++++++++++ backend/webdav/webdav.go | 11 ++++++++++ 2 files changed, 43 insertions(+) create mode 100644 backend/webdav/odrvcookie/renew.go diff --git a/backend/webdav/odrvcookie/renew.go b/backend/webdav/odrvcookie/renew.go new file mode 100644 index 000000000..dc1b142fc --- /dev/null +++ b/backend/webdav/odrvcookie/renew.go @@ -0,0 +1,32 @@ +package odrvcookie + +import ( + "time" + + "github.com/ncw/rclone/lib/rest" +) + +// CookieRenew holds information for the renew +type CookieRenew struct { + srv *rest.Client + timer *time.Ticker + renewFn func() +} + +// NewRenew returns and starts a CookieRenew +func NewRenew(interval time.Duration, renewFn func()) *CookieRenew { + renew := CookieRenew{ + timer: time.NewTicker(interval), + renewFn: renewFn, + } + go renew.Renew() + return &renew +} + +// Renew calls the renewFn for every tick +func (c *CookieRenew) Renew() { + for { + <-c.timer.C // wait for tick + c.renewFn() + } +} diff --git a/backend/webdav/webdav.go b/backend/webdav/webdav.go index 187ddad23..9ba4839a3 100644 --- a/backend/webdav/webdav.go +++ b/backend/webdav/webdav.go @@ -372,6 +372,17 @@ func (f *Fs) setQuirks(vendor string) error { if err != nil { return err } + + odrvcookie.NewRenew(12*time.Hour, func() { + spCookies, err := spCk.Cookies() + if err != nil { + fs.Errorf("could not renew cookies: %s", err.Error()) + return + } + f.srv.SetCookie(&spCookies.FedAuth, &spCookies.RtFa) + fs.Debugf(spCookies, "successfully renewed sharepoint cookies") + }) + f.srv.SetCookie(&spCookies.FedAuth, &spCookies.RtFa) // sharepoint, unlike the other vendors, only lists files if the depth header is set to 0