workspace/kcontrol/kfontinst/kio/Helper.cpp

   1 /*
   2  * KFontInst - KDE Font Installer
   3  *
   4  * Copyright 2003-2007 Craig Drummond <craig@kde.org>
   5  *
   6  * ----
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; see the file COPYING.  If not, write to
  20  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  21  * Boston, MA 02110-1301, USA.
  22  */
  23
  24 #include "Helper.h"
  25 #include "Misc.h"
  26 #include "DisabledFonts.h"
  27 #include "KfiConstants.h"
  28 #include "Socket.h"
  29 #include "kxftconfig.h"
  30 #include <QtCore/QFile>
  31 #include <QtCore/QDataStream>
  32 #include <QtCore/QVariant>
  33 #include <KDE/KGlobal>
  34 #include <KDE/KComponentData>
  35 #include <KDE/KDebug>
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38
  39 namespace KFI
  40 {
  41
  42 class CHelper : public CSocket
  43 {
  44     public:
  45
  46     CHelper(char *sock, int uid) : itsSock(sock), itsUid(uid) { }
  47
  48     int run();
  49
  50     private:
  51
  52     bool getFile(CDisabledFonts::TFont &font);
  53     bool disableFont();
  54     bool enableFont();
  55     bool deleteDisabledFont();
  56     bool addToFc();
  57     bool configureX();
  58     bool copyFile();
  59     bool moveFile();
  60     bool deleteFile();
  61     bool createDir();
  62     bool createAfm();
  63
  64     private:
  65
  66     const char     *itsSock;
  67     int            itsUid;
  68     CDisabledFonts itsDisabledFonts;
  69 };
  70
  71 int CHelper::run()
  72 {
  73     if(connectToServer(itsSock, itsUid))
  74     {
  75         for(;;)
  76         {
  77             QVariant cmd;
  78             bool     res(false);
  79
  80             kDebug() << "Waiting for cmd...";
  81             if(!read(cmd, -1))
  82                 break;
  83
  84             switch(cmd.type())
  85             {
  86                 case QVariant::Int:
  87                     switch(cmd.toInt())
  88                     {
  89                         case KFI::CMD_ENABLE_FONT:
  90                             res=enableFont();
  91                             break;
  92                         case KFI::CMD_DISABLE_FONT:
  93                             res=disableFont();
  94                             break;
  95                         case KFI::CMD_DELETE_DISABLED_FONT:
  96                             res=deleteDisabledFont();
  97                             break;
  98                         case KFI::CMD_RELOAD_DISABLED_LIST:
  99                             kDebug() << "reloadDisabled";
 100                             res=itsDisabledFonts.save();
 101                             itsDisabledFonts.load();
 102                             break;
 103                         case KFI::CMD_COPY_FILE:
 104                             res=copyFile();
 105                             break;
 106                         case KFI::CMD_MOVE_FILE:
 107                             res=moveFile();
 108                             break;
 109                         case KFI::CMD_DELETE_FILE:
 110                             res=deleteFile();
 111                             break;
 112                         case KFI::CMD_CREATE_DIR:
 113                             res=createDir();
 114                             break;
 115                         case KFI::CMD_CREATE_AFM:
 116                             res=createAfm();
 117                             break;
 118                         case KFI::CMD_FC_CACHE:
 119                             kDebug() << "fc-cache";
 120                             Misc::doCmd("fc-cache");
 121                             res=true;
 122                             break;
 123                         case KFI::CMD_ADD_DIR_TO_FONTCONFIG:
 124                             res=addToFc();
 125                             break;
 126                         case KFI::CMD_CFG_DIR_FOR_X:
 127                             res=configureX();
 128                             break;
 129                         case KFI::CMD_QUIT:
 130                             write(true);
 131                             return 0;
 132                             break;
 133                         default:
 134                             kError() << "Unknown command:" << cmd.toInt() << endl;
 135                             return -1;
 136                     }
 137                     break;
 138                 case QVariant::Invalid:
 139                     kDebug() << "Finished";
 140                     return 0; // finished!
 141                 default:
 142                     kError() << "Invalid type received when expecting command int, " << cmd.type() << endl;
 143                     break;
 144             }
 145
 146             if(!write(res))
 147             {
 148                 kError() << " Could not write result to kio_fonts" << endl;
 149                 break;
 150             }
 151         }
 152     }
 153     else
 154         kError() << " Could not connect to kio_fonts" << endl;
 155
 156     return -1;
 157 }
 158
 159 bool CHelper::getFile(CDisabledFonts::TFont &font)
 160 {
 161     kDebug() << "getFile";
 162
 163     QString file,
 164             foundry;
 165
 166     if(read(file) && read(foundry))
 167     {
 168         kDebug() << "file:" << file << " foundry:" << foundry;
 169         font.files.append(CDisabledFonts::TFile(file, 0, foundry));
 170         return true;
 171     }
 172
 173     kError() << " Invalid parameters for getFile" << endl;
 174
 175     return false;
 176 }
 177
 178 bool CHelper::disableFont()
 179 {
 180     kDebug() << "disableFont";
 181
 182     QString    family;
 183     int        style,
 184                face,
 185                numFiles;
 186     qulonglong writingSystems;
 187
 188     if(read(family) && read(style) && read(writingSystems) && read(face) && read(numFiles))
 189     {
 190         kDebug() << "family:" << family << " style:" << style << " face:" << face
 191                  << " numFiles:" << numFiles << endl;
 192
 193         CDisabledFonts::TFont font(family, style, writingSystems);
 194
 195         for(int i=0; i<numFiles; ++i)
 196             if(!getFile(font))
 197                 return false;
 198
 199         if(font.files.count())
 200         {
 201             if(1==font.files.count())
 202                 (*(font.files.begin())).face=face;
 203             return itsDisabledFonts.disable(font);
 204         }
 205         else
 206             return false;
 207     }
 208
 209     kError() << " Invalid parameters for disableFont" << endl;
 210     return false;
 211 }
 212
 213 bool CHelper::enableFont()
 214 {
 215     kDebug() << "enableFont";
 216
 217     QString family;
 218     int     style;
 219
 220     if(read(family) && read(style))
 221     {
 222         kDebug() << "family:" << family << " style:" << style;
 223
 224         return itsDisabledFonts.enable(family, style);
 225     }
 226
 227     kError() << " Invalid parameters for enableFont" << endl;
 228     return false;
 229 }
 230
 231 bool CHelper::deleteDisabledFont()
 232 {
 233     kDebug() << "deleteDisabledFont";
 234
 235     QString family;
 236     int     style;
 237
 238     if(read(family) && read(style))
 239     {
 240         kDebug() << "family:" << family << " style:" << style;
 241
 242         CDisabledFonts::TFont               font(family, style);
 243         CDisabledFonts::TFontList::Iterator it(itsDisabledFonts.items().find(font));
 244
 245         if(itsDisabledFonts.modifiable() && itsDisabledFonts.items().end()!=it)
 246         {
 247             CDisabledFonts::TFileList::ConstIterator fIt((*it).files.begin()),
 248                                                      fEnd((*it).files.end());
 249
 250             for(; fIt!=fEnd; ++fIt)
 251                 if(::unlink(QFile::encodeName((*fIt).path).constData()))
 252                     return false;
 253
 254             itsDisabledFonts.remove(it);
 255             itsDisabledFonts.refresh();
 256             return true;
 257         }
 258         else
 259             return false;
 260     }
 261
 262     kError() << " Invalid parameters for disableFont" << endl;
 263     return false;
 264 }
 265
 266 bool CHelper::addToFc()
 267 {
 268     kDebug() << "addToFc";
 269
 270     QString dir;
 271
 272     if(read(dir))
 273     {
 274         kDebug() << "dir:" << dir;
 275
 276         KXftConfig xft(KXftConfig::Dirs, KFI::Misc::root());
 277
 278         xft.addDir(dir);
 279         return xft.apply();
 280     }
 281
 282     kError() << " Invalid parameters for addToFc" << endl;
 283     return false;
 284 }
 285
 286 bool CHelper::configureX()
 287 {
 288     kDebug() << "configureX";
 289
 290     QString dir;
 291
 292     if(read(dir))
 293     {
 294         kDebug() << "dir:" << dir;
 295         return Misc::configureForX11(dir);
 296     }
 297
 298     kError() << " Invalid parameters for configureX" << endl;
 299     return false;
 300 }
 301
 302 bool CHelper::copyFile()
 303 {
 304     kDebug() << "copyFile";
 305
 306     QString from,
 307             to;
 308
 309     if(read(from) && read(to))
 310     {
 311         kDebug() << "from:" << from << " to:" << to;
 312
 313         bool rv=QFile::copy(from, to);
 314         if(rv)
 315             Misc::setFilePerms(to);
 316         return rv;
 317     }
 318
 319     kError() << " Invalid parameters for copyFile" << endl;
 320     return false;
 321 }
 322
 323 bool CHelper::moveFile()
 324 {
 325     kDebug() << "moveFile";
 326
 327     QString from,
 328             to;
 329     int     user,
 330             group;
 331
 332     if(read(from) && read(to) && read(user) && read(group))
 333     {
 334         kDebug() << "from:" << from << " to:" << to << " user:" << user << " group:" << group;
 335
 336         QByteArray toC(QFile::encodeName(to));
 337
 338         bool res(0==::rename(QFile::encodeName(from).constData(), toC.constData()));
 339
 340         if(res)
 341         {
 342             Misc::setFilePerms(toC);
 343             ::chown(toC.constData(), user, group);
 344         }
 345
 346         return res;
 347     }
 348
 349     kError() << " Invalid parameters for moveFile" << endl;
 350     return false;
 351 }
 352
 353 bool CHelper::deleteFile()
 354 {
 355     kDebug() << "deleteFile";
 356
 357     QString file;
 358
 359     if(read(file))
 360     {
 361         kDebug() << "file:" << file;
 362         return 0==::unlink(QFile::encodeName(file).constData());
 363     }
 364
 365     kError() << " Invalid parameters for deleteFile" << endl;
 366     return false;
 367 }
 368
 369 bool CHelper::createDir()
 370 {
 371     kDebug() << "createDir";
 372
 373     QString dir;
 374
 375     if(read(dir))
 376     {
 377         kDebug() << "dir:" << dir;
 378
 379         return KFI::Misc::createDir(dir);
 380     }
 381
 382     kError() << " Invalid parameters for createDir" << endl;
 383     return false;
 384 }
 385
 386 bool CHelper::createAfm()
 387 {
 388     kDebug() << "createAfm";
 389
 390     QString font;
 391
 392     if(read(font))
 393     {
 394         kDebug() << "font:" << font;
 395
 396         return Misc::doCmd("pf2afm", QFile::encodeName(font));
 397     }
 398
 399     kError() << " Invalid parameters for createAfm" << endl;
 400     return false;
 401 }
 402
 403 }
 404
 405 int main(int argc, char *argv[])
 406 {
 407     if(argc==3)
 408         return KFI::CHelper(argv[1], atoi(argv[2])).run();
 409     else
 410         fprintf(stderr, "Usage: %s <socket> <uid>\n", argv[0]);
 411     return -1;
 412 }