1 /* Ricochet - https://ricochet.im/
2 * Copyright (C) 2014, John Brooks <john.brooks@dereferenced.net>
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met:
7 *
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *
11 * * Redistributions in binary form must reproduce the above
12 * copyright notice, this list of conditions and the following disclaimer
13 * in the documentation and/or other materials provided with the
14 * distribution.
15 *
16 * * Neither the names of the copyright owners nor the names of its
17 * contributors may be used to endorse or promote products derived from
18 * this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #include "TorProcess_p.h"
34 #include "utils/CryptoKey.h"
35 #include "utils/SecureRNG.h"
36 #include <QDir>
37 #include <QDebug>
38 #include <QCoreApplication>
39
40 using namespace Tor;
41
TorProcess(QObject * parent)42 TorProcess::TorProcess(QObject *parent)
43 : QObject(parent), d(new TorProcessPrivate(this))
44 {
45 }
46
~TorProcess()47 TorProcess::~TorProcess()
48 {
49 if (state() > NotStarted)
50 stop();
51 }
52
TorProcessPrivate(TorProcess * q)53 TorProcessPrivate::TorProcessPrivate(TorProcess *q)
54 : QObject(q), q(q), state(TorProcess::NotStarted), controlPort(0), controlPortAttempts(0)
55 {
56 connect(&process, &QProcess::started, this, &TorProcessPrivate::processStarted);
57 connect(&process, (void (QProcess::*)(int, QProcess::ExitStatus))&QProcess::finished,
58 this, &TorProcessPrivate::processFinished);
59 connect(&process, (void (QProcess::*)(QProcess::ProcessError))&QProcess::error,
60 this, &TorProcessPrivate::processError);
61 connect(&process, &QProcess::readyRead, this, &TorProcessPrivate::processReadable);
62
63 controlPortTimer.setInterval(500);
64 connect(&controlPortTimer, &QTimer::timeout, this, &TorProcessPrivate::tryReadControlPort);
65 }
66
executable() const67 QString TorProcess::executable() const
68 {
69 return d->executable;
70 }
71
setExecutable(const QString & path)72 void TorProcess::setExecutable(const QString &path)
73 {
74 d->executable = path;
75 }
76
dataDir() const77 QString TorProcess::dataDir() const
78 {
79 return d->dataDir;
80 }
81
setDataDir(const QString & path)82 void TorProcess::setDataDir(const QString &path)
83 {
84 d->dataDir = path;
85 }
86
defaultTorrc() const87 QString TorProcess::defaultTorrc() const
88 {
89 return d->defaultTorrc;
90 }
91
setDefaultTorrc(const QString & path)92 void TorProcess::setDefaultTorrc(const QString &path)
93 {
94 d->defaultTorrc = path;
95 }
96
extraSettings() const97 QStringList TorProcess::extraSettings() const
98 {
99 return d->extraSettings;
100 }
101
setExtraSettings(const QStringList & settings)102 void TorProcess::setExtraSettings(const QStringList &settings)
103 {
104 d->extraSettings = settings;
105 }
106
state() const107 TorProcess::State TorProcess::state() const
108 {
109 return d->state;
110 }
111
errorMessage() const112 QString TorProcess::errorMessage() const
113 {
114 return d->errorMessage;
115 }
116
start()117 void TorProcess::start()
118 {
119 if (state() > NotStarted)
120 return;
121
122 d->errorMessage.clear();
123
124 if (d->executable.isEmpty() || d->dataDir.isEmpty()) {
125 d->errorMessage = QStringLiteral("Tor executable and data directory not specified");
126 d->state = Failed;
127 emit errorMessageChanged(d->errorMessage);
128 emit stateChanged(d->state);
129 return;
130 }
131
132 if (!d->ensureFilesExist()) {
133 d->state = Failed;
134 emit errorMessageChanged(d->errorMessage);
135 emit stateChanged(d->state);
136 return;
137 }
138
139 QByteArray password = controlPassword();
140 QByteArray hashedPassword = torControlHashedPassword(password);
141 if (password.isEmpty() || hashedPassword.isEmpty()) {
142 d->errorMessage = QStringLiteral("Random password generation failed");
143 d->state = Failed;
144 emit errorMessageChanged(d->errorMessage);
145 emit stateChanged(d->state);
146 }
147
148 QStringList args;
149 if (!d->defaultTorrc.isEmpty())
150 args << QStringLiteral("--defaults-torrc") << d->defaultTorrc;
151 args << QStringLiteral("-f") << d->torrcPath();
152 args << QStringLiteral("DataDirectory") << d->dataDir;
153 args << QStringLiteral("HashedControlPassword") << QString::fromLatin1(hashedPassword);
154 args << QStringLiteral("ControlPort") << QStringLiteral("auto");
155 args << QStringLiteral("ControlPortWriteToFile") << d->controlPortFilePath();
156 args << QStringLiteral("__OwningControllerProcess") << QString::number(qApp->applicationPid());
157 args << d->extraSettings;
158
159 d->state = Starting;
160 emit stateChanged(d->state);
161
162 if (QFile::exists(d->controlPortFilePath()))
163 QFile::remove(d->controlPortFilePath());
164 d->controlPort = 0;
165 d->controlHost.clear();
166
167 d->process.setProcessChannelMode(QProcess::MergedChannels);
168 d->process.start(d->executable, args, QIODevice::ReadOnly);
169 }
170
stop()171 void TorProcess::stop()
172 {
173 if (state() < Starting)
174 return;
175
176 d->controlPortTimer.stop();
177
178 if (d->process.state() == QProcess::Starting)
179 d->process.waitForStarted(2000);
180
181 d->state = NotStarted;
182
183 // Windows can't terminate the process well, but Tor will clean itself up
184 #ifndef Q_OS_WIN
185 if (d->process.state() == QProcess::Running) {
186 d->process.terminate();
187 if (!d->process.waitForFinished(5000)) {
188 qWarning() << "Tor process" << d->process.pid() << "did not respond to terminate, killing...";
189 d->process.kill();
190 if (!d->process.waitForFinished(2000)) {
191 qCritical() << "Tor process" << d->process.pid() << "did not respond to kill!";
192 }
193 }
194 }
195 #endif
196
197 emit stateChanged(d->state);
198 }
199
controlPassword()200 QByteArray TorProcess::controlPassword()
201 {
202 if (d->controlPassword.isEmpty())
203 d->controlPassword = SecureRNG::randomPrintable(16);
204 return d->controlPassword;
205 }
206
controlHost()207 QHostAddress TorProcess::controlHost()
208 {
209 return d->controlHost;
210 }
211
controlPort()212 quint16 TorProcess::controlPort()
213 {
214 return d->controlPort;
215 }
216
ensureFilesExist()217 bool TorProcessPrivate::ensureFilesExist()
218 {
219 QFile torrc(torrcPath());
220 if (!torrc.exists()) {
221 QDir dir(dataDir);
222 if (!dir.exists() && !dir.mkpath(QStringLiteral("."))) {
223 errorMessage = QStringLiteral("Cannot create Tor data directory: %1").arg(dataDir);
224 return false;
225 }
226
227 if (!torrc.open(QIODevice::ReadWrite)) {
228 errorMessage = QStringLiteral("Cannot create Tor configuration file: %1").arg(torrcPath());
229 return false;
230 }
231 }
232
233 return true;
234 }
235
torrcPath() const236 QString TorProcessPrivate::torrcPath() const
237 {
238 return QDir::toNativeSeparators(dataDir) + QDir::separator() + QStringLiteral("torrc");
239 }
240
controlPortFilePath() const241 QString TorProcessPrivate::controlPortFilePath() const
242 {
243 return QDir::toNativeSeparators(dataDir) + QDir::separator() + QStringLiteral("control-port");
244 }
245
processStarted()246 void TorProcessPrivate::processStarted()
247 {
248 state = TorProcess::Connecting;
249 emit q->stateChanged(state);
250
251 controlPortAttempts = 0;
252 controlPortTimer.start();
253 }
254
processFinished()255 void TorProcessPrivate::processFinished()
256 {
257 if (state < TorProcess::Starting)
258 return;
259
260 controlPortTimer.stop();
261 errorMessage = process.errorString();
262 if (errorMessage.isEmpty())
263 errorMessage = QStringLiteral("Process exited unexpectedly (code %1)").arg(process.exitCode());
264 state = TorProcess::Failed;
265 emit q->errorMessageChanged(errorMessage);
266 emit q->stateChanged(state);
267 }
268
processError(QProcess::ProcessError error)269 void TorProcessPrivate::processError(QProcess::ProcessError error)
270 {
271 if (error == QProcess::FailedToStart || error == QProcess::Crashed)
272 processFinished();
273 }
274
processReadable()275 void TorProcessPrivate::processReadable()
276 {
277 while (process.bytesAvailable() > 0) {
278 QByteArray line = process.readLine(2048).trimmed();
279 if (!line.isEmpty())
280 emit q->logMessage(QString::fromLatin1(line));
281 }
282 }
283
tryReadControlPort()284 void TorProcessPrivate::tryReadControlPort()
285 {
286 QFile file(controlPortFilePath());
287 if (file.open(QIODevice::ReadOnly)) {
288 QByteArray data = file.readLine().trimmed();
289
290 int p;
291 if (data.startsWith("PORT=") && (p = data.lastIndexOf(':')) > 0) {
292 controlHost = QHostAddress(QString::fromLatin1(data.mid(5, p - 5)));
293 controlPort = data.mid(p+1).toUShort();
294
295 if (!controlHost.isNull() && controlPort > 0) {
296 controlPortTimer.stop();
297 state = TorProcess::Ready;
298 emit q->stateChanged(state);
299 return;
300 }
301 }
302 }
303
304 if (++controlPortAttempts * controlPortTimer.interval() > 10000) {
305 errorMessage = QStringLiteral("No control port available after launching process");
306 state = TorProcess::Failed;
307 emit q->errorMessageChanged(errorMessage);
308 emit q->stateChanged(state);
309 }
310 }
311
312