575 lines
21 KiB
C++
Executable file
575 lines
21 KiB
C++
Executable file
#define _DEFAULT_SOURCE
|
|
|
|
#ifdef _WIN32
|
|
#include <Windows.h>
|
|
#include <conio.h>
|
|
#else
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/select.h>
|
|
#endif
|
|
|
|
#include <sys/time.h>
|
|
#include <iostream>
|
|
#include <iomanip>
|
|
#include <stdint.h>
|
|
#include <ctime>
|
|
#include <time.h>
|
|
|
|
#include <omp.h>
|
|
#include <queue>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <memory>
|
|
#include <cmath>
|
|
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include <deque>
|
|
|
|
#include "BiasAndAC.h"
|
|
#include "Monkey.h"
|
|
#include "Serial.h"
|
|
#include "Entropy.h"
|
|
#include "Stat.h"
|
|
#include "Gamma.h"
|
|
#include "KolmogorovSmirnov.h"
|
|
|
|
int deviceType_;
|
|
int rxBytes;
|
|
int blockIntCount;
|
|
bool doExit = false;
|
|
|
|
using namespace std;
|
|
|
|
#ifdef __linux
|
|
#include <signal.h>
|
|
// Terminal Signal Handler
|
|
void sigintevent(int)
|
|
{
|
|
doExit = true;
|
|
}
|
|
#endif
|
|
|
|
template<typename T>
|
|
class ThreadSafeQueue {
|
|
private:
|
|
mutable std::mutex mtx;
|
|
std::deque<T> dataQueue;
|
|
std::condition_variable dataCond;
|
|
|
|
public:
|
|
ThreadSafeQueue() {}
|
|
|
|
void push(T newData) {
|
|
std::lock_guard<std::mutex> lk(mtx);
|
|
dataQueue.push_back(std::move(newData));
|
|
dataCond.notify_one();
|
|
}
|
|
|
|
bool try_pop(T& value) {
|
|
std::lock_guard<std::mutex> lk(mtx);
|
|
if (dataQueue.empty()) {
|
|
return false;
|
|
}
|
|
value = std::move(dataQueue.front());
|
|
dataQueue.pop_front();
|
|
return true;
|
|
}
|
|
|
|
std::unique_lock<std::mutex> wait_and_pop(T& value) {
|
|
std::unique_lock<std::mutex> lk(mtx);
|
|
dataCond.wait(lk, [this] { return !dataQueue.empty(); });
|
|
value = std::move(dataQueue.front());
|
|
dataQueue.pop_front();
|
|
return lk;
|
|
}
|
|
|
|
bool empty() const {
|
|
std::lock_guard<std::mutex> lk(mtx);
|
|
return dataQueue.empty();
|
|
}
|
|
};
|
|
|
|
int main()
|
|
{
|
|
// try this to resize terminal
|
|
// resizeterm(42, 80);
|
|
|
|
#ifdef _WIN32
|
|
// hide cursor
|
|
CONSOLE_CURSOR_INFO info;
|
|
info.dwSize = 100;
|
|
info.bVisible = FALSE;
|
|
SetConsoleCursorInfo(GetStdHandle(STD_OUTPUT_HANDLE), &info);
|
|
// clear screen
|
|
system("cls");
|
|
#elif __linux
|
|
// clear screen
|
|
// cout << "\E[?25l\E[H\E[2J";
|
|
// Signal handler to catch CTRL-C in terminal
|
|
signal(SIGINT, (__sighandler_t)&sigintevent);
|
|
#elif MACOSX
|
|
#endif
|
|
|
|
// randtest classes
|
|
CBiasAndAC biasAndAc;
|
|
biasAndAc.ResetAll();
|
|
CMonkey oqso;
|
|
CSerial serial;
|
|
CEntropy entropy;
|
|
CGamma gamma;
|
|
CKolmogorovSmirnov ks;
|
|
|
|
double serialP = 0.5;
|
|
double serialZ = 0.0;
|
|
double KSP;
|
|
double KSN;
|
|
|
|
time_t startTime;
|
|
time(&startTime);
|
|
string sTime = ctime(&startTime);
|
|
char sStartTime[16];
|
|
sTime.copy(sStartTime, 15, 4);
|
|
|
|
time_t timeNow;
|
|
time_t timePrev;
|
|
time(&timePrev);
|
|
timePrev -= 8;
|
|
|
|
//queue<shared_ptr<vector<uint32_t> > > dataQueue;
|
|
|
|
double bitsThroughputCount = 0;
|
|
double prevBitsThroughputCount = 0;
|
|
double bitsTestedCount = 0;
|
|
double bitsTestedRatio = 1;
|
|
double prevBitsTestedCount = 0;
|
|
|
|
#ifdef _WIN32
|
|
LARGE_INTEGER countFreq;
|
|
QueryPerformanceFrequency(&countFreq);
|
|
LARGE_INTEGER prevCount;
|
|
QueryPerformanceCounter(&prevCount);
|
|
LARGE_INTEGER nowCount;
|
|
#elif __linux
|
|
timeval start, stop;
|
|
double newTimeInterval;
|
|
gettimeofday(&start, NULL);
|
|
#elif MACOSX
|
|
#endif
|
|
|
|
double throughput = 0;
|
|
|
|
vector<double> metaPs;
|
|
vector<double> meterZs;
|
|
vector<int> meterFlags(36, 0);
|
|
double meterScore = 0;
|
|
bool meterFreeze = false;
|
|
|
|
const char* fifoPath = "/tmp/QNGmeter";
|
|
#ifdef __linux
|
|
mkfifo(fifoPath, 0666); // Create FIFO if it doesn't exist
|
|
int fifoFd = open(fifoPath, O_RDONLY | O_NONBLOCK);
|
|
if (fifoFd == -1) {
|
|
std::cerr << "Failed to open named pipe for reading." << std::endl;
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
fd_set readfds;
|
|
int maxfd = max(STDIN_FILENO, fifoFd) + 1;
|
|
|
|
ThreadSafeQueue<std::string> dataQueue; // Use a thread-safe queue for shared data
|
|
|
|
omp_set_nested(true); // Ensure nested parallelism is enabled
|
|
|
|
#pragma omp parallel sections
|
|
{
|
|
#pragma omp section
|
|
{
|
|
fd_set readfds;
|
|
int fifoFd = open ("/tmp/myfifo", O_RDONLY | O_NONBLOCK);
|
|
if (fifoFd == -1)
|
|
{
|
|
perror ("Failed to open FIFO");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
|
|
int maxfd = max (STDIN_FILENO, fifoFd) + 1;
|
|
ThreadSafeQueue < std::string > dataQueue;
|
|
|
|
while (!doExit)
|
|
{
|
|
FD_ZERO (&readfds);
|
|
FD_SET (STDIN_FILENO, &readfds);
|
|
FD_SET (fifoFd, &readfds);
|
|
|
|
if (select (maxfd, &readfds, NULL, NULL, NULL) == -1)
|
|
{
|
|
perror ("select");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
|
|
char buffer[4096];
|
|
ssize_t bytesRead;
|
|
|
|
if (FD_ISSET (STDIN_FILENO, &readfds))
|
|
{
|
|
bytesRead = read (STDIN_FILENO, buffer, sizeof (buffer));
|
|
if (bytesRead > 0)
|
|
{
|
|
dataQueue.push (std::string (buffer, bytesRead));
|
|
}
|
|
}
|
|
|
|
if (FD_ISSET (fifoFd, &readfds))
|
|
{
|
|
bytesRead = read (fifoFd, buffer, sizeof (buffer));
|
|
if (bytesRead > 0)
|
|
{
|
|
dataQueue.push (std::string (buffer, bytesRead));
|
|
}
|
|
}
|
|
}
|
|
|
|
close (fifoFd);
|
|
}
|
|
|
|
#pragma omp section
|
|
{
|
|
std::string buffer; // This will hold the raw binary string data from the queue
|
|
while (!doExit)
|
|
{
|
|
if (dataQueue.try_pop (buffer))
|
|
{
|
|
// Assuming each uint32_t is stored in 4 bytes within the string
|
|
// Convert the string buffer into a vector<uint32_t> for processing
|
|
std::vector < uint32_t > values;
|
|
const size_t numValues = buffer.size () / sizeof (uint32_t);
|
|
for (size_t i = 0; i < numValues; ++i)
|
|
{
|
|
uint32_t value;
|
|
memcpy (&value, buffer.data () + i * sizeof (uint32_t),
|
|
sizeof (uint32_t));
|
|
values.push_back (value);
|
|
}
|
|
|
|
// Now, values contains the uint32_t data extracted from buffer
|
|
// Process the values as needed
|
|
#pragma omp parallel sections num_threads(4)
|
|
{
|
|
#pragma omp section
|
|
{
|
|
for (uint32_t value:values)
|
|
biasAndAc.InsertWord32 (value);
|
|
}
|
|
#pragma omp section
|
|
{
|
|
for (uint32_t value:values)
|
|
oqso.InsertWord32 (value);
|
|
}
|
|
#pragma omp section
|
|
{
|
|
for (uint32_t value:values)
|
|
serial.InsertWord32 (value);
|
|
}
|
|
#pragma omp section
|
|
{
|
|
for (uint32_t value:values)
|
|
entropy.InsertWord32 (value);
|
|
}
|
|
}
|
|
|
|
// Display results
|
|
time (&timeNow);
|
|
if (difftime (timeNow, timePrev) >= 10)
|
|
{
|
|
// calc rates
|
|
#ifdef _WIN32
|
|
QueryPerformanceCounter (&nowCount);
|
|
double newTimeInterval =
|
|
((double) nowCount.QuadPart -
|
|
prevCount.QuadPart) / countFreq.QuadPart;
|
|
prevCount = nowCount;
|
|
#elif __linux
|
|
gettimeofday (&stop, NULL);
|
|
newTimeInterval = (stop.tv_sec - start.tv_sec); // sec
|
|
newTimeInterval += (stop.tv_usec - start.tv_usec) / 1000000.0; // us to sec
|
|
start = stop;
|
|
#elif MACOSX
|
|
#endif
|
|
|
|
double newBitInterval;
|
|
double newRate;
|
|
#pragma omp critical
|
|
{
|
|
newBitInterval =
|
|
(bitsThroughputCount - prevBitsThroughputCount);
|
|
prevBitsThroughputCount = bitsThroughputCount;
|
|
}
|
|
|
|
newRate = newBitInterval / newTimeInterval;
|
|
if (throughput == 0)
|
|
throughput = newRate;
|
|
else
|
|
throughput = (2 * throughput + newRate) / 3;
|
|
|
|
double newBitsTestedRatio =
|
|
(bitsTestedCount - prevBitsTestedCount) / newBitInterval;
|
|
prevBitsTestedCount = bitsTestedCount;
|
|
bitsTestedRatio =
|
|
(2 * bitsTestedRatio + newBitsTestedRatio) / 3;
|
|
if (bitsTestedRatio > 1)
|
|
bitsTestedRatio = 1;
|
|
|
|
// meta test and meter
|
|
metaPs.clear ();
|
|
meterZs.clear ();
|
|
|
|
if (bitsTestedCount >= 65536)
|
|
{
|
|
// Autocorrelation KS test
|
|
for (int i = 0; i < 32; i++)
|
|
{
|
|
metaPs.push_back (biasAndAc.AC.P_Chi2[i]);
|
|
meterZs.push_back (biasAndAc.AC.
|
|
cumulativeACZScore[i]);
|
|
}
|
|
double AcKSP;
|
|
double AcKSN;
|
|
ks.KSUP (&AcKSP, &AcKSN, &metaPs[0], metaPs.size ());
|
|
|
|
// Combined KS test
|
|
metaPs.push_back (AcKSP);
|
|
|
|
metaPs.push_back (biasAndAc.Bias.P_Chi2);
|
|
meterZs.push_back (biasAndAc.Bias.cumulativeBiasZScore);
|
|
}
|
|
|
|
if (bitsTestedCount >= 4194304)
|
|
{
|
|
metaPs.push_back (serial.P_Chi2);
|
|
serialP = gamma.Gamma (128., serial.cumulativeSerialChi2);
|
|
serialZ = ks.PtoZ (serialP);
|
|
meterZs.push_back (serialZ);
|
|
|
|
metaPs.push_back (entropy.P_Chi2);
|
|
meterZs.push_back (entropy.cumulativeZScore);
|
|
}
|
|
|
|
if (bitsTestedCount >= 10485775)
|
|
{
|
|
metaPs.push_back (oqso.P_Chi2);
|
|
meterZs.push_back (oqso.cumulativeZScore);
|
|
}
|
|
|
|
if (bitsTestedCount >= 65536)
|
|
{
|
|
// This KS is combined AC KSP plus with other tests
|
|
ks.KSUP (&KSP, &KSN, &metaPs[32], metaPs.size () - 32);
|
|
|
|
meterFreeze = false;
|
|
for (int i = 0; i < meterZs.size (); i++)
|
|
{
|
|
// freeze condition
|
|
if (fabs (meterZs[i]) > 4.264897
|
|
|| (metaPs[i] < 0.00001 || metaPs[i] > 0.99999))
|
|
meterFlags[i] = -1;
|
|
// unfreeze condition
|
|
if (meterFlags[i] == -1)
|
|
{
|
|
if (fabs (meterZs[i]) < 2.326348
|
|
&& (metaPs[i] > 0.01 && metaPs[i] < 0.99))
|
|
meterFlags[i] = 0;
|
|
}
|
|
|
|
if (meterFlags[i] == -1)
|
|
meterFreeze = true;
|
|
}
|
|
|
|
// meter calc
|
|
if (meterFreeze == false)
|
|
meterScore = log (bitsTestedCount) / log (2.);
|
|
}
|
|
|
|
|
|
cout << endl;
|
|
cout <<
|
|
" QNGmeter Console 1.0 Test Type z-score p[z<=x] p[chi2<=x] "
|
|
<< endl;
|
|
cout <<
|
|
" +---------------------------+------------------------------------------------+"
|
|
<< endl;
|
|
cout << " | | 1/0 Balance "
|
|
<< setiosflags (ios::
|
|
fixed) << setprecision (3) << showpos <<
|
|
biasAndAc.Bias.
|
|
cumulativeBiasZScore << " " << setprecision (4) <<
|
|
noshowpos << CStat::ZtoP (biasAndAc.Bias.
|
|
cumulativeBiasZScore) << " " <<
|
|
biasAndAc.Bias.P_Chi2 << " |" << endl;
|
|
cout << " | | Serial Test "
|
|
<< setiosflags (ios::
|
|
fixed) << setprecision (3) << showpos <<
|
|
serialZ << " " << setprecision (4) << noshowpos <<
|
|
serialP << " " << serial.P_Chi2 << " |" << endl;
|
|
cout << " | | OQSO Test "
|
|
<< setiosflags (ios::
|
|
fixed) << setprecision (3) << showpos <<
|
|
oqso.
|
|
cumulativeZScore << " " << setprecision (4) << noshowpos
|
|
<< CStat::ZtoP (oqso.cumulativeZScore) << " " << oqso.
|
|
P_Chi2 << " |" << endl;
|
|
cout << " | | Entropy Test "
|
|
<< setiosflags (ios::
|
|
fixed) << setprecision (3) << showpos <<
|
|
entropy.
|
|
cumulativeZScore << " " << setprecision (4) << noshowpos
|
|
<< CStat::ZtoP (entropy.
|
|
cumulativeZScore) << " " << serial.
|
|
P_Chi2 << " |" << endl;
|
|
cout << " | | H: " <<
|
|
setiosflags (ios::fixed) << setprecision (9) << entropy.
|
|
E << " |" << endl;
|
|
|
|
cout <<
|
|
" | | |"
|
|
<< endl;
|
|
|
|
for (int i = 1; i <= 32; i++)
|
|
{
|
|
switch (i)
|
|
{
|
|
case 2:
|
|
cout << " | Start Time |";
|
|
break;
|
|
case 3:
|
|
cout << " | " << sStartTime << " |";
|
|
break;
|
|
case 5:
|
|
cout << " | Total Bits Tested |";
|
|
break;
|
|
case 6:
|
|
cout << " | " << scientific << setw (9) <<
|
|
setprecision (2) << bitsTestedCount << fixed <<
|
|
" |";
|
|
break;
|
|
case 8:
|
|
cout << " | Throughput |";
|
|
break;
|
|
case 9:
|
|
cout << " | " << setiosflags (ios::
|
|
fixed) << setw (4)
|
|
<< setprecision (1) << (double) (throughput /
|
|
1000000.0) <<
|
|
" Mbps |";
|
|
break;
|
|
case 11:
|
|
cout << " | Bits Tested Percent |";
|
|
break;
|
|
case 12:
|
|
cout << " | " << setiosflags (ios::
|
|
fixed) << setw (5)
|
|
<< setprecision (1) << (100 *
|
|
bitsTestedRatio) <<
|
|
"% |";
|
|
break;
|
|
case 18:
|
|
cout << " | Meta KS+ Test |";
|
|
break;
|
|
case 19:
|
|
cout << " | " << setiosflags (ios::
|
|
fixed) << setw (5)
|
|
<< setprecision (3) << KSP << " |";
|
|
break;
|
|
case 22:
|
|
cout << " | Meta KS- Test |";
|
|
break;
|
|
case 23:
|
|
cout << " | " << setiosflags (ios::
|
|
fixed) << setw (5)
|
|
<< setprecision (3) << KSN << " |";
|
|
break;
|
|
case 29:
|
|
cout << " | QNGmeter Score |";
|
|
break;
|
|
case 30:
|
|
cout << " | " << setiosflags (ios::
|
|
fixed) << setw (4)
|
|
<< setprecision (1) << abs (meterScore) <<
|
|
((meterScore < 0) ? "-" : (meterFreeze ==
|
|
false) ? "+" : " ") <<
|
|
" |";
|
|
break;
|
|
default:
|
|
cout << " | |";
|
|
}
|
|
cout << " " << setw (2) << i << "st AutoCorr " <<
|
|
setiosflags (ios::
|
|
fixed) << setprecision (3) << showpos <<
|
|
biasAndAc.AC.cumulativeACZScore[i -
|
|
1] << " " <<
|
|
setprecision (4) << noshowpos << CStat::ZtoP (biasAndAc.
|
|
AC.
|
|
cumulativeACZScore
|
|
[i -
|
|
1]) <<
|
|
" " << biasAndAc.AC.P_Chi2[i - 1] << " |" << endl;
|
|
}
|
|
cout <<
|
|
" +---------------------------+------------------------------------------------+"
|
|
<< endl;
|
|
|
|
#ifdef _WIN32
|
|
// put cursor in top corner
|
|
COORD coord;
|
|
coord.X = 0;
|
|
coord.Y = 0;
|
|
SetConsoleCursorPosition (GetStdHandle (STD_OUTPUT_HANDLE),
|
|
coord);
|
|
#elif __linux
|
|
// clear screen
|
|
// printf("\E[H");
|
|
#elif MACOSX
|
|
#endif
|
|
|
|
timePrev = timeNow;
|
|
cout.flush ();
|
|
buffer.clear ();
|
|
}
|
|
|
|
// End on an 'x' keypress
|
|
#ifdef _WIN32
|
|
if (kbhit ())
|
|
{
|
|
char c = getch_ ();
|
|
|
|
if (tolower (c) == 'x')
|
|
{
|
|
doExit = true;
|
|
break;
|
|
}
|
|
}
|
|
#elif __linux
|
|
// CRTL-C
|
|
#elif MACOSX
|
|
#endif
|
|
}
|
|
else
|
|
{
|
|
// give this thread a break from tight loop - waiting for data
|
|
usleep (1000);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#ifdef __linux
|
|
close(fifoFd); // Close the FIFO file descriptor when done
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|