Merge branch 'master' into interface-checks

This commit is contained in:
Gabe Rosenhouse 2016-12-15 18:20:34 -08:00 committed by GitHub
commit 1c1cf2faf7
2 changed files with 51 additions and 2 deletions

View File

@ -20,6 +20,7 @@ import (
"net" "net"
"os" "os"
"path/filepath" "path/filepath"
"strings"
"github.com/containernetworking/cni/plugins/ipam/host-local/backend" "github.com/containernetworking/cni/plugins/ipam/host-local/backend"
) )
@ -61,7 +62,7 @@ func (s *Store) Reserve(id string, ip net.IP) (bool, error) {
if err != nil { if err != nil {
return false, err return false, err
} }
if _, err := f.WriteString(id); err != nil { if _, err := f.WriteString(strings.TrimSpace(id)); err != nil {
f.Close() f.Close()
os.Remove(f.Name()) os.Remove(f.Name())
return false, err return false, err
@ -104,7 +105,7 @@ func (s *Store) ReleaseByID(id string) error {
if err != nil { if err != nil {
return nil return nil
} }
if string(data) == id { if strings.TrimSpace(string(data)) == strings.TrimSpace(id) {
if err := os.Remove(path); err != nil { if err := os.Remove(path); err != nil {
return nil return nil
} }

View File

@ -89,4 +89,52 @@ var _ = Describe("host-local Operations", func() {
_, err = os.Stat(ipFilePath) _, err = os.Stat(ipFilePath)
Expect(err).To(HaveOccurred()) Expect(err).To(HaveOccurred())
}) })
It("ignores whitespace in disk files", func() {
const ifname string = "eth0"
const nspath string = "/some/where"
tmpDir, err := ioutil.TempDir("", "host_local_artifacts")
Expect(err).NotTo(HaveOccurred())
defer os.RemoveAll(tmpDir)
conf := fmt.Sprintf(`{
"cniVersion": "0.2.0",
"name": "mynet",
"type": "ipvlan",
"master": "foo0",
"ipam": {
"type": "host-local",
"subnet": "10.1.2.0/24",
"dataDir": "%s"
}
}`, tmpDir)
args := &skel.CmdArgs{
ContainerID: " dummy\n ",
Netns: nspath,
IfName: ifname,
StdinData: []byte(conf),
}
// Allocate the IP
result, err := testutils.CmdAddWithResult(nspath, ifname, func() error {
return cmdAdd(args)
})
Expect(err).NotTo(HaveOccurred())
ipFilePath := filepath.Join(tmpDir, "mynet", result.IP4.IP.IP.String())
contents, err := ioutil.ReadFile(ipFilePath)
Expect(err).NotTo(HaveOccurred())
Expect(string(contents)).To(Equal("dummy"))
// Release the IP
err = testutils.CmdDelWithResult(nspath, ifname, func() error {
return cmdDel(args)
})
Expect(err).NotTo(HaveOccurred())
_, err = os.Stat(ipFilePath)
Expect(err).To(HaveOccurred())
})
}) })