626 lines
11 KiB
C++
626 lines
11 KiB
C++
/*
|
|
* This file is a part of Winix
|
|
* and is distributed under the 2-Clause BSD licence.
|
|
* Author: Tomasz Sowa <t.sowa@ttmath.org>
|
|
*/
|
|
|
|
/*
|
|
* Copyright (c) 2010-2018, Tomasz Sowa
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <limits.h>
|
|
#include <limits>
|
|
#include "dbbase.h"
|
|
#include "core/log.h"
|
|
#include "core/error.h"
|
|
#include "core/misc.h"
|
|
#include "utf8/utf8.h"
|
|
|
|
|
|
namespace Winix
|
|
{
|
|
|
|
|
|
|
|
DbBase::DbBase()
|
|
{
|
|
db_conn = 0;
|
|
log_queries = false;
|
|
}
|
|
|
|
|
|
|
|
void DbBase::SetConn(DbConn * conn)
|
|
{
|
|
db_conn = conn;
|
|
}
|
|
|
|
|
|
|
|
void DbBase::SetConn(DbConn & conn)
|
|
{
|
|
db_conn = &conn;
|
|
}
|
|
|
|
|
|
DbConn * DbBase::GetConn()
|
|
{
|
|
return db_conn;
|
|
}
|
|
|
|
|
|
|
|
void DbBase::LogQueries(bool log_q)
|
|
{
|
|
log_queries = log_q;
|
|
}
|
|
|
|
|
|
|
|
PGresult * DbBase::AssertQuery(const char * q)
|
|
{
|
|
if( log_queries )
|
|
log << log1 << "Db: executing query: " << q << logend;
|
|
|
|
bool bad_query = false;
|
|
PGresult * r = PQexec(db_conn->GetPgConn(), q);
|
|
|
|
if( !r )
|
|
{
|
|
bad_query = true;
|
|
|
|
if( PQstatus(db_conn->GetPgConn()) != CONNECTION_OK )
|
|
{
|
|
db_conn->AssertConnection();
|
|
r = PQexec(db_conn->GetPgConn(), q);
|
|
|
|
if( r )
|
|
bad_query = false;
|
|
}
|
|
}
|
|
|
|
|
|
if( bad_query )
|
|
{
|
|
log << log1 << "Db: Problem with this query: \"" << q << '\"' << logend;
|
|
log << log1 << "Db: " << PQerrorMessage(db_conn->GetPgConn()) << logend;
|
|
|
|
throw Error(WINIX_ERR_DB_INCORRECT_QUERY);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
|
|
|
|
|
|
PGresult * DbBase::AssertQuery(const DbTextStream & query)
|
|
{
|
|
return AssertQuery(query.CStr());
|
|
}
|
|
|
|
|
|
|
|
PGresult * DbBase::AssertQuery(const char * q, ExecStatusType t)
|
|
{
|
|
PGresult * r = AssertQuery(q);
|
|
AssertResult(r, t);
|
|
|
|
return r;
|
|
}
|
|
|
|
|
|
PGresult * DbBase::AssertQuery(const DbTextStream & query, ExecStatusType t)
|
|
{
|
|
return AssertQuery(query.CStr(), t);
|
|
}
|
|
|
|
|
|
|
|
void DbBase::AssertResult(PGresult * r, ExecStatusType t)
|
|
{
|
|
if( PQresultStatus(r) != t )
|
|
{
|
|
log << log1 << "Db: Incorrect result status: " << PQerrorMessage(db_conn->GetPgConn()) << logend;
|
|
|
|
throw Error(WINIX_ERR_DB_INCORRENT_RESULT_STATUS);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
int DbBase::AssertColumn(PGresult * r, const char * column_name)
|
|
{
|
|
int c = PQfnumber(r, column_name);
|
|
|
|
if( c == -1 )
|
|
{
|
|
log << log1 << "Db: there is no column: " << column_name << logend;
|
|
throw Error(WINIX_ERR_DB_NO_COLUMN);
|
|
}
|
|
|
|
return c;
|
|
}
|
|
|
|
|
|
const char * DbBase::AssertValue(PGresult * r, int row, int col)
|
|
{
|
|
const char * res = PQgetvalue(r, row, col);
|
|
|
|
if( !res )
|
|
{
|
|
log << log1 << "Db: there is no such an item in the result, row:" << row << ", col:" << col << logend;
|
|
throw Error(WINIX_ERR_NO_ITEM);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
const std::wstring & DbBase::AssertValueWide(PGresult * r, int row, int col)
|
|
{
|
|
const char * res = AssertValue(r, row, col);
|
|
static std::wstring temp_wide_value; // !! IMPROVE ME add as a class field (nonstatic)
|
|
|
|
PT::UTF8ToWide(res, temp_wide_value);
|
|
|
|
return temp_wide_value;
|
|
}
|
|
|
|
|
|
|
|
void DbBase::AssertValueBin(PGresult * r, int row, int col, std::string & result)
|
|
{
|
|
result.clear();
|
|
|
|
const char * res = AssertValue(r, row, col);
|
|
int len = PQgetlength(r, row, col);
|
|
|
|
if( len <= 0 )
|
|
return;
|
|
|
|
UnescapeBin(res, len, result);
|
|
}
|
|
|
|
|
|
void DbBase::AssertValueWide(PGresult * r, int row, int col, std::wstring & result)
|
|
{
|
|
const char * res = AssertValue(r, row, col);
|
|
PT::UTF8ToWide(res, result);
|
|
}
|
|
|
|
|
|
long DbBase::AssertValueLong(PGresult * r, int row, int col)
|
|
{
|
|
return strtol( AssertValue(r, row, col), 0, 10 );
|
|
}
|
|
|
|
|
|
int DbBase::AssertValueInt(PGresult * r, int row, int col)
|
|
{
|
|
return (int)strtol( AssertValue(r, row, col), 0, 10 );
|
|
}
|
|
|
|
|
|
bool DbBase::AssertValueBool(PGresult * r, int row, int col)
|
|
{
|
|
const char * s = AssertValue(r, row, col);
|
|
return (s[0]=='t' || s[0]=='y' || s[0]=='1');
|
|
}
|
|
|
|
|
|
unsigned long DbBase::AssertValueULong(PGresult * r, int row, int col)
|
|
{
|
|
return strtoul( AssertValue(r, row, col), 0, 10 );
|
|
}
|
|
|
|
|
|
unsigned int DbBase::AssertValueUInt(PGresult * r, int row, int col)
|
|
{
|
|
return (unsigned int)strtoul( AssertValue(r, row, col), 0, 10 );
|
|
}
|
|
|
|
|
|
|
|
PT::Date DbBase::AssertValueDate(PGresult * r, int row, int col)
|
|
{
|
|
PT::Date date = AssertValue(r, row, col);
|
|
|
|
return date;
|
|
}
|
|
|
|
|
|
bool DbBase::AssertValueSpace(PGresult * r, int row, int col, PT::Space & space)
|
|
{
|
|
const char * res = AssertValue(r, row, col);
|
|
|
|
conf_parser.SetSpace(space);
|
|
space.Clear();
|
|
|
|
PT::SpaceParser::Status status = conf_parser.ParseString(res);
|
|
|
|
if( status != PT::SpaceParser::ok )
|
|
{
|
|
log << log1 << "Db: a problem with parsing a PT::Space";
|
|
|
|
if( status == PT::SpaceParser::syntax_error )
|
|
log << ", syntax error at line: " << conf_parser.line;
|
|
|
|
log << logend;
|
|
|
|
space.Clear();
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void DbBase::ClearResult(PGresult * r)
|
|
{
|
|
if( r )
|
|
PQclear(r);
|
|
}
|
|
|
|
|
|
bool DbBase::IsNull(PGresult * r, int row, int col)
|
|
{
|
|
return PQgetisnull(r, row, col) == 1;
|
|
}
|
|
|
|
|
|
int DbBase::Rows(PGresult * r)
|
|
{
|
|
// PQntuples - Returns the number of rows (tuples) in the query result. Because it returns
|
|
// an integer result, large result sets might overflow the return value on 32-bit operating systems.
|
|
return PQntuples(r);
|
|
}
|
|
|
|
|
|
int DbBase::Cols(PGresult * r)
|
|
{
|
|
// PQnfields - Returns the number of columns (fields) in each row of the query result.
|
|
return PQnfields(r);
|
|
}
|
|
|
|
|
|
long DbBase::AffectedRows(PGresult * r)
|
|
{
|
|
// PQcmdTuples - This function returns a string containing the number of rows affected by the SQL
|
|
// statement that generated the PGresult. This function can only be used following the execution
|
|
// of an INSERT, UPDATE, DELETE, MOVE, FETCH, or COPY statement, or [...]
|
|
char * rows_str = PQcmdTuples(r); // can be an empty string
|
|
long rows = 0;
|
|
|
|
if( rows_str )
|
|
{
|
|
rows = strtol(rows_str, 0, 10);
|
|
// strtol - If an overflow or underflow occurs, errno is set to ERANGE
|
|
// and the function return value is clamped according to the following table:
|
|
// Function underflow overflow
|
|
// strtol() LONG_MIN LONG_MAX
|
|
|
|
if( rows < 0 )
|
|
rows = 0;
|
|
}
|
|
|
|
return rows;
|
|
}
|
|
|
|
|
|
long DbBase::AssertCurrval(const char * table)
|
|
{
|
|
PGresult * r;
|
|
|
|
bquery.Clear();
|
|
bquery << R("select currval(")
|
|
<< table
|
|
<< R(");");
|
|
|
|
r = AssertQuery(bquery);
|
|
AssertResult(r, PGRES_TUPLES_OK);
|
|
|
|
if( Rows(r) != 1 )
|
|
{
|
|
log << log1 << "Db: error (currval) for table: " << table << ", " << PQerrorMessage(db_conn->GetPgConn()) << logend;
|
|
throw Error(WINIX_ERR_DB_ERR_CURRVAL);
|
|
}
|
|
|
|
return AssertValueLong(r, 0, 0);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void DbBase::CreateIdList(const std::vector<long> & id_tab, std::wstring & list, bool add_parentheses)
|
|
{
|
|
wchar_t buffer[50];
|
|
size_t buffer_len = sizeof(buffer) / sizeof(wchar_t);
|
|
|
|
list.clear();
|
|
|
|
if( add_parentheses )
|
|
list += '(';
|
|
|
|
for(size_t i=0 ; i<id_tab.size() ; ++i)
|
|
{
|
|
Toa((unsigned long)id_tab[i], buffer, buffer_len);
|
|
list += buffer;
|
|
|
|
if( i+1 < id_tab.size() )
|
|
list += ',';
|
|
}
|
|
|
|
if( add_parentheses )
|
|
list += ')';
|
|
}
|
|
|
|
|
|
|
|
|
|
Error DbBase::DoCommand(const DbTextStream & command)
|
|
{
|
|
PGresult * r = 0;
|
|
Error status = WINIX_ERR_OK;
|
|
|
|
try
|
|
{
|
|
r = AssertQuery(command);
|
|
AssertResult(r, PGRES_COMMAND_OK);
|
|
}
|
|
catch(const Error & e)
|
|
{
|
|
status = e;
|
|
}
|
|
|
|
ClearResult(r);
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
|
|
Error DbBase::DoCommand(const char * command)
|
|
{
|
|
PGresult * r = 0;
|
|
Error status = WINIX_ERR_OK;
|
|
|
|
try
|
|
{
|
|
r = AssertQuery(command);
|
|
AssertResult(r, PGRES_COMMAND_OK);
|
|
}
|
|
catch(const Error & e)
|
|
{
|
|
status = e;
|
|
}
|
|
|
|
ClearResult(r);
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
|
|
Error DbBase::BeginTrans()
|
|
{
|
|
return DoCommand("BEGIN;");
|
|
}
|
|
|
|
|
|
|
|
Error DbBase::RollbackTrans()
|
|
{
|
|
return DoCommand("ROLLBACK;");
|
|
}
|
|
|
|
|
|
|
|
Error DbBase::CommitTrans()
|
|
{
|
|
return DoCommand("COMMIT;");
|
|
}
|
|
|
|
|
|
|
|
bool DbBase::EndTrans(bool everything_ok)
|
|
{
|
|
bool result;
|
|
|
|
if( everything_ok )
|
|
{
|
|
result = (CommitTrans() == WINIX_ERR_OK);
|
|
}
|
|
else
|
|
{
|
|
RollbackTrans();
|
|
// we return the old err code
|
|
result = false;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
Error DbBase::EndTrans(Error err)
|
|
{
|
|
if( err == WINIX_ERR_OK )
|
|
{
|
|
err = CommitTrans();
|
|
}
|
|
else
|
|
{
|
|
// we return the old err code
|
|
RollbackTrans();
|
|
}
|
|
|
|
return err;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
converting from a bytea
|
|
the old way (escape format)
|
|
*/
|
|
/*
|
|
int DbBase::CharToInt(char c)
|
|
{
|
|
return (int)(unsigned char)(c-'0');
|
|
}
|
|
|
|
bool DbBase::IsCorrectOctalDigit(char c)
|
|
{
|
|
return c>='0' && c<='7';
|
|
}
|
|
|
|
// moves 'i' at least once
|
|
// return -1 if there is en error
|
|
int DbBase::UnescapeBin(const char * str, size_t & i, size_t len)
|
|
{
|
|
if( str[i] != '\\' )
|
|
return str[i++];
|
|
|
|
i += 1;
|
|
|
|
if( i >= len )
|
|
return -1;
|
|
|
|
if( str[i] == '\\' )
|
|
return str[i++];
|
|
|
|
if( i+2 >= len )
|
|
{
|
|
i = len;
|
|
return -1;
|
|
}
|
|
|
|
if( !IsCorrectOctalDigit(str[i]) ||
|
|
!IsCorrectOctalDigit(str[i+1]) ||
|
|
!IsCorrectOctalDigit(str[i+2]) )
|
|
{
|
|
i += 3;
|
|
return -1;
|
|
}
|
|
|
|
int c = 8*8*CharToInt(str[i]) + 8*CharToInt(str[i+1]) + CharToInt(str[i+2]);
|
|
|
|
i += 3;
|
|
|
|
if( c<0 || c>255 )
|
|
return -1;
|
|
|
|
return c;
|
|
}
|
|
|
|
void DbBase::UnescapeBin(const char * str, size_t len, std::string & out, bool clear_out)
|
|
{
|
|
int c;
|
|
size_t i = 0;
|
|
|
|
if( clear_out )
|
|
out.clear();
|
|
|
|
while( i < len )
|
|
{
|
|
c = UnescapeBin(str, i, len);
|
|
|
|
if( c != -1 )
|
|
out += c;
|
|
}
|
|
}
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
converting from a bytea
|
|
the new way (hex format)
|
|
*/
|
|
|
|
|
|
char DbBase::UnescapeBinHexToDigit(char hex)
|
|
{
|
|
if( hex>='0' && hex<='9' )
|
|
return hex - '0';
|
|
|
|
if( hex>='a' && hex<='z' )
|
|
return hex - 'a' + 10;
|
|
|
|
if( hex>='A' && hex<='Z' )
|
|
return hex - 'A' + 10;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
void DbBase::UnescapeBin(const char * str, size_t len, std::string & out, bool clear_out)
|
|
{
|
|
if( clear_out )
|
|
out.clear();
|
|
|
|
if( len < 2 || str[0]!='\\' || str[1]!='x' )
|
|
{
|
|
log << log1 << "Db: unsupported binary format (skipping)" << logend;
|
|
return;
|
|
}
|
|
|
|
for(size_t i=2 ; i + 1 < len ; i+=2 )
|
|
{
|
|
int c1 = UnescapeBinHexToDigit(str[i]);
|
|
int c2 = UnescapeBinHexToDigit(str[i+1]);
|
|
|
|
out += ((c1 << 4) | c2);
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
end of converting from bytea
|
|
*/
|
|
|
|
|
|
|
|
} // namespace Winix
|
|
|
|
|
|
|