/* Try to write past end of buffer */
static int buf_stream_overfill_test(char *message, int len) {
int ret;
char buf[SHORT_BUF_LEN];
rd_t stream = open_buf_stream(buf, SHORT_BUF_LEN);
int written = fputs(stream, STREAM_MESSAGE);
if (written != SHORT_BUF_LEN-1) {
strncpy(message, "Incorrect number of characters written", len);
ret = FAILED;
goto out;
}
if (buf[SHORT_BUF_LEN-1] != '\0') {
/* Avoid using scnprintf, which will end up using buffer streams */
strncpy(message, "Terminating NULL overwritten", len);
ret = FAILED;
goto out;
}
ret = PASSED;
out:
close(stream);
return ret;
}
static int buf_stream_write_test(char *message, int len) {
int ret;
char buf[BUF_LEN];
rd_t stream = open_buf_stream(buf, BUF_LEN);
int written = fputs(stream, STREAM_MESSAGE);
if (written != ARRAY_LENGTH(STREAM_MESSAGE)-1) {
strncpy(message, "Incorrect number of characters written", len);
ret = FAILED;
goto out;
}
if (strncmp(buf, STREAM_MESSAGE, BUF_LEN) != 0) {
/* Avoid using scnprintf, which will end up using buffer streams */
strncpy(message, "Buffer does not match written contents.", len);
ret = FAILED;
goto out;
}
ret = PASSED;
out:
close(stream);
return ret;
}
/* It should not be possible to write to a length 1 buffer,
* as the last space is left for a NULL char */
static int buf_stream_single_char_test(char *message, int len) {
int ret;
char c = '\0';
rd_t stream = open_buf_stream(&c, 1);
int written = fputs(stream, STREAM_MESSAGE);
if (written != 0) {
strncpy(message, "Incorrect number of characters written", len);
ret = FAILED;
goto out;
}
if (c != '\0') {
/* Avoid using scnprintf, which will end up using buffer streams */
strncpy(message, "Stream overwrote character", len);
ret = FAILED;
goto out;
}
ret = PASSED;
out:
close(stream);
return ret;
}
int scnprintf(char *buf, uint32_t n, char *fmt, ...) {
va_list ap;
va_start(ap, fmt);
rd_t stream = open_buf_stream(buf, n);
int ret = vfprintf(stream, fmt, ap, &fputs, &fputc);
close(stream);
va_end(ap);
return ret;
}
static int buf_stream_read_test(char *message, int len) {
int ret;
char buf[BUF_LEN] = STREAM_MESSAGE;
char read_buf[BUF_LEN];
int i = 0;
rd_t stream = open_buf_stream(buf, BUF_LEN);
while ((read_buf[i++] = fgetc(stream)) && (i < BUF_LEN));
if (strncmp(read_buf, STREAM_MESSAGE, BUF_LEN) != 0) {
/* Avoid using scnprintf, which will end up using buffer streams */
strncpy(message, "Read characters do not match original buffer", len);
ret = FAILED;
goto out;
}
ret = PASSED;
out:
close(stream);
return ret;
}
