Ravada.pm 83.7 KB
Newer Older
1
2
3
4
5
package Ravada;

use warnings;
use strict;

Francesc Guasch's avatar
Francesc Guasch committed
6
our $VERSION = '0.3.0-beta6';
Francesc Guasch's avatar
Francesc Guasch committed
7

8
use Carp qw(carp croak);
9
use Data::Dumper;
10
use DBIx::Connector;
11
use File::Copy;
12
use Hash::Util qw(lock_hash);
13
use Moose;
14
use Parallel::ForkManager;
Francesc Guasch's avatar
Francesc Guasch committed
15
use POSIX qw(WNOHANG);
Francesc Guasch's avatar
Francesc Guasch committed
16
use Time::HiRes qw(gettimeofday tv_interval);
17
18
use YAML;

19
20
use Socket qw( inet_aton inet_ntoa );

Francesc Guasch's avatar
Francesc Guasch committed
21
22
23
no warnings "experimental::signatures";
use feature qw(signatures);

Francesc Guasch's avatar
Francesc Guasch committed
24
use Ravada::Auth;
25
use Ravada::Request;
26
use Ravada::VM::Void;
27

Francesc Guasch's avatar
Francesc Guasch committed
28
our %VALID_VM;
Francesc Guasch's avatar
Francesc Guasch committed
29
our %ERROR_VM;
Francesc Guasch's avatar
Francesc Guasch committed
30
31
32
33
34
35
36

eval {
    require Ravada::VM::KVM and do {
        Ravada::VM::KVM->import;
    };
    $VALID_VM{KVM} = 1;
};
Francesc Guasch's avatar
Francesc Guasch committed
37
38
39
40
41
42
43
44
45
$ERROR_VM{KVM} = $@;

eval {
    require Ravada::VM::Void and do {
        Ravada::VM::Void->import;
    };
    $VALID_VM{Void} = 1;
};
$ERROR_VM{Void} = $@;
Francesc Guasch's avatar
Francesc Guasch committed
46

47
48
49
no warnings "experimental::signatures";
use feature qw(signatures);

Francesc Guasch's avatar
Francesc Guasch committed
50
51
52
53
54
55
56
57
58
59
60
61
=head1 NAME

Ravada - Remove Virtual Desktop Manager

=head1 SYNOPSIS

  use Ravada;

  my $ravada = Ravada->new()

=cut

62
63

our $FILE_CONFIG = "/etc/ravada.conf";
64
$FILE_CONFIG = undef if ! -e $FILE_CONFIG;
65
66
67
68
69

###########################################################################

our $CONNECTOR;
our $CONFIG = {};
70
our $DEBUG;
Francesc Guasch's avatar
Francesc Guasch committed
71
our $VERBOSE;
72
our $CAN_FORK = 1;
73
our $CAN_LXC = 0;
74
75

# Seconds to wait for other long process
76
our $SECONDS_WAIT_CHILDREN = 5;
77

78
79
our $DIR_SQL = "sql/mysql";
$DIR_SQL = "/usr/share/doc/ravada/sql/mysql" if ! -e $DIR_SQL;
80

81
82
83
our $USER_DAEMON;
our $USER_DAEMON_NAME = 'daemon';

84
has 'connector' => (
85
86
87
88
89
90
        is => 'rw'
);

has 'config' => (
    is => 'ro'
    ,isa => 'Str'
91
92
);

93
94
95
96
97
98
has 'warn_error' => (
    is => 'rw'
    ,isa => 'Bool'
    ,default => sub { 1 }
);

Francesc Guasch's avatar
Francesc Guasch committed
99
100
101
102
103
104
105
=head2 BUILD

Internal constructor

=cut


106
107
sub BUILD {
    my $self = shift;
108
    if ($self->config()) {
109
        _init_config($self->config);
110
    } else {
111
        _init_config($FILE_CONFIG) if $FILE_CONFIG && -e $FILE_CONFIG;
112
    }
113

Francesc Guasch's avatar
Francesc Guasch committed
114
    if ( $self->connector ) {
joansp's avatar
joansp committed
115
        $CONNECTOR = $self->connector
Francesc Guasch's avatar
Francesc Guasch committed
116
117
    } else {
        $CONNECTOR = $self->_connect_dbh();
118
        $self->connector($CONNECTOR);
Francesc Guasch's avatar
Francesc Guasch committed
119
    }
Francesc Guasch's avatar
Francesc Guasch committed
120
    Ravada::Auth::init($CONFIG);
121

122
123
124
}

sub _install($self) {
125
    $self->_create_tables();
126
    $self->_upgrade_tables();
127
    $self->_update_data();
128
    $self->_init_user_daemon();
129
130
}

131
132
133
134
135
136
137
138
139
140
141
142
143
144
sub _init_user_daemon {
    my $self = shift;
    return if $USER_DAEMON;

    $USER_DAEMON = Ravada::Auth::SQL->new(name => $USER_DAEMON_NAME);
    if (!$USER_DAEMON->id) {
        $USER_DAEMON = Ravada::Auth::SQL::add_user(
            name => $USER_DAEMON_NAME,
            is_admin => 1
        );
        $USER_DAEMON = Ravada::Auth::SQL->new(name => $USER_DAEMON_NAME);
    }

}
145
146
sub _update_user_grants {
    my $self = shift;
Francesc Guasch's avatar
Francesc Guasch committed
147
    $self->_init_user_daemon();
148
149
150
    my $sth = $CONNECTOR->dbh->prepare("SELECT id FROM users");
    my $id;
    $sth->execute;
Francesc Guasch's avatar
Francesc Guasch committed
151
    $sth->bind_columns(\$id);
152
153
    while ($sth->fetch) {
        my $user = Ravada::Auth::SQL->search_by_id($id);
Francesc Guasch's avatar
Francesc Guasch committed
154
155
        next if $user->name() eq $USER_DAEMON_NAME;

156
        next if $user->grants();
157
158
        $USER_DAEMON->grant_user_permissions($user);
        $USER_DAEMON->grant_admin_permissions($user)    if $user->is_admin;
159
160
161
162
    }
    $sth->finish;
}

