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
|
// Copyright 2021 Canonical Ltd.
// Licensed under the LGPLv3 with static-linking exception.
// See LICENCE file for details.
package linux
import (
"errors"
"fmt"
"math"
"os"
"syscall"
"golang.org/x/sys/unix"
efi "github.com/canonical/go-efilib"
internal_unix "github.com/canonical/go-efilib/internal/unix"
)
func getSectorSize(f *os.File) (int64, error) {
fi, err := f.Stat()
if err != nil {
return 0, err
}
if fi.Mode().IsRegular() {
return 512, nil
}
if fi.Mode()&os.ModeDevice == 0 {
return 0, errors.New("not a regular file or device")
}
sz, err := unix.IoctlGetInt(int(f.Fd()), unix.BLKSSZGET)
if err != nil {
return 0, err
}
return int64(sz), nil
}
func getDeviceSize(f *os.File) (int64, error) {
fi, err := f.Stat()
if err != nil {
return 0, err
}
if fi.Mode().IsRegular() {
return fi.Size(), nil
}
if fi.Mode()&os.ModeDevice == 0 {
return 0, errors.New("not a regular file or device")
}
sz, err := internal_unix.IoctlGetUint64(int(f.Fd()), unix.BLKGETSIZE64)
switch {
case err == syscall.ENOTTY:
n, err := internal_unix.IoctlGetUint(int(f.Fd()), unix.BLKGETSIZE)
if err != nil {
return 0, err
}
if int64(n) > int64(math.MaxInt64>>9) {
return 0, errors.New("overflow")
}
return int64(n << 9), nil
case err != nil:
return 0, err
case sz > math.MaxInt64:
return 0, errors.New("overflow")
default:
return int64(sz), nil
}
}
// NewHardDriveDevicePathNodeFromDevice constructs a HardDriveDevicePathNode for the
// specified partition on the device or file at the supplied path.
func NewHardDriveDevicePathNodeFromDevice(dev string, part int) (*efi.HardDriveDevicePathNode, error) {
f, err := osOpen(dev)
if err != nil {
return nil, err
}
defer f.Close()
sz, err := getDeviceSize(f)
if err != nil {
return nil, fmt.Errorf("cannot determine device size: %w", err)
}
ssz, err := getSectorSize(f)
if err != nil {
return nil, fmt.Errorf("cannot determine logical sector size: %w", err)
}
return efi.NewHardDriveDevicePathNodeFromDevice(f, sz, ssz, part)
}
|
