Add ingest

3 files changed, 66 insertions, 1 deletions

diff --git a/.gitignore b/.gitignore
index 18271b1..5493d5f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 broadcast
+ingest
 tags
 view
diff --git a/Makefile b/Makefile
index b6c3d8d..664647c 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,4 @@
-BINS = broadcast view
+BINS = broadcast ingest view
 
 CFLAGS += -Wall -Wextra -Wpedantic
 LDLIBS = -lutil
diff --git a/ingest.c b/ingest.c
new file mode 100644
index 0000000..9fa754d
--- /dev/null
+++ b/ingest.c
@@ -0,0 +1,64 @@
+/* Copyright (c) 2018, Curtis McEnroe <programble@gmail.com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <err.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/ioctl.h>
+#include <sysexits.h>
+#include <unistd.h>
+
+int main(int argc, char *argv[]) {
+    if (argc < 2) return EX_USAGE;
+    const char *path = argv[1];
+
+    int remote = STDIN_FILENO;
+
+    int stream = open(path, O_WRONLY);
+    if (stream < 0) err(EX_NOINPUT, "%s", path);
+
+    struct winsize window;
+    ssize_t size = read(remote, &window, sizeof(window));
+    if (size < 0) err(EX_IOERR, "read(%d)", remote);
+    if ((size_t)size < sizeof(window)) errx(EX_IOERR, "short read(%d)", remote);
+
+    for (;;) {
+        int error = ftruncate(stream, 0);
+        if (error) err(EX_IOERR, "%s", path);
+
+        off_t offset = lseek(stream, 0, SEEK_SET);
+        if (offset < 0) err(EX_IOERR, "%s", path);
+
+        ssize_t size = write(stream, &window, sizeof(window));
+        if (size < 0) err(EX_IOERR, "write(%d)", stream);
+        if ((size_t)size < sizeof(window)) errx(EX_IOERR, "short write(%d)", stream);
+
+        char buf[4096];
+        ssize_t totalSize = 0;
+        while (totalSize < 1024 * 1024) {
+            ssize_t readSize = read(remote, buf, sizeof(buf));
+            if (readSize < 0) err(EX_IOERR, "read(%d)", remote);
+            if (!readSize) return EX_OK;
+
+            ssize_t writeSize = write(stream, buf, readSize);
+            if (writeSize < 0) err(EX_IOERR, "write(%d)", stream);
+            if (writeSize < readSize) errx(EX_IOERR, "short write(%d)", stream);
+
+            totalSize += readSize;
+        }
+    }
+}