mariadb-columnstore-engine/utils/funcexp/func_inet_aton.cpp

/* Copyright (C) 2014 InfiniDB, Inc.

   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; version 2 of
   the License.

   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. */

/****************************************************************************
 * $Id: func_inet_aton.cpp 3923 2013-06-19 21:43:06Z bwilkinson $
 *
 *
 ****************************************************************************/

#include "functor_real.h"

#include "calpontsystemcatalog.h"
#include "functioncolumn.h"
#include "joblisttypes.h"
#include "rowgroup.h"
// #include <iostream> // included when debugging

namespace funcexp
{
//------------------------------------------------------------------------------
// Return input argument type.
// See mcs_add in udfsdk.h for explanation of this function.
//------------------------------------------------------------------------------
execplan::CalpontSystemCatalog::ColType Func_inet_aton::operationType(
    FunctionParm& fp, execplan::CalpontSystemCatalog::ColType& /*resultType*/)
{
  return fp[0]->data()->resultType();  // input type
}

//------------------------------------------------------------------------------
// Return IP address as a long long int value.
// SELECT ... WHERE inet_aton(ipstring) = 11111111 will call getIntVal()
//------------------------------------------------------------------------------
int64_t Func_inet_aton::getIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                  execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  //	std::cout << "In Func_inet_aton::getIntVal" << std::endl;

  int64_t iValue = joblist::NULL_INT64;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      iValue = iVal;
  }

  return iValue;
}

//------------------------------------------------------------------------------
// Return IP address as a double value.
// SELECT ... WHERE inet_aton(ipstring) = '11111111' will call getDoubleVal()
//------------------------------------------------------------------------------
double Func_inet_aton::getDoubleVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                    execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  //	std::cout << "In Func_inet_aton::getDoubleVal" << std::endl;

  double dValue = doubleNullVal();

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iValue = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      dValue = iValue;
  }

  return dValue;
}

//------------------------------------------------------------------------------
// Return IP address as a string value.
// We are starting with a string value, and we want to return a string value,
// so we might be tempted to just return the result from getStrVal(), as-is.
// But we still call convertAton() to validate that the IP address we have is
// a syntactically valid one.
// Don't know if this function will ever be called.
//------------------------------------------------------------------------------
std::string Func_inet_aton::getStrVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                      execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  //	std::cout << "In Func_inet_aton::getStrVal" << std::endl;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    convertAton(sValue.unsafeStringRef(), isNull);  // ignore return valuea
    if (isNull)
    {
      return "";
    }
  }

  return sValue.safeString("");
}

//------------------------------------------------------------------------------
// Return IP address as a boolean?
// getBoolVal() makes no sense for inet_aton() but we will implement anyway.
// Don't know if this function will ever be called.
//------------------------------------------------------------------------------
bool Func_inet_aton::getBoolVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  bool bValue = false;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if ((!isNull) && (iVal != 0))
      bValue = true;
  }

  return bValue;
}

//------------------------------------------------------------------------------
// Return IP address as a decimal value.
// SELECT ... WHERE inet_aton(ipstring) = 11111111. will call getDecimalVal()
//------------------------------------------------------------------------------
execplan::IDB_Decimal Func_inet_aton::getDecimalVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                                    execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  execplan::CalpontSystemCatalog::ColType colType = fp[0]->data()->resultType();

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!datatypes::Decimal::isWideDecimalTypeByPrecision(colType.precision))
  {
    if (!sValue.isNull())
    {
      int64_t iValue = convertAton(sValue.unsafeStringRef(), isNull);

      if (!isNull)
        return execplan::IDB_Decimal(iValue, colType.scale, colType.precision);
    }

    return execplan::IDB_Decimal(joblist::NULL_INT64, colType.scale, colType.precision);
  }
  else
  {
    if (!isNull)
    {
      int64_t iValue = convertAton(sValue.unsafeStringRef(), isNull);

      if (!isNull)
        return execplan::IDB_Decimal(0, colType.scale, colType.precision, (int128_t)iValue);
    }

    return execplan::IDB_Decimal(0, colType.scale, colType.precision, datatypes::Decimal128Null);
  }
}

