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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016-2020 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package main
import (
"fmt"
"time"
"github.com/snapcore/snapd/asserts"
"github.com/snapcore/snapd/osutil/disks"
"github.com/snapcore/snapd/secboot"
)
var (
Parser = parser
DoSystemdMount = doSystemdMountImpl
)
type SystemdMountOptions = systemdMountOptions
type RecoverDegradedState = recoverDegradedState
type PartitionState = partitionState
func (r *RecoverDegradedState) Degraded(isEncrypted bool) bool {
m := recoverModeStateMachine{
isEncryptedDev: isEncrypted,
degradedState: r,
}
return m.degraded()
}
func MockTimeNow(f func() time.Time) (restore func()) {
old := timeNow
timeNow = f
return func() {
timeNow = old
}
}
func MockOsutilIsMounted(f func(string) (bool, error)) (restore func()) {
old := osutilIsMounted
osutilIsMounted = f
return func() {
osutilIsMounted = old
}
}
func MockSystemdMount(f func(_, _ string, opts *SystemdMountOptions) error) (restore func()) {
old := doSystemdMount
doSystemdMount = f
return func() {
doSystemdMount = old
}
}
func MockTriggerwatchWait(f func(_ time.Duration) error) (restore func()) {
oldTriggerwatchWait := triggerwatchWait
triggerwatchWait = f
return func() {
triggerwatchWait = oldTriggerwatchWait
}
}
var DefaultTimeout = defaultTimeout
func MockDefaultMarkerFile(p string) (restore func()) {
old := defaultMarkerFile
defaultMarkerFile = p
return func() {
defaultMarkerFile = old
}
}
func MockSecbootUnlockVolumeUsingSealedKeyIfEncrypted(f func(disk disks.Disk, name string, sealedEncryptionKeyFile string, opts *secboot.UnlockVolumeUsingSealedKeyOptions) (secboot.UnlockResult, error)) (restore func()) {
old := secbootUnlockVolumeUsingSealedKeyIfEncrypted
secbootUnlockVolumeUsingSealedKeyIfEncrypted = f
return func() {
secbootUnlockVolumeUsingSealedKeyIfEncrypted = old
}
}
func MockSecbootUnlockEncryptedVolumeUsingKey(f func(disk disks.Disk, name string, key []byte) (secboot.UnlockResult, error)) (restore func()) {
old := secbootUnlockEncryptedVolumeUsingKey
secbootUnlockEncryptedVolumeUsingKey = f
return func() {
secbootUnlockEncryptedVolumeUsingKey = old
}
}
func MockSecbootMeasureSnapSystemEpochWhenPossible(f func() error) (restore func()) {
old := secbootMeasureSnapSystemEpochWhenPossible
secbootMeasureSnapSystemEpochWhenPossible = f
return func() {
secbootMeasureSnapSystemEpochWhenPossible = old
}
}
func MockSecbootMeasureSnapModelWhenPossible(f func(findModel func() (*asserts.Model, error)) error) (restore func()) {
old := secbootMeasureSnapModelWhenPossible
secbootMeasureSnapModelWhenPossible = f
return func() {
secbootMeasureSnapModelWhenPossible = old
}
}
func MockSecbootLockSealedKeys(f func() error) (restore func()) {
old := secbootLockSealedKeys
secbootLockSealedKeys = f
return func() {
secbootLockSealedKeys = old
}
}
func MockPartitionUUIDForBootedKernelDisk(uuid string) (restore func()) {
old := bootFindPartitionUUIDForBootedKernelDisk
bootFindPartitionUUIDForBootedKernelDisk = func() (string, error) {
if uuid == "" {
// mock error
return "", fmt.Errorf("mocked error")
}
return uuid, nil
}
return func() {
bootFindPartitionUUIDForBootedKernelDisk = old
}
}
|
