180 lines
4.2 KiB
C++
180 lines
4.2 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 "Streams.hpp"
|
|
#include "IO.hpp"
|
|
#include <iostream>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <cstdio>
|
|
#include <unistd.h>
|
|
|
|
using namespace std;
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
FILE* initHTML(FILE *fPtr, const string &title);
|
|
void closeHTML(FILE *fPtr);
|
|
#endif
|
|
|
|
shared_ptr<int> getHostDescriptor()
|
|
{
|
|
static shared_ptr<int> descriptor{};
|
|
|
|
descriptor.reset(new int{ open(HID_DEV, O_RDWR | O_NONBLOCK | O_APPEND) }, [](int* fd){
|
|
close(*fd);
|
|
delete fd;
|
|
});
|
|
|
|
if (*descriptor == -1)
|
|
throw runtime_error{ "Descriptor is unavailable" };
|
|
|
|
return descriptor;
|
|
}
|
|
|
|
#ifdef DEBUG_STREAMS
|
|
shared_ptr<FILE> getComHostStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ fopen((cacheDirectory + "comhost.txt").c_str(), "wb"), [](FILE *f){
|
|
clog << "Closing comhost stream" << endl;
|
|
fclose(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
shared_ptr<FILE> getHostPacketStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ initHTML(fopen((cacheDirectory + "hostpackets.html").c_str(), "wb"), "Host Packets"), [](FILE *f){
|
|
clog << "Closing hostPackets stream" << endl;
|
|
closeHTML(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
shared_ptr<FILE> getHostAPDUStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ initHTML(fopen((cacheDirectory + "hostAPDU.html").c_str(), "wb"), "Host APDU"), [](FILE *f){
|
|
clog << "Closing host APDU stream" << endl;
|
|
closeHTML(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
shared_ptr<FILE> getComDevStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ fopen((cacheDirectory + "comdev.txt").c_str(), "wb"), [](FILE *f){
|
|
clog << "Closing comdev stream" << endl;
|
|
fclose(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
shared_ptr<FILE> getDevPacketStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ initHTML(fopen((cacheDirectory + "devpackets.html").c_str(), "wb"), "Dev Packets"), [](FILE *f){
|
|
clog << "Closing devPackets stream" << endl;
|
|
closeHTML(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
shared_ptr<FILE> getDevAPDUStream()
|
|
{
|
|
static shared_ptr<FILE> stream{ initHTML(fopen((cacheDirectory + "devAPDU.html").c_str(), "wb"), "Dev APDU"), [](FILE *f){
|
|
clog << "Closing dev APDU stream" << endl;
|
|
closeHTML(f);
|
|
} };
|
|
|
|
if (!stream)
|
|
clog << "Stream is unavailable" << endl;
|
|
|
|
return stream;
|
|
}
|
|
|
|
FILE* initHTML(FILE *fPtr, const string &title)
|
|
{
|
|
fprintf(fPtr, "<html>\n"
|
|
"\t<head>\n"
|
|
"\t\t<title>%s</title>\n"
|
|
"\t\t<style>\n"
|
|
"\t\t\ttable {\n"
|
|
"\t\t\t\tdisplay: table;\n"
|
|
"\t\t\t\twidth: 100%%;\n"
|
|
"\t\t\t\tborder-collapse: collapse;\n"
|
|
"\t\t\t\tbox-sizing: border-box;\n"
|
|
"\t\t\t}\n"
|
|
"\n"
|
|
"\t\t\tth.data {\n"
|
|
"\t\t\t\ttext-align: left;\n"
|
|
"\t\t\t}\n"
|
|
"\n"
|
|
"\t\t\ttd {\n"
|
|
"\t\t\t\tfont-family: \"Courier New\", Courier, monospace;\n"
|
|
"\t\t\t\twhite-space: pre;\n"
|
|
"\t\t\t}\n"
|
|
"\n"
|
|
"\t\t\ttd.data {\n"
|
|
"\t\t\t\toverflow: hidden;\n"
|
|
"\t\t\t\ttext-overflow: ellipsis;\n"
|
|
"\t\t\t\tmax-width:1px;\n"
|
|
"\t\t\t\twidth:100%%;\n"
|
|
"\t\t\t}\n"
|
|
"\n"
|
|
"\t\t\ttd.data:hover {\n"
|
|
"\t\t\t\twhite-space: pre-wrap;\n"
|
|
"\t\t\t}\n"
|
|
"\n"
|
|
"\t\t\ttable, th, td {\n"
|
|
"\t\t\t\tborder: 1px solid black;\n"
|
|
"\t\t\t}\n"
|
|
"\t\t</style>\n"
|
|
"\t</head>\n"
|
|
"\n"
|
|
"\t<body>", title.c_str());
|
|
|
|
return fPtr;
|
|
}
|
|
|
|
void closeHTML(FILE *fPtr)
|
|
{
|
|
fprintf(fPtr, "\t</body>\n"
|
|
"</html>");
|
|
fclose(fPtr);
|
|
}
|
|
#endif
|