//------------------------------------------------------------------------------
// Return IP address as a date?
// getDateIntVal() makes no sense for inet_aton() but we will implement anyway.
// Don't know if this function will ever be called.
//------------------------------------------------------------------------------
int32_t Func_inet_aton::getDateIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                      execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  int32_t iValue = joblist::DATENULL;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      iValue = iVal;
  }

  return iValue;
}

//------------------------------------------------------------------------------
// Return IP address as a date/time?
// getDatetimeIntVal() makes no sense for inet_aton() but we will implement
// anyway.
// Don't know if this function will ever be called.
//------------------------------------------------------------------------------
int64_t Func_inet_aton::getDatetimeIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                          execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  int64_t iValue = joblist::DATETIMENULL;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      iValue = iVal;
  }

  return iValue;
}

int64_t Func_inet_aton::getTimestampIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                           execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  int64_t iValue = joblist::TIMESTAMPNULL;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      iValue = iVal;
  }

  return iValue;
}

int64_t Func_inet_aton::getTimeIntVal(rowgroup::Row& row, FunctionParm& fp, bool& isNull,
                                      execplan::CalpontSystemCatalog::ColType& /*op_ct*/)
{
  int64_t iValue = joblist::TIMENULL;

  const auto& sValue = fp[0]->data()->getStrVal(row, isNull);

  if (!sValue.isNull())
  {
    int64_t iVal = convertAton(sValue.unsafeStringRef(), isNull);

    if (!isNull)
      iValue = iVal;
  }

  return iValue;
}

//------------------------------------------------------------------------------
// Convert an ascii IP address string to it's integer equivalent.
// isNull is set to true if the IP address string has invalid content.
// Source code based on MySQL source (Item_func_inet_aton() in item_func.cc).
//------------------------------------------------------------------------------
int64_t Func_inet_aton::convertAton(const std::string& ipString, bool& isNull)
{
  char c = '.';
  int dot_count = 0;
  unsigned int byte_result = 0;
  unsigned long long result = 0;

  const char* p = ipString.c_str();
  const char* end = p + ipString.length();

  // Loop through bytes in the IP address string
  while (p < end)
  {
    c = *p++;

    int digit = (int)(c - '0');  // Assume ascii

    if (digit >= 0 && digit <= 9)
    {
      // Add the next digit from the string to byte_result
      if ((byte_result = byte_result * 10 + digit) > 255)
      {
        // Wrong address
        isNull = true;
        return 0;
      }
    }
    // Detect end of one portion of the IP address.
    // Shift current result over 8 bits, and add next byte (byte_result)
    else if (c == '.')
    {
      dot_count++;
      result = (result << 8) + (unsigned long long)byte_result;
      byte_result = 0;
    }
    // Exit loop if/when we encounter end of string for fixed length column,
    // that is padded with '\0' at the end.
    else if (c == '\0')
    {
      break;
    }
    else
    {
      // Invalid character
      isNull = true;
      return 0;
    }
  }

  if (c != '.')  // IP number can't end on '.'
  {
    //
    // Handle short-forms addresses according to standard. Examples:
    // 127     -> 0.0.0.127
    // 127.1   -> 127.0.0.1
    // 127.2.1 -> 127.2.0.1
    //
    switch (dot_count)
    {
      case 1: result <<= 8; /* Fall through */

      case 2: result <<= 8; /* Fall through */
    }

    //		std::cout << "aton: " <<
    //			(result << 8) + (unsigned long long) byte_result << std::endl;

    return (result << 8) + (unsigned long long)byte_result;
  }

  // Invalid IP address ended in '.'
  isNull = true;
  return 0;
}

}  // namespace funcexp