src/Common/ODBC/Connection.h

Go to the documentation of this file.

#pragma once

namespace AppSecInc
{
        namespace Databases
        {
                namespace ODBC
                {
                        enum ODBCConnectionProtocol
                        {
                                ODBC_CONNECTION_UNSPECIFIED = -1,
                                ODBC_CONNECTION_TCPIP = 0,
                                ODBC_CONNECTION_NAMEDPIPES = 1,
                                ODBC_CONNECTION_RPC, // multi-protocol
                                ODBC_CONNECTION_LPC, // shared memory
                                ODBC_CONNECTION_SPX, // nwlink ipx/spx
                                ODBC_CONNECTION_VINES, // Banyan VINES
                                ODBC_CONNECTION_ADSP // Apple Talk
                        };

                        // Bit flags controlling behaviour of Get-methods:
                        const WORD ODBC_SQL_FLAG_NO_ROWS_RETURN_EMPTY = 0x0001; 


                        class ODBCConnection
                        {
                        protected:
                                SQLHDBC _hconnection;
                                SQLHENV _henvironment;
                                BOOL _connected;
                        public:
                                ODBCConnection();
                                virtual ~ODBCConnection();
                                void Initialize();
                                void Terminate();
                void Connect(const ODBCConnectionInfo& connection_info);
                                void Disconnect();
                std::wstring GetError() const { return ODBCHandle::GetError(_hconnection, SQL_HANDLE_DBC); }
                                std::vector<ODBCError> GetErrors() const { return ODBCHandle::GetErrors(_hconnection, SQL_HANDLE_DBC); }
                operator SQLHDBC() const { return _hconnection; }
                                BOOL IsConnected() const { return _connected; }
                SQLHENV GetEnvironmentHandle() const { return _henvironment; }
                                long GetScalar(const std::wstring& query) const;
                                long GetScalar(const std::wstring& query, ODBCHandle& stmt) const;
                std::wstring GetXml(const std::wstring& query) const;
                MSXML2::IXMLDOMNodePtr GetXml(const std::wstring& query, AppSecInc::Xml::XmlDocument& xmldoc, MSXML2::IXMLDOMNodePtr parent_node = NULL) const;
                void InsertXml(const AppSecInc::Xml::XmlDocument& xmldatadoc);
                void InsertXml(const AppSecInc::Xml::XmlDocument& xmldatadoc, AppSecInc::Xml::XmlDocument& xmldoc, MSXML2::IXMLDOMNodePtr parent_node);
                void InsertXml(const AppSecInc::Xml::XmlDocument& xmldatadoc, MSXML2::IXMLDOMNodePtr result_node, AppSecInc::Xml::XmlDocument& xmldoc, MSXML2::IXMLDOMNodePtr parent_node);
                std::string GetString(const std::wstring& query, WORD flags = 0) const;
                std::string GetString(const std::wstring& query, ODBCHandle& stmt, WORD flags = 0) const;
                std::wstring GetWString(const std::wstring& query, WORD flags = 0) const;
                std::wstring GetWString(const std::wstring& query, ODBCHandle& stmt, WORD flags = 0) const;
                                void Execute(const std::wstring& query) const;
                                void Execute(const std::wstring& query, ODBCHandle&) const;
                                void Execute(const std::wstring& query, ODBCRowSet&) const;
                                void Execute(const std::wstring& query, ODBCParameters&) const;
                                void Execute(const std::wstring& query, ODBCParameters&, ODBCRowSet&) const;
                        };
                }
        }
}