cmd-save-buffer.c

   1 /* $Id: cmd-save-buffer.c,v 1.4 2009-02-08 13:36:40 tcunha 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_AFLAG|CMD_ARG1,
  37         cmd_buffer_init,
  38         cmd_buffer_parse,
  39         cmd_save_buffer_exec,
  40         cmd_buffer_send,
  41         cmd_buffer_recv,
  42         cmd_buffer_free,
  43         cmd_buffer_print
  44 };
  45
  46 int
  47 cmd_save_buffer_exec(struct cmd *self, struct cmd_ctx *ctx)
  48 {
  49         struct cmd_buffer_data  *data = self->data;
  50         struct session          *s;
  51         struct paste_buffer     *pb;
  52         mode_t                  mask;
  53         FILE                    *f;
  54
  55         if ((s = cmd_find_session(ctx, data->target)) == NULL)
  56                 return (-1);
  57
  58         if (data->buffer == -1) {
  59                 if ((pb = paste_get_top(&s->buffers)) == NULL) {
  60                         ctx->error(ctx, "no buffers");
  61                         return (-1);
  62                 }
  63         } else {
  64                 if ((pb = paste_get_index(&s->buffers, data->buffer)) == NULL) {
  65                         ctx->error(ctx, "no buffer %d", data->buffer);
  66                         return (-1);
  67                 }
  68         }
  69
  70         mask = umask(S_IRWXG | S_IRWXO);
  71         if (data->flags & CMD_AFLAG)
  72                 f = fopen(data->arg, "a");
  73         else
  74                 f = fopen(data->arg, "w");
  75         if (f == NULL) {
  76                 ctx->error(ctx, "%s: %s", data->arg, strerror(errno));
  77                 return (-1);
  78         }
  79
  80         if (fwrite(pb->data, 1, strlen(pb->data), f) != strlen(pb->data)) {
  81                 ctx->error(ctx, "%s: fwrite error", data->arg);
  82                 fclose(f);
  83                 return (-1);
  84         }
  85
  86         fclose(f);
  87         umask(mask);
  88
  89         return (0);
  90 }