aboutsummaryrefslogtreecommitdiff
path: root/docs/code/uvtee/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'docs/code/uvtee/main.c')
-rw-r--r--docs/code/uvtee/main.c80
1 files changed, 80 insertions, 0 deletions
diff --git a/docs/code/uvtee/main.c b/docs/code/uvtee/main.c
new file mode 100644
index 00000000..6216c2eb
--- /dev/null
+++ b/docs/code/uvtee/main.c
@@ -0,0 +1,80 @@
+#include <stdio.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdlib.h>
+
+#include <uv.h>
+
+typedef struct {
+ uv_write_t req;
+ uv_buf_t buf;
+} write_req_t;
+
+uv_loop_t *loop;
+uv_pipe_t stdin_pipe;
+uv_pipe_t stdout_pipe;
+uv_pipe_t file_pipe;
+
+void alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {
+ *buf = uv_buf_init((char*) malloc(suggested_size), suggested_size);
+}
+
+void free_write_req(uv_write_t *req) {
+ write_req_t *wr = (write_req_t*) req;
+ free(wr->buf.base);
+ free(wr);
+}
+
+void on_stdout_write(uv_write_t *req, int status) {
+ free_write_req(req);
+}
+
+void on_file_write(uv_write_t *req, int status) {
+ free_write_req(req);
+}
+
+void write_data(uv_stream_t *dest, size_t size, uv_buf_t buf, uv_write_cb cb) {
+ write_req_t *req = (write_req_t*) malloc(sizeof(write_req_t));
+ req->buf = uv_buf_init((char*) malloc(size), size);
+ memcpy(req->buf.base, buf.base, size);
+ uv_write((uv_write_t*) req, (uv_stream_t*)dest, &req->buf, 1, cb);
+}
+
+void read_stdin(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf) {
+ if (nread < 0){
+ if (nread == UV_EOF){
+ // end of file
+ uv_close((uv_handle_t *)&stdin_pipe, NULL);
+ uv_close((uv_handle_t *)&stdout_pipe, NULL);
+ uv_close((uv_handle_t *)&file_pipe, NULL);
+ }
+ } else if (nread > 0) {
+ write_data((uv_stream_t *)&stdout_pipe, nread, *buf, on_stdout_write);
+ write_data((uv_stream_t *)&file_pipe, nread, *buf, on_file_write);
+ }
+
+ // OK to free buffer as write_data copies it.
+ if (buf->base)
+ free(buf->base);
+}
+
+int main(int argc, char **argv) {
+ loop = uv_default_loop();
+
+ uv_pipe_init(loop, &stdin_pipe, 0);
+ uv_pipe_open(&stdin_pipe, 0);
+
+ uv_pipe_init(loop, &stdout_pipe, 0);
+ uv_pipe_open(&stdout_pipe, 1);
+
+ uv_fs_t file_req;
+ int fd = uv_fs_open(loop, &file_req, argv[1], O_CREAT | O_RDWR, 0644, NULL);
+ uv_pipe_init(loop, &file_pipe, 0);
+ uv_pipe_open(&file_pipe, fd);
+
+ uv_read_start((uv_stream_t*)&stdin_pipe, alloc_buffer, read_stdin);
+
+ uv_run(loop, UV_RUN_DEFAULT);
+ return 0;
+}