main.go

// Copyright 2022 Mohammad Hadi Hosseinpour
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main

import (
	"flag"
	"os"
	"os/signal"
	"time"
)
import "github.com/sirupsen/logrus"

var (
	listenAddr        string
	targetAddr        string
	proxyAddr         string
	dialTimeout       int
	keepAliveInterval int
	showHelp          bool
	debugLog          bool
)

// Create a new instance of the logger. You can have any number of instances.
var log = logrus.New()

func init() {
	flag.BoolVar(&showHelp, "help", false, "show usage")
	flag.BoolVar(&debugLog, "debug", false, "more verbose logging")
	flag.StringVar(&listenAddr, "listen", "", "listening address (<host>:<port>)")
	flag.StringVar(&targetAddr, "target", "", "remote target (<host>:<port>)")
	flag.StringVar(&proxyAddr, "proxy", "", "proxy address (<proto>://[user[:password]@]<host>:<port>/)")
	flag.IntVar(&dialTimeout, "timeout", 10, "dial timeout")
	flag.IntVar(&keepAliveInterval, "keepalive", 30, "keep-alive interval")
}

func main() {
	flag.Parse()

	log.SetLevel(logrus.InfoLevel)
	if debugLog {
		log.SetLevel(logrus.DebugLevel)
	}

	log.Debugf("logging level set to %s", log.GetLevel())

	if showHelp || targetAddr == "" || listenAddr == "" {
		flag.Usage()
		return
	}
	signals := make(chan os.Signal, 1)

	signal.Notify(signals, os.Interrupt, os.Kill)
	client := newClient(listenAddr, targetAddr, proxyAddr, time.Duration(dialTimeout), time.Duration(keepAliveInterval), signals)
	err := client.Run()
	if err != nil {
		log.Fatalf("exiting on error: %s", err)
	}
}