2017-10-21 00:18:06 +02:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"fmt"
|
2018-06-27 05:32:54 +02:00
|
|
|
"io/ioutil"
|
2018-02-10 00:00:11 +01:00
|
|
|
"os"
|
2018-06-27 05:32:54 +02:00
|
|
|
"reflect"
|
|
|
|
"strconv"
|
2018-02-10 00:00:11 +01:00
|
|
|
"strings"
|
2018-04-13 18:31:43 +02:00
|
|
|
|
2018-06-27 05:32:54 +02:00
|
|
|
log "github.com/cihub/seelog"
|
|
|
|
"github.com/fatih/structs"
|
2018-04-13 18:31:43 +02:00
|
|
|
"github.com/urfave/cli"
|
|
|
|
"github.com/yudai/gotty/pkg/homedir"
|
2018-06-27 05:32:54 +02:00
|
|
|
"github.com/yudai/hcl"
|
|
|
|
yaml "gopkg.in/yaml.v2"
|
2017-10-21 00:18:06 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
const BUFFERSIZE = 1024
|
|
|
|
|
2018-02-09 03:17:25 +01:00
|
|
|
type AppConfig struct {
|
|
|
|
Relay bool `yaml:"relay" flagName:"relay" flagSName:"r" flagDescribe:"Run as relay" default:"false"`
|
|
|
|
Debug bool `yaml:"debug" flagName:"debug" flagSName:"d" flagDescribe:"Debug mode" default:"false"`
|
|
|
|
Wait bool `yaml:"wait" flagName:"wait" flagSName:"w" flagDescribe:"Wait for code to be sent" default:"false"`
|
|
|
|
PathSpec bool `yaml:"ask-save" flagName:"ask-save" flagSName:"q" flagDescribe:"Ask for path to save to" default:"false"`
|
|
|
|
DontEncrypt bool `yaml:"no-encrypt" flagName:"no-encrypt" flagSName:"g" flagDescribe:"Turn off encryption" default:"false"`
|
2018-04-13 23:57:34 +02:00
|
|
|
UseStdout bool `yaml:"stdout" flagName:"stdout" flagSName:"o" flagDescribe:"Use stdout" default:"false"`
|
2018-04-14 00:31:03 +02:00
|
|
|
Yes bool `yaml:"yes" flagName:"yes" flagSName:"y" flagDescribe:"Automatically accept file" default:"false"`
|
2018-04-25 09:20:09 +02:00
|
|
|
Local bool `yaml:"local" flagName:"local" flagSName:"lo" flagDescribe:"Use local relay when sending" default:"false"`
|
2018-06-26 17:55:30 +02:00
|
|
|
NoLocal bool `yaml:"no-local" flagName:"no-local" flagSName:"nlo" flagDescribe:"Don't create local relay" default:"false"`
|
2018-04-22 14:49:27 +02:00
|
|
|
Server string `yaml:"server" flagName:"server" flagSName:"l" flagDescribe:"Croc relay to use" default:"cowyo.com"`
|
2018-04-14 10:29:39 +02:00
|
|
|
File string `yaml:"send" flagName:"send" flagSName:"s" flagDescribe:"File to send default:""`
|
2018-02-09 03:17:25 +01:00
|
|
|
Path string `yaml:"save" flagName:"save" flagSName:"p" flagDescribe:"Path to save to" default:""`
|
|
|
|
Code string `yaml:"code" flagName:"code" flagSName:"c" flagDescribe:"Use your own code phrase" default:""`
|
|
|
|
Rate int `yaml:"rate" flagName:"rate" flagSName:"R" flagDescribe:"Throttle down to speed in kbps" default:"1000000"`
|
|
|
|
NumberOfConnections int `yaml:"threads" flagName:"threads" flagSName:"n" flagDescribe:"Number of threads to use" default:"4"`
|
2017-10-21 00:18:06 +02:00
|
|
|
}
|
|
|
|
|
2018-02-09 03:17:25 +01:00
|
|
|
var email string
|
|
|
|
var author string
|
2017-10-21 00:18:06 +02:00
|
|
|
var version string
|
|
|
|
|
2018-06-27 05:32:54 +02:00
|
|
|
func init() {
|
|
|
|
|
|
|
|
SetLogLevel("debug")
|
|
|
|
}
|
|
|
|
|
2017-10-21 00:18:06 +02:00
|
|
|
func main() {
|
2018-06-27 15:46:51 +02:00
|
|
|
defer log.Flush()
|
2018-02-09 03:17:25 +01:00
|
|
|
app := cli.NewApp()
|
2018-04-15 14:19:54 +02:00
|
|
|
app.Name = "croc"
|
2018-02-09 03:17:25 +01:00
|
|
|
app.Version = version
|
|
|
|
app.Author = author
|
|
|
|
app.Email = email
|
|
|
|
app.Usage = "send file by croc bridge"
|
|
|
|
app.HideHelp = true
|
|
|
|
|
|
|
|
cli.AppHelpTemplate = helpTemplate
|
|
|
|
|
|
|
|
appOptions := &AppConfig{}
|
2018-02-09 16:28:01 +01:00
|
|
|
if err := ApplyDefaultValues(appOptions); err != nil {
|
2018-02-09 03:17:25 +01:00
|
|
|
exit(err, 1)
|
|
|
|
}
|
|
|
|
|
2018-02-09 16:28:01 +01:00
|
|
|
cliFlags, flagMappings, err := GenerateFlags(appOptions)
|
2018-02-09 03:17:25 +01:00
|
|
|
if err != nil {
|
|
|
|
exit(err, 3)
|
|
|
|
}
|
2018-02-09 12:57:57 +01:00
|
|
|
|
2018-02-09 03:17:25 +01:00
|
|
|
app.Flags = append(
|
|
|
|
cliFlags,
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "config",
|
|
|
|
Value: "~/.croc",
|
|
|
|
Usage: "Config file path",
|
|
|
|
EnvVar: "CROC_CONFIG",
|
|
|
|
},
|
|
|
|
)
|
|
|
|
|
|
|
|
app.Action = func(c *cli.Context) {
|
|
|
|
|
|
|
|
configFile := c.String("config")
|
|
|
|
_, err := os.Stat(homedir.Expand(configFile))
|
|
|
|
if configFile != "~/.croc" || !os.IsNotExist(err) {
|
2018-02-09 16:28:01 +01:00
|
|
|
if err := ApplyConfigFileYaml(configFile, appOptions); err != nil {
|
2018-02-09 03:17:25 +01:00
|
|
|
exit(err, 2)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-02-09 16:28:01 +01:00
|
|
|
ApplyFlags(cliFlags, flagMappings, c, appOptions)
|
2018-02-09 03:17:25 +01:00
|
|
|
|
|
|
|
if appOptions.Relay {
|
2018-04-22 14:16:16 +02:00
|
|
|
fmt.Println("running relay on local address " + GetLocalIP())
|
2018-02-09 03:17:25 +01:00
|
|
|
r := NewRelay(appOptions)
|
|
|
|
r.Run()
|
|
|
|
} else {
|
|
|
|
c, err := NewConnection(appOptions)
|
|
|
|
if err != nil {
|
2018-06-27 15:46:51 +02:00
|
|
|
fmt.Printf("Error! Please submit the following error to https://github.com/schollz/croc/issues:\n\n'NewConnection: %s'\n\n", err.Error())
|
2018-02-09 03:17:25 +01:00
|
|
|
return
|
|
|
|
}
|
|
|
|
err = c.Run()
|
|
|
|
if err != nil {
|
2018-06-27 15:46:51 +02:00
|
|
|
fmt.Printf("Error! Please submit the following error to https://github.com/schollz/croc/issues:\n\n'Run: %s'\n\n", err.Error())
|
2018-02-09 03:17:25 +01:00
|
|
|
}
|
2017-10-21 00:18:06 +02:00
|
|
|
}
|
|
|
|
}
|
2018-02-09 03:17:25 +01:00
|
|
|
|
|
|
|
app.Run(os.Args)
|
2017-10-21 00:18:06 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func getInput(prompt string) string {
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
2018-04-13 23:57:34 +02:00
|
|
|
fmt.Fprintf(os.Stderr, "%s", prompt)
|
2017-10-21 00:18:06 +02:00
|
|
|
text, _ := reader.ReadString('\n')
|
|
|
|
return strings.TrimSpace(text)
|
|
|
|
}
|
2018-02-09 03:17:25 +01:00
|
|
|
|
|
|
|
func exit(err error, code int) {
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
}
|
|
|
|
os.Exit(code)
|
|
|
|
}
|
2018-06-27 05:32:54 +02:00
|
|
|
|
|
|
|
func ApplyDefaultValues(struct_ interface{}) (err error) {
|
|
|
|
o := structs.New(struct_)
|
|
|
|
|
|
|
|
for _, field := range o.Fields() {
|
|
|
|
defaultValue := field.Tag("default")
|
|
|
|
if defaultValue == "" {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var val interface{}
|
|
|
|
switch field.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
val = defaultValue
|
|
|
|
case reflect.Bool:
|
|
|
|
if defaultValue == "true" {
|
|
|
|
val = true
|
|
|
|
} else if defaultValue == "false" {
|
|
|
|
val = false
|
|
|
|
} else {
|
|
|
|
return fmt.Errorf("invalid bool expression: %v, use true/false", defaultValue)
|
|
|
|
}
|
|
|
|
case reflect.Int:
|
|
|
|
val, err = strconv.Atoi(defaultValue)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
val = field.Value()
|
|
|
|
}
|
|
|
|
field.Set(val)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func GenerateFlags(options ...interface{}) (flags []cli.Flag, mappings map[string]string, err error) {
|
|
|
|
mappings = make(map[string]string)
|
|
|
|
|
|
|
|
for _, struct_ := range options {
|
|
|
|
o := structs.New(struct_)
|
|
|
|
for _, field := range o.Fields() {
|
|
|
|
flagName := field.Tag("flagName")
|
|
|
|
if flagName == "" {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
envName := "CROC_" + strings.ToUpper(strings.Join(strings.Split(flagName, "-"), "_"))
|
|
|
|
mappings[flagName] = field.Name()
|
|
|
|
|
|
|
|
flagShortName := field.Tag("flagSName")
|
|
|
|
if flagShortName != "" {
|
|
|
|
flagName += ", " + flagShortName
|
|
|
|
}
|
|
|
|
|
|
|
|
flagDescription := field.Tag("flagDescribe")
|
|
|
|
|
|
|
|
switch field.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
flags = append(flags, cli.StringFlag{
|
|
|
|
Name: flagName,
|
|
|
|
Value: field.Value().(string),
|
|
|
|
Usage: flagDescription,
|
|
|
|
EnvVar: envName,
|
|
|
|
})
|
|
|
|
case reflect.Bool:
|
|
|
|
flags = append(flags, cli.BoolFlag{
|
|
|
|
Name: flagName,
|
|
|
|
Usage: flagDescription,
|
|
|
|
EnvVar: envName,
|
|
|
|
})
|
|
|
|
case reflect.Int:
|
|
|
|
flags = append(flags, cli.IntFlag{
|
|
|
|
Name: flagName,
|
|
|
|
Value: field.Value().(int),
|
|
|
|
Usage: flagDescription,
|
|
|
|
EnvVar: envName,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func ApplyFlags(
|
|
|
|
flags []cli.Flag,
|
|
|
|
mappingHint map[string]string,
|
|
|
|
c *cli.Context,
|
|
|
|
options ...interface{},
|
|
|
|
) {
|
|
|
|
objects := make([]*structs.Struct, len(options))
|
|
|
|
for i, struct_ := range options {
|
|
|
|
objects[i] = structs.New(struct_)
|
|
|
|
}
|
|
|
|
|
|
|
|
for flagName, fieldName := range mappingHint {
|
|
|
|
if !c.IsSet(flagName) {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var field *structs.Field
|
|
|
|
var ok bool
|
|
|
|
for _, o := range objects {
|
|
|
|
field, ok = o.FieldOk(fieldName)
|
|
|
|
if ok {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if field == nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var val interface{}
|
|
|
|
switch field.Kind() {
|
|
|
|
case reflect.String:
|
|
|
|
val = c.String(flagName)
|
|
|
|
case reflect.Bool:
|
|
|
|
val = c.Bool(flagName)
|
|
|
|
case reflect.Int:
|
|
|
|
val = c.Int(flagName)
|
|
|
|
}
|
|
|
|
field.Set(val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func ApplyConfigFile(filePath string, options ...interface{}) error {
|
|
|
|
filePath = homedir.Expand(filePath)
|
|
|
|
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
fileString := []byte{}
|
|
|
|
log.Debugf("Loading config file at: %s", filePath)
|
|
|
|
fileString, err := ioutil.ReadFile(filePath)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, object := range options {
|
|
|
|
if err := hcl.Decode(object, string(fileString)); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func ApplyConfigFileYaml(filePath string, options ...interface{}) error {
|
|
|
|
filePath = homedir.Expand(filePath)
|
|
|
|
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
fileString := []byte{}
|
|
|
|
log.Debugf("Loading config file at: %s", filePath)
|
|
|
|
fileString, err := ioutil.ReadFile(filePath)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, object := range options {
|
|
|
|
if err := yaml.Unmarshal(fileString, object); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func SaveConfigFileYaml(filePath string, options ...interface{}) error {
|
|
|
|
filePath = homedir.Expand(filePath)
|
|
|
|
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
fd, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE, 0644)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer fd.Close()
|
|
|
|
|
|
|
|
for _, object := range options {
|
|
|
|
if byteString, err := yaml.Marshal(object); err != nil {
|
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
fd.Write(byteString)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
var helpTemplate = `
|
|
|
|
,_
|
|
|
|
>' )
|
|
|
|
( ( \
|
|
|
|
|| \
|
|
|
|
/^^^^\ ||
|
|
|
|
/^^\________/0 \ ||
|
|
|
|
( ` + "`" + `~+++,,_||__,,++~^^^^^^^
|
|
|
|
...V^V^V^V^V^V^\...............................
|
|
|
|
|
|
|
|
|
|
|
|
NAME:
|
|
|
|
{{.Name}} - {{.Usage}}
|
|
|
|
|
|
|
|
USAGE:
|
|
|
|
{{.Name}} [options]
|
|
|
|
|
|
|
|
VERSION:
|
|
|
|
{{.Version}}{{if or .Author .Email}}
|
|
|
|
|
|
|
|
AUTHOR:{{if .Author}}
|
|
|
|
{{.Author}}{{if .Email}} - <{{.Email}}>{{end}}{{else}}
|
|
|
|
{{.Email}}{{end}}{{end}}
|
|
|
|
|
|
|
|
OPTIONS:
|
|
|
|
{{range .Flags}}{{.}}
|
|
|
|
{{end}}
|
|
|
|
`
|
|
|
|
|
|
|
|
// SetLogLevel determines the log level
|
|
|
|
func SetLogLevel(level string) (err error) {
|
|
|
|
|
|
|
|
// https://en.wikipedia.org/wiki/ANSI_escape_code#3/4_bit
|
|
|
|
// https://github.com/cihub/seelog/wiki/Log-levels
|
|
|
|
appConfig := `
|
|
|
|
<seelog minlevel="` + level + `">
|
|
|
|
<outputs formatid="stdout">
|
|
|
|
<filter levels="debug,trace">
|
|
|
|
<console formatid="debug"/>
|
|
|
|
</filter>
|
|
|
|
<filter levels="info">
|
|
|
|
<console formatid="info"/>
|
|
|
|
</filter>
|
|
|
|
<filter levels="critical,error">
|
|
|
|
<console formatid="error"/>
|
|
|
|
</filter>
|
|
|
|
<filter levels="warn">
|
|
|
|
<console formatid="warn"/>
|
|
|
|
</filter>
|
|
|
|
</outputs>
|
|
|
|
<formats>
|
|
|
|
<format id="stdout" format="%Date %Time [%LEVEL] %File %FuncShort:%Line %Msg %n" />
|
|
|
|
<format id="debug" format="%Date %Time %EscM(37)[%LEVEL]%EscM(0) %File %FuncShort:%Line %Msg %n" />
|
|
|
|
<format id="info" format="%EscM(36)[%LEVEL]%EscM(0) %Msg %n" />
|
|
|
|
<format id="warn" format="%Date %Time %EscM(33)[%LEVEL]%EscM(0) %File %FuncShort:%Line %Msg %n" />
|
|
|
|
<format id="error" format="%Date %Time %EscM(31)[%LEVEL]%EscM(0) %File %FuncShort:%Line %Msg %n" />
|
|
|
|
</formats>
|
|
|
|
</seelog>
|
|
|
|
`
|
|
|
|
logger, err := log.LoggerFromConfigAsBytes([]byte(appConfig))
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
log.ReplaceLogger(logger)
|
|
|
|
return
|
|
|
|
}
|