forked from quic-go/quic-go
introduce stream callback in session, move http2 stuff to main.go
This commit is contained in:
@@ -6,6 +6,9 @@ import (
|
||||
"net"
|
||||
"os"
|
||||
|
||||
"golang.org/x/net/http2"
|
||||
"golang.org/x/net/http2/hpack"
|
||||
|
||||
"github.com/lucas-clemente/quic-go"
|
||||
"github.com/lucas-clemente/quic-go/crypto"
|
||||
"github.com/lucas-clemente/quic-go/protocol"
|
||||
@@ -77,7 +80,7 @@ func main() {
|
||||
|
||||
session, ok := sessions[publicHeader.ConnectionID]
|
||||
if !ok {
|
||||
session = quic.NewSession(conn, publicHeader.ConnectionID, serverConfig)
|
||||
session = quic.NewSession(conn, publicHeader.ConnectionID, serverConfig, handleStream)
|
||||
sessions[publicHeader.ConnectionID] = session
|
||||
}
|
||||
err = session.HandlePacket(remoteAddr, data[0:n-r.Len()], publicHeader, r)
|
||||
@@ -86,3 +89,30 @@ func main() {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func handleStream(frame *quic.StreamFrame) *quic.StreamFrame {
|
||||
h2r := bytes.NewReader(frame.Data)
|
||||
var reply bytes.Buffer
|
||||
h2framer := http2.NewFramer(&reply, h2r)
|
||||
h2framer.ReadMetaHeaders = hpack.NewDecoder(1024, nil)
|
||||
h2frame, err := h2framer.ReadFrame()
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
h2headersFrame := h2frame.(*http2.MetaHeadersFrame)
|
||||
fmt.Printf("%#v\n", h2headersFrame)
|
||||
|
||||
var replyHeaders bytes.Buffer
|
||||
enc := hpack.NewEncoder(&replyHeaders)
|
||||
enc.WriteField(hpack.HeaderField{Name: ":status", Value: "204"})
|
||||
h2framer.WriteHeaders(http2.HeadersFrameParam{
|
||||
StreamID: h2frame.Header().StreamID,
|
||||
EndHeaders: true,
|
||||
BlockFragment: replyHeaders.Bytes(),
|
||||
})
|
||||
|
||||
return &quic.StreamFrame{
|
||||
StreamID: frame.StreamID,
|
||||
Data: reply.Bytes(),
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user