github
/
nps
mirror of https://github.com/ehang-io/nps
1
0
Fork 0
Code Issues Releases Wiki Activity

change mux data struct, fix #250

pull/310/head
ffdfgdfg 2019-12-16 23:10:44 +08:00
parent eab5708b33
commit ae28d41231
4 changed files with 71 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
lib/common/netpackager.go
View File

@ -22,24 +22,34 @@ type BasePackager struct {
Content []byte Content []byte
} }
func (Self *BasePackager) NewPac(contents ...interface{}) (err error) { func (Self *BasePackager) NewPac(content []byte) (err error) {
Self.clean() Self.clean()
for _, content := range contents { if content != nil {
switch content.(type) { n := len(content)
case nil: if n > MAXIMUM_SEGMENT_SIZE {
err = errors.New("mux:packer: newpack content segment too large")
}
Self.Content = Self.Content[:n]
copy(Self.Content, content)
} else {
Self.Content = Self.Content[:0] Self.Content = Self.Content[:0]
case []byte:
err = Self.appendByte(content.([]byte))
case string:
err = Self.appendByte([]byte(content.(string)))
if err != nil {
return
}
err = Self.appendByte([]byte(CONN_DATA_SEQ))
default:
err = Self.marshal(content)
}
} }
//for _, content := range contents {
// switch content.(type) {
// case nil:
// Self.Content = Self.Content[:0]
// case []byte:
// err = Self.appendByte(content.([]byte))
// case string:
// err = Self.appendByte([]byte(content.(string)))
// if err != nil {
// return
// }
// err = Self.appendByte([]byte(CONN_DATA_SEQ))
// default:
// err = Self.marshal(content)
// }
//}
Self.setLength() Self.setLength()
return return
} }
@ -78,6 +88,9 @@ func (Self *BasePackager) UnPack(reader io.Reader) (n uint16, err error) {
if int(Self.Length) > cap(Self.Content) { if int(Self.Length) > cap(Self.Content) {
err = errors.New("unpack err, content length too large") err = errors.New("unpack err, content length too large")
} }
if Self.Length > MAXIMUM_SEGMENT_SIZE {
err = errors.New("mux:packer: unpack content segment too large")
}
Self.Content = Self.Content[:int(Self.Length)] Self.Content = Self.Content[:int(Self.Length)]
//n, err := io.ReadFull(reader, Self.Content) //n, err := io.ReadFull(reader, Self.Content)
//if n != int(Self.Length) { //if n != int(Self.Length) {
@ -128,61 +141,49 @@ type ConnPackager struct {
BasePackager BasePackager
} }
func (Self *ConnPackager) NewPac(connType uint8, content ...interface{}) (err error) { //func (Self *ConnPackager) NewPac(connType uint8, content ...interface{}) (err error) {
Self.ConnType = connType // Self.ConnType = connType
err = Self.BasePackager.NewPac(content...) // err = Self.BasePackager.NewPac(content...)
return // return
} //}
//
func (Self *ConnPackager) Pack(writer io.Writer) (err error) { //func (Self *ConnPackager) Pack(writer io.Writer) (err error) {
err = binary.Write(writer, binary.LittleEndian, Self.ConnType) // err = binary.Write(writer, binary.LittleEndian, Self.ConnType)
if err != nil { // if err != nil {
return // return
} // }
err = Self.BasePackager.Pack(writer) // err = Self.BasePackager.Pack(writer)
return // return
} //}
//
func (Self *ConnPackager) UnPack(reader io.Reader) (n uint16, err error) { //func (Self *ConnPackager) UnPack(reader io.Reader) (n uint16, err error) {
err = binary.Read(reader, binary.LittleEndian, &Self.ConnType) // err = binary.Read(reader, binary.LittleEndian, &Self.ConnType)
if err != nil && err != io.EOF { // if err != nil && err != io.EOF {
return // return
} // }
n, err = Self.BasePackager.UnPack(reader) // n, err = Self.BasePackager.UnPack(reader)
n += 2 // n += 2
return // return
} //}
type MuxPackager struct { type MuxPackager struct {
Flag uint8 Flag uint8
Id int32 Id int32
Window uint32 RemainLength uint32
ReadLength uint32
BasePackager BasePackager
} }
func (Self *MuxPackager) NewPac(flag uint8, id int32, content ...interface{}) (err error) { func (Self *MuxPackager) NewPac(flag uint8, id int32, content interface{}) (err error) {
Self.Flag = flag Self.Flag = flag
Self.Id = id Self.Id = id
switch flag { switch flag {
case MUX_PING_FLAG, MUX_PING_RETURN, MUX_NEW_MSG, MUX_NEW_MSG_PART: case MUX_PING_FLAG, MUX_PING_RETURN, MUX_NEW_MSG, MUX_NEW_MSG_PART:
Self.Content = WindowBuff.Get() Self.Content = WindowBuff.Get()
err = Self.BasePackager.NewPac(content...) err = Self.BasePackager.NewPac(content.([]byte))
//logs.Warn(Self.Length, string(Self.Content)) //logs.Warn(Self.Length, string(Self.Content))
case MUX_MSG_SEND_OK: case MUX_MSG_SEND_OK:
// MUX_MSG_SEND_OK contains two data // MUX_MSG_SEND_OK contains two data
switch content[0].(type) { Self.RemainLength = content.(uint32)
case int:
Self.Window = uint32(content[0].(int))
case uint32:
Self.Window = content[0].(uint32)
}
switch content[1].(type) {
case int:
Self.ReadLength = uint32(content[1].(int))
case uint32:
Self.ReadLength = content[1].(uint32)
}
} }
return return
} }
@ -201,11 +202,7 @@ func (Self *MuxPackager) Pack(writer io.Writer) (err error) {
err = Self.BasePackager.Pack(writer) err = Self.BasePackager.Pack(writer)
WindowBuff.Put(Self.Content) WindowBuff.Put(Self.Content)
case MUX_MSG_SEND_OK: case MUX_MSG_SEND_OK:
err = binary.Write(writer, binary.LittleEndian, Self.Window) err = binary.Write(writer, binary.LittleEndian, Self.RemainLength)
if err != nil {
return
}
err = binary.Write(writer, binary.LittleEndian, Self.ReadLength)
} }
return return
} }
@ -226,12 +223,7 @@ func (Self *MuxPackager) UnPack(reader io.Reader) (n uint16, err error) {
n, err = Self.BasePackager.UnPack(reader) n, err = Self.BasePackager.UnPack(reader)
//logs.Warn("unpack", Self.Length, string(Self.Content)) //logs.Warn("unpack", Self.Length, string(Self.Content))
case MUX_MSG_SEND_OK: case MUX_MSG_SEND_OK:
err = binary.Read(reader, binary.LittleEndian, &Self.Window) err = binary.Read(reader, binary.LittleEndian, &Self.RemainLength)
if err != nil {
return
}
n += 4 // uint32
err = binary.Read(reader, binary.LittleEndian, &Self.ReadLength)
n += 4 // uint32 n += 4 // uint32
} }
n += 5 //uint8 int32 n += 5 //uint8 int32

