1 /* $Id: cmd-save-buffer.c,v 1.1.1.2 2011/08/17 18:40:04 jmmv 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 "ab:", 1, 1, 36 "[-a] " CMD_BUFFER_USAGE, 37 0, 38 NULL, 39 NULL, 40 cmd_save_buffer_exec 41 }; 42 43 int 44 cmd_save_buffer_exec(struct cmd *self, struct cmd_ctx *ctx) 45 { 46 struct args *args = self->args; 47 struct client *c = ctx->cmdclient; 48 struct paste_buffer *pb; 49 const char *path; 50 char *cause; 51 int buffer; 52 mode_t mask; 53 FILE *f; 54 55 if (!args_has(args, 'b')) { 56 if ((pb = paste_get_top(&global_buffers)) == NULL) { 57 ctx->error(ctx, "no buffers"); 58 return (-1); 59 } 60 } else { 61 buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause); 62 if (cause != NULL) { 63 ctx->error(ctx, "buffer %s", cause); 64 xfree(cause); 65 return (-1); 66 } 67 68 pb = paste_get_index(&global_buffers, buffer); 69 if (pb == NULL) { 70 ctx->error(ctx, "no buffer %d", buffer); 71 return (-1); 72 } 73 } 74 75 path = args->argv[0]; 76 if (strcmp(path, "-") == 0) { 77 if (c == NULL) { 78 ctx->error(ctx, "%s: can't write to stdout", path); 79 return (-1); 80 } 81 bufferevent_write(c->stdout_event, pb->data, pb->size); 82 } else { 83 mask = umask(S_IRWXG | S_IRWXO); 84 if (args_has(self->args, 'a')) 85 f = fopen(path, "ab"); 86 else 87 f = fopen(path, "wb"); 88 umask(mask); 89 if (f == NULL) { 90 ctx->error(ctx, "%s: %s", path, strerror(errno)); 91 return (-1); 92 } 93 if (fwrite(pb->data, 1, pb->size, f) != pb->size) { 94 ctx->error(ctx, "%s: fwrite error", path); 95 fclose(f); 96 return (-1); 97 } 98 fclose(f); 99 } 100 101 return (0); 102 } 103