w3mimgdisplay.c

   1 /* $Id: w3mimgdisplay.c,v 1.18 2003/07/13 16:20:42 ukai Exp $ */
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <ctype.h>
   5 #include <string.h>
   6 #include <sys/types.h>
   7 #include <unistd.h>
   8 #include "config.h"
   9 #include "w3mimg/w3mimg.h"
  10
  11 w3mimg_op *w_op;
  12 static char *background = NULL;
  13 static int offset_x = 0, offset_y = 0;
  14 static int defined_bg = 0, defined_x = 0, defined_y = 0, defined_test = 0;
  15 static int defined_debug = 0;
  16 static char *defined_size = NULL;
  17
  18 #define MAX_IMAGE 1000
  19 static W3MImage *imageBuf = NULL;
  20 static int maxImage = 0, maxAnim = 100, clearMargin = 0;
  21
  22 static void GetOption(int argc, char **argv);
  23 static void DrawImage(char *buf, int redraw);
  24 static void TermImage(void);
  25 static void ClearImage(char *buf);
  26
  27 int
  28 main(int argc, char **argv)
  29 {
  30     int len;
  31     char buf[1024 + 128];
  32 #ifdef W3MIMGDISPLAY_SETUID
  33     uid_t runner_uid = getuid();
  34     uid_t owner_uid = geteuid();
  35
  36     /* swap real and effective */
  37     setreuid(owner_uid, runner_uid);
  38 #endif
  39     GetOption(argc, argv);
  40     if (!defined_debug)
  41         freopen(DEV_NULL_PATH, "w", stderr);
  42
  43 #ifdef W3MIMGDISPLAY_SETUID
  44     /*
  45      * back real and effective
  46      * run w3mimg_open() in setuid privileges
  47      */
  48     setreuid(runner_uid, owner_uid);
  49 #endif
  50     w_op = w3mimg_open();
  51 #ifdef W3MIMGDISPLAY_SETUID
  52     /* make sure drop privileges now */
  53     setreuid(runner_uid, runner_uid);
  54 #endif
  55     if (w_op == NULL)
  56         exit(1);
  57     if (defined_x)
  58         w_op->offset_x = offset_x;
  59     if (defined_y)
  60         w_op->offset_y = offset_y;
  61
  62     w_op->max_anim = maxAnim;
  63     w_op->clear_margin = clearMargin;
  64
  65     if (defined_test) {
  66         printf("%d %d\n", w_op->width - w_op->offset_x,
  67                w_op->height - w_op->offset_y);
  68         exit(0);
  69     }
  70
  71     if (defined_size) {
  72         if (w_op->init(w_op)) {
  73             W3MImage img;
  74             int w, h;
  75             if (w_op->get_image_size(w_op, &img, defined_size, &w, &h))
  76                 printf("%d %d\n", w, h);
  77         }
  78         exit(0);
  79     }
  80
  81     w_op->set_background(w_op, background);
  82
  83     while (fgets(buf, sizeof(buf), stdin) != NULL) {
  84         if (!(isdigit(buf[0]) && buf[1] == ';')) {
  85             fputc('\n', stdout);
  86             fflush(stdout);
  87             continue;
  88         }
  89         len = strlen(buf);
  90         if (buf[len - 1] == '\n') {
  91             buf[--len] = '\0';
  92             if (buf[len - 1] == '\r')
  93                 buf[--len] = '\0';
  94         }
  95         /*
  96          * w3mimg protocol
  97          *  0  1  2 ....
  98          * +--+--+--+--+ ...... +--+--+
  99          * |op|; |args             |\n|
 100          * +--+--+--+--+ .......+--+--+
 101          *
 102          * args is separeted by ';'
 103          * op   args
 104          *  0;  params          draw image
 105          *  1;  params          redraw image
 106          *  2;  -none-          terminate drawing
 107          *  3;  -none-          sync drawing
 108          *  4;  -none-          nop, sync communication
 109          *                      response '\n'
 110          *  5;  path            get size of image,
 111          *                      response "<width> <height>\n"
 112          *  6;  params(6)       clear image
 113          *
 114          * params
 115          *      <n>;<x>;<y>;<w>;<h>;<sx>;<sy>;<sw>;<sh>;<path>
 116          * params(6)
 117          *      <x>;<y>;<w>;<h>
 118          *
 119          */
 120         switch (buf[0]) {
 121         case '0':
 122             DrawImage(&buf[2], 0);
 123             break;
 124         case '1':
 125             DrawImage(&buf[2], 1);
 126             break;
 127         case '2':
 128             TermImage();
 129             break;
 130         case '3':
 131             w_op->sync(w_op);
 132             break;
 133         case '4':
 134             fputs("\n", stdout);
 135             fflush(stdout);
 136             break;
 137         case '5':
 138             if (w_op->init(w_op)) {
 139                 W3MImage img;
 140                 int w, h;
 141                 if (w_op->get_image_size(w_op, &img, &buf[2], &w, &h)) {
 142                     fprintf(stdout, "%d %d\n", w, h);
 143                     fflush(stdout);
 144                 }
 145                 else {
 146                     fprintf(stdout, "\n");
 147                     fflush(stdout);
 148                 }
 149             }
 150             else {
 151                 fprintf(stdout, "\n");
 152                 fflush(stdout);
 153             }
 154             break;
 155         case '6':
 156             ClearImage(&buf[2]);
 157             break;
 158         }
 159     }
 160     TermImage();
 161     w_op->close(w_op);
 162     exit(0);
 163 }
 164
 165 static void
 166 GetOption(int argc, char **argv)
 167 {
 168     int i;
 169
 170     for (i = 1; i < argc; i++) {
 171         if (!strcmp("-bg", argv[i])) {
 172             if (++i >= argc)
 173                 exit(1);
 174             background = argv[i];
 175             defined_bg = 1;
 176         }
 177         else if (!strcmp("-x", argv[i])) {
 178             if (++i >= argc)
 179                 exit(1);
 180             offset_x = atoi(argv[i]);
 181             defined_x = 1;
 182         }
 183         else if (!strcmp("-y", argv[i])) {
 184             if (++i >= argc)
 185                 exit(1);
 186             offset_y = atoi(argv[i]);
 187             defined_y = 1;
 188         }
 189         else if (!strcmp("-test", argv[i])) {
 190             defined_test = 1;
 191         }
 192         else if (!strcmp("-anim", argv[i])) {
 193             if (++i >= argc)
 194                 exit(1);
 195             maxAnim = atoi(argv[i]);
 196         }
 197         else if (!strcmp("-margin", argv[i])) {
 198             if (++i >= argc)
 199                 exit(1);
 200             clearMargin = atoi(argv[i]);
 201             if (clearMargin < 0)
 202                 clearMargin = 0;
 203         }
 204         else if (!strcmp("-size", argv[i])) {
 205             if (++i >= argc)
 206                 exit(1);
 207             defined_size = argv[i];
 208         }
 209         else if (!strcmp("-debug", argv[i])) {
 210             defined_debug = 1;
 211         }
 212         else {
 213             exit(1);
 214         }
 215     }
 216 }
 217
 218 void
 219 DrawImage(char *buf, int redraw)
 220 {
 221     char *p = buf;
 222     int n = 0, x = 0, y = 0, w = 0, h = 0, sx = 0, sy = 0, sw = 0, sh = 0;
 223
 224     if (!p)
 225         return;
 226     for (; isdigit(*p); p++)
 227         n = 10 * n + (*p - '0');
 228     if (*(p++) != ';')
 229         return;
 230     for (; isdigit(*p); p++)
 231         x = 10 * x + (*p - '0');
 232     if (*(p++) != ';')
 233         return;
 234     for (; isdigit(*p); p++)
 235         y = 10 * y + (*p - '0');
 236     if (*(p++) != ';')
 237         return;
 238     for (; isdigit(*p); p++)
 239         w = 10 * w + (*p - '0');
 240     if (*(p++) != ';')
 241         return;
 242     for (; isdigit(*p); p++)
 243         h = 10 * h + (*p - '0');
 244     if (*(p++) != ';')
 245         return;
 246     for (; isdigit(*p); p++)
 247         sx = 10 * sx + (*p - '0');
 248     if (*(p++) != ';')
 249         return;
 250     for (; isdigit(*p); p++)
 251         sy = 10 * sy + (*p - '0');
 252     if (*(p++) != ';')
 253         return;
 254     for (; isdigit(*p); p++)
 255         sw = 10 * sw + (*p - '0');
 256     if (*(p++) != ';')
 257         return;
 258     for (; isdigit(*p); p++)
 259         sh = 10 * sh + (*p - '0');
 260     if (*(p++) != ';')
 261         return;
 262
 263     n--;
 264     if (n < 0 || n >= MAX_IMAGE)
 265         return;
 266     if (redraw) {
 267         if (!w_op->active(w_op) || n >= maxImage || !imageBuf[n].pixmap)
 268             return;
 269         goto draw_image;
 270     }
 271     w_op->init(w_op);
 272
 273     if (n >= maxImage) {
 274         int i = maxImage;
 275         maxImage = i ? (i * 2) : 2;
 276         if (maxImage > MAX_IMAGE)
 277             maxImage = MAX_IMAGE;
 278         else if (n >= maxImage)
 279             maxImage = n + 1;
 280         imageBuf = (W3MImage *) realloc((void *)imageBuf,
 281                                         sizeof(W3MImage) * maxImage);
 282         for (; i < maxImage; i++)
 283             imageBuf[i].pixmap = NULL;
 284     }
 285     if (imageBuf[n].pixmap) {
 286         w_op->free_image(w_op, &imageBuf[n]);
 287         imageBuf[n].pixmap = NULL;
 288     }
 289
 290     if (w_op->load_image(w_op, &imageBuf[n], p, w, h) == 0)
 291         imageBuf[n].pixmap = NULL;
 292
 293   draw_image:
 294     if (imageBuf[n].pixmap)
 295         w_op->show_image(w_op, &imageBuf[n], sx, sy, sw, sh, x, y);
 296 }
 297
 298 void
 299 TermImage(void)
 300 {
 301     w_op->finish(w_op);
 302     if (imageBuf) {
 303         int i;
 304         for (i = 0; i < maxImage; i++) {
 305             w_op->free_image(w_op, &imageBuf[i]);
 306         }
 307         free(imageBuf);
 308         imageBuf = NULL;
 309     }
 310     maxImage = 0;
 311 }
 312
 313 void
 314 ClearImage(char *buf)
 315 {
 316     char *p = buf;
 317     int x = 0, y = 0, w = 0, h = 0;
 318
 319     if (!p)
 320         return;
 321     for (; isdigit(*p); p++)
 322         x = 10 * x + (*p - '0');
 323     if (*(p++) != ';')
 324         return;
 325     for (; isdigit(*p); p++)
 326         y = 10 * y + (*p - '0');
 327     if (*(p++) != ';')
 328         return;
 329     for (; isdigit(*p); p++)
 330         w = 10 * w + (*p - '0');
 331     if (*(p++) != ';')
 332         return;
 333     for (; isdigit(*p); p++)
 334         h = 10 * h + (*p - '0');
 335
 336     w_op->clear(w_op,
 337                 x + offset_x - w_op->clear_margin,
 338                 y + offset_y - w_op->clear_margin,
 339                 w + w_op->clear_margin * 2, h + w_op->clear_margin * 2);
 340 }