54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
#include "icsneo/communication/icommunication.h"
|
|
|
|
using namespace icsneo;
|
|
|
|
bool ICommunication::read(std::vector<uint8_t>& bytes, size_t limit) {
|
|
// A limit of zero indicates no limit
|
|
if(limit == 0)
|
|
limit = (size_t)-1;
|
|
|
|
if(limit > (readQueue.size_approx() + 4))
|
|
limit = (readQueue.size_approx() + 4);
|
|
|
|
if(bytes.capacity() < limit)
|
|
bytes.resize(limit);
|
|
|
|
size_t actuallyRead = readQueue.try_dequeue_bulk(bytes.data(), limit);
|
|
|
|
if(bytes.size() > actuallyRead)
|
|
bytes.resize(actuallyRead);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ICommunication::readWait(std::vector<uint8_t>& bytes, std::chrono::milliseconds timeout, size_t limit) {
|
|
// A limit of zero indicates no limit
|
|
if(limit == 0)
|
|
limit = (size_t)-1;
|
|
|
|
if(limit > (readQueue.size_approx() + 4))
|
|
limit = (readQueue.size_approx() + 4);
|
|
|
|
bytes.resize(limit);
|
|
|
|
size_t actuallyRead = readQueue.wait_dequeue_bulk_timed(bytes.data(), limit, timeout);
|
|
|
|
bytes.resize(actuallyRead);
|
|
|
|
return actuallyRead > 0;
|
|
}
|
|
|
|
bool ICommunication::write(const std::vector<uint8_t>& bytes) {
|
|
if(writeBlocks) {
|
|
std::unique_lock<std::mutex> lk(writeMutex);
|
|
if(writeQueue.size_approx() > writeQueueSize) {
|
|
writeCV.wait(lk);
|
|
}
|
|
} else {
|
|
if(writeQueue.size_approx() > writeQueueSize) {
|
|
err(APIError::TransmitBufferFull);
|
|
return false;
|
|
}
|
|
}
|
|
return writeQueue.enqueue(WriteOperation(bytes));
|
|
} |