]>
Commit | Line | Data |
---|---|---|
9a8d51a4 DM |
1 | package PMG::Backup; |
2 | ||
3 | use strict; | |
4 | use warnings; | |
5 | use Data::Dumper; | |
6 | use File::Basename; | |
7 | use File::Path; | |
8 | ||
9 | use PVE::Tools; | |
10 | ||
11 | use PMG::pmgcfg; | |
12 | use PMG::AtomicFile; | |
6529020a | 13 | use PMG::Utils qw(postgres_admin_cmd); |
9a8d51a4 | 14 | |
29b263e5 | 15 | my $sa_custom_config_fn = "/etc/mail/spamassassin/custom.cf"; |
520f7717 | 16 | |
9a8d51a4 DM |
17 | sub dump_table { |
18 | my ($dbh, $table, $ofh, $seq, $seqcol) = @_; | |
19 | ||
20 | my $sth = $dbh->column_info(undef, undef, $table, undef); | |
21 | ||
22 | my $attrs = $sth->fetchall_arrayref({}); | |
23 | ||
24 | my @col_arr; | |
25 | foreach my $ref (@$attrs) { | |
26 | push @col_arr, $ref->{COLUMN_NAME}; | |
27 | } | |
28 | ||
29 | $sth->finish(); | |
30 | ||
31 | my $cols = join (', ', @col_arr); | |
32 | $cols || die "unable to fetch column definitions: ERROR"; | |
33 | ||
34 | print $ofh "COPY $table ($cols) FROM stdin;\n"; | |
35 | ||
36 | my $cmd = "COPY $table ($cols) TO STDOUT"; | |
37 | $dbh->do($cmd); | |
38 | ||
39 | my $data = ''; | |
40 | while ($dbh->pg_getcopydata($data) >= 0) { | |
41 | print $ofh $data; | |
42 | } | |
43 | ||
44 | print $ofh "\\.\n\n"; | |
45 | ||
46 | if ($seq && $seqcol) { | |
47 | print $ofh "SELECT setval('$seq', max($seqcol)) FROM $table;\n\n"; | |
48 | } | |
49 | } | |
50 | ||
51 | sub dumpdb { | |
52 | my ($ofh) = @_; | |
53 | ||
54 | print $ofh "SET client_encoding = 'SQL_ASCII';\n"; | |
55 | print $ofh "SET check_function_bodies = false;\n\n"; | |
56 | ||
57 | my $dbh = PMG::DBTools::open_ruledb(); | |
58 | ||
59 | print $ofh "BEGIN TRANSACTION;\n\n"; | |
60 | ||
61 | eval { | |
62 | $dbh->begin_work; | |
63 | ||
64 | # read a consistent snapshot | |
65 | $dbh->do("SET TRANSACTION ISOLATION LEVEL SERIALIZABLE"); | |
66 | ||
67 | dump_table($dbh, 'attribut', $ofh); | |
68 | dump_table($dbh, 'object', $ofh, 'object_id_seq', 'id'); | |
69 | dump_table($dbh, 'objectgroup', $ofh, 'objectgroup_id_seq', 'id'); | |
70 | dump_table($dbh, 'rule', $ofh, 'rule_id_seq', 'id'); | |
71 | dump_table($dbh, 'rulegroup', $ofh); | |
72 | dump_table($dbh, 'userprefs', $ofh); | |
73 | ||
74 | # we do not save the following tables: cgreylist, cmailstore, cmsreceivers, clusterinfo | |
75 | }; | |
76 | my $err = $@; | |
77 | ||
78 | $dbh->rollback(); # end read-only transaction | |
79 | ||
80 | $dbh->disconnect(); | |
81 | ||
82 | die $err if $err; | |
83 | ||
84 | print $ofh "COMMIT TRANSACTION;\n\n"; | |
85 | } | |
86 | ||
87 | sub dumpstatdb { | |
88 | my ($ofh) = @_; | |
89 | ||
90 | print $ofh "SET client_encoding = 'SQL_ASCII';\n"; | |
91 | print $ofh "SET check_function_bodies = false;\n\n"; | |
92 | ||
93 | my $dbh = PMG::DBTools::open_ruledb(); | |
94 | ||
95 | eval { | |
96 | $dbh->begin_work; | |
97 | ||
98 | # read a consistent snapshot | |
99 | $dbh->do("SET TRANSACTION ISOLATION LEVEL SERIALIZABLE"); | |
100 | ||
101 | print $ofh "BEGIN TRANSACTION;\n\n"; | |
102 | ||
103 | dump_table($dbh, 'dailystat', $ofh); | |
104 | dump_table($dbh, 'domainstat', $ofh); | |
105 | dump_table($dbh, 'virusinfo', $ofh); | |
106 | dump_table($dbh, 'localstat', $ofh); | |
107 | ||
108 | # drop/create the index is a little bit faster (20%) | |
109 | ||
110 | print $ofh "DROP INDEX cstatistic_time_index;\n\n"; | |
111 | print $ofh "ALTER TABLE cstatistic DROP CONSTRAINT cstatistic_id_key;\n\n"; | |
112 | print $ofh "ALTER TABLE cstatistic DROP CONSTRAINT cstatistic_pkey;\n\n"; | |
113 | dump_table($dbh, 'cstatistic', $ofh, 'cstatistic_id_seq', 'id'); | |
114 | print $ofh "ALTER TABLE ONLY cstatistic ADD CONSTRAINT cstatistic_pkey PRIMARY KEY (cid, rid);\n\n"; | |
115 | print $ofh "ALTER TABLE ONLY cstatistic ADD CONSTRAINT cstatistic_id_key UNIQUE (id);\n\n"; | |
116 | print $ofh "CREATE INDEX CStatistic_Time_Index ON CStatistic (Time);\n\n"; | |
117 | ||
118 | print $ofh "DROP INDEX CStatistic_ID_Index;\n\n"; | |
119 | dump_table($dbh, 'creceivers', $ofh); | |
120 | print $ofh "CREATE INDEX CStatistic_ID_Index ON CReceivers (CStatistic_CID, CStatistic_RID);\n\n"; | |
121 | ||
122 | dump_table($dbh, 'statinfo', $ofh); | |
123 | ||
124 | print $ofh "COMMIT TRANSACTION;\n\n"; | |
125 | }; | |
126 | my $err = $@; | |
127 | ||
128 | $dbh->rollback(); # end read-only transaction | |
129 | ||
130 | $dbh->disconnect(); | |
131 | ||
132 | die $err if $err; | |
133 | } | |
134 | ||
708cc7a9 | 135 | # this function assumes that directory $dirname exists and is empty |
9a8d51a4 | 136 | sub pmg_backup { |
29e2ff48 SI |
137 | my ($dirname, $include_statistics) = @_; |
138 | ||
139 | die "No backupdir provided!\n" if !defined($dirname); | |
9a8d51a4 DM |
140 | |
141 | my $time = time; | |
9a8d51a4 DM |
142 | my $dbfn = "Proxmox_ruledb.sql"; |
143 | my $statfn = "Proxmox_statdb.sql"; | |
144 | my $tarfn = "config_backup.tar"; | |
145 | my $sigfn = "proxmox_backup_v1.md5"; | |
146 | my $verfn = "version.txt"; | |
147 | ||
148 | eval { | |
149 | ||
9a8d51a4 DM |
150 | # dump the database first |
151 | my $fh = PMG::AtomicFile->open("$dirname/$dbfn", "w") || | |
152 | die "cant open '$dirname/$dbfn' - $! :ERROR"; | |
153 | ||
154 | dumpdb($fh); | |
155 | ||
156 | $fh->close(1); | |
157 | ||
158 | if ($include_statistics) { | |
159 | # dump the statistic db | |
160 | my $sfh = PMG::AtomicFile->open("$dirname/$statfn", "w") || | |
161 | die "cant open '$dirname/$statfn' - $! :ERROR"; | |
162 | ||
163 | dumpstatdb($sfh); | |
164 | ||
165 | $sfh->close(1); | |
166 | } | |
167 | ||
168 | my $pkg = PMG::pmgcfg::package(); | |
1c15ce0f | 169 | my $release = PMG::pmgcfg::release(); |
9a8d51a4 DM |
170 | |
171 | my $vfh = PMG::AtomicFile->open ("$dirname/$verfn", "w") || | |
172 | die "cant open '$dirname/$verfn' - $! :ERROR"; | |
173 | ||
174 | $time = time; | |
175 | my $now = localtime; | |
1c15ce0f | 176 | print $vfh "product: $pkg\nversion: $release\nbackuptime:$time:$now\n"; |
9a8d51a4 DM |
177 | $vfh->close(1); |
178 | ||
bd1325ed | 179 | my $extra_cfgs = []; |
9a8d51a4 | 180 | |
29b263e5 | 181 | push @$extra_cfgs, $sa_custom_config_fn; |
9a8d51a4 | 182 | |
9a8d51a4 DM |
183 | my $extradb = $include_statistics ? $statfn : ''; |
184 | ||
bd1325ed | 185 | my $extra = join(' ', @$extra_cfgs); |
9a8d51a4 DM |
186 | |
187 | system("/bin/tar cf $dirname/$tarfn -C / " . | |
bd1325ed | 188 | "/etc/pmg $extra>/dev/null 2>&1") == 0 || |
9a8d51a4 DM |
189 | die "unable to create system configuration backup: ERROR"; |
190 | ||
191 | system("cd $dirname; md5sum $tarfn $dbfn $extradb $verfn> $sigfn") == 0 || | |
192 | die "unable to create backup signature: ERROR"; | |
193 | ||
29e2ff48 SI |
194 | }; |
195 | my $err = $@; | |
196 | ||
197 | if ($err) { | |
198 | die $err; | |
199 | } | |
200 | } | |
201 | ||
202 | sub pmg_backup_pack { | |
203 | my ($filename, $include_statistics) = @_; | |
204 | ||
205 | my $time = time; | |
206 | my $dirname = "/tmp/proxbackup_$$.$time"; | |
207 | ||
208 | eval { | |
209 | ||
210 | my $targetdir = dirname($filename); | |
211 | mkdir $targetdir; # try to create target dir | |
212 | -d $targetdir || | |
213 | die "unable to access target directory '$targetdir'\n"; | |
214 | ||
215 | rmtree $dirname; | |
216 | # create backup directory | |
217 | mkdir $dirname; | |
218 | ||
219 | pmg_backup($dirname, $include_statistics); | |
220 | ||
221 | system("rm -f $filename; tar czf $filename --strip-components=1 -C $dirname .") == 0 || | |
9a8d51a4 DM |
222 | die "unable to create backup archive: ERROR"; |
223 | }; | |
224 | my $err = $@; | |
225 | ||
226 | rmtree $dirname; | |
227 | ||
228 | if ($err) { | |
229 | unlink $filename; | |
230 | die $err; | |
231 | } | |
232 | } | |
233 | ||
a6d276e9 DM |
234 | sub pmg_restore { |
235 | my ($filename, $restore_database, $restore_config, $restore_statistics) = @_; | |
236 | ||
237 | my $dbname = 'Proxmox_ruledb'; | |
238 | ||
239 | my $time = time; | |
240 | my $dirname = "/tmp/proxrestore_$$.$time"; | |
241 | my $dbfn = "Proxmox_ruledb.sql"; | |
242 | my $statfn = "Proxmox_statdb.sql"; | |
243 | my $tarfn = "config_backup.tar"; | |
244 | my $sigfn = "proxmox_backup_v1.md5"; | |
245 | ||
2cf02541 SI |
246 | my $untar = 1; |
247 | ||
248 | # directory indicates that the files were restored from a PBS remote | |
249 | if ( -d $filename ) { | |
250 | $dirname = $filename; | |
251 | $untar = 0; | |
252 | } | |
253 | ||
a6d276e9 | 254 | eval { |
a6d276e9 | 255 | |
2cf02541 | 256 | if ($untar) { |
e9c4929f DM |
257 | # remove any leftovers |
258 | rmtree $dirname; | |
2cf02541 SI |
259 | # create a temporary directory |
260 | mkdir $dirname; | |
261 | ||
262 | system("cd $dirname; tar xzf $filename >/dev/null 2>&1") == 0 || | |
263 | die "unable to extract backup archive: ERROR"; | |
264 | } | |
a6d276e9 DM |
265 | |
266 | system("cd $dirname; md5sum -c $sigfn") == 0 || | |
267 | die "proxmox backup signature check failed: ERROR"; | |
268 | ||
269 | if ($restore_config) { | |
270 | # restore the tar file | |
271 | mkdir "$dirname/config/"; | |
272 | system("tar xpf $dirname/$tarfn -C $dirname/config/") == 0 || | |
273 | die "unable to restore configuration tar archive: ERROR"; | |
274 | ||
275 | -d "$dirname/config/etc/pmg" || | |
276 | die "backup does not contain a valid system configuration directory (/etc/pmg)\n"; | |
277 | # unlink unneeded files | |
278 | unlink "$dirname/config/etc/pmg/cluster.conf"; # never restore cluster config | |
279 | rmtree "$dirname/config/etc/pmg/master"; | |
280 | ||
8624613e DM |
281 | # remove current config, but keep directory for INotify |
282 | rmtree("/etc/pmg", { keep_root => 1 }); | |
a6d276e9 DM |
283 | # copy files |
284 | system("cp -a $dirname/config/etc/pmg/* /etc/pmg/") == 0 || | |
285 | die "unable to restore system configuration: ERROR"; | |
286 | ||
29b263e5 | 287 | if (-f "$dirname/config/${sa_custom_config_fn}") { |
520f7717 | 288 | my $data = PVE::Tools::file_get_contents( |
29b263e5 DM |
289 | "$dirname/config/${sa_custom_config_fn}", 1024*1024); |
290 | PVE::Tools::file_set_contents($sa_custom_config_fn, $data); | |
520f7717 DM |
291 | } |
292 | ||
a6d276e9 DM |
293 | my $cfg = PMG::Config->new(); |
294 | my $ruledb = PMG::RuleDB->new(); | |
295 | my $rulecache = PMG::RuleCache->new($ruledb); | |
296 | $cfg->rewrite_config($rulecache, 1); | |
297 | } | |
298 | ||
299 | if ($restore_database) { | |
300 | # recreate the database | |
301 | ||
302 | # stop all services accessing the database | |
303 | PMG::Utils::service_wait_stopped(40, $PMG::Utils::db_service_list); | |
304 | ||
305 | print "Destroy existing rule database\n"; | |
306 | PMG::DBTools::delete_ruledb($dbname); | |
307 | ||
308 | print "Create new database\n"; | |
309 | my $dbh = PMG::DBTools::create_ruledb($dbname); | |
a6d276e9 DM |
310 | |
311 | system("cat $dirname/$dbfn|psql $dbname >/dev/null 2>&1") == 0 || | |
312 | die "unable to restore rule database: ERROR"; | |
313 | ||
314 | if ($restore_statistics) { | |
315 | if (-f "$dirname/$statfn") { | |
316 | system("cat $dirname/$statfn|psql $dbname >/dev/null 2>&1") == 0 || | |
317 | die "unable to restore statistic database: ERROR"; | |
318 | } | |
319 | } | |
320 | ||
321 | print STDERR "run analyze to speed up database queries\n"; | |
6529020a | 322 | postgres_admin_cmd('psql', { input => 'analyze;' }, $dbname); |
a6d276e9 DM |
323 | |
324 | print "Analyzing/Upgrading existing Databases..."; | |
9882b397 | 325 | my $ruledb = PMG::RuleDB->new($dbh); |
a6d276e9 DM |
326 | PMG::DBTools::upgradedb($ruledb); |
327 | print "done\n"; | |
328 | ||
329 | # cleanup old spam/virus storage | |
330 | PMG::MailQueue::create_spooldirs(0, 1); | |
331 | ||
332 | my $cfg = PMG::Config->new(); | |
333 | my $rulecache = PMG::RuleCache->new($ruledb); | |
334 | $cfg->rewrite_config($rulecache, 1); | |
335 | ||
336 | # and restart services as soon as possible | |
337 | foreach my $service (reverse @$PMG::Utils::db_service_list) { | |
338 | eval { PVE::Tools::run_command(['systemctl', 'start', $service]); }; | |
339 | warn $@ if $@; | |
340 | } | |
341 | } | |
342 | }; | |
343 | my $err = $@; | |
344 | ||
345 | rmtree $dirname; | |
346 | ||
347 | die $err if $err; | |
348 | } | |
349 | ||
9a8d51a4 | 350 | 1; |