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
147
148
149
150
151
|
package main
import (
"context"
"fmt"
"io"
"net"
"net/http"
"net/url"
"os"
"slices"
"github.com/gorilla/mux"
"github.com/pkg/sftp"
"github.com/lxc/incus/v6/internal/server/cluster"
"github.com/lxc/incus/v6/internal/server/operations"
"github.com/lxc/incus/v6/internal/server/project"
"github.com/lxc/incus/v6/internal/server/request"
"github.com/lxc/incus/v6/internal/server/response"
storagePools "github.com/lxc/incus/v6/internal/server/storage"
storageDrivers "github.com/lxc/incus/v6/internal/server/storage/drivers"
"github.com/lxc/incus/v6/shared/api"
"github.com/lxc/incus/v6/shared/logger"
)
// swagger:operation GET /1.0/storage-pools/{poolName}/volumes/{type}/{volumeName}/sftp storage storage_pool_volume_type_sftp_get
//
// Get the storage volume SFTP connection
//
// Upgrades the request to an SFTP connection of the storage volume's filesystem.
//
// ---
// produces:
// - application/json
// - application/octet-stream
// responses:
// "101":
// description: Switching protocols to SFTP
// "400":
// $ref: "#/responses/BadRequest"
// "404":
// $ref: "#/responses/NotFound"
// "403":
// $ref: "#/responses/Forbidden"
// "500":
// $ref: "#/responses/InternalServerError"
func storagePoolVolumeTypeSFTPHandler(d *Daemon, r *http.Request) response.Response {
s := d.State()
projectName := request.ProjectParam(r)
if r.Header.Get("Upgrade") != "sftp" {
return response.SmartError(api.StatusErrorf(http.StatusBadRequest, "Missing or invalid upgrade header"))
}
// Get the volume details.
volumeTypeName, err := url.PathUnescape(mux.Vars(r)["type"])
if err != nil {
return response.SmartError(err)
}
volumeName, err := url.PathUnescape(mux.Vars(r)["volumeName"])
if err != nil {
return response.SmartError(err)
}
poolName, err := url.PathUnescape(mux.Vars(r)["poolName"])
if err != nil {
return response.SmartError(err)
}
// Convert the volume type name to our internal integer representation.
volumeType, err := storagePools.VolumeTypeNameToDBType(volumeTypeName)
if err != nil {
return response.BadRequest(err)
}
// Check that the storage volume type is valid.
if !slices.Contains(supportedVolumeTypes, volumeType) {
return response.BadRequest(fmt.Errorf("Invalid storage volume type %q", volumeTypeName))
}
volumeProjectName, err := project.StorageVolumeProject(s.DB.Cluster, projectName, volumeType)
if err != nil {
return response.SmartError(err)
}
// Redirect to correct server if needed.
var conn net.Conn
// Forward the request if the instance is remote.
client, err := cluster.ConnectIfVolumeIsRemote(s, poolName, volumeProjectName, volumeName, volumeType, s.Endpoints.NetworkCert(), s.ServerCert(), r)
if err != nil {
return response.SmartError(err)
}
if client != nil {
conn, err = client.GetStoragePoolVolumeFileSFTPConn(poolName, volumeTypeName, volumeName)
if err != nil {
return response.SmartError(err)
}
} else {
pool, err := storagePools.LoadByName(s, poolName)
if err != nil {
return response.SmartError(err)
}
diskVolName := project.StorageVolume(volumeProjectName, volumeName)
vol := pool.GetVolume(storageDrivers.VolumeTypeCustom, storageDrivers.ContentTypeFS, diskVolName, nil)
serverConn, clientConn := net.Pipe()
conn = clientConn
go func() {
err := serveVolumeSFTP(context.Background(), vol, serverConn, nil)
if err != nil {
logger.Warn("volume‑SFTP server exited", logger.Ctx{
"pool": poolName,
"vol": volumeName,
"err": err,
})
}
_ = serverConn.Close()
}()
}
return response.SFTPResponse(r, conn)
}
// Serve the mounted volume over the pkg/sftp built‑in server.
func serveVolumeSFTP(ctx context.Context, vol storageDrivers.Volume, conn net.Conn, op *operations.Operation) error {
return vol.MountTask(func(mountPath string, _ *operations.Operation) error {
srv, err := sftp.NewServer(
conn,
sftp.WithServerWorkingDirectory(mountPath),
sftp.WithDebug(os.Stderr),
)
if err != nil {
return err
}
err = srv.Serve()
if err != nil && err != io.EOF {
return err
}
return nil
}, nil)
}
|
