diff options
Diffstat (limited to 'src/platform/windows')
| -rw-r--r-- | src/platform/windows/win_file.c | 21 | ||||
| -rw-r--r-- | src/platform/windows/win_thread.c | 4 |
2 files changed, 15 insertions, 10 deletions
diff --git a/src/platform/windows/win_file.c b/src/platform/windows/win_file.c index 3b6461a6..47bb87af 100644 --- a/src/platform/windows/win_file.c +++ b/src/platform/windows/win_file.c @@ -136,14 +136,19 @@ nni_plat_file_get(const char *name, void **datap, size_t *lenp) rv = nni_win_error(GetLastError()); goto done; } - if ((data = nni_alloc((size_t) sz)) == NULL) { - rv = NNG_ENOMEM; - goto done; - } - if (!ReadFile(h, data, sz, &nread, NULL)) { - rv = nni_win_error(GetLastError()); - nni_free(data, sz); - goto done; + if (sz > 0) { + if ((data = nni_alloc((size_t) sz)) == NULL) { + rv = NNG_ENOMEM; + goto done; + } + if (!ReadFile(h, data, sz, &nread, NULL)) { + rv = nni_win_error(GetLastError()); + nni_free(data, sz); + goto done; + } + } else { + data = NULL; + nread = 0; } // We can get a short read, indicating end of file. We return diff --git a/src/platform/windows/win_thread.c b/src/platform/windows/win_thread.c index d6363207..3b4e63bc 100644 --- a/src/platform/windows/win_thread.c +++ b/src/platform/windows/win_thread.c @@ -19,13 +19,13 @@ void * nni_alloc(size_t sz) { - return (malloc(sz)); + return (sz > 0 ? malloc(sz) : NULL); } void * nni_zalloc(size_t sz) { - return (calloc(1, sz)); + return (sz > 0 ? calloc(1, sz) : NULL); } void |
