新增：连接验权 auth 处理

git-svn-id: svn://47.119.165.148/zhub@156 e63fbceb-bcc3-4977-ac22-735b83d8d0f4

cmd/client.go

@@ -34,6 +34,7 @@ type Client struct {
 	chReceive    chan []string    // chan of receive message
 	timerReceive chan []string    // chan of timer
 	lockFlag     map[string]*Lock // chan of lock
+	auth         string
 }
 
 type Lock struct {
@@ -44,7 +45,7 @@ type Lock struct {
 	// duration  int    // lock duration
 }
 
-func Create(appname string, addr string, groupid string) (*Client, error) {
+func Create(appname, addr, groupid, auth string) (*Client, error) {
 	conn, err := net.Dial("tcp", addr)
 	if err != nil {
 		return &Client{}, err
@@ -65,8 +66,10 @@ func Create(appname string, addr string, groupid string) (*Client, error) {
 		chReceive:    make(chan []string, 100),
 		timerReceive: make(chan []string, 100),
 		lockFlag:     make(map[string]*Lock),
+		auth:         auth,
 	}
 
+	client.send("auth", auth)
 	client.send("groupid " + groupid)
 	client.init()
 	return &client, err
@@ -81,6 +84,7 @@ func (c *Client) reconn() (err error) {
 			continue
 		} else if err == nil {
 			c.conn = conn
+			c.send("auth", c.auth)
 			c.send("groupid " + c.groupid)
 			go c.receive()
 

go.mod

@@ -1,6 +1,6 @@
 module zhub
 
-go 1.16
+go 1.18
 
 require (
 	github.com/go-basic/uuid v1.0.0 // indirect

main.go

@@ -29,7 +29,7 @@ func main() {
 	}
 
 	if len(os.Args) == 3 && strings.EqualFold(os.Args[1], "-r") {
-		if cli, err := cmd.Create("zhub-local", addr, "group-admin"); err != nil {
+		if cli, err := cmd.Create("zhub-local", addr, "group-admin", "zchd@123456"); err != nil {
 			log.Println(err)
 		} else {
 			switch os.Args[2] {

pkg.bat

@@ -1,10 +1,6 @@
 SET GOOS=linux
 SET GOARCH=amd64
-go build -o zhub.sh -ldflags "-s -w" ./app.go
+go build -o zhub.sh -ldflags "-s -w" ./main.go
 upx -9 zhub.sh
 
-scp zhub.sh  pro:/opt/zhub
+rem scp zhub.sh  dev:/opt/zhub
-scp zhub.sh  dev:/opt/zhub
-scp zhub.sh  qc:/opt/zhub
-scp zhub.sh  my:/opt/zhub
-del zhub.sh

zsub/msg-consumer.go

@@ -32,6 +32,10 @@ func msgAccept(v Message) {
 	if LogDebug {
 		log.Println("[", v.Conn.sn, "] rcmd: "+strings.Join(rcmd, " "))
 	}
+	if !c.auth && !strings.EqualFold("auth", rcmd[0]) && strings.EqualFold(GetStr("service.auth", "0"), "1") {
+		c.send("-Auth: NOAUTH Authentication required:" + rcmd[0])
+		return
+	}
 
 	if len(rcmd) == 1 {
 		switch strings.ToLower(rcmd[0]) {
@@ -149,6 +153,21 @@ func msgAccept(v Message) {
 				return
 			}
 			zsub._unlock(Lock{key: rcmd[1], uuid: rcmd[2]})
+		case "auth":
+			if len(rcmd) != 2 || strings.IndexAny(rcmd[1], "@") == -1 {
+				c.send("-Error: invalid password!")
+				return
+			}
+
+			inx := strings.IndexAny(rcmd[1], "@") //user@pwd
+
+			if strings.EqualFold(GetStr("auth."+rcmd[1][:inx], ""), rcmd[1][inx+1:]) {
+				c.auth = true
+				c.send("+Auth: ok!")
+			} else {
+				c.send("-Auth: invalid password!")
+			}
+			return
 		default:
 			c.send("-Error: default not supported:[" + strings.Join(rcmd, " ") + "]")
 			return

zsub/zsub.go

@@ -87,6 +87,7 @@ type ZConn struct { //ZConn
 	substoped map[string]chan int // 关闭信号量
 	ping      int64               // 最后心跳时间
 	pong      int64               // 最后心跳回复时间
+	auth      bool                // 是否已验证授权
 }
 
 type Lock struct {
@@ -483,6 +484,7 @@ func Info() map[string]interface{} {
 		m["groupid"] = c.groupid
 		m["topics"] = c.topics
 		m["timers"] = c.timers
+		m["auth"] = c.auth
 		conns = append(conns, m)
 	}