util/wxcopy.c

   1 /* wxcopy.c- copy stdin or file into cutbuffer
   2  *
   3  *  Copyright (c) 1997-2003 Alfredo K. Kojima
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19
  20 #define PROG_VERSION "wxcopy 0.3"
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <errno.h>
  27 #include <X11/Xlib.h>
  28 #include <X11/Xatom.h>
  29
  30 #define LINESIZE        (4*1024)
  31 #define MAXDATA         (64*1024)
  32
  33 void help(char *progn)
  34 {
  35         printf("Usage: %s [OPTIONS] [FILE]\n", progn);
  36         puts("Copies data from FILE or stdin into X cut buffer.");
  37         puts("");
  38         puts("  -display <display>              display to use");
  39         puts("  --cutbuffer <number>            cutbuffer number to put data");
  40         puts("  --no-limit                      do not limit size of input data");
  41         puts("  --clear-selection               clears the current PRIMARY selection");
  42         puts("  --help                  display this help and exit");
  43         puts("  --version                       output version information and exit");
  44 }
  45
  46 static int errorHandler(Display * dpy, XErrorEvent * err)
  47 {
  48         /* ignore all errors */
  49         return 0;
  50 }
  51
  52 int main(int argc, char **argv)
  53 {
  54         Display *dpy;
  55         int i;
  56         int buffer = -1;
  57         char *filename = NULL;
  58         FILE *file = stdin;
  59         char *buf = NULL;
  60         char *display_name = "";
  61         int l = 0;
  62         int buf_len = 0;
  63         int limit_check = 1;
  64         int clear_selection = 0;
  65
  66         for (i = 1; i < argc; i++) {
  67                 if (argv[i][0] == '-') {
  68                         if (strcmp(argv[i], "--help") == 0) {
  69                                 help(argv[0]);
  70                                 exit(0);
  71                         } else if (strcmp(argv[i], "--version") == 0) {
  72                                 puts(PROG_VERSION);
  73                                 exit(0);
  74                         } else if (strcmp(argv[i], "-cutbuffer") == 0 || strcmp(argv[i], "--cutbuffer") == 0) {
  75                                 if (i < argc - 1) {
  76                                         i++;
  77                                         if (sscanf(argv[i], "%i", &buffer) != 1) {
  78                                                 fprintf(stderr, "%s: could not convert '%s' to int\n",
  79                                                         argv[0], argv[i]);
  80                                                 exit(1);
  81                                         }
  82                                         if (buffer < 0 || buffer > 7) {
  83                                                 fprintf(stderr, "%s: invalid buffer number %i\n", argv[0], buffer);
  84                                                 exit(1);
  85                                         }
  86                                 } else {
  87                                         printf("%s: missing argument for '%s'\n", argv[0], argv[i]);
  88                                         printf("Try '%s --help' for more information\n", argv[0]);
  89                                         exit(1);
  90                                 }
  91                         } else if (strcmp(argv[i], "-display") == 0) {
  92                                 if (i < argc - 1) {
  93                                         display_name = argv[++i];
  94                                 } else {
  95                                         printf("%s: missing argument for '%s'\n", argv[0], argv[i]);
  96                                         printf("Try '%s --help' for more information\n", argv[0]);
  97                                         exit(1);
  98                                 }
  99                         } else if (strcmp(argv[i], "-clearselection") == 0
 100                                    || strcmp(argv[i], "--clear-selection") == 0) {
 101                                 clear_selection = 1;
 102                         } else if (strcmp(argv[i], "-nolimit") == 0 || strcmp(argv[i], "--no-limit") == 0) {
 103                                 limit_check = 0;
 104                         } else {
 105                                 printf("%s: invalid argument '%s'\n", argv[0], argv[i]);
 106                                 printf("Try '%s --help' for more information\n", argv[0]);
 107                                 exit(1);
 108                         }
 109                 } else {
 110                         filename = argv[i];
 111                 }
 112         }
 113         if (filename) {
 114                 file = fopen(filename, "rb");
 115                 if (!file) {
 116                         char line[1024];
 117                         sprintf(line, "%s: could not open \"%s\"", argv[0], filename);
 118                         perror(line);
 119                         exit(1);
 120                 }
 121         }
 122
 123         dpy = XOpenDisplay(display_name);
 124         XSetErrorHandler(errorHandler);
 125         if (!dpy) {
 126                 fprintf(stderr, "%s: could not open display \"%s\"\n", argv[0], XDisplayName(display_name));
 127                 exit(1);
 128         }
 129
 130         if (buffer < 0) {
 131                 Atom *rootWinProps;
 132                 int exists[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
 133                 int i, count;
 134
 135                 /* Create missing CUT_BUFFERs */
 136                 rootWinProps = XListProperties(dpy, DefaultRootWindow(dpy), &count);
 137                 for (i = 0; i < count; i++) {
 138                         switch (rootWinProps[i]) {
 139                         case XA_CUT_BUFFER0:
 140                                 exists[0] = 1;
 141                                 break;
 142                         case XA_CUT_BUFFER1:
 143                                 exists[1] = 1;
 144                                 break;
 145                         case XA_CUT_BUFFER2:
 146                                 exists[2] = 1;
 147                                 break;
 148                         case XA_CUT_BUFFER3:
 149                                 exists[3] = 1;
 150                                 break;
 151                         case XA_CUT_BUFFER4:
 152                                 exists[4] = 1;
 153                                 break;
 154                         case XA_CUT_BUFFER5:
 155                                 exists[5] = 1;
 156                                 break;
 157                         case XA_CUT_BUFFER6:
 158                                 exists[6] = 1;
 159                                 break;
 160                         case XA_CUT_BUFFER7:
 161                                 exists[7] = 1;
 162                                 break;
 163                         default:
 164                                 break;
 165                         }
 166                 }
 167                 if (rootWinProps) {
 168                         XFree(rootWinProps);
 169                 }
 170                 for (i = 0; i < 8; i++) {
 171                         if (!exists[i]) {
 172                                 XStoreBuffer(dpy, "", 0, i);
 173                         }
 174                 }
 175
 176                 XRotateBuffers(dpy, 1);
 177                 buffer = 0;
 178         }
 179
 180         while (!feof(file)) {
 181                 char *nbuf;
 182                 char tmp[LINESIZE + 2];
 183                 int nl = 0;
 184
 185                 /*
 186                  * Use read() instead of fgets() to preserve NULLs, since
 187                  * especially since there's no reason to read one line at a time.
 188                  */
 189                 if ((nl = fread(tmp, 1, LINESIZE, file)) <= 0) {
 190                         break;
 191                 }
 192                 if (buf_len == 0) {
 193                         nbuf = malloc(buf_len = l + nl + 1);
 194                 } else if (buf_len < l + nl + 1) {
 195                         /*
 196                          * To avoid terrible performance on big input buffers,
 197                          * grow by doubling, not by the minimum needed for the
 198                          * current line.
 199                          */
 200                         buf_len = 2 * buf_len + nl + 1;
 201                         /* some realloc implementations don't do malloc if buf==NULL */
 202                         if (buf == NULL) {
 203                                 nbuf = malloc(buf_len);
 204                         } else {
 205                                 nbuf = realloc(buf, buf_len);
 206                         }
 207                 } else {
 208                         nbuf = buf;
 209                 }
 210                 if (!nbuf) {
 211                         fprintf(stderr, "%s: out of memory\n", argv[0]);
 212                         exit(1);
 213                 }
 214                 buf = nbuf;
 215                 /*
 216                  * Don't strcat, since it would make the algorithm n-squared.
 217                  * Don't use strcpy, since it stops on a NUL.
 218                  */
 219                 memcpy(buf + l, tmp, nl);
 220                 l += nl;
 221                 if (limit_check && l >= MAXDATA) {
 222                         fprintf
 223                             (stderr,
 224                              "%s: too much data in input - more than %d bytes\n"
 225                              "  use the -nolimit argument to remove the limit check.\n", argv[0], MAXDATA);
 226                         exit(1);
 227                 }
 228         }
 229
 230         if (clear_selection) {
 231                 XSetSelectionOwner(dpy, XA_PRIMARY, None, CurrentTime);
 232         }
 233         if (buf) {
 234                 XStoreBuffer(dpy, buf, l, buffer);
 235         }
 236         XFlush(dpy);
 237         XCloseDisplay(dpy);
 238         exit(buf == NULL || errno != 0);
 239 }