blob: 9a307ea0ebedb0edd35ce91afec6bca614247b4e [file] [log] [blame]
// Copyright 2020 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
#pragma once
#include <cstddef>
#include "pw_kvs/alignment.h"
#include "pw_span/span.h"
#include "pw_status/status.h"
namespace pw {
namespace kvs {
class ChecksumAlgorithm {
public:
// Resets the checksum to its initial state.
virtual void Reset() = 0;
// Updates the checksum with the provided data.
virtual void Update(span<const std::byte> data) = 0;
// Updates the checksum from a pointer and size.
void Update(const void* data, size_t size_bytes) {
return Update(
span<const std::byte>(static_cast<const std::byte*>(data), size_bytes));
}
// Returns the final result of the checksum. Update() can no longer be called
// after this. The returned span is valid until a call to Reset().
//
// Finish MUST be called before calling Verify.
span<const std::byte> Finish() {
Finalize(); // Implemented by derived classes, if required.
return state();
}
// Returns the size of the checksum state.
constexpr size_t size_bytes() const { return state_.size(); }
// Compares a calculated checksum to this checksum's state. The checksum must
// be at least as large as size_bytes(). If it is larger, bytes beyond
// size_bytes() are ignored.
//
// Finish MUST be called before calling Verify.
Status Verify(span<const std::byte> checksum) const;
protected:
// A derived class provides a span of its state buffer.
constexpr ChecksumAlgorithm(span<const std::byte> state) : state_(state) {}
// Protected destructor prevents deleting ChecksumAlgorithms from the base
// class, so that it is safe to have a non-virtual destructor.
~ChecksumAlgorithm() = default;
// Returns the current checksum state.
constexpr span<const std::byte> state() const { return state_; }
private:
// Checksums that require finalizing operations may override this method.
virtual void Finalize() {}
span<const std::byte> state_;
};
// A checksum algorithm for which Verify always passes. This can be used to
// disable checksum verification for a particular entry format.
class IgnoreChecksum final : public ChecksumAlgorithm {
public:
constexpr IgnoreChecksum() : ChecksumAlgorithm({}) {}
void Reset() override {}
void Update(span<const std::byte>) override {}
};
// Calculates a checksum in kAlignmentBytes chunks. Checksum classes can inherit
// from this and implement UpdateAligned and FinalizeAligned instead of Update
// and Finalize.
template <size_t kAlignmentBytes, size_t kBufferSize = kAlignmentBytes>
class AlignedChecksum : public ChecksumAlgorithm {
public:
void Update(span<const std::byte> data) final {
writer_.Write(data)
.IgnoreError(); // TODO(b/242598609): Handle Status properly
}
protected:
constexpr AlignedChecksum(span<const std::byte> state)
: ChecksumAlgorithm(state),
output_(*this),
writer_(kAlignmentBytes, output_) {}
~AlignedChecksum() = default;
private:
static_assert(kBufferSize >= kAlignmentBytes);
void Finalize() final {
writer_.Flush().IgnoreError(); // TODO(b/242598609): Handle Status properly
FinalizeAligned();
}
virtual void UpdateAligned(span<const std::byte> data) = 0;
virtual void FinalizeAligned() = 0;
class CallUpdateAligned final : public Output {
public:
constexpr CallUpdateAligned(AlignedChecksum& object) : object_(object) {}
private:
StatusWithSize DoWrite(span<const std::byte> data) override {
object_.UpdateAligned(data);
return StatusWithSize(data.size());
}
AlignedChecksum& object_;
};
CallUpdateAligned output_;
AlignedWriterBuffer<kBufferSize> writer_;
};
} // namespace kvs
} // namespace pw