/* pdfgen.c provided courtesy of https://github.com/AndreRenaud/PDFGen * * Thank You ! */ /** * Simple engine for creating PDF files. * It supports text, shapes, images etc... * Capable of handling millions of objects without too much performance * penalty. * Public domain license - no warrenty implied; use at your own risk. */ /** * PDF HINTS & TIPS * The specification can be found at * https://www.adobe.com/content/dam/acom/en/devnet/pdf/pdfs/pdf_reference_archives/PDFReference.pdf * The following sites have various bits & pieces about PDF document * generation * http://www.mactech.com/articles/mactech/Vol.15/15.09/PDFIntro/index.html * http://gnupdf.org/Introduction_to_PDF * http://www.planetpdf.com/mainpage.asp?WebPageID=63 * http://archive.vector.org.uk/art10008970 * http://www.adobe.com/devnet/acrobat/pdfs/pdf_reference_1-7.pdf * https://blog.idrsolutions.com/2013/01/understanding-the-pdf-file-format-overview/ * * To validate the PDF output, there are several online validators: * http://www.validatepdfa.com/online.htm * http://www.datalogics.com/products/callas/callaspdfA-onlinedemo.asp * http://www.pdf-tools.com/pdf/validate-pdfa-online.aspx * * In addition the 'pdftk' server can be used to analyse the output: * https://www.pdflabs.com/docs/pdftk-cli-examples/ * * PDF page markup operators: * b closepath, fill,and stroke path. * B fill and stroke path. * b* closepath, eofill,and stroke path. * B* eofill and stroke path. * BI begin image. * BMC begin marked content. * BT begin text object. * BX begin section allowing undefined operators. * c curveto. * cm concat. Concatenates the matrix to the current transform. * cs setcolorspace for fill. * CS setcolorspace for stroke. * d setdash. * Do execute the named XObject. * DP mark a place in the content stream, with a dictionary. * EI end image. * EMC end marked content. * ET end text object. * EX end section that allows undefined operators. * f fill path. * f* eofill Even/odd fill path. * g setgray (fill). * G setgray (stroke). * gs set parameters in the extended graphics state. * h closepath. * i setflat. * ID begin image data. * j setlinejoin. * J setlinecap. * k setcmykcolor (fill). * K setcmykcolor (stroke). * l lineto. * m moveto. * M setmiterlimit. * n end path without fill or stroke. * q save graphics state. * Q restore graphics state. * re rectangle. * rg setrgbcolor (fill). * RG setrgbcolor (stroke). * s closepath and stroke path. * S stroke path. * sc setcolor (fill). * SC setcolor (stroke). * sh shfill (shaded fill). * Tc set character spacing. * Td move text current point. * TD move text current point and set leading. * Tf set font name and size. * Tj show text. * TJ show text, allowing individual character positioning. * TL set leading. * Tm set text matrix. * Tr set text rendering mode. * Ts set super/subscripting text rise. * Tw set word spacing. * Tz set horizontal scaling. * T* move to start of next line. * v curveto. * w setlinewidth. * W clip. * y curveto. */ #if defined(_MSC_VER) #define _CRT_SECURE_NO_WARNINGS 1 // Drop the MSVC complaints about snprintf #define _USE_MATH_DEFINES #include typedef SSIZE_T ssize_t; #else #ifndef _POSIX_SOURCE #define _POSIX_SOURCE /* For localtime_r */ #endif #ifndef _XOPEN_SOURCE #define _XOPEN_SOURCE 600 /* for M_SQRT2 */ #endif #include /* for ssize_t */ #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include "pdfgen.h" #define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0])) #define PDF_RGB_R(c) (float)((((c) >> 16) & 0xff) / 255.0) #define PDF_RGB_G(c) (float)((((c) >> 8) & 0xff) / 255.0) #define PDF_RGB_B(c) (float)((((c) >> 0) & 0xff) / 255.0) #define PDF_IS_TRANSPARENT(c) (((c) >> 24) == 0xff) #if defined(_MSC_VER) #define inline __inline #define snprintf _snprintf #define strcasecmp _stricmp #define strncasecmp _strnicmp #ifdef stat #undef stat #endif #define stat _stat #define SKIP_ATTRIBUTE #else #include // strcasecmp #endif /** * Try and support big & little endian machines */ static inline uint32_t bswap32(uint32_t x) { return (((x & 0xff000000u) >> 24) | ((x & 0x00ff0000u) >> 8) | ((x & 0x0000ff00u) << 8) | ((x & 0x000000ffu) << 24)); } #ifdef __has_include // C++17, supported as extension to C++11 in clang, GCC // 5+, vs2015 #if __has_include() #include // gnu libc normally provides, linux #elif __has_include() #include //open bsd, macos #elif __has_include() #include // mingw, some bsd (not open/macos) #elif __has_include() #include // solaris #endif #endif #if !defined(__LITTLE_ENDIAN__) && !defined(__BIG_ENDIAN__) #ifndef __BYTE_ORDER__ /* Fall back to little endian by default */ #define __LITTLE_ENDIAN__ #elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ || __BYTE_ORDER == __BIG_ENDIAN #define __BIG_ENDIAN__ #else #define __LITTLE_ENDIAN__ #endif #endif #if defined(__LITTLE_ENDIAN__) #define ntoh32(x) bswap32((x)) #elif defined(__BIG_ENDIAN__) #define ntoh32(x) (x) #endif // Limits on image sizes for sanity checking & to avoid plausible overflow // issues #define MAX_IMAGE_WIDTH (16 * 1024) #define MAX_IMAGE_HEIGHT (16 * 1024) // Signatures for various image formats static const uint8_t bmp_signature[] = {'B', 'M'}; static const uint8_t png_signature[] = {0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A}; static const uint8_t jpeg_signature[] = {0xff, 0xd8}; static const uint8_t ppm_signature[] = {'P', '6'}; static const uint8_t pgm_signature[] = {'P', '5'}; // Special signatures for PNG chunks static const char png_chunk_header[] = "IHDR"; static const char png_chunk_palette[] = "PLTE"; static const char png_chunk_data[] = "IDAT"; static const char png_chunk_end[] = "IEND"; typedef struct pdf_object pdf_object; enum { OBJ_none, /* skipped */ OBJ_info, OBJ_stream, OBJ_font, OBJ_page, OBJ_bookmark, OBJ_outline, OBJ_catalog, OBJ_pages, OBJ_image, OBJ_link, OBJ_count, }; struct flexarray { void ***bins; int item_count; int bin_count; }; /** * Simple dynamic string object. Tries to store a reasonable amount on the * stack before falling back to malloc once things get large */ struct dstr { char static_data[128]; char *data; size_t alloc_len; size_t used_len; }; struct pdf_object { int type; /* See OBJ_xxxx */ int index; /* PDF output index */ int offset; /* Byte position within the output file */ struct pdf_object *prev; /* Previous of this type */ struct pdf_object *next; /* Next of this type */ union { struct { struct pdf_object *page; char name[64]; struct pdf_object *parent; struct flexarray children; } bookmark; struct dstr stream; struct { float width; float height; struct flexarray children; struct flexarray annotations; } page; struct pdf_info *info; struct { char name[64]; int index; } font; struct { struct pdf_object *page; /* Page containing link */ float llx; /* Clickable rectangle */ float lly; float urx; float ury; struct pdf_object *target_page; /* Target page */ float target_x; /* Target location */ float target_y; } link; }; }; struct pdf_doc { char errstr[128]; int errval; struct flexarray objects; float width; float height; struct pdf_object *current_font; struct pdf_object *last_objects[OBJ_count]; struct pdf_object *first_objects[OBJ_count]; }; /** * Since we're casting random areas of memory to these, make sure * they're packed properly to match the image format requirements */ #pragma pack(push, 1) struct png_chunk { uint32_t length; // chunk type, see png_chunk_header, png_chunk_data, png_chunk_end char type[4]; }; #pragma pack(pop) // Simple data container to store a single 24 Bit RGB value, used for // processing PNG images struct rgb_value { uint8_t red; uint8_t blue; uint8_t green; }; /** * Simple flexible resizing array implementation * The bins get larger in powers of two * bin 0 = 1024 items * 1 = 2048 items * 2 = 4096 items * etc... */ /* What is the first index that will be in the given bin? */ #define MIN_SHIFT 10 #define MIN_OFFSET ((1 << MIN_SHIFT) - 1) static int bin_offset[] = { (1 << (MIN_SHIFT + 0)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 1)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 2)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 3)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 4)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 5)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 6)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 7)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 8)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 9)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 10)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 11)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 12)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 13)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 14)) - 1 - MIN_OFFSET, (1 << (MIN_SHIFT + 15)) - 1 - MIN_OFFSET, }; static inline int flexarray_get_bin(const struct flexarray *flex, int index) { (void)flex; for (size_t i = 0; i < ARRAY_SIZE(bin_offset); i++) if (index < bin_offset[i]) return i - 1; return -1; } static inline int flexarray_get_bin_size(const struct flexarray *flex, int bin) { (void)flex; if (bin >= (int)ARRAY_SIZE(bin_offset) - 1) return -1; int next = bin_offset[bin + 1]; return next - bin_offset[bin]; } static inline int flexarray_get_bin_offset(const struct flexarray *flex, int bin, int index) { (void)flex; return index - bin_offset[bin]; } static void flexarray_clear(struct flexarray *flex) { for (int i = 0; i < flex->bin_count; i++) free(flex->bins[i]); free(flex->bins); flex->bin_count = 0; flex->item_count = 0; } static inline int flexarray_size(const struct flexarray *flex) { return flex->item_count; } static int flexarray_set(struct flexarray *flex, int index, void *data) { int bin = flexarray_get_bin(flex, index); if (bin < 0) return -EINVAL; if (bin >= flex->bin_count) { void ***bins = (void ***)realloc(flex->bins, (flex->bin_count + 1) * sizeof(*flex->bins)); if (!bins) return -ENOMEM; flex->bin_count++; flex->bins = bins; flex->bins[flex->bin_count - 1] = (void **)calloc(flexarray_get_bin_size(flex, flex->bin_count - 1), sizeof(void *)); if (!flex->bins[flex->bin_count - 1]) { flex->bin_count--; return -ENOMEM; } } flex->item_count++; flex->bins[bin][flexarray_get_bin_offset(flex, bin, index)] = data; return flex->item_count - 1; } static inline int flexarray_append(struct flexarray *flex, void *data) { return flexarray_set(flex, flexarray_size(flex), data); } static inline void *flexarray_get(const struct flexarray *flex, int index) { int bin; if (index >= flex->item_count) return NULL; bin = flexarray_get_bin(flex, index); if (bin < 0 || bin >= flex->bin_count) return NULL; return flex->bins[bin][flexarray_get_bin_offset(flex, bin, index)]; } /** * Simple dynamic string object. Tries to store a reasonable amount on the * stack before falling back to malloc once things get large */ #define INIT_DSTR \ (struct dstr) \ { \ .static_data = {0}, .data = NULL, .alloc_len = 0, .used_len = 0 \ } static char *dstr_data(struct dstr *str) { return str->data ? str->data : str->static_data; } static size_t dstr_len(const struct dstr *str) { return str->used_len; } static ssize_t dstr_ensure(struct dstr *str, size_t len) { if (len <= str->alloc_len) return 0; if (!str->data && len <= sizeof(str->static_data)) str->alloc_len = len; else if (str->alloc_len < len) { size_t new_len; new_len = len + 4096; if (str->data) { char *new_data = (char *)realloc((void *)str->data, new_len); if (!new_data) return -ENOMEM; str->data = new_data; } else { str->data = (char *)malloc(new_len); if (!str->data) return -ENOMEM; if (str->used_len) memcpy(str->data, str->static_data, str->used_len + 1); } str->alloc_len = new_len; } return 0; } // Locales can replace the decimal character with a ','. // This breaks the PDF output, so we force a 'safe' locale. static void force_locale(char *buf, int len) { char *saved_locale = setlocale(LC_ALL, NULL); if (!saved_locale) { *buf = '\0'; } else { strncpy(buf, saved_locale, len - 1); buf[len - 1] = '\0'; } setlocale(LC_NUMERIC, "POSIX"); } static void restore_locale(char *buf) { setlocale(LC_ALL, buf); } #ifndef SKIP_ATTRIBUTE static int dstr_printf(struct dstr *str, const char *fmt, ...) __attribute__((format(printf, 2, 3))); #endif static int dstr_printf(struct dstr *str, const char *fmt, ...) { va_list ap, aq; int len; char saved_locale[32]; force_locale(saved_locale, sizeof(saved_locale)); va_start(ap, fmt); va_copy(aq, ap); len = vsnprintf(NULL, 0, fmt, ap); if (dstr_ensure(str, str->used_len + len + 1) < 0) { va_end(ap); va_end(aq); restore_locale(saved_locale); return -ENOMEM; } vsprintf(dstr_data(str) + str->used_len, fmt, aq); str->used_len += len; va_end(ap); va_end(aq); restore_locale(saved_locale); return len; } static ssize_t dstr_append_data(struct dstr *str, const void *extend, size_t len) { if (dstr_ensure(str, str->used_len + len + 1) < 0) return -ENOMEM; memcpy(dstr_data(str) + str->used_len, extend, len); str->used_len += len; dstr_data(str)[str->used_len] = '\0'; return len; } static ssize_t dstr_append(struct dstr *str, const char *extend) { return dstr_append_data(str, extend, strlen(extend)); } static void dstr_free(struct dstr *str) { if (str->data) free(str->data); *str = INIT_DSTR; } /** * PDF Implementation */ #ifndef SKIP_ATTRIBUTE static int pdf_set_err(struct pdf_doc *doc, int errval, const char *buffer, ...) __attribute__((format(printf, 3, 4))); #endif static int pdf_set_err(struct pdf_doc *doc, int errval, const char *buffer, ...) { va_list ap; int len; va_start(ap, buffer); len = vsnprintf(doc->errstr, sizeof(doc->errstr) - 1, buffer, ap); va_end(ap); if (len < 0) { doc->errstr[0] = '\0'; return errval; } if (len >= (int)(sizeof(doc->errstr) - 1)) len = (int)(sizeof(doc->errstr) - 1); doc->errstr[len] = '\0'; doc->errval = errval; return errval; } const char *pdf_get_err(const struct pdf_doc *pdf, int *errval) { if (!pdf) return NULL; if (pdf->errstr[0] == '\0') return NULL; if (errval) *errval = pdf->errval; return pdf->errstr; } void pdf_clear_err(struct pdf_doc *pdf) { if (!pdf) return; pdf->errstr[0] = '\0'; pdf->errval = 0; } static int pdf_get_errval(struct pdf_doc *pdf) { if (!pdf) return 0; return pdf->errval; } static struct pdf_object *pdf_get_object(const struct pdf_doc *pdf, int index) { return (struct pdf_object *)flexarray_get(&pdf->objects, index); } static int pdf_append_object(struct pdf_doc *pdf, struct pdf_object *obj) { int index = flexarray_append(&pdf->objects, obj); if (index < 0) return index; obj->index = index; if (pdf->last_objects[obj->type]) { obj->prev = pdf->last_objects[obj->type]; pdf->last_objects[obj->type]->next = obj; } pdf->last_objects[obj->type] = obj; if (!pdf->first_objects[obj->type]) pdf->first_objects[obj->type] = obj; return 0; } static void pdf_object_destroy(struct pdf_object *object) { switch (object->type) { case OBJ_stream: case OBJ_image: dstr_free(&object->stream); break; case OBJ_page: flexarray_clear(&object->page.children); flexarray_clear(&object->page.annotations); break; case OBJ_info: free(object->info); break; case OBJ_bookmark: flexarray_clear(&object->bookmark.children); break; } free(object); } static struct pdf_object *pdf_add_object(struct pdf_doc *pdf, int type) { struct pdf_object *obj; if (!pdf) return NULL; obj = (struct pdf_object *)calloc(1, sizeof(*obj)); if (!obj) { pdf_set_err(pdf, -errno, "Unable to allocate object %d of type %d: %s", flexarray_size(&pdf->objects) + 1, type, strerror(errno)); return NULL; } obj->type = type; if (pdf_append_object(pdf, obj) < 0) { free(obj); return NULL; } return obj; } static void pdf_del_object(struct pdf_doc *pdf, struct pdf_object *obj) { int type = obj->type; flexarray_set(&pdf->objects, obj->index, NULL); if (pdf->last_objects[type] == obj) { pdf->last_objects[type] = NULL; for (int i = 0; i < flexarray_size(&pdf->objects); i++) { struct pdf_object *o = pdf_get_object(pdf, i); if (o && o->type == type) pdf->last_objects[type] = o; } } if (pdf->first_objects[type] == obj) { pdf->first_objects[type] = NULL; for (int i = 0; i < flexarray_size(&pdf->objects); i++) { struct pdf_object *o = pdf_get_object(pdf, i); if (o && o->type == type) { pdf->first_objects[type] = o; break; } } } pdf_object_destroy(obj); } struct pdf_doc *pdf_create(float width, float height, const struct pdf_info *info) { struct pdf_doc *pdf; struct pdf_object *obj; pdf = (struct pdf_doc *)calloc(1, sizeof(*pdf)); if (!pdf) return NULL; pdf->width = width; pdf->height = height; /* We don't want to use ID 0 */ pdf_add_object(pdf, OBJ_none); /* Create the 'info' object */ obj = pdf_add_object(pdf, OBJ_info); if (!obj) { pdf_destroy(pdf); return NULL; } obj->info = (struct pdf_info *)calloc(sizeof(*obj->info), 1); if (!obj->info) { pdf_destroy(pdf); return NULL; } if (info) { *obj->info = *info; obj->info->creator[sizeof(obj->info->creator) - 1] = '\0'; obj->info->producer[sizeof(obj->info->producer) - 1] = '\0'; obj->info->title[sizeof(obj->info->title) - 1] = '\0'; obj->info->author[sizeof(obj->info->author) - 1] = '\0'; obj->info->subject[sizeof(obj->info->subject) - 1] = '\0'; obj->info->date[sizeof(obj->info->date) - 1] = '\0'; } /* FIXME: Should be quoting PDF strings? */ if (!obj->info->date[0]) { time_t now = time(NULL); struct tm tm; #ifdef _WIN32 struct tm *tmp; tmp = localtime(&now); tm = *tmp; #else localtime_r(&now, &tm); #endif strftime(obj->info->date, sizeof(obj->info->date), "%Y%m%d%H%M%SZ", &tm); } if (!pdf_add_object(pdf, OBJ_pages)) { pdf_destroy(pdf); return NULL; } if (!pdf_add_object(pdf, OBJ_catalog)) { pdf_destroy(pdf); return NULL; } if (pdf_set_font(pdf, "Times-Roman") < 0) { pdf_destroy(pdf); return NULL; } return pdf; } float pdf_width(const struct pdf_doc *pdf) { return pdf->width; } float pdf_height(const struct pdf_doc *pdf) { return pdf->height; } float pdf_page_width(const struct pdf_object *page) { return page->page.width; } float pdf_page_height(const struct pdf_object *page) { return page->page.height; } void pdf_destroy(struct pdf_doc *pdf) { if (pdf) { for (int i = 0; i < flexarray_size(&pdf->objects); i++) pdf_object_destroy(pdf_get_object(pdf, i)); flexarray_clear(&pdf->objects); free(pdf); } } static struct pdf_object *pdf_find_first_object(const struct pdf_doc *pdf, int type) { if (!pdf) return NULL; return pdf->first_objects[type]; } static struct pdf_object *pdf_find_last_object(const struct pdf_doc *pdf, int type) { if (!pdf) return NULL; return pdf->last_objects[type]; } int pdf_set_font(struct pdf_doc *pdf, const char *font) { struct pdf_object *obj; int last_index = 0; /* See if we've used this font before */ for (obj = pdf_find_first_object(pdf, OBJ_font); obj; obj = obj->next) { if (strcmp(obj->font.name, font) == 0) break; last_index = obj->font.index; } /* Create a new font object if we need it */ if (!obj) { obj = pdf_add_object(pdf, OBJ_font); if (!obj) return pdf->errval; strncpy(obj->font.name, font, sizeof(obj->font.name) - 1); obj->font.name[sizeof(obj->font.name) - 1] = '\0'; obj->font.index = last_index + 1; } pdf->current_font = obj; return 0; } struct pdf_object *pdf_append_page(struct pdf_doc *pdf) { struct pdf_object *page; page = pdf_add_object(pdf, OBJ_page); if (!page) return NULL; page->page.width = pdf->width; page->page.height = pdf->height; return page; } int pdf_page_set_size(struct pdf_doc *pdf, struct pdf_object *page, float width, float height) { if (!page) page = pdf_find_last_object(pdf, OBJ_page); if (!page || page->type != OBJ_page) return pdf_set_err(pdf, -EINVAL, "Invalid PDF page"); page->page.width = width; page->page.height = height; return 0; } // Recursively scan for the number of children static int pdf_get_bookmark_count(const struct pdf_object *obj) { int count = 0; if (obj->type == OBJ_bookmark) { int nchildren = flexarray_size(&obj->bookmark.children); count += nchildren; for (int i = 0; i < nchildren; i++) { count += pdf_get_bookmark_count( (const struct pdf_object *)flexarray_get( &obj->bookmark.children, i)); } } return count; } static int pdf_save_object(struct pdf_doc *pdf, FILE *fp, int index) { struct pdf_object *object = pdf_get_object(pdf, index); if (!object) return -ENOENT; if (object->type == OBJ_none) return -ENOENT; object->offset = ftell(fp); fprintf(fp, "%d 0 obj\r\n", index); switch (object->type) { case OBJ_stream: case OBJ_image: { fwrite(dstr_data(&object->stream), dstr_len(&object->stream), 1, fp); break; } case OBJ_info: { struct pdf_info *info = object->info; fprintf(fp, "<<\r\n"); if (info->creator[0]) fprintf(fp, " /Creator (%s)\r\n", info->creator); if (info->producer[0]) fprintf(fp, " /Producer (%s)\r\n", info->producer); if (info->title[0]) fprintf(fp, " /Title (%s)\r\n", info->title); if (info->author[0]) fprintf(fp, " /Author (%s)\r\n", info->author); if (info->subject[0]) fprintf(fp, " /Subject (%s)\r\n", info->subject); if (info->date[0]) fprintf(fp, " /CreationDate (D:%s)\r\n", info->date); fprintf(fp, ">>\r\n"); break; } case OBJ_page: { struct pdf_object *pages = pdf_find_first_object(pdf, OBJ_pages); struct pdf_object *image = pdf_find_first_object(pdf, OBJ_image); fprintf(fp, "<<\r\n" " /Type /Page\r\n" " /Parent %d 0 R\r\n", pages->index); fprintf(fp, " /MediaBox [0 0 %f %f]\r\n", object->page.width, object->page.height); fprintf(fp, " /Resources <<\r\n"); fprintf(fp, " /Font <<\r\n"); for (struct pdf_object *font = pdf_find_first_object(pdf, OBJ_font); font; font = font->next) fprintf(fp, " /F%d %d 0 R\r\n", font->font.index, font->index); fprintf(fp, " >>\r\n"); // We trim transparency to just 4-bits fprintf(fp, " /ExtGState <<\r\n"); for (int i = 0; i < 16; i++) { fprintf(fp, " /GS%d <>\r\n", i, (float)(15 - i) / 15); } fprintf(fp, " >>\r\n"); if (image) { fprintf(fp, " /XObject <<"); for (; image; image = image->next) fprintf(fp, " /Image%d %d 0 R ", image->index, image->index); fprintf(fp, " >>\r\n"); } fprintf(fp, " >>\r\n"); fprintf(fp, " /Contents [\r\n"); for (int i = 0; i < flexarray_size(&object->page.children); i++) { struct pdf_object *child = (struct pdf_object *)flexarray_get(&object->page.children, i); fprintf(fp, "%d 0 R\r\n", child->index); } fprintf(fp, "]\r\n"); if (flexarray_size(&object->page.annotations)) { fprintf(fp, " /Annots [\r\n"); for (int i = 0; i < flexarray_size(&object->page.annotations); i++) { struct pdf_object *child = (struct pdf_object *)flexarray_get( &object->page.annotations, i); fprintf(fp, "%d 0 R\r\n", child->index); } fprintf(fp, "]\r\n"); } fprintf(fp, ">>\r\n"); break; } case OBJ_bookmark: { struct pdf_object *parent, *other; parent = object->bookmark.parent; if (!parent) parent = pdf_find_first_object(pdf, OBJ_outline); if (!object->bookmark.page) break; fprintf(fp, "<<\r\n" " /Dest [%d 0 R /XYZ 0 %f null]\r\n" " /Parent %d 0 R\r\n" " /Title (%s)\r\n", object->bookmark.page->index, pdf->height, parent->index, object->bookmark.name); int nchildren = flexarray_size(&object->bookmark.children); if (nchildren > 0) { struct pdf_object *f, *l; f = (struct pdf_object *)flexarray_get(&object->bookmark.children, 0); l = (struct pdf_object *)flexarray_get(&object->bookmark.children, nchildren - 1); fprintf(fp, " /First %d 0 R\r\n", f->index); fprintf(fp, " /Last %d 0 R\r\n", l->index); fprintf(fp, " /Count %d\r\n", pdf_get_bookmark_count(object)); } // Find the previous bookmark with the same parent for (other = object->prev; other && other->bookmark.parent != object->bookmark.parent; other = other->prev) ; if (other) fprintf(fp, " /Prev %d 0 R\r\n", other->index); // Find the next bookmark with the same parent for (other = object->next; other && other->bookmark.parent != object->bookmark.parent; other = other->next) ; if (other) fprintf(fp, " /Next %d 0 R\r\n", other->index); fprintf(fp, ">>\r\n"); break; } case OBJ_outline: { struct pdf_object *first, *last, *cur; first = pdf_find_first_object(pdf, OBJ_bookmark); last = pdf_find_last_object(pdf, OBJ_bookmark); if (first && last) { int count = 0; cur = first; while (cur) { if (!cur->bookmark.parent) count += pdf_get_bookmark_count(cur) + 1; cur = cur->next; } /* Bookmark outline */ fprintf(fp, "<<\r\n" " /Count %d\r\n" " /Type /Outlines\r\n" " /First %d 0 R\r\n" " /Last %d 0 R\r\n" ">>\r\n", count, first->index, last->index); } break; } case OBJ_font: fprintf(fp, "<<\r\n" " /Type /Font\r\n" " /Subtype /Type1\r\n" " /BaseFont /%s\r\n" " /Encoding /WinAnsiEncoding\r\n" ">>\r\n", object->font.name); break; case OBJ_pages: { int npages = 0; fprintf(fp, "<<\r\n" " /Type /Pages\r\n" " /Kids [ "); for (struct pdf_object *page = pdf_find_first_object(pdf, OBJ_page); page; page = page->next) { npages++; fprintf(fp, "%d 0 R ", page->index); } fprintf(fp, "]\r\n"); fprintf(fp, " /Count %d\r\n", npages); fprintf(fp, ">>\r\n"); break; } case OBJ_catalog: { struct pdf_object *outline = pdf_find_first_object(pdf, OBJ_outline); struct pdf_object *pages = pdf_find_first_object(pdf, OBJ_pages); fprintf(fp, "<<\r\n" " /Type /Catalog\r\n"); if (outline) fprintf(fp, " /Outlines %d 0 R\r\n" " /PageMode /UseOutlines\r\n", outline->index); fprintf(fp, " /Pages %d 0 R\r\n" ">>\r\n", pages->index); break; } case OBJ_link: { fprintf(fp, "<<\r\n" " /Type /Annot\r\n" " /Subtype /Link\r\n" " /Rect [%f %f %f %f]\r\n" " /Dest [%u 0 R /XYZ %f %f null]\r\n" " /Border [0 0 0]\r\n" ">>\r\n", object->link.llx, object->link.lly, object->link.urx, object->link.ury, object->link.target_page->index, object->link.target_x, object->link.target_y); break; } default: return pdf_set_err(pdf, -EINVAL, "Invalid PDF object type %d", object->type); } fprintf(fp, "endobj\r\n"); return 0; } // Slightly modified djb2 hash algorithm to get pseudo-random ID static uint64_t hash(uint64_t hash, const void *data, size_t len) { const uint8_t *d8 = (const uint8_t *)data; for (; len; len--) { hash = (((hash & 0x03ffffffffffffff) << 5) + (hash & 0x7fffffffffffffff)) + *d8++; } return hash; } int pdf_save_file(struct pdf_doc *pdf, FILE *fp) { struct pdf_object *obj; int xref_offset; int xref_count = 0; uint64_t id1, id2; time_t now = time(NULL); char saved_locale[32]; force_locale(saved_locale, sizeof(saved_locale)); fprintf(fp, "%%PDF-1.3\r\n"); /* Hibit bytes */ fprintf(fp, "%c%c%c%c%c\r\n", 0x25, 0xc7, 0xec, 0x8f, 0xa2); /* Dump all the objects & get their file offsets */ for (int i = 0; i < flexarray_size(&pdf->objects); i++) if (pdf_save_object(pdf, fp, i) >= 0) xref_count++; /* xref */ xref_offset = ftell(fp); fprintf(fp, "xref\r\n"); fprintf(fp, "0 %d\r\n", xref_count + 1); fprintf(fp, "0000000000 65535 f\r\n"); for (int i = 0; i < flexarray_size(&pdf->objects); i++) { obj = pdf_get_object(pdf, i); if (obj->type != OBJ_none) fprintf(fp, "%10.10d 00000 n\r\n", obj->offset); } fprintf(fp, "trailer\r\n" "<<\r\n" "/Size %d\r\n", xref_count + 1); obj = pdf_find_first_object(pdf, OBJ_catalog); fprintf(fp, "/Root %d 0 R\r\n", obj->index); obj = pdf_find_first_object(pdf, OBJ_info); fprintf(fp, "/Info %d 0 R\r\n", obj->index); /* Generate document unique IDs */ id1 = hash(5381, obj->info, sizeof(struct pdf_info)); id1 = hash(id1, &xref_count, sizeof(xref_count)); id2 = hash(5381, &now, sizeof(now)); fprintf(fp, "/ID [<%16.16" PRIx64 "> <%16.16" PRIx64 ">]\r\n", id1, id2); fprintf(fp, ">>\r\n" "startxref\r\n"); fprintf(fp, "%d\r\n", xref_offset); fprintf(fp, "%%%%EOF\r\n"); restore_locale(saved_locale); return 0; } int pdf_save(struct pdf_doc *pdf, const char *filename) { FILE *fp; int e; if (filename == NULL) fp = stdout; else if ((fp = fopen(filename, "wb")) == NULL) return pdf_set_err(pdf, -errno, "Unable to open '%s': %s", filename, strerror(errno)); e = pdf_save_file(pdf, fp); if (fp != stdout) if (fclose(fp) != 0 && e >= 0) return pdf_set_err(pdf, -errno, "Unable to close '%s': %s", filename, strerror(errno)); return e; } static int pdf_add_stream(struct pdf_doc *pdf, struct pdf_object *page, const char *buffer) { struct pdf_object *obj; size_t len; if (!page) page = pdf_find_last_object(pdf, OBJ_page); if (!page) return pdf_set_err(pdf, -EINVAL, "Invalid pdf page"); len = strlen(buffer); /* We don't want any trailing whitespace in the stream */ while (len >= 1 && (buffer[len - 1] == '\r' || buffer[len - 1] == '\n')) len--; obj = pdf_add_object(pdf, OBJ_stream); if (!obj) return pdf->errval; dstr_printf(&obj->stream, "<< /Length %zu >>stream\r\n", len); dstr_append_data(&obj->stream, buffer, len); dstr_append(&obj->stream, "\r\nendstream\r\n"); return flexarray_append(&page->page.children, obj); } int pdf_add_bookmark(struct pdf_doc *pdf, struct pdf_object *page, int parent, const char *name) { struct pdf_object *obj, *outline = NULL; if (!page) page = pdf_find_last_object(pdf, OBJ_page); if (!page) return pdf_set_err(pdf, -EINVAL, "Unable to add bookmark, no pages available"); if (!pdf_find_first_object(pdf, OBJ_outline)) { outline = pdf_add_object(pdf, OBJ_outline); if (!outline) return pdf->errval; } obj = pdf_add_object(pdf, OBJ_bookmark); if (!obj) { if (outline) pdf_del_object(pdf, outline); return pdf->errval; } strncpy(obj->bookmark.name, name, sizeof(obj->bookmark.name) - 1); obj->bookmark.name[sizeof(obj->bookmark.name) - 1] = '\0'; obj->bookmark.page = page; if (parent >= 0) { struct pdf_object *parent_obj = pdf_get_object(pdf, parent); if (!parent_obj) return pdf_set_err(pdf, -EINVAL, "Invalid parent ID %d supplied", parent); obj->bookmark.parent = parent_obj; flexarray_append(&parent_obj->bookmark.children, obj); } return obj->index; } int pdf_add_link(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, struct pdf_object *target_page, float target_x, float target_y) { struct pdf_object *obj; if (!page) page = pdf_find_last_object(pdf, OBJ_page); if (!page) return pdf_set_err(pdf, -EINVAL, "Unable to add link, no pages available"); if (!target_page) return pdf_set_err(pdf, -EINVAL, "Unable to link, no target page"); obj = pdf_add_object(pdf, OBJ_link); if (!obj) { return pdf->errval; } obj->link.target_page = target_page; obj->link.target_x = target_x; obj->link.target_y = target_y; obj->link.llx = x; obj->link.lly = y; obj->link.urx = x + width; obj->link.ury = y + height; flexarray_append(&page->page.annotations, obj); return obj->index; } static int utf8_to_utf32(const char *utf8, int len, uint32_t *utf32) { uint32_t ch; uint8_t mask; if (len <= 0 || !utf8 || !utf32) return -EINVAL; ch = *(uint8_t *)utf8; if ((ch & 0x80) == 0) { len = 1; mask = 0x7f; } else if ((ch & 0xe0) == 0xc0 && len >= 2) { len = 2; mask = 0x1f; } else if ((ch & 0xf0) == 0xe0 && len >= 3) { len = 3; mask = 0xf; } else if ((ch & 0xf8) == 0xf0 && len >= 4) { len = 4; mask = 0x7; } else return -EINVAL; ch = 0; for (int i = 0; i < len; i++) { int shift = (len - i - 1) * 6; if (!*utf8) return -EINVAL; if (i == 0) ch |= ((uint32_t)(*utf8++) & mask) << shift; else ch |= ((uint32_t)(*utf8++) & 0x3f) << shift; } *utf32 = ch; return len; } static int utf8_to_pdfencoding(struct pdf_doc *pdf, const char *utf8, int len, uint8_t *res) { uint32_t code; int code_len; *res = 0; code_len = utf8_to_utf32(utf8, len, &code); if (code_len < 0) { return pdf_set_err(pdf, -EINVAL, "Invalid UTF-8 encoding"); } if (code > 255) { /* We support *some* minimal UTF-8 characters */ // See Appendix D of // https://www.adobe.com/content/dam/acom/en/devnet/pdf/pdfs/pdf_reference_archives/PDFReference.pdf // These are all in WinAnsiEncoding switch (code) { case 0x160: // Latin Capital Letter S with Caron *res = 0212; break; case 0x161: // Latin Small Letter S with Caron *res = 0232; break; case 0x17d: // Latin Capital Letter Z with Caron *res = 0216; break; case 0x17e: // Latin Small Letter Z with Caron *res = 0236; break; case 0x2014: // emdash *res = 0227; break; case 0x2018: // left single quote *res = 0221; break; case 0x2019: // right single quote *res = 0222; break; case 0x201c: // left double quote *res = 0223; break; case 0x201d: // right double quote *res = 0224; break; case 0x20ac: // Euro *res = 0200; break; default: return pdf_set_err(pdf, -EINVAL, "Unsupported UTF-8 character: 0x%x 0o%o", code, code); } } else { *res = code; } return code_len; } static int pdf_add_text_spacing(struct pdf_doc *pdf, struct pdf_object *page, const char *text, float size, float xoff, float yoff, uint32_t colour, float spacing) { int ret; size_t len = text ? strlen(text) : 0; struct dstr str = INIT_DSTR; int alpha = (colour >> 24) >> 4; /* Don't bother adding empty/null strings */ if (!len) return 0; dstr_append(&str, "BT "); dstr_printf(&str, "/GS%d gs ", alpha); dstr_printf(&str, "%f %f TD ", xoff, yoff); dstr_printf(&str, "/F%d %f Tf ", pdf->current_font->font.index, size); dstr_printf(&str, "%f %f %f rg ", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f Tc ", spacing); dstr_append(&str, "("); /* Escape magic characters properly */ for (size_t i = 0; i < len;) { int code_len; uint8_t pdf_char; code_len = utf8_to_pdfencoding(pdf, &text[i], len - i, &pdf_char); if (code_len < 0) { dstr_free(&str); return code_len; } if (strchr("()\\", pdf_char)) { char buf[3]; /* Escape some characters */ buf[0] = '\\'; buf[1] = pdf_char; buf[2] = '\0'; dstr_append(&str, buf); } else if (strrchr("\n\r\t\b\f", pdf_char)) { /* Skip over these characters */ ; } else { dstr_append_data(&str, &pdf_char, 1); } i += code_len; } dstr_append(&str, ") Tj "); dstr_append(&str, "ET"); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_text(struct pdf_doc *pdf, struct pdf_object *page, const char *text, float size, float xoff, float yoff, uint32_t colour) { return pdf_add_text_spacing(pdf, page, text, size, xoff, yoff, colour, 0); } /* How wide is each character, in points, at size 14 */ static const uint16_t helvetica_widths[256] = { 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 357, 560, 560, 896, 672, 192, 335, 335, 392, 588, 280, 335, 280, 280, 560, 560, 560, 560, 560, 560, 560, 560, 560, 560, 280, 280, 588, 588, 588, 560, 1023, 672, 672, 727, 727, 672, 615, 784, 727, 280, 504, 672, 560, 839, 727, 784, 672, 784, 727, 672, 615, 727, 672, 951, 672, 672, 615, 280, 280, 280, 472, 560, 335, 560, 560, 504, 560, 560, 280, 560, 560, 223, 223, 504, 223, 839, 560, 560, 560, 560, 335, 504, 280, 560, 504, 727, 504, 504, 504, 336, 262, 336, 588, 352, 560, 352, 223, 560, 335, 1008, 560, 560, 335, 1008, 672, 335, 1008, 352, 615, 352, 352, 223, 223, 335, 335, 352, 560, 1008, 335, 1008, 504, 335, 951, 352, 504, 672, 280, 335, 560, 560, 560, 560, 262, 560, 335, 742, 372, 560, 588, 335, 742, 335, 403, 588, 335, 335, 335, 560, 541, 280, 335, 335, 367, 560, 840, 840, 840, 615, 672, 672, 672, 672, 672, 672, 1008, 727, 672, 672, 672, 672, 280, 280, 280, 280, 727, 727, 784, 784, 784, 784, 784, 588, 784, 727, 727, 727, 727, 672, 672, 615, 560, 560, 560, 560, 560, 560, 896, 504, 560, 560, 560, 560, 280, 280, 280, 280, 560, 560, 560, 560, 560, 560, 560, 588, 615, 560, 560, 560, 560, 504, 560, 504, }; static const uint16_t helvetica_bold_widths[256] = { 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 335, 477, 560, 560, 896, 727, 239, 335, 335, 392, 588, 280, 335, 280, 280, 560, 560, 560, 560, 560, 560, 560, 560, 560, 560, 335, 335, 588, 588, 588, 615, 982, 727, 727, 727, 727, 672, 615, 784, 727, 280, 560, 727, 615, 839, 727, 784, 672, 784, 727, 672, 615, 727, 672, 951, 672, 672, 615, 335, 280, 335, 588, 560, 335, 560, 615, 560, 615, 560, 335, 615, 615, 280, 280, 560, 280, 896, 615, 615, 615, 615, 392, 560, 335, 615, 560, 784, 560, 560, 504, 392, 282, 392, 588, 352, 560, 352, 280, 560, 504, 1008, 560, 560, 335, 1008, 672, 335, 1008, 352, 615, 352, 352, 280, 280, 504, 504, 352, 560, 1008, 335, 1008, 560, 335, 951, 352, 504, 672, 280, 335, 560, 560, 560, 560, 282, 560, 335, 742, 372, 560, 588, 335, 742, 335, 403, 588, 335, 335, 335, 615, 560, 280, 335, 335, 367, 560, 840, 840, 840, 615, 727, 727, 727, 727, 727, 727, 1008, 727, 672, 672, 672, 672, 280, 280, 280, 280, 727, 727, 784, 784, 784, 784, 784, 588, 784, 727, 727, 727, 727, 672, 672, 615, 560, 560, 560, 560, 560, 560, 896, 560, 560, 560, 560, 560, 280, 280, 280, 280, 615, 615, 615, 615, 615, 615, 615, 588, 615, 615, 615, 615, 615, 560, 615, 560, }; static const uint16_t helvetica_bold_oblique_widths[256] = { 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 335, 477, 560, 560, 896, 727, 239, 335, 335, 392, 588, 280, 335, 280, 280, 560, 560, 560, 560, 560, 560, 560, 560, 560, 560, 335, 335, 588, 588, 588, 615, 982, 727, 727, 727, 727, 672, 615, 784, 727, 280, 560, 727, 615, 839, 727, 784, 672, 784, 727, 672, 615, 727, 672, 951, 672, 672, 615, 335, 280, 335, 588, 560, 335, 560, 615, 560, 615, 560, 335, 615, 615, 280, 280, 560, 280, 896, 615, 615, 615, 615, 392, 560, 335, 615, 560, 784, 560, 560, 504, 392, 282, 392, 588, 352, 560, 352, 280, 560, 504, 1008, 560, 560, 335, 1008, 672, 335, 1008, 352, 615, 352, 352, 280, 280, 504, 504, 352, 560, 1008, 335, 1008, 560, 335, 951, 352, 504, 672, 280, 335, 560, 560, 560, 560, 282, 560, 335, 742, 372, 560, 588, 335, 742, 335, 403, 588, 335, 335, 335, 615, 560, 280, 335, 335, 367, 560, 840, 840, 840, 615, 727, 727, 727, 727, 727, 727, 1008, 727, 672, 672, 672, 672, 280, 280, 280, 280, 727, 727, 784, 784, 784, 784, 784, 588, 784, 727, 727, 727, 727, 672, 672, 615, 560, 560, 560, 560, 560, 560, 896, 560, 560, 560, 560, 560, 280, 280, 280, 280, 615, 615, 615, 615, 615, 615, 615, 588, 615, 615, 615, 615, 615, 560, 615, 560, }; static const uint16_t helvetica_oblique_widths[256] = { 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 280, 357, 560, 560, 896, 672, 192, 335, 335, 392, 588, 280, 335, 280, 280, 560, 560, 560, 560, 560, 560, 560, 560, 560, 560, 280, 280, 588, 588, 588, 560, 1023, 672, 672, 727, 727, 672, 615, 784, 727, 280, 504, 672, 560, 839, 727, 784, 672, 784, 727, 672, 615, 727, 672, 951, 672, 672, 615, 280, 280, 280, 472, 560, 335, 560, 560, 504, 560, 560, 280, 560, 560, 223, 223, 504, 223, 839, 560, 560, 560, 560, 335, 504, 280, 560, 504, 727, 504, 504, 504, 336, 262, 336, 588, 352, 560, 352, 223, 560, 335, 1008, 560, 560, 335, 1008, 672, 335, 1008, 352, 615, 352, 352, 223, 223, 335, 335, 352, 560, 1008, 335, 1008, 504, 335, 951, 352, 504, 672, 280, 335, 560, 560, 560, 560, 262, 560, 335, 742, 372, 560, 588, 335, 742, 335, 403, 588, 335, 335, 335, 560, 541, 280, 335, 335, 367, 560, 840, 840, 840, 615, 672, 672, 672, 672, 672, 672, 1008, 727, 672, 672, 672, 672, 280, 280, 280, 280, 727, 727, 784, 784, 784, 784, 784, 588, 784, 727, 727, 727, 727, 672, 672, 615, 560, 560, 560, 560, 560, 560, 896, 504, 560, 560, 560, 560, 280, 280, 280, 280, 560, 560, 560, 560, 560, 560, 560, 588, 615, 560, 560, 560, 560, 504, 560, 504, }; static const uint16_t symbol_widths[256] = { 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 335, 718, 504, 553, 839, 784, 442, 335, 335, 504, 553, 252, 553, 252, 280, 504, 504, 504, 504, 504, 504, 504, 504, 504, 504, 280, 280, 553, 553, 553, 447, 553, 727, 672, 727, 616, 615, 769, 607, 727, 335, 636, 727, 691, 896, 727, 727, 774, 746, 560, 596, 615, 695, 442, 774, 650, 801, 615, 335, 869, 335, 663, 504, 504, 636, 553, 553, 497, 442, 525, 414, 607, 331, 607, 553, 553, 580, 525, 553, 553, 525, 553, 607, 442, 580, 718, 691, 496, 691, 497, 483, 201, 483, 553, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 756, 624, 248, 553, 168, 718, 504, 759, 759, 759, 759, 1050, 994, 607, 994, 607, 403, 553, 414, 553, 553, 718, 497, 463, 553, 553, 553, 553, 1008, 607, 1008, 663, 829, 691, 801, 994, 774, 774, 829, 774, 774, 718, 718, 718, 718, 718, 718, 718, 774, 718, 796, 796, 897, 829, 553, 252, 718, 607, 607, 1050, 994, 607, 994, 607, 497, 331, 796, 796, 792, 718, 387, 387, 387, 387, 387, 387, 497, 497, 497, 497, 0, 331, 276, 691, 691, 691, 387, 387, 387, 387, 387, 387, 497, 497, 497, 0, }; static const uint16_t times_widths[256] = { 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 335, 411, 504, 504, 839, 784, 181, 335, 335, 504, 568, 252, 335, 252, 280, 504, 504, 504, 504, 504, 504, 504, 504, 504, 504, 280, 280, 568, 568, 568, 447, 928, 727, 672, 672, 727, 615, 560, 727, 727, 335, 392, 727, 615, 896, 727, 727, 560, 727, 672, 560, 615, 727, 727, 951, 727, 727, 615, 335, 280, 335, 472, 504, 335, 447, 504, 447, 504, 447, 335, 504, 504, 280, 280, 504, 280, 784, 504, 504, 504, 504, 335, 392, 280, 504, 504, 727, 504, 504, 447, 483, 201, 483, 545, 352, 504, 352, 335, 504, 447, 1008, 504, 504, 335, 1008, 560, 335, 896, 352, 615, 352, 352, 335, 335, 447, 447, 352, 504, 1008, 335, 987, 392, 335, 727, 352, 447, 727, 252, 335, 504, 504, 504, 504, 201, 504, 335, 766, 278, 504, 568, 335, 766, 335, 403, 568, 302, 302, 335, 504, 456, 252, 335, 302, 312, 504, 756, 756, 756, 447, 727, 727, 727, 727, 727, 727, 896, 672, 615, 615, 615, 615, 335, 335, 335, 335, 727, 727, 727, 727, 727, 727, 727, 568, 727, 727, 727, 727, 727, 727, 560, 504, 447, 447, 447, 447, 447, 447, 672, 447, 447, 447, 447, 447, 280, 280, 280, 280, 504, 504, 504, 504, 504, 504, 504, 568, 504, 504, 504, 504, 504, 504, 504, 504, }; static const uint16_t times_bold_widths[256] = { 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 335, 559, 504, 504, 1008, 839, 280, 335, 335, 504, 574, 252, 335, 252, 280, 504, 504, 504, 504, 504, 504, 504, 504, 504, 504, 335, 335, 574, 574, 574, 504, 937, 727, 672, 727, 727, 672, 615, 784, 784, 392, 504, 784, 672, 951, 727, 784, 615, 784, 727, 560, 672, 727, 727, 1008, 727, 727, 672, 335, 280, 335, 585, 504, 335, 504, 560, 447, 560, 447, 335, 504, 560, 280, 335, 560, 280, 839, 560, 504, 560, 560, 447, 392, 335, 560, 504, 727, 504, 504, 447, 397, 221, 397, 524, 352, 504, 352, 335, 504, 504, 1008, 504, 504, 335, 1008, 560, 335, 1008, 352, 672, 352, 352, 335, 335, 504, 504, 352, 504, 1008, 335, 1008, 392, 335, 727, 352, 447, 727, 252, 335, 504, 504, 504, 504, 221, 504, 335, 752, 302, 504, 574, 335, 752, 335, 403, 574, 302, 302, 335, 560, 544, 252, 335, 302, 332, 504, 756, 756, 756, 504, 727, 727, 727, 727, 727, 727, 1008, 727, 672, 672, 672, 672, 392, 392, 392, 392, 727, 727, 784, 784, 784, 784, 784, 574, 784, 727, 727, 727, 727, 727, 615, 560, 504, 504, 504, 504, 504, 504, 727, 447, 447, 447, 447, 447, 280, 280, 280, 280, 504, 560, 504, 504, 504, 504, 504, 574, 504, 560, 560, 560, 560, 504, 560, 504, }; static const uint16_t times_bold_italic_widths[256] = { 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 392, 559, 504, 504, 839, 784, 280, 335, 335, 504, 574, 252, 335, 252, 280, 504, 504, 504, 504, 504, 504, 504, 504, 504, 504, 335, 335, 574, 574, 574, 504, 838, 672, 672, 672, 727, 672, 672, 727, 784, 392, 504, 672, 615, 896, 727, 727, 615, 727, 672, 560, 615, 727, 672, 896, 672, 615, 615, 335, 280, 335, 574, 504, 335, 504, 504, 447, 504, 447, 335, 504, 560, 280, 280, 504, 280, 784, 560, 504, 504, 504, 392, 392, 280, 560, 447, 672, 504, 447, 392, 350, 221, 350, 574, 352, 504, 352, 335, 504, 504, 1008, 504, 504, 335, 1008, 560, 335, 951, 352, 615, 352, 352, 335, 335, 504, 504, 352, 504, 1008, 335, 1008, 392, 335, 727, 352, 392, 615, 252, 392, 504, 504, 504, 504, 221, 504, 335, 752, 268, 504, 610, 335, 752, 335, 403, 574, 302, 302, 335, 580, 504, 252, 335, 302, 302, 504, 756, 756, 756, 504, 672, 672, 672, 672, 672, 672, 951, 672, 672, 672, 672, 672, 392, 392, 392, 392, 727, 727, 727, 727, 727, 727, 727, 574, 727, 727, 727, 727, 727, 615, 615, 504, 504, 504, 504, 504, 504, 504, 727, 447, 447, 447, 447, 447, 280, 280, 280, 280, 504, 560, 504, 504, 504, 504, 504, 574, 504, 560, 560, 560, 560, 447, 504, 447, }; static const uint16_t times_italic_widths[256] = { 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 252, 335, 423, 504, 504, 839, 784, 215, 335, 335, 504, 680, 252, 335, 252, 280, 504, 504, 504, 504, 504, 504, 504, 504, 504, 504, 335, 335, 680, 680, 680, 504, 927, 615, 615, 672, 727, 615, 615, 727, 727, 335, 447, 672, 560, 839, 672, 727, 615, 727, 615, 504, 560, 727, 615, 839, 615, 560, 560, 392, 280, 392, 425, 504, 335, 504, 504, 447, 504, 447, 280, 504, 504, 280, 280, 447, 280, 727, 504, 504, 504, 504, 392, 392, 280, 504, 447, 672, 447, 447, 392, 403, 277, 403, 545, 352, 504, 352, 335, 504, 560, 896, 504, 504, 335, 1008, 504, 335, 951, 352, 560, 352, 352, 335, 335, 560, 560, 352, 504, 896, 335, 987, 392, 335, 672, 352, 392, 560, 252, 392, 504, 504, 504, 504, 277, 504, 335, 766, 278, 504, 680, 335, 766, 335, 403, 680, 302, 302, 335, 504, 527, 252, 335, 302, 312, 504, 756, 756, 756, 504, 615, 615, 615, 615, 615, 615, 896, 672, 615, 615, 615, 615, 335, 335, 335, 335, 727, 672, 727, 727, 727, 727, 727, 680, 727, 727, 727, 727, 727, 560, 615, 504, 504, 504, 504, 504, 504, 504, 672, 447, 447, 447, 447, 447, 280, 280, 280, 280, 504, 504, 504, 504, 504, 504, 504, 680, 504, 504, 504, 504, 504, 447, 504, 447, }; static const uint16_t zapfdingbats_widths[256] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 280, 981, 968, 981, 987, 724, 795, 796, 797, 695, 967, 946, 553, 861, 918, 940, 918, 952, 981, 761, 852, 768, 767, 575, 682, 769, 766, 765, 760, 497, 556, 541, 581, 697, 792, 794, 794, 796, 799, 800, 822, 829, 795, 847, 829, 839, 822, 837, 930, 749, 728, 754, 796, 798, 700, 782, 774, 798, 765, 712, 713, 687, 706, 832, 821, 795, 795, 712, 692, 701, 694, 792, 793, 718, 797, 791, 797, 879, 767, 768, 768, 765, 765, 899, 899, 794, 790, 441, 139, 279, 418, 395, 395, 673, 673, 0, 393, 393, 319, 319, 278, 278, 513, 513, 413, 413, 235, 235, 336, 336, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 737, 548, 548, 917, 672, 766, 766, 782, 599, 699, 631, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 794, 901, 844, 1024, 461, 753, 931, 753, 925, 934, 935, 935, 840, 879, 834, 931, 931, 924, 937, 938, 466, 890, 842, 842, 873, 873, 701, 701, 880, 0, 880, 766, 953, 777, 871, 777, 895, 974, 895, 837, 879, 934, 977, 925, 0, }; static const uint16_t courier_widths[256] = { 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, 604, }; static int pdf_text_point_width(struct pdf_doc *pdf, const char *text, ptrdiff_t text_len, float size, const uint16_t *widths, float *point_width) { uint32_t len = 0; if (text_len < 0) text_len = strlen(text); *point_width = 0.0f; for (int i = 0; i < (int)text_len;) { uint8_t pdf_char = 0; int code_len; code_len = utf8_to_pdfencoding(pdf, &text[i], text_len - i, &pdf_char); if (code_len < 0) return pdf_set_err(pdf, code_len, "Invalid unicode string at position %d in %s", i, text); i += code_len; if (pdf_char != '\n' && pdf_char != '\r') len += widths[pdf_char]; } /* Our widths arrays are for 14pt fonts */ *point_width = len * size / (14.0f * 72.0f); return 0; } static const uint16_t *find_font_widths(const char *font_name) { if (strcasecmp(font_name, "Helvetica") == 0) return helvetica_widths; if (strcasecmp(font_name, "Helvetica-Bold") == 0) return helvetica_bold_widths; if (strcasecmp(font_name, "Helvetica-BoldOblique") == 0) return helvetica_bold_oblique_widths; if (strcasecmp(font_name, "Helvetica-Oblique") == 0) return helvetica_oblique_widths; if (strcasecmp(font_name, "Courier") == 0 || strcasecmp(font_name, "Courier-Bold") == 0 || strcasecmp(font_name, "Courier-BoldOblique") == 0 || strcasecmp(font_name, "Courier-Oblique") == 0) return courier_widths; if (strcasecmp(font_name, "Times-Roman") == 0) return times_widths; if (strcasecmp(font_name, "Times-Bold") == 0) return times_bold_widths; if (strcasecmp(font_name, "Times-Italic") == 0) return times_italic_widths; if (strcasecmp(font_name, "Times-BoldItalic") == 0) return times_bold_italic_widths; if (strcasecmp(font_name, "Symbol") == 0) return symbol_widths; if (strcasecmp(font_name, "ZapfDingbats") == 0) return zapfdingbats_widths; return NULL; } int pdf_get_font_text_width(struct pdf_doc *pdf, const char *font_name, const char *text, float size, float *text_width) { if (!font_name) font_name = pdf->current_font->font.name; const uint16_t *widths = find_font_widths(font_name); if (!widths) return pdf_set_err(pdf, -EINVAL, "Unable to determine width for font '%s'", pdf->current_font->font.name); return pdf_text_point_width(pdf, text, -1, size, widths, text_width); } static const char *find_word_break(const char *string) { /* Skip over the actual word */ while (string && *string && (*string < 0 || !isspace(*string))) string++; return string; } int pdf_add_text_wrap(struct pdf_doc *pdf, struct pdf_object *page, const char *text, float size, float xoff, float yoff, uint32_t colour, float wrap_width, int align, float *height) { /* Move through the text string, stopping at word boundaries, * trying to find the longest text string we can fit in the given width */ const char *start = text; const char *last_best = text; const char *end = text; char line[512]; const uint16_t *widths; float orig_yoff = yoff; widths = find_font_widths(pdf->current_font->font.name); if (!widths) return pdf_set_err(pdf, -EINVAL, "Unable to determine width for font '%s'", pdf->current_font->font.name); while (start && *start) { const char *new_end = find_word_break(end + 1); float line_width; int output = 0; float xoff_align = xoff; int e; end = new_end; e = pdf_text_point_width(pdf, start, end - start, size, widths, &line_width); if (e < 0) return e; if (line_width >= wrap_width) { if (last_best == start) { /* There is a single word that is too long for the line */ ptrdiff_t i; /* Find the best character to chop it at */ for (i = end - start - 1; i > 0; i--) { float this_width; // Don't look at places that are in the middle of a utf-8 // sequence if ((start[i - 1] & 0xc0) == 0xc0 || ((start[i - 1] & 0xc0) == 0x80 && (start[i] & 0xc0) == 0x80)) continue; e = pdf_text_point_width(pdf, start, i, size, widths, &this_width); if (e < 0) return e; if (this_width < wrap_width) break; } if (i == 0) return pdf_set_err(pdf, -EINVAL, "Unable to find suitable line break"); end = start + i; } else end = last_best; output = 1; } if (*end == '\0') output = 1; if (*end == '\n' || *end == '\r') output = 1; if (output) { int len = end - start; float char_spacing = 0; if (len >= (int)sizeof(line)) len = (int)sizeof(line) - 1; strncpy(line, start, len); line[len] = '\0'; e = pdf_text_point_width(pdf, start, len, size, widths, &line_width); if (e < 0) return e; switch (align) { case PDF_ALIGN_RIGHT: xoff_align += wrap_width - line_width; break; case PDF_ALIGN_CENTER: xoff_align += (wrap_width - line_width) / 2; break; case PDF_ALIGN_JUSTIFY: if ((len - 1) > 0 && *end != '\r' && *end != '\n' && *end != '\0') char_spacing = (wrap_width - line_width) / (len - 2); break; case PDF_ALIGN_JUSTIFY_ALL: if ((len - 1) > 0) char_spacing = (wrap_width - line_width) / (len - 2); break; } if (align != PDF_ALIGN_NO_WRITE) { pdf_add_text_spacing(pdf, page, line, size, xoff_align, yoff, colour, char_spacing); } if (*end == ' ') end++; start = last_best = end; yoff -= size; } else last_best = end; } if (height) *height = orig_yoff - yoff; return 0; } int pdf_add_line(struct pdf_doc *pdf, struct pdf_object *page, float x1, float y1, float x2, float y2, float width, uint32_t colour) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f w\r\n", width); dstr_printf(&str, "%f %f m\r\n", x1, y1); dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f RG\r\n", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f %f l S\r\n", x2, y2); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_cubic_bezier(struct pdf_doc *pdf, struct pdf_object *page, float x1, float y1, float x2, float y2, float xq1, float yq1, float xq2, float yq2, float width, uint32_t colour) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f w\r\n", width); dstr_printf(&str, "%f %f m\r\n", x1, y1); dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f RG\r\n", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f %f %f %f %f %f c S\r\n", xq1, yq1, xq2, yq2, x2, y2); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_quadratic_bezier(struct pdf_doc *pdf, struct pdf_object *page, float x1, float y1, float x2, float y2, float xq1, float yq1, float width, uint32_t colour) { float xc1 = x1 + (xq1 - x1) * (2.0f / 3.0f); float yc1 = y1 + (yq1 - y1) * (2.0f / 3.0f); float xc2 = x2 + (xq1 - x2) * (2.0f / 3.0f); float yc2 = y2 + (yq1 - y2) * (2.0f / 3.0f); return pdf_add_cubic_bezier(pdf, page, x1, y1, x2, y2, xc1, yc1, xc2, yc2, width, colour); } int pdf_add_custom_path(struct pdf_doc *pdf, struct pdf_object *page, struct pdf_path_operation *operations, int operation_count, float stroke_width, uint32_t stroke_colour, uint32_t fill_colour) { int ret; struct dstr str = INIT_DSTR; if (!PDF_IS_TRANSPARENT(fill_colour)) { dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f rg\r\n", PDF_RGB_R(fill_colour), PDF_RGB_G(fill_colour), PDF_RGB_B(fill_colour)); } dstr_printf(&str, "%f w\r\n", stroke_width); dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f RG\r\n", PDF_RGB_R(stroke_colour), PDF_RGB_G(stroke_colour), PDF_RGB_B(stroke_colour)); for (int i = 0; i < operation_count; i++) { struct pdf_path_operation operation = operations[i]; switch (operation.op) { case 'm': dstr_printf(&str, "%f %f m\r\n", operation.x1, operation.y1); break; case 'l': dstr_printf(&str, "%f %f l\r\n", operation.x1, operation.y1); break; case 'c': dstr_printf(&str, "%f %f %f %f %f %f c\r\n", operation.x1, operation.y1, operation.x2, operation.y2, operation.x3, operation.y3); break; case 'v': dstr_printf(&str, "%f %f %f %f v\r\n", operation.x1, operation.y1, operation.x2, operation.y2); break; case 'y': dstr_printf(&str, "%f %f %f %f y\r\n", operation.x1, operation.y1, operation.x2, operation.y2); break; case 'h': dstr_printf(&str, "h\r\n"); break; default: return pdf_set_err(pdf, -errno, "Invalid operation"); break; } } if (PDF_IS_TRANSPARENT(fill_colour)) dstr_printf(&str, "%s", "S "); else dstr_printf(&str, "%s", "B "); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_ellipse(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float xradius, float yradius, float width, uint32_t colour, uint32_t fill_colour) { int ret; struct dstr str = INIT_DSTR; float lx, ly; lx = (4.0f / 3.0f) * (M_SQRT2 - 1) * xradius; ly = (4.0f / 3.0f) * (M_SQRT2 - 1) * yradius; if (!PDF_IS_TRANSPARENT(fill_colour)) { dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f rg\r\n", PDF_RGB_R(fill_colour), PDF_RGB_G(fill_colour), PDF_RGB_B(fill_colour)); } /* stroke color */ dstr_printf(&str, "/DeviceRGB CS\r\n"); dstr_printf(&str, "%f %f %f RG\r\n", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f w ", width); dstr_printf(&str, "%.2f %.2f m ", (x + xradius), (y)); dstr_printf(&str, "%.2f %.2f %.2f %.2f %.2f %.2f c ", (x + xradius), (y - ly), (x + lx), (y - yradius), x, (y - yradius)); dstr_printf(&str, "%.2f %.2f %.2f %.2f %.2f %.2f c ", (x - lx), (y - yradius), (x - xradius), (y - ly), (x - xradius), y); dstr_printf(&str, "%.2f %.2f %.2f %.2f %.2f %.2f c ", (x - xradius), (y + ly), (x - lx), (y + yradius), x, (y + yradius)); dstr_printf(&str, "%.2f %.2f %.2f %.2f %.2f %.2f c ", (x + lx), (y + yradius), (x + xradius), (y + ly), (x + xradius), y); if (PDF_IS_TRANSPARENT(fill_colour)) dstr_printf(&str, "%s", "S "); else dstr_printf(&str, "%s", "B "); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_circle(struct pdf_doc *pdf, struct pdf_object *page, float xr, float yr, float radius, float width, uint32_t colour, uint32_t fill_colour) { return pdf_add_ellipse(pdf, page, xr, yr, radius, radius, width, colour, fill_colour); } int pdf_add_rectangle(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, float border_width, uint32_t colour) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f %f %f RG ", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f w ", border_width); dstr_printf(&str, "%f %f %f %f re S ", x, y, width, height); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_filled_rectangle(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, float border_width, uint32_t colour_fill, uint32_t colour_border) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f %f %f rg ", PDF_RGB_R(colour_fill), PDF_RGB_G(colour_fill), PDF_RGB_B(colour_fill)); if (border_width > 0) { dstr_printf(&str, "%f %f %f RG ", PDF_RGB_R(colour_border), PDF_RGB_G(colour_border), PDF_RGB_B(colour_border)); dstr_printf(&str, "%f w ", border_width); dstr_printf(&str, "%f %f %f %f re B ", x, y, width, height); } else { dstr_printf(&str, "%f %f %f %f re f ", x, y, width, height); } ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_polygon(struct pdf_doc *pdf, struct pdf_object *page, float x[], float y[], int count, float border_width, uint32_t colour) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f %f %f RG ", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f w ", border_width); dstr_printf(&str, "%f %f m ", x[0], y[0]); for (int i = 1; i < count; i++) { dstr_printf(&str, "%f %f l ", x[i], y[i]); } dstr_printf(&str, "h S "); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } int pdf_add_filled_polygon(struct pdf_doc *pdf, struct pdf_object *page, float x[], float y[], int count, float border_width, uint32_t colour) { int ret; struct dstr str = INIT_DSTR; dstr_printf(&str, "%f %f %f RG ", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f %f %f rg ", PDF_RGB_R(colour), PDF_RGB_G(colour), PDF_RGB_B(colour)); dstr_printf(&str, "%f w ", border_width); dstr_printf(&str, "%f %f m ", x[0], y[0]); for (int i = 1; i < count; i++) { dstr_printf(&str, "%f %f l ", x[i], y[i]); } dstr_printf(&str, "h f "); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } static const struct { uint32_t code; char ch; } code_128a_encoding[] = { {0x212222, ' '}, {0x222122, '!'}, {0x222221, '"'}, {0x121223, '#'}, {0x121322, '$'}, {0x131222, '%'}, {0x122213, '&'}, {0x122312, '\''}, {0x132212, '('}, {0x221213, ')'}, {0x221312, '*'}, {0x231212, '+'}, {0x112232, ','}, {0x122132, '-'}, {0x122231, '.'}, {0x113222, '/'}, {0x123122, '0'}, {0x123221, '1'}, {0x223211, '2'}, {0x221132, '3'}, {0x221231, '4'}, {0x213212, '5'}, {0x223112, '6'}, {0x312131, '7'}, {0x311222, '8'}, {0x321122, '9'}, {0x321221, ':'}, {0x312212, ';'}, {0x322112, '<'}, {0x322211, '='}, {0x212123, '>'}, {0x212321, '?'}, {0x232121, '@'}, {0x111323, 'A'}, {0x131123, 'B'}, {0x131321, 'C'}, {0x112313, 'D'}, {0x132113, 'E'}, {0x132311, 'F'}, {0x211313, 'G'}, {0x231113, 'H'}, {0x231311, 'I'}, {0x112133, 'J'}, {0x112331, 'K'}, {0x132131, 'L'}, {0x113123, 'M'}, {0x113321, 'N'}, {0x133121, 'O'}, {0x313121, 'P'}, {0x211331, 'Q'}, {0x231131, 'R'}, {0x213113, 'S'}, {0x213311, 'T'}, {0x213131, 'U'}, {0x311123, 'V'}, {0x311321, 'W'}, {0x331121, 'X'}, {0x312113, 'Y'}, {0x312311, 'Z'}, {0x332111, '['}, {0x314111, '\\'}, {0x221411, ']'}, {0x431111, '^'}, {0x111224, '_'}, {0x111422, '`'}, {0x121124, 'a'}, {0x121421, 'b'}, {0x141122, 'c'}, {0x141221, 'd'}, {0x112214, 'e'}, {0x112412, 'f'}, {0x122114, 'g'}, {0x122411, 'h'}, {0x142112, 'i'}, {0x142211, 'j'}, {0x241211, 'k'}, {0x221114, 'l'}, {0x413111, 'm'}, {0x241112, 'n'}, {0x134111, 'o'}, {0x111242, 'p'}, {0x121142, 'q'}, {0x121241, 'r'}, {0x114212, 's'}, {0x124112, 't'}, {0x124211, 'u'}, {0x411212, 'v'}, {0x421112, 'w'}, {0x421211, 'x'}, {0x212141, 'y'}, {0x214121, 'z'}, {0x412121, '{'}, {0x111143, '|'}, {0x111341, '}'}, {0x131141, '~'}, {0x114113, '\0'}, {0x114311, '\0'}, {0x411113, '\0'}, {0x411311, '\0'}, {0x113141, '\0'}, {0x114131, '\0'}, {0x311141, '\0'}, {0x411131, '\0'}, {0x211412, '\0'}, {0x211214, '\0'}, {0x211232, '\0'}, {0x2331112, '\0'}, }; static int find_128_encoding(char ch) { for (int i = 0; i < (int)ARRAY_SIZE(code_128a_encoding); i++) { if (code_128a_encoding[i].ch == ch) return i; } return -1; } static float pdf_barcode_128a_ch(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, uint32_t colour, int index, int code_len) { uint32_t code = code_128a_encoding[index].code; float line_width = width / 11.0f; for (int i = 0; i < code_len; i++) { uint8_t shift = (code_len - 1 - i) * 4; uint8_t mask = (code >> shift) & 0xf; if (!(i % 2)) pdf_add_filled_rectangle(pdf, page, x, y, line_width * mask, height, 0, colour, PDF_TRANSPARENT); x += line_width * mask; } return x; } static int pdf_add_barcode_128a(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { const char *s; size_t len = strlen(string) + 3; float char_width = width / len; int checksum, i; if (char_width / 11.0f <= 0) return pdf_set_err(pdf, -EINVAL, "Insufficient width to draw barcode"); for (s = string; *s; s++) if (find_128_encoding(*s) < 0) return pdf_set_err(pdf, -EINVAL, "Invalid barcode character 0x%x", *s); x = pdf_barcode_128a_ch(pdf, page, x, y, char_width, height, colour, 104, 6); checksum = 104; for (i = 1, s = string; *s; s++, i++) { int index = find_128_encoding(*s); // This should be impossible, due to the checks above, but confirm // here anyway to stop coverity complaining if (index < 0) return pdf_set_err(pdf, -EINVAL, "Invalid 128a barcode character 0x%x", *s); x = pdf_barcode_128a_ch(pdf, page, x, y, char_width, height, colour, index, 6); checksum += index * i; } x = pdf_barcode_128a_ch(pdf, page, x, y, char_width, height, colour, checksum % 103, 6); pdf_barcode_128a_ch(pdf, page, x, y, char_width, height, colour, 106, 7); return 0; } /* Code 39 character encoding. Each 4-bit value indicates: * 0 => wide bar * 1 => narrow bar * 2 => wide space */ static const struct { int code; char ch; } code_39_encoding[] = { {0x012110, '1'}, {0x102110, '2'}, {0x002111, '3'}, {0x112010, '4'}, {0x012011, '5'}, {0x102011, '6'}, {0x112100, '7'}, {0x012101, '8'}, {0x102101, '9'}, {0x112001, '0'}, {0x011210, 'A'}, {0x101210, 'B'}, {0x001211, 'C'}, {0x110210, 'D'}, {0x010211, 'E'}, {0x100211, 'F'}, {0x111200, 'G'}, {0x011201, 'H'}, {0x101201, 'I'}, {0x110201, 'J'}, {0x011120, 'K'}, {0x101120, 'L'}, {0x001121, 'M'}, {0x110120, 'N'}, {0x010121, 'O'}, {0x100121, 'P'}, {0x111020, 'Q'}, {0x011021, 'R'}, {0x101021, 'S'}, {0x110021, 'T'}, {0x021110, 'U'}, {0x120110, 'V'}, {0x020111, 'W'}, {0x121010, 'X'}, {0x021011, 'Y'}, {0x120011, 'Z'}, {0x121100, '-'}, {0x021101, '.'}, {0x120101, ' '}, {0x121001, '*'}, // 'stop' character }; static int find_39_encoding(char ch) { for (int i = 0; i < (int)ARRAY_SIZE(code_39_encoding); i++) { if (code_39_encoding[i].ch == ch) { return code_39_encoding[i].code; } } return -1; } static int pdf_barcode_39_ch(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float char_width, float height, uint32_t colour, char ch, float *new_x) { float nw = char_width / 12.0f; float ww = char_width / 4.0f; int code = 0; code = find_39_encoding(ch); if (code < 0) return pdf_set_err(pdf, -EINVAL, "Invalid Code 39 character %c 0x%x", ch, ch); for (int i = 5; i >= 0; i--) { int pattern = (code >> i * 4) & 0xf; if (pattern == 0) { // wide if (pdf_add_filled_rectangle(pdf, page, x, y, ww - 1, height, 0, colour, PDF_TRANSPARENT) < 0) return pdf->errval; x += ww; } if (pattern == 1) { // narrow if (pdf_add_filled_rectangle(pdf, page, x, y, nw - 1, height, 0, colour, PDF_TRANSPARENT) < 0) return pdf->errval; x += nw; } if (pattern == 2) { // space x += nw; } } if (new_x) *new_x = x; return 0; } static int pdf_add_barcode_39(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { size_t len = strlen(string); float char_width = width / (len + 2); int e; e = pdf_barcode_39_ch(pdf, page, x, y, char_width, height, colour, '*', &x); if (e < 0) return e; while (string && *string) { e = pdf_barcode_39_ch(pdf, page, x, y, char_width, height, colour, *string, &x); if (e < 0) return e; string++; } e = pdf_barcode_39_ch(pdf, page, x, y, char_width, height, colour, '*', NULL); if (e < 0) return e; return 0; } /* EAN/UPC character encoding. Each 4-bit value indicates width in x units. * Elements are SBSB (Space, Bar, Space, Bar) for LHS digits * Elements are inverted for RHS digits */ static const int code_eanupc_encoding[] = { 0x3211, // 0 0x2221, // 1 0x2122, // 2 0x1411, // 3 0x1132, // 4 0x1231, // 5 0x1114, // 6 0x1312, // 7 0x1213, // 8 0x3112, // 9 }; /** * List of different barcode guard patterns that are supported */ enum { GUARD_NORMAL, //!< Produce normal guard pattern GUARD_CENTRE, //!< Produce centre guard pattern GUARD_SPECIAL, //!< Produce special guard pattern GUARD_ADDON, GUARD_ADDON_DELIN, }; static const int code_eanupc_aux_encoding[] = { 0x150, // Normal guard 0x554, // Centre guard 0x555, // Special guard 0x160, // Add-on guard 0x500, // Add-on delineator }; static const int set_ean13_encoding[] = { 0x00, // 0 0x34, // 1 0x2c, // 2 0x1c, // 3 0x32, // 4 0x26, // 5 0x0e, // 6 0x2a, // 7 0x1a, // 8 0x16, // 9 }; static const int set_upce_encoding[] = { 0x07, // 0 0x0b, // 1 0x13, // 2 0x23, // 3 0x0d, // 4 0x19, // 5 0x31, // 6 0x15, // 7 0x25, // 8 0x29, // 9 }; #define EANUPC_X PDF_MM_TO_POINT(0.330f) static const struct { unsigned modules; float height_bar; float height_outer; unsigned quiet_left; unsigned quiet_right; } eanupc_dimensions[] = { {113, PDF_MM_TO_POINT(22.85), PDF_MM_TO_POINT(25.93), 11, 7}, // EAN-13 {113, PDF_MM_TO_POINT(22.85), PDF_MM_TO_POINT(25.93), 9, 9}, // UPC-A {81, PDF_MM_TO_POINT(18.23), PDF_MM_TO_POINT(21.31), 7, 7}, // EAN-8 {67, PDF_MM_TO_POINT(22.85), PDF_MM_TO_POINT(25.93), 9, 7}, // UPC-E }; static void pdf_barcode_eanupc_calc_dims(int type, float width, float height, float *x_off, float *y_off, float *new_width, float *new_height, float *x, float *bar_height, float *bar_ext, float *font_size) { float aspectBarcode, aspectRect, scale; aspectRect = width / height; aspectBarcode = eanupc_dimensions[type - PDF_BARCODE_EAN13].modules * EANUPC_X / eanupc_dimensions[type - PDF_BARCODE_EAN13].height_outer; if (aspectRect > aspectBarcode) { *new_height = height; *new_width = height * aspectBarcode; } else if (aspectRect < aspectBarcode) { *new_width = width; *new_height = width / aspectBarcode; } else { *new_width = width; *new_height = height; } scale = *new_height / eanupc_dimensions[type - PDF_BARCODE_EAN13].height_outer; *x = *new_width / eanupc_dimensions[type - PDF_BARCODE_EAN13].modules; *bar_ext = *x * 5; *bar_height = eanupc_dimensions[type - PDF_BARCODE_EAN13].height_bar * scale; *font_size = 8.0 * scale; *x_off = (width - *new_width) / 2.0; *y_off = (height - *new_height) / 2.0; } static int pdf_barcode_eanupc_ch(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float x_width, float height, uint32_t colour, char ch, int set, float *new_x) { if ('0' > ch || ch > '9') return pdf_set_err(pdf, -EINVAL, "Invalid EAN/UPC character %c 0x%x", ch, ch); int code; code = code_eanupc_encoding[ch - '0']; for (int i = 3; i >= 0; i--) { int shift = (set == 1 ? 3 - i : i) * 4; int bar = (set == 2 && i & 0x1) || (set != 2 && (i & 0x1) == 0); float width = (code >> shift) & 0xf; switch (ch) { case '1': case '2': if ((set == 0 && bar) || (set != 0 && !bar)) { width -= 1.0 / 13.0; } else { width += 1.0 / 13.0; } break; case '7': case '8': if ((set == 0 && bar) || (set != 0 && !bar)) { width += 1.0 / 13.0; } else { width -= 1.0 / 13.0; } break; } width *= x_width; if (bar) { if (pdf_add_filled_rectangle(pdf, page, x, y, width, height, 0, colour, PDF_TRANSPARENT) < 0) return pdf->errval; } x += width; } if (new_x) *new_x = x; return 0; } static int pdf_barcode_eanupc_aux(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float x_width, float height, uint32_t colour, int guard_type, float *new_x) { int code = code_eanupc_aux_encoding[guard_type]; for (int i = 5; i >= 0; i--) { int value = code >> i * 2 & 0x3; if (value) { if ((i & 0x1) == 0) { if (pdf_add_filled_rectangle(pdf, page, x, y, x_width * value, height, 0, colour, PDF_TRANSPARENT) < 0) return pdf->errval; } x += x_width * value; } } if (new_x) *new_x = x; return 0; } static int pdf_add_barcode_ean13(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { if (!string) return 0; size_t len = strlen(string); int lead = 0; if (len == 13) { char ch = string[0]; if (!isdigit(*string)) return pdf_set_err(pdf, -EINVAL, "Invalid EAN13 character %c 0x%x", ch, ch); lead = ch - '0'; ++string; } else if (len != 12) return pdf_set_err(pdf, -EINVAL, "Invalid EAN13 string length %lu", len); /* Scale and calculate dimensions */ float x_off, y_off, new_width, new_height, x_width, bar_height, bar_ext, font; pdf_barcode_eanupc_calc_dims(PDF_BARCODE_EAN13, width, height, &x_off, &y_off, &new_width, &new_height, &x_width, &bar_height, &bar_ext, &font); x += x_off; y += y_off; float bar_y = y + new_height - bar_height; int e; const char *save_font = pdf->current_font->font.name; e = pdf_set_font(pdf, "Courier"); /* Built-in monospace font */ if (e < 0) return e; char text[2]; text[1] = 0; text[0] = lead + '0'; e = pdf_add_text(pdf, page, text, font, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } x += eanupc_dimensions[0].quiet_left * x_width; e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 6; i++) { text[0] = *string; e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } int set = (set_ean13_encoding[lead] & 1 << i) ? 1 : 0; e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y, x_width, bar_height, colour, *string, set, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_CENTRE, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 6; i++) { text[0] = *string; e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y, x_width, bar_height, colour, *string, 2, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } text[0] = '>'; x += eanupc_dimensions[0].quiet_right * x_width - 604.0f * font / (14.0f * 72.0f); e = pdf_add_text(pdf, page, text, font, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } pdf_set_font(pdf, save_font); return 0; } static int pdf_add_barcode_upca(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { if (!string) return 0; size_t len = strlen(string); if (len != 12) return pdf_set_err(pdf, -EINVAL, "Invalid UPCA string length %lu", len); /* Scale and calculate dimensions */ float x_off, y_off, new_width, new_height; float x_width, bar_height, bar_ext, font; pdf_barcode_eanupc_calc_dims(PDF_BARCODE_UPCA, width, height, &x_off, &y_off, &new_width, &new_height, &x_width, &bar_height, &bar_ext, &font); x += x_off; y += y_off; float bar_y = y + new_height - bar_height; int e; const char *save_font = pdf->current_font->font.name; e = pdf_set_font(pdf, "Courier"); if (e < 0) return e; char text[2]; text[1] = 0; text[0] = *string; e = pdf_add_text(pdf, page, text, font * 4.0 / 7.0, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } x += eanupc_dimensions[1].quiet_left * x_width; e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 6; i++) { text[0] = *string; if (i) { e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } } e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y - (i ? 0 : bar_ext), x_width, bar_height + (i ? 0 : bar_ext), colour, *string, 0, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_CENTRE, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 6; i++) { text[0] = *string; if (i != 5) { e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } } e = pdf_barcode_eanupc_ch( pdf, page, x, bar_y - (i != 5 ? 0 : bar_ext), x_width, bar_height + (i != 5 ? 0 : bar_ext), colour, *string, 2, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } text[0] = *--string; x += eanupc_dimensions[1].quiet_right * x_width - 604.0f * font * 4.0f / 7.0f / (14.0f * 72.0f); e = pdf_add_text(pdf, page, text, font * 4.0 / 7.0, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } pdf_set_font(pdf, save_font); return 0; } static int pdf_add_barcode_ean8(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { if (!string) return 0; size_t len = strlen(string); if (len != 8) return pdf_set_err(pdf, -EINVAL, "Invalid EAN8 string length %lu", len); /* Scale and calculate dimensions */ float x_off, y_off, new_width, new_height, x_width, bar_height, bar_ext, font; pdf_barcode_eanupc_calc_dims(PDF_BARCODE_EAN8, width, height, &x_off, &y_off, &new_width, &new_height, &x_width, &bar_height, &bar_ext, &font); x += x_off; y += y_off; float bar_y = y + new_height - bar_height; int e; const char *save_font = pdf->current_font->font.name; e = pdf_set_font(pdf, "Courier"); /* Built-in monospace font */ if (e < 0) return e; char text[2]; text[1] = 0; text[0] = '<'; e = pdf_add_text(pdf, page, text, font, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } x += eanupc_dimensions[2].quiet_left * x_width; e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 4; i++) { text[0] = *string; e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y, x_width, bar_height, colour, *string, 0, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_CENTRE, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } for (int i = 0; i != 4; i++) { text[0] = *string; e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y, x_width, bar_height, colour, *string, 2, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } string++; } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y - bar_ext, x_width, bar_height + bar_ext, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } text[0] = '>'; x += eanupc_dimensions[0].quiet_right * x_width - 604.0f * font / (14.0f * 72.0f); e = pdf_add_text(pdf, page, text, font, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } pdf_set_font(pdf, save_font); return 0; } static int pdf_add_barcode_upce(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float width, float height, const char *string, uint32_t colour) { if (!string) return 0; size_t len = strlen(string); if (len != 12) return pdf_set_err(pdf, -EINVAL, "Invalid UPCE string length %lu", len); if (*string != '0') return pdf_set_err(pdf, -EINVAL, "Invalid UPCE char %c at start", *string); for (size_t i = 0; i < len; i++) { if (!isdigit(string[i])) return pdf_set_err(pdf, -EINVAL, "Invalid UPCE char 0x%x at %zd", string[i], i); } /* Scale and calculate dimensions */ float x_off, y_off, new_width, new_height; float x_width, bar_height, bar_ext, font; pdf_barcode_eanupc_calc_dims(PDF_BARCODE_UPCE, width, height, &x_off, &y_off, &new_width, &new_height, &x_width, &bar_height, &bar_ext, &font); x += x_off; y += y_off; float bar_y = y + new_height - bar_height; int e; const char *save_font = pdf->current_font->font.name; e = pdf_set_font(pdf, "Courier"); if (e < 0) return e; char text[2]; text[1] = 0; text[0] = string[0]; e = pdf_add_text(pdf, page, text, font * 4.0 / 7.0, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } x += eanupc_dimensions[2].quiet_left * x_width; e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y, x_width, bar_height, colour, GUARD_NORMAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } char X[6]; if (string[5] && memcmp(string + 6, "0000", 4) == 0 && '5' <= string[10] && string[10] <= '9') { memcpy(X, string + 1, 5); X[5] = string[10]; } else if (string[4] && memcmp(string + 5, "00000", 5) == 0) { memcpy(X, string + 1, 4); X[4] = string[11]; X[5] = 4; } else if ('0' <= string[3] && string[3] <= '2' && memcmp(string + 4, "0000", 4) == 0) { X[0] = string[1]; X[1] = string[2]; X[2] = string[8]; X[3] = string[9]; X[4] = string[10]; X[5] = string[3]; } else if ('3' <= string[3] && string[3] <= '9' && memcmp(string + 4, "00000", 5) == 0) { memcpy(X, string + 1, 3); X[3] = string[9]; X[4] = string[10]; X[5] = 3; } else { pdf_set_font(pdf, save_font); return pdf_set_err(pdf, -EINVAL, "Invalid UPCE string format"); } for (int i = 0; i != 6; i++) { text[0] = X[i]; e = pdf_add_text_wrap(pdf, page, text, font, x, y, colour, 7 * x_width, PDF_ALIGN_CENTER, NULL); if (e < 0) { pdf_set_font(pdf, save_font); return e; } int set = (set_upce_encoding[string[11] - '0'] & 1 << i) ? 1 : 0; e = pdf_barcode_eanupc_ch(pdf, page, x, bar_y, x_width, bar_height, colour, X[i], set, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } } e = pdf_barcode_eanupc_aux(pdf, page, x, bar_y, x_width, bar_height, colour, GUARD_SPECIAL, &x); if (e < 0) { pdf_set_font(pdf, save_font); return e; } text[0] = string[11]; x += eanupc_dimensions[0].quiet_right * x_width - 604.0f * font * 4.0f / 7.0f / (14.0f * 72.0f); e = pdf_add_text(pdf, page, text, font * 4.0 / 7.0, x, y, colour); if (e < 0) { pdf_set_font(pdf, save_font); return e; } pdf_set_font(pdf, save_font); return 0; } int pdf_add_barcode(struct pdf_doc *pdf, struct pdf_object *page, int code, float x, float y, float width, float height, const char *string, uint32_t colour) { if (!string || !*string) return 0; switch (code) { case PDF_BARCODE_128A: return pdf_add_barcode_128a(pdf, page, x, y, width, height, string, colour); case PDF_BARCODE_39: return pdf_add_barcode_39(pdf, page, x, y, width, height, string, colour); case PDF_BARCODE_EAN13: return pdf_add_barcode_ean13(pdf, page, x, y, width, height, string, colour); case PDF_BARCODE_UPCA: return pdf_add_barcode_upca(pdf, page, x, y, width, height, string, colour); case PDF_BARCODE_EAN8: return pdf_add_barcode_ean8(pdf, page, x, y, width, height, string, colour); case PDF_BARCODE_UPCE: return pdf_add_barcode_upce(pdf, page, x, y, width, height, string, colour); default: return pdf_set_err(pdf, -EINVAL, "Invalid barcode code %d", code); } } static pdf_object *pdf_add_raw_grayscale8(struct pdf_doc *pdf, const uint8_t *data, uint32_t width, uint32_t height) { struct pdf_object *obj; size_t len; const char *endstream = ">\r\nendstream\r\n"; struct dstr str = INIT_DSTR; size_t data_len = (size_t)width * (size_t)height; dstr_printf(&str, "<<\r\n" " /Type /XObject\r\n" " /Name /Image%d\r\n" " /Subtype /Image\r\n" " /ColorSpace /DeviceGray\r\n" " /Height %d\r\n" " /Width %d\r\n" " /BitsPerComponent 8\r\n" " /Length %zu\r\n" ">>stream\r\n", flexarray_size(&pdf->objects), height, width, data_len + 1); len = dstr_len(&str) + data_len + strlen(endstream) + 1; if (dstr_ensure(&str, len) < 0) { dstr_free(&str); pdf_set_err(pdf, -ENOMEM, "Unable to allocate %zu bytes memory for image", len); return NULL; } dstr_append_data(&str, data, data_len); dstr_append(&str, endstream); obj = pdf_add_object(pdf, OBJ_image); if (!obj) { dstr_free(&str); return NULL; } obj->stream = str; return obj; } static struct pdf_object *pdf_add_raw_rgb24(struct pdf_doc *pdf, const uint8_t *data, uint32_t width, uint32_t height) { struct pdf_object *obj; size_t len; const char *endstream = ">\r\nendstream\r\n"; struct dstr str = INIT_DSTR; size_t data_len = (size_t)width * (size_t)height * 3; dstr_printf(&str, "<<\r\n" " /Type /XObject\r\n" " /Name /Image%d\r\n" " /Subtype /Image\r\n" " /ColorSpace /DeviceRGB\r\n" " /Height %d\r\n" " /Width %d\r\n" " /BitsPerComponent 8\r\n" " /Length %zu\r\n" ">>stream\r\n", flexarray_size(&pdf->objects), height, width, data_len + 1); len = dstr_len(&str) + data_len + strlen(endstream) + 1; if (dstr_ensure(&str, len) < 0) { dstr_free(&str); pdf_set_err(pdf, -ENOMEM, "Unable to allocate %zu bytes memory for image", len); return NULL; } dstr_append_data(&str, data, data_len); dstr_append(&str, endstream); obj = pdf_add_object(pdf, OBJ_image); if (!obj) { dstr_free(&str); return NULL; } obj->stream = str; return obj; } static uint8_t *get_file(struct pdf_doc *pdf, const char *file_name, size_t *length) { FILE *fp; uint8_t *file_data; struct stat buf; off_t len; if ((fp = fopen(file_name, "rb")) == NULL) { pdf_set_err(pdf, -errno, "Unable to open %s: %s", file_name, strerror(errno)); return NULL; } if (fstat(fileno(fp), &buf) < 0) { pdf_set_err(pdf, -errno, "Unable to access %s: %s", file_name, strerror(errno)); fclose(fp); return NULL; } len = buf.st_size; file_data = (uint8_t *)malloc(len); if (!file_data) { pdf_set_err(pdf, -ENOMEM, "Unable to allocate: %d", (int)len); fclose(fp); return NULL; } if (fread(file_data, len, 1, fp) != 1) { pdf_set_err(pdf, -errno, "Unable to read full data: %s", file_name); free(file_data); fclose(fp); return NULL; } fclose(fp); *length = len; return file_data; } static struct pdf_object * pdf_add_raw_jpeg_data(struct pdf_doc *pdf, const struct pdf_img_info *info, const uint8_t *jpeg_data, size_t len) { struct pdf_object *obj = pdf_add_object(pdf, OBJ_image); if (!obj) return NULL; dstr_printf(&obj->stream, "<<\r\n" " /Type /XObject\r\n" " /Name /Image%d\r\n" " /Subtype /Image\r\n" " /ColorSpace %s\r\n" " /Width %d\r\n" " /Height %d\r\n" " /BitsPerComponent 8\r\n" " /Filter /DCTDecode\r\n" " /Length %zu\r\n" ">>stream\r\n", flexarray_size(&pdf->objects), (info->jpeg.ncolours == 1) ? "/DeviceGray" : "/DeviceRGB", info->width, info->height, len); dstr_append_data(&obj->stream, jpeg_data, len); dstr_printf(&obj->stream, "\r\nendstream\r\n"); return obj; } /** * Get the display dimensions of an image, respecting the images aspect ratio * if only one desired display dimension is defined. * The pdf parameter is only used for setting the error value. */ static int get_img_display_dimensions(struct pdf_doc *pdf, uint32_t img_width, uint32_t img_height, float *display_width, float *display_height) { if (!display_height || !display_width) { return pdf_set_err( pdf, -EINVAL, "display_width and display_height may not be null pointers"); } const float display_width_in = *display_width; const float display_height_in = *display_height; if (display_width_in < 0 && display_height_in < 0) { return pdf_set_err(pdf, -EINVAL, "Unable to determine image display dimensions, " "display_width and display_height are both < 0"); } if (img_width == 0 || img_height == 0) { return pdf_set_err(pdf, -EINVAL, "Invalid image dimensions received, the loaded " "image appears to be empty."); } if (display_width_in < 0) { // Set width, keeping aspect ratio *display_width = display_height_in * ((float)img_width / img_height); } else if (display_height_in < 0) { // Set height, keeping aspect ratio *display_height = display_width_in * ((float)img_height / img_width); } return 0; } static int pdf_add_image(struct pdf_doc *pdf, struct pdf_object *page, struct pdf_object *image, float x, float y, float width, float height) { int ret; struct dstr str = INIT_DSTR; dstr_append(&str, "q "); dstr_printf(&str, "%f 0 0 %f %f %f cm ", width, height, x, y); dstr_printf(&str, "/Image%d Do ", image->index); dstr_append(&str, "Q"); ret = pdf_add_stream(pdf, page, dstr_data(&str)); dstr_free(&str); return ret; } // Works like fgets, except it's for a fixed in-memory buffer of data static size_t dgets(const uint8_t *data, size_t *pos, size_t len, char *line, size_t line_len) { size_t line_pos = 0; if (*pos >= len) return 0; while ((*pos) < len) { if (line_pos < line_len) { // Reject non-ascii data if (data[*pos] & 0x80) { return 0; } line[line_pos] = data[*pos]; line_pos++; } if (data[*pos] == '\n') { (*pos)++; break; } (*pos)++; } if (line_pos < line_len) { line[line_pos] = '\0'; } return *pos; } static int parse_ppm_header(struct pdf_img_info *info, const uint8_t *data, size_t length, char *err_msg, size_t err_msg_length) { char line[1024]; memset(line, '\0', sizeof(line)); size_t pos = 0; // Load the PPM file if (!dgets(data, &pos, length, line, sizeof(line) - 1)) { snprintf(err_msg, err_msg_length, "Invalid PPM file"); return -EINVAL; } // Determine number of color channels (Also, we only support binary ppms) int ncolors; if (strncmp(line, "P6", 2) == 0) { info->ppm.color_space = PPM_BINARY_COLOR_RGB; ncolors = 3; } else if (strncmp(line, "P5", 2) == 0) { info->ppm.color_space = PPM_BINARY_COLOR_GRAY; ncolors = 1; } else { snprintf(err_msg, err_msg_length, "Only binary PPM files (grayscale, RGB) supported"); return -EINVAL; } // Skip comments before header do { if (!dgets(data, &pos, length, line, sizeof(line) - 1)) { snprintf(err_msg, err_msg_length, "Unable to find PPM header"); return -EINVAL; } } while (line[0] == '#'); // Read image dimensions if (sscanf(line, "%u %u\n", &(info->width), &(info->height)) != 2) { snprintf(err_msg, err_msg_length, "Unable to find PPM size"); return -EINVAL; } if (info->width > MAX_IMAGE_WIDTH || info->height > MAX_IMAGE_HEIGHT) { snprintf(err_msg, err_msg_length, "Invalid width/height: %ux%u", info->width, info->height); return -EINVAL; } info->ppm.size = (size_t)(info->width * info->height * ncolors); info->ppm.data_begin_pos = pos; return 0; } static int pdf_add_ppm_data(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const struct pdf_img_info *info, const uint8_t *ppm_data, size_t len) { char line[1024]; // We start reading at the position delivered by parse_ppm_header, // since we already parsed the header of the file there. size_t pos = info->ppm.data_begin_pos; /* Skip over the byte-size line */ if (!dgets(ppm_data, &pos, len, line, sizeof(line) - 1)) return pdf_set_err(pdf, -EINVAL, "No byte-size line in PPM file"); /* Try and limit the memory usage to sane images */ if (info->width > MAX_IMAGE_WIDTH || info->height > MAX_IMAGE_HEIGHT) { return pdf_set_err(pdf, -EINVAL, "Invalid width/height in PPM file: %ux%u", info->width, info->height); } if (info->ppm.size > len - pos) { return pdf_set_err(pdf, -EINVAL, "Insufficient image data available"); } switch (info->ppm.color_space) { case PPM_BINARY_COLOR_GRAY: return pdf_add_grayscale8(pdf, page, x, y, display_width, display_height, &ppm_data[pos], info->width, info->height); break; case PPM_BINARY_COLOR_RGB: return pdf_add_rgb24(pdf, page, x, y, display_width, display_height, &ppm_data[pos], info->width, info->height); break; default: return pdf_set_err(pdf, -EINVAL, "Invalid color space in ppm file: %i", info->ppm.color_space); break; } } static int parse_jpeg_header(struct pdf_img_info *info, const uint8_t *data, size_t length, char *err_msg, size_t err_msg_length) { // See http://www.videotechnology.com/jpeg/j1.html for details if (length >= 4 && data[0] == 0xFF && data[1] == 0xD8) { for (size_t i = 2; i < length; i++) { if (data[i] != 0xff) { break; } while (++i < length && data[i] == 0xff) ; if (i + 2 >= length) { break; } int len = data[i + 1] * 256 + data[i + 2]; /* Search for SOFn marker and decode jpeg details */ if ((data[i] & 0xf4) == 0xc0) { if (len >= 9 && i + len + 1 < length) { info->height = data[i + 4] * 256 + data[i + 5]; info->width = data[i + 6] * 256 + data[i + 7]; info->jpeg.ncolours = data[i + 8]; return 0; } break; } i += len; } } snprintf(err_msg, err_msg_length, "Error parsing JPEG header"); return -EINVAL; } static int pdf_add_jpeg_data(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, struct pdf_img_info *info, const uint8_t *jpeg_data, size_t len) { struct pdf_object *obj; obj = pdf_add_raw_jpeg_data(pdf, info, jpeg_data, len); if (!obj) return pdf->errval; if (get_img_display_dimensions(pdf, info->width, info->height, &display_width, &display_height)) { return pdf->errval; } return pdf_add_image(pdf, page, obj, x, y, display_width, display_height); } int pdf_add_rgb24(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const uint8_t *data, uint32_t width, uint32_t height) { struct pdf_object *obj; obj = pdf_add_raw_rgb24(pdf, data, width, height); if (!obj) return pdf->errval; if (get_img_display_dimensions(pdf, width, height, &display_width, &display_height)) { return pdf->errval; } return pdf_add_image(pdf, page, obj, x, y, display_width, display_height); } int pdf_add_grayscale8(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const uint8_t *data, uint32_t width, uint32_t height) { struct pdf_object *obj; obj = pdf_add_raw_grayscale8(pdf, data, width, height); if (!obj) return pdf->errval; if (get_img_display_dimensions(pdf, width, height, &display_width, &display_height)) { return pdf->errval; } return pdf_add_image(pdf, page, obj, x, y, display_width, display_height); } static int parse_png_header(struct pdf_img_info *info, const uint8_t *data, size_t length, char *err_msg, size_t err_msg_length) { if (length <= sizeof(png_signature)) { snprintf(err_msg, err_msg_length, "PNG file too short"); return -EINVAL; } if (memcmp(data, png_signature, sizeof(png_signature))) { snprintf(err_msg, err_msg_length, "File is not correct PNG file"); return -EINVAL; } // process first PNG chunk uint32_t pos = sizeof(png_signature); const struct png_chunk *chunk = (const struct png_chunk *)&data[pos]; pos += sizeof(struct png_chunk); if (pos > length) { snprintf(err_msg, err_msg_length, "PNG file too short"); return -EINVAL; } if (strncmp(chunk->type, png_chunk_header, 4) == 0) { // header found, process width and height, check errors struct png_header *header = &info->png; if (pos + sizeof(struct png_header) > length) { snprintf(err_msg, err_msg_length, "PNG file too short"); return -EINVAL; } memcpy(header, &data[pos], sizeof(struct png_header)); if (header->deflate != 0) { snprintf(err_msg, err_msg_length, "Deflate wrong in PNG header"); return -EINVAL; } if (header->bitDepth == 0) { snprintf(err_msg, err_msg_length, "PNG file has zero bit depth"); return -EINVAL; } // ensure the width and height values have the proper byte order // and copy them into the info struct. header->width = ntoh32(header->width); header->height = ntoh32(header->height); info->width = header->width; info->height = header->height; return 0; } snprintf(err_msg, err_msg_length, "Failed to read PNG file header"); return -EINVAL; } static int pdf_add_png_data(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const struct pdf_img_info *img_info, const uint8_t *png_data, size_t png_data_length) { // indicates if we return an error or add the img at the // end of the function bool success = false; // string stream used for writing color space (and palette) info // into the pdf struct dstr colour_space = INIT_DSTR; struct pdf_object *obj = NULL; uint8_t *final_data = NULL; int written = 0; uint32_t pos; uint8_t *png_data_temp = NULL; size_t png_data_total_length = 0; uint8_t ncolours; // Stores palette information for indexed PNGs struct rgb_value *palette_buffer = NULL; size_t palette_buffer_length = 0; const struct png_header *header = &img_info->png; // Father info from png header switch (header->colorType) { case PNG_COLOR_GREYSCALE: ncolours = 1; break; case PNG_COLOR_RGB: ncolours = 3; break; case PNG_COLOR_INDEXED: ncolours = 1; break; // PNG_COLOR_RGBA and PNG_COLOR_GREYSCALE_A are unsupported default: pdf_set_err(pdf, -EINVAL, "PNG has unsupported color type: %d", header->colorType); goto free_buffers; break; } /* process PNG chunks */ pos = sizeof(png_signature); while (1) { const struct png_chunk *chunk; chunk = (const struct png_chunk *)&png_data[pos]; pos += sizeof(struct png_chunk); if (pos > png_data_length - 4) { pdf_set_err(pdf, -EINVAL, "PNG file too short"); goto free_buffers; } const uint32_t chunk_length = ntoh32(chunk->length); // chunk length + 4-bytes of CRC if (chunk_length > png_data_length - pos - 4) { pdf_set_err(pdf, -EINVAL, "PNG chunk exceeds file: %d vs %ld", chunk_length, png_data_length - pos - 4); goto free_buffers; } if (strncmp(chunk->type, png_chunk_header, 4) == 0) { // Ignoring the header, since it was parsed // before calling this function. } else if (strncmp(chunk->type, png_chunk_palette, 4) == 0) { // Palette chunk if (header->colorType == PNG_COLOR_INDEXED) { // palette chunk is needed for indexed images if (palette_buffer) { pdf_set_err(pdf, -EINVAL, "PNG contains multiple palette chunks"); goto free_buffers; } if (chunk_length % 3 != 0) { pdf_set_err(pdf, -EINVAL, "PNG format error: palette chunk length is " "not divisbly by 3!"); goto free_buffers; } palette_buffer_length = (size_t)(chunk_length / 3); if (palette_buffer_length > 256 || palette_buffer_length == 0) { pdf_set_err(pdf, -EINVAL, "PNG palette length invalid: %zd", palette_buffer_length); goto free_buffers; } palette_buffer = (struct rgb_value *)malloc( palette_buffer_length * sizeof(struct rgb_value)); if (!palette_buffer) { pdf_set_err(pdf, -ENOMEM, "Could not allocate memory for indexed color " "palette (%zu bytes)", palette_buffer_length * sizeof(struct rgb_value)); goto free_buffers; } for (size_t i = 0; i < palette_buffer_length; i++) { size_t offset = (i * 3) + pos; palette_buffer[i].red = png_data[offset]; palette_buffer[i].green = png_data[offset + 1]; palette_buffer[i].blue = png_data[offset + 2]; } } else if (header->colorType == PNG_COLOR_RGB || header->colorType == PNG_COLOR_RGBA) { // palette chunk is optional for RGB(A) images // but we do not process them } else { pdf_set_err(pdf, -EINVAL, "Unexpected palette chunk for color type %d", header->colorType); goto free_buffers; } } else if (strncmp(chunk->type, png_chunk_data, 4) == 0) { if (chunk_length > 0 && chunk_length < png_data_length - pos) { uint8_t *data = (uint8_t *)realloc( png_data_temp, png_data_total_length + chunk_length); // (uint8_t *)realloc(info.data, info.length + chunk_length); if (!data) { pdf_set_err(pdf, -ENOMEM, "No memory for PNG data"); goto free_buffers; } png_data_temp = data; memcpy(&png_data_temp[png_data_total_length], &png_data[pos], chunk_length); png_data_total_length += chunk_length; } } else if (strncmp(chunk->type, png_chunk_end, 4) == 0) { /* end of file, exit */ break; } if (chunk_length >= png_data_length) { pdf_set_err(pdf, -EINVAL, "PNG chunk length larger than file"); goto free_buffers; } pos += chunk_length; // add chunk length pos += sizeof(uint32_t); // add CRC length } /* if no length was found */ if (png_data_total_length == 0) { pdf_set_err(pdf, -EINVAL, "PNG file has zero length"); goto free_buffers; } switch (header->colorType) { case PNG_COLOR_GREYSCALE: dstr_append(&colour_space, "/DeviceGray"); break; case PNG_COLOR_RGB: dstr_append(&colour_space, "/DeviceRGB"); break; case PNG_COLOR_INDEXED: if (palette_buffer_length == 0) { pdf_set_err(pdf, -EINVAL, "Indexed PNG contains no palette"); goto free_buffers; } // Write the color palette to the color_palette buffer dstr_printf(&colour_space, "[ /Indexed\r\n" " /DeviceRGB\r\n" " %zu\r\n" " <", palette_buffer_length - 1); // write individual paletter values // the index value for every RGB value is determined by its position // (0, 1, 2, ...) for (size_t i = 0; i < palette_buffer_length; i++) { dstr_printf(&colour_space, "%02X%02X%02X ", palette_buffer[i].red, palette_buffer[i].green, palette_buffer[i].blue); } dstr_append(&colour_space, ">\r\n]"); break; default: pdf_set_err(pdf, -EINVAL, "Cannot map PNG color type %d to PDF color space", header->colorType); goto free_buffers; break; } final_data = (uint8_t *)malloc(png_data_total_length + 1024 + dstr_len(&colour_space)); if (!final_data) { pdf_set_err(pdf, -ENOMEM, "Unable to allocate PNG data %zu", png_data_total_length + 1024 + dstr_len(&colour_space)); goto free_buffers; } // Write image information to PDF written = sprintf((char *)final_data, "<<\r\n" " /Type /XObject\r\n" " /Name /Image%d\r\n" " /Subtype /Image\r\n" " /ColorSpace %s\r\n" " /Width %u\r\n" " /Height %u\r\n" " /Interpolate true\r\n" " /BitsPerComponent %u\r\n" " /Filter /FlateDecode\r\n" " /DecodeParms << /Predictor 15 /Colors %d " "/BitsPerComponent %u /Columns %u >>\r\n" " /Length %zu\r\n" ">>stream\r\n", flexarray_size(&pdf->objects), dstr_data(&colour_space), header->width, header->height, header->bitDepth, ncolours, header->bitDepth, header->width, png_data_total_length); memcpy(&final_data[written], png_data_temp, png_data_total_length); written += png_data_total_length; written += sprintf((char *)&final_data[written], "\r\nendstream\r\n"); obj = pdf_add_object(pdf, OBJ_image); if (!obj) { goto free_buffers; } dstr_append_data(&obj->stream, final_data, written); if (get_img_display_dimensions(pdf, header->width, header->height, &display_width, &display_height)) { goto free_buffers; } success = true; free_buffers: if (final_data) free(final_data); if (palette_buffer) free(palette_buffer); if (png_data_temp) free(png_data_temp); dstr_free(&colour_space); if (success) return pdf_add_image(pdf, page, obj, x, y, display_width, display_height); else return pdf->errval; } static int parse_bmp_header(struct pdf_img_info *info, const uint8_t *data, size_t data_length, char *err_msg, size_t err_msg_length) { if (data_length < sizeof(bmp_signature) + sizeof(struct bmp_header)) { snprintf(err_msg, err_msg_length, "File is too short"); return -EINVAL; } if (memcmp(data, bmp_signature, sizeof(bmp_signature))) { snprintf(err_msg, err_msg_length, "File is not correct BMP file"); return -EINVAL; } memcpy(&info->bmp, &data[sizeof(bmp_signature)], sizeof(struct bmp_header)); if (info->bmp.biWidth < 0) { snprintf(err_msg, err_msg_length, "BMP has negative width"); return -EINVAL; } if (info->bmp.biHeight == -2147483648) { // 1 << 31 snprintf(err_msg, err_msg_length, "BMP height overflow"); return -EINVAL; } info->width = info->bmp.biWidth; // biHeight might be negative (positive indicates vertically mirrored // lines) info->height = abs(info->bmp.biHeight); return 0; } static int pdf_add_bmp_data(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const struct pdf_img_info *info, const uint8_t *data, const size_t len) { const struct bmp_header *header = &info->bmp; uint8_t *bmp_data = NULL; uint8_t row_padding; uint32_t bpp; size_t data_len; int retval; const uint32_t width = info->width; const uint32_t height = info->height; if (header->bfSize != len) return pdf_set_err(pdf, -EINVAL, "BMP file seems to have wrong length"); if (header->biSize != 40) return pdf_set_err(pdf, -EINVAL, "Wrong BMP header: biSize"); if (header->biCompression != 0) return pdf_set_err(pdf, -EINVAL, "Wrong BMP compression value: %d", header->biCompression); if (header->biWidth > MAX_IMAGE_WIDTH || header->biWidth <= 0 || width > MAX_IMAGE_WIDTH || width == 0) return pdf_set_err(pdf, -EINVAL, "BMP has invalid width: %d", header->biWidth); if (header->biHeight > MAX_IMAGE_HEIGHT || header->biHeight < -MAX_IMAGE_HEIGHT || header->biHeight == 0 || height > MAX_IMAGE_HEIGHT || height == 0) return pdf_set_err(pdf, -EINVAL, "BMP has invalid height: %d", header->biHeight); if (header->biBitCount != 24 && header->biBitCount != 32) return pdf_set_err(pdf, -EINVAL, "Unsupported BMP bitdepth: %d", header->biBitCount); bpp = header->biBitCount / 8; /* BMP rows are 4-bytes padded! */ row_padding = (width * bpp) & 3; data_len = (size_t)width * (size_t)height * 3; if (header->bfOffBits >= len) return pdf_set_err(pdf, -EINVAL, "Invalid BMP image offset"); if (len - header->bfOffBits < (size_t)height * (width + row_padding) * bpp) return pdf_set_err(pdf, -EINVAL, "Wrong BMP image size"); if (bpp == 3) { /* 24 bits: change R and B colors */ bmp_data = (uint8_t *)malloc(data_len); if (!bmp_data) return pdf_set_err(pdf, -ENOMEM, "Insufficient memory for bitmap"); for (uint32_t pos = 0; pos < width * height; pos++) { uint32_t src_pos = header->bfOffBits + 3 * (pos + (pos / width) * row_padding); bmp_data[pos * 3] = data[src_pos + 2]; bmp_data[pos * 3 + 1] = data[src_pos + 1]; bmp_data[pos * 3 + 2] = data[src_pos]; } } else if (bpp == 4) { /* 32 bits: change R and B colors, remove key color */ int offs = 0; bmp_data = (uint8_t *)malloc(data_len); if (!bmp_data) return pdf_set_err(pdf, -ENOMEM, "Insufficient memory for bitmap"); for (uint32_t pos = 0; pos < width * height * 4; pos += 4) { bmp_data[offs] = data[header->bfOffBits + pos + 2]; bmp_data[offs + 1] = data[header->bfOffBits + pos + 1]; bmp_data[offs + 2] = data[header->bfOffBits + pos]; offs += 3; } } else { return pdf_set_err(pdf, -EINVAL, "Unsupported BMP bitdepth: %d", header->biBitCount); } if (header->biHeight >= 0) { // BMP has vertically mirrored representation of lines, so swap them uint8_t *line = (uint8_t *)malloc(width * 3); if (!line) { free(bmp_data); return pdf_set_err(pdf, -ENOMEM, "Unable to allocate memory for bitmap mirror"); } for (uint32_t pos = 0; pos < (height / 2); pos++) { memcpy(line, &bmp_data[pos * width * 3], width * 3); memcpy(&bmp_data[pos * width * 3], &bmp_data[(height - pos - 1) * width * 3], width * 3); memcpy(&bmp_data[(height - pos - 1) * width * 3], line, width * 3); } free(line); } retval = pdf_add_rgb24(pdf, page, x, y, display_width, display_height, bmp_data, width, height); free(bmp_data); return retval; } static int determine_image_format(const uint8_t *data, size_t length) { if (length >= sizeof(png_signature) && memcmp(data, png_signature, sizeof(png_signature)) == 0) return IMAGE_PNG; if (length >= sizeof(bmp_signature) && memcmp(data, bmp_signature, sizeof(bmp_signature)) == 0) return IMAGE_BMP; if (length >= sizeof(jpeg_signature) && memcmp(data, jpeg_signature, sizeof(jpeg_signature)) == 0) return IMAGE_JPG; if (length >= sizeof(ppm_signature) && memcmp(data, ppm_signature, sizeof(ppm_signature)) == 0) return IMAGE_PPM; if (length >= sizeof(pgm_signature) && memcmp(data, pgm_signature, sizeof(pgm_signature)) == 0) return IMAGE_PPM; return IMAGE_UNKNOWN; } int pdf_parse_image_header(struct pdf_img_info *info, const uint8_t *data, size_t length, char *err_msg, size_t err_msg_length) { const int image_format = determine_image_format(data, length); info->image_format = image_format; switch (image_format) { case IMAGE_PNG: return parse_png_header(info, data, length, err_msg, err_msg_length); case IMAGE_BMP: return parse_bmp_header(info, data, length, err_msg, err_msg_length); case IMAGE_JPG: return parse_jpeg_header(info, data, length, err_msg, err_msg_length); case IMAGE_PPM: return parse_ppm_header(info, data, length, err_msg, err_msg_length); case IMAGE_UNKNOWN: default: snprintf(err_msg, err_msg_length, "Unknown file format"); return -EINVAL; } } int pdf_add_image_data(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const uint8_t *data, size_t len) { struct pdf_img_info info = { .image_format = IMAGE_UNKNOWN, .width = 0, .height = 0, .jpeg = {0}, }; int ret = pdf_parse_image_header(&info, data, len, pdf->errstr, sizeof(pdf->errstr)); if (ret) return ret; // Try and determine which image format it is based on the content switch (info.image_format) { case IMAGE_PNG: return pdf_add_png_data(pdf, page, x, y, display_width, display_height, &info, data, len); case IMAGE_BMP: return pdf_add_bmp_data(pdf, page, x, y, display_width, display_height, &info, data, len); case IMAGE_JPG: return pdf_add_jpeg_data(pdf, page, x, y, display_width, display_height, &info, data, len); case IMAGE_PPM: return pdf_add_ppm_data(pdf, page, x, y, display_width, display_height, &info, data, len); // This case should be caught in parse_image_header, but is checked // here again for safety case IMAGE_UNKNOWN: default: return pdf_set_err(pdf, -EINVAL, "Unable to determine image format"); } } int pdf_add_image_file(struct pdf_doc *pdf, struct pdf_object *page, float x, float y, float display_width, float display_height, const char *image_filename) { size_t len; uint8_t *data; int ret = 0; data = get_file(pdf, image_filename, &len); if (data == NULL) return pdf_get_errval(pdf); ret = pdf_add_image_data(pdf, page, x, y, display_width, display_height, data, len); free(data); return ret; }