main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"runtime"

	"dfss"
	dapi "dfss/dfssd/api"
	"dfss/dfssp/authority"
	"dfss/dfssp/server"
	"dfss/net"
)

var (
	verbose                                                  bool
	path, country, org, unit, cn, port, address, dbURI, demo string
	keySize, rootValidity, certValidity                      int
)

func init() {

	flag.BoolVar(&verbose, "v", false, "Print verbose messages")

	flag.StringVar(&demo, "d", "", "Demonstrator address and port (empty string disables debug)")
	flag.StringVar(&port, "p", "9000", "Default port listening")
	flag.StringVar(&address, "a", "0.0.0.0", "Default address to bind for listening")

	flag.StringVar(&path, "path", ".", "Path for the platform's private key and root certificate")
	flag.StringVar(&country, "country", "France", "Country for the root certificate")
	flag.StringVar(&org, "org", "DFSS", "Organization for the root certificate")
	flag.StringVar(&unit, "unit", "INSA Rennes", "Organizational unit for the root certificate")
	flag.StringVar(&cn, "cn", "dfssp", "Common name for the root certificate")

	flag.IntVar(&keySize, "keySize", 512, "Encoding size for the private key")
	flag.IntVar(&rootValidity, "rootValidity", 365, "Root certificate's validity duration (days)")
	flag.IntVar(&certValidity, "certValidity", 365, "Validity duration for the certificates generated by this platform (days)")

	flag.StringVar(&dbURI, "db", "mongodb://localhost/dfss", "Name of the environment variable containing the server url in standard MongoDB format")

	flag.Usage = func() {
		fmt.Println("DFSS platform v" + dfss.Version)
		fmt.Println("Users and contracts manager")

		fmt.Println("\nUsage:")
		fmt.Println("  dfssp [flags] command")

		fmt.Println("\nThe commands are:")
		fmt.Println("  init     [cn, country, keySize, org, path, unit, rootValidity]")
		fmt.Println("           create and save the platform's private key and root certificate")
		fmt.Println("  ttp      [cn, country, keySize, org, path, unit, certValidity]")
		fmt.Println("           create and save the TTP's private key and certificate")
		fmt.Println("  start    [path, db, a, p, certValidity]")
		fmt.Println("           start the platform after loading its private key and root certificate")
		fmt.Println("  help     print this help")
		fmt.Println("  version  print dfss client version")

		fmt.Println("\nFlags:")
		flag.PrintDefaults()

		fmt.Println()
	}
}

func main() {
	flag.Parse()
	command := flag.Arg(0)
	dapi.Configure(demo != "", demo, "platform")

	switch command {
	case "version":
		fmt.Println("v"+dfss.Version, runtime.GOOS, runtime.GOARCH)
	case "init":
		err := authority.Initialize(keySize, rootValidity, country, org, unit, cn, path, nil, nil)
		if err != nil {
			fmt.Fprintln(os.Stderr, "An error occured during the initialization operation:", err)
			os.Exit(1)
		}
		dapi.DLog("Private key and root certificate generated")
	case "ttp":
		pid, err := authority.Start(path)
		if err != nil {
			fmt.Fprintln(os.Stderr, "Bad root CA or key; please use the `init` command before the `ttp` one.\n", err)
		}
		ttpPath := filepath.Join(path, "ttp")
		err = authority.Initialize(keySize, certValidity, country, org, unit, cn, ttpPath, pid.RootCA, pid.Pkey)
		if err != nil {
			fmt.Fprintln(os.Stderr, "An error occured during TTP credentials generation:", err)
		}
		dapi.DLog("Private key and certificate generated for TTP")
	case "start":
		srv := server.GetServer(path, dbURI, certValidity, verbose)
		fmt.Println("Listening on " + address + ":" + port)
		dapi.DLog("Platform server started on " + address + ":" + port)
		err := net.Listen(address+":"+port, srv)
		if err != nil {
			fmt.Fprintln(os.Stderr, err)
		}
	default:
		flag.Usage()
	}

	dapi.DClose()
}