forked from keybase/client
-
Notifications
You must be signed in to change notification settings - Fork 0
/
block_getter.go
58 lines (49 loc) · 1.88 KB
/
block_getter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// Copyright 2016 Keybase Inc. All rights reserved.
// Use of this source code is governed by a BSD
// license that can be found in the LICENSE file.
package libkbfs
import (
"fmt"
"github.com/keybase/client/go/kbfs/kbfsblock"
"github.com/keybase/client/go/kbfs/kbfscrypto"
"github.com/keybase/client/go/kbfs/libkey"
"golang.org/x/net/context"
)
// blockGetter provides the API for the block retrieval worker to obtain blocks.
type blockGetter interface {
getBlock(
context.Context, libkey.KeyMetadata, BlockPointer, Block,
DiskBlockCacheType) error
assembleBlock(context.Context, libkey.KeyMetadata, BlockPointer, Block, []byte,
kbfscrypto.BlockCryptKeyServerHalf) error
}
// realBlockGetter obtains real blocks using the APIs available in Config.
type realBlockGetter struct {
config blockOpsConfig
}
// getBlock implements the interface for realBlockGetter.
func (bg *realBlockGetter) getBlock(
ctx context.Context, kmd libkey.KeyMetadata, blockPtr BlockPointer,
block Block, cacheType DiskBlockCacheType) error {
bserv := bg.config.BlockServer()
buf, blockServerHalf, err := bserv.Get(
ctx, kmd.TlfID(), blockPtr.ID, blockPtr.Context, cacheType)
if err != nil {
// Temporary code to track down bad block
// requests. Remove when not needed anymore.
if _, ok := err.(kbfsblock.ServerErrorBadRequest); ok {
panic(fmt.Sprintf("Bad BServer request detected: err=%s, blockPtr=%s",
err, blockPtr))
}
return err
}
return assembleBlock(
ctx, bg.config.keyGetter(), bg.config.Codec(), bg.config.cryptoPure(),
kmd, blockPtr, block, buf, blockServerHalf)
}
func (bg *realBlockGetter) assembleBlock(ctx context.Context,
kmd libkey.KeyMetadata, ptr BlockPointer, block Block, buf []byte,
serverHalf kbfscrypto.BlockCryptKeyServerHalf) error {
return assembleBlock(ctx, bg.config.keyGetter(), bg.config.Codec(),
bg.config.cryptoPure(), kmd, ptr, block, buf, serverHalf)
}