|
160
|
1 #include <stdio.h>
|
|
|
2 #include <fcntl.h>
|
|
|
3 #include <string.h>
|
|
|
4 #include <stdlib.h>
|
|
|
5
|
|
|
6 #include <uv.h>
|
|
|
7
|
|
|
8 typedef struct {
|
|
|
9 uv_write_t req;
|
|
|
10 uv_buf_t buf;
|
|
|
11 } write_req_t;
|
|
|
12
|
|
|
13 uv_loop_t *loop;
|
|
|
14 uv_pipe_t stdin_pipe;
|
|
|
15 uv_pipe_t stdout_pipe;
|
|
|
16 uv_pipe_t file_pipe;
|
|
|
17
|
|
|
18 void alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {
|
|
|
19 *buf = uv_buf_init((char*) malloc(suggested_size), suggested_size);
|
|
|
20 }
|
|
|
21
|
|
|
22 void free_write_req(uv_write_t *req) {
|
|
|
23 write_req_t *wr = (write_req_t*) req;
|
|
|
24 free(wr->buf.base);
|
|
|
25 free(wr);
|
|
|
26 }
|
|
|
27
|
|
|
28 void on_stdout_write(uv_write_t *req, int status) {
|
|
|
29 free_write_req(req);
|
|
|
30 }
|
|
|
31
|
|
|
32 void on_file_write(uv_write_t *req, int status) {
|
|
|
33 free_write_req(req);
|
|
|
34 }
|
|
|
35
|
|
|
36 void write_data(uv_stream_t *dest, size_t size, uv_buf_t buf, uv_write_cb cb) {
|
|
|
37 write_req_t *req = (write_req_t*) malloc(sizeof(write_req_t));
|
|
|
38 req->buf = uv_buf_init((char*) malloc(size), size);
|
|
|
39 memcpy(req->buf.base, buf.base, size);
|
|
|
40 uv_write((uv_write_t*) req, (uv_stream_t*)dest, &req->buf, 1, cb);
|
|
|
41 }
|
|
|
42
|
|
|
43 void read_stdin(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf) {
|
|
|
44 if (nread < 0){
|
|
|
45 if (nread == UV_EOF){
|
|
|
46 // end of file
|
|
|
47 uv_close((uv_handle_t *)&stdin_pipe, NULL);
|
|
|
48 uv_close((uv_handle_t *)&stdout_pipe, NULL);
|
|
|
49 uv_close((uv_handle_t *)&file_pipe, NULL);
|
|
|
50 }
|
|
|
51 } else if (nread > 0) {
|
|
|
52 write_data((uv_stream_t *)&stdout_pipe, nread, *buf, on_stdout_write);
|
|
|
53 write_data((uv_stream_t *)&file_pipe, nread, *buf, on_file_write);
|
|
|
54 }
|
|
|
55
|
|
|
56 // OK to free buffer as write_data copies it.
|
|
|
57 if (buf->base)
|
|
|
58 free(buf->base);
|
|
|
59 }
|
|
|
60
|
|
|
61 int main(int argc, char **argv) {
|
|
|
62 loop = uv_default_loop();
|
|
|
63
|
|
|
64 uv_pipe_init(loop, &stdin_pipe, 0);
|
|
|
65 uv_pipe_open(&stdin_pipe, 0);
|
|
|
66
|
|
|
67 uv_pipe_init(loop, &stdout_pipe, 0);
|
|
|
68 uv_pipe_open(&stdout_pipe, 1);
|
|
|
69
|
|
|
70 uv_fs_t file_req;
|
|
|
71 int fd = uv_fs_open(loop, &file_req, argv[1], O_CREAT | O_RDWR, 0644, NULL);
|
|
|
72 uv_pipe_init(loop, &file_pipe, 0);
|
|
|
73 uv_pipe_open(&file_pipe, fd);
|
|
|
74
|
|
|
75 uv_read_start((uv_stream_t*)&stdin_pipe, alloc_buffer, read_stdin);
|
|
|
76
|
|
|
77 uv_run(loop, UV_RUN_DEFAULT);
|
|
|
78 return 0;
|
|
|
79 }
|
