vol-upload-nonblock.pl 1.22 KB
Newer Older
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
#!/usr/bin/perl

use strict;
use warnings;

use Sys::Virt;

Sys::Virt::Event::register_default();

my $c = Sys::Virt->new(uri => "qemu:///session");

my $st = $c->new_stream(Sys::Virt::Stream::NONBLOCK);

die "syntax: $0 VOL-PATH INPUT-FILE" unless int(@ARGV) == 2;

my $volpath = shift @ARGV;
my $f = shift @ARGV;

my $vol = $c->get_storage_volume_by_path($volpath);

my $quit = 0;

open FILE, "<$f" or die "cannot open $f: $!";

my $nextdata;
my $nextdatalen;

sub moredata {
    return if $nextdatalen;
    $nextdatalen = sysread FILE, $nextdata, 1024;
}

sub streamevent {
    my $st = shift;
    my $events = shift;

    if ($events & (Sys::Virt::Stream::EVENT_HANGUP |
		   Sys::Virt::Stream::EVENT_ERROR)) {
	$quit = 1;
	return;
    }

    &moredata;
    if ($nextdatalen == 0) {
	$quit = 1;
	$st->remove_callback();
	return;
    }
    my $rv = $st->send($nextdata, $nextdatalen);

    if ($rv > 0) {
	$nextdata = substr $nextdata, $rv;
	$nextdatalen -= $rv;
    }
}

eval {
    $vol->upload($st, 0, 0);

    $st->add_callback(Sys::Virt::Stream::EVENT_WRITABLE, \&streamevent);

    while (!$quit) {
	Sys::Virt::Event::run_default();
    }
    $st->finish();
};
if ($@) {
    close FILE;
    die $@;
}
close FILE or die "cannot save $f: $!";