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
|
package stats
import (
"encoding/binary"
"errors"
"os"
"path/filepath"
"google.golang.org/protobuf/proto"
)
func readStatisticsFile(imgDir *os.File, fileName string) (*StatsEntry, error) {
buf, err := os.ReadFile(filepath.Join(imgDir.Name(), fileName))
if err != nil {
return nil, err
}
if binary.LittleEndian.Uint32(buf[PrimaryMagicOffset:SecondaryMagicOffset]) != ImgServiceMagic {
return nil, errors.New("primary magic not found")
}
if binary.LittleEndian.Uint32(buf[SecondaryMagicOffset:SizeOffset]) != StatsMagic {
return nil, errors.New("secondary magic not found")
}
payloadSize := binary.LittleEndian.Uint32(buf[SizeOffset:PayloadOffset])
st := &StatsEntry{}
if err := proto.Unmarshal(buf[PayloadOffset:PayloadOffset+payloadSize], st); err != nil {
return nil, err
}
return st, nil
}
func CriuGetDumpStats(imgDir *os.File) (*DumpStatsEntry, error) {
st, err := readStatisticsFile(imgDir, StatsDump)
if err != nil {
return nil, err
}
return st.GetDump(), nil
}
func CriuGetRestoreStats(imgDir *os.File) (*RestoreStatsEntry, error) {
st, err := readStatisticsFile(imgDir, StatsRestore)
if err != nil {
return nil, err
}
return st.GetRestore(), nil
}
|
