example: add change-html

examples/change-html/main.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	"regexp"
+	"strconv"
+	"strings"
+
+	log "github.com/sirupsen/logrus"
+
+	"github.com/lqqyt2423/go-mitmproxy/addon"
+	"github.com/lqqyt2423/go-mitmproxy/flow"
+	"github.com/lqqyt2423/go-mitmproxy/proxy"
+)
+
+var titleRegexp = regexp.MustCompile("(<title>)(.*?)(</title>)")
+
+type ChangeHtml struct {
+	addon.Base
+}
+
+func (c *ChangeHtml) Response(f *flow.Flow) {
+	contentType := f.Response.Header.Get("Content-Type")
+	if !strings.Contains(contentType, "text/html") {
+		return
+	}
+
+	// change html <title> end with: " - go-mitmproxy"
+	f.Response.ReplaceToDecodedBody()
+	f.Response.Body = titleRegexp.ReplaceAll(f.Response.Body, []byte("${1}${2} - go-mitmproxy${3}"))
+	f.Response.Header.Set("Content-Length", strconv.Itoa(len(f.Response.Body)))
+}
+
+func main() {
+	opts := &proxy.Options{
+		Addr:              ":9080",
+		StreamLargeBodies: 1024 * 1024 * 5,
+	}
+
+	p, err := proxy.NewProxy(opts)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	p.AddAddon(&ChangeHtml{})
+
+	log.Fatal(p.Start())
+}