utils.c

Go to the documentation of this file.

/***************************************************************
 Copyright (C) 2013 Hewlett-Packard Development Company, L.P.

 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 version 2 as published by the Free Software Foundation.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License along
 with this program; if not, write to the Free Software Foundation, Inc.,
 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

 ***************************************************************/
#include "demomod.h"

/**********  Globals  *************/
extern psqlCopy_t psqlcpy;       // fo_sqlCopy struct used for fast data insertion
extern PGconn    *pgConn;        // database connection

FUNCTION void CheckTable(char *AgentARSName)
{
  PGresult* result; // the result of the database access
  int rv;             // generic return value
  char *TableName = "demomod";
  char *CreateTableSQL =
       "CREATE TABLE demomod ( \
          demomod_pk serial NOT NULL PRIMARY KEY, \
          pfile_fk integer, \
          agent_fk integer, \
          firstbytes character(65), \
        FOREIGN KEY (pfile_fk) REFERENCES pfile(pfile_pk) ON DELETE CASCADE, \
        FOREIGN KEY (agent_fk) REFERENCES agent(agent_pk) ON DELETE CASCADE \
        ); \
        COMMENT ON TABLE demomod IS 'table for demo module'; \
        COMMENT ON COLUMN demomod.firstbytes IS 'Hex string of first 32 bytes of the pfile'; \
      ";


  /* Check if the demomod_ars table exists.  If not, create it.
   * The _ars tables serve as an audit trail.  They tell you when an agent
   * has run and what the parameters were.  They also provide a database service.
   * Without the _ars tables, the agent would have to run in a single transaction,
   * potentially inserting millions of records in that single transaction.  This
   * can result in deadlocks, large memory consumption, and slower performance.
   * Because of the _ars tables, we can run in multiple transactions.
   */
  rv = fo_tableExists(pgConn, AgentARSName);
  if (!rv)
  {
    rv = fo_CreateARSTable(pgConn, AgentARSName);
    if (!rv)
    {
      LOG_FATAL("%s table could not be created", AgentARSName);
      ExitNow(-10);
    }
  }

  /* Check if TableName exists.  If not, create it
   * An easy way to get all the table creation sql is to create the table with all your
   * constraints manually or with phppgadmin, then export the table definition.
   */
  rv = fo_tableExists(pgConn, TableName);
  if (!rv)
  {
     result = PQexec(pgConn, CreateTableSQL);
     if (fo_checkPQresult(pgConn, result, CreateTableSQL, __FILE__, __LINE__))
     {
       LOG_ERROR("Failed to create %s table.", TableName);
       ExitNow(-11);
     }
  }

}


FUNCTION void Char2Hex(char *InBuf, int NumBytes, char *OutBuf)
{
  int i;
  char* pbuf = OutBuf;
  for (i=0; i<NumBytes; i++) pbuf += sprintf(pbuf, "%02X", (unsigned char)InBuf[i]);

  *(pbuf) = '\0';
} /* Char2Hex() */


FUNCTION void ExitNow(int ExitVal)
{
  if (pgConn) PQfinish(pgConn);

  if (ExitVal) LOG_ERROR("Exiting with status %d", ExitVal);

  fo_scheduler_disconnect(ExitVal);
  exit(ExitVal);
} /* ExitNow() */