56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/tardisx/netgiv/secure"
|
|
)
|
|
|
|
type Client struct {
|
|
port int
|
|
}
|
|
|
|
func (c *Client) Connect() error {
|
|
address := fmt.Sprintf("127.0.0.1:%d", c.port)
|
|
serverAddress, _ := net.ResolveTCPAddr("tcp", address)
|
|
|
|
conn, err := net.DialTCP("tcp", nil, serverAddress)
|
|
if err != nil {
|
|
return errors.New("Problem connecting to server, is it running?\n")
|
|
}
|
|
defer conn.Close()
|
|
|
|
fmt.Printf("Connection on %s\n", address)
|
|
|
|
sharedKey := secure.Handshake(conn)
|
|
secureConnection := secure.SecureConnection{Conn: conn, SharedKey: sharedKey}
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
|
|
for {
|
|
fmt.Print("> ")
|
|
// Read up to the newline character
|
|
msg, _ := reader.ReadBytes(0xA)
|
|
// Kill the newline char
|
|
msg = msg[:len(msg)-1]
|
|
|
|
_, err := secureConnection.Write(msg)
|
|
|
|
response := make([]byte, 1024)
|
|
|
|
_, err = secureConnection.Read(response)
|
|
if err != nil {
|
|
fmt.Print("Connection to the server was closed.\n")
|
|
break
|
|
}
|
|
|
|
fmt.Printf("%s\n", response)
|
|
}
|
|
|
|
return nil
|
|
}
|