163
164
165
166
167
sub _update_isos {
    my $self = shift;
    my $table = 'iso_images';
    my $field = 'name';
    my %data = (
fv3rdugo's avatar
fv3rdugo committed
168
        mate_artful => {
Fernando Verdugo's avatar
Fernando Verdugo committed
169
                    name => 'Ubuntu Mate Artful'
fv3rdugo's avatar
fv3rdugo committed
170
171
172
173
174
175
            ,description => 'Ubuntu Mate 17.10.1 (Artful) 64 bits'
                   ,arch => 'amd64'
                    ,xml => 'yakkety64-amd64.xml'
             ,xml_volume => 'yakkety64-volume.xml'
                    ,url => 'http://cdimage.ubuntu.com/ubuntu-mate/releases/17.10.*/release/ubuntu-mate-17.10.*-desktop-amd64.iso'
                ,md5_url => '$url/MD5SUMS'
176
                ,min_disk_size => '10'
fv3rdugo's avatar
fv3rdugo committed
177
        },
Joel Alarcón's avatar
Joel Alarcón committed
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
        mate_bionic => {
                    name => 'Ubuntu Mate Bionic 64 bits'
            ,description => 'Ubuntu Mate 18.04 (Bionic Beaver) 64 bits'
                   ,arch => 'amd64'
                    ,xml => 'bionic-amd64.xml'
             ,xml_volume => 'bionic64-volume.xml'
                    ,url => 'http://cdimage.ubuntu.com/ubuntu-mate/releases/18.04.*/release/ubuntu-mate-18.04.*-desktop-amd64.iso'
                ,md5_url => '$url/MD5SUMS'
        },
        mate_bionic_i386 => {
                    name => 'Ubuntu Mate Bionic 32 bits'
            ,description => 'Ubuntu Mate 18.04 (Bionic Beaver) 32 bits'
                   ,arch => 'i386'
                    ,xml => 'bionic-i386.xml'
             ,xml_volume => 'bionic32-volume.xml'
                    ,url => 'http://cdimage.ubuntu.com/ubuntu-mate/releases/18.04.*/release/ubuntu-mate-18.04.*-desktop-i386.iso'
                ,md5_url => '$url/MD5SUMS'
        },

fv3rdugo's avatar
fv3rdugo committed
197
        mate_xenial => {
Fernando Verdugo's avatar
Fernando Verdugo committed
198
                    name => 'Ubuntu Mate Xenial'
fv3rdugo's avatar
fv3rdugo committed
199
200
201
202
203
204
            ,description => 'Ubuntu Mate 16.04.3 (Xenial) 64 bits'
                   ,arch => 'amd64'
                    ,xml => 'yakkety64-amd64.xml'
             ,xml_volume => 'yakkety64-volume.xml'
                    ,url => 'http://cdimage.ubuntu.com/ubuntu-mate/releases/16.04.*/release/ubuntu-mate-16.04.*-desktop-amd64.iso'
                ,md5_url => '$url/MD5SUMS'
205
                ,min_disk_size => '10'
fv3rdugo's avatar
fv3rdugo committed
206
        },
207
208
209
210
211
212
213
214
        alpine_37 => {
                    name => 'Alpine 3.7'
            ,description => 'Alpine Linux 3.7 64 bits ( Minimal Linux Distribution)'
                   ,arch => 'amd64'
                    ,xml => 'yakkety64-amd64.xml'
             ,xml_volume => 'yakkety64-volume.xml'
                    ,url => 'http://dl-cdn.alpinelinux.org/alpine/v3.7/releases/x86_64/'
                ,file_re => 'alpine-virt-3.7.\d+-x86_64.iso'
215
                ,sha256_url => 'http://dl-cdn.alpinelinux.org/alpine/v3.7/releases/x86_64/alpine-virt-3.7.\d+-x86_64.iso.sha256'
Francesc Guasch's avatar
Francesc Guasch committed
216
                ,min_disk_size => '1'
217
        }
Joel Alarcón's avatar
Joel Alarcón committed
218
219
220
221
222
223
224
        ,bionic=> {
                    name => 'Ubuntu Bionic Beaver'
            ,description => 'Ubuntu 18.04 Bionic Beaver 64 bits'
                   ,arch => 'amd64'
                    ,xml => 'bionic-amd64.xml'
             ,xml_volume => 'bionic64-volume.xml'
                    ,url => 'http://releases.ubuntu.com/18.04/'
225
                ,file_re => '^ubuntu-18.04.*desktop-amd64.iso'
Joel Alarcón's avatar
Joel Alarcón committed
226
227
228
229
                ,md5_url => '$url/MD5SUMS'
          ,min_disk_size => '9'
        }

230
        ,zesty => {
231
232
233
234
235
236
                    name => 'Ubuntu Zesty Zapus'
            ,description => 'Ubuntu 17.04 Zesty Zapus 64 bits'
                   ,arch => 'amd64'
                    ,xml => 'yakkety64-amd64.xml'
             ,xml_volume => 'yakkety64-volume.xml'
                    ,url => 'http://releases.ubuntu.com/17.04/'
joelalju's avatar
joelalju committed
237
238
                ,file_re => 'ubuntu-17.04.*desktop-amd64.iso'
                ,md5_url => 'http://releases.ubuntu.com/17.04/MD5SUMS'
239
                ,min_disk_size => '10'
240
        }
Francesc Guasch's avatar
Francesc Guasch committed
241
242
243
244
245
246
247
248
249
250
        ,serena64 => {
            name => 'Mint 18.1 Mate 64 bits'
    ,description => 'Mint Serena 18.1 with Mate Desktop based on Ubuntu Xenial 64 bits'
           ,arch => 'amd64'
            ,xml => 'xenial64-amd64.xml'
     ,xml_volume => 'xenial64-volume.xml'
            ,url => 'http://mirrors.evowise.com/linuxmint/stable/18.1/'
        ,file_re => 'linuxmint-18.1-mate-64bit.iso'
        ,md5_url => ''
            ,md5 => 'c5cf5c5d568e2dfeaf705cfa82996d93'
251
            ,min_disk_size => '10'
Francesc Guasch's avatar
Francesc Guasch committed
252
253

        }
254
255
256
257
        ,fedora_27 => {
            name => 'Fedora 27'
            ,description => 'RedHat Fedora 27 Workstation 64 bits'
            ,url => 'http://ftp.halifax.rwth-aachen.de/fedora/linux/releases/27/Workstation/x86_64/iso/Fedora-Workstation-netinst-x86_64-27-.*\.iso'
Francesc Guasch's avatar
Francesc Guasch committed
258
259
260
            ,arch => 'amd64'
            ,xml => 'xenial64-amd64.xml'
            ,xml_volume => 'xenial64-volume.xml'
261
            ,sha256_url => '$url/Fedora-Workstation-27-.*-x86_64-CHECKSUM'
262
            ,min_disk_size => '10'
Francesc Guasch's avatar
Francesc Guasch committed
263
        }
264
265
266
267
        ,fedora_28 => {
            name => 'Fedora 28'
            ,description => 'RedHat Fedora 28 Workstation 64 bits'
            ,url => 'http://ftp.halifax.rwth-aachen.de/fedora/linux/releases/28/Workstation/x86_64/iso/Fedora-Workstation-netinst-x86_64-28-.*\.iso'
Francesc Guasch's avatar
Francesc Guasch committed
268
269
270
            ,arch => 'amd64'
            ,xml => 'xenial64-amd64.xml'
            ,xml_volume => 'xenial64-volume.xml'
271
            ,sha256_url => 'http://fedora.mirrors.ovh.net/linux/releases/28/Workstation/x86_64/iso/Fedora-Workstation-28-.*-x86_64-CHECKSUM'
272
            ,min_disk_size => '10'
Francesc Guasch's avatar
Francesc Guasch committed
273
        }
Joel Alarcón's avatar
Joel Alarcón committed
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
        ,kubuntu_64 => {
            name => 'Kubuntu Bionic Beaver 64 bits'
            ,description => 'Kubuntu 18.04 Bionic Beaver 64 bits'
            ,arch => 'amd64'
            ,xml => 'bionic-amd64.xml'
            ,xml_volume => 'bionic64-volume.xml'
            ,md5_url => '$url/MD5SUMS'
            ,url => 'http://cdimage.ubuntu.com/kubuntu/releases/18.04/release/'
            ,file_re => 'kubuntu-18.04-desktop-amd64.iso'
            ,rename_file => 'kubuntu_bionic_64.iso'
        }
        ,kubuntu_32 => {
            name => 'Kubuntu Bionic Beaver 32 bits'
            ,description => 'Kubuntu 18.04 Bionic Beaver 32 bits'
            ,arch => 'i386'
            ,xml => 'bionic-i386.xml'
            ,xml_volume => 'bionic32-volume.xml'
            ,md5_url => '$url/MD5SUMS'
            ,url => 'http://cdimage.ubuntu.com/kubuntu/releases/18.04/release/'
            ,file_re => 'kubuntu-18.04-desktop-i386.iso'
            ,rename_file => 'kubuntu_bionic_32.iso'
        }
296
297
298
299
300
301
302
        ,suse_15 => {
            name => "openSUSE Leap 15"
            ,description => "openSUSE Leap 15 64 bits"
            ,arch => 'amd64'
            ,xml => 'bionic-amd64.xml'
            ,xml_volume => 'bionic64-volume.xml'
            ,url => 'https://download.opensuse.org/distribution/leap/15.0/iso/'
303
            ,sha256_url => '$url/openSUSE-Leap-15.\d+-NET-x86_64.iso.sha256'
304
305
306
            ,file_re => 'openSUSE-Leap-15.\d+-NET-x86_64.iso'

        }
Joel Alarcón's avatar
Joel Alarcón committed
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
        ,xubuntu_beaver_64 => {
            name => 'Xubuntu Bionic Beaver 64 bits'
            ,description => 'Xubuntu 18.04 Bionic Beaver 64 bits'
            ,arch => 'amd64'
            ,xml => 'bionic-amd64.xml'
            ,xml_volume => 'bionic64-volume.xml'
            ,md5_url => '$url/../MD5SUMS'
            ,url => 'http://archive.ubuntu.com/ubuntu/dists/bionic/main/installer-amd64/current/images/netboot/'
            ,file_re => 'mini.iso'
            ,rename_file => 'xubuntu_bionic_64.iso'
        }
        ,xubuntu_beaver_32 => {
            name => 'Xubuntu Bionic Beaver 32 bits'
            ,description => 'Xubuntu 18.04 Bionic Beaver 32 bits'
            ,arch => 'amd64'
            ,xml => 'bionic-i386.xml'
            ,xml_volume => 'bionic32-volume.xml'
            ,md5_url => '$url/../MD5SUMS'
            ,url => 'http://archive.ubuntu.com/ubuntu/dists/bionic/main/installer-i386/current/images/netboot/'
            ,file_re => 'mini.iso'
            ,rename_file => 'xubuntu_bionic_32.iso'
        }
329
330
331
332
333
334
        ,xubuntu_artful => {
            name => 'Xubuntu Artful Aardvark'
            ,description => 'Xubuntu 17.10 Artful Aardvark 64 bits'
            ,arch => 'amd64'
            ,xml => 'yakkety64-amd64.xml'
            ,xml_volume => 'yakkety64-volume.xml'
Francesc Guasch's avatar
Francesc Guasch committed
335
336
337
            ,md5_url => '$url/../MD5SUMS'
            ,url => 'http://archive.ubuntu.com/ubuntu/dists/artful/main/installer-amd64/current/images/netboot/'
            ,file_re => 'mini.iso'
338
            ,rename_file => 'xubuntu_artful.iso'
339
            ,min_disk_size => '10'
340
        }
Francesc Guasch's avatar
Francesc Guasch committed
341
342
343
344
345
346
        ,xubuntu_zesty => {
            name => 'Xubuntu Zesty Zapus'
            ,description => 'Xubuntu 17.04 Zesty Zapus 64 bits'
            ,arch => 'amd64'
            ,xml => 'yakkety64-amd64.xml'
            ,xml_volume => 'yakkety64-volume.xml'
Francesc Guasch's avatar
Francesc Guasch committed
347
348
349
            ,md5_url => '$url/../MD5SUMS'
            ,url => 'http://archive.ubuntu.com/ubuntu/dists/zesty/main/installer-amd64/current/images/netboot'
            ,file_re => 'mini.iso'
350
            ,rename_file => 'xubuntu_zesty_mini.iso'
351
            ,min_disk_size => '10'
Francesc Guasch's avatar
Francesc Guasch committed
352
353
354
        }
        ,xubuntu_xenial => {
            name => 'Xubuntu Xenial Xerus'
355
            ,description => 'Xubuntu 16.04 Xenial Xerus 64 bits (LTS)'
Francesc Guasch's avatar
Francesc Guasch committed
356
357
358
359
            ,url => 'http://archive.ubuntu.com/ubuntu/dists/xenial/main/installer-amd64/current/images/netboot/mini.iso'
           ,xml => 'yakkety64-amd64.xml'
            ,xml_volume => 'yakkety64-volume.xml'
            ,md5 => 'fe495d34188a9568c8d166efc5898d22'
360
            ,rename_file => 'xubuntu_xenial_mini.iso'
361
            ,min_disk_size => '10'
Francesc Guasch's avatar
Francesc Guasch committed
362
        }
Joel Alarcón's avatar
Joel Alarcón committed
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
        ,lubuntu_bionic_64 => {
             name => 'Lubuntu Bionic Beaver 64 bits'
             ,description => 'Lubuntu 18.04 Bionic Beaver 64 bits'
             ,url => 'http://cdimage.ubuntu.com/lubuntu/releases/18.04.*/release/lubuntu-18.04.*-desktop-amd64.iso'
             ,md5_url => '$url/MD5SUMS'
             ,xml => 'bionic-amd64.xml'
             ,xml_volume => 'bionic64-volume.xml'
         }
         ,lubuntu_bionic_32 => {
             name => 'Lubuntu Bionic Beaver 32 bits'
             ,description => 'Lubuntu 18.04 Bionic Beaver 32 bits'
             ,arch => 'i386'
             ,url => 'http://cdimage.ubuntu.com/lubuntu/releases/18.04.*/release/lubuntu-18.04.*-desktop-i386.iso'
             ,md5_url => '$url/MD5SUMS'
             ,xml => 'bionic-i386.xml'
             ,xml_volume => 'bionic32-volume.xml'
        }
380
381
382
        ,lubuntu_xenial => {
            name => 'Lubuntu Xenial Xerus'
            ,description => 'Xubuntu 16.04 Xenial Xerus 64 bits (LTS)'
383
384
385
            ,url => 'http://cdimage.ubuntu.com/lubuntu/releases/16.04.*/release/'
            ,file_re => 'lubuntu-16.04.*-desktop-amd64.iso'
            ,md5_url => '$url/MD5SUMS'
386
387
            ,xml => 'yakkety64-amd64.xml'
            ,xml_volume => 'yakkety64-volume.xml'
388
            ,min_disk_size => '10'
389
        }
390
391
392
393
        ,debian_jessie_32 => {
            name =>'Debian Jessie 32 bits'
            ,description => 'Debian 8 Jessie 32 bits'
            ,url => 'http://cdimage.debian.org/cdimage/archive/^8\..*/i386/iso-cd/'
394
            ,file_re => 'debian-8.[\d\.]+-i386-xfce-CD-1.iso'
395
            ,md5_url => '$url/MD5SUMS'
396
            ,xml => 'jessie-i386.xml'
397
            ,xml_volume => 'jessie-volume.xml'
398
            ,min_disk_size => '10'
399
400
401
402
403
404
405
406
407
        }
        ,debian_jessie_64 => {
            name =>'Debian Jessie 64 bits'
            ,description => 'Debian 8 Jessie 64 bits'
            ,url => 'http://cdimage.debian.org/cdimage/archive/^8\..*/amd64/iso-cd/'
            ,file_re => 'debian-8.[\d\.]+-amd64-xfce-CD-1.iso'
            ,md5_url => '$url/MD5SUMS'
            ,xml => 'jessie-amd64.xml'
            ,xml_volume => 'jessie-volume.xml'
408
            ,min_disk_size => '10'
409
        }
410
        ,debian_stretch => {
411
            name =>'Debian Stretch 64 bits'
Francesc Guasch's avatar
Francesc Guasch committed
412
            ,description => 'Debian 9 Stretch 64 bits (XFCE desktop)'
413
            ,url => 'https://cdimage.debian.org/debian-cd/^9\..*/amd64/iso-cd/'
414
            ,file_re => 'debian-9.[\d\.]+-amd64-xfce-CD-1.iso'
415
            ,md5_url => '$url/MD5SUMS'
416
417
            ,xml => 'jessie-amd64.xml'
            ,xml_volume => 'jessie-volume.xml'
418
            ,min_disk_size => '10'
419
        }
joelalju's avatar
joelalju committed
420
        ,windows_7 => {
Francesc Guasch's avatar
Francesc Guasch committed
421
          name => 'Windows 7'
422
423
          ,description => 'Windows 7 64 bits. Requires an user provided ISO image.'
            .'<a target="_blank" href="http://ravada.readthedocs.io/en/latest/docs/new_iso_image.html">[help]</a>'
joelalju's avatar
joelalju committed
424
          ,xml => 'windows_7.xml'
joelalju's avatar
joelalju committed
425
          ,xml_volume => 'wisuvolume.xml'
426
          ,min_disk_size => '21'
joelalju's avatar
joelalju committed
427
428
429
430
431
432
        }
        ,windows_10 => {
          name => 'Windows 10'
          ,description => 'Windows 10 64 bits. Requires an user provided ISO image.'
          .'<a target="_blank" href="http://ravada.readthedocs.io/en/latest/docs/new_iso_image.html">[help]</a>'
          ,xml => 'windows_10.xml'
433
          ,xml_volume => 'windows10-volume.xml'
434
          ,min_disk_size => '21'
joelalju's avatar
joelalju committed
435
        }
joelalju's avatar
joelalju committed
436
437
438
439
440
441
        ,windows_xp => {
          name => 'Windows XP'
          ,description => 'Windows XP 64 bits. Requires an user provided ISO image.'
          .'<a target="_blank" href="http://ravada.readthedocs.io/en/latest/docs/new_iso_image.html">[help]</a>'
          ,xml => 'windows_xp.xml'
          ,xml_volume => 'wisuvolume.xml'
442
          ,min_disk_size => '3'
joelalju's avatar
joelalju committed
443
444
445
446
447
448
449
        }
        ,windows_12 => {
          name => 'Windows 2012'
          ,description => 'Windows 2012 64 bits. Requires an user provided ISO image.'
          .'<a target="_blank" href="http://ravada.readthedocs.io/en/latest/docs/new_iso_image.html">[help]</a>'
          ,xml => 'windows_12.xml'
          ,xml_volume => 'wisuvolume.xml'
450
          ,min_disk_size => '21'
joelalju's avatar
joelalju committed
451
452
453
454
455
456
457
        }
        ,windows_8 => {
          name => 'Windows 8.1'
          ,description => 'Windows 8.1 64 bits. Requires an user provided ISO image.'
          .'<a target="_blank" href="http://ravada.readthedocs.io/en/latest/docs/new_iso_image.html">[help]</a>'
          ,xml => 'windows_8.xml'
          ,xml_volume => 'wisuvolume.xml'
458
          ,min_disk_size => '21'
joelalju's avatar
joelalju committed
459
        }
460
461
    );

462
463
464
465
466
467
468
469
470
471
472
    $self->_update_table($table, $field, \%data);

}

