2 files changed, 87 insertions, 0 deletions
diff --git a/buffer.c b/buffer.c
new file mode 100644
index 0000000..71d51f8
--- /dev/null
+++ b/buffer.c
@@ -0,0 +1,43 @@
+#include "buffer.h"
+
+#include <stdlib.h>
+#include <string.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+void BufferCreate(struct Buffer* buffer) {
+  memset(buffer, 0, sizeof(struct Buffer));
+}
+
+void* BufferReserve(struct Buffer* buffer, size_t size) {
+  size_t alloc = buffer->size + size;
+  if (alloc > buffer->alloc) {
+    void* data = realloc(buffer->data, alloc);
+    if (!data) return NULL;
+    buffer->data = data;
+    buffer->alloc = alloc;
+  }
+  return (char*)buffer->data + buffer->size;
+}
+
+int BufferAppendFrom(struct Buffer* buffer, int fd) {
+  int nbytes;
+  if (ioctl(fd, FIONREAD, &nbytes) == -1) return -1;
+  if (!nbytes) return 0;
+  void* data = BufferReserve(buffer, (size_t)nbytes);
+  if (!data) return -1;
+  ssize_t result = read(fd, data, (size_t)nbytes);
+  if (result > 0) buffer->size += (size_t)result;
+  return (int)result;
+}
+
+void BufferDiscard(struct Buffer* buffer, size_t size) {
+  size_t tail_size = buffer->size - size;
+  if (tail_size) {
+    const char* tail = (const char*)buffer->data + size;
+    memmove(buffer->data, tail, tail_size);
+  }
+  buffer->size = tail_size;
+}
+
+void BufferDestroy(struct Buffer* buffer) { free(buffer->data); }
diff --git a/buffer.h b/buffer.h
new file mode 100644
index 0000000..40ce4cc
--- /dev/null
+++ b/buffer.h
@@ -0,0 +1,44 @@
+/*
+ * Copyright (C) 2022 Mikhail Burakov. This file is part of toolbox.
+ *
+ * toolbox is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * toolbox is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with toolbox.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#ifndef TOOLBOX_BUFFER_H_
+#define TOOLBOX_BUFFER_H_
+
+#include <stdbool.h>
+#include <stddef.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif  // __cplusplus
+
+struct Buffer {
+  void* data;
+  size_t size;
+  size_t alloc;
+};
+
+void BufferCreate(struct Buffer* buffer);
+void* BufferReserve(struct Buffer* buffer, size_t size);
+int BufferAppendFrom(struct Buffer* buffer, int fd);
+void BufferDiscard(struct Buffer* buffer, size_t size);
+void BufferDestroy(struct Buffer* buffer);
+
+#ifdef __cplusplus
+}  // extern "C"
+#endif  // __cplusplus
+
+#endif  // TOOLBOX_BUFFER_H_