blob: a208a11f8bc1e3df42279fa71cc43ed483b3a9d8 (
plain)
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
101
102
|
/******************************************************************************
**
** Copyright (C) 2016 The Qt Company Ltd.
** Contact: http://www.qt.io/licensing/
**
** This file is part of the Qt Debug Bridge.
**
** $QT_BEGIN_LICENSE:COMM$
**
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see http://www.qt.io/terms-conditions. For further
** information use the contact form at http://www.qt.io/contact-us.
**
** $QT_END_LICENSE$
**
******************************************************************************/
#include "filepushexecutor.h"
#include "../utils/make_unique.h"
#include "filepushcommon.h"
#include "stream.h"
#include <QtCore/qdatastream.h>
#include <QtCore/qdebug.h>
#include <QtCore/qfile.h>
FilePushExecutor::FilePushExecutor(Stream *stream)
: m_stream{stream},
m_sink{nullptr}
{
if (m_stream)
connect(m_stream, &Stream::packetAvailable, this, &Executor::receive);
}
void FilePushExecutor::receive(StreamPacket packet)
{
uint32_t typeValue;
packet >> typeValue;
auto type = toFilePushPacketType(typeValue);
switch (type) {
case FilePushOpen: {
QString sinkPath;
packet >> sinkPath;
openSink(sinkPath);
break;
}
case FilePushWrite: {
QByteArray fileData;
packet >> fileData;
writeToSink(fileData);
break;
}
case FilePushEnd:
closeSink();
break;
case FilePushError:
qDebug() << "FilePushError from host";
if (m_sink)
m_sink->remove();
break;
default:
qFatal("Unsupported FilePushPacketType %d in ProcessExecutor::receive", type);
}
}
void FilePushExecutor::openSink(const QString &path)
{
qDebug() << "Opening sink file" << path;
m_sink = make_unique<QFile>(path);
StreamPacket packet;
if (!m_sink->open(QIODevice::WriteOnly))
packet << FilePushError;
else
packet << FilePushOpened;
m_stream->write(packet);
}
void FilePushExecutor::writeToSink(const QByteArray &data)
{
auto written = m_sink->write(data);
StreamPacket packet;
if (written != data.size())
packet << FilePushError;
else
packet << FilePushWritten;
m_stream->write(packet);
}
void FilePushExecutor::closeSink()
{
qDebug() << "Closing sink file" << m_sink->fileName();
m_sink->close();
}
|