diff options
Diffstat (limited to 'cqrlogo.c')
| -rw-r--r-- | cqrlogo.c | 324 |
1 files changed, 20 insertions, 304 deletions
@@ -5,285 +5,22 @@ * of the GNU General Public License, incorporated herein by reference. */ -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <inttypes.h> -#include <regex.h> - -#include <iniparser.h> -#include <png.h> -#include <zlib.h> -#include <qrencode.h> - -#if HAVE_FCGI -#include <fcgi_stdio.h> -#endif - #include "config.h" #include "version.h" -/* define structs and functions */ #include "cqrlogo.h" -#define URLPATTERN "^[hH][tT][tT][pP][sS]\\?://%s/" -#define TEXTSTOLEN "This QR Code has been stolen from http%s://%s/!" - -#if defined PNG_TEXT_SUPPORTED && PNG_ENABLE_TEXT -/*** add_png_text ***/ -png_text * add_png_text(png_text *pngtext, unsigned int *textcount, char *key, char *text) { - pngtext = realloc(pngtext, ((*textcount) + 1) * sizeof(png_text)); - - pngtext[*textcount].compression = PNG_TEXT_COMPRESSION_zTXt; - pngtext[*textcount].key = key; - pngtext[*textcount].text = text; - - (*textcount)++; - return pngtext; -} -#endif - -#if HAVE_FCGI -/*** png_write_stdout ***/ -void png_write_stdout(png_structp png_ptr, png_bytep data, png_size_t length) { - if (length != fwrite(data, 1, length, png_get_io_ptr(png_ptr))) - png_error(png_ptr, "Write Error"); -} - -/*** png_flush_stdout ***/ -void png_flush_stdout(png_structp png_ptr) { - fflush(stdout); -} -#endif - -/*** generate_png ***/ -int generate_png (struct bitmap_t *bitmap, const char *uri) { - png_structp png_ptr = NULL; - png_infop info_ptr = NULL; - png_byte ** row_pointers = NULL; - unsigned int x, y; - uint8_t bit, byte; - - if ((png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL)) == NULL) - return 1; - - if ((info_ptr = png_create_info_struct (png_ptr)) == NULL || - (setjmp (png_jmpbuf (png_ptr)))) { - png_destroy_write_struct (&png_ptr, &info_ptr); - return 1; - } - - png_set_IHDR (png_ptr, info_ptr, bitmap->width, bitmap->height, 1 /* depth */, - PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT); - - /* use best compression */ - png_set_compression_level(png_ptr, Z_BEST_COMPRESSION); - - /* use compression strategy filtered - * this way pngcrush can not optimize any more */ - png_set_compression_strategy(png_ptr, Z_FILTERED); - -#if defined PNG_TEXT_SUPPORTED && PNG_ENABLE_TEXT - unsigned int textcount = 0; - png_text *pngtext = NULL; - - pngtext = add_png_text(pngtext, &textcount, "comment", "QR-Code created by cqrlogo - https://github.com/eworm-de/cqrlogo"); -# if PNG_ENABLE_TEXT_REFERER - pngtext = add_png_text(pngtext, &textcount, "referer", (char *)uri); -# endif - -# if PNG_ENABLE_TEXT_VERSIONS -#if HAVE_FCGI -# define VERSIONSTR VERSION " (FastCGI) (" __DATE__ ", " __TIME__ ")" -#else -# define VERSIONSTR VERSION " (CGI) (" __DATE__ ", " __TIME__ ")" -#endif -# define LIBSSTR "libqrencode %s, libpng %s, zlib %s" - char *libsstr, *qrver = QRcode_APIVersionString(); - - libsstr = malloc(sizeof(LIBSSTR) + strlen(qrver) + strlen(png_libpng_ver) + strlen(zlib_version)); - sprintf(libsstr, LIBSSTR, qrver, png_libpng_ver, zlib_version); - - pngtext = add_png_text(pngtext, &textcount, "version", VERSIONSTR); - pngtext = add_png_text(pngtext, &textcount, "libs", libsstr); -# endif - - png_set_text(png_ptr, info_ptr, pngtext, textcount); - png_free (png_ptr, pngtext); -# if PNG_ENABLE_TEXT_VERSIONS - free(libsstr); -# endif -#endif - - row_pointers = png_malloc (png_ptr, bitmap->height * sizeof (png_byte *)); - for (y = 0; y < bitmap->height; ++y) { - /* we need to round up, need a complete byte for less than eight bits */ - row_pointers[y] = png_malloc (png_ptr, (sizeof(uint8_t) * bitmap->width + 7) / 8); - for (x = 0; x < bitmap->width; ++x) { - /* bit are written in reverse order! */ - bit = 7 - (x % 8); - byte = x / 8; - if (bitmap->pixel[y * bitmap->width + x]) - row_pointers[y][byte] |= 1 << (bit); - else - row_pointers[y][byte] &= ~(1 << (bit)); - } - } - -#if HAVE_FCGI - /* with fastcgi we can not just open stdout for writing... - * define a write function instead */ - png_set_write_fn(png_ptr, (png_voidp)stdout, png_write_stdout, png_flush_stdout); -#else - png_init_io (png_ptr, stdout); -#endif - - png_set_rows (png_ptr, info_ptr, row_pointers); - png_write_png (png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL); - - for (y = 0; y < bitmap->height; ++y) - png_free (png_ptr, row_pointers[y]); - png_free (png_ptr, row_pointers); - png_destroy_write_struct(&png_ptr, &info_ptr); - - return 0; -} - -/*** bitmap_new ***/ -struct bitmap_t * bitmap_new(int width, int height) { - struct bitmap_t *bitmap; - - if ((bitmap = malloc(sizeof(struct bitmap_t))) == NULL) - return NULL; - - bitmap->width = width; - bitmap->height = height; - if ((bitmap->pixel = malloc(width * height * sizeof(uint8_t))) == NULL) { - free(bitmap); - return NULL; - } - - /* initialize with white */ - memset(bitmap->pixel, 0xff, width * height); - - return bitmap; -} - -/*** bitmap_free ***/ -void bitmap_free(struct bitmap_t * bitmap) { - free(bitmap->pixel); - free(bitmap); -} - -/*** encode_qrcode ***/ -struct bitmap_t * encode_qrcode (const char *text, unsigned int scale, - unsigned int border, unsigned int level) { - QRcode *qrcode; - struct bitmap_t *bitmap, *scaled; - int i, j, k, l; - unsigned char *data; - - qrcode = QRcode_encodeString8bit(text, 0, level); - - /* this happens if the string is too long - * possibly we have an URL (referer) that is too long, so the code - * automatically falls back to http_server (see main()) */ - if (qrcode == NULL) - return NULL; - - data = qrcode->data; - - /* wirte QR code to bitmap */ - if ((bitmap = bitmap_new(qrcode->width + border * 2, qrcode->width + border * 2)) == NULL) - return NULL; - for (i = border; i < qrcode->width + border; i++) - for (j = border; j < qrcode->width + border; j++) { - bitmap->pixel[i * (qrcode->width + border * 2) + j] = !(*data & 0x1) * 0xff; - data++; - } - - QRcode_free(qrcode); - - if (scale == 1) - return bitmap; - - /* scale bitmap */ - if ((scaled = bitmap_new(bitmap->width * scale, bitmap->height * scale)) == NULL) - return NULL; - for (i = 0; i < bitmap->height; i++) - for (j = 0; j < bitmap->width; j++) - for (k = 0; k < scale; k++) - for (l = 0; l < scale; l++) - scaled->pixel[i * bitmap->width * scale * scale + k * bitmap->width * scale + j * scale + l] = - bitmap->pixel[i * bitmap->width + j]; - - - bitmap_free(bitmap); - - return scaled; -} - -/*** get_query_value ***/ -unsigned int get_query_value(const char *query_string, const char *pattern, - unsigned int value, unsigned int min, unsigned int max) { - char *match = NULL, *newpattern = NULL; - unsigned int length; - int tmp = -1; - - newpattern = strdup(pattern); - - length = strlen(newpattern); - /* length is without null termination, allocacte 4 bytes so we - * have "=", "%u" and null termination */ - newpattern = realloc(newpattern, length + 4); - sprintf(newpattern + length, "="); - - if ((match = strstr(query_string, newpattern)) != NULL) { - sprintf(newpattern + length + 1, "%%u"); - - if ((sscanf(match, newpattern, &tmp)) > 0) - if (tmp >= min && tmp <= max) - value = tmp; - } - - free(newpattern); - - return value; -} - -/*** get_ini_value ***/ -unsigned int get_ini_value(dictionary * ini, uint8_t type, const char * section, const char * parameter, - unsigned int value, unsigned int min, unsigned int max) { - char * key; - unsigned int tmp; - - key = malloc(strlen(section) + strlen(parameter) + 2); - sprintf(key, "%s:%s", section, parameter); - - if (type) - tmp = iniparser_getint(ini, key, value); - else - tmp = iniparser_getboolean(ini, key, value); - - if (tmp >= min && tmp <= max) - value = tmp; - - free(key); - - return value; -} - /*** main ***/ int main(int argc, char **argv) { - dictionary * ini; const char * http_referer, * server_name, * query_string, * uri; char * uri_server_name, * uri_png, * pattern, * stolen; regex_t preg; regmatch_t pmatch[1]; - uint8_t https, overwrite; + uint8_t https; + struct png_t * png; struct bitmap_t * bitmap; - unsigned int scale, border, level; + struct cqrconf_t cqrconf; #if HAVE_FCGI /* loop for requests */ @@ -296,11 +33,12 @@ int main(int argc, char **argv) { stolen = NULL; https = 0; - overwrite = ALLOW_OVERWRITE; - scale = QRCODE_SCALE; - border = QRCODE_BORDER; - level = QRCODE_LEVEL; + /* these default values are defined in config.h */ + cqrconf.scale = QRCODE_SCALE; + cqrconf.border = QRCODE_BORDER; + cqrconf.level = QRCODE_LEVEL; + cqrconf.overwrite = ALLOW_OVERWRITE; /* check if we have environment variables from CGI */ if ((server_name = getenv("SERVER_NAME")) == NULL) { @@ -344,42 +82,14 @@ int main(int argc, char **argv) { uri = uri_server_name; } - /* parse config file */ - if ((ini = iniparser_load(CONFIGFILE)) == NULL) { - fprintf(stderr, "cannot parse file " CONFIGFILE ", continue anyway\n"); - /* continue anyway, there is nothing essential in the config file */ - } else { - scale = get_ini_value(ini, 1, "general", "scale", scale, 1, QRCODE_MAX_SCALE); - border = get_ini_value(ini, 1, "general", "border", border, 0, QRCODE_MAX_BORDER); - level = get_ini_value(ini, 1, "general", "level", level, QR_ECLEVEL_L, QR_ECLEVEL_H); - overwrite = get_ini_value(ini, 0, "general", "allow overwrite", overwrite, 0, 1); - - scale = get_ini_value(ini, 1, server_name, "scale", scale, 1, QRCODE_MAX_SCALE); - border = get_ini_value(ini, 1, server_name, "border", border, 0, QRCODE_MAX_BORDER); - level = get_ini_value(ini, 1, server_name, "level", level, QR_ECLEVEL_L, QR_ECLEVEL_H); - overwrite = get_ini_value(ini, 0, server_name, "allow overwrite", overwrite, 0, 1); - - /* done reading config file, free */ - iniparser_freedict(ini); - } - - /* get query string and read settings */ - if (overwrite && (query_string = getenv("QUERY_STRING")) != NULL) { - /* do we have a special scale? */ - scale = get_query_value(query_string, "scale", scale, 1, QRCODE_MAX_SCALE); - - /* width of the border? */ - border = get_query_value(query_string, "border", border, 0, QRCODE_MAX_BORDER); - - /* error correction level? */ - level = get_query_value(query_string, "level", level, QR_ECLEVEL_L, QR_ECLEVEL_H); - } + cqrconf_file(server_name, &cqrconf); + cqrconf_string(getenv("QUERY_STRING"), &cqrconf); /* encode the QR-Code */ - if ((bitmap = encode_qrcode(uri, scale, border, level)) == NULL) { + if ((bitmap = encode_qrcode(uri, cqrconf)) == NULL) { /* uri too long? retry with uri from server name */ uri = uri_server_name; - if ((bitmap = encode_qrcode(uri, scale, border, level)) == NULL) { + if ((bitmap = encode_qrcode(uri, cqrconf)) == NULL) { fprintf(stderr, "Could not generate QR-Code.\n"); return EXIT_FAILURE; } @@ -395,12 +105,15 @@ int main(int argc, char **argv) { uri = uri_png; } - /* print PNG data */ - if (generate_png(bitmap, uri)) { + /* generate PNG data */ + if ((png = generate_png(bitmap, CQR_COMMENT|CQR_REFERER|CQR_VERSION|CQR_LIBVERSION, uri)) == NULL) { fprintf(stderr, "Failed to generate PNG.\n"); return EXIT_FAILURE; } + /* write PNG data to stdout */ + fwrite(png->buffer, png->size, 1, stdout); + /* free memory we no longer need */ if (uri_server_name) free(uri_server_name); @@ -409,6 +122,9 @@ int main(int argc, char **argv) { if (uri_png) free(uri_png); bitmap_free(bitmap); + if (png->size > 0) + free(png->buffer); + free(png); #if HAVE_FCGI /* end of loop */ |