Initial commit, based mostly on https://github.com/austburn/gocrypt

server.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	"net"
+	"os"
+
+	"github.com/tardisx/netgiv/secure"
+)
+
+type Server struct {
+	port int
+}
+
+func (s *Server) Run() {
+	address := fmt.Sprintf("127.0.0.1:%d", s.port)
+	networkAddress, _ := net.ResolveTCPAddr("tcp", address)
+
+	listener, err := net.ListenTCP("tcp", networkAddress)
+	if err != nil {
+		fmt.Print(err)
+		os.Exit(2)
+	}
+
+	for {
+		conn, err := listener.AcceptTCP()
+
+		if err != nil {
+			fmt.Print(err)
+		}
+
+		go handleConnection(conn)
+	}
+}
+
+func handleConnection(conn *net.TCPConn) {
+	defer conn.Close()
+	sharedKey := secure.Handshake(conn)
+	secureConnection := secure.SecureConnection{Conn: conn, SharedKey: sharedKey}
+
+	for {
+		msg := make([]byte, 1024)
+		_, err := secureConnection.Read(msg)
+
+		if err != nil {
+			break
+		}
+
+		secureConnection.Write(msg)
+	}
+}