1*0a6a1f1dSLionel Sambuc /* Id */
2eda6f593SDavid van Moolenbroek
3eda6f593SDavid van Moolenbroek /*
4eda6f593SDavid van Moolenbroek * Copyright (c) 2009 Tiago Cunha <me@tiagocunha.org>
5eda6f593SDavid van Moolenbroek *
6eda6f593SDavid van Moolenbroek * Permission to use, copy, modify, and distribute this software for any
7eda6f593SDavid van Moolenbroek * purpose with or without fee is hereby granted, provided that the above
8eda6f593SDavid van Moolenbroek * copyright notice and this permission notice appear in all copies.
9eda6f593SDavid van Moolenbroek *
10eda6f593SDavid van Moolenbroek * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11eda6f593SDavid van Moolenbroek * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12eda6f593SDavid van Moolenbroek * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13eda6f593SDavid van Moolenbroek * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14eda6f593SDavid van Moolenbroek * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15eda6f593SDavid van Moolenbroek * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16eda6f593SDavid van Moolenbroek * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17eda6f593SDavid van Moolenbroek */
18eda6f593SDavid van Moolenbroek
19eda6f593SDavid van Moolenbroek #include <sys/types.h>
20eda6f593SDavid van Moolenbroek
21eda6f593SDavid van Moolenbroek #include <errno.h>
22*0a6a1f1dSLionel Sambuc #include <fcntl.h>
23eda6f593SDavid van Moolenbroek #include <stdio.h>
24eda6f593SDavid van Moolenbroek #include <stdlib.h>
25eda6f593SDavid van Moolenbroek #include <string.h>
26eda6f593SDavid van Moolenbroek #include <unistd.h>
27eda6f593SDavid van Moolenbroek
28eda6f593SDavid van Moolenbroek #include "tmux.h"
29eda6f593SDavid van Moolenbroek
30eda6f593SDavid van Moolenbroek /*
31*0a6a1f1dSLionel Sambuc * Loads a paste buffer from a file.
32eda6f593SDavid van Moolenbroek */
33eda6f593SDavid van Moolenbroek
34*0a6a1f1dSLionel Sambuc enum cmd_retval cmd_load_buffer_exec(struct cmd *, struct cmd_q *);
35*0a6a1f1dSLionel Sambuc void cmd_load_buffer_callback(struct client *, int, void *);
36eda6f593SDavid van Moolenbroek
37eda6f593SDavid van Moolenbroek const struct cmd_entry cmd_load_buffer_entry = {
38eda6f593SDavid van Moolenbroek "load-buffer", "loadb",
39eda6f593SDavid van Moolenbroek "b:", 1, 1,
40eda6f593SDavid van Moolenbroek CMD_BUFFER_USAGE " path",
41eda6f593SDavid van Moolenbroek 0,
42eda6f593SDavid van Moolenbroek NULL,
43eda6f593SDavid van Moolenbroek cmd_load_buffer_exec
44eda6f593SDavid van Moolenbroek };
45eda6f593SDavid van Moolenbroek
46*0a6a1f1dSLionel Sambuc enum cmd_retval
cmd_load_buffer_exec(struct cmd * self,struct cmd_q * cmdq)47*0a6a1f1dSLionel Sambuc cmd_load_buffer_exec(struct cmd *self, struct cmd_q *cmdq)
48eda6f593SDavid van Moolenbroek {
49eda6f593SDavid van Moolenbroek struct args *args = self->args;
50*0a6a1f1dSLionel Sambuc struct client *c = cmdq->client;
51*0a6a1f1dSLionel Sambuc struct session *s;
52eda6f593SDavid van Moolenbroek FILE *f;
53eda6f593SDavid van Moolenbroek const char *path;
54eda6f593SDavid van Moolenbroek char *pdata, *new_pdata, *cause;
55eda6f593SDavid van Moolenbroek size_t psize;
56eda6f593SDavid van Moolenbroek u_int limit;
57*0a6a1f1dSLionel Sambuc int ch, error, buffer, *buffer_ptr, cwd, fd;
58eda6f593SDavid van Moolenbroek
59eda6f593SDavid van Moolenbroek if (!args_has(args, 'b'))
60eda6f593SDavid van Moolenbroek buffer = -1;
61eda6f593SDavid van Moolenbroek else {
62eda6f593SDavid van Moolenbroek buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause);
63eda6f593SDavid van Moolenbroek if (cause != NULL) {
64*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "buffer %s", cause);
65*0a6a1f1dSLionel Sambuc free(cause);
66*0a6a1f1dSLionel Sambuc return (CMD_RETURN_ERROR);
67eda6f593SDavid van Moolenbroek }
68eda6f593SDavid van Moolenbroek }
69eda6f593SDavid van Moolenbroek
70eda6f593SDavid van Moolenbroek path = args->argv[0];
71eda6f593SDavid van Moolenbroek if (strcmp(path, "-") == 0) {
72eda6f593SDavid van Moolenbroek buffer_ptr = xmalloc(sizeof *buffer_ptr);
73eda6f593SDavid van Moolenbroek *buffer_ptr = buffer;
74eda6f593SDavid van Moolenbroek
75*0a6a1f1dSLionel Sambuc error = server_set_stdin_callback(c, cmd_load_buffer_callback,
76*0a6a1f1dSLionel Sambuc buffer_ptr, &cause);
77*0a6a1f1dSLionel Sambuc if (error != 0) {
78*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "%s: %s", path, cause);
79*0a6a1f1dSLionel Sambuc free(cause);
80*0a6a1f1dSLionel Sambuc return (CMD_RETURN_ERROR);
81*0a6a1f1dSLionel Sambuc }
82*0a6a1f1dSLionel Sambuc return (CMD_RETURN_WAIT);
83eda6f593SDavid van Moolenbroek }
84eda6f593SDavid van Moolenbroek
85*0a6a1f1dSLionel Sambuc if (c != NULL && c->session == NULL)
86*0a6a1f1dSLionel Sambuc cwd = c->cwd;
87*0a6a1f1dSLionel Sambuc else if ((s = cmd_current_session(cmdq, 0)) != NULL)
88*0a6a1f1dSLionel Sambuc cwd = s->cwd;
89*0a6a1f1dSLionel Sambuc else
90*0a6a1f1dSLionel Sambuc cwd = AT_FDCWD;
91*0a6a1f1dSLionel Sambuc
92*0a6a1f1dSLionel Sambuc if ((fd = openat(cwd, path, O_RDONLY)) == -1 ||
93*0a6a1f1dSLionel Sambuc (f = fdopen(fd, "rb")) == NULL) {
94*0a6a1f1dSLionel Sambuc if (fd != -1)
95*0a6a1f1dSLionel Sambuc close(fd);
96*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "%s: %s", path, strerror(errno));
97*0a6a1f1dSLionel Sambuc return (CMD_RETURN_ERROR);
98eda6f593SDavid van Moolenbroek }
99eda6f593SDavid van Moolenbroek
100eda6f593SDavid van Moolenbroek pdata = NULL;
101eda6f593SDavid van Moolenbroek psize = 0;
102eda6f593SDavid van Moolenbroek while ((ch = getc(f)) != EOF) {
103eda6f593SDavid van Moolenbroek /* Do not let the server die due to memory exhaustion. */
104eda6f593SDavid van Moolenbroek if ((new_pdata = realloc(pdata, psize + 2)) == NULL) {
105*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "realloc error: %s", strerror(errno));
106eda6f593SDavid van Moolenbroek goto error;
107eda6f593SDavid van Moolenbroek }
108eda6f593SDavid van Moolenbroek pdata = new_pdata;
109eda6f593SDavid van Moolenbroek pdata[psize++] = ch;
110eda6f593SDavid van Moolenbroek }
111eda6f593SDavid van Moolenbroek if (ferror(f)) {
112*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "%s: read error", path);
113eda6f593SDavid van Moolenbroek goto error;
114eda6f593SDavid van Moolenbroek }
115eda6f593SDavid van Moolenbroek if (pdata != NULL)
116eda6f593SDavid van Moolenbroek pdata[psize] = '\0';
117eda6f593SDavid van Moolenbroek
118eda6f593SDavid van Moolenbroek fclose(f);
119eda6f593SDavid van Moolenbroek
120eda6f593SDavid van Moolenbroek limit = options_get_number(&global_options, "buffer-limit");
121eda6f593SDavid van Moolenbroek if (buffer == -1) {
122eda6f593SDavid van Moolenbroek paste_add(&global_buffers, pdata, psize, limit);
123*0a6a1f1dSLionel Sambuc return (CMD_RETURN_NORMAL);
124eda6f593SDavid van Moolenbroek }
125eda6f593SDavid van Moolenbroek if (paste_replace(&global_buffers, buffer, pdata, psize) != 0) {
126*0a6a1f1dSLionel Sambuc cmdq_error(cmdq, "no buffer %d", buffer);
127*0a6a1f1dSLionel Sambuc free(pdata);
128*0a6a1f1dSLionel Sambuc return (CMD_RETURN_ERROR);
129eda6f593SDavid van Moolenbroek }
130eda6f593SDavid van Moolenbroek
131*0a6a1f1dSLionel Sambuc return (CMD_RETURN_NORMAL);
132eda6f593SDavid van Moolenbroek
133eda6f593SDavid van Moolenbroek error:
134*0a6a1f1dSLionel Sambuc free(pdata);
135eda6f593SDavid van Moolenbroek if (f != NULL)
136eda6f593SDavid van Moolenbroek fclose(f);
137*0a6a1f1dSLionel Sambuc return (CMD_RETURN_ERROR);
138eda6f593SDavid van Moolenbroek }
139eda6f593SDavid van Moolenbroek
140eda6f593SDavid van Moolenbroek void
cmd_load_buffer_callback(struct client * c,int closed,void * data)141*0a6a1f1dSLionel Sambuc cmd_load_buffer_callback(struct client *c, int closed, void *data)
142eda6f593SDavid van Moolenbroek {
143eda6f593SDavid van Moolenbroek int *buffer = data;
144eda6f593SDavid van Moolenbroek char *pdata;
145eda6f593SDavid van Moolenbroek size_t psize;
146eda6f593SDavid van Moolenbroek u_int limit;
147eda6f593SDavid van Moolenbroek
148*0a6a1f1dSLionel Sambuc if (!closed)
149eda6f593SDavid van Moolenbroek return;
150*0a6a1f1dSLionel Sambuc c->stdin_callback = NULL;
151*0a6a1f1dSLionel Sambuc
152*0a6a1f1dSLionel Sambuc c->references--;
153*0a6a1f1dSLionel Sambuc if (c->flags & CLIENT_DEAD)
154*0a6a1f1dSLionel Sambuc return;
155*0a6a1f1dSLionel Sambuc
156*0a6a1f1dSLionel Sambuc psize = EVBUFFER_LENGTH(c->stdin_data);
157*0a6a1f1dSLionel Sambuc if (psize == 0 || (pdata = malloc(psize + 1)) == NULL) {
158*0a6a1f1dSLionel Sambuc free(data);
159*0a6a1f1dSLionel Sambuc goto out;
160eda6f593SDavid van Moolenbroek }
161*0a6a1f1dSLionel Sambuc memcpy(pdata, EVBUFFER_DATA(c->stdin_data), psize);
162eda6f593SDavid van Moolenbroek pdata[psize] = '\0';
163*0a6a1f1dSLionel Sambuc evbuffer_drain(c->stdin_data, psize);
164eda6f593SDavid van Moolenbroek
165eda6f593SDavid van Moolenbroek limit = options_get_number(&global_options, "buffer-limit");
166eda6f593SDavid van Moolenbroek if (*buffer == -1)
167eda6f593SDavid van Moolenbroek paste_add(&global_buffers, pdata, psize, limit);
168eda6f593SDavid van Moolenbroek else if (paste_replace(&global_buffers, *buffer, pdata, psize) != 0) {
169eda6f593SDavid van Moolenbroek /* No context so can't use server_client_msg_error. */
170*0a6a1f1dSLionel Sambuc evbuffer_add_printf(c->stderr_data, "no buffer %d\n", *buffer);
171*0a6a1f1dSLionel Sambuc server_push_stderr(c);
172*0a6a1f1dSLionel Sambuc free(pdata);
173eda6f593SDavid van Moolenbroek }
174eda6f593SDavid van Moolenbroek
175*0a6a1f1dSLionel Sambuc free(data);
176*0a6a1f1dSLionel Sambuc
177*0a6a1f1dSLionel Sambuc out:
178*0a6a1f1dSLionel Sambuc cmdq_continue(c->cmdq);
179eda6f593SDavid van Moolenbroek }
180