1: <?php // mysql.inc.php -- MySQL Test
  2: /* -------------------------------------------------------------
  3: This file is part of FreeNATS
  4: 
  5: FreeNATS is (C) Copyright 2008 PurplePixie Systems
  6: 
  7: FreeNATS is free software: you can redistribute it and/or modify
  8: it under the terms of the GNU General Public License as published by
  9: the Free Software Foundation, either version 3 of the License, or
 10: (at your option) any later version.
 11: 
 12: FreeNATS is distributed in the hope that it will be useful,
 13: but WITHOUT ANY WARRANTY; without even the implied warranty of
 14: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 15: GNU General Public License for more details.
 16: 
 17: You should have received a copy of the GNU General Public License
 18: along with FreeNATS.  If not, see www.gnu.org/licenses
 19: 
 20: For more information see www.purplepixie.org/freenats
 21: -------------------------------------------------------------- */
 22: 
 23: function fnmysql_error($error)
 24: {
 25: global $NATS;
 26: if (!isset($NATS)) return false;
 27: $NATS->Event("MySQL Test: ".$error,2,"Test","MySQL");
 28: }
 29: 
 30: 
 31: 
 32: // Data Test
 33: function mysql_test_data($host,$user,$pass,$database="",$timeout=0,$query="",$column=0,$debug=false)
 34: {
 35: global $NATS;
 36: if ($timeout>0) $timeout=$timeout; // use specific for test if set
 37: else
 38: 	{
 39: 	// otherwise use system if available
 40: 	if (isset($NATS)) $timeout=$NATS->Cfg->Get("test.mysql.timeout",0);
 41: 	if ($timeout<=0) $timeout=0; // unset specifically or in environment
 42: 	}
 43: // this will return a 0 at any stage if connect etc is ok but no rows are returned
 44: // negative if something actually fails
 45: if ($timeout>0)
 46: 	{
 47: 	$oldtimeout=ini_get("mysql.timeout");
 48: 	ini_set("mysql.timeout",$timeout);
 49: 	}
 50: 
 51: if (!is_numeric($column)) $column=0;
 52: if ($column<0) $column=0;
 53: 	
 54: $sql=@mysql_connect($host,$user,$pass,true);
 55: 
 56: if ((!$sql)||($database==""))
 57: 	{
 58: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
 59: 	if (!$sql) 
 60: 		{
 61: 		if ($debug) echo "Connect Error: Failed to Connect\n";
 62: 		fnmysql_error("Failed to Connect");
 63: 		return -1; // total connect failed
 64: 		}
 65: 	// otherwise is no database so close and return -1 to indicate failure (for the data requires a DB+qry etc)
 66: 	@mysql_close($sql);
 67: 	return -1;
 68: 	}
 69: 
 70: @mysql_select_db($database,$sql);
 71: 
 72: if (mysql_errno($sql)!=0) // failed to select DB
 73: 	{
 74: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
 75: 	fnmysql_error(mysql_error($sql));
 76: 	@mysql_close($sql);
 77: 	return -2;	// select database failed
 78: 	}
 79: 
 80: if ($query=="")
 81: 	{ // no query to perform
 82: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);	
 83: 	@mysql_close($sql);
 84: 	return -1; // all ok but no query/rows
 85: 	}
 86: 	
 87: $r=@mysql_query($query,$sql);
 88: if (mysql_errno($sql)==0) // successful query
 89: 	{
 90: 	if (is_bool($r)) // didn't return any data
 91: 		{
 92: 		$return=-4; // so for this purpose (data) the query failed
 93: 		}
 94: 	else 
 95: 		{
 96: 		if ($row=mysql_fetch_array($r))
 97: 			{ // got data ok
 98: 			if (isset($row[$column])) $return=$row[$column];
 99: 			else $return=$row[0];
100: 			}
101: 		else $return=-5; // query seemed to succeed but no data at all here
102: 		@mysql_free_result($r,$sql); // free if a result
103: 		}
104: 	}
105: else 
106: 	{
107: 	fnmysql_error(mysql_error($sql));
108: 	$return=-3; // query failed
109: 	}
110: 
111: @mysql_close($sql);
112: if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
113: return $return;
114: }
115: 
116: 
117: 
118: // Row Test
119: function mysql_test_rows($host,$user,$pass,$database="",$timeout=0,$query="",$debug=false)
120: {
121: global $NATS;
122: if ($timeout>0) $timeout=$timeout; // use specific for test if set
123: else
124: 	{
125: 	// otherwise use system if available
126: 	if (isset($NATS)) $timeout=$NATS->Cfg->Get("test.mysql.timeout",0);
127: 	if ($timeout<=0) $timeout=0; // unset specifically or in environment
128: 	}
129: // this will return a 0 at any stage if connect etc is ok but no rows are returned
130: // negative if something actually fails
131: if ($timeout>0)
132: 	{
133: 	$oldtimeout=ini_get("mysql.timeout");
134: 	ini_set("mysql.timeout",$timeout);
135: 	}
136: 
137: if ($debug) echo "mysql://".$user.":".$pass."@".$host."/".$database."\n";
138: 
139: $sql=@mysql_connect($host,$user,$pass,true);
140: 
141: if ((!$sql)||($database==""))
142: 	{
143: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
144: 	if (!$sql) 
145: 		{
146: 		if ($debug) echo "Connect Error: Failed to Connect\n";
147: 		fnmysql_error("Failed to Connect");
148: 		return -1; // total connect failed
149: 		}
150: 	// otherwise is no database so close and return 0
151: 	@mysql_close($sql);
152: 	return 0;
153: 	}
154: 
155: @mysql_select_db($database,$sql);
156: 
157: if (mysql_errno($sql)!=0) // failed to select DB
158: 	{
159: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
160: 	fnmysql_error(mysql_error($sql));
161: 	@mysql_close($sql);
162: 	return -2;	// select database failed
163: 	}
164: 
165: if ($query=="")
166: 	{ // no query to perform
167: 	if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);	
168: 	@mysql_close($sql);
169: 	return 0; // all ok but no query/rows
170: 	}
171: 	
172: $r=@mysql_query($query,$sql);
173: if (mysql_errno($sql)==0) // successful query
174: 	{
175: 	if (is_bool($r)) // didn't return any daya
176: 		{
177: 		$return=mysql_affected_rows($sql);
178: 		}
179: 	else 
180: 		{
181: 		$return=mysql_num_rows($r);
182: 		@mysql_free_result($r,$sql); // free if a result
183: 		}
184: 	}
185: else 
186: 	{
187: 	fnmysql_error(mysql_error($sql));
188: 	$return=-3; // query failed
189: 	}
190: 
191: @mysql_close($sql);
192: if ($timeout>0) ini_set("mysql.timeout",$oldtimeout);
193: return $return;
194: }
195: 
196: function mysql_test_time($host,$user,$pass,$database="",$timeout=0,$query="",$debug=false)
197: {
198: $timer=new TFNTimer();
199: $timer->Start();
200: $val=mysql_test_rows($host,$user,$pass,$database,$timeout,$query,$debug);
201: $time=$timer->Stop();
202: 
203: if ($val<0) return $val; // connect/select/query failed
204: 
205: // if $val is 0 then nothing was returned - maybe check the query here? Complicates the idea
206: // though for the user so left. Will have to do two tests for both time and rows 0 as fails
207: 
208: $time=round($time,4);
209: if ($time==0) return "0.0001";
210: return $time;
211: }
212: 
213: 
214: if (isset($NATS))
215: {
216: class FreeNATS_MySQL_Test extends FreeNATS_Local_Test
217: 	{
218: 	function DoTest($testname,$param,$hostname,$timeout,$params)
219: 		{ // 0: host, 1: user, 2: pass, 3: database, 4: query
220: 		
221: 		if ($testname=="mysql")
222: 			{
223: 			$ip=ip_lookup($param);
224: 			if ($ip=="0") return -1; // cache only as 127.0.0.1 is not the same connection as localhost for MySQL auth!
225: 	
226: 			return mysql_test_time($param,$params[1],$params[2],$params[3],$timeout,$params[4]);
227: 			}
228: 			
229: 		else if ($testname=="mysqlrows")
230: 			{	
231: 			$ip=ip_lookup($param);
232: 			if ($ip=="0") return -1; // cache only - see above
233: 			return mysql_test_rows($param,$params[1],$params[2],$params[3],$timeout,$params[4]);
234: 			}
235: 			
236: 		else if ($testname=="mysqldata")
237: 			{
238: 			$ip=ip_lookup($param);
239: 			if ($ip=="0") return -1;
240: 			return mysql_test_data($param,$params[1],$params[2],$params[3],$timeout,$params[4],$params[5]);
241: 			}
242: 			
243: 		else return -1;
244: 		
245: 		}
246: 	function Evaluate($result) 
247: 		{ // same for all types
248: 		if ($result<0) return 2; // failed
249: 		return 0; // passed
250: 		}
251: 	
252: 	function ProtectOutput(&$test)
253: 		{
254: 		$test['testparam2']="";
255: 		}
256: 		
257: 	function DisplayForm(&$row)
258: 		{
259: 		$optional=true;
260: 		if ($row['testtype']=="mysqldata") $optional=false;
261: 		echo "<table border=0>";
262: 		echo "<tr><td align=left>";
263: 		echo "Hostname :";
264: 		echo "</td><td align=left>";
265: 		echo "<input type=text name=testparam size=30 maxlength=128 value=\"".$row['testparam']."\">";
266: 		echo "</td></tr>";
267: 		echo "<tr><td align=left>";
268: 		echo "Username :";
269: 		echo "</td><td align=left>";
270: 		echo "<input type=text name=testparam1 size=30 maxlength=128 value=\"".$row['testparam1']."\">";
271: 		echo "</td></tr>";
272: 		echo "<tr><td align=left>";
273: 		echo "Password :";
274: 		echo "</td><td align=left>";
275: 		echo "<input type=text name=testparam2 size=30 maxlength=128 value=\"\">";
276: 		echo "<input type=hidden name=keepparam2 value=1>";
277: 		echo "</td></tr>";
278: 		echo "<tr><td colspan=2><i>Leave blank to not change or <input type=checkbox name=clearparam2 value=1> click to clear</i></td></tr>";
279: 		echo "<tr><td align=left>";
280: 		echo "Database :";
281: 		echo "</td><td align=left>";
282: 		echo "<input type=text name=testparam3 size=30 maxlength=128 value=\"".$row['testparam3']."\">";
283: 		echo "</td></tr>";
284: 		if ($optional) echo "<tr><td colspan=2><i>Optional - leave blank to not bother with select_db</td></tr>";
285: 		echo "<tr><td align=left>";
286: 		echo "Query :";
287: 		echo "</td><td align=left>";
288: 		echo "<input type=text name=testparam4 size=30 maxlength=128 value=\"".$row['testparam4']."\">";
289: 		echo "</td></tr>";
290: 		if ($optional) echo "<tr><td colspan=2><i>Optional - leave blank to not bother with a query</td></tr>";
291: 		if ($row['testtype']=="mysqldata")
292: 			{
293: 			echo "<tr><td align=left>";
294: 			echo "Column :";
295: 			echo "</td><td align=left>";
296: 			echo "<input type=text name=testparam5 size=2 maxlength=8 value=\"".$row['testparam5']."\">";
297: 			echo "</td></tr>";
298: 			echo "<tr><td colspan=2><i>Which field (0 is first and the default) of the first record to use</i></td></tr>";
299: 			}
300: 		echo "</table>";
301: 		}
302: 		
303: 	}
304: $params=array();
305: $NATS->Tests->Register("mysql","FreeNATS_MySQL_Test",$params,"MySQL Connect",2,"FreeNATS MySQL Tester");
306: $NATS->Tests->SetUnits("mysql","Seconds","s");
307: $NATS->Tests->Register("mysqlrows","FreeNATS_MySQL_Test",$params,"MySQL Rows",2,"FreeNATS MySQL Tester");
308: $NATS->Tests->SetUnits("mysqlrows","Rows","rows");
309: $NATS->Tests->Register("mysqldata","FreeNATS_MySQL_Test",$params,"MySQL Data",2,"FreeNATS MySQL Tester");
310: }
311: 
312: 
313: ?>