Files
im_server/server/session.go
T
Austin 753a04ed12 fix: broadcast offline presence as soon as the last session drops
Logout left contacts seeing the user online until the me topic idle timeout
fired presUsersOfInterest(off), and even that frame was silently dropped
for p2p routes by handlePresence's xoriginal check. MarkUserOffline now
returns the remaining session count and cleanUp notifies users of interest
immediately once it hits zero; p2p presence is delivered straight to the
contact's me topic, mirroring presSingleUserOffline. Zentao #22.
2026-09-18 17:42:18 +08:00

1762 lines
53 KiB
Go

/******************************************************************************
*
* Description :
*
* Handling of user sessions/connections. One user may have multiple sesions.
* Each session may handle multiple topics
*
*****************************************************************************/
package main
import (
"container/list"
"encoding/json"
"fmt"
"net/http"
"os"
"strings"
"sync"
"sync/atomic"
"time"
"github.com/gorilla/websocket"
"github.com/tinode/chat/pbx"
"github.com/tinode/chat/server/auth"
"github.com/tinode/chat/server/logs"
"github.com/tinode/chat/server/store"
"github.com/tinode/chat/server/store/types"
"golang.org/x/text/language"
)
// Maximum number of queued messages before session is considered stale and dropped.
const sendQueueLimit = 128
// Time given to a background session to terminate to avoid tiggering presence notifications.
// If session terminates (or unsubscribes from topic) in this time frame notifications are not sent at all.
const deferredNotificationsTimeout = time.Second * 5
var minSupportedVersionValue = parseVersion(minSupportedVersion)
// SessionProto is the type of the wire transport.
type SessionProto int
// Constants defining individual types of wire transports.
const (
// NONE is undefined/not set.
NONE SessionProto = iota
// WEBSOCK represents websocket connection.
WEBSOCK
// LPOLL represents a long polling connection.
LPOLL
// GRPC is a gRPC connection
GRPC
// PROXY is temporary session used as a proxy at master node.
PROXY
// MULTIPLEX is a multiplexing session reprsenting a connection from proxy topic to master.
MULTIPLEX
)
// Session represents a single WS connection or a long polling session. A user may have multiple
// sessions.
type Session struct {
// protocol - NONE (unset), WEBSOCK, LPOLL, GRPC, PROXY, MULTIPLEX
proto SessionProto
// Session ID
sid string
// Websocket. Set only for websocket sessions.
ws *websocket.Conn
// Pointer to session's record in sessionStore. Set only for Long Poll sessions.
lpTracker *list.Element
// gRPC handle. Set only for gRPC clients.
grpcnode pbx.Node_MessageLoopServer
// Reference to the cluster node where the session has originated. Set only for cluster RPC sessions.
clnode *ClusterNode
// Reference to multiplexing session. Set only for proxy sessions.
multi *Session
proxiedTopic string
// IP address of the client. For long polling this is the IP of the last poll.
remoteAddr string
// User agent, a string provived by an authenticated client in {login} packet.
userAgent string
// Protocol version of the client: ((major & 0xff) << 8) | (minor & 0xff).
ver int
// Device ID of the client
deviceID string
// Platform: web, ios, android
platf string
// Human language of the client
lang string
// Country code of the client
countryCode string
// ID of the current user. Could be zero if session is not authenticated
// or for multiplexing sessions.
uid types.Uid
// Tenant context is resolved from the first hi packet and cannot change.
tenantID types.TenantID
tenantCode string
// Authentication level - NONE (unset), ANON, AUTH, ROOT.
authLvl auth.Level
// Time when the long polling session was last refreshed
lastTouched time.Time
// Time when the session received any packer from client
lastAction int64
// Timer which triggers after some seconds to mark background session as foreground.
bkgTimer *time.Timer
// Number of subscribe/unsubscribe requests in flight.
inflightReqs *boundedWaitGroup
// Synchronizes access to session store in cluster mode:
// subscribe/unsubscribe replies are asynchronous.
sessionStoreLock sync.Mutex
// Indicates that the session is terminating.
// After this flag's been flipped to true, there must not be any more writes
// into the session's send channel.
// Read/written atomically.
// 0 = false
// 1 = true
terminating int32
// Background session: subscription presence notifications and online status are delayed.
background bool
// Outbound mesages, buffered.
// The content must be serialized in format suitable for the session.
send chan any
// Channel for shutting down the session, buffer 1.
// Content in the same format as for 'send'
stop chan any
// detach - channel for detaching session from topic, buffered.
// Content is topic name to detach from.
detach chan string
// Map of topic subscriptions, indexed by topic name.
// Don't access directly. Use getters/setters.
subs map[string]*Subscription
// Mutex for subs access: both topic go routines and network go routines access
// subs concurrently.
subsLock sync.RWMutex
// Needed for long polling and grpc.
lock sync.Mutex
// Field used only in cluster mode by topic master node.
// Type of proxy to master request being handled.
proxyReq ProxyReqType
}
// Subscription is a mapper of sessions to topics.
type Subscription struct {
// Channel to communicate with the topic, copy of Topic.clientMsg
broadcast chan<- *ClientComMessage
// Session sends a signal to Topic when this session is unsubscribed
// This is a copy of Topic.unreg
done chan<- *ClientComMessage
// Channel to send {meta} requests, copy of Topic.meta
meta chan<- *ClientComMessage
// Channel to ping topic with session's updates, copy of Topic.supd
supd chan<- *sessionUpdate
}
func (s *Session) addSub(topic string, sub *Subscription) {
if s.multi != nil {
s.multi.addSub(topic, sub)
return
}
s.subsLock.Lock()
// Sessions that serve as an interface between proxy topics and their masters (proxy sessions)
// may have only one subscription, that is, to its master topic.
// Normal sessions may be subscribed to multiple topics.
if !s.isMultiplex() || s.countSub() == 0 {
s.subs[topic] = sub
}
s.subsLock.Unlock()
}
func (s *Session) getSub(topic string) *Subscription {
// Don't check s.multi here. Let it panic if called for proxy session.
s.subsLock.RLock()
defer s.subsLock.RUnlock()
return s.subs[topic]
}
func (s *Session) hasSub(topic string) bool {
if s.multi != nil {
return s.multi.hasSub(topic)
}
s.subsLock.RLock()
defer s.subsLock.RUnlock()
_, ok := s.subs[topic]
return ok
}
func (s *Session) groupSubs() []string {
if s.multi != nil {
return s.multi.groupSubs()
}
s.subsLock.RLock()
defer s.subsLock.RUnlock()
topics := make([]string, 0, len(s.subs))
for topic := range s.subs {
if len(topic) >= 3 && types.GetTopicCat(topic) == types.TopicCatGrp {
topics = append(topics, topic)
}
}
return topics
}
func (s *Session) delSub(topic string) {
if s.multi != nil {
s.multi.delSub(topic)
return
}
s.subsLock.Lock()
delete(s.subs, topic)
s.subsLock.Unlock()
}
func (s *Session) countSub() int {
if s.multi != nil {
return s.multi.countSub()
}
return len(s.subs)
}
// Inform topics that the session is being terminated.
// No need to check for s.multi because it's not called for PROXY sessions.
func (s *Session) unsubAll() {
s.subsLock.RLock()
defer s.subsLock.RUnlock()
for _, sub := range s.subs {
// sub.done is the same as topic.unreg
// The whole session is being dropped; ClientComMessage is a wrapper for session, ClientComMessage.init is false.
// keep redundant init: false so it can be searched for.
sub.done <- &ClientComMessage{sess: s, init: false}
}
}
// Indicates whether this session is a local interface for a remote proxy topic.
// It multiplexes multiple sessions.
func (s *Session) isMultiplex() bool {
return s.proto == MULTIPLEX
}
// Indicates whether this session is a short-lived proxy for a remote session.
func (s *Session) isProxy() bool {
return s.proto == PROXY
}
// Cluster session: either a proxy or a multiplexing session.
func (s *Session) isCluster() bool {
return s.isProxy() || s.isMultiplex()
}
func (s *Session) scheduleClusterWriteLoop() {
if globals.cluster != nil && globals.cluster.proxyEventQueue != nil {
globals.cluster.proxyEventQueue.Schedule(
func() { s.clusterWriteLoop(s.proxiedTopic) })
}
}
func (s *Session) supportsMessageBatching() bool {
switch s.proto {
case WEBSOCK:
return true
case GRPC:
return true
default:
return false
}
}
// queueOut attempts to send a list of ServerComMessages to a session write loop;
// it fails if the send buffer is full.
func (s *Session) queueOutBatch(msgs []*ServerComMessage) bool {
if s == nil {
return true
}
if atomic.LoadInt32(&s.terminating) > 0 {
return true
}
for _, msg := range msgs {
if msg == nil {
continue
}
if msg.TenantID.IsZero() {
msg.TenantID = s.tenantID
} else if !s.tenantID.IsZero() && msg.TenantID != s.tenantID {
logs.Err.Printf("s.queueOutBatch: rejected cross-tenant message: session=%d message=%d", s.tenantID, msg.TenantID)
return false
}
}
if s.multi != nil {
// In case of a cluster we need to pass a copy of the actual session.
for i := range msgs {
msgs[i].sess = s
}
if s.multi.queueOutBatch(msgs) {
s.multi.scheduleClusterWriteLoop()
return true
}
return false
}
if s.supportsMessageBatching() {
select {
case s.send <- msgs:
default:
// Never block here since it may also block the topic's run() goroutine.
logs.Err.Println("s.queueOut: session's send queue2 full", s.sid)
return false
}
if s.isMultiplex() {
s.scheduleClusterWriteLoop()
}
} else {
for _, msg := range msgs {
s.queueOut(msg)
}
}
return true
}
// queueOut attempts to send a ServerComMessage to a session write loop;
// it fails, if the send buffer is full.
func (s *Session) queueOut(msg *ServerComMessage) bool {
if s == nil {
return true
}
if atomic.LoadInt32(&s.terminating) > 0 {
return true
}
if msg == nil {
return true
}
if msg.TenantID.IsZero() {
msg.TenantID = s.tenantID
} else if !s.tenantID.IsZero() && msg.TenantID != s.tenantID {
logs.Err.Printf("s.queueOut: rejected cross-tenant message: session=%d message=%d", s.tenantID, msg.TenantID)
return false
}
if s.multi != nil {
// In case of a cluster we need to pass a copy of the actual session.
msg.sess = s
if s.multi.queueOut(msg) {
s.multi.scheduleClusterWriteLoop()
return true
}
return false
}
// Record latency only on {ctrl} messages and end-user sessions.
if msg.Ctrl != nil && msg.Id != "" {
if !msg.Ctrl.Timestamp.IsZero() && !s.isCluster() {
duration := time.Since(msg.Ctrl.Timestamp).Milliseconds()
statsAddHistSample("RequestLatency", float64(duration))
}
if 200 <= msg.Ctrl.Code && msg.Ctrl.Code < 600 {
statsInc(fmt.Sprintf("CtrlCodesTotal%dxx", msg.Ctrl.Code/100), 1)
} else {
logs.Warn.Println("Invalid response code: ", msg.Ctrl.Code)
}
}
select {
case s.send <- msg:
default:
// Never block here since it may also block the topic's run() goroutine.
logs.Err.Println("s.queueOut: session's send queue full", s.sid)
return false
}
if s.isMultiplex() {
s.scheduleClusterWriteLoop()
}
return true
}
// queueOutBytes attempts to send a ServerComMessage already serialized to []byte.
// If the send buffer is full, it fails.
func (s *Session) queueOutBytes(data []byte) bool {
if s == nil || atomic.LoadInt32(&s.terminating) > 0 {
return true
}
select {
case s.send <- data:
default:
logs.Err.Println("s.queueOutBytes: session's send queue full", s.sid)
return false
}
if s.isMultiplex() {
s.scheduleClusterWriteLoop()
}
return true
}
func (s *Session) maybeScheduleClusterWriteLoop() {
if s.multi != nil {
s.multi.scheduleClusterWriteLoop()
return
}
if s.isMultiplex() {
s.scheduleClusterWriteLoop()
}
}
func (s *Session) detachSession(fromTopic string) {
if atomic.LoadInt32(&s.terminating) == 0 {
s.detach <- fromTopic
s.maybeScheduleClusterWriteLoop()
}
}
func (s *Session) stopSession(data any) {
s.stop <- data
s.maybeScheduleClusterWriteLoop()
}
func (s *Session) purgeChannels() {
for len(s.send) > 0 {
<-s.send
}
for len(s.stop) > 0 {
<-s.stop
}
for len(s.detach) > 0 {
<-s.detach
}
}
// cleanUp is called when the session is terminated to perform resource cleanup.
func (s *Session) cleanUp(expired bool) {
atomic.StoreInt32(&s.terminating, 1)
if !s.uid.IsZero() && s.tenantID.IsValid() {
if err := store.SecurityAudit.CloseLogin(s.tenantID, s.sid, time.Now().UTC().Round(time.Millisecond)); err != nil {
logs.Warn.Println("s.cleanUp: failed to close login audit", s.uid, err, s.sid)
}
remaining, err := store.MarkUserOffline(s.tenantID, s.uid, s.sid, currentNodeID())
if err != nil {
logs.Warn.Println("s.cleanUp: failed to mark user offline", s.uid, err, s.sid)
}
// 最后一个会话已断开(如主动登出、断网):立即通知关注者该用户离线,
// 不再等待 me topic 闲置超时后的 presUsersOfInterest("off")。
if err == nil && remaining == 0 && globals.hub != nil {
if uoi, uerr := store.Users.GetSubs(s.tenantID, s.uid); uerr == nil {
presUsersOfInterestOffline(s.tenantID, s.uid, uoi, "off")
} else {
logs.Warn.Println("s.cleanUp: failed to notify users of interest", s.uid, uerr, s.sid)
}
}
}
s.purgeChannels()
s.inflightReqs.Wait()
s.inflightReqs = nil
if !expired {
s.sessionStoreLock.Lock()
globals.sessionStore.Delete(s)
s.sessionStoreLock.Unlock()
}
s.background = false
s.bkgTimer.Stop()
s.unsubAll()
// Stop the write loop.
s.stopSession(nil)
}
// Message received, convert bytes to ClientComMessage and dispatch
func (s *Session) dispatchRaw(raw []byte) {
now := types.TimeNow()
var msg ClientComMessage
if atomic.LoadInt32(&s.terminating) > 0 {
logs.Warn.Println("s.dispatch: message received on a terminating session", s.sid)
s.queueOut(ErrLocked("", "", now))
return
}
if len(raw) == 1 && raw[0] == 0x31 {
// 0x31 == '1'. This is a network probe message. Respond with a '0':
s.queueOutBytes([]byte{0x30})
return
}
if err := json.Unmarshal(raw, &msg); err != nil {
// Malformed message
logs.Warn.Println("s.dispatch", err, s.sid)
s.queueOut(ErrMalformed("", "", now))
return
}
toLog := redactClientMessageForLog(raw)
truncated := ""
if len(toLog) > 512 {
toLog = toLog[:512]
truncated = "<...>"
}
logs.Info.Printf("in: '%s%s' sid='%s' uid='%s'", toLog, truncated, s.sid, s.uid)
s.dispatch(&msg)
}
func redactClientMessageForLog(raw []byte) []byte {
var payload map[string]any
if json.Unmarshal(raw, &payload) != nil {
return []byte("[malformed payload]")
}
for _, section := range []string{"login", "acc"} {
body, ok := payload[section].(map[string]any)
if !ok {
continue
}
for _, field := range []string{"secret", "tmpsecret"} {
if _, exists := body[field]; exists {
body[field] = "[REDACTED]"
}
}
}
redacted, err := json.Marshal(payload)
if err != nil {
return []byte("[unprintable payload]")
}
return redacted
}
func (s *Session) dispatch(msg *ClientComMessage) {
now := types.TimeNow()
atomic.StoreInt64(&s.lastAction, now.UnixNano())
s.refreshPresence("s.dispatch")
// This should be the first block here, before any other checks.
var resp *ServerComMessage
if msg, resp = pluginFireHose(s, msg); resp != nil {
// Plugin provided a response. No further processing is needed.
s.queueOut(resp)
return
} else if msg == nil {
// Plugin requested to silently drop the request.
return
}
authLvl := auth.LevelNone
if msg.Extra != nil {
authLvl = auth.ParseAuthLevel(msg.Extra.AuthLevel)
}
if msg.Extra == nil || (msg.Extra.AsUser == "" && authLvl == auth.LevelNone) {
// Use current user's ID and auth level.
msg.AsUser = s.uid.UserId()
msg.AuthLvl = int(s.authLvl)
} else if s.authLvl != auth.LevelRoot {
// Only root user can set alternative user ID and auth level values.
s.queueOut(ErrPermissionDenied("", "", now))
logs.Warn.Println("s.dispatch: non-root assigned asUser", s.sid)
return
} else if fromUid := types.ParseUserId(msg.Extra.AsUser); fromUid.IsZero() {
// Invalid msg.Extra.AsUser.
s.queueOut(ErrMalformed("", "", now))
logs.Warn.Println("s.dispatch: malformed asUser: ", msg.Extra.AsUser, s.sid)
return
} else {
state, err := userGetState(s.tenantID, fromUid)
if err != nil || state != types.StateOK {
s.queueOut(ErrPermissionDenied("", "", now))
logs.Warn.Println("s.dispatch: cross-tenant or inactive asUser", s.sid)
return
}
// Use provided msg.Extra.AsUser
msg.AsUser = msg.Extra.AsUser
// Assign auth level, if one is provided. Ignore invalid strings.
if authLvl == auth.LevelNone {
// AuthLvl is not set by the caller, assign default LevelAuth.
msg.AuthLvl = int(auth.LevelAuth)
} else {
msg.AuthLvl = int(authLvl)
}
}
msg.Timestamp = now
var handler func(*ClientComMessage)
var uaRefresh bool
// Check if s.ver is defined
checkVers := func(handler func(*ClientComMessage)) func(*ClientComMessage) {
return func(m *ClientComMessage) {
if s.ver == 0 {
logs.Warn.Println("s.dispatch: {hi} is missing", s.sid)
s.queueOut(ErrCommandOutOfSequence(m.Id, m.Original, msg.Timestamp))
return
}
handler(m)
}
}
// Check if the session has an immutable tenant context.
checkTenant := func(handler func(*ClientComMessage)) func(*ClientComMessage) {
return func(m *ClientComMessage) {
if s.tenantID.IsZero() {
logs.Warn.Println("s.dispatch: tenant context is missing", s.sid)
s.queueOut(ErrCommandOutOfSequence(m.Id, m.Original, msg.Timestamp))
return
}
handler(m)
}
}
// Check if user is logged in
checkUser := func(handler func(*ClientComMessage)) func(*ClientComMessage) {
return func(m *ClientComMessage) {
if msg.AsUser == "" {
logs.Warn.Println("s.dispatch: authentication required", s.sid)
s.queueOut(ErrAuthRequiredReply(m, m.Timestamp))
return
}
handler(m)
}
}
switch {
case msg.Pub != nil:
handler = checkVers(checkTenant(checkUser(s.publish)))
msg.Id = msg.Pub.Id
msg.Original = msg.Pub.Topic
uaRefresh = true
case msg.Sub != nil:
handler = checkVers(checkTenant(checkUser(s.subscribe)))
msg.Id = msg.Sub.Id
msg.Original = msg.Sub.Topic
uaRefresh = true
case msg.Leave != nil:
handler = checkVers(checkTenant(checkUser(s.leave)))
msg.Id = msg.Leave.Id
msg.Original = msg.Leave.Topic
case msg.Hi != nil:
handler = s.hello
msg.Id = msg.Hi.Id
case msg.Login != nil:
handler = checkVers(checkTenant(s.login))
msg.Id = msg.Login.Id
case msg.Get != nil:
handler = checkVers(checkTenant(checkUser(s.get)))
msg.Id = msg.Get.Id
msg.Original = msg.Get.Topic
uaRefresh = true
case msg.Set != nil:
handler = checkVers(checkTenant(checkUser(s.set)))
msg.Id = msg.Set.Id
msg.Original = msg.Set.Topic
uaRefresh = true
case msg.Del != nil:
handler = checkVers(checkTenant(checkUser(s.del)))
msg.Id = msg.Del.Id
msg.Original = msg.Del.Topic
case msg.Acc != nil:
handler = checkVers(checkTenant(s.acc))
msg.Id = msg.Acc.Id
case msg.Note != nil:
// If user is not authenticated or version not set the {note} is silently ignored.
handler = s.note
msg.Original = msg.Note.Topic
uaRefresh = true
default:
// Unknown message
s.queueOut(ErrMalformed("", "", msg.Timestamp))
logs.Warn.Println("s.dispatch: unknown message", s.sid)
return
}
if globals.cluster.isPartitioned() {
// The cluster is partitioned due to network or other failure and this node is a part of the smaller partition.
// In order to avoid data inconsistency across the cluster we must reject all requests.
s.queueOut(ErrClusterUnreachableReply(msg, msg.Timestamp))
return
}
msg.sess = s
msg.init = true
msg.TenantID = s.tenantID
handler(msg)
// Notify 'me' topic that this session is currently active.
if uaRefresh && msg.AsUser != "" && s.userAgent != "" {
if sub := s.getSub(msg.AsUser); sub != nil {
// The chan is buffered. If the buffer is exhaused, the session will wait for 'me' to become available
sub.supd <- &sessionUpdate{userAgent: s.userAgent}
}
}
}
func (s *Session) refreshPresence(context string) {
if s.uid.IsZero() || !s.tenantID.IsValid() {
return
}
if err := store.RefreshUserPresence(s.tenantID, s.uid, s.sid, currentNodeID()); err != nil {
logs.Warn.Println(context+": failed to refresh user presence", s.uid, err, s.sid)
}
for _, topic := range s.groupSubs() {
if err := store.MarkTopicMemberOnline(s.tenantID, topic, s.uid, currentNodeID()); err != nil {
logs.Warn.Println(context+": failed to refresh group member presence", topic, s.uid, err, s.sid)
}
}
}
// Request to subscribe to a topic.
func (s *Session) subscribe(msg *ClientComMessage) {
if strings.HasPrefix(msg.Original, "new") || strings.HasPrefix(msg.Original, "nch") {
// Request to create a new group/channel topic.
// If we are in a cluster, make sure the new topic belongs to the current node.
msg.RcptTo = globals.cluster.genLocalTopicName(msg.TenantID)
} else {
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
}
s.inflightReqs.Add(1)
// Session can subscribe to topic on behalf of a single user at a time.
// Fast-path the duplicate attach only when the client used the canonical
// topic name itself. A p2p topic may also be addressed by an alias (the
// peer's user id, e.g. "usrXXX") while the session is already attached to
// the canonical topic ("p2pXXX"): forwarding it to the topic lets the
// client complete the alias->canonical rename with a 200 reply instead of
// a 304 that would leave the client-side alias topic object unattached
// (publishes on it then fail with "Cannot publish on inactive topic").
if sub := s.getSub(msg.RcptTo); sub != nil && canReplyAlreadySubscribed(msg.Original, msg.RcptTo) {
s.queueOut(InfoAlreadySubscribed(msg.Id, msg.Original, msg.Timestamp))
s.inflightReqs.Done()
} else {
select {
case globals.hub.join <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
s.inflightReqs.Done()
logs.Err.Println("s.subscribe: hub.join queue full, topic ", msg.RcptTo, s.sid)
}
// Hub will send Ctrl success/failure packets back to session
}
}
func canReplyAlreadySubscribed(original, routeTo string) bool {
if original == routeTo {
return true
}
switch original {
case "me", "fnd", "slf":
return true
default:
return false
}
}
// Leave/Unsubscribe a topic
func (s *Session) leave(msg *ClientComMessage) {
// Expand topic name
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
s.inflightReqs.Add(1)
if sub := s.getSub(msg.RcptTo); sub != nil {
// Session is attached to the topic.
if (msg.Original == "me" || msg.Original == "fnd") && msg.Leave.Unsub {
// User should not unsubscribe from 'me' or 'find'. Just leaving is fine.
s.queueOut(ErrPermissionDeniedReply(msg, msg.Timestamp))
s.inflightReqs.Done()
} else {
// Unlink from topic, topic will send a reply.
sub.done <- msg
}
return
}
s.inflightReqs.Done()
if !msg.Leave.Unsub {
// Session is not attached to the topic, wants to leave - fine, no change
s.queueOut(InfoNotJoined(msg.Id, msg.Original, msg.Timestamp))
} else {
// Session wants to unsubscribe from the topic it did not join
// TODO(gene): allow topic to unsubscribe without joining first; send to hub to unsub
logs.Warn.Println("s.leave:", "must attach first", s.sid)
s.queueOut(ErrAttachFirst(msg, msg.Timestamp))
}
}
// Broadcast a message to all topic subscribers
func (s *Session) publish(msg *ClientComMessage) {
// TODO(gene): Check for repeated messages with the same ID
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
// Add "sender" header if the message is sent on behalf of another user.
if msg.AsUser != s.uid.UserId() {
if msg.Pub.Head == nil {
msg.Pub.Head = make(map[string]any)
}
msg.Pub.Head["sender"] = s.uid.UserId()
} else if msg.Pub.Head != nil {
// Clear potentially false "sender" field.
delete(msg.Pub.Head, "sender")
if len(msg.Pub.Head) == 0 {
msg.Pub.Head = nil
}
}
if sub := s.getSub(msg.RcptTo); sub != nil {
// This is a post to a subscribed topic. The message is sent to the topic only
select {
case sub.broadcast <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.publish: sub.broadcast channel full, topic ", msg.RcptTo, s.sid)
}
} else if msg.RcptTo == "sys" {
if s.authLvl != auth.LevelRoot {
s.queueOut(ErrPermissionDeniedReply(msg, msg.Timestamp))
return
}
// Publishing to "sys" topic requires no subscription.
select {
case globals.hub.routeCli <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.publish: hub.route channel full", s.sid)
}
} else {
// Publish request received without attaching to topic first.
s.queueOut(ErrAttachFirst(msg, msg.Timestamp))
logs.Warn.Printf("s.publish[%s]: must attach first %s", msg.RcptTo, s.sid)
}
}
// Client metadata
func (s *Session) hello(msg *ClientComMessage) {
var params map[string]any
var deviceIDUpdate bool
if s.ver == 0 {
clientVersion := parseVersion(msg.Hi.Version)
if clientVersion == 0 {
logs.Warn.Println("s.hello:", "failed to parse version", s.sid)
s.queueOut(ErrMalformed(msg.Id, "", msg.Timestamp))
return
}
// Check version compatibility
if versionCompare(clientVersion, minSupportedVersionValue) < 0 {
s.queueOut(ErrVersionNotSupported(msg.Id, msg.Timestamp))
logs.Warn.Println("s.hello:", "unsupported version", s.sid)
return
}
tenantCode := strings.TrimSpace(msg.Hi.Tenant)
if tenantCode == "" {
// Clients that predate the tenant-aware protocol (stock Tinode
// webapp/mobile SDKs never send this field) fall back to the
// default tenant rather than being rejected. Clients that do
// send an explicit code are unaffected.
tenantCode = "default"
} else if len(tenantCode) > 64 {
s.queueOut(ErrMalformed(msg.Id, "", msg.Timestamp))
logs.Warn.Println("s.hello: tenant code is invalid", s.sid)
return
}
if store.Tenants == nil {
s.queueOut(ErrUnknown(msg.Id, "", msg.Timestamp))
logs.Err.Println("s.hello: tenant store is not initialized", s.sid)
return
}
tenant, err := store.Tenants.GetByCode(tenantCode)
if err != nil {
s.queueOut(decodeStoreError(err, msg.Id, msg.Timestamp, nil))
logs.Warn.Println("s.hello: failed to resolve tenant", tenantCode, err, s.sid)
return
}
if tenant == nil || tenant.ID.IsZero() {
s.queueOut(ErrUnknown(msg.Id, "", msg.Timestamp))
logs.Err.Println("s.hello: tenant store returned an invalid record", tenantCode, s.sid)
return
}
if !tenant.IsActive() {
s.queueOut(ErrPermissionDenied(msg.Id, "", msg.Timestamp))
logs.Warn.Println("s.hello: tenant is not active", tenant.Code, s.sid)
return
}
s.ver = clientVersion
s.tenantID = tenant.ID
s.tenantCode = tenant.Code
params = map[string]any{
"ver": currentVersion,
"build": store.Store.GetAdapterName() + ":" + buildstamp,
"maxMessageSize": globals.maxMessageSize,
"maxSubscriberCount": globals.maxSubscriberCount,
"minTagLength": minTagLength,
"maxTagLength": maxTagLength,
"maxTagCount": globals.maxTagCount,
"maxFileUploadSize": globals.maxFileUploadSize,
"reqCred": globals.validatorClientConfig,
"msgDelAge": globals.msgDeleteAge.Seconds(),
"tenant": map[string]any{
"code": tenant.Code,
"name": tenant.Name,
},
}
if len(globals.iceServers) > 0 {
params["iceServers"] = globals.iceServers
}
if timeout := imCallTimeoutSeconds(); timeout > 0 {
params["callTimeout"] = timeout
}
if s.proto == GRPC {
// gRPC client may need server address to be able to fetch large files over http(s).
// TODO: add support for fetching files over gRPC, then remove this parameter.
params["servingAt"] = globals.servingAt
// Report cluster size.
if globals.cluster != nil {
params["clusterSize"] = len(globals.cluster.nodes) + 1
} else {
params["clusterSize"] = 1
}
}
// Set ua & platform in the beginning of the session.
// Don't change them later.
s.userAgent = msg.Hi.UserAgent
s.platf = msg.Hi.Platform
if s.platf == "" {
s.platf = platformFromUA(msg.Hi.UserAgent)
}
// This is a background session. Start a timer.
if msg.Hi.Background {
s.bkgTimer.Reset(deferredNotificationsTimeout)
}
} else if msg.Hi.Version == "" || parseVersion(msg.Hi.Version) == s.ver {
if tenantCode := strings.TrimSpace(msg.Hi.Tenant); tenantCode != "" && tenantCode != s.tenantCode {
s.queueOut(ErrCommandOutOfSequence(msg.Id, "", msg.Timestamp))
logs.Warn.Println("s.hello: tenant cannot be changed", s.sid)
return
}
// Save changed device ID+Lang or delete earlier specified device ID.
// Platform cannot be changed.
if !s.uid.IsZero() {
var err error
if msg.Hi.DeviceID == types.NullValue {
// User wants to delete device ID.
deviceIDUpdate = true
if s.deviceID != "" {
err = store.Devices.Delete(s.tenantID, s.uid, s.deviceID)
}
} else if msg.Hi.DeviceID != "" && s.deviceID != msg.Hi.DeviceID {
deviceIDUpdate = true
err = store.Devices.Update(s.tenantID, s.uid, s.deviceID, &types.DeviceDef{
TenantID: s.tenantID,
DeviceId: msg.Hi.DeviceID,
Platform: s.platf,
LastSeen: msg.Timestamp,
Lang: msg.Hi.Lang,
})
userChannelsSubUnsub(s.tenantID, s.uid, msg.Hi.DeviceID, true)
}
if err != nil {
s.queueOut(decodeStoreError(err, msg.Id, msg.Timestamp, nil))
logs.Warn.Println("s.hello:", "device ID", err, s.sid)
return
}
} else {
// Session is not authenticated, report an error. Otherwise,
// the client may think that the device ID was updated successfully,
// but it will not be saved in the database.
s.queueOut(ErrAuthRequiredReply(msg, msg.Timestamp))
logs.Warn.Println("s.hello:", "device ID update requires authentication", s.sid)
return
}
} else {
// Version cannot be changed mid-session.
s.queueOut(ErrCommandOutOfSequence(msg.Id, "", msg.Timestamp))
logs.Warn.Println("s.hello:", "version cannot be changed", s.sid)
return
}
if msg.Hi.DeviceID == types.NullValue {
msg.Hi.DeviceID = ""
}
s.deviceID = msg.Hi.DeviceID
s.lang = msg.Hi.Lang
// Try to deduce the country from the locale.
// Tag may be well-defined even if err != nil. For example, for 'zh_CN_#Hans'
// the tag is 'zh-CN' exact but the err is 'tag is not well-formed'.
if tag, _ := language.Parse(s.lang); tag != language.Und {
if region, conf := tag.Region(); region.IsCountry() && conf >= language.High {
s.countryCode = region.String()
}
}
if s.countryCode == "" {
if len(s.lang) > 2 {
// Logging strings longer than 2 b/c language.Parse(XX) always succeeds
// returning confidence Low.
logs.Warn.Println("s.hello:", "could not parse locale ", s.lang)
}
s.countryCode = globals.defaultCountryCode
}
var httpStatus int
var httpStatusText string
if s.proto == LPOLL || deviceIDUpdate {
// In case of long polling StatusCreated was reported earlier.
// In case of deviceID update just report success.
httpStatus = http.StatusOK
httpStatusText = "ok"
} else {
httpStatus = http.StatusCreated
httpStatusText = "created"
}
ctrl := &MsgServerCtrl{Id: msg.Id, Code: httpStatus, Text: httpStatusText, Timestamp: msg.Timestamp}
if len(params) > 0 {
ctrl.Params = params
}
s.queueOut(&ServerComMessage{Ctrl: ctrl})
}
// Account creation
func (s *Session) acc(msg *ClientComMessage) {
newAcc := strings.HasPrefix(msg.Acc.User, "new")
// If temporary auth parameters are provided, get the user ID from them.
var rec *auth.Rec
if !newAcc && msg.Acc.TmpScheme != "" {
if !s.uid.IsZero() {
s.queueOut(ErrAlreadyAuthenticated(msg.Acc.Id, "", msg.Timestamp))
logs.Warn.Println("s.acc: got temp auth while already authenticated", s.sid)
return
}
authHdl := store.Store.GetLogicalAuthHandler(msg.Acc.TmpScheme)
if authHdl == nil {
logs.Warn.Println("s.acc: unknown authentication scheme", msg.Acc.TmpScheme, s.sid)
s.queueOut(ErrAuthUnknownScheme(msg.Id, "", msg.Timestamp))
}
var err error
rec, _, err = authHdl.Authenticate(auth.AuthContext{TenantID: s.tenantID, RemoteAddr: s.remoteAddr},
msg.Acc.TmpSecret)
if err != nil {
s.queueOut(decodeStoreError(err, msg.Acc.Id, msg.Timestamp,
map[string]any{"what": "auth"}))
logs.Warn.Println("s.acc: invalid temp auth", err, s.sid)
return
}
if rec == nil || rec.TenantID != s.tenantID {
s.queueOut(decodeStoreError(types.ErrFailed, msg.Acc.Id, msg.Timestamp,
map[string]any{"what": "auth"}))
logs.Warn.Println("s.acc: temp auth returned invalid tenant", s.sid)
return
}
}
if newAcc {
// New account
replyCreateUser(s, msg, rec)
} else {
// Existing account.
replyUpdateUser(s, msg, rec)
}
}
// Authenticate
func (s *Session) login(msg *ClientComMessage) {
// msg.from is ignored here
if msg.Login.Scheme == "reset" {
if err := s.authSecretReset(msg.Login.Secret); err != nil {
s.queueOut(decodeStoreError(err, msg.Id, msg.Timestamp, nil))
} else {
s.queueOut(InfoAuthReset(msg.Id, msg.Timestamp))
}
return
}
if !s.uid.IsZero() {
// TODO: change error to notice InfoNoChange and return current user ID & auth level
// params := map[string]interface{}{"user": s.uid.UserId(), "authlvl": s.authLevel.String()}
s.queueOut(ErrAlreadyAuthenticated(msg.Id, "", msg.Timestamp))
return
}
handler := store.Store.GetLogicalAuthHandler(msg.Login.Scheme)
if handler == nil {
recordSessionLoginFailure(s, msg.Login.Scheme, msg.Login.Secret, nil, msg.Id, types.ErrUnsupported, msg.Timestamp)
logs.Warn.Println("s.login: unknown authentication scheme", msg.Login.Scheme, s.sid)
s.queueOut(ErrAuthUnknownScheme(msg.Id, "", msg.Timestamp))
return
}
rec, challenge, err := handler.Authenticate(
auth.AuthContext{TenantID: s.tenantID, RemoteAddr: s.remoteAddr}, msg.Login.Secret)
if err != nil {
recordSessionLoginFailure(s, msg.Login.Scheme, msg.Login.Secret, rec, msg.Id, err, msg.Timestamp)
resp := decodeStoreError(err, msg.Id, msg.Timestamp, nil)
if resp.Ctrl.Code >= 500 {
// Log internal errors
logs.Warn.Println("s.login: internal", err, s.sid)
}
s.queueOut(resp)
return
}
if rec == nil || rec.TenantID != s.tenantID {
recordSessionLoginFailure(s, msg.Login.Scheme, msg.Login.Secret, rec, msg.Id, types.ErrFailed, msg.Timestamp)
logs.Warn.Println("s.login: authenticator returned invalid tenant", s.sid)
s.queueOut(decodeStoreError(types.ErrFailed, msg.Id, msg.Timestamp, nil))
return
}
// If authenticator did not check user state, it returns state "undef". If so, check user state here.
if rec.State == types.StateUndefined {
rec.State, err = userGetState(s.tenantID, rec.Uid)
}
if err == nil && rec.State != types.StateOK {
err = types.ErrPermissionDenied
}
if err == nil {
err = checkTenantLoginIPWhitelist(s.tenantID, s.platf, s.remoteAddr)
}
if err == nil {
err = checkUserLoginIPWhitelist(s.tenantID, rec.Uid, s.remoteAddr)
}
if err != nil {
recordSessionLoginFailure(s, msg.Login.Scheme, msg.Login.Secret, rec, msg.Id, err, msg.Timestamp)
logs.Warn.Println("s.login: user state check failed", rec.Uid, err, s.sid)
s.queueOut(decodeStoreError(err, msg.Id, msg.Timestamp, nil))
return
}
if challenge != nil {
// Multi-stage authentication. Issue challenge to the client.
s.queueOut(InfoChallenge(msg.Id, msg.Timestamp, challenge))
return
}
var missing []string
if rec.Features&auth.FeatureValidated == 0 && len(globals.authValidators[rec.AuthLevel]) > 0 {
var validated []string
// Check responses. Ignore invalid responses, just keep cred unvalidated.
if validated, _, err = validatedCreds(s.tenantID, rec.Uid, rec.AuthLevel, msg.Login.Cred, false); err == nil {
// Get a list of credentials which have not been validated.
_, missing, _ = stringSliceDelta(globals.authValidators[rec.AuthLevel], validated)
}
}
if err != nil {
recordSessionLoginFailure(s, msg.Login.Scheme, msg.Login.Secret, rec, msg.Id, err, msg.Timestamp)
logs.Warn.Println("s.login: failed to validate credentials:", err, s.sid)
s.queueOut(decodeStoreError(err, msg.Id, msg.Timestamp, nil))
} else {
s.queueOut(s.onLogin(msg.Id, msg.Timestamp, rec, missing, msg.Login.Scheme, strings.SplitN(string(msg.Login.Secret), ":", 2)[0]))
}
}
// authSecretReset resets an authentication secret;
// params: "auth-method-to-reset:credential-method:credential-value",
// for example: "basic:email:alice@example.com".
func (s *Session) authSecretReset(params []byte) error {
var authScheme, credMethod, credValue string
if parts := strings.Split(string(params), ":"); len(parts) >= 3 {
authScheme, credMethod, credValue = parts[0], parts[1], parts[2]
} else {
return types.ErrMalformed
}
// Technically we don't need to check it here, but we are going to mail the 'authScheme' string to the user.
// We have to make sure it does not contain any exploits. This is the simplest check.
auther := store.Store.GetLogicalAuthHandler(authScheme)
if auther == nil {
return types.ErrUnsupported
}
validator := store.Store.GetValidator(credMethod)
if validator == nil {
return types.ErrUnsupported
}
uid, err := store.Users.GetByCred(s.tenantID, credMethod, credValue)
if err != nil {
return err
}
if uid.IsZero() {
// Prevent discovery of existing contacts: report "no error" if contact is not found.
return nil
}
resetParams, err := auther.GetResetParams(s.tenantID, uid)
if err != nil {
return err
}
tempScheme, err := validator.TempAuthScheme()
if err != nil {
return err
}
tempAuth := store.Store.GetLogicalAuthHandler(tempScheme)
if tempAuth == nil || !tempAuth.IsInitialized() {
logs.Err.Println("s.authSecretReset: validator with missing temp auth", credMethod, tempScheme, s.sid)
return types.ErrInternal
}
code, _, err := tempAuth.GenSecret(&auth.Rec{
TenantID: s.tenantID,
Uid: uid,
AuthLevel: auth.LevelAuth,
Features: auth.FeatureNoLogin,
Credential: credMethod + ":" + credValue,
})
if err != nil {
return err
}
return validator.ResetSecret(s.tenantID, credValue, authScheme, s.lang, code, resetParams)
}
// onLogin performs steps after successful authentication.
func (s *Session) onLogin(msgID string, timestamp time.Time, rec *auth.Rec, missing []string, scheme, account string) *ServerComMessage {
var reply *ServerComMessage
var params map[string]any
features := rec.Features
params = map[string]any{
"user": rec.Uid.UserId(),
"authlvl": rec.AuthLevel.String(),
}
if len(missing) > 0 {
// Some credentials are not validated yet. Respond with request for validation.
reply = InfoValidateCredentials(msgID, timestamp)
params["cred"] = missing
} else {
// Everything is fine, authenticate the session.
reply = NoErr(msgID, "", timestamp)
// Check if the token is suitable for session authentication.
if features&auth.FeatureNoLogin == 0 {
if store.Security == nil {
logs.Warn.Println("s.onLogin: user security store is unavailable", s.sid)
reply = decodeStoreError(types.ErrInternal, msgID, timestamp, nil)
return reply
}
authVersion, err := store.Security.GetAuthVersion(s.tenantID, rec.Uid)
if err != nil {
logs.Warn.Println("s.onLogin: failed to read auth version", rec.Uid, err, s.sid)
reply = decodeStoreError(err, msgID, timestamp, nil)
return reply
}
rec.AuthVersion = authVersion
// Authenticate the session.
s.uid = rec.Uid
s.authLvl = rec.AuthLevel
if err := store.MarkUserOnline(s.tenantID, s.uid, s.sid, currentNodeID(), s.platf, s.userAgent); err != nil {
logs.Warn.Println("s.onLogin: failed to mark user online", s.uid, err, s.sid)
}
// 同平台抢登互踢:新登录踢掉同平台旧会话,异平台共存。
if globals.sessionStore != nil {
globals.sessionStore.EvictUserPlatform(rec.Uid, s.platf, s.sid, "replaced_by_new_login")
}
publishRealtimeEvictUserPlatform(s.tenantID, rec.Uid, s.platf, "replaced_by_new_login", s.sid)
// Reset expiration time.
rec.Lifetime = 0
}
features |= auth.FeatureValidated
// Record deviceId used in this session
if s.deviceID != "" {
isNewDevice, deviceErr := isNewLoginDevice(s.tenantID, rec.Uid, s.deviceID)
if deviceErr != nil {
logs.Warn.Println("s.onLogin: failed to check known device", rec.Uid, deviceErr, s.sid)
}
if err := store.Devices.Update(s.tenantID, rec.Uid, "", &types.DeviceDef{
TenantID: s.tenantID,
DeviceId: s.deviceID,
Platform: s.platf,
LastSeen: timestamp,
Lang: s.lang,
}); err != nil {
logs.Warn.Println("failed to update device record", err)
} else if isNewDevice {
createLoginDeviceSecurityNotice(s.tenantID, rec.Uid, s.deviceID, s.platf, s.userAgent, timestamp)
}
}
if !s.uid.IsZero() {
recordSessionLoginSuccess(s, rec, scheme, account, msgID, timestamp)
}
}
// GenSecret fails only if tokenLifetime is < 0. It can't be < 0 here,
// otherwise login would have failed earlier.
rec.Features = features
token, expires, _ := store.Store.GetLogicalAuthHandler("token").GenSecret(rec)
params["token"], params["expires"] = token, expires
if !s.uid.IsZero() && features&auth.FeatureNoLogin == 0 {
if enabled, err := store.StoreLoginSession(token, store.LoginSession{
UserID: store.DecodeUid(rec.Uid),
TenantID: int64(s.tenantID),
TenantCode: s.tenantCode,
DeviceID: s.deviceID,
AuthLevel: rec.AuthLevel.String(),
State: "active",
ExpiresAt: expires.Format(time.RFC3339),
IssuedAt: time.Now().UTC().Format(time.RFC3339),
}, time.Until(expires)); enabled && err != nil {
logs.Warn.Println("s.onLogin: failed to store login session", rec.Uid, err, s.sid)
}
}
reply.Ctrl.Params = params
return reply
}
func isNewLoginDevice(tenantID types.TenantID, uid types.Uid, deviceID string) (bool, error) {
if store.Devices == nil || strings.TrimSpace(deviceID) == "" || uid.IsZero() {
return false, nil
}
devices, _, err := store.Devices.GetAll(tenantID, uid)
if err != nil {
return false, err
}
for _, device := range devices[uid] {
if device.DeviceId == deviceID {
return false, nil
}
}
return true, nil
}
func createLoginDeviceSecurityNotice(tenantID types.TenantID, uid types.Uid, deviceID, platform, userAgent string, at time.Time) {
if store.Security == nil || uid.IsZero() || strings.TrimSpace(deviceID) == "" {
return
}
eventID := stableIMEventID("security.login_device", uid.UserId(), deviceID)
if err := createIMSystemNotification(tenantID, uid, eventID, "security.login_device",
"新设备登录提醒", loginDeviceSecurityNoticeSummary(platform, userAgent),
map[string]any{
"scene": "login_device",
"device_id": deviceID,
"platform": strings.TrimSpace(platform),
"user_agent": strings.TrimSpace(userAgent),
"login_at": at.UTC().Format(time.RFC3339Nano),
}); err != nil {
logs.Warn.Println("s.onLogin: failed to create login device notification", uid, err)
}
}
func loginDeviceSecurityNoticeSummary(platform, userAgent string) string {
platform = strings.TrimSpace(platform)
if platform == "" {
platform = "未知设备"
}
userAgent = strings.TrimSpace(userAgent)
if userAgent == "" {
return platform + " 登录了你的账号"
}
if len([]rune(userAgent)) > 80 {
userAgent = string([]rune(userAgent)[:80])
}
return platform + " 登录了你的账号:" + userAgent
}
func currentNodeID() string {
if globals.cluster != nil && globals.cluster.thisNodeName != "" {
return globals.cluster.thisNodeName
}
if globals.nodeID != "" {
return globals.nodeID
}
return resolveIMNodeID("")
}
func resolveIMNodeID(configured string) string {
if configured = strings.TrimSpace(configured); configured != "" {
return configured
}
if value := strings.TrimSpace(os.Getenv("IM_NODE_ID")); value != "" {
return value
}
if value := strings.TrimSpace(os.Getenv("HOSTNAME")); value != "" {
return value
}
return "local"
}
func redisUserOnline(tenantID types.TenantID, uid types.Uid, fallback bool) bool {
if !store.UserPresenceEnabled() || !tenantID.IsValid() || uid.IsZero() {
return fallback
}
presence, err := store.GetUserPresence(tenantID, uid)
if err != nil {
logs.Warn.Printf("presence[%s]: failed to read redis presence: %v", uid.UserId(), err)
return fallback
}
return presence.Online
}
func (s *Session) get(msg *ClientComMessage) {
// Expand topic name.
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
msg.MetaWhat = parseMsgClientMeta(msg.Get.What)
sub := s.getSub(msg.RcptTo)
if msg.MetaWhat == 0 {
s.queueOut(ErrMalformedReply(msg, msg.Timestamp))
logs.Warn.Println("s.get: invalid Get message action", msg.Get.What)
} else if sub != nil {
select {
case sub.meta <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.get: sub.meta channel full, topic ", msg.RcptTo, s.sid)
}
} else if msg.MetaWhat&(constMsgMetaDesc|constMsgMetaSub) != 0 {
// Request some minimal info from a topic not currently attached to.
select {
case globals.hub.meta <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.get: hub.meta channel full", s.sid)
}
} else {
logs.Warn.Println("s.get: subscribe first to get=", msg.Get.What)
s.queueOut(ErrPermissionDeniedReply(msg, msg.Timestamp))
}
}
func (s *Session) set(msg *ClientComMessage) {
// Expand topic name.
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
if msg.Set.Desc != nil {
msg.MetaWhat = constMsgMetaDesc
}
if msg.Set.Sub != nil {
msg.MetaWhat |= constMsgMetaSub
}
if msg.Set.Tags != nil {
msg.MetaWhat |= constMsgMetaTags
}
if msg.Set.Cred != nil {
msg.MetaWhat |= constMsgMetaCred
}
if msg.Set.Aux != nil {
msg.MetaWhat |= constMsgMetaAux
}
if msg.MetaWhat == 0 {
s.queueOut(ErrMalformedReply(msg, msg.Timestamp))
logs.Warn.Println("s.set: nil Set action")
} else if sub := s.getSub(msg.RcptTo); sub != nil {
select {
case sub.meta <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.set: sub.meta channel full, topic ", msg.RcptTo, s.sid)
}
} else if msg.MetaWhat&(constMsgMetaTags|constMsgMetaCred|constMsgMetaAux) != 0 {
logs.Warn.Println("s.set: setting tags/creds/aux is allowed for subscribed topics only", msg.MetaWhat)
s.queueOut(ErrPermissionDeniedReply(msg, msg.Timestamp))
} else {
// Desc.Private and Sub updates are possible without the subscription.
select {
case globals.hub.meta <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.set: hub.meta channel full", s.sid)
}
}
}
func (s *Session) del(msg *ClientComMessage) {
msg.MetaWhat = parseMsgClientDel(msg.Del.What)
// Delete user
if msg.MetaWhat == constMsgDelUser {
replyDelUser(s, msg)
return
}
// Delete something other than user: topic, subscription, message(s)
// Expand topic name and validate request.
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
s.queueOut(resp)
return
}
if msg.MetaWhat == 0 {
s.queueOut(ErrMalformedReply(msg, msg.Timestamp))
logs.Warn.Println("s.del: invalid Del action", msg.Del.What, s.sid)
return
}
if msg.MetaWhat == constMsgDelTopic {
// Deleting topic: for sessions attached or not attached, send request to hub first.
// Hub will forward to topic, if appropriate.
select {
case globals.hub.unreg <- &topicUnreg{
tenantID: s.tenantID,
rcptTo: msg.RcptTo,
pkt: msg,
sess: s,
del: true,
}:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.del: hub.unreg channel full", s.sid)
}
} else if sub := s.getSub(msg.RcptTo); sub != nil {
// Session is attached, deleting subscription or messages. Send to topic.
select {
case sub.meta <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.del: sub.meta channel full, topic ", msg.RcptTo, s.sid)
}
} else {
// Must join the topic to delete messages or subscriptions.
s.queueOut(ErrAttachFirst(msg, msg.Timestamp))
logs.Warn.Println("s.del: invalid Del action while unsubbed", msg.Del.What, s.sid)
}
}
// Broadcast a transient message to active topic subscribers.
// Not reporting any errors.
func (s *Session) note(msg *ClientComMessage) {
if s.ver == 0 || msg.AsUser == "" {
// Silently ignore the message: have not received {hi} or don't know who sent the message.
return
}
// Expand topic name and validate request.
var resp *ServerComMessage
msg.RcptTo, resp = s.expandTopicName(msg)
if resp != nil {
// Silently ignoring the message
return
}
switch msg.Note.What {
case "data":
if msg.Note.Payload == nil {
// Payload must be present in 'data' notifications.
return
}
case "kp", "kpa", "kpv":
if msg.Note.SeqId != 0 {
return
}
case "call":
if types.GetTopicCat(msg.RcptTo) != types.TopicCatP2P {
// Calls are only available in P2P topics.
return
}
fallthrough
case "read", "recv":
if msg.Note.SeqId <= 0 {
return
}
default:
return
}
if sub := s.getSub(msg.RcptTo); sub != nil {
// Pings can be sent to subscribed topics only
select {
case sub.broadcast <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.note: sub.broacast channel full, topic ", msg.RcptTo, s.sid)
}
} else if msg.Note.What == "recv" || (msg.Note.What == "call" && isDetachedCallEvent(msg.Note.Event)) {
// One of the following events happened:
// 1. Client received a pres notification about a new message, initiated a fetch
// from the server (and detached from the topic) and acknowledges receipt.
// 2. Client is either accepting or terminating the current video call or
// letting the initiator of the call know that it is ringing/notifying
// the user about the call.
//
// Hub will forward to topic, if appropriate.
if msg.Note.What == "call" && globals.cluster != nil &&
globals.cluster.isRemoteTopic(types.TopicKey{TenantID: msg.TenantID, Topic: msg.RcptTo}) {
if err := globals.cluster.routeToTopicMaster(ProxyReqCall, msg, msg.RcptTo, s); err != nil {
logs.Warn.Printf("s.note: failed routing detached call event to topic master topic=%s sid=%s err=%v", msg.RcptTo, s.sid, err)
}
return
}
select {
case globals.hub.routeCli <- msg:
default:
// Reply with a 503 to the user.
s.queueOut(ErrServiceUnavailableReply(msg, msg.Timestamp))
logs.Err.Println("s.note: hub.route channel full", s.sid)
}
} else {
s.queueOut(ErrAttachFirst(msg, msg.Timestamp))
logs.Warn.Println("s.note: note to invalid topic - must subscribe first", msg.Note.What, s.sid)
}
}
// expandTopicName expands session specific topic name to global name
// Returns
//
// topic: session-specific topic name the message recipient should see
// routeTo: routable global topic name
// err: *ServerComMessage with an error to return to the sender
func (s *Session) expandTopicName(msg *ClientComMessage) (string, *ServerComMessage) {
if msg.Original == "" {
logs.Warn.Println("s.etn: empty topic name", s.sid)
return "", ErrMalformed(msg.Id, "", msg.Timestamp)
}
// Expanded name of the topic to route to i.e. rcptto: or s.subs[routeTo]
var routeTo string
if msg.Original == "me" {
routeTo = msg.AsUser
} else if msg.Original == "fnd" {
routeTo = types.ParseUserId(msg.AsUser).FndName()
} else if msg.Original == "slf" {
routeTo = types.ParseUserId(msg.AsUser).SlfName()
} else if strings.HasPrefix(msg.Original, "usr") {
// p2p topic
uid1 := types.ParseUserId(msg.AsUser)
uid2 := types.ParseUserId(msg.Original)
if uid2.IsZero() {
// Ensure the user id is valid.
logs.Warn.Println("s.etn: failed to parse p2p topic name", s.sid)
return "", ErrMalformed(msg.Id, msg.Original, msg.Timestamp)
} else if uid2 == uid1 {
// Use 'me' to access self-topic.
logs.Warn.Println("s.etn: invalid p2p self-subscription", s.sid)
return "", ErrPermissionDeniedReply(msg, msg.Timestamp)
}
routeTo = uid1.P2PName(uid2)
} else if tmp := types.ChnToGrp(msg.Original); tmp != "" {
routeTo = tmp
} else {
routeTo = msg.Original
}
return routeTo, nil
}
func (s *Session) serializeAndUpdateStats(msg *ServerComMessage) any {
dataSize, data := s.serialize(msg)
if dataSize >= 0 {
statsAddHistSample("OutgoingMessageSize", float64(dataSize))
}
return data
}
func (s *Session) serialize(msg *ServerComMessage) (int, any) {
if s.proto == GRPC {
msg := pbServSerialize(msg)
// TODO: calculate and return the size of `msg`.
return -1, msg
}
if s.isMultiplex() {
// No need to serialize the message to bytes within the cluster.
return -1, msg
}
out, _ := json.Marshal(msg)
return len(out), out
}
// onBackgroundTimer marks background session as foreground and informs topics it's subscribed to.
func (s *Session) onBackgroundTimer() {
s.subsLock.RLock()
defer s.subsLock.RUnlock()
update := &sessionUpdate{sess: s}
for _, sub := range s.subs {
if sub.supd != nil {
sub.supd <- update
}
}
}