summaryrefslogtreecommitdiffstats
path: root/Demo/rpc/rnusersclient.py
blob: 4d540f85922ee298340ca5e1c619f952019f2be6 (plain)
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
# Remote nusers client interface

import rpc
from rpc import Packer, Unpacker, UDPClient, BroadcastUDPClient


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 PartialRnusersClient:

	def addpackers(self):
		self.packer = RnusersPacker().init()
		self.unpacker = RnusersUnpacker().init('')

	def Num(self):
		return self.make_call(1, None, None, self.unpacker.unpack_int)

	def Names(self):
		return self.make_call(2, None, \
			None, self.unpacker.unpack_utmpidlearr)

	def Allnames(self):
		return self.make_call(3, None, \
			None, self.unpacker.unpack_utmpidlearr)


class RnusersClient(PartialRnusersClient, UDPClient):

	def init(self, host):
		return UDPClient.init(self, host, 100002, 2)


class BroadcastRnusersClient(PartialRnusersClient, BroadcastUDPClient):

	def init(self, bcastaddr):
		return BroadcastUDPClient.init(self, bcastaddr, 100002, 2)


def test():
	import sys
	if not sys.argv[1:]:
		testbcast()
		return
	else:
		host = sys.argv[1]
	c = RnusersClient().init(host)
	list = c.Names()
	for (line, name, host, time), idle in list:
		line = strip0(line)
		name = strip0(name)
		host = strip0(host)
		print `name`, `host`, `line`, time, idle

def testbcast():
	c = BroadcastRnusersClient().init('<broadcast>')
	def listit(list, fromaddr):
		host, port = fromaddr
		print host + '\t:',
		for (line, name, host, time), idle in list:
			print strip0(name),
		print
	c.set_reply_handler(listit)
	all = c.Names()
	print 'Total Count:', len(all)

def strip0(s):
	while s and s[-1] == '\0': s = s[:-1]
	return s

test()