package bootstrap import ( "encoding/json" "io" "net/http" "os" "path/filepath" "time" "github.com/k3s-io/k3s/pkg/daemons/config" "github.com/pkg/errors" "github.com/sirupsen/logrus" ) func Handler(bootstrap *config.ControlRuntimeBootstrap) http.Handler { return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { rw.Header().Set("Content-Type", "application/json") ReadFromDisk(rw, bootstrap) }) } // ReadFromDisk reads the bootstrap data from the files on disk and // writes their content in JSON form to the given io.Writer. func ReadFromDisk(w io.Writer, bootstrap *config.ControlRuntimeBootstrap) error { paths, err := ObjToMap(bootstrap) if err != nil { return nil } dataMap := make(map[string]File) for pathKey, path := range paths { if path == "" { continue } info, err := os.Stat(path) if err != nil { logrus.Warnf("failed to stat %s: %v", pathKey, err) continue } data, err := os.ReadFile(path) if err != nil { return err } dataMap[pathKey] = File{ Timestamp: info.ModTime(), Content: data, } } return json.NewEncoder(w).Encode(dataMap) } // File is a representation of a certificate // or key file within the bootstrap context that contains // the contents of the file as well as a timestamp from // when the file was last modified. type File struct { Timestamp time.Time Content []byte } type PathsDataformat map[string]File // WriteToDiskFromStorage writes the contents of the given reader to the paths // derived from within the ControlRuntimeBootstrap. func WriteToDiskFromStorage(files PathsDataformat, bootstrap *config.ControlRuntimeBootstrap) error { paths, err := ObjToMap(bootstrap) if err != nil { return err } for pathKey, bsf := range files { path, ok := paths[pathKey] if !ok || path == "" { continue } if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil { return errors.Wrapf(err, "failed to mkdir %s", filepath.Dir(path)) } if err := os.WriteFile(path, bsf.Content, 0600); err != nil { return errors.Wrapf(err, "failed to write to %s", path) } if err := os.Chtimes(path, bsf.Timestamp, bsf.Timestamp); err != nil { return errors.Wrapf(err, "failed to update modified time on %s", path) } } return nil } func ObjToMap(obj interface{}) (map[string]string, error) { bytes, err := json.Marshal(obj) if err != nil { return nil, err } data := map[string]string{} return data, json.Unmarshal(bytes, &data) }