diff options
| -rw-r--r-- | tests/unit/unit1604.c | 33 |
1 files changed, 19 insertions, 14 deletions
diff --git a/tests/unit/unit1604.c b/tests/unit/unit1604.c index a7c2a2543..3c0c5d121 100644 --- a/tests/unit/unit1604.c +++ b/tests/unit/unit1604.c @@ -45,26 +45,28 @@ static void unit_stop(void) static char *getflagstr(int flags) { char *buf = malloc(256); - abort_unless(buf, "out of memory"); - snprintf(buf, 256, "%s,%s,%s,%s", - ((flags & SANITIZE_ALLOW_COLONS) ? "SANITIZE_ALLOW_COLONS" : ""), - ((flags & SANITIZE_ALLOW_PATH) ? "SANITIZE_ALLOW_PATH" : ""), - ((flags & SANITIZE_ALLOW_RESERVED) ? "SANITIZE_ALLOW_RESERVED" : ""), - ((flags & SANITIZE_ALLOW_TRUNCATE) ? "SANITIZE_ALLOW_TRUNCATE" : "")); + if(buf) { + snprintf(buf, 256, "%s,%s,%s,%s", + ((flags & SANITIZE_ALLOW_COLONS) ? "SANITIZE_ALLOW_COLONS" : ""), + ((flags & SANITIZE_ALLOW_PATH) ? "SANITIZE_ALLOW_PATH" : ""), + ((flags & SANITIZE_ALLOW_RESERVED) ? "SANITIZE_ALLOW_RESERVED" : ""), + ((flags & SANITIZE_ALLOW_TRUNCATE) ? "SANITIZE_ALLOW_TRUNCATE" : "")); + } return buf; } static char *getcurlcodestr(int cc) { char *buf = malloc(256); - abort_unless(buf, "out of memory"); - snprintf(buf, 256, "%s (%d)", - (cc == SANITIZE_ERR_OK ? "SANITIZE_ERR_OK" : - cc == SANITIZE_ERR_BAD_ARGUMENT ? "SANITIZE_ERR_BAD_ARGUMENT" : - cc == SANITIZE_ERR_INVALID_PATH ? "SANITIZE_ERR_INVALID_PATH" : - cc == SANITIZE_ERR_OUT_OF_MEMORY ? "SANITIZE_ERR_OUT_OF_MEMORY" : - "unexpected error code - add name"), - cc); + if(buf) { + snprintf(buf, 256, "%s (%d)", + (cc == SANITIZE_ERR_OK ? "SANITIZE_ERR_OK" : + cc == SANITIZE_ERR_BAD_ARGUMENT ? "SANITIZE_ERR_BAD_ARGUMENT" : + cc == SANITIZE_ERR_INVALID_PATH ? "SANITIZE_ERR_INVALID_PATH" : + cc == SANITIZE_ERR_OUT_OF_MEMORY ? "SANITIZE_ERR_OUT_OF_MEMORY" : + "unexpected error code - add name"), + cc); + } return buf; } @@ -310,8 +312,11 @@ UNITTEST_START } flagstr = getflagstr(data[i].flags); + abort_unless(flagstr, "out of memory"); received_ccstr = getcurlcodestr(res); + abort_unless(received_ccstr, "out of memory"); expected_ccstr = getcurlcodestr(data[i].expected_result); + abort_unless(expected_ccstr, "out of memory"); unitfail++; fprintf(stderr, "\n" |