CWE-ou-minkata/Sources/Plasma/NucleusLib/pnSqlLib/Private/pnSqlUtil.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, 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, either version 3 of the License, or
(at your option) any later version.

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, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
/*****************************************************************************
*
*   $/Plasma20/Sources/Plasma/NucleusLib/pnSqlLib/Private/pnSqlUtil.cpp
*   
***/

#include "../Pch.h"
#pragma hdrstop


/*****************************************************************************
*
*   Exports
*
***/

//===========================================================================
void SqlConnBindParameterInputGuid (
    SqlStmt *       stmt,
    Uuid *          uuid,
    SQLINTEGER *    uuidLen
) {
    SqlConnBindParameter(
        stmt,
        SQL_PARAM_INPUT,
        SQL_C_BINARY,
        SQL_BINARY,
        sizeof(Uuid),
        0,
        uuid,
        sizeof(Uuid),
        uuidLen
    );
}

//===========================================================================
void SqlConnBindParameterOutputGuid (
    SqlStmt *       stmt,
    Uuid *          uuid,
    SQLINTEGER *    uuidLen
) {
    SqlConnBindParameter(
        stmt,
        SQL_PARAM_OUTPUT,
        SQL_C_BINARY,
        SQL_BINARY,
        sizeof(Uuid),
        0,
        uuid,
        sizeof(Uuid),
        uuidLen
    );
}

//===========================================================================
void SqlConnBindParameterBigInt (
    SqlStmt *       stmt,
    SQLINTEGER      inputOutputType,
    SQLBIGINT *     parameterValuePtr,
    SQLINTEGER *    indPtr
) {
    SqlConnBindParameter(
        stmt,
        inputOutputType,
        SQL_C_SBIGINT,
        SQL_BIGINT,
        sizeof(qword),
        0,
        parameterValuePtr,
        0,
        indPtr
    );
}

//===========================================================================
bool SqlConnGetBlobData (
    SqlStmt *       stmt,
    unsigned        colIndex,
    ARRAY(byte) *   buffer,
    unsigned *      bytesAdded
) {
    // Since a number of routines use this function, set the chunk size to be
    // fairly large to avoid the necessity of reallocing the block, and further
    // to prevent memory fragmentation by allocating variously sized blocks.
    // In for a penny, in for a pound...
    const unsigned CHUNK_SIZE   = 32 * 1024;
    SQLINTEGER length           = CHUNK_SIZE - 1;
    unsigned bytes              = buffer->Bytes();
    *bytesAdded                 = 0 - bytes;
    for (;;) {
        // Allocate memory that is evenly aligned to CHUNK_SIZE, which
        // reduces memory fragmentation by allocating on 32K multiples.
        // Use SetCount() instead of New() to ensure that this is true
        // even if the buffer already contains data.
        buffer->SetCount(
            MathNextMultiplePow2(
                buffer->Bytes() + length,
                CHUNK_SIZE
            )
        );

        // Read column data
        int result = SqlConnGetData(
            stmt,
            colIndex,
            SQL_C_BINARY,
            buffer->Ptr() + bytes,
            buffer->Bytes() - bytes,
            &length
        );

        // Handle SQL_NO_TOTAL and SQL_NULL_DATA values
        if (length < 0) {
            buffer->ZeroCount();
            *bytesAdded = 0;
            return result == SQL_SUCCESS;
        }

        // Keep getting data until we have the entire buffer
        if (result == SQL_SUCCESS_WITH_INFO) {
            // length contains the number of remaining bytes
            bytes = buffer->Bytes();
            continue;
        }

        // Is the buffer complete?
        if (result == SQL_SUCCESS) {
            buffer->SetCountFewer(length + bytes);
            *bytesAdded += buffer->Bytes();
            return true;
        }

        // An error occurred
        break;
    }

    buffer->ZeroCount();
    *bytesAdded = 0;
    return false;
}

//===========================================================================
int SqlConnPutBlobData (
    SqlStmt *   stmt,
    unsigned    bytes,
    const byte  data[]
) {
    int result;
    SQLPOINTER putPtr;
    while (SQL_NEED_DATA == (result = SqlConnParamData(stmt, &putPtr))) {
        if (!SqlConnPutData(stmt, const_cast<byte *>(data), bytes))
            return SQL_ERROR;
    }
    return result;
}