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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
# -*- coding: utf-8 -*-
#
# File: portato/listener.py
# This file is part of the Portato-Project, a graphical portage-frontend.
#
# Copyright (C) 2006-2009 René 'Necoro' Neumann
# This is free software. You may redistribute copies of it under the terms of
# the GNU General Public License version 2.
# There is NO WARRANTY, to the extent permitted by law.
#
# Written by René 'Necoro' Neumann <necoro@necoro.net>
from __future__ import absolute_import
import os
from subprocess import Popen
try:
import pynotify
except ImportError:
pynotify = None
from .constants import APP
from .helper import debug, warning
class Listener (object):
"""This class handles the communication between the "listener" and the GUI.
This listener starts programs as the user while the GUI runs as root.
@ivar _recv: listener socket
@type _recv: int
@ivar _send: sender socket
@type _send: int"""
def set_recv (self, mem, sig, rw):
self._mem = mem
self._sig = sig
self._rw = rw
while True:
try:
try:
self._sig.P()
self._rw.P()
len = self._mem.read(NumberOfBytes = 4)
string = self._mem.read(NumberOfBytes = int(len), offset = 4)
finally:
self._rw.V()
data = string.split("\0")
debug("Listener received: %s", data)
if data[0] == "notify":
self.do_notify(*data[1:])
elif data[0] == "cmd":
self.do_cmd(data[1:])
elif data[0] == "close":
break
except KeyboardInterrupt:
debug("Got KeyboardInterrupt. Aborting.")
break
self._mem.remove()
self._sig.remove()
self._rw.remove()
self._mem = None
self._sig = None
self._rw = None
def do_cmd (self, cmdlist):
"""Starts a command as the user.
@param cmdlist: list of command (options)
@type cmdlist: string[]"""
Popen(cmdlist)
def do_notify(self, base, descr, icon, urgency = None):
"""Displays a notify.
This will do nothing if pynotify is not present and/or root is running the listener."""
if pynotify and os.getuid() != 0:
if not pynotify.is_initted():
pynotify.init(APP)
n = pynotify.Notification(base, descr, icon)
if urgency is not None and urgency != "":
n.set_urgency(int(urgency))
n.show()
def set_send (self, mem = None, sig = None, rw = None):
if mem is None or sig is None or rw is None:
warning(_("Listener has not been started."))
self._mem = None
self._sig = None
self._rw = None
else:
import shm_wrapper as shm
self._mem = shm.SharedMemoryHandle(mem)
self._sig = shm.SemaphoreHandle(sig)
self._rw = shm.SemaphoreHandle(rw)
def __send (self, string):
self._rw.P()
self._sig.Z()
try:
self._mem.write("%4d%s" % (len(string), string))
self._sig.V()
finally:
self._rw.V()
def send_notify (self, base = "", descr = "", icon = "", urgency = None):
if self._sig is None:
self.do_notify(base, descr, icon, urgency)
else:
string = "\0".join(["notify", base, descr, icon])
if urgency is not None:
string += "\0%d" % urgency
else:
string += "\0"
self.__send(string)
def send_cmd (self, cmdlist):
if self._sig is None:
self.do_cmd(cmdlist)
else:
self.__send("\0".join(["cmd"] +cmdlist))
def close (self):
if self._sig is not None:
self.__send("close")
|