sub _update_domain_drivers_types($self) {

    my $data = {
        image => {
            id => 4,
            ,name => 'image'
           ,description => 'Graphics Options'
473
           ,vm => 'KVM'
474
475
476
477
478
        },
        jpeg => {
            id => 5,
            ,name => 'jpeg'
           ,description => 'Graphics Options'
479
           ,vm => 'KVM'
480
481
482
483
484
        },
        zlib => {
            id => 6,
            ,name => 'zlib'
           ,description => 'Graphics Options'
485
           ,vm => 'KVM'
486
487
488
489
490
        },
        playback => {
            id => 7,
            ,name => 'playback'
           ,description => 'Graphics Options'
491
           ,vm => 'KVM'
492
493
494
495
496
497

        },
        streaming => {
            id => 8,
            ,name => 'streaming'
           ,description => 'Graphics Options'
498
           ,vm => 'KVM'
499
500
501
502

        }
    };
    $self->_update_table('domain_drivers_types','id',$data);
503
504
505
506
507
508

    my $sth = $CONNECTOR->dbh->prepare(
        "UPDATE domain_drivers_types SET vm='KVM' WHERE vm='qemu'"
    );
    $sth->execute;
    $sth->finish;
509
510
}

511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
sub _update_domain_drivers_options($self) {

    my $data = {
        qxl => {
            id => 1,
            ,id_driver_type => 1,
            ,name => 'QXL'
           ,value => 'type="qxl" ram="65536" vram="65536" vgamem="16384" heads="1" primary="yes"'
        },
        vmvga => {
            id => 2,
            ,id_driver_type => 1,
            ,name => 'VMVGA'
           ,value => 'type="vmvga" vram="16384" heads="1" primary="yes"'
        },
        cirrus => {
            id => 3,
            ,id_driver_type => 1,
            ,name => 'Cirrus'
           ,value => 'type="cirrus" vram="16384" heads="1" primary="yes"'
        },
        vga => {
            id => 4,
            ,id_driver_type => 1,
            ,name => 'VGA'
           ,value => 'type="vga" vram="16384" heads="1" primary="yes"'
        },
        ich6 => {
            id => 6,
            ,id_driver_type => 2,
            ,name => 'ich6'
           ,value => 'model="ich6"'
        },
        ac97 => {
            id => 7,
            ,id_driver_type => 2,
            ,name => 'ac97'
           ,value => 'model="ac97"'
        },
        virtio => {
            id => 8,
            ,id_driver_type => 3,
            ,name => 'virtio'
           ,value => 'type="virtio"'
        },
        e1000 => {
            id => 9,
            ,id_driver_type => 3,
            ,name => 'e1000'
           ,value => 'type="e1000"'
        },
        rtl8139 => {
            id => 10,
            ,id_driver_type => 3,
            ,name => 'rtl8139'
           ,value => 'type="rtl8139"'
        },
        auto_glz => {
            id => 11,
            ,id_driver_type => 4,
            ,name => 'auto_glz'
           ,value => 'compression="auto_glz"'
        },
        auto_lz => {
            id => 12,
            ,id_driver_type => 4,
            ,name => 'auto_lz'
           ,value => 'compression="auto_lz"'
        },
        quic => {
            id => 13,
            ,id_driver_type => 4,
            ,name => 'quic'
           ,value => 'compression="quic"'
        },
        glz => {
            id => 14,
            ,id_driver_type => 4,
            ,name => 'glz'
           ,value => 'compression="glz"'
        },
        lz => {
            id => 15,
            ,id_driver_type => 4,
            ,name => 'lz'
           ,value => 'compression="lz"'
        },
        off => {
            id => 16,
            ,id_driver_type => 4,
            ,name => 'off'
           ,value => 'compression="off"'
        },
        auto => {
            id => 17,
            ,id_driver_type => 5,
            ,name => 'auto'
           ,value => 'compression="auto"'
        },
        never => {
            id => 18,
            ,id_driver_type => 5,
            ,name => 'never'
           ,value => 'compression="never"'
        },
        always => {
            id => 19,
            ,id_driver_type => 5,
            ,name => 'always'
           ,value => 'compression="always"'
        },
        auto1 => {
            id => 20,
            ,id_driver_type => 6,
            ,name => 'auto'
           ,value => 'compression="auto"'
        },
        never1 => {
            id => 21,
            ,id_driver_type => 6,
            ,name => 'never'
           ,value => 'compression="never"'
        },
        always1 => {
            id => 22,
            ,id_driver_type => 6,
            ,name => 'always'
           ,value => 'compression="always"'
        },
        on => {
            id => 23,
            ,id_driver_type => 7,
            ,name => 'on'
           ,value => 'compression="on"'
        },
        off1 => {
            id => 24,
            ,id_driver_type => 7,
            ,name => 'off'
           ,value => 'compression="off"'
        },
        filter => {
            id => 25,
            ,id_driver_type => 8,
            ,name => 'filter'
           ,value => 'mode="filter"'
        },
        all => {
            id => 26,
            ,id_driver_type => 8,
            ,name => 'all'
           ,value => 'mode="all"'
        },
        off2 => {
            id => 27,
            ,id_driver_type => 8,
            ,name => 'off'
           ,value => 'mode="off"'
        }
    };
    $self->_update_table('domain_drivers_options','id',$data);
}

