goircd.go

   1 /*
   2 goircd -- minimalistic simple Internet Relay Chat (IRC) server
   3 Copyright (C) 2014-2015 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
  19 package main
  20
  21 import (
  22         "crypto/tls"
  23         "flag"
  24         "io/ioutil"
  25         "log"
  26         "net"
  27         "path"
  28         "path/filepath"
  29         "strings"
  30 )
  31
  32 var (
  33         version   string
  34         hostname  = flag.String("hostname", "localhost", "Hostname")
  35         bind      = flag.String("bind", ":6667", "Address to bind to")
  36         motd      = flag.String("motd", "", "Path to MOTD file")
  37         logdir    = flag.String("logdir", "", "Absolute path to directory for logs")
  38         statedir  = flag.String("statedir", "", "Absolute path to directory for states")
  39         passwords = flag.String("passwords", "", "Optional path to passwords file")
  40
  41         tlsBind = flag.String("tlsbind", "", "TLS address to bind to")
  42         tlsPEM  = flag.String("tlspem", "", "Path to TLS certificat+key PEM file")
  43
  44         verbose = flag.Bool("v", false, "Enable verbose logging.")
  45 )
  46
  47 func listenerLoop(sock net.Listener, events chan<- ClientEvent) {
  48         for {
  49                 conn, err := sock.Accept()
  50                 if err != nil {
  51                         log.Println("Error during accepting connection", err)
  52                         continue
  53                 }
  54                 client := NewClient(hostname, conn)
  55                 go client.Processor(events)
  56         }
  57 }
  58
  59 func Run() {
  60         events := make(chan ClientEvent)
  61         log.SetFlags(log.Ldate | log.Lmicroseconds | log.Lshortfile)
  62
  63         logSink := make(chan LogEvent)
  64         if *logdir == "" {
  65                 // Dummy logger
  66                 go func() {
  67                         for _ = range logSink {
  68                         }
  69                 }()
  70         } else {
  71                 if !path.IsAbs(*logdir) {
  72                         log.Fatalln("Need absolute path for logdir")
  73                         return
  74                 }
  75                 go Logger(*logdir, logSink)
  76                 log.Println(*logdir, "logger initialized")
  77         }
  78
  79         stateSink := make(chan StateEvent)
  80         daemon := NewDaemon(version, hostname, motd, passwords, logSink, stateSink)
  81         daemon.Verbose = *verbose
  82         log.Println("goircd " + daemon.version + " is starting")
  83         if *statedir == "" {
  84                 // Dummy statekeeper
  85                 go func() {
  86                         for _ = range stateSink {
  87                         }
  88                 }()
  89         } else {
  90                 if !path.IsAbs(*statedir) {
  91                         log.Fatalln("Need absolute path for statedir")
  92                 }
  93                 states, err := filepath.Glob(path.Join(*statedir, "#*"))
  94                 if err != nil {
  95                         log.Fatalln("Can not read statedir", err)
  96                 }
  97                 for _, state := range states {
  98                         buf, err := ioutil.ReadFile(state)
  99                         if err != nil {
 100                                 log.Fatalf("Can not read state %s: %v", state, err)
 101                         }
 102                         room, _ := daemon.RoomRegister(path.Base(state))
 103                         contents := strings.Split(string(buf), "\n")
 104                         if len(contents) < 2 {
 105                                 log.Printf("State corrupted for %s: %q", room.name, contents)
 106                         } else {
 107                                 room.topic = contents[0]
 108                                 room.key = contents[1]
 109                                 log.Println("Loaded state for room", room.name)
 110                         }
 111                 }
 112                 go StateKeeper(*statedir, stateSink)
 113                 log.Println(*statedir, "statekeeper initialized")
 114         }
 115
 116         if *bind != "" {
 117                 listener, err := net.Listen("tcp", *bind)
 118                 if err != nil {
 119                         log.Fatalf("Can not listen on %s: %v", *bind, err)
 120                 }
 121                 log.Println("Raw listening on", *bind)
 122                 go listenerLoop(listener, events)
 123         }
 124         if *tlsBind != "" {
 125                 cert, err := tls.LoadX509KeyPair(*tlsPEM, *tlsPEM)
 126                 if err != nil {
 127                         log.Fatalf("Could not load TLS keys from %s: %s", *tlsPEM, err)
 128                 }
 129                 config := tls.Config{Certificates: []tls.Certificate{cert}}
 130                 listenerTLS, err := tls.Listen("tcp", *tlsBind, &config)
 131                 if err != nil {
 132                         log.Fatalf("Can not listen on %s: %v", *tlsBind, err)
 133                 }
 134                 log.Println("TLS listening on", *tlsBind)
 135                 go listenerLoop(listenerTLS, events)
 136         }
 137
 138         daemon.Processor(events)
 139 }
 140
 141 func main() {
 142         flag.Parse()
 143         Run()
 144 }