File: reflink_linux.go

package info (click to toggle)
golang-github-karpeleslab-reflink 1.0.1-1.1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 120 kB
  • sloc: makefile: 13
file content (123 lines) | stat: -rw-r--r-- 2,277 bytes parent folder | download
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
//go:build linux

package reflink

import (
	"errors"
	"os"

	"golang.org/x/sys/unix"
)

// reflinkInternal performs the actual reflink action without worrying about fallback
func reflinkInternal(d, s *os.File) error {
	ss, err := s.SyscallConn()
	if err != nil {
		return err
	}
	sd, err := d.SyscallConn()
	if err != nil {
		return err
	}

	var err2, err3 error

	err = sd.Control(func(dfd uintptr) {
		err2 = ss.Control(func(sfd uintptr) {
			// int ioctl(int dest_fd, FICLONE, int src_fd);
			err3 = unix.IoctlFileClone(int(dfd), int(sfd))
		})
	})

	if err != nil {
		// sd.Control failed
		return err
	}
	if err2 != nil {
		// ss.Control failed
		return err2
	}

	if err3 != nil && errors.Is(err3, unix.ENOTSUP) {
		return ErrReflinkFailed
	}

	// err3 is ioctl() response
	return err3
}

func reflinkRangeInternal(dst, src *os.File, dstOffset, srcOffset, n int64) error {
	ss, err := src.SyscallConn()
	if err != nil {
		return err
	}
	sd, err := dst.SyscallConn()
	if err != nil {
		return err
	}

	var err2, err3 error

	err = sd.Control(func(dfd uintptr) {
		err2 = ss.Control(func(sfd uintptr) {
			req := &unix.FileCloneRange{
				Src_fd:      int64(sfd),
				Src_offset:  uint64(srcOffset),
				Src_length:  uint64(n),
				Dest_offset: uint64(dstOffset),
			}

			// int ioctl(int dest_fd, FICLONE, int src_fd);
			err3 = unix.IoctlFileCloneRange(int(dfd), req)
		})
	})

	if err != nil {
		// sd.Control failed
		return err
	}
	if err2 != nil {
		// ss.Control failed
		return err2
	}
	if err3 != nil && errors.Is(err3, unix.ENOTSUP) {
		return ErrReflinkFailed
	}

	// err3 is ioctl() response
	return err3
}

func copyFileRange(dst, src *os.File, dstOffset, srcOffset, n int64) (int64, error) {
	ss, err := src.SyscallConn()
	if err != nil {
		return 0, err
	}
	sd, err := dst.SyscallConn()
	if err != nil {
		return 0, err
	}

	var resN int
	var err2, err3 error

	err = sd.Control(func(dfd uintptr) {
		err2 = ss.Control(func(sfd uintptr) {
			// call syscall
			resN, err3 = unix.CopyFileRange(int(sfd), &srcOffset, int(dfd), &dstOffset, int(n), 0)
		})
	})

	if err != nil {
		// sd.Control failed
		return int64(resN), err
	}
	if err2 != nil {
		// ss.Control failed
		return int64(resN), err2
	}

	// err3 is ioctl() response
	return int64(resN), err3

}