1 # Remote nusers client interface

     2 

     3 import rpc

     4 from rpc import Packer, Unpacker, UDPClient, BroadcastUDPClient

     5 

     6 

     7 class RnusersPacker(Packer):

     8     def pack_utmp(self, ui):

     9         ut_line, ut_name, ut_host, ut_time = utmp

    10         self.pack_string(ut_line)

    11         self.pack_string(ut_name)

    12         self.pack_string(ut_host)

    13         self.pack_int(ut_time)

    14     def pack_utmpidle(self, ui):

    15         ui_itmp, ui_idle = ui

    16         self.pack_utmp(ui_utmp)

    17         self.pack_uint(ui_idle)

    18     def pack_utmpidlearr(self, list):

    19         self.pack_array(list, self.pack_itmpidle)

    20 

    21 

    22 class RnusersUnpacker(Unpacker):

    23     def unpack_utmp(self):

    24         ut_line = self.unpack_string()

    25         ut_name = self.unpack_string()

    26         ut_host = self.unpack_string()

    27         ut_time = self.unpack_int()

    28         return ut_line, ut_name, ut_host, ut_time

    29     def unpack_utmpidle(self):

    30         ui_utmp = self.unpack_utmp()

    31         ui_idle = self.unpack_uint()

    32         return ui_utmp, ui_idle

    33     def unpack_utmpidlearr(self):

    34         return self.unpack_array(self.unpack_utmpidle)

    35 

    36 

    37 class PartialRnusersClient:

    38 

    39     def addpackers(self):

    40         self.packer = RnusersPacker()

    41         self.unpacker = RnusersUnpacker('')

    42 

    43     def Num(self):

    44         return self.make_call(1, None, None, self.unpacker.unpack_int)

    45 

    46     def Names(self):

    47         return self.make_call(2, None, \

    48                 None, self.unpacker.unpack_utmpidlearr)

    49 

    50     def Allnames(self):

    51         return self.make_call(3, None, \

    52                 None, self.unpacker.unpack_utmpidlearr)

    53 

    54 

    55 class RnusersClient(PartialRnusersClient, UDPClient):

    56 

    57     def __init__(self, host):

    58         UDPClient.__init__(self, host, 100002, 2)

    59 

    60 

    61 class BroadcastRnusersClient(PartialRnusersClient, BroadcastUDPClient):

    62 

    63     def __init__(self, bcastaddr):

    64         BroadcastUDPClient.__init__(self, bcastaddr, 100002, 2)

    65 

    66 

    67 def test():

    68     import sys

    69     if not sys.argv[1:]:

    70         testbcast()

    71         return

    72     else:

    73         host = sys.argv[1]

    74     c = RnusersClient(host)

    75     list = c.Names()

    76     for (line, name, host, time), idle in list:

    77         line = strip0(line)

    78         name = strip0(name)

    79         host = strip0(host)

    80         print "%r %r %r %s %s" % (name, host, line, time, idle)

    81 

    82 def testbcast():

    83     c = BroadcastRnusersClient('<broadcast>')

    84     def listit(list, fromaddr):

    85         host, port = fromaddr

    86         print host + '\t:',

    87         for (line, name, host, time), idle in list:

    88             print strip0(name),

    89         print

    90     c.set_reply_handler(listit)

    91     all = c.Names()

    92     print 'Total Count:', len(all)

    93 

    94 def strip0(s):

    95     while s and s[-1] == '\0': s = s[:-1]

    96     return s

    97 

    98 test()