4 * Copyright (c) 2009 Tiago Cunha <me@tiagocunha.org>
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.
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.
19 #include <sys/types.h>
30 * Loads a paste buffer from a file.
33 int cmd_load_buffer_exec(struct cmd
*, struct cmd_ctx
*);
34 void cmd_load_buffer_callback(struct client
*, void *);
36 const struct cmd_entry cmd_load_buffer_entry
= {
37 "load-buffer", "loadb",
39 CMD_BUFFER_USAGE
" path",
47 cmd_load_buffer_exec(struct cmd
*self
, struct cmd_ctx
*ctx
)
49 struct args
*args
= self
->args
;
50 struct client
*c
= ctx
->cmdclient
;
53 const char *path
, *newpath
, *wd
;
54 char *pdata
, *new_pdata
, *cause
;
60 if (!args_has(args
, 'b'))
63 buffer
= args_strtonum(args
, 'b', 0, INT_MAX
, &cause
);
65 ctx
->error(ctx
, "buffer %s", cause
);
72 if (strcmp(path
, "-") == 0) {
74 ctx
->error(ctx
, "%s: can't read from stdin", path
);
77 if (c
->flags
& CLIENT_TERMINAL
) {
78 ctx
->error(ctx
, "%s: stdin is a tty", path
);
81 if (c
->stdin_fd
== -1) {
82 ctx
->error(ctx
, "%s: can't read from stdin", path
);
86 buffer_ptr
= xmalloc(sizeof *buffer_ptr
);
89 c
->stdin_data
= buffer_ptr
;
90 c
->stdin_callback
= cmd_load_buffer_callback
;
93 bufferevent_enable(c
->stdin_event
, EV_READ
);
99 else if ((s
= cmd_current_session(ctx
, 0)) != NULL
) {
100 wd
= options_get_string(&s
->options
, "default-path");
105 if (wd
!= NULL
&& *wd
!= '\0') {
106 newpath
= get_full_path(wd
, path
);
110 if ((f
= fopen(path
, "rb")) == NULL
) {
111 ctx
->error(ctx
, "%s: %s", path
, strerror(errno
));
117 while ((ch
= getc(f
)) != EOF
) {
118 /* Do not let the server die due to memory exhaustion. */
119 if ((new_pdata
= realloc(pdata
, psize
+ 2)) == NULL
) {
120 ctx
->error(ctx
, "realloc error: %s", strerror(errno
));
127 ctx
->error(ctx
, "%s: read error", path
);
135 limit
= options_get_number(&global_options
, "buffer-limit");
137 paste_add(&global_buffers
, pdata
, psize
, limit
);
140 if (paste_replace(&global_buffers
, buffer
, pdata
, psize
) != 0) {
141 ctx
->error(ctx
, "no buffer %d", buffer
);
157 cmd_load_buffer_callback(struct client
*c
, void *data
)
165 * Event callback has already checked client is not dead and reduced
166 * its reference count. But tell it to exit.
168 c
->flags
|= CLIENT_EXIT
;
170 psize
= EVBUFFER_LENGTH(c
->stdin_event
->input
);
171 if (psize
== 0 || (pdata
= malloc(psize
+ 1)) == NULL
) {
175 bufferevent_read(c
->stdin_event
, pdata
, psize
);
178 limit
= options_get_number(&global_options
, "buffer-limit");
180 paste_add(&global_buffers
, pdata
, psize
, limit
);
181 else if (paste_replace(&global_buffers
, *buffer
, pdata
, psize
) != 0) {
182 /* No context so can't use server_client_msg_error. */
184 c
->stderr_event
->output
, "no buffer %d\n", *buffer
);
185 bufferevent_enable(c
->stderr_event
, EV_WRITE
);