674
675
sub _update_table($self, $table, $field, $data) {

676
    my $sth_search = $CONNECTOR->dbh->prepare("SELECT id FROM $table WHERE $field = ?");
677
678
    for my $name (keys %$data) {
        my $row = $data->{$name};
679
680
681
        $sth_search->execute($row->{$field});
        my ($id) = $sth_search->fetchrow;
        next if $id;
Francesc Guasch's avatar
Francesc Guasch committed
682
        warn("INFO: updating $table : $row->{$field}\n")    if $0 !~ /\.t$/;
683
684

        my $sql =
685
            "INSERT INTO $table "
686
            ."("
687
            .join(" , ", sort keys %{$data->{$name}})
688
689
            .")"
            ." VALUES ( "
690
            .join(" , ", map { "?" } keys %{$data->{$name}})
691
692
693
            ." )"
        ;
        my $sth = $CONNECTOR->dbh->prepare($sql);
694
        $sth->execute(map { $data->{$name}->{$_} } sort keys %{$data->{$name}});
695
696
697
698
        $sth->finish;
    }
}

699
700
sub _remove_old_isos {
    my $self = shift;
701
702
703
704
705
706
    for my $sql (
        "DELETE FROM iso_images "
            ."  WHERE url like '%debian-9.0%iso'"
        ,"DELETE FROM iso_images"
            ."  WHERE name like 'Debian%' "
            ."      AND NOT url  like '%*%' "
707
708
709
710
711
712
        ,"DELETE FROM iso_images "
            ."  WHERE name like 'Lubuntu Artful%'"
            ."      AND url NOT LIKE '%*%' "
        ,"DELETE FROM iso_images "
            ."  WHERE name like 'Lubuntu Zesty%'"

713
714
715
716
717
    ) {
        my $sth = $CONNECTOR->dbh->prepare($sql);
        $sth->execute();
        $sth->finish;
    }
718
719
}

