annotate plugins/plugin_xep_0065.py @ 40:6f0699ba0329

plugin XEP-0077: minor log fix
author Goffi <goffi@goffi.org>
date Sat, 19 Dec 2009 18:13:04 +1100
parents 3e24753b9e0b
children 4392f1fdb064
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
goffi@necton2
parents:
diff changeset
1 #!/usr/bin/python
goffi@necton2
parents:
diff changeset
2 #-*- coding: utf-8 -*-
goffi@necton2
parents:
diff changeset
3 """
goffi@necton2
parents:
diff changeset
4 SAT plugin for managing xep-0065
goffi@necton2
parents:
diff changeset
5
goffi@necton2
parents:
diff changeset
6 Copyright (C)
goffi@necton2
parents:
diff changeset
7 2002-2004 Dave Smith (dizzyd@jabber.org)
goffi@necton2
parents:
diff changeset
8 2007-2008 Fabio Forno (xmpp:ff@jabber.bluendo.com)
goffi@necton2
parents:
diff changeset
9 2009 Jérôme Poisson (goffi@goffi.org)
goffi@necton2
parents:
diff changeset
10
goffi@necton2
parents:
diff changeset
11 This program is free software: you can redistribute it and/or modify
goffi@necton2
parents:
diff changeset
12 it under the terms of the GNU General Public License as published by
goffi@necton2
parents:
diff changeset
13 the Free Software Foundation, either version 3 of the License, or
goffi@necton2
parents:
diff changeset
14 (at your option) any later version.
goffi@necton2
parents:
diff changeset
15
goffi@necton2
parents:
diff changeset
16 This program is distributed in the hope that it will be useful,
goffi@necton2
parents:
diff changeset
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
goffi@necton2
parents:
diff changeset
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
goffi@necton2
parents:
diff changeset
19 GNU General Public License for more details.
goffi@necton2
parents:
diff changeset
20
goffi@necton2
parents:
diff changeset
21 You should have received a copy of the GNU General Public License
goffi@necton2
parents:
diff changeset
22 along with this program. If not, see <http://www.gnu.org/licenses/>.
goffi@necton2
parents:
diff changeset
23
goffi@necton2
parents:
diff changeset
24 --
goffi@necton2
parents:
diff changeset
25
goffi@necton2
parents:
diff changeset
26 This program is based on proxy65 (http://code.google.com/p/proxy65),
goffi@necton2
parents:
diff changeset
27 originaly written by David Smith and modified by Fabio Forno.
goffi@necton2
parents:
diff changeset
28 It is sublicensed under GPL v3 (or any later version) as allowed by the original
goffi@necton2
parents:
diff changeset
29 license.
goffi@necton2
parents:
diff changeset
30
goffi@necton2
parents:
diff changeset
31 --
goffi@necton2
parents:
diff changeset
32
goffi@necton2
parents:
diff changeset
33 Here is a copy of the original license:
goffi@necton2
parents:
diff changeset
34
goffi@necton2
parents:
diff changeset
35 Copyright (C)
goffi@necton2
parents:
diff changeset
36 2002-2004 Dave Smith (dizzyd@jabber.org)
goffi@necton2
parents:
diff changeset
37 2007-2008 Fabio Forno (xmpp:ff@jabber.bluendo.com)
goffi@necton2
parents:
diff changeset
38
goffi@necton2
parents:
diff changeset
39 Permission is hereby granted, free of charge, to any person obtaining a copy
goffi@necton2
parents:
diff changeset
40 of this software and associated documentation files (the "Software"), to deal
goffi@necton2
parents:
diff changeset
41 in the Software without restriction, including without limitation the rights
goffi@necton2
parents:
diff changeset
42 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
goffi@necton2
parents:
diff changeset
43 copies of the Software, and to permit persons to whom the Software is
goffi@necton2
parents:
diff changeset
44 furnished to do so, subject to the following conditions:
goffi@necton2
parents:
diff changeset
45
goffi@necton2
parents:
diff changeset
46 The above copyright notice and this permission notice shall be included in
goffi@necton2
parents:
diff changeset
47 all copies or substantial portions of the Software.
goffi@necton2
parents:
diff changeset
48
goffi@necton2
parents:
diff changeset
49 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
goffi@necton2
parents:
diff changeset
50 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
goffi@necton2
parents:
diff changeset
51 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
goffi@necton2
parents:
diff changeset
52 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
goffi@necton2
parents:
diff changeset
53 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
goffi@necton2
parents:
diff changeset
54 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
goffi@necton2
parents:
diff changeset
55 THE SOFTWARE.
goffi@necton2
parents:
diff changeset
56 """
goffi@necton2
parents:
diff changeset
57
goffi@necton2
parents:
diff changeset
58 from logging import debug, info, error
goffi@necton2
parents:
diff changeset
59 from twisted.internet import protocol, reactor
goffi@necton2
parents:
diff changeset
60 from twisted.protocols.basic import FileSender
goffi@necton2
parents:
diff changeset
61 from twisted.words.xish import domish
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
62 from twisted.web.client import getPage
0
goffi@necton2
parents:
diff changeset
63 import struct
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
64 import urllib
0
goffi@necton2
parents:
diff changeset
65 import hashlib, pdb
goffi@necton2
parents:
diff changeset
66
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
67 from zope.interface import implements
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
68
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
69 try:
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
70 from twisted.words.protocols.xmlstream import XMPPHandler
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
71 except ImportError:
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
72 from wokkel.subprotocols import XMPPHandler
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
73
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
74 from wokkel import disco, iwokkel
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
75
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
76 IQ_SET = '/iq[@type="set"]'
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
77 NS_BS = 'http://jabber.org/protocol/bytestreams'
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
78 BS_REQUEST = IQ_SET + '/query[@xmlns="' + NS_BS + '"]'
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
79
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
80
0
goffi@necton2
parents:
diff changeset
81
goffi@necton2
parents:
diff changeset
82 PLUGIN_INFO = {
goffi@necton2
parents:
diff changeset
83 "name": "XEP 0065 Plugin",
goffi@necton2
parents:
diff changeset
84 "import_name": "XEP_0065",
goffi@necton2
parents:
diff changeset
85 "type": "XEP",
goffi@necton2
parents:
diff changeset
86 "main": "XEP_0065",
9
63ab3d8058f4 description of plugin_xep_0065
Goffi <goffi@goffi.org>
parents: 8
diff changeset
87 "description": """Implementation of SOCKS5 Bytestreams"""
0
goffi@necton2
parents:
diff changeset
88 }
goffi@necton2
parents:
diff changeset
89
goffi@necton2
parents:
diff changeset
90 STATE_INITIAL = 0
goffi@necton2
parents:
diff changeset
91 STATE_AUTH = 1
goffi@necton2
parents:
diff changeset
92 STATE_REQUEST = 2
goffi@necton2
parents:
diff changeset
93 STATE_READY = 3
goffi@necton2
parents:
diff changeset
94 STATE_AUTH_USERPASS = 4
goffi@necton2
parents:
diff changeset
95 STATE_TARGET_INITIAL = 5
goffi@necton2
parents:
diff changeset
96 STATE_TARGET_AUTH = 6
goffi@necton2
parents:
diff changeset
97 STATE_TARGET_REQUEST = 7
goffi@necton2
parents:
diff changeset
98 STATE_TARGET_READY = 8
goffi@necton2
parents:
diff changeset
99 STATE_LAST = 9
goffi@necton2
parents:
diff changeset
100
goffi@necton2
parents:
diff changeset
101 STATE_CONNECT_PENDING = STATE_LAST + 1
goffi@necton2
parents:
diff changeset
102
goffi@necton2
parents:
diff changeset
103 SOCKS5_VER = 0x05
goffi@necton2
parents:
diff changeset
104
goffi@necton2
parents:
diff changeset
105 ADDR_IPV4 = 0x01
goffi@necton2
parents:
diff changeset
106 ADDR_DOMAINNAME = 0x03
goffi@necton2
parents:
diff changeset
107 ADDR_IPV6 = 0x04
goffi@necton2
parents:
diff changeset
108
goffi@necton2
parents:
diff changeset
109 CMD_CONNECT = 0x01
goffi@necton2
parents:
diff changeset
110 CMD_BIND = 0x02
goffi@necton2
parents:
diff changeset
111 CMD_UDPASSOC = 0x03
goffi@necton2
parents:
diff changeset
112
goffi@necton2
parents:
diff changeset
113 AUTHMECH_ANON = 0x00
goffi@necton2
parents:
diff changeset
114 AUTHMECH_USERPASS = 0x02
goffi@necton2
parents:
diff changeset
115 AUTHMECH_INVALID = 0xFF
goffi@necton2
parents:
diff changeset
116
goffi@necton2
parents:
diff changeset
117 REPLY_SUCCESS = 0x00
goffi@necton2
parents:
diff changeset
118 REPLY_GENERAL_FAILUR = 0x01
goffi@necton2
parents:
diff changeset
119 REPLY_CONN_NOT_ALLOWED = 0x02
goffi@necton2
parents:
diff changeset
120 REPLY_NETWORK_UNREACHABLE = 0x03
goffi@necton2
parents:
diff changeset
121 REPLY_HOST_UNREACHABLE = 0x04
goffi@necton2
parents:
diff changeset
122 REPLY_CONN_REFUSED = 0x05
goffi@necton2
parents:
diff changeset
123 REPLY_TTL_EXPIRED = 0x06
goffi@necton2
parents:
diff changeset
124 REPLY_CMD_NOT_SUPPORTED = 0x07
goffi@necton2
parents:
diff changeset
125 REPLY_ADDR_NOT_SUPPORTED = 0x08
goffi@necton2
parents:
diff changeset
126
goffi@necton2
parents:
diff changeset
127
goffi@necton2
parents:
diff changeset
128
goffi@necton2
parents:
diff changeset
129
goffi@necton2
parents:
diff changeset
130
goffi@necton2
parents:
diff changeset
131 class SOCKSv5(protocol.Protocol, FileSender):
goffi@necton2
parents:
diff changeset
132 def __init__(self):
goffi@necton2
parents:
diff changeset
133 debug("Protocol init")
goffi@necton2
parents:
diff changeset
134 self.state = STATE_INITIAL
goffi@necton2
parents:
diff changeset
135 self.buf = ""
goffi@necton2
parents:
diff changeset
136 self.supportedAuthMechs = [ AUTHMECH_ANON ]
goffi@necton2
parents:
diff changeset
137 self.supportedAddrs = [ ADDR_DOMAINNAME ]
goffi@necton2
parents:
diff changeset
138 self.enabledCommands = [ CMD_CONNECT ]
goffi@necton2
parents:
diff changeset
139 self.peersock = None
goffi@necton2
parents:
diff changeset
140 self.addressType = 0
goffi@necton2
parents:
diff changeset
141 self.requestType = 0
goffi@necton2
parents:
diff changeset
142 self.activeConns = {}
goffi@necton2
parents:
diff changeset
143 self.pendingConns = {}
goffi@necton2
parents:
diff changeset
144 self.transfered = 0 #nb of bytes already copied
goffi@necton2
parents:
diff changeset
145
goffi@necton2
parents:
diff changeset
146 def _startNegotiation(self):
goffi@necton2
parents:
diff changeset
147 debug("_startNegotiation")
goffi@necton2
parents:
diff changeset
148 self.state = STATE_TARGET_AUTH
goffi@necton2
parents:
diff changeset
149 self.transport.write(struct.pack('!3B', SOCKS5_VER, 1, AUTHMECH_ANON))
goffi@necton2
parents:
diff changeset
150
goffi@necton2
parents:
diff changeset
151 def _parseNegotiation(self):
goffi@necton2
parents:
diff changeset
152 debug("_parseNegotiation")
goffi@necton2
parents:
diff changeset
153 try:
goffi@necton2
parents:
diff changeset
154 # Parse out data
goffi@necton2
parents:
diff changeset
155 ver, nmethod = struct.unpack('!BB', self.buf[:2])
goffi@necton2
parents:
diff changeset
156 methods = struct.unpack('%dB' % nmethod, self.buf[2:nmethod+2])
goffi@necton2
parents:
diff changeset
157
goffi@necton2
parents:
diff changeset
158 # Ensure version is correct
goffi@necton2
parents:
diff changeset
159 if ver != 5:
goffi@necton2
parents:
diff changeset
160 self.transport.write(struct.pack('!BB', SOCKS5_VER, AUTHMECH_INVALID))
goffi@necton2
parents:
diff changeset
161 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
162 return
goffi@necton2
parents:
diff changeset
163
goffi@necton2
parents:
diff changeset
164 # Trim off front of the buffer
goffi@necton2
parents:
diff changeset
165 self.buf = self.buf[nmethod+2:]
goffi@necton2
parents:
diff changeset
166
goffi@necton2
parents:
diff changeset
167 # Check for supported auth mechs
goffi@necton2
parents:
diff changeset
168 for m in self.supportedAuthMechs:
goffi@necton2
parents:
diff changeset
169 if m in methods:
goffi@necton2
parents:
diff changeset
170 # Update internal state, according to selected method
goffi@necton2
parents:
diff changeset
171 if m == AUTHMECH_ANON:
goffi@necton2
parents:
diff changeset
172 self.state = STATE_REQUEST
goffi@necton2
parents:
diff changeset
173 elif m == AUTHMECH_USERPASS:
goffi@necton2
parents:
diff changeset
174 self.state = STATE_AUTH_USERPASS
goffi@necton2
parents:
diff changeset
175 # Complete negotiation w/ this method
goffi@necton2
parents:
diff changeset
176 self.transport.write(struct.pack('!BB', SOCKS5_VER, m))
goffi@necton2
parents:
diff changeset
177 return
goffi@necton2
parents:
diff changeset
178
goffi@necton2
parents:
diff changeset
179 # No supported mechs found, notify client and close the connection
goffi@necton2
parents:
diff changeset
180 self.transport.write(struct.pack('!BB', SOCKS5_VER, AUTHMECH_INVALID))
goffi@necton2
parents:
diff changeset
181 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
182 except struct.error:
goffi@necton2
parents:
diff changeset
183 pass
goffi@necton2
parents:
diff changeset
184
goffi@necton2
parents:
diff changeset
185 def _parseUserPass(self):
goffi@necton2
parents:
diff changeset
186 debug("_parseUserPass")
goffi@necton2
parents:
diff changeset
187 try:
goffi@necton2
parents:
diff changeset
188 # Parse out data
goffi@necton2
parents:
diff changeset
189 ver, ulen = struct.unpack('BB', self.buf[:2])
goffi@necton2
parents:
diff changeset
190 uname, = struct.unpack('%ds' % ulen, self.buf[2:ulen + 2])
goffi@necton2
parents:
diff changeset
191 plen, = struct.unpack('B', self.buf[ulen + 2])
goffi@necton2
parents:
diff changeset
192 password, = struct.unpack('%ds' % plen, self.buf[ulen + 3:ulen + 3 + plen])
goffi@necton2
parents:
diff changeset
193 # Trim off fron of the buffer
goffi@necton2
parents:
diff changeset
194 self.buf = self.buf[3 + ulen + plen:]
goffi@necton2
parents:
diff changeset
195 # Fire event to authenticate user
goffi@necton2
parents:
diff changeset
196 if self.authenticateUserPass(uname, password):
goffi@necton2
parents:
diff changeset
197 # Signal success
goffi@necton2
parents:
diff changeset
198 self.state = STATE_REQUEST
goffi@necton2
parents:
diff changeset
199 self.transport.write(struct.pack('!BB', SOCKS5_VER, 0x00))
goffi@necton2
parents:
diff changeset
200 else:
goffi@necton2
parents:
diff changeset
201 # Signal failure
goffi@necton2
parents:
diff changeset
202 self.transport.write(struct.pack('!BB', SOCKS5_VER, 0x01))
goffi@necton2
parents:
diff changeset
203 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
204 except struct.error:
goffi@necton2
parents:
diff changeset
205 pass
goffi@necton2
parents:
diff changeset
206
goffi@necton2
parents:
diff changeset
207 def sendErrorReply(self, errorcode):
goffi@necton2
parents:
diff changeset
208 debug("sendErrorReply")
goffi@necton2
parents:
diff changeset
209 # Any other address types are not supported
goffi@necton2
parents:
diff changeset
210 result = struct.pack('!BBBBIH', SOCKS5_VER, errorcode, 0, 1, 0, 0)
goffi@necton2
parents:
diff changeset
211 self.transport.write(result)
goffi@necton2
parents:
diff changeset
212 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
213
goffi@necton2
parents:
diff changeset
214 def addConnection(self, address, connection):
goffi@necton2
parents:
diff changeset
215 info("Adding connection: %s, %s", address, connection)
goffi@necton2
parents:
diff changeset
216 olist = self.pendingConns.get(address, [])
goffi@necton2
parents:
diff changeset
217 if len(olist) <= 1:
goffi@necton2
parents:
diff changeset
218 olist.append(connection)
goffi@necton2
parents:
diff changeset
219 self.pendingConns[address] = olist
goffi@necton2
parents:
diff changeset
220 return True
goffi@necton2
parents:
diff changeset
221 else:
goffi@necton2
parents:
diff changeset
222 return False
goffi@necton2
parents:
diff changeset
223
goffi@necton2
parents:
diff changeset
224 def removePendingConnection(self, address, connection):
goffi@necton2
parents:
diff changeset
225 olist = self.pendingConns[address]
goffi@necton2
parents:
diff changeset
226 if len(olist) == 1:
goffi@necton2
parents:
diff changeset
227 del self.pendingConns[address]
goffi@necton2
parents:
diff changeset
228 else:
goffi@necton2
parents:
diff changeset
229 olist.remove(connection)
goffi@necton2
parents:
diff changeset
230 self.pendingConns[address] = olist
goffi@necton2
parents:
diff changeset
231
goffi@necton2
parents:
diff changeset
232 def removeActiveConnection(self, address):
goffi@necton2
parents:
diff changeset
233 del self.activeConns[address]
goffi@necton2
parents:
diff changeset
234
goffi@necton2
parents:
diff changeset
235 def _parseRequest(self):
goffi@necton2
parents:
diff changeset
236 debug("_parseRequest")
goffi@necton2
parents:
diff changeset
237 try:
goffi@necton2
parents:
diff changeset
238 # Parse out data and trim buffer accordingly
goffi@necton2
parents:
diff changeset
239 ver, cmd, rsvd, self.addressType = struct.unpack('!BBBB', self.buf[:4])
goffi@necton2
parents:
diff changeset
240
goffi@necton2
parents:
diff changeset
241 # Ensure we actually support the requested address type
goffi@necton2
parents:
diff changeset
242 if self.addressType not in self.supportedAddrs:
goffi@necton2
parents:
diff changeset
243 self.sendErrorReply(REPLY_ADDR_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
244 return
goffi@necton2
parents:
diff changeset
245
goffi@necton2
parents:
diff changeset
246 # Deal with addresses
goffi@necton2
parents:
diff changeset
247 if self.addressType == ADDR_IPV4:
goffi@necton2
parents:
diff changeset
248 addr, port = struct.unpack('!IH', self.buf[4:10])
goffi@necton2
parents:
diff changeset
249 self.buf = self.buf[10:]
goffi@necton2
parents:
diff changeset
250 elif self.addressType == ADDR_DOMAINNAME:
goffi@necton2
parents:
diff changeset
251 nlen = ord(self.buf[4])
goffi@necton2
parents:
diff changeset
252 addr, port = struct.unpack('!%dsH' % nlen, self.buf[5:])
goffi@necton2
parents:
diff changeset
253 self.buf = self.buf[7 + len(addr):]
goffi@necton2
parents:
diff changeset
254 else:
goffi@necton2
parents:
diff changeset
255 # Any other address types are not supported
goffi@necton2
parents:
diff changeset
256 self.sendErrorReply(REPLY_ADDR_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
257 return
goffi@necton2
parents:
diff changeset
258
goffi@necton2
parents:
diff changeset
259 # Ensure command is supported
goffi@necton2
parents:
diff changeset
260 if cmd not in self.enabledCommands:
goffi@necton2
parents:
diff changeset
261 # Send a not supported error
goffi@necton2
parents:
diff changeset
262 self.sendErrorReply(REPLY_CMD_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
263 return
goffi@necton2
parents:
diff changeset
264
goffi@necton2
parents:
diff changeset
265 # Process the command
goffi@necton2
parents:
diff changeset
266 if cmd == CMD_CONNECT:
goffi@necton2
parents:
diff changeset
267 self.connectRequested(addr, port)
goffi@necton2
parents:
diff changeset
268 elif cmd == CMD_BIND:
goffi@necton2
parents:
diff changeset
269 self.bindRequested(addr, port)
goffi@necton2
parents:
diff changeset
270 else:
goffi@necton2
parents:
diff changeset
271 # Any other command is not supported
goffi@necton2
parents:
diff changeset
272 self.sendErrorReply(REPLY_CMD_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
273
goffi@necton2
parents:
diff changeset
274 except struct.error, why:
goffi@necton2
parents:
diff changeset
275 return None
goffi@necton2
parents:
diff changeset
276
goffi@necton2
parents:
diff changeset
277 def _makeRequest(self):
goffi@necton2
parents:
diff changeset
278 debug("_makeRequest")
goffi@necton2
parents:
diff changeset
279 self.state = STATE_TARGET_REQUEST
goffi@necton2
parents:
diff changeset
280 sha1 = hashlib.sha1(self.sid + self.initiator_jid + self.target_jid).hexdigest()
goffi@necton2
parents:
diff changeset
281 request = struct.pack('!5B%dsH' % len(sha1), SOCKS5_VER, CMD_CONNECT, 0, ADDR_DOMAINNAME, len(sha1), sha1, 0)
goffi@necton2
parents:
diff changeset
282 self.transport.write(request)
goffi@necton2
parents:
diff changeset
283
goffi@necton2
parents:
diff changeset
284 def _parseRequestReply(self):
goffi@necton2
parents:
diff changeset
285 debug("_parseRequestReply")
goffi@necton2
parents:
diff changeset
286 try:
goffi@necton2
parents:
diff changeset
287 ver, rep, rsvd, self.addressType = struct.unpack('!BBBB', self.buf[:4])
goffi@necton2
parents:
diff changeset
288 # Ensure we actually support the requested address type
goffi@necton2
parents:
diff changeset
289 if self.addressType not in self.supportedAddrs:
goffi@necton2
parents:
diff changeset
290 self.sendErrorReply(REPLY_ADDR_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
291 return
goffi@necton2
parents:
diff changeset
292
goffi@necton2
parents:
diff changeset
293 # Deal with addresses
goffi@necton2
parents:
diff changeset
294 if self.addressType == ADDR_IPV4:
goffi@necton2
parents:
diff changeset
295 addr, port = struct.unpack('!IH', self.buf[4:10])
goffi@necton2
parents:
diff changeset
296 self.buf = self.buf[10:]
goffi@necton2
parents:
diff changeset
297 elif self.addressType == ADDR_DOMAINNAME:
goffi@necton2
parents:
diff changeset
298 nlen = ord(self.buf[4])
goffi@necton2
parents:
diff changeset
299 addr, port = struct.unpack('!%dsH' % nlen, self.buf[5:])
goffi@necton2
parents:
diff changeset
300 self.buf = self.buf[7 + len(addr):]
goffi@necton2
parents:
diff changeset
301 else:
goffi@necton2
parents:
diff changeset
302 # Any other address types are not supported
goffi@necton2
parents:
diff changeset
303 self.sendErrorReply(REPLY_ADDR_NOT_SUPPORTED)
goffi@necton2
parents:
diff changeset
304 return
goffi@necton2
parents:
diff changeset
305
goffi@necton2
parents:
diff changeset
306 # Ensure reply is OK
goffi@necton2
parents:
diff changeset
307 if rep != REPLY_SUCCESS:
goffi@necton2
parents:
diff changeset
308 self.loseConnection()
goffi@necton2
parents:
diff changeset
309 return
goffi@necton2
parents:
diff changeset
310
goffi@necton2
parents:
diff changeset
311 debug("Saving file in %s.", self.data["dest_path"])
goffi@necton2
parents:
diff changeset
312 self.dest_file = open(self.data["dest_path"], 'w')
goffi@necton2
parents:
diff changeset
313 self.state = STATE_TARGET_READY
goffi@necton2
parents:
diff changeset
314 self.activateCB(self.target_jid, self.initiator_jid, self.sid, self.IQ_id)
goffi@necton2
parents:
diff changeset
315
goffi@necton2
parents:
diff changeset
316
goffi@necton2
parents:
diff changeset
317 except struct.error, why:
goffi@necton2
parents:
diff changeset
318 return None
goffi@necton2
parents:
diff changeset
319
goffi@necton2
parents:
diff changeset
320 def connectionMade(self):
goffi@necton2
parents:
diff changeset
321 debug("connectionMade (mode = %s)" % self.mode)
goffi@necton2
parents:
diff changeset
322 self.host.registerProgressCB(self.transfert_id, self.getProgress)
goffi@necton2
parents:
diff changeset
323
goffi@necton2
parents:
diff changeset
324 if self.mode == "target":
goffi@necton2
parents:
diff changeset
325 self.state = STATE_TARGET_INITIAL
goffi@necton2
parents:
diff changeset
326 self._startNegotiation()
goffi@necton2
parents:
diff changeset
327
goffi@necton2
parents:
diff changeset
328 def connectRequested(self, addr, port):
goffi@necton2
parents:
diff changeset
329 debug(("connectRequested"))
goffi@necton2
parents:
diff changeset
330 # Check for special connect to the namespace -- this signifies that the client
goffi@necton2
parents:
diff changeset
331 # is just checking to ensure it can connect to the streamhost
goffi@necton2
parents:
diff changeset
332 if addr == "http://jabber.org/protocol/bytestreams":
goffi@necton2
parents:
diff changeset
333 self.connectCompleted(addr, 0)
goffi@necton2
parents:
diff changeset
334 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
335 return
goffi@necton2
parents:
diff changeset
336
goffi@necton2
parents:
diff changeset
337 # Save addr, for cleanup
goffi@necton2
parents:
diff changeset
338 self.addr = addr
goffi@necton2
parents:
diff changeset
339
goffi@necton2
parents:
diff changeset
340 # Check to see if the requested address is already
goffi@necton2
parents:
diff changeset
341 # activated -- send an error if so
goffi@necton2
parents:
diff changeset
342 if addr in self.activeConns:
goffi@necton2
parents:
diff changeset
343 self.sendErrorReply(socks5.REPLY_CONN_NOT_ALLOWED)
goffi@necton2
parents:
diff changeset
344 return
goffi@necton2
parents:
diff changeset
345
goffi@necton2
parents:
diff changeset
346 # Add this address to the pending connections
goffi@necton2
parents:
diff changeset
347 if self.addConnection(addr, self):
goffi@necton2
parents:
diff changeset
348 self.connectCompleted(addr, 0)
goffi@necton2
parents:
diff changeset
349 self.transport.stopReading()
goffi@necton2
parents:
diff changeset
350 else:
goffi@necton2
parents:
diff changeset
351 self.sendErrorReply(socks5.REPLY_CONN_REFUSED)
goffi@necton2
parents:
diff changeset
352
goffi@necton2
parents:
diff changeset
353 def getProgress(self, data):
goffi@necton2
parents:
diff changeset
354 """Fill data with position of current transfert"""
goffi@necton2
parents:
diff changeset
355 data["size"] = self.filesize
goffi@necton2
parents:
diff changeset
356 try:
goffi@necton2
parents:
diff changeset
357 data["position"] = str(self.dest_file.tell())
goffi@necton2
parents:
diff changeset
358 except (ValueError, AttributeError):
goffi@necton2
parents:
diff changeset
359 data["position"] = ""
goffi@necton2
parents:
diff changeset
360
goffi@necton2
parents:
diff changeset
361 def fileTransfered(self, d):
goffi@necton2
parents:
diff changeset
362 info("File transfer completed, closing connection")
goffi@necton2
parents:
diff changeset
363 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
364
goffi@necton2
parents:
diff changeset
365 def updateTransfered(self, data):
goffi@necton2
parents:
diff changeset
366 self.transfered+=len(data)
goffi@necton2
parents:
diff changeset
367 return data
goffi@necton2
parents:
diff changeset
368
goffi@necton2
parents:
diff changeset
369 def connectCompleted(self, remotehost, remoteport):
goffi@necton2
parents:
diff changeset
370 debug("connectCompleted")
goffi@necton2
parents:
diff changeset
371 if self.addressType == ADDR_IPV4:
goffi@necton2
parents:
diff changeset
372 result = struct.pack('!BBBBIH', SOCKS5_VER, REPLY_SUCCESS, 0, 1, remotehost, remoteport)
goffi@necton2
parents:
diff changeset
373 elif self.addressType == ADDR_DOMAINNAME:
goffi@necton2
parents:
diff changeset
374 result = struct.pack('!BBBBB%dsH' % len(remotehost), SOCKS5_VER, REPLY_SUCCESS, 0,
goffi@necton2
parents:
diff changeset
375 ADDR_DOMAINNAME, len(remotehost), remotehost, remoteport)
goffi@necton2
parents:
diff changeset
376 self.transport.write(result)
goffi@necton2
parents:
diff changeset
377 self.state = STATE_READY
goffi@necton2
parents:
diff changeset
378 self.dest_file=open(self.filepath)
goffi@necton2
parents:
diff changeset
379 d=self.beginFileTransfer(self.dest_file, self.transport, self.updateTransfered)
goffi@necton2
parents:
diff changeset
380 d.addCallback(self.fileTransfered)
goffi@necton2
parents:
diff changeset
381
goffi@necton2
parents:
diff changeset
382 def bindRequested(self, addr, port):
goffi@necton2
parents:
diff changeset
383 pass
goffi@necton2
parents:
diff changeset
384
goffi@necton2
parents:
diff changeset
385 def authenticateUserPass(self, user, passwd):
goffi@necton2
parents:
diff changeset
386 debug("User/pass: %s/%s", user, passwd)
goffi@necton2
parents:
diff changeset
387 return True
goffi@necton2
parents:
diff changeset
388
goffi@necton2
parents:
diff changeset
389 def dataReceived(self, buf):
goffi@necton2
parents:
diff changeset
390 if self.state == STATE_TARGET_READY:
goffi@necton2
parents:
diff changeset
391 self.dest_file.write(buf)
goffi@necton2
parents:
diff changeset
392 self.transfered+=len(buf)
goffi@necton2
parents:
diff changeset
393 return
goffi@necton2
parents:
diff changeset
394
goffi@necton2
parents:
diff changeset
395 self.buf = self.buf + buf
goffi@necton2
parents:
diff changeset
396 if self.state == STATE_INITIAL:
goffi@necton2
parents:
diff changeset
397 self._parseNegotiation()
goffi@necton2
parents:
diff changeset
398 if self.state == STATE_AUTH_USERPASS:
goffi@necton2
parents:
diff changeset
399 self._parseUserPass()
goffi@necton2
parents:
diff changeset
400 if self.state == STATE_REQUEST:
goffi@necton2
parents:
diff changeset
401 self._parseRequest()
goffi@necton2
parents:
diff changeset
402 if self.state == STATE_TARGET_AUTH:
goffi@necton2
parents:
diff changeset
403 ver, method = struct.unpack('!BB', buf)
goffi@necton2
parents:
diff changeset
404 self.buf = self.buf[2:]
goffi@necton2
parents:
diff changeset
405 if ver!=SOCKS5_VER or method!=AUTHMECH_ANON:
goffi@necton2
parents:
diff changeset
406 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
407 else:
goffi@necton2
parents:
diff changeset
408 self._makeRequest()
goffi@necton2
parents:
diff changeset
409 if self.state == STATE_TARGET_REQUEST:
goffi@necton2
parents:
diff changeset
410 self._parseRequestReply()
goffi@necton2
parents:
diff changeset
411
goffi@necton2
parents:
diff changeset
412
goffi@necton2
parents:
diff changeset
413 def clientConnectionLost(self, reason):
goffi@necton2
parents:
diff changeset
414 debug("clientConnectionLost")
goffi@necton2
parents:
diff changeset
415 self.transport.loseConnection()
goffi@necton2
parents:
diff changeset
416
goffi@necton2
parents:
diff changeset
417 def connectionLost(self, reason):
goffi@necton2
parents:
diff changeset
418 debug("connectionLost")
goffi@necton2
parents:
diff changeset
419 self.host.removeProgressCB(self.transfert_id)
goffi@necton2
parents:
diff changeset
420 if self.state == STATE_CONNECT_PENDING:
goffi@necton2
parents:
diff changeset
421 self.removePendingConnection(self.addr, self)
goffi@necton2
parents:
diff changeset
422 else:
goffi@necton2
parents:
diff changeset
423 self.transport.unregisterProducer()
goffi@necton2
parents:
diff changeset
424 if self.peersock != None:
goffi@necton2
parents:
diff changeset
425 self.peersock.peersock = None
goffi@necton2
parents:
diff changeset
426 self.peersock.transport.unregisterProducer()
goffi@necton2
parents:
diff changeset
427 self.peersock = None
goffi@necton2
parents:
diff changeset
428 self.removeActiveConnection(self.addr)
goffi@necton2
parents:
diff changeset
429
goffi@necton2
parents:
diff changeset
430 class Socks5ServerFactory(protocol.ServerFactory):
goffi@necton2
parents:
diff changeset
431 protocol = SOCKSv5
goffi@necton2
parents:
diff changeset
432 protocol.mode = "initiator" #FIXME: Q&D way, fix it
goffi@necton2
parents:
diff changeset
433
goffi@necton2
parents:
diff changeset
434
goffi@necton2
parents:
diff changeset
435 def startedConnecting(self, connector):
goffi@necton2
parents:
diff changeset
436 debug ("Socks 5 server connection started")
goffi@necton2
parents:
diff changeset
437
goffi@necton2
parents:
diff changeset
438 def clientConnectionLost(self, connector, reason):
goffi@necton2
parents:
diff changeset
439 debug ("Socks 5 server connection lost (reason: %s)", reason)
goffi@necton2
parents:
diff changeset
440
goffi@necton2
parents:
diff changeset
441 class Socks5ClientFactory(protocol.ClientFactory):
goffi@necton2
parents:
diff changeset
442 protocol = SOCKSv5
goffi@necton2
parents:
diff changeset
443 protocol.mode = "target" #FIXME: Q&D way, fix it
goffi@necton2
parents:
diff changeset
444
goffi@necton2
parents:
diff changeset
445 def startedConnecting(self, connector):
goffi@necton2
parents:
diff changeset
446 debug ("Socks 5 client connection started")
goffi@necton2
parents:
diff changeset
447
goffi@necton2
parents:
diff changeset
448 def clientConnectionLost(self, connector, reason):
goffi@necton2
parents:
diff changeset
449 debug ("Socks 5 client connection lost (reason: %s)", reason)
goffi@necton2
parents:
diff changeset
450
goffi@necton2
parents:
diff changeset
451
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
452 class XEP_0065(XMPPHandler):
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
453 implements(iwokkel.IDisco)
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
454
19
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
455 params = """
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
456 <params>
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
457 <category name="File Transfert">
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
458 <param name="IP" value='0.0.0.0' default_cb='yes' type="string" />
19
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
459 <param name="Port" value="28915" type="string" />
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
460 </category>
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
461 </params>
f2a745ca0fbc refactoring: using xml params part III (parameters import)
Goffi <goffi@goffi.org>
parents: 15
diff changeset
462 """
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
463
0
goffi@necton2
parents:
diff changeset
464 def __init__(self, host):
goffi@necton2
parents:
diff changeset
465 info("Plugin XEP_0065 initialization")
goffi@necton2
parents:
diff changeset
466 self.host = host
goffi@necton2
parents:
diff changeset
467 debug("registering")
goffi@necton2
parents:
diff changeset
468 self.server_factory = Socks5ServerFactory()
goffi@necton2
parents:
diff changeset
469 self.server_factory.protocol.host = self.host #needed for progress CB
goffi@necton2
parents:
diff changeset
470 self.client_factory = Socks5ClientFactory()
21
633c5ed65701 parameters: new button type (not finished)
Goffi <goffi@goffi.org>
parents: 20
diff changeset
471
633c5ed65701 parameters: new button type (not finished)
Goffi <goffi@goffi.org>
parents: 20
diff changeset
472 #parameters
38
3e24753b9e0b Fixed parameters loading/saving
Goffi <goffi@goffi.org>
parents: 22
diff changeset
473 host.memory.importParams(XEP_0065.params)
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
474 host.memory.setDefault("IP", "File Transfert", self.getExternalIP)
21
633c5ed65701 parameters: new button type (not finished)
Goffi <goffi@goffi.org>
parents: 20
diff changeset
475
22
bb72c29f3432 added action cb mechanism for buttons. Tested with a temporary new user registration button.
Goffi <goffi@goffi.org>
parents: 21
diff changeset
476 port = int(self.host.memory.getParamA("Port", "File Transfert"))
0
goffi@necton2
parents:
diff changeset
477 info("Launching Socks5 Stream server on port %d", port)
goffi@necton2
parents:
diff changeset
478 reactor.listenTCP(port, self.server_factory)
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
479
20
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
480 def getExternalIP(self):
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
481 """Return IP visible from outside, by asking to a website"""
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
482 return getPage("http://www.goffi.org/sat_tools/get_ip.php")
fc8c202cda87 refactoring: using xml params part IV (default values)
Goffi <goffi@goffi.org>
parents: 19
diff changeset
483
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
484 def connectionInitialized(self):
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
485 self.xmlstream.addObserver(BS_REQUEST, self.getFile)
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
486
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
487
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
488 def getDiscoInfo(self, requestor, target, nodeIdentifier=''):
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
489 return [disco.DiscoFeature(NS_BS)]
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
490
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
491 def getDiscoItems(self, requestor, target, nodeIdentifier=''):
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
492 return []
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
493
0
goffi@necton2
parents:
diff changeset
494
goffi@necton2
parents:
diff changeset
495 def setData(self, data, id):
goffi@necton2
parents:
diff changeset
496 self.data = data
goffi@necton2
parents:
diff changeset
497 self.transfert_id = id
goffi@necton2
parents:
diff changeset
498
goffi@necton2
parents:
diff changeset
499 def sendFile(self, id, filepath, size):
goffi@necton2
parents:
diff changeset
500 #lauching socks5 initiator
8
4b05308d45f9 fixed bad namespace
Goffi <goffi@goffi.org>
parents: 0
diff changeset
501 debug("Launching socks5 initiator")
0
goffi@necton2
parents:
diff changeset
502 self.server_factory.protocol.mode = "initiator"
goffi@necton2
parents:
diff changeset
503 self.server_factory.protocol.filepath = filepath
goffi@necton2
parents:
diff changeset
504 self.server_factory.protocol.filesize = size
goffi@necton2
parents:
diff changeset
505 self.server_factory.protocol.transfert_id = id
goffi@necton2
parents:
diff changeset
506
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
507 def getFile(self, iq):
0
goffi@necton2
parents:
diff changeset
508 """Get file using byte stream"""
15
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
509 iq.handled = True
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
510 SI_elem = iq.firstChildElement()
218ec9984fa5 wokkel integration part III + memory saved again
Goffi <goffi@goffi.org>
parents: 9
diff changeset
511 IQ_id = iq['id']
0
goffi@necton2
parents:
diff changeset
512 for element in SI_elem.elements():
goffi@necton2
parents:
diff changeset
513 if element.name == "streamhost":
goffi@necton2
parents:
diff changeset
514 info ("Stream proposed: host=[%s] port=[%s]", element['host'], element['port'])
goffi@necton2
parents:
diff changeset
515 factory = self.client_factory
goffi@necton2
parents:
diff changeset
516 self.server_factory.protocol.mode = "target"
goffi@necton2
parents:
diff changeset
517 factory.protocol.host = self.host #needed for progress CB
goffi@necton2
parents:
diff changeset
518 factory.protocol.data = self.data
goffi@necton2
parents:
diff changeset
519 factory.protocol.transfert_id = self.transfert_id
goffi@necton2
parents:
diff changeset
520 factory.protocol.filesize = self.data["size"]
goffi@necton2
parents:
diff changeset
521 factory.protocol.sid = SI_elem['sid']
goffi@necton2
parents:
diff changeset
522 factory.protocol.initiator_jid = element['jid']
goffi@necton2
parents:
diff changeset
523 factory.protocol.target_jid = self.host.me.full()
goffi@necton2
parents:
diff changeset
524 factory.protocol.IQ_id = IQ_id
goffi@necton2
parents:
diff changeset
525 factory.protocol.activateCB = self.activateStream
goffi@necton2
parents:
diff changeset
526 reactor.connectTCP(element['host'], int(element['port']), factory)
goffi@necton2
parents:
diff changeset
527
goffi@necton2
parents:
diff changeset
528 def activateStream(self, from_jid, to_jid, sid, IQ_id):
goffi@necton2
parents:
diff changeset
529 debug("activating stream")
goffi@necton2
parents:
diff changeset
530 result = domish.Element(('', 'iq'))
goffi@necton2
parents:
diff changeset
531 result['type'] = 'result'
goffi@necton2
parents:
diff changeset
532 result['id'] = IQ_id
goffi@necton2
parents:
diff changeset
533 result['from'] = from_jid
goffi@necton2
parents:
diff changeset
534 result['to'] = to_jid
goffi@necton2
parents:
diff changeset
535 query = result.addElement('query', 'http://jabber.org/protocol/bytestreams')
goffi@necton2
parents:
diff changeset
536 query['sid'] = sid
goffi@necton2
parents:
diff changeset
537 streamhost = query.addElement('streamhost-used')
goffi@necton2
parents:
diff changeset
538 streamhost['jid'] = to_jid #FIXME: use real streamhost
goffi@necton2
parents:
diff changeset
539 self.host.xmlstream.send(result)
goffi@necton2
parents:
diff changeset
540