aboutsummaryrefslogtreecommitdiff
path: root/cmd
diff options
context:
space:
mode:
authorJames Barnett <noreply@jamesbarnett.xyz>2018-12-27 21:52:25 +0000
committerJames Barnett <noreply@jamesbarnett.xyz>2018-12-27 21:52:25 +0000
commit7672da39aafb3e9a2fcbb0bce236dff8b7613355 (patch)
tree19e8619a6ff94136cf38e871c40ff8acdd612e8b /cmd
parent370f8120e4ddb8cef35c62f77b060599c92a0c53 (diff)
downloadslowpoke-7672da39aafb3e9a2fcbb0bce236dff8b7613355.tar.xz
slowpoke-7672da39aafb3e9a2fcbb0bce236dff8b7613355.zip
Remove cmd package. Cleanup logging.
Diffstat (limited to 'cmd')
-rw-r--r--cmd/slowpoke/main.go93
1 files changed, 0 insertions, 93 deletions
diff --git a/cmd/slowpoke/main.go b/cmd/slowpoke/main.go
deleted file mode 100644
index 93f62b0..0000000
--- a/cmd/slowpoke/main.go
+++ /dev/null
@@ -1,93 +0,0 @@
-package main
-
-import (
- "fmt"
- "net"
- "os"
- "time"
-
- "github.com/jessevdk/go-flags"
- "github.com/op/go-logging"
-
- "github.com/jamesbarnett91/slowpoke"
-)
-
-var log = logging.MustGetLogger("main")
-
-var opts struct {
- TargetAddress string `short:"t" long:"target" description:"The target address in host:port form" required:"true"`
- Port int `short:"p" long:"port" description:"The port Slowpoke should listen for connections on" required:"true"`
- Verbose []bool `short:"v" long:"verbose" description:"Log verbosity level. -v or -vv"`
- Latency time.Duration `short:"l" long:"latency" default:"0ms" description:"The amount of latency to apply to data packets, specified as a number and unit. E.g. 15ms or 2s. Supported units are 'us', 'ms', 's', 'm' and 'h'"`
- BufferSize int `short:"b" long:"buffer" default:"1500" description:"The size of the transfer buffer in bytes. Latency is applied between each buffer flush. Therefore total latency applied is equal to '(totalDataTransferred/bufferSize) * latency'"`
-}
-
-func init() {
- _, err := flags.Parse(&opts)
- if err != nil {
- log.Error(err)
- os.Exit(1)
- }
-
- configureLogger()
-}
-
-func configureLogger() {
- logBackend := logging.NewLogBackend(os.Stderr, "", 0)
- logFormat := logging.MustStringFormatter(`%{color}%{time:15:04:05.000} [%{level:.3s}]%{color:reset} - %{message}`)
- logger := logging.AddModuleLevel(logging.NewBackendFormatter(logBackend, logFormat))
-
- if len(opts.Verbose) == 0 {
- logger.SetLevel(logging.WARNING, "")
- } else if len(opts.Verbose) == 1 {
- logger.SetLevel(logging.INFO, "")
- } else {
- logger.SetLevel(logging.DEBUG, "")
- }
-
- logging.SetBackend(logger)
-}
-
-func main() {
- log.Infof("Proxying between :%d and %s with %s of latency", opts.Port, opts.TargetAddress, opts.Latency)
- log.Debugf("Transfer buffer size set to %d bytes", opts.BufferSize)
-
- listener := getListener(opts.Port)
- targetAddr := resolveTarget(opts.TargetAddress)
- waitForClients(listener, targetAddr)
-}
-
-func getListener(port int) net.Listener {
- listener, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
- if err != nil {
- log.Errorf("Failed to start listening on port %d:\n%v", port, err)
- os.Exit(1)
- }
- log.Debugf("Waiting for connections on port %d", port)
-
- return listener
-}
-
-func resolveTarget(targetAddress string) *net.TCPAddr {
- tcpAddr, err := net.ResolveTCPAddr("tcp", opts.TargetAddress)
- if err != nil {
- log.Errorf("Failed to resolve target address %s:\n%v", opts.TargetAddress, err)
- os.Exit(1)
- }
- return tcpAddr
-}
-
-func waitForClients(listener net.Listener, targetAddr *net.TCPAddr) {
- for {
- client, err := listener.Accept()
- if err != nil {
- log.Errorf("Failed to accept connection:\n%v", err)
- break
- }
- log.Infof("Accepted connection from client %v\n", client.RemoteAddr())
-
- s := slowpoke.New(client, targetAddr, opts.Latency, opts.BufferSize, log)
-
- go s.StartTransfer()
- }
-}