From d06548e4f4492c181f04a5fc9bf994764278e68b Mon Sep 17 00:00:00 2001
From: Ondrej Fabry <ofabry@cisco.com>
Date: Mon, 9 Sep 2019 06:16:18 +0200
Subject: Use new default binapi socket with fallback to legacy

Change-Id: I1a6839e3bbb8d599fcbdbc2b1aff7c81f7d263c9
Signed-off-by: Ondrej Fabry <ofabry@cisco.com>
---
 adapter/socketclient/socketclient.go | 78 +++++++++++++++++++++++++++---------
 adapter/vpp_api.go                   |  2 +-
 2 files changed, 59 insertions(+), 21 deletions(-)

(limited to 'adapter')

diff --git a/adapter/socketclient/socketclient.go b/adapter/socketclient/socketclient.go
index 2144d24..1c425ba 100644
--- a/adapter/socketclient/socketclient.go
+++ b/adapter/socketclient/socketclient.go
@@ -38,22 +38,9 @@ import (
 const (
 	// DefaultSocketName is default VPP API socket file path.
 	DefaultSocketName = adapter.DefaultBinapiSocket
+	legacySocketName  = "/run/vpp-api.sock"
 )
 
-const socketMissing = `
-------------------------------------------------------------
- VPP binary API socket file %s is missing!
-
-  - is VPP running with socket for binapi enabled?
-  - is the correct socket name configured?
-
- To enable it add following section to your VPP config:
-   socksvr {
-     default
-   }
-------------------------------------------------------------
-`
-
 var (
 	// DefaultConnectTimeout is default timeout for connecting
 	DefaultConnectTimeout = time.Second * 3
@@ -85,6 +72,27 @@ func init() {
 	}
 }
 
+const socketMissing = `
+------------------------------------------------------------
+ No socket file found at: %s
+ VPP binary API socket file is missing!
+
+  - is VPP running with socket for binapi enabled?
+  - is the correct socket name configured?
+
+ To enable it add following section to your VPP config:
+   socksvr {
+     default
+   }
+------------------------------------------------------------
+`
+
+var warnOnce sync.Once
+
+func (c *vppClient) printMissingSocketMsg() {
+	fmt.Fprintf(os.Stderr, socketMissing, c.sockAddr)
+}
+
 type vppClient struct {
 	sockAddr string
 
@@ -129,6 +137,32 @@ func (c *vppClient) SetDisconnectTimeout(t time.Duration) {
 	c.disconnectTimeout = t
 }
 
+func (c *vppClient) SetMsgCallback(cb adapter.MsgCallback) {
+	Log.Debug("SetMsgCallback")
+	c.cb = cb
+}
+
+func (c *vppClient) checkLegacySocket() bool {
+	if c.sockAddr == legacySocketName {
+		return false
+	}
+	Log.Debugf("checking legacy socket: %s", legacySocketName)
+	// check if socket exists
+	if _, err := os.Stat(c.sockAddr); err == nil {
+		return false // socket exists
+	} else if !os.IsNotExist(err) {
+		return false // some other error occurred
+	}
+	// check if legacy socket exists
+	if _, err := os.Stat(legacySocketName); err == nil {
+		// legacy socket exists, update sockAddr
+		c.sockAddr = legacySocketName
+		return true
+	}
+	// no socket socket found
+	return false
+}
+
 // WaitReady checks socket file existence and waits for it if necessary
 func (c *vppClient) WaitReady() error {
 	// check if socket already exists
@@ -138,6 +172,10 @@ func (c *vppClient) WaitReady() error {
 		return err // some other error occurred
 	}
 
+	if c.checkLegacySocket() {
+		return nil
+	}
+
 	// socket does not exist, watch for it
 	watcher, err := fsnotify.NewWatcher()
 	if err != nil {
@@ -158,6 +196,9 @@ func (c *vppClient) WaitReady() error {
 	for {
 		select {
 		case <-timeout.C:
+			if c.checkLegacySocket() {
+				return nil
+			}
 			return fmt.Errorf("timeout waiting (%s) for socket file: %s", MaxWaitReady, c.sockAddr)
 
 		case e := <-watcher.Errors:
@@ -173,15 +214,12 @@ func (c *vppClient) WaitReady() error {
 	}
 }
 
-func (c *vppClient) SetMsgCallback(cb adapter.MsgCallback) {
-	Log.Debug("SetMsgCallback")
-	c.cb = cb
-}
-
 func (c *vppClient) Connect() error {
+	c.checkLegacySocket()
+
 	// check if socket exists
 	if _, err := os.Stat(c.sockAddr); os.IsNotExist(err) {
-		fmt.Fprintf(os.Stderr, socketMissing, c.sockAddr)
+		warnOnce.Do(c.printMissingSocketMsg)
 		return fmt.Errorf("VPP API socket file %s does not exist", c.sockAddr)
 	} else if err != nil {
 		return fmt.Errorf("VPP API socket error: %v", err)
diff --git a/adapter/vpp_api.go b/adapter/vpp_api.go
index 2391420..71bf7c2 100644
--- a/adapter/vpp_api.go
+++ b/adapter/vpp_api.go
@@ -20,7 +20,7 @@ import (
 
 const (
 	// DefaultBinapiSocket defines a default socket file path for VPP binary API.
-	DefaultBinapiSocket = "/run/vpp-api.sock"
+	DefaultBinapiSocket = "/run/vpp/api.sock"
 )
 
 var (
-- 
cgit 1.2.3-korg