diff options
| author | Jeff Halter <868228+jhalter@users.noreply.github.com> | 2021-07-28 18:21:52 -0700 |
|---|---|---|
| committer | Jeff Halter <868228+jhalter@users.noreply.github.com> | 2021-07-28 14:21:52 -0700 |
| commit | 22c599abc18895f73e96095f35b71cf3357d41b4 (patch) | |
| tree | 482ef57d386c955692ea43c43e4655b3c3763499 /client_conn.go | |
| parent | 71c56068adca18f76ebee86355f000a3e51d3127 (diff) | |
Move code to hotline dir
Diffstat (limited to 'client_conn.go')
| -rw-r--r-- | client_conn.go | 247 |
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)), - }, - } -} |