278 lines
10 KiB
Rust
278 lines
10 KiB
Rust
use super::*;
|
||
|
||
fn find(hay: &[u8], needle: &[u8]) -> Option<usize> {
|
||
hay.windows(needle.len()).position(|w| w == needle)
|
||
}
|
||
fn rfind(hay: &[u8], needle: &[u8]) -> Option<usize> {
|
||
hay.windows(needle.len()).rposition(|w| w == needle)
|
||
}
|
||
|
||
/// Parses the xref table back and checks each in-use entry points at `N 0 obj`.
|
||
fn check_xref(pdf: &[u8]) -> usize {
|
||
let sx = rfind(pdf, b"startxref\n").unwrap();
|
||
let rest = std::str::from_utf8(&pdf[sx + 10..]).unwrap();
|
||
let off: usize = rest.lines().next().unwrap().trim().parse().unwrap();
|
||
assert!(pdf[off..].starts_with(b"xref\n"), "startxref points at xref");
|
||
let table = std::str::from_utf8(&pdf[off..sx]).unwrap();
|
||
let mut lines = table.lines();
|
||
assert_eq!(lines.next(), Some("xref"));
|
||
let hdr: Vec<usize> = lines.next().unwrap().split(' ').map(|x| x.parse().unwrap()).collect();
|
||
assert_eq!(hdr[0], 0);
|
||
let count = hdr[1];
|
||
let mut checked = 0;
|
||
for i in 0..count {
|
||
let e = lines.next().unwrap();
|
||
assert_eq!(e.len(), 19, "entry `{e}` (20 bytes with newline)");
|
||
let o: usize = e[..10].parse().unwrap();
|
||
if e.trim_end().ends_with('n') {
|
||
let want = format!("{i} 0 obj");
|
||
assert!(pdf[o..].starts_with(want.as_bytes()), "object {i} at {o}");
|
||
checked += 1;
|
||
}
|
||
}
|
||
assert!(table.contains("trailer"));
|
||
assert!(table.contains(&format!("/Size {count}")));
|
||
checked
|
||
}
|
||
|
||
#[test]
|
||
fn minimal_document() {
|
||
let pdf = PdfDoc::new().finish();
|
||
assert!(pdf.starts_with(b"%PDF-1.7"));
|
||
assert!(pdf.ends_with(b"%%EOF\n"));
|
||
assert!(check_xref(&pdf) >= 5);
|
||
}
|
||
|
||
#[test]
|
||
fn drawing_and_text() {
|
||
let mut d = PdfDoc::new();
|
||
d.info.title = "Report (Q1)".into();
|
||
let p = d.add_page(612.0, 792.0);
|
||
p.fill_rect(10.0, 10.0, 100.0, 20.0, [255, 255, 0]);
|
||
p.rect_stroke(10.0, 10.0, 100.0, 20.0, 0.5, [0, 0, 0]);
|
||
p.stroke_line(0.0, 0.0, 612.0, 792.0, 1.0, [255, 0, 0]);
|
||
p.stroke_line_dashed(0.0, 5.0, 100.0, 5.0, 1.0, [0, 0, 0], &[3.0, 2.0]);
|
||
p.polygon(&[[0.0, 0.0], [10.0, 0.0], [5.0, 8.0]], Some([0, 0, 255]), Some(([0, 0, 0], 1.0)));
|
||
p.circle(300.0, 300.0, 20.0, Some([0, 128, 0]), None);
|
||
p.save();
|
||
p.clip_rect(0.0, 0.0, 50.0, 50.0);
|
||
p.text(72.0, 72.0, 12.0, Font::Helvetica, [0, 0, 0], "Hello (world) \\ café €5");
|
||
p.restore();
|
||
p.text_rotated(100.0, 100.0, 9.0, Font::TimesBold, [0, 0, 0], "Up", -std::f32::consts::FRAC_PI_2);
|
||
let p2 = d.add_page(842.0, 595.0);
|
||
p2.text(10.0, 20.0, 10.0, Font::CourierBold, [0, 0, 0], "Second");
|
||
assert_eq!(d.page_count(), 2);
|
||
let pdf = d.finish();
|
||
check_xref(&pdf);
|
||
assert!(find(&pdf, b"(Hello \\(world\\) \\\\ caf\\351 \\2005) Tj").is_some());
|
||
assert!(find(&pdf, b"(Second) Tj").is_some());
|
||
assert!(find(&pdf, b"/BaseFont /Helvetica /Encoding /WinAnsiEncoding").is_some());
|
||
assert!(find(&pdf, b"/BaseFont /Courier-Bold").is_some());
|
||
assert!(find(&pdf, b"/BaseFont /Times-Bold").is_some());
|
||
assert!(find(&pdf, b"/Count 2").is_some());
|
||
assert!(find(&pdf, b"/MediaBox [0 0 842 595]").is_some());
|
||
assert!(find(&pdf, b"/Title (Report \\(Q1\\))").is_some());
|
||
// Top-left origin: a line from (0,0) starts at the top of the page.
|
||
assert!(find(&pdf, b"0 792 m 612 0 l S").is_some());
|
||
}
|
||
|
||
#[test]
|
||
fn images_with_alpha() {
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(200.0, 200.0);
|
||
let mut px = Vec::new();
|
||
for i in 0..16u8 {
|
||
px.extend_from_slice(&[i * 10, 0, 0, if i < 8 { 255 } else { 128 }]);
|
||
}
|
||
p.image_rgba(10.0, 10.0, 40.0, 40.0, &px, 4, 4);
|
||
p.image_rgba(10.0, 10.0, 40.0, 40.0, &[255; 16], 2, 2); // opaque: no mask
|
||
p.image_rgba(0.0, 0.0, 1.0, 1.0, &[1, 2, 3], 4, 4); // too short: ignored
|
||
let pdf = d.finish();
|
||
check_xref(&pdf);
|
||
assert!(find(&pdf, b"/SMask").is_some());
|
||
assert!(find(&pdf, b"/Im1 Do").is_some());
|
||
assert!(find(&pdf, b"/Im2 Do").is_some());
|
||
assert!(find(&pdf, b"/Im3").is_none());
|
||
}
|
||
|
||
#[test]
|
||
fn run_length_roundtrip() {
|
||
fn decode(d: &[u8]) -> Vec<u8> {
|
||
let mut out = Vec::new();
|
||
let mut i = 0;
|
||
while i < d.len() {
|
||
let l = d[i];
|
||
i += 1;
|
||
if l == 128 {
|
||
break;
|
||
} else if l < 128 {
|
||
out.extend_from_slice(&d[i..i + l as usize + 1]);
|
||
i += l as usize + 1;
|
||
} else {
|
||
out.extend(std::iter::repeat_n(d[i], 257 - l as usize));
|
||
i += 1;
|
||
}
|
||
}
|
||
out
|
||
}
|
||
let cases: Vec<Vec<u8>> = vec![
|
||
vec![],
|
||
vec![1],
|
||
vec![1, 1],
|
||
vec![1, 2, 3, 3, 3, 3, 4, 5, 5],
|
||
(0..1000).map(|i| (i / 7) as u8).collect(),
|
||
(0..1000).map(|i| (i * 31 % 251) as u8).collect(),
|
||
vec![9; 300],
|
||
];
|
||
for c in cases {
|
||
assert_eq!(decode(&run_length(&c)), c);
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn widths() {
|
||
assert!((text_width(Font::Helvetica, 10.0, "Hello") - 22.78).abs() < 0.01);
|
||
assert!(text_width(Font::HelveticaBold, 10.0, "Hello") > text_width(Font::Helvetica, 10.0, "Hello"));
|
||
assert_eq!(text_width(Font::Courier, 10.0, "abc"), 18.0);
|
||
assert!(text_width(Font::TimesRoman, 12.0, "é") > 0.0);
|
||
assert_eq!(text_width(Font::Helvetica, 10.0, ""), 0.0);
|
||
assert_eq!(Font::family_for("Calibri"), Family::Sans);
|
||
assert_eq!(Font::family_for("Times New Roman"), Family::Serif);
|
||
assert_eq!(Font::family_for("Consolas"), Family::Mono);
|
||
assert_eq!(Font::styled(Family::Serif, true, true), Font::TimesBoldItalic);
|
||
}
|
||
|
||
#[test]
|
||
fn hostile_inputs_never_panic() {
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(f32::NAN, -5.0);
|
||
p.text(f32::INFINITY, f32::NAN, 1e30, Font::Helvetica, [0, 0, 0], "x");
|
||
p.fill_rect(0.0, 0.0, -1.0, 5.0, [0, 0, 0]);
|
||
p.polygon(&[], None, None);
|
||
p.circle(0.0, 0.0, -3.0, Some([0, 0, 0]), None);
|
||
p.restore();
|
||
p.restore();
|
||
p.save();
|
||
p.image_rgba(0.0, 0.0, 10.0, 10.0, &[], u32::MAX, u32::MAX);
|
||
let pdf = d.finish();
|
||
check_xref(&pdf);
|
||
}
|
||
|
||
fn page_content(d: &PdfDoc) -> &str {
|
||
&d.pages[0].content
|
||
}
|
||
|
||
#[test]
|
||
fn text_width_does_not_overflow() {
|
||
// 8,000,000 x 556 units exceeds u32::MAX: a u32 sum panics in debug builds.
|
||
let long = "a".repeat(8_000_000);
|
||
let w = text_width(Font::Helvetica, 10.0, &long);
|
||
assert!(w.is_finite() && w > 0.0);
|
||
assert!(w > text_width(Font::Helvetica, 10.0, &"a".repeat(7_000_000)));
|
||
}
|
||
|
||
#[test]
|
||
fn all_zero_dash_is_drawn_solid() {
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(100.0, 100.0);
|
||
p.stroke_line_dashed(0.0, 0.0, 50.0, 0.0, 1.0, [0, 0, 0], &[0.0, 0.0]);
|
||
p.polyline(&[[0.0, 0.0], [5.0, 5.0]], 1.0, [0, 0, 0], &[0.0, 0.0001, f32::NAN]);
|
||
p.stroke_line_dashed(0.0, 0.0, 50.0, 0.0, 1.0, [0, 0, 0], &[3.0, 2.0]);
|
||
let c = page_content(&d);
|
||
assert_eq!(c.matches("[] 0 d").count(), 2);
|
||
assert!(!c.contains("[0 0]"));
|
||
assert!(c.contains("[3 2] 0 d"));
|
||
}
|
||
|
||
#[test]
|
||
fn negative_and_nan_stroke_widths_become_zero() {
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(100.0, 100.0);
|
||
p.rect_stroke(1.0, 1.0, 10.0, 10.0, -3.0, [0, 0, 0]);
|
||
p.stroke_line(0.0, 0.0, 5.0, 5.0, f32::NAN, [0, 0, 0]);
|
||
p.polygon(&[[0.0, 0.0], [5.0, 0.0], [5.0, 5.0]], None, Some(([0, 0, 0], -1.0)));
|
||
p.circle(20.0, 20.0, 5.0, None, Some(([0, 0, 0], f32::NEG_INFINITY)));
|
||
let c = page_content(&d);
|
||
assert_eq!(c.matches("\n0 w ").count(), 4);
|
||
assert!(!c.contains("-1 w") && !c.contains("-3 w"));
|
||
}
|
||
|
||
#[test]
|
||
fn non_finite_image_size_stores_nothing() {
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(200.0, 200.0);
|
||
let px = [255u8; 16];
|
||
for (w, h) in [(f32::NAN, 10.0), (10.0, f32::NAN), (f32::INFINITY, 10.0), (-1.0, 10.0)] {
|
||
p.image_rgba(0.0, 0.0, w, h, &px, 2, 2);
|
||
}
|
||
p.image_rgba(f32::NAN, 0.0, 10.0, 10.0, &px, 2, 2);
|
||
assert!(p.images.is_empty());
|
||
assert!(p.content.is_empty());
|
||
}
|
||
|
||
#[test]
|
||
fn every_draw_method_survives_hostile_numbers() {
|
||
let bad = [f32::NAN, f32::INFINITY, f32::NEG_INFINITY, -7.0, 0.0, 1.0e30, -1.0e30];
|
||
let px = [200u8; 16];
|
||
let mut d = PdfDoc::new();
|
||
let p = d.add_page(300.0, 300.0);
|
||
for &a in &bad {
|
||
for &b in &bad {
|
||
p.clip_rect(a, b, a, b);
|
||
p.fill_rect(a, b, b, a, [1, 2, 3]);
|
||
p.rect_stroke(a, b, a, b, a, [1, 2, 3]);
|
||
p.stroke_line(a, b, b, a, a, [1, 2, 3]);
|
||
p.stroke_line_dashed(a, b, b, a, b, [1, 2, 3], &[a, b, 0.0]);
|
||
p.polyline(&[[a, b], [b, a], [1.0, 1.0]], a, [1, 2, 3], &[b, a]);
|
||
p.polygon(&[[a, b], [b, a], [1.0, 1.0]], Some([1, 2, 3]), Some(([4, 5, 6], a)));
|
||
p.circle(a, b, b, Some([1, 2, 3]), Some(([4, 5, 6], a)));
|
||
p.text(a, b, a, Font::Helvetica, [0, 0, 0], "x");
|
||
p.text_rotated(a, b, 10.0, Font::TimesRoman, [0, 0, 0], "y", b);
|
||
p.image_rgba(a, b, b, a, &px, 2, 2);
|
||
}
|
||
}
|
||
p.stroke_line_dashed(0.0, 0.0, 1.0, 1.0, 1.0, [0, 0, 0], &[0.0; 100]);
|
||
let c = page_content(&d);
|
||
assert!(!c.contains("NaN") && !c.contains("inf"));
|
||
let pdf = d.finish();
|
||
check_xref(&pdf);
|
||
/// Decodes the `<FEFF…>` hex string following `key` in the file bytes.
|
||
fn utf16_entry(pdf: &[u8], key: &str) -> String {
|
||
let at = find(pdf, key.as_bytes()).unwrap() + key.len();
|
||
let rest = std::str::from_utf8(&pdf[at..at + 9000.min(pdf.len() - at)]).unwrap_or("");
|
||
let hex = rest.trim_start().strip_prefix("<FEFF").unwrap();
|
||
let hex = &hex[..hex.find('>').unwrap()];
|
||
let units: Vec<u16> = (0..hex.len()).step_by(4).map(|i| u16::from_str_radix(&hex[i..i + 4], 16).unwrap()).collect();
|
||
String::from_utf16(&units).unwrap()
|
||
}
|
||
|
||
#[test]
|
||
fn info_strings_with_non_ascii_are_utf16be() {
|
||
let mut doc = PdfDoc::new();
|
||
doc.add_page(200.0, 200.0);
|
||
doc.info.title = "Übersicht €5 – “Q1” 日本 😀".into();
|
||
doc.info.author = "Müller & Søn".into();
|
||
let pdf = doc.finish();
|
||
assert_eq!(utf16_entry(&pdf, "/Title"), "Übersicht €5 – “Q1” 日本 😀");
|
||
assert_eq!(utf16_entry(&pdf, "/Author"), "Müller & Søn");
|
||
}
|
||
|
||
#[test]
|
||
fn ascii_info_strings_stay_literal_and_escaped() {
|
||
let mut doc = PdfDoc::new();
|
||
doc.add_page(200.0, 200.0);
|
||
doc.info.title = r"Q1 (draft) a\b".into();
|
||
let pdf = doc.finish();
|
||
assert!(find(&pdf, br"/Title (Q1 \(draft\) a\\b)").is_some());
|
||
assert!(find(&pdf, b"/Producer (GridCraft PDF)").is_some());
|
||
}
|
||
|
||
#[test]
|
||
fn info_strings_are_capped() {
|
||
let mut doc = PdfDoc::new();
|
||
doc.add_page(200.0, 200.0);
|
||
doc.info.title = "日".repeat(100_000);
|
||
let pdf = doc.finish();
|
||
assert_eq!(utf16_entry(&pdf, "/Title").chars().count(), MAX_INFO_CHARS);
|
||
}
|