blob: 246ef27ffc241aee630f53928a32321af22b3e97 [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
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300135from decimal import Decimal
Georg Brandl24420152008-05-26 16:32:26 +0000136import http.client
Ezio Melotti91932da2012-02-24 12:44:04 +0200137import urllib.parse
Georg Brandlcef803f2009-06-04 09:04:53 +0000138from xml.parsers import expat
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000139import errno
140from io import BytesIO
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000141try:
142 import gzip
Brett Cannoncd171c82013-07-04 17:43:24 -0400143except ImportError:
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000144 gzip = None #python can be built without zlib/gzip support
Fredrik Lundh1538c232001-10-01 19:42:03 +0000145
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000146# --------------------------------------------------------------------
147# Internal stuff
148
Neal Norwitzff113342007-04-17 08:42:15 +0000149def escape(s):
150 s = s.replace("&", "&")
151 s = s.replace("<", "&lt;")
152 return s.replace(">", "&gt;",)
Fredrik Lundh1538c232001-10-01 19:42:03 +0000153
Florent Xicluna75861df2011-10-30 20:39:24 +0100154# used in User-Agent header sent
Serhiy Storchaka885bdc42016-02-11 13:10:36 +0200155__version__ = '%d.%d' % sys.version_info[:2]
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000156
157# xmlrpc integer limits
Guido van Rossume2a383d2007-01-15 16:59:06 +0000158MAXINT = 2**31-1
159MININT = -2**31
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000160
161# --------------------------------------------------------------------
162# Error constants (from Dan Libby's specification at
163# http://xmlrpc-epi.sourceforge.net/specs/rfc.fault_codes.php)
164
165# Ranges of errors
166PARSE_ERROR = -32700
167SERVER_ERROR = -32600
168APPLICATION_ERROR = -32500
169SYSTEM_ERROR = -32400
170TRANSPORT_ERROR = -32300
171
172# Specific errors
173NOT_WELLFORMED_ERROR = -32700
174UNSUPPORTED_ENCODING = -32701
175INVALID_ENCODING_CHAR = -32702
176INVALID_XMLRPC = -32600
177METHOD_NOT_FOUND = -32601
178INVALID_METHOD_PARAMS = -32602
179INTERNAL_ERROR = -32603
Fredrik Lundhb9056332001-07-11 17:42:21 +0000180
181# --------------------------------------------------------------------
182# Exceptions
183
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000184##
185# Base class for all kinds of client-side errors.
186
Fredrik Lundh78eedce2001-08-23 20:04:33 +0000187class Error(Exception):
Fred Drake1b410792001-09-04 18:55:03 +0000188 """Base class for client errors."""
Serhiy Storchaka96aeaec2019-05-06 22:29:40 +0300189 __str__ = object.__str__
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 (
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300211 "<%s for %s: %s %s>" %
212 (self.__class__.__name__, self.url, self.errcode, self.errmsg)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000213 )
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):
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300240 return "<%s %s: %r>" % (self.__class__.__name__,
241 self.faultCode, self.faultString)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000242
243# --------------------------------------------------------------------
244# Special values
245
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000246##
Georg Brandl38eceaa2008-05-26 11:14:17 +0000247# Backwards compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000248
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000249boolean = Boolean = bool
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000250
251##
252# Wrapper for XML-RPC DateTime values. This converts a time value to
253# the format used by XML-RPC.
254# <p>
Florent Xicluna2738a642011-11-01 00:06:58 +0100255# The value can be given as a datetime object, as a string in the
256# format "yyyymmddThh:mm:ss", as a 9-item time tuple (as returned by
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000257# time.localtime()), or an integer value (as returned by time.time()).
258# The wrapper uses time.localtime() to convert an integer to a time
259# tuple.
260#
Florent Xicluna2738a642011-11-01 00:06:58 +0100261# @param value The time, given as a datetime object, an ISO 8601 string,
262# a time tuple, or an integer time value.
263
264
265# Issue #13305: different format codes across platforms
266_day0 = datetime(1, 1, 1)
267if _day0.strftime('%Y') == '0001': # Mac OS X
268 def _iso8601_format(value):
269 return value.strftime("%Y%m%dT%H:%M:%S")
270elif _day0.strftime('%4Y') == '0001': # Linux
271 def _iso8601_format(value):
272 return value.strftime("%4Y%m%dT%H:%M:%S")
273else:
274 def _iso8601_format(value):
275 return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
276del _day0
277
Fredrik Lundhb9056332001-07-11 17:42:21 +0000278
Christian Heimesdae2a892008-04-19 00:55:37 +0000279def _strftime(value):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100280 if isinstance(value, datetime):
Florent Xicluna2738a642011-11-01 00:06:58 +0100281 return _iso8601_format(value)
Christian Heimesdae2a892008-04-19 00:55:37 +0000282
283 if not isinstance(value, (tuple, time.struct_time)):
284 if value == 0:
285 value = time.time()
286 value = time.localtime(value)
287
288 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
289
Fredrik Lundhb9056332001-07-11 17:42:21 +0000290class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000291 """DateTime wrapper for an ISO 8601 string or time tuple or
292 localtime integer value to generate 'dateTime.iso8601' XML-RPC
293 value.
294 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000295
296 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000297 if isinstance(value, str):
298 self.value = value
299 else:
300 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000301
Christian Heimes05e8be12008-02-23 18:30:17 +0000302 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000303 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000304 s = self.value
305 o = other.value
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100306 elif isinstance(other, datetime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000307 s = self.value
Florent Xicluna2738a642011-11-01 00:06:58 +0100308 o = _iso8601_format(other)
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100309 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000310 s = self.value
311 o = other
312 elif hasattr(other, "timetuple"):
313 s = self.timetuple()
314 o = other.timetuple()
315 else:
Serhiy Storchaka662db122019-08-08 08:42:54 +0300316 s = self
317 o = NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000318 return s, o
319
320 def __lt__(self, other):
321 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300322 if o is NotImplemented:
323 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000324 return s < o
325
326 def __le__(self, other):
327 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300328 if o is NotImplemented:
329 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000330 return s <= o
331
332 def __gt__(self, other):
333 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300334 if o is NotImplemented:
335 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000336 return s > o
337
338 def __ge__(self, other):
339 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300340 if o is NotImplemented:
341 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000342 return s >= o
343
344 def __eq__(self, other):
345 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300346 if o is NotImplemented:
347 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000348 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000349
Christian Heimes05e8be12008-02-23 18:30:17 +0000350 def timetuple(self):
351 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
352
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000353 ##
354 # Get date/time value.
355 #
356 # @return Date/time value, as an ISO 8601 string.
357
358 def __str__(self):
359 return self.value
360
Fredrik Lundhb9056332001-07-11 17:42:21 +0000361 def __repr__(self):
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300362 return "<%s %r at %#x>" % (self.__class__.__name__, self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000363
364 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000365 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000366
367 def encode(self, out):
368 out.write("<value><dateTime.iso8601>")
369 out.write(self.value)
370 out.write("</dateTime.iso8601></value>\n")
371
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000372def _datetime(data):
373 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000374 value = DateTime()
375 value.decode(data)
376 return value
377
Skip Montanaro174dd222005-05-14 20:54:16 +0000378def _datetime_type(data):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100379 return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
Skip Montanaro174dd222005-05-14 20:54:16 +0000380
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000381##
382# Wrapper for binary data. This can be used to transport any kind
383# of binary data over XML-RPC, using BASE64 encoding.
384#
385# @param data An 8-bit string containing arbitrary data.
386
Fredrik Lundhb9056332001-07-11 17:42:21 +0000387class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000388 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000389
390 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000391 if data is None:
392 data = b""
393 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100394 if not isinstance(data, (bytes, bytearray)):
395 raise TypeError("expected bytes or bytearray, not %s" %
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000396 data.__class__.__name__)
397 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000398 self.data = data
399
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000400 ##
401 # Get buffer contents.
402 #
403 # @return Buffer contents, as an 8-bit string.
404
405 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000406 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000407
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000408 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000409 if isinstance(other, Binary):
410 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000411 return self.data == other
412
Fredrik Lundhb9056332001-07-11 17:42:21 +0000413 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000414 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000415
416 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000417 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000418 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000419 out.write(encoded.decode('ascii'))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000420 out.write("</base64></value>\n")
421
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000422def _binary(data):
423 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000424 value = Binary()
425 value.decode(data)
426 return value
427
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000428WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000429
430# --------------------------------------------------------------------
431# XML parsers
432
Georg Brandlcef803f2009-06-04 09:04:53 +0000433class ExpatParser:
434 # fast expat parser for Python 2.0 and later.
435 def __init__(self, target):
436 self._parser = parser = expat.ParserCreate(None, None)
437 self._target = target
438 parser.StartElementHandler = target.start
439 parser.EndElementHandler = target.end
440 parser.CharacterDataHandler = target.data
441 encoding = None
442 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000443
Georg Brandlcef803f2009-06-04 09:04:53 +0000444 def feed(self, data):
445 self._parser.Parse(data, 0)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000446
Georg Brandlcef803f2009-06-04 09:04:53 +0000447 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300448 try:
449 parser = self._parser
450 except AttributeError:
451 pass
452 else:
453 del self._target, self._parser # get rid of circular references
454 parser.Parse(b"", True) # end of data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000455
Fredrik Lundhb9056332001-07-11 17:42:21 +0000456# --------------------------------------------------------------------
457# XML-RPC marshalling and unmarshalling code
458
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000459##
460# XML-RPC marshaller.
461#
462# @param encoding Default encoding for 8-bit strings. The default
463# value is None (interpreted as UTF-8).
464# @see dumps
465
Fredrik Lundhb9056332001-07-11 17:42:21 +0000466class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000467 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000468
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000469 Create a Marshaller instance for each set of parameters, and use
470 the "dumps" method to convert your data (represented as a tuple)
471 to an XML-RPC params chunk. To write a fault response, pass a
472 Fault instance instead. You may prefer to use the "dumps" module
473 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000474 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000475
476 # by the way, if you don't understand what's going on in here,
477 # that's perfectly ok.
478
Georg Brandlfe991052009-09-16 15:54:04 +0000479 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000480 self.memo = {}
481 self.data = None
482 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000483 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000484
Fredrik Lundhb9056332001-07-11 17:42:21 +0000485 dispatch = {}
486
487 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000488 out = []
489 write = out.append
490 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000491 if isinstance(values, Fault):
492 # fault instance
493 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000494 dump({'faultCode': values.faultCode,
495 'faultString': values.faultString},
496 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000497 write("</fault>\n")
498 else:
499 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000500 # FIXME: the xml-rpc specification allows us to leave out
501 # the entire <params> block if there are no parameters.
502 # however, changing this may break older code (including
503 # old versions of xmlrpclib.py), so this is better left as
504 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000505 write("<params>\n")
506 for v in values:
507 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000508 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000509 write("</param>\n")
510 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000511 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000512 return result
513
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000514 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000515 try:
516 f = self.dispatch[type(value)]
517 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000518 # check if this object can be marshalled as a structure
Florent Xicluna93dfee12011-10-30 20:22:25 +0100519 if not hasattr(value, '__dict__'):
Collin Winterce36ad82007-08-30 01:19:48 +0000520 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000521 # check if this class is a sub-class of a basic type,
522 # because we don't know how to marshal these types
523 # (e.g. a string sub-class)
524 for type_ in type(value).__mro__:
525 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000526 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000527 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
528 # for the p3yk merge, this should probably be fixed more neatly.
529 f = self.dispatch["_arbitrary_instance"]
530 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000531
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000532 def dump_nil (self, value, write):
533 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000534 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000535 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000536 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000537
Georg Brandl38eceaa2008-05-26 11:14:17 +0000538 def dump_bool(self, value, write):
539 write("<value><boolean>")
540 write(value and "1" or "0")
541 write("</boolean></value>\n")
542 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000543
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000544 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000545 if value > MAXINT or value < MININT:
Serhiy Storchaka95949422013-08-27 19:40:23 +0300546 raise OverflowError("int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000547 write("<value><int>")
548 write(str(int(value)))
549 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000550 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000551
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200552 # backward compatible
553 dump_int = dump_long
554
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000555 def dump_double(self, value, write):
556 write("<value><double>")
557 write(repr(value))
558 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000559 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000560
Guido van Rossum54ad5232007-05-27 09:17:48 +0000561 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000562 write("<value><string>")
563 write(escape(value))
564 write("</string></value>\n")
565 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000566
Florent Xicluna61665192011-11-15 20:53:25 +0100567 def dump_bytes(self, value, write):
568 write("<value><base64>\n")
569 encoded = base64.encodebytes(value)
570 write(encoded.decode('ascii'))
571 write("</base64></value>\n")
572 dispatch[bytes] = dump_bytes
573 dispatch[bytearray] = dump_bytes
574
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000575 def dump_array(self, value, write):
576 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000577 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000578 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000579 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000580 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000581 write("<value><array><data>\n")
582 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000583 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000584 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000585 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000586 dispatch[tuple] = dump_array
587 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000588
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000589 def dump_struct(self, value, write, escape=escape):
590 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000591 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000592 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000593 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000594 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000595 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000596 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000597 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000598 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000599 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000600 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000601 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000602 write("</member>\n")
603 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000604 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000605 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000606
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100607 def dump_datetime(self, value, write):
608 write("<value><dateTime.iso8601>")
609 write(_strftime(value))
610 write("</dateTime.iso8601></value>\n")
611 dispatch[datetime] = dump_datetime
Fred Drakeba613c32005-02-10 18:33:30 +0000612
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000613 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000614 # check for special wrappers
615 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000616 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000617 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000618 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000619 else:
620 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000621 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000622 dispatch[DateTime] = dump_instance
623 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000624 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
625 # for the p3yk merge, this should probably be fixed more neatly.
626 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000627
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000628##
629# XML-RPC unmarshaller.
630#
631# @see loads
632
Fredrik Lundhb9056332001-07-11 17:42:21 +0000633class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000634 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000635 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000636 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000637
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000638 Note that this reader is fairly tolerant, and gladly accepts bogus
639 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000640 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000641
642 # and again, if you don't understand what's going on in here,
643 # that's perfectly ok.
644
Florent Xicluna61665192011-11-15 20:53:25 +0100645 def __init__(self, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000646 self._type = None
647 self._stack = []
648 self._marks = []
649 self._data = []
Serhiy Storchaka290fed42016-05-04 11:26:42 +0300650 self._value = False
Fredrik Lundhb9056332001-07-11 17:42:21 +0000651 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
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300677 if ':' in tag:
678 tag = tag.split(':')[-1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000679 if tag == "array" or tag == "struct":
680 self._marks.append(len(self._stack))
681 self._data = []
Serhiy Storchaka290fed42016-05-04 11:26:42 +0300682 if self._value and tag not in self.dispatch:
683 raise ResponseError("unknown tag %r" % tag)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000684 self._value = (tag == "value")
685
686 def data(self, text):
687 self._data.append(text)
688
Neal Norwitzff113342007-04-17 08:42:15 +0000689 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000690 # call the appropriate end tag handler
691 try:
692 f = self.dispatch[tag]
693 except KeyError:
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300694 if ':' not in tag:
695 return # unknown tag ?
696 try:
697 f = self.dispatch[tag.split(':')[-1]]
698 except KeyError:
699 return # unknown tag ?
700 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000701
702 #
703 # accelerator support
704
705 def end_dispatch(self, tag, data):
706 # dispatch data
707 try:
708 f = self.dispatch[tag]
709 except KeyError:
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300710 if ':' not in tag:
711 return # unknown tag ?
712 try:
713 f = self.dispatch[tag.split(':')[-1]]
714 except KeyError:
715 return # unknown tag ?
716 return f(self, data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000717
718 #
719 # element decoders
720
721 dispatch = {}
722
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000723 def end_nil (self, data):
724 self.append(None)
725 self._value = 0
726 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000727
Fredrik Lundh1538c232001-10-01 19:42:03 +0000728 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000729 if data == "0":
730 self.append(False)
731 elif data == "1":
732 self.append(True)
733 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000734 raise TypeError("bad boolean value")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000735 self._value = 0
736 dispatch["boolean"] = end_boolean
737
Fredrik Lundh1538c232001-10-01 19:42:03 +0000738 def end_int(self, data):
739 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000740 self._value = 0
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300741 dispatch["i1"] = end_int
742 dispatch["i2"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000743 dispatch["i4"] = end_int
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000744 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000745 dispatch["int"] = end_int
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300746 dispatch["biginteger"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000747
Fredrik Lundh1538c232001-10-01 19:42:03 +0000748 def end_double(self, data):
749 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000750 self._value = 0
751 dispatch["double"] = end_double
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300752 dispatch["float"] = end_double
753
754 def end_bigdecimal(self, data):
755 self.append(Decimal(data))
756 self._value = 0
757 dispatch["bigdecimal"] = end_bigdecimal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000758
Fredrik Lundh1538c232001-10-01 19:42:03 +0000759 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000760 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000761 data = data.decode(self._encoding)
762 self.append(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000763 self._value = 0
764 dispatch["string"] = end_string
765 dispatch["name"] = end_string # struct keys are always strings
766
767 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000768 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769 # map arrays to Python lists
770 self._stack[mark:] = [self._stack[mark:]]
771 self._value = 0
772 dispatch["array"] = end_array
773
774 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000775 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000776 # map structs to Python dictionaries
777 dict = {}
778 items = self._stack[mark:]
779 for i in range(0, len(items), 2):
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000780 dict[items[i]] = items[i+1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000781 self._stack[mark:] = [dict]
782 self._value = 0
783 dispatch["struct"] = end_struct
784
Fredrik Lundh1538c232001-10-01 19:42:03 +0000785 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000786 value = Binary()
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000787 value.decode(data.encode("ascii"))
Florent Xicluna61665192011-11-15 20:53:25 +0100788 if self._use_bytes:
789 value = value.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000790 self.append(value)
791 self._value = 0
792 dispatch["base64"] = end_base64
793
Fredrik Lundh1538c232001-10-01 19:42:03 +0000794 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000795 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000796 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000797 if self._use_datetime:
798 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000799 self.append(value)
800 dispatch["dateTime.iso8601"] = end_dateTime
801
802 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000803 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000804 # elements, treat it as a string element
805 if self._value:
806 self.end_string(data)
807 dispatch["value"] = end_value
808
809 def end_params(self, data):
810 self._type = "params"
811 dispatch["params"] = end_params
812
813 def end_fault(self, data):
814 self._type = "fault"
815 dispatch["fault"] = end_fault
816
Fredrik Lundh1538c232001-10-01 19:42:03 +0000817 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000818 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000819 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000820 self._methodname = data
821 self._type = "methodName" # no params
822 dispatch["methodName"] = end_methodName
823
Martin v. Löwis45394c22003-10-31 13:49:36 +0000824## Multicall support
825#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000826
Martin v. Löwis45394c22003-10-31 13:49:36 +0000827class _MultiCallMethod:
828 # some lesser magic to store calls made to a MultiCall object
829 # for batch execution
830 def __init__(self, call_list, name):
831 self.__call_list = call_list
832 self.__name = name
833 def __getattr__(self, name):
834 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
835 def __call__(self, *args):
836 self.__call_list.append((self.__name, args))
837
Martin v. Löwis12237b32004-08-22 16:04:50 +0000838class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000839 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov737fb892012-12-18 21:14:22 +0200840 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000841
Martin v. Löwis12237b32004-08-22 16:04:50 +0000842 def __init__(self, results):
843 self.results = results
844
845 def __getitem__(self, i):
846 item = self.results[i]
847 if type(item) == type({}):
848 raise Fault(item['faultCode'], item['faultString'])
849 elif type(item) == type([]):
850 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000851 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000852 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000853
Martin v. Löwis45394c22003-10-31 13:49:36 +0000854class MultiCall:
Martin Panter7462b6492015-11-02 03:37:02 +0000855 """server -> an object used to boxcar method calls
Martin v. Löwis45394c22003-10-31 13:49:36 +0000856
857 server should be a ServerProxy object.
858
859 Methods can be added to the MultiCall using normal
860 method call syntax e.g.:
861
862 multicall = MultiCall(server_proxy)
863 multicall.add(2,3)
864 multicall.get_address("Guido")
865
866 To execute the multicall, call the MultiCall object e.g.:
867
868 add_result, address = multicall()
869 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000870
Martin v. Löwis45394c22003-10-31 13:49:36 +0000871 def __init__(self, server):
872 self.__server = server
873 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000874
Martin v. Löwis45394c22003-10-31 13:49:36 +0000875 def __repr__(self):
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300876 return "<%s at %#x>" % (self.__class__.__name__, id(self))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000877
Martin v. Löwis45394c22003-10-31 13:49:36 +0000878 def __getattr__(self, name):
879 return _MultiCallMethod(self.__call_list, name)
880
881 def __call__(self):
882 marshalled_list = []
883 for name, args in self.__call_list:
884 marshalled_list.append({'methodName' : name, 'params' : args})
885
886 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000887
Fredrik Lundhb9056332001-07-11 17:42:21 +0000888# --------------------------------------------------------------------
889# convenience functions
890
Georg Brandl38eceaa2008-05-26 11:14:17 +0000891FastMarshaller = FastParser = FastUnmarshaller = None
892
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000893##
894# Create a parser object, and connect it to an unmarshalling instance.
895# This function picks the fastest available XML parser.
896#
897# return A (parser, unmarshaller) tuple.
898
Florent Xicluna61665192011-11-15 20:53:25 +0100899def getparser(use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000900 """getparser() -> parser, unmarshaller
901
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000902 Create an instance of the fastest available parser, and attach it
903 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000904 """
905 if FastParser and FastUnmarshaller:
Florent Xicluna61665192011-11-15 20:53:25 +0100906 if use_builtin_types:
Skip Montanaro174dd222005-05-14 20:54:16 +0000907 mkdatetime = _datetime_type
Florent Xicluna61665192011-11-15 20:53:25 +0100908 mkbytes = base64.decodebytes
909 elif use_datetime:
910 mkdatetime = _datetime_type
911 mkbytes = _binary
Skip Montanaro174dd222005-05-14 20:54:16 +0000912 else:
913 mkdatetime = _datetime
Florent Xicluna61665192011-11-15 20:53:25 +0100914 mkbytes = _binary
915 target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000916 parser = FastParser(target)
917 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100918 target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000919 if FastParser:
920 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000921 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000922 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000923 return parser, target
924
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000925##
926# Convert a Python tuple or a Fault instance to an XML-RPC packet.
927#
928# @def dumps(params, **options)
929# @param params A tuple or Fault instance.
930# @keyparam methodname If given, create a methodCall request for
931# this method name.
932# @keyparam methodresponse If given, create a methodResponse packet.
933# If used with a tuple, the tuple must be a singleton (that is,
934# it must contain exactly one element).
935# @keyparam encoding The packet encoding.
936# @return A string containing marshalled data.
937
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000938def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000939 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000940 """data [,options] -> marshalled data
941
942 Convert an argument tuple or a Fault instance to an XML-RPC
943 request (or response, if the methodresponse option is used).
944
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000945 In addition to the data object, the following options can be given
946 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000947
948 methodname: the method name for a methodCall packet
949
950 methodresponse: true to create a methodResponse packet.
951 If this option is used with a tuple, the tuple must be
952 a singleton (i.e. it can contain only one element).
953
954 encoding: the packet encoding (default is UTF-8)
955
Florent Xicluna61665192011-11-15 20:53:25 +0100956 All byte strings in the data structure are assumed to use the
Fredrik Lundhb9056332001-07-11 17:42:21 +0000957 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000958 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000959 """
960
Guido van Rossum13257902007-06-07 23:15:56 +0000961 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000962 if isinstance(params, Fault):
963 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000964 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000965 assert len(params) == 1, "response tuple must be a singleton"
966
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000967 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000968 encoding = "utf-8"
969
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000970 if FastMarshaller:
971 m = FastMarshaller(encoding)
972 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000973 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000974
Fredrik Lundhb9056332001-07-11 17:42:21 +0000975 data = m.dumps(params)
976
977 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000978 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000979 else:
980 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
981
982 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000983 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000984 # a method call
Fredrik Lundhb9056332001-07-11 17:42:21 +0000985 data = (
986 xmlheader,
987 "<methodCall>\n"
988 "<methodName>", methodname, "</methodName>\n",
989 data,
990 "</methodCall>\n"
991 )
992 elif methodresponse:
993 # a method response, or a fault structure
994 data = (
995 xmlheader,
996 "<methodResponse>\n",
997 data,
998 "</methodResponse>\n"
999 )
1000 else:
1001 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +00001002 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001003
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001004##
1005# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1006# represents a fault condition, this function raises a Fault exception.
1007#
1008# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001009# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001010# (None if not present).
1011# @see Fault
1012
Florent Xicluna61665192011-11-15 20:53:25 +01001013def loads(data, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001014 """data -> unmarshalled data, method name
1015
1016 Convert an XML-RPC packet to unmarshalled data plus a method
1017 name (None if not present).
1018
1019 If the XML-RPC packet represents a fault condition, this function
1020 raises a Fault exception.
1021 """
Florent Xicluna61665192011-11-15 20:53:25 +01001022 p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001023 p.feed(data)
1024 p.close()
1025 return u.close(), u.getmethodname()
1026
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001027##
1028# Encode a string using the gzip content encoding such as specified by the
1029# Content-Encoding: gzip
1030# in the HTTP header, as described in RFC 1952
1031#
1032# @param data the unencoded data
1033# @return the encoded data
1034
1035def gzip_encode(data):
1036 """data -> gzip encoded data
1037
1038 Encode data using the gzip content encoding as described in RFC 1952
1039 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001040 if not gzip:
1041 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001042 f = BytesIO()
Serhiy Storchaka46ba6c82015-04-04 11:01:02 +03001043 with gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1) as gzf:
1044 gzf.write(data)
1045 return f.getvalue()
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001046
1047##
1048# Decode a string using the gzip content encoding such as specified by the
1049# Content-Encoding: gzip
1050# in the HTTP header, as described in RFC 1952
1051#
1052# @param data The encoded data
Victor Stinner8c663fd2017-11-08 14:44:44 -08001053# @keyparam max_decode Maximum bytes to decode (20 MiB default), use negative
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001054# values for unlimited decoding
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001055# @return the unencoded data
1056# @raises ValueError if data is not correctly coded.
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001057# @raises ValueError if max gzipped payload length exceeded
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001058
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001059def gzip_decode(data, max_decode=20971520):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001060 """gzip encoded data -> unencoded data
1061
1062 Decode data using the gzip content encoding as described in RFC 1952
1063 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001064 if not gzip:
1065 raise NotImplementedError
Serhiy Storchaka46ba6c82015-04-04 11:01:02 +03001066 with gzip.GzipFile(mode="rb", fileobj=BytesIO(data)) as gzf:
1067 try:
1068 if max_decode < 0: # no limit
1069 decoded = gzf.read()
1070 else:
1071 decoded = gzf.read(max_decode + 1)
1072 except OSError:
1073 raise ValueError("invalid data")
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001074 if max_decode >= 0 and len(decoded) > max_decode:
1075 raise ValueError("max gzipped payload length exceeded")
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001076 return decoded
1077
1078##
1079# Return a decoded file-like object for the gzip encoding
1080# as described in RFC 1952.
1081#
1082# @param response A stream supporting a read() method
1083# @return a file-like object that the decoded data can be read() from
1084
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001085class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001086 """a file-like object to decode a response encoded with the gzip
1087 method, as described in RFC 1952.
1088 """
1089 def __init__(self, response):
1090 #response doesn't support tell() and read(), required by
1091 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001092 if not gzip:
1093 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001094 self.io = BytesIO(response.read())
1095 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1096
1097 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001098 try:
1099 gzip.GzipFile.close(self)
1100 finally:
1101 self.io.close()
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001102
Fredrik Lundhb9056332001-07-11 17:42:21 +00001103
1104# --------------------------------------------------------------------
1105# request dispatcher
1106
1107class _Method:
1108 # some magic to bind an XML-RPC method to an RPC server.
1109 # supports "nested" methods (e.g. examples.getStateName)
1110 def __init__(self, send, name):
1111 self.__send = send
1112 self.__name = name
1113 def __getattr__(self, name):
1114 return _Method(self.__send, "%s.%s" % (self.__name, name))
1115 def __call__(self, *args):
1116 return self.__send(self.__name, args)
1117
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001118##
1119# Standard transport class for XML-RPC over HTTP.
1120# <p>
1121# You can create custom transports by subclassing this method, and
1122# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001123
1124class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001125 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001126
1127 # client identifier (may be overridden)
Florent Xicluna75861df2011-10-30 20:39:24 +01001128 user_agent = "Python-xmlrpc/%s" % __version__
Fredrik Lundhb9056332001-07-11 17:42:21 +00001129
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001130 #if true, we'll request gzip encoding
1131 accept_gzip_encoding = True
1132
1133 # if positive, encode request using gzip if it exceeds this threshold
Mike53f7a7c2017-12-14 14:04:53 +03001134 # note that many servers will get confused, so only use it if you know
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001135 # that they can decode such a request
1136 encode_threshold = None #None = don't encode
1137
Cédric Krierbeda52e2019-02-19 17:18:50 +01001138 def __init__(self, use_datetime=False, use_builtin_types=False,
1139 *, headers=()):
Skip Montanaro174dd222005-05-14 20:54:16 +00001140 self._use_datetime = use_datetime
Florent Xicluna61665192011-11-15 20:53:25 +01001141 self._use_builtin_types = use_builtin_types
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001142 self._connection = (None, None)
Cédric Krierbeda52e2019-02-19 17:18:50 +01001143 self._headers = list(headers)
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001144 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001145
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001146 ##
1147 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001148 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001149 #
1150 # @param host Target host.
1151 # @param handler Target PRC handler.
1152 # @param request_body XML-RPC request body.
1153 # @param verbose Debugging flag.
1154 # @return Parsed response.
1155
Georg Brandlfe991052009-09-16 15:54:04 +00001156 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001157 #retry request once if cached connection has gone cold
1158 for i in (0, 1):
1159 try:
1160 return self.single_request(host, handler, request_body, verbose)
Martin Pantereae33362016-02-25 11:53:40 +00001161 except http.client.RemoteDisconnected:
1162 if i:
1163 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001164 except OSError as e:
1165 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
1166 errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001167 raise
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001168
Georg Brandlfe991052009-09-16 15:54:04 +00001169 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001170 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001171 try:
1172 http_conn = self.send_request(host, handler, request_body, verbose)
1173 resp = http_conn.getresponse()
1174 if resp.status == 200:
1175 self.verbose = verbose
1176 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001177
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001178 except Fault:
1179 raise
1180 except Exception:
1181 #All unexpected errors leave connection in
1182 # a strange state, so we clear it.
1183 self.close()
1184 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001185
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001186 #We got an error response.
1187 #Discard any response data and raise exception
1188 if resp.getheader("content-length", ""):
1189 resp.read()
1190 raise ProtocolError(
1191 host + handler,
1192 resp.status, resp.reason,
1193 dict(resp.getheaders())
1194 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001195
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001196
1197 ##
1198 # Create parser.
1199 #
Martin Panter7462b6492015-11-02 03:37:02 +00001200 # @return A 2-tuple containing a parser and an unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001201
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001202 def getparser(self):
1203 # get parser and unmarshaller
Florent Xicluna61665192011-11-15 20:53:25 +01001204 return getparser(use_datetime=self._use_datetime,
1205 use_builtin_types=self._use_builtin_types)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001206
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001207 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001208 # Get authorization info from host parameter
1209 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1210 # it is checked for a "user:pw@host" format, and a "Basic
1211 # Authentication" header is added if appropriate.
1212 #
1213 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1214 # @return A 3-tuple containing (actual host, extra headers,
1215 # x509 info). The header and x509 fields may be None.
1216
1217 def get_host_info(self, host):
1218
1219 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001220 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001221 host, x509 = host
1222
Cheryl Sabella0250de42018-04-25 16:51:54 -07001223 auth, host = urllib.parse._splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001224
1225 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001226 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001227 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001228 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001229 extra_headers = [
1230 ("Authorization", "Basic " + auth)
1231 ]
1232 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001233 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001234
1235 return host, extra_headers, x509
1236
1237 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001238 # Connect to server.
1239 #
1240 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001241 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001242
Fredrik Lundhb9056332001-07-11 17:42:21 +00001243 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001244 #return an existing connection if possible. This allows
1245 #HTTP/1.1 keep-alive.
1246 if self._connection and host == self._connection[0]:
1247 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001248 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001249 chost, self._extra_headers, x509 = self.get_host_info(host)
1250 self._connection = host, http.client.HTTPConnection(chost)
1251 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001252
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001253 ##
1254 # Clear any cached connection object.
1255 # Used in the event of socket errors.
1256 #
1257 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001258 host, connection = self._connection
1259 if connection:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001260 self._connection = (None, None)
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001261 connection.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001262
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001263 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001264 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001265 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001266 # @param host Host descriptor (URL or (URL, x509 info) tuple).
Mike53f7a7c2017-12-14 14:04:53 +03001267 # @param handler Target RPC handler (a path relative to host)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001268 # @param request_body The XML-RPC request body
1269 # @param debug Enable debugging if debug is true.
1270 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001271
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001272 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001273 connection = self.make_connection(host)
Cédric Krierbeda52e2019-02-19 17:18:50 +01001274 headers = self._headers + self._extra_headers
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001275 if debug:
1276 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001277 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001278 connection.putrequest("POST", handler, skip_accept_encoding=True)
1279 headers.append(("Accept-Encoding", "gzip"))
1280 else:
1281 connection.putrequest("POST", handler)
1282 headers.append(("Content-Type", "text/xml"))
1283 headers.append(("User-Agent", self.user_agent))
1284 self.send_headers(connection, headers)
1285 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001286 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001287
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001288 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001289 # Send request headers.
1290 # This function provides a useful hook for subclassing
1291 #
1292 # @param connection httpConnection.
1293 # @param headers list of key,value pairs for HTTP headers
1294
1295 def send_headers(self, connection, headers):
1296 for key, val in headers:
1297 connection.putheader(key, val)
1298
1299 ##
1300 # Send request body.
1301 # This function provides a useful hook for subclassing
1302 #
1303 # @param connection httpConnection.
1304 # @param request_body XML-RPC request body.
1305
1306 def send_content(self, connection, request_body):
1307 #optionally encode the request
1308 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001309 self.encode_threshold < len(request_body) and
1310 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001311 connection.putheader("Content-Encoding", "gzip")
1312 request_body = gzip_encode(request_body)
1313
1314 connection.putheader("Content-Length", str(len(request_body)))
1315 connection.endheaders(request_body)
1316
1317 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001318 # Parse response.
1319 #
1320 # @param file Stream.
1321 # @return Response tuple and target method.
1322
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001323 def parse_response(self, response):
1324 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001325 # Check for new http response object, otherwise it is a file object.
1326 if hasattr(response, 'getheader'):
1327 if response.getheader("Content-Encoding", "") == "gzip":
1328 stream = GzipDecodedResponse(response)
1329 else:
1330 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001331 else:
1332 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001333
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001334 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001335
1336 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001337 data = stream.read(1024)
1338 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001339 break
1340 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001341 print("body:", repr(data))
1342 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001343
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001344 if stream is not response:
1345 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001346 p.close()
1347
1348 return u.close()
1349
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001350##
1351# Standard transport class for XML-RPC over HTTPS.
1352
Fredrik Lundhb9056332001-07-11 17:42:21 +00001353class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001354 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001355
Cédric Krierbeda52e2019-02-19 17:18:50 +01001356 def __init__(self, use_datetime=False, use_builtin_types=False,
1357 *, headers=(), context=None):
1358 super().__init__(use_datetime=use_datetime,
1359 use_builtin_types=use_builtin_types,
1360 headers=headers)
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001361 self.context = context
1362
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001363 # FIXME: mostly untested
1364
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001365 def make_connection(self, host):
1366 if self._connection and host == self._connection[0]:
1367 return self._connection[1]
1368
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001369 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001370 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001371 "your version of http.client doesn't support HTTPS")
1372 # create a HTTPS connection object from a host descriptor
1373 # host may be a string, or a (host, x509-dict) tuple
1374 chost, self._extra_headers, x509 = self.get_host_info(host)
1375 self._connection = host, http.client.HTTPSConnection(chost,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001376 None, context=self.context, **(x509 or {}))
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001377 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001378
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001379##
1380# Standard server proxy. This class establishes a virtual connection
1381# to an XML-RPC server.
1382# <p>
1383# This class is available as ServerProxy and Server. New code should
1384# use ServerProxy, to avoid confusion.
1385#
1386# @def ServerProxy(uri, **options)
1387# @param uri The connection point on the server.
1388# @keyparam transport A transport factory, compatible with the
1389# standard transport class.
1390# @keyparam encoding The default encoding used for 8-bit strings
1391# (default is UTF-8).
1392# @keyparam verbose Use a true value to enable debugging output.
1393# (printed to standard output).
1394# @see Transport
1395
Fredrik Lundhb9056332001-07-11 17:42:21 +00001396class ServerProxy:
1397 """uri [,options] -> a logical connection to an XML-RPC server
1398
1399 uri is the connection point on the server, given as
1400 scheme://host/target.
1401
1402 The standard implementation always supports the "http" scheme. If
1403 SSL socket support is available (Python 2.0), it also supports
1404 "https".
1405
1406 If the target part and the slash preceding it are both omitted,
1407 "/RPC2" is assumed.
1408
1409 The following options can be given as keyword arguments:
1410
1411 transport: a transport factory
1412 encoding: the request encoding (default is UTF-8)
1413
1414 All 8-bit strings passed to the server proxy are assumed to use
1415 the given encoding.
1416 """
1417
Georg Brandlfe991052009-09-16 15:54:04 +00001418 def __init__(self, uri, transport=None, encoding=None, verbose=False,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001419 allow_none=False, use_datetime=False, use_builtin_types=False,
Cédric Krierbeda52e2019-02-19 17:18:50 +01001420 *, headers=(), context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001421 # establish a "logical" server connection
1422
1423 # get the url
Cheryl Sabella0250de42018-04-25 16:51:54 -07001424 type, uri = urllib.parse._splittype(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001425 if type not in ("http", "https"):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001426 raise OSError("unsupported XML-RPC protocol")
Cheryl Sabella0250de42018-04-25 16:51:54 -07001427 self.__host, self.__handler = urllib.parse._splithost(uri)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001428 if not self.__handler:
1429 self.__handler = "/RPC2"
1430
1431 if transport is None:
1432 if type == "https":
Florent Xicluna61665192011-11-15 20:53:25 +01001433 handler = SafeTransport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001434 extra_kwargs = {"context": context}
Fredrik Lundhb9056332001-07-11 17:42:21 +00001435 else:
Florent Xicluna61665192011-11-15 20:53:25 +01001436 handler = Transport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001437 extra_kwargs = {}
Florent Xicluna61665192011-11-15 20:53:25 +01001438 transport = handler(use_datetime=use_datetime,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001439 use_builtin_types=use_builtin_types,
Cédric Krierbeda52e2019-02-19 17:18:50 +01001440 headers=headers,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001441 **extra_kwargs)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001442 self.__transport = transport
1443
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001444 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001445 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001446 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001447
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001448 def __close(self):
1449 self.__transport.close()
1450
Fredrik Lundhb9056332001-07-11 17:42:21 +00001451 def __request(self, methodname, params):
1452 # call a method on the remote server
1453
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001454 request = dumps(params, methodname, encoding=self.__encoding,
Serhiy Storchakaaebb6d32016-01-20 10:34:27 +02001455 allow_none=self.__allow_none).encode(self.__encoding, 'xmlcharrefreplace')
Fredrik Lundhb9056332001-07-11 17:42:21 +00001456
1457 response = self.__transport.request(
1458 self.__host,
1459 self.__handler,
1460 request,
1461 verbose=self.__verbose
1462 )
1463
1464 if len(response) == 1:
1465 response = response[0]
1466
1467 return response
1468
1469 def __repr__(self):
1470 return (
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001471 "<%s for %s%s>" %
1472 (self.__class__.__name__, self.__host, self.__handler)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001473 )
1474
Fredrik Lundhb9056332001-07-11 17:42:21 +00001475 def __getattr__(self, name):
1476 # magic method dispatcher
1477 return _Method(self.__request, name)
1478
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +03001479 # note: to call a remote object with a non-standard name, use
Fredrik Lundhb9056332001-07-11 17:42:21 +00001480 # result getattr(server, "strange-python-name")(args)
1481
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001482 def __call__(self, attr):
1483 """A workaround to get special attributes on the ServerProxy
1484 without interfering with the magic __getattr__
1485 """
1486 if attr == "close":
1487 return self.__close
1488 elif attr == "transport":
1489 return self.__transport
1490 raise AttributeError("Attribute %r not found" % (attr,))
1491
Brett Cannon33a40002014-03-21 11:24:40 -04001492 def __enter__(self):
1493 return self
1494
1495 def __exit__(self, *args):
1496 self.__close()
1497
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001498# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001499
Fredrik Lundhb9056332001-07-11 17:42:21 +00001500Server = ServerProxy
1501
1502# --------------------------------------------------------------------
1503# test code
1504
1505if __name__ == "__main__":
1506
1507 # simple test program (from the XML-RPC specification)
1508
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001509 # local server, available from Lib/xmlrpc/server.py
1510 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001511
Fredrik Lundhb9056332001-07-11 17:42:21 +00001512 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001513 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001514 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001515 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001516
1517 multi = MultiCall(server)
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001518 multi.getData()
1519 multi.pow(2,9)
1520 multi.add(1,2)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001521 try:
1522 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001523 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001524 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001525 print("ERROR", v)