You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

server.go 35KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999
  1. // Copyright (c) 2012-2014 Jeremy Latt
  2. // Copyright (c) 2014-2015 Edmund Huber
  3. // Copyright (c) 2016-2017 Daniel Oaks <daniel@danieloaks.net>
  4. // released under the MIT license
  5. package irc
  6. import (
  7. "fmt"
  8. "net"
  9. "net/http"
  10. _ "net/http/pprof"
  11. "os"
  12. "os/signal"
  13. "strconv"
  14. "strings"
  15. "sync"
  16. "syscall"
  17. "time"
  18. "unsafe"
  19. "github.com/goshuirc/irc-go/ircfmt"
  20. "github.com/oragono/oragono/irc/caps"
  21. "github.com/oragono/oragono/irc/connection_limits"
  22. "github.com/oragono/oragono/irc/flatip"
  23. "github.com/oragono/oragono/irc/history"
  24. "github.com/oragono/oragono/irc/logger"
  25. "github.com/oragono/oragono/irc/modes"
  26. "github.com/oragono/oragono/irc/mysql"
  27. "github.com/oragono/oragono/irc/sno"
  28. "github.com/oragono/oragono/irc/utils"
  29. "github.com/tidwall/buntdb"
  30. )
  31. var (
  32. // common error line to sub values into
  33. errorMsg = "ERROR :%s\r\n"
  34. // three final parameters of 004 RPL_MYINFO, enumerating our supported modes
  35. rplMyInfo1, rplMyInfo2, rplMyInfo3 = modes.RplMyInfo()
  36. // CHANMODES isupport token
  37. chanmodesToken = modes.ChanmodesToken()
  38. // whitelist of caps to serve on the STS-only listener. In particular,
  39. // never advertise SASL, to discourage people from sending their passwords:
  40. stsOnlyCaps = caps.NewSet(caps.STS, caps.MessageTags, caps.ServerTime, caps.Batch, caps.LabeledResponse, caps.EchoMessage, caps.Nope)
  41. // we only have standard channels for now. TODO: any updates to this
  42. // will also need to be reflected in CasefoldChannel
  43. chanTypes = "#"
  44. throttleMessage = "You have attempted to connect too many times within a short duration. Wait a while, and you will be able to connect."
  45. )
  46. // Server is the main Oragono server.
  47. type Server struct {
  48. accounts AccountManager
  49. channels ChannelManager
  50. channelRegistry ChannelRegistry
  51. clients ClientManager
  52. config unsafe.Pointer
  53. configFilename string
  54. connectionLimiter connection_limits.Limiter
  55. ctime time.Time
  56. dlines *DLineManager
  57. helpIndexManager HelpIndexManager
  58. klines *KLineManager
  59. listeners map[string]IRCListener
  60. logger *logger.Manager
  61. monitorManager MonitorManager
  62. name string
  63. nameCasefolded string
  64. rehashMutex sync.Mutex // tier 4
  65. rehashSignal chan os.Signal
  66. pprofServer *http.Server
  67. resumeManager ResumeManager
  68. signals chan os.Signal
  69. snomasks SnoManager
  70. store *buntdb.DB
  71. historyDB mysql.MySQL
  72. torLimiter connection_limits.TorLimiter
  73. whoWas WhoWasList
  74. stats Stats
  75. semaphores ServerSemaphores
  76. defcon uint32
  77. }
  78. // NewServer returns a new Oragono server.
  79. func NewServer(config *Config, logger *logger.Manager) (*Server, error) {
  80. // initialize data structures
  81. server := &Server{
  82. ctime: time.Now().UTC(),
  83. listeners: make(map[string]IRCListener),
  84. logger: logger,
  85. rehashSignal: make(chan os.Signal, 1),
  86. signals: make(chan os.Signal, len(ServerExitSignals)),
  87. defcon: 5,
  88. }
  89. server.clients.Initialize()
  90. server.semaphores.Initialize()
  91. server.resumeManager.Initialize(server)
  92. server.whoWas.Initialize(config.Limits.WhowasEntries)
  93. server.monitorManager.Initialize()
  94. server.snomasks.Initialize()
  95. if err := server.applyConfig(config); err != nil {
  96. return nil, err
  97. }
  98. // Attempt to clean up when receiving these signals.
  99. signal.Notify(server.signals, ServerExitSignals...)
  100. signal.Notify(server.rehashSignal, syscall.SIGHUP)
  101. return server, nil
  102. }
  103. // Shutdown shuts down the server.
  104. func (server *Server) Shutdown() {
  105. //TODO(dan): Make sure we disallow new nicks
  106. for _, client := range server.clients.AllClients() {
  107. client.Notice("Server is shutting down")
  108. if client.AlwaysOn() {
  109. client.Store(IncludeLastSeen)
  110. }
  111. }
  112. if err := server.store.Close(); err != nil {
  113. server.logger.Error("shutdown", fmt.Sprintln("Could not close datastore:", err))
  114. }
  115. server.historyDB.Close()
  116. }
  117. // Run starts the server.
  118. func (server *Server) Run() {
  119. // defer closing db/store
  120. defer server.store.Close()
  121. for {
  122. select {
  123. case <-server.signals:
  124. server.Shutdown()
  125. return
  126. case <-server.rehashSignal:
  127. go func() {
  128. server.logger.Info("server", "Rehashing due to SIGHUP")
  129. server.rehash()
  130. }()
  131. }
  132. }
  133. }
  134. func (server *Server) checkBans(config *Config, ipaddr net.IP, checkScripts bool) (banned bool, requireSASL bool, message string) {
  135. if server.Defcon() == 1 {
  136. if !(ipaddr.IsLoopback() || utils.IPInNets(ipaddr, server.Config().Server.secureNets)) {
  137. return true, false, "New connections to this server are temporarily restricted"
  138. }
  139. }
  140. flat := flatip.FromNetIP(ipaddr)
  141. // check DLINEs
  142. isBanned, info := server.dlines.CheckIP(flat)
  143. if isBanned {
  144. server.logger.Info("connect-ip", "Client rejected by d-line", ipaddr.String())
  145. return true, false, info.BanMessage("You are banned from this server (%s)")
  146. }
  147. // check connection limits
  148. err := server.connectionLimiter.AddClient(flat)
  149. if err == connection_limits.ErrLimitExceeded {
  150. // too many connections from one client, tell the client and close the connection
  151. server.logger.Info("connect-ip", "Client rejected for connection limit", ipaddr.String())
  152. return true, false, "Too many clients from your network"
  153. } else if err == connection_limits.ErrThrottleExceeded {
  154. server.logger.Info("connect-ip", "Client exceeded connection throttle", ipaddr.String())
  155. return true, false, throttleMessage
  156. } else if err != nil {
  157. server.logger.Warning("internal", "unexpected ban result", err.Error())
  158. }
  159. if checkScripts && config.Server.IPCheckScript.Enabled {
  160. output, err := CheckIPBan(server.semaphores.IPCheckScript, config.Server.IPCheckScript, ipaddr)
  161. if err != nil {
  162. server.logger.Error("internal", "couldn't check IP ban script", ipaddr.String(), err.Error())
  163. return false, false, ""
  164. }
  165. // TODO: currently no way to cache results other than IPBanned
  166. if output.Result == IPBanned && output.CacheSeconds != 0 {
  167. network, err := utils.NormalizedNetFromString(output.CacheNet)
  168. if err != nil {
  169. server.logger.Error("internal", "invalid dline net from IP ban script", ipaddr.String(), output.CacheNet)
  170. } else {
  171. dlineDuration := time.Duration(output.CacheSeconds) * time.Second
  172. err := server.dlines.AddNetwork(network, dlineDuration, output.BanMessage, "", "")
  173. if err != nil {
  174. server.logger.Error("internal", "couldn't set dline from IP ban script", ipaddr.String(), err.Error())
  175. }
  176. }
  177. }
  178. if output.Result == IPBanned {
  179. // XXX roll back IP connection/throttling addition for the IP
  180. server.connectionLimiter.RemoveClient(flat)
  181. server.logger.Info("connect-ip", "Rejected client due to ip-check-script", ipaddr.String())
  182. return true, false, output.BanMessage
  183. } else if output.Result == IPRequireSASL {
  184. server.logger.Info("connect-ip", "Requiring SASL from client due to ip-check-script", ipaddr.String())
  185. return false, true, output.BanMessage
  186. }
  187. }
  188. return false, false, ""
  189. }
  190. func (server *Server) checkTorLimits() (banned bool, message string) {
  191. switch server.torLimiter.AddClient() {
  192. case connection_limits.ErrLimitExceeded:
  193. return true, "Too many clients from the Tor network"
  194. case connection_limits.ErrThrottleExceeded:
  195. return true, "Exceeded connection throttle for the Tor network"
  196. default:
  197. return false, ""
  198. }
  199. }
  200. //
  201. // server functionality
  202. //
  203. func (server *Server) tryRegister(c *Client, session *Session) (exiting bool) {
  204. // if the session just sent us a RESUME line, try to resume
  205. if session.resumeDetails != nil {
  206. session.tryResume()
  207. return // whether we succeeded or failed, either way `c` is not getting registered
  208. }
  209. // XXX PROXY or WEBIRC MUST be sent as the first line of the session;
  210. // if we are here at all that means we have the final value of the IP
  211. if session.rawHostname == "" {
  212. session.client.lookupHostname(session, false)
  213. }
  214. // try to complete registration normally
  215. // XXX(#1057) username can be filled in by an ident query without the client
  216. // having sent USER: check for both username and realname to ensure they did
  217. if c.preregNick == "" || c.username == "" || c.realname == "" || session.capState == caps.NegotiatingState {
  218. return
  219. }
  220. if c.isSTSOnly {
  221. server.playSTSBurst(session)
  222. return true
  223. }
  224. // client MUST send PASS if necessary, or authenticate with SASL if necessary,
  225. // before completing the other registration commands
  226. config := server.Config()
  227. authOutcome := c.isAuthorized(server, config, session, c.requireSASL)
  228. var quitMessage string
  229. switch authOutcome {
  230. case authFailPass:
  231. quitMessage = c.t("Password incorrect")
  232. c.Send(nil, server.name, ERR_PASSWDMISMATCH, "*", quitMessage)
  233. case authFailSaslRequired, authFailTorSaslRequired:
  234. quitMessage = c.requireSASLMessage
  235. if quitMessage == "" {
  236. quitMessage = c.t("You must log in with SASL to join this server")
  237. }
  238. c.Send(nil, c.server.name, "FAIL", "*", "ACCOUNT_REQUIRED", quitMessage)
  239. }
  240. if authOutcome != authSuccess {
  241. c.Quit(quitMessage, nil)
  242. return true
  243. }
  244. c.requireSASLMessage = ""
  245. rb := NewResponseBuffer(session)
  246. nickError := performNickChange(server, c, c, session, c.preregNick, rb)
  247. rb.Send(true)
  248. if nickError == errInsecureReattach {
  249. c.Quit(c.t("You can't mix secure and insecure connections to this account"), nil)
  250. return true
  251. } else if nickError != nil {
  252. c.preregNick = ""
  253. return false
  254. }
  255. if session.client != c {
  256. // reattached, bail out.
  257. // we'll play the reg burst later, on the new goroutine associated with
  258. // (thisSession, otherClient). This is to avoid having to transfer state
  259. // like nickname, hostname, etc. to show the correct values in the reg burst.
  260. return false
  261. }
  262. // Apply default user modes (without updating the invisible counter)
  263. // The number of invisible users will be updated by server.stats.Register
  264. // if we're using default user mode +i.
  265. for _, defaultMode := range config.Accounts.defaultUserModes {
  266. c.SetMode(defaultMode, true)
  267. }
  268. // count new user in statistics (before checking KLINEs, see #1303)
  269. server.stats.Register(c.HasMode(modes.Invisible))
  270. // check KLINEs
  271. isBanned, info := server.klines.CheckMasks(c.AllNickmasks()...)
  272. if isBanned {
  273. c.Quit(info.BanMessage(c.t("You are banned from this server (%s)")), nil)
  274. return true
  275. }
  276. server.playRegistrationBurst(session)
  277. return false
  278. }
  279. func (server *Server) playSTSBurst(session *Session) {
  280. nick := utils.SafeErrorParam(session.client.preregNick)
  281. session.Send(nil, server.name, RPL_WELCOME, nick, fmt.Sprintf("Welcome to the Internet Relay Network %s", nick))
  282. session.Send(nil, server.name, RPL_YOURHOST, nick, fmt.Sprintf("Your host is %[1]s, running version %[2]s", server.name, "oragono"))
  283. session.Send(nil, server.name, RPL_CREATED, nick, fmt.Sprintf("This server was created %s", time.Time{}.Format(time.RFC1123)))
  284. session.Send(nil, server.name, RPL_MYINFO, nick, server.name, "oragono", "o", "o", "o")
  285. session.Send(nil, server.name, RPL_ISUPPORT, nick, "CASEMAPPING=ascii", "are supported by this server")
  286. session.Send(nil, server.name, ERR_NOMOTD, nick, "MOTD is unavailable")
  287. for _, line := range server.Config().Server.STS.bannerLines {
  288. session.Send(nil, server.name, "NOTICE", nick, line)
  289. }
  290. }
  291. func (server *Server) playRegistrationBurst(session *Session) {
  292. c := session.client
  293. // continue registration
  294. d := c.Details()
  295. server.logger.Info("connect", fmt.Sprintf("Client connected [%s] [u:%s] [r:%s]", d.nick, d.username, d.realname))
  296. server.snomasks.Send(sno.LocalConnects, fmt.Sprintf("Client connected [%s] [u:%s] [h:%s] [ip:%s] [r:%s]", d.nick, d.username, session.rawHostname, session.IP().String(), d.realname))
  297. if d.account != "" {
  298. server.sendLoginSnomask(d.nickMask, d.accountName)
  299. }
  300. // send welcome text
  301. //NOTE(dan): we specifically use the NICK here instead of the nickmask
  302. // see http://modern.ircdocs.horse/#rplwelcome-001 for details on why we avoid using the nickmask
  303. config := server.Config()
  304. session.Send(nil, server.name, RPL_WELCOME, d.nick, fmt.Sprintf(c.t("Welcome to the %s IRC Network %s"), config.Network.Name, d.nick))
  305. session.Send(nil, server.name, RPL_YOURHOST, d.nick, fmt.Sprintf(c.t("Your host is %[1]s, running version %[2]s"), server.name, Ver))
  306. session.Send(nil, server.name, RPL_CREATED, d.nick, fmt.Sprintf(c.t("This server was created %s"), server.ctime.Format(time.RFC1123)))
  307. session.Send(nil, server.name, RPL_MYINFO, d.nick, server.name, Ver, rplMyInfo1, rplMyInfo2, rplMyInfo3)
  308. rb := NewResponseBuffer(session)
  309. server.RplISupport(c, rb)
  310. server.Lusers(c, rb)
  311. server.MOTD(c, rb)
  312. rb.Send(true)
  313. modestring := c.ModeString()
  314. if modestring != "+" {
  315. session.Send(nil, server.name, RPL_UMODEIS, d.nick, modestring)
  316. }
  317. c.attemptAutoOper(session)
  318. if server.logger.IsLoggingRawIO() {
  319. session.Send(nil, c.server.name, "NOTICE", d.nick, c.t("This server is in debug mode and is logging all user I/O. If you do not wish for everything you send to be readable by the server owner(s), please disconnect."))
  320. }
  321. }
  322. // RplISupport outputs our ISUPPORT lines to the client. This is used on connection and in VERSION responses.
  323. func (server *Server) RplISupport(client *Client, rb *ResponseBuffer) {
  324. translatedISupport := client.t("are supported by this server")
  325. nick := client.Nick()
  326. config := server.Config()
  327. for _, cachedTokenLine := range config.Server.isupport.CachedReply {
  328. length := len(cachedTokenLine) + 2
  329. tokenline := make([]string, length)
  330. tokenline[0] = nick
  331. copy(tokenline[1:], cachedTokenLine)
  332. tokenline[length-1] = translatedISupport
  333. rb.Add(nil, server.name, RPL_ISUPPORT, tokenline...)
  334. }
  335. }
  336. func (server *Server) Lusers(client *Client, rb *ResponseBuffer) {
  337. nick := client.Nick()
  338. stats := server.stats.GetValues()
  339. rb.Add(nil, server.name, RPL_LUSERCLIENT, nick, fmt.Sprintf(client.t("There are %[1]d users and %[2]d invisible on %[3]d server(s)"), stats.Total-stats.Invisible, stats.Invisible, 1))
  340. rb.Add(nil, server.name, RPL_LUSEROP, nick, strconv.Itoa(stats.Operators), client.t("IRC Operators online"))
  341. rb.Add(nil, server.name, RPL_LUSERUNKNOWN, nick, strconv.Itoa(stats.Unknown), client.t("unregistered connections"))
  342. rb.Add(nil, server.name, RPL_LUSERCHANNELS, nick, strconv.Itoa(server.channels.Len()), client.t("channels formed"))
  343. rb.Add(nil, server.name, RPL_LUSERME, nick, fmt.Sprintf(client.t("I have %[1]d clients and %[2]d servers"), stats.Total, 0))
  344. total := strconv.Itoa(stats.Total)
  345. max := strconv.Itoa(stats.Max)
  346. rb.Add(nil, server.name, RPL_LOCALUSERS, nick, total, max, fmt.Sprintf(client.t("Current local users %[1]s, max %[2]s"), total, max))
  347. rb.Add(nil, server.name, RPL_GLOBALUSERS, nick, total, max, fmt.Sprintf(client.t("Current global users %[1]s, max %[2]s"), total, max))
  348. }
  349. // MOTD serves the Message of the Day.
  350. func (server *Server) MOTD(client *Client, rb *ResponseBuffer) {
  351. motdLines := server.Config().Server.motdLines
  352. if len(motdLines) < 1 {
  353. rb.Add(nil, server.name, ERR_NOMOTD, client.nick, client.t("MOTD File is missing"))
  354. return
  355. }
  356. rb.Add(nil, server.name, RPL_MOTDSTART, client.nick, fmt.Sprintf(client.t("- %s Message of the day - "), server.name))
  357. for _, line := range motdLines {
  358. rb.Add(nil, server.name, RPL_MOTD, client.nick, line)
  359. }
  360. rb.Add(nil, server.name, RPL_ENDOFMOTD, client.nick, client.t("End of MOTD command"))
  361. }
  362. // WhoisChannelsNames returns the common channel names between two users.
  363. func (client *Client) WhoisChannelsNames(target *Client, multiPrefix bool) []string {
  364. var chstrs []string
  365. for _, channel := range target.Channels() {
  366. // channel is secret and the target can't see it
  367. if !client.HasMode(modes.Operator) {
  368. if (target.HasMode(modes.Invisible) || channel.flags.HasMode(modes.Secret)) && !channel.hasClient(client) {
  369. continue
  370. }
  371. }
  372. chstrs = append(chstrs, channel.ClientPrefixes(target, multiPrefix)+channel.name)
  373. }
  374. return chstrs
  375. }
  376. func (client *Client) getWhoisOf(target *Client, hasPrivs bool, rb *ResponseBuffer) {
  377. cnick := client.Nick()
  378. targetInfo := target.Details()
  379. rb.Add(nil, client.server.name, RPL_WHOISUSER, cnick, targetInfo.nick, targetInfo.username, targetInfo.hostname, "*", targetInfo.realname)
  380. tnick := targetInfo.nick
  381. whoischannels := client.WhoisChannelsNames(target, rb.session.capabilities.Has(caps.MultiPrefix))
  382. if whoischannels != nil {
  383. rb.Add(nil, client.server.name, RPL_WHOISCHANNELS, cnick, tnick, strings.Join(whoischannels, " "))
  384. }
  385. if target.HasMode(modes.Operator) && operStatusVisible(client, target, hasPrivs) {
  386. tOper := target.Oper()
  387. if tOper != nil {
  388. rb.Add(nil, client.server.name, RPL_WHOISOPERATOR, cnick, tnick, tOper.WhoisLine)
  389. }
  390. }
  391. if client == target || hasPrivs {
  392. rb.Add(nil, client.server.name, RPL_WHOISACTUALLY, cnick, tnick, fmt.Sprintf("%s@%s", targetInfo.username, target.RawHostname()), target.IPString(), client.t("Actual user@host, Actual IP"))
  393. rb.Add(nil, client.server.name, RPL_WHOISMODES, cnick, tnick, fmt.Sprintf(client.t("is using modes +%s"), target.modes.String()))
  394. }
  395. if target.HasMode(modes.TLS) {
  396. rb.Add(nil, client.server.name, RPL_WHOISSECURE, cnick, tnick, client.t("is using a secure connection"))
  397. }
  398. if targetInfo.accountName != "*" {
  399. rb.Add(nil, client.server.name, RPL_WHOISACCOUNT, cnick, tnick, targetInfo.accountName, client.t("is logged in as"))
  400. }
  401. if target.HasMode(modes.Bot) {
  402. rb.Add(nil, client.server.name, RPL_WHOISBOT, cnick, tnick, ircfmt.Unescape(fmt.Sprintf(client.t("is a $bBot$b on %s"), client.server.Config().Network.Name)))
  403. }
  404. if client == target || hasPrivs {
  405. for _, session := range target.Sessions() {
  406. if session.certfp != "" {
  407. rb.Add(nil, client.server.name, RPL_WHOISCERTFP, cnick, tnick, fmt.Sprintf(client.t("has client certificate fingerprint %s"), session.certfp))
  408. }
  409. }
  410. }
  411. rb.Add(nil, client.server.name, RPL_WHOISIDLE, cnick, tnick, strconv.FormatUint(target.IdleSeconds(), 10), strconv.FormatInt(target.SignonTime(), 10), client.t("seconds idle, signon time"))
  412. if away, awayMessage := target.Away(); away {
  413. rb.Add(nil, client.server.name, RPL_AWAY, cnick, tnick, awayMessage)
  414. }
  415. }
  416. // rehash reloads the config and applies the changes from the config file.
  417. func (server *Server) rehash() error {
  418. server.logger.Info("server", "Attempting rehash")
  419. // only let one REHASH go on at a time
  420. server.rehashMutex.Lock()
  421. defer server.rehashMutex.Unlock()
  422. server.logger.Debug("server", "Got rehash lock")
  423. config, err := LoadConfig(server.configFilename)
  424. if err != nil {
  425. server.logger.Error("server", "failed to load config file", err.Error())
  426. return err
  427. }
  428. err = server.applyConfig(config)
  429. if err != nil {
  430. server.logger.Error("server", "Failed to rehash", err.Error())
  431. return err
  432. }
  433. server.logger.Info("server", "Rehash completed successfully")
  434. return nil
  435. }
  436. func (server *Server) applyConfig(config *Config) (err error) {
  437. oldConfig := server.Config()
  438. initial := oldConfig == nil
  439. if initial {
  440. server.configFilename = config.Filename
  441. server.name = config.Server.Name
  442. server.nameCasefolded = config.Server.nameCasefolded
  443. globalCasemappingSetting = config.Server.Casemapping
  444. globalUtf8EnforcementSetting = config.Server.EnforceUtf8
  445. } else {
  446. // enforce configs that can't be changed after launch:
  447. if server.name != config.Server.Name {
  448. return fmt.Errorf("Server name cannot be changed after launching the server, rehash aborted")
  449. } else if oldConfig.Datastore.Path != config.Datastore.Path {
  450. return fmt.Errorf("Datastore path cannot be changed after launching the server, rehash aborted")
  451. } else if globalCasemappingSetting != config.Server.Casemapping {
  452. return fmt.Errorf("Casemapping cannot be changed after launching the server, rehash aborted")
  453. } else if globalUtf8EnforcementSetting != config.Server.EnforceUtf8 {
  454. return fmt.Errorf("UTF-8 enforcement cannot be changed after launching the server, rehash aborted")
  455. } else if oldConfig.Accounts.Multiclient.AlwaysOn != config.Accounts.Multiclient.AlwaysOn {
  456. return fmt.Errorf("Default always-on setting cannot be changed after launching the server, rehash aborted")
  457. } else if oldConfig.Server.Relaymsg.Enabled != config.Server.Relaymsg.Enabled {
  458. return fmt.Errorf("Cannot enable or disable relaying after launching the server, rehash aborted")
  459. } else if oldConfig.Server.Relaymsg.Separators != config.Server.Relaymsg.Separators {
  460. return fmt.Errorf("Cannot change relaying separators after launching the server, rehash aborted")
  461. } else if oldConfig.Server.IPCheckScript.MaxConcurrency != config.Server.IPCheckScript.MaxConcurrency ||
  462. oldConfig.Accounts.AuthScript.MaxConcurrency != config.Accounts.AuthScript.MaxConcurrency {
  463. return fmt.Errorf("Cannot change max-concurrency for scripts after launching the server, rehash aborted")
  464. } else if oldConfig.Server.OverrideServicesHostname != config.Server.OverrideServicesHostname {
  465. return fmt.Errorf("Cannot change override-services-hostname after launching the server, rehash aborted")
  466. } else if !oldConfig.Datastore.MySQL.Enabled && config.Datastore.MySQL.Enabled {
  467. return fmt.Errorf("Cannot enable MySQL after launching the server, rehash aborted")
  468. }
  469. }
  470. server.logger.Info("server", "Using config file", server.configFilename)
  471. // first, reload config sections for functionality implemented in subpackages:
  472. wasLoggingRawIO := !initial && server.logger.IsLoggingRawIO()
  473. err = server.logger.ApplyConfig(config.Logging)
  474. if err != nil {
  475. return err
  476. }
  477. nowLoggingRawIO := server.logger.IsLoggingRawIO()
  478. // notify existing clients if raw i/o logging was enabled by a rehash
  479. sendRawOutputNotice := !wasLoggingRawIO && nowLoggingRawIO
  480. server.connectionLimiter.ApplyConfig(&config.Server.IPLimits)
  481. tlConf := &config.Server.TorListeners
  482. server.torLimiter.Configure(tlConf.MaxConnections, tlConf.ThrottleDuration, tlConf.MaxConnectionsPerDuration)
  483. // Translations
  484. server.logger.Debug("server", "Regenerating HELP indexes for new languages")
  485. server.helpIndexManager.GenerateIndices(config.languageManager)
  486. if initial {
  487. maxIPConc := int(config.Server.IPCheckScript.MaxConcurrency)
  488. if maxIPConc != 0 {
  489. server.semaphores.IPCheckScript.Initialize(maxIPConc)
  490. }
  491. maxAuthConc := int(config.Accounts.AuthScript.MaxConcurrency)
  492. if maxAuthConc != 0 {
  493. server.semaphores.AuthScript.Initialize(maxAuthConc)
  494. }
  495. if err := overrideServicePrefixes(config.Server.OverrideServicesHostname); err != nil {
  496. return err
  497. }
  498. }
  499. if oldConfig != nil {
  500. // if certain features were enabled by rehash, we need to load the corresponding data
  501. // from the store
  502. if !oldConfig.Accounts.NickReservation.Enabled {
  503. server.accounts.buildNickToAccountIndex(config)
  504. }
  505. if !oldConfig.Channels.Registration.Enabled {
  506. server.channels.loadRegisteredChannels(config)
  507. }
  508. // resize history buffers as needed
  509. if config.historyChangedFrom(oldConfig) {
  510. for _, channel := range server.channels.Channels() {
  511. channel.resizeHistory(config)
  512. }
  513. for _, client := range server.clients.AllClients() {
  514. client.resizeHistory(config)
  515. }
  516. }
  517. if oldConfig.Accounts.Registration.Throttling != config.Accounts.Registration.Throttling {
  518. server.accounts.resetRegisterThrottle(config)
  519. }
  520. }
  521. server.logger.Info("server", "Using datastore", config.Datastore.Path)
  522. if initial {
  523. if err := server.loadDatastore(config); err != nil {
  524. return err
  525. }
  526. } else {
  527. if config.Datastore.MySQL.Enabled && config.Datastore.MySQL != oldConfig.Datastore.MySQL {
  528. server.historyDB.SetConfig(config.Datastore.MySQL)
  529. }
  530. }
  531. // now that the datastore is initialized, we can load the cloak secret from it
  532. // XXX this modifies config after the initial load, which is naughty,
  533. // but there's no data race because we haven't done SetConfig yet
  534. config.Server.Cloaks.SetSecret(LoadCloakSecret(server.store))
  535. // activate the new config
  536. server.SetConfig(config)
  537. // load [dk]-lines, registered users and channels, etc.
  538. if initial {
  539. if err := server.loadFromDatastore(config); err != nil {
  540. return err
  541. }
  542. }
  543. // burst new and removed caps
  544. addedCaps, removedCaps := config.Diff(oldConfig)
  545. var capBurstSessions []*Session
  546. added := make(map[caps.Version][]string)
  547. var removed []string
  548. if !addedCaps.Empty() || !removedCaps.Empty() {
  549. capBurstSessions = server.clients.AllWithCapsNotify()
  550. added[caps.Cap301] = addedCaps.Strings(caps.Cap301, config.Server.capValues, 0)
  551. added[caps.Cap302] = addedCaps.Strings(caps.Cap302, config.Server.capValues, 0)
  552. // removed never has values, so we leave it as Cap301
  553. removed = removedCaps.Strings(caps.Cap301, config.Server.capValues, 0)
  554. }
  555. for _, sSession := range capBurstSessions {
  556. // DEL caps and then send NEW ones so that updated caps get removed/added correctly
  557. if !removedCaps.Empty() {
  558. for _, capStr := range removed {
  559. sSession.Send(nil, server.name, "CAP", sSession.client.Nick(), "DEL", capStr)
  560. }
  561. }
  562. if !addedCaps.Empty() {
  563. for _, capStr := range added[sSession.capVersion] {
  564. sSession.Send(nil, server.name, "CAP", sSession.client.Nick(), "NEW", capStr)
  565. }
  566. }
  567. }
  568. server.setupPprofListener(config)
  569. // set RPL_ISUPPORT
  570. var newISupportReplies [][]string
  571. if oldConfig != nil {
  572. newISupportReplies = oldConfig.Server.isupport.GetDifference(&config.Server.isupport)
  573. }
  574. if len(config.Server.ProxyAllowedFrom) != 0 {
  575. server.logger.Info("server", "Proxied IPs will be accepted from", strings.Join(config.Server.ProxyAllowedFrom, ", "))
  576. }
  577. // we are now open for business
  578. err = server.setupListeners(config)
  579. if !initial {
  580. // push new info to all of our clients
  581. for _, sClient := range server.clients.AllClients() {
  582. for _, tokenline := range newISupportReplies {
  583. sClient.Send(nil, server.name, RPL_ISUPPORT, append([]string{sClient.nick}, tokenline...)...)
  584. }
  585. if sendRawOutputNotice {
  586. sClient.Notice(sClient.t("This server is in debug mode and is logging all user I/O. If you do not wish for everything you send to be readable by the server owner(s), please disconnect."))
  587. }
  588. }
  589. }
  590. return err
  591. }
  592. func (server *Server) setupPprofListener(config *Config) {
  593. pprofListener := ""
  594. if config.Debug.PprofListener != nil {
  595. pprofListener = *config.Debug.PprofListener
  596. }
  597. if server.pprofServer != nil {
  598. if pprofListener == "" || (pprofListener != server.pprofServer.Addr) {
  599. server.logger.Info("server", "Stopping pprof listener", server.pprofServer.Addr)
  600. server.pprofServer.Close()
  601. server.pprofServer = nil
  602. }
  603. }
  604. if pprofListener != "" && server.pprofServer == nil {
  605. ps := http.Server{
  606. Addr: pprofListener,
  607. }
  608. go func() {
  609. if err := ps.ListenAndServe(); err != nil {
  610. server.logger.Error("server", "pprof listener failed", err.Error())
  611. }
  612. }()
  613. server.pprofServer = &ps
  614. server.logger.Info("server", "Started pprof listener", server.pprofServer.Addr)
  615. }
  616. }
  617. func (server *Server) loadDatastore(config *Config) error {
  618. // open the datastore and load server state for which it (rather than config)
  619. // is the source of truth
  620. _, err := os.Stat(config.Datastore.Path)
  621. if os.IsNotExist(err) {
  622. server.logger.Warning("server", "database does not exist, creating it", config.Datastore.Path)
  623. err = initializeDB(config.Datastore.Path)
  624. if err != nil {
  625. return err
  626. }
  627. }
  628. db, err := OpenDatabase(config)
  629. if err == nil {
  630. server.store = db
  631. return nil
  632. } else {
  633. return fmt.Errorf("Failed to open datastore: %s", err.Error())
  634. }
  635. }
  636. func (server *Server) loadFromDatastore(config *Config) (err error) {
  637. // load *lines (from the datastores)
  638. server.logger.Debug("server", "Loading D/Klines")
  639. server.loadDLines()
  640. server.loadKLines()
  641. server.channelRegistry.Initialize(server)
  642. server.channels.Initialize(server)
  643. server.accounts.Initialize(server)
  644. if config.Datastore.MySQL.Enabled {
  645. server.historyDB.Initialize(server.logger, config.Datastore.MySQL)
  646. err = server.historyDB.Open()
  647. if err != nil {
  648. server.logger.Error("internal", "could not connect to mysql", err.Error())
  649. return err
  650. }
  651. }
  652. return nil
  653. }
  654. func (server *Server) setupListeners(config *Config) (err error) {
  655. logListener := func(addr string, config utils.ListenerConfig) {
  656. server.logger.Info("listeners",
  657. fmt.Sprintf("now listening on %s, tls=%t, proxy=%t, tor=%t, websocket=%t.", addr, (config.TLSConfig != nil), config.RequireProxy, config.Tor, config.WebSocket),
  658. )
  659. }
  660. // update or destroy all existing listeners
  661. for addr := range server.listeners {
  662. currentListener := server.listeners[addr]
  663. newConfig, stillConfigured := config.Server.trueListeners[addr]
  664. if stillConfigured {
  665. if reloadErr := currentListener.Reload(newConfig); reloadErr == nil {
  666. logListener(addr, newConfig)
  667. } else {
  668. // stop the listener; we will attempt to replace it below
  669. currentListener.Stop()
  670. delete(server.listeners, addr)
  671. }
  672. } else {
  673. currentListener.Stop()
  674. delete(server.listeners, addr)
  675. server.logger.Info("listeners", fmt.Sprintf("stopped listening on %s.", addr))
  676. }
  677. }
  678. publicPlaintextListener := ""
  679. // create new listeners that were not previously configured,
  680. // or that couldn't be reloaded above:
  681. for newAddr, newConfig := range config.Server.trueListeners {
  682. if strings.HasPrefix(newAddr, ":") && !newConfig.Tor && !newConfig.STSOnly && newConfig.TLSConfig == nil {
  683. publicPlaintextListener = newAddr
  684. }
  685. _, exists := server.listeners[newAddr]
  686. if !exists {
  687. // make a new listener
  688. newListener, newErr := NewListener(server, newAddr, newConfig, config.Server.UnixBindMode)
  689. if newErr == nil {
  690. server.listeners[newAddr] = newListener
  691. logListener(newAddr, newConfig)
  692. } else {
  693. server.logger.Error("server", "couldn't listen on", newAddr, newErr.Error())
  694. err = newErr
  695. }
  696. }
  697. }
  698. if publicPlaintextListener != "" {
  699. server.logger.Warning("listeners", fmt.Sprintf("Your server is configured with public plaintext listener %s. Consider disabling it for improved security and privacy.", publicPlaintextListener))
  700. }
  701. return
  702. }
  703. // Gets the abstract sequence from which we're going to query history;
  704. // we may already know the channel we're querying, or we may have
  705. // to look it up via a string query. This function is responsible for
  706. // privilege checking.
  707. func (server *Server) GetHistorySequence(providedChannel *Channel, client *Client, query string) (channel *Channel, sequence history.Sequence, err error) {
  708. config := server.Config()
  709. // 4 cases: {persistent, ephemeral} x {normal, conversation}
  710. // with ephemeral history, target is implicit in the choice of `hist`,
  711. // and correspondent is "" if we're retrieving a channel or *, and the correspondent's name
  712. // if we're retrieving a DM conversation ("query buffer"). with persistent history,
  713. // target is always nonempty, and correspondent is either empty or nonempty as before.
  714. var status HistoryStatus
  715. var target, correspondent string
  716. var hist *history.Buffer
  717. channel = providedChannel
  718. if channel == nil {
  719. if strings.HasPrefix(query, "#") {
  720. channel = server.channels.Get(query)
  721. if channel == nil {
  722. return
  723. }
  724. }
  725. }
  726. if channel != nil {
  727. if !channel.hasClient(client) {
  728. err = errInsufficientPrivs
  729. return
  730. }
  731. status, target = channel.historyStatus(config)
  732. switch status {
  733. case HistoryEphemeral:
  734. hist = &channel.history
  735. case HistoryPersistent:
  736. // already set `target`
  737. default:
  738. return
  739. }
  740. } else {
  741. status, target = client.historyStatus(config)
  742. if query != "*" {
  743. correspondent, err = CasefoldName(query)
  744. if err != nil {
  745. return
  746. }
  747. }
  748. switch status {
  749. case HistoryEphemeral:
  750. hist = &client.history
  751. case HistoryPersistent:
  752. // already set `target`, and `correspondent` if necessary
  753. default:
  754. return
  755. }
  756. }
  757. var cutoff time.Time
  758. if config.History.Restrictions.ExpireTime != 0 {
  759. cutoff = time.Now().UTC().Add(-time.Duration(config.History.Restrictions.ExpireTime))
  760. }
  761. // #836: registration date cutoff is always enforced for DMs
  762. if config.History.Restrictions.EnforceRegistrationDate || channel == nil {
  763. regCutoff := client.historyCutoff()
  764. // take the later of the two cutoffs
  765. if regCutoff.After(cutoff) {
  766. cutoff = regCutoff
  767. }
  768. }
  769. // #836 again: grace period is never applied to DMs
  770. if !cutoff.IsZero() && channel != nil {
  771. cutoff = cutoff.Add(-time.Duration(config.History.Restrictions.GracePeriod))
  772. }
  773. if hist != nil {
  774. sequence = hist.MakeSequence(correspondent, cutoff)
  775. } else if target != "" {
  776. sequence = server.historyDB.MakeSequence(target, correspondent, cutoff)
  777. }
  778. return
  779. }
  780. func (server *Server) ForgetHistory(accountName string) {
  781. // sanity check
  782. if accountName == "*" {
  783. return
  784. }
  785. config := server.Config()
  786. if !config.History.Enabled {
  787. return
  788. }
  789. if cfAccount, err := CasefoldName(accountName); err == nil {
  790. server.historyDB.Forget(cfAccount)
  791. }
  792. persistent := config.History.Persistent
  793. if persistent.Enabled && persistent.UnregisteredChannels && persistent.RegisteredChannels == PersistentMandatory && persistent.DirectMessages == PersistentMandatory {
  794. return
  795. }
  796. predicate := func(item *history.Item) bool { return item.AccountName == accountName }
  797. for _, channel := range server.channels.Channels() {
  798. channel.history.Delete(predicate)
  799. }
  800. for _, client := range server.clients.AllClients() {
  801. client.history.Delete(predicate)
  802. }
  803. }
  804. // deletes a message. target is a hint about what buffer it's in (not required for
  805. // persistent history, where all the msgids are indexed together). if accountName
  806. // is anything other than "*", it must match the recorded AccountName of the message
  807. func (server *Server) DeleteMessage(target, msgid, accountName string) (err error) {
  808. config := server.Config()
  809. var hist *history.Buffer
  810. if target != "" {
  811. if target[0] == '#' {
  812. channel := server.channels.Get(target)
  813. if channel != nil {
  814. if status, _ := channel.historyStatus(config); status == HistoryEphemeral {
  815. hist = &channel.history
  816. }
  817. }
  818. } else {
  819. client := server.clients.Get(target)
  820. if client != nil {
  821. if status, _ := client.historyStatus(config); status == HistoryEphemeral {
  822. hist = &client.history
  823. }
  824. }
  825. }
  826. }
  827. if hist == nil {
  828. err = server.historyDB.DeleteMsgid(msgid, accountName)
  829. } else {
  830. count := hist.Delete(func(item *history.Item) bool {
  831. return item.Message.Msgid == msgid && (accountName == "*" || item.AccountName == accountName)
  832. })
  833. if count == 0 {
  834. err = errNoop
  835. }
  836. }
  837. return
  838. }
  839. // elistMatcher takes and matches ELIST conditions
  840. type elistMatcher struct {
  841. MinClientsActive bool
  842. MinClients int
  843. MaxClientsActive bool
  844. MaxClients int
  845. }
  846. // Matches checks whether the given channel matches our matches.
  847. func (matcher *elistMatcher) Matches(channel *Channel) bool {
  848. if matcher.MinClientsActive {
  849. if len(channel.Members()) < matcher.MinClients {
  850. return false
  851. }
  852. }
  853. if matcher.MaxClientsActive {
  854. if len(channel.Members()) < len(channel.members) {
  855. return false
  856. }
  857. }
  858. return true
  859. }
  860. var (
  861. infoString1 = strings.Split(` ▄▄▄ ▄▄▄· ▄▄ • ▐ ▄
  862. ▪ ▀▄ █·▐█ ▀█ ▐█ ▀ ▪▪ •█▌▐█▪
  863. ▄█▀▄ ▐▀▀▄ ▄█▀▀█ ▄█ ▀█▄ ▄█▀▄▪▐█▐▐▌ ▄█▀▄
  864. ▐█▌.▐▌▐█•█▌▐█ ▪▐▌▐█▄▪▐█▐█▌ ▐▌██▐█▌▐█▌.▐▌
  865. ▀█▄▀▪.▀ ▀ ▀ ▀ ·▀▀▀▀ ▀█▄▀ ▀▀ █▪ ▀█▄▀▪
  866. https://oragono.io/
  867. https://github.com/oragono/oragono
  868. https://crowdin.com/project/oragono
  869. `, "\n")
  870. infoString2 = strings.Split(` Daniel Oakley, DanielOaks, <daniel@danieloaks.net>
  871. Shivaram Lingamneni, slingamn, <slingamn@cs.stanford.edu>
  872. `, "\n")
  873. infoString3 = strings.Split(` Jeremy Latt, jlatt
  874. Edmund Huber, edmund-huber
  875. `, "\n")
  876. )