/
remoteiodevice.h
100 lines (92 loc) · 2.37 KB
/
remoteiodevice.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#ifndef REMOTEIODEVICE_H
#define REMOTEIODEVICE_H
#include <QIODevice>
#include <QMutex>
#include <QWaitCondition>
#include <QDebug>
class RemoteIODevice : public QIODevice
{
Q_OBJECT
public:
RemoteIODevice() :
m_eof(false), m_headers(false), m_filesize(0)
{};
// abuse this to only block until headers are available:
bool waitForReadyRead ( int msecs )
{
if(!m_headers)
{
qDebug() << "RemoteIODevice::waitForReadyRead";
m_wait.wait(&m_mut);
}else
qDebug() << "RemoteIODevice - no waiting required";
return true;
}
qint64 size () const
{
return m_filesize;
};
qint64 readData ( char * data, qint64 maxSize )
{
qDebug() << "RemIO::readData";
m_mut_recv.lock();
if(m_eof && m_buffer.length() == 0)
{
// eof
m_mut_recv.unlock();
return 0;
}
if(!m_buffer.length())
{
// wait for more data to arrive
m_mut_recv.unlock();
m_wait.wait(&m_mut);
m_mut_recv.lock();
}
int len;
if(maxSize>=m_buffer.length()) // whole buffer
{
len = m_buffer.length();
memcpy(data, m_buffer.constData(), len);
m_buffer.clear();
} else { // partial
len = maxSize;
memcpy(data, m_buffer.constData(), len);
m_buffer.remove(0,len);
}
m_mut_recv.unlock();
return len;
};
qint64 writeData ( const char * data, qint64 maxSize )
{
Q_ASSERT(false);
return 0;
};
void addData(QByteArray msg)
{
qDebug() << "RemIO::addData";
m_mut_recv.lock();
if(msg.length()==0) m_eof=true;
else
{
if(!m_headers)
{
m_filesize = QString::fromAscii(msg).toInt();
m_headers = true;
qDebug() << "RemIODev, filesize: " << m_filesize;
}
else m_buffer.append(msg);
}
m_mut_recv.unlock();
m_wait.wakeAll();
qDebug() << "just read: " << msg.length();
//emit QIODevice::readyRead();
}
private:
QByteArray m_buffer;
QMutex m_mut, m_mut_recv;
QWaitCondition m_wait;
bool m_eof, m_headers;
int m_filesize;
};
#endif // REMOTEIODEVICE_H