gvusershape.c

Go to the documentation of this file.

/*************************************************************************
 * Copyright (c) 2011 AT&T Intellectual Property 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors: Details at https://graphviz.org
 *************************************************************************/
 
#include "config.h"
#include <assert.h>
#include <limits.h>
#include <math.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <util/gv_fopen.h>
 
#ifdef _WIN32
#include <windows.h>
#define GLOB_NOSPACE    1   /* Ran out of memory.  */
#define GLOB_ABORTED    2   /* Read error.  */
#define GLOB_NOMATCH    3   /* No matches found.  */
#define GLOB_NOSORT     4
#endif
 
#include <common/types.h>
#include <common/usershape.h>
#include <common/utils.h>
#include <gvc/gvplugin_loadimage.h>
#include <gvc/gvplugin.h>
#include <gvc/gvcint.h>
#include <gvc/gvcproc.h>
#include <util/agxbuf.h>
#include <util/alloc.h>
#include <util/gv_ctype.h>
#include <util/strview.h>
 
extern char *Gvimagepath;
extern char *HTTPServerEnVar;
extern shape_desc *find_user_shape(const char *);
 
static Dict_t *ImageDict;
 
typedef struct {
    char *template;
    size_t size;
    imagetype_t type;
    char *stringtype;
} knowntype_t;
 
#define HDRLEN 20
 
#define PNG_MAGIC  "\x89PNG\x0D\x0A\x1A\x0A"
#define PS_MAGIC   "%!PS-Adobe-"
#define BMP_MAGIC  "BM"
#define GIF_MAGIC  "GIF8"
#define JPEG_MAGIC "\xFF\xD8\xFF"
#define PDF_MAGIC  "%PDF-"
#define EPS_MAGIC  "\xC5\xD0\xD3\xC6"
#define XML_MAGIC  "<?xml"
#define SVG_MAGIC  "<svg"
#define RIFF_MAGIC "RIFF"
#define WEBP_MAGIC "WEBP"
#define ICO_MAGIC  "\x00\x00\x01\x00"
 
static knowntype_t knowntypes[] = {
    { PNG_MAGIC,  sizeof(PNG_MAGIC)-1,   FT_PNG,  "png",  },
    { PS_MAGIC,   sizeof(PS_MAGIC)-1,    FT_PS,   "ps",   },
    { BMP_MAGIC,  sizeof(BMP_MAGIC)-1,   FT_BMP,  "bmp",  },
    { GIF_MAGIC,  sizeof(GIF_MAGIC)-1,   FT_GIF,  "gif",  },
    { JPEG_MAGIC, sizeof(JPEG_MAGIC)-1,  FT_JPEG, "jpeg", },
    { PDF_MAGIC,  sizeof(PDF_MAGIC)-1,   FT_PDF,  "pdf",  },
    { EPS_MAGIC,  sizeof(EPS_MAGIC)-1,   FT_EPS,  "eps",  },
    { XML_MAGIC,  sizeof(XML_MAGIC)-1,   FT_XML,  "xml",  },
    { RIFF_MAGIC, sizeof(RIFF_MAGIC)-1,  FT_RIFF, "riff", },
    { ICO_MAGIC,  sizeof(ICO_MAGIC)-1,   FT_ICO,  "ico",  },
};
 
