157 lines
3.0 KiB
Go
157 lines
3.0 KiB
Go
/*
|
|
* Copyright 2022 Oleg Borodin <borodin@unix7.org>
|
|
*/
|
|
|
|
package dsrpc
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
type Context struct {
|
|
start time.Time
|
|
remoteHost string
|
|
sockReader io.Reader
|
|
sockWriter io.Writer
|
|
|
|
reqHeader *Header
|
|
reqRPC *Request
|
|
reqPacket *Packet
|
|
|
|
resPacket *Packet
|
|
resHeader *Header
|
|
resRPC *Response
|
|
|
|
binReader io.Reader
|
|
binWriter io.Writer
|
|
}
|
|
|
|
|
|
func NewContext() *Context {
|
|
context := &Context{}
|
|
context.start = time.Now()
|
|
return context
|
|
}
|
|
|
|
func CreateContext(conn net.Conn) *Context {
|
|
context := &Context{}
|
|
context.start = time.Now()
|
|
context.sockReader = conn
|
|
context.sockWriter = conn
|
|
|
|
context.reqPacket = NewPacket()
|
|
context.resPacket = NewPacket()
|
|
|
|
context.reqHeader = NewHeader()
|
|
context.reqRPC = NewRequest()
|
|
|
|
context.resHeader = NewHeader()
|
|
context.resRPC = NewResponse()
|
|
context.resRPC = NewResponse()
|
|
|
|
return context
|
|
}
|
|
|
|
func (context *Context) Request() *Request {
|
|
return context.reqRPC
|
|
}
|
|
|
|
func (context *Context) RemoteHost() string {
|
|
return context.remoteHost
|
|
}
|
|
|
|
func (context *Context) Start() time.Time {
|
|
return context.start
|
|
}
|
|
|
|
func (context *Context) Method() string {
|
|
var method string
|
|
if context.reqRPC != nil {
|
|
method = context.reqRPC.Method
|
|
}
|
|
return method
|
|
}
|
|
|
|
func (context *Context) ReqRpcSize() int64 {
|
|
var size int64
|
|
if context.reqHeader != nil {
|
|
size = context.reqHeader.rpcSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
|
|
func (context *Context) ReqBinSize() int64 {
|
|
var size int64
|
|
if context.reqHeader != nil {
|
|
size = context.reqHeader.binSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
func (context *Context) ResBinSize() int64 {
|
|
var size int64
|
|
if context.resHeader != nil {
|
|
size = context.resHeader.binSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
func (context *Context) ResRpcSize() int64 {
|
|
var size int64
|
|
if context.resHeader != nil {
|
|
size = context.resHeader.rpcSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
func (context *Context) ReqSize() int64 {
|
|
var size int64
|
|
if context.reqHeader != nil {
|
|
size += context.reqHeader.binSize
|
|
size += context.reqHeader.rpcSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
func (context *Context) ResSize() int64 {
|
|
var size int64
|
|
if context.resHeader != nil {
|
|
size += context.resHeader.binSize
|
|
size += context.resHeader.rpcSize
|
|
}
|
|
return size
|
|
}
|
|
|
|
|
|
|
|
func (context *Context) SetAuthIdent(ident []byte) {
|
|
context.reqRPC.Auth.Ident = ident
|
|
}
|
|
|
|
func (context *Context) SetAuthSalt(salt []byte) {
|
|
context.reqRPC.Auth.Salt = salt
|
|
}
|
|
|
|
func (context *Context) SetAuthHash(hash []byte) {
|
|
context.reqRPC.Auth.Hash = hash
|
|
}
|
|
|
|
func (context *Context) AuthIdent() []byte {
|
|
return context.reqRPC.Auth.Ident
|
|
}
|
|
|
|
func (context *Context) AuthSalt() []byte {
|
|
return context.reqRPC.Auth.Salt
|
|
}
|
|
|
|
func (context *Context) AuthHash() []byte {
|
|
return context.reqRPC.Auth.Hash
|
|
}
|
|
|
|
func (context *Context) Auth() *Auth {
|
|
return context.reqRPC.Auth
|
|
}
|