try repeating write

This commit is contained in:
vanhauser-thc
2022-04-22 10:28:14 +02:00
parent 5798c686b2
commit 7f26d13345
4 changed files with 36 additions and 17 deletions

View File

@ -145,7 +145,7 @@ else
CFLAGS ?= -O2 $(CFLAGS_OPT) # -funroll-loops is slower on modern compilers
endif
override CFLAGS += -g -Wno-pointer-sign -Wno-variadic-macros -Wall -Wextra -Wpointer-arith \
override CFLAGS += -g -Wno-pointer-sign -Wno-variadic-macros -Wall -Wextra -Wno-pointer-arith \
-fPIC -I include/ -DAFL_PATH=\"$(HELPER_PATH)\" \
-DBIN_PATH=\"$(BIN_PATH)\" -DDOC_PATH=\"$(DOC_PATH)\"
# -fstack-protector

View File

@ -358,17 +358,36 @@ static inline const char *colorfilter(const char *x) {
#define ck_write(fd, buf, len, fn) \
do { \
\
if (len <= 0) break; \
int _fd = (fd); \
s32 _written = 0, _off = 0, _len = (s32)(len); \
\
s32 _len = (s32)(len); \
s32 _res = write(_fd, (buf), _len); \
if (_res != _len) { \
do { \
\
s32 _res = write(_fd, (buf) + _off, _len); \
if (_res != _len && (_res > 0 && _written + _res != _len)) { \
\
if (_res > 0) { \
\
_written += _res; \
_len -= _res; \
_off += _res; \
\
} else { \
\
RPFATAL(_res, "Short write to %s, fd %d (%d of %d bytes)", fn, _fd, \
_res, _len); \
\
} \
\
} else { \
\
break; \
\
} \
\
} while (1); \
\
} while (0)
#define ck_read(fd, buf, len, fn) \