diff options
Diffstat (limited to 'libwaitress')
| -rw-r--r-- | libwaitress/src/waitress.c | 14 | ||||
| -rw-r--r-- | libwaitress/src/waitress.h | 4 |
2 files changed, 11 insertions, 7 deletions
diff --git a/libwaitress/src/waitress.c b/libwaitress/src/waitress.c index 70a1547..72f5f0f 100644 --- a/libwaitress/src/waitress.c +++ b/libwaitress/src/waitress.c @@ -180,7 +180,7 @@ inline void WaitressSetHPP (WaitressHandle_t *waith, const char *host, * @param data size * @param buffer structure */ -static char WaitressFetchBufCb (void *recvData, size_t recvDataSize, +static WaitressCbReturn_t WaitressFetchBufCb (void *recvData, size_t recvDataSize, void *extraData) { char *recvBytes = recvData; WaitressFetchBufCbBuffer_t *buffer = extraData; @@ -188,7 +188,7 @@ static char WaitressFetchBufCb (void *recvData, size_t recvDataSize, if (buffer->buf == NULL) { if ((buffer->buf = malloc (sizeof (*buffer->buf) * (recvDataSize + 1))) == NULL) { - return 0; + return WAITRESS_CB_RET_ERR; } } else { char *newbuf; @@ -196,7 +196,7 @@ static char WaitressFetchBufCb (void *recvData, size_t recvDataSize, sizeof (*buffer->buf) * (buffer->pos + recvDataSize + 1))) == NULL) { free (buffer->buf); - return 0; + return WAITRESS_CB_RET_ERR; } buffer->buf = newbuf; } @@ -204,7 +204,7 @@ static char WaitressFetchBufCb (void *recvData, size_t recvDataSize, buffer->pos += recvDataSize; *(buffer->buf+buffer->pos) = '\0'; - return 1; + return WAITRESS_CB_RET_OK; } /* Fetch string. Beware! This overwrites your waith->data pointer @@ -457,7 +457,8 @@ WaitressReturn_t WaitressFetchCall (WaitressHandle_t *waith) { /* push remaining bytes */ if (bufFilled > 0) { waith->contentReceived += bufFilled; - if (!waith->callback (thisLine, bufFilled, waith->data)) { + if (waith->callback (thisLine, bufFilled, waith->data) == + WAITRESS_CB_RET_ERR) { CLOSE_RET (WAITRESS_RET_CB_ABORT); } } @@ -467,7 +468,8 @@ WaitressReturn_t WaitressFetchCall (WaitressHandle_t *waith) { READ_RET (recvBuf, sizeof (recvBuf), &recvSize); if (recvSize > 0) { waith->contentReceived += recvSize; - if (!waith->callback (recvBuf, recvSize, waith->data)) { + if (waith->callback (recvBuf, recvSize, waith->data) == + WAITRESS_CB_RET_ERR) { wRet = WAITRESS_RET_CB_ABORT; break; } diff --git a/libwaitress/src/waitress.h b/libwaitress/src/waitress.h index 5fa1874..ed5679f 100644 --- a/libwaitress/src/waitress.h +++ b/libwaitress/src/waitress.h @@ -34,6 +34,8 @@ THE SOFTWARE. typedef enum {WAITRESS_METHOD_GET = 0, WAITRESS_METHOD_POST} WaitressMethod_t; +typedef enum {WAITRESS_CB_RET_ERR, WAITRESS_CB_RET_OK} WaitressCbReturn_t; + typedef struct { char host[WAITRESS_HOST_SIZE]; char port[WAITRESS_PORT_SIZE]; @@ -47,7 +49,7 @@ typedef struct { char proxyPort[WAITRESS_PORT_SIZE]; /* extra data handed over to callback function */ void *data; - char (*callback) (void *, size_t, void *); + WaitressCbReturn_t (*callback) (void *, size_t, void *); int socktimeout; } WaitressHandle_t; |