/src/librawspeed/fuzz/librawspeed/decompressors/PentaxDecompressor.cpp
Line | Count | Source |
1 | | /* |
2 | | RawSpeed - RAW file decoder. |
3 | | |
4 | | Copyright (C) 2017 Roman Lebedev |
5 | | |
6 | | This library is free software; you can redistribute it and/or |
7 | | modify it under the terms of the GNU Lesser General Public |
8 | | License as published by the Free Software Foundation; either |
9 | | version 2 of the License, or (at your option) any later version. |
10 | | |
11 | | This library is distributed in the hope that it will be useful, |
12 | | but WITHOUT ANY WARRANTY; without even the implied warranty of |
13 | | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
14 | | Lesser General Public License for more details. |
15 | | |
16 | | You should have received a copy of the GNU Lesser General Public |
17 | | License along with this library; if not, write to the Free Software |
18 | | Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
19 | | */ |
20 | | |
21 | | #include "decompressors/PentaxDecompressor.h" |
22 | | #include "MemorySanitizer.h" |
23 | | #include "adt/Casts.h" |
24 | | #include "adt/Optional.h" |
25 | | #include "common/RawImage.h" |
26 | | #include "common/RawspeedException.h" |
27 | | #include "fuzz/Common.h" |
28 | | #include "io/Buffer.h" |
29 | | #include "io/ByteStream.h" |
30 | | #include "io/Endianness.h" |
31 | | #include <cassert> |
32 | | #include <cstdint> |
33 | | #include <cstdio> |
34 | | |
35 | | extern "C" int LLVMFuzzerTestOneInput(const uint8_t* Data, size_t Size); |
36 | | |
37 | 915 | extern "C" int LLVMFuzzerTestOneInput(const uint8_t* Data, size_t Size) { |
38 | 915 | assert(Data); |
39 | | |
40 | 915 | try { |
41 | 915 | const rawspeed::Buffer b( |
42 | 915 | Data, rawspeed::implicit_cast<rawspeed::Buffer::size_type>(Size)); |
43 | 915 | const rawspeed::DataBuffer db(b, rawspeed::Endianness::little); |
44 | 915 | rawspeed::ByteStream bs(db); |
45 | | |
46 | 915 | rawspeed::RawImage mRaw(CreateRawImage(bs)); |
47 | | |
48 | 915 | rawspeed::Optional<rawspeed::ByteStream> metaData; |
49 | | |
50 | 915 | const bool haveMetadata = bs.get<uint32_t>(); |
51 | 915 | if (haveMetadata) { |
52 | 360 | const auto medataLength = bs.get<uint32_t>(); |
53 | 360 | metaData = bs.getStream(medataLength); |
54 | 360 | } |
55 | | |
56 | 915 | rawspeed::ByteStream rawData = bs.getStream(bs.getRemainSize()); |
57 | | |
58 | 915 | rawspeed::PentaxDecompressor p(mRaw, metaData); |
59 | 915 | mRaw->createData(); |
60 | 915 | p.decompress(rawData); |
61 | | |
62 | 915 | rawspeed::MSan::CheckMemIsInitialized( |
63 | 915 | mRaw->getByteDataAsUncroppedArray2DRef()); |
64 | 915 | } catch (const rawspeed::RawspeedException&) { // NOLINT(bugprone-empty-catch) |
65 | | // Exceptions are good, crashes are bad. |
66 | 833 | } |
67 | | |
68 | 915 | return 0; |
69 | 915 | } |