]> git.proxmox.com Git - pmg-api.git/blame - PMG/Cluster.pm
do not use 'last' inside do/while
[pmg-api.git] / PMG / Cluster.pm
CommitLineData
0854fb22
DM
1package PMG::Cluster;
2
3use strict;
4use warnings;
45e68618 5use Data::Dumper;
0854fb22 6use Socket;
cfdf6608 7use File::Path;
9430b6d4 8use Time::HiRes qw (gettimeofday tv_interval);
45e68618 9
a7c7cad7 10use PVE::SafeSyslog;
0854fb22
DM
11use PVE::Tools;
12use PVE::INotify;
0757859a 13use PVE::APIClient::LWP;
0854fb22 14
1fb2ab76 15use PMG::Utils;
a7c7cad7 16use PMG::Config;
9f67f5b3 17use PMG::ClusterConfig;
db303db4
DM
18use PMG::RuleDB;
19use PMG::RuleCache;
9430b6d4 20use PMG::MailQueue;
0757859a 21use PMG::Fetchmail;
0854fb22 22
8737f93a
DM
23sub remote_node_ip {
24 my ($nodename, $noerr) = @_;
25
d8782874 26 my $cinfo = PMG::ClusterConfig->new();
8737f93a 27
45e68618 28 foreach my $entry (values %{$cinfo->{ids}}) {
8737f93a
DM
29 if ($entry->{name} eq $nodename) {
30 my $ip = $entry->{ip};
31 return $ip if !wantarray;
32 my $family = PVE::Tools::get_host_address_family($ip);
33 return ($ip, $family);
34 }
35 }
36
37 # fallback: try to get IP by other means
9f67f5b3 38 return PMG::Utils::lookup_node_ip($nodename, $noerr);
8737f93a
DM
39}
40
d2e43f9e
DM
41sub get_master_node {
42 my ($cinfo) = @_;
43
d8782874 44 $cinfo = PMG::ClusterConfig->new() if !$cinfo;
d2e43f9e
DM
45
46 return $cinfo->{master}->{name} if defined($cinfo->{master});
47
48 return 'localhost';
49}
50
cba17aeb
DM
51sub read_local_ssl_cert_fingerprint {
52 my $cert_path = "/etc/pmg/pmg-api.pem";
0854fb22 53
cba17aeb
DM
54 my $cert;
55 eval {
56 my $bio = Net::SSLeay::BIO_new_file($cert_path, 'r');
57 $cert = Net::SSLeay::PEM_read_bio_X509($bio);
58 Net::SSLeay::BIO_free($bio);
59 };
60 if (my $err = $@) {
61 die "unable to read certificate '$cert_path' - $err\n";
62 }
0854fb22 63
cba17aeb
DM
64 if (!defined($cert)) {
65 die "unable to read certificate '$cert_path' - got empty value\n";
66 }
67
68 my $fp;
69 eval {
70 $fp = Net::SSLeay::X509_get_fingerprint($cert, 'sha256');
71 };
72 if (my $err = $@) {
73 die "unable to get fingerprint for '$cert_path' - $err\n";
74 }
0854fb22 75
cba17aeb
DM
76 if (!defined($fp) || $fp eq '') {
77 die "unable to get fingerprint for '$cert_path' - got empty value\n";
78 }
0854fb22 79
cba17aeb
DM
80 return $fp;
81}
0854fb22 82
cba17aeb
DM
83my $hostrsapubkey_fn = '/etc/ssh/ssh_host_rsa_key.pub';
84my $rootrsakey_fn = '/root/.ssh/id_rsa';
85my $rootrsapubkey_fn = '/root/.ssh/id_rsa.pub';
0854fb22 86
cba17aeb
DM
87sub read_local_cluster_info {
88
89 my $res = {};
90
91 my $hostrsapubkey = PVE::Tools::file_read_firstline($hostrsapubkey_fn);
92 $hostrsapubkey =~ s/^.*ssh-rsa\s+//i;
93 $hostrsapubkey =~ s/\s+root\@\S+\s*$//i;
94
95 die "unable to parse ${hostrsapubkey_fn}\n"
96 if $hostrsapubkey !~ m/^[A-Za-z0-9\.\/\+]{200,}$/;
0854fb22
DM
97
98 my $nodename = PVE::INotify::nodename();
99
cba17aeb 100 $res->{name} = $nodename;
0854fb22 101
cba17aeb 102 $res->{ip} = PMG::Utils::lookup_node_ip($nodename);
0854fb22 103
cba17aeb 104 $res->{hostrsapubkey} = $hostrsapubkey;
0854fb22 105
cba17aeb
DM
106 if (! -f $rootrsapubkey_fn) {
107 unlink $rootrsakey_fn;
108 my $cmd = ['ssh-keygen', '-t', 'rsa', '-N', '', '-b', '2048',
109 '-f', $rootrsakey_fn];
1fb2ab76 110 PMG::Utils::run_silent_cmd($cmd);
cba17aeb
DM
111 }
112
113 my $rootrsapubkey = PVE::Tools::file_read_firstline($rootrsapubkey_fn);
114 $rootrsapubkey =~ s/^.*ssh-rsa\s+//i;
115 $rootrsapubkey =~ s/\s+root\@\S+\s*$//i;
116
117 die "unable to parse ${rootrsapubkey_fn}\n"
118 if $rootrsapubkey !~ m/^[A-Za-z0-9\.\/\+]{200,}$/;
119
120 $res->{rootrsapubkey} = $rootrsapubkey;
121
122 $res->{fingerprint} = read_local_ssl_cert_fingerprint();
123
124 return $res;
125}
126
127# X509 Certificate cache helper
128
129my $cert_cache_nodes = {};
130my $cert_cache_timestamp = time();
131my $cert_cache_fingerprints = {};
0854fb22 132
cba17aeb 133sub update_cert_cache {
0854fb22 134
cba17aeb
DM
135 $cert_cache_timestamp = time();
136
137 $cert_cache_fingerprints = {};
138 $cert_cache_nodes = {};
139
d8782874 140 my $cinfo = PMG::ClusterConfig->new();
cba17aeb
DM
141
142 foreach my $entry (values %{$cinfo->{ids}}) {
143 my $node = $entry->{name};
144 my $fp = $entry->{fingerprint};
145 if ($node && $fp) {
146 $cert_cache_fingerprints->{$fp} = 1;
147 $cert_cache_nodes->{$node} = $fp;
0854fb22
DM
148 }
149 }
150}
151
152# load and cache cert fingerprint once
153sub initialize_cert_cache {
154 my ($node) = @_;
155
cba17aeb 156 update_cert_cache()
0854fb22
DM
157 if defined($node) && !defined($cert_cache_nodes->{$node});
158}
159
160sub check_cert_fingerprint {
161 my ($cert) = @_;
162
163 # clear cache every 30 minutes at least
cba17aeb 164 update_cert_cache() if time() - $cert_cache_timestamp >= 60*30;
0854fb22
DM
165
166 # get fingerprint of server certificate
167 my $fp;
168 eval {
169 $fp = Net::SSLeay::X509_get_fingerprint($cert, 'sha256');
170 };
171 return 0 if $@ || !defined($fp) || $fp eq ''; # error
172
173 my $check = sub {
174 for my $expected (keys %$cert_cache_fingerprints) {
175 return 1 if $fp eq $expected;
176 }
177 return 0;
178 };
179
cba17aeb 180 return 1 if $check->();
0854fb22
DM
181
182 # clear cache and retry at most once every minute
183 if (time() - $cert_cache_timestamp >= 60) {
184 syslog ('info', "Could not verify remote node certificate '$fp' with list of pinned certificates, refreshing cache");
185 update_cert_cache();
cba17aeb 186 return $check->();
0854fb22
DM
187 }
188
189 return 0;
190}
191
58072364
DM
192my $sshglobalknownhosts = "/etc/ssh/ssh_known_hosts2";
193my $rootsshauthkeys = "/root/.ssh/authorized_keys";
194my $ssh_rsa_id = "/root/.ssh/id_rsa.pub";
195
196sub update_ssh_keys {
197 my ($cinfo) = @_;
198
809dd92a 199 my $old = '';
58072364 200 my $data = '';
809dd92a 201
58072364
DM
202 foreach my $node (values %{$cinfo->{ids}}) {
203 $data .= "$node->{ip} ssh-rsa $node->{hostrsapubkey}\n";
204 $data .= "$node->{name} ssh-rsa $node->{hostrsapubkey}\n";
205 }
206
809dd92a
DM
207 $old = PVE::Tools::file_get_contents($sshglobalknownhosts, 1024*1024)
208 if -f $sshglobalknownhosts;
209
210 PVE::Tools::file_set_contents($sshglobalknownhosts, $data)
211 if $old ne $data;
58072364
DM
212
213 $data = '';
809dd92a 214 $old = '';
58072364
DM
215
216 # always add ourself
217 if (-f $ssh_rsa_id) {
218 my $pub = PVE::Tools::file_get_contents($ssh_rsa_id);
219 chomp($pub);
220 $data .= "$pub\n";
221 }
222
223 foreach my $node (values %{$cinfo->{ids}}) {
224 $data .= "ssh-rsa $node->{rootrsapubkey} root\@$node->{name}\n";
225 }
226
227 if (-f $rootsshauthkeys) {
a7c7cad7
DM
228 my $mykey = PVE::Tools::file_get_contents($rootsshauthkeys, 128*1024);
229 chomp($mykey);
230 $data .= "$mykey\n";
58072364
DM
231 }
232
233 my $newdata = "";
234 my $vhash = {};
235 my @lines = split(/\n/, $data);
236 foreach my $line (@lines) {
237 if ($line !~ /^#/ && $line =~ m/(^|\s)ssh-(rsa|dsa)\s+(\S+)\s+\S+$/) {
238 next if $vhash->{$3}++;
239 }
240 $newdata .= "$line\n";
241 }
242
809dd92a
DM
243 $old = PVE::Tools::file_get_contents($rootsshauthkeys, 1024*1024)
244 if -f $rootsshauthkeys;
245
246 PVE::Tools::file_set_contents($rootsshauthkeys, $newdata, 0600)
247 if $old ne $newdata;
58072364
DM
248}
249
a7c7cad7
DM
250my $cfgdir = '/etc/pmg';
251my $syncdir = "$cfgdir/master";
252
253my $cond_commit_synced_file = sub {
254 my ($filename, $dstfn) = @_;
255
256 $dstfn = "$cfgdir/$filename" if !defined($dstfn);
257 my $srcfn = "$syncdir/$filename";
258
259 if (! -f $srcfn) {
260 unlink $dstfn;
261 return;
262 }
263
264 my $new = PVE::Tools::file_get_contents($srcfn, 1024*1024);
265
266 if (-f $dstfn) {
267 my $old = PVE::Tools::file_get_contents($dstfn, 1024*1024);
268 return 0 if $new eq $old;
269 }
270
1dc946d4
DM
271 # set mtime (touch) to avoid time drift problems
272 utime(undef, undef, $srcfn);
273
a7c7cad7
DM
274 rename($srcfn, $dstfn) ||
275 die "cond_rename_file '$filename' failed - $!\n";
276
277 print STDERR "updated $dstfn\n";
278
279 return 1;
280};
281
c9dae0df
DM
282my $rsync_command = sub {
283 my ($host_key_alias, @args) = @_;
284
285 my $ssh_cmd = '--rsh=ssh -l root -o BatchMode=yes';
286 $ssh_cmd .= " -o HostKeyAlias=${host_key_alias}" if $host_key_alias;
287
288 my $cmd = ['rsync', $ssh_cmd, '-q', @args];
289
290 return $cmd;
291};
292
293sub sync_quarantine_files {
89eefd36 294 my ($host_ip, $host_name, $flistname, $rcid) = @_;
c9dae0df 295
9430b6d4
DM
296 my $spooldir = $PMG::MailQueue::spooldir;
297
89eefd36
DM
298 mkdir "$spooldir/cluster/";
299 my $syncdir = "$spooldir/cluster/$rcid";
300 mkdir $syncdir;
301
c9dae0df
DM
302 my $cmd = $rsync_command->(
303 $host_name, '--timeout', '10', "${host_ip}:$spooldir", $spooldir,
304 '--files-from', $flistname);
305
9430b6d4 306 PVE::Tools::run_command($cmd);
c9dae0df
DM
307}
308
309sub sync_spooldir {
310 my ($host_ip, $host_name, $rcid) = @_;
311
9430b6d4
DM
312 my $spooldir = $PMG::MailQueue::spooldir;
313
c9dae0df
DM
314 mkdir "$spooldir/cluster/";
315 my $syncdir = "$spooldir/cluster/$rcid";
316 mkdir $syncdir;
317
318 my $cmd = $rsync_command->(
319 $host_name, '-aq', '--timeout', '10', "${host_ip}:$syncdir/", $syncdir);
320
321 foreach my $incl (('spam/', 'spam/*', 'spam/*/*', 'virus/', 'virus/*', 'virus/*/*')) {
322 push @$cmd, '--include', $incl;
323 }
324
325 push @$cmd, '--exclude', '*';
326
327 PVE::Tools::run_command($cmd);
328}
329
330sub sync_master_quar {
331 my ($host_ip, $host_name) = @_;
332
9430b6d4
DM
333 my $spooldir = $PMG::MailQueue::spooldir;
334
c9dae0df
DM
335 my $syncdir = "$spooldir/cluster/";
336 mkdir $syncdir;
337
338 my $cmd = $rsync_command->(
339 $host_name, '-aq', '--timeout', '10', "${host_ip}:$syncdir", $syncdir);
340
341 PVE::Tools::run_command($cmd);
342}
343
58072364 344sub sync_config_from_master {
809ae8f4 345 my ($master_name, $master_ip, $noreload) = @_;
58072364 346
58072364 347 mkdir $syncdir;
a7c7cad7 348 File::Path::remove_tree($syncdir, {keep_root => 1});
58072364 349
a7c7cad7
DM
350 my $sa_conf_dir = "/etc/mail/spamassassin";
351 my $sa_custom_cf = "custom.cf";
58072364 352
c9dae0df 353 my $cmd = $rsync_command->(
f4b7112b 354 $master_name, '-aq',
c9dae0df
DM
355 "${master_ip}:$cfgdir/* ${sa_conf_dir}/${sa_custom_cf}",
356 "$syncdir/",
f4b7112b 357 '--exclude', 'master/',
c9dae0df
DM
358 '--exclude', '*~',
359 '--exclude', '*.db',
360 '--exclude', 'pmg-api.pem',
361 '--exclude', 'pmg-tls.pem',
362 );
58072364
DM
363
364 my $errmsg = "syncing master configuration from '${master_ip}' failed";
365 PVE::Tools::run_command($cmd, errmsg => $errmsg);
a7c7cad7
DM
366
367 # verify that the remote host is cluster master
368 open (my $fh, '<', "$syncdir/cluster.conf") ||
369 die "unable to open synced cluster.conf - $!\n";
a7c7cad7 370
809ae8f4
DM
371 my $cinfo = PMG::ClusterConfig::read_cluster_conf('cluster.conf', $fh);
372
373 if (!$cinfo->{master} || ($cinfo->{master}->{ip} ne $master_ip)) {
a7c7cad7
DM
374 die "host '$master_ip' is not cluster master\n";
375 }
376
809ae8f4
DM
377 my $role = $cinfo->{'local'}->{type} // '-';
378 die "local node '$cinfo->{local}->{name}' not part of cluster\n"
a7c7cad7
DM
379 if $role eq '-';
380
809ae8f4 381 die "local node '$cinfo->{local}->{name}' is new cluster master\n"
a7c7cad7
DM
382 if $role eq 'master';
383
a7c7cad7 384 $cond_commit_synced_file->('cluster.conf');
a7c7cad7 385
809dd92a
DM
386 update_ssh_keys($cinfo); # rewrite ssh keys
387
0757859a
DM
388 PMG::Fetchmail::update_fetchmail_default(0); # disable on slave
389
a7c7cad7
DM
390 my $files = [
391 'pmg-authkey.key',
392 'pmg-authkey.pub',
393 'pmg-csrf.key',
394 'ldap.conf',
395 'user.conf',
7cac3e28
DM
396 'domains',
397 'mynetworks',
398 'transport',
959aaeba 399 'tls_policy',
fd6feef4 400 'fetchmailrc',
a7c7cad7
DM
401 ];
402
403 foreach my $filename (@$files) {
404 $cond_commit_synced_file->($filename);
405 }
406
0757859a 407
a7c7cad7
DM
408 my $force_restart = {};
409
410 if ($cond_commit_synced_file->($sa_custom_cf, "${sa_conf_dir}/${sa_custom_cf}")) {
411 $force_restart->{spam} = 1;
412 }
413
414 $cond_commit_synced_file->('pmg.conf');
f4b7112b
DM
415
416 # sync user templates files/symlinks (not recursive)
417 my $srcdir = "$syncdir/templates";
418 if (-d $srcdir) {
419 my $dstdir = "$cfgdir/templates";
420 mkdir $dstdir;
421 my $names_hash = {};
422 foreach my $fn (<$srcdir/*>) {
423 next if $fn !~ m|^($srcdir/(.*))$|;
424 $fn = $1; # untaint;
425 my $name = $2;
426 $names_hash->{$name} = 1;
427 my $target = "$dstdir/$name";
428 if (-f $fn) {
429 $cond_commit_synced_file->("templates/$name", $target);
430 } elsif (-l $fn) {
431 warn "update $target failed - $!\n" if !rename($fn, $target);
432 }
433 }
434 # remove vanished files
435 foreach my $fn (<$dstdir/*>) {
436 next if $fn !~ m|^($dstdir/(.*))$|;
437 $fn = $1; # untaint;
438 my $name = $2;
439 next if $names_hash->{$name};
440 warn "unlink $fn failed - $!\n" if !unlink($fn);
441 }
442 }
58072364
DM
443}
444
db303db4
DM
445sub sync_ruledb_from_master {
446 my ($ldb, $rdb, $ni, $ticket) = @_;
447
448 my $ruledb = PMG::RuleDB->new($ldb);
449 my $rulecache = PMG::RuleCache->new($ruledb);
450
451 my $conn = PVE::APIClient::LWP->new(
452 ticket => $ticket,
453 cookie_name => 'PMGAuthCookie',
454 host => $ni->{ip},
455 cached_fingerprints => {
456 $ni->{fingerprint} => 1,
457 });
458
459 my $digest = $conn->get("/config/ruledb/digest", {});
460
461 return if $digest eq $rulecache->{digest}; # no changes
462
463 syslog('info', "detected rule database changes - starting sync from '$ni->{ip}'");
464
465 eval {
466 $ldb->begin_work;
467
468 $ldb->do("DELETE FROM Rule");
469 $ldb->do("DELETE FROM RuleGroup");
470 $ldb->do("DELETE FROM ObjectGroup");
471 $ldb->do("DELETE FROM Object");
472 $ldb->do("DELETE FROM Attribut");
473
474 eval {
475 $rdb->begin_work;
476
477 # read a consistent snapshot
478 $rdb->do("SET TRANSACTION ISOLATION LEVEL SERIALIZABLE");
479
480 PMG::DBTools::copy_table($ldb, $rdb, "Rule");
481 PMG::DBTools::copy_table($ldb, $rdb, "RuleGroup");
482 PMG::DBTools::copy_table($ldb, $rdb, "ObjectGroup");
483 PMG::DBTools::copy_table($ldb, $rdb, "Object", 'value');
484 PMG::DBTools::copy_table($ldb, $rdb, "Attribut", 'value');
485 };
486
487 $rdb->rollback; # end transaction
488
489 die $@ if $@;
490
491 # update sequences
492
493 $ldb->do("SELECT setval('rule_id_seq', max(id)+1) FROM Rule");
494 $ldb->do("SELECT setval('object_id_seq', max(id)+1) FROM Object");
495 $ldb->do("SELECT setval('objectgroup_id_seq', max(id)+1) FROM ObjectGroup");
496
497 $ldb->commit;
498 };
499 if (my $err = $@) {
500 $ldb->rollback;
501 die $err;
502 }
503
504 syslog('info', "finished rule database sync from host '$ni->{ip}'");
505}
506
9430b6d4
DM
507sub sync_quarantine_db {
508 my ($ldb, $rdb, $ni, $rsynctime_ref) = @_;
509
510 my $rcid = $ni->{cid};
511
512 my $maxmails = 100000;
513
514 my $mscount = 0;
515
516 my $ctime = PMG::DBTools::get_remote_time($rdb);
517
518 my $maxcount = 1000;
519
520 my $count;
521
522 PMG::DBTools::create_clusterinfo_default($ldb, $rcid, 'lastid_CMailStore', -1, undef);
523
524 do { # get new values
525
526 $count = 0;
527
528 my $flistname = "/tmp/quarantinefilelist.$$";
529
530 eval {
531 $ldb->begin_work;
532
533 open(my $flistfh, '>', $flistname) ||
534 die "unable to open file '$flistname' - $!\n";
535
536 my $lastid = PMG::DBTools::read_int_clusterinfo($ldb, $rcid, 'lastid_CMailStore');
537
538 # sync CMailStore
539
540 my $sth = $rdb->prepare(
541 "SELECT * from CMailstore WHERE cid = ? AND rid > ? " .
542 "ORDER BY cid,rid LIMIT ?");
543 $sth->execute($rcid, $lastid, $maxcount);
544
545 my $maxid;
546 my $callback = sub {
547 my $ref = shift;
548 $maxid = $ref->{rid};
89eefd36
DM
549 my $filename = $ref->{file};
550 # skip files generated before cluster was created
551 return if $filename !~ m!^cluster/!;
552 print $flistfh "$filename\n";
9430b6d4
DM
553 };
554
555 my $attrs = [qw(cid rid time qtype bytes spamlevel info sender header file)];
556 $count += PMG::DBTools::copy_selected_data($ldb, $sth, 'CMailStore', $attrs, $callback);
557
558 close($flistfh);
559
560 my $starttime = [ gettimeofday() ];
89eefd36 561 sync_quarantine_files($ni->{ip}, $ni->{name}, $flistname, $rcid);
9430b6d4
DM
562 $$rsynctime_ref += tv_interval($starttime);
563
564 if ($maxid) {
565 # sync CMSReceivers
566
567 $sth = $rdb->prepare(
568 "SELECT * from CMSReceivers WHERE " .
569 "CMailStore_CID = ? AND CMailStore_RID > ? " .
570 "AND CMailStore_RID <= ?");
571 $sth->execute($rcid, $lastid, $maxid);
572
666b5e8f 573 $attrs = [qw(cmailstore_cid cmailstore_rid pmail receiver ticketid status mtime)];
9430b6d4
DM
574 PMG::DBTools::copy_selected_data($ldb, $sth, 'CMSReceivers', $attrs);
575
576 PMG::DBTools::write_maxint_clusterinfo($ldb, $rcid, 'lastid_CMailStore', $maxid);
577 }
578
579 $ldb->commit;
580 };
581 my $err = $@;
582
583 unlink $flistname;
584
585 if ($err) {
586 $ldb->rollback;
587 die $err;
588 }
589
590 $mscount += $count;
591
d4c2f646 592 } while (($count >= $maxcount) && ($mscount < $maxmails));
9430b6d4
DM
593
594 PMG::DBTools::create_clusterinfo_default($ldb, $rcid, 'lastmt_CMSReceivers', 0, undef);
595
596 eval { # synchronize status updates
597 $ldb->begin_work;
598
599 my $lastmt = PMG::DBTools::read_int_clusterinfo($ldb, $rcid, 'lastmt_CMSReceivers');
600
601 my $sth = $rdb->prepare ("SELECT * from CMSReceivers WHERE mtime >= ? AND status != 'N'");
602 $sth->execute($lastmt);
603
604 my $update_sth = $ldb->prepare(
605 "UPDATE CMSReceivers SET status = ? WHERE " .
666b5e8f 606 "CMailstore_CID = ? AND CMailstore_RID = ? AND TicketID = ?");
9430b6d4
DM
607 while (my $ref = $sth->fetchrow_hashref()) {
608 $update_sth->execute($ref->{status}, $ref->{cmailstore_cid},
666b5e8f 609 $ref->{cmailstore_rid}, $ref->{ticketid});
9430b6d4
DM
610 }
611
612 PMG::DBTools::write_maxint_clusterinfo($ldb, $rcid, 'lastmt_CMSReceivers', $ctime);
613
614 $ldb->commit;
615 };
616 if (my $err = $@) {
617 $ldb->rollback;
618 die $err;
619 }
620
621 return $mscount;
622}
623
624sub sync_statistic_db {
625 my ($ldb, $rdb, $ni) = @_;
626
627 my $rcid = $ni->{cid};
628
629 my $maxmails = 100000;
630
631 my $mscount = 0;
632
633 my $maxcount = 1000;
634
635 my $count;
636
637 PMG::DBTools::create_clusterinfo_default(
638 $ldb, $rcid, 'lastid_CStatistic', -1, undef);
639
640 do { # get new values
641
642 $count = 0;
643
644 eval {
645 $ldb->begin_work;
646
647 my $lastid = PMG::DBTools::read_int_clusterinfo(
648 $ldb, $rcid, 'lastid_CStatistic');
649
650 # sync CStatistic
651
652 my $sth = $rdb->prepare(
653 "SELECT * from CStatistic " .
654 "WHERE cid = ? AND rid > ? " .
655 "ORDER BY cid, rid LIMIT ?");
656 $sth->execute($rcid, $lastid, $maxcount);
657
658 my $maxid;
659 my $callback = sub {
660 my $ref = shift;
661 $maxid = $ref->{rid};
662 };
663
664 my $attrs = [qw(cid rid time bytes direction spamlevel ptime virusinfo sender)];
665 $count += PMG::DBTools::copy_selected_data($ldb, $sth, 'CStatistic', $attrs, $callback);
666
667 if ($maxid) {
668 # sync CReceivers
669
670 $sth = $rdb->prepare(
671 "SELECT * from CReceivers WHERE " .
672 "CStatistic_CID = ? AND CStatistic_RID > ? AND CStatistic_RID <= ?");
673 $sth->execute($rcid, $lastid, $maxid);
674
675 $attrs = [qw(cstatistic_cid cstatistic_rid blocked receiver)];
676 PMG::DBTools::copy_selected_data($ldb, $sth, 'CReceivers', $attrs);
9430b6d4
DM
677 }
678
64b7a2ea
DM
679 PMG::DBTools::write_maxint_clusterinfo ($ldb, $rcid, 'lastid_CStatistic', $maxid);
680
9430b6d4
DM
681 $ldb->commit;
682 };
683 if (my $err = $@) {
684 $ldb->rollback;
685 die $err;
686 }
687
688 $mscount += $count;
689
d4c2f646 690 } while (($count >= $maxcount) && ($mscount < $maxmails));
e86b85a3
DM
691
692 return $mscount;
9430b6d4
DM
693}
694
986eec31 695my $sync_generic_mtime_db = sub {
9430b6d4
DM
696 my ($ldb, $rdb, $ni, $table, $selectfunc, $mergefunc) = @_;
697
9430b6d4
DM
698 my $ctime = PMG::DBTools::get_remote_time($rdb);
699
700 PMG::DBTools::create_clusterinfo_default($ldb, $ni->{cid}, "lastmt_$table", 0, undef);
701
702 my $lastmt = PMG::DBTools::read_int_clusterinfo($ldb, $ni->{cid}, "lastmt_$table");
703
4c93256a 704 my $sql_cmd = $selectfunc->($ctime, $lastmt);
9430b6d4 705
4c93256a 706 my $sth = $rdb->prepare($sql_cmd);
9430b6d4
DM
707
708 $sth->execute();
709
986eec31
DM
710 my $updates = 0;
711
4c93256a
DM
712 eval {
713 # use transaction to speedup things
714 my $max = 1000; # UPDATE MAX ENTRIES AT ONCE
4c93256a
DM
715 my $count = 0;
716 while (my $ref = $sth->fetchrow_hashref()) {
9d5bdd2a
DM
717 $ldb->begin_work if !$count;
718 $mergefunc->($ref);
4c93256a
DM
719 if (++$count >= $max) {
720 $count = 0;
721 $ldb->commit;
9430b6d4 722 }
986eec31 723 $updates++;
9430b6d4 724 }
9d5bdd2a
DM
725
726 $ldb->commit if $count;
9430b6d4 727 };
4c93256a
DM
728 if (my $err = $@) {
729 $ldb->rollback;
730 die $err;
9430b6d4
DM
731 }
732
9430b6d4
DM
733 PMG::DBTools::write_maxint_clusterinfo($ldb, $ni->{cid}, "lastmt_$table", $ctime);
734
986eec31
DM
735 return $updates;
736};
9430b6d4 737
5e1408fd
DM
738sub sync_localstat_db {
739 my ($dbh, $rdb, $ni) = @_;
740
741 my $rcid = $ni->{cid};
742
743 my $selectfunc = sub {
744 my ($ctime, $lastmt) = @_;
745 return "SELECT * from LocalStat WHERE mtime >= $lastmt AND cid = $rcid";
746 };
747
748 my $merge_sth = $dbh->prepare(
ebd19c79
DM
749 'INSERT INTO LocalStat (Time, RBLCount, PregreetCount, CID, MTime) ' .
750 'VALUES (?, ?, ?, ?, ?) ' .
5e1408fd 751 'ON CONFLICT (Time, CID) DO UPDATE SET ' .
ebd19c79 752 'RBLCount = excluded.RBLCount, PregreetCount = excluded.PregreetCount, MTime = excluded.MTime');
5e1408fd
DM
753
754 my $mergefunc = sub {
755 my ($ref) = @_;
756
ebd19c79 757 $merge_sth->execute($ref->{time}, $ref->{rblcount}, $ref->{pregreetcount}, $ref->{cid}, $ref->{mtime});
5e1408fd
DM
758 };
759
760 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'LocalStat', $selectfunc, $mergefunc);
761}
762
9430b6d4
DM
763sub sync_greylist_db {
764 my ($dbh, $rdb, $ni) = @_;
765
766 my $selectfunc = sub {
767 my ($ctime, $lastmt) = @_;
768 return "SELECT * from CGreylist WHERE extime >= $ctime AND " .
769 "mtime >= $lastmt AND CID != 0";
770 };
771
2e049252 772 my $merge_sth = $dbh->prepare($PMG::DBTools::cgreylist_merge_sql);
9430b6d4 773 my $mergefunc = sub {
986eec31 774 my ($ref) = @_;
9430b6d4 775
f413f920
DM
776 $merge_sth->execute(
777 $ref->{ipnet}, $ref->{host}, $ref->{sender}, $ref->{receiver},
778 $ref->{instance}, $ref->{rctime}, $ref->{extime}, $ref->{delay},
779 $ref->{blocked}, $ref->{passed}, 0, $ref->{cid});
9430b6d4
DM
780 };
781
986eec31 782 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'CGreylist', $selectfunc, $mergefunc);
9430b6d4
DM
783}
784
785sub sync_userprefs_db {
786 my ($dbh, $rdb, $ni) = @_;
787
788 my $selectfunc = sub {
789 my ($ctime, $lastmt) = @_;
790
791 return "SELECT * from UserPrefs WHERE mtime >= $lastmt";
792 };
793
7d13157e 794 my $merge_sth = $dbh->prepare(
471b05ed 795 "INSERT INTO UserPrefs (PMail, Name, Data, MTime) " .
286bc590 796 'VALUES (?, ?, ?, ?) ' .
471b05ed 797 'ON CONFLICT (PMail, Name) DO UPDATE SET ' .
dbdf1298 798 # Note: MTime = 0 ==> this is just a copy from somewhere else, not modified
3a9ffd85 799 'MTime = CASE WHEN excluded.MTime >= UserPrefs.MTime THEN 0 ELSE UserPrefs.MTime END, ' .
0527ea1a 800 'Data = CASE WHEN excluded.MTime >= UserPrefs.MTime THEN excluded.Data ELSE UserPrefs.Data END');
471b05ed 801
9430b6d4 802 my $mergefunc = sub {
986eec31 803 my ($ref) = @_;
9430b6d4 804
286bc590 805 $merge_sth->execute($ref->{pmail}, $ref->{name}, $ref->{data}, $ref->{mtime});
9430b6d4
DM
806 };
807
986eec31 808 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'UserPrefs', $selectfunc, $mergefunc);
9430b6d4
DM
809}
810
811sub sync_domainstat_db {
812 my ($dbh, $rdb, $ni) = @_;
813
814 my $selectfunc = sub {
815 my ($ctime, $lastmt) = @_;
816 return "SELECT * from DomainStat WHERE mtime >= $lastmt";
817 };
818
1c7ea32c
DM
819 my $merge_sth = $dbh->prepare(
820 'INSERT INTO Domainstat ' .
821 '(Time,Domain,CountIn,CountOut,BytesIn,BytesOut,VirusIn,VirusOut,SpamIn,SpamOut,' .
822 'BouncesIn,BouncesOut,PTimeSum,Mtime) ' .
823 'VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?) ' .
824 'ON CONFLICT (Time, Domain) DO UPDATE SET ' .
825 'CountIn = excluded.CountIn, CountOut = excluded.CountOut, ' .
826 'BytesIn = excluded.BytesIn, BytesOut = excluded.BytesOut, ' .
827 'VirusIn = excluded.VirusIn, VirusOut = excluded.VirusOut, ' .
828 'SpamIn = excluded.SpamIn, SpamOut = excluded.SpamOut, ' .
829 'BouncesIn = excluded.BouncesIn, BouncesOut = excluded.BouncesOut, ' .
830 'PTimeSum = excluded.PTimeSum, MTime = excluded.MTime');
831
9430b6d4 832 my $mergefunc = sub {
986eec31 833 my ($ref) = @_;
9430b6d4 834
1c7ea32c
DM
835 $merge_sth->execute(
836 $ref->{time}, $ref->{domain}, $ref->{countin}, $ref->{countout},
837 $ref->{bytesin}, $ref->{bytesout},
838 $ref->{virusin}, $ref->{virusout}, $ref->{spamin}, $ref->{spamout},
839 $ref->{bouncesin}, $ref->{bouncesout}, $ref->{ptimesum}, $ref->{mtime});
9430b6d4
DM
840 };
841
986eec31 842 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'DomainStat', $selectfunc, $mergefunc);
9430b6d4
DM
843}
844
845sub sync_dailystat_db {
846 my ($dbh, $rdb, $ni) = @_;
847
848 my $selectfunc = sub {
849 my ($ctime, $lastmt) = @_;
850 return "SELECT * from DailyStat WHERE mtime >= $lastmt";
8bf584ae
DM
851 };
852
853 my $merge_sth = $dbh->prepare(
854 'INSERT INTO DailyStat ' .
855 '(Time,CountIn,CountOut,BytesIn,BytesOut,VirusIn,VirusOut,SpamIn,SpamOut,' .
856 'BouncesIn,BouncesOut,GreylistCount,SPFCount,RBLCount,PTimeSum,Mtime) ' .
857 'VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?) ' .
858 'ON CONFLICT (Time) DO UPDATE SET ' .
859 'CountIn = excluded.CountIn, CountOut = excluded.CountOut, ' .
860 'BytesIn = excluded.BytesIn, BytesOut = excluded.BytesOut, ' .
861 'VirusIn = excluded.VirusIn, VirusOut = excluded.VirusOut, ' .
862 'SpamIn = excluded.SpamIn, SpamOut = excluded.SpamOut, ' .
863 'BouncesIn = excluded.BouncesIn, BouncesOut = excluded.BouncesOut, ' .
864 'GreylistCount = excluded.GreylistCount, SPFCount = excluded.SpfCount, ' .
865 'RBLCount = excluded.RBLCount, ' .
866 'PTimeSum = excluded.PTimeSum, MTime = excluded.MTime');
9430b6d4
DM
867
868 my $mergefunc = sub {
986eec31 869 my ($ref) = @_;
9430b6d4 870
dbdf1298 871 $merge_sth->execute(
8bf584ae
DM
872 $ref->{time}, $ref->{countin}, $ref->{countout},
873 $ref->{bytesin}, $ref->{bytesout},
874 $ref->{virusin}, $ref->{virusout}, $ref->{spamin}, $ref->{spamout},
875 $ref->{bouncesin}, $ref->{bouncesout}, $ref->{greylistcount},
876 $ref->{spfcount}, $ref->{rblcount}, $ref->{ptimesum}, $ref->{mtime});
9430b6d4
DM
877 };
878
986eec31 879 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'DailyStat', $selectfunc, $mergefunc);
9430b6d4
DM
880}
881
882sub sync_virusinfo_db {
883 my ($dbh, $rdb, $ni) = @_;
884
885 my $selectfunc = sub {
886 my ($ctime, $lastmt) = @_;
887 return "SELECT * from VirusInfo WHERE mtime >= $lastmt";
888 };
889
80615ad6
DM
890 my $merge_sth = $dbh->prepare(
891 'INSERT INTO VirusInfo (Time,Name,Count,MTime) ' .
892 'VALUES (?,?,?,?) ' .
893 'ON CONFLICT (Time,Name) DO UPDATE SET ' .
894 'Count = excluded.Count , MTime = excluded.MTime');
895
9430b6d4 896 my $mergefunc = sub {
986eec31 897 my ($ref) = @_;
9430b6d4 898
80615ad6 899 $merge_sth->execute($ref->{time}, $ref->{name}, $ref->{count}, $ref->{mtime});
9430b6d4
DM
900 };
901
986eec31 902 return $sync_generic_mtime_db->($dbh, $rdb, $ni, 'VirusInfo', $selectfunc, $mergefunc);
9430b6d4
DM
903}
904
905sub sync_deleted_nodes_from_master {
906 my ($ldb, $masterdb, $cinfo, $masterni, $rsynctime_ref) = @_;
907
908 my $rsynctime = 0;
909
910 my $cid_hash = {}; # fast lookup
5c6e6eeb 911 foreach my $ni (values %{$cinfo->{ids}}) {
9430b6d4
DM
912 $cid_hash->{$ni->{cid}} = $ni;
913 }
914
915 my $spooldir = $PMG::MailQueue::spooldir;
916
5c6e6eeb
DM
917 my $maxcid = $cinfo->{master}->{maxcid} // 0;
918
919 for (my $rcid = 1; $rcid <= $maxcid; $rcid++) {
9430b6d4
DM
920 next if $cid_hash->{$rcid};
921
922 my $done_marker = "$spooldir/cluster/$rcid/.synced-deleted-node";
923
924 next if -f $done_marker; # already synced
925
926 syslog('info', "syncing deleted node $rcid from master '$masterni->{ip}'");
927
928 my $starttime = [ gettimeofday() ];
929 sync_spooldir($masterni->{ip}, $masterni->{name}, $rcid);
930 $$rsynctime_ref += tv_interval($starttime);
931
932 my $fake_ni = {
933 ip => $masterni->{ip},
934 name => $masterni->{name},
935 cid => $rcid,
936 };
937
938 sync_quarantine_db($ldb, $masterdb, $fake_ni);
939
940 sync_statistic_db ($ldb, $masterdb, $fake_ni);
941
942 open(my $fh, ">>", $done_marker);
943 }
944}
945
946
0854fb22 9471;