| // Copyright 2019 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| package packages |
| |
| import ( |
| "context" |
| "crypto/sha256" |
| "encoding/hex" |
| "encoding/json" |
| "fmt" |
| "log" |
| "net" |
| "net/http" |
| "os" |
| "path/filepath" |
| "strings" |
| "time" |
| |
| tuf_data "github.com/flynn/go-tuf/data" |
| |
| "go.fuchsia.dev/fuchsia/src/sys/pkg/lib/repo" |
| ) |
| |
| type Server struct { |
| Dir string |
| URL string |
| Hash string |
| server *http.Server |
| } |
| |
| func newServer(dir string, localHostname string, repoName string) (*Server, error) { |
| listener, err := net.Listen("tcp", ":0") |
| if err != nil { |
| return nil, err |
| } |
| |
| port := listener.Addr().(*net.TCPAddr).Port |
| log.Printf("Serving %s on :%d", dir, port) |
| |
| configURL, configHash, config, err := genConfig(dir, localHostname, repoName, port) |
| if err != nil { |
| listener.Close() |
| return nil, err |
| } |
| |
| mux := http.NewServeMux() |
| mux.HandleFunc("/host_target_testing/config.json", func(w http.ResponseWriter, r *http.Request) { |
| w.Header().Set("Content-Type", "application/json") |
| w.WriteHeader(200) |
| w.Write(config) |
| }) |
| mux.Handle("/", http.FileServer(http.Dir(dir))) |
| |
| server := &http.Server{ |
| Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| lw := &loggingWriter{w, 0} |
| mux.ServeHTTP(lw, r) |
| log.Printf("%s [pm serve] %d %s\n", |
| time.Now().Format("2006-01-02 15:04:05"), lw.status, r.RequestURI) |
| }), |
| } |
| |
| go func() { |
| if err := server.Serve(listener); err != http.ErrServerClosed { |
| log.Fatalf("failed to shutdown server: %s", err) |
| } |
| }() |
| |
| return &Server{ |
| Dir: dir, |
| URL: configURL, |
| Hash: configHash, |
| server: server, |
| }, err |
| } |
| |
| func (s *Server) Shutdown(ctx context.Context) { |
| s.server.Shutdown(ctx) |
| } |
| |
| type loggingWriter struct { |
| http.ResponseWriter |
| status int |
| } |
| |
| func (lw *loggingWriter) WriteHeader(status int) { |
| lw.status = status |
| lw.ResponseWriter.WriteHeader(status) |
| } |
| |
| // writeConfig writes the source config to the repository. |
| func genConfig(dir string, localHostname string, repoName string, port int) (configURL string, configHash string, config []byte, err error) { |
| type statusConfig struct { |
| Enabled bool |
| } |
| |
| type sourceConfig struct { |
| ID string `json:"id"` |
| RepoURL string `json:"repoUrl"` |
| BlobRepoURL string `json:"blobRepoUrl"` |
| RootKeys []repo.KeyConfig `json:"rootKeys"` |
| StatusConfig statusConfig `json:"statusConfig"` |
| } |
| |
| f, err := os.Open(filepath.Join(dir, "root.json")) |
| if err != nil { |
| return "", "", nil, err |
| } |
| defer f.Close() |
| |
| var signed tuf_data.Signed |
| if err := json.NewDecoder(f).Decode(&signed); err != nil { |
| return "", "", nil, err |
| } |
| |
| var root tuf_data.Root |
| if err := json.Unmarshal(signed.Signed, &root); err != nil { |
| return "", "", nil, err |
| } |
| |
| rootKeys, err := repo.GetRootKeys(&root) |
| if err != nil { |
| return "", "", nil, err |
| } |
| |
| hostname := strings.ReplaceAll(localHostname, "%", "%25") |
| |
| var repoURL string |
| if strings.Contains(hostname, ":") { |
| // This is an IPv6 address, use brackets for an IPv6 literal |
| repoURL = fmt.Sprintf("http://[%s]:%d", hostname, port) |
| } else { |
| repoURL = fmt.Sprintf("http://%s:%d", hostname, port) |
| } |
| configURL = fmt.Sprintf("%s/host_target_testing/config.json", repoURL) |
| |
| config, err = json.Marshal(&sourceConfig{ |
| ID: repoName, |
| RepoURL: repoURL, |
| BlobRepoURL: fmt.Sprintf("%s/blobs", repoURL), |
| RootKeys: rootKeys, |
| StatusConfig: statusConfig{ |
| Enabled: true, |
| }, |
| }) |
| if err != nil { |
| return "", "", nil, err |
| } |
| h := sha256.Sum256(config) |
| configHash = hex.EncodeToString(h[:]) |
| |
| return configURL, configHash, config, nil |
| } |