/src/qpdf/libqpdf/QPDFFormFieldObjectHelper.cc
Line | Count | Source |
1 | | #include <qpdf/QPDFFormFieldObjectHelper.hh> |
2 | | |
3 | | #include <qpdf/AcroForm.hh> |
4 | | |
5 | | #include <qpdf/Pipeline_private.hh> |
6 | | #include <qpdf/Pl_QPDFTokenizer.hh> |
7 | | #include <qpdf/QIntC.hh> |
8 | | #include <qpdf/QPDFAcroFormDocumentHelper.hh> |
9 | | #include <qpdf/QPDFAnnotationObjectHelper.hh> |
10 | | #include <qpdf/QPDFObjectHandle_private.hh> |
11 | | #include <qpdf/QPDF_private.hh> |
12 | | #include <qpdf/QTC.hh> |
13 | | #include <qpdf/QUtil.hh> |
14 | | #include <cstdlib> |
15 | | |
16 | | #include <memory> |
17 | | |
18 | | using namespace qpdf; |
19 | | |
20 | | using FormNode = qpdf::impl::FormNode; |
21 | | |
22 | | const QPDFObjectHandle FormNode::null_oh; |
23 | | |
24 | | class QPDFFormFieldObjectHelper::Members: public FormNode |
25 | | { |
26 | | public: |
27 | | Members(QPDFObjectHandle const& oh) : |
28 | 48.2k | FormNode(oh) |
29 | 48.2k | { |
30 | 48.2k | } |
31 | | }; |
32 | | |
33 | | QPDFFormFieldObjectHelper::QPDFFormFieldObjectHelper(QPDFObjectHandle o) : |
34 | 25.5k | QPDFObjectHelper(o), |
35 | 25.5k | m(std::make_shared<Members>(oh())) |
36 | 25.5k | { |
37 | 25.5k | } |
38 | | |
39 | | QPDFFormFieldObjectHelper::QPDFFormFieldObjectHelper() : |
40 | 22.7k | QPDFObjectHelper(Null::temp()), |
41 | 22.7k | m(std::make_shared<Members>(QPDFObjectHandle())) |
42 | 22.7k | { |
43 | 22.7k | } |
44 | | |
45 | | bool |
46 | | QPDFFormFieldObjectHelper::isNull() |
47 | 0 | { |
48 | 0 | return m->null(); |
49 | 0 | } |
50 | | |
51 | | QPDFFormFieldObjectHelper |
52 | | QPDFFormFieldObjectHelper::getParent() |
53 | 0 | { |
54 | 0 | return {Null::if_null(m->Parent().oh())}; |
55 | 0 | } |
56 | | |
57 | | QPDFFormFieldObjectHelper |
58 | | QPDFFormFieldObjectHelper::getTopLevelField(bool* is_different) |
59 | 0 | { |
60 | 0 | return Null::if_null(m->root_field(is_different).oh()); |
61 | 0 | } |
62 | | |
63 | | FormNode |
64 | | FormNode::root_field(bool* is_different) |
65 | 0 | { |
66 | 0 | if (is_different) { |
67 | 0 | *is_different = false; |
68 | 0 | } |
69 | 0 | if (!obj) { |
70 | 0 | return {}; |
71 | 0 | } |
72 | 0 | auto rf = *this; |
73 | 0 | size_t depth = 0; // Don't bother with loop detection until depth becomes suspicious |
74 | 0 | QPDFObjGen::set seen; |
75 | 0 | while (rf.Parent() && (++depth < 10 || seen.add(rf))) { |
76 | 0 | rf = rf.Parent(); |
77 | 0 | if (is_different) { |
78 | 0 | *is_different = true; |
79 | 0 | } |
80 | 0 | } |
81 | 0 | return rf; |
82 | 0 | } |
83 | | |
84 | | QPDFObjectHandle |
85 | | QPDFFormFieldObjectHelper::getInheritableFieldValue(std::string const& name) |
86 | 0 | { |
87 | 0 | return Null::if_null(m->inheritable_value<QPDFObjectHandle>(name)); |
88 | 0 | } |
89 | | |
90 | | QPDFObjectHandle const& |
91 | | FormNode::inherited(std::string const& name, bool acroform) const |
92 | 19.9k | { |
93 | 19.9k | if (!obj) { |
94 | 794 | return null_oh; |
95 | 794 | } |
96 | 19.1k | auto node = *this; |
97 | 19.1k | QPDFObjGen::set seen; |
98 | 19.1k | size_t depth = 0; // Don't bother with loop detection until depth becomes suspicious |
99 | 25.8k | while (node.Parent() && (++depth < 10 || seen.add(node))) { |
100 | 9.28k | node = node.Parent(); |
101 | 9.28k | if (auto const& result = node[name]) { |
102 | 2.57k | return {result}; |
103 | 2.57k | } |
104 | 9.28k | } |
105 | 16.5k | return acroform ? from_AcroForm(name) : null_oh; |
106 | 19.1k | } |
107 | | |
108 | | std::string |
109 | | QPDFFormFieldObjectHelper::getInheritableFieldValueAsString(std::string const& name) |
110 | 0 | { |
111 | 0 | return m->inheritable_string(name); |
112 | 0 | } |
113 | | |
114 | | std::string |
115 | | FormNode::inheritable_string(std::string const& name) const |
116 | 9.34k | { |
117 | 9.34k | if (auto fv = inheritable_value<String>(name)) { |
118 | 842 | return fv.utf8_value(); |
119 | 842 | } |
120 | 8.50k | return {}; |
121 | 9.34k | } |
122 | | |
123 | | std::string |
124 | | QPDFFormFieldObjectHelper::getInheritableFieldValueAsName(std::string const& name) |
125 | 0 | { |
126 | 0 | if (auto fv = m->inheritable_value<Name>(name)) { |
127 | 0 | return fv; |
128 | 0 | } |
129 | 0 | return {}; |
130 | 0 | } |
131 | | |
132 | | std::string |
133 | | QPDFFormFieldObjectHelper::getFieldType() |
134 | 11.1k | { |
135 | 11.1k | if (auto ft = m->FT()) { |
136 | 10.4k | return ft; |
137 | 10.4k | } |
138 | 681 | return {}; |
139 | 11.1k | } |
140 | | |
141 | | std::string |
142 | | QPDFFormFieldObjectHelper::getFullyQualifiedName() |
143 | 0 | { |
144 | 0 | return m->fully_qualified_name(); |
145 | 0 | } |
146 | | |
147 | | std::string |
148 | | FormNode::fully_qualified_name() const |
149 | 9.37k | { |
150 | 9.37k | std::string result; |
151 | 9.37k | auto node = *this; |
152 | 9.37k | QPDFObjGen::set seen; |
153 | 9.37k | size_t depth = 0; // Don't bother with loop detection until depth becomes suspicious |
154 | 43.3k | while (node && (++depth < 10 || seen.add(node))) { |
155 | 33.9k | if (auto T = node.T()) { |
156 | 30.8k | if (!result.empty()) { |
157 | 21.3k | result.insert(0, 1, '.'); |
158 | 21.3k | } |
159 | 30.8k | result.insert(0, T.utf8_value()); |
160 | 30.8k | } |
161 | 33.9k | node = node.Parent(); |
162 | 33.9k | } |
163 | 9.37k | return result; |
164 | 9.37k | } |
165 | | |
166 | | std::string |
167 | | QPDFFormFieldObjectHelper::getPartialName() |
168 | 0 | { |
169 | 0 | return m->partial_name(); |
170 | 0 | } |
171 | | |
172 | | std::string |
173 | | FormNode::partial_name() const |
174 | 0 | { |
175 | 0 | if (auto pn = T()) { |
176 | 0 | return pn.utf8_value(); |
177 | 0 | } |
178 | 0 | return {}; |
179 | 0 | } |
180 | | |
181 | | std::string |
182 | | QPDFFormFieldObjectHelper::getAlternativeName() |
183 | 0 | { |
184 | 0 | return m->alternative_name(); |
185 | 0 | } |
186 | | |
187 | | std::string |
188 | | FormNode::alternative_name() const |
189 | 0 | { |
190 | 0 | if (auto an = TU()) { |
191 | 0 | return an.utf8_value(); |
192 | 0 | } |
193 | 0 | return fully_qualified_name(); |
194 | 0 | } |
195 | | |
196 | | std::string |
197 | | QPDFFormFieldObjectHelper::getMappingName() |
198 | 0 | { |
199 | 0 | return m->mapping_name(); |
200 | 0 | } |
201 | | |
202 | | std::string |
203 | | FormNode::mapping_name() const |
204 | 0 | { |
205 | 0 | if (auto mn = TM()) { |
206 | 0 | return mn.utf8_value(); |
207 | 0 | } |
208 | 0 | return alternative_name(); |
209 | 0 | } |
210 | | |
211 | | QPDFObjectHandle |
212 | | QPDFFormFieldObjectHelper::getValue() |
213 | 821 | { |
214 | 821 | return Null::if_null(m->V<QPDFObjectHandle>()); |
215 | 821 | } |
216 | | |
217 | | std::string |
218 | | QPDFFormFieldObjectHelper::getValueAsString() |
219 | 0 | { |
220 | 0 | return m->value(); |
221 | 0 | } |
222 | | |
223 | | std::string |
224 | | FormNode::value() const |
225 | 9.34k | { |
226 | 9.34k | return inheritable_string("/V"); |
227 | 9.34k | } |
228 | | |
229 | | QPDFObjectHandle |
230 | | QPDFFormFieldObjectHelper::getDefaultValue() |
231 | 0 | { |
232 | 0 | return Null::if_null(m->DV()); |
233 | 0 | } |
234 | | |
235 | | std::string |
236 | | QPDFFormFieldObjectHelper::getDefaultValueAsString() |
237 | 0 | { |
238 | 0 | return m->default_value(); |
239 | 0 | } |
240 | | |
241 | | std::string |
242 | | FormNode::default_value() const |
243 | 0 | { |
244 | 0 | return inheritable_string("/DV"); |
245 | 0 | } |
246 | | |
247 | | QPDFObjectHandle |
248 | | QPDFFormFieldObjectHelper::getDefaultResources() |
249 | 9.63k | { |
250 | 9.63k | return Null::if_null(m->getDefaultResources()); |
251 | 9.63k | } |
252 | | |
253 | | QPDFObjectHandle |
254 | | FormNode::getDefaultResources() |
255 | 15.9k | { |
256 | 15.9k | return from_AcroForm("/DR"); |
257 | 15.9k | } |
258 | | |
259 | | std::string |
260 | | QPDFFormFieldObjectHelper::getDefaultAppearance() |
261 | 0 | { |
262 | 0 | return m->default_appearance(); |
263 | 0 | } |
264 | | |
265 | | std::string |
266 | | FormNode::default_appearance() const |
267 | 9.34k | { |
268 | 9.34k | if (auto DA = inheritable_value<String>("/DA")) { |
269 | 7.13k | return DA.utf8_value(); |
270 | 7.13k | } |
271 | 2.20k | if (String DA = from_AcroForm("/DA")) { |
272 | 210 | return DA.utf8_value(); |
273 | 210 | } |
274 | 1.99k | return {}; |
275 | 2.20k | } |
276 | | |
277 | | int |
278 | | QPDFFormFieldObjectHelper::getQuadding() |
279 | 0 | { |
280 | 0 | return m->getQuadding(); |
281 | 0 | } |
282 | | |
283 | | int |
284 | | FormNode::getQuadding() |
285 | 0 | { |
286 | 0 | auto fv = inheritable_value<QPDFObjectHandle>("/Q"); |
287 | 0 | bool looked_in_acroform = false; |
288 | 0 | if (!fv.isInteger()) { |
289 | 0 | fv = from_AcroForm("/Q"); |
290 | 0 | looked_in_acroform = true; |
291 | 0 | } |
292 | 0 | if (fv.isInteger()) { |
293 | 0 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper Q present", looked_in_acroform ? 0 : 1); |
294 | 0 | return QIntC::to_int(fv.getIntValue()); |
295 | 0 | } |
296 | 0 | return 0; |
297 | 0 | } |
298 | | |
299 | | int |
300 | | QPDFFormFieldObjectHelper::getFlags() |
301 | 0 | { |
302 | 0 | return m->getFlags(); |
303 | 0 | } |
304 | | |
305 | | int |
306 | | FormNode::getFlags() |
307 | 4.21k | { |
308 | 4.21k | auto f = inheritable_value<QPDFObjectHandle>("/Ff"); |
309 | 4.21k | return f.isInteger() ? f.getIntValueAsInt() : 0; |
310 | 4.21k | } |
311 | | |
312 | | bool |
313 | | QPDFFormFieldObjectHelper::isText() |
314 | 0 | { |
315 | 0 | return m->isText(); |
316 | 0 | } |
317 | | |
318 | | bool |
319 | | FormNode::isText() |
320 | 0 | { |
321 | 0 | return FT() == "/Tx"; |
322 | 0 | } |
323 | | |
324 | | bool |
325 | | QPDFFormFieldObjectHelper::isCheckbox() |
326 | 555 | { |
327 | 555 | return m->isCheckbox(); |
328 | 555 | } |
329 | | |
330 | | bool |
331 | | FormNode::isCheckbox() |
332 | 1.37k | { |
333 | 1.37k | return FT() == "/Btn" && (getFlags() & (ff_btn_radio | ff_btn_pushbutton)) == 0; |
334 | 1.37k | } |
335 | | |
336 | | bool |
337 | | QPDFFormFieldObjectHelper::isChecked() |
338 | 0 | { |
339 | 0 | return m->isChecked(); |
340 | 0 | } |
341 | | |
342 | | bool |
343 | | FormNode::isChecked() |
344 | 0 | { |
345 | 0 | return isCheckbox() && V<Name>() != "/Off"; |
346 | 0 | } |
347 | | |
348 | | bool |
349 | | QPDFFormFieldObjectHelper::isRadioButton() |
350 | 861 | { |
351 | 861 | return m->isRadioButton(); |
352 | 861 | } |
353 | | |
354 | | bool |
355 | | FormNode::isRadioButton() |
356 | 1.41k | { |
357 | 1.41k | return FT() == "/Btn" && (getFlags() & ff_btn_radio) == ff_btn_radio; |
358 | 1.41k | } |
359 | | |
360 | | bool |
361 | | QPDFFormFieldObjectHelper::isPushbutton() |
362 | 0 | { |
363 | 0 | return m->isPushbutton(); |
364 | 0 | } |
365 | | |
366 | | bool |
367 | | FormNode::isPushbutton() |
368 | 0 | { |
369 | 0 | return FT() == "/Btn" && (getFlags() & ff_btn_pushbutton) == ff_btn_pushbutton; |
370 | 0 | } |
371 | | |
372 | | bool |
373 | | QPDFFormFieldObjectHelper::isChoice() |
374 | 0 | { |
375 | 0 | return m->isChoice(); |
376 | 0 | } |
377 | | |
378 | | bool |
379 | | FormNode::isChoice() |
380 | 10.6k | { |
381 | 10.6k | return FT() == "/Ch"; |
382 | 10.6k | } |
383 | | |
384 | | std::vector<std::string> |
385 | | QPDFFormFieldObjectHelper::getChoices() |
386 | 0 | { |
387 | 0 | return m->getChoices(); |
388 | 0 | } |
389 | | |
390 | | std::vector<std::string> |
391 | | FormNode::getChoices() |
392 | 1.30k | { |
393 | 1.30k | if (!isChoice()) { |
394 | 0 | return {}; |
395 | 0 | } |
396 | 1.30k | std::vector<std::string> result; |
397 | 32.2k | for (auto const& item: inheritable_value<Array>("/Opt")) { |
398 | 32.2k | if (item.isString()) { |
399 | 17.0k | result.emplace_back(item.getUTF8Value()); |
400 | 17.0k | } else if (item.size() == 2) { |
401 | 130 | auto display = item.getArrayItem(1); |
402 | 130 | if (display.isString()) { |
403 | 64 | result.emplace_back(display.getUTF8Value()); |
404 | 64 | } |
405 | 130 | } |
406 | 32.2k | } |
407 | 1.30k | return result; |
408 | 1.30k | } |
409 | | |
410 | | void |
411 | | QPDFFormFieldObjectHelper::setFieldAttribute(std::string const& key, QPDFObjectHandle value) |
412 | 0 | { |
413 | 0 | m->setFieldAttribute(key, value); |
414 | 0 | } |
415 | | |
416 | | void |
417 | | FormNode::setFieldAttribute(std::string const& key, QPDFObjectHandle value) |
418 | 0 | { |
419 | 0 | replace(key, value); |
420 | 0 | } |
421 | | |
422 | | void |
423 | | FormNode::setFieldAttribute(std::string const& key, Name const& value) |
424 | 340 | { |
425 | 340 | replace(key, value); |
426 | 340 | } |
427 | | |
428 | | void |
429 | | QPDFFormFieldObjectHelper::setFieldAttribute(std::string const& key, std::string const& utf8_value) |
430 | 0 | { |
431 | 0 | m->setFieldAttribute(key, utf8_value); |
432 | 0 | } |
433 | | |
434 | | void |
435 | | FormNode::setFieldAttribute(std::string const& key, std::string const& utf8_value) |
436 | 0 | { |
437 | 0 | replace(key, String::utf16(utf8_value)); |
438 | 0 | } |
439 | | |
440 | | void |
441 | | QPDFFormFieldObjectHelper::setV(QPDFObjectHandle value, bool need_appearances) |
442 | 820 | { |
443 | 820 | m->setV(value, need_appearances); |
444 | 820 | } |
445 | | |
446 | | void |
447 | | FormNode::setV(QPDFObjectHandle value, bool need_appearances) |
448 | 820 | { |
449 | 820 | if (FT() == "/Btn") { |
450 | 820 | Name name = value; |
451 | 820 | if (isCheckbox()) { |
452 | 515 | if (!name) { |
453 | 173 | warn("ignoring attempt to set a checkbox field to a value whose type is not name"); |
454 | 173 | return; |
455 | 173 | } |
456 | | // Accept any value other than /Off to mean checked. Files have been seen that use |
457 | | // /1 or other values. |
458 | 342 | setCheckBoxValue(name != "/Off"); |
459 | 342 | return; |
460 | 515 | } |
461 | 305 | if (isRadioButton()) { |
462 | 305 | if (!name) { |
463 | 52 | warn( |
464 | 52 | "ignoring attempt to set a radio button field to an object that is not a name"); |
465 | 52 | return; |
466 | 52 | } |
467 | 253 | setRadioButtonValue(name); |
468 | 253 | return; |
469 | 305 | } |
470 | 0 | if (isPushbutton()) { |
471 | 0 | warn("ignoring attempt set the value of a pushbutton field"); |
472 | 0 | } |
473 | 0 | return; |
474 | 305 | } |
475 | 0 | if (value.isString()) { |
476 | 0 | setFieldAttribute("/V", QPDFObjectHandle::newUnicodeString(value.getUTF8Value())); |
477 | 0 | } else { |
478 | 0 | setFieldAttribute("/V", value); |
479 | 0 | } |
480 | 0 | if (need_appearances) { |
481 | 0 | QPDF& qpdf = oh().getQPDF( |
482 | 0 | "QPDFFormFieldObjectHelper::setV called with need_appearances = " |
483 | 0 | "true on an object that is not associated with an owning QPDF"); |
484 | 0 | qpdf.doc().acroform().setNeedAppearances(true); |
485 | 0 | } |
486 | 0 | } |
487 | | |
488 | | void |
489 | | QPDFFormFieldObjectHelper::setV(std::string const& utf8_value, bool need_appearances) |
490 | 0 | { |
491 | 0 | m->setV(utf8_value, need_appearances); |
492 | 0 | } |
493 | | |
494 | | void |
495 | | FormNode::setV(std::string const& utf8_value, bool need_appearances) |
496 | 0 | { |
497 | 0 | setV(QPDFObjectHandle::newUnicodeString(utf8_value), need_appearances); |
498 | 0 | } |
499 | | |
500 | | void |
501 | | FormNode::setRadioButtonValue(Name const& name) |
502 | 405 | { |
503 | 405 | qpdf_expect(name); |
504 | | // Set the value of a radio button field. This has the following specific behavior: |
505 | | // * If this is a node without /Kids, assume this is a individual radio button widget and call |
506 | | // itself on the parent |
507 | | // * If this is a radio button field with children, set /V to the given value. Then, for each |
508 | | // child, if the child has the specified value as one of its keys in the /N subdictionary of |
509 | | // its /AP (i.e. its normal appearance stream dictionary), set /AS to name; otherwise, if /Off |
510 | | // is a member, set /AS to /Off. |
511 | 405 | auto kids = Kids(); |
512 | 405 | if (!kids) { |
513 | | // This is most likely one of the individual buttons. Try calling on the parent. |
514 | 188 | auto parent = Parent(); |
515 | 188 | if (parent.Kids()) { |
516 | 152 | parent.setRadioButtonValue(name); |
517 | 152 | return; |
518 | 152 | } |
519 | 188 | } |
520 | 253 | if (!isRadioButton() || !kids) { |
521 | 49 | warn("don't know how to set the value of this field as a radio button"); |
522 | 49 | return; |
523 | 49 | } |
524 | 204 | replace("/V", name); |
525 | 1.73k | for (FormNode kid: kids) { |
526 | 1.73k | auto ap = kid.AP(); |
527 | 1.73k | QPDFObjectHandle annot; |
528 | 1.73k | if (!ap) { |
529 | | // The widget may be below. If there is more than one, just find the first one. |
530 | 1.38k | for (FormNode grandkid: kid.Kids()) { |
531 | 927 | ap = grandkid.AP(); |
532 | 927 | if (ap) { |
533 | 36 | annot = grandkid; |
534 | 36 | break; |
535 | 36 | } |
536 | 927 | } |
537 | 1.38k | } else { |
538 | 354 | annot = kid; |
539 | 354 | } |
540 | 1.73k | if (!annot) { |
541 | 1.34k | warn("unable to set the value of this radio button"); |
542 | 1.34k | continue; |
543 | 1.34k | } |
544 | 392 | if (ap["/N"].contains(name.value())) { |
545 | 43 | annot.replace("/AS", name); |
546 | 349 | } else { |
547 | 349 | annot.replace("/AS", Name("/Off")); |
548 | 349 | } |
549 | 392 | } |
550 | 204 | } |
551 | | |
552 | | void |
553 | | FormNode::setCheckBoxValue(bool value) |
554 | 342 | { |
555 | 342 | auto ap = AP(); |
556 | 342 | QPDFObjectHandle annot; |
557 | 342 | if (ap) { |
558 | 222 | annot = oh(); |
559 | 222 | } else { |
560 | | // The widget may be below. If there is more than one, just find the first one. |
561 | 8.41k | for (FormNode kid: Kids()) { |
562 | 8.41k | ap = kid.AP(); |
563 | 8.41k | if (ap) { |
564 | 25 | annot = kid; |
565 | 25 | break; |
566 | 25 | } |
567 | 8.41k | } |
568 | 120 | } |
569 | 342 | std::string on_value; |
570 | 342 | if (value) { |
571 | | // Set the "on" value to the first value in the appearance stream's normal state dictionary |
572 | | // that isn't /Off. If not found, fall back to /Yes. |
573 | 268 | if (ap) { |
574 | 233 | for (auto const& item: Dictionary(ap["/N"])) { |
575 | 233 | if (item.first != "/Off") { |
576 | 138 | on_value = item.first; |
577 | 138 | break; |
578 | 138 | } |
579 | 233 | } |
580 | 192 | } |
581 | 268 | if (on_value.empty()) { |
582 | 129 | on_value = "/Yes"; |
583 | 129 | } |
584 | 268 | } |
585 | | |
586 | | // Set /AS to the on value or /Off in addition to setting /V. |
587 | 342 | auto name = Name(value ? on_value : "/Off"); |
588 | 342 | setFieldAttribute("/V", name); |
589 | 342 | if (!annot) { |
590 | 94 | warn("unable to set the value of this checkbox"); |
591 | 94 | return; |
592 | 94 | } |
593 | 248 | annot.replace("/AS", name); |
594 | 248 | } |
595 | | |
596 | | void |
597 | | QPDFFormFieldObjectHelper::generateAppearance(QPDFAnnotationObjectHelper& aoh) |
598 | 10.2k | { |
599 | 10.2k | m->generateAppearance(aoh); |
600 | 10.2k | } |
601 | | |
602 | | void |
603 | | FormNode::generateAppearance(QPDFAnnotationObjectHelper& aoh) |
604 | 10.2k | { |
605 | | // Ignore field types we don't know how to generate appearances for. Button fields don't really |
606 | | // need them -- see code in QPDFAcroFormDocumentHelper::generateAppearancesIfNeeded. |
607 | 10.2k | auto ft = FT(); |
608 | 10.2k | if (ft == "/Tx" || ft == "/Ch") { |
609 | 9.44k | generateTextAppearance(aoh); |
610 | 9.44k | } |
611 | 10.2k | } |
612 | | |
613 | | namespace |
614 | | { |
615 | | class ValueSetter: public QPDFObjectHandle::TokenFilter |
616 | | { |
617 | | public: |
618 | | ValueSetter( |
619 | | std::string const& DA, |
620 | | std::string const& V, |
621 | | std::vector<std::string> const& opt, |
622 | | double tf, |
623 | | QPDFObjectHandle::Rectangle const& bbox); |
624 | 9.33k | ~ValueSetter() override = default; |
625 | | void handleToken(QPDFTokenizer::Token const&) override; |
626 | | void handleEOF() override; |
627 | | void writeAppearance(); |
628 | | |
629 | | private: |
630 | | std::string DA; |
631 | | std::string V; |
632 | | std::vector<std::string> opt; |
633 | | double tf; |
634 | | QPDFObjectHandle::Rectangle bbox; |
635 | | enum { st_top, st_bmc, st_emc, st_end } state{st_top}; |
636 | | bool replaced{false}; |
637 | | }; |
638 | | } // namespace |
639 | | |
640 | | ValueSetter::ValueSetter( |
641 | | std::string const& DA, |
642 | | std::string const& V, |
643 | | std::vector<std::string> const& opt, |
644 | | double tf, |
645 | | QPDFObjectHandle::Rectangle const& bbox) : |
646 | 9.33k | DA(DA), |
647 | 9.33k | V(V), |
648 | 9.33k | opt(opt), |
649 | 9.33k | tf(tf), |
650 | 9.33k | bbox(bbox) |
651 | 9.33k | { |
652 | 9.33k | } |
653 | | |
654 | | void |
655 | | ValueSetter::handleToken(QPDFTokenizer::Token const& token) |
656 | 263k | { |
657 | 263k | QPDFTokenizer::token_type_e ttype = token.getType(); |
658 | 263k | std::string value = token.getValue(); |
659 | 263k | bool do_replace = false; |
660 | 263k | switch (state) { |
661 | 53.2k | case st_top: |
662 | 53.2k | writeToken(token); |
663 | 53.2k | if (token.isWord("BMC")) { |
664 | 8.50k | state = st_bmc; |
665 | 8.50k | } |
666 | 53.2k | break; |
667 | | |
668 | 17.0k | case st_bmc: |
669 | 17.0k | if ((ttype == QPDFTokenizer::tt_space) || (ttype == QPDFTokenizer::tt_comment)) { |
670 | 8.53k | writeToken(token); |
671 | 8.53k | } else { |
672 | 8.50k | state = st_emc; |
673 | 8.50k | } |
674 | | // fall through to emc |
675 | | |
676 | 191k | case st_emc: |
677 | 191k | if (token.isWord("EMC")) { |
678 | 8.48k | do_replace = true; |
679 | 8.48k | state = st_end; |
680 | 8.48k | } |
681 | 191k | break; |
682 | | |
683 | 18.6k | case st_end: |
684 | 18.6k | writeToken(token); |
685 | 18.6k | break; |
686 | 263k | } |
687 | 263k | if (do_replace) { |
688 | 8.48k | writeAppearance(); |
689 | 8.48k | } |
690 | 263k | } |
691 | | |
692 | | void |
693 | | ValueSetter::handleEOF() |
694 | 9.04k | { |
695 | 9.04k | if (!replaced) { |
696 | 571 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper replaced BMC at EOF"); |
697 | 571 | write("/Tx BMC\n"); |
698 | 571 | writeAppearance(); |
699 | 571 | } |
700 | 9.04k | } |
701 | | |
702 | | void |
703 | | ValueSetter::writeAppearance() |
704 | 9.05k | { |
705 | 9.05k | replaced = true; |
706 | | |
707 | | // This code does not take quadding into consideration because doing so requires font metric |
708 | | // information, which we don't have in many cases. |
709 | | |
710 | 9.05k | double tfh = 1.2 * tf; |
711 | 9.05k | int dx = 1; |
712 | | |
713 | | // Write one or more lines, centered vertically, possibly with one row highlighted. |
714 | | |
715 | 9.05k | auto max_rows = static_cast<size_t>((bbox.ury - bbox.lly) / tfh); |
716 | 9.05k | bool highlight = false; |
717 | 9.05k | size_t highlight_idx = 0; |
718 | | |
719 | 9.05k | std::vector<std::string> lines; |
720 | 9.05k | if (opt.empty() || (max_rows < 2)) { |
721 | 8.56k | lines.push_back(V); |
722 | 8.56k | } else { |
723 | | // Figure out what rows to write |
724 | 488 | size_t nopt = opt.size(); |
725 | 488 | size_t found_idx = 0; |
726 | 488 | bool found = false; |
727 | 5.57k | for (found_idx = 0; found_idx < nopt; ++found_idx) { |
728 | 5.22k | if (opt.at(found_idx) == V) { |
729 | 144 | found = true; |
730 | 144 | break; |
731 | 144 | } |
732 | 5.22k | } |
733 | 488 | if (found) { |
734 | | // Try to make the found item the second one, but adjust for under/overflow. |
735 | 144 | int wanted_first = QIntC::to_int(found_idx) - 1; |
736 | 144 | int wanted_last = QIntC::to_int(found_idx + max_rows) - 2; |
737 | 144 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper list found"); |
738 | 144 | if (wanted_first < 0) { |
739 | 24 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper list first too low"); |
740 | 24 | wanted_last -= wanted_first; |
741 | 24 | wanted_first = 0; |
742 | 24 | } |
743 | 144 | if (wanted_last >= QIntC::to_int(nopt)) { |
744 | 113 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper list last too high"); |
745 | 113 | auto diff = wanted_last - QIntC::to_int(nopt) + 1; |
746 | 113 | wanted_first = std::max(0, wanted_first - diff); |
747 | 113 | wanted_last -= diff; |
748 | 113 | } |
749 | 144 | highlight = true; |
750 | 144 | highlight_idx = found_idx - QIntC::to_size(wanted_first); |
751 | 2.25k | for (size_t i = QIntC::to_size(wanted_first); i <= QIntC::to_size(wanted_last); ++i) { |
752 | 2.11k | lines.push_back(opt.at(i)); |
753 | 2.11k | } |
754 | 344 | } else { |
755 | 344 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper list not found"); |
756 | | // include our value and the first n-1 rows |
757 | 344 | highlight_idx = 0; |
758 | 344 | highlight = true; |
759 | 344 | lines.push_back(V); |
760 | 3.64k | for (size_t i = 0; ((i < nopt) && (i < (max_rows - 1))); ++i) { |
761 | 3.30k | lines.push_back(opt.at(i)); |
762 | 3.30k | } |
763 | 344 | } |
764 | 488 | } |
765 | | |
766 | | // Write the lines centered vertically, highlighting if needed |
767 | 9.05k | size_t nlines = lines.size(); |
768 | 9.05k | double dy = bbox.ury - ((bbox.ury - bbox.lly - (static_cast<double>(nlines) * tfh)) / 2.0); |
769 | 9.05k | if (highlight) { |
770 | 478 | write( |
771 | 478 | "q\n0.85 0.85 0.85 rg\n" + QUtil::double_to_string(bbox.llx) + " " + |
772 | 478 | QUtil::double_to_string( |
773 | 478 | bbox.lly + dy - (tfh * (static_cast<double>(highlight_idx + 1)))) + |
774 | 478 | " " + QUtil::double_to_string(bbox.urx - bbox.llx) + " " + |
775 | 478 | QUtil::double_to_string(tfh) + " re f\nQ\n"); |
776 | 478 | } |
777 | 9.05k | dy -= tf; |
778 | 9.05k | write("q\nBT\n" + DA + "\n"); |
779 | 23.3k | for (size_t i = 0; i < nlines; ++i) { |
780 | | // We could adjust Tm to translate to the beginning the first line, set TL to tfh, and use |
781 | | // T* for each subsequent line, but doing this would require extracting any Tm from DA, |
782 | | // which doesn't seem really worth the effort. |
783 | 14.3k | if (i == 0) { |
784 | 9.04k | write( |
785 | 9.04k | QUtil::double_to_string(bbox.llx + static_cast<double>(dx)) + " " + |
786 | 9.04k | QUtil::double_to_string(bbox.lly + static_cast<double>(dy)) + " Td\n"); |
787 | 9.04k | } else { |
788 | 5.28k | write("0 " + QUtil::double_to_string(-tfh) + " Td\n"); |
789 | 5.28k | } |
790 | 14.3k | write(QPDFObjectHandle::newString(lines.at(i)).unparse() + " Tj\n"); |
791 | 14.3k | } |
792 | 9.05k | write("ET\nQ\nEMC"); |
793 | 9.05k | } |
794 | | |
795 | | namespace |
796 | | { |
797 | | class TfFinder final: public QPDFObjectHandle::TokenFilter |
798 | | { |
799 | | public: |
800 | 9.34k | TfFinder() = default; |
801 | 9.34k | ~TfFinder() final = default; |
802 | | |
803 | | void |
804 | | handleToken(QPDFTokenizer::Token const& token) final |
805 | 370k | { |
806 | 370k | auto ttype = token.getType(); |
807 | 370k | auto const& value = token.getValue(); |
808 | 370k | DA.emplace_back(token.getRawValue()); |
809 | 370k | switch (ttype) { |
810 | 53.2k | case QPDFTokenizer::tt_integer: |
811 | 74.5k | case QPDFTokenizer::tt_real: |
812 | 74.5k | last_num = strtod(value.c_str(), nullptr); |
813 | 74.5k | last_num_idx = QIntC::to_int(DA.size() - 1); |
814 | 74.5k | break; |
815 | | |
816 | 49.4k | case QPDFTokenizer::tt_name: |
817 | 49.4k | last_name = value; |
818 | 49.4k | break; |
819 | | |
820 | 62.4k | case QPDFTokenizer::tt_word: |
821 | 62.4k | if (token.isWord("Tf")) { |
822 | 8.06k | if ((last_num > 1.0) && (last_num < 1000.0)) { |
823 | | // These ranges are arbitrary but keep us from doing insane things or |
824 | | // suffering from over/underflow |
825 | 6.58k | tf = last_num; |
826 | 6.58k | } |
827 | 8.06k | tf_idx = last_num_idx; |
828 | 8.06k | font_name = last_name; |
829 | 8.06k | } |
830 | 62.4k | break; |
831 | | |
832 | 183k | default: |
833 | 183k | break; |
834 | 370k | } |
835 | 370k | } |
836 | | |
837 | | double |
838 | | getTf() const |
839 | 9.34k | { |
840 | 9.34k | return tf; |
841 | 9.34k | } |
842 | | std::string |
843 | | getFontName() const |
844 | 9.34k | { |
845 | 9.34k | return font_name; |
846 | 9.34k | } |
847 | | |
848 | | std::string |
849 | | getDA() |
850 | 9.34k | { |
851 | 9.34k | std::string result; |
852 | 9.34k | int i = -1; |
853 | 370k | for (auto const& cur: DA) { |
854 | 370k | if (++i == tf_idx) { |
855 | 6.56k | double delta = strtod(cur.c_str(), nullptr) - tf; |
856 | 6.56k | if (delta > 0.001 || delta < -0.001) { |
857 | | // tf doesn't match the font size passed to Tf, so substitute. |
858 | 618 | QTC::TC("qpdf", "QPDFFormFieldObjectHelper fallback Tf"); |
859 | 618 | result += QUtil::double_to_string(tf); |
860 | 618 | continue; |
861 | 618 | } |
862 | 6.56k | } |
863 | 369k | result += cur; |
864 | 369k | } |
865 | 9.34k | return result; |
866 | 9.34k | } |
867 | | |
868 | | private: |
869 | | double tf{11.0}; |
870 | | int tf_idx{-1}; |
871 | | std::string font_name; |
872 | | double last_num{0.0}; |
873 | | int last_num_idx{-1}; |
874 | | std::string last_name; |
875 | | std::vector<std::string> DA; |
876 | | }; |
877 | | } // namespace |
878 | | |
879 | | void |
880 | | FormNode::generateTextAppearance(QPDFAnnotationObjectHelper& aoh) |
881 | 9.44k | { |
882 | 9.44k | no_ci_warn_if( |
883 | 9.44k | !Dictionary(aoh), // There is no guarantee that aoh is a dictionary |
884 | 9.44k | "cannot generate appearance for non-dictionary annotation" // |
885 | 9.44k | ); |
886 | 9.44k | Stream AS = aoh.getAppearanceStream("/N"); // getAppearanceStream returns a stream or null. |
887 | 9.44k | if (!AS) { |
888 | 5.87k | QPDFObjectHandle::Rectangle rect = aoh.getRect(); // may silently be invalid / all zeros |
889 | 5.87k | QPDFObjectHandle::Rectangle bbox(0, 0, rect.urx - rect.llx, rect.ury - rect.lly); |
890 | 5.87k | auto* pdf = qpdf(); |
891 | 5.87k | no_ci_stop_damaged_if(!pdf, "unable to get owning QPDF for appearance generation"); |
892 | 5.87k | AS = pdf->newStream("/Tx BMC\nEMC\n"); |
893 | 5.87k | AS.replaceDict(Dictionary( |
894 | 5.87k | {{"/BBox", QPDFObjectHandle::newFromRectangle(bbox)}, |
895 | 5.87k | {"/Resources", Dictionary({{"/ProcSet", Array({Name("/PDF"), Name("/Text")})}})}, |
896 | 5.87k | {"/Type", Name("/XObject")}, |
897 | 5.87k | {"/Subtype", Name("/Form")}})); |
898 | 5.87k | if (auto ap = AP()) { |
899 | 966 | ap.replace("/N", AS); |
900 | 4.91k | } else { |
901 | 4.91k | aoh.replace("/AP", Dictionary({{"/N", AS}})); |
902 | 4.91k | } |
903 | 5.87k | } |
904 | | |
905 | 9.44k | if (AS.obj_sp().use_count() > 3) { |
906 | | // Ensures that the appearance stream is not shared by copying it if the threshold of 3 is |
907 | | // exceeded. The threshold is based on the current implementation details: |
908 | | // - One reference from the local variable AS |
909 | | // - One reference from the appearance dictionary (/AP) |
910 | | // - One reference from the object table |
911 | | // If use_count() is greater than 3, it means the appearance stream is shared elsewhere, |
912 | | // and updating it could have unintended side effects. This threshold may need to be updated |
913 | | // if the internal reference counting changes in the future. |
914 | | // |
915 | | // There is currently no explicit CI test for this code. It has been manually tested by |
916 | | // running it through CI with a threshold of 0, unconditionally copying streams. |
917 | 184 | auto data = AS.getStreamData(qpdf_dl_all); |
918 | 184 | AS = AS.copy(); |
919 | 184 | AS.replaceStreamData(std::move(data), Null::temp(), Null::temp()); |
920 | 184 | if (Dictionary AP = aoh.getAppearanceDictionary()) { |
921 | 172 | AP.replace("/N", AS); |
922 | 172 | } else { |
923 | 12 | aoh.replace("/AP", Dictionary({{"/N", AS}})); |
924 | | // aoh is a dictionary, so insertion will succeed. No need to check by retrieving it. |
925 | 12 | } |
926 | 184 | } |
927 | 9.44k | QPDFObjectHandle bbox_obj = AS.getDict()["/BBox"]; |
928 | 9.44k | if (!bbox_obj.isRectangle()) { |
929 | 85 | aoh.warn("unable to get appearance stream bounding box"); |
930 | 85 | return; |
931 | 85 | } |
932 | 9.36k | QPDFObjectHandle::Rectangle bbox = bbox_obj.getArrayAsRectangle(); |
933 | 9.36k | std::string DA = default_appearance(); |
934 | 9.36k | std::string V = value(); |
935 | | |
936 | 9.36k | TfFinder tff; |
937 | 9.36k | Pl_QPDFTokenizer tok("tf", &tff); |
938 | 9.36k | tok.writeString(DA); |
939 | 9.36k | tok.finish(); |
940 | 9.36k | double tf = tff.getTf(); |
941 | 9.36k | DA = tff.getDA(); |
942 | | |
943 | 9.36k | std::string (*encoder)(std::string const&, char) = &QUtil::utf8_to_ascii; |
944 | 9.36k | std::string font_name = tff.getFontName(); |
945 | 9.36k | if (!font_name.empty()) { |
946 | | // See if the font is encoded with something we know about. |
947 | 6.61k | Dictionary resources = AS.getDict()["/Resources"]; |
948 | 6.61k | Dictionary font = resources["/Font"][font_name]; |
949 | 6.61k | if (!font) { |
950 | 6.27k | font = getDefaultResources()["/Font"][font_name]; |
951 | 6.27k | if (resources) { |
952 | 5.35k | if (resources.indirect()) { |
953 | 190 | resources = resources.qpdf()->makeIndirectObject(resources.copy()); |
954 | 190 | AS.getDict().replace("/Resources", resources); |
955 | 190 | } |
956 | | // Use mergeResources to force /Font to be local |
957 | 5.35k | QPDFObjectHandle res = resources; |
958 | 5.35k | res.mergeResources(Dictionary({{"/Font", Dictionary::empty()}})); |
959 | 5.35k | res.getKey("/Font").replace(font_name, font); |
960 | 5.35k | } |
961 | 6.27k | } |
962 | | |
963 | 6.61k | if (Name Encoding = font["/Encoding"]) { |
964 | 973 | if (Encoding == "/WinAnsiEncoding") { |
965 | 328 | encoder = &QUtil::utf8_to_win_ansi; |
966 | 645 | } else if (Encoding == "/MacRomanEncoding") { |
967 | 41 | encoder = &QUtil::utf8_to_mac_roman; |
968 | 41 | } |
969 | 973 | } |
970 | 6.61k | } |
971 | | |
972 | 9.36k | V = (*encoder)(V, '?'); |
973 | | |
974 | 9.36k | std::vector<std::string> opt; |
975 | 9.36k | if (isChoice() && (getFlags() & ff_ch_combo) == 0) { |
976 | 1.30k | opt = getChoices(); |
977 | 17.1k | for (auto& o: opt) { |
978 | 17.1k | o = (*encoder)(o, '?'); |
979 | 17.1k | } |
980 | 1.30k | } |
981 | | |
982 | 9.36k | std::string result; |
983 | 9.36k | pl::String pl(result); |
984 | 9.36k | ValueSetter vs(DA, V, opt, tf, bbox); |
985 | 9.36k | Pl_QPDFTokenizer vs_tok("", &vs, &pl); |
986 | 9.36k | vs_tok.writeString(AS.getStreamData(qpdf_dl_all)); |
987 | 9.36k | vs_tok.finish(); |
988 | 9.36k | AS.replaceStreamData(std::move(result), Null::temp(), Null::temp()); |
989 | 9.36k | } |