xref: /openbsd-src/usr.bin/tmux/cmd-save-buffer.c (revision 50027fe110c3c362514cbbf1128910104a00203e)
1 /* $OpenBSD: cmd-save-buffer.c,v 1.6 2009/11/13 19:53:29 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 <string.h>
24 
25 #include "tmux.h"
26 
27 /*
28  * Saves a session paste buffer to a file.
29  */
30 
31 int	cmd_save_buffer_exec(struct cmd *, struct cmd_ctx *);
32 
33 const struct cmd_entry cmd_save_buffer_entry = {
34 	"save-buffer", "saveb",
35 	"[-a] " CMD_BUFFER_SESSION_USAGE " path",
36 	CMD_ARG1, "a",
37 	cmd_buffer_init,
38 	cmd_buffer_parse,
39 	cmd_save_buffer_exec,
40 	cmd_buffer_free,
41 	cmd_buffer_print
42 };
43 
44 int
45 cmd_save_buffer_exec(struct cmd *self, struct cmd_ctx *ctx)
46 {
47 	struct cmd_buffer_data	*data = self->data;
48 	struct session		*s;
49 	struct paste_buffer	*pb;
50 	mode_t			mask;
51 	FILE			*f;
52 
53 	if ((s = cmd_find_session(ctx, data->target)) == NULL)
54 		return (-1);
55 
56 	if (data->buffer == -1) {
57 		if ((pb = paste_get_top(&s->buffers)) == NULL) {
58 			ctx->error(ctx, "no buffers");
59 			return (-1);
60 		}
61 	} else {
62 		if ((pb = paste_get_index(&s->buffers, data->buffer)) == NULL) {
63 			ctx->error(ctx, "no buffer %d", data->buffer);
64 			return (-1);
65 		}
66 	}
67 
68 	mask = umask(S_IRWXG | S_IRWXO);
69 	if (cmd_check_flag(data->chflags, 'a'))
70 		f = fopen(data->arg, "ab");
71 	else
72 		f = fopen(data->arg, "wb");
73 	umask(mask);
74 	if (f == NULL) {
75 		ctx->error(ctx, "%s: %s", data->arg, strerror(errno));
76 		return (-1);
77 	}
78 
79 	if (fwrite(pb->data, 1, pb->size, f) != pb->size) {
80 	    	ctx->error(ctx, "%s: fwrite error", data->arg);
81 	    	fclose(f);
82 	    	return (-1);
83 	}
84 
85 	fclose(f);
86 
87 	return (0);
88 }
89