mirror of
https://github.com/kemko/nomad.git
synced 2026-01-05 09:55:44 +03:00
716df52788
In #24650 we switched to using ephemeral state for CNI plugins, so that when a host reboots and we lose all the allocations we don't end up trying to use IPs we created in network namespaces we just destroyed. Unfortunately upgrade testing missed that in a non-reboot scenario, the existing CNI state was being used by plugins like the ipam plugin to hand out the "next available" IP address. So with no state carried over, we might allocate new addresses that conflict with existing allocations. (This can be avoided by draining the node first.) As a compatibility shim, copy the old CNI state directory to the new CNI state directory during agent startup, if the new CNI state directory doesn't already exist. Ref: https://github.com/hashicorp/nomad/pull/24650
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package escapingfs
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// CopyDir copies a directory's contents to a new location, returning an error
|
|
// on symlinks. This implementation is roughly the same as the stdlib os.CopyDir
|
|
// but with th e important difference that we preserve file modes.
|
|
func CopyDir(src, dst string) error {
|
|
srcFs := os.DirFS(src)
|
|
|
|
return fs.WalkDir(srcFs, ".", func(oldPath string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
newPath := filepath.Join(dst, oldPath)
|
|
if d.IsDir() {
|
|
info, err := d.Info()
|
|
if err != nil {
|
|
return fmt.Errorf("could not stat directory: %v", err)
|
|
}
|
|
return os.MkdirAll(newPath, info.Mode())
|
|
}
|
|
if !d.Type().IsRegular() {
|
|
return fmt.Errorf("copying cannot traverse symlinks")
|
|
}
|
|
|
|
r, err := srcFs.Open(oldPath)
|
|
if err != nil {
|
|
return fmt.Errorf("could not open existing file: %v", err)
|
|
}
|
|
defer r.Close()
|
|
info, err := r.Stat()
|
|
if err != nil {
|
|
return fmt.Errorf("could not stat file: %v", err)
|
|
}
|
|
|
|
w, err := os.OpenFile(newPath, os.O_CREATE|os.O_EXCL|os.O_WRONLY, info.Mode())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := io.Copy(w, r); err != nil {
|
|
w.Close()
|
|
return fmt.Errorf("could not copy file: %v", err)
|
|
}
|
|
return w.Close()
|
|
})
|
|
}
|