handler.go 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242
  1. package oscar
  2. import (
  3. "bytes"
  4. "context"
  5. "encoding/binary"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "log/slog"
  10. "github.com/mk6i/open-oscar-server/config"
  11. "github.com/mk6i/open-oscar-server/server/oscar/middleware"
  12. "github.com/mk6i/open-oscar-server/state"
  13. "github.com/mk6i/open-oscar-server/wire"
  14. )
  15. var (
  16. // ErrRouteNotFound is an error that indicates a failure to find a matching
  17. // route for an OSCAR protocol request.
  18. ErrRouteNotFound = errors.New("route not found")
  19. errUnknownICQMetaReqType = errors.New("unknown ICQ request type")
  20. )
  21. // ResponseWriter is the interface for sending a SNAC response to the client
  22. // from the server handlers.
  23. type ResponseWriter interface {
  24. SendSNAC(frame wire.SNACFrame, body any) error
  25. }
  26. // Handler defines a structure for routing OSCAR protocol requests to
  27. // appropriate handlers based on group:subGroup identifiers.
  28. type Handler struct {
  29. AdminService
  30. BARTService
  31. BuddyService
  32. ChatNavService
  33. ChatService
  34. FeedbagService
  35. ICBMService
  36. ICQService
  37. LocateService
  38. ODirService
  39. OServiceService
  40. PermitDenyService
  41. StatsService
  42. UserLookupService
  43. middleware.RouteLogger
  44. }
  45. func (rt Handler) AdminConfirmRequest(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  46. outSNAC, err := rt.ConfirmRequest(ctx, instance, inFrame)
  47. if err != nil {
  48. return err
  49. }
  50. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  51. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  52. }
  53. func (rt Handler) AdminInfoQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  54. inBody := wire.SNAC_0x07_0x02_AdminInfoQuery{}
  55. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  56. return err
  57. }
  58. outSNAC, err := rt.AdminService.InfoQuery(ctx, instance, inFrame, inBody)
  59. if err != nil {
  60. return err
  61. }
  62. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  63. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  64. }
  65. func (rt Handler) AdminInfoChangeRequest(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  66. inBody := wire.SNAC_0x07_0x04_AdminInfoChangeRequest{}
  67. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  68. return err
  69. }
  70. outSNAC, err := rt.InfoChangeRequest(ctx, instance, inFrame, inBody)
  71. if err != nil {
  72. return err
  73. }
  74. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  75. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  76. }
  77. func (rt Handler) AlertNotifyCapabilities(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  78. rt.LogRequest(ctx, inFrame, nil)
  79. return nil
  80. }
  81. func (rt Handler) AlertNotifyDisplayCapabilities(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  82. rt.LogRequest(ctx, inFrame, nil)
  83. return nil
  84. }
  85. func (rt Handler) InviteRequest(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  86. rt.LogRequest(ctx, inFrame, nil)
  87. return nil
  88. }
  89. func (rt Handler) PluginRequest(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  90. rt.LogRequest(ctx, inFrame, nil)
  91. return nil
  92. }
  93. func (rt Handler) MDirRequest(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  94. rt.LogRequest(ctx, inFrame, nil)
  95. return nil
  96. }
  97. func (rt Handler) BARTUploadQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  98. inBody := wire.SNAC_0x10_0x02_BARTUploadQuery{}
  99. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  100. return err
  101. }
  102. outSNAC, err := rt.BARTService.UpsertItem(ctx, instance, inFrame, inBody)
  103. if err != nil {
  104. return err
  105. }
  106. rt.LogRequestAndResponse(ctx, inFrame, outSNAC, outSNAC.Frame, outSNAC.Body)
  107. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  108. }
  109. func (rt Handler) BARTDownloadQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  110. inBody := wire.SNAC_0x10_0x04_BARTDownloadQuery{}
  111. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  112. return err
  113. }
  114. outSNAC, err := rt.RetrieveItem(ctx, inFrame, inBody)
  115. if err != nil {
  116. return err
  117. }
  118. rt.LogRequestAndResponse(ctx, inFrame, outSNAC, outSNAC.Frame, outSNAC.Body)
  119. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  120. }
  121. func (rt Handler) BARTDownload2Query(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  122. inBody := wire.SNAC_0x10_0x06_BARTDownload2Query{}
  123. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  124. return err
  125. }
  126. outSNACS, err := rt.RetrieveItemV2(ctx, inFrame, inBody)
  127. if err != nil {
  128. return err
  129. }
  130. for _, snac := range outSNACS {
  131. rt.LogRequestAndResponse(ctx, inFrame, snac, snac.Frame, snac.Body)
  132. if err := rw.SendSNAC(snac.Frame, snac.Body); err != nil {
  133. return err
  134. }
  135. }
  136. return nil
  137. }
  138. func (rt Handler) BuddyRightsQuery(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  139. inSNAC := wire.SNAC_0x03_0x02_BuddyRightsQuery{}
  140. if err := wire.UnmarshalBE(&inSNAC, r); err != nil {
  141. return err
  142. }
  143. outSNAC := rt.BuddyService.RightsQuery(ctx, inFrame)
  144. rt.LogRequestAndResponse(ctx, inFrame, inSNAC, outSNAC.Frame, outSNAC.Body)
  145. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  146. }
  147. func (rt Handler) BuddyAddBuddies(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  148. inSNAC := wire.SNAC_0x03_0x04_BuddyAddBuddies{}
  149. if err := wire.UnmarshalBE(&inSNAC, r); err != nil {
  150. return err
  151. }
  152. rejectSNAC, err := rt.AddBuddies(ctx, instance, inFrame, inSNAC)
  153. if err != nil {
  154. return err
  155. }
  156. if rejectSNAC != nil {
  157. rt.LogRequestAndResponse(ctx, inFrame, inSNAC, rejectSNAC.Frame, rejectSNAC.Body)
  158. if sendErr := rw.SendSNAC(rejectSNAC.Frame, rejectSNAC.Body); sendErr != nil {
  159. return sendErr
  160. }
  161. } else {
  162. rt.LogRequest(ctx, inFrame, inSNAC)
  163. }
  164. return nil
  165. }
  166. func (rt Handler) BuddyDelBuddies(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  167. inSNAC := wire.SNAC_0x03_0x05_BuddyDelBuddies{}
  168. if err := wire.UnmarshalBE(&inSNAC, r); err != nil {
  169. return err
  170. }
  171. rt.LogRequest(ctx, inFrame, inSNAC)
  172. return rt.DelBuddies(ctx, instance, inSNAC)
  173. }
  174. func (rt Handler) BuddyAddTempBuddies(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  175. inSNAC := wire.SNAC_0x03_0x0F_BuddyAddTempBuddies{}
  176. if err := wire.UnmarshalBE(&inSNAC, r); err != nil {
  177. return err
  178. }
  179. rejectSNAC, err := rt.AddTempBuddies(ctx, instance, inFrame, inSNAC)
  180. if err != nil {
  181. return err
  182. }
  183. if rejectSNAC != nil {
  184. rt.LogRequestAndResponse(ctx, inFrame, inSNAC, rejectSNAC.Frame, rejectSNAC.Body)
  185. if sendErr := rw.SendSNAC(rejectSNAC.Frame, rejectSNAC.Body); sendErr != nil {
  186. return sendErr
  187. }
  188. } else {
  189. rt.LogRequest(ctx, inFrame, inSNAC)
  190. }
  191. return nil
  192. }
  193. func (rt Handler) BuddyDelTempBuddies(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  194. inSNAC := wire.SNAC_0x03_0x10_BuddyDelTempBuddies{}
  195. if err := wire.UnmarshalBE(&inSNAC, r); err != nil {
  196. return err
  197. }
  198. rt.LogRequest(ctx, inFrame, inSNAC)
  199. return rt.DelTempBuddies(ctx, instance, inSNAC)
  200. }
  201. func (rt Handler) ChatChannelMsgToHost(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  202. inBody := wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{}
  203. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  204. return err
  205. }
  206. outSNAC, err := rt.ChatService.ChannelMsgToHost(ctx, instance, inFrame, inBody)
  207. if err != nil {
  208. return err
  209. }
  210. if outSNAC == nil {
  211. return nil
  212. }
  213. rt.Logger.InfoContext(ctx, "user sent a chat message")
  214. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  215. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  216. }
  217. func (rt Handler) ChatNavRequestChatRights(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  218. outSNAC := rt.RequestChatRights(ctx, inFrame)
  219. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  220. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  221. }
  222. func (rt Handler) ChatNavRequestExchangeInfo(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  223. inBody := wire.SNAC_0x0D_0x03_ChatNavRequestExchangeInfo{}
  224. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  225. return err
  226. }
  227. outSNAC, err := rt.ExchangeInfo(ctx, inFrame, inBody)
  228. if err != nil {
  229. return err
  230. }
  231. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  232. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  233. }
  234. func (rt Handler) ChatNavRequestRoomInfo(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  235. inBody := wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{}
  236. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  237. return err
  238. }
  239. outSNAC, err := rt.RequestRoomInfo(ctx, inFrame, inBody)
  240. if err != nil {
  241. return err
  242. }
  243. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  244. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  245. }
  246. func (rt Handler) ChatNavCreateRoom(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  247. inBody := wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{}
  248. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  249. return err
  250. }
  251. outSNAC, err := rt.CreateRoom(ctx, instance, inFrame, inBody)
  252. if err != nil {
  253. return err
  254. }
  255. roomName, _ := inBody.String(wire.ChatRoomTLVRoomName)
  256. rt.Logger.InfoContext(ctx, "user started a chat room", slog.String("roomName", roomName))
  257. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  258. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  259. }
  260. func (rt Handler) FeedbagRightsQuery(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  261. inBody := wire.SNAC_0x13_0x02_FeedbagRightsQuery{}
  262. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  263. return err
  264. }
  265. outSNAC := rt.FeedbagService.RightsQuery(ctx, inFrame)
  266. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  267. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  268. }
  269. func (rt Handler) FeedbagQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  270. outSNAC, err := rt.Query(ctx, instance, inFrame)
  271. if err != nil {
  272. return err
  273. }
  274. rt.LogRequest(ctx, inFrame, outSNAC)
  275. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  276. }
  277. func (rt Handler) FeedbagQueryIfModified(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  278. inBody := wire.SNAC_0x13_0x05_FeedbagQueryIfModified{}
  279. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  280. return err
  281. }
  282. outSNAC, err := rt.QueryIfModified(ctx, instance, inFrame, inBody)
  283. if err != nil {
  284. return err
  285. }
  286. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  287. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  288. }
  289. func (rt Handler) FeedbagUse(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, _ ResponseWriter) error {
  290. rt.LogRequest(ctx, inFrame, nil)
  291. return rt.Use(ctx, instance)
  292. }
  293. func (rt Handler) FeedbagInsertItem(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  294. inBody := wire.SNAC_0x13_0x08_FeedbagInsertItem{}
  295. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  296. return err
  297. }
  298. outSNAC, err := rt.FeedbagService.UpsertItem(ctx, instance, inFrame, inBody.Items)
  299. if err != nil {
  300. return err
  301. }
  302. if outSNAC == nil {
  303. rt.LogRequest(ctx, inFrame, inBody)
  304. return nil
  305. }
  306. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  307. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  308. }
  309. func (rt Handler) FeedbagUpdateItem(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  310. inBody := wire.SNAC_0x13_0x09_FeedbagUpdateItem{}
  311. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  312. return err
  313. }
  314. outSNAC, err := rt.FeedbagService.UpsertItem(ctx, instance, inFrame, inBody.Items)
  315. if err != nil {
  316. return err
  317. }
  318. if outSNAC == nil {
  319. rt.LogRequest(ctx, inFrame, inBody)
  320. return nil
  321. }
  322. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  323. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  324. }
  325. func (rt Handler) FeedbagDeleteItem(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  326. inBody := wire.SNAC_0x13_0x0A_FeedbagDeleteItem{}
  327. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  328. return err
  329. }
  330. outSNAC, err := rt.DeleteItem(ctx, instance, inFrame, inBody)
  331. if err != nil {
  332. return err
  333. }
  334. if outSNAC == nil {
  335. rt.LogRequest(ctx, inFrame, inBody)
  336. return nil
  337. }
  338. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  339. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  340. }
  341. func (rt Handler) FeedbagStartCluster(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  342. inBody := wire.SNAC_0x13_0x11_FeedbagStartCluster{}
  343. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  344. return err
  345. }
  346. rt.StartCluster(ctx, instance, inFrame, inBody)
  347. rt.LogRequest(ctx, inFrame, inBody)
  348. return nil
  349. }
  350. func (rt Handler) FeedbagEndCluster(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  351. if err := rt.EndCluster(ctx, instance, inFrame); err != nil {
  352. return err
  353. }
  354. rt.LogRequest(ctx, inFrame, nil)
  355. return nil
  356. }
  357. func (rt Handler) FeedbagPreAuthorizeBuddy(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  358. inBody := wire.SNAC_0x13_0x14_FeedbagPreAuthorizeBuddy{}
  359. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  360. return err
  361. }
  362. outSNAC, err := rt.PreAuthorizeBuddy(ctx, instance, inFrame, inBody)
  363. if err != nil {
  364. return err
  365. }
  366. if outSNAC != nil {
  367. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  368. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  369. }
  370. rt.LogRequest(ctx, inFrame, inBody)
  371. return nil
  372. }
  373. func (rt Handler) FeedbagRequestAuthorizeToHost(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  374. inBody := wire.SNAC_0x13_0x18_FeedbagRequestAuthorizationToHost{}
  375. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  376. return err
  377. }
  378. if err := rt.RequestAuthorizeToHost(ctx, instance, inFrame, inBody); err != nil {
  379. return err
  380. }
  381. rt.LogRequest(ctx, inFrame, inBody)
  382. return nil
  383. }
  384. func (rt Handler) FeedbagRespondAuthorizeToHost(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  385. inBody := wire.SNAC_0x13_0x1A_FeedbagRespondAuthorizeToHost{}
  386. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  387. return err
  388. }
  389. if err := rt.RespondAuthorizeToHost(ctx, instance.IdentScreenName(), inFrame, inBody); err != nil {
  390. return err
  391. }
  392. rt.LogRequest(ctx, inFrame, inBody)
  393. return nil
  394. }
  395. func (rt Handler) ICBMAddParameters(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  396. inBody := wire.SNAC_0x04_0x02_ICBMAddParameters{}
  397. rt.LogRequest(ctx, inFrame, inBody)
  398. return wire.UnmarshalBE(&inBody, r)
  399. }
  400. func (rt Handler) ICBMParameterQuery(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  401. outSNAC := rt.ParameterQuery(ctx, inFrame)
  402. rt.LogRequestAndResponse(ctx, inFrame, outSNAC, outSNAC.Frame, outSNAC.Body)
  403. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  404. }
  405. func (rt Handler) ICBMChannelMsgToHost(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  406. inBody := wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{}
  407. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  408. return err
  409. }
  410. outSNAC, err := rt.ICBMService.ChannelMsgToHost(ctx, instance, inFrame, inBody)
  411. if err != nil {
  412. return err
  413. }
  414. if outSNAC == nil {
  415. rt.LogRequest(ctx, inFrame, inBody)
  416. return nil
  417. }
  418. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  419. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  420. }
  421. func (rt Handler) ICBMEvilRequest(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  422. inBody := wire.SNAC_0x04_0x08_ICBMEvilRequest{}
  423. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  424. return err
  425. }
  426. outSNAC, err := rt.EvilRequest(ctx, instance, inFrame, inBody)
  427. if err != nil {
  428. return err
  429. }
  430. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  431. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  432. }
  433. func (rt Handler) ICBMClientErr(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  434. inBody := wire.SNAC_0x04_0x0B_ICBMClientErr{}
  435. rt.LogRequest(ctx, inFrame, inBody)
  436. err := wire.UnmarshalBE(&inBody, r)
  437. if err != nil {
  438. return err
  439. }
  440. return rt.ClientErr(ctx, instance, inFrame, inBody)
  441. }
  442. func (rt Handler) ICBMClientEvent(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  443. inBody := wire.SNAC_0x04_0x14_ICBMClientEvent{}
  444. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  445. return err
  446. }
  447. rt.LogRequest(ctx, inFrame, inBody)
  448. return rt.ClientEvent(ctx, instance, inFrame, inBody)
  449. }
  450. func (rt Handler) ICBMOfflineRetrieve(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, rw ResponseWriter) error {
  451. outSNAC, err := rt.OfflineRetrieve(ctx, instance, inFrame)
  452. if err != nil {
  453. return err
  454. }
  455. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  456. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  457. }
  458. func (rt Handler) ICQDBQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  459. inBody := wire.SNAC_0x15_0x02_BQuery{}
  460. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  461. return err
  462. }
  463. md, ok := inBody.Bytes(wire.ICQTLVTagsMetadata)
  464. if !ok {
  465. return errors.New("invalid ICQ frame")
  466. }
  467. icqChunk := wire.ICQMessageRequestEnvelope{}
  468. if err := wire.UnmarshalLE(&icqChunk, bytes.NewBuffer(md)); err != nil {
  469. return err
  470. }
  471. buf := bytes.NewBuffer(icqChunk.Body)
  472. icqMD := wire.ICQMetadataWithSubType{}
  473. if err := wire.UnmarshalLE(&icqMD, buf); err != nil {
  474. return err
  475. }
  476. switch icqMD.ReqType {
  477. case wire.ICQDBQueryOfflineMsgReq:
  478. return rt.OfflineMsgReq(ctx, inFrame, instance, icqMD.Seq)
  479. case wire.ICQDBQueryDeleteMsgReq:
  480. return rt.DeleteMsgReq(ctx, instance, icqMD.Seq)
  481. case wire.ICQDBQueryMetaReq:
  482. if icqMD.Optional == nil {
  483. return errors.New("got req without subtype")
  484. }
  485. rt.Logger.Debug("ICQ client request",
  486. "query_name", wire.ICQDBQueryName(icqMD.ReqType),
  487. "query_type", wire.ICQDBQueryMetaName(icqMD.Optional.ReqSubType),
  488. "uin", instance.UIN())
  489. switch icqMD.Optional.ReqSubType {
  490. case wire.ICQDBQueryMetaReqShortInfo:
  491. userInfo := wire.ICQ_0x07D0_0x04BA_DBQueryMetaReqShortInfo{}
  492. if err := binary.Read(buf, binary.LittleEndian, &userInfo); err != nil {
  493. return nil
  494. }
  495. return rt.ShortUserInfo(ctx, instance, inFrame, userInfo, icqMD.Seq)
  496. case wire.ICQDBQueryMetaReqFullInfo, wire.ICQDBQueryMetaReqFullInfo2:
  497. userInfo := wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{}
  498. if err := binary.Read(buf, binary.LittleEndian, &userInfo); err != nil {
  499. return nil
  500. }
  501. return rt.FullUserInfo(ctx, instance, inFrame, userInfo, icqMD.Seq)
  502. case wire.ICQDBQueryMetaReqXMLReq:
  503. req := wire.ICQ_0x07D0_0x0898_DBQueryMetaReqXMLReq{}
  504. if err := wire.UnmarshalLE(&req, buf); err != nil {
  505. return err
  506. }
  507. if err := rt.XMLReqData(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  508. return err
  509. }
  510. case wire.ICQDBQueryMetaReqSetPermissions:
  511. req := wire.ICQ_0x07D0_0x0424_DBQueryMetaReqSetPermissions{}
  512. if err := wire.UnmarshalLE(&req, buf); err != nil {
  513. return err
  514. }
  515. if err := rt.SetPermissions(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  516. return err
  517. }
  518. case wire.ICQDBQueryMetaReqSetICQPhone:
  519. req := wire.ICQ_0x07D0_0x0654_DBQueryMetaReqSetICQPhone{}
  520. if err := wire.UnmarshalLE(&req, buf); err != nil {
  521. return err
  522. }
  523. if err := rt.SetICQPhone(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  524. return err
  525. }
  526. case wire.ICQDBQueryMetaReqSearchByUIN:
  527. req := wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{}
  528. if err := wire.UnmarshalLE(&req, buf); err != nil {
  529. return err
  530. }
  531. if err := rt.FindByUIN(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  532. return err
  533. }
  534. case wire.ICQDBQueryMetaReqSearchByUIN2:
  535. req := wire.ICQ_0x07D0_0x0569_DBQueryMetaReqSearchByUIN2{}
  536. if err := wire.UnmarshalLE(&req, buf); err != nil {
  537. return err
  538. }
  539. if err := rt.FindByUIN2(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  540. return err
  541. }
  542. case wire.ICQDBQueryMetaReqSearchByEmail:
  543. req := wire.ICQ_0x07D0_0x0529_DBQueryMetaReqSearchByEmail{}
  544. if err := wire.UnmarshalLE(&req, buf); err != nil {
  545. return err
  546. }
  547. if err := rt.FindByICQEmail(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  548. return err
  549. }
  550. case wire.ICQDBQueryMetaReqSearchByEmail3:
  551. req := wire.ICQ_0x07D0_0x0573_DBQueryMetaReqSearchByEmail3{}
  552. if err := wire.UnmarshalLE(&req, buf); err != nil {
  553. return err
  554. }
  555. if err := rt.FindByEmail3(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  556. return err
  557. }
  558. case wire.ICQDBQueryMetaReqSearchByDetails:
  559. req := wire.ICQ_0x07D0_0x0515_DBQueryMetaReqSearchByDetails{}
  560. if err := wire.UnmarshalLE(&req, buf); err != nil {
  561. return err
  562. }
  563. if err := rt.FindByICQName(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  564. return err
  565. }
  566. case wire.ICQDBQueryMetaReqSearchWhitePages:
  567. req := wire.ICQ_0x07D0_0x0533_DBQueryMetaReqSearchWhitePages{}
  568. if err := wire.UnmarshalLE(&req, buf); err != nil {
  569. return err
  570. }
  571. if err := rt.FindByICQInterests(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  572. return err
  573. }
  574. case wire.ICQDBQueryMetaReqSearchWhitePages2:
  575. req := wire.ICQ_0x07D0_0x055F_DBQueryMetaReqSearchWhitePages2{}
  576. if err := wire.UnmarshalLE(&req, buf); err != nil {
  577. return err
  578. }
  579. if err := rt.FindByWhitePages2(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  580. return err
  581. }
  582. case wire.ICQDBQueryMetaReqSetBasicInfo:
  583. req := wire.ICQ_0x07D0_0x03EA_DBQueryMetaReqSetBasicInfo{}
  584. if err := wire.UnmarshalLE(&req, buf); err != nil {
  585. return err
  586. }
  587. if err := rt.SetBasicInfo(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  588. return err
  589. }
  590. case wire.ICQDBQueryMetaReqSetWorkInfo:
  591. req := wire.ICQ_0x07D0_0x03F3_DBQueryMetaReqSetWorkInfo{}
  592. if err := wire.UnmarshalLE(&req, buf); err != nil {
  593. return err
  594. }
  595. if err := rt.SetWorkInfo(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  596. return err
  597. }
  598. case wire.ICQDBQueryMetaReqSetMoreInfo:
  599. req := wire.ICQ_0x07D0_0x03FD_DBQueryMetaReqSetMoreInfo{}
  600. if err := wire.UnmarshalLE(&req, buf); err != nil {
  601. return err
  602. }
  603. if err := rt.SetMoreInfo(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  604. return err
  605. }
  606. case wire.ICQDBQueryMetaReqSetNotes:
  607. req := wire.ICQ_0x07D0_0x0406_DBQueryMetaReqSetNotes{}
  608. if err := wire.UnmarshalLE(&req, buf); err != nil {
  609. return err
  610. }
  611. if err := rt.SetUserNotes(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  612. return err
  613. }
  614. case wire.ICQDBQueryMetaReqSetEmails:
  615. req := wire.ICQ_0x07D0_0x040B_DBQueryMetaReqSetEmails{}
  616. if err := wire.UnmarshalLE(&req, buf); err != nil {
  617. return err
  618. }
  619. if err := rt.SetEmails(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  620. return err
  621. }
  622. case wire.ICQDBQueryMetaReqSetInterests:
  623. req := wire.ICQ_0x07D0_0x0410_DBQueryMetaReqSetInterests{}
  624. if err := wire.UnmarshalLE(&req, buf); err != nil {
  625. return err
  626. }
  627. if err := rt.SetInterests(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  628. return err
  629. }
  630. case wire.ICQDBQueryMetaReqSetAffiliations:
  631. req := wire.ICQ_0x07D0_0x041A_DBQueryMetaReqSetAffiliations{}
  632. if err := wire.UnmarshalLE(&req, buf); err != nil {
  633. return err
  634. }
  635. if err := rt.SetAffiliations(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  636. return err
  637. }
  638. case wire.ICQDBQueryMetaReqSetFullInfo:
  639. req := wire.ICQ_0x07D0_0x0C3A_DBQueryMetaReqSetFullInfo{}
  640. if err := wire.UnmarshalLE(&req, buf); err != nil {
  641. return err
  642. }
  643. if err := rt.SetICQInfo(ctx, instance, inFrame, req, icqMD.Seq); err != nil {
  644. return err
  645. }
  646. default:
  647. rt.Logger.Debug("unsupported ICQDBQueryMetaReqType", "type", icqMD.Optional.ReqSubType)
  648. return nil
  649. }
  650. default:
  651. return fmt.Errorf("%w: %X", errUnknownICQMetaReqType, icqMD.ReqType)
  652. }
  653. return nil
  654. }
  655. func (rt Handler) LocateRightsQuery(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  656. outSNAC := rt.LocateService.RightsQuery(ctx, inFrame)
  657. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  658. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  659. }
  660. func (rt Handler) LocateSetInfo(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  661. inBody := wire.SNAC_0x02_0x04_LocateSetInfo{}
  662. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  663. return err
  664. }
  665. rt.LogRequest(ctx, inFrame, inBody)
  666. return rt.SetInfo(ctx, instance, inBody)
  667. }
  668. func (rt Handler) LocateSetDirInfo(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  669. inBody := wire.SNAC_0x02_0x09_LocateSetDirInfo{}
  670. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  671. return err
  672. }
  673. outSNAC, err := rt.SetDirInfo(ctx, instance, inFrame, inBody)
  674. if err != nil {
  675. return err
  676. }
  677. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  678. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  679. }
  680. func (rt Handler) LocateGetDirInfo(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  681. inBody := wire.SNAC_0x02_0x0B_LocateGetDirInfo{}
  682. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  683. return err
  684. }
  685. outSNAC, err := rt.DirInfo(ctx, inFrame, inBody)
  686. if err != nil {
  687. return err
  688. }
  689. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  690. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  691. }
  692. func (rt Handler) LocateSetKeywordInfo(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  693. inBody := wire.SNAC_0x02_0x0F_LocateSetKeywordInfo{}
  694. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  695. return err
  696. }
  697. outSNAC, err := rt.SetKeywordInfo(ctx, instance, inFrame, inBody)
  698. if err != nil {
  699. return err
  700. }
  701. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  702. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  703. }
  704. func (rt Handler) LocateUserInfoQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  705. inBody := wire.SNAC_0x02_0x05_LocateUserInfoQuery{}
  706. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  707. return err
  708. }
  709. outSNAC, err := rt.LocateService.UserInfoQuery(ctx, instance, inFrame, inBody)
  710. if err != nil {
  711. return err
  712. }
  713. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  714. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  715. }
  716. func (rt Handler) LocateUserInfoQuery2(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  717. inBody := wire.SNAC_0x02_0x15_LocateUserInfoQuery2{}
  718. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  719. return err
  720. }
  721. // SNAC functionality for LocateUserInfoQuery and LocateUserInfoQuery2 is
  722. // identical except for the Type field data type (uint16 vs uint32).
  723. wrappedBody := wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  724. Type: uint16(inBody.Type2),
  725. ScreenName: inBody.ScreenName,
  726. }
  727. outSNAC, err := rt.LocateService.UserInfoQuery(ctx, instance, inFrame, wrappedBody)
  728. if err != nil {
  729. return err
  730. }
  731. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  732. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  733. }
  734. func (rt Handler) ODirInfoQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  735. inBody := wire.SNAC_0x0F_0x02_InfoQuery{}
  736. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  737. return err
  738. }
  739. outSNAC, err := rt.ODirService.InfoQuery(ctx, inFrame, inBody)
  740. if err != nil {
  741. return err
  742. }
  743. rt.LogRequestAndResponse(ctx, inFrame, outSNAC, outSNAC.Frame, outSNAC.Body)
  744. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  745. }
  746. func (rt Handler) ODirKeywordListQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  747. outSNAC, err := rt.KeywordListQuery(ctx, inFrame)
  748. if err != nil {
  749. return err
  750. }
  751. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  752. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  753. }
  754. func (rt Handler) OServiceRateParamsQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  755. outSNAC := rt.RateParamsQuery(ctx, instance, inFrame)
  756. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  757. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  758. }
  759. func (rt Handler) OServiceRateParamsSubAdd(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  760. inBody := wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd{}
  761. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  762. return err
  763. }
  764. rt.RateParamsSubAdd(ctx, instance, inBody)
  765. rt.LogRequest(ctx, inFrame, inBody)
  766. return nil
  767. }
  768. func (rt Handler) OServiceUserInfoQuery(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  769. outSNAC := rt.OServiceService.UserInfoQuery(ctx, instance, inFrame)
  770. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  771. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  772. }
  773. func (rt Handler) OServiceProbeReq(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  774. outSNAC := rt.ProbeReq(ctx, inFrame)
  775. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  776. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  777. }
  778. func (rt Handler) OServiceIdleNotification(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  779. inBody := wire.SNAC_0x01_0x11_OServiceIdleNotification{}
  780. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  781. return err
  782. }
  783. rt.LogRequest(ctx, inFrame, inBody)
  784. return rt.IdleNotification(ctx, instance, inBody)
  785. }
  786. func (rt Handler) OServiceClientVersions(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  787. inBody := wire.SNAC_0x01_0x17_OServiceClientVersions{}
  788. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  789. return err
  790. }
  791. outSNACs := rt.ClientVersions(ctx, instance, inFrame, inBody)
  792. for _, snac := range outSNACs {
  793. rt.LogRequestAndResponse(ctx, inFrame, inBody, snac.Frame, snac.Body)
  794. if err := rw.SendSNAC(snac.Frame, snac.Body); err != nil {
  795. return err
  796. }
  797. }
  798. return nil
  799. }
  800. func (rt Handler) OServiceSetUserInfoFields(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  801. inBody := wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields{}
  802. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  803. return err
  804. }
  805. outSNAC, err := rt.SetUserInfoFields(ctx, instance, inFrame, inBody)
  806. if err != nil {
  807. return err
  808. }
  809. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  810. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  811. }
  812. func (rt Handler) OServiceNoop(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  813. // no-op keep-alive
  814. rt.LogRequest(ctx, inFrame, nil)
  815. return nil
  816. }
  817. func (rt Handler) OServiceSetPrivacyFlags(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  818. inBody := wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags{}
  819. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  820. return err
  821. }
  822. rt.SetPrivacyFlags(ctx, inBody)
  823. rt.LogRequest(ctx, inFrame, inBody)
  824. return nil
  825. }
  826. func (rt Handler) OServiceServiceRequest(ctx context.Context, service uint16, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter, listener config.Listener) error {
  827. inBody := wire.SNAC_0x01_0x04_OServiceServiceRequest{}
  828. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  829. return err
  830. }
  831. outSNAC, err := rt.ServiceRequest(ctx, service, instance, inFrame, inBody, listener)
  832. if err != nil {
  833. return err
  834. }
  835. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  836. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  837. }
  838. func (rt Handler) OServiceClientOnline(ctx context.Context, service uint16, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, _ ResponseWriter) error {
  839. inBody := wire.SNAC_0x01_0x02_OServiceClientOnline{}
  840. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  841. return err
  842. }
  843. rt.Logger.InfoContext(ctx, "user signed on")
  844. rt.LogRequest(ctx, inFrame, inBody)
  845. return rt.ClientOnline(ctx, service, inBody, instance)
  846. }
  847. func (rt Handler) PermitDenyRightsQuery(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, _ io.Reader, rw ResponseWriter) error {
  848. outSNAC := rt.PermitDenyService.RightsQuery(ctx, inFrame)
  849. rt.LogRequestAndResponse(ctx, inFrame, nil, outSNAC.Frame, outSNAC.Body)
  850. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  851. }
  852. func (rt Handler) PermitDenyAddDenyListEntries(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  853. inBody := wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{}
  854. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  855. return err
  856. }
  857. rt.LogRequest(ctx, inFrame, inBody)
  858. return rt.AddDenyListEntries(ctx, instance, inBody)
  859. }
  860. func (rt Handler) PermitDenyDelDenyListEntries(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  861. inBody := wire.SNAC_0x09_0x08_PermitDenyDelDenyListEntries{}
  862. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  863. return err
  864. }
  865. rt.LogRequest(ctx, inFrame, inBody)
  866. return rt.DelDenyListEntries(ctx, instance, inBody)
  867. }
  868. func (rt Handler) PermitDenyAddPermListEntries(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  869. inBody := wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{}
  870. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  871. return err
  872. }
  873. rt.LogRequest(ctx, inFrame, inBody)
  874. return rt.AddPermListEntries(ctx, instance, inBody)
  875. }
  876. func (rt Handler) PermitDenyDelPermListEntries(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  877. inBody := wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries{}
  878. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  879. return err
  880. }
  881. rt.LogRequest(ctx, inFrame, inBody)
  882. return rt.DelPermListEntries(ctx, instance, inBody)
  883. }
  884. // PermitDenySetGroupPermitMask sets the classes of users I can interact with. We don't
  885. // apply any of these settings to the privacy mechanism, so just log them for
  886. // now.
  887. func (rt Handler) PermitDenySetGroupPermitMask(ctx context.Context, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  888. inBody := wire.SNAC_0x09_0x04_PermitDenySetGroupPermitMask{}
  889. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  890. return err
  891. }
  892. var flags []string
  893. if inBody.IsFlagSet(wire.OServiceUserFlagUnconfirmed) {
  894. flags = append(flags, "wire.OServiceUserFlagUnconfirmed")
  895. }
  896. if inBody.IsFlagSet(wire.OServiceUserFlagAdministrator) {
  897. flags = append(flags, "wire.OServiceUserFlagAdministrator")
  898. }
  899. if inBody.IsFlagSet(wire.OServiceUserFlagAOL) {
  900. flags = append(flags, "wire.OServiceUserFlagAOL")
  901. }
  902. if inBody.IsFlagSet(wire.OServiceUserFlagOSCARPay) {
  903. flags = append(flags, "wire.OServiceUserFlagOSCARPay")
  904. }
  905. if inBody.IsFlagSet(wire.OServiceUserFlagOSCARFree) {
  906. flags = append(flags, "wire.OServiceUserFlagOSCARFree")
  907. }
  908. if inBody.IsFlagSet(wire.OServiceUserFlagUnavailable) {
  909. flags = append(flags, "wire.OServiceUserFlagUnavailable")
  910. }
  911. if inBody.IsFlagSet(wire.OServiceUserFlagICQ) {
  912. flags = append(flags, "wire.OServiceUserFlagICQ")
  913. }
  914. if inBody.IsFlagSet(wire.OServiceUserFlagWireless) {
  915. flags = append(flags, "wire.OServiceUserFlagWireless")
  916. }
  917. if inBody.IsFlagSet(wire.OServiceUserFlagInternal) {
  918. flags = append(flags, "wire.OServiceUserFlagInternal")
  919. }
  920. if inBody.IsFlagSet(wire.OServiceUserFlagFish) {
  921. flags = append(flags, "wire.OServiceUserFlagFish")
  922. }
  923. if inBody.IsFlagSet(wire.OServiceUserFlagBot) {
  924. flags = append(flags, "wire.OServiceUserFlagBot")
  925. }
  926. if inBody.IsFlagSet(wire.OServiceUserFlagBeast) {
  927. flags = append(flags, "wire.OServiceUserFlagBeast")
  928. }
  929. if inBody.IsFlagSet(wire.OServiceUserFlagOneWayWireless) {
  930. flags = append(flags, "wire.OServiceUserFlagOneWayWireless")
  931. }
  932. if inBody.IsFlagSet(wire.OServiceUserFlagOfficial) {
  933. flags = append(flags, "wire.OServiceUserFlagOfficial")
  934. }
  935. rt.Logger.Info("set pd group mask", "flags", flags)
  936. rt.LogRequest(ctx, inFrame, inBody)
  937. return nil
  938. }
  939. func (rt Handler) StatsReportEvents(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  940. inBody := wire.SNAC_0x0B_0x03_StatsReportEvents{}
  941. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  942. return err
  943. }
  944. outSNAC := rt.ReportEvents(ctx, inFrame, inBody)
  945. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  946. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  947. }
  948. func (rt Handler) UserLookupFindByEmail(ctx context.Context, _ *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter) error {
  949. inBody := wire.SNAC_0x0A_0x02_UserLookupFindByEmail{}
  950. if err := wire.UnmarshalBE(&inBody, r); err != nil {
  951. return err
  952. }
  953. outSNAC, err := rt.FindByEmail(ctx, inFrame, inBody)
  954. if err != nil {
  955. return err
  956. }
  957. rt.LogRequestAndResponse(ctx, inFrame, inBody, outSNAC.Frame, outSNAC.Body)
  958. return rw.SendSNAC(outSNAC.Frame, outSNAC.Body)
  959. }
  960. // Handle directs an incoming OSCAR request to the appropriate handler based on
  961. // its group and subGroup identifiers found in the SNAC frame. It returns an
  962. // ErrRouteNotFound error if no matching handler is found for the group:subGroup
  963. // pair in the request.
  964. func (rt Handler) Handle(ctx context.Context, server uint16, instance *state.SessionInstance, inFrame wire.SNACFrame, r io.Reader, rw ResponseWriter, listener config.Listener) error {
  965. switch inFrame.FoodGroup {
  966. case wire.Admin:
  967. switch inFrame.SubGroup {
  968. case wire.AdminAcctConfirmRequest:
  969. return rt.AdminConfirmRequest(ctx, instance, inFrame, r, rw)
  970. case wire.AdminInfoChangeRequest:
  971. return rt.AdminInfoChangeRequest(ctx, instance, inFrame, r, rw)
  972. case wire.AdminInfoQuery:
  973. return rt.AdminInfoQuery(ctx, instance, inFrame, r, rw)
  974. }
  975. case wire.Alert:
  976. switch inFrame.SubGroup {
  977. case wire.AlertNotifyCapabilities:
  978. return rt.AlertNotifyCapabilities(ctx, instance, inFrame, r, rw)
  979. case wire.AlertNotifyDisplayCapabilities:
  980. return rt.AlertNotifyDisplayCapabilities(ctx, instance, inFrame, r, rw)
  981. }
  982. case wire.BART:
  983. switch inFrame.SubGroup {
  984. case wire.BARTDownloadQuery:
  985. return rt.BARTDownloadQuery(ctx, instance, inFrame, r, rw)
  986. case wire.BARTDownload2Query:
  987. return rt.BARTDownload2Query(ctx, instance, inFrame, r, rw)
  988. case wire.BARTUploadQuery:
  989. return rt.BARTUploadQuery(ctx, instance, inFrame, r, rw)
  990. }
  991. case wire.Buddy:
  992. switch inFrame.SubGroup {
  993. case wire.BuddyAddBuddies:
  994. return rt.BuddyAddBuddies(ctx, instance, inFrame, r, rw)
  995. case wire.BuddyDelBuddies:
  996. return rt.BuddyDelBuddies(ctx, instance, inFrame, r, rw)
  997. case wire.BuddyRightsQuery:
  998. return rt.BuddyRightsQuery(ctx, instance, inFrame, r, rw)
  999. case wire.BuddyAddTempBuddies:
  1000. return rt.BuddyAddTempBuddies(ctx, instance, inFrame, r, rw)
  1001. case wire.BuddyDelTempBuddies:
  1002. return rt.BuddyDelTempBuddies(ctx, instance, inFrame, r, rw)
  1003. }
  1004. case wire.Chat:
  1005. switch inFrame.SubGroup {
  1006. case wire.ChatChannelMsgToHost:
  1007. return rt.ChatChannelMsgToHost(ctx, instance, inFrame, r, rw)
  1008. }
  1009. case wire.ChatNav:
  1010. switch inFrame.SubGroup {
  1011. case wire.ChatNavCreateRoom:
  1012. return rt.ChatNavCreateRoom(ctx, instance, inFrame, r, rw)
  1013. case wire.ChatNavRequestChatRights:
  1014. return rt.ChatNavRequestChatRights(ctx, instance, inFrame, r, rw)
  1015. case wire.ChatNavRequestExchangeInfo:
  1016. return rt.ChatNavRequestExchangeInfo(ctx, instance, inFrame, r, rw)
  1017. case wire.ChatNavRequestRoomInfo:
  1018. return rt.ChatNavRequestRoomInfo(ctx, instance, inFrame, r, rw)
  1019. }
  1020. case wire.Feedbag:
  1021. switch inFrame.SubGroup {
  1022. case wire.FeedbagDeleteItem:
  1023. return rt.FeedbagDeleteItem(ctx, instance, inFrame, r, rw)
  1024. case wire.FeedbagEndCluster:
  1025. return rt.FeedbagEndCluster(ctx, instance, inFrame, r, rw)
  1026. case wire.FeedbagInsertItem:
  1027. return rt.FeedbagInsertItem(ctx, instance, inFrame, r, rw)
  1028. case wire.FeedbagQuery:
  1029. return rt.FeedbagQuery(ctx, instance, inFrame, r, rw)
  1030. case wire.FeedbagQueryIfModified:
  1031. return rt.FeedbagQueryIfModified(ctx, instance, inFrame, r, rw)
  1032. case wire.FeedbagPreAuthorizeBuddy:
  1033. return rt.FeedbagPreAuthorizeBuddy(ctx, instance, inFrame, r, rw)
  1034. case wire.FeedbagRequestAuthorizeToHost:
  1035. return rt.FeedbagRequestAuthorizeToHost(ctx, instance, inFrame, r, rw)
  1036. case wire.FeedbagRespondAuthorizeToHost:
  1037. return rt.FeedbagRespondAuthorizeToHost(ctx, instance, inFrame, r, rw)
  1038. case wire.FeedbagRightsQuery:
  1039. return rt.FeedbagRightsQuery(ctx, instance, inFrame, r, rw)
  1040. case wire.FeedbagStartCluster:
  1041. return rt.FeedbagStartCluster(ctx, instance, inFrame, r, rw)
  1042. case wire.FeedbagUpdateItem:
  1043. return rt.FeedbagUpdateItem(ctx, instance, inFrame, r, rw)
  1044. case wire.FeedbagUse:
  1045. return rt.FeedbagUse(ctx, instance, inFrame, r, rw)
  1046. }
  1047. case wire.Invite:
  1048. switch inFrame.SubGroup {
  1049. case wire.InviteRequestQuery, wire.InviteRequestReply:
  1050. return rt.InviteRequest(ctx, instance, inFrame, r, rw)
  1051. }
  1052. case wire.ICQ:
  1053. switch inFrame.SubGroup {
  1054. case wire.ICQDBQuery:
  1055. return rt.ICQDBQuery(ctx, instance, inFrame, r, rw)
  1056. }
  1057. case wire.ICBM:
  1058. switch inFrame.SubGroup {
  1059. case wire.ICBMAddParameters:
  1060. return rt.ICBMAddParameters(ctx, instance, inFrame, r, rw)
  1061. case wire.ICBMChannelMsgToHost:
  1062. return rt.ICBMChannelMsgToHost(ctx, instance, inFrame, r, rw)
  1063. case wire.ICBMClientErr:
  1064. return rt.ICBMClientErr(ctx, instance, inFrame, r, rw)
  1065. case wire.ICBMClientEvent:
  1066. return rt.ICBMClientEvent(ctx, instance, inFrame, r, rw)
  1067. case wire.ICBMEvilRequest:
  1068. return rt.ICBMEvilRequest(ctx, instance, inFrame, r, rw)
  1069. case wire.ICBMParameterQuery:
  1070. return rt.ICBMParameterQuery(ctx, instance, inFrame, r, rw)
  1071. case wire.ICBMOfflineRetrieve:
  1072. return rt.ICBMOfflineRetrieve(ctx, instance, inFrame, rw)
  1073. }
  1074. case wire.Locate:
  1075. switch inFrame.SubGroup {
  1076. case wire.LocateGetDirInfo:
  1077. return rt.LocateGetDirInfo(ctx, instance, inFrame, r, rw)
  1078. case wire.LocateRightsQuery:
  1079. return rt.LocateRightsQuery(ctx, instance, inFrame, r, rw)
  1080. case wire.LocateSetDirInfo:
  1081. return rt.LocateSetDirInfo(ctx, instance, inFrame, r, rw)
  1082. case wire.LocateSetInfo:
  1083. return rt.LocateSetInfo(ctx, instance, inFrame, r, rw)
  1084. case wire.LocateSetKeywordInfo:
  1085. return rt.LocateSetKeywordInfo(ctx, instance, inFrame, r, rw)
  1086. case wire.LocateUserInfoQuery:
  1087. return rt.LocateUserInfoQuery(ctx, instance, inFrame, r, rw)
  1088. case wire.LocateUserInfoQuery2:
  1089. return rt.LocateUserInfoQuery2(ctx, instance, inFrame, r, rw)
  1090. }
  1091. case wire.MDir:
  1092. return rt.MDirRequest(ctx, instance, inFrame, r, rw)
  1093. case wire.ODir:
  1094. switch inFrame.SubGroup {
  1095. case wire.ODirInfoQuery:
  1096. return rt.ODirInfoQuery(ctx, instance, inFrame, r, rw)
  1097. case wire.ODirKeywordListQuery:
  1098. return rt.ODirKeywordListQuery(ctx, instance, inFrame, r, rw)
  1099. }
  1100. case wire.OService:
  1101. switch inFrame.SubGroup {
  1102. case wire.OServiceClientOnline:
  1103. return rt.OServiceClientOnline(ctx, server, instance, inFrame, r, rw)
  1104. case wire.OServiceClientVersions:
  1105. return rt.OServiceClientVersions(ctx, instance, inFrame, r, rw)
  1106. case wire.OServiceIdleNotification:
  1107. return rt.OServiceIdleNotification(ctx, instance, inFrame, r, rw)
  1108. case wire.OServiceNoop:
  1109. return rt.OServiceNoop(ctx, instance, inFrame, r, rw)
  1110. case wire.OServiceRateParamsQuery:
  1111. return rt.OServiceRateParamsQuery(ctx, instance, inFrame, r, rw)
  1112. case wire.OServiceRateParamsSubAdd:
  1113. return rt.OServiceRateParamsSubAdd(ctx, instance, inFrame, r, rw)
  1114. case wire.OServiceServiceRequest:
  1115. return rt.OServiceServiceRequest(ctx, server, instance, inFrame, r, rw, listener)
  1116. case wire.OServiceSetPrivacyFlags:
  1117. return rt.OServiceSetPrivacyFlags(ctx, instance, inFrame, r, rw)
  1118. case wire.OServiceSetUserInfoFields:
  1119. return rt.OServiceSetUserInfoFields(ctx, instance, inFrame, r, rw)
  1120. case wire.OServiceUserInfoQuery:
  1121. return rt.OServiceUserInfoQuery(ctx, instance, inFrame, r, rw)
  1122. case wire.OServiceProbeReq:
  1123. return rt.OServiceProbeReq(ctx, instance, inFrame, r, rw)
  1124. }
  1125. case wire.PermitDeny:
  1126. switch inFrame.SubGroup {
  1127. case wire.PermitDenyAddDenyListEntries:
  1128. return rt.PermitDenyAddDenyListEntries(ctx, instance, inFrame, r, rw)
  1129. case wire.PermitDenyAddPermListEntries:
  1130. return rt.PermitDenyAddPermListEntries(ctx, instance, inFrame, r, rw)
  1131. case wire.PermitDenyDelDenyListEntries:
  1132. return rt.PermitDenyDelDenyListEntries(ctx, instance, inFrame, r, rw)
  1133. case wire.PermitDenyDelPermListEntries:
  1134. return rt.PermitDenyDelPermListEntries(ctx, instance, inFrame, r, rw)
  1135. case wire.PermitDenyRightsQuery:
  1136. return rt.PermitDenyRightsQuery(ctx, instance, inFrame, r, rw)
  1137. case wire.PermitDenySetGroupPermitMask:
  1138. return rt.PermitDenySetGroupPermitMask(ctx, instance, inFrame, r, rw)
  1139. }
  1140. case wire.Plugin:
  1141. return rt.PluginRequest(ctx, instance, inFrame, r, rw)
  1142. case wire.Stats:
  1143. switch inFrame.SubGroup {
  1144. case wire.StatsReportEvents:
  1145. return rt.StatsReportEvents(ctx, instance, inFrame, r, rw)
  1146. }
  1147. case wire.UserLookup:
  1148. switch inFrame.SubGroup {
  1149. case wire.UserLookupFindByEmail:
  1150. return rt.UserLookupFindByEmail(ctx, instance, inFrame, r, rw)
  1151. }
  1152. }
  1153. return ErrRouteNotFound
  1154. }