static void test_printbuf_memappend(int *before_resize) { struct printbuf *pb; int initial_size; printf("%s: starting test\n", __func__); pb = printbuf_new(); printf("Buffer length: %d\n", printbuf_length(pb)); initial_size = pb->size; while(pb->size == initial_size) { printbuf_memappend_fast(pb, "x", 1); } *before_resize = printbuf_length(pb) - 1; printf("Appended %d bytes for resize: [%s]\n", *before_resize + 1, pb->buf); printbuf_reset(pb); printbuf_memappend_fast(pb, "bluexyz123", 3); printf("Partial append: %d, [%s]\n", printbuf_length(pb), pb->buf); char with_nulls[] = { 'a', 'b', '\0', 'c' }; printbuf_reset(pb); printbuf_memappend_fast(pb, with_nulls, (int)sizeof(with_nulls)); printf("With embedded \\0 character: %d, [%s]\n", printbuf_length(pb), pb->buf); printbuf_free(pb); pb = printbuf_new(); char *data = malloc(*before_resize); memset(data, 'X', *before_resize); printbuf_memappend_fast(pb, data, *before_resize); printf("Append to just before resize: %d, [%s]\n", printbuf_length(pb), pb->buf); free(data); printbuf_free(pb); pb = printbuf_new(); data = malloc(*before_resize + 1); memset(data, 'X', *before_resize + 1); printbuf_memappend_fast(pb, data, *before_resize + 1); printf("Append to just after resize: %d, [%s]\n", printbuf_length(pb), pb->buf); free(data); printbuf_free(pb); #define SA_TEST_STR "XXXXXXXXXXXXXXXX" pb = printbuf_new(); printbuf_strappend(pb, SA_TEST_STR); printf("Buffer size after printbuf_strappend(): %d, [%s]\n", printbuf_length(pb), pb->buf); printbuf_free(pb); #undef SA_TEST_STR printf("%s: end test\n", __func__); }
static void test_sprintbuf(int before_resize) { struct printbuf *pb; printf("%s: starting test\n", __func__); pb = printbuf_new(); printf("Buffer length: %d\n", printbuf_length(pb)); char *data = malloc(before_resize + 1 + 1); memset(data, 'X', before_resize + 1 + 1); data[before_resize + 1] = '\0'; sprintbuf(pb, "%s", data); free(data); printf("sprintbuf to just after resize(%d+1): %d, [%s], strlen(buf)=%d\n", before_resize, printbuf_length(pb), pb->buf, (int)strlen(pb->buf)); printbuf_reset(pb); sprintbuf(pb, "plain"); printf("%d, [%s]\n", printbuf_length(pb), pb->buf); sprintbuf(pb, "%d", 1); printf("%d, [%s]\n", printbuf_length(pb), pb->buf); sprintbuf(pb, "%d", INT_MAX); printf("%d, [%s]\n", printbuf_length(pb), pb->buf); sprintbuf(pb, "%d", INT_MIN); printf("%d, [%s]\n", printbuf_length(pb), pb->buf); sprintbuf(pb, "%s", "%s"); printf("%d, [%s]\n", printbuf_length(pb), pb->buf); printbuf_free(pb); printf("%s: end test\n", __func__); }
/** * Read a line from a socket to a print buffer buf. * Calls error() in case of any problems. */ static void read_line(int fd, struct printbuf *buf){ printbuf_reset(buf); char c; do{ ssize_t status = read(fd, &c, 1); if(status == -1){ error(errno, NULL); }else if(status < 1){ error(0, "Remote side closed the connection up."); } if(printbuf_memappend(buf, &c, 1) == -1){ error(errno, NULL); } }while(c != '\n'); }