]>
Commit | Line | Data |
---|---|---|
1 | package PVE::Storage; | |
2 | ||
3 | use strict; | |
4 | use warnings; | |
5 | use Data::Dumper; | |
6 | ||
7 | use POSIX; | |
8 | use IO::Select; | |
9 | use IO::File; | |
10 | use IO::Socket::IP; | |
11 | use File::Basename; | |
12 | use File::Path; | |
13 | use Cwd 'abs_path'; | |
14 | use Socket; | |
15 | ||
16 | use PVE::Tools qw(run_command file_read_firstline dir_glob_foreach $IPV6RE); | |
17 | use PVE::Cluster qw(cfs_read_file cfs_write_file cfs_lock_file); | |
18 | use PVE::DataCenterConfig; | |
19 | use PVE::Exception qw(raise_param_exc); | |
20 | use PVE::JSONSchema; | |
21 | use PVE::INotify; | |
22 | use PVE::RPCEnvironment; | |
23 | use PVE::SSHInfo; | |
24 | ||
25 | use PVE::Storage::Plugin; | |
26 | use PVE::Storage::DirPlugin; | |
27 | use PVE::Storage::LVMPlugin; | |
28 | use PVE::Storage::LvmThinPlugin; | |
29 | use PVE::Storage::NFSPlugin; | |
30 | use PVE::Storage::CIFSPlugin; | |
31 | use PVE::Storage::ISCSIPlugin; | |
32 | use PVE::Storage::RBDPlugin; | |
33 | use PVE::Storage::CephFSPlugin; | |
34 | use PVE::Storage::ISCSIDirectPlugin; | |
35 | use PVE::Storage::GlusterfsPlugin; | |
36 | use PVE::Storage::ZFSPoolPlugin; | |
37 | use PVE::Storage::ZFSPlugin; | |
38 | use PVE::Storage::DRBDPlugin; | |
39 | use PVE::Storage::PBSPlugin; | |
40 | ||
41 | # Storage API version. Icrement it on changes in storage API interface. | |
42 | use constant APIVER => 5; | |
43 | # Age is the number of versions we're backward compatible with. | |
44 | # This is like having 'current=APIVER' and age='APIAGE' in libtool, | |
45 | # see https://www.gnu.org/software/libtool/manual/html_node/Libtool-versioning.html | |
46 | use constant APIAGE => 4; | |
47 | ||
48 | # load standard plugins | |
49 | PVE::Storage::DirPlugin->register(); | |
50 | PVE::Storage::LVMPlugin->register(); | |
51 | PVE::Storage::LvmThinPlugin->register(); | |
52 | PVE::Storage::NFSPlugin->register(); | |
53 | PVE::Storage::CIFSPlugin->register(); | |
54 | PVE::Storage::ISCSIPlugin->register(); | |
55 | PVE::Storage::RBDPlugin->register(); | |
56 | PVE::Storage::CephFSPlugin->register(); | |
57 | PVE::Storage::ISCSIDirectPlugin->register(); | |
58 | PVE::Storage::GlusterfsPlugin->register(); | |
59 | PVE::Storage::ZFSPoolPlugin->register(); | |
60 | PVE::Storage::ZFSPlugin->register(); | |
61 | PVE::Storage::DRBDPlugin->register(); | |
62 | PVE::Storage::PBSPlugin->register(); | |
63 | ||
64 | # load third-party plugins | |
65 | if ( -d '/usr/share/perl5/PVE/Storage/Custom' ) { | |
66 | dir_glob_foreach('/usr/share/perl5/PVE/Storage/Custom', '.*\.pm$', sub { | |
67 | my ($file) = @_; | |
68 | my $modname = 'PVE::Storage::Custom::' . $file; | |
69 | $modname =~ s!\.pm$!!; | |
70 | $file = 'PVE/Storage/Custom/' . $file; | |
71 | ||
72 | eval { | |
73 | require $file; | |
74 | ||
75 | # Check perl interface: | |
76 | die "not derived from PVE::Storage::Plugin\n" | |
77 | if !$modname->isa('PVE::Storage::Plugin'); | |
78 | die "does not provide an api() method\n" | |
79 | if !$modname->can('api'); | |
80 | # Check storage API version and that file is really storage plugin. | |
81 | my $version = $modname->api(); | |
82 | die "implements an API version newer than current ($version > " . APIVER . ")\n" | |
83 | if $version > APIVER; | |
84 | my $min_version = (APIVER - APIAGE); | |
85 | die "API version too old, please update the plugin ($version < $min_version)\n" | |
86 | if $version < $min_version; | |
87 | import $file; | |
88 | $modname->register(); | |
89 | ||
90 | # If we got this far and the API version is not the same, make some | |
91 | # noise: | |
92 | warn "Plugin \"$modname\" is implementing an older storage API, an upgrade is recommended\n" | |
93 | if $version != APIVER; | |
94 | }; | |
95 | if ($@) { | |
96 | warn "Error loading storage plugin \"$modname\": $@"; | |
97 | } | |
98 | }); | |
99 | } | |
100 | ||
101 | # initialize all plugins | |
102 | PVE::Storage::Plugin->init(); | |
103 | ||
104 | my $UDEVADM = '/sbin/udevadm'; | |
105 | ||
106 | our $iso_extension_re = qr/\.(?:iso|img)/i; | |
107 | ||
108 | # PVE::Storage utility functions | |
109 | ||
110 | sub config { | |
111 | return cfs_read_file("storage.cfg"); | |
112 | } | |
113 | ||
114 | sub write_config { | |
115 | my ($cfg) = @_; | |
116 | ||
117 | cfs_write_file('storage.cfg', $cfg); | |
118 | } | |
119 | ||
120 | sub lock_storage_config { | |
121 | my ($code, $errmsg) = @_; | |
122 | ||
123 | cfs_lock_file("storage.cfg", undef, $code); | |
124 | my $err = $@; | |
125 | if ($err) { | |
126 | $errmsg ? die "$errmsg: $err" : die $err; | |
127 | } | |
128 | } | |
129 | ||
130 | sub storage_config { | |
131 | my ($cfg, $storeid, $noerr) = @_; | |
132 | ||
133 | die "no storage ID specified\n" if !$storeid; | |
134 | ||
135 | my $scfg = $cfg->{ids}->{$storeid}; | |
136 | ||
137 | die "storage '$storeid' does not exist\n" if (!$noerr && !$scfg); | |
138 | ||
139 | return $scfg; | |
140 | } | |
141 | ||
142 | sub storage_check_node { | |
143 | my ($cfg, $storeid, $node, $noerr) = @_; | |
144 | ||
145 | my $scfg = storage_config($cfg, $storeid); | |
146 | ||
147 | if ($scfg->{nodes}) { | |
148 | $node = PVE::INotify::nodename() if !$node || ($node eq 'localhost'); | |
149 | if (!$scfg->{nodes}->{$node}) { | |
150 | die "storage '$storeid' is not available on node '$node'\n" if !$noerr; | |
151 | return undef; | |
152 | } | |
153 | } | |
154 | ||
155 | return $scfg; | |
156 | } | |
157 | ||
158 | sub storage_check_enabled { | |
159 | my ($cfg, $storeid, $node, $noerr) = @_; | |
160 | ||
161 | my $scfg = storage_config($cfg, $storeid); | |
162 | ||
163 | if ($scfg->{disable}) { | |
164 | die "storage '$storeid' is disabled\n" if !$noerr; | |
165 | return undef; | |
166 | } | |
167 | ||
168 | return storage_check_node($cfg, $storeid, $node, $noerr); | |
169 | } | |
170 | ||
171 | # storage_can_replicate: | |
172 | # return true if storage supports replication | |
173 | # (volumes alocated with vdisk_alloc() has replication feature) | |
174 | sub storage_can_replicate { | |
175 | my ($cfg, $storeid, $format) = @_; | |
176 | ||
177 | my $scfg = storage_config($cfg, $storeid); | |
178 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
179 | return $plugin->storage_can_replicate($scfg, $storeid, $format); | |
180 | } | |
181 | ||
182 | sub storage_ids { | |
183 | my ($cfg) = @_; | |
184 | ||
185 | return keys %{$cfg->{ids}}; | |
186 | } | |
187 | ||
188 | sub file_size_info { | |
189 | my ($filename, $timeout) = @_; | |
190 | ||
191 | return PVE::Storage::Plugin::file_size_info($filename, $timeout); | |
192 | } | |
193 | ||
194 | sub volume_size_info { | |
195 | my ($cfg, $volid, $timeout) = @_; | |
196 | ||
197 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
198 | if ($storeid) { | |
199 | my $scfg = storage_config($cfg, $storeid); | |
200 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
201 | return $plugin->volume_size_info($scfg, $storeid, $volname, $timeout); | |
202 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
203 | return file_size_info($volid, $timeout); | |
204 | } else { | |
205 | return 0; | |
206 | } | |
207 | } | |
208 | ||
209 | sub volume_resize { | |
210 | my ($cfg, $volid, $size, $running) = @_; | |
211 | ||
212 | my $padding = (1024 - $size % 1024) % 1024; | |
213 | $size = $size + $padding; | |
214 | ||
215 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
216 | if ($storeid) { | |
217 | my $scfg = storage_config($cfg, $storeid); | |
218 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
219 | return $plugin->volume_resize($scfg, $storeid, $volname, $size, $running); | |
220 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
221 | die "resize file/device '$volid' is not possible\n"; | |
222 | } else { | |
223 | die "unable to parse volume ID '$volid'\n"; | |
224 | } | |
225 | } | |
226 | ||
227 | sub volume_rollback_is_possible { | |
228 | my ($cfg, $volid, $snap) = @_; | |
229 | ||
230 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
231 | if ($storeid) { | |
232 | my $scfg = storage_config($cfg, $storeid); | |
233 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
234 | return $plugin->volume_rollback_is_possible($scfg, $storeid, $volname, $snap); | |
235 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
236 | die "snapshot rollback file/device '$volid' is not possible\n"; | |
237 | } else { | |
238 | die "unable to parse volume ID '$volid'\n"; | |
239 | } | |
240 | } | |
241 | ||
242 | sub volume_snapshot { | |
243 | my ($cfg, $volid, $snap) = @_; | |
244 | ||
245 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
246 | if ($storeid) { | |
247 | my $scfg = storage_config($cfg, $storeid); | |
248 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
249 | return $plugin->volume_snapshot($scfg, $storeid, $volname, $snap); | |
250 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
251 | die "snapshot file/device '$volid' is not possible\n"; | |
252 | } else { | |
253 | die "unable to parse volume ID '$volid'\n"; | |
254 | } | |
255 | } | |
256 | ||
257 | sub volume_snapshot_rollback { | |
258 | my ($cfg, $volid, $snap) = @_; | |
259 | ||
260 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
261 | if ($storeid) { | |
262 | my $scfg = storage_config($cfg, $storeid); | |
263 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
264 | $plugin->volume_rollback_is_possible($scfg, $storeid, $volname, $snap); | |
265 | return $plugin->volume_snapshot_rollback($scfg, $storeid, $volname, $snap); | |
266 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
267 | die "snapshot rollback file/device '$volid' is not possible\n"; | |
268 | } else { | |
269 | die "unable to parse volume ID '$volid'\n"; | |
270 | } | |
271 | } | |
272 | ||
273 | sub volume_snapshot_delete { | |
274 | my ($cfg, $volid, $snap, $running) = @_; | |
275 | ||
276 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
277 | if ($storeid) { | |
278 | my $scfg = storage_config($cfg, $storeid); | |
279 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
280 | return $plugin->volume_snapshot_delete($scfg, $storeid, $volname, $snap, $running); | |
281 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
282 | die "snapshot delete file/device '$volid' is not possible\n"; | |
283 | } else { | |
284 | die "unable to parse volume ID '$volid'\n"; | |
285 | } | |
286 | } | |
287 | ||
288 | # check if a volume or snapshot supports a given feature | |
289 | # $feature - one of: | |
290 | # clone - linked clone is possible | |
291 | # copy - full clone is possible | |
292 | # replicate - replication is possible | |
293 | # snapshot - taking a snapshot is possible | |
294 | # sparseinit - volume is sparsely initialized | |
295 | # template - conversion to base image is possible | |
296 | # $snap - check if the feature is supported for a given snapshot | |
297 | # $running - if the guest owning the volume is running | |
298 | # $opts - hash with further options: | |
299 | # valid_target_formats - list of formats for the target of a copy/clone | |
300 | # operation that the caller could work with. The | |
301 | # format of $volid is always considered valid and if | |
302 | # no list is specified, all formats are considered valid. | |
303 | sub volume_has_feature { | |
304 | my ($cfg, $feature, $volid, $snap, $running, $opts) = @_; | |
305 | ||
306 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
307 | if ($storeid) { | |
308 | my $scfg = storage_config($cfg, $storeid); | |
309 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
310 | return $plugin->volume_has_feature($scfg, $feature, $storeid, $volname, $snap, $running, $opts); | |
311 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
312 | return undef; | |
313 | } else { | |
314 | return undef; | |
315 | } | |
316 | } | |
317 | ||
318 | sub volume_snapshot_list { | |
319 | my ($cfg, $volid) = @_; | |
320 | ||
321 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
322 | if ($storeid) { | |
323 | my $scfg = storage_config($cfg, $storeid); | |
324 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
325 | return $plugin->volume_snapshot_list($scfg, $storeid, $volname); | |
326 | } elsif ($volid =~ m|^(/.+)$| && -e $volid) { | |
327 | die "send file/device '$volid' is not possible\n"; | |
328 | } else { | |
329 | die "unable to parse volume ID '$volid'\n"; | |
330 | } | |
331 | # return an empty array if dataset does not exist. | |
332 | } | |
333 | ||
334 | sub get_image_dir { | |
335 | my ($cfg, $storeid, $vmid) = @_; | |
336 | ||
337 | my $scfg = storage_config($cfg, $storeid); | |
338 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
339 | ||
340 | my $path = $plugin->get_subdir($scfg, 'images'); | |
341 | ||
342 | return $vmid ? "$path/$vmid" : $path; | |
343 | } | |
344 | ||
345 | sub get_private_dir { | |
346 | my ($cfg, $storeid, $vmid) = @_; | |
347 | ||
348 | my $scfg = storage_config($cfg, $storeid); | |
349 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
350 | ||
351 | my $path = $plugin->get_subdir($scfg, 'rootdir'); | |
352 | ||
353 | return $vmid ? "$path/$vmid" : $path; | |
354 | } | |
355 | ||
356 | sub get_iso_dir { | |
357 | my ($cfg, $storeid) = @_; | |
358 | ||
359 | my $scfg = storage_config($cfg, $storeid); | |
360 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
361 | ||
362 | return $plugin->get_subdir($scfg, 'iso'); | |
363 | } | |
364 | ||
365 | sub get_vztmpl_dir { | |
366 | my ($cfg, $storeid) = @_; | |
367 | ||
368 | my $scfg = storage_config($cfg, $storeid); | |
369 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
370 | ||
371 | return $plugin->get_subdir($scfg, 'vztmpl'); | |
372 | } | |
373 | ||
374 | sub get_backup_dir { | |
375 | my ($cfg, $storeid) = @_; | |
376 | ||
377 | my $scfg = storage_config($cfg, $storeid); | |
378 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
379 | ||
380 | return $plugin->get_subdir($scfg, 'backup'); | |
381 | } | |
382 | ||
383 | # library implementation | |
384 | ||
385 | sub parse_vmid { | |
386 | my $vmid = shift; | |
387 | ||
388 | die "VMID '$vmid' contains illegal characters\n" if $vmid !~ m/^\d+$/; | |
389 | ||
390 | return int($vmid); | |
391 | } | |
392 | ||
393 | # NOTE: basename and basevmid are always undef for LVM-thin, where the | |
394 | # clone -> base reference is not encoded in the volume ID. | |
395 | # see note in PVE::Storage::LvmThinPlugin for details. | |
396 | sub parse_volname { | |
397 | my ($cfg, $volid) = @_; | |
398 | ||
399 | my ($storeid, $volname) = parse_volume_id($volid); | |
400 | ||
401 | my $scfg = storage_config($cfg, $storeid); | |
402 | ||
403 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
404 | ||
405 | # returns ($vtype, $name, $vmid, $basename, $basevmid, $isBase, $format) | |
406 | ||
407 | return $plugin->parse_volname($volname); | |
408 | } | |
409 | ||
410 | sub parse_volume_id { | |
411 | my ($volid, $noerr) = @_; | |
412 | ||
413 | return PVE::Storage::Plugin::parse_volume_id($volid, $noerr); | |
414 | } | |
415 | ||
416 | # test if we have read access to volid | |
417 | sub check_volume_access { | |
418 | my ($rpcenv, $user, $cfg, $vmid, $volid) = @_; | |
419 | ||
420 | my ($sid, $volname) = parse_volume_id($volid, 1); | |
421 | if ($sid) { | |
422 | my ($vtype, undef, $ownervm) = parse_volname($cfg, $volid); | |
423 | if ($vtype eq 'iso' || $vtype eq 'vztmpl') { | |
424 | # require at least read access to storage, (custom) templates/ISOs could be sensitive | |
425 | $rpcenv->check_any($user, "/storage/$sid", ['Datastore.AllocateSpace', 'Datastore.Audit']); | |
426 | } elsif (defined($ownervm) && defined($vmid) && ($ownervm == $vmid)) { | |
427 | # we are owner - allow access | |
428 | } elsif ($vtype eq 'backup' && $ownervm) { | |
429 | $rpcenv->check($user, "/storage/$sid", ['Datastore.AllocateSpace']); | |
430 | $rpcenv->check($user, "/vms/$ownervm", ['VM.Backup']); | |
431 | } else { | |
432 | # allow if we are Datastore administrator | |
433 | $rpcenv->check($user, "/storage/$sid", ['Datastore.Allocate']); | |
434 | } | |
435 | } else { | |
436 | die "Only root can pass arbitrary filesystem paths." | |
437 | if $user ne 'root@pam'; | |
438 | } | |
439 | ||
440 | return undef; | |
441 | } | |
442 | ||
443 | my $volume_is_base_and_used__no_lock = sub { | |
444 | my ($scfg, $storeid, $plugin, $volname) = @_; | |
445 | ||
446 | my ($vtype, $name, $vmid, undef, undef, $isBase, undef) = | |
447 | $plugin->parse_volname($volname); | |
448 | ||
449 | if ($isBase) { | |
450 | my $vollist = $plugin->list_images($storeid, $scfg); | |
451 | foreach my $info (@$vollist) { | |
452 | my (undef, $tmpvolname) = parse_volume_id($info->{volid}); | |
453 | my $basename = undef; | |
454 | my $basevmid = undef; | |
455 | ||
456 | eval{ | |
457 | (undef, undef, undef, $basename, $basevmid) = | |
458 | $plugin->parse_volname($tmpvolname); | |
459 | }; | |
460 | ||
461 | if ($basename && defined($basevmid) && $basevmid == $vmid && $basename eq $name) { | |
462 | return 1; | |
463 | } | |
464 | } | |
465 | } | |
466 | return 0; | |
467 | }; | |
468 | ||
469 | # NOTE: this check does not work for LVM-thin, where the clone -> base | |
470 | # reference is not encoded in the volume ID. | |
471 | # see note in PVE::Storage::LvmThinPlugin for details. | |
472 | sub volume_is_base_and_used { | |
473 | my ($cfg, $volid) = @_; | |
474 | ||
475 | my ($storeid, $volname) = parse_volume_id($volid); | |
476 | my $scfg = storage_config($cfg, $storeid); | |
477 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
478 | ||
479 | $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { | |
480 | return &$volume_is_base_and_used__no_lock($scfg, $storeid, $plugin, $volname); | |
481 | }); | |
482 | } | |
483 | ||
484 | # try to map a filesystem path to a volume identifier | |
485 | sub path_to_volume_id { | |
486 | my ($cfg, $path) = @_; | |
487 | ||
488 | my $ids = $cfg->{ids}; | |
489 | ||
490 | my ($sid, $volname) = parse_volume_id($path, 1); | |
491 | if ($sid) { | |
492 | if (my $scfg = $ids->{$sid}) { | |
493 | if ($scfg->{path}) { | |
494 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
495 | my ($vtype, $name, $vmid) = $plugin->parse_volname($volname); | |
496 | return ($vtype, $path); | |
497 | } | |
498 | } | |
499 | return (''); | |
500 | } | |
501 | ||
502 | # Note: abs_path() return undef if $path doesn not exist | |
503 | # for example when nfs storage is not mounted | |
504 | $path = abs_path($path) || $path; | |
505 | ||
506 | foreach my $sid (keys %$ids) { | |
507 | my $scfg = $ids->{$sid}; | |
508 | next if !$scfg->{path}; | |
509 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
510 | my $imagedir = $plugin->get_subdir($scfg, 'images'); | |
511 | my $isodir = $plugin->get_subdir($scfg, 'iso'); | |
512 | my $tmpldir = $plugin->get_subdir($scfg, 'vztmpl'); | |
513 | my $backupdir = $plugin->get_subdir($scfg, 'backup'); | |
514 | my $privatedir = $plugin->get_subdir($scfg, 'rootdir'); | |
515 | my $snippetsdir = $plugin->get_subdir($scfg, 'snippets'); | |
516 | ||
517 | if ($path =~ m!^$imagedir/(\d+)/([^/\s]+)$!) { | |
518 | my $vmid = $1; | |
519 | my $name = $2; | |
520 | ||
521 | my $vollist = $plugin->list_images($sid, $scfg, $vmid); | |
522 | foreach my $info (@$vollist) { | |
523 | my ($storeid, $volname) = parse_volume_id($info->{volid}); | |
524 | my $volpath = $plugin->path($scfg, $volname, $storeid); | |
525 | if ($volpath eq $path) { | |
526 | return ('images', $info->{volid}); | |
527 | } | |
528 | } | |
529 | } elsif ($path =~ m!^$isodir/([^/]+$iso_extension_re)$!) { | |
530 | my $name = $1; | |
531 | return ('iso', "$sid:iso/$name"); | |
532 | } elsif ($path =~ m!^$tmpldir/([^/]+\.tar\.gz)$!) { | |
533 | my $name = $1; | |
534 | return ('vztmpl', "$sid:vztmpl/$name"); | |
535 | } elsif ($path =~ m!^$privatedir/(\d+)$!) { | |
536 | my $vmid = $1; | |
537 | return ('rootdir', "$sid:rootdir/$vmid"); | |
538 | } elsif ($path =~ m!^$backupdir/([^/]+\.(?:tgz|(?:(?:tar|vma)(?:\.(?:${\PVE::Storage::Plugin::COMPRESSOR_RE}))?)))$!) { | |
539 | my $name = $1; | |
540 | return ('backup', "$sid:backup/$name"); | |
541 | } elsif ($path =~ m!^$snippetsdir/([^/]+)$!) { | |
542 | my $name = $1; | |
543 | return ('snippets', "$sid:snippets/$name"); | |
544 | } | |
545 | } | |
546 | ||
547 | # can't map path to volume id | |
548 | return (''); | |
549 | } | |
550 | ||
551 | sub path { | |
552 | my ($cfg, $volid, $snapname) = @_; | |
553 | ||
554 | my ($storeid, $volname) = parse_volume_id($volid); | |
555 | ||
556 | my $scfg = storage_config($cfg, $storeid); | |
557 | ||
558 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
559 | my ($path, $owner, $vtype) = $plugin->path($scfg, $volname, $storeid, $snapname); | |
560 | return wantarray ? ($path, $owner, $vtype) : $path; | |
561 | } | |
562 | ||
563 | sub abs_filesystem_path { | |
564 | my ($cfg, $volid) = @_; | |
565 | ||
566 | my $path; | |
567 | if (parse_volume_id ($volid, 1)) { | |
568 | activate_volumes($cfg, [ $volid ]); | |
569 | $path = PVE::Storage::path($cfg, $volid); | |
570 | } else { | |
571 | if (-f $volid) { | |
572 | my $abspath = abs_path($volid); | |
573 | if ($abspath && $abspath =~ m|^(/.+)$|) { | |
574 | $path = $1; # untaint any path | |
575 | } | |
576 | } | |
577 | } | |
578 | ||
579 | die "can't find file '$volid'\n" if !($path && -f $path); | |
580 | ||
581 | return $path; | |
582 | } | |
583 | ||
584 | my $volname_for_storage = sub { | |
585 | my ($cfg, $volid, $target_storeid) = @_; | |
586 | ||
587 | my (undef, $name, $vmid, undef, undef, undef, $format) = parse_volname($cfg, $volid); | |
588 | my $target_scfg = storage_config($cfg, $target_storeid); | |
589 | ||
590 | my (undef, $valid_formats) = PVE::Storage::Plugin::default_format($target_scfg); | |
591 | my $format_is_valid = grep { $_ eq $format } @$valid_formats; | |
592 | die "unsupported format '$format' for storage type $target_scfg->{type}\n" if !$format_is_valid; | |
593 | ||
594 | (my $name_without_extension = $name) =~ s/\.$format$//; | |
595 | ||
596 | if ($target_scfg->{path}) { | |
597 | return "$vmid/$name_without_extension.$format"; | |
598 | } else { | |
599 | return "$name_without_extension"; | |
600 | } | |
601 | }; | |
602 | ||
603 | sub storage_migrate { | |
604 | my ($cfg, $volid, $target_sshinfo, $target_storeid, $opts, $logfunc) = @_; | |
605 | ||
606 | my $base_snapshot = $opts->{base_snapshot}; | |
607 | my $snapshot = $opts->{snapshot}; | |
608 | my $ratelimit_bps = $opts->{ratelimit_bps}; | |
609 | my $insecure = $opts->{insecure}; | |
610 | my $with_snapshots = $opts->{with_snapshots} ? 1 : 0; | |
611 | my $allow_rename = $opts->{allow_rename} ? 1 : 0; | |
612 | ||
613 | my ($storeid, $volname) = parse_volume_id($volid); | |
614 | ||
615 | my $scfg = storage_config($cfg, $storeid); | |
616 | ||
617 | # no need to migrate shared content | |
618 | return $volid if $storeid eq $target_storeid && $scfg->{shared}; | |
619 | ||
620 | my $tcfg = storage_config($cfg, $target_storeid); | |
621 | ||
622 | my $vtype = (parse_volname($cfg, $volid))[0]; | |
623 | ||
624 | die "content type '$vtype' is not available on storage '$target_storeid'\n" | |
625 | if !$tcfg->{content}->{$vtype}; | |
626 | ||
627 | my $target_volname; | |
628 | if ($opts->{target_volname}) { | |
629 | $target_volname = $opts->{target_volname}; | |
630 | } elsif ($scfg->{type} eq $tcfg->{type}) { | |
631 | $target_volname = $volname; | |
632 | } else { | |
633 | $target_volname = $volname_for_storage->($cfg, $volid, $target_storeid); | |
634 | } | |
635 | ||
636 | my $target_volid = "${target_storeid}:${target_volname}"; | |
637 | ||
638 | my $target_ip = $target_sshinfo->{ip}; | |
639 | ||
640 | my $ssh = PVE::SSHInfo::ssh_info_to_command($target_sshinfo); | |
641 | my $ssh_base = PVE::SSHInfo::ssh_info_to_command_base($target_sshinfo); | |
642 | local $ENV{RSYNC_RSH} = PVE::Tools::cmd2string($ssh_base); | |
643 | ||
644 | my @cstream = ([ '/usr/bin/cstream', '-t', $ratelimit_bps ]) | |
645 | if defined($ratelimit_bps); | |
646 | ||
647 | my $migration_snapshot; | |
648 | if (!defined($snapshot)) { | |
649 | if ($scfg->{type} eq 'zfspool') { | |
650 | $migration_snapshot = 1; | |
651 | $snapshot = '__migration__'; | |
652 | } | |
653 | } | |
654 | ||
655 | my @formats = volume_transfer_formats($cfg, $volid, $target_volid, $snapshot, $base_snapshot, $with_snapshots); | |
656 | die "cannot migrate from storage type '$scfg->{type}' to '$tcfg->{type}'\n" if !@formats; | |
657 | my $format = $formats[0]; | |
658 | ||
659 | my $import_fn = '-'; # let pvesm import read from stdin per default | |
660 | if ($insecure) { | |
661 | my $net = $target_sshinfo->{network} // $target_sshinfo->{ip}; | |
662 | $import_fn = "tcp://$net"; | |
663 | } | |
664 | ||
665 | my $target_apiver = 1; # if there is no apiinfo call, assume 1 | |
666 | my $get_api_version = [@$ssh, 'pvesm', 'apiinfo']; | |
667 | my $match_api_version = sub { $target_apiver = $1 if $_[0] =~ m!^APIVER (\d+)$!; }; | |
668 | eval { run_command($get_api_version, logfunc => $match_api_version); }; | |
669 | ||
670 | my $send = ['pvesm', 'export', $volid, $format, '-', '-with-snapshots', $with_snapshots]; | |
671 | my $recv = [@$ssh, '--', 'pvesm', 'import', $target_volid, $format, $import_fn, '-with-snapshots', $with_snapshots]; | |
672 | if (defined($snapshot)) { | |
673 | push @$send, '-snapshot', $snapshot | |
674 | } | |
675 | if ($migration_snapshot) { | |
676 | push @$recv, '-delete-snapshot', $snapshot; | |
677 | } | |
678 | push @$recv, '-allow-rename', $allow_rename if $target_apiver >= 5; | |
679 | ||
680 | if (defined($base_snapshot)) { | |
681 | # Check if the snapshot exists on the remote side: | |
682 | push @$send, '-base', $base_snapshot; | |
683 | push @$recv, '-base', $base_snapshot; | |
684 | } | |
685 | ||
686 | my $new_volid; | |
687 | my $pattern = volume_imported_message(undef, 1); | |
688 | my $match_volid_and_log = sub { | |
689 | my $line = shift; | |
690 | ||
691 | $new_volid = $1 if ($line =~ $pattern); | |
692 | ||
693 | if ($logfunc) { | |
694 | chomp($line); | |
695 | $logfunc->($line); | |
696 | } | |
697 | }; | |
698 | ||
699 | volume_snapshot($cfg, $volid, $snapshot) if $migration_snapshot; | |
700 | eval { | |
701 | if ($insecure) { | |
702 | open(my $info, '-|', @$recv) | |
703 | or die "receive command failed: $!\n"; | |
704 | my ($ip) = <$info> =~ /^($PVE::Tools::IPRE)$/ or die "no tunnel IP received\n"; | |
705 | my ($port) = <$info> =~ /^(\d+)$/ or die "no tunnel port received\n"; | |
706 | my $socket = IO::Socket::IP->new(PeerHost => $ip, PeerPort => $port, Type => SOCK_STREAM) | |
707 | or die "failed to connect to tunnel at $ip:$port\n"; | |
708 | # we won't be reading from the socket | |
709 | shutdown($socket, 0); | |
710 | run_command([$send, @cstream], output => '>&'.fileno($socket), errfunc => $logfunc); | |
711 | # don't close the connection entirely otherwise the receiving end | |
712 | # might not get all buffered data (and fails with 'connection reset by peer') | |
713 | shutdown($socket, 1); | |
714 | ||
715 | # wait for the remote process to finish | |
716 | while (my $line = <$info>) { | |
717 | $match_volid_and_log->("[$target_sshinfo->{name}] $line"); | |
718 | } | |
719 | ||
720 | # now close the socket | |
721 | close($socket); | |
722 | if (!close($info)) { # does waitpid() | |
723 | die "import failed: $!\n" if $!; | |
724 | die "import failed: exit code ".($?>>8)."\n"; | |
725 | } | |
726 | } else { | |
727 | run_command([$send, @cstream, $recv], logfunc => $match_volid_and_log); | |
728 | } | |
729 | ||
730 | die "unable to get ID of the migrated volume\n" | |
731 | if !defined($new_volid) && $target_apiver >= 5; | |
732 | }; | |
733 | my $err = $@; | |
734 | warn "send/receive failed, cleaning up snapshot(s)..\n" if $err; | |
735 | if ($migration_snapshot) { | |
736 | eval { volume_snapshot_delete($cfg, $volid, $snapshot, 0) }; | |
737 | warn "could not remove source snapshot: $@\n" if $@; | |
738 | } | |
739 | die $err if $err; | |
740 | ||
741 | return $new_volid // $target_volid; | |
742 | } | |
743 | ||
744 | sub vdisk_clone { | |
745 | my ($cfg, $volid, $vmid, $snap) = @_; | |
746 | ||
747 | my ($storeid, $volname) = parse_volume_id($volid); | |
748 | ||
749 | my $scfg = storage_config($cfg, $storeid); | |
750 | ||
751 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
752 | ||
753 | activate_storage($cfg, $storeid); | |
754 | ||
755 | # lock shared storage | |
756 | return $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { | |
757 | my $volname = $plugin->clone_image($scfg, $storeid, $volname, $vmid, $snap); | |
758 | return "$storeid:$volname"; | |
759 | }); | |
760 | } | |
761 | ||
762 | sub vdisk_create_base { | |
763 | my ($cfg, $volid) = @_; | |
764 | ||
765 | my ($storeid, $volname) = parse_volume_id($volid); | |
766 | ||
767 | my $scfg = storage_config($cfg, $storeid); | |
768 | ||
769 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
770 | ||
771 | activate_storage($cfg, $storeid); | |
772 | ||
773 | # lock shared storage | |
774 | return $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { | |
775 | my $volname = $plugin->create_base($storeid, $scfg, $volname); | |
776 | return "$storeid:$volname"; | |
777 | }); | |
778 | } | |
779 | ||
780 | sub map_volume { | |
781 | my ($cfg, $volid, $snapname) = @_; | |
782 | ||
783 | my ($storeid, $volname) = parse_volume_id($volid); | |
784 | ||
785 | my $scfg = storage_config($cfg, $storeid); | |
786 | ||
787 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
788 | ||
789 | return $plugin->map_volume($storeid, $scfg, $volname, $snapname); | |
790 | } | |
791 | ||
792 | sub unmap_volume { | |
793 | my ($cfg, $volid, $snapname) = @_; | |
794 | ||
795 | my ($storeid, $volname) = parse_volume_id($volid); | |
796 | ||
797 | my $scfg = storage_config($cfg, $storeid); | |
798 | ||
799 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
800 | ||
801 | return $plugin->unmap_volume($storeid, $scfg, $volname, $snapname); | |
802 | } | |
803 | ||
804 | sub vdisk_alloc { | |
805 | my ($cfg, $storeid, $vmid, $fmt, $name, $size) = @_; | |
806 | ||
807 | die "no storage ID specified\n" if !$storeid; | |
808 | ||
809 | PVE::JSONSchema::parse_storage_id($storeid); | |
810 | ||
811 | my $scfg = storage_config($cfg, $storeid); | |
812 | ||
813 | die "no VMID specified\n" if !$vmid; | |
814 | ||
815 | $vmid = parse_vmid($vmid); | |
816 | ||
817 | my $defformat = PVE::Storage::Plugin::default_format($scfg); | |
818 | ||
819 | $fmt = $defformat if !$fmt; | |
820 | ||
821 | activate_storage($cfg, $storeid); | |
822 | ||
823 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
824 | ||
825 | # lock shared storage | |
826 | return $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { | |
827 | my $old_umask = umask(umask|0037); | |
828 | my $volname = eval { $plugin->alloc_image($storeid, $scfg, $vmid, $fmt, $name, $size) }; | |
829 | my $err = $@; | |
830 | umask $old_umask; | |
831 | die $err if $err; | |
832 | return "$storeid:$volname"; | |
833 | }); | |
834 | } | |
835 | ||
836 | sub vdisk_free { | |
837 | my ($cfg, $volid) = @_; | |
838 | ||
839 | my ($storeid, $volname) = parse_volume_id($volid); | |
840 | my $scfg = storage_config($cfg, $storeid); | |
841 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
842 | ||
843 | activate_storage($cfg, $storeid); | |
844 | ||
845 | my $cleanup_worker; | |
846 | ||
847 | # lock shared storage | |
848 | $plugin->cluster_lock_storage($storeid, $scfg->{shared}, undef, sub { | |
849 | # LVM-thin allows deletion of still referenced base volumes! | |
850 | die "base volume '$volname' is still in use by linked clones\n" | |
851 | if &$volume_is_base_and_used__no_lock($scfg, $storeid, $plugin, $volname); | |
852 | ||
853 | my (undef, undef, undef, undef, undef, $isBase, $format) = | |
854 | $plugin->parse_volname($volname); | |
855 | $cleanup_worker = $plugin->free_image($storeid, $scfg, $volname, $isBase, $format); | |
856 | }); | |
857 | ||
858 | return if !$cleanup_worker; | |
859 | ||
860 | my $rpcenv = PVE::RPCEnvironment::get(); | |
861 | my $authuser = $rpcenv->get_user(); | |
862 | ||
863 | $rpcenv->fork_worker('imgdel', undef, $authuser, $cleanup_worker); | |
864 | } | |
865 | ||
866 | sub vdisk_list { | |
867 | my ($cfg, $storeid, $vmid, $vollist) = @_; | |
868 | ||
869 | my $ids = $cfg->{ids}; | |
870 | ||
871 | storage_check_enabled($cfg, $storeid) if ($storeid); | |
872 | ||
873 | my $res = {}; | |
874 | ||
875 | # prepare/activate/refresh all storages | |
876 | ||
877 | my $storage_list = []; | |
878 | if ($vollist) { | |
879 | foreach my $volid (@$vollist) { | |
880 | my ($sid, undef) = parse_volume_id($volid); | |
881 | next if !defined($ids->{$sid}); | |
882 | next if !storage_check_enabled($cfg, $sid, undef, 1); | |
883 | push @$storage_list, $sid; | |
884 | } | |
885 | } else { | |
886 | foreach my $sid (keys %$ids) { | |
887 | next if $storeid && $storeid ne $sid; | |
888 | next if !storage_check_enabled($cfg, $sid, undef, 1); | |
889 | push @$storage_list, $sid; | |
890 | } | |
891 | } | |
892 | ||
893 | my $cache = {}; | |
894 | ||
895 | activate_storage_list($cfg, $storage_list, $cache); | |
896 | ||
897 | foreach my $sid (keys %$ids) { | |
898 | next if $storeid && $storeid ne $sid; | |
899 | next if !storage_check_enabled($cfg, $sid, undef, 1); | |
900 | ||
901 | my $scfg = $ids->{$sid}; | |
902 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
903 | $res->{$sid} = $plugin->list_images($sid, $scfg, $vmid, $vollist, $cache); | |
904 | @{$res->{$sid}} = sort {lc($a->{volid}) cmp lc ($b->{volid}) } @{$res->{$sid}} if $res->{$sid}; | |
905 | } | |
906 | ||
907 | return $res; | |
908 | } | |
909 | ||
910 | sub template_list { | |
911 | my ($cfg, $storeid, $tt) = @_; | |
912 | ||
913 | die "unknown template type '$tt'\n" | |
914 | if !($tt eq 'iso' || $tt eq 'vztmpl' || $tt eq 'backup' || $tt eq 'snippets'); | |
915 | ||
916 | my $ids = $cfg->{ids}; | |
917 | ||
918 | storage_check_enabled($cfg, $storeid) if ($storeid); | |
919 | ||
920 | my $res = {}; | |
921 | ||
922 | # query the storage | |
923 | foreach my $sid (keys %$ids) { | |
924 | next if $storeid && $storeid ne $sid; | |
925 | ||
926 | my $scfg = $ids->{$sid}; | |
927 | my $type = $scfg->{type}; | |
928 | ||
929 | next if !$scfg->{content}->{$tt}; | |
930 | ||
931 | next if !storage_check_enabled($cfg, $sid, undef, 1); | |
932 | ||
933 | $res->{$sid} = volume_list($cfg, $sid, undef, $tt); | |
934 | } | |
935 | ||
936 | return $res; | |
937 | } | |
938 | ||
939 | sub volume_list { | |
940 | my ($cfg, $storeid, $vmid, $content) = @_; | |
941 | ||
942 | my @ctypes = qw(rootdir images vztmpl iso backup snippets); | |
943 | ||
944 | my $cts = $content ? [ $content ] : [ @ctypes ]; | |
945 | ||
946 | my $scfg = PVE::Storage::storage_config($cfg, $storeid); | |
947 | ||
948 | $cts = [ grep { defined($scfg->{content}->{$_}) } @$cts ]; | |
949 | ||
950 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
951 | ||
952 | activate_storage($cfg, $storeid); | |
953 | ||
954 | my $res = $plugin->list_volumes($storeid, $scfg, $vmid, $cts); | |
955 | ||
956 | @$res = sort {lc($a->{volid}) cmp lc ($b->{volid}) } @$res; | |
957 | ||
958 | return $res; | |
959 | } | |
960 | ||
961 | sub uevent_seqnum { | |
962 | ||
963 | my $filename = "/sys/kernel/uevent_seqnum"; | |
964 | ||
965 | my $seqnum = 0; | |
966 | if (my $fh = IO::File->new($filename, "r")) { | |
967 | my $line = <$fh>; | |
968 | if ($line =~ m/^(\d+)$/) { | |
969 | $seqnum = int($1); | |
970 | } | |
971 | close ($fh); | |
972 | } | |
973 | return $seqnum; | |
974 | } | |
975 | ||
976 | sub activate_storage { | |
977 | my ($cfg, $storeid, $cache) = @_; | |
978 | ||
979 | $cache = {} if !$cache; | |
980 | ||
981 | my $scfg = storage_check_enabled($cfg, $storeid); | |
982 | ||
983 | return if $cache->{activated}->{$storeid}; | |
984 | ||
985 | $cache->{uevent_seqnum} = uevent_seqnum() if !$cache->{uevent_seqnum}; | |
986 | ||
987 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
988 | ||
989 | if ($scfg->{base}) { | |
990 | my ($baseid, undef) = parse_volume_id ($scfg->{base}); | |
991 | activate_storage($cfg, $baseid, $cache); | |
992 | } | |
993 | ||
994 | if (!$plugin->check_connection($storeid, $scfg)) { | |
995 | die "storage '$storeid' is not online\n"; | |
996 | } | |
997 | ||
998 | $plugin->activate_storage($storeid, $scfg, $cache); | |
999 | ||
1000 | my $newseq = uevent_seqnum (); | |
1001 | ||
1002 | # only call udevsettle if there are events | |
1003 | if ($newseq > $cache->{uevent_seqnum}) { | |
1004 | my $timeout = 30; | |
1005 | system ("$UDEVADM settle --timeout=$timeout"); # ignore errors | |
1006 | $cache->{uevent_seqnum} = $newseq; | |
1007 | } | |
1008 | ||
1009 | $cache->{activated}->{$storeid} = 1; | |
1010 | } | |
1011 | ||
1012 | sub activate_storage_list { | |
1013 | my ($cfg, $storeid_list, $cache) = @_; | |
1014 | ||
1015 | $cache = {} if !$cache; | |
1016 | ||
1017 | foreach my $storeid (@$storeid_list) { | |
1018 | activate_storage($cfg, $storeid, $cache); | |
1019 | } | |
1020 | } | |
1021 | ||
1022 | sub deactivate_storage { | |
1023 | my ($cfg, $storeid) = @_; | |
1024 | ||
1025 | my $scfg = storage_config ($cfg, $storeid); | |
1026 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1027 | ||
1028 | my $cache = {}; | |
1029 | $plugin->deactivate_storage($storeid, $scfg, $cache); | |
1030 | } | |
1031 | ||
1032 | sub activate_volumes { | |
1033 | my ($cfg, $vollist, $snapname) = @_; | |
1034 | ||
1035 | return if !($vollist && scalar(@$vollist)); | |
1036 | ||
1037 | my $storagehash = {}; | |
1038 | foreach my $volid (@$vollist) { | |
1039 | my ($storeid, undef) = parse_volume_id($volid); | |
1040 | $storagehash->{$storeid} = 1; | |
1041 | } | |
1042 | ||
1043 | my $cache = {}; | |
1044 | ||
1045 | activate_storage_list($cfg, [keys %$storagehash], $cache); | |
1046 | ||
1047 | foreach my $volid (@$vollist) { | |
1048 | my ($storeid, $volname) = parse_volume_id($volid); | |
1049 | my $scfg = storage_config($cfg, $storeid); | |
1050 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1051 | $plugin->activate_volume($storeid, $scfg, $volname, $snapname, $cache); | |
1052 | } | |
1053 | } | |
1054 | ||
1055 | sub deactivate_volumes { | |
1056 | my ($cfg, $vollist, $snapname) = @_; | |
1057 | ||
1058 | return if !($vollist && scalar(@$vollist)); | |
1059 | ||
1060 | my $cache = {}; | |
1061 | ||
1062 | my @errlist = (); | |
1063 | foreach my $volid (@$vollist) { | |
1064 | my ($storeid, $volname) = parse_volume_id($volid); | |
1065 | ||
1066 | my $scfg = storage_config($cfg, $storeid); | |
1067 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1068 | ||
1069 | eval { | |
1070 | $plugin->deactivate_volume($storeid, $scfg, $volname, $snapname, $cache); | |
1071 | }; | |
1072 | if (my $err = $@) { | |
1073 | warn $err; | |
1074 | push @errlist, $volid; | |
1075 | } | |
1076 | } | |
1077 | ||
1078 | die "volume deactivation failed: " . join(' ', @errlist) | |
1079 | if scalar(@errlist); | |
1080 | } | |
1081 | ||
1082 | sub storage_info { | |
1083 | my ($cfg, $content, $includeformat) = @_; | |
1084 | ||
1085 | my $ids = $cfg->{ids}; | |
1086 | ||
1087 | my $info = {}; | |
1088 | ||
1089 | my @ctypes = PVE::Tools::split_list($content); | |
1090 | ||
1091 | my $slist = []; | |
1092 | foreach my $storeid (keys %$ids) { | |
1093 | my $storage_enabled = defined(storage_check_enabled($cfg, $storeid, undef, 1)); | |
1094 | ||
1095 | if (defined($content)) { | |
1096 | my $want_ctype = 0; | |
1097 | foreach my $ctype (@ctypes) { | |
1098 | if ($ids->{$storeid}->{content}->{$ctype}) { | |
1099 | $want_ctype = 1; | |
1100 | last; | |
1101 | } | |
1102 | } | |
1103 | next if !$want_ctype || !$storage_enabled; | |
1104 | } | |
1105 | ||
1106 | my $type = $ids->{$storeid}->{type}; | |
1107 | ||
1108 | $info->{$storeid} = { | |
1109 | type => $type, | |
1110 | total => 0, | |
1111 | avail => 0, | |
1112 | used => 0, | |
1113 | shared => $ids->{$storeid}->{shared} ? 1 : 0, | |
1114 | content => PVE::Storage::Plugin::content_hash_to_string($ids->{$storeid}->{content}), | |
1115 | active => 0, | |
1116 | enabled => $storage_enabled ? 1 : 0, | |
1117 | }; | |
1118 | ||
1119 | push @$slist, $storeid; | |
1120 | } | |
1121 | ||
1122 | my $cache = {}; | |
1123 | ||
1124 | foreach my $storeid (keys %$ids) { | |
1125 | my $scfg = $ids->{$storeid}; | |
1126 | ||
1127 | next if !$info->{$storeid}; | |
1128 | next if !$info->{$storeid}->{enabled}; | |
1129 | ||
1130 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1131 | if ($includeformat) { | |
1132 | my $pd = $plugin->plugindata(); | |
1133 | $info->{$storeid}->{format} = $pd->{format} | |
1134 | if $pd->{format}; | |
1135 | $info->{$storeid}->{select_existing} = $pd->{select_existing} | |
1136 | if $pd->{select_existing}; | |
1137 | } | |
1138 | ||
1139 | eval { activate_storage($cfg, $storeid, $cache); }; | |
1140 | if (my $err = $@) { | |
1141 | warn $err; | |
1142 | next; | |
1143 | } | |
1144 | ||
1145 | my ($total, $avail, $used, $active) = eval { $plugin->status($storeid, $scfg, $cache); }; | |
1146 | warn $@ if $@; | |
1147 | next if !$active; | |
1148 | $info->{$storeid}->{total} = int($total); | |
1149 | $info->{$storeid}->{avail} = int($avail); | |
1150 | $info->{$storeid}->{used} = int($used); | |
1151 | $info->{$storeid}->{active} = $active; | |
1152 | } | |
1153 | ||
1154 | return $info; | |
1155 | } | |
1156 | ||
1157 | sub resolv_server { | |
1158 | my ($server) = @_; | |
1159 | ||
1160 | my ($packed_ip, $family); | |
1161 | eval { | |
1162 | my @res = PVE::Tools::getaddrinfo_all($server); | |
1163 | $family = $res[0]->{family}; | |
1164 | $packed_ip = (PVE::Tools::unpack_sockaddr_in46($res[0]->{addr}))[2]; | |
1165 | }; | |
1166 | if (defined $packed_ip) { | |
1167 | return Socket::inet_ntop($family, $packed_ip); | |
1168 | } | |
1169 | return undef; | |
1170 | } | |
1171 | ||
1172 | sub scan_nfs { | |
1173 | my ($server_in) = @_; | |
1174 | ||
1175 | my $server; | |
1176 | if (!($server = resolv_server ($server_in))) { | |
1177 | die "unable to resolve address for server '${server_in}'\n"; | |
1178 | } | |
1179 | ||
1180 | my $cmd = ['/sbin/showmount', '--no-headers', '--exports', $server]; | |
1181 | ||
1182 | my $res = {}; | |
1183 | run_command($cmd, outfunc => sub { | |
1184 | my $line = shift; | |
1185 | ||
1186 | # note: howto handle white spaces in export path?? | |
1187 | if ($line =~ m!^(/\S+)\s+(.+)$!) { | |
1188 | $res->{$1} = $2; | |
1189 | } | |
1190 | }); | |
1191 | ||
1192 | return $res; | |
1193 | } | |
1194 | ||
1195 | sub scan_cifs { | |
1196 | my ($server_in, $user, $password, $domain) = @_; | |
1197 | ||
1198 | my $server; | |
1199 | if (!($server = resolv_server ($server_in))) { | |
1200 | die "unable to resolve address for server '${server_in}'\n"; | |
1201 | } | |
1202 | ||
1203 | # we support only Windows grater than 2012 cifsscan so use smb3 | |
1204 | my $cmd = ['/usr/bin/smbclient', '-m', 'smb3', '-d', '0', '-L', $server]; | |
1205 | if (defined($user)) { | |
1206 | die "password is required" if !defined($password); | |
1207 | push @$cmd, '-U', "$user\%$password"; | |
1208 | push @$cmd, '-W', $domain if defined($domain); | |
1209 | } else { | |
1210 | push @$cmd, '-N'; | |
1211 | } | |
1212 | ||
1213 | my $res = {}; | |
1214 | run_command($cmd, | |
1215 | outfunc => sub { | |
1216 | my $line = shift; | |
1217 | if ($line =~ m/(\S+)\s*Disk\s*(\S*)/) { | |
1218 | $res->{$1} = $2; | |
1219 | } elsif ($line =~ m/(NT_STATUS_(\S*))/) { | |
1220 | $res->{$1} = ''; | |
1221 | } | |
1222 | }, | |
1223 | errfunc => sub {}, | |
1224 | noerr => 1 | |
1225 | ); | |
1226 | ||
1227 | return $res; | |
1228 | } | |
1229 | ||
1230 | sub scan_zfs { | |
1231 | ||
1232 | my $cmd = ['zfs', 'list', '-t', 'filesystem', '-Hp', '-o', 'name,avail,used']; | |
1233 | ||
1234 | my $res = []; | |
1235 | run_command($cmd, outfunc => sub { | |
1236 | my $line = shift; | |
1237 | ||
1238 | if ($line =~m/^(\S+)\s+(\S+)\s+(\S+)$/) { | |
1239 | my ($pool, $size_str, $used_str) = ($1, $2, $3); | |
1240 | my $size = $size_str + 0; | |
1241 | my $used = $used_str + 0; | |
1242 | # ignore subvolumes generated by our ZFSPoolPlugin | |
1243 | return if $pool =~ m!/subvol-\d+-[^/]+$!; | |
1244 | return if $pool =~ m!/basevol-\d+-[^/]+$!; | |
1245 | push @$res, { pool => $pool, size => $size, free => $size-$used }; | |
1246 | } | |
1247 | }); | |
1248 | ||
1249 | return $res; | |
1250 | } | |
1251 | ||
1252 | sub resolv_portal { | |
1253 | my ($portal, $noerr) = @_; | |
1254 | ||
1255 | my ($server, $port) = PVE::Tools::parse_host_and_port($portal); | |
1256 | if ($server) { | |
1257 | if (my $ip = resolv_server($server)) { | |
1258 | $server = $ip; | |
1259 | $server = "[$server]" if $server =~ /^$IPV6RE$/; | |
1260 | return $port ? "$server:$port" : $server; | |
1261 | } | |
1262 | } | |
1263 | return undef if $noerr; | |
1264 | ||
1265 | raise_param_exc({ portal => "unable to resolve portal address '$portal'" }); | |
1266 | } | |
1267 | ||
1268 | ||
1269 | sub scan_iscsi { | |
1270 | my ($portal_in) = @_; | |
1271 | ||
1272 | my $portal; | |
1273 | if (!($portal = resolv_portal($portal_in))) { | |
1274 | die "unable to parse/resolve portal address '${portal_in}'\n"; | |
1275 | } | |
1276 | ||
1277 | return PVE::Storage::ISCSIPlugin::iscsi_discovery($portal); | |
1278 | } | |
1279 | ||
1280 | sub storage_default_format { | |
1281 | my ($cfg, $storeid) = @_; | |
1282 | ||
1283 | my $scfg = storage_config ($cfg, $storeid); | |
1284 | ||
1285 | return PVE::Storage::Plugin::default_format($scfg); | |
1286 | } | |
1287 | ||
1288 | sub vgroup_is_used { | |
1289 | my ($cfg, $vgname) = @_; | |
1290 | ||
1291 | foreach my $storeid (keys %{$cfg->{ids}}) { | |
1292 | my $scfg = storage_config($cfg, $storeid); | |
1293 | if ($scfg->{type} eq 'lvm' && $scfg->{vgname} eq $vgname) { | |
1294 | return 1; | |
1295 | } | |
1296 | } | |
1297 | ||
1298 | return undef; | |
1299 | } | |
1300 | ||
1301 | sub target_is_used { | |
1302 | my ($cfg, $target) = @_; | |
1303 | ||
1304 | foreach my $storeid (keys %{$cfg->{ids}}) { | |
1305 | my $scfg = storage_config($cfg, $storeid); | |
1306 | if ($scfg->{type} eq 'iscsi' && $scfg->{target} eq $target) { | |
1307 | return 1; | |
1308 | } | |
1309 | } | |
1310 | ||
1311 | return undef; | |
1312 | } | |
1313 | ||
1314 | sub volume_is_used { | |
1315 | my ($cfg, $volid) = @_; | |
1316 | ||
1317 | foreach my $storeid (keys %{$cfg->{ids}}) { | |
1318 | my $scfg = storage_config($cfg, $storeid); | |
1319 | if ($scfg->{base} && $scfg->{base} eq $volid) { | |
1320 | return 1; | |
1321 | } | |
1322 | } | |
1323 | ||
1324 | return undef; | |
1325 | } | |
1326 | ||
1327 | sub storage_is_used { | |
1328 | my ($cfg, $storeid) = @_; | |
1329 | ||
1330 | foreach my $sid (keys %{$cfg->{ids}}) { | |
1331 | my $scfg = storage_config($cfg, $sid); | |
1332 | next if !$scfg->{base}; | |
1333 | my ($st) = parse_volume_id($scfg->{base}); | |
1334 | return 1 if $st && $st eq $storeid; | |
1335 | } | |
1336 | ||
1337 | return undef; | |
1338 | } | |
1339 | ||
1340 | sub foreach_volid { | |
1341 | my ($list, $func) = @_; | |
1342 | ||
1343 | return if !$list; | |
1344 | ||
1345 | foreach my $sid (keys %$list) { | |
1346 | foreach my $info (@{$list->{$sid}}) { | |
1347 | my $volid = $info->{volid}; | |
1348 | my ($sid1, $volname) = parse_volume_id($volid, 1); | |
1349 | if ($sid1 && $sid1 eq $sid) { | |
1350 | &$func ($volid, $sid, $info); | |
1351 | } else { | |
1352 | warn "detected strange volid '$volid' in volume list for '$sid'\n"; | |
1353 | } | |
1354 | } | |
1355 | } | |
1356 | } | |
1357 | ||
1358 | sub decompressor_info { | |
1359 | my ($format, $comp) = @_; | |
1360 | ||
1361 | if ($format eq 'tgz' && !defined($comp)) { | |
1362 | ($format, $comp) = ('tar', 'gz'); | |
1363 | } | |
1364 | ||
1365 | my $decompressor = { | |
1366 | tar => { | |
1367 | gz => ['tar', '-z'], | |
1368 | lzo => ['tar', '--lzop'], | |
1369 | zst => ['tar', '--zstd'], | |
1370 | }, | |
1371 | vma => { | |
1372 | gz => ['zcat'], | |
1373 | lzo => ['lzop', '-d', '-c'], | |
1374 | zst => ['zstd', '-q', '-d', '-c'], | |
1375 | }, | |
1376 | }; | |
1377 | ||
1378 | die "ERROR: archive format not defined\n" | |
1379 | if !defined($decompressor->{$format}); | |
1380 | ||
1381 | my $decomp = $decompressor->{$format}->{$comp} if $comp; | |
1382 | ||
1383 | my $info = { | |
1384 | format => $format, | |
1385 | compression => $comp, | |
1386 | decompressor => $decomp, | |
1387 | }; | |
1388 | ||
1389 | return $info; | |
1390 | } | |
1391 | ||
1392 | sub archive_info { | |
1393 | my ($archive) = shift; | |
1394 | my $info; | |
1395 | ||
1396 | my $volid = basename($archive); | |
1397 | if ($volid =~ /\.(tgz$|tar|vma)(?:\.(${\PVE::Storage::Plugin::COMPRESSOR_RE}))?$/) { | |
1398 | $info = decompressor_info($1, $2); | |
1399 | $info->{type} = 'unknown'; | |
1400 | ||
1401 | if ($volid =~ /vzdump-(lxc|openvz|qemu)-\d+-(?:\d{4})_(?:\d{2})_(?:\d{2})-(?:\d{2})_(?:\d{2})_(?:\d{2})/) { | |
1402 | $info->{type} = $1; | |
1403 | } | |
1404 | } else { | |
1405 | die "ERROR: couldn't determine format and compression type\n"; | |
1406 | } | |
1407 | ||
1408 | return $info; | |
1409 | } | |
1410 | ||
1411 | sub extract_vzdump_config_tar { | |
1412 | my ($archive, $conf_re) = @_; | |
1413 | ||
1414 | die "ERROR: file '$archive' does not exist\n" if ! -f $archive; | |
1415 | ||
1416 | my $pid = open(my $fh, '-|', 'tar', 'tf', $archive) || | |
1417 | die "unable to open file '$archive'\n"; | |
1418 | ||
1419 | my $file; | |
1420 | while (defined($file = <$fh>)) { | |
1421 | if ($file =~ $conf_re) { | |
1422 | $file = $1; # untaint | |
1423 | last; | |
1424 | } | |
1425 | } | |
1426 | ||
1427 | kill 15, $pid; | |
1428 | waitpid $pid, 0; | |
1429 | close $fh; | |
1430 | ||
1431 | die "ERROR: archive contains no configuration file\n" if !$file; | |
1432 | chomp $file; | |
1433 | ||
1434 | my $raw = ''; | |
1435 | my $out = sub { | |
1436 | my $output = shift; | |
1437 | $raw .= "$output\n"; | |
1438 | }; | |
1439 | ||
1440 | PVE::Tools::run_command(['tar', '-xpOf', $archive, $file, '--occurrence'], outfunc => $out); | |
1441 | ||
1442 | return wantarray ? ($raw, $file) : $raw; | |
1443 | } | |
1444 | ||
1445 | sub extract_vzdump_config_vma { | |
1446 | my ($archive, $comp) = @_; | |
1447 | ||
1448 | my $cmd; | |
1449 | my $raw = ''; | |
1450 | my $out = sub { | |
1451 | my $output = shift; | |
1452 | $raw .= "$output\n"; | |
1453 | }; | |
1454 | ||
1455 | ||
1456 | my $info = archive_info($archive); | |
1457 | $comp //= $info->{compression}; | |
1458 | my $decompressor = $info->{decompressor}; | |
1459 | ||
1460 | if ($comp) { | |
1461 | $cmd = [ [@$decompressor, $archive], ["vma", "config", "-"] ]; | |
1462 | ||
1463 | # in some cases, lzop/zcat exits with 1 when its stdout pipe is | |
1464 | # closed early by vma, detect this and ignore the exit code later | |
1465 | my $broken_pipe; | |
1466 | my $errstring; | |
1467 | my $err = sub { | |
1468 | my $output = shift; | |
1469 | if ($output =~ m/lzop: Broken pipe: <stdout>/ || $output =~ m/gzip: stdout: Broken pipe/ || $output =~ m/zstd: error 70 : Write error : Broken pipe/) { | |
1470 | $broken_pipe = 1; | |
1471 | } elsif (!defined ($errstring) && $output !~ m/^\s*$/) { | |
1472 | $errstring = "Failed to extract config from VMA archive: $output\n"; | |
1473 | } | |
1474 | }; | |
1475 | ||
1476 | # in other cases, the pipeline will exit with exit code 141 | |
1477 | # because of the broken pipe, handle / ignore this as well | |
1478 | my $rc; | |
1479 | eval { | |
1480 | $rc = PVE::Tools::run_command($cmd, outfunc => $out, errfunc => $err, noerr => 1); | |
1481 | }; | |
1482 | my $rerr = $@; | |
1483 | ||
1484 | # use exit code if no stderr output and not just broken pipe | |
1485 | if (!$errstring && !$broken_pipe && $rc != 0 && $rc != 141) { | |
1486 | die "$rerr\n" if $rerr; | |
1487 | die "config extraction failed with exit code $rc\n"; | |
1488 | } | |
1489 | die "$errstring\n" if $errstring; | |
1490 | } else { | |
1491 | # simple case without compression and weird piping behaviour | |
1492 | PVE::Tools::run_command(["vma", "config", $archive], outfunc => $out); | |
1493 | } | |
1494 | ||
1495 | return wantarray ? ($raw, undef) : $raw; | |
1496 | } | |
1497 | ||
1498 | sub extract_vzdump_config { | |
1499 | my ($cfg, $volid) = @_; | |
1500 | ||
1501 | my ($storeid, $volname) = parse_volume_id($volid); | |
1502 | if (defined($storeid)) { | |
1503 | my $scfg = storage_config($cfg, $storeid); | |
1504 | if ($scfg->{type} eq 'pbs') { | |
1505 | storage_check_enabled($cfg, $storeid); | |
1506 | return PVE::Storage::PBSPlugin->extract_vzdump_config($scfg, $volname, $storeid); | |
1507 | } | |
1508 | } | |
1509 | ||
1510 | my $archive = abs_filesystem_path($cfg, $volid); | |
1511 | my $info = archive_info($archive); | |
1512 | my $format = $info->{format}; | |
1513 | my $comp = $info->{compression}; | |
1514 | my $type = $info->{type}; | |
1515 | ||
1516 | if ($type eq 'lxc' || $type eq 'openvz') { | |
1517 | return extract_vzdump_config_tar($archive, qr!^(\./etc/vzdump/(pct|vps)\.conf)$!); | |
1518 | } elsif ($type eq 'qemu') { | |
1519 | if ($format eq 'tar') { | |
1520 | return extract_vzdump_config_tar($archive, qr!\(\./qemu-server\.conf\)!); | |
1521 | } else { | |
1522 | return extract_vzdump_config_vma($archive, $comp); | |
1523 | } | |
1524 | } else { | |
1525 | die "cannot determine backup guest type for backup archive '$volid'\n"; | |
1526 | } | |
1527 | } | |
1528 | ||
1529 | sub volume_export { | |
1530 | my ($cfg, $fh, $volid, $format, $snapshot, $base_snapshot, $with_snapshots) = @_; | |
1531 | ||
1532 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
1533 | die "cannot export volume '$volid'\n" if !$storeid; | |
1534 | my $scfg = storage_config($cfg, $storeid); | |
1535 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1536 | return $plugin->volume_export($scfg, $storeid, $fh, $volname, $format, | |
1537 | $snapshot, $base_snapshot, $with_snapshots); | |
1538 | } | |
1539 | ||
1540 | sub volume_import { | |
1541 | my ($cfg, $fh, $volid, $format, $base_snapshot, $with_snapshots, $allow_rename) = @_; | |
1542 | ||
1543 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
1544 | die "cannot import into volume '$volid'\n" if !$storeid; | |
1545 | my $scfg = storage_config($cfg, $storeid); | |
1546 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1547 | return $plugin->volume_import($scfg, $storeid, $fh, $volname, $format, | |
1548 | $base_snapshot, $with_snapshots, $allow_rename) // $volid; | |
1549 | } | |
1550 | ||
1551 | sub volume_export_formats { | |
1552 | my ($cfg, $volid, $snapshot, $base_snapshot, $with_snapshots) = @_; | |
1553 | ||
1554 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
1555 | return if !$storeid; | |
1556 | my $scfg = storage_config($cfg, $storeid); | |
1557 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1558 | return $plugin->volume_export_formats($scfg, $storeid, $volname, | |
1559 | $snapshot, $base_snapshot, | |
1560 | $with_snapshots); | |
1561 | } | |
1562 | ||
1563 | sub volume_import_formats { | |
1564 | my ($cfg, $volid, $base_snapshot, $with_snapshots) = @_; | |
1565 | ||
1566 | my ($storeid, $volname) = parse_volume_id($volid, 1); | |
1567 | return if !$storeid; | |
1568 | my $scfg = storage_config($cfg, $storeid); | |
1569 | my $plugin = PVE::Storage::Plugin->lookup($scfg->{type}); | |
1570 | return $plugin->volume_import_formats($scfg, $storeid, $volname, | |
1571 | $base_snapshot, $with_snapshots); | |
1572 | } | |
1573 | ||
1574 | sub volume_transfer_formats { | |
1575 | my ($cfg, $src_volid, $dst_volid, $snapshot, $base_snapshot, $with_snapshots) = @_; | |
1576 | my @export_formats = volume_export_formats($cfg, $src_volid, $snapshot, $base_snapshot, $with_snapshots); | |
1577 | my @import_formats = volume_import_formats($cfg, $dst_volid, $base_snapshot, $with_snapshots); | |
1578 | my %import_hash = map { $_ => 1 } @import_formats; | |
1579 | my @common = grep { $import_hash{$_} } @export_formats; | |
1580 | return @common; | |
1581 | } | |
1582 | ||
1583 | sub volume_imported_message { | |
1584 | my ($volid, $want_pattern) = @_; | |
1585 | ||
1586 | if ($want_pattern) { | |
1587 | return qr/successfully imported '([^']*)'$/; | |
1588 | } else { | |
1589 | return "successfully imported '$volid'\n"; | |
1590 | } | |
1591 | } | |
1592 | ||
1593 | # bash completion helper | |
1594 | ||
1595 | sub complete_storage { | |
1596 | my ($cmdname, $pname, $cvalue) = @_; | |
1597 | ||
1598 | my $cfg = PVE::Storage::config(); | |
1599 | ||
1600 | return $cmdname eq 'add' ? [] : [ PVE::Storage::storage_ids($cfg) ]; | |
1601 | } | |
1602 | ||
1603 | sub complete_storage_enabled { | |
1604 | my ($cmdname, $pname, $cvalue) = @_; | |
1605 | ||
1606 | my $res = []; | |
1607 | ||
1608 | my $cfg = PVE::Storage::config(); | |
1609 | foreach my $sid (keys %{$cfg->{ids}}) { | |
1610 | next if !storage_check_enabled($cfg, $sid, undef, 1); | |
1611 | push @$res, $sid; | |
1612 | } | |
1613 | return $res; | |
1614 | } | |
1615 | ||
1616 | sub complete_content_type { | |
1617 | my ($cmdname, $pname, $cvalue) = @_; | |
1618 | ||
1619 | return [qw(rootdir images vztmpl iso backup snippets)]; | |
1620 | } | |
1621 | ||
1622 | sub complete_volume { | |
1623 | my ($cmdname, $pname, $cvalue) = @_; | |
1624 | ||
1625 | my $cfg = config(); | |
1626 | ||
1627 | my $storage_list = complete_storage_enabled(); | |
1628 | ||
1629 | if ($cvalue =~ m/^([^:]+):/) { | |
1630 | $storage_list = [ $1 ]; | |
1631 | } else { | |
1632 | if (scalar(@$storage_list) > 1) { | |
1633 | # only list storage IDs to avoid large listings | |
1634 | my $res = []; | |
1635 | foreach my $storeid (@$storage_list) { | |
1636 | # Hack: simply return 2 artificial values, so that | |
1637 | # completions does not finish | |
1638 | push @$res, "$storeid:volname", "$storeid:..."; | |
1639 | } | |
1640 | return $res; | |
1641 | } | |
1642 | } | |
1643 | ||
1644 | my $res = []; | |
1645 | foreach my $storeid (@$storage_list) { | |
1646 | my $vollist = PVE::Storage::volume_list($cfg, $storeid); | |
1647 | ||
1648 | foreach my $item (@$vollist) { | |
1649 | push @$res, $item->{volid}; | |
1650 | } | |
1651 | } | |
1652 | ||
1653 | return $res; | |
1654 | } | |
1655 | ||
1656 | # Various io-heavy operations require io/bandwidth limits which can be | |
1657 | # configured on multiple levels: The global defaults in datacenter.cfg, and | |
1658 | # per-storage overrides. When we want to do a restore from storage A to storage | |
1659 | # B, we should take the smaller limit defined for storages A and B, and if no | |
1660 | # such limit was specified, use the one from datacenter.cfg. | |
1661 | sub get_bandwidth_limit { | |
1662 | my ($operation, $storage_list, $override) = @_; | |
1663 | ||
1664 | # called for each limit (global, per-storage) with the 'default' and the | |
1665 | # $operation limit and should udpate $override for every limit affecting | |
1666 | # us. | |
1667 | my $use_global_limits = 0; | |
1668 | my $apply_limit = sub { | |
1669 | my ($bwlimit) = @_; | |
1670 | if (defined($bwlimit)) { | |
1671 | my $limits = PVE::JSONSchema::parse_property_string('bwlimit', $bwlimit); | |
1672 | my $limit = $limits->{$operation} // $limits->{default}; | |
1673 | if (defined($limit)) { | |
1674 | if (!$override || $limit < $override) { | |
1675 | $override = $limit; | |
1676 | } | |
1677 | return; | |
1678 | } | |
1679 | } | |
1680 | # If there was no applicable limit, try to apply the global ones. | |
1681 | $use_global_limits = 1; | |
1682 | }; | |
1683 | ||
1684 | my ($rpcenv, $authuser); | |
1685 | if (defined($override)) { | |
1686 | $rpcenv = PVE::RPCEnvironment->get(); | |
1687 | $authuser = $rpcenv->get_user(); | |
1688 | } | |
1689 | ||
1690 | # Apply per-storage limits - if there are storages involved. | |
1691 | if (defined($storage_list) && @$storage_list) { | |
1692 | my $config = config(); | |
1693 | ||
1694 | # The Datastore.Allocate permission allows us to modify the per-storage | |
1695 | # limits, therefore it also allows us to override them. | |
1696 | # Since we have most likely multiple storages to check, do a quick check on | |
1697 | # the general '/storage' path to see if we can skip the checks entirely: | |
1698 | return $override if $rpcenv && $rpcenv->check($authuser, '/storage', ['Datastore.Allocate'], 1); | |
1699 | ||
1700 | my %done; | |
1701 | foreach my $storage (@$storage_list) { | |
1702 | next if !defined($storage); | |
1703 | # Avoid duplicate checks: | |
1704 | next if $done{$storage}; | |
1705 | $done{$storage} = 1; | |
1706 | ||
1707 | # Otherwise we may still have individual /storage/$ID permissions: | |
1708 | if (!$rpcenv || !$rpcenv->check($authuser, "/storage/$storage", ['Datastore.Allocate'], 1)) { | |
1709 | # And if not: apply the limits. | |
1710 | my $storecfg = storage_config($config, $storage); | |
1711 | $apply_limit->($storecfg->{bwlimit}); | |
1712 | } | |
1713 | } | |
1714 | ||
1715 | # Storage limits take precedence over the datacenter defaults, so if | |
1716 | # a limit was applied: | |
1717 | return $override if !$use_global_limits; | |
1718 | } | |
1719 | ||
1720 | # Sys.Modify on '/' means we can change datacenter.cfg which contains the | |
1721 | # global default limits. | |
1722 | if (!$rpcenv || !$rpcenv->check($authuser, '/', ['Sys.Modify'], 1)) { | |
1723 | # So if we cannot modify global limits, apply them to our currently | |
1724 | # requested override. | |
1725 | my $dc = cfs_read_file('datacenter.cfg'); | |
1726 | $apply_limit->($dc->{bwlimit}); | |
1727 | } | |
1728 | ||
1729 | return $override; | |
1730 | } | |
1731 | ||
1732 | # checks if the storage id is available and dies if not | |
1733 | sub assert_sid_unused { | |
1734 | my ($sid) = @_; | |
1735 | ||
1736 | my $cfg = config(); | |
1737 | if (my $scfg = storage_config($cfg, $sid, 1)) { | |
1738 | die "storage ID '$sid' already defined\n"; | |
1739 | } | |
1740 | ||
1741 | return undef; | |
1742 | } | |
1743 | ||
1744 | 1; |