infosecn1nja
/
C3
mirror of https://github.com/infosecn1nja/C3.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify MSSQL::OnSend

dependabot/npm_and_yarn/Src/WebController/UI/websocket-extensions-0.1.4
Grzegorz Rychlik 2020-05-14 16:36:59 +02:00
parent 1e5b12d0c7
commit 737322dbb5
1 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Src/Common/FSecure/C3/Interfaces/Channels/MSSQL.cpp
View File

@ -59,27 +59,15 @@ size_t FSecure::C3::Interfaces::Channels::MSSQL::OnSendToChannel(FSecure::ByteVi
Sql::Enviroment env; Sql::Enviroment env;
auto conn = env.Connect(m_servername, m_databasename, m_username, m_password, m_useSSPI, m_impersonationToken.get()); auto conn = env.Connect(m_servername, m_databasename, m_username, m_password, m_useSSPI, m_impersonationToken.get());
size_t bytesWritten = 0;
std::string b64packet = "";
//Rounded down: Max size of bytes that can be put into a MSSQL database before base64 encoding //Rounded down: Max size of bytes that can be put into a MSSQL database before base64 encoding
if (packet.size() > MAX_MSG_BYTES) packet = packet.SubString(0, MAX_MSG_BYTES);
{
auto strpacket = packet.SubString(0, MAX_MSG_BYTES);
b64packet = cppcodec::base64_rfc4648::encode(strpacket.data(), strpacket.size());
bytesWritten = strpacket.size();
}
else
{
b64packet = cppcodec::base64_rfc4648::encode(packet.data(), packet.size());
bytesWritten = packet.size();
}
std::string stmtString = OBF("INSERT into dbo.") + this->m_tablename + OBF(" (MSGID, MSG) VALUES ('") + this->m_outboundDirectionName + "', '" + b64packet + OBF("');"); std::string stmtString = OBF("INSERT into dbo.") + this->m_tablename + OBF(" (MSGID, MSG) VALUES ('") + this->m_outboundDirectionName + "', '" + cppcodec::base64_rfc4648::encode(packet) + OBF("');");
auto hStmt = conn.MakeStatement(stmtString); auto hStmt = conn.MakeStatement(stmtString);
hStmt.Execute(); hStmt.Execute();
return bytesWritten; // packet was trimmed if it was too large
return packet.size();
} }
std::vector<FSecure::ByteVector> FSecure::C3::Interfaces::Channels::MSSQL::OnReceiveFromChannel() std::vector<FSecure::ByteVector> FSecure::C3::Interfaces::Channels::MSSQL::OnReceiveFromChannel()