6
lib/mux/conn.go
View File

@ -265,7 +265,7 @@ start:
Self.bufQueue.Push(element) Self.bufQueue.Push(element)
// status check finish, now we can push the element into the queue // status check finish, now we can push the element into the queue
if wait == 0 { if wait == 0 {
Self.mux.sendInfo(common.MUX_MSG_SEND_OK, id, Self.maxSize, newRemaining) Self.mux.sendInfo(common.MUX_MSG_SEND_OK, id, newRemaining)
// send the remaining window size, not including zero size // send the remaining window size, not including zero size
} }
return nil return nil
@ -333,7 +333,7 @@ func (Self *ReceiveWindow) sendStatus(id int32, l uint16) {
// now we get the current window status success // now we get the current window status success
if wait == 1 { if wait == 1 {
//logs.Warn("send the wait status", remaining) //logs.Warn("send the wait status", remaining)
Self.mux.sendInfo(common.MUX_MSG_SEND_OK, id, atomic.LoadUint32(&Self.maxSize), remaining) Self.mux.sendInfo(common.MUX_MSG_SEND_OK, id, remaining)
} }
return return
} }
@ -394,7 +394,7 @@ func (Self *SendWindow) SetSendBuf(buf []byte) {
Self.off = 0 Self.off = 0
} }
func (Self *SendWindow) SetSize(windowSize, newRemaining uint32) (closed bool) { func (Self *SendWindow) SetSize(newRemaining uint32) (closed bool) {
// set the window size from receive window // set the window size from receive window
defer func() { defer func() {
if recover() != nil { if recover() != nil {

6
lib/mux/mux.go
View File

@ -92,13 +92,13 @@ func (s *Mux) Addr() net.Addr {
return s.conn.LocalAddr() return s.conn.LocalAddr()
} }
func (s *Mux) sendInfo(flag uint8, id int32, data ...interface{}) { func (s *Mux) sendInfo(flag uint8, id int32, data interface{}) {
if s.IsClose { if s.IsClose {
return return
} }
var err error var err error
pack := common.MuxPack.Get() pack := common.MuxPack.Get()
err = pack.NewPac(flag, id, data...) err = pack.NewPac(flag, id, data)
if err != nil { if err != nil {
common.MuxPack.Put(pack) common.MuxPack.Put(pack)
logs.Error("mux: new pack err", err) logs.Error("mux: new pack err", err)
@ -297,7 +297,7 @@ func (s *Mux) readSession() {
if connection.isClose { if connection.isClose {
continue continue
} }
connection.sendWindow.SetSize(pack.Window, pack.ReadLength) connection.sendWindow.SetSize(pack.RemainLength)
continue continue
case common.MUX_CONN_CLOSE: //close the connection case common.MUX_CONN_CLOSE: //close the connection
connection.closeFlag = true connection.closeFlag = true

2
lib/version/version.go
View File

@ -4,5 +4,5 @@ const VERSION = "0.25.2"
// Compulsory minimum version, Minimum downward compatibility to this version // Compulsory minimum version, Minimum downward compatibility to this version
func GetVersion() string { func GetVersion() string {
return "0.25.0" return "0.25.2"
} }