diff options
author | AdamKorcz <44787359+AdamKorcz@users.noreply.github.com> | 2021-03-31 15:28:20 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-03-31 07:28:20 -0700 |
commit | 9e96a69018343aad6a0ac106fd7ca41298385841 (patch) | |
tree | 451b33a70796c181ffa51f8384146cb80b9facc3 | |
parent | c06a4c0324d3bf11ac8d683420c474e4b79c84f5 (diff) | |
download | oss-fuzz-9e96a69018343aad6a0ac106fd7ca41298385841.tar.gz |
[hugo] Fix build (#5538)
-rw-r--r-- | projects/hugo/Dockerfile | 3 | ||||
-rwxr-xr-x | projects/hugo/build.sh | 1 | ||||
-rw-r--r-- | projects/hugo/fuzz.go | 61 |
3 files changed, 63 insertions, 2 deletions
diff --git a/projects/hugo/Dockerfile b/projects/hugo/Dockerfile index 3584ce980..bfba1a283 100644 --- a/projects/hugo/Dockerfile +++ b/projects/hugo/Dockerfile @@ -16,7 +16,6 @@ FROM gcr.io/oss-fuzz-base/base-builder -ENV GO111MODULE on RUN git clone https://github.com/gohugoio/hugo -COPY build.sh $SRC/ +COPY build.sh fuzz.go $SRC/ WORKDIR $SRC/hugo diff --git a/projects/hugo/build.sh b/projects/hugo/build.sh index e9015c5ac..c1950ca1e 100755 --- a/projects/hugo/build.sh +++ b/projects/hugo/build.sh @@ -15,5 +15,6 @@ # ################################################################################ +mv $SRC/fuzz.go $SRC/hugo/tpl/transform/ compile_go_fuzzer github.com/gohugoio/hugo/tpl/transform FuzzMarkdownify fuzzmarkdownify diff --git a/projects/hugo/fuzz.go b/projects/hugo/fuzz.go new file mode 100644 index 000000000..7a22e6d50 --- /dev/null +++ b/projects/hugo/fuzz.go @@ -0,0 +1,61 @@ +// +build gofuzz + +// Copyright 2021 Google LLC +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package transform + +import ( + "github.com/gohugoio/hugo/common/loggers" + "github.com/gohugoio/hugo/config" + "github.com/gohugoio/hugo/deps" + "github.com/gohugoio/hugo/helpers" + "github.com/gohugoio/hugo/hugofs" + "github.com/gohugoio/hugo/langs" + "github.com/spf13/afero" + "github.com/spf13/viper" +) + +func newFuzzDeps(cfg config.Provider) *deps.Deps { + cfg.Set("contentDir", "content") + cfg.Set("i18nDir", "i18n") + + l := langs.NewLanguage("en", cfg) + + cs, _ := helpers.NewContentSpec(l, loggers.NewErrorLogger(), afero.NewMemMapFs()) + + return &deps.Deps{ + Cfg: cfg, + Fs: hugofs.NewMem(l), + ContentSpec: cs, + } +} + +func FuzzMarkdownify(data []byte) int { + v := viper.New() + v.Set("contentDir", "content") + ns := New(newFuzzDeps(v)) + + for _, test := range []struct { + s interface{} + }{ + {string(data)}, + } { + _, err := ns.Markdownify(test.s) + if err != nil { + return 0 + } + } + return 1 +} |