mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2026-08-31 05:07:01 +00:00
* ec: uniform shard block layout An EC volume is striped as 1GiB blocks until less than one row remains, then 1MiB blocks, and consecutive blocks land on different shards. With ec.encode's -fullPercent 95 against the 30GiB default limit, ~30% of every volume sits in that 1MiB tail, so a 4MB filer chunk there is five stripes on five servers. New encodes now use one block per shard, sized ceil(datSize/dataShards) rounded up to 1MiB and recorded in the .vif (EcShardConfig.block_size, also carried by the .ecsum manifest). A needle now maps to one shard unless it is larger than the block or straddles a boundary. The chosen size equals the legacy layout's padded shard length for every input, so shard sizes, capacity math, and the shard-size credibility checks are unchanged; only the byte placement moved. Reads, decode, and scrub resolve the block sizes from the volume's .vif; absence keeps the legacy interpretation, so existing EC volumes read exactly as before. Rebuild is layout-agnostic. weed fix -ecx recovers the layout from the .vif, else the .ecsum sidecar, and with neither de-stripes under both candidate layouts and keeps the one that indexes more valid needles. Same change in the Rust volume server, which now also streams the encode in 256KB sub-batches like Go instead of allocating whole blocks, and computes the large-row count as shardSize/largeBlock to match Go on exact multiples. On a 26MB fixture both encoders produce byte-identical shards, and a Go-written .vif parses in Rust with the block size intact. * ec: resolve the rust ecx rebuild through the recorded layout The Rust rebuild path regenerated a lost .ecx by scanning the logical .dat through a hand-rolled pure-1MiB striping, which was already wrong for legacy volumes with large-block rows and is wrong for any uniform volume with a block past 1MiB. Route the scan through locate_data with the .vif-recorded block size, the same mapping the read path uses. Also seed the new tests' random data instead of the deprecated global math/rand.Read. * ec: fail the Rust ecx rebuild on any shard read error A read error mid-scan published the entries collected so far as a successful .ecx, and read_at's byte count was ignored so a legal short read passed as complete — a truncated or failing shard could produce a silently incomplete recovery index. Exact-read semantics in read_from_data_shards, error propagation in the needle walk, and a truncated-shard regression test. * ec: fail the mount on an unreadable or malformed vif Both servers silently fell back to the legacy layout when an existing .vif could not be read or parsed. Every new encode records a positive uniform block size there, so the fallback mounted the same shards with legacy offset math and could return wrong data. Absent stays legal (legacy volumes predate the sidecar), and a zero-byte stub still reads as absent (Go's MaybeLoadVolumeInfo convention, now mirrored in Rust); a present-but-unreadable or malformed .vif fails the mount instead. * ec: bound the reconstruct fan-out of one needle's intervals A degraded interval fans out a read to every reachable shard location, each with a buffer the size of the interval. Reading a needle's intervals in parallel multiplied that by the interval concurrency: a needle spanning 8 blocks could hold 8 x MaxShardCount remote reads and buffers at once, where the sequential version peaked at MaxShardCount. Give each needle a single reconstruct budget its intervals share, held for the buffer's lifetime, so separate reads stay independent but one read cannot multiply its own fan-out. * ec: drop the duplicated shard-size formula calculateExpectedShardSize reimplemented the padding rule that UniformBlockSize already owns — TestUniformBlockSizeMatchesLegacyShardSize asserts the two agree for every input — so a change to the rule would have had to be made in both. Defer to the helper, keeping the historic answer for an empty .dat. * ec: resolve the shard block layout from whatever records it Four places still answered the layout question by inference when a record of it was available, or accepted an answer that was not one: - A mount with no .vif defaulted to the legacy layout; the bitrot sidecar records the same config at encode time, so take it when present, as weed fix -ecx already does. The vif itself is now parsed once per mount rather than twice. - The Rust ecx rebuild derived its row count from the padded shard extent, which under the legacy layout reads a shard that is an exact large-block multiple as one row too many. Pass the encode-time .dat size from the .vif and keep the extent as the fallback. - weed fix -ecx read the block size outside the EC-config guard (collapsing the unknown sentinel into a definitive legacy), only wrote the recovered layout back when the .vif was absent rather than unusable, and broke a scan tie by candidate order instead of the documented reach. - The uniform layout tripped writeDatFile's large-block ambiguity guard, which cannot apply when the large and small blocks are the same size. * ec: give the index-recovery tests a parseable vif The fixtures wrote the literal bytes "volinfo" as the source .vif and the recovery copies it verbatim, so the receiving server then mounted the volume from a .vif it could not parse. That used to pass by silently defaulting to the legacy layout; a mount now refuses a vif it cannot read, which is what the tests were exercising all along without meaning to. * ec: validate the layout a vif records, not just its syntax Review follow-ups on the mount-strictness change: - A .vif can parse and still record a block size no encoder could have produced (negative, or not a whole number of small blocks). Both servers took it and mapped every read through it. ValidateBlockSize / the Rust mirror now refuse the mount, the same way an unparseable vif does; 0 stays valid as the legacy two-tier layout. - The bitrot-sidecar fallback accepted parity_shards == 0 and summed the counts in their own width, so values near the ceiling wrapped past the MaxShardCount bound. Require both counts and sum in a wider type. - weed fix -ecx treated a config with only DataShards > 0 as usable, so a half-written .vif suppressed the recovery paths AND survived the rewrite. Require a complete, in-range config before trusting it. - Returning the vif-load error left the .ecx and .ecj descriptors open; repeated mount attempts on malformed metadata could exhaust them. * ec: refuse to act on a layout the metadata does not establish - The worker encode only logged a failed .vif write and skipped it in the distribution set, and treated the .ecsum write as best-effort. A worker whose disk filled after the much larger shards landed could still distribute, mount, verify shard inventory, and delete the source replicas — leaving holders with shards whose geometry nothing records. Both writes and both inclusions are encode success conditions now. - A generation-matching .ecsum that disagreed with the .vif geometry only disabled checksums in Go, and in Rust was not compared at all, so protection stayed On while reads used the other layout. Both files record the layout their generation was encoded with, so a disagreement now fails the mount. * ec: reject an invalid recorded block size in weed fix -ecx A .vif with valid shard counts but a negative or unaligned block size was marked usable: a positive invalid value pinned the scan to a geometry that de-stripes to garbage, and a negative one ran the dual scan but left the invalid .vif in place afterwards. Validate it with the same rule the mount applies, and when it fails leave the layout unknown so the scan recovers it and the file is rewritten. * ec: validate the sidecar layout weed fix -ecx recovers from The .ecsum fallback was taken on DataShards > 0 alone, so a CRC-valid sidecar carrying the wrong generation, an incomplete ratio, or an unaligned block size would pin the reconstruction to one incorrect uniform-layout candidate instead of letting the dual scan decide. Require generation 0, a complete in-range ratio, and a valid block size; anything less leaves the layout unknown, which is the answer that still recovers by scanning. * ec: let only a genuinely absent sidecar choose the legacy layout With no .vif the bitrot sidecar is the only record of a volume's layout, and the mount fallback read a failed load, an unusable config, or a sidecar stamped for another generation as "assume legacy". A uniform generation-0 volume could therefore mount with legacy or another generation's geometry and answer reads with the wrong bytes. Present-but-unusable now fails the mount; only actual absence keeps the legacy defaults. Shared as EcShardConfigFromSidecar so every caller reads the sidecar the same way. * ec: treat a recorded-but-impossible layout as corruption, not as legacy - A .vif whose ecShardConfig is PRESENT but records an impossible ratio was answered with the default 10+4 and the legacy block layout, in both languages. That reads a uniform volume's shards at the wrong offsets and returns the wrong bytes. Only an entirely absent config still means "this predates the record"; a present one that cannot be true fails the mount. - The shard-count bound summed two uint32 counts as int, which wraps on a 32-bit build: 0x7fffffff + 0x7fffffff lands at -2 and slips under MaxShardCount. ValidEcShardCounts sums in uint64, and every EC call site that checked a recorded ratio now goes through it. * ec: rebuild on the geometry the sidecar records, and flag it when it disagrees The rebuild RPC passes BackgroundECContext, so RebuildEcFiles resolves the layout itself — and it resolved a missing or invalid .vif to the default 10+4 with the legacy block size. Two consequences: a 12+4 volume was reconstructed through a 10+4 matrix, which produces wrong bytes and never regenerates shards 14-15; and the chosen geometry then contradicted a valid uniform sidecar, which loadRebuildSidecar reported as BitrotOff — silently skipping the input and regenerated-shard checksum checks precisely when the volume had already lost its metadata. The layout now resolves from the bitrot sidecar (found across the server's disks, not just beside the base name) before falling back to the defaults, and a present-but-impossible ratio fails instead of being replaced. A sidecar that contradicts the chosen geometry is BitrotInvalid, which the existing unsafeIgnoreSidecar override still lets an operator push past. * ec: let the Rust rebuild read metadata off a sibling disk read_ec_shard_config searches only the location the rebuild writes into, so a volume whose .vif or generation-0 .ecsum sits on another of the server's disks resolved to the default 10+4 with the legacy block layout — the Rust half of the geometry-guessing the Go rebuild just stopped doing. It then reconstructs a custom-ratio or uniform volume through the wrong Reed-Solomon matrix and de-striping geometry. The rebuild now looks for the .vif in its own location and then each sibling, falls back to the generation-0 sidecar wherever that lives, and only defaults when neither exists anywhere. The encode-time .dat size the ecx rebuild needs is resolved the same way. * ec: resolve a rebuild's vif from every directory that may hold it RebuildEcFiles probed only <data-base>.vif. The caller knows the selected location's index directory and the sibling locations, but passed neither for metadata: additionalDirs carried shard directories only, and were searched for shards and the checksum sidecar. A split -dir/-dir.idx layout, or a disk holding only shards, therefore resolved a pre-sidecar custom-ratio volume to 10+4 and reconstructed through the wrong matrix — never regenerating shards 14-15. The caller now hands over the index and sibling directories, and the resolver probes the vif across all of them, matching what the Rust resolver already does for both the vif and the sidecar. * ec: make every rebuild consumer agree on the layout it resolved - The post-rebuild bitrot backfill re-derived the geometry from this directory's .vif alone and dropped the block size entirely, so a rebuild that resolved its layout from a sibling, the sidecar, or a uniform vif wrote a manifest describing a DIFFERENT layout — one later mounts reject, or that covers only the default shard count. The layout is resolved once now, through an exported ResolveRebuildECContext, and the rebuild and the backfill share that answer. - The Rust rebuild collected only each location's data directory, so a sibling's INDEX directory — where a split -dir/-dir.idx layout keeps .ecx/.ecj/.vif — was never probed, and a custom-ratio volume still resolved to 10+4 with the legacy layout. Both directories of every location are carried now, deduped against the rebuild's own. - A shard delivery can bring the checksum manifest with it, but the receive path only writes the file: a server that already had the volume mounted kept its resolved protection state (off) until a remount. The mount RPC re-resolves it once the shards it describes have been added. * ec: cover the rebuild's directory search with tests Reviewers flagged the sibling index directory twice, and the fix that closed it had no test of its own: the assembly sat inline in the rebuild handler, reachable only through a gRPC call against a populated store. Lifting it into rebuildSearchDirs / select_rebuild_location makes the rule assertable — a sibling contributes BOTH its data and its index directory, a shared index directory is listed once, and the rebuild's own data directory never repeats. Writing the Rust cases surfaced that the two implementations do not agree on where the rebuild's own index directory belongs, and both are right: Go's resolver takes a single directory list, so that directory has to be inside it, while Rust's takes the rebuild's data and index directories as their own arguments and would search them twice. The tests now state which contract each side is holding to, so neither drifts into the other's shape. Pure refactor otherwise; no behaviour change. * ec: search the index directory for the layout sidecar The Rust resolver looked for the generation-0 .ecsum in the rebuild's data directory and the sibling list, but not in the rebuild's own index directory — while the .vif lookup directly above it did, and Go's findBitrotSidecar has always checked both bases. On a split -dir/-dir.idx location that directory is where the metadata lives, and callers leave it out of the sibling list precisely because it is passed here separately, so nothing searched it. With no .vif anywhere the sidecar is the only surviving record of the layout. Missing it resolved a 12+4 uniform volume to 10+4 with the legacy striping — the test added here fails with (10, 4, 0) against the old code — and the rebuild then reconstructs through the wrong matrix and writes .ecx offsets that no reader can follow. * ec: let the rebuild see its own index directory The Rust rebuild takes a single flat directory list — the shape Go's RebuildEcFiles uses — so it cannot be handed the rebuild location's index directory separately the way the layout resolvers are, and the handler was passing the sibling list, which deliberately omits exactly that directory. On a split -dir/-dir.idx location that is where .ecx and .vif live, so the shard and index lookups could not see them. Go has always carried that directory in additionalDirs; this lines the two call sites up. * ec: let a config-free vif fall through to the layout sidecar A .vif that carries no ecShardConfig answers nothing about the layout, so it is no more informative than an absent one — but both trees treated its mere existence as the end of the search. Go went straight to the 10+4 legacy defaults without consulting the sidecar at all; Rust returned whatever ec_shard_config_from could make of a single directory. A 12+4 uniform volume with a legacy config-free vif therefore resolved as 10+4 legacy, and every read landed at the wrong shard offset. The sidecar lookup was also single-directory on both sides, while a split -dir/-dir.idx layout keeps .vif and .ecsum with the INDEX. Go's findBitrotSidecar has always taken both bases; the callers here passed only the data base, and the Rust bitrot resolver derived its path from the data base alone. Rust's layout resolver now takes a candidate directory list — data, index, then any siblings — and searches all of it, which also removes the early return that made the vif's presence decisive. load_vif_info_across_dirs reported `dir` even when load_vif_info had found the vif in `dir_idx`. Nothing reads that field today, so this changes no behaviour; it stops the next caller that resolves the rest of the volume's metadata against the answer from being sent to a disk holding none of it. Absence stays legal throughout: a volume with neither record is genuinely legacy. Present-but-unusable still fails the mount, now in the config-free-vif branch too. * ec: activate a delivered sidecar on every per-disk runtime A vid mounts as one EcVolume per disk, each with its own resolved protection state, but the post-delivery reload used the first-match lookup and so touched exactly one of them. The siblings kept reporting no protection until a remount — and since shard distribution deduplicates the metadata files onto the first target disk for a node, the runtime that got the .ecsum is not necessarily the one the lookup returns. Iterate every runtime instead, via a new FindAllEcVolumes and its Rust mut equivalent. Combined with each runtime now resolving its sidecar against its index directory as well as its data directory, a server sharing one -dir.idx across its disks activates all of them from the single delivered copy. The Rust volume server had no post-mount reload at all; it gets one here, matching Go. * ec: resolve the delivered sidecar across every EC metadata directory Reloading every per-disk runtime, added last round, did not by itself make the delivered manifest reachable. Startup mirroring copies .ecx/.ecj/.vif to every shard-bearing disk so each mounts self-contained, but deliberately not .ecsum, and a repair delivers exactly one copy. Each runtime was resolving against its own two directories, so every sibling of the disk that received the file kept reporting no protection however often it reloaded. Resolve one authoritative copy across every EC metadata directory instead of duplicating the file. Mirroring .ecsum would have to keep pace with a file that is rewritten as shards are repaired, and would not help the reported case at all: the delivery happens at runtime, and mirroring only runs at startup. The regression test pins both halves — a reload restricted to the volume's own directories still finds nothing, and the same reload given the server's metadata directories turns protection on. * ec: ask every directory before writing a TOFU baseline After a rebuild the opportunistic backfill asks whether this volume already has a checksum manifest, and answered from the data base alone. A split -dir/-dir.idx layout keeps the sidecar with the index, and a multi-disk server may keep it on a sibling, so an existing manifest read as absent. The consequence is worse than a missed read. On a false "no" the backfill writes a fresh sidecar at the data base from whatever the shards say right now — and the data base is the first candidate every resolver checks, so that TOFU baseline shadows the real manifest rather than sitting beside it. A shard that was silently corrupt gets blessed, and the record that would have caught it stops being consulted. FindBitrotSidecar exports the search the package already used internally, so the question is asked of the data base, the index base and the sibling disks — the same candidates the rebuild resolves its layout from. * ec: refuse a shard block size no encoder could have produced weed fix -ecx derived one from the raw shard extent, so a truncated or partially copied shard wrote a .vif that NewEcVolume then permanently refuses — the volume the tool was run to rescue could never mount again. An extent that is not a whole number of small blocks cannot have come from a uniform encode, so it is no longer offered as a candidate, and nothing unvalidated reaches the .vif. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: derive the .vif's dat size and block size from one measurement VolumeEcShardsGenerate stat'ed the .dat before the encode while WriteEcFiles stat'ed it again to size the blocks. A write landing between the two produced a .vif whose own two fields describe different files. WriteEcFiles now leaves both on the context, and fills a placeholder context in place so the caller can read them back. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: keep the source volume until every holder serves its shard layout The uniform layout rides in a .vif field older volume servers never knew: they discard it, mount the shards as legacy and return wrong bytes with nothing erroring, and the shard files are the same length either way so no other check notices. The upgrade order lived only in the release note. VolumeEcShardsInfo now reports the block size the holder actually serves, in both the Go and Rust servers, and the pre-delete verification refuses to drop the source unless every reachable holder echoes the one the shards were encoded with — while a rollback still exists. A server that predates the field answers 0, which is the negative answer. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: drop the rebuild's dead block-size parameters generateMissingEcFiles never reads largeBlockSize/smallBlockSize — Reed-Solomon reconstruction is layout-agnostic — so passing the legacy constants only advertised a layout the rebuild does not use. Also move UniformBlockSize's doc off ValidateBlockSize. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: warn about EC defaults only when the mount used them The "vif file not found, using defaults" warning fired even after the bitrot sidecar supplied a non-default layout, sending anyone triaging wrong bytes after the legacy layout the volume never mounted on. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: stat the distributed bitrot sidecar once The strict check re-stat'ed the file immediately before the stat that already gates inclusion, and a failed sidecar write now fails the encode outright, so the first could only fire on a deletion between the two lines. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * ec: say what the reconstruct budget actually bounds A shard's buffer stays in bufs until its interval reconstructs, which is after the read that filled it released its permit, so the semaphore bounds round trips in flight and not retained bytes. Peak memory is the intervals reconstructing at once times the shards each reaches times the interval size. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7 * test: let the fake volume server report its delivered EC layout The pre-delete verification now asks each holder which shard block layout it serves, and a fake that always answered "unset" looked exactly like a volume server too old to know the field. Distribution ships the .vif to every holder alongside its shards, so read the layout back out of it as a real holder does. Claude-Session: https://claude.ai/code/session_011FRRoNKBiGbH58rs2AQyA7
658 lines
25 KiB
Go
658 lines
25 KiB
Go
package erasure_coding
|
|
|
|
// EC bitrot detection — checksum sidecar.
|
|
//
|
|
// A per-volume sidecar file stores a CRC32C (Castagnoli) checksum for every
|
|
// fixed-size block of every EC shard, so a scrub (and the reconstruction path)
|
|
// can detect silent disk corruption in any shard — including cold parity shards
|
|
// that are never read during normal serving.
|
|
//
|
|
// The sidecar is OPTIONAL: an absent or generation-mismatched sidecar simply
|
|
// means "feature off" for that generation, so old binaries, JSON-only Rust
|
|
// nodes, and rollback deployments ignore it and degrade gracefully. See
|
|
// EC_BITROT_DETECTION_DESIGN.md.
|
|
//
|
|
// On-disk layout of <base>.ecsum (legacy/generation 0) and <base>.ecsum.v<N>:
|
|
//
|
|
// [ magic(4) | format_version(2) | payload_len(4) | payload_crc32c(4) ] [ proto payload ]
|
|
//
|
|
// The header's payload_crc32c lets a loader detect corruption of the sidecar
|
|
// itself BEFORE trusting any contents, so a rotted sidecar can never be
|
|
// mistaken for shard corruption.
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
"math/bits"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/volume_server_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/storage/needle"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
const (
|
|
// BitrotSidecarExt is the canonical extension for the checksum sidecar.
|
|
// Generation 0 (legacy/fresh encode) uses "<base>.ecsum"; vacuum generation
|
|
// N uses "<base>.ecsum.v<N>", mirroring the .vif/.ecx versioned convention.
|
|
BitrotSidecarExt = ".ecsum"
|
|
|
|
// DefaultBitrotBlockSize is the default checksum granularity (16 MiB). It is
|
|
// a power-of-two multiple of ErasureCodingSmallBlockSize (1 MiB) and keeps
|
|
// the sidecar tiny (~11 KB for a 30 GB volume) while localizing corruption
|
|
// to a 16 MiB region.
|
|
DefaultBitrotBlockSize = 16 * 1024 * 1024
|
|
|
|
// MaxBitrotBlockSize caps the block granularity so a loaded sidecar cannot
|
|
// force a huge scrub/verify scratch buffer. Power-of-two multiple of 1 MiB.
|
|
MaxBitrotBlockSize = 64 * 1024 * 1024
|
|
|
|
bitrotMagic uint32 = 0x45435355 // "ECSU"
|
|
bitrotFormatVersion uint16 = 1
|
|
bitrotHeaderSize = 14 // magic(4)+version(2)+payload_len(4)+payload_crc32c(4)
|
|
)
|
|
|
|
// Config knobs (wired to volume-server flags). Defaults: protection on for new
|
|
// encodes, 16 MiB block granularity.
|
|
var (
|
|
// BitrotProtectionEnabled controls whether encode/vacuum write a sidecar.
|
|
// When false, generations are produced unprotected (no sidecar), which is a
|
|
// legitimate "off" state. Detection on already-protected generations is
|
|
// unaffected.
|
|
BitrotProtectionEnabled = true
|
|
// BitrotBlockSize is the checksum block granularity used for new sidecars.
|
|
// Must remain a power-of-two multiple of 1 MiB.
|
|
BitrotBlockSize int64 = DefaultBitrotBlockSize
|
|
)
|
|
|
|
// BitrotStatus is the resolved protection state of an EC volume's active
|
|
// generation after loading and validating its sidecar.
|
|
type BitrotStatus int
|
|
|
|
const (
|
|
// BitrotOff: no sidecar, or a sidecar that does not describe the active
|
|
// generation. The generation is unprotected; this is NOT corruption.
|
|
BitrotOff BitrotStatus = iota
|
|
// BitrotOn: a complete, well-formed, generation-matching sidecar is loaded.
|
|
BitrotOn
|
|
// BitrotInvalid: a generation-matching sidecar that is malformed, incomplete,
|
|
// or self-integrity-failed. The generation is unprotected pending repair of
|
|
// the sidecar, and an integrity alarm should fire. The rebuild path treats
|
|
// this as fail-closed (see ec_encoder rebuild verify).
|
|
BitrotInvalid
|
|
)
|
|
|
|
func (s BitrotStatus) String() string {
|
|
switch s {
|
|
case BitrotOn:
|
|
return "on"
|
|
case BitrotInvalid:
|
|
return "invalid"
|
|
default:
|
|
return "off"
|
|
}
|
|
}
|
|
|
|
// BitrotSidecarPath returns the sidecar path for a base file name and EC
|
|
// generation. Generation 0 is the un-suffixed legacy path; generation N>0 is
|
|
// the versioned path, consistent with how .vif/.ecx are versioned by the 2PC
|
|
// switch.
|
|
func BitrotSidecarPath(baseFileName string, generation uint32) string {
|
|
if generation == 0 {
|
|
return baseFileName + BitrotSidecarExt
|
|
}
|
|
return fmt.Sprintf("%s%s.v%d", baseFileName, BitrotSidecarExt, generation)
|
|
}
|
|
|
|
// NewEncodeUUID returns a fresh random per-encode identity used to detect a
|
|
// stale sidecar left behind by an in-place re-encode.
|
|
func NewEncodeUUID() []byte {
|
|
b := make([]byte, 16)
|
|
if _, err := rand.Read(b); err != nil {
|
|
// rand.Read essentially never fails; fall back to a zero UUID rather
|
|
// than aborting an encode over entropy. Stale detection still has the
|
|
// other defenses (wholesale-mismatch cap + RS arbiter).
|
|
glog.Warningf("ec bitrot: failed to read encode uuid entropy: %v", err)
|
|
}
|
|
return b
|
|
}
|
|
|
|
// isPow2MultipleOf1MiB reports whether block_size is a power of two in
|
|
// [1 MiB, MaxBitrotBlockSize].
|
|
func isPow2MultipleOf1MiB(blockSize uint32) bool {
|
|
return blockSize >= (1<<20) && blockSize <= MaxBitrotBlockSize && bits.OnesCount32(blockSize) == 1
|
|
}
|
|
|
|
// shardChecksumBuilder accumulates the per-block CRC32C of a single shard's byte
|
|
// stream as it is written. It tolerates arbitrary chunk sizes that cross block
|
|
// boundaries, so it works for both the 256 KiB encode buffers and the 1 MiB
|
|
// rebuild buffers.
|
|
type shardChecksumBuilder struct {
|
|
blockSize int64
|
|
cur needle.CRC
|
|
curLen int64
|
|
total int64
|
|
blocks []uint32
|
|
}
|
|
|
|
func newShardChecksumBuilder(blockSize int64) *shardChecksumBuilder {
|
|
return &shardChecksumBuilder{blockSize: blockSize}
|
|
}
|
|
|
|
func (b *shardChecksumBuilder) write(p []byte) {
|
|
for len(p) > 0 {
|
|
room := b.blockSize - b.curLen
|
|
n := int64(len(p))
|
|
if n > room {
|
|
n = room
|
|
}
|
|
b.cur = b.cur.Update(p[:n])
|
|
b.curLen += n
|
|
b.total += n
|
|
p = p[n:]
|
|
if b.curLen == b.blockSize {
|
|
b.blocks = append(b.blocks, uint32(b.cur))
|
|
b.cur = 0
|
|
b.curLen = 0
|
|
}
|
|
}
|
|
}
|
|
|
|
// finalize flushes any partial last block and returns the covered size and the
|
|
// packed little-endian uint32 CRC array.
|
|
func (b *shardChecksumBuilder) finalize() (coveredSize int64, packed []byte) {
|
|
if b.curLen > 0 {
|
|
b.blocks = append(b.blocks, uint32(b.cur))
|
|
b.cur = 0
|
|
b.curLen = 0
|
|
}
|
|
return b.total, packUint32LE(b.blocks)
|
|
}
|
|
|
|
// buildProtectionFromBuilders assembles a generation-0 EcBitrotProtection from
|
|
// the per-shard checksum builders produced during an encode pass. Callers that
|
|
// produce a versioned generation (vacuum) set prot.Generation afterwards. A
|
|
// fresh encode_uuid is minted on every call so an in-place re-encode is
|
|
// distinguishable from a stale sidecar.
|
|
func buildProtectionFromBuilders(ctx *ECContext, builders []*shardChecksumBuilder, blockSize int64) *volume_server_pb.EcBitrotProtection {
|
|
shards := make([]*volume_server_pb.EcShardChecksums, 0, len(builders))
|
|
for id, b := range builders {
|
|
covered, packed := b.finalize()
|
|
shards = append(shards, &volume_server_pb.EcShardChecksums{
|
|
ShardId: uint32(id),
|
|
CoveredSize: covered,
|
|
BlockCrc32C: packed,
|
|
})
|
|
}
|
|
return &volume_server_pb.EcBitrotProtection{
|
|
Algorithm: volume_server_pb.ChecksumAlgorithm_CHECKSUM_CRC32C,
|
|
BlockSize: uint32(blockSize),
|
|
Generation: 0,
|
|
EcShardConfig: &volume_server_pb.EcShardConfig{
|
|
DataShards: uint32(ctx.DataShards),
|
|
ParityShards: uint32(ctx.ParityShards),
|
|
BlockSize: ctx.BlockSize,
|
|
},
|
|
Shards: shards,
|
|
EncodeUuid: NewEncodeUUID(),
|
|
}
|
|
}
|
|
|
|
func packUint32LE(vals []uint32) []byte {
|
|
out := make([]byte, len(vals)*4)
|
|
for i, v := range vals {
|
|
binary.LittleEndian.PutUint32(out[i*4:], v)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func unpackUint32LE(b []byte) []uint32 {
|
|
out := make([]uint32, len(b)/4)
|
|
for i := range out {
|
|
out[i] = binary.LittleEndian.Uint32(b[i*4:])
|
|
}
|
|
return out
|
|
}
|
|
|
|
// expectedBlockCount returns ceil(coveredSize/blockSize).
|
|
func expectedBlockCount(coveredSize, blockSize int64) int {
|
|
if blockSize <= 0 {
|
|
return 0
|
|
}
|
|
return int((coveredSize + blockSize - 1) / blockSize)
|
|
}
|
|
|
|
// SaveBitrotSidecar atomically writes prot to path, wrapped in the on-disk
|
|
// header with a CRC32C over the serialized payload.
|
|
func SaveBitrotSidecar(path string, prot *volume_server_pb.EcBitrotProtection) error {
|
|
payload, err := proto.Marshal(prot)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal bitrot sidecar: %w", err)
|
|
}
|
|
// The header records payload_len as a uint32 and the allocation below adds
|
|
// it to a constant. Bound the payload well under any overflow (a real
|
|
// manifest is a few KB) so neither the length field nor make() can wrap.
|
|
const maxBitrotPayloadSize = 1 << 30 // 1 GiB, vastly above any real sidecar
|
|
if len(payload) > maxBitrotPayloadSize {
|
|
return fmt.Errorf("bitrot sidecar payload too large: %d bytes", len(payload))
|
|
}
|
|
buf := make([]byte, bitrotHeaderSize+len(payload))
|
|
binary.BigEndian.PutUint32(buf[0:4], bitrotMagic)
|
|
binary.BigEndian.PutUint16(buf[4:6], bitrotFormatVersion)
|
|
binary.BigEndian.PutUint32(buf[6:10], uint32(len(payload)))
|
|
binary.BigEndian.PutUint32(buf[10:14], uint32(needle.NewCRC(payload)))
|
|
copy(buf[bitrotHeaderSize:], payload)
|
|
|
|
tmp := path + ".tmp"
|
|
if err := os.WriteFile(tmp, buf, 0644); err != nil {
|
|
return fmt.Errorf("write bitrot sidecar tmp %s: %w", tmp, err)
|
|
}
|
|
if err := os.Rename(tmp, path); err != nil {
|
|
os.Remove(tmp)
|
|
return fmt.Errorf("rename bitrot sidecar %s: %w", path, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// LoadBitrotSidecar reads and self-integrity-checks a sidecar file. It returns
|
|
// the parsed message, or an error if the file is missing, truncated, has a bad
|
|
// magic/version, or fails the payload CRC. A self-integrity failure is a
|
|
// sidecar-integrity problem (caller maps it to BitrotInvalid), never a shard
|
|
// corruption signal.
|
|
func LoadBitrotSidecar(path string) (*volume_server_pb.EcBitrotProtection, error) {
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(data) < bitrotHeaderSize {
|
|
return nil, fmt.Errorf("bitrot sidecar %s too short (%d bytes)", path, len(data))
|
|
}
|
|
if magic := binary.BigEndian.Uint32(data[0:4]); magic != bitrotMagic {
|
|
return nil, fmt.Errorf("bitrot sidecar %s bad magic %#x", path, magic)
|
|
}
|
|
if ver := binary.BigEndian.Uint16(data[4:6]); ver != bitrotFormatVersion {
|
|
return nil, fmt.Errorf("bitrot sidecar %s unsupported format version %d", path, ver)
|
|
}
|
|
payloadLen := binary.BigEndian.Uint32(data[6:10])
|
|
wantCRC := binary.BigEndian.Uint32(data[10:14])
|
|
payload := data[bitrotHeaderSize:]
|
|
if int(payloadLen) != len(payload) {
|
|
return nil, fmt.Errorf("bitrot sidecar %s length mismatch: header %d, actual %d", path, payloadLen, len(payload))
|
|
}
|
|
if got := uint32(needle.NewCRC(payload)); got != wantCRC {
|
|
return nil, fmt.Errorf("bitrot sidecar %s self-integrity CRC mismatch: header %#x, computed %#x", path, wantCRC, got)
|
|
}
|
|
prot := &volume_server_pb.EcBitrotProtection{}
|
|
if err := proto.Unmarshal(payload, prot); err != nil {
|
|
return nil, fmt.Errorf("unmarshal bitrot sidecar %s: %w", path, err)
|
|
}
|
|
return prot, nil
|
|
}
|
|
|
|
// ValidateBitrotManifest performs the disk-free manifest/syntax checks that
|
|
// every loader runs: supported algorithm, valid block size, exactly one entry
|
|
// per shard id in the active layout (no duplicates, no out-of-range ids),
|
|
// positive covered_size, and a packed-CRC count consistent with covered_size.
|
|
// It does NOT compare covered_size against on-disk shard lengths — that is a
|
|
// per-node physical check done only for locally-held shards.
|
|
func ValidateBitrotManifest(prot *volume_server_pb.EcBitrotProtection, dataShards, parityShards int) error {
|
|
if prot.Algorithm != volume_server_pb.ChecksumAlgorithm_CHECKSUM_CRC32C {
|
|
return fmt.Errorf("unsupported checksum algorithm %v", prot.Algorithm)
|
|
}
|
|
bs := int64(prot.BlockSize)
|
|
if !isPow2MultipleOf1MiB(prot.BlockSize) {
|
|
return fmt.Errorf("invalid block_size %d (must be a power-of-two multiple of 1 MiB, at most %d)", prot.BlockSize, MaxBitrotBlockSize)
|
|
}
|
|
total := dataShards + parityShards
|
|
if total <= 0 || total > MaxShardCount {
|
|
return fmt.Errorf("invalid active layout: data=%d parity=%d", dataShards, parityShards)
|
|
}
|
|
if len(prot.Shards) != total {
|
|
return fmt.Errorf("incomplete manifest: %d shard entries, expected %d", len(prot.Shards), total)
|
|
}
|
|
seen := make([]bool, MaxShardCount)
|
|
for _, s := range prot.Shards {
|
|
if s.ShardId >= uint32(total) {
|
|
return fmt.Errorf("shard id %d out of range [0,%d)", s.ShardId, total)
|
|
}
|
|
if seen[s.ShardId] {
|
|
return fmt.Errorf("duplicate shard id %d", s.ShardId)
|
|
}
|
|
seen[s.ShardId] = true
|
|
if s.CoveredSize <= 0 {
|
|
return fmt.Errorf("shard %d has non-positive covered_size %d", s.ShardId, s.CoveredSize)
|
|
}
|
|
wantCount := expectedBlockCount(s.CoveredSize, bs)
|
|
if len(s.BlockCrc32C) != wantCount*4 {
|
|
return fmt.Errorf("shard %d crc count mismatch: %d bytes, expected %d (covered_size=%d block_size=%d)",
|
|
s.ShardId, len(s.BlockCrc32C), wantCount*4, s.CoveredSize, prot.BlockSize)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// shardChecksums returns the EcShardChecksums entry for a shard id, or nil.
|
|
func shardChecksums(prot *volume_server_pb.EcBitrotProtection, shardId uint32) *volume_server_pb.EcShardChecksums {
|
|
for _, s := range prot.Shards {
|
|
if s.ShardId == shardId {
|
|
return s
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// verifyShardFileBlocks reads a shard file at path in block_size chunks and
|
|
// compares each block's CRC32C against the manifest entry. It returns the list
|
|
// of mismatching block indices (empty == clean) and a fatal error only for I/O
|
|
// problems or a covered_size/length mismatch (which is itself corruption of the
|
|
// shard). It does not interpret the result — the caller (scrub / rebuild)
|
|
// arbitrates shard-vs-sidecar via Reed-Solomon before acting.
|
|
func verifyShardFileBlocks(path string, entry *volume_server_pb.EcShardChecksums, blockSize int64) (mismatched []int, err error) {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
fi, err := f.Stat()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if fi.Size() != entry.CoveredSize {
|
|
// Length drift (truncation or unexpected trailing bytes) is shard
|
|
// corruption: report every block as mismatched so the caller treats the
|
|
// shard as bad.
|
|
want := unpackUint32LE(entry.BlockCrc32C)
|
|
all := make([]int, len(want))
|
|
for i := range all {
|
|
all[i] = i
|
|
}
|
|
return all, nil
|
|
}
|
|
want := unpackUint32LE(entry.BlockCrc32C)
|
|
buf := make([]byte, blockSize)
|
|
var offset int64
|
|
for i := 0; i < len(want); i++ {
|
|
toRead := blockSize
|
|
if rem := entry.CoveredSize - offset; rem < toRead {
|
|
toRead = rem
|
|
}
|
|
// os.File.ReadAt fills the slice or returns an error; a full read that
|
|
// lands exactly on EOF may still report io.EOF, which is not corruption.
|
|
// A short read (n < toRead) means the shard is truncated and the EOF is
|
|
// propagated as an error.
|
|
n, rerr := f.ReadAt(buf[:toRead], offset)
|
|
if rerr == io.EOF && int64(n) == toRead {
|
|
rerr = nil
|
|
}
|
|
if rerr != nil {
|
|
return nil, rerr
|
|
}
|
|
if uint32(needle.NewCRC(buf[:n])) != want[i] {
|
|
mismatched = append(mismatched, i)
|
|
}
|
|
offset += int64(n)
|
|
}
|
|
return mismatched, nil
|
|
}
|
|
|
|
// ComputeProtectionFromShards builds a complete EcBitrotProtection by reading
|
|
// the on-disk bytes of every shard for a generation — the trust-on-first-use
|
|
// backfill primitive for EC volumes that were encoded before this feature. It
|
|
// requires EVERY shard to be reachable (locally or in additionalDirs) and
|
|
// returns an error otherwise, so a partial sidecar is never produced. This is
|
|
// the per-holder building block a coordinator assembles across servers; on a
|
|
// server that holds (or can reach) all shards it yields a complete manifest
|
|
// directly. Caveat: it blesses whatever bytes exist now and cannot detect
|
|
// pre-baseline corruption.
|
|
func ComputeProtectionFromShards(baseFileName string, ctx *ECContext, generation uint32, additionalDirs []string) (*volume_server_pb.EcBitrotProtection, error) {
|
|
shards := make([]*volume_server_pb.EcShardChecksums, 0, ctx.Total())
|
|
for id := 0; id < ctx.Total(); id++ {
|
|
path := findShardFile(baseFileName, ctx.ToExt(id), additionalDirs)
|
|
if path == "" {
|
|
return nil, fmt.Errorf("bitrot backfill: shard %d missing for %s; refusing to write a partial sidecar", id, baseFileName)
|
|
}
|
|
covered, packed, err := computeShardFileCRCs(path, BitrotBlockSize)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("bitrot backfill: read shard %d (%s): %w", id, path, err)
|
|
}
|
|
shards = append(shards, &volume_server_pb.EcShardChecksums{
|
|
ShardId: uint32(id),
|
|
CoveredSize: covered,
|
|
BlockCrc32C: packed,
|
|
})
|
|
}
|
|
return &volume_server_pb.EcBitrotProtection{
|
|
Algorithm: volume_server_pb.ChecksumAlgorithm_CHECKSUM_CRC32C,
|
|
BlockSize: uint32(BitrotBlockSize),
|
|
Generation: generation,
|
|
EcShardConfig: &volume_server_pb.EcShardConfig{
|
|
DataShards: uint32(ctx.DataShards),
|
|
ParityShards: uint32(ctx.ParityShards),
|
|
BlockSize: ctx.BlockSize,
|
|
},
|
|
Shards: shards,
|
|
EncodeUuid: NewEncodeUUID(),
|
|
}, nil
|
|
}
|
|
|
|
// computeShardFileCRCs reads a shard file sequentially and returns its size and
|
|
// packed per-block CRC32C array.
|
|
func computeShardFileCRCs(path string, blockSize int64) (coveredSize int64, packed []byte, err error) {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
defer f.Close()
|
|
b := newShardChecksumBuilder(blockSize)
|
|
buf := make([]byte, blockSize)
|
|
for {
|
|
n, rerr := f.Read(buf)
|
|
if n > 0 {
|
|
b.write(buf[:n])
|
|
}
|
|
if rerr == io.EOF {
|
|
break
|
|
}
|
|
if rerr != nil {
|
|
return 0, nil, rerr
|
|
}
|
|
if n == 0 {
|
|
break
|
|
}
|
|
}
|
|
coveredSize, packed = b.finalize()
|
|
return coveredSize, packed, nil
|
|
}
|
|
|
|
// BitrotProtection returns the active-generation bitrot protection and its
|
|
// status. The returned message is read-only; callers must not mutate it.
|
|
func (ev *EcVolume) BitrotProtection() (*volume_server_pb.EcBitrotProtection, BitrotStatus) {
|
|
ev.bitrotLock.RLock()
|
|
defer ev.bitrotLock.RUnlock()
|
|
return ev.bitrot, ev.bitrotStatus
|
|
}
|
|
|
|
// ReloadBitrotSidecar re-resolves the checksum sidecar for a volume that is
|
|
// already mounted — a shard delivery can bring the manifest with it, and the
|
|
// receive path only writes the file, so without this the in-memory volume
|
|
// keeps the protection state it resolved at mount (off) until a remount.
|
|
func (ev *EcVolume) ReloadBitrotSidecar(additionalDirs ...string) {
|
|
if err := ev.loadActiveBitrotSidecar(additionalDirs...); err != nil {
|
|
glog.Warningf("ec volume %d: reload bitrot sidecar: %v", ev.VolumeId, err)
|
|
}
|
|
}
|
|
|
|
// loadActiveBitrotSidecar loads the generation-0 checksum sidecar into the
|
|
// volume. Best-effort: any failure leaves protection off/invalid without
|
|
// failing the mount. OSS only produces generation-0 (fresh-encode) sidecars.
|
|
func (ev *EcVolume) loadActiveBitrotSidecar(additionalDirs ...string) error {
|
|
return ev.loadBitrotForGeneration(0, additionalDirs...)
|
|
}
|
|
|
|
// loadBitrotForGeneration loads and validates the sidecar describing generation
|
|
// `generation`, setting ev.bitrot/ev.bitrotStatus. Called at mount. Absent or
|
|
// generation-mismatched => BitrotOff; self-integrity/manifest failure =>
|
|
// BitrotInvalid; usable => BitrotOn.
|
|
//
|
|
// A sidecar written FOR THIS generation that disagrees with the volume's shard
|
|
// geometry is different in kind from those: both files record the layout the
|
|
// generation was encoded with, so a disagreement means one of them is wrong and
|
|
// reads through the other would land at the wrong shard offsets. That returns
|
|
// an error and fails the mount instead of quietly dropping to unprotected
|
|
// reads.
|
|
func (ev *EcVolume) loadBitrotForGeneration(generation uint32, additionalDirs ...string) error {
|
|
ev.bitrotLock.Lock()
|
|
defer ev.bitrotLock.Unlock()
|
|
ev.bitrot = nil
|
|
ev.bitrotStatus = BitrotOff
|
|
|
|
if ev.ECContext == nil {
|
|
return nil
|
|
}
|
|
path := findBitrotSidecar(generation, ev.DataBaseFileName(), ev.IndexBaseFileName(), additionalDirs...)
|
|
if path == "" {
|
|
return nil
|
|
}
|
|
prot, err := LoadBitrotSidecar(path)
|
|
if err != nil {
|
|
glog.Warningf("ec volume %d: bitrot sidecar %s self-integrity failed: %v", ev.VolumeId, path, err)
|
|
ev.bitrotStatus = BitrotInvalid
|
|
return nil
|
|
}
|
|
if prot.Generation != generation {
|
|
return nil // not for this generation -> off, not corruption
|
|
}
|
|
if prot.EcShardConfig == nil {
|
|
return nil // records no geometry -> nothing to contradict
|
|
}
|
|
if int(prot.EcShardConfig.DataShards) != ev.ECContext.DataShards ||
|
|
int(prot.EcShardConfig.ParityShards) != ev.ECContext.ParityShards ||
|
|
prot.EcShardConfig.BlockSize != ev.ECContext.BlockSize {
|
|
return fmt.Errorf("ec volume %d generation %d: %s records layout %d+%d block %d but the volume is mounted as %d+%d block %d; refusing to serve one of the two layouts",
|
|
ev.VolumeId, generation, path,
|
|
prot.EcShardConfig.DataShards, prot.EcShardConfig.ParityShards, prot.EcShardConfig.BlockSize,
|
|
ev.ECContext.DataShards, ev.ECContext.ParityShards, ev.ECContext.BlockSize)
|
|
}
|
|
if err := ValidateBitrotManifest(prot, ev.ECContext.DataShards, ev.ECContext.ParityShards); err != nil {
|
|
glog.Warningf("ec volume %d: bitrot sidecar %s manifest invalid: %v", ev.VolumeId, path, err)
|
|
ev.bitrotStatus = BitrotInvalid
|
|
return nil
|
|
}
|
|
ev.bitrot = prot
|
|
ev.bitrotStatus = BitrotOn
|
|
glog.V(1).Infof("ec volume %d: loaded bitrot protection generation %d (%d shards, block_size %d)",
|
|
ev.VolumeId, generation, len(prot.Shards), prot.BlockSize)
|
|
return nil
|
|
}
|
|
|
|
// layoutFromSidecar resolves a volume's EC layout from its generation-0
|
|
// checksum sidecar, searching the data base and then the index base. A split
|
|
// -dir/-dir.idx layout keeps the sidecar with the index, so probing the data
|
|
// base alone reports "absent" for a volume that has the record right there —
|
|
// and absent is the one answer that selects the legacy 10+4 layout.
|
|
//
|
|
// The three answers stay distinct: (nil, false, nil) means genuinely absent
|
|
// and the caller may fall back to the defaults; a non-nil error means the
|
|
// record exists but cannot establish the layout, which must fail rather than
|
|
// default; otherwise the config is usable.
|
|
func layoutFromSidecar(dataBaseFileName, indexBaseFileName string) (cfg *volume_server_pb.EcShardConfig, found bool, err error) {
|
|
path := findBitrotSidecar(0, dataBaseFileName, indexBaseFileName)
|
|
if path == "" {
|
|
return nil, false, nil
|
|
}
|
|
cfg, err = EcShardConfigFromSidecarPath(path)
|
|
return cfg, true, err
|
|
}
|
|
|
|
// EcShardConfigFromSidecar reads the generation-0 bitrot sidecar's record of a
|
|
// volume's EC config. The three answers are distinct on purpose: absent means
|
|
// the volume may genuinely predate the sidecar, which is the only case a caller
|
|
// may answer with the legacy layout; present-but-unusable means the one
|
|
// surviving record of the geometry is corrupt, and guessing from there returns
|
|
// wrong bytes rather than no bytes.
|
|
func EcShardConfigFromSidecar(baseFileName string) (cfg *volume_server_pb.EcShardConfig, found bool, err error) {
|
|
path := BitrotSidecarPath(baseFileName, 0)
|
|
if _, statErr := os.Stat(path); statErr != nil {
|
|
if os.IsNotExist(statErr) {
|
|
return nil, false, nil
|
|
}
|
|
return nil, false, fmt.Errorf("stat %s: %w", path, statErr)
|
|
}
|
|
cfg, err = EcShardConfigFromSidecarPath(path)
|
|
return cfg, true, err
|
|
}
|
|
|
|
// EcShardConfigFromSidecarPath is EcShardConfigFromSidecar for a sidecar whose
|
|
// path the caller already resolved — the rebuild finds it across a multi-disk
|
|
// server's directories, not only next to the base name.
|
|
func EcShardConfigFromSidecarPath(path string) (*volume_server_pb.EcShardConfig, error) {
|
|
prot, loadErr := LoadBitrotSidecar(path)
|
|
if loadErr != nil {
|
|
return nil, fmt.Errorf("read %s: %w", path, loadErr)
|
|
}
|
|
// The un-suffixed sidecar describes generation 0 and nothing else; one
|
|
// stamped for another generation is not a record of these shards.
|
|
if prot.GetGeneration() != 0 {
|
|
return nil, fmt.Errorf("%s records generation %d, not generation 0", path, prot.GetGeneration())
|
|
}
|
|
cfg := prot.GetEcShardConfig()
|
|
if cfg == nil {
|
|
return nil, fmt.Errorf("%s records no EC config", path)
|
|
}
|
|
if !ValidEcShardCounts(cfg.GetDataShards(), cfg.GetParityShards()) {
|
|
return nil, fmt.Errorf("%s records invalid shard counts %d+%d",
|
|
path, cfg.GetDataShards(), cfg.GetParityShards())
|
|
}
|
|
if bsErr := ValidateBlockSize(cfg.GetBlockSize()); bsErr != nil {
|
|
return nil, fmt.Errorf("%s: %w", path, bsErr)
|
|
}
|
|
return cfg, nil
|
|
}
|
|
|
|
// RemoveBitrotSidecars removes the legacy <base>.ecsum and any versioned
|
|
// <base>.ecsum.v<N> sidecars for a base file name. Best-effort.
|
|
func RemoveBitrotSidecars(base string) {
|
|
os.Remove(base + BitrotSidecarExt)
|
|
if matches, _ := filepath.Glob(base + BitrotSidecarExt + ".v*"); matches != nil {
|
|
for _, m := range matches {
|
|
os.Remove(m)
|
|
}
|
|
}
|
|
}
|
|
|
|
// FindBitrotSidecar is findBitrotSidecar for callers outside this package. It
|
|
// answers "does this volume already have a manifest, and where" — a question
|
|
// that cannot be settled from one base name, because a split -dir/-dir.idx
|
|
// layout keeps the sidecar with the index and a multi-disk server may keep it
|
|
// on a sibling. Returns "" when no candidate exists.
|
|
func FindBitrotSidecar(generation uint32, dataBase, indexBase string, additionalDirs ...string) string {
|
|
return findBitrotSidecar(generation, dataBase, indexBase, additionalDirs...)
|
|
}
|
|
|
|
// findBitrotSidecar resolves the sidecar path for a generation, searching the
|
|
// data base, the index base, and any additional directories — mirroring how
|
|
// shard/.vif lookups handle split data/idx layouts and per-disk mirrors.
|
|
func findBitrotSidecar(generation uint32, dataBase, indexBase string, additionalDirs ...string) string {
|
|
candidates := []string{
|
|
BitrotSidecarPath(dataBase, generation),
|
|
BitrotSidecarPath(indexBase, generation),
|
|
}
|
|
base := filepath.Base(dataBase)
|
|
for _, dir := range additionalDirs {
|
|
candidates = append(candidates, BitrotSidecarPath(filepath.Join(dir, base), generation))
|
|
}
|
|
for _, c := range candidates {
|
|
if c == "" {
|
|
continue
|
|
}
|
|
if _, err := os.Stat(c); err == nil {
|
|
return c
|
|
}
|
|
}
|
|
return ""
|
|
}
|