login_info.cpp

Go to the documentation of this file.

/*****************************************************************************\
*                                                                             *
*  Name   : login_info                                                        *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*******************************************************************************
* Copyright (c) 2008-$now By Author.  This program is free software; you can  *
* redistribute it and/or modify it under the terms of the GNU General Public  *
* License as published by the Free Software Foundation; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/

#include "login_info.h"

#include <basis/function.h>
#include <basis/log_base.h>
#include <data_struct/static_memory_gremlin.h>
#include <opsystem/byte_filer.h>
#include <opsystem/filename.h>
#include <opsystem/ini_config.h>

#include <stdio.h>

#define LOG(s) CLASS_EMERGENCY_LOG(program_wide_logger(), filename(portable::application_name()).basename().raw() + " " + s)


// select a default config file based on platform considerations.
SAFE_STATIC(istring_object, database_login_info::freetds_config_file,
#ifdef EMBEDDED_BUILD
    ("/etc/config/freetds.conf")
#else
    ("/usr/local/etc/freetds.conf")
#endif
);

database_login_info::database_login_info(const istring &db_section_in)
: db_section(db_section_in)
{
  reset();
}

database_login_info::~database_login_info() { reset(); }

bool database_login_info::identical(const database_login_info &to_compare)
{
  return (user_name == to_compare.user_name)
      && (password == to_compare.password)
      && (db_host == to_compare.db_host)
      && (db_port == to_compare.db_port)
      && (db_instance == to_compare.db_instance)
      && (database == to_compare.database);
}

void database_login_info::reset()
{
  user_name = istring::empty_string();
  password = istring::empty_string();
  db_host = istring::empty_string();
  db_port = 0;
  db_instance = istring::empty_string();
  database = istring::empty_string();
  table_name = istring::empty_string();
}

bool database_login_info::read_info(ini_configurator &ini)
{
  reset();  // clear before reading anything.

  const istring login_section = "database";
  const istring bogus = "b0gus";

  user_name = ini.load(login_section, "user", bogus);
  password = ini.load(login_section, "passwd", bogus);
  db_host = ini.load(login_section, "host", bogus);
  db_port = ini.load(login_section, "port", 0);
  db_instance = ini.load(login_section, "instance", bogus);
  database = ini.load(login_section, "database", bogus);
  table_name = ini.load(login_section, "table", bogus);

  // table is optional.
  if ( (user_name == bogus)
      || (password == bogus)
      || (db_host == bogus)
      || ( (db_instance == bogus) && (db_port == 0) )
      || (database == bogus) ) {
    reset();
    return false;
  }

  // reset optional or alternate parameters rather than leaving set to
  // sentinel values.
  if (table_name == bogus) table_name = "";
  if (db_instance == bogus) db_instance = "";

  return true;
}


#define CHECK_INDY(indy, what) \
  if (indy < 0) { \
    LOG(istring("ERROR: failed to find ") + what); \
    return false; \
  }

bool database_login_info::replace_value(istring &modify, const istring &tag,
    const istring &replacement)
{
  FUNCDEF("replace_value");
//LOG(isprintf("looking for replace value now, section=%s", db_section.s()));
  istring section_tag = istring("[") + db_section + "]";
  int sect_indy = modify.find(section_tag, 0);
  if (sect_indy < 0) {
    modify.insert(modify.length(), istring(log_base::platform_ending())
        + section_tag + log_base::platform_ending());
    // try finding it again.
    sect_indy = modify.find(section_tag, 0);
  }
  CHECK_INDY(sect_indy, istring(db_section) + " section in config.");
  int cr_indy_1 = modify.find("\n", sect_indy);
  CHECK_INDY(cr_indy_1, istring("carriage return in ")
      + db_section + " section.");
  // find start of next section, if it exists.
  int next_sect = modify.find("[", cr_indy_1);
  if (next_sect < 0) next_sect = modify.length() - 1;
  int tag_indy = modify.find(tag, sect_indy);
  if ( (tag_indy < 0) || (tag_indy > next_sect) ) {
    // in this case it's not an error, it means we need to add the tag.
    modify.insert(cr_indy_1 + 1, tag + " = " + replacement
        + log_base::platform_ending());
    return true;
  }
  int cr_indy_2 = modify.find("\n", tag_indy);
  CHECK_INDY(cr_indy_2, istring("carriage return for tag ") + tag
      + " entry in " + db_section + " section");
  modify.zap(tag_indy + tag.length(), cr_indy_2);
  modify.insert(tag_indy + tag.length(), istring(" = ") + replacement
        + log_base::platform_ending());
  return true;
}

// simple wrapper, saved due to history.
bool database_login_info::read_login_info(ini_configurator &ini,
    database_login_info &to_fill)
{ return to_fill.read_info(ini); }

bool database_login_info::update_freetds_configuration()
{
  FUNCDEF("update_freetds_configuration");
//hmmm: this piece would need to be done each time if we move to only
//      reading configs on startup.

  // write the info into freetds.conf for the section.
  byte_filer initial(freetds_config_file(), "rb");
const int MAXIMUM_FREETDS_FILE = 8192;
  istring current_content;
  int chars_read = initial.read(current_content, MAXIMUM_FREETDS_FILE);
  initial.close();
  if (!chars_read) {
    LOG(istring("ERROR: failed to read freetds configuration from ")
        + freetds_config_file());
    return false;
  }

  bool to_return = replace_value(current_content, "host", db_host);
  if (!to_return) {
    LOG(istring("ERROR: failed to find or add host value to ")
        + freetds_config_file());
    return false;
  }

  // instance and port are only written if they are not at their default and
  // unusable value.
  if (db_instance.t()) {
    to_return = replace_value(current_content, "instance", db_instance);
    if (!to_return) {
      LOG(istring("ERROR: failed to find or add instance value to ")
          + freetds_config_file());
      return false;
    }
  }
  if (db_port != 0) {
    to_return = replace_value(current_content, "port", isprintf("%d", db_port));
    if (!to_return) {
      LOG(istring("ERROR: failed to find or add port value to ")
          + freetds_config_file());
      return false;
    }
  }

  byte_filer new_version(freetds_config_file(), "wb");
  int chars_written = new_version.write(current_content);
  new_version.close();
  if (chars_written < current_content.length()) {
    LOG(istring("ERROR: failed to write new version of ") + freetds_config_file());
    return false;
  }

  fflush(0);  // flush all changes out.

  return to_return;
}

---