patialtech/mux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

option handler

Browse Source
This commit is contained in:
Ankit Patial 2024-11-04 11:06:49 +05:30
parent 894614cd54
commit 859d4fa458
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
router_serve.go
View File

@ -16,7 +16,18 @@ type ServeCB func(srv *http.Server) error
func (r *Router) Serve(cb ServeCB) { func (r *Router) Serve(cb ServeCB) {
// catch all options // catch all options
// lets get it thorugh all middlewares // lets get it thorugh all middlewares
r.mux.Handle("OPTIONS /", optionsHandler{}) r.OPTIONS("/", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Length", "0")
if r.ContentLength != 0 {
// Read up to 4KB of OPTIONS body (as mentioned in the
// spec as being reserved for future use), but anything
// over that is considered a waste of server resources
// (or an attack) and we abort and close the connection,
// courtesy of MaxBytesReader's EOF behavior.
mb := http.MaxBytesReader(w, r.Body, 4<<10)
io.Copy(io.Discard, mb)
}
})
srv := &http.Server{ srv := &http.Server{
Handler: r, Handler: r,
@ -45,18 +56,3 @@ func (r *Router) Serve(cb ServeCB) {
<-idleConnsClosed <-idleConnsClosed
} }
type optionsHandler struct{}
func (optionsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Length", "0")
if r.ContentLength != 0 {
// Read up to 4KB of OPTIONS body (as mentioned in the
// spec as being reserved for future use), but anything
// over that is considered a waste of server resources
// (or an attack) and we abort and close the connection,
// courtesy of MaxBytesReader's EOF behavior.
mb := http.MaxBytesReader(w, r.Body, 4<<10)
io.Copy(io.Discard, mb)
}
}