goircd.go

   1 /*
   2 goircd -- minimalistic simple Internet Relay Chat (IRC) server
   3 Copyright (C) 2014 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18 package main
  19
  20 import (
  21         "crypto/tls"
  22         "flag"
  23         "io/ioutil"
  24         "log"
  25         "net"
  26         "os"
  27         "os/signal"
  28         "path"
  29         "path/filepath"
  30         "strings"
  31         "syscall"
  32 )
  33
  34 var (
  35         hostname  = flag.String("hostname", "localhost", "Hostname")
  36         bind      = flag.String("bind", ":6667", "Address to bind to")
  37         motd      = flag.String("motd", "", "Path to MOTD file")
  38         logdir    = flag.String("logdir", "", "Absolute path to directory for logs")
  39         statedir  = flag.String("statedir", "", "Absolute path to directory for states")
  40         passwords = flag.String("passwords", "", "Optional path to passwords file")
  41
  42         tlsKey  = flag.String("tls_key", "", "TLS keyfile")
  43         tlsCert = flag.String("tls_cert", "", "TLS certificate")
  44
  45         verbose = flag.Bool("v", false, "Enable verbose logging.")
  46 )
  47
  48 func Run() {
  49         var client *Client
  50         events := make(chan ClientEvent)
  51         log.SetFlags(log.Ldate | log.Lmicroseconds | log.Lshortfile)
  52
  53         logSink := make(chan LogEvent)
  54         if *logdir == "" {
  55                 // Dummy logger
  56                 go func() {
  57                         for _ = range logSink {
  58                         }
  59                 }()
  60         } else {
  61                 if !path.IsAbs(*logdir) {
  62                         log.Fatalln("Need absolute path for logdir")
  63                         return
  64                 }
  65                 go Logger(*logdir, logSink)
  66                 log.Println(*logdir, "logger initialized")
  67         }
  68
  69         stateSink := make(chan StateEvent)
  70         daemon := NewDaemon(*hostname, *motd, logSink, stateSink)
  71         daemon.Verbose = *verbose
  72         if *statedir == "" {
  73                 // Dummy statekeeper
  74                 go func() {
  75                         for _ = range stateSink {
  76                         }
  77                 }()
  78         } else {
  79                 if !path.IsAbs(*statedir) {
  80                         log.Fatalln("Need absolute path for statedir")
  81                 }
  82                 states, err := filepath.Glob(path.Join(*statedir, "#*"))
  83                 if err != nil {
  84                         log.Fatalln("Can not read statedir", err)
  85                 }
  86                 for _, state := range states {
  87                         buf, err := ioutil.ReadFile(state)
  88                         if err != nil {
  89                                 log.Fatalf("Can not read state %s: %v", state, err)
  90                         }
  91                         room, _ := daemon.RoomRegister(path.Base(state))
  92                         contents := strings.Split(string(buf), "\n")
  93                         if len(contents) < 2 {
  94                                 log.Printf("State corrupted for %s: %q", room.name, contents)
  95                         } else {
  96                                 room.topic = contents[0]
  97                                 room.key = contents[1]
  98                                 log.Println("Loaded state for room", room.name)
  99                         }
 100                 }
 101                 go StateKeeper(*statedir, stateSink)
 102                 log.Println(*statedir, "statekeeper initialized")
 103         }
 104
 105         var listener net.Listener
 106         if *tlsKey != "" {
 107                 cert, err := tls.LoadX509KeyPair(*tlsCert, *tlsKey)
 108                 if err != nil {
 109                         log.Fatalf("Could not load TLS keys from %s and %s: %s", *tlsCert, *tlsKey, err)
 110                 }
 111                 config := tls.Config{Certificates: []tls.Certificate{cert}}
 112                 listener, err = tls.Listen("tcp", *bind, &config)
 113                 if err != nil {
 114                         log.Fatalf("Can not listen on %s: %v", *bind, err)
 115                 }
 116         } else {
 117                 var err error
 118                 listener, err = net.Listen("tcp", *bind)
 119                 if err != nil {
 120                         log.Fatalf("Can not listen on %s: %v", *bind, err)
 121                 }
 122         }
 123         log.Println("Listening on", *bind)
 124
 125         if *passwords != "" {
 126                 daemon.PasswordsRefresh()
 127                 hups := make(chan os.Signal)
 128                 signal.Notify(hups, syscall.SIGHUP)
 129                 go func() {
 130                         for {
 131                                 <-hups
 132                                 daemon.PasswordsRefresh()
 133                         }
 134                 }()
 135         }
 136
 137         go daemon.Processor(events)
 138         for {
 139                 conn, err := listener.Accept()
 140                 if err != nil {
 141                         log.Println("Error during accepting connection", err)
 142                         continue
 143                 }
 144                 client = NewClient(*hostname, conn)
 145                 go client.Processor(events)
 146         }
 147 }
 148
 149 func main() {
 150         flag.Parse()
 151         Run()
 152 }