You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
609 B
Go
38 lines
609 B
Go
3 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
|
||
|
|
"github.com/lqqyt2423/go-mitmproxy/addon"
|
||
|
|
"github.com/lqqyt2423/go-mitmproxy/flow"
|
||
|
|
"github.com/lqqyt2423/go-mitmproxy/proxy"
|
||
|
|
)
|
||
|
|
|
||
|
|
type AddHeader struct {
|
||
|
|
addon.Base
|
||
|
|
count int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a *AddHeader) Responseheaders(f *flow.Flow) {
|
||
|
|
a.count += 1
|
||
|
|
f.Response.Header.Add("x-count", strconv.Itoa(a.count))
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
opts := &proxy.Options{
|
||
|
|
Addr: ":9080",
|
||
|
|
StreamLargeBodies: 1024 * 1024 * 5,
|
||
|
|
}
|
||
|
|
|
||
|
|
p, err := proxy.NewProxy(opts)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
p.AddAddon(&AddHeader{})
|
||
|
|
|
||
|
|
log.Fatal(p.Start())
|
||
|
|
}
|