1 files changed, 111 insertions, 0 deletions
diff --git a/src/Common/ResponseAPDU.cpp b/src/Common/ResponseAPDU.cpp
new file mode 100644
index 00000000..8a7f069c
--- /dev/null
+++ b/src/Common/ResponseAPDU.cpp
@@ -0,0 +1,111 @@
+#include "ResponseAPDU.h"
+#include <string.h>
+
+using namespace std;
+
+namespace VeraCrypt
+{
+	uint16 BytesToUInt16(const vector<byte>& buff)
+	{
+		uint16 value = 0;
+		for (uint16 i = 0; i < buff.size(); i++)
+		{
+			value <<= 8;
+			value |= (uint16)buff.at(i);
+		}
+
+		return value;
+	}
+
+	void AppendData (vector<byte>& buffer, const byte* pbData, size_t cbData)
+	{
+		size_t orgSize = buffer.size ();
+		buffer.resize (orgSize + cbData);
+		memcpy (buffer.data () + orgSize, pbData, cbData);
+	}
+
+	/*********************************************************************************/
+
+	void ResponseAPDU::clear()
+	{
+		m_data.clear();
+		m_SW = 0;
+	}
+
+	ResponseAPDU::ResponseAPDU() : m_SW(0)
+	{
+	}
+
+	ResponseAPDU::ResponseAPDU(const vector<byte>& data, uint16 SW)
+	{
+		m_data = data;
+		m_SW = SW;
+	}
+
+	uint32 ResponseAPDU::getNr()
+	{
+		return (uint32)m_data.size();
+	}
+
+	const vector<byte> ResponseAPDU::getData()
+	{
+		return m_data;
+	}
+
+	byte ResponseAPDU::getSW1()
+	{
+		return (byte)((0xFF00 & m_SW) >> 8);
+	}
+
+	byte ResponseAPDU::getSW2()
+	{
+		return (byte)(0x00FF & m_SW);
+	}
+
+	uint16 ResponseAPDU::getSW()
+	{
+		return m_SW;
+	}
+
+	const vector<byte> ResponseAPDU::getBytes()
+	{
+		vector<byte> apdu;
+
+		AppendData(apdu, m_data.data(), m_data.size());
+		apdu.push_back((byte)getSW1());
+		apdu.push_back((byte)getSW2());
+
+		return apdu;
+	}
+
+	void ResponseAPDU::appendData(const vector<byte>& data)
+	{
+		appendData(data.data(), data.size());
+	}
+
+	void ResponseAPDU::appendData(const byte* data, size_t dataLen)
+	{
+		AppendData(m_data, data, dataLen);
+	}
+
+	void ResponseAPDU::setSW(uint16 SW)
+	{
+		m_SW = SW;
+	}
+
+	void ResponseAPDU::setBytes(const vector<byte>& bytes)
+	{
+		clear();
+		if (bytes.size() >= 2)
+		{
+			vector<byte> SWBytes;
+			m_data.resize(bytes.size() - 2);
+			SWBytes.resize(2);
+
+			memcpy(m_data.data(), bytes.data(), bytes.size() - 2);
+			memcpy(SWBytes.data(), bytes.data() + bytes.size() - 2, 2);
+			m_SW = BytesToUInt16(SWBytes);
+		}
+	}
+}
+