croc/main.go

26 lines
327 B
Go
Raw Normal View History

2018-06-28 15:38:07 +02:00
package main
2018-06-29 03:47:34 +02:00
import (
"flag"
croc "github.com/schollz/croc/src"
)
2018-06-28 15:38:07 +02:00
func main() {
2018-06-29 03:47:34 +02:00
var err error
role := flag.Int("role", 0, "role number")
flag.Parse()
2018-06-28 17:05:31 +02:00
c := croc.Init()
2018-06-29 15:23:00 +02:00
if *role == -1 {
2018-06-29 03:47:34 +02:00
err = c.Relay()
2018-06-29 15:23:00 +02:00
} else if *role == 0 {
2018-06-29 03:47:34 +02:00
err = c.Send("foo")
2018-06-29 15:23:00 +02:00
} else {
err = c.Receive()
2018-06-29 03:47:34 +02:00
}
2018-06-28 15:38:07 +02:00
if err != nil {
panic(err)
}
}