util/wxpaste.c

   1 /* wxpaste.c- paste contents of cutbuffer to stdout
   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., 51 Franklin St, Fifth Floor, Boston,
  18  *  MA 02110-1301, USA.
  19  */
  20
  21 #include "../config.h"
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <unistd.h>
  26 #include <string.h>
  27 #include <errno.h>
  28 #include <X11/Xlib.h>
  29 #include <X11/Xatom.h>
  30 #include <sys/time.h>
  31 #include <sys/types.h>
  32 #ifdef HAVE_SYS_SELECT_H
  33 # include <sys/select.h>
  34 #endif
  35
  36 #define MAXDATA         (4*1024*1024)
  37
  38 static const char *prog_name;
  39
  40 static void print_help(void)
  41 {
  42         printf("Usage: %s [OPTIONS] [FILE]\n", prog_name);
  43         puts("Copies data from X selection or cutbuffer to FILE or stdout.");
  44         puts("");
  45         puts("  -display display         display to use");
  46         puts("  --cutbuffer number       cutbuffer number to get data from");
  47         puts("  --selection [selection]  reads data from named selection instead of cutbuffer");
  48         puts("  -h, --help               display this help and exit");
  49         puts("  -v, --version            output version information and exit");
  50 }
  51
  52 static Time getTimestamp(Display * dpy, Window win)
  53 {
  54         XEvent ev;
  55
  56         /* So we do this trickery to get a time stamp:
  57          *
  58          * 1. Grab the server because we are paranoid and don't want to
  59          * get in a race with another instance of wxpaste being ran at the
  60          * same time.
  61          *
  62          * 2. Set a dummy property in our window.
  63          *
  64          * 3. Get the PropertyNotify event and get it's timestamp.
  65          *
  66          * 4. Ungrab the server.
  67          */
  68
  69         XSelectInput(dpy, win, PropertyChangeMask);
  70
  71         /* Generate a PropertyNotify event */
  72         XStoreName(dpy, win, "shit");
  73
  74         /* wait for the event */
  75         while (1) {
  76                 XNextEvent(dpy, &ev);
  77                 if (ev.type == PropertyNotify)
  78                         break;
  79         }
  80
  81         return ev.xproperty.time;
  82 }
  83
  84 static char *fetchSelection(Display *dpy, const char *selection)
  85 {
  86         Atom selatom = XInternAtom(dpy, selection, False);
  87         Atom clipatom = XInternAtom(dpy, "CLIPBOARD", False);
  88         Time now;
  89         XEvent ev;
  90         Window win;
  91         int ok = 0;
  92         struct timeval timeout;
  93         fd_set fdset;
  94
  95         win = XCreateSimpleWindow(dpy, DefaultRootWindow(dpy), 0, 0, 1, 1, 0, 0, 0);
  96         /*
  97          * The ICCCM says that we can't pass CurrentTime as the timestamp
  98          * for XConvertSelection(), but we don't have anything to use as
  99          * a timestamp...
 100          */
 101         now = getTimestamp(dpy, win);
 102
 103         XConvertSelection(dpy, selatom, XA_STRING, clipatom, win, now);
 104
 105         timeout.tv_sec = 1;
 106         timeout.tv_usec = 0;
 107
 108         /* wait for the convertion */
 109         while (0) {
 110                 int res;
 111
 112                 if (XPending(dpy) == 0) {
 113                         FD_ZERO(&fdset);
 114                         FD_SET(ConnectionNumber(dpy), &fdset);
 115                         res = select(ConnectionNumber(dpy) + 1, &fdset, NULL, NULL, &timeout);
 116                         if (res <= 0) {
 117                                 ok = 0;
 118                                 break;
 119                         }
 120                 }
 121                 if (res > 0 || XPending(dpy) > 0) {
 122                         XNextEvent(dpy, &ev);
 123                         if (ev.type == SelectionNotify && ev.xany.window == win) {
 124                                 ok = 1;
 125                                 break;
 126                         }
 127                 }
 128         }
 129
 130         /* if success, return the data */
 131         if (ok) {
 132                 Atom rtype;
 133                 int bits;
 134                 unsigned long len, bytes;
 135                 unsigned char *data;
 136
 137                 if (XGetWindowProperty(dpy, win, clipatom, 0, MAXDATA / 4, False,
 138                                        XA_STRING, &rtype, &bits, &len, &bytes, &data) != 0)
 139                         return NULL;
 140
 141                 if ((rtype != XA_STRING) || (bits != 8)) {
 142                         return NULL;
 143                 } else {
 144                         return (char *)data;
 145                 }
 146         }
 147         return NULL;
 148 }
 149
 150 int main(int argc, char **argv)
 151 {
 152         Display *dpy;
 153         int i, l;
 154         int buffer = 0;
 155         char *buf;
 156         int status;
 157         char *display_name = "";
 158         char *selection_name = NULL;
 159
 160         prog_name = argv[0];
 161         for (i = 1; i < argc; i++) {
 162                 if (argv[i][0] == '-') {
 163                         if (argv[i][1] == 'h' || strcmp(argv[i], "--help") == 0) {
 164                                 print_help();
 165                                 exit(0);
 166                         } else if (argv[i][1] == 'v' || strcmp(argv[i], "--version") == 0) {
 167                                 printf("%s (Window Maker %s)\n", prog_name, VERSION);
 168                                 exit(0);
 169                         } else if (strcmp(argv[i], "-selection") == 0 || strcmp(argv[i], "--selection") == 0) {
 170                                 if (i < argc - 1) {
 171                                         selection_name = argv[++i];
 172                                 } else {
 173                                         selection_name = "PRIMARY";
 174                                 }
 175                         } else if (strcmp(argv[i], "-display") == 0) {
 176                                 if (i < argc - 1) {
 177                                         display_name = argv[++i];
 178                                 } else {
 179                                         print_help();
 180                                         exit(0);
 181                                 }
 182                         } else if (strcmp(argv[i], "-cutbuffer") == 0 || strcmp(argv[i], "--cutbuffer") == 0) {
 183                                 if (i < argc - 1) {
 184                                         i++;
 185                                         if (sscanf(argv[i], "%i", &buffer) != 1) {
 186                                                 fprintf(stderr, "%s: could not convert \"%s\" to int\n",
 187                                                         prog_name, argv[i]);
 188                                                 exit(1);
 189                                         }
 190                                         if (buffer < 0 || buffer > 7) {
 191                                                 fprintf(stderr, "%s: invalid buffer number %i\n", prog_name, buffer);
 192                                                 exit(1);
 193                                         }
 194                                 } else {
 195                                         fprintf(stderr, "%s: invalid argument '%s'\n", prog_name, argv[i]);
 196                                         fprintf(stderr, "Try '%s --help' for more information.\n", prog_name);
 197                                         exit(1);
 198                                 }
 199                         }
 200                 } else {
 201                         fprintf(stderr, "%s: invalid argument '%s'\n", prog_name, argv[i]);
 202                         fprintf(stderr, "Try '%s --help' for more information.\n", prog_name);
 203                         exit(1);
 204                 }
 205         }
 206         dpy = XOpenDisplay(display_name);
 207         if (!dpy) {
 208                 fprintf(stderr, "%s: could not open display \"%s\"\n", prog_name, XDisplayName(display_name));
 209                 exit(1);
 210         }
 211
 212         if (selection_name) {
 213                 buf = fetchSelection(dpy, selection_name);
 214         } else {
 215                 buf = NULL;
 216         }
 217
 218         if (buf == NULL) {
 219                 buf = XFetchBuffer(dpy, &l, buffer);
 220         }
 221
 222         if (buf == NULL) {
 223                 status = 1;
 224         } else {
 225                 if (write(STDOUT_FILENO, buf, l) == -1)
 226                         status = errno;
 227                 else
 228                         status = 0;
 229         }
 230         XCloseDisplay(dpy);
 231         exit(status);
 232 }