static imagetype_t imagetype(usershape_t *us) {
    char header[HDRLEN] = {0};
 
    if (us->f && fread(header, 1, HDRLEN, us->f) == HDRLEN) {
        for (size_t i = 0; i < sizeof(knowntypes) / sizeof(knowntype_t); i++) {
            if (!memcmp (header, knowntypes[i].template, knowntypes[i].size)) {
                us->stringtype = knowntypes[i].stringtype;
                us->type = knowntypes[i].type;
                if (us->type == FT_XML) {
                    // if we did not see the closing of the XML declaration, scan for it
                    if (memchr(header, '>', HDRLEN) == NULL) {
                        while (true) {
                            int c = fgetc(us->f);
                            if (c == EOF) {
                                return us->type;
                            } else if (c == '>') {
                                break;
                            }
                        }
                    }
                    /* check for SVG in case of XML */
                    char tag[sizeof(SVG_MAGIC) - 1] = {0};
                    if (fread(tag, 1, sizeof(tag), us->f) != sizeof(tag)) {
                        return us->type;
                    }
                    while (true) {
                        if (memcmp(tag, SVG_MAGIC, sizeof(SVG_MAGIC) - 1) == 0) {
                            us->stringtype = "svg";
                            return (us->type = FT_SVG);
                        }
                        int c = fgetc(us->f);
                        if (c == EOF) {
                            return us->type;
                        }
                        memmove(&tag[0], &tag[1], sizeof(tag) - 1);
                        tag[sizeof(tag) - 1] = (char)c;
                    }
                }
                else if (us->type == FT_RIFF) {
                    /* check for WEBP in case of RIFF */
                    if (!memcmp(header+8, WEBP_MAGIC, sizeof(WEBP_MAGIC)-1)) {
                        us->stringtype = "webp";
                        return (us->type = FT_WEBP);
                    }
                }
                return us->type;
            }
        }
    }
 
    us->stringtype = "(lib)";
    us->type = FT_NULL;
 
    return FT_NULL;
}
    
static bool get_int_lsb_first(FILE *f, size_t sz, int *val) {
    int ch;
 
    unsigned value = 0;
    for (size_t i = 0; i < sz; i++) {
        ch = fgetc(f);
        if (feof(f))
            return false;
        value |= (unsigned)ch << 8 * i;
    }
    if (value > INT_MAX) {
        return false;
    }
    *val = (int)value;
    return true;
}
        
static bool get_int_msb_first(FILE *f, size_t sz, int *val) {
    int ch;
 
    unsigned value = 0;
    for (size_t i = 0; i < sz; i++) {
        ch = fgetc(f);
        if (feof(f))
            return false;
        value <<= 8;
        value |= (unsigned)ch;
    }
    if (value > INT_MAX) {
        return false;
    }
    *val = (int)value;
    return true;
}
 
static double svg_units_convert(double n, char *u) {
    if (strcmp(u, "in") == 0)
        return round(n * POINTS_PER_INCH);
    if (strcmp(u, "px") == 0)
        return round(n * POINTS_PER_INCH / 96);
    if (strcmp(u, "pc") == 0)
        return round(n * POINTS_PER_INCH / 6); 
    if (strcmp(u, "pt") == 0 || strcmp(u, "\"") == 0)   /* ugly!!  - if there are no inits then the %2s get the trailing '"' */
        return round(n);
    if (strcmp(u, "cm") == 0)
        return round(n * POINTS_PER_CM);
    if (strcmp(u, "mm") == 0)
        return round(n * POINTS_PER_MM);
    return 0;
}
 
typedef struct {
  strview_t key;
  strview_t value;
} match_t;
 
static int find_attribute(const char *s, match_t *result) {
 
  // look for an attribute string matching ([a-z][a-zA-Z]*)="([^"]*)"
  for (size_t i = 0; s[i] != '\0'; ) {
    if (s[i] >= 'a' && s[i] <= 'z') {
      result->key.data = &s[i];
      result->key.size = 1;
      ++i;
      while ((s[i] >= 'a' && s[i] <= 'z') || (s[i] >= 'A' && s[i] <= 'Z')) {
        ++i;
        ++result->key.size;
      }
      if (s[i] == '=' && s[i + 1] == '"') {
        i += 2;
        result->value.data = &s[i];
        result->value.size = 0;
        while (s[i] != '"' && s[i] != '\0') {
          ++i;
          ++result->value.size;
        }
        if (s[i] == '"') {
          // found a valid attribute
          return 0;
        }
      }
    } else {
      ++i;
    }
  }
 
  // no attribute found
  return -1;
}
 
