Fuzzing: Add archive fuzzer
Signed-off-by: AdamKorcz <adam@adalogics.com>
This commit is contained in:
		
							
								
								
									
										52
									
								
								contrib/fuzz/archive_fuzzer.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								contrib/fuzz/archive_fuzzer.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,52 @@ | ||||
| // +build gofuzz | ||||
|  | ||||
| /* | ||||
|    Copyright The containerd Authors. | ||||
|    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 fuzz | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
| 	"io/ioutil" | ||||
| 	"os" | ||||
|  | ||||
| 	fuzz "github.com/AdaLogics/go-fuzz-headers" | ||||
|  | ||||
| 	"github.com/containerd/containerd/archive" | ||||
| ) | ||||
|  | ||||
| // FuzzApply implements a fuzzer that applies | ||||
| // a fuzzed tar archive on a directory | ||||
| func FuzzApply(data []byte) int { | ||||
| 	f := fuzz.NewConsumer(data) | ||||
| 	iters, err := f.GetInt() | ||||
| 	if err != nil { | ||||
| 		return 0 | ||||
| 	} | ||||
| 	maxIters := 20 | ||||
| 	tmpDir, err := ioutil.TempDir("", "prefix-test") | ||||
| 	if err != nil { | ||||
| 		return 0 | ||||
| 	} | ||||
| 	defer os.RemoveAll(tmpDir) | ||||
| 	for i := 0; i < iters%maxIters; i++ { | ||||
| 		rBytes, err := f.TarBytes() | ||||
| 		if err != nil { | ||||
| 			return 0 | ||||
| 		} | ||||
| 		r := bytes.NewReader(rBytes) | ||||
| 		_, _ = archive.Apply(context.Background(), tmpDir, r) | ||||
| 	} | ||||
| 	return 1 | ||||
| } | ||||
| @@ -17,9 +17,10 @@ | ||||
| cd "$(dirname "${BASH_SOURCE[0]}")" | ||||
| cd ../../ | ||||
|  | ||||
| compile_go_fuzzer github.com/containerd/containerd/contrib/fuzz FuzzFiltersParse fuzz_filters_parse | ||||
| compile_go_fuzzer github.com/containerd/containerd/contrib/fuzz FuzzPlatformsParse fuzz_platforms_parse | ||||
|  | ||||
| # Don't move docker_fuzzer.go back into contrib/fuzz | ||||
| mv contrib/fuzz/docker_fuzzer.go remotes/docker/ | ||||
| compile_go_fuzzer github.com/containerd/containerd/remotes/docker FuzzFetcher fuzz_fetcher | ||||
| mv remotes/docker/docker_fuzzer.go contrib/fuzz/ | ||||
|  | ||||
| compile_go_fuzzer github.com/containerd/containerd/contrib/fuzz FuzzFiltersParse fuzz_filters_parse | ||||
| compile_go_fuzzer github.com/containerd/containerd/contrib/fuzz FuzzPlatformsParse fuzz_platforms_parse | ||||
| compile_go_fuzzer github.com/containerd/containerd/contrib/fuzz FuzzApply fuzz_apply | ||||
|   | ||||
							
								
								
									
										1
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										1
									
								
								go.mod
									
									
									
									
									
								
							| @@ -3,6 +3,7 @@ module github.com/containerd/containerd | ||||
| go 1.16 | ||||
|  | ||||
| require ( | ||||
| 	github.com/AdaLogics/go-fuzz-headers v0.0.0-20210715213245-6c3934b029d8 | ||||
| 	github.com/Microsoft/go-winio v0.5.0 | ||||
| 	github.com/Microsoft/hcsshim v0.8.18 | ||||
| 	github.com/containerd/aufs v1.0.0 | ||||
|   | ||||
							
								
								
									
										2
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								go.sum
									
									
									
									
									
								
							| @@ -22,6 +22,8 @@ cloud.google.com/go/storage v1.0.0/go.mod h1:IhtSnM/ZTZV8YYJWCY8RULGVqBDmpoyjwiy | ||||
| cloud.google.com/go/storage v1.5.0/go.mod h1:tpKbwo567HUNpVclU5sGELwQWBDZ8gh0ZeosJ0Rtdos= | ||||
| cloud.google.com/go/storage v1.6.0/go.mod h1:N7U0C8pVQ/+NIKOBQyamJIeKQKkZ+mxpohlUTyfDhBk= | ||||
| dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU= | ||||
| github.com/AdaLogics/go-fuzz-headers v0.0.0-20210715213245-6c3934b029d8 h1:V8krnnfGj4pV65YLUm3C0/8bl7V5Nry2Pwvy3ru/wLc= | ||||
| github.com/AdaLogics/go-fuzz-headers v0.0.0-20210715213245-6c3934b029d8/go.mod h1:CzsSbkDixRphAF5hS6wbMKq0eI6ccJRb7/A0M6JBnwg= | ||||
| github.com/Azure/go-ansiterm v0.0.0-20170929234023-d6e3b3328b78/go.mod h1:LmzpDX56iTiv29bbRTIsUNlaFfuhWRQBWjQdVyAevI8= | ||||
| github.com/Azure/go-autorest v14.2.0+incompatible/go.mod h1:r+4oMnoxhatjLLJ6zxSWATqVooLgysK6ZNox3g/xq24= | ||||
| github.com/Azure/go-autorest/autorest v0.11.1/go.mod h1:JFgpikqFJ/MleTTxwepExTKnFUKKszPS8UavbQYUMuw= | ||||
|   | ||||
							
								
								
									
										201
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										201
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,201 @@ | ||||
|                                  Apache License | ||||
|                            Version 2.0, January 2004 | ||||
|                         http://www.apache.org/licenses/ | ||||
|  | ||||
|    TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION | ||||
|  | ||||
|    1. Definitions. | ||||
|  | ||||
|       "License" shall mean the terms and conditions for use, reproduction, | ||||
|       and distribution as defined by Sections 1 through 9 of this document. | ||||
|  | ||||
|       "Licensor" shall mean the copyright owner or entity authorized by | ||||
|       the copyright owner that is granting the License. | ||||
|  | ||||
|       "Legal Entity" shall mean the union of the acting entity and all | ||||
|       other entities that control, are controlled by, or are under common | ||||
|       control with that entity. For the purposes of this definition, | ||||
|       "control" means (i) the power, direct or indirect, to cause the | ||||
|       direction or management of such entity, whether by contract or | ||||
|       otherwise, or (ii) ownership of fifty percent (50%) or more of the | ||||
|       outstanding shares, or (iii) beneficial ownership of such entity. | ||||
|  | ||||
|       "You" (or "Your") shall mean an individual or Legal Entity | ||||
|       exercising permissions granted by this License. | ||||
|  | ||||
|       "Source" form shall mean the preferred form for making modifications, | ||||
|       including but not limited to software source code, documentation | ||||
|       source, and configuration files. | ||||
|  | ||||
|       "Object" form shall mean any form resulting from mechanical | ||||
|       transformation or translation of a Source form, including but | ||||
|       not limited to compiled object code, generated documentation, | ||||
|       and conversions to other media types. | ||||
|  | ||||
|       "Work" shall mean the work of authorship, whether in Source or | ||||
|       Object form, made available under the License, as indicated by a | ||||
|       copyright notice that is included in or attached to the work | ||||
|       (an example is provided in the Appendix below). | ||||
|  | ||||
|       "Derivative Works" shall mean any work, whether in Source or Object | ||||
|       form, that is based on (or derived from) the Work and for which the | ||||
|       editorial revisions, annotations, elaborations, or other modifications | ||||
|       represent, as a whole, an original work of authorship. For the purposes | ||||
|       of this License, Derivative Works shall not include works that remain | ||||
|       separable from, or merely link (or bind by name) to the interfaces of, | ||||
|       the Work and Derivative Works thereof. | ||||
|  | ||||
|       "Contribution" shall mean any work of authorship, including | ||||
|       the original version of the Work and any modifications or additions | ||||
|       to that Work or Derivative Works thereof, that is intentionally | ||||
|       submitted to Licensor for inclusion in the Work by the copyright owner | ||||
|       or by an individual or Legal Entity authorized to submit on behalf of | ||||
|       the copyright owner. For the purposes of this definition, "submitted" | ||||
|       means any form of electronic, verbal, or written communication sent | ||||
|       to the Licensor or its representatives, including but not limited to | ||||
|       communication on electronic mailing lists, source code control systems, | ||||
|       and issue tracking systems that are managed by, or on behalf of, the | ||||
|       Licensor for the purpose of discussing and improving the Work, but | ||||
|       excluding communication that is conspicuously marked or otherwise | ||||
|       designated in writing by the copyright owner as "Not a Contribution." | ||||
|  | ||||
|       "Contributor" shall mean Licensor and any individual or Legal Entity | ||||
|       on behalf of whom a Contribution has been received by Licensor and | ||||
|       subsequently incorporated within the Work. | ||||
|  | ||||
|    2. Grant of Copyright License. Subject to the terms and conditions of | ||||
|       this License, each Contributor hereby grants to You a perpetual, | ||||
|       worldwide, non-exclusive, no-charge, royalty-free, irrevocable | ||||
|       copyright license to reproduce, prepare Derivative Works of, | ||||
|       publicly display, publicly perform, sublicense, and distribute the | ||||
|       Work and such Derivative Works in Source or Object form. | ||||
|  | ||||
|    3. Grant of Patent License. Subject to the terms and conditions of | ||||
|       this License, each Contributor hereby grants to You a perpetual, | ||||
|       worldwide, non-exclusive, no-charge, royalty-free, irrevocable | ||||
|       (except as stated in this section) patent license to make, have made, | ||||
|       use, offer to sell, sell, import, and otherwise transfer the Work, | ||||
|       where such license applies only to those patent claims licensable | ||||
|       by such Contributor that are necessarily infringed by their | ||||
|       Contribution(s) alone or by combination of their Contribution(s) | ||||
|       with the Work to which such Contribution(s) was submitted. If You | ||||
|       institute patent litigation against any entity (including a | ||||
|       cross-claim or counterclaim in a lawsuit) alleging that the Work | ||||
|       or a Contribution incorporated within the Work constitutes direct | ||||
|       or contributory patent infringement, then any patent licenses | ||||
|       granted to You under this License for that Work shall terminate | ||||
|       as of the date such litigation is filed. | ||||
|  | ||||
|    4. Redistribution. You may reproduce and distribute copies of the | ||||
|       Work or Derivative Works thereof in any medium, with or without | ||||
|       modifications, and in Source or Object form, provided that You | ||||
|       meet the following conditions: | ||||
|  | ||||
|       (a) You must give any other recipients of the Work or | ||||
|           Derivative Works a copy of this License; and | ||||
|  | ||||
|       (b) You must cause any modified files to carry prominent notices | ||||
|           stating that You changed the files; and | ||||
|  | ||||
|       (c) You must retain, in the Source form of any Derivative Works | ||||
|           that You distribute, all copyright, patent, trademark, and | ||||
|           attribution notices from the Source form of the Work, | ||||
|           excluding those notices that do not pertain to any part of | ||||
|           the Derivative Works; and | ||||
|  | ||||
|       (d) If the Work includes a "NOTICE" text file as part of its | ||||
|           distribution, then any Derivative Works that You distribute must | ||||
|           include a readable copy of the attribution notices contained | ||||
|           within such NOTICE file, excluding those notices that do not | ||||
|           pertain to any part of the Derivative Works, in at least one | ||||
|           of the following places: within a NOTICE text file distributed | ||||
|           as part of the Derivative Works; within the Source form or | ||||
|           documentation, if provided along with the Derivative Works; or, | ||||
|           within a display generated by the Derivative Works, if and | ||||
|           wherever such third-party notices normally appear. The contents | ||||
|           of the NOTICE file are for informational purposes only and | ||||
|           do not modify the License. You may add Your own attribution | ||||
|           notices within Derivative Works that You distribute, alongside | ||||
|           or as an addendum to the NOTICE text from the Work, provided | ||||
|           that such additional attribution notices cannot be construed | ||||
|           as modifying the License. | ||||
|  | ||||
|       You may add Your own copyright statement to Your modifications and | ||||
|       may provide additional or different license terms and conditions | ||||
|       for use, reproduction, or distribution of Your modifications, or | ||||
|       for any such Derivative Works as a whole, provided Your use, | ||||
|       reproduction, and distribution of the Work otherwise complies with | ||||
|       the conditions stated in this License. | ||||
|  | ||||
|    5. Submission of Contributions. Unless You explicitly state otherwise, | ||||
|       any Contribution intentionally submitted for inclusion in the Work | ||||
|       by You to the Licensor shall be under the terms and conditions of | ||||
|       this License, without any additional terms or conditions. | ||||
|       Notwithstanding the above, nothing herein shall supersede or modify | ||||
|       the terms of any separate license agreement you may have executed | ||||
|       with Licensor regarding such Contributions. | ||||
|  | ||||
|    6. Trademarks. This License does not grant permission to use the trade | ||||
|       names, trademarks, service marks, or product names of the Licensor, | ||||
|       except as required for reasonable and customary use in describing the | ||||
|       origin of the Work and reproducing the content of the NOTICE file. | ||||
|  | ||||
|    7. Disclaimer of Warranty. Unless required by applicable law or | ||||
|       agreed to in writing, Licensor provides the Work (and each | ||||
|       Contributor provides its Contributions) on an "AS IS" BASIS, | ||||
|       WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or | ||||
|       implied, including, without limitation, any warranties or conditions | ||||
|       of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A | ||||
|       PARTICULAR PURPOSE. You are solely responsible for determining the | ||||
|       appropriateness of using or redistributing the Work and assume any | ||||
|       risks associated with Your exercise of permissions under this License. | ||||
|  | ||||
|    8. Limitation of Liability. In no event and under no legal theory, | ||||
|       whether in tort (including negligence), contract, or otherwise, | ||||
|       unless required by applicable law (such as deliberate and grossly | ||||
|       negligent acts) or agreed to in writing, shall any Contributor be | ||||
|       liable to You for damages, including any direct, indirect, special, | ||||
|       incidental, or consequential damages of any character arising as a | ||||
|       result of this License or out of the use or inability to use the | ||||
|       Work (including but not limited to damages for loss of goodwill, | ||||
|       work stoppage, computer failure or malfunction, or any and all | ||||
|       other commercial damages or losses), even if such Contributor | ||||
|       has been advised of the possibility of such damages. | ||||
|  | ||||
|    9. Accepting Warranty or Additional Liability. While redistributing | ||||
|       the Work or Derivative Works thereof, You may choose to offer, | ||||
|       and charge a fee for, acceptance of support, warranty, indemnity, | ||||
|       or other liability obligations and/or rights consistent with this | ||||
|       License. However, in accepting such obligations, You may act only | ||||
|       on Your own behalf and on Your sole responsibility, not on behalf | ||||
|       of any other Contributor, and only if You agree to indemnify, | ||||
|       defend, and hold each Contributor harmless for any liability | ||||
|       incurred by, or claims asserted against, such Contributor by reason | ||||
|       of your accepting any such warranty or additional liability. | ||||
|  | ||||
|    END OF TERMS AND CONDITIONS | ||||
|  | ||||
|    APPENDIX: How to apply the Apache License to your work. | ||||
|  | ||||
|       To apply the Apache License to your work, attach the following | ||||
|       boilerplate notice, with the fields enclosed by brackets "[]" | ||||
|       replaced with your own identifying information. (Don't include | ||||
|       the brackets!)  The text should be enclosed in the appropriate | ||||
|       comment syntax for the file format. We also recommend that a | ||||
|       file or class name and description of purpose be included on the | ||||
|       same "printed page" as the copyright notice for easier | ||||
|       identification within third-party archives. | ||||
|  | ||||
|    Copyright [yyyy] [name of copyright owner] | ||||
|  | ||||
|    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. | ||||
							
								
								
									
										49
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/README.md
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/README.md
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,49 @@ | ||||
| # go-fuzz-headers | ||||
| This repository contains various helper functions to be used with [go-fuzz](https://github.com/dvyukov/go-fuzz). | ||||
|  | ||||
| ## Goal | ||||
|  | ||||
| The current goal of go-fuzz-headers is: | ||||
| To maintain a series of helper utilities that can be used for golang projects that are integrated into OSS-fuzz and use the go-fuzz engine to fuzz more complicated types than merely strings and data arrays.  | ||||
| While go-fuzz-headers can be used when using go-fuzz outside of OSS-fuzz, we do not test such usage and cannot confirm that it is supported. | ||||
|  | ||||
| ## Status | ||||
|  | ||||
| The project is under development and will be updated regularly. | ||||
|  | ||||
| Fuzzers that use `GenerateStruct` will not require modifications as more types get supported. | ||||
|  | ||||
| ## Usage | ||||
| To make use of the helper functions, a ConsumeFuzzer has to be instantiated: | ||||
| ```go | ||||
| f := NewConsumer(data) | ||||
| ``` | ||||
| To split the input data from the fuzzer into a random set of equally large chunks: | ||||
| ```go | ||||
| err := f.Split(3, 6) | ||||
| ``` | ||||
| ...after which the consumer has the following available attributes: | ||||
| ```go | ||||
| f.CommandPart = commandPart | ||||
| f.RestOfArray = restOfArray | ||||
| f.NumberOfCalls = numberOfCalls | ||||
| ``` | ||||
| To pass the input data from the fuzzer into a struct: | ||||
| ```go | ||||
| ts := new(target_struct) | ||||
| err :=f.GenerateStruct(ts) | ||||
| ``` | ||||
| or: | ||||
| ```go | ||||
| ts := target_struct{} | ||||
| err = f.GenerateStruct(&ts) | ||||
| ``` | ||||
| `GenerateStruct` will pass data from the input data to the targeted struct. Currently the following field types are supported: | ||||
| 1. `string` | ||||
| 2. `bool` | ||||
| 3. `int` | ||||
| 4. `[]string` | ||||
| 5. `byte` | ||||
| 5. `[]byte` | ||||
| 6. custom structures | ||||
| 7. `map` | ||||
							
								
								
									
										339
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/consumer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										339
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/consumer.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,339 @@ | ||||
| package gofuzzheaders | ||||
|  | ||||
| import ( | ||||
| 	"archive/tar" | ||||
| 	"bytes" | ||||
| 	"errors" | ||||
| 	//"fmt" | ||||
| 	"reflect" | ||||
| 	"unsafe" | ||||
| ) | ||||
|  | ||||
| type ConsumeFuzzer struct { | ||||
| 	data                 []byte | ||||
| 	CommandPart          []byte | ||||
| 	RestOfArray          []byte | ||||
| 	NumberOfCalls        int | ||||
| 	position             int | ||||
| 	fuzzUnexportedFields bool | ||||
| } | ||||
|  | ||||
| func IsDivisibleBy(n int, divisibleby int) bool { | ||||
| 	return (n % divisibleby) == 0 | ||||
| } | ||||
|  | ||||
| func NewConsumer(fuzzData []byte) *ConsumeFuzzer { | ||||
| 	f := &ConsumeFuzzer{data: fuzzData, position: 0} | ||||
| 	return f | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) Split(minCalls, maxCalls int) error { | ||||
| 	if len(f.data) == 0 { | ||||
| 		return errors.New("Could not split") | ||||
| 	} | ||||
| 	numberOfCalls := int(f.data[0]) | ||||
| 	if numberOfCalls < minCalls || numberOfCalls > maxCalls { | ||||
| 		return errors.New("Bad number of calls") | ||||
|  | ||||
| 	} | ||||
| 	if len(f.data) < numberOfCalls+numberOfCalls+1 { | ||||
| 		return errors.New("Length of data does not match required parameters") | ||||
| 	} | ||||
|  | ||||
| 	// Define part 2 and 3 of the data array | ||||
| 	commandPart := f.data[1 : numberOfCalls+1] | ||||
| 	restOfArray := f.data[numberOfCalls+1:] | ||||
|  | ||||
| 	// Just a small check. It is necessary | ||||
| 	if len(commandPart) != numberOfCalls { | ||||
| 		return errors.New("Length of commandPart does not match number of calls") | ||||
| 	} | ||||
|  | ||||
| 	// Check if restOfArray is divisible by numberOfCalls | ||||
| 	if !IsDivisibleBy(len(restOfArray), numberOfCalls) { | ||||
| 		return errors.New("Length of commandPart does not match number of calls") | ||||
| 	} | ||||
| 	f.CommandPart = commandPart | ||||
| 	f.RestOfArray = restOfArray | ||||
| 	f.NumberOfCalls = numberOfCalls | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) AllowUnexportedFields() { | ||||
| 	f.fuzzUnexportedFields = true | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) DisallowUnexportedFields() { | ||||
| 	f.fuzzUnexportedFields = false | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GenerateStruct(targetStruct interface{}) error { | ||||
| 	v := reflect.ValueOf(targetStruct) | ||||
| 	/*if !v.CanSet() { | ||||
| 		return errors.New("This interface cannot be set") | ||||
| 	}*/ | ||||
| 	e := v.Elem() | ||||
| 	err := f.fuzzStruct(e) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) fuzzStruct(e reflect.Value) error { | ||||
| 	switch e.Kind() { | ||||
| 	case reflect.Struct: | ||||
| 		for i := 0; i < e.NumField(); i++ { | ||||
| 			// Useful for debugging, so we leave it for now: | ||||
| 			//vt := e.Type().Field(i).Name | ||||
| 			//fmt.Println("vt:::::::::::::::::::::: ", vt) | ||||
| 			var v reflect.Value | ||||
| 			if !e.Field(i).CanSet() { | ||||
| 				if f.fuzzUnexportedFields { | ||||
| 					v = reflect.NewAt(e.Field(i).Type(), unsafe.Pointer(e.Field(i).UnsafeAddr())).Elem() | ||||
| 				} | ||||
| 			} else { | ||||
| 				v = e.Field(i) | ||||
| 			} | ||||
| 			err := f.fuzzStruct(v) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
| 	case reflect.String: | ||||
| 		str, err := f.GetString() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		if e.CanSet() { | ||||
| 			e.SetString(str) | ||||
| 		} | ||||
| 	case reflect.Slice: | ||||
| 		maxElements := 50 | ||||
| 		randQty, err := f.GetInt() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		numOfElements := randQty % maxElements | ||||
|  | ||||
| 		uu := reflect.MakeSlice(e.Type(), numOfElements, numOfElements) | ||||
|  | ||||
| 		for i := 0; i < numOfElements; i++ { | ||||
| 			err := f.fuzzStruct(uu.Index(i)) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
| 		if e.CanSet() { | ||||
| 			e.Set(uu) | ||||
| 		} | ||||
| 	case reflect.Uint64: | ||||
| 		newInt, err := f.GetInt() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		if e.CanSet() { | ||||
| 			e.SetUint(uint64(newInt)) | ||||
| 		} | ||||
| 	case reflect.Int: | ||||
| 		newInt, err := f.GetInt() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		if e.CanSet() { | ||||
| 			e.SetInt(int64(newInt)) | ||||
| 		} | ||||
| 	case reflect.Map: | ||||
| 		if e.CanSet() { | ||||
| 			e.Set(reflect.MakeMap(e.Type())) | ||||
| 			maxElements := 50 | ||||
| 			randQty, err := f.GetInt() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			numOfElements := randQty % maxElements | ||||
| 			for i := 0; i < numOfElements; i++ { | ||||
| 				key := reflect.New(e.Type().Key()).Elem() | ||||
| 				err := f.fuzzStruct(key) | ||||
| 				if err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 				val := reflect.New(e.Type().Elem()).Elem() | ||||
| 				err = f.fuzzStruct(val) | ||||
| 				if err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 				e.SetMapIndex(key, val) | ||||
| 			} | ||||
| 		} | ||||
| 	case reflect.Ptr: | ||||
| 		if e.CanSet() { | ||||
| 			e.Set(reflect.New(e.Type().Elem())) | ||||
| 			err := f.fuzzStruct(e.Elem()) | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			return nil | ||||
| 		} | ||||
| 	case reflect.Uint8: | ||||
| 		b, err := f.GetByte() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		if e.CanSet() { | ||||
| 			e.SetUint(uint64(b)) | ||||
| 		} | ||||
| 	default: | ||||
| 		return nil | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetStringArray() (reflect.Value, error) { | ||||
| 	// The max size of the array: | ||||
| 	max := 20 | ||||
|  | ||||
| 	arraySize := f.position | ||||
| 	if arraySize > max { | ||||
| 		arraySize = max | ||||
| 	} | ||||
| 	elemType := reflect.TypeOf("string") | ||||
| 	stringArray := reflect.MakeSlice(reflect.SliceOf(elemType), arraySize, arraySize) | ||||
| 	if f.position+arraySize >= len(f.data) { | ||||
| 		return stringArray, errors.New("Could not make string array") | ||||
| 	} | ||||
|  | ||||
| 	for i := 0; i < arraySize; i++ { | ||||
| 		stringSize := int(f.data[f.position]) | ||||
|  | ||||
| 		if f.position+stringSize >= len(f.data) { | ||||
| 			return stringArray, nil | ||||
| 		} | ||||
| 		stringToAppend := string(f.data[f.position : f.position+stringSize]) | ||||
| 		strVal := reflect.ValueOf(stringToAppend) | ||||
| 		stringArray = reflect.Append(stringArray, strVal) | ||||
| 		f.position = f.position + stringSize | ||||
| 	} | ||||
| 	return stringArray, nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetInt() (int, error) { | ||||
| 	if f.position >= len(f.data) { | ||||
| 		return 0, errors.New("Not enough bytes to create int") | ||||
| 	} | ||||
| 	returnInt := int(f.data[f.position]) | ||||
| 	f.position++ | ||||
| 	return returnInt, nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetByte() (byte, error) { | ||||
| 	if len(f.data) == 0 { | ||||
| 		return 0x00, errors.New("Not enough bytes to get byte") | ||||
| 	} | ||||
| 	if f.position >= len(f.data) { | ||||
| 		return 0x00, errors.New("Not enough bytes to get byte") | ||||
| 	} | ||||
| 	returnByte := f.data[f.position] | ||||
| 	f.position += 1 | ||||
| 	return returnByte, nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetBytes() ([]byte, error) { | ||||
| 	if len(f.data) == 0 || f.position >= len(f.data) { | ||||
| 		return nil, errors.New("Not enough bytes to create byte array") | ||||
| 	} | ||||
| 	length := int(f.data[f.position]) | ||||
| 	byteBegin := f.position + 1 | ||||
| 	if byteBegin >= len(f.data) { | ||||
| 		return nil, errors.New("Not enough bytes to create byte array") | ||||
| 	} | ||||
| 	if length == 0 { | ||||
| 		return nil, errors.New("Zero-length is not supported") | ||||
| 	} | ||||
| 	if byteBegin+length >= len(f.data) { | ||||
| 		return nil, errors.New("Not enough bytes to create byte array") | ||||
| 	} | ||||
| 	b := f.data[byteBegin : byteBegin+length] | ||||
| 	f.position = byteBegin + length | ||||
| 	return b, nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetString() (string, error) { | ||||
| 	if f.position >= len(f.data) { | ||||
| 		return "nil", errors.New("Not enough bytes to create string") | ||||
| 	} | ||||
| 	length := int(f.data[f.position]) | ||||
| 	byteBegin := f.position + 1 | ||||
| 	if byteBegin >= len(f.data) { | ||||
| 		return "nil", errors.New("Not enough bytes to create string") | ||||
| 	} | ||||
| 	if byteBegin+length > len(f.data) { | ||||
| 		return "nil", errors.New("Not enough bytes to create string") | ||||
| 	} | ||||
| 	str := string(f.data[byteBegin : byteBegin+length]) | ||||
| 	f.position = byteBegin + length | ||||
| 	return str, nil | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) GetBool() (bool, error) { | ||||
| 	if f.position >= len(f.data) { | ||||
| 		return false, errors.New("Not enough bytes to create bool") | ||||
| 	} | ||||
| 	if IsDivisibleBy(int(f.data[f.position]), 2) { | ||||
| 		f.position++ | ||||
| 		return true, nil | ||||
| 	} else { | ||||
| 		f.position++ | ||||
| 		return false, nil | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (f *ConsumeFuzzer) FuzzMap(m interface{}) error { | ||||
| 	err := f.GenerateStruct(m) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // TarBytes returns valid tar bytes for a tar archive | ||||
| func (f *ConsumeFuzzer) TarBytes() ([]byte, error) { | ||||
| 	var buf bytes.Buffer | ||||
| 	tw := tar.NewWriter(&buf) | ||||
|  | ||||
| 	numberOfFiles, err := f.GetInt() | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	maxNoOfFiles := 100000 | ||||
| 	for i := 0; i < numberOfFiles%maxNoOfFiles; i++ { | ||||
| 		filename, err := f.GetString() | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		filebody, err := f.GetBytes() | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		hdr := &tar.Header{} | ||||
| 		err = f.GenerateStruct(hdr) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		hdr.Name = filename | ||||
| 		hdr.Size = int64(len(filebody)) | ||||
| 		hdr.Mode = 0600 | ||||
|  | ||||
| 		if err := tw.WriteHeader(hdr); err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		if _, err := tw.Write(filebody); err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 	} | ||||
| 	if err := tw.Close(); err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	//fmt.Println(string(buf.Bytes())) | ||||
| 	return buf.Bytes(), nil | ||||
| } | ||||
							
								
								
									
										3
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								vendor/github.com/AdaLogics/go-fuzz-headers/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,3 @@ | ||||
| module github.com/AdaLogics/go-fuzz-headers | ||||
|  | ||||
| go 1.13 | ||||
							
								
								
									
										3
									
								
								vendor/modules.txt
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								vendor/modules.txt
									
									
									
									
										vendored
									
									
								
							| @@ -1,3 +1,6 @@ | ||||
| # github.com/AdaLogics/go-fuzz-headers v0.0.0-20210715213245-6c3934b029d8 | ||||
| ## explicit | ||||
| github.com/AdaLogics/go-fuzz-headers | ||||
| # github.com/Microsoft/go-winio v0.5.0 | ||||
| ## explicit | ||||
| github.com/Microsoft/go-winio | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 AdamKorcz
					AdamKorcz