From 8a3e19f0c9993c2b9052503071ff104616248a2c Mon Sep 17 00:00:00 2001
From: Lei Xue <vfs@live.com>
Date: Sat, 10 Dec 2022 22:05:19 +0800
Subject: [PATCH] fix daemon 's host flag

---
 cmd/daemon.go                   | 10 ++++++++--
 pkg/port/iscsit/iscsid.go       |  2 +-
 pkg/scsi/backingstore/common.go | 15 ++++++++-------
 3 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/cmd/daemon.go b/cmd/daemon.go
index c9087ae..83b0799 100644
--- a/cmd/daemon.go
+++ b/cmd/daemon.go
@@ -23,13 +23,15 @@ import (
 	"strings"
 	"syscall"
 
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+
 	"github.com/gostor/gotgt/pkg/apiserver"
 	"github.com/gostor/gotgt/pkg/config"
 	_ "github.com/gostor/gotgt/pkg/port/iscsit"
 	"github.com/gostor/gotgt/pkg/scsi"
 	_ "github.com/gostor/gotgt/pkg/scsi/backingstore"
-	log "github.com/sirupsen/logrus"
-	"github.com/spf13/cobra"
 )
 
 func newDaemonCommand() *cobra.Command {
@@ -41,7 +43,11 @@ func newDaemonCommand() *cobra.Command {
 		Use:   "daemon",
 		Short: "Setup a daemon",
 		Long:  `Setup the Gotgt's daemon`,
+		PreRun: func(cmd *cobra.Command, args []string) {
+			viper.BindPFlags(cmd.Flags())
+		},
 		RunE: func(cmd *cobra.Command, args []string) error {
+			host = viper.GetString("host")
 			return createDaemon(host, driver, logLevel, blockMultipleHosts)
 		},
 	}
diff --git a/pkg/port/iscsit/iscsid.go b/pkg/port/iscsit/iscsid.go
index efd6415..afa98de 100644
--- a/pkg/port/iscsit/iscsid.go
+++ b/pkg/port/iscsit/iscsid.go
@@ -339,7 +339,7 @@ func (s *ISCSITargetDriver) rxHandler(conn *iscsiConnection) {
 			log.Debug("rx handler: IOSTATE_RX_BHS")
 			length, err = conn.readData(buf)
 			if err != nil {
-				log.Error("read BHS failed:", err)
+				log.Error("read BHS failed: ", err)
 				conn.state = CONN_STATE_CLOSE
 				return
 			}
diff --git a/pkg/scsi/backingstore/common.go b/pkg/scsi/backingstore/common.go
index eca1f3f..3160a78 100644
--- a/pkg/scsi/backingstore/common.go
+++ b/pkg/scsi/backingstore/common.go
@@ -21,10 +21,11 @@ import (
 	"io"
 	"os"
 
+	log "github.com/sirupsen/logrus"
+
 	"github.com/gostor/gotgt/pkg/api"
 	"github.com/gostor/gotgt/pkg/scsi"
 	"github.com/gostor/gotgt/pkg/util"
-	log "github.com/sirupsen/logrus"
 )
 
 const (
@@ -51,13 +52,10 @@ func new() (api.BackingStore, error) {
 }
 
 func (bs *FileBackingStore) Open(dev *api.SCSILu, path string) error {
+	var mode os.FileMode
 
-	var (
-		mode  os.FileMode
-		finfo os.FileInfo
-	)
-
-	if finfo, err := os.Stat(path); err != nil {
+	finfo, err := os.Stat(path)
+	if err != nil {
 		return err
 	} else {
 		// determine file type
@@ -75,6 +73,9 @@ func (bs *FileBackingStore) Open(dev *api.SCSILu, path string) error {
 			}
 			bs.DataSize = uint64(pos)
 		} else {
+			if finfo == nil {
+				log.Infof("finfo is nil")
+			}
 			bs.DataSize = uint64(finfo.Size())
 		}
 	}