blob: 45213258d77507ca2105375e61fbf8b5b1bb6783 [file] [log] [blame]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001#
2# XML-RPC CLIENT LIBRARY
3# $Id$
4#
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00005# an XML-RPC client interface for Python.
6#
7# the marshalling and response parser code can also be used to
8# implement XML-RPC servers.
9#
10# Notes:
Martin v. Löwis37af9862004-08-20 07:31:37 +000011# this version is designed to work with Python 2.1 or newer.
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000012#
Fredrik Lundhb9056332001-07-11 17:42:21 +000013# History:
14# 1999-01-14 fl Created
15# 1999-01-15 fl Changed dateTime to use localtime
16# 1999-01-16 fl Added Binary/base64 element, default to RPC2 service
17# 1999-01-19 fl Fixed array data element (from Skip Montanaro)
18# 1999-01-21 fl Fixed dateTime constructor, etc.
19# 1999-02-02 fl Added fault handling, handle empty sequences, etc.
20# 1999-02-10 fl Fixed problem with empty responses (from Skip Montanaro)
21# 1999-06-20 fl Speed improvements, pluggable parsers/transports (0.9.8)
22# 2000-11-28 fl Changed boolean to check the truth value of its argument
23# 2001-02-24 fl Added encoding/Unicode/SafeTransport patches
24# 2001-02-26 fl Added compare support to wrappers (0.9.9/1.0b1)
25# 2001-03-28 fl Make sure response tuple is a singleton
26# 2001-03-29 fl Don't require empty params element (from Nicholas Riley)
Fredrik Lundh78eedce2001-08-23 20:04:33 +000027# 2001-06-10 fl Folded in _xmlrpclib accelerator support (1.0b2)
28# 2001-08-20 fl Base xmlrpclib.Error on built-in Exception (from Paul Prescod)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000029# 2001-09-03 fl Allow Transport subclass to override getparser
30# 2001-09-10 fl Lazy import of urllib, cgi, xmllib (20x import speedup)
Fredrik Lundh1538c232001-10-01 19:42:03 +000031# 2001-10-01 fl Remove containers from memo cache when done with them
32# 2001-10-01 fl Use faster escape method (80% dumps speedup)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000033# 2001-10-02 fl More dumps microtuning
34# 2001-10-04 fl Make sure import expat gets a parser (from Guido van Rossum)
Skip Montanaro5e9c71b2001-10-10 15:56:34 +000035# 2001-10-10 sm Allow long ints to be passed as ints if they don't overflow
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000036# 2001-10-17 sm Test for int and long overflow (allows use on 64-bit systems)
Fredrik Lundhb6ab93f2001-12-19 21:40:04 +000037# 2001-11-12 fl Use repr() to marshal doubles (from Paul Felix)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000038# 2002-03-17 fl Avoid buffered read when possible (from James Rucker)
39# 2002-04-07 fl Added pythondoc comments
40# 2002-04-16 fl Added __str__ methods to datetime/binary wrappers
41# 2002-05-15 fl Added error constants (from Andrew Kuchling)
42# 2002-06-27 fl Merged with Python CVS version
Fredrik Lundh1303c7c2002-10-22 18:23:00 +000043# 2002-10-22 fl Added basic authentication (based on code from Phillip Eby)
Martin v. Löwis541342f2003-07-12 07:53:04 +000044# 2003-01-22 sm Add support for the bool type
45# 2003-02-27 gvr Remove apply calls
46# 2003-04-24 sm Use cStringIO if available
47# 2003-04-25 ak Add support for nil
48# 2003-06-15 gn Add support for time.struct_time
49# 2003-07-12 gp Correct marshalling of Faults
Martin v. Löwis45394c22003-10-31 13:49:36 +000050# 2003-10-31 mvl Add multicall support
Martin v. Löwis37af9862004-08-20 07:31:37 +000051# 2004-08-20 mvl Bump minimum supported Python version to 2.1
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -050052# 2014-12-02 ch/doko Add workaround for gzip bomb vulnerability
Fredrik Lundhb9056332001-07-11 17:42:21 +000053#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000054# Copyright (c) 1999-2002 by Secret Labs AB.
55# Copyright (c) 1999-2002 by Fredrik Lundh.
Fredrik Lundhb9056332001-07-11 17:42:21 +000056#
57# info@pythonware.com
58# http://www.pythonware.com
59#
60# --------------------------------------------------------------------
61# The XML-RPC client interface is
62#
Fredrik Lundh3d9addd2002-06-27 21:36:21 +000063# Copyright (c) 1999-2002 by Secret Labs AB
64# Copyright (c) 1999-2002 by Fredrik Lundh
Fredrik Lundhb9056332001-07-11 17:42:21 +000065#
66# By obtaining, using, and/or copying this software and/or its
67# associated documentation, you agree that you have read, understood,
68# and will comply with the following terms and conditions:
69#
70# Permission to use, copy, modify, and distribute this software and
71# its associated documentation for any purpose and without fee is
72# hereby granted, provided that the above copyright notice appears in
73# all copies, and that both that copyright notice and this permission
74# notice appear in supporting documentation, and that the name of
75# Secret Labs AB or the author not be used in advertising or publicity
76# pertaining to distribution of the software without specific, written
77# prior permission.
78#
79# SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD
80# TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANT-
81# ABILITY AND FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR
82# BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY
83# DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
84# WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
85# ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
86# OF THIS SOFTWARE.
87# --------------------------------------------------------------------
88
Fred Drake1b410792001-09-04 18:55:03 +000089"""
90An XML-RPC client interface for Python.
91
92The marshalling and response parser code can also be used to
93implement XML-RPC servers.
94
Fred Drake1b410792001-09-04 18:55:03 +000095Exported exceptions:
96
Fredrik Lundhc4c062f2001-09-10 19:45:02 +000097 Error Base class for client errors
98 ProtocolError Indicates an HTTP protocol error
99 ResponseError Indicates a broken response package
100 Fault Indicates an XML-RPC fault package
Fred Drake1b410792001-09-04 18:55:03 +0000101
102Exported classes:
103
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000104 ServerProxy Represents a logical connection to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000105
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000106 MultiCall Executor of boxcared xmlrpc requests
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000107 DateTime dateTime wrapper for an ISO 8601 string or time tuple or
108 localtime integer value to generate a "dateTime.iso8601"
109 XML-RPC value
110 Binary binary data wrapper
Fred Drake1b410792001-09-04 18:55:03 +0000111
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000112 Marshaller Generate an XML-RPC params chunk from a Python data structure
113 Unmarshaller Unmarshal an XML-RPC response from incoming XML event message
114 Transport Handles an HTTP transaction to an XML-RPC server
115 SafeTransport Handles an HTTPS transaction to an XML-RPC server
Fred Drake1b410792001-09-04 18:55:03 +0000116
117Exported constants:
118
Florent Xiclunac4fec932011-10-30 20:19:32 +0100119 (none)
Fred Drake1b410792001-09-04 18:55:03 +0000120
121Exported functions:
122
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000123 getparser Create instance of the fastest available parser & attach
124 to an unmarshalling object
125 dumps Convert an argument tuple or a Fault instance to an XML-RPC
126 request (or response, if the methodresponse option is used).
127 loads Convert an XML-RPC packet to unmarshalled data plus a method
128 name (None if not present).
Fred Drake1b410792001-09-04 18:55:03 +0000129"""
130
Florent Xiclunac4fec932011-10-30 20:19:32 +0100131import base64
Florent Xicluna75861df2011-10-30 20:39:24 +0100132import sys
Florent Xiclunac4fec932011-10-30 20:19:32 +0100133import time
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100134from datetime import datetime
Georg Brandl24420152008-05-26 16:32:26 +0000135import http.client
Ezio Melotti91932da2012-02-24 12:44:04 +0200136import urllib.parse
Georg Brandlcef803f2009-06-04 09:04:53 +0000137from xml.parsers import expat
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000138import errno
139from io import BytesIO
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000140try:
141 import gzip
Brett Cannoncd171c82013-07-04 17:43:24 -0400142except ImportError:
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000143 gzip = None #python can be built without zlib/gzip support
Fredrik Lundh1538c232001-10-01 19:42:03 +0000144
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000145# --------------------------------------------------------------------
146# Internal stuff
147
Neal Norwitzff113342007-04-17 08:42:15 +0000148def escape(s):
149 s = s.replace("&", "&")
150 s = s.replace("<", "&lt;")
151 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000152
Florent Xicluna75861df2011-10-30 20:39:24 +0100153# used in User-Agent header sent
154__version__ = sys.version[:3]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000155
156# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000157MAXINT = 2**31-1
158MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000159
160# --------------------------------------------------------------------
161# Error constants (from Dan Libby's specification at
162# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
163
164# Ranges of errors
165PARSE_ERROR = -32700
166SERVER_ERROR = -32600
167APPLICATION_ERROR = -32500
168SYSTEM_ERROR = -32400
169TRANSPORT_ERROR = -32300
170
171# Specific errors
172NOT_WELLFORMED_ERROR = -32700
173UNSUPPORTED_ENCODING = -32701
174INVALID_ENCODING_CHAR = -32702
175INVALID_XMLRPC = -32600
176METHOD_NOT_FOUND = -32601
177INVALID_METHOD_PARAMS = -32602
178INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000179
180# --------------------------------------------------------------------
181# Exceptions
182
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000183##
184# Base class for all kinds of client-side errors.
185
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000186class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000187 """Base class for client errors."""
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000188 def __str__(self):
189 return repr(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000190
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000191##
192# Indicates an HTTP-level protocol error. This is raised by the HTTP
193# transport layer, if the server returns an error code other than 200
194# (OK).
195#
196# @param url The target URL.
197# @param errcode The HTTP error code.
198# @param errmsg The HTTP error message.
199# @param headers The HTTP header dictionary.
200
Fredrik Lundhb9056332001-07-11 17:42:21 +0000201class ProtocolError(Error):
Fred Drake1b410792001-09-04 18:55:03 +0000202 """Indicates an HTTP protocol error."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000203 def __init__(self, url, errcode, errmsg, headers):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000204 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000205 self.url = url
206 self.errcode = errcode
207 self.errmsg = errmsg
208 self.headers = headers
209 def __repr__(self):
210 return (
211 "<ProtocolError for %s: %s %s>" %
212 (self.url, self.errcode, self.errmsg)
213 )
214
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000215##
216# Indicates a broken XML-RPC response package. This exception is
217# raised by the unmarshalling layer, if the XML-RPC response is
218# malformed.
219
Fredrik Lundhb9056332001-07-11 17:42:21 +0000220class ResponseError(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000221 """Indicates a broken response package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000222 pass
223
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000224##
225# Indicates an XML-RPC fault response package. This exception is
226# raised by the unmarshalling layer, if the XML-RPC response contains
Florent Xiclunac4fec932011-10-30 20:19:32 +0100227# a fault string. This exception can also be used as a class, to
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000228# generate a fault XML-RPC message.
229#
230# @param faultCode The XML-RPC fault code.
231# @param faultString The XML-RPC fault string.
232
Fredrik Lundhb9056332001-07-11 17:42:21 +0000233class Fault(Error):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000234 """Indicates an XML-RPC fault package."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000235 def __init__(self, faultCode, faultString, **extra):
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000236 Error.__init__(self)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000237 self.faultCode = faultCode
238 self.faultString = faultString
239 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100240 return "<Fault %s: %r>" % (self.faultCode, self.faultString)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000241
242# --------------------------------------------------------------------
243# Special values
244
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000245##
Georg Brandl38eceaa2008-05-26 11:14:17 +0000246# Backwards compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000247
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000248boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000249
250##
251# Wrapper for XML-RPC DateTime values. This converts a time value to
252# the format used by XML-RPC.
253# <p>
Florent Xicluna2738a642011-11-01 00:06:58 +0100254# The value can be given as a datetime object, as a string in the
255# format "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000256# time.localtime()), or an integer value (as returned by time.time()).
257# The wrapper uses time.localtime() to convert an integer to a time
258# tuple.
259#
Florent Xicluna2738a642011-11-01 00:06:58 +0100260# @param value The time, given as a datetime object, an ISO 8601 string,
261# a time tuple, or an integer time value.
262
263
264# Issue #13305: different format codes across platforms
265_day0 = datetime(1, 1, 1)
266if _day0.strftime('%Y') == '0001': # Mac OS X
267 def _iso8601_format(value):
268 return value.strftime("%Y%m%dT%H:%M:%S")
269elif _day0.strftime('%4Y') == '0001': # Linux
270 def _iso8601_format(value):
271 return value.strftime("%4Y%m%dT%H:%M:%S")
272else:
273 def _iso8601_format(value):
274 return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
275del _day0
276
Fredrik Lundhb9056332001-07-11 17:42:21 +0000277
Christian Heimesdae2a892008-04-19 00:55:37 +0000278def _strftime(value):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100279 if isinstance(value, datetime):
Florent Xicluna2738a642011-11-01 00:06:58 +0100280 return _iso8601_format(value)
Christian Heimesdae2a892008-04-19 00:55:37 +0000281
282 if not isinstance(value, (tuple, time.struct_time)):
283 if value == 0:
284 value = time.time()
285 value = time.localtime(value)
286
287 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
288
Fredrik Lundhb9056332001-07-11 17:42:21 +0000289class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000290 """DateTime wrapper for an ISO 8601 string or time tuple or
291 localtime integer value to generate 'dateTime.iso8601' XML-RPC
292 value.
293 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000294
295 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000296 if isinstance(value, str):
297 self.value = value
298 else:
299 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000300
Christian Heimes05e8be12008-02-23 18:30:17 +0000301 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000302 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000303 s = self.value
304 o = other.value
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100305 elif isinstance(other, datetime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000306 s = self.value
Florent Xicluna2738a642011-11-01 00:06:58 +0100307 o = _iso8601_format(other)
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100308 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000309 s = self.value
310 o = other
311 elif hasattr(other, "timetuple"):
312 s = self.timetuple()
313 o = other.timetuple()
314 else:
315 otype = (hasattr(other, "__class__")
316 and other.__class__.__name__
317 or type(other))
318 raise TypeError("Can't compare %s and %s" %
319 (self.__class__.__name__, otype))
320 return s, o
321
322 def __lt__(self, other):
323 s, o = self.make_comparable(other)
324 return s < o
325
326 def __le__(self, other):
327 s, o = self.make_comparable(other)
328 return s <= o
329
330 def __gt__(self, other):
331 s, o = self.make_comparable(other)
332 return s > o
333
334 def __ge__(self, other):
335 s, o = self.make_comparable(other)
336 return s >= o
337
338 def __eq__(self, other):
339 s, o = self.make_comparable(other)
340 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000341
342 def __ne__(self, other):
Christian Heimes05e8be12008-02-23 18:30:17 +0000343 s, o = self.make_comparable(other)
344 return s != o
345
346 def timetuple(self):
347 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
348
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000349 ##
350 # Get date/time value.
351 #
352 # @return Date/time value, as an ISO 8601 string.
353
354 def __str__(self):
355 return self.value
356
Fredrik Lundhb9056332001-07-11 17:42:21 +0000357 def __repr__(self):
Florent Xiclunac4fec932011-10-30 20:19:32 +0100358 return "<DateTime %r at %x>" % (self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000359
360 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000361 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000362
363 def encode(self, out):
364 out.write("<value><dateTime.iso8601>")
365 out.write(self.value)
366 out.write("</dateTime.iso8601></value>\n")
367
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000368def _datetime(data):
369 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000370 value = DateTime()
371 value.decode(data)
372 return value
373
Skip Montanaro174dd222005-05-14 20:54:16 +0000374def _datetime_type(data):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100375 return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
Skip Montanaro174dd222005-05-14 20:54:16 +0000376
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000377##
378# Wrapper for binary data. This can be used to transport any kind
379# of binary data over XML-RPC, using BASE64 encoding.
380#
381# @param data An 8-bit string containing arbitrary data.
382
Fredrik Lundhb9056332001-07-11 17:42:21 +0000383class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000384 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000385
386 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000387 if data is None:
388 data = b""
389 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100390 if not isinstance(data, (bytes, bytearray)):
391 raise TypeError("expected bytes or bytearray, not %s" %
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000392 data.__class__.__name__)
393 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000394 self.data = data
395
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000396 ##
397 # Get buffer contents.
398 #
399 # @return Buffer contents, as an 8-bit string.
400
401 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000402 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000403
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000404 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000405 if isinstance(other, Binary):
406 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000407 return self.data == other
408
409 def __ne__(self, other):
410 if isinstance(other, Binary):
411 other = other.data
412 return self.data != other
Fredrik Lundhb9056332001-07-11 17:42:21 +0000413
414 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000415 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000416
417 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000418 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000419 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000420 out.write(encoded.decode('ascii'))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000421 out.write("</base64></value>\n")
422
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000423def _binary(data):
424 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000425 value = Binary()
426 value.decode(data)
427 return value
428
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000429WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000430
431# --------------------------------------------------------------------
432# XML parsers
433
Georg Brandlcef803f2009-06-04 09:04:53 +0000434class ExpatParser:
435 # fast expat parser for Python 2.0 and later.
436 def __init__(self, target):
437 self._parser = parser = expat.ParserCreate(None, None)
438 self._target = target
439 parser.StartElementHandler = target.start
440 parser.EndElementHandler = target.end
441 parser.CharacterDataHandler = target.data
442 encoding = None
443 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000444
Georg Brandlcef803f2009-06-04 09:04:53 +0000445 def feed(self, data):
446 self._parser.Parse(data, 0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000447
Georg Brandlcef803f2009-06-04 09:04:53 +0000448 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300449 try:
450 parser = self._parser
451 except AttributeError:
452 pass
453 else:
454 del self._target, self._parser # get rid of circular references
455 parser.Parse(b"", True) # end of data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000456
Fredrik Lundhb9056332001-07-11 17:42:21 +0000457# --------------------------------------------------------------------
458# XML-RPC marshalling and unmarshalling code
459
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000460##
461# XML-RPC marshaller.
462#
463# @param encoding Default encoding for 8-bit strings. The default
464# value is None (interpreted as UTF-8).
465# @see dumps
466
Fredrik Lundhb9056332001-07-11 17:42:21 +0000467class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000468 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000469
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000470 Create a Marshaller instance for each set of parameters, and use
471 the "dumps" method to convert your data (represented as a tuple)
472 to an XML-RPC params chunk. To write a fault response, pass a
473 Fault instance instead. You may prefer to use the "dumps" module
474 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000475 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000476
477 # by the way, if you don't understand what's going on in here,
478 # that's perfectly ok.
479
Georg Brandlfe991052009-09-16 15:54:04 +0000480 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000481 self.memo = {}
482 self.data = None
483 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000484 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000485
Fredrik Lundhb9056332001-07-11 17:42:21 +0000486 dispatch = {}
487
488 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000489 out = []
490 write = out.append
491 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000492 if isinstance(values, Fault):
493 # fault instance
494 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000495 dump({'faultCode': values.faultCode,
496 'faultString': values.faultString},
497 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000498 write("</fault>\n")
499 else:
500 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000501 # FIXME: the xml-rpc specification allows us to leave out
502 # the entire <params> block if there are no parameters.
503 # however, changing this may break older code (including
504 # old versions of xmlrpclib.py), so this is better left as
505 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000506 write("<params>\n")
507 for v in values:
508 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000509 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000510 write("</param>\n")
511 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000512 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000513 return result
514
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000515 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000516 try:
517 f = self.dispatch[type(value)]
518 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000519 # check if this object can be marshalled as a structure
Florent Xicluna93dfee12011-10-30 20:22:25 +0100520 if not hasattr(value, '__dict__'):
Collin Winterce36ad82007-08-30 01:19:48 +0000521 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000522 # check if this class is a sub-class of a basic type,
523 # because we don't know how to marshal these types
524 # (e.g. a string sub-class)
525 for type_ in type(value).__mro__:
526 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000527 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000528 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
529 # for the p3yk merge, this should probably be fixed more neatly.
530 f = self.dispatch["_arbitrary_instance"]
531 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000532
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000533 def dump_nil (self, value, write):
534 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000535 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000536 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000537 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000538
Georg Brandl38eceaa2008-05-26 11:14:17 +0000539 def dump_bool(self, value, write):
540 write("<value><boolean>")
541 write(value and "1" or "0")
542 write("</boolean></value>\n")
543 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000544
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000545 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000546 if value > MAXINT or value < MININT:
Serhiy Storchaka95949422013-08-27 19:40:23 +0300547 raise OverflowError("int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000548 write("<value><int>")
549 write(str(int(value)))
550 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000551 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000552
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200553 # backward compatible
554 dump_int = dump_long
555
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000556 def dump_double(self, value, write):
557 write("<value><double>")
558 write(repr(value))
559 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000560 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000561
Guido van Rossum54ad5232007-05-27 09:17:48 +0000562 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000563 write("<value><string>")
564 write(escape(value))
565 write("</string></value>\n")
566 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000567
Florent Xicluna61665192011-11-15 20:53:25 +0100568 def dump_bytes(self, value, write):
569 write("<value><base64>\n")
570 encoded = base64.encodebytes(value)
571 write(encoded.decode('ascii'))
572 write("</base64></value>\n")
573 dispatch[bytes] = dump_bytes
574 dispatch[bytearray] = dump_bytes
575
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000576 def dump_array(self, value, write):
577 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000578 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000579 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000580 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000581 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000582 write("<value><array><data>\n")
583 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000584 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000585 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000586 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000587 dispatch[tuple] = dump_array
588 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000589
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000590 def dump_struct(self, value, write, escape=escape):
591 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000592 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000593 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000594 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000595 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000596 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000597 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000598 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000599 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000600 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000601 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000602 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000603 write("</member>\n")
604 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000605 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000606 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000607
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100608 def dump_datetime(self, value, write):
609 write("<value><dateTime.iso8601>")
610 write(_strftime(value))
611 write("</dateTime.iso8601></value>\n")
612 dispatch[datetime] = dump_datetime
Fred Drakeba613c32005-02-10 18:33:30 +0000613
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000614 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000615 # check for special wrappers
616 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000617 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000618 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000619 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000620 else:
621 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000622 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000623 dispatch[DateTime] = dump_instance
624 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000625 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
626 # for the p3yk merge, this should probably be fixed more neatly.
627 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000628
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000629##
630# XML-RPC unmarshaller.
631#
632# @see loads
633
Fredrik Lundhb9056332001-07-11 17:42:21 +0000634class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000635 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000636 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000637 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000638
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000639 Note that this reader is fairly tolerant, and gladly accepts bogus
640 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000641 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000642
643 # and again, if you don't understand what's going on in here,
644 # that's perfectly ok.
645
Florent Xicluna61665192011-11-15 20:53:25 +0100646 def __init__(self, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000647 self._type = None
648 self._stack = []
649 self._marks = []
650 self._data = []
651 self._methodname = None
652 self._encoding = "utf-8"
653 self.append = self._stack.append
Florent Xicluna61665192011-11-15 20:53:25 +0100654 self._use_datetime = use_builtin_types or use_datetime
655 self._use_bytes = use_builtin_types
Fredrik Lundhb9056332001-07-11 17:42:21 +0000656
657 def close(self):
658 # return response tuple and target method
659 if self._type is None or self._marks:
660 raise ResponseError()
661 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000662 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000663 return tuple(self._stack)
664
665 def getmethodname(self):
666 return self._methodname
667
668 #
669 # event handlers
670
671 def xml(self, encoding, standalone):
672 self._encoding = encoding
673 # FIXME: assert standalone == 1 ???
674
675 def start(self, tag, attrs):
676 # prepare to handle this element
677 if tag == "array" or tag == "struct":
678 self._marks.append(len(self._stack))
679 self._data = []
680 self._value = (tag == "value")
681
682 def data(self, text):
683 self._data.append(text)
684
Neal Norwitzff113342007-04-17 08:42:15 +0000685 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000686 # call the appropriate end tag handler
687 try:
688 f = self.dispatch[tag]
689 except KeyError:
690 pass # unknown tag ?
691 else:
Neal Norwitzff113342007-04-17 08:42:15 +0000692 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000693
694 #
695 # accelerator support
696
697 def end_dispatch(self, tag, data):
698 # dispatch data
699 try:
700 f = self.dispatch[tag]
701 except KeyError:
702 pass # unknown tag ?
703 else:
704 return f(self, data)
705
706 #
707 # element decoders
708
709 dispatch = {}
710
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000711 def end_nil (self, data):
712 self.append(None)
713 self._value = 0
714 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000715
Fredrik Lundh1538c232001-10-01 19:42:03 +0000716 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000717 if data == "0":
718 self.append(False)
719 elif data == "1":
720 self.append(True)
721 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000722 raise TypeError("bad boolean value")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000723 self._value = 0
724 dispatch["boolean"] = end_boolean
725
Fredrik Lundh1538c232001-10-01 19:42:03 +0000726 def end_int(self, data):
727 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000728 self._value = 0
729 dispatch["i4"] = end_int
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000730 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000731 dispatch["int"] = end_int
732
Fredrik Lundh1538c232001-10-01 19:42:03 +0000733 def end_double(self, data):
734 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000735 self._value = 0
736 dispatch["double"] = end_double
737
Fredrik Lundh1538c232001-10-01 19:42:03 +0000738 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000739 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000740 data = data.decode(self._encoding)
741 self.append(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000742 self._value = 0
743 dispatch["string"] = end_string
744 dispatch["name"] = end_string # struct keys are always strings
745
746 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000747 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000748 # map arrays to Python lists
749 self._stack[mark:] = [self._stack[mark:]]
750 self._value = 0
751 dispatch["array"] = end_array
752
753 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000754 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000755 # map structs to Python dictionaries
756 dict = {}
757 items = self._stack[mark:]
758 for i in range(0, len(items), 2):
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000759 dict[items[i]] = items[i+1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000760 self._stack[mark:] = [dict]
761 self._value = 0
762 dispatch["struct"] = end_struct
763
Fredrik Lundh1538c232001-10-01 19:42:03 +0000764 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000765 value = Binary()
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000766 value.decode(data.encode("ascii"))
Florent Xicluna61665192011-11-15 20:53:25 +0100767 if self._use_bytes:
768 value = value.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769 self.append(value)
770 self._value = 0
771 dispatch["base64"] = end_base64
772
Fredrik Lundh1538c232001-10-01 19:42:03 +0000773 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000774 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000775 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000776 if self._use_datetime:
777 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000778 self.append(value)
779 dispatch["dateTime.iso8601"] = end_dateTime
780
781 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000782 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000783 # elements, treat it as a string element
784 if self._value:
785 self.end_string(data)
786 dispatch["value"] = end_value
787
788 def end_params(self, data):
789 self._type = "params"
790 dispatch["params"] = end_params
791
792 def end_fault(self, data):
793 self._type = "fault"
794 dispatch["fault"] = end_fault
795
Fredrik Lundh1538c232001-10-01 19:42:03 +0000796 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000797 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000798 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000799 self._methodname = data
800 self._type = "methodName" # no params
801 dispatch["methodName"] = end_methodName
802
Martin v. Löwis45394c22003-10-31 13:49:36 +0000803## Multicall support
804#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000805
Martin v. Löwis45394c22003-10-31 13:49:36 +0000806class _MultiCallMethod:
807 # some lesser magic to store calls made to a MultiCall object
808 # for batch execution
809 def __init__(self, call_list, name):
810 self.__call_list = call_list
811 self.__name = name
812 def __getattr__(self, name):
813 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
814 def __call__(self, *args):
815 self.__call_list.append((self.__name, args))
816
Martin v. Löwis12237b32004-08-22 16:04:50 +0000817class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000818 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov737fb892012-12-18 21:14:22 +0200819 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000820
Martin v. Löwis12237b32004-08-22 16:04:50 +0000821 def __init__(self, results):
822 self.results = results
823
824 def __getitem__(self, i):
825 item = self.results[i]
826 if type(item) == type({}):
827 raise Fault(item['faultCode'], item['faultString'])
828 elif type(item) == type([]):
829 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000830 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000831 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000832
Martin v. Löwis45394c22003-10-31 13:49:36 +0000833class MultiCall:
834 """server -> a object used to boxcar method calls
835
836 server should be a ServerProxy object.
837
838 Methods can be added to the MultiCall using normal
839 method call syntax e.g.:
840
841 multicall = MultiCall(server_proxy)
842 multicall.add(2,3)
843 multicall.get_address("Guido")
844
845 To execute the multicall, call the MultiCall object e.g.:
846
847 add_result, address = multicall()
848 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000849
Martin v. Löwis45394c22003-10-31 13:49:36 +0000850 def __init__(self, server):
851 self.__server = server
852 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000853
Martin v. Löwis45394c22003-10-31 13:49:36 +0000854 def __repr__(self):
855 return "<MultiCall at %x>" % id(self)
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000856
Martin v. Löwis45394c22003-10-31 13:49:36 +0000857 __str__ = __repr__
858
859 def __getattr__(self, name):
860 return _MultiCallMethod(self.__call_list, name)
861
862 def __call__(self):
863 marshalled_list = []
864 for name, args in self.__call_list:
865 marshalled_list.append({'methodName' : name, 'params' : args})
866
867 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000868
Fredrik Lundhb9056332001-07-11 17:42:21 +0000869# --------------------------------------------------------------------
870# convenience functions
871
Georg Brandl38eceaa2008-05-26 11:14:17 +0000872FastMarshaller = FastParser = FastUnmarshaller = None
873
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000874##
875# Create a parser object, and connect it to an unmarshalling instance.
876# This function picks the fastest available XML parser.
877#
878# return A (parser, unmarshaller) tuple.
879
Florent Xicluna61665192011-11-15 20:53:25 +0100880def getparser(use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000881 """getparser() -> parser, unmarshaller
882
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000883 Create an instance of the fastest available parser, and attach it
884 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000885 """
886 if FastParser and FastUnmarshaller:
Florent Xicluna61665192011-11-15 20:53:25 +0100887 if use_builtin_types:
Skip Montanaro174dd222005-05-14 20:54:16 +0000888 mkdatetime = _datetime_type
Florent Xicluna61665192011-11-15 20:53:25 +0100889 mkbytes = base64.decodebytes
890 elif use_datetime:
891 mkdatetime = _datetime_type
892 mkbytes = _binary
Skip Montanaro174dd222005-05-14 20:54:16 +0000893 else:
894 mkdatetime = _datetime
Florent Xicluna61665192011-11-15 20:53:25 +0100895 mkbytes = _binary
896 target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000897 parser = FastParser(target)
898 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100899 target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000900 if FastParser:
901 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000902 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000903 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000904 return parser, target
905
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000906##
907# Convert a Python tuple or a Fault instance to an XML-RPC packet.
908#
909# @def dumps(params, **options)
910# @param params A tuple or Fault instance.
911# @keyparam methodname If given, create a methodCall request for
912# this method name.
913# @keyparam methodresponse If given, create a methodResponse packet.
914# If used with a tuple, the tuple must be a singleton (that is,
915# it must contain exactly one element).
916# @keyparam encoding The packet encoding.
917# @return A string containing marshalled data.
918
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000919def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000920 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000921 """data [,options] -> marshalled data
922
923 Convert an argument tuple or a Fault instance to an XML-RPC
924 request (or response, if the methodresponse option is used).
925
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000926 In addition to the data object, the following options can be given
927 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000928
929 methodname: the method name for a methodCall packet
930
931 methodresponse: true to create a methodResponse packet.
932 If this option is used with a tuple, the tuple must be
933 a singleton (i.e. it can contain only one element).
934
935 encoding: the packet encoding (default is UTF-8)
936
Florent Xicluna61665192011-11-15 20:53:25 +0100937 All byte strings in the data structure are assumed to use the
Fredrik Lundhb9056332001-07-11 17:42:21 +0000938 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000939 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000940 """
941
Guido van Rossum13257902007-06-07 23:15:56 +0000942 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000943 if isinstance(params, Fault):
944 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000945 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000946 assert len(params) == 1, "response tuple must be a singleton"
947
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000948 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000949 encoding = "utf-8"
950
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000951 if FastMarshaller:
952 m = FastMarshaller(encoding)
953 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000954 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000955
Fredrik Lundhb9056332001-07-11 17:42:21 +0000956 data = m.dumps(params)
957
958 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000959 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000960 else:
961 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
962
963 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000964 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000965 # a method call
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000966 if not isinstance(methodname, str):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000967 methodname = methodname.encode(encoding)
968 data = (
969 xmlheader,
970 "<methodCall>\n"
971 "<methodName>", methodname, "</methodName>\n",
972 data,
973 "</methodCall>\n"
974 )
975 elif methodresponse:
976 # a method response, or a fault structure
977 data = (
978 xmlheader,
979 "<methodResponse>\n",
980 data,
981 "</methodResponse>\n"
982 )
983 else:
984 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +0000985 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000986
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000987##
988# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
989# represents a fault condition, this function raises a Fault exception.
990#
991# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000992# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000993# (None if not present).
994# @see Fault
995
Florent Xicluna61665192011-11-15 20:53:25 +0100996def loads(data, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000997 """data -> unmarshalled data, method name
998
999 Convert an XML-RPC packet to unmarshalled data plus a method
1000 name (None if not present).
1001
1002 If the XML-RPC packet represents a fault condition, this function
1003 raises a Fault exception.
1004 """
Florent Xicluna61665192011-11-15 20:53:25 +01001005 p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001006 p.feed(data)
1007 p.close()
1008 return u.close(), u.getmethodname()
1009
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001010##
1011# Encode a string using the gzip content encoding such as specified by the
1012# Content-Encoding: gzip
1013# in the HTTP header, as described in RFC 1952
1014#
1015# @param data the unencoded data
1016# @return the encoded data
1017
1018def gzip_encode(data):
1019 """data -> gzip encoded data
1020
1021 Encode data using the gzip content encoding as described in RFC 1952
1022 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001023 if not gzip:
1024 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001025 f = BytesIO()
1026 gzf = gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1)
1027 gzf.write(data)
1028 gzf.close()
1029 encoded = f.getvalue()
1030 f.close()
1031 return encoded
1032
1033##
1034# Decode a string using the gzip content encoding such as specified by the
1035# Content-Encoding: gzip
1036# in the HTTP header, as described in RFC 1952
1037#
1038# @param data The encoded data
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001039# @keyparam max_decode Maximum bytes to decode (20MB default), use negative
1040# values for unlimited decoding
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001041# @return the unencoded data
1042# @raises ValueError if data is not correctly coded.
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001043# @raises ValueError if max gzipped payload length exceeded
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001044
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001045def gzip_decode(data, max_decode=20971520):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001046 """gzip encoded data -> unencoded data
1047
1048 Decode data using the gzip content encoding as described in RFC 1952
1049 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001050 if not gzip:
1051 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001052 f = BytesIO(data)
1053 gzf = gzip.GzipFile(mode="rb", fileobj=f)
1054 try:
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001055 if max_decode < 0: # no limit
1056 decoded = gzf.read()
1057 else:
1058 decoded = gzf.read(max_decode + 1)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001059 except OSError:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001060 raise ValueError("invalid data")
1061 f.close()
1062 gzf.close()
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001063 if max_decode >= 0 and len(decoded) > max_decode:
1064 raise ValueError("max gzipped payload length exceeded")
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001065 return decoded
1066
1067##
1068# Return a decoded file-like object for the gzip encoding
1069# as described in RFC 1952.
1070#
1071# @param response A stream supporting a read() method
1072# @return a file-like object that the decoded data can be read() from
1073
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001074class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001075 """a file-like object to decode a response encoded with the gzip
1076 method, as described in RFC 1952.
1077 """
1078 def __init__(self, response):
1079 #response doesn't support tell() and read(), required by
1080 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001081 if not gzip:
1082 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001083 self.io = BytesIO(response.read())
1084 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1085
1086 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001087 try:
1088 gzip.GzipFile.close(self)
1089 finally:
1090 self.io.close()
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001091
Fredrik Lundhb9056332001-07-11 17:42:21 +00001092
1093# --------------------------------------------------------------------
1094# request dispatcher
1095
1096class _Method:
1097 # some magic to bind an XML-RPC method to an RPC server.
1098 # supports "nested" methods (e.g. examples.getStateName)
1099 def __init__(self, send, name):
1100 self.__send = send
1101 self.__name = name
1102 def __getattr__(self, name):
1103 return _Method(self.__send, "%s.%s" % (self.__name, name))
1104 def __call__(self, *args):
1105 return self.__send(self.__name, args)
1106
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001107##
1108# Standard transport class for XML-RPC over HTTP.
1109# <p>
1110# You can create custom transports by subclassing this method, and
1111# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001112
1113class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001114 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001115
1116 # client identifier (may be overridden)
Florent Xicluna75861df2011-10-30 20:39:24 +01001117 user_agent = "Python-xmlrpc/%s" % __version__
Fredrik Lundhb9056332001-07-11 17:42:21 +00001118
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001119 #if true, we'll request gzip encoding
1120 accept_gzip_encoding = True
1121
1122 # if positive, encode request using gzip if it exceeds this threshold
1123 # note that many server will get confused, so only use it if you know
1124 # that they can decode such a request
1125 encode_threshold = None #None = don't encode
1126
Florent Xicluna61665192011-11-15 20:53:25 +01001127 def __init__(self, use_datetime=False, use_builtin_types=False):
Skip Montanaro174dd222005-05-14 20:54:16 +00001128 self._use_datetime = use_datetime
Florent Xicluna61665192011-11-15 20:53:25 +01001129 self._use_builtin_types = use_builtin_types
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001130 self._connection = (None, None)
1131 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001132
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001133 ##
1134 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001135 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001136 #
1137 # @param host Target host.
1138 # @param handler Target PRC handler.
1139 # @param request_body XML-RPC request body.
1140 # @param verbose Debugging flag.
1141 # @return Parsed response.
1142
Georg Brandlfe991052009-09-16 15:54:04 +00001143 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001144 #retry request once if cached connection has gone cold
1145 for i in (0, 1):
1146 try:
1147 return self.single_request(host, handler, request_body, verbose)
Andrew Svetlov0832af62012-12-18 23:10:48 +02001148 except OSError as e:
1149 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
1150 errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001151 raise
1152 except http.client.BadStatusLine: #close after we sent request
1153 if i:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001154 raise
1155
Georg Brandlfe991052009-09-16 15:54:04 +00001156 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001157 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001158 try:
1159 http_conn = self.send_request(host, handler, request_body, verbose)
1160 resp = http_conn.getresponse()
1161 if resp.status == 200:
1162 self.verbose = verbose
1163 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001164
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001165 except Fault:
1166 raise
1167 except Exception:
1168 #All unexpected errors leave connection in
1169 # a strange state, so we clear it.
1170 self.close()
1171 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001172
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001173 #We got an error response.
1174 #Discard any response data and raise exception
1175 if resp.getheader("content-length", ""):
1176 resp.read()
1177 raise ProtocolError(
1178 host + handler,
1179 resp.status, resp.reason,
1180 dict(resp.getheaders())
1181 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001182
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001183
1184 ##
1185 # Create parser.
1186 #
1187 # @return A 2-tuple containing a parser and a unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001188
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001189 def getparser(self):
1190 # get parser and unmarshaller
Florent Xicluna61665192011-11-15 20:53:25 +01001191 return getparser(use_datetime=self._use_datetime,
1192 use_builtin_types=self._use_builtin_types)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001193
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001194 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001195 # Get authorization info from host parameter
1196 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1197 # it is checked for a "user:pw@host" format, and a "Basic
1198 # Authentication" header is added if appropriate.
1199 #
1200 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1201 # @return A 3-tuple containing (actual host, extra headers,
1202 # x509 info). The header and x509 fields may be None.
1203
1204 def get_host_info(self, host):
1205
1206 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001207 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001208 host, x509 = host
1209
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001210 auth, host = urllib.parse.splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001211
1212 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001213 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001214 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001215 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001216 extra_headers = [
1217 ("Authorization", "Basic " + auth)
1218 ]
1219 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001220 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001221
1222 return host, extra_headers, x509
1223
1224 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001225 # Connect to server.
1226 #
1227 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001228 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001229
Fredrik Lundhb9056332001-07-11 17:42:21 +00001230 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001231 #return an existing connection if possible. This allows
1232 #HTTP/1.1 keep-alive.
1233 if self._connection and host == self._connection[0]:
1234 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001235 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001236 chost, self._extra_headers, x509 = self.get_host_info(host)
1237 self._connection = host, http.client.HTTPConnection(chost)
1238 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001239
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001240 ##
1241 # Clear any cached connection object.
1242 # Used in the event of socket errors.
1243 #
1244 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001245 host, connection = self._connection
1246 if connection:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001247 self._connection = (None, None)
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001248 connection.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001249
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001250 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001251 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001252 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001253 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1254 # @param handler Targer RPC handler (a path relative to host)
1255 # @param request_body The XML-RPC request body
1256 # @param debug Enable debugging if debug is true.
1257 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001258
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001259 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001260 connection = self.make_connection(host)
1261 headers = self._extra_headers[:]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001262 if debug:
1263 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001264 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001265 connection.putrequest("POST", handler, skip_accept_encoding=True)
1266 headers.append(("Accept-Encoding", "gzip"))
1267 else:
1268 connection.putrequest("POST", handler)
1269 headers.append(("Content-Type", "text/xml"))
1270 headers.append(("User-Agent", self.user_agent))
1271 self.send_headers(connection, headers)
1272 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001273 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001274
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001275 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001276 # Send request headers.
1277 # This function provides a useful hook for subclassing
1278 #
1279 # @param connection httpConnection.
1280 # @param headers list of key,value pairs for HTTP headers
1281
1282 def send_headers(self, connection, headers):
1283 for key, val in headers:
1284 connection.putheader(key, val)
1285
1286 ##
1287 # Send request body.
1288 # This function provides a useful hook for subclassing
1289 #
1290 # @param connection httpConnection.
1291 # @param request_body XML-RPC request body.
1292
1293 def send_content(self, connection, request_body):
1294 #optionally encode the request
1295 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001296 self.encode_threshold < len(request_body) and
1297 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001298 connection.putheader("Content-Encoding", "gzip")
1299 request_body = gzip_encode(request_body)
1300
1301 connection.putheader("Content-Length", str(len(request_body)))
1302 connection.endheaders(request_body)
1303
1304 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001305 # Parse response.
1306 #
1307 # @param file Stream.
1308 # @return Response tuple and target method.
1309
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001310 def parse_response(self, response):
1311 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001312 # Check for new http response object, otherwise it is a file object.
1313 if hasattr(response, 'getheader'):
1314 if response.getheader("Content-Encoding", "") == "gzip":
1315 stream = GzipDecodedResponse(response)
1316 else:
1317 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001318 else:
1319 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001320
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001321 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001322
1323 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001324 data = stream.read(1024)
1325 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001326 break
1327 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001328 print("body:", repr(data))
1329 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001330
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001331 if stream is not response:
1332 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001333 p.close()
1334
1335 return u.close()
1336
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001337##
1338# Standard transport class for XML-RPC over HTTPS.
1339
Fredrik Lundhb9056332001-07-11 17:42:21 +00001340class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001341 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001342
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001343 def __init__(self, use_datetime=False, use_builtin_types=False, *,
1344 context=None):
1345 super().__init__(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
1346 self.context = context
1347
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001348 # FIXME: mostly untested
1349
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001350 def make_connection(self, host):
1351 if self._connection and host == self._connection[0]:
1352 return self._connection[1]
1353
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001354 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001355 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001356 "your version of http.client doesn't support HTTPS")
1357 # create a HTTPS connection object from a host descriptor
1358 # host may be a string, or a (host, x509-dict) tuple
1359 chost, self._extra_headers, x509 = self.get_host_info(host)
1360 self._connection = host, http.client.HTTPSConnection(chost,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001361 None, context=self.context, **(x509 or {}))
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001362 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001363
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001364##
1365# Standard server proxy. This class establishes a virtual connection
1366# to an XML-RPC server.
1367# <p>
1368# This class is available as ServerProxy and Server. New code should
1369# use ServerProxy, to avoid confusion.
1370#
1371# @def ServerProxy(uri, **options)
1372# @param uri The connection point on the server.
1373# @keyparam transport A transport factory, compatible with the
1374# standard transport class.
1375# @keyparam encoding The default encoding used for 8-bit strings
1376# (default is UTF-8).
1377# @keyparam verbose Use a true value to enable debugging output.
1378# (printed to standard output).
1379# @see Transport
1380
Fredrik Lundhb9056332001-07-11 17:42:21 +00001381class ServerProxy:
1382 """uri [,options] -> a logical connection to an XML-RPC server
1383
1384 uri is the connection point on the server, given as
1385 scheme://host/target.
1386
1387 The standard implementation always supports the "http" scheme. If
1388 SSL socket support is available (Python 2.0), it also supports
1389 "https".
1390
1391 If the target part and the slash preceding it are both omitted,
1392 "/RPC2" is assumed.
1393
1394 The following options can be given as keyword arguments:
1395
1396 transport: a transport factory
1397 encoding: the request encoding (default is UTF-8)
1398
1399 All 8-bit strings passed to the server proxy are assumed to use
1400 the given encoding.
1401 """
1402
Georg Brandlfe991052009-09-16 15:54:04 +00001403 def __init__(self, uri, transport=None, encoding=None, verbose=False,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001404 allow_none=False, use_datetime=False, use_builtin_types=False,
1405 *, context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001406 # establish a "logical" server connection
1407
1408 # get the url
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001409 type, uri = urllib.parse.splittype(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001410 if type not in ("http", "https"):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001411 raise OSError("unsupported XML-RPC protocol")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001412 self.__host, self.__handler = urllib.parse.splithost(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001413 if not self.__handler:
1414 self.__handler = "/RPC2"
1415
1416 if transport is None:
1417 if type == "https":
Florent Xicluna61665192011-11-15 20:53:25 +01001418 handler = SafeTransport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001419 extra_kwargs = {"context": context}
Fredrik Lundhb9056332001-07-11 17:42:21 +00001420 else:
Florent Xicluna61665192011-11-15 20:53:25 +01001421 handler = Transport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001422 extra_kwargs = {}
Florent Xicluna61665192011-11-15 20:53:25 +01001423 transport = handler(use_datetime=use_datetime,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001424 use_builtin_types=use_builtin_types,
1425 **extra_kwargs)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001426 self.__transport = transport
1427
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001428 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001429 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001430 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001431
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001432 def __close(self):
1433 self.__transport.close()
1434
Fredrik Lundhb9056332001-07-11 17:42:21 +00001435 def __request(self, methodname, params):
1436 # call a method on the remote server
1437
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001438 request = dumps(params, methodname, encoding=self.__encoding,
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001439 allow_none=self.__allow_none).encode(self.__encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001440
1441 response = self.__transport.request(
1442 self.__host,
1443 self.__handler,
1444 request,
1445 verbose=self.__verbose
1446 )
1447
1448 if len(response) == 1:
1449 response = response[0]
1450
1451 return response
1452
1453 def __repr__(self):
1454 return (
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001455 "<ServerProxy for %s%s>" %
Fredrik Lundhb9056332001-07-11 17:42:21 +00001456 (self.__host, self.__handler)
1457 )
1458
1459 __str__ = __repr__
Raymond Hettingercc523fc2003-11-02 09:47:05 +00001460
Fredrik Lundhb9056332001-07-11 17:42:21 +00001461 def __getattr__(self, name):
1462 # magic method dispatcher
1463 return _Method(self.__request, name)
1464
1465 # note: to call a remote object with an non-standard name, use
1466 # result getattr(server, "strange-python-name")(args)
1467
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001468 def __call__(self, attr):
1469 """A workaround to get special attributes on the ServerProxy
1470 without interfering with the magic __getattr__
1471 """
1472 if attr == "close":
1473 return self.__close
1474 elif attr == "transport":
1475 return self.__transport
1476 raise AttributeError("Attribute %r not found" % (attr,))
1477
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001478# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001479
Fredrik Lundhb9056332001-07-11 17:42:21 +00001480Server = ServerProxy
1481
1482# --------------------------------------------------------------------
1483# test code
1484
1485if __name__ == "__main__":
1486
1487 # simple test program (from the XML-RPC specification)
1488
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001489 # local server, available from Lib/xmlrpc/server.py
1490 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001491
Fredrik Lundhb9056332001-07-11 17:42:21 +00001492 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001493 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001494 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001495 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001496
1497 multi = MultiCall(server)
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001498 multi.getData()
1499 multi.pow(2,9)
1500 multi.add(1,2)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001501 try:
1502 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001503 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001504 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001505 print("ERROR", v)