Store fixes; Windows compile

cni/plugin.go

@@ -161,14 +161,14 @@ func (plugin *Plugin) InitializeKeyValueStore(config *common.PluginConfig) error
 		var err error
 		plugin.Store, err = store.NewJsonFileStore(platform.CNIRuntimePath + plugin.Name + ".json")
 		if err != nil {
-			log.Printf("[cni] Failed to create store, err:%v.", err)
+			log.Printf("[cni] Failed to create store: %v.", err)
 			return err
 		}
 	}
 
 	// Acquire store lock.
 	if err := plugin.Store.Lock(true); err != nil {
-		log.Printf("[cni] Timed out on locking store, err:%v.", err)
+		log.Printf("[cni] Failed to lock store: %v.", err)
 		return err
 	}
 
@@ -182,7 +182,7 @@ func (plugin *Plugin) UninitializeKeyValueStore() error {
 	if plugin.Store != nil {
 		err := plugin.Store.Unlock()
 		if err != nil {
-			log.Printf("[cni] Failed to unlock store, err:%v.", err)
+			log.Printf("[cni] Failed to unlock store: %v.", err)
 			return err
 		}
 	}

netlink/ip.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

netlink/link.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

netlink/netlink.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

netlink/netlink_test.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

netlink/netlink_windows.go

@@ -0,0 +1,4 @@
+// Copyright 2017 Microsoft. All rights reserved.
+// MIT License
+
+package netlink

netlink/protocol.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

netlink/socket.go

@@ -1,6 +1,8 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
+// +build linux
+
 package netlink
 
 import (

network/endpoint_linux.go

@@ -1,8 +1,6 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
-// +build linux
-
 package network
 
 import (

network/endpoint_windows.go

@@ -1,8 +1,6 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
-// +build windows
-
 package network
 
 import (

network/epcommon/endpoint_common_windows.go

@@ -0,0 +1 @@
+package epcommon

network/namespace.go → network/namespace_linux.go

@@ -1,8 +1,6 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
-// +build linux
-
 package network
 
 import (

network/network_linux.go

@@ -1,8 +1,6 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
-// +build linux
-
 package network
 
 import (

network/network_windows.go

@@ -1,8 +1,6 @@
 // Copyright 2017 Microsoft. All rights reserved.
 // MIT License
 
-// +build windows
-
 package network
 
 import (

store/json.go

@@ -21,10 +21,10 @@ const (
 	lockExtension = ".lock"
 
 	// Maximum number of retries before failing a lock call.
-	lockMaxRetries = 20
+	lockMaxRetries = 200
 
 	// Delay between lock retries.
-	lockRetryDelayInMilliseconds = 100
+	lockRetryDelay = 100 * time.Millisecond
 )
 
 // jsonFileStore is an implementation of KeyValueStore using a local JSON file.
@@ -61,27 +61,22 @@ func (kvs *jsonFileStore) Read(key string, value interface{}) error {
 		file, err := os.Open(kvs.fileName)
 		if err != nil {
 			if os.IsNotExist(err) {
-				return nil
+				return ErrKeyNotFound
 			}
 			return err
 		}
+		defer file.Close()
 
 		// Decode to raw JSON messages.
-		err = json.NewDecoder(file).Decode(&kvs.data)
-		if err != nil {
-			return err
-		}
-
-		err = file.Close()
-		if err != nil {
+		if err := json.NewDecoder(file).Decode(&kvs.data); err != nil {
 			return err
 		}
 
 		kvs.inSync = true
 	}
 
-	raw := kvs.data[key]
-	if raw == nil {
+	raw, ok := kvs.data[key]
+	if !ok {
 		return ErrKeyNotFound
 	}
 
@@ -118,18 +113,17 @@ func (kvs *jsonFileStore) flush() error {
 	if err != nil {
 		return err
 	}
+	defer file.Close()
 
 	buf, err := json.MarshalIndent(&kvs.data, "", "\t")
 	if err != nil {
 		return err
 	}
 
-	_, err = file.Write(buf)
-	if err != nil {
+	if _, err := file.Write(buf); err != nil {
 		return err
 	}
-
-	return file.Close()
+	return nil
 }
 
 // Lock locks the store for exclusive access.
@@ -153,21 +147,20 @@ func (kvs *jsonFileStore) Lock(block bool) error {
 			break
 		}
 
-		if !block || i == lockMaxRetries {
-			return ErrStoreLocked
+		if !block {
+			return ErrNonBlockingLockIsAlreadyLocked
 		}
 
-		time.Sleep(time.Duration(lockRetryDelayInMilliseconds) * time.Millisecond)
-	}
+		if i == lockMaxRetries {
+			return ErrTimeoutLockingStore
+		}
 
-	// Write the process ID for easy identification.
-	_, err = lockFile.WriteString(strconv.Itoa(os.Getpid()))
-	if err != nil {
-		return err
+		time.Sleep(lockRetryDelay)
 	}
+	defer lockFile.Close()
 
-	err = lockFile.Close()
-	if err != nil {
+	// Write the process ID for easy identification.
+	if _, err = lockFile.WriteString(strconv.Itoa(os.Getpid())); err != nil {
 		return err
 	}
 
@@ -198,9 +191,12 @@ func (kvs *jsonFileStore) Unlock() error {
 
 // GetModificationTime returns the modification time of the persistent store.
 func (kvs *jsonFileStore) GetModificationTime() (time.Time, error) {
+	kvs.Mutex.Lock()
+	defer kvs.Mutex.Unlock()
+
 	info, err := os.Stat(kvs.fileName)
 	if err != nil {
-		log.Printf("os.stat() for file %v failed with error %v", kvs.fileName, err)
+		log.Printf("os.stat() for file %v failed: %v", kvs.fileName, err)
 		return time.Time{}.UTC(), err
 	}
 

store/store.go

@@ -20,7 +20,9 @@ type KeyValueStore interface {
 
 var (
 	// Errors returned by KeyValueStore methods.
-	ErrKeyNotFound    = fmt.Errorf("Key not found")
-	ErrStoreLocked    = fmt.Errorf("Store is locked")
-	ErrStoreNotLocked = fmt.Errorf("Store is not locked")
+	ErrKeyNotFound                    = fmt.Errorf("key not found")
+	ErrStoreLocked                    = fmt.Errorf("store is already locked")
+	ErrStoreNotLocked                 = fmt.Errorf("store is not locked")
+	ErrTimeoutLockingStore            = fmt.Errorf("timed out locking store")
+	ErrNonBlockingLockIsAlreadyLocked = fmt.Errorf("attempted to perform non-blocking lock on an already locked store")
 )