File:
1.14.1a/server/base/freenats.inc.php (
View as Code)
1: 2: /* -------------------------------------------------------------
3: This file is part of FreeNATS
4:
5: FreeNATS is (C) Copyright 2008-2011 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: class TFreeNATS
24: {
25: var $init=false;
26: var $DB;
27: var $Cfg;
28: var $Tests;
29: var $Lang;
30: var $RSS;
31: var $Version="1.14.1";
32: var $Release="a";
33: var $EventHandlers=array();
34:
35: var $PageErrors=array();
36:
37: function TFreeNATS()
38: {
39: $this->Tests=new TNATS_Tests(); // need this available during the include phase pre-start
40: }
41:
42: function Start()
43: {
44: if ($this->init) return 0;
45: global $BaseDir,$fnSkipSiteIncludes;
46: if ( (!isset($BaseDir)) || ($BaseDir=="") ) $BaseDir="./";
47: $this->DB=new TNATS_DB();
48: $this->Cfg=new TNATS_Cfg();
49: $this->DB->Connect();
50: $this->Cfg->Load($this->DB);
51: // Site Includes
52: // Tests
53: if ( (!isset($fnSkipSiteIncludes)) || ($fnSkipSiteIncludes === false) )
54: {
55:
56: if ($this->Cfg->Get("site.include.tests",0)==1)
57: {
58: foreach(glob($BaseDir."site/tests/*.php") as $phpfile)
59: {
60: $this->Event("Including ".$phpfile,10,"NATS","Start");
61: include_once($phpfile);
62: }
63: }
64: // Events
65: if ($this->Cfg->Get("site.include.events",0)==1)
66: {
67: foreach(glob($BaseDir."site/events/*.php") as $phpfile)
68: {
69: $this->Event("Including ".$phpfile,10,"NATS","Start");
70: include_once($phpfile);
71: }
72: }
73: }
74:
75: // RSS
76: $this->RSS = new NATS_RSS();
77:
78: // Language
79: $this->Lang=new TNATS_Lang();
80: if (isset($_COOKIE['fn_lang']) && ($_COOKIE['fn_lang']!="") ) $l=$_COOKIE['fn_lang'];
81: else $l=$this->Cfg->Get("site.language","");
82: $this->Lang->Load($l);
83:
84:
85: $this->init=true;
86: }
87:
88: function Stop()
89: {
90: $t=$this->Cfg->Get("freenats.tracker","");
91: if ( ($t>0) && ($t!="") )
92: {
93: $day=date("Ymd");
94: if ($this->Cfg->Get("freenats.tracker.last","")!=$day)
95: {
96: $this->Cfg->Set("freenats.tracker.last",$day);
97: $this->PhoneHome();
98: }
99: }
100:
101: $this->DB->Disconnect();
102: $this->init=false;
103: }
104:
105: function Event($logevent,$loglevel=1,$modid="NONE",$catid="NONE")
106: {
107: global $NATS_Session;
108: if ((isset($NATS_Session))&&($NATS_Session->auth)) $username=$NATS_Session->username;
109: else $username="";
110: $l=$this->Cfg->Get("log.level");
111: //echo "** $l **\n";
112: if ( $l=="" ) $l=10; // debug logging if no variable
113: if ( $l < $loglevel ) return false;
114: if (strlen($logevent)>249) $logevent=substr($logevent,0,245)."...";
115: $q="INSERT INTO fnlog(postedx,modid,catid,loglevel,logevent,username) VALUES(".time().",";
116: $q.="\"".ss($modid)."\",\"".ss($catid)."\",".ss($loglevel).",\"".ss($logevent)."\",\"".ss($username)."\")";
117: //echo $q;
118: $this->DB->Query($q,false);
119: }
120:
121: function AlertAction($nodeid,$alertlevel,$change,$alerttext)
122: {
123: //echo "Called for node: ".$nodeid."\n";
124: if ($change==0) return false; // no change
125: if (trim($alerttext)=="") return false; // no text so a bogus event i.e. -1 to 0
126: //echo $nodeid.":".$alertlevel.":".$change.":".$alerttext."\n";
127: // get all the alertactions for this node id
128: $q="SELECT aaid FROM fnnalink WHERE nodeid=\"".ss($nodeid)."\"";
129: $r=$this->DB->Query($q);
130: while ($arow=$this->DB->Fetch_Array($r))
131: {
132: // get details for this alert action
133: $aq="SELECT * FROM fnalertaction WHERE aaid=".$arow['aaid']." LIMIT 0,1";
134:
135: $ar=$this->DB->Query($aq);
136: $aa=$this->DB->Fetch_Array($ar);
137: $this->DB->Free($ar);
138:
139: // UGGGGGGGG continue!!
140: // if the type is blank or disabled skip
141: if ( ($aa['atype']=="") || ($aa['atype']=="Disabled") ) continue;
142: // if warnings aren't set and it is a warning skip
143: if ( ($aa['awarnings']==0) && ($alertlevel==1) ) continue;
144: // if decreases aren't set and it is a decrease skip
145: if ( ($aa['adecrease']==0) && ($change<1) ) continue;
146: // if has a schedule and it dictates not to run now then skip
147: if (($aa['scheduleid']!=0)&&(!run_x_in_schedule(time(),$aa['scheduleid']))) continue;
148:
149: // made it this far
150:
151: $ndata=$nodeid.": ".$alerttext;
152: $this->ActionAddData($arow['aaid'],$ndata);
153:
154: /* // spun to ActionAddData
155: if ($aa['mdata']!="") $ndata=$aa['mdata']."\n".$nodeid.": ".$alerttext;
156: else $ndata=$nodeid.": ".$alerttext;
157: $uq="UPDATE fnalertaction SET mdata=\"".ss($ndata)."\" WHERE aaid=".$arow['aaid'];
158: //echo $uq."\n";
159: $this->DB->Query($uq);
160: */
161: }
162: }
163:
164: function ActionAddData($aaid, $newmdata)
165: {
166: $q="SELECT aaid,mdata FROM fnalertaction WHERE aaid=".ss($aaid)." LIMIT 0,1";
167: $r=$this->DB->Query($q);
168: if (!$row=$this->DB->Fetch_Array($r)) return false;
169: $this->DB->Free($r);
170:
171: if ($row['mdata']!="") $ndata=$row['mdata']."\n".$newmdata;
172: else $ndata=$newmdata;
173:
174: $uq="UPDATE fnalertaction SET mdata=\"".ss($ndata)."\" WHERE aaid=".$row['aaid'];
175: $this->DB->Query($uq);
176: return true;
177: }
178:
179: function ActionFlush()
180: {
181: global $allowed,$BaseDir; // allowed chars from screen in YA BODGE
182: $q="SELECT * FROM fnalertaction WHERE mdata!=\"\"";
183: $r=$this->DB->Query($q);
184: while ($row=$this->DB->Fetch_Array($r))
185: {
186:
187: $doalert=true;
188:
189: // clear mdata right at the start to get around duplicate emails whilst processing
190: $q="UPDATE fnalertaction SET mdata=\"\" WHERE aaid=".$row['aaid'];
191: $this->DB->Query($q);
192:
193: if ($this->DB->Affected_Rows()<=0) // already flushed or failed to flush
194: {
195: $doalert=false;
196: $this->Event("Alert Action Already Flushed - Skipping",8,"Flush","Action");
197: }
198:
199: // alert counter
200: $td=date("Ymd");
201: if ($td!=$row['ctrdate']) // new day or no flush record
202: {
203: $q="UPDATE fnalertaction SET ctrdate=\"".$td."\",ctrtoday=1 WHERE aaid=".$row['aaid'];
204: $this->DB->Query($q);
205: }
206: else
207: {
208:
209: if ( ($row['ctrlimit']==0) || ($row['ctrlimit']>$row['ctrtoday']) ) // no limit or below
210: {
211: $q="UPDATE fnalertaction SET ctrtoday=ctrtoday+1 WHERE aaid=".$row['aaid'];
212: $this->DB->Query($q);
213: }
214: else // at or over limit
215: {
216: $this->Event("Alert Action Limit Reached - Skipping",2,"Flush","Action");
217: $doalert=false;
218: }
219:
220: }
221:
222:
223: if ($row['atype']=="email")
224: {
225: if ($row['esubject']==0) $sub="";
226: else if ($row['esubject']==1) $sub=$this->Cfg->Get("alert.subject.short","FreeNATS Alert");
227: else $sub=$this->Cfg->Get("alert.subject.long","** FreeNATS Alert **");
228: $body="";
229: if ($row['etype']==0) $body=$row['mdata'];
230: else
231: {
232: $body=$this->Cfg->Get("alert.body.header","FreeNATS Alert,");
233: $body.="\r\n\r\n".$row['mdata']."\r\n\r\n";
234: $body.=$this->Cfg->Get("alert.body.footer","");
235: $body.="\r\n-- FreeNATS @ ".nicedt(time());
236: }
237: //$tolist=preg_split("[\n\r]",$row['etolist']);
238: $tolist=array();
239: $f=0;
240: $tolist[0]="";
241: for ($a=0; $a242: {
243: $chr=$row['etolist'][$a];
244: //echo $chr;
245: if (strpos($allowed,$chr)===false) // special char
246: {
247: $f++;
248: $tolist[$f]="";
249: }
250: else
251: {
252: $tolist[$f].=$chr;
253: }
254: }
255:
256: foreach($tolist as $toaddr)
257: {
258: $toaddr=nices($toaddr);
259: if ($toaddr!="")
260: {
261: $smtpserver=$this->Cfg->Get("mail.smtpserver","");
262: if ($smtpserver=="")
263: {
264: // mail() method - local delivery
265: $header="From: ".$row['efrom']."\r\n";
266: if ($doalert)
267: {
268: mail($toaddr,$sub,$body,$header);
269: $this->Event("Sent alert email to ".$toaddr,4,"Flush","Email");
270: }
271: }
272: else // use phpmailer direct SMTP delivery
273: {
274: include_once($BaseDir."phpmailer/class.phpmailer.php");
275: $fromname=$this->Cfg->Get("mail.fromname","");
276: if ($fromname=="") $fromname="FreeNATS";
277: $smtpusername=$this->Cfg->Get("mail.smtpusername",""); // removed .
278: if ($smtpusername!="") $smtpauth=true;
279: else $smtpauth=false;
280: $smtppassword=$this->Cfg->Get("mail.smtppassword",""); // removed .
281: $smtphostname=$this->Cfg->Get("mail.smtphostname",""); // removed .
282: $smtpsec=$this->Cfg->Get("mail.smtpsecure","");
283: $mail=new PHPMailer();
284: $mail->IsSMTP();
285: $mail->Host=$smtpserver;
286: $mail->From=$row['efrom'];
287: $mail->FromName=$fromname;
288: $mail->AddAddress($toaddr);
289: $mail->Subject=$sub;
290: $mail->Body=$body;
291: $mail->WordWrap=50;
292: if ($smtphostname!="") $mail->Hostname=$smtphostname;
293: if ($smtpauth)
294: {
295: $mail->SMTPAuth=true;
296: $mail->Username=$smtpusername;
297: $mail->Password=$smtppassword;
298: }
299: if ($smtpsec!="") $mail->SMTPSecure=$smtpsec;
300: if (!$mail->Send())
301: { // failed
302: $this->Event("phpMailer to ".$toaddr." failed",1,"Flush","Email");
303: $this->Event("phpMailer Error: ".$mail->ErrorInfo,1,"Flush","Email");
304: }
305: else
306: {
307: $this->Event("phpMailer Sent Email To ".$toaddr,4,"Flush","Email");
308: }
309: }
310:
311: }
312: }
313:
314:
315:
316: }
317: else if ($row['atype']=="url")
318: {
319: // url send
320: if ($row['etype']==0) $body=$row['mdata'];
321: else $body="FreeNATS Alert,\r\n".$row['mdata']."\r\n--FreeNATS @ ".nicedt(time());
322:
323: $body=urlencode($body);
324: $tolist=array();
325: $f=0;
326: $tolist[0]="";
327: for ($a=0; $a328: {
329: $chr=$row['etolist'][$a];
330: //echo $chr;
331: if (strpos($allowed,$chr)===false) // special char
332: {
333: $f++;
334: $tolist[$f]="";
335: }
336: else
337: {
338: $tolist[$f].=$chr;
339: }
340: }
341:
342: foreach($tolist as $tourl)
343: {
344: if ($doalert)
345: {
346: $url=$tourl.$body;
347: $fp=@fopen($url,"r");
348: if ($fp>0) fclose($fp);
349: else $this->Event("URL Alert Failed ".$url,1,"Flush","URL");
350: $this->Event("URL Alert ".$url,4,"Flush","URL");
351: }
352: }
353:
354:
355: }
356: else if ($row['atype']=="mqueue")
357: {
358: // message queue
359: $eventdata=array("aaid"=>$row['aaid'],"name"=>$row['aname'],"data"=>$row['mdata']);
360: $result=$this->EventHandler("alert_action",$eventdata);
361: if ($result===false) // put the data back into the queue
362: {
363: $q="UPDATE fnalertaction SET mdata=\"".$row['mdata']."\"+mdata WHERE aaid=".$row['aaid'];
364: $this->DB->Query($q);
365: if ($this->DB->Affected_Rows()<=0)
366: $this->Event("Persist MDATA Failed for AAID ".$row['aaid'],2,"Flush","MQueue");
367: }
368: else $this->Event("Queue Cleared for AAID ".$row['aaid']." by Handler",4,"Flush","MQueue");
369: }
370:
371: }
372: }
373:
374: function GetAlerts()
375: {
376: $q="SELECT nodeid,alertlevel FROM fnalert WHERE closedx=0";
377: $r=$this->DB->Query($q);
378: $c=0;
379: $al=array();
380: while ($row=$this->DB->Fetch_Array($r))
381: {
382: $al[$c]['nodeid']=$row['nodeid'];
383: $al[$c]['alertlevel']=$row['alertlevel'];
384: $c++;
385: }
386: if ($c>0) return $al;
387: else return false;
388: }
389:
390: function SetAlerts($nodeid,$alertlevel,$alerts="")
391: {
392: if ($alerts=="") $alerts=array();
393: // get current alert level
394: $q="SELECT alertlevel,nodealert FROM fnnode WHERE nodeid=\"".ss($nodeid)."\"";
395: $r=$this->DB->Query($q);
396: $row=$this->DB->Fetch_Array($r);
397: $this->DB->Free($r);
398: $cal=$row['alertlevel'];
399:
400: $eventdata=array("nodeid"=>$nodeid,"alertlevel"=>$alertlevel,
401: "oldalertlevel"=>$cal);
402: $this->EventHandler("set_alerts",$eventdata);
403:
404: if ($alertlevel!=$cal)
405: {
406: // update table
407: $q="UPDATE fnnode SET alertlevel=".ss($alertlevel)." WHERE nodeid=\"".ss($nodeid)."\"";
408: $this->DB->Query($q);
409: }
410:
411: // do not continue if node alert isn't set
412: if ($row['nodealert']!=1) return 0;
413: // or if untested
414: if ($alertlevel<0) return 0;
415:
416: // ALERTS
417: // is there an existing alert for this node
418: $q="SELECT alertid,alertlevel FROM fnalert WHERE nodeid=\"".ss($nodeid)."\" AND closedx=0";
419: $r=$this->DB->Query($q);
420: if ($row=$this->DB->Fetch_Array($r))
421: { // yes there is
422: // if new alert level is 0 let's close it
423: if ($alertlevel==0)
424: {
425: $alertid=$row['alertid'];
426: $q="UPDATE fnalert SET closedx=".time()." WHERE alertid=".$row['alertid'];
427: $this->DB->Query($q);
428: $closetext=$this->Cfg->Get("site.text.closed","Alert Closed");
429: if (is_array($alerts)) $alerts[]=$closetext;
430: else
431: {
432: $alerts=array($alerts); // add as first element to new array
433: $alerts[]=$closetext;
434: }
435: $eventdata=array("nodeid"=>$nodeid);
436: $this->EventHandler("alert_close",$eventdata);
437: }
438: else
439: {
440: $alertid=$row['alertid'];
441: // otherwise update the alert to the new value (was: regardless, now just if not a 0)
442: $q="UPDATE fnalert SET alertlevel=".ss($alertlevel)." WHERE alertid=".$alertid;
443: $this->DB->Query($q);
444: }
445: }
446: else
447: { // no there's not
448: $cal=0; // the cal (current alert level) goes to zero if it's a new alert so alert_actions fire ok
449: if ($alertlevel>0) // only if an actual alert
450: {
451: $q="INSERT INTO fnalert(nodeid,alertlevel,openedx) VALUES(";
452: $q.="\"".ss($nodeid)."\",".ss($alertlevel).",".time().")";
453: $this->DB->Query($q);
454: $alertid=$this->DB->Insert_Id();
455: $eventdata=array("nodeid"=>$nodeid);
456: $this->EventHandler("alert_open",$eventdata);
457: }
458: }
459: // ALERT LOG with $alertid
460: $t=time();
461: $at="";
462: if (is_array($alerts))
463: {
464: foreach($alerts as $alert)
465: {
466: if (isset($alertid)) // misses on manual runs methinx
467: {
468: if ($at!="") $at.=", ";
469: $at.=$alert;
470: //echo $at."\n";
471: $iq="INSERT INTO fnalertlog(alertid,postedx,logentry) VALUES(";
472: $iq.=$alertid.",".$t.",\"".ss($alert)."\")";
473: //echo $iq;
474: $this->DB->Query($iq);
475: }
476: }
477: }
478:
479: $this->AlertAction($nodeid,$alertlevel,$alertlevel-$cal,$at);
480:
481:
482:
483: }
484:
485: function NodeAlertLevel($nodeid)
486: {
487: $q="SELECT alertlevel FROM fnnode WHERE nodeid=\"".ss($nodeid)."\"";
488: $r=$this->DB->Query($q);
489: if ($row=$this->DB->Fetch_Array($r)) return $row['alertlevel'];
490: else return -1;
491: }
492:
493: function GroupAlertLevel($groupid)
494: {
495: $lvl=-1;
496: $q="SELECT nodeid FROM fngrouplink WHERE groupid=\"".ss($groupid)."\"";
497: $r=$this->DB->Query($q);
498: while ($row=$this->DB->Fetch_Array($r))
499: {
500: $nl=$this->NodeAlertLevel($row['nodeid']);
501: if ($nl>$lvl) $lvl=$nl;
502: }
503: $this->DB->Free($r);
504: return $lvl;
505: }
506:
507: function PhoneHome($mode=0,$type="ping") // 0 - php, 1 - html, 2 - data
508: {
509: if ($mode<2)
510: {
511: $qs="?type=".$type."&data=v=".$this->Version;
512: if (isset($_SERVER['REMOTE_ADDR']))
513: $qs.=",ip=".$_SERVER['REMOTE_ADDR'];
514: $ploc="http://www.purplepixie.org/freenats/report/";
515: if ($mode==1) $ploc.="ping.html";
516: else $ploc.="ping.php";
517:
518: $ploc.=$qs;
519:
520: $lp=@fopen($ploc,"r");
521: if ($lp>0) @fclose($lp);
522: }
523: else
524: {
525: // data post -- !!
526: }
527: }
528:
529: function GetNode($nodeid)
530: {
531: $return_row=false;
532: $q="SELECT * FROM fnnode WHERE nodeid=\"".ss($nodeid)."\" LIMIT 0,1";
533: $r=$this->DB->Query($q);
534: if ($row=$this->DB->Fetch_Array($r))
535: $return_row=true;
536:
537: $this->DB->Free($r);
538: if ($return_row) // found a valid
539: {
540: if ($row['nodename']!="") $row['name']=$row['nodename']; // make a "nice" name for it
541: else $row['name']=$row['nodeid'];
542:
543: $row['alerttext']=oText($row['alertlevel']); // textual alert status
544:
545: $row['lastrundt']=nicedt($row['lastrunx']); // text date-time last run
546: $row['lastrunago']=dtago($row['lastrunx'],false); // last run ago
547:
548: // protection
549: $row['nskey']="";
550:
551: return $row;
552: }
553: else
554: return false; // or failed
555: }
556:
557: function GetNodes()
558: {
559: $out=array();
560: $q="SELECT * FROM fnnode";
561: $r=$this->DB->Query($q);
562:
563: while ($row=$this->DB->Fetch_Array($r))
564: {
565: if ($row['nodename']!="") $row['name']=$row['nodename']; // make a "nice" name for it
566: else $row['name']=$row['nodeid'];
567:
568: $row['alerttext']=oText($row['alertlevel']); // textual alert status
569:
570: $row['lastrundt']=nicedt($row['lastrunx']); // text date-time last run
571: $row['lastrunago']=dtago($row['lastrunx'],false); // last run ago
572:
573: // protection
574: $row['nskey']="";
575:
576: $out[$row['nodeid']]=$row;
577: }
578: $this->DB->Free($r);
579:
580: return $out;
581: }
582:
583:
584: function GetNodeTests($nodeid)
585: { // returns an array of testids for the node (enabled tests only)
586: $tests=array();
587:
588: // local tests
589: $q="SELECT localtestid FROM fnlocaltest WHERE testenabled=1 AND nodeid=\"".ss($nodeid)."\" ORDER BY localtestid ASC";
590: $r=$this->DB->Query($q);
591: while ($row=$this->DB->Fetch_Array($r))
592: {
593: $tests[]="L".$row['localtestid'];
594: }
595: $this->DB->Free($r);
596:
597: // nodeside
598: $q="SELECT nstestid FROM fnnstest WHERE testenabled=1 AND nodeid=\"".ss($nodeid)."\" ORDER BY testtype ASC";
599: $r=$this->DB->Query($q);
600: while ($row=$this->DB->Fetch_Array($r))
601: {
602: $tests[]="N".$row['nstestid'];
603: }
604: $this->DB->Free($r);
605:
606: return $tests;
607: }
608:
609: function SetNode($nodeid,$data)
610: {
611: $q="UPDATE fnnode SET ";
612: $first=true;
613: foreach($data as $key => $val)
614: {
615: if ($first) $first=false;
616: else $q.=",";
617: $q.=ss($key)."=\"".ss($val)."\"";
618: }
619: $q.=" WHERE nodeid=\"".ss($nodeid)."\"";
620: $this->DB->Query($q);
621: if ($this->DB->Affected_Rows()>0) return true;
622:
623: if ($this->DB->Error()) // query failed
624: {
625: $errstr1="Query Failed: ".$q;
626: $errstr2="Query Failed: ".$this->DB->Error_String();
627: $this->Event($errstr1,2,"Node","Set");
628: $this->Event($errstr1,2,"Node","Set");
629: return false;
630: }
631: return true; // query succeeded but nothing was updated
632: }
633:
634: function EnableNode($nodeid,$enabled=true)
635: {
636: if ($enabled) $ne=1;
637: else $ne=0;
638: $data=array("nodeenabled"=>$ne);
639: return $this->SetNode($nodeid,$data);
640: }
641:
642: function DisableNode($nodeid)
643: {
644: return $this->EnableNode($nodeid,false);
645: }
646:
647: function SetNodeSchedule($nodeid,$scheduleid)
648: {
649: $data=array("scheduleid"=>$scheduleid);
650: return $this->SetNode($nodeid,$data);
651: }
652:
653: function GetGroup($groupid)
654: {
655: $q="SELECT * FROM fngroup WHERE groupid=".ss($groupid)." LIMIT 0,1";
656: $r=$this->DB->Query($q);
657: if (!$row=$this->DB->Fetch_Array($r)) return false;
658:
659: $this->DB->Free($r);
660: $row['alertlevel']=$this->GroupAlertLevel($groupid);
661: $row['alerttext']=oText($row['alertlevel']);
662: return $row;
663: }
664:
665: function GetGroups()
666: {
667: $out=array();
668: $q="SELECT * FROM fngroup";
669: $r=$this->DB->Query($q);
670:
671: while ($row=$this->DB->Fetch_Array($r))
672: {
673: $row['alertlevel']=$this->GroupAlertLevel($row['groupid']);
674: $row['alerttext']=oText($row['alertlevel']);
675: $out[$row['groupid']]=$row;
676: }
677:
678: $this->DB->Free($r);
679: return $out;
680: }
681:
682: function GetTest($testid,$protect=false)
683: {
684: if ($testid=="") return false;
685: $class=$testid[0];
686: if (is_numeric($class))
687: {
688: // test ID will stay the same
689: $class="L";
690: $anytestid=$testid;
691: }
692: else
693: {
694: //$testid=substr($testid,1); // as it will here also so direct use to graphs can be made
695: $anytestid=substr($testid,1); // the classless version
696: }
697:
698: $q="";
699: switch($class)
700: {
701: case "L": // local tests
702: $q="SELECT * FROM fnlocaltest WHERE localtestid=".ss($anytestid)." LIMIT 0,1";
703: break;
704: case "N": // node-side test
705: $q="SELECT * FROM fnnstest WHERE nstestid=".ss($anytestid)." LIMIT 0,1";
706: break;
707: default:
708: return false; // can't lookup this class
709: }
710:
711: if ($q=="") return false;
712:
713: $r=$this->DB->Query($q);
714:
715: if (!$row=$this->DB->Fetch_Array($r)) return false;
716:
717: $row['class']=$class;
718: $row['testid']=$testid;
719: $row['anytestid']=$anytestid;
720: $row['alerttext']=oText($row['alertlevel']);
721: $row['lastrundt']=nicedt($row['lastrunx']);
722: $row['lastrunago']=dtago($row['lastrunx'],false);
723:
724: if ($row['testname']!="") $row['name']=$row['testname'];
725: else
726: {
727: if ($class=="L")
728: {
729: $row['name']=lText($row['testtype']); // TODO OTHER TESTS
730: if ($row['testparam']!="") $row['name'].=" (".$row['testparam'].")";
731: }
732: else if ($class=="N")
733: {
734: if ($row['testdesc']!="") $row['name']=$row['testdesc'];
735: else $row['name']=$row['testtype'];
736: }
737: }
738:
739: if ($protect&&($class=="L")) // module test protection
740: {
741: if ($this->Tests->Exists($row['testtype'])) // in the module register
742: {
743: $this->Tests->Tests[$row['testtype']]->Create();
744: $this->Tests->Tests[$row['testtype']]->instance->ProtectOutput($row);
745: }
746: }
747:
748: $this->DB->Free($r);
749:
750: return $row;
751: }
752:
753:
754: function DeleteTest($testid)
755: {
756: if ($testid=="") return false;
757: $class=$testid[0];
758: if (is_numeric($class))
759: {
760: // test ID will stay the same
761: $class="L";
762: $anytestid=$testid;
763: }
764: else
765: {
766: $anytestid=substr($testid,1); // the classless version
767: }
768:
769: $q="";
770: switch($class)
771: {
772: case "L": // local tests
773: $q="DELETE FROM fnlocaltest WHERE localtestid=".ss($anytestid);
774: break;
775: case "N": // node-side test
776: $q="DELETE FROM fnnstest WHERE nstestid=".ss($anytestid);
777: break;
778: default:
779: return false; // can't lookup this class
780: }
781:
782: if ($q=="") return false;
783:
784: $this->DB->Query($q);
785: $tests=$this->DB->Affected_Rows();
786:
787: $rq="DELETE FROM fnrecord WHERE testid=\"".ss($testid)."\"";
788: $this->DB->Query($rq);
789: $records=$this->DB->Affected_Rows();
790:
791: $eq="DELETE FROM fneval WHERE testid=\"".ss($testid)."\"";
792: $this->DB->Query($eq);
793: $eval=$this->DB->Affected_Rows();
794:
795: $s="Deleted test ".$testid." (".$tests." tests, ".$records." records, ".$eval." evaluators)";
796: $this->Event($s,6,"Test","Delete");
797: }
798:
799:
800:
801: function InvalidateTest($testid,$rightnow=false)
802: {
803: $class=$testid[0];
804: if (is_numeric($class)) $class="L";
805: else $testid=substr($testid,1);
806: if ($rightnow)
807: {
808: $nextx=time();
809: $q="UPDATE ";
810: if ($class=="L") $q.="fnlocaltest";
811: // other ones here
812:
813: $q.=" SET nextrunx=".$nextx." WHERE ";
814:
815: if ($class=="L") $q.="localtestid=".$testid;
816: // other ones here
817:
818: $this->DB->Query($q);
819: return true;
820: }
821: // otherwise use it's interval
822: $q="SELECT testinterval FROM ";
823:
824: if ($class=="L") $q.="fnlocaltest WHERE localtestid=";
825: // other ones here
826:
827: $q.=$testid;
828: $r=$this->DB->Query($q);
829: if ($row=$this->DB->Fetch_Array($r))
830: {
831: $this->DB->Free($r);
832: $nextx=next_run_x($row['testinterval']);
833: $q="UPDATE ";
834: if ($class=="L") $q.="fnlocaltest";
835: // other ones here
836:
837: $q.=" SET nextrunx=".$nextx." WHERE ";
838:
839: if ($class=="L") $q.="localtestid=".$testid;
840: // other ones here
841:
842: $this->DB->Query($q);
843: return true;
844: }
845: return false;
846: }
847:
848: function InvalidateNode($nodeid,$rightnow=false,$testsaswell=false)
849: {
850: if ($rightnow)
851: {
852: $nextx=time();
853: $q="UPDATE fnnode SET nextrunx=".$nextx." WHERE nodeid=\"".ss($nodeid)."\"";
854: $this->DB->Query($q);
855: if ($testsaswell)
856: {
857: $q="UPDATE fnlocaltest SET nextrunx=".$nextx." WHERE nodeid=\"".ss($nodeid)."\"";
858: $this->DB->Query($q);
859: }
860: return true;
861: }
862: // otherwise set to it's interval
863: $q="SELECT testinterval FROM fnnode WHERE nodeid=\"".ss($nodeid)."\"";
864: $r=$this->DB->Query($q);
865: if ($row=$this->DB->Fetch_Array($r))
866: {
867: $nextx=next_run_x($row['testinterval']);
868: $uq="UPDATE fnnode SET nextrunx=".$nextx." WHERE nodeid=\"".ss($nodeid)."\"";
869: $this->DB->Query($uq);
870: $this->DB->Free($r);
871: if ($testsaswell)
872: {
873: $uq="UPDATE fnlocaltest SET nextrunx=".$nextx." WHERE nodeid=\"".ss($nodeid)."\"";
874: $this->DB->Query($uq);
875: }
876: return true;
877: }
878: return false;
879: }
880:
881:
882: function NodeSide_Pull($nodeid)
883: {
884: $eventdata=array("nodeid"=>$nodeid,"success"=>false);
885: $q="SELECT nsenabled,nspullenabled,nsurl,nskey,nsinterval FROM fnnode WHERE nodeid=\"".ss($nodeid)."\" LIMIT 0,1";
886: $r=$this->DB->Query($q);
887: if (!$row=$this->DB->Fetch_Array($r)) return false;
888:
889: $this->DB->Free($r);
890:
891: $url=$row['nsurl'];
892: if ($row['nskey']!="") $url.="?nodekey=".$row['nskey'];
893: //echo $url."\n";
894: $this->Event("NodeSide_Pull Started for ".$nodeid,10,"Node","Pull");
895:
896: $xmlobj=new TNodeXML();
897:
898: $fp=@fopen($url,"r");
899: if ($fp<=0)
900: {
901: $this->Event("Pull Failed URL ".$url,1,"Node","Pull");
902: $this->EventHandler("nodeside_pull",$eventdata);
903: return false;
904: }
905: $xml="";
906: while (!feof($fp))
907: {
908: $xml.=fgets($fp,4096);
909: }
910: if ($xml=="")
911: {
912: $this->EventHandler("nodeside_pull",$eventdata);
913: return false;
914: }
915:
916: //echo $xml;
917:
918: $result=$xmlobj->Parse($xml);
919:
920: if ($xmlobj->Error()!="")
921: {
922: $this->Event("NodeXML Error: ".$xmlobj->Error(),1,"Node","Pull");
923: $this->EventHandler("nodeside_pull",$eventdata);
924: return false;
925: }
926: $this->Event("NodeSide_Pull Fetched ".$xmlobj->Tests." tests for ".$nodeid,10,"Node","Pull");
927: // Now just to actually process it...
928: $eventdata['success']=true;
929: $this->EventHandler("nodeside_pull",$eventdata);
930: $this->NodeSide_Process($nodeid,$xmlobj);
931: return true;
932: }
933:
934: function NodeSide_Process($nodeid,&$xmlobj)
935: { // nodeid + takes a TNodeXML Object
936: $alvl=0;
937: $this->Event("NodeSide_Process for ".$nodeid,10,"Node","Pull");
938: $q="SELECT * FROM fnnstest WHERE nodeid=\"".ss($nodeid)."\"";
939: $r=$this->DB->Query($q);
940: $tests=array();
941: while ($row=$this->DB->Fetch_Array($r))
942: {
943: $tests[$row['testtype']]=$row;
944: if (isset($xmlobj->Catalogue[$row['testtype']]))
945: { // this test is in the DB and catalogue
946: $tests[$row['testtype']]['incat']=true;
947: if ($row['testenabled']==1) // it is enabled - so we test it
948: {
949: if ($row['simpleeval']==1) $level=$xmlobj->Catalogue[$row['testtype']]['ALERTLEVEL']; // use provided level
950: else $level=nats_eval("N".$row['nstestid'],$xmlobj->Catalogue[$row['testtype']]['VALUE']);
951: $dbs="Nodeside ".$row['testtype']." on ".$row['nodeid']." = ".$level;
952: if ($level==0) $debuglev=8;
953: else if ($level>0) $debuglev=5;
954: else $debuglev=2;
955: $this->Event($dbs,$debuglev,"Node","Process");
956:
957: if ($level>$alvl) $alvl=$level;
958:
959: if ($row['testrecord']==1) // record it
960: {
961: $testvalue=$xmlobj->Catalogue[$row['testtype']]['VALUE'];
962: $testvalue=str_replace(",",".",$testvalue);
963: if (!is_numeric($testvalue)) $testvalue=0;
964: $iq="INSERT INTO fnrecord(testid,alertlevel,recordx,nodeid,testvalue) VALUES(";
965: $iq.="\"N".$row['nstestid']."\",".$level.",".time().",\"".$row['nodeid']."\",".$testvalue.")";
966: $this->DB->Query($iq);
967: if ($this->DB->Affected_Rows()<=0)
968: $this->Event("Nodeside ".$row['testtype']." Failed to Record",1,"Node","Process");
969: }
970:
971: // We don't do any alerting here - the tester will do that for us!
972: $uq="UPDATE fnnstest SET lastrunx=".time().",lastvalue=\"".ss($xmlobj->Catalogue[$row['testtype']]['VALUE'])."\",alertlevel=".$level." ";
973: $uq.="WHERE nstestid=".$row['nstestid'];
974: $this->DB->Query($uq);
975: if ($this->DB->Affected_Rows()<=0)
976: $this->Event("Nodeside ".$row['testtype']." Failed to Update or Same Values",5,"Node","Process");
977:
978: }
979:
980: // check to see if the desc has changed
981: if ($row['testdesc']!=$xmlobj->Catalogue[$row['testtype']]['DESC'])
982: {
983: $duq="UPDATE fnnstest SET testdesc=\"".ss($xmlobj->Catalogue[$row['testtype']]['DESC'])."\" WHERE nstestid=".$row['nstestid'];
984: $this->DB->Query($duq);
985: }
986:
987: }
988: else
989: {
990: // test in the DB but NOT in the catalogue
991: //$xmlobj->Catalogue[$row['testtype']]['incat']=false;
992: if ($row['testenabled']==1)
993: { // enabled so shown in lists etc
994: // Update it to show failed status
995: $this->Event("No nodeside data for test N".$row['nstestid'],3,"Node","Process");
996: $uq="UPDATE fnnstest SET alertlevel=2,lastvalue=-1 WHERE nstestid=".$row['nstestid'];
997: $this->DB->Query($uq);
998: $alvl=2;
999: if ($row['testrecord']==1) // record it
1000: {
1001: $testvalue=-1;
1002: $iq="INSERT INTO fnrecord(testid,alertlevel,recordx,nodeid,testvalue) VALUES(";
1003: $iq.="\"N".$row['nstestid']."\",2,".time().",\"".$row['nodeid']."\",".$testvalue.")";
1004: $this->DB->Query($iq);
1005: if ($this->DB->Affected_Rows()<=0)
1006: $this->Event("Nodeside ".$row['testtype']." Failed to Record",1,"Node","Process");
1007: }
1008: }
1009: else // not enabled so simply delete
1010: {
1011: $this->DeleteTest("N".$row['nstestid']);
1012: }
1013: }
1014: }
1015: $this->DB->Free($r);
1016:
1017:
1018: // and finally we look for new tests i.e. in the cat but not in the DB
1019: foreach($xmlobj->Catalogue as $val)
1020: {
1021: $key=$val['NAME'];
1022: if (!isset($tests[$key])) // not in the DB
1023: {
1024: $q="INSERT INTO fnnstest(nodeid,testtype,testdesc,lastvalue,lastrunx,alertlevel) ";
1025: $q.="VALUES(\"".ss($nodeid)."\",\"".$key."\",\"".ss($val['DESC'])."\",\"".ss($val['VALUE'])."\",".time().",".ss($val['ALERTLEVEL']).")";
1026: //echo $q."
";
1027: $this->DB->Query($q);
1028: }
1029: }
1030:
1031: $eventdata=array("nodeid"=>$nodeid,"alertlevel"=>$alvl);
1032: $this->EventHandler("nodeside_process",$eventdata);
1033:
1034:
1035: }
1036:
1037:
1038: function AddEventHandler($event,$function)
1039: {
1040: if (!isset($this->EventHandlers[$event])) $this->EventHandlers[$event]=array();
1041: $this->EventHandlers[$event][]=$function;
1042: }
1043:
1044: function EventHandler($event,$data)
1045: {
1046: $res=false;
1047: if ( isset($data) && is_array($data) ) $data['event']=$event;
1048:
1049: if (isset($this->EventHandlers[$event])) // handler(s) exist
1050: {
1051: for($a=0; $aEventHandlers[$event]); $a++)
1052: {
1053: if (function_exists($this->EventHandlers[$event][$a]))
1054: {
1055: $this->Event("Event ".$event." -> ".$this->EventHandlers[$event][$a],6,"Event","Handler");
1056: if($this->EventHandlers[$event][$a]($data)) $res=true; // persist true only
1057: }
1058: else
1059: {
1060: $t="Illegal Handler ".$this->EventHandlers[$event][$a]." for ".$event;
1061: $this->Event($t,2,"Event","Handler");
1062: //return false;
1063: }
1064: }
1065: }
1066: else return $res;
1067: }
1068:
1069: function StripGPC($data)
1070: {
1071: if (get_magic_quotes_gpc()) return stripslashes($data);
1072: else return $data;
1073: }
1074:
1075: function PageError($code,$desc)
1076: {
1077: $this->PageErrors[]=array( "code" => $code, "desc" => $desc );
1078: }
1079:
1080: }
1081: ?>
1082: