mirror of
https://github.com/storytold/pdfcraft.git
synced 2026-10-25 10:39:01 -06:00
Brandon ThomasandClaude Opus 5.5
1449b71a19
Fuzzed inputs drove one check-one process to 2-6 GB (the 16 GB CI runner running two of them was killed every night). Three causes, each with a test: - lopdf (bootstrap inspector) sized PNG predictor rows from /Columns before reading data: /Columns 4294967295 allocated two 4 GiB rows. Vendored lopdf 0.45.0 refuses rows longer than the data (and uses saturating row widths); its load-time stream decoding also gets a 256 MiB limit (it had none). - hayro sized a tiling pattern's cell pixmap from /XStep x scale, bounded only by u16: a step far beyond /BBox allocated 2 GB. The vendored patch keeps the cell within 3000 px a side, lowering the resolution instead. - cos decoded cross-reference streams up to the general 1 GiB limit; they now stop at what 8,388,607 objects' rows can hold (~200 MB at most), and the section is reconstructed (logged) instead. On the nightly's seed (41) the worst captured input fell from 6.2 GB to 6 MB, and a 300 s fuzz run had no process above 300 MB (before: many at 2-5 GB). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
97 lines
2.8 KiB
Rust
97 lines
2.8 KiB
Rust
// Display a summary of the annotations in a PDF file to the terminal
|
|
//
|
|
// Run with `cargo run --example print_annotations <pdf-file>`
|
|
|
|
use core::str;
|
|
use env_logger::Env;
|
|
use lopdf::{Document, Object};
|
|
use std::env;
|
|
use std::process;
|
|
|
|
fn logging() {
|
|
env_logger::Builder::from_env(Env::default().default_filter_or("info")).init()
|
|
}
|
|
|
|
fn args() -> Vec<String> {
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
if args.len() != 2 {
|
|
eprintln!("Usage: {} <pdf-file>", args[0]);
|
|
process::exit(1);
|
|
}
|
|
|
|
args
|
|
}
|
|
|
|
fn handle_pdf_page(doc: Document) -> u32 {
|
|
let mut page_counter = 1;
|
|
|
|
for page in doc.page_iter() {
|
|
for a in doc.get_page_annotations(page).unwrap() {
|
|
let subtype = a.get_deref(b"Subtype", &doc).and_then(Object::as_name).unwrap_or(b"");
|
|
println!(
|
|
"Page {}, {} annotation at {:?}",
|
|
page_counter,
|
|
str::from_utf8(subtype).unwrap(),
|
|
a.get_deref(b"Rect", &doc).and_then(Object::as_array).unwrap()
|
|
);
|
|
if let Ok(Object::String(c, _)) = a.get_deref(b"Contents", &doc) {
|
|
println!(" Contents: {:.60}", String::from_utf8_lossy(c).lines().next().unwrap());
|
|
}
|
|
if subtype == b"Link"
|
|
&& let Ok(ahref) = a.get_deref(b"A", &doc).and_then(Object::as_dict)
|
|
{
|
|
print!(
|
|
" {} -> ",
|
|
ahref
|
|
.get_deref(b"S", &doc)
|
|
.and_then(Object::as_name)
|
|
.map(str::from_utf8)
|
|
.unwrap()
|
|
.unwrap()
|
|
);
|
|
if let Ok(d) = ahref.get_deref(b"D", &doc).and_then(Object::as_array) {
|
|
println!("{:?}", d);
|
|
} else if let Ok(Object::String(u, _)) = ahref.get_deref(b"URI", &doc) {
|
|
println!("{}", String::from_utf8_lossy(u));
|
|
} else if let Ok(n) = ahref
|
|
.get_deref(b"N", &doc)
|
|
.and_then(Object::as_name)
|
|
.map(str::from_utf8)
|
|
.unwrap()
|
|
{
|
|
println!("{}", n);
|
|
}
|
|
}
|
|
}
|
|
page_counter += 1;
|
|
}
|
|
|
|
page_counter
|
|
}
|
|
|
|
#[cfg(not(feature = "async"))]
|
|
fn main() {
|
|
logging();
|
|
|
|
let args: Vec<String> = args();
|
|
|
|
match Document::load(&args[1]) {
|
|
Ok(doc) => _ = handle_pdf_page(doc),
|
|
Err(e) => eprintln!("Error opening {:?}: {:?}", args[1], e),
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "async")]
|
|
#[tokio::main]
|
|
async fn main() {
|
|
logging();
|
|
|
|
let args: Vec<String> = args();
|
|
|
|
match Document::load(&args[1]).await {
|
|
Ok(doc) => _ = handle_pdf_page(doc),
|
|
Err(e) => eprintln!("Error opening {:?}: {:?}", args[1], e),
|
|
}
|
|
}
|