| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829 |
- package toc
- import (
- "bytes"
- "context"
- "encoding/base64"
- "encoding/csv"
- "encoding/hex"
- "errors"
- "fmt"
- "log/slog"
- "net/url"
- "strconv"
- "strings"
- "sync"
- "time"
- "github.com/google/uuid"
- "github.com/mk6i/open-oscar-server/config"
- "github.com/mk6i/open-oscar-server/state"
- "github.com/mk6i/open-oscar-server/wire"
- )
- // NewChatRegistry creates a new ChatRegistry instances.
- func NewChatRegistry() *ChatRegistry {
- chatRegistry := &ChatRegistry{
- lookup: make(map[int]wire.ICBMRoomInfo),
- sessions: make(map[int]*state.SessionInstance),
- m: sync.RWMutex{},
- }
- return chatRegistry
- }
- // ChatRegistry manages the chat rooms that a user is connected to during a TOC
- // session. It maintains mappings between chat room identifiers, metadata, and
- // active chat sessions.
- //
- // This struct provides thread-safe operations for adding, retrieving, and managing
- // chat room metadata and associated sessions.
- type ChatRegistry struct {
- lookup map[int]wire.ICBMRoomInfo // Maps chat room IDs to their metadata.
- sessions map[int]*state.SessionInstance // Tracks active chat sessions by chat room ID.
- nextID int // Incremental identifier for newly added chat rooms.
- m sync.RWMutex // Synchronization primitive for concurrent access.
- }
- // Add registers metadata for a newly joined chat room and returns a unique
- // identifier for it. If the room is already registered, it returns the existing ID.
- func (c *ChatRegistry) Add(room wire.ICBMRoomInfo) int {
- c.m.Lock()
- defer c.m.Unlock()
- for chatID, r := range c.lookup {
- if r == room {
- return chatID
- }
- }
- id := c.nextID
- c.lookup[id] = room
- c.nextID++
- return id
- }
- // LookupRoom retrieves metadata for the chat room registered with chatID.
- // It returns the room metadata and a boolean indicating whether the chat ID
- // was found.
- func (c *ChatRegistry) LookupRoom(chatID int) (wire.ICBMRoomInfo, bool) {
- c.m.RLock()
- defer c.m.RUnlock()
- room, found := c.lookup[chatID]
- return room, found
- }
- // RegisterSess associates a chat session with a chat room. If a session is
- // already registered for the given chat ID, it will be overwritten.
- func (c *ChatRegistry) RegisterSess(chatID int, instance *state.SessionInstance) {
- c.m.Lock()
- defer c.m.Unlock()
- c.sessions[chatID] = instance
- }
- // RetrieveSess retrieves the chat session associated with the given chat ID.
- // If no session is registered for the chat ID, it returns nil.
- func (c *ChatRegistry) RetrieveSess(chatID int) *state.SessionInstance {
- c.m.RLock()
- defer c.m.RUnlock()
- return c.sessions[chatID]
- }
- // RemoveSess removes a chat session.
- func (c *ChatRegistry) RemoveSess(chatID int) {
- c.m.Lock()
- defer c.m.Unlock()
- delete(c.sessions, chatID)
- }
- // Sessions retrieves all the chat sessions.
- func (c *ChatRegistry) Sessions() []*state.SessionInstance {
- c.m.RLock()
- defer c.m.RUnlock()
- sessions := make([]*state.SessionInstance, 0, len(c.sessions))
- for _, s := range c.sessions {
- sessions = append(sessions, s)
- }
- return sessions
- }
- // OSCARProxy acts as a bridge between TOC clients and the OSCAR server,
- // translating protocol messages between the two.
- //
- // It performs the following functions:
- // - Receives TOC messages from the client, converts them into SNAC messages,
- // and forwards them to the OSCAR server. The SNAC response is then converted
- // back into a TOC response for the client.
- // - Receives incoming messages from the OSCAR server and translates them into
- // TOC responses for the client.
- type OSCARProxy struct {
- AdminService AdminService
- AuthService AuthService
- BuddyListRegistry BuddyListRegistry
- BuddyService BuddyService
- ChatNavService ChatNavService
- ChatService ChatService
- CookieBaker CookieBaker
- DirSearchService DirSearchService
- ICBMService ICBMService
- LocateService LocateService
- Logger *slog.Logger
- OServiceService OServiceService
- PermitDenyService PermitDenyService
- TOCConfigStore TOCConfigStore
- SessionRetriever SessionRetriever
- SNACRateLimits wire.SNACRateLimits
- HTTPIPRateLimiter *IPRateLimiter
- }
- // RecvClientCmd processes a client TOC command and returns a server reply.
- //
- // * sessBOS is the current user's session.
- // * chatRegistry manages the current user's chat sessions
- // * payload is the command + arguments
- // * toCh is the channel that transports messages to client
- // * doAsync performs async tasks, is auto-cleaned up by caller
- //
- // It returns true if the server can continue processing commands.
- func (s OSCARProxy) RecvClientCmd(
- ctx context.Context,
- sessBOS *state.SessionInstance,
- chatRegistry *ChatRegistry,
- payload []byte,
- toCh chan<- []byte,
- doAsync func(f func() error),
- ) (reply string) {
- cmd := payload
- var args []byte
- if idx := bytes.IndexByte(payload, ' '); idx > -1 {
- cmd, args = payload[:idx], payload[idx:]
- }
- if s.Logger.Enabled(ctx, slog.LevelDebug) {
- s.Logger.DebugContext(ctx, "client request", "command", payload)
- } else {
- s.Logger.InfoContext(ctx, "client request", "command", cmd)
- }
- switch string(cmd) {
- case "toc_send_im":
- return s.SendIM(ctx, sessBOS, args)
- case "toc_init_done":
- return s.InitDone(ctx, sessBOS)
- case "toc_add_buddy":
- return s.AddBuddy(ctx, sessBOS, args)
- case "toc_get_status":
- return s.GetStatus(ctx, sessBOS, args)
- case "toc_remove_buddy":
- return s.RemoveBuddy(ctx, sessBOS, args)
- case "toc_add_permit":
- return s.AddPermit(ctx, sessBOS, args)
- case "toc_add_deny":
- return s.AddDeny(ctx, sessBOS, args)
- case "toc_set_away":
- return s.SetAway(ctx, sessBOS, args)
- case "toc_set_caps":
- return s.SetCaps(ctx, sessBOS, args)
- case "toc_evil":
- return s.Evil(ctx, sessBOS, args)
- case "toc_get_info":
- return s.GetInfoURL(ctx, sessBOS, args)
- case "toc_change_passwd":
- return s.ChangePassword(ctx, sessBOS, args)
- case "toc_format_nickname":
- return s.FormatNickname(ctx, sessBOS, args)
- case "toc_chat_join", "toc_chat_accept":
- var chatID int
- var msg string
- if string(cmd) == "toc_chat_join" {
- chatID, msg = s.ChatJoin(ctx, sessBOS, chatRegistry, args)
- } else {
- chatID, msg = s.ChatAccept(ctx, sessBOS, chatRegistry, args)
- }
- if msg == cmdInternalSvcErr {
- return msg
- }
- doAsync(func() error {
- sess := chatRegistry.RetrieveSess(chatID)
- s.RecvChat(ctx, sess, chatID, toCh)
- return nil
- })
- return msg
- case "toc_chat_send":
- return s.ChatSend(ctx, chatRegistry, args)
- case "toc_chat_whisper":
- return s.ChatWhisper(ctx, chatRegistry, args)
- case "toc_chat_leave":
- return s.ChatLeave(ctx, chatRegistry, args)
- case "toc_set_info":
- return s.SetInfo(ctx, sessBOS, args)
- case "toc_set_dir":
- return s.SetDir(ctx, sessBOS, args)
- case "toc_set_idle":
- return s.SetIdle(ctx, sessBOS, args)
- case "toc_set_config":
- return s.SetConfig(ctx, sessBOS, args)
- case "toc_chat_invite":
- return s.ChatInvite(ctx, sessBOS, chatRegistry, args)
- case "toc_dir_search":
- return s.GetDirSearchURL(ctx, sessBOS, args)
- case "toc_get_dir":
- return s.GetDirURL(ctx, sessBOS, args)
- case "toc_rvous_accept":
- return s.RvousAccept(ctx, sessBOS, args)
- case "toc_rvous_cancel":
- return s.RvousCancel(ctx, sessBOS, args)
- }
- s.Logger.ErrorContext(ctx, fmt.Sprintf("unsupported TOC command %s", cmd))
- return cmdInternalSvcErr
- }
- // AddBuddy handles the toc_add_buddy TOC command.
- //
- // From the TiK documentation:
- //
- // Add buddies to your buddy list. This does not change your saved config.
- //
- // Command syntax: toc_add_buddy <Buddy User 1> [<Buddy User2> [<Buddy User 3> [...]]]
- func (s OSCARProxy) AddBuddy(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.Buddy, wire.BuddyAddBuddies); isLimited {
- return msg
- }
- users, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- snac := wire.SNAC_0x03_0x04_BuddyAddBuddies{}
- for _, sn := range users {
- snac.Buddies = append(snac.Buddies, struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{ScreenName: sn})
- }
- if err := s.BuddyService.AddBuddies(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("BuddyService.AddBuddies: %w", err))
- }
- return ""
- }
- // AddPermit handles the toc_add_permit TOC command.
- //
- // From the TiK documentation:
- //
- // ADD the following people to your permit mode. If you are in deny mode it
- // will switch you to permit mode first. With no arguments and in deny mode
- // this will switch you to permit none. If already in permit mode, no
- // arguments does nothing and your permit list remains the same.
- //
- // Command syntax: toc_add_permit [ <User 1> [<User 2> [...]]]
- func (s OSCARProxy) AddPermit(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.PermitDeny, wire.PermitDenyAddDenyListEntries); isLimited {
- return msg
- }
- users, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- snac := wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{}
- for _, sn := range users {
- snac.Users = append(snac.Users, struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{ScreenName: sn})
- }
- if err := s.PermitDenyService.AddPermListEntries(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("PermitDenyService.AddPermListEntries: %w", err))
- }
- return ""
- }
- // AddDeny handles the toc_add_deny TOC command.
- //
- // From the TiK documentation:
- //
- // ADD the following people to your deny mode. If you are in permit mode it
- // will switch you to deny mode first. With no arguments and in permit mode,
- // this will switch you to deny none. If already in deny mode, no arguments
- // does nothing and your deny list remains unchanged.
- //
- // Command syntax: toc_add_deny [ <User 1> [<User 2> [...]]]
- func (s OSCARProxy) AddDeny(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.PermitDeny, wire.PermitDenyAddDenyListEntries); isLimited {
- return msg
- }
- users, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- snac := wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{}
- for _, sn := range users {
- snac.Users = append(snac.Users, struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{ScreenName: sn})
- }
- if err := s.PermitDenyService.AddDenyListEntries(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("PermitDenyService.AddDenyListEntries: %w", err))
- }
- return ""
- }
- // ChangePassword handles the toc_change_passwd TOC command.
- //
- // From the TiK documentation:
- //
- // Change a user's password. An ADMIN_PASSWD_STATUS or ERROR message will be
- // sent back to the client.
- //
- // Command syntax: toc_change_passwd <existing_passwd> <new_passwd>
- func (s OSCARProxy) ChangePassword(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.Admin, wire.AdminInfoChangeRequest); isLimited {
- return msg
- }
- var oldPass, newPass string
- if _, err := parseArgs(args, &oldPass, &newPass); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- oldPass = unescape(oldPass)
- newPass = unescape(newPass)
- reqSNAC := wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVOldPassword, oldPass),
- wire.NewTLVBE(wire.AdminTLVNewPassword, newPass),
- },
- },
- }
- reply, err := s.AdminService.InfoChangeRequest(ctx, me, wire.SNACFrame{}, reqSNAC)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("AdminService.InfoChangeRequest: %w", err))
- }
- replyBody, ok := reply.Body.(wire.SNAC_0x07_0x05_AdminChangeReply)
- if !ok {
- return s.runtimeErr(ctx, fmt.Errorf("AdminService.InfoChangeRequest: unexpected response type %v", replyBody))
- }
- code, ok := replyBody.Uint16BE(wire.AdminTLVErrorCode)
- if ok {
- switch code {
- case wire.AdminInfoErrorInvalidPasswordLength:
- return "ERROR:911"
- case wire.AdminInfoErrorValidatePassword:
- return "ERROR:912"
- default:
- return "ERROR:913"
- }
- }
- return "ADMIN_PASSWD_STATUS:0"
- }
- // ChatAccept handles the toc_chat_accept TOC command.
- //
- // From the TiK documentation:
- //
- // Accept a CHAT_INVITE message from TOC. The server will send a CHAT_JOIN in
- // response.
- //
- // Command syntax: toc_chat_accept <Chat Room ID>
- func (s OSCARProxy) ChatAccept(
- ctx context.Context,
- me *state.SessionInstance,
- chatRegistry *ChatRegistry,
- args []byte,
- ) (int, string) {
- var chatIDStr string
- if _, err := parseArgs(args, &chatIDStr); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- chatID, err := strconv.Atoi(chatIDStr)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- chatInfo, found := chatRegistry.LookupRoom(chatID)
- if !found {
- return 0, s.runtimeErr(ctx, fmt.Errorf("chatRegistry.LookupRoom: no chat found for ID %d", chatID))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.ChatNav, wire.ChatNavRequestRoomInfo); isLimited {
- return 0, msg
- }
- reqRoomSNAC := wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
- Cookie: chatInfo.Cookie,
- Exchange: chatInfo.Exchange,
- InstanceNumber: chatInfo.Instance,
- }
- reqRoomReply, err := s.ChatNavService.RequestRoomInfo(ctx, wire.SNACFrame{}, reqRoomSNAC)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("ChatNavService.RequestRoomInfo: %w", err))
- }
- reqRoomReplyBody, ok := reqRoomReply.Body.(wire.SNAC_0x0D_0x09_ChatNavNavInfo)
- if !ok {
- return 0, s.runtimeErr(
- ctx,
- fmt.Errorf("chatNavService.RequestRoomInfo: unexpected response type %v", reqRoomReplyBody),
- )
- }
- b, hasInfo := reqRoomReplyBody.Bytes(wire.ChatNavTLVRoomInfo)
- if !hasInfo {
- return 0, s.runtimeErr(ctx, errors.New("reqRoomReplyBody.Bytes: missing wire.ChatNavTLVRoomInfo"))
- }
- roomInfo := wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{}
- if err := wire.UnmarshalBE(&roomInfo, bytes.NewReader(b)); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("wire.UnmarshalBE: %w", err))
- }
- roomName, hasName := roomInfo.Bytes(wire.ChatRoomTLVRoomName)
- if !hasName {
- return 0, s.runtimeErr(ctx, errors.New("roomInfo.Bytes: missing wire.ChatRoomTLVRoomName"))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.OService, wire.OServiceServiceRequest); isLimited {
- return 0, msg
- }
- svcReqSNAC := wire.SNAC_0x01_0x04_OServiceServiceRequest{
- FoodGroup: wire.Chat,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, wire.SNAC_0x01_0x04_TLVRoomInfo{
- Cookie: chatInfo.Cookie,
- }),
- },
- },
- }
- svcReqReply, err := s.OServiceService.ServiceRequest(ctx, wire.BOS, me, wire.SNACFrame{}, svcReqSNAC, config.Listener{})
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("OServiceServiceBOS.ServiceRequest: %w", err))
- }
- svcReqReplyBody, ok := svcReqReply.Body.(wire.SNAC_0x01_0x05_OServiceServiceResponse)
- if !ok {
- return 0, s.runtimeErr(
- ctx,
- fmt.Errorf("OServiceServiceBOS.ServiceRequest: unexpected response type %v", svcReqReplyBody),
- )
- }
- loginCookie, hasCookie := svcReqReplyBody.Bytes(wire.OServiceTLVTagsLoginCookie)
- if !hasCookie {
- return 0, s.runtimeErr(ctx, errors.New("missing wire.OServiceTLVTagsLoginCookie"))
- }
- // todo: naming for cookie: login cookie, server cookie, or auth cookie?
- serverCookie, err := s.AuthService.CrackCookie(loginCookie)
- chatSess, err := s.AuthService.RegisterChatSession(ctx, serverCookie)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("AuthService.RegisterChatSession: %w", err))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.OService, wire.OServiceClientOnline); isLimited {
- return 0, msg
- }
- if err := s.OServiceService.ClientOnline(ctx, wire.Chat, wire.SNAC_0x01_0x02_OServiceClientOnline{}, chatSess); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("OServiceServiceChat.ClientOnline: %w", err))
- }
- chatRegistry.RegisterSess(chatID, chatSess)
- return chatID, fmt.Sprintf("CHAT_JOIN:%d:%s", chatID, roomName)
- }
- // ChatInvite handles the toc_chat_invite TOC command.
- //
- // From the TiK documentation:
- //
- // Once you are inside a chat room you can invite other people into that room.
- // Remember to quote and encode the invite message.
- //
- // Command syntax: toc_chat_invite <Chat Room ID> <Invite Msg> <buddy1> [<buddy2> [<buddy3> [...]]]
- func (s OSCARProxy) ChatInvite(ctx context.Context, me *state.SessionInstance, chatRegistry *ChatRegistry, args []byte) string {
- var chatRoomIDStr, msg string
- users, err := parseArgs(args, &chatRoomIDStr, &msg)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- msg = unescape(msg)
- chatID, err := strconv.Atoi(chatRoomIDStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- roomInfo, found := chatRegistry.LookupRoom(chatID)
- if !found {
- return s.runtimeErr(ctx, fmt.Errorf("chatRegistry.LookupRoom: chat ID `%d` not found", chatID))
- }
- for _, guest := range users {
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.ICBM, wire.ICBMChannelMsgToHost); isLimited {
- return msg
- }
- snac := wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ChannelID: wire.ICBMChannelRendezvous,
- ScreenName: guest,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
- Type: wire.ICBMRdvMessagePropose,
- Capability: wire.CapChat,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMRdvTLVTagsSeqNum, uint16(1)),
- wire.NewTLVBE(wire.ICBMRdvTLVTagsInvitation, msg),
- wire.NewTLVBE(wire.ICBMRdvTLVTagsInviteMIMECharset, "us-ascii"),
- wire.NewTLVBE(wire.ICBMRdvTLVTagsInviteMIMELang, "en"),
- wire.NewTLVBE(wire.ICBMRdvTLVTagsSvcData, roomInfo),
- },
- },
- }),
- },
- },
- }
- if _, err := s.ICBMService.ChannelMsgToHost(ctx, me, wire.SNACFrame{}, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ICBMService.ChannelMsgToHost: %w", err))
- }
- }
- return ""
- }
- // ChatJoin handles the toc_chat_join TOC command.
- //
- // From the TiK documentation:
- //
- // Join a chat room in the given exchange. Exchange is an integer that
- // represents a group of chat rooms. Different exchanges have different
- // properties. For example some exchanges might have room replication (ie a
- // room never fills up, there are just multiple instances.) and some exchanges
- // might have navigational information. Currently, exchange should always be
- // 4, however this may change in the future. You will either receive an ERROR
- // if the room couldn't be joined or a CHAT_JOIN message. The Chat Room Name
- // is case-insensitive and consecutive spaces are removed.
- //
- // Command syntax: toc_chat_join <Exchange> <Chat Room Name>
- func (s OSCARProxy) ChatJoin(
- ctx context.Context,
- me *state.SessionInstance,
- chatRegistry *ChatRegistry,
- args []byte,
- ) (int, string) {
- var exchangeStr, roomName string
- if _, err := parseArgs(args, &exchangeStr, &roomName); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- roomName = unescape(roomName)
- // create room or retrieve the room if it already exists
- exchange, err := strconv.Atoi(exchangeStr)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.Chat, wire.ChatRoomInfoUpdate); isLimited {
- return 0, msg
- }
- mkRoomReq := wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
- Exchange: uint16(exchange),
- Cookie: "create",
- TLVBlock: wire.TLVBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ChatRoomTLVRoomName, roomName),
- },
- },
- }
- mkRoomReply, err := s.ChatNavService.CreateRoom(ctx, me, wire.SNACFrame{}, mkRoomReq)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("ChatNavService.CreateRoom: %w", err))
- }
- mkRoomReplyBody, ok := mkRoomReply.Body.(wire.SNAC_0x0D_0x09_ChatNavNavInfo)
- if !ok {
- return 0, s.runtimeErr(
- ctx,
- fmt.Errorf("chatNavService.CreateRoom: unexpected response type %v", mkRoomReplyBody),
- )
- }
- buf, ok := mkRoomReplyBody.Bytes(wire.ChatNavTLVRoomInfo)
- if !ok {
- return 0, s.runtimeErr(ctx, errors.New("mkRoomReplyBody.Bytes: missing wire.ChatNavTLVRoomInfo"))
- }
- inBody := wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{}
- if err := wire.UnmarshalBE(&inBody, bytes.NewReader(buf)); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("wire.UnmarshalBE: %w", err))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.OService, wire.OServiceServiceRequest); isLimited {
- return 0, msg
- }
- svcReqSNAC := wire.SNAC_0x01_0x04_OServiceServiceRequest{
- FoodGroup: wire.Chat,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, wire.SNAC_0x01_0x04_TLVRoomInfo{
- Cookie: inBody.Cookie,
- }),
- },
- },
- }
- svcReqReply, err := s.OServiceService.ServiceRequest(ctx, wire.BOS, me, wire.SNACFrame{}, svcReqSNAC, config.Listener{})
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("OServiceServiceBOS.ServiceRequest: %w", err))
- }
- svcReqReplyBody, ok := svcReqReply.Body.(wire.SNAC_0x01_0x05_OServiceServiceResponse)
- if !ok {
- return 0, s.runtimeErr(
- ctx,
- fmt.Errorf("OServiceServiceBOS.ServiceRequest: unexpected response type %v", svcReqReplyBody),
- )
- }
- loginCookie, hasCookie := svcReqReplyBody.Bytes(wire.OServiceTLVTagsLoginCookie)
- if !hasCookie {
- return 0, s.runtimeErr(ctx, errors.New("svcReqReplyBody.Bytes: missing wire.OServiceTLVTagsLoginCookie"))
- }
- // todo: naming for cookie: login cookie, server cookie, or auth cookie?
- serverCookie, err := s.AuthService.CrackCookie(loginCookie)
- chatSess, err := s.AuthService.RegisterChatSession(ctx, serverCookie)
- if err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("AuthService.RegisterChatSession: %w", err))
- }
- if msg, isLimited := s.checkRateLimit(ctx, me, wire.OService, wire.OServiceClientOnline); isLimited {
- return 0, msg
- }
- if err := s.OServiceService.ClientOnline(ctx, wire.Chat, wire.SNAC_0x01_0x02_OServiceClientOnline{}, chatSess); err != nil {
- return 0, s.runtimeErr(ctx, fmt.Errorf("OServiceServiceChat.ClientOnline: %w", err))
- }
- roomInfo := wire.ICBMRoomInfo{
- Exchange: inBody.Exchange,
- Cookie: inBody.Cookie,
- Instance: inBody.InstanceNumber,
- }
- chatID := chatRegistry.Add(roomInfo)
- chatRegistry.RegisterSess(chatID, chatSess)
- return chatID, fmt.Sprintf("CHAT_JOIN:%d:%s", chatID, roomName)
- }
- // ChatLeave handles the toc_chat_leave TOC command.
- //
- // From the TiK documentation:
- //
- // Leave the chat room.
- //
- // Command syntax: toc_chat_leave <Chat Room ID>
- func (s OSCARProxy) ChatLeave(ctx context.Context, chatRegistry *ChatRegistry, args []byte) string {
- var chatIDStr string
- if _, err := parseArgs(args, &chatIDStr); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- chatID, err := strconv.Atoi(chatIDStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- me := chatRegistry.RetrieveSess(chatID)
- if me == nil {
- return s.runtimeErr(ctx, fmt.Errorf("chatRegistry.RetrieveSess: chat session `%d` not found", chatID))
- }
- s.AuthService.SignoutChat(ctx, me)
- me.CloseInstance() // stop async server SNAC reply handler for this chat room
- chatRegistry.RemoveSess(chatID)
- return fmt.Sprintf("CHAT_LEFT:%d", chatID)
- }
- // ChatSend handles the toc_chat_send TOC command.
- //
- // From the TiK documentation:
- //
- // Send a message in a chat room using the chat room id from CHAT_JOIN. Since
- // reflection is always on in TOC, you do not need to add the message to your
- // chat UI, since you will get a CHAT_IN with the message. Remember to quote
- // and encode the message.
- //
- // Command syntax: toc_chat_send <Chat Room ID> <Message>
- func (s OSCARProxy) ChatSend(ctx context.Context, chatRegistry *ChatRegistry, args []byte) string {
- var chatIDStr, msg string
- if _, err := parseArgs(args, &chatIDStr, &msg); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- msg = unescape(msg)
- chatID, err := strconv.Atoi(chatIDStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- me := chatRegistry.RetrieveSess(chatID)
- if me == nil {
- return s.runtimeErr(ctx, fmt.Errorf("chatRegistry.RetrieveSess: session for chat ID `%d` not found", chatID))
- }
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Chat, wire.ChatChannelMsgToHost); isLimited {
- return errMsg
- }
- block := wire.TLVRestBlock{}
- // the order of these TLVs matters for AIM 2.x. if out of order, screen
- // names do not appear with each chat message.
- block.Append(wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)))
- block.Append(wire.NewTLVBE(wire.ChatTLVSenderInformation, me.Session().TLVUserInfo()))
- block.Append(wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}))
- block.Append(wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ChatTLVMessageInfoText, msg),
- },
- }))
- snac := wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: wire.ICBMChannelMIME,
- TLVRestBlock: block,
- }
- reply, err := s.ChatService.ChannelMsgToHost(ctx, me, wire.SNACFrame{}, snac)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ChatService.ChannelMsgToHost: %w", err))
- }
- if reply == nil {
- return s.runtimeErr(ctx, errors.New("ChatService.ChannelMsgToHost: missing response "))
- }
- switch v := reply.Body.(type) {
- case wire.SNAC_0x0E_0x06_ChatChannelMsgToClient:
- msgInfo, ok := v.Bytes(wire.ChatTLVMessageInfo)
- if !ok {
- return s.runtimeErr(ctx, errors.New("ChatService.ChannelMsgToHost: missing wire.ChatTLVMessageInfo"))
- }
- reflectMsg, err := wire.UnmarshalChatMessageText(msgInfo)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("wire.UnmarshalChatMessageText: %w", err))
- }
- senderInfo, ok := v.Bytes(wire.ChatTLVSenderInformation)
- if !ok {
- return s.runtimeErr(ctx, errors.New("ChatService.ChannelMsgToHost: missing wire.ChatTLVSenderInformation"))
- }
- var userInfo wire.TLVUserInfo
- if err := wire.UnmarshalBE(&userInfo, bytes.NewReader(senderInfo)); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("wire.UnmarshalBE: %w", err))
- }
- return fmt.Sprintf("CHAT_IN:%d:%s:F:%s", chatID, userInfo.ScreenName, reflectMsg)
- default:
- return s.runtimeErr(ctx, errors.New("ChatService.ChannelMsgToHost: unexpected response"))
- }
- }
- // ChatWhisper handles the toc_chat_send TOC command.
- //
- // From the TiK documentation:
- //
- // Send a message in a chat room using the chat room id from CHAT_JOIN.
- // This message is directed at only one person. (Currently you DO need to add
- // this to your UI.) Remember to quote and encode the message. Chat whispering
- // is different from IMs since it is linked to a chat room, and should usually
- // be displayed in the chat room UI.
- //
- // Command syntax: toc_chat_whisper <Chat Room ID> <dst_user> <Message>
- func (s OSCARProxy) ChatWhisper(ctx context.Context, chatRegistry *ChatRegistry, args []byte) string {
- var chatIDStr, recip, msg string
- if _, err := parseArgs(args, &chatIDStr, &recip, &msg); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- msg = unescape(msg)
- chatID, err := strconv.Atoi(chatIDStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- me := chatRegistry.RetrieveSess(chatID)
- if me == nil {
- return s.runtimeErr(ctx, fmt.Errorf("chatRegistry.RetrieveSess: session for chat ID `%d` not found", chatID))
- }
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Chat, wire.ChatChannelMsgToHost); isLimited {
- return errMsg
- }
- block := wire.TLVRestBlock{}
- block.Append(wire.NewTLVBE(wire.ChatTLVSenderInformation, me.Session().TLVUserInfo()))
- block.Append(wire.NewTLVBE(wire.ChatTLVWhisperToUser, recip))
- block.Append(wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ChatTLVMessageInfoText, msg),
- },
- }))
- snac := wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: wire.ICBMChannelMIME,
- TLVRestBlock: block,
- }
- if _, err = s.ChatService.ChannelMsgToHost(ctx, me, wire.SNACFrame{}, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ChatService.ChannelMsgToHost: %w", err))
- }
- return ""
- }
- // Evil handles the toc_evil TOC command.
- //
- // From the TiK documentation:
- //
- // Evil/Warn someone else. The 2nd argument is either the string "norm" for a
- // normal warning, or "anon" for an anonymous warning. You can only evil
- // people who have recently sent you ims. The higher someones evil level, the
- // slower they can send message.
- //
- // Command syntax: toc_evil <User> <norm|anon>
- func (s OSCARProxy) Evil(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.ICBM, wire.ICBMEvilRequest); isLimited {
- return errMsg
- }
- var user, scope string
- if _, err := parseArgs(args, &user, &scope); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- snac := wire.SNAC_0x04_0x08_ICBMEvilRequest{
- ScreenName: user,
- }
- switch scope {
- case "anon":
- snac.SendAs = 1
- case "norm":
- snac.SendAs = 0
- default:
- return s.runtimeErr(ctx, fmt.Errorf("incorrect warning type `%s`. allowed values: anon, norm", scope))
- }
- response, err := s.ICBMService.EvilRequest(ctx, me, wire.SNACFrame{}, snac)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ICBMService.EvilRequest: %w", err))
- }
- switch v := response.Body.(type) {
- case wire.SNAC_0x04_0x09_ICBMEvilReply:
- return ""
- case wire.SNACError:
- s.Logger.InfoContext(ctx, "unable to warn user", "code", v.Code)
- default:
- return s.runtimeErr(ctx, errors.New("unexpected response"))
- }
- return ""
- }
- // FormatNickname handles the toc_format_nickname TOC command.
- //
- // From the TiK documentation:
- //
- // Reformat a user's nickname. An ADMIN_NICK_STATUS or ERROR message will be
- // sent back to the client.
- //
- // Command syntax: toc_format_nickname <new_format>
- func (s OSCARProxy) FormatNickname(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Admin, wire.AdminInfoChangeRequest); isLimited {
- return errMsg
- }
- var newFormat string
- if _, err := parseArgs(args, &newFormat); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- // remove curly braces added by TiK
- newFormat = strings.Trim(newFormat, "{}")
- reqSNAC := wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, newFormat),
- },
- },
- }
- reply, err := s.AdminService.InfoChangeRequest(ctx, me, wire.SNACFrame{}, reqSNAC)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("AdminService.InfoChangeRequest: %w", err))
- }
- replyBody, ok := reply.Body.(wire.SNAC_0x07_0x05_AdminChangeReply)
- if !ok {
- return s.runtimeErr(ctx, fmt.Errorf("AdminService.InfoChangeRequest: unexpected response type %v", replyBody))
- }
- code, ok := replyBody.Uint16BE(wire.AdminTLVErrorCode)
- if ok {
- switch code {
- case wire.AdminInfoErrorInvalidNickNameLength, wire.AdminInfoErrorInvalidNickName:
- return "ERROR:911"
- default:
- return "ERROR:913"
- }
- }
- return "ADMIN_NICK_STATUS:0"
- }
- // GetDirSearchURL handles the toc_dir_search TOC command.
- //
- // From the TiK documentation:
- //
- // Perform a search of the Oscar Directory, using colon separated fields as in:
- //
- // "first name":"middle name":"last name":"maiden name":"city":"state":"country":"email"
- //
- // You can search by keyword by setting search terms in the 11th position (this
- // feature is not in the TiK docs but is present in the code):
- //
- // ::::::::::"search kw"
- //
- // Returns either a GOTO_URL or ERROR msg.
- //
- // Command syntax: toc_dir_search <info information>
- func (s OSCARProxy) GetDirSearchURL(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if status := me.Session().EvaluateRateLimit(time.Now(), 1); status == wire.RateLimitStatusLimited {
- return rateLimitExceededErr
- }
- var info string
- if _, err := parseArgs(args, &info); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- info = unescape(info)
- params := strings.Split(info, ":")
- labels := []string{
- "first_name",
- "middle_name",
- "last_name",
- "maiden_name",
- "city",
- "state",
- "country",
- "email",
- "nop", // unused placeholder
- "nop",
- "keyword",
- }
- // map labels to param values at their corresponding positions
- p := url.Values{}
- for i, param := range params {
- if i >= len(labels) {
- break
- }
- if param != "" {
- p.Add(labels[i], strings.Trim(param, "\""))
- }
- }
- if len(p) == 0 {
- return s.runtimeErr(ctx, errors.New("no search fields found"))
- }
- cookie, err := s.newHTTPAuthToken(me.IdentScreenName())
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("newHTTPAuthToken: %w", err))
- }
- p.Add("cookie", cookie)
- return fmt.Sprintf("GOTO_URL:search results:dir_search?%s", p.Encode())
- }
- // GetDirURL handles the toc_get_dir TOC command.
- //
- // From the TiK documentation:
- //
- // Gets a user's dir info a GOTO_URL or ERROR message will be sent back to the client.
- //
- // Command syntax: toc_get_dir <username>
- func (s OSCARProxy) GetDirURL(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if status := me.Session().EvaluateRateLimit(time.Now(), 1); status == wire.RateLimitStatusLimited {
- return rateLimitExceededErr
- }
- var user string
- if _, err := parseArgs(args, &user); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- cookie, err := s.newHTTPAuthToken(me.IdentScreenName())
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("newHTTPAuthToken: %w", err))
- }
- p := url.Values{}
- p.Add("cookie", cookie)
- p.Add("user", user)
- return fmt.Sprintf("GOTO_URL:directory info:dir_info?%s", p.Encode())
- }
- // GetInfoURL handles the toc_get_info TOC command.
- //
- // From the TiK documentation:
- //
- // Gets a user's info a GOTO_URL or ERROR message will be sent back to the client.
- //
- // Command syntax: toc_get_info <username>
- func (s OSCARProxy) GetInfoURL(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if status := me.Session().EvaluateRateLimit(time.Now(), 1); status == wire.RateLimitStatusLimited {
- return rateLimitExceededErr
- }
- var user string
- if _, err := parseArgs(args, &user); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- cookie, err := s.newHTTPAuthToken(me.IdentScreenName())
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("newHTTPAuthToken: %w", err))
- }
- p := url.Values{}
- p.Add("cookie", cookie)
- p.Add("from", me.IdentScreenName().String())
- p.Add("user", user)
- return fmt.Sprintf("GOTO_URL:profile:info?%s", p.Encode())
- }
- // GetStatus handles the toc_get_status TOC command.
- //
- // From the TOC2 documentation:
- //
- // This useful command wasn't ever really documented. It returns either an
- // UPDATE_BUDDY message or an ERROR message depending on whether or not the
- // guy appears to be online.
- //
- // Command syntax: toc_get_status <screenname>
- func (s OSCARProxy) GetStatus(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Locate, wire.LocateUserInfoQuery); isLimited {
- return errMsg
- }
- var them string
- if _, err := parseArgs(args, &them); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- inBody := wire.SNAC_0x02_0x05_LocateUserInfoQuery{
- ScreenName: them,
- }
- info, err := s.LocateService.UserInfoQuery(ctx, me, wire.SNACFrame{}, inBody)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.UserInfoQuery: %w", err))
- }
- switch v := info.Body.(type) {
- case wire.SNACError:
- if v.Code == wire.ErrorCodeNotLoggedOn {
- return fmt.Sprintf("ERROR:901:%s", them)
- } else {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.UserInfoQuery error code: %d", v.Code))
- }
- case wire.SNAC_0x02_0x06_LocateUserInfoReply:
- return userInfoToUpdateBuddy(v.TLVUserInfo)
- default:
- return s.runtimeErr(ctx, fmt.Errorf("AdminService.InfoChangeRequest: unexpected response type %v", v))
- }
- }
- // InitDone handles the toc_init_done TOC command.
- //
- // From the TiK documentation:
- //
- // Tells TOC that we are ready to go online. TOC clients should first send TOC
- // the buddy list and any permit/deny lists. However, toc_init_done must be
- // called within 30 seconds after toc_signon, or the connection will be
- // dropped. Remember, it can't be called until after the SIGN_ON message is
- // received. Calling this before or multiple times after a SIGN_ON will cause
- // the connection to be dropped.
- //
- // Note: The business logic described in the last 3 sentences are not yet
- // implemented.
- //
- // Command syntax: toc_init_done
- func (s OSCARProxy) InitDone(ctx context.Context, instance *state.SessionInstance) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, instance, wire.OService, wire.OServiceClientOnline); isLimited {
- return errMsg
- }
- if err := s.OServiceService.ClientOnline(ctx, wire.BOS, wire.SNAC_0x01_0x02_OServiceClientOnline{}, instance); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("OServiceServiceBOS.ClientOnliney: %w", err))
- }
- return ""
- }
- // RemoveBuddy handles the toc_remove_buddy TOC command.
- //
- // From the TiK documentation:
- //
- // Remove buddies from your buddy list. This does not change your saved config.
- //
- // Command syntax: toc_remove_buddy <Buddy User 1> [<Buddy User2> [<Buddy User 3> [...]]]
- func (s OSCARProxy) RemoveBuddy(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Buddy, wire.BuddyDelBuddies); isLimited {
- return errMsg
- }
- users, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- snac := wire.SNAC_0x03_0x05_BuddyDelBuddies{}
- for _, sn := range users {
- snac.Buddies = append(snac.Buddies, struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{ScreenName: sn})
- }
- if err := s.BuddyService.DelBuddies(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("BuddyService.DelBuddies: %w", err))
- }
- return ""
- }
- // RvousAccept handles the toc_rvous_accept TOC command.
- //
- // From the TiK documentation:
- //
- // Accept a rendezvous proposal from the user <nick>. <cookie> is the cookie
- // from the RVOUS_PROPOSE message. <service> is the UUID the proposal was for.
- // <tlvlist> contains a list of tlv tags followed by base64 encoded values.
- //
- // Note: This method does not actually process the TLV list param, as it's not
- // passed in the TiK client, the reference implementation.
- //
- // Command syntax: toc_rvous_accept <nick> <cookie> <service>
- func (s OSCARProxy) RvousAccept(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.ICBM, wire.ICBMChannelMsgToHost); isLimited {
- return errMsg
- }
- var nick, cookie, service string
- if _, err := parseArgs(args, &nick, &cookie, &service); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- cbytes, err := base64.StdEncoding.DecodeString(cookie)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("base64.Decode: %w", err))
- }
- var arr [8]byte
- copy(arr[:], cbytes) // copy slice into array
- svcUUID, err := uuid.Parse(service)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("uuid.Parse: %w", err))
- }
- snac := wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ChannelID: wire.ICBMChannelRendezvous,
- ScreenName: nick,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
- Type: wire.ICBMRdvMessageAccept,
- Cookie: arr,
- Capability: svcUUID,
- }),
- },
- },
- }
- if _, err = s.ICBMService.ChannelMsgToHost(ctx, me, wire.SNACFrame{}, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ICBMService.ChannelMsgToHost: %w", err))
- }
- return ""
- }
- // RvousCancel handles the toc_rvous_cancel TOC command.
- //
- // From the TiK documentation:
- //
- // Cancel a rendezvous proposal from the user <nick>. <cookie> is the cookie
- // from the RVOUS_PROPOSE message. <service> is the UUID the proposal was for.
- // <tlvlist> contains a list of tlv tags followed by base64 encoded values.
- //
- // Note: This method does not actually process the TLV list param, as it's not
- // passed in the TiK client, the reference implementation.
- //
- // Command syntax: toc_rvous_cancel <nick> <cookie> <service>
- func (s OSCARProxy) RvousCancel(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.ICBM, wire.ICBMChannelMsgToHost); isLimited {
- return errMsg
- }
- var nick, cookie, service string
- if _, err := parseArgs(args, &nick, &cookie, &service); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- cbytes, err := base64.StdEncoding.DecodeString(cookie)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("base64.Decode: %w", err))
- }
- var arr [8]byte
- copy(arr[:], cbytes) // copy slice into array
- svcUUID, err := uuid.Parse(service)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("uuid.Parse: %w", err))
- }
- snac := wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ChannelID: wire.ICBMChannelRendezvous,
- ScreenName: nick,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
- Type: wire.ICBMRdvMessageCancel,
- Cookie: arr,
- Capability: svcUUID,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMRdvTLVTagsCancelReason, wire.ICBMRdvCancelReasonsUserCancel),
- },
- },
- }),
- },
- },
- }
- if _, err = s.ICBMService.ChannelMsgToHost(ctx, me, wire.SNACFrame{}, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ICBMService.ChannelMsgToHost: %w", err))
- }
- return ""
- }
- // SendIM handles the toc_send_im TOC command.
- //
- // From the TiK documentation:
- //
- // Send a message to a remote user. Remember to quote and encode the message.
- // If the optional string "auto" is the last argument, then the auto response
- // flag will be turned on for the IM.
- //
- // Command syntax: toc_send_im <Destination User> <Message> [auto]
- func (s OSCARProxy) SendIM(ctx context.Context, sender *state.SessionInstance, args []byte) string {
- if msg, isLimited := s.checkRateLimit(ctx, sender, wire.ICBM, wire.ICBMChannelMsgToHost); isLimited {
- return msg
- }
- var recip, msg string
- autoReply, err := parseArgs(args, &recip, &msg)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- msg = unescape(msg)
- frags, err := wire.ICBMFragmentList(msg)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("wire.ICBMFragmentList: %w", err))
- }
- snac := wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ChannelID: wire.ICBMChannelIM,
- ScreenName: recip,
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICBMTLVAOLIMData, frags),
- },
- },
- }
- if len(autoReply) > 0 && autoReply[0] == "auto" {
- snac.Append(wire.NewTLVBE(wire.ICBMTLVAutoResponse, []byte{}))
- }
- // send message and ignore response since there is no TOC error code to
- // handle errors such as "user is offline", etc.
- _, err = s.ICBMService.ChannelMsgToHost(ctx, sender, wire.SNACFrame{}, snac)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("ICBMService.ChannelMsgToHost: %w", err))
- }
- return ""
- }
- // SetAway handles the toc_chat_join TOC command.
- //
- // From the TiK documentation:
- //
- // If the away message is present, then the unavailable status flag is set for
- // the user. If the away message is not present, then the unavailable status
- // flag is unset. The away message is basic HTML, remember to encode the
- // information.
- //
- // Command syntax: toc_set_away [<away message>]
- func (s OSCARProxy) SetAway(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Locate, wire.LocateSetInfo); isLimited {
- return errMsg
- }
- maybeMsg, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- var msg string
- if len(maybeMsg) > 0 {
- msg = unescape(maybeMsg[0])
- }
- snac := wire.SNAC_0x02_0x04_LocateSetInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.LocateTLVTagsInfoUnavailableData, msg),
- },
- },
- }
- if err := s.LocateService.SetInfo(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.SetInfo: %w", err))
- }
- return ""
- }
- // SetCaps handles the toc_set_caps TOC command.
- //
- // From the TiK documentation:
- //
- // Set my capabilities. All capabilities that we support need to be sent at
- // the same time. Capabilities are represented by UUIDs.
- //
- // This method automatically adds the "chat" capability since it doesn't seem
- // to be sent explicitly by the official clients, even though they support
- // chat.
- //
- // Command syntax: toc_set_caps [ <Capability 1> [<Capability 2> [...]]]
- func (s OSCARProxy) SetCaps(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Locate, wire.LocateSetInfo); isLimited {
- return errMsg
- }
- params, err := parseArgs(args)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- caps := make([]uuid.UUID, 0, 16*(len(params)+1))
- for _, capStr := range params {
- uid, err := uuid.Parse(capStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("UUID.Parse: %w", err))
- }
- caps = append(caps, uid)
- }
- // assume client supports chat, although we may want to do this according
- // to client ID
- caps = append(caps, wire.CapChat)
- snac := wire.SNAC_0x02_0x04_LocateSetInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, caps),
- },
- },
- }
- if err := s.LocateService.SetInfo(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.SetInfo: %w", err))
- }
- return ""
- }
- // SetConfig handles the toc_set_config TOC command.
- //
- // From the TiK documentation:
- //
- // Set the config information for this user. The config information is line
- // oriented with the first character being the item type, followed by a space,
- // with the rest of the line being the item value. Only letters, numbers, and
- // spaces should be used. Remember you will have to enclose the entire config
- // in quotes.
- //
- // Item Types:
- // - g - Buddy Group (All Buddies until the next g or the end of config are in this group.)
- // - b - A Buddy
- // - p - Person on permit list
- // - d - Person on deny list
- // - m - Permit/Deny Mode. Possible values are
- // - 1 - Permit All
- // - 2 - Deny All
- // - 3 - Permit Some
- // - 4 - Deny Some
- //
- // This method doesn't attempt to validate any of the configuration--it saves
- // the config as received from the client.
- //
- // Command syntax: toc_set_config <Config Info>
- func (s OSCARProxy) SetConfig(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if status := me.Session().EvaluateRateLimit(time.Now(), 1); status == wire.RateLimitStatusLimited {
- return rateLimitExceededErr
- }
- // most TOC clients don't quote the config info argument, despite what the
- // documentation specifies. this makes the argument payload incompatible
- // for CSV parsing. since this command takes a single argument, we can get
- // away with trimming quotes and spaces from the byte slice before passing
- // it to the config store.
- args = bytes.Trim(args, "'\" ")
- config := string(args)
- if config == "" {
- return s.runtimeErr(ctx, fmt.Errorf("empty config"))
- }
- if err := s.TOCConfigStore.SetTOCConfig(ctx, me.IdentScreenName(), config); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("TOCConfigStore.SaveTOCConfig: %w", err))
- }
- return ""
- }
- // SetDir handles the toc_set_dir TOC command.
- //
- // From the TiK documentation:
- //
- // Set the DIR user information. This is a colon separated fields as in:
- //
- // "first name":"middle name":"last name":"maiden name":"city":"state":"country":"email":"allow web searches".
- //
- // Should return a DIR_STATUS msg. Having anything in the "allow web searches"
- // field allows people to use web-searches to find your directory info.
- // Otherwise, they'd have to use the client.
- //
- // The fields "email" and "allow web searches" are ignored by this method.
- //
- // Command syntax: toc_set_dir <info information>
- func (s OSCARProxy) SetDir(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Locate, wire.LocateSetDirInfo); isLimited {
- return errMsg
- }
- var info string
- if _, err := parseArgs(args, &info); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- info = unescape(info)
- rawFields := strings.Split(info, ":")
- var finalFields [9]string
- if len(rawFields) > len(finalFields) {
- return s.runtimeErr(ctx, fmt.Errorf("expected at most %d params, got %d", len(finalFields), len(rawFields)))
- }
- for i, a := range rawFields {
- finalFields[i] = strings.Trim(a, "\"")
- }
- snac := wire.SNAC_0x02_0x09_LocateSetDirInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ODirTLVFirstName, finalFields[0]),
- wire.NewTLVBE(wire.ODirTLVMiddleName, finalFields[1]),
- wire.NewTLVBE(wire.ODirTLVLastName, finalFields[2]),
- wire.NewTLVBE(wire.ODirTLVMaidenName, finalFields[3]),
- wire.NewTLVBE(wire.ODirTLVCountry, finalFields[6]),
- wire.NewTLVBE(wire.ODirTLVState, finalFields[5]),
- wire.NewTLVBE(wire.ODirTLVCity, finalFields[4]),
- },
- },
- }
- if _, err := s.LocateService.SetDirInfo(ctx, me, wire.SNACFrame{}, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.SetDirInfo: %w", err))
- }
- return ""
- }
- // SetIdle handles the toc_set_idle TOC command.
- //
- // From the TiK documentation:
- //
- // Set idle information. If <idle secs> is 0 then the user isn't idle at all.
- // If <idle secs> is greater than 0 then the user has already been idle for
- // <idle secs> number of seconds. The server will automatically keep
- // incrementing this number, so do not repeatedly call with new idle times.
- //
- // Command syntax: toc_set_idle <idle secs>
- func (s OSCARProxy) SetIdle(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.OService, wire.OServiceIdleNotification); isLimited {
- return errMsg
- }
- var idleTimeStr string
- if _, err := parseArgs(args, &idleTimeStr); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- time, err := strconv.Atoi(idleTimeStr)
- if err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("strconv.Atoi: %w", err))
- }
- snac := wire.SNAC_0x01_0x11_OServiceIdleNotification{
- IdleTime: uint32(time),
- }
- if err := s.OServiceService.IdleNotification(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("OServiceServiceBOS.IdleNotification: %w", err))
- }
- return ""
- }
- // SetInfo handles the toc_set_info TOC command.
- //
- // From the TiK documentation:
- //
- // Set the LOCATE user information. This is basic HTML. Remember to encode the info.
- //
- // Command syntax: toc_set_info <info information>
- func (s OSCARProxy) SetInfo(ctx context.Context, me *state.SessionInstance, args []byte) string {
- if errMsg, isLimited := s.checkRateLimit(ctx, me, wire.Locate, wire.LocateSetInfo); isLimited {
- return errMsg
- }
- var info string
- if _, err := parseArgs(args, &info); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))
- }
- info = unescape(info)
- snac := wire.SNAC_0x02_0x04_LocateSetInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.LocateTLVTagsInfoSigData, info),
- },
- },
- }
- if err := s.LocateService.SetInfo(ctx, me, snac); err != nil {
- return s.runtimeErr(ctx, fmt.Errorf("LocateService.SetInfo: %w", err))
- }
- return ""
- }
- // Signon handles the toc_signon TOC command.
- //
- // From the TiK documentation:
- //
- // The password needs to be roasted with the Roasting String if coming over a
- // FLAP connection, CP connections don't use roasted passwords. The language
- // specified will be used when generating web pages, such as the get info
- // pages. Currently, the only supported language is "english". If the language
- // sent isn't found, the default "english" language will be used. The version
- // string will be used for the client identity, and must be less than 50
- // characters.
- //
- // Passwords are roasted when sent to the host. This is done so they aren't
- // sent in "clear text" over the wire, although they are still trivial to
- // decode. Roasting is performed by first xoring each byte in the password
- // with the equivalent modulo byte in the roasting string. The result is then
- // converted to ascii hex, and prepended with "0x". So for example the
- // password "password" roasts to "0x2408105c23001130".
- //
- // The Roasting String is Tic/Toc.
- //
- // Command syntax: toc_signon <authorizer host> <authorizer port> <User Name> <Password> <language> <version>
- func (s OSCARProxy) Signon(ctx context.Context, args []byte) (*state.SessionInstance, []string) {
- var userName, password string
- if _, err := parseArgs(args, nil, nil, &userName, &password); err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("parseArgs: %w", err))}
- }
- passwordHash, err := hex.DecodeString(password[2:])
- if err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("hex.DecodeString: %w", err))}
- }
- signonFrame := wire.FLAPSignonFrame{}
- signonFrame.Append(wire.NewTLVBE(wire.LoginTLVTagsScreenName, userName))
- signonFrame.Append(wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, passwordHash))
- block, err := s.AuthService.FLAPLogin(ctx, signonFrame, state.NewStubUser, "")
- if err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("AuthService.FLAPLogin: %w", err))}
- }
- if block.HasTag(wire.LoginTLVTagsErrorSubcode) {
- s.Logger.DebugContext(ctx, "login failed")
- return nil, []string{"ERROR:980"} // bad username/password
- }
- authCookie, ok := block.Bytes(wire.OServiceTLVTagsLoginCookie)
- if !ok {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("unable to get session id from payload"))}
- }
- // todo: naming for cookie: login cookie, server cookie, or auth cookie?
- serverCookie, err := s.AuthService.CrackCookie(authCookie)
- sess, err := s.AuthService.RegisterBOSSession(ctx, serverCookie)
- if err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("AuthService.RegisterBOSSession: %w", err))}
- }
- // set chat capability so that... tk
- sess.SetCaps([][16]byte{wire.CapChat})
- if err := s.BuddyListRegistry.RegisterBuddyList(ctx, sess.IdentScreenName()); err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("BuddyListRegistry.RegisterBuddyList: %w", err))}
- }
- u, err := s.TOCConfigStore.User(ctx, sess.IdentScreenName())
- if err != nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("TOCConfigStore.User: %w", err))}
- }
- if u == nil {
- return nil, []string{s.runtimeErr(ctx, fmt.Errorf("TOCConfigStore.User: user not found"))}
- }
- return sess, []string{"SIGN_ON:TOC1.0", fmt.Sprintf("CONFIG:%s", u.TOCConfig)}
- }
- // Signout terminates a TOC session. It sends departure notifications to
- // buddies, de-registers buddy list and session.
- func (s OSCARProxy) Signout(ctx context.Context, me *state.SessionInstance, chatRegistry *ChatRegistry) {
- if err := s.BuddyService.BroadcastBuddyDeparted(ctx, me); err != nil {
- s.Logger.ErrorContext(ctx, "error sending departure notifications", "err", err.Error())
- }
- if err := s.BuddyListRegistry.UnregisterBuddyList(ctx, me.IdentScreenName()); err != nil {
- s.Logger.ErrorContext(ctx, "error removing buddy list entry", "err", err.Error())
- }
- s.AuthService.Signout(ctx, me)
- for _, sess := range chatRegistry.Sessions() {
- s.AuthService.SignoutChat(ctx, sess)
- sess.CloseInstance() // stop async server SNAC reply handler for this chat room
- }
- }
- // newHTTPAuthToken creates a HMAC token for authenticating TOC HTTP requests
- func (s OSCARProxy) newHTTPAuthToken(me state.IdentScreenName) (string, error) {
- cookie, err := s.CookieBaker.Issue([]byte(me.String()))
- if err != nil {
- return "", err
- }
- // trim padding so that gaim doesn't choke on the long value
- cookie = bytes.TrimRight(cookie, "\x00")
- return hex.EncodeToString(cookie), nil
- }
- // parseArgs extracts arguments from a TOC command. Each positional argument is
- // assigned to its corresponding args pointer. It returns the remaining
- // arguments as varargs.
- func parseArgs(payload []byte, args ...*string) (varArgs []string, err error) {
- if len(payload) == 0 && len(args) == 0 {
- return []string{}, nil
- }
- reader := csv.NewReader(bytes.NewReader(payload))
- reader.Comma = ' '
- reader.LazyQuotes = true
- reader.TrimLeadingSpace = true
- segs, err := reader.Read()
- if err != nil {
- return []string{}, fmt.Errorf("CSV reader error: %w", err)
- }
- if len(segs) < len(args) {
- return []string{}, fmt.Errorf("command contains fewer arguments than expected")
- }
- // populate placeholder pointers with their corresponding values
- for i, arg := range args {
- if arg != nil {
- *arg = strings.TrimSpace(segs[i])
- }
- }
- // dump remaining arguments as varargs
- return segs[len(args):], err
- }
- // runtimeErr is a convenience function that logs an error and returns a TOC
- // internal server error.
- func (s OSCARProxy) runtimeErr(ctx context.Context, err error) string {
- s.Logger.ErrorContext(ctx, "internal service error", "err", err.Error())
- return cmdInternalSvcErr
- }
- func (s OSCARProxy) checkRateLimit(ctx context.Context, sender *state.SessionInstance, foodGroup uint16, subGroup uint16) (string, bool) {
- rateClassID, ok := s.SNACRateLimits.RateClassLookup(foodGroup, subGroup)
- if !ok {
- s.Logger.ErrorContext(ctx, "rate limit not found, allowing request through")
- return "", false
- }
- if status := sender.Session().EvaluateRateLimit(time.Now(), rateClassID); status == wire.RateLimitStatusLimited {
- s.Logger.DebugContext(ctx, "(toc) rate limit exceeded, dropping SNAC",
- "foodgroup", wire.FoodGroupName(foodGroup),
- "subgroup", wire.SubGroupName(foodGroup, subGroup))
- return rateLimitExceededErr, true
- }
- return "", false
- }
- // unescape removes escaping from the following TOC characters: \ { } ( ) [ ] $ "
- func unescape(encoded string) string {
- if !strings.ContainsRune(encoded, '\\') {
- return encoded
- }
- var result strings.Builder
- result.Grow(len(encoded))
- escaped := false
- for i := 0; i < len(encoded); i++ {
- ch := encoded[i]
- if escaped {
- // append escaped character without the backslash
- result.WriteByte(ch)
- escaped = false
- } else if ch == '\\' {
- escaped = true
- } else {
- result.WriteByte(ch)
- }
- }
- return result.String()
- }
|