xref: /openbsd-src/usr.bin/tmux/cmd-save-buffer.c (revision 505ee9ea3b177e2387d907a91ca7da069f3f14d8)
1 /* $OpenBSD: cmd-save-buffer.c,v 1.50 2020/04/13 20:51:57 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 <fcntl.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27 #include <vis.h>
28 
29 #include "tmux.h"
30 
31 /*
32  * Saves a paste buffer to a file.
33  */
34 
35 static enum cmd_retval	cmd_save_buffer_exec(struct cmd *, struct cmdq_item *);
36 
37 const struct cmd_entry cmd_save_buffer_entry = {
38 	.name = "save-buffer",
39 	.alias = "saveb",
40 
41 	.args = { "ab:", 1, 1 },
42 	.usage = "[-a] " CMD_BUFFER_USAGE " path",
43 
44 	.flags = CMD_AFTERHOOK,
45 	.exec = cmd_save_buffer_exec
46 };
47 
48 const struct cmd_entry cmd_show_buffer_entry = {
49 	.name = "show-buffer",
50 	.alias = "showb",
51 
52 	.args = { "b:", 0, 0 },
53 	.usage = CMD_BUFFER_USAGE,
54 
55 	.flags = CMD_AFTERHOOK,
56 	.exec = cmd_save_buffer_exec
57 };
58 
59 static void
60 cmd_save_buffer_done(__unused struct client *c, const char *path, int error,
61     __unused int closed, __unused struct evbuffer *buffer, void *data)
62 {
63 	struct cmdq_item	*item = data;
64 
65 	if (!closed)
66 		return;
67 
68 	if (error != 0)
69 		cmdq_error(item, "%s: %s", path, strerror(error));
70 	cmdq_continue(item);
71 }
72 
73 static enum cmd_retval
74 cmd_save_buffer_exec(struct cmd *self, struct cmdq_item *item)
75 {
76 	struct args		*args = cmd_get_args(self);
77 	struct paste_buffer	*pb;
78 	int			 flags;
79 	const char		*bufname = args_get(args, 'b'), *bufdata;
80 	size_t			 bufsize;
81 	char			*path;
82 
83 	if (bufname == NULL) {
84 		if ((pb = paste_get_top(NULL)) == NULL) {
85 			cmdq_error(item, "no buffers");
86 			return (CMD_RETURN_ERROR);
87 		}
88 	} else {
89 		pb = paste_get_name(bufname);
90 		if (pb == NULL) {
91 			cmdq_error(item, "no buffer %s", bufname);
92 			return (CMD_RETURN_ERROR);
93 		}
94 	}
95 	bufdata = paste_buffer_data(pb, &bufsize);
96 
97 	if (cmd_get_entry(self) == &cmd_show_buffer_entry)
98 		path = xstrdup("-");
99 	else
100 		path = format_single_from_target(item, args->argv[0]);
101 	if (args_has(args, 'a'))
102 		flags = O_APPEND;
103 	else
104 		flags = 0;
105 	file_write(cmdq_get_client(item), path, flags, bufdata, bufsize,
106 	    cmd_save_buffer_done, item);
107 	free(path);
108 
109 	return (CMD_RETURN_WAIT);
110 }
111