static void svg_size (usershape_t *us)
{
    double w = 0, h = 0;
    double n, x0, y0, x1, y1;
    char u[10];
    agxbuf line = {0};
    bool eof = false;
    bool wFlag = false, hFlag = false;
 
    rewind(us->f);
    while (!eof && (!wFlag || !hFlag)) {
        // read next line
        while (true) {
            int c = fgetc(us->f);
            if (c == EOF) {
                eof = true;
                break;
            } else if (c == '\n') {
                break;
            }
            agxbputc(&line, (char)c);
        }
 
        const char *re_string = agxbuse(&line);
        match_t match;
        while (find_attribute(re_string, &match) == 0) {
            re_string = match.value.data + match.value.size + 1;
 
            if (strview_str_eq(match.key, "width")) {
                char *value = strview_str(match.value);
                if (sscanf(value, "%lf%2s", &n, u) == 2) {
                    w = svg_units_convert(n, u);
                    wFlag = true;
                }
                else if (sscanf(value, "%lf", &n) == 1) {
                    w = svg_units_convert(n, "pt");
                    wFlag = true;
                }
                free(value);
                if (hFlag)
                    break;
            }
            else if (strview_str_eq(match.key, "height")) {
                char *value = strview_str(match.value);
                if (sscanf(value, "%lf%2s", &n, u) == 2) {
                    h = svg_units_convert(n, u);
                    hFlag = true;
                }
                else if (sscanf(value, "%lf", &n) == 1) {
                    h = svg_units_convert(n, "pt");
                    hFlag = true;
                }
                free(value);
                if (wFlag)
                    break;
            }
            else if (strview_str_eq(match.key, "viewBox")) {
                char *value = strview_str(match.value);
                if (sscanf(value, "%lf %lf %lf %lf", &x0, &y0, &x1, &y1) == 4) {
                    w = x1 - x0 + 1;
                    h = y1 - y0 + 1;
                    wFlag = true;
                    hFlag = true;
                    free(value);
                    break;
                }
                free(value);
            }
        }
    }
    us->dpi = 0;
    assert(w >= 0 && w <= INT_MAX);
    us->w = (int)w;
    assert(h >= 0 && h <= INT_MAX);
    us->h = (int)h;
    agxbfree(&line);
}
 
static void png_size (usershape_t *us)
{
    int w, h;
 
    us->dpi = 0;
    fseek(us->f, 16, SEEK_SET);
    if (get_int_msb_first(us->f, 4, &w) && get_int_msb_first(us->f, 4, &h)) {
        us->w = w;
        us->h = h;
    }
}
 
static void ico_size (usershape_t *us)
{
    int w, h;
 
    us->dpi = 0;
    fseek(us->f, 6, SEEK_SET);
    if (get_int_msb_first(us->f, 1, &w) && get_int_msb_first(us->f, 1, &h)) {
        us->w = w;
        us->h = h;
    }
}
 
static void webp_size (usershape_t *us)
{
    int w, h;
 
    us->dpi = 0;
    fseek(us->f, 15, SEEK_SET);
    if (fgetc(us->f) == 'X') { //VP8X
        fseek(us->f, 24, SEEK_SET);
        if (get_int_lsb_first(us->f, 4, &w) && get_int_lsb_first(us->f, 4, &h)) {
            us->w = w;
            us->h = h;
        }
    }
    else { //VP8
        fseek(us->f, 26, SEEK_SET);
        if (get_int_lsb_first(us->f, 2, &w) && get_int_lsb_first(us->f, 2, &h)) {
            us->w = w;
            us->h = h;
        }
    }
}
 
