You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.4 KiB
Go
75 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
// "net"
|
|
"os"
|
|
)
|
|
|
|
const keyLength = 2048
|
|
|
|
var (
|
|
info_log *log.Logger
|
|
error_log *log.Logger
|
|
)
|
|
|
|
var options struct {
|
|
port int
|
|
host string
|
|
key string
|
|
publicKey string
|
|
nick string
|
|
debug bool
|
|
}
|
|
|
|
func exit(status int, template string, args ...interface{}) {
|
|
if status == 0 {
|
|
info_log.Printf(template, args...)
|
|
} else {
|
|
error_log.Printf(template, args...)
|
|
}
|
|
os.Exit(status)
|
|
}
|
|
|
|
func usage(template string, args ...interface{}) {
|
|
fmt.Fprintf(os.Stdout, template, args...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
info_log, error_log = log.New(os.Stdout, "", 0), log.New(os.Stderr, "", 0)
|
|
|
|
if flag.NArg() < 1 {
|
|
usage("client or server?")
|
|
}
|
|
|
|
switch flag.Arg(0) {
|
|
case "client":
|
|
connect()
|
|
case "server":
|
|
serve()
|
|
case "generate":
|
|
generate()
|
|
case "encrypt":
|
|
encrypt()
|
|
case "decrypt":
|
|
decrypt()
|
|
case "get-public":
|
|
getPublic()
|
|
default:
|
|
usage("i dunno what you mean with %v", flag.Arg(0))
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
flag.IntVar(&options.port, "port", 9000, "port number")
|
|
flag.StringVar(&options.host, "host", "localhost", "host to connect to")
|
|
flag.StringVar(&options.key, "key", "whisper_key", "rsa key to use")
|
|
flag.StringVar(&options.publicKey, "public-key", "", "public rsa key to use")
|
|
flag.StringVar(&options.nick, "nick", "", "nick to use in chat")
|
|
flag.BoolVar(&options.debug, "debug", false, "include debug messages")
|
|
}
|