forked from openkylin/qpdf
110 lines
3.1 KiB
C++
110 lines
3.1 KiB
C++
#include <qpdf/BufferInputSource.hh>
|
|
#include <qpdf/PointerHolder.hh>
|
|
#include <qpdf/Buffer.hh>
|
|
#include <qpdf/QPDFTokenizer.hh>
|
|
#include <iostream>
|
|
#include <cstring>
|
|
|
|
static PointerHolder<Buffer>
|
|
get_buffer()
|
|
{
|
|
size_t size = 3172;
|
|
PointerHolder<Buffer> b(new Buffer(size));
|
|
unsigned char* p = b->getBuffer();
|
|
for (size_t i = 0; i < size; ++i)
|
|
{
|
|
p[i] = static_cast<unsigned char>(i & 0xff);
|
|
}
|
|
return b;
|
|
}
|
|
|
|
class Finder: public InputSource::Finder
|
|
{
|
|
public:
|
|
Finder(PointerHolder<InputSource> is, std::string const& after) :
|
|
is(is),
|
|
after(after)
|
|
{
|
|
}
|
|
virtual ~Finder()
|
|
{
|
|
}
|
|
virtual bool check();
|
|
|
|
private:
|
|
PointerHolder<InputSource> is;
|
|
std::string after;
|
|
};
|
|
|
|
bool
|
|
Finder::check()
|
|
{
|
|
QPDFTokenizer tokenizer;
|
|
QPDFTokenizer::Token t = tokenizer.readToken(is, "finder", true);
|
|
if (t == QPDFTokenizer::Token(QPDFTokenizer::tt_word, "potato"))
|
|
{
|
|
t = tokenizer.readToken(is, "finder", true);
|
|
return (t == QPDFTokenizer::Token(QPDFTokenizer::tt_word, after));
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void check(char const* description, bool expected, bool actual)
|
|
{
|
|
std::cout << description << ": "
|
|
<< ((actual == expected) ? "PASS" : "FAIL")
|
|
<< std::endl;
|
|
}
|
|
|
|
int main()
|
|
{
|
|
PointerHolder<Buffer> b1 = get_buffer();
|
|
unsigned char* b = b1->getBuffer();
|
|
// Straddle block boundaries
|
|
memcpy(b + 1022, "potato", 6);
|
|
// Overlap so that the first check() would advance past the start
|
|
// of the next match
|
|
memcpy(b + 2037, "potato potato salad ", 20);
|
|
PointerHolder<InputSource> is =
|
|
new BufferInputSource("test buffer input source", b1.getPointer());
|
|
Finder f1(is, "salad");
|
|
check("find potato salad", true,
|
|
is->findFirst("potato", 0, 0, f1));
|
|
check("barely find potato salad", true,
|
|
is->findFirst("potato", 1100, 945, f1));
|
|
check("barely find potato salad", true,
|
|
is->findFirst("potato", 2000, 45, f1));
|
|
check("potato salad is too late", false,
|
|
is->findFirst("potato", 1100, 944, f1));
|
|
check("potato salad is too late", false,
|
|
is->findFirst("potato", 2000, 44, f1));
|
|
check("potato salad not found", false,
|
|
is->findFirst("potato", 2045, 0, f1));
|
|
check("potato salad not found", false,
|
|
is->findFirst("potato", 0, 1, f1));
|
|
|
|
// Put one more right at EOF
|
|
memcpy(b + b1->getSize() - 12, "potato salad", 12);
|
|
check("potato salad at EOF", true,
|
|
is->findFirst("potato", 3000, 0, f1));
|
|
|
|
is->findFirst("potato", 0, 0, f1);
|
|
check("findFirst found first", true,
|
|
is->tell() == 2056);
|
|
check("findLast found potato salad", true,
|
|
is->findLast("potato", 0, 0, f1));
|
|
check("findLast found at EOF", true,
|
|
is->tell() == 3172);
|
|
|
|
// Make check() bump into EOF
|
|
memcpy(b + b1->getSize() - 6, "potato", 6);
|
|
check("potato but not salad salad at EOF", false,
|
|
is->findFirst("potato", 3000, 0, f1));
|
|
check("findLast found potato salad", true,
|
|
is->findLast("potato", 0, 0, f1));
|
|
check("findLast found first one", true,
|
|
is->tell() == 2056);
|
|
|
|
return 0;
|
|
}
|