static void gif_size (usershape_t *us)
{
    int w, h;
 
    us->dpi = 0;
    fseek(us->f, 6, SEEK_SET);
    if (get_int_lsb_first(us->f, 2, &w) && get_int_lsb_first(us->f, 2, &h)) {
        us->w = w;
        us->h = h;
    }
}
 
static void bmp_size (usershape_t *us) {
    int size_x_msw, size_x_lsw, size_y_msw, size_y_lsw;
 
    us->dpi = 0;
    fseek (us->f, 16, SEEK_SET);
    if ( get_int_lsb_first (us->f, 2, &size_x_msw) &&
         get_int_lsb_first (us->f, 2, &size_x_lsw) &&
         get_int_lsb_first (us->f, 2, &size_y_msw) &&
         get_int_lsb_first (us->f, 2, &size_y_lsw) ) {
        us->w = size_x_msw << 16 | size_x_lsw;
        us->h = size_y_msw << 16 | size_y_lsw;
    }
}
 
static void jpeg_size (usershape_t *us) {
    int marker, length, size_x, size_y;
 
    /* These are the markers that follow 0xff in the file.
     * Other markers implicitly have a 2-byte length field that follows.
     */
    static const unsigned char standalone_markers[] = {
        0x01,                       /* Temporary */
        0xd0, 0xd1, 0xd2, 0xd3,     /* Reset */
            0xd4, 0xd5, 0xd6,
            0xd7,
        0xd8,                       /* Start of image */
        0xd9,                       /* End of image */
    };
 
    us->dpi = 0;
    while (true) {
        /* Now we must be at a 0xff or at a series of 0xff's.
         * If that is not the case, or if we're at EOF, then there's
         * a parsing error.
         */
        if (! get_int_msb_first (us->f, 1, &marker))
            return;
 
        if (marker == 0xff)
            continue;
 
        /* Ok.. marker now read. If it is not a stand-alone marker,
         * then continue. If it's a Start Of Frame (0xc?), then we're there.
         * If it's another marker with a length field, then skip ahead
         * over that length field.
         */
 
        /* A stand-alone... */
        if (memchr(standalone_markers, marker, sizeof(standalone_markers)))
            continue;
 
        /* Incase of a 0xc0 marker: */
        if (marker == 0xc0) {
            /* Skip length and 2 lengths. */
            if (fseek(us->f, 3, SEEK_CUR) == 0 &&
                 get_int_msb_first (us->f, 2, &size_x) &&
                 get_int_msb_first (us->f, 2, &size_y) ) {
 
            /* Store length. */
                us->h = size_x;
                us->w = size_y;
            }
            return;
        }
 
        /* Incase of a 0xc2 marker: */
        if (marker == 0xc2) {
            /* Skip length and one more byte */
            if (fseek(us->f, 3, SEEK_CUR) != 0)
                return;
 
            /* Get length and store. */
            if ( get_int_msb_first (us->f, 2, &size_x) &&
                 get_int_msb_first (us->f, 2, &size_y) ) {
                us->h = size_x;
                us->w = size_y;
            }
            return;
        }
 
        /* Any other marker is assumed to be followed by 2 bytes length. */
        if (! get_int_msb_first (us->f, 2, &length))
            return;
 
        fseek (us->f, length - 2, SEEK_CUR);
    }
}
 
static void ps_size (usershape_t *us)
{
    char line[BUFSIZ];
    int lx, ly, ux, uy;
    char* linep;
 
    us->dpi = 72;
    rewind(us->f);
    bool saw_bb = false;
    while (fgets(line, sizeof(line), us->f)) {
        /* PostScript accepts \r as EOL, so using fgets () and looking for a
         * bounding box comment at the beginning doesn't work in this case. 
         * As a heuristic, we first search for a bounding box comment in line.
         * This obviously fails if not all of the numbers make it into the
         * current buffer. This shouldn't be a problem, as the comment is
         * typically near the beginning, and so should be read within the first
         * BUFSIZ bytes (even on Windows where this is 512).
         */
        if (!(linep = strstr (line, "%%BoundingBox:")))
            continue;
        if (sscanf (linep, "%%%%BoundingBox: %d %d %d %d", &lx, &ly, &ux, &uy) == 4) {
            saw_bb = true;
            break;
        }
    }
    if (saw_bb) {
        us->x = lx;
        us->y = ly;
        us->w = ux - lx;
        us->h = uy - ly;
    }
}
 
