mirror of https://github.com/schollz/croc.git
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const BUFFERSIZE = 1024
|
|
const numberConnections = 4
|
|
|
|
// Build flags
|
|
var server, file string
|
|
|
|
// Global varaibles
|
|
var serverAddress, fileName, codePhraseFlag, connectionTypeFlag string
|
|
var runAsRelay, debugFlag bool
|
|
|
|
func main() {
|
|
flag.BoolVar(&runAsRelay, "relay", false, "run as relay")
|
|
flag.BoolVar(&debugFlag, "debug", false, "debug mode")
|
|
flag.StringVar(&serverAddress, "server", "", "(run as client) server address to connect to")
|
|
flag.StringVar(&fileName, "file", "", "(run as server) file to serve")
|
|
flag.StringVar(&codePhraseFlag, "code", "", "(run as server) file to serve")
|
|
flag.Parse()
|
|
// Check build flags too, which take precedent
|
|
if server != "" {
|
|
serverAddress = server
|
|
}
|
|
if file != "" {
|
|
fileName = file
|
|
}
|
|
|
|
if len(fileName) > 0 {
|
|
_, err := os.Open(fileName)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
connectionTypeFlag = "s" // sender
|
|
} else {
|
|
connectionTypeFlag = "r" //receiver
|
|
}
|
|
|
|
if !runAsRelay {
|
|
if len(codePhraseFlag) == 0 {
|
|
codePhraseFlag = getInput("What is your code phrase? ")
|
|
if len(codePhraseFlag) < 5 {
|
|
codePhraseFlag = GetRandomName()
|
|
fmt.Println("Your code phrase is now " + codePhraseFlag)
|
|
}
|
|
}
|
|
}
|
|
|
|
log.SetFormatter(&log.TextFormatter{})
|
|
if debugFlag {
|
|
log.SetLevel(log.DebugLevel)
|
|
} else {
|
|
log.SetLevel(log.WarnLevel)
|
|
}
|
|
|
|
if runAsRelay {
|
|
runServer()
|
|
} else if len(serverAddress) != 0 {
|
|
runClient(connectionTypeFlag, codePhraseFlag)
|
|
} else {
|
|
fmt.Println("You must specify either -file (for running as a server) or -server (for running as a client)")
|
|
}
|
|
}
|
|
|
|
func getInput(prompt string) string {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
fmt.Print(prompt)
|
|
text, _ := reader.ReadString('\n')
|
|
return strings.TrimSpace(text)
|
|
}
|