X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/1b7ed029c3d9d50b275573b65e8fbf4943e76bcb..e4fca76d5db9bd844530454894d07ddc729b4a9a:/services/keep-web/handler.go diff --git a/services/keep-web/handler.go b/services/keep-web/handler.go index 5ab4f70cde..912398fa64 100644 --- a/services/keep-web/handler.go +++ b/services/keep-web/handler.go @@ -10,10 +10,10 @@ import ( "html" "html/template" "io" - "log" "net/http" "net/url" "os" + "path/filepath" "sort" "strconv" "strings" @@ -25,11 +25,13 @@ import ( "git.curoverse.com/arvados.git/sdk/go/health" "git.curoverse.com/arvados.git/sdk/go/httpserver" "git.curoverse.com/arvados.git/sdk/go/keepclient" + log "github.com/Sirupsen/logrus" "golang.org/x/net/webdav" ) type handler struct { Config *Config + MetricsAPI http.Handler clientPool *arvadosclient.ClientPool setupOnce sync.Once healthHandler http.Handler @@ -89,14 +91,7 @@ func (h *handler) setup() { } func (h *handler) serveStatus(w http.ResponseWriter, r *http.Request) { - status := struct { - cacheStats - Version string - }{ - cacheStats: h.Config.Cache.Stats(), - Version: version, - } - json.NewEncoder(w).Encode(status) + json.NewEncoder(w).Encode(struct{ Version string }{version}) } // updateOnSuccess wraps httpserver.ResponseWriter. If the handler @@ -112,12 +107,12 @@ type updateOnSuccess struct { } func (uos *updateOnSuccess) Write(p []byte) (int, error) { - if uos.err != nil { - return 0, uos.err - } if !uos.sentHeader { uos.WriteHeader(http.StatusOK) } + if uos.err != nil { + return 0, uos.err + } return uos.ResponseWriter.Write(p) } @@ -140,6 +135,11 @@ func (uos *updateOnSuccess) WriteHeader(code int) { } var ( + corsAllowHeadersHeader = strings.Join([]string{ + "Authorization", "Content-Type", "Range", + // WebDAV request headers: + "Depth", "Destination", "If", "Lock-Token", "Overwrite", "Timeout", + }, ", ") writeMethod = map[string]bool{ "COPY": true, "DELETE": true, @@ -163,6 +163,12 @@ var ( "HEAD": true, "POST": true, } + // top-level dirs to serve with siteFS + siteFSDir = map[string]bool{ + "": true, // root directory + "by_id": true, + "users": true, + } ) // ServeHTTP implements http.Handler. @@ -176,6 +182,9 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { if xff := r.Header.Get("X-Forwarded-For"); xff != "" { remoteAddr = xff + "," + remoteAddr } + if xfp := r.Header.Get("X-Forwarded-Proto"); xfp != "" && xfp != "http" { + r.URL.Scheme = xfp + } w := httpserver.WrapResponseWriter(wOrig) defer func() { @@ -184,13 +193,12 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { } else if w.WroteStatus() == 0 { w.WriteHeader(statusCode) } else if w.WroteStatus() != statusCode { - httpserver.Log(r.RemoteAddr, "WARNING", + log.WithField("RequestID", r.Header.Get("X-Request-Id")).Warn( fmt.Sprintf("Our status changed from %d to %d after we sent headers", w.WroteStatus(), statusCode)) } if statusText == "" { statusText = http.StatusText(statusCode) } - httpserver.Log(remoteAddr, statusCode, statusText, w.WroteBodyBytes(), r.Method, r.Host, r.URL.Path, r.URL.RawQuery) }() if strings.HasPrefix(r.URL.Path, "/_health/") && r.Method == "GET" { @@ -203,7 +211,7 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { statusCode = http.StatusMethodNotAllowed return } - w.Header().Set("Access-Control-Allow-Headers", "Authorization, Content-Type, Range") + w.Header().Set("Access-Control-Allow-Headers", corsAllowHeadersHeader) w.Header().Set("Access-Control-Allow-Methods", "COPY, DELETE, GET, MKCOL, MOVE, OPTIONS, POST, PROPFIND, PUT, RMCOL") w.Header().Set("Access-Control-Allow-Origin", "*") w.Header().Set("Access-Control-Max-Age", "86400") @@ -226,13 +234,6 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { w.Header().Set("Access-Control-Expose-Headers", "Content-Range") } - arv := h.clientPool.Get() - if arv == nil { - statusCode, statusText = http.StatusInternalServerError, "Pool failed: "+h.clientPool.Err().Error() - return - } - defer h.clientPool.Put(arv) - pathParts := strings.Split(r.URL.Path[1:], "/") var stripParts int @@ -241,6 +242,7 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { var reqTokens []string var pathToken bool var attachment bool + var useSiteFS bool credentialsOK := h.Config.TrustAllContent if r.Host != "" && r.Host == h.Config.AttachmentOnlyHost { @@ -256,6 +258,11 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { } else if r.URL.Path == "/status.json" { h.serveStatus(w, r) return + } else if strings.HasPrefix(r.URL.Path, "/metrics") { + h.MetricsAPI.ServeHTTP(w, r) + return + } else if siteFSDir[pathParts[0]] { + useSiteFS = true } else if len(pathParts) >= 1 && strings.HasPrefix(pathParts[0], "c=") { // /c=ID[/PATH...] collectionID = parseCollectionIDFromURL(pathParts[0][2:]) @@ -275,6 +282,16 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { } } + if collectionID == "" && !useSiteFS { + statusCode = http.StatusNotFound + return + } + + forceReload := false + if cc := r.Header.Get("Cache-Control"); strings.Contains(cc, "no-cache") || strings.Contains(cc, "must-revalidate") { + forceReload = true + } + formToken := r.FormValue("api_token") if formToken != "" && r.Header.Get("Origin") != "" && attachment && r.URL.Query().Get("api_token") == "" { // The client provided an explicit token in the POST @@ -301,6 +318,14 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { return } + if useSiteFS { + if tokens == nil { + tokens = auth.NewCredentialsFromHTTPRequest(r).Tokens + } + h.serveSiteFS(w, r, tokens, credentialsOK, attachment) + return + } + targetPath := pathParts[stripParts:] if tokens == nil && len(targetPath) > 0 && strings.HasPrefix(targetPath[0], "t=") { // http://ID.example/t=TOKEN/PATH... @@ -333,10 +358,12 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { stripParts++ } - forceReload := false - if cc := r.Header.Get("Cache-Control"); strings.Contains(cc, "no-cache") || strings.Contains(cc, "must-revalidate") { - forceReload = true + arv := h.clientPool.Get() + if arv == nil { + statusCode, statusText = http.StatusInternalServerError, "Pool failed: "+h.clientPool.Err().Error() + return } + defer h.clientPool.Put(arv) var collection *arvados.Collection tokenResult := make(map[string]int) @@ -397,6 +424,7 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { statusCode, statusText = http.StatusInternalServerError, err.Error() return } + kc.RequestID = r.Header.Get("X-Request-Id") var basename string if len(targetPath) > 0 { @@ -404,18 +432,13 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { } applyContentDispositionHdr(w, r, basename, attachment) - client := &arvados.Client{ + client := (&arvados.Client{ APIHost: arv.ApiServer, AuthToken: arv.ApiToken, Insecure: arv.ApiInsecure, - } + }).WithRequestID(r.Header.Get("X-Request-Id")) - var fs arvados.FileSystem - if collectionID == "" { - fs = client.SiteFileSystem(kc) - } else { - fs, err = collection.FileSystem(client, kc) - } + fs, err := collection.FileSystem(client, kc) if err != nil { statusCode, statusText = http.StatusInternalServerError, err.Error() return @@ -475,7 +498,7 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { // "dirname/fnm". h.seeOtherWithCookie(w, r, r.URL.Path+"/", credentialsOK) } else if stat.IsDir() { - h.serveDirectory(w, r, collection.Name, fs, openPath, stripParts) + h.serveDirectory(w, r, collection.Name, fs, openPath, true) } else { http.ServeContent(w, r, basename, stat.ModTime(), f) if r.Header.Get("Range") == "" && int64(w.WroteBodyBytes()) != stat.Size() { @@ -491,10 +514,78 @@ func (h *handler) ServeHTTP(wOrig http.ResponseWriter, r *http.Request) { } } +func (h *handler) serveSiteFS(w http.ResponseWriter, r *http.Request, tokens []string, credentialsOK, attachment bool) { + if len(tokens) == 0 { + w.Header().Add("WWW-Authenticate", "Basic realm=\"collections\"") + http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized) + return + } + if writeMethod[r.Method] { + http.Error(w, errReadOnly.Error(), http.StatusMethodNotAllowed) + return + } + arv := h.clientPool.Get() + if arv == nil { + http.Error(w, "Pool failed: "+h.clientPool.Err().Error(), http.StatusInternalServerError) + return + } + defer h.clientPool.Put(arv) + arv.ApiToken = tokens[0] + + kc, err := keepclient.MakeKeepClient(arv) + if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + kc.RequestID = r.Header.Get("X-Request-Id") + client := (&arvados.Client{ + APIHost: arv.ApiServer, + AuthToken: arv.ApiToken, + Insecure: arv.ApiInsecure, + }).WithRequestID(r.Header.Get("X-Request-Id")) + fs := client.SiteFileSystem(kc) + f, err := fs.Open(r.URL.Path) + if os.IsNotExist(err) { + http.Error(w, err.Error(), http.StatusNotFound) + return + } else if err != nil { + http.Error(w, err.Error(), http.StatusInternalServerError) + return + } + defer f.Close() + if fi, err := f.Stat(); err == nil && fi.IsDir() && r.Method == "GET" { + if !strings.HasSuffix(r.URL.Path, "/") { + h.seeOtherWithCookie(w, r, r.URL.Path+"/", credentialsOK) + } else { + h.serveDirectory(w, r, fi.Name(), fs, r.URL.Path, false) + } + return + } + if r.Method == "GET" { + _, basename := filepath.Split(r.URL.Path) + applyContentDispositionHdr(w, r, basename, attachment) + } + wh := webdav.Handler{ + Prefix: "/", + FileSystem: &webdavFS{ + collfs: fs, + writing: writeMethod[r.Method], + alwaysReadEOF: r.Method == "PROPFIND", + }, + LockSystem: h.webdavLS, + Logger: func(_ *http.Request, err error) { + if err != nil { + log.Printf("error from webdav handler: %q", err) + } + }, + } + wh.ServeHTTP(w, r) +} + var dirListingTemplate = ` - {{ .Collection.Name }} + {{ .CollectionName }} +

{{ .CollectionName }}

This collection of data files is being shared with you through Arvados. You can download individual files listed below. To download -the entire collection with wget, try:

+the entire directory tree with wget, try:

-
$ wget --mirror --no-parent --no-host --cut-dirs={{ .StripParts }} https://{{ .Request.Host }}{{ .Request.URL }}
+
$ wget --mirror --no-parent --no-host --cut-dirs={{ .StripParts }} https://{{ .Request.Host }}{{ .Request.URL.Path }}

File Listing

{{if .Files}} {{else}}

(No files; this collection is empty.)

@@ -550,11 +648,12 @@ the entire collection with wget, try:

` type fileListEnt struct { - Name string - Size int64 + Name string + Size int64 + IsDir bool } -func (h *handler) serveDirectory(w http.ResponseWriter, r *http.Request, collectionName string, fs http.FileSystem, base string, stripParts int) { +func (h *handler) serveDirectory(w http.ResponseWriter, r *http.Request, collectionName string, fs http.FileSystem, base string, recurse bool) { var files []fileListEnt var walk func(string) error if !strings.HasSuffix(base, "/") { @@ -574,15 +673,16 @@ func (h *handler) serveDirectory(w http.ResponseWriter, r *http.Request, collect return err } for _, ent := range ents { - if ent.IsDir() { + if recurse && ent.IsDir() { err = walk(path + ent.Name() + "/") if err != nil { return err } } else { files = append(files, fileListEnt{ - Name: path + ent.Name(), - Size: ent.Size(), + Name: path + ent.Name(), + Size: ent.Size(), + IsDir: ent.IsDir(), }) } } @@ -611,7 +711,7 @@ func (h *handler) serveDirectory(w http.ResponseWriter, r *http.Request, collect "CollectionName": collectionName, "Files": files, "Request": r, - "StripParts": stripParts, + "StripParts": strings.Count(strings.TrimRight(r.URL.Path, "/"), "/"), }) } @@ -678,6 +778,7 @@ func (h *handler) seeOtherWithCookie(w http.ResponseWriter, r *http.Request, loc u = newu } redir := (&url.URL{ + Scheme: r.URL.Scheme, Host: r.Host, Path: u.Path, RawQuery: redirQuery.Encode(),