720
721
sub _update_data {
    my $self = shift;
722

723
    $self->_remove_old_isos();
724
    $self->_update_isos();
725

726
727
728
    $self->_rename_grants();
    $self->_alias_grants();
    $self->_add_grants();
729
    $self->_enable_grants();
730
    $self->_update_user_grants();
731

732
    $self->_update_domain_drivers_types();
733
    $self->_update_domain_drivers_options();
Francesc Guasch's avatar
Francesc Guasch committed
734
    $self->_update_old_qemus();
735

736
737
738
739
740
    $self->_add_indexes();
}

sub _add_indexes($self) {
    return if $CONNECTOR->dbh->{Driver}{Name} !~ /mysql/i;
741
    $self->_add_indexes_vms();
742
743
744
745
    $self->_add_indexes_domains();
    $self->_add_indexes_requests();
}

746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
sub _add_indexes_vms($self) {
    my %index;
    my $sth = $CONNECTOR->dbh->prepare("show index from vms");
    $sth->execute;
    while (my $row = $sth->fetchrow_hashref) {
        $index{$row->{Key_name}}->{$row->{Column_name}}++;
    }

    my $index_name = 'hostname_vm_type';
    return if $index{$index_name};
    warn "INFO: Adding index to vms: $index_name";
    $sth = $CONNECTOR->dbh->prepare("ALTER TABLE vms add unique $index_name"
        ." (hostname, vm_type)");
    $sth->execute;
}

