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 <stdlib.h>
  24 #include <string.h>
  25
  26 #include "tmux.h"
  27
  28 /*
  29  * Saves a paste buffer to a file.
  30  */
  31
  32 enum cmd_retval  cmd_save_buffer_exec(struct cmd *, struct cmd_ctx *);
  33
  34 const struct cmd_entry cmd_save_buffer_entry = {
  35         "save-buffer", "saveb",
  36         "ab:", 1, 1,
  37         "[-a] " CMD_BUFFER_USAGE " path",
  38         0,
  39         NULL,
  40         NULL,
  41         cmd_save_buffer_exec
  42 };
  43
  44 enum cmd_retval
  45 cmd_save_buffer_exec(struct cmd *self, struct cmd_ctx *ctx)
  46 {
  47         struct args             *args = self->args;
  48         struct client           *c;
  49         struct session          *s;
  50         struct paste_buffer     *pb;
  51         const char              *path, *newpath, *wd;
  52         char                    *cause;
  53         int                      buffer;
  54         mode_t                   mask;
  55         FILE                    *f;
  56
  57         if (!args_has(args, 'b')) {
  58                 if ((pb = paste_get_top(&global_buffers)) == NULL) {
  59                         ctx->error(ctx, "no buffers");
  60                         return (CMD_RETURN_ERROR);
  61                 }
  62         } else {
  63                 buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause);
  64                 if (cause != NULL) {
  65                         ctx->error(ctx, "buffer %s", cause);
  66                         free(cause);
  67                         return (CMD_RETURN_ERROR);
  68                 }
  69
  70                 pb = paste_get_index(&global_buffers, buffer);
  71                 if (pb == NULL) {
  72                         ctx->error(ctx, "no buffer %d", buffer);
  73                         return (CMD_RETURN_ERROR);
  74                 }
  75         }
  76
  77         path = args->argv[0];
  78         if (strcmp(path, "-") == 0) {
  79                 c = ctx->curclient;
  80                 if (c == NULL || !(c->flags & CLIENT_CONTROL))
  81                         c = ctx->cmdclient;
  82                 if (c == NULL) {
  83                         ctx->error(ctx, "can't write to stdout");
  84                         return (CMD_RETURN_ERROR);
  85                 }
  86                 evbuffer_add(c->stdout_data, pb->data, pb->size);
  87                 server_push_stdout(c);
  88         } else {
  89                 c = ctx->cmdclient;
  90                 if (c != NULL)
  91                         wd = c->cwd;
  92                 else if ((s = cmd_current_session(ctx, 0)) != NULL) {
  93                         wd = options_get_string(&s->options, "default-path");
  94                         if (*wd == '\0')
  95                                 wd = s->cwd;
  96                 } else
  97                         wd = NULL;
  98                 if (wd != NULL && *wd != '\0') {
  99                         newpath = get_full_path(wd, path);
 100                         if (newpath != NULL)
 101                                 path = newpath;
 102                 }
 103
 104                 mask = umask(S_IRWXG | S_IRWXO);
 105                 if (args_has(self->args, 'a'))
 106                         f = fopen(path, "ab");
 107                 else
 108                         f = fopen(path, "wb");
 109                 umask(mask);
 110                 if (f == NULL) {
 111                         ctx->error(ctx, "%s: %s", path, strerror(errno));
 112                         return (CMD_RETURN_ERROR);
 113                 }
 114                 if (fwrite(pb->data, 1, pb->size, f) != pb->size) {
 115                         ctx->error(ctx, "%s: fwrite error", path);
 116                         fclose(f);
 117                         return (CMD_RETURN_ERROR);
 118                 }
 119                 fclose(f);
 120         }
 121
 122         return (CMD_RETURN_NORMAL);
 123 }