319 lines
7.5 KiB
C++
319 lines
7.5 KiB
C++
/*
|
|
U2FDevice - A program to allow Raspberry Pi Zeros to act as U2F tokens
|
|
Copyright (C) 2018 Michael Kuc
|
|
|
|
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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "Packet.hpp"
|
|
#include "IO.hpp"
|
|
#include "Streams.hpp"
|
|
#include "u2f.hpp"
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <unistd.h>
|
|
|
|
using namespace std;
|
|
|
|
shared_ptr<InitPacket> InitPacket::getPacket(const uint32_t rCID, const uint8_t rCMD) {
|
|
static size_t bytesRead = 0;
|
|
static uint8_t bcnth;
|
|
static uint8_t bcntl;
|
|
static decltype(InitPacket::data) dataBytes;
|
|
vector<uint8_t> bytes{};
|
|
|
|
switch (bytesRead) {
|
|
case 0:
|
|
bytes = readNonBlock(1);
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
bcnth = bytes[0];
|
|
bytesRead += bytes.size();
|
|
[[fallthrough]];
|
|
|
|
case 1:
|
|
bytes = readNonBlock(1);
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
bcntl = bytes[0];
|
|
bytesRead += bytes.size();
|
|
[[fallthrough]];
|
|
|
|
case 2:
|
|
bytes = readNonBlock(dataBytes.size());
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
copy(bytes.begin(), bytes.end(), dataBytes.begin());
|
|
;
|
|
bytesRead += bytes.size();
|
|
[[fallthrough]];
|
|
|
|
case 2 + dataBytes.size():
|
|
break;
|
|
|
|
default:
|
|
throw runtime_error{ "Unknown stage in InitPacket construction" };
|
|
}
|
|
|
|
auto p = make_shared<InitPacket>();
|
|
p->cid = rCID;
|
|
p->cmd = rCMD;
|
|
p->bcnth = bcnth;
|
|
p->bcntl = bcntl;
|
|
p->data = dataBytes;
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
auto hPStream = getHostPacketStream().get();
|
|
fprintf(hPStream,
|
|
"\t\t<table>\n"
|
|
"\t\t\t<thead>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<th>CID</th>\n"
|
|
"\t\t\t\t\t<th>CMD</th>\n"
|
|
"\t\t\t\t\t<th>BCNTH</th>\n"
|
|
"\t\t\t\t\t<th>BCNTL</th>\n"
|
|
"\t\t\t\t\t<th class=\"data\">DATA</th>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</thead>\n"
|
|
"\t\t\t<tbody>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<td>0x%08X</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td class=\"data\">",
|
|
p->cid, p->cmd, p->bcnth, p->bcntl);
|
|
|
|
for (auto elem : dataBytes)
|
|
fprintf(hPStream, "%3u ", elem);
|
|
|
|
fprintf(hPStream, "</td>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</tbody>\n"
|
|
"\t\t</table>"
|
|
"\t\t<br />");
|
|
#endif
|
|
|
|
bytesRead = 0;
|
|
return p;
|
|
}
|
|
|
|
shared_ptr<ContPacket> ContPacket::getPacket(const uint32_t rCID, const uint8_t rSeq) {
|
|
static size_t readBytes = 0;
|
|
static decltype(ContPacket::data) dataBytes;
|
|
|
|
vector<uint8_t> bytes{};
|
|
auto p = make_shared<ContPacket>();
|
|
|
|
if (readBytes != dataBytes.size()) {
|
|
dataBytes = {};
|
|
bytes = readNonBlock(dataBytes.size());
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
copy(bytes.begin(), bytes.end(), dataBytes.begin());
|
|
readBytes += bytes.size();
|
|
}
|
|
|
|
p->cid = rCID;
|
|
p->seq = rSeq;
|
|
p->data = dataBytes;
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
auto hPStream = getHostPacketStream().get();
|
|
fprintf(hPStream,
|
|
"\t\t<table>\n"
|
|
"\t\t\t<thead>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<th>CID</th>\n"
|
|
"\t\t\t\t\t<th>SEQ</th>\n"
|
|
"\t\t\t\t\t<th class=\"data\">DATA</th>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</thead>\n"
|
|
"\t\t\t<tbody>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<td>0x%08X</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td class=\"data\">",
|
|
p->cid, p->seq);
|
|
|
|
for (auto elem : dataBytes)
|
|
fprintf(hPStream, "%3u ", elem);
|
|
|
|
fprintf(hPStream, "</td>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</tbody>\n"
|
|
"\t\t</table>\n"
|
|
"\t\t<br />");
|
|
#endif
|
|
|
|
readBytes = 0;
|
|
return p;
|
|
}
|
|
|
|
shared_ptr<Packet> Packet::getPacket() {
|
|
static size_t bytesRead = 0;
|
|
vector<uint8_t> bytes{};
|
|
|
|
static uint32_t cid;
|
|
static uint8_t b;
|
|
shared_ptr<Packet> packet{};
|
|
|
|
switch (bytesRead) {
|
|
case 0:
|
|
bytes = readNonBlock(4);
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
cid = *reinterpret_cast<uint32_t*>(bytes.data());
|
|
bytesRead += bytes.size();
|
|
[[fallthrough]];
|
|
|
|
case 4:
|
|
bytes = readNonBlock(1);
|
|
|
|
if (bytes.size() == 0)
|
|
return {};
|
|
|
|
b = bytes[0];
|
|
bytesRead += bytes.size();
|
|
[[fallthrough]];
|
|
|
|
case 5:
|
|
if (b & TYPE_MASK) {
|
|
// Init packet
|
|
packet = InitPacket::getPacket(cid, b);
|
|
|
|
if (packet)
|
|
bytesRead = 0;
|
|
|
|
return packet;
|
|
} else {
|
|
// Cont packet
|
|
packet = ContPacket::getPacket(cid, b);
|
|
|
|
if (packet)
|
|
bytesRead = 0;
|
|
|
|
return packet;
|
|
}
|
|
default:
|
|
throw runtime_error{ "Unknown stage in Packet construction" };
|
|
}
|
|
}
|
|
|
|
void Packet::writePacket() {
|
|
memset(this->buf, 0, HID_RPT_SIZE);
|
|
memcpy(this->buf, &cid, 4);
|
|
}
|
|
|
|
void InitPacket::writePacket() {
|
|
Packet::writePacket();
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
auto devStream = getComDevStream().get();
|
|
#endif
|
|
|
|
memcpy(this->buf + 4, &cmd, 1);
|
|
memcpy(this->buf + 5, &bcnth, 1);
|
|
memcpy(this->buf + 6, &bcntl, 1);
|
|
memcpy(this->buf + 7, data.data(), data.size());
|
|
write(vector<uint8_t>{ this->buf, this->buf + sizeof(this->buf) });
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
fwrite(this->buf, 1, sizeof(this->buf), devStream);
|
|
|
|
auto dPStream = getDevPacketStream().get();
|
|
fprintf(dPStream,
|
|
"\t\t<table>\n"
|
|
"\t\t\t<thead>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<th>CID</th>\n"
|
|
"\t\t\t\t\t<th>CMD</th>\n"
|
|
"\t\t\t\t\t<th>BCNTH</th>\n"
|
|
"\t\t\t\t\t<th>BCNTL</th>\n"
|
|
"\t\t\t\t\t<th class=\"data\">DATA</th>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</thead>\n"
|
|
"\t\t\t<tbody>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<td>0x%08X</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td class=\"data\">",
|
|
cid, cmd, bcnth, bcntl);
|
|
|
|
for (auto elem : data)
|
|
fprintf(dPStream, "%3u ", elem);
|
|
|
|
fprintf(dPStream, "</td>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</tbody>\n"
|
|
"\t\t</table>"
|
|
"\t\t<br />");
|
|
#endif
|
|
}
|
|
|
|
void ContPacket::writePacket() {
|
|
Packet::writePacket();
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
auto devStream = getComDevStream().get();
|
|
#endif
|
|
|
|
memcpy(this->buf + 4, &seq, 1);
|
|
memcpy(this->buf + 5, data.data(), data.size());
|
|
write(vector<uint8_t>{ this->buf, this->buf + HID_RPT_SIZE });
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
fwrite(this->buf, HID_RPT_SIZE, 1, devStream);
|
|
|
|
auto dPStream = getDevPacketStream().get();
|
|
|
|
fprintf(dPStream,
|
|
"\t\t<table>\n"
|
|
"\t\t\t<thead>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<th>CID</th>\n"
|
|
"\t\t\t\t\t<th>SEQ</th>\n"
|
|
"\t\t\t\t\t<th class=\"data\">DATA</th>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</thead>\n"
|
|
"\t\t\t<tbody>\n"
|
|
"\t\t\t\t<tr>\n"
|
|
"\t\t\t\t\t<td>0x%08X</td>\n"
|
|
"\t\t\t\t\t<td>%u</td>\n"
|
|
"\t\t\t\t\t<td class=\"data\">",
|
|
cid, seq);
|
|
|
|
for (auto elem : data)
|
|
fprintf(dPStream, "%3u ", elem);
|
|
|
|
fprintf(dPStream, "</td>\n"
|
|
"\t\t\t\t</tr>\n"
|
|
"\t\t\t</tbody>\n"
|
|
"\t\t</table>\n"
|
|
"\t\t<br />");
|
|
#endif
|
|
}
|