rdp_channel/main.go

package main

import (
	"flag"
	"log"
	"rdp_channel/app"
)

func main() {
	mode := flag.String("mode", "server", "server or client")
	host := flag.String("host", "127.0.0.1", "server or client")
	port := flag.Int("port", 8080, "server or client")
	flag.Parse()

	var a app.App
	switch *mode {
	case "server":
		a = app.NewServer(*host, *port)
	case "client":
		a = app.NewClient(*host, *port)
	default:
		log.Fatal("[APP] invalid mode: " + *mode)
	}

	err := a.Start()
	if err != nil {
		log.Fatal(err)
	}
}