From 0cf415d9a4252e818ea4383bbd155715c9160185 Mon Sep 17 00:00:00 2001 From: Janne Koschinski <janne@kuschku.de> Date: Mon, 18 Mar 2019 23:54:01 +0100 Subject: [PATCH] Get hostname and port from CLI --- main.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 8d11c7b..eb2835c 100644 --- a/main.go +++ b/main.go @@ -4,6 +4,7 @@ import ( "bufio" "crypto/tls" "encoding/binary" + "flag" "fmt" "net" ) @@ -117,7 +118,11 @@ func parseProtocolInfo(data []byte) protocolInfo { } func main() { - conn := makeConnection("kuschku.de", 4242) + hostname := flag.String("hostname", "", "address of server to connect to") + port := flag.Int("port", 4242, "port of server to connect to") + flag.Parse() + + conn := makeConnection(*hostname, *port) conn.write(PROTOCOL_MAGIC | uint32(PROTOCOL_FEATURE_TLS)) supportedProtocols := []uint32{ PROTOCOL_DATASTREAM, @@ -128,7 +133,6 @@ func main() { conn.write(1 << 31) conn.flush() protocolInfo := parseProtocolInfo(conn.read(4)) - println("Read Protocol Info") if protocolInfo.flagTLS { conn.withTLS(false) } -- GitLab