[PD1] Generic networking server and client implemented
This commit is contained in:
parent
8553e1674e
commit
278b8e1a73
4 changed files with 114 additions and 38 deletions
|
@ -5,5 +5,5 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func main(){
|
func main(){
|
||||||
server.Run()
|
server.Run(8080)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,8 +1,6 @@
|
||||||
package protocol
|
package protocol
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -12,8 +10,9 @@ const (
|
||||||
ReqPK PacketType = iota
|
ReqPK PacketType = iota
|
||||||
ReqAllMsg
|
ReqAllMsg
|
||||||
ReqMsg
|
ReqMsg
|
||||||
|
SubmitMsg
|
||||||
SendPK
|
SendPK
|
||||||
UsrMsg
|
Msg
|
||||||
)
|
)
|
||||||
|
|
||||||
type PacketBody interface {}
|
type PacketBody interface {}
|
||||||
|
@ -39,18 +38,19 @@ type RequestMsg struct {
|
||||||
Num uint16
|
Num uint16
|
||||||
}
|
}
|
||||||
|
|
||||||
// Server --> Client: Send the client the requested public key
|
// Client --> Server: Send message from client to server
|
||||||
type SendPubKey struct {
|
type SubmitMessage struct {
|
||||||
Key []byte
|
|
||||||
}
|
|
||||||
|
|
||||||
// Bidirectional: Send messages between server and clients
|
|
||||||
type SendMessage struct {
|
|
||||||
ToUID string
|
ToUID string
|
||||||
Subject []byte
|
Subject []byte
|
||||||
Body []byte
|
Body []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Server --> Client: Send the client the requested public key
|
||||||
|
type SendPubKey struct {
|
||||||
|
Key []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// Server --> Client: Send the client a message
|
||||||
type Message struct {
|
type Message struct {
|
||||||
FromUID string
|
FromUID string
|
||||||
ToUID string
|
ToUID string
|
||||||
|
@ -58,11 +58,3 @@ type Message struct {
|
||||||
Body []byte
|
Body []byte
|
||||||
Timestamp time.Time
|
Timestamp time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p Packet) Marshal() ([]byte, error) {
|
|
||||||
return json.Marshal(p)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *Packet) Unmarshal(data []byte) error {
|
|
||||||
return json.Unmarshal(data, p)
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,30 +1,43 @@
|
||||||
package server
|
package server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
_ "PD1/internal/utils/cryptoUtils"
|
"PD1/internal/protocol"
|
||||||
_ "PD1/internal/utils/networking"
|
"PD1/internal/utils/networking"
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func clientHandler(conn net.Conn) {
|
func clientHandler(connection networking.Connection[protocol.Packet]) {
|
||||||
|
defer connection.Conn.Close()
|
||||||
|
|
||||||
|
jd := json.NewDecoder(connection.Conn)
|
||||||
|
|
||||||
|
for {
|
||||||
|
var pac protocol.Packet
|
||||||
|
jd.Decode(&pac)
|
||||||
|
|
||||||
|
switch pac.Flag {
|
||||||
|
case protocol.ReqPK:
|
||||||
|
fmt.Println("ReqPK")
|
||||||
|
case protocol.ReqAllMsg:
|
||||||
|
fmt.Println("ReqAllMsg")
|
||||||
|
case protocol.ReqMsg:
|
||||||
|
fmt.Println("ReqMsg")
|
||||||
|
case protocol.SubmitMsg:
|
||||||
|
fmt.Println("SubmitMsh")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func Run(port int) {
|
func Run(port int) {
|
||||||
ip, err := net.ResolveTCPAddr("tcp", ":8080")
|
server := networking.NewServer[protocol.Packet](port)
|
||||||
if err != nil {
|
go server.ListenLoop()
|
||||||
panic("Server could not bind to address")
|
|
||||||
}
|
|
||||||
listener, err := net.ListenTCP("tcp", ip)
|
|
||||||
if err != nil {
|
|
||||||
panic("Server could not listen on address")
|
|
||||||
}
|
|
||||||
for {
|
for {
|
||||||
conn, err := listener.Accept()
|
//Receive connection via channel
|
||||||
if err!=nil{
|
conn := <-server.C
|
||||||
panic("Server could not accept connection")
|
//Launch client handler via clientHandler
|
||||||
}
|
go clientHandler(conn)
|
||||||
go clientHandler(conn)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,5 +1,76 @@
|
||||||
package networking
|
package networking
|
||||||
|
|
||||||
func sendPacket(){
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Server[T any] struct {
|
||||||
|
listener net.Listener
|
||||||
|
C chan Connection[T]
|
||||||
|
}
|
||||||
|
|
||||||
|
type Client[T any] struct {
|
||||||
|
conn net.Conn
|
||||||
|
}
|
||||||
|
|
||||||
|
type Connection[T any] struct {
|
||||||
|
Conn net.Conn
|
||||||
|
encoder *json.Encoder
|
||||||
|
decoder *json.Decoder
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewServer[T any](port int) Server[T]{
|
||||||
|
|
||||||
|
listener, err := net.Listen("tcp", fmt.Sprintf("0.0.0.0:%d", port))
|
||||||
|
if err != nil {
|
||||||
|
panic("Server could not bind to address")
|
||||||
|
}
|
||||||
|
return Server[T]{
|
||||||
|
listener:listener,
|
||||||
|
C: make(chan Connection[T]),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server[T]) ListenLoop() {
|
||||||
|
|
||||||
|
for {
|
||||||
|
listenerConn, err := s.listener.Accept()
|
||||||
|
if err != nil {
|
||||||
|
panic("Server could not accept connection")
|
||||||
|
}
|
||||||
|
conn := NewConnection[T](listenerConn)
|
||||||
|
s.C <- conn
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Client[T]) Connect() Connection[T] {
|
||||||
|
dialConn, err := net.Dial("tcp", "localhost:8080")
|
||||||
|
if err != nil {
|
||||||
|
panic("Could not open connection to server")
|
||||||
|
}
|
||||||
|
return NewConnection[T](dialConn)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewConnection[T any](netConn net.Conn) Connection[T] {
|
||||||
|
return Connection[T]{
|
||||||
|
Conn: netConn,
|
||||||
|
encoder: json.NewEncoder(netConn),
|
||||||
|
decoder: json.NewDecoder(netConn),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (jc Connection[T]) Send(obj T) {
|
||||||
|
if err := jc.encoder.Encode(&obj); err != nil {
|
||||||
|
panic("Failed encoding data or sending it to connection")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (jc Connection[T]) Receive() T {
|
||||||
|
var obj T
|
||||||
|
if err := jc.decoder.Decode(&obj); err != nil {
|
||||||
|
panic("Failed decoding data or reading it from connection")
|
||||||
|
}
|
||||||
|
return obj
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue