diff options
Diffstat (limited to 'portato')
-rw-r--r-- | portato/constants.py | 9 | ||||
-rw-r--r-- | portato/plistener.py | 53 |
2 files changed, 21 insertions, 41 deletions
diff --git a/portato/constants.py b/portato/constants.py index d871e78..3aea05d 100644 --- a/portato/constants.py +++ b/portato/constants.py @@ -40,8 +40,6 @@ These should be set during the installation. @type STD_FRONTEND: string @var SU_COMMAND: command to execute to "su" @type SU_COMMAND: string -@var SOCKET: path to socket for communication between listener and GUI -@type SOCKET: string """ import os from os.path import join as pjoin @@ -56,6 +54,8 @@ SETTINGS_DIR = pjoin(HOME, "."+APP) CONFIG_DIR = "/etc/portato/" CONFIG_LOCATION = pjoin(CONFIG_DIR, "portato.cfg") +VAR_DIR = "/var/portato/" + DATA_DIR = "portato/gui/templates/" PLUGIN_DIR = "plugins/" @@ -67,10 +67,9 @@ APP_ICON = pjoin(ICON_DIR, "portato-icon.png") LOCALE_DIR = "i18n/" -FRONTENDS = ["gtk" ,"qt"] +FRONTENDS = ["gtk"] STD_FRONTEND = "gtk" SU_COMMAND = "gksu -D 'Portato'" -SOCKET = "/tmp/portato.socket" -USE_CATAPULT = False +USE_CATAPULT = True diff --git a/portato/plistener.py b/portato/plistener.py index 452bee6..a24a262 100644 --- a/portato/plistener.py +++ b/portato/plistener.py @@ -21,7 +21,7 @@ try: except ImportError: pynotify = None -from .constants import SOCKET, APP +from .constants import APP from .helper import debug, warning class PListener (object): @@ -29,29 +29,17 @@ class PListener (object): This listener starts programs as the user while the GUI runs as root. @ivar _recv: listener socket - @type _recv: socket.socket + @type _recv: int @ivar _send: sender socket - @type _send: socket.socket""" + @type _send: int""" - def set_recv (self): - self._recv = socket.socket(socket.AF_UNIX) + def set_recv (self, pipe): + self._recv = pipe - try: - self._recv.bind(SOCKET) - except socket.error, e: - if int(e[0]) == 98: # already existing - delete - os.unlink(SOCKET) - self._recv.bind(SOCKET) - else: - raise - - self._recv.listen(1) - con, addr = self._recv.accept() - while True: try: - len = con.recv(4) - string = con.recv(int(len)) + len = os.read(self._recv, 4) + string = os.read(self._recv, int(len)) data = string.split("\0") @@ -62,10 +50,9 @@ class PListener (object): elif data[0] == "close": break except KeyboardInterrupt: - pass + break - con.close() - self._recv.close() + os.close(self._recv) def do_cmd (self, cmdlist): """Starts a command as the user. @@ -87,20 +74,15 @@ class PListener (object): n.set_urgency(int(urgency)) n.show() - def set_send (self): - self._send = socket.socket(socket.AF_UNIX) - try: - self._send.connect(SOCKET) - except socket.error, e: - if e[0] in [111, 2]: # can't connect - warning(_("Listener has not been started.")) - self._send = None - else: - raise + def set_send (self, pipe = None): + if pipe is None: + warning(_("Listener has not been started.")) + + self._send = pipe def __send (self, string): - self._send.sendall("%4d" % len(string)) - self._send.sendall(string) + os.write(self._send, "%4d" % len(string)) + os.write(self._send, string) def send_notify (self, base = "", descr = "", icon = "", urgency = None): if self._send is None: @@ -124,5 +106,4 @@ class PListener (object): def close (self): if self._send is not None: self.__send("close") - self._send.close() - os.unlink(SOCKET) + os.close(self._send) |