src/core/oidentdconfiggenerator.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include "oidentdconfiggenerator.h"
  22 #include "corenetwork.h"
  23
  24 #include <QString>
  25
  26 OidentdConfigGenerator::OidentdConfigGenerator(bool strict, QObject *parent) :
  27     QObject(parent),
  28     _initialized(false),
  29     _strict(strict)
  30 {
  31     if (!_initialized)
  32         init();
  33 }
  34
  35
  36 OidentdConfigGenerator::~OidentdConfigGenerator()
  37 {
  38     _quasselConfig.clear();
  39     writeConfig();
  40     _configFile->deleteLater();
  41 }
  42
  43
  44 bool OidentdConfigGenerator::init()
  45 {
  46     _configDir = QDir::homePath();
  47     _configFileName = ".oidentd.conf";
  48
  49     if (Quassel::isOptionSet("oidentd-conffile"))
  50         _configPath = Quassel::optionValue("oidentd-conffile");
  51     else
  52         _configPath = _configDir.absoluteFilePath(_configFileName);
  53
  54     _configTag = " stanza created by Quassel";
  55
  56     _configFile = new QFile(_configPath);
  57
  58     // Rx has to match Template in order for cleanup to work.
  59     // Template should be enhanced with the "from" parameter as soon as Quassel gains
  60     // the ability to bind to an IP on client sockets.
  61
  62     _quasselStanzaTemplate = QString("lport %1 { reply \"%2\" } #%3\n");
  63     _quasselStanzaRx = QRegExp(QString("^lport .* \\{ .* \\} #%1\\r?\\n").arg(_configTag));
  64
  65     // initially remove all Quassel stanzas that might be present
  66     if (parseConfig(false) && writeConfig())
  67         _initialized = true;
  68
  69     return _initialized;
  70 }
  71
  72 const QString OidentdConfigGenerator::sysidentForIdentity(const CoreIdentity *identity) {
  73     if (!_strict) {
  74         return identity->ident();
  75     }
  76     const CoreNetwork *network = qobject_cast<CoreNetwork *>(sender());
  77     return network->coreSession()->strictSysident();
  78 }
  79
  80 bool OidentdConfigGenerator::addSocket(const CoreIdentity *identity, const QHostAddress &localAddress, quint16 localPort, const QHostAddress &peerAddress, quint16 peerPort)
  81 {
  82     Q_UNUSED(localAddress) Q_UNUSED(peerAddress) Q_UNUSED(peerPort)
  83     const QString ident = sysidentForIdentity(identity);
  84
  85     _quasselConfig.append(_quasselStanzaTemplate.arg(localPort).arg(ident).arg(_configTag).toLatin1());
  86
  87     bool ret = writeConfig();
  88
  89     return ret;
  90 }
  91
  92
  93 //! not yet implemented
  94 bool OidentdConfigGenerator::removeSocket(const CoreIdentity *identity, const QHostAddress &localAddress, quint16 localPort, const QHostAddress &peerAddress, quint16 peerPort)
  95 {
  96     Q_UNUSED(identity) Q_UNUSED(localAddress) Q_UNUSED(localPort) Q_UNUSED(peerAddress) Q_UNUSED(peerPort)
  97     return true;
  98 }
  99
 100
 101 bool OidentdConfigGenerator::parseConfig(bool readQuasselStanzas)
 102 {
 103     if (!_configFile->exists())
 104         return true;
 105
 106     if (!_configFile->isOpen() && !_configFile->open(QIODevice::ReadOnly))
 107         return false;
 108     _mutex.lock();
 109
 110     _parsedConfig.clear();
 111     _configFile->seek(0);
 112     while (!_configFile->atEnd()) {
 113         QByteArray line = _configFile->readLine();
 114
 115         if (!lineByUs(line))
 116             _parsedConfig.append(line);
 117         else if (readQuasselStanzas)
 118             _quasselConfig.append(line);
 119     }
 120
 121     _configFile->close();
 122     _mutex.unlock();
 123     return true;
 124 }
 125
 126
 127 bool OidentdConfigGenerator::writeConfig()
 128 {
 129 #ifdef HAVE_UMASK
 130     mode_t prev_umask = umask(S_IXUSR | S_IWGRP | S_IXGRP | S_IWOTH | S_IXOTH); // == 0133, rw-r--r--
 131 #endif
 132     bool not_open = (!_configFile->isOpen() && !_configFile->open(QIODevice::ReadWrite | QIODevice::Text));
 133 #ifdef HAVE_UMASK
 134     umask(prev_umask);
 135 #endif
 136
 137     if (not_open)
 138         return false;
 139
 140     _mutex.lock();
 141
 142     _configFile->seek(0);
 143     _configFile->resize(0);
 144     _configFile->write(_parsedConfig);
 145     _configFile->write(_quasselConfig);
 146
 147     _configFile->close();
 148     _mutex.unlock();
 149     return true;
 150 }
 151
 152
 153 bool OidentdConfigGenerator::lineByUs(const QByteArray &line)
 154 {
 155     return _quasselStanzaRx.exactMatch(line);
 156 }