utils.c

Go to the documentation of this file.

/*
 SPDX-FileCopyrightText: © 2013 Hewlett-Packard Development Company, L.P.
 
 SPDX-License-Identifier: GPL-2.0-only
*/
#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() */