]> source.dussan.org Git - gitea.git/commitdiff
Fix go-fuzz (#15596)
author6543 <6543@obermui.de>
Fri, 23 Apr 2021 08:22:52 +0000 (10:22 +0200)
committerGitHub <noreply@github.com>
Fri, 23 Apr 2021 08:22:52 +0000 (16:22 +0800)
* Fix go-fuzz

followup of https://github.com/go-gitea/gitea/pull/15175

* simplify

* enhance

tools/fuzz.go

index 6ed1b40003d8c68fba03343760607da2bdce862d..4b5b72d1d0d52a865a80be0f99c8a95b6c95176f 100644 (file)
@@ -7,6 +7,9 @@
 package fuzz
 
 import (
+       "bytes"
+       "io"
+
        "code.gitea.io/gitea/modules/markup"
        "code.gitea.io/gitea/modules/markup/markdown"
 )
@@ -18,17 +21,26 @@ import (
 // (for example, the input is lexically correct and was parsed successfully).
 // -1 if the input must not be added to corpus even if gives new coverage and 0 otherwise.
 
+var (
+       renderContext = markup.RenderContext{
+               URLPrefix: "https://example.com",
+               Metas: map[string]string{
+                       "user": "go-gitea",
+                       "repo": "gitea",
+               },
+       }
+)
+
 func FuzzMarkdownRenderRaw(data []byte) int {
-       _ = markdown.RenderRaw(data, "", false)
+       err := markdown.RenderRaw(&renderContext, bytes.NewReader(data), io.Discard)
+       if err != nil {
+               return 0
+       }
        return 1
 }
 
 func FuzzMarkupPostProcess(data []byte) int {
-       var localMetas = map[string]string{
-               "user": "go-gitea",
-               "repo": "gitea",
-       }
-       _, err := markup.PostProcess(data, "https://example.com", localMetas, false)
+       err := markup.PostProcess(&renderContext, bytes.NewReader(data), io.Discard)
        if err != nil {
                return 0
        }