main.go

package main

import (
	"crypto/sha1"
	"crypto/tls"
	"fmt"
	"log"
	"os"
	"syscall"

	"gobroker/actions"
	"gobroker/db"
)

func setRLimit() {
	var rLimit syscall.Rlimit
	or := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit)
	if err(or) {
		os.Exit(3)
	}
	rLimit.Max = 20000
	rLimit.Cur = 20000
	or = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit)
	if err(or) {
		os.Exit(3)
	}
	or = syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit)
	if err(or) {
		os.Exit(3)
	}
	log.Println("TCP limit set to:", rLimit.Max)
}

func err(e error) bool {
	if e != nil {
		log.Print(e)
		return true
	}
	return false
}

func main() {
	cert, or := tls.LoadX509KeyPair("cert/cert.pem", "cert/key.pem")
	if err(or) {
		createCertificate()
		cert, or = tls.LoadX509KeyPair("cert/cert.pem", "cert/key.pem")
		if err(or) {
			os.Exit(1)
		}
	}
	var fingerprint []byte
	for _, s := range cert.Certificate {
		fingerprint = append(fingerprint, s...)
	}
	sha := ""
	for _, s := range sha1.Sum(fingerprint) {
		if sha != "" {
			sha += ":"
		}
		sha += fmt.Sprintf("%02X", s)
	}
	log.Println(sha)
	cfg := &tls.Config{Certificates: []tls.Certificate{cert}}
	listen, or := tls.Listen("tcp4", db.Conf.Host, cfg)
	if err(or) {
		os.Exit(1)
	}
	log.Printf("Listening on %s", listen.Addr())
	for {
		conn, or := listen.Accept()
		if or != nil {
			if db.Conf.Debug {
				log.Println(or)
			}
			continue
		}
		go actions.Handle(conn)
	}
}