r4178 bugfix: PHP warnings were appearing before rackspace on SLB object page, defaul...
[racktables] / syncdomain.php
1 #!/usr/local/bin/php
2 <?php
3
4 $script_mode = TRUE;
5 require 'inc/init.php';
6
7 function usage()
8 {
9 echo "Usage: <this file> <options>\n";
10 echo "\t\t--vdid=<VLAN domain ID>\n";
11 echo "\t\t--mode=pull\n";
12 echo "\t\t--mode=pullall\n";
13 echo "\t\t--mode=push\n";
14 echo "\t\t[--max=<max_to_do>]\n";
15 echo "\t\t[--verbose]\n";
16 exit (1);
17 }
18
19 $options = getopt ('', array ('vdid:', 'max::', 'mode:', 'verbose'));
20 if (!array_key_exists ('vdid', $options) or !array_key_exists ('mode', $options))
21 usage();
22
23 switch ($options['mode'])
24 {
25 case 'pullall':
26 $do_push = FALSE;
27 break;
28 case 'pull':
29 $do_push = FALSE;
30 break;
31 case 'push':
32 $do_push = TRUE;
33 break;
34 default:
35 usage();
36 }
37
38 $max = array_key_exists ('max', $options) ? $options['max'] : 0;
39 $verbose = array_key_exists ('verbose', $options);
40
41 try
42 {
43 $mydomain = getVLANDomain ($options['vdid']);
44 }
45 catch (RackTablesError $e)
46 {
47 echo "Cannot load domain data with ID ${options['vdid']}\n";
48 echo $e->getMessage() . "\n";
49 exit (1);
50 }
51
52 $todo = array
53 (
54 'pull' => array ('sync_ready', 'resync_ready'),
55 'push' => array ('sync_ready'),
56 'pullall' => array ('sync_ready', 'resync_ready', 'sync_aging', 'resync_aging', 'done'),
57 );
58
59 $filename = '/var/tmp/RackTables-syncdomain-' . $options['vdid'] . '.pid';
60 if (FALSE === $fp = @fopen ($filename, 'x+'))
61 {
62 if (FALSE === $pidfile_mtime = filemtime ($filename))
63 {
64 echo "Failed to obtain mtime of ${filename}\n";
65 exit (1);
66 }
67 $current_time = time();
68 if ($current_time < $pidfile_mtime)
69 {
70 echo "Warning: pidfile ${filename} mtime is in future!\n";
71 exit (1);
72 }
73 // don't indicate failure unless the pidfile is 15 minutes or more old
74 if ($current_time < $pidfile_mtime + 15 * 60)
75 exit (0);
76 echo "Failed to lock ${filename}, already locked by PID " . mb_substr (file_get_contents ($filename), 0, 6) . "\n";
77 exit (1);
78 }
79
80 ftruncate ($fp, 0);
81 fwrite ($fp, getmypid() . "\n");
82 fclose ($fp);
83
84 // fetch all the needed data from DB (preparing for DB connection loss)
85 $switch_queue = array();
86 foreach ($mydomain['switchlist'] as $switch)
87 if (in_array (detectVLANSwitchQueue (getVLANSwitchInfo ($switch['object_id'])), $todo[$options['mode']]))
88 $switch_queue[] = spotEntity ('object', $switch['object_id']);
89
90 // YOU SHOULD NOT USE DB FUNCTIONS BELOW IN THE PARENT PROCESS
91 // THE PARENT'S DB CONNECTION IS LOST DUE TO RECONNECTING IN THE CHILD
92 $fork_slots = getConfigVar ('SYNCDOMAIN_MAX_PROCESSES');
93 $do_fork = ($fork_slots > 1) and extension_loaded ('pcntl');
94 if ($fork_slots > 1 and ! $do_fork)
95 throw new RackTablesError ('PHP extension \'pcntl\' not found, can not use childs', RackTablesError::MISCONFIGURED);
96 $switches_working = 0;
97 $switchesdone = 0;
98 foreach ($switch_queue as $object)
99 {
100 if ($do_fork)
101 {
102 // wait for the next free slot
103 while ($fork_slots <= $switches_working)
104 {
105 pcntl_waitpid (-1, $wait_status);
106 --$switches_working;
107 }
108 $i_am_child = (0 === $fork_res = pcntl_fork());
109 }
110 if (! $do_fork or $i_am_child)
111 {
112 try
113 {
114 // make a separate DB connection for correct concurrent transactions handling
115 if ($i_am_child)
116 connectDB();
117 $portsdone = exec8021QDeploy ($object['id'], $do_push);
118 if ($portsdone or $verbose)
119 echo "Done '${object['dname']}': ${portsdone}\n";
120 }
121 catch (RackTablesError $e)
122 {
123 echo "FAILED '${object['dname']}': " . $e->getMessage() . "\n";
124 }
125 if ($i_am_child)
126 exit (0);
127 }
128 if (isset ($fork_res) and $fork_res > 0)
129 ++$switches_working;
130
131 if (++$switchesdone == $max)
132 {
133 if ($verbose)
134 echo "Maximum of ${max} items reached, terminating\n";
135 break;
136 }
137 }
138
139 // wait for all childs to exit
140 while ($switches_working > 0)
141 {
142 --$switches_working;
143 pcntl_waitpid (-1, $wait_status);
144 }
145
146 if (FALSE === unlink ($filename))
147 {
148 echo "Failed removing pidfile ${filename}\n";
149 exit (1);
150 }
151 exit (0);
152 ?>