#define KEY "/MediaBox"
 
typedef struct {
    char* s;
    char* buf;
    FILE* fp;
} stream_t;
 
static char nxtc(stream_t *str) {
    if (fgets(str->buf, BUFSIZ, str->fp)) {
        str->s = str->buf;
        return *(str->s);
    }
    return '\0';
        
}
 
#define strc(x) (*(x->s)?*(x->s):nxtc(x))
#define stradv(x) (x->s++)
 
static void
skipWS (stream_t* str)
{
    char c;
    while ((c = strc(str))) {
        if (gv_isspace(c)) stradv(str);
        else return;
    }
}
 
static int
scanNum (char* tok, double* dp)
{
    char* endp;
    double d = strtod(tok, &endp);
 
    if (tok == endp) return 1;
    *dp = d;
    return 0;
}
 
static void
getNum (stream_t* str, char* buf)
{
    int len = 0;
    char c;
    skipWS(str);
    while ((c = strc(str)) && (gv_isdigit(c) || (c == '.'))) {
        buf[len++] = c;
        stradv(str);
        if (len == BUFSIZ-1) break;
    }
    buf[len] = '\0';
 
    return;
}
 
static int
boxof (stream_t* str, boxf* bp)
{
        char tok[BUFSIZ];
 
        skipWS(str);
        if (strc(str) != '[') return 1;
        stradv(str);
        getNum(str, tok);
        if (scanNum(tok,&bp->LL.x)) return 1;
        getNum(str, tok);
        if (scanNum(tok,&bp->LL.y)) return 1;
        getNum(str, tok);
        if (scanNum(tok,&bp->UR.x)) return 1;
        getNum(str, tok);
        if (scanNum(tok,&bp->UR.y)) return 1;
        return 0;
}
 
static int
bboxPDF (FILE* fp, boxf* bp)
{
        stream_t str;
        char* s;
        char buf[BUFSIZ];
        while (fgets(buf, BUFSIZ, fp)) {
                if ((s = strstr(buf,KEY))) {
                        str.buf = buf;
                        str.s = s+(sizeof(KEY)-1);
                        str.fp = fp;
                        return boxof(&str,bp);
                } 
        } 
        return 1;
}
 
static void pdf_size (usershape_t *us)
{
    boxf bb;
 
    us->dpi = 0;
    rewind(us->f);
    if ( ! bboxPDF (us->f, &bb)) {
        us->x = bb.LL.x;
        us->y = bb.LL.y;
        us->w = bb.UR.x - bb.LL.x;
        us->h = bb.UR.y - bb.LL.y;
    }
}
 
static void usershape_close(void *p) {
    usershape_t *us = p;
 
    if (us->f)
        fclose(us->f);
    if (us->data && us->datafree)
        us->datafree(us);
    free (us);
}
 
static Dtdisc_t ImageDictDisc = {
    .key = offsetof(usershape_t, name),
    .size = -1,
    .freef = usershape_close,
};
 
usershape_t *gvusershape_find(const char *name)
{
    usershape_t *us;
 
    assert(name);
    assert(name[0]);
 
    if (!ImageDict)
        return NULL;
 
    us = dtmatch(ImageDict, name);
    return us;
}
 
