xref: /openbsd-src/usr.bin/tmux/cmd-load-buffer.c (revision 8a0ccaa27ef88657cfabc0103b87ab8472478d3b)
1 /* $OpenBSD: cmd-load-buffer.c,v 1.5 2009/07/30 17:46:12 nicm Exp $ */
2 
3 /*
4  * Copyright (c) 2009 Tiago Cunha <me@tiagocunha.org>
5  *
6  * Permission to use, copy, modify, and distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 
19 #include <sys/types.h>
20 #include <sys/stat.h>
21 
22 #include <errno.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26 
27 #include "tmux.h"
28 
29 /*
30  * Loads a session paste buffer from a file.
31  */
32 
33 int	cmd_load_buffer_exec(struct cmd *, struct cmd_ctx *);
34 
35 const struct cmd_entry cmd_load_buffer_entry = {
36 	"load-buffer", "loadb",
37 	CMD_BUFFER_SESSION_USAGE " path",
38 	CMD_ARG1, 0,
39 	cmd_buffer_init,
40 	cmd_buffer_parse,
41 	cmd_load_buffer_exec,
42 	cmd_buffer_free,
43 	cmd_buffer_print
44 };
45 
46 int
47 cmd_load_buffer_exec(struct cmd *self, struct cmd_ctx *ctx)
48 {
49 	struct cmd_buffer_data	*data = self->data;
50 	struct session		*s;
51 	struct stat		statbuf;
52 	FILE			*f;
53 	char			*buf;
54 	u_int			limit;
55 
56 	if ((s = cmd_find_session(ctx, data->target)) == NULL)
57 		return (-1);
58 
59 	if (stat(data->arg, &statbuf) < 0) {
60 		ctx->error(ctx, "%s: %s", data->arg, strerror(errno));
61 		return (-1);
62 	}
63 
64 	if ((f = fopen(data->arg, "rb")) == NULL) {
65 		ctx->error(ctx, "%s: %s", data->arg, strerror(errno));
66 		return (-1);
67 	}
68 
69 	/*
70 	 * We don't want to die due to memory exhaustion, hence xmalloc can't
71 	 * be used here.
72 	 */
73 	if ((buf = malloc(statbuf.st_size + 1)) == NULL) {
74 		ctx->error(ctx, "malloc error: %s", strerror(errno));
75 		fclose(f);
76 		return (-1);
77 	}
78 
79 	if (fread(buf, 1, statbuf.st_size, f) != (size_t) statbuf.st_size) {
80 		ctx->error(ctx, "%s: fread error", data->arg);
81 		xfree(buf);
82 		fclose(f);
83 		return (-1);
84 	}
85 
86 	buf[statbuf.st_size] = '\0';
87 	fclose(f);
88 
89 	limit = options_get_number(&s->options, "buffer-limit");
90 	if (data->buffer == -1) {
91 		paste_add(&s->buffers, buf, limit);
92 		return (0);
93 	}
94 	if (paste_replace(&s->buffers, data->buffer, buf) != 0) {
95 		ctx->error(ctx, "no buffer %d", data->buffer);
96 		xfree(buf);
97 		return (-1);
98 	}
99 
100 	return (0);
101 }
102