Ejemplo n.º 1
0
// auth connection
func connAuth(s *nc.SocketServer, conn net.Conn, line string) error {
	if strings.HasPrefix(line, "AUTH:") {
		arr := strings.Split(line[5:], "#") // AUTH:API_ID#SECRET#VERSION
		if len(arr) == 3 {
			var af nc.AuthFunc = func() (int, error) {
				merchantId := dps.MerchantService.GetMerchantIdByApiId(arr[0])
				apiInfo := dps.MerchantService.GetApiInfo(merchantId)
				if apiInfo != nil && apiInfo.ApiSecret == arr[1] {
					if apiInfo.Enabled == 0 {
						return merchantId, errors.New("api has exipres")
					}
				}
				return merchantId, nil
			}

			if err := s.Auth(conn, af); err != nil {
				return err
			}

			s.Printf("[ CLIENT] - Version = %s", arr[2])
			return nil
		}
	}
	return errors.New("conn reject")
}
Ejemplo n.º 2
0
// push member summary to tcp client
func pushMemberSummary(s *nc.SocketServer, connList []net.Conn, memberId int) {
	s.Printf("[ TCP][ NOTIFY] - notify member update - %d", memberId)
	sm := GetMemberSummary(memberId, 0)
	if d, err := json.Marshal(sm); err == nil {
		d = append([]byte("MSUM:"), d...)
		for _, conn := range connList {
			conn.Write(append(d, '\n'))
		}
	}
}
Ejemplo n.º 3
0
// push member summary to tcp client
func pushMemberAccount(s *nc.SocketServer, connList []net.Conn, memberId int) {
	s.Printf("[ TCP][ NOTIFY] - notify account update - %d", memberId)
	sm := getMemberAccount(memberId, 0)
	if sm != nil {
		if d, err := json.Marshal(sm); err == nil {
			d = append([]byte("MACC:"), d...)
			for _, conn := range connList {
				conn.Write(append(d, '\n'))
			}
		}
	}
}
Ejemplo n.º 4
0
func MemberSummaryNotifyJob(s *nc.SocketServer) {
	conn := core.GetRedisConn()
	defer conn.Close()
	for {
		values, err := redis.Values(conn.Do("BLPOP",
			variable.KvMemberUpdateTcpNotifyQueue, 0))
		if err == nil {
			id, err := strconv.Atoi(string(values[1].([]byte)))
			if err == nil {
				connList := s.GetConnections(id)
				if len(connList) > 0 {
					go pushMemberSummary(s, connList, id)
				}
			}
		}
	}
}
Ejemplo n.º 5
0
// Handle command of client sending.
func handleCommand(s *nc.SocketServer, ci *nc.Client, cmd string) ([]byte, error) {
	if time.Now().Sub(ci.LatestConnectTime) > disconnectDuration { //主动关闭没有活动的连接
		//s.Print("--disconnect ---",ci.Addr.String())
		ci.Conn.Close()
		return nil, nil
	}
	if !strings.HasPrefix(cmd, "PING") {
		s.Printf("[ CLIENT][ MESSAGE] - send by %d ; %s", ci.Source, cmd)
		ci.LatestConnectTime = time.Now()
	}
	i := strings.Index(cmd, ":")
	if i != -1 {
		plan := cmd[i+1:]
		if v, ok := handlers[cmd[:i]]; ok {
			return v(ci, plan)
		}
	}
	return nil, errors.New("unknown command:" + cmd)
}
Ejemplo n.º 6
0
// member auth,command like 'MAUTH:jarrysix#3234234242342342'
func memberAuth(s *nc.SocketServer, id *nc.Client, param string) ([]byte, error) {
	var err error
	arr := strings.Split(param, "#")
	if len(arr) == 2 {

		f := func() (int, error) {
			memberId, _ := strconv.Atoi(arr[0])
			authOk := util.CompareMemberApiToken(gof.CurrentApp.Storage(),
				memberId, arr[1])
			if !authOk {
				return memberId, errors.New("auth fail")
			}
			return memberId, nil
		}

		if err = s.UAuth(id.Conn, f); err == nil { //验证成功
			return []byte("ok"), nil
		}
	}
	return nil, err
}
Ejemplo n.º 7
0
func NewServe(output bool) *nc.SocketServer {
	var s *nc.SocketServer
	r := func(conn net.Conn, b []byte) ([]byte, error) {
		cmd := string(b)
		id, ok := s.GetCli(conn)
		if !ok {
			// not join,auth first!
			if err := connAuth(s, conn, cmd); err != nil {
				return nil, err
			}
			return []byte("ok"), nil
		}
		if strings.HasPrefix(cmd, "MAUTH:") {
			//auth member
			return memberAuth(s, id, cmd[6:])
		}
		return handleCommand(s, id, cmd)
	}

	s = nc.NewSocketServer(r)
	s.ReadDeadLine = defaultReadDeadLine
	if !output {
		s.OutputOff()
	}
	return s
}