blob: a614cef6ab2f1a52b3ece45f9a3c2e6920802e15 [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)
Miss Islington (bot)f40b2302021-10-13 10:59:45 -0700267def _try(fmt):
268 try:
269 return _day0.strftime(fmt) == '0001'
270 except ValueError:
271 return False
272if _try('%Y'): # Mac OS X
Florent Xicluna2738a642011-11-01 00:06:58 +0100273 def _iso8601_format(value):
274 return value.strftime("%Y%m%dT%H:%M:%S")
Miss Islington (bot)f40b2302021-10-13 10:59:45 -0700275elif _try('%4Y'): # Linux
Florent Xicluna2738a642011-11-01 00:06:58 +0100276 def _iso8601_format(value):
277 return value.strftime("%4Y%m%dT%H:%M:%S")
278else:
279 def _iso8601_format(value):
280 return value.strftime("%Y%m%dT%H:%M:%S").zfill(17)
281del _day0
Miss Islington (bot)f40b2302021-10-13 10:59:45 -0700282del _try
Florent Xicluna2738a642011-11-01 00:06:58 +0100283
Fredrik Lundhb9056332001-07-11 17:42:21 +0000284
Christian Heimesdae2a892008-04-19 00:55:37 +0000285def _strftime(value):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100286 if isinstance(value, datetime):
Florent Xicluna2738a642011-11-01 00:06:58 +0100287 return _iso8601_format(value)
Christian Heimesdae2a892008-04-19 00:55:37 +0000288
289 if not isinstance(value, (tuple, time.struct_time)):
290 if value == 0:
291 value = time.time()
292 value = time.localtime(value)
293
294 return "%04d%02d%02dT%02d:%02d:%02d" % value[:6]
295
Fredrik Lundhb9056332001-07-11 17:42:21 +0000296class DateTime:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000297 """DateTime wrapper for an ISO 8601 string or time tuple or
298 localtime integer value to generate 'dateTime.iso8601' XML-RPC
299 value.
300 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000301
302 def __init__(self, value=0):
Christian Heimesdae2a892008-04-19 00:55:37 +0000303 if isinstance(value, str):
304 self.value = value
305 else:
306 self.value = _strftime(value)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000307
Christian Heimes05e8be12008-02-23 18:30:17 +0000308 def make_comparable(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000309 if isinstance(other, DateTime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000310 s = self.value
311 o = other.value
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100312 elif isinstance(other, datetime):
Christian Heimes05e8be12008-02-23 18:30:17 +0000313 s = self.value
Florent Xicluna2738a642011-11-01 00:06:58 +0100314 o = _iso8601_format(other)
Florent Xicluna3fa29f72011-10-30 20:18:50 +0100315 elif isinstance(other, str):
Christian Heimes05e8be12008-02-23 18:30:17 +0000316 s = self.value
317 o = other
318 elif hasattr(other, "timetuple"):
319 s = self.timetuple()
320 o = other.timetuple()
321 else:
Serhiy Storchaka662db122019-08-08 08:42:54 +0300322 s = self
323 o = NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000324 return s, o
325
326 def __lt__(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 __le__(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 __gt__(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 __ge__(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
349
350 def __eq__(self, other):
351 s, o = self.make_comparable(other)
Serhiy Storchaka662db122019-08-08 08:42:54 +0300352 if o is NotImplemented:
353 return NotImplemented
Christian Heimes05e8be12008-02-23 18:30:17 +0000354 return s == o
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000355
Christian Heimes05e8be12008-02-23 18:30:17 +0000356 def timetuple(self):
357 return time.strptime(self.value, "%Y%m%dT%H:%M:%S")
358
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000359 ##
360 # Get date/time value.
361 #
362 # @return Date/time value, as an ISO 8601 string.
363
364 def __str__(self):
365 return self.value
366
Fredrik Lundhb9056332001-07-11 17:42:21 +0000367 def __repr__(self):
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300368 return "<%s %r at %#x>" % (self.__class__.__name__, self.value, id(self))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000369
370 def decode(self, data):
Neal Norwitzff113342007-04-17 08:42:15 +0000371 self.value = str(data).strip()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000372
373 def encode(self, out):
374 out.write("<value><dateTime.iso8601>")
375 out.write(self.value)
376 out.write("</dateTime.iso8601></value>\n")
377
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000378def _datetime(data):
379 # decode xml element contents into a DateTime structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000380 value = DateTime()
381 value.decode(data)
382 return value
383
Skip Montanaro174dd222005-05-14 20:54:16 +0000384def _datetime_type(data):
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100385 return datetime.strptime(data, "%Y%m%dT%H:%M:%S")
Skip Montanaro174dd222005-05-14 20:54:16 +0000386
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000387##
388# Wrapper for binary data. This can be used to transport any kind
389# of binary data over XML-RPC, using BASE64 encoding.
390#
391# @param data An 8-bit string containing arbitrary data.
392
Fredrik Lundhb9056332001-07-11 17:42:21 +0000393class Binary:
Fred Drake1b410792001-09-04 18:55:03 +0000394 """Wrapper for binary data."""
Fredrik Lundhb9056332001-07-11 17:42:21 +0000395
396 def __init__(self, data=None):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000397 if data is None:
398 data = b""
399 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100400 if not isinstance(data, (bytes, bytearray)):
401 raise TypeError("expected bytes or bytearray, not %s" %
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000402 data.__class__.__name__)
403 data = bytes(data) # Make a copy of the bytes!
Fredrik Lundhb9056332001-07-11 17:42:21 +0000404 self.data = data
405
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000406 ##
407 # Get buffer contents.
408 #
409 # @return Buffer contents, as an 8-bit string.
410
411 def __str__(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000412 return str(self.data, "latin-1") # XXX encoding?!
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000413
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000414 def __eq__(self, other):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000415 if isinstance(other, Binary):
416 other = other.data
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000417 return self.data == other
418
Fredrik Lundhb9056332001-07-11 17:42:21 +0000419 def decode(self, data):
Georg Brandlb54d8012009-06-04 09:11:51 +0000420 self.data = base64.decodebytes(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000421
422 def encode(self, out):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000423 out.write("<value><base64>\n")
Georg Brandlb54d8012009-06-04 09:11:51 +0000424 encoded = base64.encodebytes(self.data)
Brett Cannon2dbde5e2007-07-30 03:50:35 +0000425 out.write(encoded.decode('ascii'))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000426 out.write("</base64></value>\n")
427
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000428def _binary(data):
429 # decode xml element contents into a Binary structure
Fredrik Lundhb9056332001-07-11 17:42:21 +0000430 value = Binary()
431 value.decode(data)
432 return value
433
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000434WRAPPERS = (DateTime, Binary)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000435
436# --------------------------------------------------------------------
437# XML parsers
438
Georg Brandlcef803f2009-06-04 09:04:53 +0000439class ExpatParser:
440 # fast expat parser for Python 2.0 and later.
441 def __init__(self, target):
442 self._parser = parser = expat.ParserCreate(None, None)
443 self._target = target
444 parser.StartElementHandler = target.start
445 parser.EndElementHandler = target.end
446 parser.CharacterDataHandler = target.data
447 encoding = None
448 target.xml(encoding, None)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000449
Georg Brandlcef803f2009-06-04 09:04:53 +0000450 def feed(self, data):
Serhiy Storchakaeb897462019-09-01 12:11:43 +0300451 self._parser.Parse(data, False)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000452
Georg Brandlcef803f2009-06-04 09:04:53 +0000453 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300454 try:
455 parser = self._parser
456 except AttributeError:
457 pass
458 else:
459 del self._target, self._parser # get rid of circular references
460 parser.Parse(b"", True) # end of data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000461
Fredrik Lundhb9056332001-07-11 17:42:21 +0000462# --------------------------------------------------------------------
463# XML-RPC marshalling and unmarshalling code
464
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000465##
466# XML-RPC marshaller.
467#
468# @param encoding Default encoding for 8-bit strings. The default
469# value is None (interpreted as UTF-8).
470# @see dumps
471
Fredrik Lundhb9056332001-07-11 17:42:21 +0000472class Marshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000473 """Generate an XML-RPC params chunk from a Python data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000474
Fredrik Lundhc4c062f2001-09-10 19:45:02 +0000475 Create a Marshaller instance for each set of parameters, and use
476 the "dumps" method to convert your data (represented as a tuple)
477 to an XML-RPC params chunk. To write a fault response, pass a
478 Fault instance instead. You may prefer to use the "dumps" module
479 function for this purpose.
Fred Drake1b410792001-09-04 18:55:03 +0000480 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000481
482 # by the way, if you don't understand what's going on in here,
483 # that's perfectly ok.
484
Georg Brandlfe991052009-09-16 15:54:04 +0000485 def __init__(self, encoding=None, allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000486 self.memo = {}
487 self.data = None
488 self.encoding = encoding
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000489 self.allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +0000490
Fredrik Lundhb9056332001-07-11 17:42:21 +0000491 dispatch = {}
492
493 def dumps(self, values):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000494 out = []
495 write = out.append
496 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000497 if isinstance(values, Fault):
498 # fault instance
499 write("<fault>\n")
Martin v. Löwis541342f2003-07-12 07:53:04 +0000500 dump({'faultCode': values.faultCode,
501 'faultString': values.faultString},
502 write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000503 write("</fault>\n")
504 else:
505 # parameter block
Fredrik Lundhc266bb02001-08-23 20:13:08 +0000506 # FIXME: the xml-rpc specification allows us to leave out
507 # the entire <params> block if there are no parameters.
508 # however, changing this may break older code (including
509 # old versions of xmlrpclib.py), so this is better left as
510 # is for now. See @XMLRPC3 for more information. /F
Fredrik Lundhb9056332001-07-11 17:42:21 +0000511 write("<params>\n")
512 for v in values:
513 write("<param>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000514 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000515 write("</param>\n")
516 write("</params>\n")
Neal Norwitzff113342007-04-17 08:42:15 +0000517 result = "".join(out)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000518 return result
519
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000520 def __dump(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000521 try:
522 f = self.dispatch[type(value)]
523 except KeyError:
Thomas Wouters89f507f2006-12-13 04:49:30 +0000524 # check if this object can be marshalled as a structure
Florent Xicluna93dfee12011-10-30 20:22:25 +0100525 if not hasattr(value, '__dict__'):
Collin Winterce36ad82007-08-30 01:19:48 +0000526 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000527 # check if this class is a sub-class of a basic type,
528 # because we don't know how to marshal these types
529 # (e.g. a string sub-class)
530 for type_ in type(value).__mro__:
531 if type_ in self.dispatch.keys():
Collin Winterce36ad82007-08-30 01:19:48 +0000532 raise TypeError("cannot marshal %s objects" % type(value))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000533 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
534 # for the p3yk merge, this should probably be fixed more neatly.
535 f = self.dispatch["_arbitrary_instance"]
536 f(self, value, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000537
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000538 def dump_nil (self, value, write):
539 if not self.allow_none:
Collin Winterce36ad82007-08-30 01:19:48 +0000540 raise TypeError("cannot marshal None unless allow_none is enabled")
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000541 write("<value><nil/></value>")
Guido van Rossum13257902007-06-07 23:15:56 +0000542 dispatch[type(None)] = dump_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000543
Georg Brandl38eceaa2008-05-26 11:14:17 +0000544 def dump_bool(self, value, write):
545 write("<value><boolean>")
546 write(value and "1" or "0")
547 write("</boolean></value>\n")
548 dispatch[bool] = dump_bool
Skip Montanaro9a7c96a2003-01-22 18:17:25 +0000549
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000550 def dump_long(self, value, write):
Skip Montanaro5449e082001-10-17 22:53:33 +0000551 if value > MAXINT or value < MININT:
Serhiy Storchaka95949422013-08-27 19:40:23 +0300552 raise OverflowError("int exceeds XML-RPC limits")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000553 write("<value><int>")
554 write(str(int(value)))
555 write("</int></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000556 dispatch[int] = dump_long
Skip Montanaro5e9c71b2001-10-10 15:56:34 +0000557
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200558 # backward compatible
559 dump_int = dump_long
560
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000561 def dump_double(self, value, write):
562 write("<value><double>")
563 write(repr(value))
564 write("</double></value>\n")
Guido van Rossum13257902007-06-07 23:15:56 +0000565 dispatch[float] = dump_double
Fredrik Lundhb9056332001-07-11 17:42:21 +0000566
Guido van Rossum54ad5232007-05-27 09:17:48 +0000567 def dump_unicode(self, value, write, escape=escape):
Guido van Rossum54ad5232007-05-27 09:17:48 +0000568 write("<value><string>")
569 write(escape(value))
570 write("</string></value>\n")
571 dispatch[str] = dump_unicode
Fredrik Lundhb9056332001-07-11 17:42:21 +0000572
Florent Xicluna61665192011-11-15 20:53:25 +0100573 def dump_bytes(self, value, write):
574 write("<value><base64>\n")
575 encoded = base64.encodebytes(value)
576 write(encoded.decode('ascii'))
577 write("</base64></value>\n")
578 dispatch[bytes] = dump_bytes
579 dispatch[bytearray] = dump_bytes
580
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000581 def dump_array(self, value, write):
582 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000583 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000584 raise TypeError("cannot marshal recursive sequences")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000585 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000586 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000587 write("<value><array><data>\n")
588 for v in value:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000589 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000590 write("</data></array></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000591 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000592 dispatch[tuple] = dump_array
593 dispatch[list] = dump_array
Fredrik Lundhb9056332001-07-11 17:42:21 +0000594
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000595 def dump_struct(self, value, write, escape=escape):
596 i = id(value)
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000597 if i in self.memo:
Collin Winterce36ad82007-08-30 01:19:48 +0000598 raise TypeError("cannot marshal recursive dictionaries")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000599 self.memo[i] = None
Fredrik Lundh1538c232001-10-01 19:42:03 +0000600 dump = self.__dump
Fredrik Lundhb9056332001-07-11 17:42:21 +0000601 write("<value><struct>\n")
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000602 for k, v in value.items():
Fredrik Lundhb9056332001-07-11 17:42:21 +0000603 write("<member>\n")
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000604 if not isinstance(k, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000605 raise TypeError("dictionary key must be string")
Fredrik Lundh1538c232001-10-01 19:42:03 +0000606 write("<name>%s</name>\n" % escape(k))
Andrew M. Kuchling5962f452004-06-05 12:35:58 +0000607 dump(v, write)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000608 write("</member>\n")
609 write("</struct></value>\n")
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000610 del self.memo[i]
Guido van Rossum13257902007-06-07 23:15:56 +0000611 dispatch[dict] = dump_struct
Fredrik Lundhb9056332001-07-11 17:42:21 +0000612
Florent Xiclunab6f019a2011-10-30 23:54:17 +0100613 def dump_datetime(self, value, write):
614 write("<value><dateTime.iso8601>")
615 write(_strftime(value))
616 write("</dateTime.iso8601></value>\n")
617 dispatch[datetime] = dump_datetime
Fred Drakeba613c32005-02-10 18:33:30 +0000618
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000619 def dump_instance(self, value, write):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000620 # check for special wrappers
621 if value.__class__ in WRAPPERS:
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000622 self.write = write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000623 value.encode(self)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000624 del self.write
Fredrik Lundhb9056332001-07-11 17:42:21 +0000625 else:
626 # store instance attributes as a struct (really?)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000627 self.dump_struct(value.__dict__, write)
Guido van Rossume4dea982006-04-21 13:45:00 +0000628 dispatch[DateTime] = dump_instance
629 dispatch[Binary] = dump_instance
Thomas Wouters89f507f2006-12-13 04:49:30 +0000630 # XXX(twouters): using "_arbitrary_instance" as key as a quick-fix
631 # for the p3yk merge, this should probably be fixed more neatly.
632 dispatch["_arbitrary_instance"] = dump_instance
Fredrik Lundhb9056332001-07-11 17:42:21 +0000633
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000634##
635# XML-RPC unmarshaller.
636#
637# @see loads
638
Fredrik Lundhb9056332001-07-11 17:42:21 +0000639class Unmarshaller:
Fred Drake1b410792001-09-04 18:55:03 +0000640 """Unmarshal an XML-RPC response, based on incoming XML event
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000641 messages (start, data, end). Call close() to get the resulting
Fred Drake1b410792001-09-04 18:55:03 +0000642 data structure.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000643
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000644 Note that this reader is fairly tolerant, and gladly accepts bogus
645 XML-RPC data without complaining (but not bogus XML).
Fred Drake1b410792001-09-04 18:55:03 +0000646 """
Fredrik Lundhb9056332001-07-11 17:42:21 +0000647
648 # and again, if you don't understand what's going on in here,
649 # that's perfectly ok.
650
Florent Xicluna61665192011-11-15 20:53:25 +0100651 def __init__(self, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000652 self._type = None
653 self._stack = []
654 self._marks = []
655 self._data = []
Serhiy Storchaka290fed42016-05-04 11:26:42 +0300656 self._value = False
Fredrik Lundhb9056332001-07-11 17:42:21 +0000657 self._methodname = None
658 self._encoding = "utf-8"
659 self.append = self._stack.append
Florent Xicluna61665192011-11-15 20:53:25 +0100660 self._use_datetime = use_builtin_types or use_datetime
661 self._use_bytes = use_builtin_types
Fredrik Lundhb9056332001-07-11 17:42:21 +0000662
663 def close(self):
664 # return response tuple and target method
665 if self._type is None or self._marks:
666 raise ResponseError()
667 if self._type == "fault":
Guido van Rossum68468eb2003-02-27 20:14:51 +0000668 raise Fault(**self._stack[0])
Fredrik Lundhb9056332001-07-11 17:42:21 +0000669 return tuple(self._stack)
670
671 def getmethodname(self):
672 return self._methodname
673
674 #
675 # event handlers
676
677 def xml(self, encoding, standalone):
678 self._encoding = encoding
679 # FIXME: assert standalone == 1 ???
680
681 def start(self, tag, attrs):
682 # prepare to handle this element
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300683 if ':' in tag:
684 tag = tag.split(':')[-1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000685 if tag == "array" or tag == "struct":
686 self._marks.append(len(self._stack))
687 self._data = []
Serhiy Storchaka290fed42016-05-04 11:26:42 +0300688 if self._value and tag not in self.dispatch:
689 raise ResponseError("unknown tag %r" % tag)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000690 self._value = (tag == "value")
691
692 def data(self, text):
693 self._data.append(text)
694
Neal Norwitzff113342007-04-17 08:42:15 +0000695 def end(self, tag):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000696 # call the appropriate end tag handler
697 try:
698 f = self.dispatch[tag]
699 except KeyError:
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300700 if ':' not in tag:
701 return # unknown tag ?
702 try:
703 f = self.dispatch[tag.split(':')[-1]]
704 except KeyError:
705 return # unknown tag ?
706 return f(self, "".join(self._data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000707
708 #
709 # accelerator support
710
711 def end_dispatch(self, tag, data):
712 # dispatch data
713 try:
714 f = self.dispatch[tag]
715 except KeyError:
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300716 if ':' not in tag:
717 return # unknown tag ?
718 try:
719 f = self.dispatch[tag.split(':')[-1]]
720 except KeyError:
721 return # unknown tag ?
722 return f(self, data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000723
724 #
725 # element decoders
726
727 dispatch = {}
728
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000729 def end_nil (self, data):
730 self.append(None)
731 self._value = 0
732 dispatch["nil"] = end_nil
Tim Petersc2659cf2003-05-12 20:19:37 +0000733
Fredrik Lundh1538c232001-10-01 19:42:03 +0000734 def end_boolean(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000735 if data == "0":
736 self.append(False)
737 elif data == "1":
738 self.append(True)
739 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000740 raise TypeError("bad boolean value")
Fredrik Lundhb9056332001-07-11 17:42:21 +0000741 self._value = 0
742 dispatch["boolean"] = end_boolean
743
Fredrik Lundh1538c232001-10-01 19:42:03 +0000744 def end_int(self, data):
745 self.append(int(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000746 self._value = 0
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300747 dispatch["i1"] = end_int
748 dispatch["i2"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000749 dispatch["i4"] = end_int
Georg Brandlf08a9dd2008-06-10 16:57:31 +0000750 dispatch["i8"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000751 dispatch["int"] = end_int
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300752 dispatch["biginteger"] = end_int
Fredrik Lundhb9056332001-07-11 17:42:21 +0000753
Fredrik Lundh1538c232001-10-01 19:42:03 +0000754 def end_double(self, data):
755 self.append(float(data))
Fredrik Lundhb9056332001-07-11 17:42:21 +0000756 self._value = 0
757 dispatch["double"] = end_double
Serhiy Storchaka352601c2016-09-11 11:23:38 +0300758 dispatch["float"] = end_double
759
760 def end_bigdecimal(self, data):
761 self.append(Decimal(data))
762 self._value = 0
763 dispatch["bigdecimal"] = end_bigdecimal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000764
Fredrik Lundh1538c232001-10-01 19:42:03 +0000765 def end_string(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000766 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000767 data = data.decode(self._encoding)
768 self.append(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000769 self._value = 0
770 dispatch["string"] = end_string
771 dispatch["name"] = end_string # struct keys are always strings
772
773 def end_array(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000774 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000775 # map arrays to Python lists
776 self._stack[mark:] = [self._stack[mark:]]
777 self._value = 0
778 dispatch["array"] = end_array
779
780 def end_struct(self, data):
Raymond Hettinger46ac8eb2002-06-30 03:39:14 +0000781 mark = self._marks.pop()
Fredrik Lundhb9056332001-07-11 17:42:21 +0000782 # map structs to Python dictionaries
783 dict = {}
784 items = self._stack[mark:]
785 for i in range(0, len(items), 2):
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000786 dict[items[i]] = items[i+1]
Fredrik Lundhb9056332001-07-11 17:42:21 +0000787 self._stack[mark:] = [dict]
788 self._value = 0
789 dispatch["struct"] = end_struct
790
Fredrik Lundh1538c232001-10-01 19:42:03 +0000791 def end_base64(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000792 value = Binary()
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000793 value.decode(data.encode("ascii"))
Florent Xicluna61665192011-11-15 20:53:25 +0100794 if self._use_bytes:
795 value = value.data
Fredrik Lundhb9056332001-07-11 17:42:21 +0000796 self.append(value)
797 self._value = 0
798 dispatch["base64"] = end_base64
799
Fredrik Lundh1538c232001-10-01 19:42:03 +0000800 def end_dateTime(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000801 value = DateTime()
Fredrik Lundh1538c232001-10-01 19:42:03 +0000802 value.decode(data)
Skip Montanaro174dd222005-05-14 20:54:16 +0000803 if self._use_datetime:
804 value = _datetime_type(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000805 self.append(value)
806 dispatch["dateTime.iso8601"] = end_dateTime
807
808 def end_value(self, data):
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000809 # if we stumble upon a value element with no internal
Fredrik Lundhb9056332001-07-11 17:42:21 +0000810 # elements, treat it as a string element
811 if self._value:
812 self.end_string(data)
813 dispatch["value"] = end_value
814
815 def end_params(self, data):
816 self._type = "params"
817 dispatch["params"] = end_params
818
819 def end_fault(self, data):
820 self._type = "fault"
821 dispatch["fault"] = end_fault
822
Fredrik Lundh1538c232001-10-01 19:42:03 +0000823 def end_methodName(self, data):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000824 if self._encoding:
Alexandre Vassalottie671fd22009-07-22 02:32:34 +0000825 data = data.decode(self._encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000826 self._methodname = data
827 self._type = "methodName" # no params
828 dispatch["methodName"] = end_methodName
829
Martin v. Löwis45394c22003-10-31 13:49:36 +0000830## Multicall support
831#
Fredrik Lundhb9056332001-07-11 17:42:21 +0000832
Martin v. Löwis45394c22003-10-31 13:49:36 +0000833class _MultiCallMethod:
834 # some lesser magic to store calls made to a MultiCall object
835 # for batch execution
836 def __init__(self, call_list, name):
837 self.__call_list = call_list
838 self.__name = name
839 def __getattr__(self, name):
840 return _MultiCallMethod(self.__call_list, "%s.%s" % (self.__name, name))
841 def __call__(self, *args):
842 self.__call_list.append((self.__name, args))
843
Martin v. Löwis12237b32004-08-22 16:04:50 +0000844class MultiCallIterator:
Martin v. Löwis45394c22003-10-31 13:49:36 +0000845 """Iterates over the results of a multicall. Exceptions are
Andrew Svetlov737fb892012-12-18 21:14:22 +0200846 raised in response to xmlrpc faults."""
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000847
Martin v. Löwis12237b32004-08-22 16:04:50 +0000848 def __init__(self, results):
849 self.results = results
850
851 def __getitem__(self, i):
852 item = self.results[i]
853 if type(item) == type({}):
854 raise Fault(item['faultCode'], item['faultString'])
855 elif type(item) == type([]):
856 return item[0]
Martin v. Löwis45394c22003-10-31 13:49:36 +0000857 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000858 raise ValueError("unexpected type in multicall result")
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000859
Martin v. Löwis45394c22003-10-31 13:49:36 +0000860class MultiCall:
Martin Panter7462b6492015-11-02 03:37:02 +0000861 """server -> an object used to boxcar method calls
Martin v. Löwis45394c22003-10-31 13:49:36 +0000862
863 server should be a ServerProxy object.
864
865 Methods can be added to the MultiCall using normal
866 method call syntax e.g.:
867
868 multicall = MultiCall(server_proxy)
869 multicall.add(2,3)
870 multicall.get_address("Guido")
871
872 To execute the multicall, call the MultiCall object e.g.:
873
874 add_result, address = multicall()
875 """
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000876
Martin v. Löwis45394c22003-10-31 13:49:36 +0000877 def __init__(self, server):
878 self.__server = server
879 self.__call_list = []
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000880
Martin v. Löwis45394c22003-10-31 13:49:36 +0000881 def __repr__(self):
Serhiy Storchaka465e60e2014-07-25 23:36:00 +0300882 return "<%s at %#x>" % (self.__class__.__name__, id(self))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000883
Martin v. Löwis45394c22003-10-31 13:49:36 +0000884 def __getattr__(self, name):
885 return _MultiCallMethod(self.__call_list, name)
886
887 def __call__(self):
888 marshalled_list = []
889 for name, args in self.__call_list:
890 marshalled_list.append({'methodName' : name, 'params' : args})
891
892 return MultiCallIterator(self.__server.system.multicall(marshalled_list))
Raymond Hettingercc523fc2003-11-02 09:47:05 +0000893
Fredrik Lundhb9056332001-07-11 17:42:21 +0000894# --------------------------------------------------------------------
895# convenience functions
896
Georg Brandl38eceaa2008-05-26 11:14:17 +0000897FastMarshaller = FastParser = FastUnmarshaller = None
898
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000899##
900# Create a parser object, and connect it to an unmarshalling instance.
901# This function picks the fastest available XML parser.
902#
903# return A (parser, unmarshaller) tuple.
904
Florent Xicluna61665192011-11-15 20:53:25 +0100905def getparser(use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000906 """getparser() -> parser, unmarshaller
907
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000908 Create an instance of the fastest available parser, and attach it
909 to an unmarshalling object. Return both objects.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000910 """
911 if FastParser and FastUnmarshaller:
Florent Xicluna61665192011-11-15 20:53:25 +0100912 if use_builtin_types:
Skip Montanaro174dd222005-05-14 20:54:16 +0000913 mkdatetime = _datetime_type
Florent Xicluna61665192011-11-15 20:53:25 +0100914 mkbytes = base64.decodebytes
915 elif use_datetime:
916 mkdatetime = _datetime_type
917 mkbytes = _binary
Skip Montanaro174dd222005-05-14 20:54:16 +0000918 else:
919 mkdatetime = _datetime
Florent Xicluna61665192011-11-15 20:53:25 +0100920 mkbytes = _binary
921 target = FastUnmarshaller(True, False, mkbytes, mkdatetime, Fault)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000922 parser = FastParser(target)
923 else:
Florent Xicluna61665192011-11-15 20:53:25 +0100924 target = Unmarshaller(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000925 if FastParser:
926 parser = FastParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000927 else:
Georg Brandlcef803f2009-06-04 09:04:53 +0000928 parser = ExpatParser(target)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000929 return parser, target
930
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000931##
932# Convert a Python tuple or a Fault instance to an XML-RPC packet.
933#
934# @def dumps(params, **options)
935# @param params A tuple or Fault instance.
936# @keyparam methodname If given, create a methodCall request for
937# this method name.
938# @keyparam methodresponse If given, create a methodResponse packet.
939# If used with a tuple, the tuple must be a singleton (that is,
940# it must contain exactly one element).
941# @keyparam encoding The packet encoding.
942# @return A string containing marshalled data.
943
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000944def dumps(params, methodname=None, methodresponse=None, encoding=None,
Georg Brandlfe991052009-09-16 15:54:04 +0000945 allow_none=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000946 """data [,options] -> marshalled data
947
948 Convert an argument tuple or a Fault instance to an XML-RPC
949 request (or response, if the methodresponse option is used).
950
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000951 In addition to the data object, the following options can be given
952 as keyword arguments:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000953
954 methodname: the method name for a methodCall packet
955
956 methodresponse: true to create a methodResponse packet.
957 If this option is used with a tuple, the tuple must be
958 a singleton (i.e. it can contain only one element).
959
960 encoding: the packet encoding (default is UTF-8)
961
Florent Xicluna61665192011-11-15 20:53:25 +0100962 All byte strings in the data structure are assumed to use the
Fredrik Lundhb9056332001-07-11 17:42:21 +0000963 packet encoding. Unicode strings are automatically converted,
Fredrik Lundhb0e8e9b2001-09-10 21:45:42 +0000964 where necessary.
Fredrik Lundhb9056332001-07-11 17:42:21 +0000965 """
966
Guido van Rossum13257902007-06-07 23:15:56 +0000967 assert isinstance(params, (tuple, Fault)), "argument must be tuple or Fault instance"
Fredrik Lundhb9056332001-07-11 17:42:21 +0000968 if isinstance(params, Fault):
969 methodresponse = 1
Guido van Rossum13257902007-06-07 23:15:56 +0000970 elif methodresponse and isinstance(params, tuple):
Fredrik Lundhb9056332001-07-11 17:42:21 +0000971 assert len(params) == 1, "response tuple must be a singleton"
972
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000973 if not encoding:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000974 encoding = "utf-8"
975
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000976 if FastMarshaller:
977 m = FastMarshaller(encoding)
978 else:
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +0000979 m = Marshaller(encoding, allow_none)
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000980
Fredrik Lundhb9056332001-07-11 17:42:21 +0000981 data = m.dumps(params)
982
983 if encoding != "utf-8":
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000984 xmlheader = "<?xml version='1.0' encoding='%s'?>\n" % str(encoding)
Fredrik Lundhb9056332001-07-11 17:42:21 +0000985 else:
986 xmlheader = "<?xml version='1.0'?>\n" # utf-8 is default
987
988 # standard XML-RPC wrappings
Fredrik Lundh3d9addd2002-06-27 21:36:21 +0000989 if methodname:
Fredrik Lundhb9056332001-07-11 17:42:21 +0000990 # a method call
Fredrik Lundhb9056332001-07-11 17:42:21 +0000991 data = (
992 xmlheader,
993 "<methodCall>\n"
994 "<methodName>", methodname, "</methodName>\n",
995 data,
996 "</methodCall>\n"
997 )
998 elif methodresponse:
999 # a method response, or a fault structure
1000 data = (
1001 xmlheader,
1002 "<methodResponse>\n",
1003 data,
1004 "</methodResponse>\n"
1005 )
1006 else:
1007 return data # return as is
Neal Norwitzff113342007-04-17 08:42:15 +00001008 return "".join(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001009
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001010##
1011# Convert an XML-RPC packet to a Python object. If the XML-RPC packet
1012# represents a fault condition, this function raises a Fault exception.
1013#
1014# @param data An XML-RPC packet, given as an 8-bit string.
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00001015# @return A tuple containing the unpacked data, and the method name
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001016# (None if not present).
1017# @see Fault
1018
Florent Xicluna61665192011-11-15 20:53:25 +01001019def loads(data, use_datetime=False, use_builtin_types=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001020 """data -> unmarshalled data, method name
1021
1022 Convert an XML-RPC packet to unmarshalled data plus a method
1023 name (None if not present).
1024
1025 If the XML-RPC packet represents a fault condition, this function
1026 raises a Fault exception.
1027 """
Florent Xicluna61665192011-11-15 20:53:25 +01001028 p, u = getparser(use_datetime=use_datetime, use_builtin_types=use_builtin_types)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001029 p.feed(data)
1030 p.close()
1031 return u.close(), u.getmethodname()
1032
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001033##
1034# Encode 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 unencoded data
1039# @return the encoded data
1040
1041def gzip_encode(data):
1042 """data -> gzip encoded data
1043
1044 Encode data using the gzip content encoding as described in RFC 1952
1045 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001046 if not gzip:
1047 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001048 f = BytesIO()
Serhiy Storchaka46ba6c82015-04-04 11:01:02 +03001049 with gzip.GzipFile(mode="wb", fileobj=f, compresslevel=1) as gzf:
1050 gzf.write(data)
1051 return f.getvalue()
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001052
1053##
1054# Decode a string using the gzip content encoding such as specified by the
1055# Content-Encoding: gzip
1056# in the HTTP header, as described in RFC 1952
1057#
1058# @param data The encoded data
Victor Stinner8c663fd2017-11-08 14:44:44 -08001059# @keyparam max_decode Maximum bytes to decode (20 MiB default), use negative
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001060# values for unlimited decoding
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001061# @return the unencoded data
1062# @raises ValueError if data is not correctly coded.
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001063# @raises ValueError if max gzipped payload length exceeded
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001064
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001065def gzip_decode(data, max_decode=20971520):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001066 """gzip encoded data -> unencoded data
1067
1068 Decode data using the gzip content encoding as described in RFC 1952
1069 """
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001070 if not gzip:
1071 raise NotImplementedError
Serhiy Storchaka46ba6c82015-04-04 11:01:02 +03001072 with gzip.GzipFile(mode="rb", fileobj=BytesIO(data)) as gzf:
1073 try:
1074 if max_decode < 0: # no limit
1075 decoded = gzf.read()
1076 else:
1077 decoded = gzf.read(max_decode + 1)
1078 except OSError:
1079 raise ValueError("invalid data")
Benjamin Peterson4e9cefa2014-12-05 20:15:15 -05001080 if max_decode >= 0 and len(decoded) > max_decode:
1081 raise ValueError("max gzipped payload length exceeded")
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001082 return decoded
1083
1084##
1085# Return a decoded file-like object for the gzip encoding
1086# as described in RFC 1952.
1087#
1088# @param response A stream supporting a read() method
1089# @return a file-like object that the decoded data can be read() from
1090
Kristján Valur Jónsson9ab07312009-07-19 22:38:38 +00001091class GzipDecodedResponse(gzip.GzipFile if gzip else object):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001092 """a file-like object to decode a response encoded with the gzip
1093 method, as described in RFC 1952.
1094 """
1095 def __init__(self, response):
1096 #response doesn't support tell() and read(), required by
1097 #GzipFile
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001098 if not gzip:
1099 raise NotImplementedError
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001100 self.io = BytesIO(response.read())
1101 gzip.GzipFile.__init__(self, mode="rb", fileobj=self.io)
1102
1103 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001104 try:
1105 gzip.GzipFile.close(self)
1106 finally:
1107 self.io.close()
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001108
Fredrik Lundhb9056332001-07-11 17:42:21 +00001109
1110# --------------------------------------------------------------------
1111# request dispatcher
1112
1113class _Method:
1114 # some magic to bind an XML-RPC method to an RPC server.
1115 # supports "nested" methods (e.g. examples.getStateName)
1116 def __init__(self, send, name):
1117 self.__send = send
1118 self.__name = name
1119 def __getattr__(self, name):
1120 return _Method(self.__send, "%s.%s" % (self.__name, name))
1121 def __call__(self, *args):
1122 return self.__send(self.__name, args)
1123
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001124##
1125# Standard transport class for XML-RPC over HTTP.
1126# <p>
1127# You can create custom transports by subclassing this method, and
1128# overriding selected methods.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001129
1130class Transport:
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001131 """Handles an HTTP transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001132
1133 # client identifier (may be overridden)
Florent Xicluna75861df2011-10-30 20:39:24 +01001134 user_agent = "Python-xmlrpc/%s" % __version__
Fredrik Lundhb9056332001-07-11 17:42:21 +00001135
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001136 #if true, we'll request gzip encoding
1137 accept_gzip_encoding = True
1138
1139 # if positive, encode request using gzip if it exceeds this threshold
Mike53f7a7c2017-12-14 14:04:53 +03001140 # note that many servers will get confused, so only use it if you know
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001141 # that they can decode such a request
1142 encode_threshold = None #None = don't encode
1143
Cédric Krierbeda52e2019-02-19 17:18:50 +01001144 def __init__(self, use_datetime=False, use_builtin_types=False,
1145 *, headers=()):
Skip Montanaro174dd222005-05-14 20:54:16 +00001146 self._use_datetime = use_datetime
Florent Xicluna61665192011-11-15 20:53:25 +01001147 self._use_builtin_types = use_builtin_types
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001148 self._connection = (None, None)
Cédric Krierbeda52e2019-02-19 17:18:50 +01001149 self._headers = list(headers)
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001150 self._extra_headers = []
Skip Montanaro174dd222005-05-14 20:54:16 +00001151
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001152 ##
1153 # Send a complete request, and parse the response.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001154 # Retry request if a cached connection has disconnected.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001155 #
1156 # @param host Target host.
1157 # @param handler Target PRC handler.
1158 # @param request_body XML-RPC request body.
1159 # @param verbose Debugging flag.
1160 # @return Parsed response.
1161
Georg Brandlfe991052009-09-16 15:54:04 +00001162 def request(self, host, handler, request_body, verbose=False):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001163 #retry request once if cached connection has gone cold
1164 for i in (0, 1):
1165 try:
1166 return self.single_request(host, handler, request_body, verbose)
Martin Pantereae33362016-02-25 11:53:40 +00001167 except http.client.RemoteDisconnected:
1168 if i:
1169 raise
Andrew Svetlov0832af62012-12-18 23:10:48 +02001170 except OSError as e:
1171 if i or e.errno not in (errno.ECONNRESET, errno.ECONNABORTED,
1172 errno.EPIPE):
Kristján Valur Jónsson43535d92009-07-03 23:23:50 +00001173 raise
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001174
Georg Brandlfe991052009-09-16 15:54:04 +00001175 def single_request(self, host, handler, request_body, verbose=False):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001176 # issue XML-RPC request
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001177 try:
1178 http_conn = self.send_request(host, handler, request_body, verbose)
1179 resp = http_conn.getresponse()
1180 if resp.status == 200:
1181 self.verbose = verbose
1182 return self.parse_response(resp)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001183
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001184 except Fault:
1185 raise
1186 except Exception:
1187 #All unexpected errors leave connection in
1188 # a strange state, so we clear it.
1189 self.close()
1190 raise
Fredrik Lundhb9056332001-07-11 17:42:21 +00001191
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001192 #We got an error response.
1193 #Discard any response data and raise exception
1194 if resp.getheader("content-length", ""):
1195 resp.read()
1196 raise ProtocolError(
1197 host + handler,
1198 resp.status, resp.reason,
1199 dict(resp.getheaders())
1200 )
Fredrik Lundhb9056332001-07-11 17:42:21 +00001201
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001202
1203 ##
1204 # Create parser.
1205 #
Martin Panter7462b6492015-11-02 03:37:02 +00001206 # @return A 2-tuple containing a parser and an unmarshaller.
Fredrik Lundhb9056332001-07-11 17:42:21 +00001207
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001208 def getparser(self):
1209 # get parser and unmarshaller
Florent Xicluna61665192011-11-15 20:53:25 +01001210 return getparser(use_datetime=self._use_datetime,
1211 use_builtin_types=self._use_builtin_types)
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001212
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001213 ##
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001214 # Get authorization info from host parameter
1215 # Host may be a string, or a (host, x509-dict) tuple; if a string,
1216 # it is checked for a "user:pw@host" format, and a "Basic
1217 # Authentication" header is added if appropriate.
1218 #
1219 # @param host Host descriptor (URL or (URL, x509 info) tuple).
1220 # @return A 3-tuple containing (actual host, extra headers,
1221 # x509 info). The header and x509 fields may be None.
1222
1223 def get_host_info(self, host):
1224
1225 x509 = {}
Guido van Rossum13257902007-06-07 23:15:56 +00001226 if isinstance(host, tuple):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001227 host, x509 = host
1228
Cheryl Sabella0250de42018-04-25 16:51:54 -07001229 auth, host = urllib.parse._splituser(host)
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001230
1231 if auth:
Georg Brandlc8dcfb62009-02-13 10:50:01 +00001232 auth = urllib.parse.unquote_to_bytes(auth)
Georg Brandlb54d8012009-06-04 09:11:51 +00001233 auth = base64.encodebytes(auth).decode("utf-8")
Neal Norwitzff113342007-04-17 08:42:15 +00001234 auth = "".join(auth.split()) # get rid of whitespace
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001235 extra_headers = [
1236 ("Authorization", "Basic " + auth)
1237 ]
1238 else:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001239 extra_headers = []
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001240
1241 return host, extra_headers, x509
1242
1243 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001244 # Connect to server.
1245 #
1246 # @param host Target host.
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001247 # @return An HTTPConnection object
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001248
Fredrik Lundhb9056332001-07-11 17:42:21 +00001249 def make_connection(self, host):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001250 #return an existing connection if possible. This allows
1251 #HTTP/1.1 keep-alive.
1252 if self._connection and host == self._connection[0]:
1253 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001254 # create a HTTP connection object from a host descriptor
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001255 chost, self._extra_headers, x509 = self.get_host_info(host)
1256 self._connection = host, http.client.HTTPConnection(chost)
1257 return self._connection[1]
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001258
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001259 ##
1260 # Clear any cached connection object.
1261 # Used in the event of socket errors.
1262 #
1263 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001264 host, connection = self._connection
1265 if connection:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001266 self._connection = (None, None)
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001267 connection.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001268
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001269 ##
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001270 # Send HTTP request.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001271 #
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001272 # @param host Host descriptor (URL or (URL, x509 info) tuple).
Mike53f7a7c2017-12-14 14:04:53 +03001273 # @param handler Target RPC handler (a path relative to host)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001274 # @param request_body The XML-RPC request body
1275 # @param debug Enable debugging if debug is true.
1276 # @return An HTTPConnection.
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001277
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001278 def send_request(self, host, handler, request_body, debug):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001279 connection = self.make_connection(host)
Cédric Krierbeda52e2019-02-19 17:18:50 +01001280 headers = self._headers + self._extra_headers
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001281 if debug:
1282 connection.set_debuglevel(1)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001283 if self.accept_gzip_encoding and gzip:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001284 connection.putrequest("POST", handler, skip_accept_encoding=True)
1285 headers.append(("Accept-Encoding", "gzip"))
1286 else:
1287 connection.putrequest("POST", handler)
1288 headers.append(("Content-Type", "text/xml"))
1289 headers.append(("User-Agent", self.user_agent))
1290 self.send_headers(connection, headers)
1291 self.send_content(connection, request_body)
Jeremy Hylton5d8a88a2007-08-14 16:47:39 +00001292 return connection
Fredrik Lundhb9056332001-07-11 17:42:21 +00001293
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001294 ##
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001295 # Send request headers.
1296 # This function provides a useful hook for subclassing
1297 #
1298 # @param connection httpConnection.
1299 # @param headers list of key,value pairs for HTTP headers
1300
1301 def send_headers(self, connection, headers):
1302 for key, val in headers:
1303 connection.putheader(key, val)
1304
1305 ##
1306 # Send request body.
1307 # This function provides a useful hook for subclassing
1308 #
1309 # @param connection httpConnection.
1310 # @param request_body XML-RPC request body.
1311
1312 def send_content(self, connection, request_body):
1313 #optionally encode the request
1314 if (self.encode_threshold is not None and
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +00001315 self.encode_threshold < len(request_body) and
1316 gzip):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001317 connection.putheader("Content-Encoding", "gzip")
1318 request_body = gzip_encode(request_body)
1319
1320 connection.putheader("Content-Length", str(len(request_body)))
1321 connection.endheaders(request_body)
1322
1323 ##
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001324 # Parse response.
1325 #
1326 # @param file Stream.
1327 # @return Response tuple and target method.
1328
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001329 def parse_response(self, response):
1330 # read response data from httpresponse, and parse it
Senthil Kumaranf34445f2010-12-08 08:04:49 +00001331 # Check for new http response object, otherwise it is a file object.
1332 if hasattr(response, 'getheader'):
1333 if response.getheader("Content-Encoding", "") == "gzip":
1334 stream = GzipDecodedResponse(response)
1335 else:
1336 stream = response
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001337 else:
1338 stream = response
Fredrik Lundhb9056332001-07-11 17:42:21 +00001339
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001340 p, u = self.getparser()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001341
1342 while 1:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001343 data = stream.read(1024)
1344 if not data:
Fredrik Lundhb9056332001-07-11 17:42:21 +00001345 break
1346 if self.verbose:
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001347 print("body:", repr(data))
1348 p.feed(data)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001349
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001350 if stream is not response:
1351 stream.close()
Fredrik Lundhb9056332001-07-11 17:42:21 +00001352 p.close()
1353
1354 return u.close()
1355
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001356##
1357# Standard transport class for XML-RPC over HTTPS.
1358
Fredrik Lundhb9056332001-07-11 17:42:21 +00001359class SafeTransport(Transport):
Fredrik Lundhc4c062f2001-09-10 19:45:02 +00001360 """Handles an HTTPS transaction to an XML-RPC server."""
Fredrik Lundhb9056332001-07-11 17:42:21 +00001361
Cédric Krierbeda52e2019-02-19 17:18:50 +01001362 def __init__(self, use_datetime=False, use_builtin_types=False,
1363 *, headers=(), context=None):
1364 super().__init__(use_datetime=use_datetime,
1365 use_builtin_types=use_builtin_types,
1366 headers=headers)
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001367 self.context = context
1368
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001369 # FIXME: mostly untested
1370
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001371 def make_connection(self, host):
1372 if self._connection and host == self._connection[0]:
1373 return self._connection[1]
1374
Senthil Kumaran6a0b5c42010-11-18 17:08:48 +00001375 if not hasattr(http.client, "HTTPSConnection"):
Fredrik Lundh1303c7c2002-10-22 18:23:00 +00001376 raise NotImplementedError(
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001377 "your version of http.client doesn't support HTTPS")
1378 # create a HTTPS connection object from a host descriptor
1379 # host may be a string, or a (host, x509-dict) tuple
1380 chost, self._extra_headers, x509 = self.get_host_info(host)
1381 self._connection = host, http.client.HTTPSConnection(chost,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001382 None, context=self.context, **(x509 or {}))
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001383 return self._connection[1]
Fredrik Lundhb9056332001-07-11 17:42:21 +00001384
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001385##
1386# Standard server proxy. This class establishes a virtual connection
1387# to an XML-RPC server.
1388# <p>
1389# This class is available as ServerProxy and Server. New code should
1390# use ServerProxy, to avoid confusion.
1391#
1392# @def ServerProxy(uri, **options)
1393# @param uri The connection point on the server.
1394# @keyparam transport A transport factory, compatible with the
1395# standard transport class.
1396# @keyparam encoding The default encoding used for 8-bit strings
1397# (default is UTF-8).
1398# @keyparam verbose Use a true value to enable debugging output.
1399# (printed to standard output).
1400# @see Transport
1401
Fredrik Lundhb9056332001-07-11 17:42:21 +00001402class ServerProxy:
1403 """uri [,options] -> a logical connection to an XML-RPC server
1404
1405 uri is the connection point on the server, given as
1406 scheme://host/target.
1407
1408 The standard implementation always supports the "http" scheme. If
1409 SSL socket support is available (Python 2.0), it also supports
1410 "https".
1411
1412 If the target part and the slash preceding it are both omitted,
1413 "/RPC2" is assumed.
1414
1415 The following options can be given as keyword arguments:
1416
1417 transport: a transport factory
1418 encoding: the request encoding (default is UTF-8)
1419
1420 All 8-bit strings passed to the server proxy are assumed to use
1421 the given encoding.
1422 """
1423
Georg Brandlfe991052009-09-16 15:54:04 +00001424 def __init__(self, uri, transport=None, encoding=None, verbose=False,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001425 allow_none=False, use_datetime=False, use_builtin_types=False,
Cédric Krierbeda52e2019-02-19 17:18:50 +01001426 *, headers=(), context=None):
Fredrik Lundhb9056332001-07-11 17:42:21 +00001427 # establish a "logical" server connection
1428
1429 # get the url
Serhiy Storchakac1b073a2021-03-29 16:39:31 +03001430 p = urllib.parse.urlsplit(uri)
Dong-hee Na9c4c4592019-09-08 17:54:02 +09001431 if p.scheme not in ("http", "https"):
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001432 raise OSError("unsupported XML-RPC protocol")
Dong-hee Na9c4c4592019-09-08 17:54:02 +09001433 self.__host = p.netloc
Serhiy Storchakac1b073a2021-03-29 16:39:31 +03001434 self.__handler = urllib.parse.urlunsplit(["", "", *p[2:]])
1435 if not self.__handler:
1436 self.__handler = "/RPC2"
Fredrik Lundhb9056332001-07-11 17:42:21 +00001437
1438 if transport is None:
Dong-hee Na9c4c4592019-09-08 17:54:02 +09001439 if p.scheme == "https":
Florent Xicluna61665192011-11-15 20:53:25 +01001440 handler = SafeTransport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001441 extra_kwargs = {"context": context}
Fredrik Lundhb9056332001-07-11 17:42:21 +00001442 else:
Florent Xicluna61665192011-11-15 20:53:25 +01001443 handler = Transport
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001444 extra_kwargs = {}
Florent Xicluna61665192011-11-15 20:53:25 +01001445 transport = handler(use_datetime=use_datetime,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001446 use_builtin_types=use_builtin_types,
Cédric Krierbeda52e2019-02-19 17:18:50 +01001447 headers=headers,
Benjamin Petersonc1da3d12014-11-29 23:32:57 -05001448 **extra_kwargs)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001449 self.__transport = transport
1450
Senthil Kumaranb3af08f2009-04-01 20:20:43 +00001451 self.__encoding = encoding or 'utf-8'
Fredrik Lundhb9056332001-07-11 17:42:21 +00001452 self.__verbose = verbose
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001453 self.__allow_none = allow_none
Tim Petersc2659cf2003-05-12 20:19:37 +00001454
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001455 def __close(self):
1456 self.__transport.close()
1457
Fredrik Lundhb9056332001-07-11 17:42:21 +00001458 def __request(self, methodname, params):
1459 # call a method on the remote server
1460
Andrew M. Kuchlinga4c2b742003-04-25 00:26:51 +00001461 request = dumps(params, methodname, encoding=self.__encoding,
Serhiy Storchakaaebb6d32016-01-20 10:34:27 +02001462 allow_none=self.__allow_none).encode(self.__encoding, 'xmlcharrefreplace')
Fredrik Lundhb9056332001-07-11 17:42:21 +00001463
1464 response = self.__transport.request(
1465 self.__host,
1466 self.__handler,
1467 request,
1468 verbose=self.__verbose
1469 )
1470
1471 if len(response) == 1:
1472 response = response[0]
1473
1474 return response
1475
1476 def __repr__(self):
1477 return (
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001478 "<%s for %s%s>" %
1479 (self.__class__.__name__, self.__host, self.__handler)
Fredrik Lundhb9056332001-07-11 17:42:21 +00001480 )
1481
Fredrik Lundhb9056332001-07-11 17:42:21 +00001482 def __getattr__(self, name):
1483 # magic method dispatcher
1484 return _Method(self.__request, name)
1485
Serhiy Storchaka6a7b3a72016-04-17 08:32:47 +03001486 # note: to call a remote object with a non-standard name, use
Fredrik Lundhb9056332001-07-11 17:42:21 +00001487 # result getattr(server, "strange-python-name")(args)
1488
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +00001489 def __call__(self, attr):
1490 """A workaround to get special attributes on the ServerProxy
1491 without interfering with the magic __getattr__
1492 """
1493 if attr == "close":
1494 return self.__close
1495 elif attr == "transport":
1496 return self.__transport
1497 raise AttributeError("Attribute %r not found" % (attr,))
1498
Brett Cannon33a40002014-03-21 11:24:40 -04001499 def __enter__(self):
1500 return self
1501
1502 def __exit__(self, *args):
1503 self.__close()
1504
Fredrik Lundh78eedce2001-08-23 20:04:33 +00001505# compatibility
Fredrik Lundh3d9addd2002-06-27 21:36:21 +00001506
Fredrik Lundhb9056332001-07-11 17:42:21 +00001507Server = ServerProxy
1508
1509# --------------------------------------------------------------------
1510# test code
1511
1512if __name__ == "__main__":
1513
1514 # simple test program (from the XML-RPC specification)
1515
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001516 # local server, available from Lib/xmlrpc/server.py
1517 server = ServerProxy("http://localhost:8000")
Fredrik Lundhb9056332001-07-11 17:42:21 +00001518
Fredrik Lundhb9056332001-07-11 17:42:21 +00001519 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001520 print(server.currentTime.getCurrentTime())
Guido van Rossumb940e112007-01-10 16:19:56 +00001521 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001522 print("ERROR", v)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001523
1524 multi = MultiCall(server)
Senthil Kumaran939e2db2014-01-12 16:06:58 -08001525 multi.getData()
1526 multi.pow(2,9)
1527 multi.add(1,2)
Martin v. Löwis12237b32004-08-22 16:04:50 +00001528 try:
1529 for response in multi():
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001530 print(response)
Guido van Rossumb940e112007-01-10 16:19:56 +00001531 except Error as v:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001532 print("ERROR", v)