gecko-dev/tools/tests/viewer/mtbf/mtbf.cpp

316 строки
8.5 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/*
* The contents of this file are subject to the Netscape Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is mozilla.org code.
*
* The Initial Developer of the Original Code is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
*/
/***************************************************************************************
* MODULE NOTES -- MTBF -- Mean Time Between Failures
* @update gess 01/04/99
* @update cyeh 03/15/99
*
* MTBF is a simple WinNT application that runs our viewer.exe remotely.
* After viewer crashes (inevitably) the total elapsed time is recorded, and send
* in an HTML mail message to the appropriate recipient. By default, the message
* is sent to rickg, but you can cause it to be sent to anywhere you like.
*
* Note: To compile this program using the existing project file, simply create a
* subst drive on your system using s:\ as the root. Place this project
* there as s:\mtbf. Of course, if this doesn't work for you, it's easy
* to change since there's only one file in the whole project anyway.
*
* Note: We depend on the availability of the blat.exe somewhere in your path.
* You can get your own copy of BLAT at http://gepasi.dbs.aber.ac.uk/softw/Blat.html.
*
***************************************************************************************/
#include <iostream.h>
#include <fstream.h>
#include <process.h>
#include <stdlib.h>
#include <errno.h>
#include <time.h>
#include <stdio.h>
#include <string.h>
#include <direct.h>
const char* gStatus[]={"abnormally","normally"};
bool gSendMail=false;
const char* gDefDir="s:";
char gMTBFDirectory[500]="";
char gDirectory[10];
const char* gDefProgram="viewer.exe";
char gProgram[500];
char gSubject[]="\"Automated MTBF Report\"";
char gRecipient[500]="rgess@san.rr.com";
char gReportFilename[20]="report.html";
char gReportPath[500]="";
const char* gURLFilename="urls.txt";
char gURLPath[500];
int gLoop = 100;
void ExitWithError(const char* aMsg1,const char* aMsg2){
cout << endl << "Error: " << aMsg1;
if(aMsg2)
cout<< " " << aMsg2;
cout <<"."<< endl;
cout << "Process terminated." << endl;
exit(2000);
}
//-d 10 -f c:/apps/utils/urls.txt -m
/**
*
* @update gess 01/04/99
* @param
* @return
*/
void ComputeEXEDirectory(char* aProgramPath,char* anOutputFilename){
char* thePtr=strrchr(aProgramPath,'\\');
if(thePtr) {
strcpy(anOutputFilename,aProgramPath);
anOutputFilename[thePtr-aProgramPath+1]=0;
}
thePtr=strchr(anOutputFilename,'\\');
while(thePtr) {
*thePtr='/';
thePtr=strchr(anOutputFilename,'\\');
}
}
/**
*
* @update gess 01/04/99
* @param
* @return
*/
void initializeSettings(int argc,char* argv[]){
bool theURLFileIsValid=false;
ComputeEXEDirectory(argv[0],gMTBFDirectory);
sprintf(gReportPath,"%s%s",gMTBFDirectory,gReportFilename);
sprintf(gURLPath,"%s%s",gMTBFDirectory,gURLFilename);
for(int i=0;i<argc;i++){
if(0==strnicmp(argv[i],"-f",2)){
//determine which url file to use...
strcpy(gURLPath, argv[i+1]);
fstream in(gURLPath,ios::in);
char buf[500];
buf[0]=0;
in>>buf;
if(!buf[0]){
ExitWithError("Unable to read the given URL file",gURLPath);
}
else {
theURLFileIsValid=true;
}
}
else if(0==strnicmp(argv[i],"-r",2)){
//determine email recipient...
strcpy(gRecipient,argv[i+1]);
gSendMail=true;
}
else if(0==strnicmp(argv[i],"-l",2)){
gLoop = atoi(argv[i+1]);
if ((gLoop > 200) || (gLoop < 0))
ExitWithError("Loop must be a positive value and less than 200", 0);
}
}
if(!theURLFileIsValid)
ExitWithError("You must provide a valid URL textfile",0);
}
/**
*
* @update gess 01/04/99
* @param
* @return
*/
void PrintResultsAsHTML(ostream& aStream,time_t& aStart,time_t& aEnd,int aResult){
time_t ltime;
time( &ltime );
//begin by dumping the MIME header...
/*
aStream << "MIME-Version: 1.0" << endl;
aStream << "Content-Type: text/html; charset=us-ascii" << endl;
aStream << "Subject: Automated MTBF Report" << endl;
// aStream << "To: rick@gessner.com rickg@netscape.com mozilla-layout@mozilla.org" << endl;
*/
int total=aEnd-aStart;
int mins=(aEnd-aStart)/60;
char buffer[50];
sprintf(buffer,"%i mins %i sec",mins,total%60);
aStream << "Automated MTBF Report"<< endl;
aStream << ctime(&ltime) << endl;
aStream << "Running: "<< gDirectory << gProgram << endl;
aStream << "Elapsed: "<< buffer << endl;
aStream << "Exit: "<< gStatus[0==aResult] << endl;
}
/**
*
* @update gess 01/04/99
* @param
* @return
*/
void PrintResults(ostream& aStream,time_t& aStart,time_t& aEnd,int aResult){
time_t ltime;
time( &ltime );
aStream << "=========================================" << endl;
aStream << "Gecko \"Mean Time Between Failure\" Report" << endl;
aStream << "=========================================" << endl;
aStream << "Date: " << ctime( &ltime);
aStream << "Running: " << gDirectory << gProgram << endl;
aStream << "File: " << gURLPath << endl;
int total=aEnd-aStart;
int mins=(aEnd-aStart)/60;
aStream << "MTBF: ";
if(mins>0){
aStream << mins << " min ";
}
const char* status[]={"abnormally","normally"};
aStream << total%60 << " seconds" << endl;
aStream << "Exit: Program terminated " << status[0==aResult] << endl;
}
/**
*
* @update gess 01/04/99
* @param
* @return
*/
bool getPath(char* aBuffer,const char* aFilename,int index){
bool result=false;
if(aFilename){
int len=strlen(aFilename);
if(len>0){
int hit=0;
int pos=len;
while((hit<index) && (--pos>-1)){
if((aFilename[pos]=='\\') || (aFilename[pos]=='/')){
hit++;
}//if
}//while
if(hit==index){
strncpy(aBuffer,aFilename,pos);
aBuffer[pos]=0;
strcat(aBuffer,"\\cache");
result=true;
}
}//if
}//if
return result;
}
/**
*
* @update gess 01/04/99
* @param
* @return
*/
int main(int argc,char* argv[]) {
int result=0;
//result=system("sendmail -t -messagefile=s:/mtbf/msg.txt ");
if(argc>1) {
/*
char* theDir = getenv("moz_src");
if(!theDir)
theDir=getenv("homedrive");
if(theDir)
strcpy(gDirectory,theDir);
else strcpy(gDirectory,gDefDir);
*/
strcpy(gProgram,gDefProgram);
initializeSettings(argc,argv);
// example for CTime::CTime
time_t startTime; // C run-time time (defined in <time.h>)
time( &startTime ) ; // Get the current time from the
char buffer[500];
sprintf(buffer,"%s%s -f %s -r %d",gDirectory,gProgram,gURLPath,gLoop);
result=system(buffer);
time_t endTime; // C run-time time (defined in <time.h>)
time( &endTime ) ; // Get the current time from system
if(gSendMail) {
{
//Write the message out in html form...
fstream out(gReportPath,ios::out);
PrintResultsAsHTML(out,startTime,endTime,result);
}
sprintf(buffer,"blat %s -s %s -t %s",gReportPath,gSubject,gRecipient);
result=system(buffer);
}
else {
cout << endl << endl;
PrintResults(cout,startTime,endTime,result);
}
int err=errno;
switch(errno){
case E2BIG:
cout << "too big" << endl; break;
case EACCES:
cout << "access" << endl; break;
case EMFILE:
cout << "file" << endl; break;
case ENOENT:
cout << "bad file or directory" << endl; break;
case ENOEXEC:
cout << "exec" << endl; break;
case ENOMEM:
cout << "no mem" << endl; break;
default:
break;
}
}
else {
printf("MTBF (Mean time between failures) version 1.1.\n");
printf("This program requires BLAT.EXE, which you can find on the net.\n");
printf("http://gepasi.dbs.aber.ac.uk/softw/Blat.html \n");
printf("Optional arguments: \n");
printf(" -f filename containing URL list\n");
printf(" -r email recipient\n");
printf(" -l loop # of times (default is 100, max is 200)\n");
}
return 0;
}