← Back to branch summary

~ubuntu-branches/ubuntu/utopic/kde-workspace/utopic-proposed

« back to all changes in this revision

Viewing changes to kwrited/kwrited.cpp

  • Committer: Bazaar Package Importer
  • Author(s): Michał Zając
  • Date: 2011-07-09 08:31:15 UTC
  • Revision ID: james.westby@ubuntu.com-20110709083115-ohyxn6z93mily9fc
Tags: upstream-4.6.90
Import upstream version 4.6.90

Show diffs side-by-side

added added

removed removed

Lines of Context:
kwrited/kwrited.cpp
 
1
/*
 
2
    kwrited is a write(1) receiver for KDE.
 
3
    Copyright 1997,1998 by Lars Doelle <lars.doelle@on-line.de>
 
4
    Copyright 2008 by George Kiagiadakis <gkiagia@users.sourceforge.net>
 
5
 
 
6
    This program is free software; you can redistribute it and/or modify
 
7
    it under the terms of the GNU General Public License as published by
 
8
    the Free Software Foundation; either version 2 of the License, or
 
9
    (at your option) any later version.
 
10
 
 
11
    This program is distributed in the hope that it will be useful,
 
12
    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
    GNU General Public License for more details.
 
15
 
 
16
    You should have received a copy of the GNU General Public License
 
17
    along with this program; if not, write to the Free Software
 
18
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 
19
    02110-1301  USA.
 
20
*/
 
21
 
 
22
// Own
 
23
#include "kwrited.h"
 
24
 
 
25
#include <kdebug.h>
 
26
#include <kptydevice.h>
 
27
#include <kuser.h>
 
28
#include <knotification.h>
 
29
#include <klocalizedstring.h>
 
30
#include <kaboutdata.h>
 
31
#include <kdeversion.h>
 
32
 
 
33
#if defined(BUILD_AS_EXECUTABLE)
 
34
# include <QtCore/QCoreApplication>
 
35
# include <kcomponentdata.h>
 
36
# include <signal.h>
 
37
# include <sys/types.h>
 
38
# include <unistd.h>
 
39
#else
 
40
# include <kpluginfactory.h>
 
41
# include <kpluginloader.h>
 
42
#endif
 
43
 
 
44
static inline KAboutData aboutData()
 
45
{
 
46
    return KAboutData("kwrited", 0, ki18n("kwrited"), KDE_VERSION_STRING);
 
47
}
 
48
 
 
49
#if defined(BUILD_AS_EXECUTABLE)
 
50
 
 
51
static uid_t original_euid;
 
52
static gid_t original_egid;
 
53
 
 
54
static void sigterm_handler(int signal)
 
55
{
 
56
    kDebug() << "Caught signal" << signal << ", exiting...";
 
57
    QCoreApplication::quit();
 
58
}
 
59
 
 
60
int main(int argc, char **argv)
 
61
{
 
62
    //drop elevated privileges temporarily
 
63
    original_euid = geteuid();
 
64
    original_egid = getegid();
 
65
    seteuid(getuid());
 
66
    setegid(getgid());
 
67
 
 
68
    //install a signal handler to exit gracefully
 
69
    //(so that pty->logout() is called in ~KWrited())
 
70
    signal(SIGTERM, sigterm_handler);
 
71
    signal(SIGINT, sigterm_handler);
 
72
    signal(SIGHUP, sigterm_handler);
 
73
 
 
74
    KComponentData kcompdata(aboutData());
 
75
    QCoreApplication a(argc, argv);
 
76
    KWrited w;
 
77
    return a.exec();
 
78
}
 
79
 
 
80
#else // BUILD_AS_EXECUTABLE
 
81
 
 
82
KWritedModule::KWritedModule(QObject* parent, const QList<QVariant>&)
 
83
    : KDEDModule(parent)
 
84
{
 
85
    pro = new KWrited;
 
86
}
 
87
 
 
88
KWritedModule::~KWritedModule()
 
89
{
 
90
    delete pro;
 
91
}
 
92
 
 
93
K_PLUGIN_FACTORY(KWritedFactory,
 
94
                 registerPlugin<KWritedModule>();
 
95
    )
 
96
K_EXPORT_PLUGIN(KWritedFactory(aboutData()))
 
97
 
 
98
#endif //BUILD_AS_EXECUTABLE
 
99
 
 
100
 
 
101
KWrited::KWrited() : QObject()
 
102
{
 
103
  pty = new KPtyDevice();
 
104
  pty->open();
 
105
 
 
106
#if defined(BUILD_AS_EXECUTABLE)
 
107
  dup2(pty->slaveFd(), 0); //HACK: login() from glibc requires this to login correctly
 
108
  //restore elevated privileges
 
109
  seteuid(original_euid);
 
110
  setegid(original_egid);
 
111
#endif
 
112
 
 
113
  pty->login(KUser(KUser::UseRealUserID).loginName().toLocal8Bit().data(), qgetenv("DISPLAY"));
 
114
  
 
115
#if defined(BUILD_AS_EXECUTABLE)
 
116
  //drop privileges again
 
117
  seteuid(getuid());
 
118
  setegid(getgid());
 
119
#endif
 
120
 
 
121
  connect(pty, SIGNAL(readyRead()), this, SLOT(block_in()));
 
122
  kDebug() << "listening on device" << pty->ttyName();
 
123
}
 
124
 
 
125
KWrited::~KWrited()
 
126
{
 
127
#if defined(BUILD_AS_EXECUTABLE)
 
128
  //restore elevated privileges
 
129
  seteuid(original_euid);
 
130
  setegid(original_egid);
 
131
#endif
 
132
 
 
133
    pty->logout();
 
134
 
 
135
#if defined(BUILD_AS_EXECUTABLE)
 
136
  //drop privileges again
 
137
  seteuid(getuid());
 
138
  setegid(getgid());
 
139
#endif
 
140
 
 
141
    delete pty;
 
142
}
 
143
 
 
144
void KWrited::block_in()
 
145
{
 
146
  QByteArray buf = pty->readAll();
 
147
  QString msg = QString::fromLocal8Bit( buf.constData(), buf.size() );
 
148
  msg.remove('\r');
 
149
  msg.remove('\a');
 
150
 
 
151
  KNotification *notification = new KNotification("NewMessage", 0, KNotification::Persistent);
 
152
#if !defined(BUILD_AS_EXECUTABLE)
 
153
  notification->setComponentData( KWritedFactory::componentData() );
 
154
#endif
 
155
  notification->setText( msg );
 
156
  connect(notification, SIGNAL(closed()), notification, SLOT(deleteLater()) );
 
157
  notification->sendEvent();
 
158
}
 
159
 
 
160
#include "kwrited.moc"