Branch data Line data Source code
# 1 : : // Copyright (c) 2011 The LevelDB Authors. All rights reserved.
# 2 : : // Use of this source code is governed by a BSD-style license that can be
# 3 : : // found in the LICENSE file. See the AUTHORS file for names of contributors.
# 4 : :
# 5 : : #include "table/format.h"
# 6 : :
# 7 : : #include "leveldb/env.h"
# 8 : : #include "port/port.h"
# 9 : : #include "table/block.h"
# 10 : : #include "util/coding.h"
# 11 : : #include "util/crc32c.h"
# 12 : :
# 13 : : namespace leveldb {
# 14 : :
# 15 : 6209 : void BlockHandle::EncodeTo(std::string* dst) const {
# 16 : : // Sanity check that all fields have been set
# 17 : 6209 : assert(offset_ != ~static_cast<uint64_t>(0));
# 18 : 6209 : assert(size_ != ~static_cast<uint64_t>(0));
# 19 : 6209 : PutVarint64(dst, offset_);
# 20 : 6209 : PutVarint64(dst, size_);
# 21 : 6209 : }
# 22 : :
# 23 : 2547260 : Status BlockHandle::DecodeFrom(Slice* input) {
# 24 [ + - ][ + - ]: 2547260 : if (GetVarint64(input, &offset_) && GetVarint64(input, &size_)) {
# 25 : 2547260 : return Status::OK();
# 26 : 2547260 : } else {
# 27 : 0 : return Status::Corruption("bad block handle");
# 28 : 0 : }
# 29 : 2547260 : }
# 30 : :
# 31 : 842 : void Footer::EncodeTo(std::string* dst) const {
# 32 : 842 : const size_t original_size = dst->size();
# 33 : 842 : metaindex_handle_.EncodeTo(dst);
# 34 : 842 : index_handle_.EncodeTo(dst);
# 35 : 842 : dst->resize(2 * BlockHandle::kMaxEncodedLength); // Padding
# 36 : 842 : PutFixed32(dst, static_cast<uint32_t>(kTableMagicNumber & 0xffffffffu));
# 37 : 842 : PutFixed32(dst, static_cast<uint32_t>(kTableMagicNumber >> 32));
# 38 : 842 : assert(dst->size() == original_size + kEncodedLength);
# 39 : 842 : (void)original_size; // Disable unused variable warning.
# 40 : 842 : }
# 41 : :
# 42 : 1439 : Status Footer::DecodeFrom(Slice* input) {
# 43 : 1439 : const char* magic_ptr = input->data() + kEncodedLength - 8;
# 44 : 1439 : const uint32_t magic_lo = DecodeFixed32(magic_ptr);
# 45 : 1439 : const uint32_t magic_hi = DecodeFixed32(magic_ptr + 4);
# 46 : 1439 : const uint64_t magic = ((static_cast<uint64_t>(magic_hi) << 32) |
# 47 : 1439 : (static_cast<uint64_t>(magic_lo)));
# 48 [ - + ]: 1439 : if (magic != kTableMagicNumber) {
# 49 : 0 : return Status::Corruption("not an sstable (bad magic number)");
# 50 : 0 : }
# 51 : :
# 52 : 1439 : Status result = metaindex_handle_.DecodeFrom(input);
# 53 [ + - ]: 1439 : if (result.ok()) {
# 54 : 1439 : result = index_handle_.DecodeFrom(input);
# 55 : 1439 : }
# 56 [ + - ]: 1439 : if (result.ok()) {
# 57 : : // We skip over any leftover data (just padding for now) in "input"
# 58 : 1439 : const char* end = magic_ptr + 8;
# 59 : 1439 : *input = Slice(end, input->data() + input->size() - end);
# 60 : 1439 : }
# 61 : 1439 : return result;
# 62 : 1439 : }
# 63 : :
# 64 : : Status ReadBlock(RandomAccessFile* file, const ReadOptions& options,
# 65 : 19678 : const BlockHandle& handle, BlockContents* result) {
# 66 : 19678 : result->data = Slice();
# 67 : 19678 : result->cachable = false;
# 68 : 19678 : result->heap_allocated = false;
# 69 : :
# 70 : : // Read the block contents as well as the type/crc footer.
# 71 : : // See table_builder.cc for the code that built this structure.
# 72 : 19678 : size_t n = static_cast<size_t>(handle.size());
# 73 : 19678 : char* buf = new char[n + kBlockTrailerSize];
# 74 : 19678 : Slice contents;
# 75 : 19678 : Status s = file->Read(handle.offset(), n + kBlockTrailerSize, &contents, buf);
# 76 [ - + ]: 19678 : if (!s.ok()) {
# 77 : 0 : delete[] buf;
# 78 : 0 : return s;
# 79 : 0 : }
# 80 [ - + ]: 19678 : if (contents.size() != n + kBlockTrailerSize) {
# 81 : 0 : delete[] buf;
# 82 : 0 : return Status::Corruption("truncated block read", file->GetName());
# 83 : 0 : }
# 84 : :
# 85 : : // Check the crc of the type and the block contents
# 86 : 19678 : const char* data = contents.data(); // Pointer to where Read put the data
# 87 [ + - ]: 19678 : if (options.verify_checksums) {
# 88 : 19678 : const uint32_t crc = crc32c::Unmask(DecodeFixed32(data + n + 1));
# 89 : 19678 : const uint32_t actual = crc32c::Value(data, n + 1);
# 90 [ - + ]: 19678 : if (actual != crc) {
# 91 : 0 : delete[] buf;
# 92 : 0 : s = Status::Corruption("block checksum mismatch", file->GetName());
# 93 : 0 : return s;
# 94 : 0 : }
# 95 : 19678 : }
# 96 : :
# 97 : 19678 : switch (data[n]) {
# 98 [ + - ]: 19678 : case kNoCompression:
# 99 [ + + ]: 19678 : if (data != buf) {
# 100 : : // File implementation gave us pointer to some other data.
# 101 : : // Use it directly under the assumption that it will be live
# 102 : : // while the file is open.
# 103 : 19192 : delete[] buf;
# 104 : 19192 : result->data = Slice(data, n);
# 105 : 19192 : result->heap_allocated = false;
# 106 : 19192 : result->cachable = false; // Do not double-cache
# 107 : 19192 : } else {
# 108 : 486 : result->data = Slice(buf, n);
# 109 : 486 : result->heap_allocated = true;
# 110 : 486 : result->cachable = true;
# 111 : 486 : }
# 112 : :
# 113 : : // Ok
# 114 : 19678 : break;
# 115 [ - + ]: 0 : case kSnappyCompression: {
# 116 : 0 : size_t ulength = 0;
# 117 [ # # ]: 0 : if (!port::Snappy_GetUncompressedLength(data, n, &ulength)) {
# 118 : 0 : delete[] buf;
# 119 : 0 : return Status::Corruption("corrupted compressed block contents", file->GetName());
# 120 : 0 : }
# 121 : 0 : char* ubuf = new char[ulength];
# 122 [ # # ]: 0 : if (!port::Snappy_Uncompress(data, n, ubuf)) {
# 123 : 0 : delete[] buf;
# 124 : 0 : delete[] ubuf;
# 125 : 0 : return Status::Corruption("corrupted compressed block contents", file->GetName());
# 126 : 0 : }
# 127 : 0 : delete[] buf;
# 128 : 0 : result->data = Slice(ubuf, ulength);
# 129 : 0 : result->heap_allocated = true;
# 130 : 0 : result->cachable = true;
# 131 : 0 : break;
# 132 : 0 : }
# 133 [ - + ]: 0 : default:
# 134 : 0 : delete[] buf;
# 135 : 0 : return Status::Corruption("bad block type", file->GetName());
# 136 : 19678 : }
# 137 : :
# 138 : 19678 : return Status::OK();
# 139 : 19678 : }
# 140 : :
# 141 : : } // namespace leveldb
|