Viktor Svensson пре 7 година
родитељ
комит
751e71e394
1 измењених фајлова са 25 додато и 4 уклоњено
  1. 25 4
      oftp.go

+ 25 - 4
oftp.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"crypto/tls"
 	"encoding/binary"
 	"encoding/json"
 	"errors"
@@ -57,12 +58,32 @@ func (o Oftp) Call() error {
 	// 6. Send ESID
 
 	// 1. Open connection
+	var conn net.Conn
+	var err error
+
+	if o.NetworkTLS {
+		/*
+			netDialer := &net.Dialer{
+				Timeout: time.Duration(o.NetworkTimeout) * time.Second,
+			}
+		*/
+		tlsConfig := &tls.Config{
+			InsecureSkipVerify: true,
+		}
 
-	conn, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", o.NetworkHost, o.NetworkPort), time.Duration(o.NetworkTimeout)*time.Second)
-	if err != nil {
-		return err
+		//conn, err = tls.DialWithDialer(netDialer, "tcp", fmt.Sprintf("%s:%d", o.NetworkHost, o.NetworkPort), tlsConfig)
+		conn, err = tls.Dial("tcp", fmt.Sprintf("%s:%d", o.NetworkHost, o.NetworkPort), tlsConfig)
+		if err != nil {
+			return err
+		}
+		defer conn.Close()
+	} else {
+		conn, err = net.DialTimeout("tcp", fmt.Sprintf("%s:%d", o.NetworkHost, o.NetworkPort), time.Duration(o.NetworkTimeout)*time.Second)
+		if err != nil {
+			return err
+		}
+		defer conn.Close()
 	}
-	defer conn.Close()
 
 	fmt.Printf("Connected to %s:%d\n", o.NetworkHost, o.NetworkPort)