762
763
764
765
766
767
768
769
770
771
772
773
774
sub _add_indexes_domains($self) {
    my %index;
    my $sth = $CONNECTOR->dbh->prepare("show index from domains");
    $sth->execute;
    while (my $row = $sth->fetchrow_hashref) {
        $index{$row->{Key_name}}->{$row->{Column_name}}++;
    }
    return if $index{id_base_index};
    warn "INFO: Adding domains . id_base index";
    $sth = $CONNECTOR->dbh->prepare("ALTER TABLE domains add index id_base_index "
        ."(id_base)");
    $sth->execute;
}
775

776
777
778
779
780
781
782
783
784
785
786
787
sub _add_indexes_requests($self) {
    my %index;
    my $sth = $CONNECTOR->dbh->prepare("show index from requests");
    $sth->execute;
    while (my $row = $sth->fetchrow_hashref) {
        $index{$row->{Key_name}}->{$row->{Column_name}}++;
    }
    return if $index{domain_status};
    warn "INFO: Adding requests . id_domain,status index";
    $sth = $CONNECTOR->dbh->prepare("ALTER TABLE requests add index domain_status "
        ."(id_domain, status)");
    $sth->execute;
788
789
}

790
sub _rename_grants($self) {
Francesc Guasch's avatar
Francesc Guasch committed
791
792
793

    my %rename = (
        create_domain => 'create_machine'
794
795
796
    );
    my $sth_old = $CONNECTOR->dbh->prepare("SELECT id FROM grant_types"
            ." WHERE name=?"
Francesc Guasch's avatar
Francesc Guasch committed
797
798
    );
    for my $old ( keys %rename ) {
799
800
801
        $sth_old->execute($rename{$old});
        next if $sth_old->fetchrow;

Francesc Guasch's avatar
Francesc Guasch committed
802
        my $sth = $CONNECTOR->dbh->prepare(
803
                 "UPDATE grant_types"
Francesc Guasch's avatar
Francesc Guasch committed
804
805
806
807
                ." SET name=? "
                ." WHERE name = ?"
        );
        $sth->execute($rename{$old}, $old);
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
        warn "INFO: renaming grant $old to $rename{$old}\n";
    }
}

sub _alias_grants($self) {

    my %alias= (
        remove_clone => 'remove_clones'
        ,shutdown_clone => 'shutdown_clones'
    );

    my $sth_old = $CONNECTOR->dbh->prepare("SELECT id FROM grant_types_alias"
            ." WHERE name=? AND alias=?"
    );
    while (my ($old, $new) =  each(%alias)) {
        $sth_old->execute($old, $new);
        return if $sth_old->fetch;
        my $sth = $CONNECTOR->dbh->prepare(
                 "INSERT INTO grant_types_alias (name,alias)"
                 ." VALUES(?,?) "
        );
        $sth->execute($old, $new);
Francesc Guasch's avatar
Francesc Guasch committed
830
    }
831
}
Francesc Guasch's avatar
Francesc Guasch committed
832

833
sub _add_grants($self) {
Francesc Guasch's avatar
Francesc Guasch committed
834
    $self->_add_grant('shutdown', 1);
Francesc Guasch's avatar
Francesc Guasch committed
835
    $self->_add_grant('screenshot', 1);
Francesc Guasch's avatar
Francesc Guasch committed
836
837
838
839
840
841
}

sub _add_grant($self, $grant, $allowed) {

    my $sth = $CONNECTOR->dbh->prepare(
        "SELECT id FROM grant_types WHERE name=?"
842
    );
Francesc Guasch's avatar
Francesc Guasch committed
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
    $sth->execute($grant);
    my ($id) = $sth->fetchrow();
    $sth->finish;

    return if $id;

    $sth = $CONNECTOR->dbh->prepare("INSERT INTO grant_types (name, description)"
        ." VALUES (?,?)");
    $sth->execute($grant,"can shutdown any virtual machine owned by the user");
    $sth->finish;

    return if !$allowed;

    $sth = $CONNECTOR->dbh->prepare("SELECT id FROM grant_types WHERE name=?");
    $sth->execute($grant);
    my ($id_grant) = $sth->fetchrow;
    $sth->finish;

    my $sth_insert = $CONNECTOR->dbh->prepare(
        "INSERT INTO grants_user (id_user, id_grant, allowed) VALUES(?,?,?) ");

    $sth = $CONNECTOR->dbh->prepare("SELECT id FROM users ");
    $sth->execute;

    while (my ($id_user) = $sth->fetchrow ) {
        eval { $sth_insert->execute($id_user, $id_grant, $allowed) };
        die $@ if $@ && $@ !~/Duplicate entry /;
    }
871
872
}

873
874
875
876
877
878
879
880
881
882
883
sub _null_grants($self) {
    my $sth = $CONNECTOR->dbh->prepare("SELECT count(*) FROM grant_types "
            ." WHERE enabled = NULL "
        );
    $sth->execute;
    my ($count) = $sth->fetchrow;

    exit if !$count && $self->{_null}++;
    return $count;
}

884
sub _enable_grants($self) {
885
886
887

    return if $self->_null_grants();

888
889
890
    my $sth = $CONNECTOR->dbh->prepare(
        "UPDATE grant_types set enabled=0"
    );
891
    $sth->execute;
892
    my @grants = (
893
894
        'change_settings',  'change_settings_all',  'change_settings_clones'
        ,'clone',           'clone_all',            'create_base', 'create_machine'
895
        ,'grant'
896
        ,'manage_users'
897
        ,'remove',          'remove_all',   'remove_clone',     'remove_clone_all'
Francesc Guasch's avatar
Francesc Guasch committed
898
        ,'screenshot'
899
        ,'shutdown',        'shutdown_all',    'shutdown_clone'
Francesc Guasch's avatar
Francesc Guasch committed
900
        ,'screenshot'
901
902
903
904
905
906
907
908
909
910
911
912
913
    );

    $sth = $CONNECTOR->dbh->prepare("SELECT id,name FROM grant_types");
    $sth->execute;
    my %grant_exists;
    while (my ($id, $name) = $sth->fetchrow ) {
        $grant_exists{$name} = $id;
    }

    $sth = $CONNECTOR->dbh->prepare(
        "UPDATE grant_types set enabled=1 WHERE name=?"
    );
    my %done;
914
    for my $name ( sort @grants ) {
915
916
917
918
919
920
921
922
        die "Duplicate grant $name "    if $done{$name};
        die "Permission $name doesn't exist at table grant_types"
                ."\n".Dumper(\%grant_exists)
            if !$grant_exists{$name};

        $sth->execute($name);

    }
923

Francesc Guasch's avatar
Francesc Guasch committed
924
925
926
927
928
929
930
931
}

