]>
Commit | Line | Data |
---|---|---|
e4fc8228 AD |
1 | package PVE::Storage::SheepdogPlugin; |
2 | ||
3 | use strict; | |
4 | use warnings; | |
5 | use IO::File; | |
6 | use PVE::Tools qw(run_command trim); | |
7 | use PVE::Storage::Plugin; | |
8 | use PVE::JSONSchema qw(get_standard_option); | |
9 | ||
10 | use base qw(PVE::Storage::Plugin); | |
11 | ||
be6339fc DM |
12 | my $collie_cmd = sub { |
13 | my ($scfg, $class, $op, @options) = @_; | |
e4fc8228 AD |
14 | |
15 | my $portal = $scfg->{portal}; | |
16 | my ($server, $port) = split(':', $portal); | |
2622a5ca | 17 | my $cmd = ['/usr/bin/dog', $class, $op, '-a', $server]; |
be6339fc | 18 | push @$cmd, '-p', $port if $port; |
7f8373d2 | 19 | |
be6339fc DM |
20 | push @$cmd, @options if scalar(@options); |
21 | ||
22 | return $cmd; | |
23 | }; | |
e4fc8228 | 24 | |
be6339fc DM |
25 | sub sheepdog_ls { |
26 | my ($scfg, $storeid) = @_; | |
e4fc8228 | 27 | |
ba4ee9ba AD |
28 | my $cmd = &$collie_cmd($scfg, 'vdi', 'graph'); |
29 | ||
30 | my $relationship = {}; | |
31 | my $child = undef; | |
32 | ||
33 | run_command($cmd, outfunc => sub { | |
34 | my $line = shift; | |
35 | ||
36 | my $parent = undef; | |
37 | my $name = undef; | |
38 | ||
39 | $line = trim($line); | |
40 | ||
41 | if ($line =~ /\"(\S+)\"\s->\s\"(\S+)\"/) { | |
42 | $parent = $1; | |
43 | $child = $2; | |
44 | $relationship->{$child}->{parent} = $parent; | |
45 | } | |
46 | elsif ($line =~ /group\s\=\s\"(\S+)\",/) { | |
47 | $name = $1; | |
48 | $relationship->{$child}->{name} = $name if $child; | |
49 | ||
50 | } | |
51 | ||
52 | }); | |
53 | ||
54 | ||
55 | $cmd = &$collie_cmd($scfg, 'vdi', 'list', '-r'); | |
be6339fc DM |
56 | |
57 | my $list = {}; | |
58 | ||
59 | run_command($cmd, outfunc => sub { | |
e4fc8228 AD |
60 | my $line = shift; |
61 | $line = trim($line); | |
a76d3c97 | 62 | if ($line =~ /(=|c) ((vm|base)-(\d+)-\S+)\s+(\d+)\s+(\d+)\s(\d+)\s(\d+)\s(\d+)\s(\S+)\s(\d+)/) { |
ba4ee9ba | 63 | my $image = $2; |
a76d3c97 | 64 | my $owner = $4; |
9538d047 | 65 | my $size = $6; |
a76d3c97 | 66 | my $idvdi = $10; |
ba4ee9ba | 67 | my $parentid = $relationship->{$idvdi}->{parent} if $relationship->{$idvdi}->{parent}; |
73b6f89b | 68 | my $parent = $relationship->{$parentid}->{name} if $parentid; |
e4fc8228 AD |
69 | $list->{$storeid}->{$image} = { |
70 | name => $image, | |
71 | size => $size, | |
ba4ee9ba | 72 | parent => $parent, |
e4fc8228 AD |
73 | vmid => $owner |
74 | }; | |
e4fc8228 AD |
75 | } |
76 | }); | |
77 | ||
78 | return $list; | |
e4fc8228 AD |
79 | } |
80 | ||
3cb21703 AD |
81 | sub sheepdog_snapshot_ls { |
82 | my ($scfg, $volname) = @_; | |
83 | ||
84 | my $cmd = &$collie_cmd($scfg, 'vdi', 'list', '-r'); | |
85 | ||
86 | my $list = {}; | |
87 | run_command($cmd, outfunc => sub { | |
88 | my $line = shift; | |
89 | $line = trim($line); | |
a602d957 AD |
90 | if ($line =~ m/s $volname (\d+) (\d+) (\d+) (\d+) (\d+) (\S+) (\d+) (\S+)/) { |
91 | $list->{$8} = 1; | |
3cb21703 AD |
92 | } |
93 | }); | |
94 | ||
95 | return $list; | |
96 | } | |
97 | ||
e4fc8228 AD |
98 | # Configuration |
99 | ||
100 | ||
101 | sub type { | |
102 | return 'sheepdog'; | |
103 | } | |
104 | ||
105 | sub plugindata { | |
106 | return { | |
107 | content => [ {images => 1}, { images => 1 }], | |
108 | }; | |
109 | } | |
110 | ||
111 | ||
112 | sub options { | |
113 | return { | |
911ce850 AD |
114 | nodes => { optional => 1 }, |
115 | disable => { optional => 1 }, | |
e4fc8228 AD |
116 | portal => { fixed => 1 }, |
117 | content => { optional => 1 }, | |
9edb99a5 | 118 | bwlimit => { optional => 1 }, |
e4fc8228 AD |
119 | }; |
120 | } | |
121 | ||
122 | # Storage implementation | |
123 | ||
124 | sub parse_volname { | |
125 | my ($class, $volname) = @_; | |
126 | ||
648095a9 | 127 | if ($volname =~ m/^((base-(\d+)-\S+)\/)?((base)?(vm)?-(\d+)-\S+)$/) { |
7800e84d | 128 | return ('images', $4, $7, $2, $3, $5, 'raw'); |
e4fc8228 AD |
129 | } |
130 | ||
9165b015 | 131 | die "unable to parse sheepdog volume name '$volname'\n"; |
e4fc8228 AD |
132 | } |
133 | ||
134 | sub path { | |
24cec3aa | 135 | my ($class, $scfg, $volname, $storeid, $snapname) = @_; |
e4fc8228 AD |
136 | |
137 | my ($vtype, $name, $vmid) = $class->parse_volname($volname); | |
138 | ||
139 | my $portal = $scfg->{portal}; | |
f7a1e35f DM |
140 | my ($server, $port) = split(':', $portal); |
141 | $port = 7000 if !$port; | |
24cec3aa | 142 | $name .= ':'.$snapname if $snapname; |
7f8373d2 | 143 | |
f7a1e35f | 144 | my $path = "sheepdog:$server:$port:$name"; |
e4fc8228 AD |
145 | |
146 | return ($path, $vmid, $vtype); | |
147 | } | |
148 | ||
19860f5b AD |
149 | my $find_free_diskname = sub { |
150 | my ($storeid, $scfg, $vmid) = @_; | |
151 | ||
152 | my $sheepdog = sheepdog_ls($scfg, $storeid); | |
9538d047 | 153 | my $dat = $sheepdog->{$storeid}; |
19860f5b AD |
154 | my $disk_ids = {}; |
155 | ||
156 | foreach my $image (keys %$dat) { | |
157 | my $volname = $dat->{$image}->{name}; | |
158 | if ($volname =~ m/(vm|base)-$vmid-disk-(\d+)/){ | |
159 | $disk_ids->{$2} = 1; | |
160 | } | |
161 | } | |
162 | ||
163 | for (my $i = 1; $i < 100; $i++) { | |
164 | if (!$disk_ids->{$i}) { | |
165 | return "vm-$vmid-disk-$i"; | |
166 | } | |
167 | } | |
168 | ||
169 | die "unable to allocate an image name for VM $vmid in storage '$storeid'\n"; | |
170 | }; | |
171 | ||
5eab0272 DM |
172 | sub create_base { |
173 | my ($class, $storeid, $scfg, $volname) = @_; | |
174 | ||
8a3c8d0e AD |
175 | my $snap = '__base__'; |
176 | ||
177 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = | |
178 | $class->parse_volname($volname); | |
179 | ||
180 | die "create_base not possible with base image\n" if $isBase; | |
181 | ||
182 | my $sheepdog = sheepdog_ls($scfg, $storeid); | |
9165b015 AD |
183 | die "sheepdog volume info on '$name' failed\n" if !($sheepdog->{$storeid}->{$name}); |
184 | my $parent = $sheepdog->{$storeid}->{$name}->{parent}; | |
8a3c8d0e AD |
185 | |
186 | die "volname '$volname' contains wrong information about parent $parent $basename\n" | |
187 | if $basename && (!$parent || $parent ne $basename); | |
188 | ||
189 | my $newname = $name; | |
190 | $newname =~ s/^vm-/base-/; | |
191 | ||
192 | my $newvolname = $basename ? "$basename/$newname" : "$newname"; | |
193 | ||
194 | #sheepdog can't rename, so we clone then delete the parent | |
195 | ||
196 | my $tempsnap = '__tempforbase__'; | |
197 | ||
198 | my $cmd = &$collie_cmd($scfg, 'vdi', 'snapshot', '-s', $tempsnap, $name); | |
199 | run_command($cmd, errmsg => "sheepdog snapshot $volname' error"); | |
200 | ||
201 | $cmd = &$collie_cmd($scfg, 'vdi', 'clone', '-s', $tempsnap, $name, $newname); | |
202 | run_command($cmd, errmsg => "sheepdog clone $volname' error"); | |
203 | ||
204 | $cmd = &$collie_cmd($scfg, 'vdi', 'delete', '-s', $tempsnap, $name); | |
205 | run_command($cmd, errmsg => "sheepdog delete snapshot $volname' error"); | |
206 | ||
207 | $cmd = &$collie_cmd($scfg, 'vdi', 'delete' , $name); | |
208 | run_command($cmd, errmsg => "sheepdog delete $volname' error"); | |
209 | ||
210 | #create the base snapshot | |
211 | my $running = undef; #fixme : is create_base always offline ? | |
212 | ||
213 | $class->volume_snapshot($scfg, $storeid, $newname, $snap, $running); | |
214 | ||
215 | return $newvolname; | |
5eab0272 DM |
216 | } |
217 | ||
218 | sub clone_image { | |
f236eaf8 | 219 | my ($class, $scfg, $storeid, $volname, $vmid, $snap) = @_; |
5eab0272 | 220 | |
f236eaf8 | 221 | $snap ||= '__base__'; |
87157341 AD |
222 | |
223 | my ($vtype, $basename, $basevmid, undef, undef, $isBase) = | |
224 | $class->parse_volname($volname); | |
225 | ||
226 | die "clone_image only works on base images\n" if !$isBase; | |
227 | ||
228 | my $name = &$find_free_diskname($storeid, $scfg, $vmid); | |
229 | ||
230 | warn "clone $volname: $basename to $name\n"; | |
231 | ||
232 | my $newvol = "$basename/$name"; | |
233 | ||
234 | my $cmd = &$collie_cmd($scfg, 'vdi', 'clone', '-s', $snap, $basename, $name); | |
235 | run_command($cmd, errmsg => "sheepdog clone $volname' error"); | |
236 | ||
237 | return $newvol; | |
5eab0272 DM |
238 | } |
239 | ||
e4fc8228 AD |
240 | sub alloc_image { |
241 | my ($class, $storeid, $scfg, $vmid, $fmt, $name, $size) = @_; | |
242 | ||
e4fc8228 AD |
243 | die "illegal name '$name' - sould be 'vm-$vmid-*'\n" |
244 | if $name && $name !~ m/^vm-$vmid-/; | |
e4fc8228 | 245 | |
3fad2603 | 246 | $name = &$find_free_diskname($storeid, $scfg, $vmid) if !$name; |
be6339fc | 247 | |
76f60be8 | 248 | my $cmd = &$collie_cmd($scfg, 'vdi', 'create', $name , "${size}k"); |
be6339fc | 249 | |
e4fc8228 AD |
250 | run_command($cmd, errmsg => "sheepdog create $name' error"); |
251 | ||
252 | return $name; | |
253 | } | |
254 | ||
255 | sub free_image { | |
32437ed2 | 256 | my ($class, $storeid, $scfg, $volname, $isBase) = @_; |
e4fc8228 | 257 | |
b2f975cf AD |
258 | my ($vtype, $name, $vmid, undef, undef, undef) = |
259 | $class->parse_volname($volname); | |
260 | ||
261 | my $snapshots = sheepdog_snapshot_ls($scfg, $name); | |
3cb21703 | 262 | while (my ($snapname) = each %$snapshots) { |
b2f975cf AD |
263 | my $cmd = &$collie_cmd($scfg, 'vdi', 'delete' , '-s', $snapname, $name); |
264 | run_command($cmd, errmsg => "sheepdog delete snapshot $snapname $name' error"); | |
3cb21703 AD |
265 | } |
266 | ||
b2f975cf | 267 | my $cmd = &$collie_cmd($scfg, 'vdi', 'delete' , $name); |
e4fc8228 | 268 | |
b2f975cf | 269 | run_command($cmd, errmsg => "sheepdog delete $name' error"); |
e4fc8228 AD |
270 | |
271 | return undef; | |
272 | } | |
273 | ||
274 | sub list_images { | |
275 | my ($class, $storeid, $scfg, $vmid, $vollist, $cache) = @_; | |
276 | ||
277 | $cache->{sheepdog} = sheepdog_ls($scfg, $storeid) if !$cache->{sheepdog}; | |
278 | my $res = []; | |
279 | ||
280 | if (my $dat = $cache->{sheepdog}->{$storeid}) { | |
281 | foreach my $image (keys %$dat) { | |
282 | ||
283 | my $volname = $dat->{$image}->{name}; | |
b22af8ee AD |
284 | my $parent = $dat->{$image}->{parent}; |
285 | ||
286 | my $volid = undef; | |
287 | if ($parent && $parent ne $volname) { | |
288 | $volid = "$storeid:$parent/$volname"; | |
289 | } else { | |
290 | $volid = "$storeid:$volname"; | |
291 | } | |
e4fc8228 | 292 | |
e4fc8228 AD |
293 | my $owner = $dat->{$volname}->{vmid}; |
294 | if ($vollist) { | |
295 | my $found = grep { $_ eq $volid } @$vollist; | |
296 | next if !$found; | |
297 | } else { | |
298 | next if defined ($vmid) && ($owner ne $vmid); | |
299 | } | |
300 | ||
301 | my $info = $dat->{$volname}; | |
302 | $info->{volid} = $volid; | |
be6339fc | 303 | $info->{format} = 'raw'; |
e4fc8228 AD |
304 | push @$res, $info; |
305 | } | |
306 | } | |
7f8373d2 | 307 | |
e4fc8228 AD |
308 | return $res; |
309 | } | |
310 | ||
311 | ||
312 | sub status { | |
313 | my ($class, $storeid, $scfg, $cache) = @_; | |
314 | ||
315 | my $total = 0; | |
316 | my $free = 0; | |
317 | my $used = 0; | |
318 | my $active = 1; | |
e9256323 DM |
319 | |
320 | my $cmd = &$collie_cmd($scfg, 'node', 'info' , '-r'); | |
321 | ||
322 | my $parser = sub { | |
323 | my $line = shift; | |
324 | if ($line =~ m/^Total\s(\d+)\s(\d+)\s/) { | |
325 | $total = $1; | |
326 | $used = $2; | |
327 | $free = $total - $used; | |
328 | } | |
329 | }; | |
330 | ||
331 | run_command($cmd, outfunc => $parser, errmsg => "sheepdog node info error"); | |
332 | ||
e4fc8228 AD |
333 | return ($total,$free,$used,$active); |
334 | ||
335 | return undef; | |
336 | } | |
337 | ||
338 | sub activate_storage { | |
339 | my ($class, $storeid, $scfg, $cache) = @_; | |
340 | return 1; | |
341 | } | |
342 | ||
343 | sub deactivate_storage { | |
344 | my ($class, $storeid, $scfg, $cache) = @_; | |
345 | return 1; | |
346 | } | |
347 | ||
348 | sub activate_volume { | |
02e797b8 | 349 | my ($class, $storeid, $scfg, $volname, $snapname, $cache) = @_; |
e4fc8228 AD |
350 | return 1; |
351 | } | |
352 | ||
353 | sub deactivate_volume { | |
02e797b8 | 354 | my ($class, $storeid, $scfg, $volname, $snapname, $cache) = @_; |
e4fc8228 AD |
355 | return 1; |
356 | } | |
357 | ||
27923b43 AD |
358 | sub volume_size_info { |
359 | my ($class, $scfg, $storeid, $volname, $timeout) = @_; | |
360 | ||
61b8ccc9 AD |
361 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
362 | $class->parse_volname($volname); | |
363 | ||
c732d5db AD |
364 | my $size = undef; |
365 | ||
73b6f89b | 366 | my $cmd = &$collie_cmd($scfg, 'vdi', 'list', '-r'); |
c732d5db AD |
367 | |
368 | run_command($cmd, outfunc => sub { | |
369 | my $line = shift; | |
370 | $line = trim($line); | |
61b8ccc9 | 371 | if ($line =~ /(=|c) $name\s+(\d+)\s+(\d+)\s(\d+)\s(\d+)\s/) { |
73b6f89b | 372 | $size = $3; |
c732d5db AD |
373 | |
374 | } | |
375 | }); | |
376 | ||
377 | return $size; | |
27923b43 AD |
378 | } |
379 | ||
9ffffc2e AD |
380 | sub volume_resize { |
381 | my ($class, $scfg, $storeid, $volname, $size, $running) = @_; | |
382 | ||
e2e74320 AD |
383 | return 1 if $running; |
384 | ||
fddfc538 AD |
385 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
386 | $class->parse_volname($volname); | |
387 | ||
388 | my $cmd = &$collie_cmd($scfg, 'vdi', 'resize' , $name, $size); | |
389 | run_command($cmd, errmsg => "sheepdog resize $name' error"); | |
9ffffc2e AD |
390 | |
391 | return undef; | |
392 | } | |
393 | ||
50a19c09 | 394 | sub volume_snapshot { |
f5640e7d | 395 | my ($class, $scfg, $storeid, $volname, $snap) = @_; |
50a19c09 | 396 | |
d719effa AD |
397 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
398 | $class->parse_volname($volname); | |
399 | ||
400 | my $cmd = &$collie_cmd($scfg, 'vdi', 'snapshot', '-s', $snap, $name); | |
50a19c09 AD |
401 | run_command($cmd, errmsg => "sheepdog snapshot $volname' error"); |
402 | ||
403 | return undef; | |
404 | } | |
405 | ||
948b2e22 AD |
406 | sub volume_snapshot_rollback { |
407 | my ($class, $scfg, $storeid, $volname, $snap) = @_; | |
408 | ||
b8f4f5be AD |
409 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
410 | $class->parse_volname($volname); | |
411 | ||
76f60be8 | 412 | my $cmd = &$collie_cmd($scfg, 'vdi', 'rollback', '-f', '-s', $snap, $name); |
b8f4f5be | 413 | run_command($cmd, errmsg => "sheepdog snapshot $name' error"); |
948b2e22 AD |
414 | |
415 | } | |
416 | ||
9cd89ee3 AD |
417 | sub volume_snapshot_delete { |
418 | my ($class, $scfg, $storeid, $volname, $snap, $running) = @_; | |
419 | ||
420 | return 1 if $running; | |
421 | ||
399581a2 WB |
422 | $class->deactivate_volume($storeid, $scfg, $volname, $snap, {}); |
423 | ||
d99eee93 AD |
424 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
425 | $class->parse_volname($volname); | |
426 | ||
427 | my $cmd = &$collie_cmd($scfg, 'vdi', 'delete', '-s', $snap, $name); | |
428 | run_command($cmd, errmsg => "sheepdog snapshot $name' error"); | |
9cd89ee3 AD |
429 | |
430 | return undef; | |
431 | } | |
432 | ||
c0235499 AD |
433 | sub volume_has_feature { |
434 | my ($class, $scfg, $feature, $storeid, $volname, $snapname, $running) = @_; | |
435 | ||
75168177 | 436 | my $features = { |
c0235499 | 437 | snapshot => { current => 1, snap => 1}, |
75168177 | 438 | clone => { base => 1}, |
5649ccfe | 439 | template => { current => 1}, |
62ebea1e | 440 | copy => { base => 1, current => 1, snap => 1}, |
baafddbd | 441 | sparseinit => { base => 1, current => 1 }, |
c0235499 AD |
442 | }; |
443 | ||
75168177 AD |
444 | my ($vtype, $name, $vmid, $basename, $basevmid, $isBase) = |
445 | $class->parse_volname($volname); | |
446 | ||
447 | my $key = undef; | |
448 | if($snapname){ | |
2c5a7097 | 449 | $key = 'snap'; |
75168177 AD |
450 | }else{ |
451 | $key = $isBase ? 'base' : 'current'; | |
452 | } | |
453 | return 1 if $features->{$feature}->{$key}; | |
c0235499 AD |
454 | |
455 | return undef; | |
456 | } | |
457 | ||
e4fc8228 | 458 | 1; |