aboutsummaryrefslogtreecommitdiff
path: root/client_conn.go
diff options
context:
space:
mode:
authorJeff Halter <868228+jhalter@users.noreply.github.com>2021-07-28 18:21:52 -0700
committerJeff Halter <868228+jhalter@users.noreply.github.com>2021-07-28 14:21:52 -0700
commit22c599abc18895f73e96095f35b71cf3357d41b4 (patch)
tree482ef57d386c955692ea43c43e4655b3c3763499 /client_conn.go
parent71c56068adca18f76ebee86355f000a3e51d3127 (diff)
Move code to hotline dir
Diffstat (limited to 'client_conn.go')
-rw-r--r--client_conn.go247
1 files changed, 0 insertions, 247 deletions
diff --git a/client_conn.go b/client_conn.go
deleted file mode 100644
index 5c25f92..0000000
--- a/client_conn.go
+++ /dev/null
@@ -1,247 +0,0 @@
-package hotline
-
-import (
- "bytes"
- "encoding/binary"
- "errors"
- "golang.org/x/crypto/bcrypt"
- "math/big"
- "net"
-)
-
-type byClientID []*ClientConn
-
-func (s byClientID) Len() int {
- return len(s)
-}
-
-func (s byClientID) Swap(i, j int) {
- s[i], s[j] = s[j], s[i]
-}
-
-func (s byClientID) Less(i, j int) bool {
- return s[i].uint16ID() < s[j].uint16ID()
-}
-
-// ClientConn represents a client connected to a Server
-type ClientConn struct {
- Connection net.Conn
- ID *[]byte
- Icon *[]byte
- Flags *[]byte
- UserName *[]byte
- Account *Account
- IdleTime *int
- Server *Server
- Version *[]byte
- Idle bool
- AutoReply *[]byte
- Transfers map[int][]*FileTransfer
-}
-
-func (cc *ClientConn) sendAll(t int, fields ...Field) {
- for _, c := range sortedClients(cc.Server.Clients) {
- cc.Server.outbox <- *NewTransaction(t, c.ID, fields...)
- }
-}
-
-func (cc *ClientConn) handleTransaction(transaction *Transaction) error {
- requestNum := binary.BigEndian.Uint16(transaction.Type)
- if handler, ok := TransactionHandlers[requestNum]; ok {
- for _, reqField := range handler.RequiredFields {
- field := transaction.GetField(reqField.ID)
-
- // Validate that required field is present
- if field.ID == nil {
- cc.Server.Logger.Infow(
- "Missing required field",
- "Account", cc.Account.Login, "UserName", string(*cc.UserName), "RequestType", handler.Name, "FieldID", reqField.ID,
- )
- return nil
- }
-
- if len(field.Data) < reqField.minLen {
- cc.Server.Logger.Infow(
- "Field does not meet minLen",
- "Account", cc.Account.Login, "UserName", string(*cc.UserName), "RequestType", handler.Name, "FieldID", reqField.ID,
- )
- return nil
- }
- }
- if !authorize(cc.Account.Access, handler.Access) {
- cc.Server.Logger.Infow(
- "Unauthorized Action",
- "Account", cc.Account.Login, "UserName", string(*cc.UserName), "RequestType", handler.Name,
- )
- cc.Server.outbox <- cc.NewErrReply(transaction, handler.DenyMsg)
-
- return nil
- }
-
- cc.Server.Logger.Infow(
- "Received Transaction",
- "login", cc.Account.Login,
- "name", string(*cc.UserName),
- "RequestType", handler.Name,
- )
-
- transactions, err := handler.Handler(cc, transaction)
- if err != nil {
- return err
- }
- for _, t := range transactions {
- cc.Server.outbox <- t
- }
- } else {
- cc.Server.Logger.Errorw(
- "Unimplemented transaction type received",
- "UserName", string(*cc.UserName), "RequestID", requestNum,
- )
- }
-
- cc.Server.mux.Lock()
- defer cc.Server.mux.Unlock()
-
- // if user was idle and this is a non-keepalive transaction
- if *cc.IdleTime > userIdleSeconds && requestNum != tranKeepAlive {
- flagBitmap := big.NewInt(int64(binary.BigEndian.Uint16(*cc.Flags)))
- flagBitmap.SetBit(flagBitmap, userFlagAway, 0)
- binary.BigEndian.PutUint16(*cc.Flags, uint16(flagBitmap.Int64()))
- cc.Idle = false
- //*cc.IdleTime = 0
-
- cc.sendAll(
- tranNotifyChangeUser,
- NewField(fieldUserID, *cc.ID),
- NewField(fieldUserFlags, *cc.Flags),
- NewField(fieldUserName, *cc.UserName),
- NewField(fieldUserIconID, *cc.Icon),
- )
-
- //return nil
- }
-
- // TODO: Don't we need to skip this if requestNum == tranKeepalive ??
- *cc.IdleTime = 0
-
- return nil
-}
-
-func (cc *ClientConn) Authenticate(login string, password []byte) bool {
- if account, ok := cc.Server.Accounts[login]; ok {
- return bcrypt.CompareHashAndPassword([]byte(account.Password), password) == nil
- }
-
- return false
-}
-
-func (cc *ClientConn) uint16ID() uint16 {
- id, _ := byteToInt(*cc.ID)
- return uint16(id)
-}
-
-// Authorize checks if the user account has the specified permission
-func (cc *ClientConn) Authorize(access int) bool {
- if access == 0 {
- return true
- }
-
- accessBitmap := big.NewInt(int64(binary.BigEndian.Uint64(*cc.Account.Access)))
-
- return accessBitmap.Bit(63-access) == 1
-}
-
-// Disconnect notifies other clients that a client has disconnected
-func (cc ClientConn) Disconnect() {
- cc.Server.mux.Lock()
- defer cc.Server.mux.Unlock()
-
- delete(cc.Server.Clients, binary.BigEndian.Uint16(*cc.ID))
-
- cc.NotifyOthers(*NewTransaction(tranNotifyDeleteUser, nil, NewField(fieldUserID, *cc.ID)))
-
- if err := cc.Connection.Close(); err != nil {
- cc.Server.Logger.Errorw("error closing client connection", "RemoteAddr", cc.Connection.RemoteAddr())
- }
-}
-
-// NotifyOthers sends transaction t to other clients connected to the server
-func (cc ClientConn) NotifyOthers(t Transaction) {
- for _, c := range sortedClients(cc.Server.Clients) {
- if c.ID != cc.ID {
- t.clientID = c.ID
- cc.Server.outbox <- t
- }
- }
-}
-
-type handshake struct {
- Protocol [4]byte // Must be 0x54525450 TRTP
- SubProtocol [4]byte
- Version [2]byte // Always 1
- SubVersion [2]byte
-}
-
-// Handshake
-// After establishing TCP connection, both client and server start the handshake process
-// in order to confirm that each of them comply with requirements of the other.
-// The information provided in this initial data exchange identifies protocols,
-// and their versions, used in the communication. In the case where, after inspection,
-// the capabilities of one of the subjects do not comply with the requirements of the other,
-// the connection is dropped.
-//
-// The following information is sent to the server:
-// Description Size Data Note
-// Protocol ID 4 TRTP 0x54525450
-// Sub-protocol ID 4 HOTL User defined
-// VERSION 2 1 Currently 1
-// Sub-version 2 2 User defined
-//
-// The server replies with the following:
-// Description Size Data Note
-// Protocol ID 4 TRTP
-//Error code 4 Error code returned by the server (0 = no error)
-func Handshake(conn net.Conn, buf []byte) error {
- var h handshake
- r := bytes.NewReader(buf)
- if err := binary.Read(r, binary.BigEndian, &h); err != nil {
- return err
- }
-
- if h.Protocol != [4]byte{0x54, 0x52, 0x54, 0x50} {
- return errors.New("invalid handshake")
- }
-
- _, err := conn.Write([]byte{84, 82, 84, 80, 0, 0, 0, 0})
- return err
-}
-
-// NewReply returns a reply Transaction with fields for the ClientConn
-func (cc *ClientConn) NewReply(t *Transaction, fields ...Field) Transaction {
- reply := Transaction{
- Flags: 0x00,
- IsReply: 0x01,
- Type: t.Type,
- ID: t.ID,
- clientID: cc.ID,
- ErrorCode: []byte{0, 0, 0, 0},
- Fields: fields,
- }
-
- return reply
-}
-
-// NewErrReply returns an error reply Transaction with errMsg
-func (cc *ClientConn) NewErrReply(t *Transaction, errMsg string) Transaction {
- return Transaction{
- clientID: cc.ID,
- Flags: 0x00,
- IsReply: 0x01,
- Type: []byte{0, 0},
- ID: t.ID,
- ErrorCode: []byte{0, 0, 0, 1},
- Fields: []Field{
- NewField(fieldError, []byte(errMsg)),
- },
- }
-}