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

Merge pull request #40526 from grasskin:patch-1 

PiperOrigin-RevId: 316909136
Change-Id: Ic626efedf856449575e5f730119bbbf3c95a054a
This commit is contained in:
TensorFlower Gardener 2020-06-17 10:07:27 -07:00
parent 406d9b5521 e3a423a949
commit 2779d9e29d
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include <fuzzer/FuzzedDataProvider.h>
#include <cstdint> #include <cstdint>
#include <cstdlib> #include <cstdlib>
@ -26,10 +28,7 @@ limitations under the License.
namespace { namespace {
tensorflow::error::Code BuildRandomErrorCode(uint8_t a, uint8_t b, uint8_t c, tensorflow::error::Code BuildRandomErrorCode(uint32_t code) {
uint8_t d) {
int code = (a << 24) | (b << 16) | (c << 8) | d;
// We cannot build a `Status` with error_code of 0 and a message, so force // We cannot build a `Status` with error_code of 0 and a message, so force
// error code to be non-zero. // error code to be non-zero.
if (code == 0) { if (code == 0) {
@ -39,22 +38,16 @@ tensorflow::error::Code BuildRandomErrorCode(uint8_t a, uint8_t b, uint8_t c,
return static_cast<tensorflow::error::Code>(code); return static_cast<tensorflow::error::Code>(code);
} }
std::string GetRandomErrorString(const uint8_t *data, size_t size) {
const char *p = reinterpret_cast<const char *>(data);
return std::string(p, size);
}
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
// TODO(mihaimaruseac): Use `FuzzedDataProvider` and then make these `const`
tensorflow::error::Code error_code; tensorflow::error::Code error_code;
std::string error_message; std::string error_message;
if (size < 4) {
error_code = BuildRandomErrorCode(0, 0, 0, 0); FuzzedDataProvider fuzzed_data(data, size);
error_message = GetRandomErrorString(data, size);
} else { uint32_t code = fuzzed_data.ConsumeIntegral<uint32_t>();
error_code = BuildRandomErrorCode(data[0], data[1], data[2], data[3]); error_code = BuildRandomErrorCode(code);
error_message = GetRandomErrorString(data + 4, size - 4);
} error_message = fuzzed_data.ConsumeRemainingBytesAsString();
tensorflow::Status s = tensorflow::Status(error_code, error_message); tensorflow::Status s = tensorflow::Status(error_code, error_message);
const std::string actual_message = s.ToString(); const std::string actual_message = s.ToString();