pl_function.cc
1.96 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#include <qpdf/assert_test.h>
#include <qpdf/Pl_Base64.hh>
#include <qpdf/Pl_Function.hh>
#include <qpdf/Pl_String.hh>
#include <iostream>
namespace
{
struct Count
{
int count{0};
};
} // namespace
int
f(unsigned char const* data, size_t len, void* udata)
{
auto c = reinterpret_cast<Count*>(udata);
++c->count;
std::cout << "got " << data << "(" << len << ")" << std::endl;
if (c->count == 3) {
return 1;
}
return 0;
}
int
g(char const* data, size_t len, void* udata)
{
auto c = reinterpret_cast<Count*>(udata);
++c->count;
std::cout << "signed got " << data << "(" << len << ")" << std::endl;
if (c->count == 2) {
return 2;
}
return 0;
}
int
main(int argc, char* argv[])
{
Pl_Function p1("p1", nullptr, [](unsigned char const* data, size_t len) {
std::cout << "p1: " << len << ": " << data << std::endl;
});
p1.write(reinterpret_cast<unsigned char const*>("potato"), 6);
std::string s;
Pl_String ps("string", nullptr, s);
Pl_Base64 b("base64", &ps, Pl_Base64::a_encode);
Pl_Function p2("p2", &b, [](unsigned char const* data, size_t len) {
std::cout << "p2: " << len << ": " << data << std::endl;
});
p2.write(reinterpret_cast<unsigned char const*>("salad"), 5);
p2.finish();
assert(s == "c2FsYWQ=");
Count c;
Pl_Function p3("c-function", nullptr, f, &c);
p3 << "one";
p3 << "two";
try {
p3 << "three";
assert(false);
} catch (std::runtime_error& e) {
std::cout << "three threw " << e.what() << std::endl;
}
p3 << "four";
p3.finish();
assert(c.count == 4);
c.count = 0;
Pl_Function p4("c-function", nullptr, g, &c);
p4 << "potato";
try {
p4 << "salad";
assert(false);
} catch (std::runtime_error& e) {
std::cout << "salad threw " << e.what() << std::endl;
}
p4 << "quack";
p4.finish();
assert(c.count == 3);
return 0;
}