cmd-save-buffer.c

   1 /* $OpenBSD$ */
   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 }