diff options
| -rw-r--r-- | mupdf/pdf_function.c | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/mupdf/pdf_function.c b/mupdf/pdf_function.c index e0a606f6..7f819a69 100644 --- a/mupdf/pdf_function.c +++ b/mupdf/pdf_function.c @@ -495,6 +495,9 @@ loadpostscriptfunc(pdf_function *func, pdf_xref *xref, fz_obj *dict, int num, in fz_error error; fz_stream *stream; int codeptr; + char buf[64]; + pdf_token_e tok; + int len; pdf_logrsrc("load postscript function (%d %d R)\n", num, gen); @@ -502,7 +505,14 @@ loadpostscriptfunc(pdf_function *func, pdf_xref *xref, fz_obj *dict, int num, in if (error) return fz_rethrow(error, "cannot open calculator function stream"); - if (fz_readbyte(stream) != '{') + error = pdf_lex(&tok, stream, buf, sizeof buf, &len); + if (error) + { + fz_dropstream(stream); + return fz_rethrow(error, "stream is not a calculator function"); + } + + if (tok != PDF_TOBRACE) { fz_dropstream(stream); return fz_throw("stream is not a calculator function"); |