experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update spacing and variable declaration

Browse Source
This commit is contained in:
Gabriel Rasskin 2020-06-18 15:31:26 -07:00 committed by GitHub
parent 852cde437f
commit 14e942faaa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tensorflow/security/fuzzing/status_group_fuzz.cc
View File

@ -14,9 +14,7 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include <cstdint> #include <cstdint>
#include <cstdlib> #include <cstdlib>
#include "tensorflow/core/platform/status.h" #include "tensorflow/core/platform/status.h"
#include <fuzzer/FuzzedDataProvider.h> #include <fuzzer/FuzzedDataProvider.h>
// This is a fuzzer for `tensorflow::StatusGroup`. Since `Status` is used almost // This is a fuzzer for `tensorflow::StatusGroup`. Since `Status` is used almost
@ -37,37 +35,22 @@ tensorflow::error::Code BuildRandomErrorCode(uint32_t code){
} }
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
tensorflow::error::Code error_code;
const std::string error_message = "ERROR"; const std::string error_message = "ERROR";
tensorflow::Status s, derived_s;
tensorflow::StatusGroup sg; tensorflow::StatusGroup sg;
bool is_derived;
uint32_t code;
FuzzedDataProvider fuzzed_data(data, size); FuzzedDataProvider fuzzed_data(data, size);
while(fuzzed_data.remaining_bytes() > 0) { while(fuzzed_data.remaining_bytes() > 0) {
code = fuzzed_data.ConsumeIntegral<uint32_t>(); uint32_t code = fuzzed_data.ConsumeIntegral<uint32_t>();
tensorflow::error::Code error_code = BuildRandomErrorCode(code);
bool is_derived = fuzzed_data.ConsumeBool();
error_code = BuildRandomErrorCode(code); tensorflow::Status s = tensorflow::Status(error_code, error_message);
is_derived = fuzzed_data.ConsumeBool();
s = tensorflow::Status(error_code, error_message);
if(is_derived) { if(is_derived) {
derived_s = tensorflow::StatusGroup::MakeDerived(s); tensorflow::Status derived_s = tensorflow::StatusGroup::MakeDerived(s);
sg.Update(derived_s); sg.Update(derived_s);
} else { } else {
sg.Update(s); sg.Update(s);
} }
} }