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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
|
## features.py
##
## Copyright (C) 2003-2004 Alexey "Snake" Nezhdanov
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2, or (at your option)
## any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
## GNU General Public License for more details.
# $Id: features.py, v1.26 2013/10/21 alkorgun Exp $
"""
This module contains variable stuff that is not worth splitting into separate modules.
Here is:
DISCO client and agents-to-DISCO and browse-to-DISCO emulators.
IBR and password manager.
jabber:iq:privacy methods
All these methods takes "disp" first argument that should be already connected
(and in most cases already authorised) dispatcher instance.
"""
from .protocol import *
REGISTER_DATA_RECEIVED = "REGISTER DATA RECEIVED"
def _discover(disp, ns, jid, node=None, fb2b=0, fb2a=1):
"""
Try to obtain info from the remote object.
If remote object doesn't support disco fall back to browse (if fb2b is true)
and if it doesnt support browse (or fb2b is not true) fall back to agents protocol
(if gb2a is true). Returns obtained info. Used internally.
"""
iq = Iq(to=jid, typ="get", queryNS=ns)
if node:
iq.setQuerynode(node)
rep = disp.SendAndWaitForResponse(iq)
if fb2b and not isResultNode(rep):
rep = disp.SendAndWaitForResponse(Iq(to=jid, typ="get", queryNS=NS_BROWSE)) # Fallback to browse
if fb2a and not isResultNode(rep):
rep = disp.SendAndWaitForResponse(Iq(to=jid, typ="get", queryNS=NS_AGENTS)) # Fallback to agents
if isResultNode(rep):
return [n for n in rep.getQueryPayload() if isinstance(n, Node)]
return []
def discoverItems(disp, jid, node=None):
"""
Query remote object about any items that it contains. Return items list.
"""
ret = []
for i in _discover(disp, NS_DISCO_ITEMS, jid, node):
if i.getName() == "agent" and i.getTag("name"):
i.setAttr("name", i.getTagData("name"))
ret.append(i.attrs)
return ret
def discoverInfo(disp, jid, node=None):
"""
Query remote object about info that it publishes. Returns identities and features lists.
"""
identities, features = [], []
for i in _discover(disp, NS_DISCO_INFO, jid, node):
if i.getName() == "identity":
identities.append(i.attrs)
elif i.getName() == "feature":
features.append(i.getAttr("var"))
elif i.getName() == "agent":
if i.getTag("name"):
i.setAttr("name", i.getTagData("name"))
if i.getTag("description"):
i.setAttr("name", i.getTagData("description"))
identities.append(i.attrs)
if i.getTag("groupchat"):
features.append(NS_GROUPCHAT)
if i.getTag("register"):
features.append(NS_REGISTER)
if i.getTag("search"):
features.append(NS_SEARCH)
return identities, features
def getRegInfo(disp, host, info={}, sync=True):
"""
Gets registration form from remote host.
You can pre-fill the info dictionary.
F.e. if you are requesting info on registering user joey than specify
info as {"username": "joey"}. See JEP-0077 for details.
"disp" must be connected dispatcher instance.
"""
iq = Iq("get", NS_REGISTER, to=host)
for i in info.keys():
iq.setTagData(i, info[i])
if sync:
resp = disp.SendAndWaitForResponse(iq)
_ReceivedRegInfo(disp.Dispatcher, resp, host)
return resp
else:
disp.SendAndCallForResponse(iq, _ReceivedRegInfo, {"agent": host})
def _ReceivedRegInfo(con, resp, agent):
iq = Iq("get", NS_REGISTER, to=agent)
if not isResultNode(resp):
return None
df = resp.getTag("query", namespace=NS_REGISTER).getTag("x", namespace=NS_DATA)
if df:
con.Event(NS_REGISTER, REGISTER_DATA_RECEIVED, (agent, DataForm(node=df)))
return None
df = DataForm(typ="form")
for i in resp.getQueryPayload():
if not isinstance(i, Iq):
pass
elif i.getName() == "instructions":
df.addInstructions(i.getData())
else:
df.setField(i.getName()).setValue(i.getData())
con.Event(NS_REGISTER, REGISTER_DATA_RECEIVED, (agent, df))
def register(disp, host, info):
"""
Perform registration on remote server with provided info.
disp must be connected dispatcher instance.
Returns true or false depending on registration result.
If registration fails you can get additional info from the dispatcher's owner
attributes lastErrNode, lastErr and lastErrCode.
"""
iq = Iq("set", NS_REGISTER, to=host)
if not isinstance(info, dict):
info = info.asDict()
for i in info.keys():
iq.setTag("query").setTagData(i, info[i])
resp = disp.SendAndWaitForResponse(iq)
if isResultNode(resp):
return 1
def unregister(disp, host):
"""
Unregisters with host (permanently removes account).
disp must be connected and authorized dispatcher instance.
Returns true on success.
"""
resp = disp.SendAndWaitForResponse(Iq("set", NS_REGISTER, to=host, payload=[Node("remove")]))
if isResultNode(resp):
return 1
def changePasswordTo(disp, newpassword, host=None):
"""
Changes password on specified or current (if not specified) server.
disp must be connected and authorized dispatcher instance.
Returns true on success."""
if not host:
host = disp._owner.Server
resp = disp.SendAndWaitForResponse(Iq("set", NS_REGISTER, to=host,
payload=[
Node("username", payload=[disp._owner.User]),
Node("password", payload=[newpassword])
]))
if isResultNode(resp):
return 1
def getPrivacyLists(disp):
"""
Requests privacy lists from connected server.
Returns dictionary of existing lists on success.
"""
dict = {"lists": []}
try:
resp = disp.SendAndWaitForResponse(Iq("get", NS_PRIVACY))
if not isResultNode(resp):
return None
for list in resp.getQueryPayload():
if list.getName() == "list":
dict["lists"].append(list.getAttr("name"))
else:
dict[list.getName()] = list.getAttr("name")
except Exception:
pass
else:
return dict
def getPrivacyList(disp, listname):
"""
Requests specific privacy list listname. Returns list of XML nodes (rules)
taken from the server responce.
"""
try:
resp = disp.SendAndWaitForResponse(Iq("get", NS_PRIVACY, payload=[Node("list", {"name": listname})]))
if isResultNode(resp):
return resp.getQueryPayload()[0]
except Exception:
pass
def setActivePrivacyList(disp, listname=None, typ="active"):
"""
Switches privacy list "listname" to specified type.
By default the type is "active". Returns true on success.
"""
if listname:
attrs = {"name": listname}
else:
attrs = {}
resp = disp.SendAndWaitForResponse(Iq("set", NS_PRIVACY, payload=[Node(typ, attrs)]))
if isResultNode(resp):
return 1
def setDefaultPrivacyList(disp, listname=None):
"""
Sets the default privacy list as "listname". Returns true on success.
"""
return setActivePrivacyList(disp, listname, "default")
def setPrivacyList(disp, list):
"""
Set the ruleset. "list" should be the simpleXML node formatted
according to RFC 3921 (XMPP-IM) (I.e. Node("list", {"name": listname}, payload=[...]) )
Returns true on success.
"""
resp = disp.SendAndWaitForResponse(Iq("set", NS_PRIVACY, payload=[list]))
if isResultNode(resp):
return 1
def delPrivacyList(disp, listname):
"""
Deletes privacy list "listname". Returns true on success.
"""
resp = disp.SendAndWaitForResponse(Iq("set", NS_PRIVACY, payload=[Node("list", {"name": listname})]))
if isResultNode(resp):
return 1
|
