82 lines
1.8 KiB
Python
82 lines
1.8 KiB
Python
# Remote nusers client interface
|
|
|
|
import rpc
|
|
from rpc import Packer, Unpacker, TCPClient, UDPClient
|
|
|
|
|
|
class RnusersPacker(Packer):
|
|
def pack_utmp(self, ui):
|
|
ut_line, ut_name, ut_host, ut_time = utmp
|
|
self.pack_string(ut_line)
|
|
self.pack_string(ut_name)
|
|
self.pack_string(ut_host)
|
|
self.pack_int(ut_time)
|
|
def pack_utmpidle(self, ui):
|
|
ui_itmp, ui_idle = ui
|
|
self.pack_utmp(ui_utmp)
|
|
self.pack_uint(ui_idle)
|
|
def pack_utmpidlearr(self, list):
|
|
self.pack_array(list, self.pack_itmpidle)
|
|
|
|
|
|
class RnusersUnpacker(Unpacker):
|
|
def unpack_utmp(self):
|
|
ut_line = self.unpack_string()
|
|
ut_name = self.unpack_string()
|
|
ut_host = self.unpack_string()
|
|
ut_time = self.unpack_int()
|
|
return ut_line, ut_name, ut_host, ut_time
|
|
def unpack_utmpidle(self):
|
|
ui_utmp = self.unpack_utmp()
|
|
ui_idle = self.unpack_uint()
|
|
return ui_utmp, ui_idle
|
|
def unpack_utmpidlearr(self):
|
|
return self.unpack_array(self.unpack_utmpidle)
|
|
|
|
|
|
class RnusersClient(UDPClient):
|
|
|
|
def addpackers(self):
|
|
self.packer = RnusersPacker().init()
|
|
self.unpacker = RnusersUnpacker().init('')
|
|
|
|
def init(self, host):
|
|
return UDPClient.init(self, host, 100002, 2)
|
|
|
|
def Num(self):
|
|
self.start_call(1)
|
|
self.do_call()
|
|
n = self.unpacker.unpack_int()
|
|
self.end_call()
|
|
return n
|
|
|
|
def Names(self):
|
|
self.start_call(2)
|
|
self.do_call()
|
|
list = self.unpacker.unpack_utmpidlearr()
|
|
self.end_call()
|
|
return list
|
|
|
|
def Allnames(self):
|
|
self.start_call(3)
|
|
self.do_call()
|
|
list = self.unpacker.unpack_utmpidlearr()
|
|
self.end_call()
|
|
return list
|
|
|
|
|
|
def test():
|
|
import sys
|
|
host = ''
|
|
if sys.argv[1:]: host = sys.argv[1]
|
|
c = RnusersClient().init(host)
|
|
list = c.Names()
|
|
def strip0(s):
|
|
while s and s[-1] == '\0': s = s[:-1]
|
|
return s
|
|
for (line, name, host, time), idle in list:
|
|
line = strip0(line)
|
|
name = strip0(name)
|
|
host = strip0(host)
|
|
print name, host, line, time, idle
|