HeaderInfo.C

Go to the documentation of this file.

#include <errno.h>

#include "SpamUtil.h"
#include "Logger.h"
#include "HeaderInfo.h"

void HeaderInfo::init()
{
  mSubject[0] = '\0';
  mTo[0] = '\0';
  mFrom[0] = '\0';
  mFromNoColon[0] = '\0';
  mDate[0] = '\0';
  mReason[0] = '\0';
}



void HeaderInfo::subject( const char *pBuf )
{
  if (pBuf != 0) {
    strncpy(mSubject, pBuf, sizeof(mSubject) );
    if (mSubject[0] == '\0') {
      mSubject[0] = ' ';
      mSubject[1] = '\0';
    }
    else {
      SpamUtil().trimEnd(mSubject);
    }
  }
} // subject


FILE *HeaderInfo::openTraceFile()
{
  const char *traceFileName = "garbage_trace";
  const size_t MAX_FILE_SIZE = 50 * 1024; // file size in K bytes
  Logger log = pLogger->getLogger("HeaderInfo");
  char msgbuf[128];
  FILE *fp = 0;
  char *mode = "a"; // open for append
  if ((fp = fopen(traceFileName, mode)) != 0) {
    int fileSize = SpamUtil().getFileSize( fp, traceFileName, msgbuf );
    if (fileSize >= 0) {
      if (fileSize > MAX_FILE_SIZE) {
        mode = "w"; // open for truncate
        fclose(fp);
        // open and truncate the file
        if ((fp = fopen(traceFileName, mode)) == 0) {
          char *err_reason = strerror( errno);
          sprintf( msgbuf, "could not re-open %s.  Reason: %s", 
                   traceFileName, err_reason );
          log.log(Logger::ERROR, "openTraceFile", msgbuf);
        }
      }
    }
    else { // error getting stat on file
      log.log(Logger::ERROR, "openTraceFile", msgbuf);
    }
  }
  else { // error opening file
    char *err_reason = strerror( errno);
    sprintf( msgbuf, "could not open %s.  Reason: %s", 
             traceFileName, err_reason );
    log.log(Logger::ERROR, "openTraceFile", msgbuf);
  }

  return fp;
} // openTraceFile


HeaderInfo::~HeaderInfo()
{
  if (mDoTrace && klass() == MailFilter::GARBAGE) {
    FILE *fp = openTraceFile();
    if (fp != 0) {
      fprintf(fp, "\n");
      fprintf(fp, "From %s\n", SpamUtil().trimEnd(mFromNoColon));
      fprintf(fp, "From: %s\n", SpamUtil().trimEnd(mFrom));
      fprintf(fp, "To: %s\n", SpamUtil().trimEnd(mTo));
      fprintf(fp, "Subject: %s\n", SpamUtil().trimEnd(mSubject));
      fprintf(fp, "Date: %s\n", SpamUtil().trimEnd(mDate));
      fprintf(fp, "Reason-its-garbage: %s\n", SpamUtil().trimEnd(mReason));
      fclose(fp);
    }
  }
}

---