sub _update_old_qemus($self) {
    my $sth = $CONNECTOR->dbh->prepare("UPDATE vms SET vm_type='KVM'"
        ." WHERE vm_type='qemu' AND name ='KVM_localhost'"
    );
    $sth->execute;

932
933
}

934
935
sub _set_url_isos($self, $new_url='http://localhost/iso/') {
    $new_url .= '/' if $new_url !~ m{/$};
936
937
938
939
940
941
942
943
944
    my $sth = $CONNECTOR->dbh->prepare(
        "SELECT id,url FROM iso_images "
        ."WHERE url is NOT NULL"
    );
    my $sth_update = $CONNECTOR->dbh->prepare(
        "UPDATE iso_images set url=? WHERE id=?"
    );
    $sth->execute();
    while ( my ($id, $url) = $sth->fetchrow) {
945
        $url =~ s{\w+://(.*?)/(.*)}{$new_url$2};
946
947
948
949
950
        $sth_update->execute($url, $id);
    }
    $sth->finish;

}
951
952
953
954
955
956
957
958
959
960
sub _upgrade_table {
    my $self = shift;
    my ($table, $field, $definition) = @_;
    my $dbh = $CONNECTOR->dbh;

    my $sth = $dbh->column_info(undef,undef,$table,$field);
    my $row = $sth->fetchrow_hashref;
    $sth->finish;
    return if $row;

Francesc Guasch's avatar
Francesc Guasch committed
961
    warn "INFO: adding $field $definition to $table\n"  if $0 !~ /\.t$/;
962
    $dbh->do("alter table $table add $field $definition");
963
    return 1;
964
965
}

966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
sub _remove_field {
    my $self = shift;
    my ($table, $field ) = @_;

    my $dbh = $CONNECTOR->dbh;
    return if $CONNECTOR->dbh->{Driver}{Name} !~ /mysql/i;

    my $sth = $dbh->column_info(undef,undef,$table,$field);
    my $row = $sth->fetchrow_hashref;
    $sth->finish;
    return if !$row;

    warn "INFO: removing $field to $table\n"  if $0 !~ /\.t$/;
    $dbh->do("alter table $table drop column $field");
    return 1;

}

984
985
986
987
988
989
990
991
992
sub _create_table {
    my $self = shift;
    my $table = shift;

    my $sth = $CONNECTOR->dbh->table_info('%',undef,$table,'TABLE');
    my $info = $sth->fetchrow_hashref();
    $sth->finish;
    return if keys %$info;

993
    warn "INFO: creating table $table\n"    if $0 !~ /\.t$/;
994

995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
    my $file_sql = "$DIR_SQL/$table.sql";
    open my $in,'<',$file_sql or die "$! $file_sql";
    my $sql = join " ",<$in>;
    close $in;

    $CONNECTOR->dbh->do($sql);
    return 1;
}

sub _insert_data {
    my $self = shift;
    my $table = shift;

    my $file_sql =  "$DIR_SQL/../data/insert_$table.sql";
    return if ! -e $file_sql;

    warn "INFO: inserting data for $table\n";
    open my $in,'<',$file_sql or die "$! $file_sql";
    my $sql = '';
    while (my $line = <$in>) {
        $sql .= $line;
        next if $sql !~ /\w/ || $sql !~ /;\s*$/;
        $CONNECTOR->dbh->do($sql);
        $sql = '';
    }
    close $in;

}

sub _create_tables {
    my $self = shift;
1026
1027
1028
1029
#    return if $CONNECTOR->dbh->{Driver}{Name} !~ /mysql/i;

    my $driver = lc($CONNECTOR->dbh->{Driver}{Name});
    $DIR_SQL =~ s{(.*)/.*}{$1/$driver};
1030

1031
1032
1033
1034
    opendir my $ls,$DIR_SQL or die "$! $DIR_SQL";
    while (my $file = readdir $ls) {
        my ($table) = $file =~ m{(.*)\.sql$};
        next if !$table;
1035
        next if $table =~ /^insert/;
1036
1037
1038
1039
1040
        $self->_insert_data($table)     if $self->_create_table($table);
    }
    closedir $ls;
}

1041
1042
1043
1044
sub _clean_iso_mini {
    my $sth = $CONNECTOR->dbh->prepare("DELETE FROM iso_images WHERE device like ?");
    $sth->execute('%/mini.iso');
    $sth->finish;
1045
1046
1047
1048

    $sth = $CONNECTOR->dbh->prepare("DELETE FROM iso_images WHERE url like ? AND rename_file = NULL");
    $sth->execute('%/mini.iso');
    $sth->finish;
1049
1050
}

1051
1052
sub _upgrade_tables {
    my $self = shift;
Francesc Guasch's avatar
Francesc Guasch committed
1053
#    return if $CONNECTOR->dbh->{Driver}{Name} !~ /mysql/i;
1054

1055
    $self->_upgrade_table('file_base_images','target','varchar(64) DEFAULT NULL');
1056

1057
    $self->_upgrade_table('vms','vm_type',"char(20) NOT NULL DEFAULT 'KVM'");
1058
    $self->_upgrade_table('vms','connection_args',"text DEFAULT NULL");
1059
1060
    $self->_upgrade_table('vms','cached_active_time',"integer DEFAULT 0");
    $self->_upgrade_table('vms','public_ip',"varchar(128) DEFAULT NULL");
Francesc Guasch's avatar
Francesc Guasch committed
1061
    $self->_upgrade_table('vms','is_active',"int DEFAULT 0");
1062
    $self->_upgrade_table('vms','enabled',"int DEFAULT 1");
1063

1064
    $self->_upgrade_table('vms','min_free_memory',"text DEFAULT NULL");
Francesc Guasch's avatar
Francesc Guasch committed
1065
1066
    $self->_upgrade_table('vms', 'max_load', 'int not null default 10');
    $self->_upgrade_table('vms', 'active_limit','int DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1067
1068
    $self->_upgrade_table('vms', 'base_storage','varchar(64) DEFAULT NULL');
    $self->_upgrade_table('vms', 'clone_storage','varchar(64) DEFAULT NULL');
1069

1070
    $self->_upgrade_table('requests','at_time','int(11) DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1071
1072
    $self->_upgrade_table('requests','pid','int(11) DEFAULT NULL');
    $self->_upgrade_table('requests','start_time','int(11) DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1073

1074
    $self->_upgrade_table('requests','at_time','int(11) DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1075
    $self->_upgrade_table('requests','run_time','float DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1076

1077
    $self->_upgrade_table('iso_images','rename_file','varchar(80) DEFAULT NULL');
1078
    $self->_clean_iso_mini();
1079
    $self->_upgrade_table('iso_images','md5_url','varchar(255)');
Francesc Guasch's avatar
Francesc Guasch committed
1080
1081
    $self->_upgrade_table('iso_images','sha256','varchar(255)');
    $self->_upgrade_table('iso_images','sha256_url','varchar(255)');
1082
    $self->_upgrade_table('iso_images','file_re','char(64)');
1083
    $self->_upgrade_table('iso_images','device','varchar(255)');
joelalju's avatar
joelalju committed
1084
    $self->_upgrade_table('iso_images','min_disk_size','int (11) DEFAULT NULL');
1085
1086

    $self->_upgrade_table('users','language','char(3) DEFAULT NULL');
1087
1088
1089
1090
1091
1092
    if ( $self->_upgrade_table('users','is_external','int(11) DEFAULT 0')) {
        my $sth = $CONNECTOR->dbh->prepare(
            "UPDATE users set is_external=1 WHERE password='*LK* no pss'"
        );
        $sth->execute;
    }
Francesc Guasch's avatar
Francesc Guasch committed
1093

1094
    $self->_upgrade_table('networks','requires_password','int(11)');
1095
1096
    $self->_upgrade_table('networks','n_order','int(11) not null default 0');

1097
    $self->_upgrade_table('domains','spice_password','varchar(20) DEFAULT NULL');
fv3rdugo's avatar
fv3rdugo committed
1098
    $self->_upgrade_table('domains','description','text DEFAULT NULL');
1099
    $self->_upgrade_table('domains','run_timeout','int DEFAULT NULL');
1100
    $self->_upgrade_table('domains','id_vm','int DEFAULT NULL');
1101
1102
    $self->_upgrade_table('domains','start_time','int DEFAULT 0');
    $self->_upgrade_table('domains','is_volatile','int NOT NULL DEFAULT 0');
1103
    $self->_upgrade_table('domains','autostart','int NOT NULL DEFAULT 0');
1104
1105
1106

    $self->_upgrade_table('domains','status','varchar(32) DEFAULT "shutdown"');
    $self->_upgrade_table('domains','display','varchar(128) DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1107
    $self->_upgrade_table('domains','info','varchar(255) DEFAULT NULL');
Francesc Guasch's avatar
Francesc Guasch committed
1108
    $self->_upgrade_table('domains','internal_id','varchar(64) DEFAULT NULL');
1109
    $self->_upgrade_table('domains','id_vm','int default null');
1110
    $self->_upgrade_table('domains','volatile_clones','int NOT NULL default 0');
1111

1112
1113
    $self->_upgrade_table('domains','client_status','varchar(32)');
    $self->_upgrade_table('domains','client_status_time_checked','int NOT NULL default 0');
1114

Francesc Guasch's avatar
Francesc Guasch committed
1115
    $self->_upgrade_table('domains','needs_restart','int not null default 0');
1116
    $self->_upgrade_table('domains_network','allowed','int not null default 1');
1117

1118
    $self->_upgrade_table('iptables','id_vm','int DEFAULT NULL');
1119
    $self->_upgrade_table('vms','security','varchar(255) default NULL');
1120
1121
    $self->_upgrade_table('grant_types','enabled','int not null default 1');

1122
1123
}

1124

1125
1126
1127
1128
sub _connect_dbh {
    my $driver= ($CONFIG->{db}->{driver} or 'mysql');;
    my $db_user = ($CONFIG->{db}->{user} or getpwnam($>));;
    my $db_pass = ($CONFIG->{db}->{password} or undef);
Francesc Guasch's avatar
Francesc Guasch committed
1129
    my $db = ( $CONFIG->{db}->{db} or 'ravada' );
1130
1131
1132
    my $host = $CONFIG->{db}->{host};

    my $data_source = "DBI:$driver:$db";
Joel Alarcón's avatar
Joel Alarcón committed
1133
    $data_source = "DBI:$driver:database=$db;host=$host"
1134
1135
        if $host && $host ne 'localhost';

1136
1137
1138
    my $con;
    for my $try ( 1 .. 10 ) {
        eval { $con = DBIx::Connector->new($data_source
1139
1140
                        ,$db_user,$db_pass,{RaiseError => 1
                        , PrintError=> 0 });
1141
1142
1143
1144
1145
1146
1147
            $con->dbh();
        };
        return $con if $con && !$@;
        sleep 1;
        warn "Try $try $@\n";
    }
    die ($@ or "Can't connect to $driver $db at $host");
1148
1149
}

1150
=head2 display_ip
1151

1152
Returns the default display IP read from the config file
1153

1154
=cut
1155

1156
1157
1158
sub display_ip {
    my $ip = $CONFIG->{display_ip};
    return $ip if $ip;
1159
1160
}

1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
=head2 nat_ip

Returns the IP for NATed environments

=cut

sub nat_ip {
    return $CONFIG->{nat_ip} if exists $CONFIG->{nat_ip};
}

1171
sub _init_config {
Francesc Guasch's avatar
Francesc Guasch committed
1172
    my $file = shift or confess "ERROR: Missing config file";
1173
1174

    my $connector = shift;
1175
    confess "Deprecated connector" if $connector;
1176

1177
    confess "ERROR: Missing config file $file\n"
Francesc Guasch's avatar
[    
Francesc Guasch committed
1178
1179
1180
1181
1182
1183
        if !-e $file;

    eval { $CONFIG = YAML::LoadFile($file) };

    die "ERROR: Format error in config file $file\n$@"  if $@;

Francesc Guasch's avatar
Francesc Guasch committed
1184
1185
1186
1187
1188
    if ( !$CONFIG->{vm} ) {
        my %default_vms = %VALID_VM;
        delete $default_vms{Void};
        $CONFIG->{vm} = [keys %default_vms];
    }
1189
#    $CONNECTOR = ( $connector or _connect_dbh());
1190

Francesc Guasch's avatar
Francesc Guasch committed
1191
    _init_config_vm();
1192
1193
1194
}

sub _init_config_vm {
Francesc Guasch's avatar
Francesc Guasch committed
1195

1196
    for my $vm ( @{$CONFIG->{vm}} ) {
Francesc Guasch's avatar
Francesc Guasch committed
1197
1198
        die "$vm not available in this system.\n".($ERROR_VM{$vm})
            if !exists $VALID_VM{$vm} || !$VALID_VM{$vm};
1199
    }
Francesc Guasch's avatar
Francesc Guasch committed
1200

1201
1202
    for my $vm ( keys %VALID_VM ) {
        delete $VALID_VM{$vm}