Add files

git-svn-id: svn+ssh://asteria.noreply.org/svn/weaselutils/trunk@41 bc3d92e2-beff-0310-a7cd-cc87d7ac0ede

1 files changed, 122 insertions, 0 deletions

diff --git a/Push b/Push
new file mode 100755
index 0000000..9b9846f
--- /dev/null
+++ b/Push
@@ -0,0 +1,122 @@
+#!/usr/bin/ruby
+
+require 'yaml';
+
+$config = YAML::load( File.open( 'Hosts' ) )
+
+def sys(command)
+	puts "} #{command}"
+	system(command) or throw "Command '#{command}' failed"
+end
+
+
+def getFilesMD5(host, ssh_hostname)
+	cmd = ssh_hostname ?
+		"ssh root@#{ssh_hostname} /bin/ls -1 /etc/openvpn" :
+		"ls -1 noreply2-#{host}"
+	puts "| " + cmd
+	f = IO.popen( cmd, aModeString="r" );
+	files = f.readlines.delete_if{|e| not e =~ /^noreply2[.-]/ }.collect{|e| e.chomp}
+	f.close
+
+	md5 = {}
+	if (files.length > 0)
+		cmd = ssh_hostname ?
+			"ssh root@#{ssh_hostname} 'cd /etc/openvpn && md5sum " + files.join(' ') + "'" :
+			"cd noreply2-#{host} && md5sum " + files.join(' ')
+		puts "| " + cmd
+		f = IO.popen( cmd, aModeString="r" );
+		f.each_line{ |l|
+			l.chomp!
+			(h,file) = l.split(/\s+/,2)
+			md5[file] = h
+		}
+		f.close
+	end
+
+	return md5
+end
+
+hosts = ARGV
+if hosts.size == 0
+	hosts = $config['hosts'].keys
+end
+hosts.each{ |hostname|
+	puts "Pushing to #{hostname}."
+	throw "Host #{hostname} not in config" unless $config['hosts'][hostname]
+	ssh_hostnames = $config['hosts'][hostname]['ssh_hostname']
+	unless ssh_hostnames
+		STDERR.puts "*** #{hostname} has no ssh_hostname!"
+		next
+	end
+
+	ssh_hostname = nil
+	ssh_hostnames.delete(' ').split(',').each{ |host|
+		if not system("ping -W2 -c3 #{host}")
+			STDERR.puts "*** #{host} is not reachable!"
+			next
+		else
+			ssh_hostname = host
+			break
+		end
+	}
+	unless ssh_hostname
+		STDERR.puts "*** no reachable host found for #{hostname}!"
+		next
+	end
+
+
+	remote = getFilesMD5(hostname, ssh_hostname)
+	local = getFilesMD5(hostname, nil)
+
+	delete = []
+	copy = []
+	stop = []
+	restart = []
+	do_iptables = nil
+	do_ip6tables = nil
+	do_quagga = nil
+	remote.each_pair{|f,h|
+		if ! local.has_key?(f)
+			delete << f
+			if f =~ /\.conf$/
+				stop << f.gsub(/\.conf$/, '')
+			end
+		end
+	}
+	local.each_pair{|f,h|
+		if ! remote.has_key?(f) || local[f] != remote[f]
+			copy << f
+			if f =~ /\.iptables\.sh$/
+				do_iptables = f
+			elsif f =~ /\.ip6tables\.sh$/
+				do_ip6tables = f
+			elsif f =~ /\.quagga\.(bgpd|zebra|daemons)$/
+				do_quagga = true
+			elsif f =~ /\.conf$/
+				restart << f.gsub(/\.conf$/, '')
+			end
+		end
+	}
+
+	dir = "noreply2-#{hostname}/"
+	sys("scp "+copy.collect{|f| dir+f }.join(' ')+" root@#{ssh_hostname}:/etc/openvpn/") if copy.size > 0
+	commands = []
+	commands << "sh ./#{do_iptables} &&\n"                                if do_iptables
+	commands << "sh ./#{do_ip6tables} &&\n"                               if do_ip6tables
+	commands << "/etc/init.d/openvpn stop "+stop.join(' ')+"  &&\n"       if stop.size > 0
+	commands << "/etc/init.d/openvpn restart "+restart.join(' ')+" && \n" if restart.size > 0
+	commands << "rm "+delete.join(' ')+" &&\n"                            if delete.size > 0
+	command = "cd /etc/openvpn && \n";
+	command << commands.join('')
+	command << "echo 'all done'"
+	sys("ssh root@#{ssh_hostname} '#{command}'") if commands.size > 0
+	if (do_quagga)
+		sys("ssh root@#{ssh_hostname} 'cp -a /etc/openvpn/noreply2.quagga.bgpd /etc/quagga/bgpd.conf &&
+		                               cp -a /etc/openvpn/noreply2.quagga.zebra /etc/quagga/zebra.conf &&
+		                               cp -a /etc/openvpn/noreply2.quagga.daemons /etc/quagga/daemons &&
+		                               chmod 640 /etc/quagga/bgpd.conf /etc/quagga/zebra.conf &&
+		                               chgrp quagga /etc/quagga/bgpd.conf /etc/quagga/zebra.conf &&
+		                               /etc/init.d/quagga restart'")
+	end
+}