#define MAX_USERSHAPE_FILES_OPEN 50
bool gvusershape_file_access(usershape_t *us)
{
    static int usershape_files_open_cnt;
    const char *fn;
 
    assert(us);
    assert(us->name);
    assert(us->name[0]);
 
    if (us->f)
        rewind(us->f);
    else {
        if (! (fn = safefile(us->name))) {
            agwarningf("Filename \"%s\" is unsafe\n", us->name);
            return false;
        }
        us->f = gv_fopen(fn, "rb");
        if (us->f == NULL) {
            agwarningf("%s while opening %s\n", strerror(errno), fn);
            return false;
        }
        if (usershape_files_open_cnt >= MAX_USERSHAPE_FILES_OPEN)
            us->nocache = true;
        else
            usershape_files_open_cnt++;
    }
    assert(us->f);
    return true;
}
 
void gvusershape_file_release(usershape_t *us)
{
    if (us->nocache) {
        if (us->f) {
            fclose(us->f);
            us->f = NULL;
        }
    }
}
 
static void freeUsershape (usershape_t* us)
{
    if (us->name) agstrfree(0, us->name);
    free (us);
}
 
static usershape_t *gvusershape_open (const char *name)
{
    usershape_t *us;
 
    assert(name);
 
    if (!ImageDict)
        ImageDict = dtopen(&ImageDictDisc, Dttree);
 
    if (! (us = gvusershape_find(name))) {
        us = gv_alloc(sizeof(usershape_t));
 
        us->name = agstrdup(0, name);
        if (!gvusershape_file_access(us)) {
            freeUsershape (us);
            return NULL;
        }
 
        assert(us->f);
 
        switch(imagetype(us)) {
            case FT_NULL:
                if (!(us->data = find_user_shape(us->name))) {
                    agwarningf("\"%s\" was not found as a file or as a shape library member\n", us->name);
                    freeUsershape (us);
                    return NULL;
                }
                break;
            case FT_GIF:
                gif_size(us);
                break;
            case FT_PNG:
                png_size(us);
                break;
            case FT_BMP:
                bmp_size(us);
                break;
            case FT_JPEG:
                jpeg_size(us);
                break;
            case FT_PS:
                ps_size(us);
                break;
            case FT_WEBP:
                webp_size(us);
                break;
            case FT_SVG:
                svg_size(us);
                break;
            case FT_PDF:
                pdf_size(us);
                break;
            case FT_ICO:
                ico_size(us);
                break;
            case FT_EPS:   /* no eps_size code available */
            default:
                break;
        }
        gvusershape_file_release(us);
        dtinsert(ImageDict, us);
        return us;
    }
    gvusershape_file_release(us);
    return us;
}
 
/* gvusershape_size_dpi:
 * Return image size in points.
 */
point 
gvusershape_size_dpi (usershape_t* us, pointf dpi)
{
    point rv;
 
    if (!us) {
        rv.x = rv.y = -1;
    }
    else {
        if (us->dpi != 0) {
            dpi.x = dpi.y = us->dpi;
        }
        rv.x = us->w * POINTS_PER_INCH / dpi.x;
        rv.y = us->h * POINTS_PER_INCH / dpi.y;
    }
    return rv;
}
 
/* gvusershape_size:
 * Loads user image from file name if not already loaded.
 * Return image size in points.
 */
point gvusershape_size(graph_t * g, char *name)
{
    point rv;
    pointf dpi;
    static char* oldpath;
    usershape_t* us;
 
    /* no shape file, no shape size */
    if (!name || (*name == '\0')) {
        rv.x = rv.y = -1;
        return rv;
    }
 
    if (!HTTPServerEnVar && (oldpath != Gvimagepath)) {
        oldpath = Gvimagepath;
        if (ImageDict) {
            dtclose(ImageDict);
            ImageDict = NULL;
        }
    }
 
    if ((dpi.y = GD_drawing(g)->dpi) >= 1.0)
        dpi.x = dpi.y;
    else
        dpi.x = dpi.y = DEFAULT_DPI;
 
    us = gvusershape_open (name);
    rv = gvusershape_size_dpi (us, dpi);
    return rv;
}