2019-11-12 00:28:07 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
|
2019-11-12 07:03:12 +01:00
|
|
|
"github.com/schollz/croc/v7/src/croc"
|
2019-11-12 00:28:07 +01:00
|
|
|
"github.com/schollz/croc/v7/src/relay"
|
|
|
|
log "github.com/schollz/logger"
|
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2019-11-12 07:03:12 +01:00
|
|
|
var startRelay, startSend, startReceive bool
|
2019-11-12 00:28:07 +01:00
|
|
|
flag.BoolVar(&startRelay, "relay", false, "start relay")
|
2019-11-12 07:03:12 +01:00
|
|
|
flag.BoolVar(&startSend, "send", false, "send")
|
|
|
|
flag.BoolVar(&startReceive, "receive", false, "receive")
|
2019-11-12 00:28:07 +01:00
|
|
|
flag.Parse()
|
|
|
|
log.SetLevel("debug")
|
|
|
|
if startRelay {
|
|
|
|
relay.Run()
|
2019-11-12 07:03:12 +01:00
|
|
|
} else if startSend {
|
|
|
|
c, err := croc.New(croc.Options{
|
|
|
|
IsSender: true,
|
2019-11-13 01:44:09 +01:00
|
|
|
SharedSecret: "test1",
|
2019-11-12 07:03:12 +01:00
|
|
|
RelayAddress: "ws://localhost:8005/ws",
|
|
|
|
Debug: true,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
err = c.Send(croc.TransferOptions{})
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
} else if startReceive {
|
|
|
|
c, err := croc.New(croc.Options{
|
|
|
|
IsSender: false,
|
2019-11-13 01:44:09 +01:00
|
|
|
SharedSecret: "test1",
|
2019-11-12 07:03:12 +01:00
|
|
|
RelayAddress: "ws://localhost:8005/ws",
|
|
|
|
Debug: true,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
err = c.Receive()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2019-11-12 00:28:07 +01:00
|
|
|
}
|
|
|
|
}
|