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         "bytes"
  22         "crypto/tls"
  23         "flag"
  24         "log"
  25         "net"
  26         "os"
  27         "path"
  28         "path/filepath"
  29         "strings"
  30 )
  31
  32 var (
  33         hostname = flag.String("hostname", "localhost", "Hostname")
  34         bind     = flag.String("bind", ":6667", "Address to bind to")
  35         motd     = flag.String("motd", "", "Path to MOTD file")
  36         logdir   = flag.String("logdir", "", "Absolute path to directory for logs")
  37         statedir = flag.String("statedir", "", "Absolute path to directory for states")
  38
  39         ssl     = flag.Bool("ssl", false, "Use SSL only.")
  40         sslKey  = flag.String("ssl_key", "", "SSL keyfile.")
  41         sslCert = flag.String("ssl_cert", "", "SSL certificate.")
  42 )
  43
  44 func Run() {
  45         var client *Client
  46         events := make(chan ClientEvent)
  47         log.SetFlags(log.Ldate | log.Lmicroseconds | log.Lshortfile)
  48
  49         log_sink := make(chan LogEvent)
  50         if *logdir == "" {
  51                 // Dummy logger
  52                 go func() {
  53                         for _ = range log_sink {
  54                         }
  55                 }()
  56         } else {
  57                 if !path.IsAbs(*logdir) {
  58                         log.Fatalln("Need absolute path for logdir")
  59                         return
  60                 }
  61                 go Logger(*logdir, log_sink)
  62                 log.Println(*logdir, "logger initialized")
  63         }
  64
  65         state_sink := make(chan StateEvent)
  66         daemon := NewDaemon(*hostname, *motd, log_sink, state_sink)
  67         if *statedir == "" {
  68                 // Dummy statekeeper
  69                 go func() {
  70                         for _ = range state_sink {
  71                         }
  72                 }()
  73         } else {
  74                 if !path.IsAbs(*statedir) {
  75                         log.Fatalln("Need absolute path for statedir")
  76                 }
  77                 states, err := filepath.Glob(*statedir + "/#*")
  78                 if err != nil {
  79                         log.Fatalln("Can not read statedir", err)
  80                 }
  81                 for _, state := range states {
  82                         fd, err := os.Open(state)
  83                         if err != nil {
  84                                 log.Fatalln("Can not open state", state, err)
  85                         }
  86                         buf := make([]byte, 1024)
  87                         _, err = fd.Read(buf)
  88                         fd.Close()
  89                         if err != nil {
  90                                 log.Fatalln("Can not read state", state, err)
  91                         }
  92                         room, _ := daemon.RoomRegister(path.Base(state))
  93                         buf = bytes.TrimRight(buf, "\x00")
  94                         contents := strings.Split(string(buf), "\n")
  95                         room.topic = contents[0]
  96                         room.key = contents[1]
  97                         log.Println("Loaded state for room", room.name)
  98                 }
  99                 go StateKeeper(*statedir, state_sink)
 100                 log.Println(*statedir, "statekeeper initialized")
 101         }
 102
 103         var listener net.Listener
 104         if *ssl {
 105                 cert, err := tls.LoadX509KeyPair(*sslCert, *sslKey)
 106                 if err != nil {
 107                         log.Fatalf("Could not load SSL keys from %s and %s: %s", *sslCert, *sslKey, err)
 108                 }
 109                 config := tls.Config{Certificates: []tls.Certificate{cert}}
 110                 listener, err = tls.Listen("tcp", *bind, &config)
 111                 if err != nil {
 112                         log.Fatalf("Can not listen on %s: %v", *bind, err)
 113                 }
 114         } else {
 115                 var err error
 116                 listener, err = net.Listen("tcp", *bind)
 117                 if err != nil {
 118                         log.Fatalf("Can not listen on %s: %v", *bind, err)
 119                 }
 120         }
 121         log.Println("Listening on", *bind)
 122
 123         go daemon.Processor(events)
 124         for {
 125                 conn, err := listener.Accept()
 126                 if err != nil {
 127                         log.Println("Error during accepting connection", err)
 128                         continue
 129                 }
 130                 client = NewClient(*hostname, conn)
 131                 go client.Processor(events)
 132         }
 133 }
 134
 135 func main() {
 136         flag.Parse()
 137         Run()
 138 }