blob: aff5fe36ca38b4c78dd7e0b21b4a01636c883918 [file] [log] [blame]
Benjamin Petersonef3e4c22009-04-11 19:48:14 +00001r"""plistlib.py -- a tool to generate and parse MacOSX .plist files.
Christian Heimes7e182542008-01-27 15:20:13 +00002
Ezio Melotti6e9b1df2009-09-16 00:49:03 +00003The property list (.plist) file format is a simple XML pickle supporting
Christian Heimes7e182542008-01-27 15:20:13 +00004basic object types, like dictionaries, lists, numbers and strings.
5Usually the top level object is a dictionary.
6
Ronald Oussorenc5cf7972013-11-21 15:46:49 +01007To write out a plist file, use the dump(value, file)
8function. 'value' is the top level object, 'file' is
9a (writable) file object.
Christian Heimes7e182542008-01-27 15:20:13 +000010
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010011To parse a plist from a file, use the load(file) function,
12with a (readable) file object as the only argument. It
Christian Heimes7e182542008-01-27 15:20:13 +000013returns the top level object (again, usually a dictionary).
14
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010015To work with plist data in bytes objects, you can use loads()
16and dumps().
Christian Heimes7e182542008-01-27 15:20:13 +000017
18Values can be strings, integers, floats, booleans, tuples, lists,
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010019dictionaries (but only with string keys), Data, bytes, bytearray, or
20datetime.datetime objects.
Christian Heimes7e182542008-01-27 15:20:13 +000021
22Generate Plist example:
23
24 pl = dict(
Ezio Melotti6e9b1df2009-09-16 00:49:03 +000025 aString = "Doodah",
26 aList = ["A", "B", 12, 32.1, [1, 2, 3]],
Christian Heimes7e182542008-01-27 15:20:13 +000027 aFloat = 0.1,
28 anInt = 728,
Ezio Melotti6e9b1df2009-09-16 00:49:03 +000029 aDict = dict(
30 anotherString = "<hello & hi there!>",
31 aUnicodeValue = "M\xe4ssig, Ma\xdf",
32 aTrueValue = True,
33 aFalseValue = False,
Christian Heimes7e182542008-01-27 15:20:13 +000034 ),
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010035 someData = b"<binary gunk>",
36 someMoreData = b"<lots of binary gunk>" * 10,
Christian Heimes7e182542008-01-27 15:20:13 +000037 aDate = datetime.datetime.fromtimestamp(time.mktime(time.gmtime())),
38 )
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010039 with open(fileName, 'wb') as fp:
40 dump(pl, fp)
Christian Heimes7e182542008-01-27 15:20:13 +000041
42Parse Plist example:
43
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010044 with open(fileName, 'rb') as fp:
45 pl = load(fp)
46 print(pl["aKey"])
Christian Heimes7e182542008-01-27 15:20:13 +000047"""
Christian Heimes7e182542008-01-27 15:20:13 +000048__all__ = [
Jon Janzence81a922019-09-05 03:11:35 -050049 "InvalidFileException", "FMT_XML", "FMT_BINARY", "load", "dump", "loads", "dumps", "UID"
Christian Heimes7e182542008-01-27 15:20:13 +000050]
Christian Heimes7e182542008-01-27 15:20:13 +000051
52import binascii
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010053import codecs
Christian Heimes7e182542008-01-27 15:20:13 +000054import datetime
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010055import enum
Christian Heimes7e182542008-01-27 15:20:13 +000056from io import BytesIO
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010057import itertools
58import os
Christian Heimes7e182542008-01-27 15:20:13 +000059import re
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010060import struct
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010061from xml.parsers.expat import ParserCreate
Christian Heimes7e182542008-01-27 15:20:13 +000062
63
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010064PlistFormat = enum.Enum('PlistFormat', 'FMT_XML FMT_BINARY', module=__name__)
65globals().update(PlistFormat.__members__)
Christian Heimes7e182542008-01-27 15:20:13 +000066
67
Jon Janzenc981ad12019-05-15 22:14:38 +020068class UID:
69 def __init__(self, data):
70 if not isinstance(data, int):
71 raise TypeError("data must be an int")
72 if data >= 1 << 64:
73 raise ValueError("UIDs cannot be >= 2**64")
74 if data < 0:
75 raise ValueError("UIDs must be positive")
76 self.data = data
77
78 def __index__(self):
79 return self.data
80
81 def __repr__(self):
82 return "%s(%s)" % (self.__class__.__name__, repr(self.data))
83
84 def __reduce__(self):
85 return self.__class__, (self.data,)
86
87 def __eq__(self, other):
88 if not isinstance(other, UID):
89 return NotImplemented
90 return self.data == other.data
91
92 def __hash__(self):
93 return hash(self.data)
94
Ronald Oussorenc5cf7972013-11-21 15:46:49 +010095#
96# XML support
97#
98
99
100# XML 'header'
101PLISTHEADER = b"""\
102<?xml version="1.0" encoding="UTF-8"?>
103<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
104"""
105
106
107# Regex to find any control chars, except for \t \n and \r
108_controlCharPat = re.compile(
109 r"[\x00\x01\x02\x03\x04\x05\x06\x07\x08\x0b\x0c\x0e\x0f"
110 r"\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f]")
111
112def _encode_base64(s, maxlinelength=76):
113 # copied from base64.encodebytes(), with added maxlinelength argument
114 maxbinsize = (maxlinelength//4)*3
115 pieces = []
116 for i in range(0, len(s), maxbinsize):
117 chunk = s[i : i + maxbinsize]
118 pieces.append(binascii.b2a_base64(chunk))
119 return b''.join(pieces)
120
121def _decode_base64(s):
122 if isinstance(s, str):
123 return binascii.a2b_base64(s.encode("utf-8"))
124
125 else:
126 return binascii.a2b_base64(s)
127
128# Contents should conform to a subset of ISO 8601
129# (in particular, YYYY '-' MM '-' DD 'T' HH ':' MM ':' SS 'Z'. Smaller units
130# may be omitted with # a loss of precision)
131_dateParser = re.compile(r"(?P<year>\d\d\d\d)(?:-(?P<month>\d\d)(?:-(?P<day>\d\d)(?:T(?P<hour>\d\d)(?::(?P<minute>\d\d)(?::(?P<second>\d\d))?)?)?)?)?Z", re.ASCII)
132
133
134def _date_from_string(s):
135 order = ('year', 'month', 'day', 'hour', 'minute', 'second')
136 gd = _dateParser.match(s).groupdict()
137 lst = []
138 for key in order:
139 val = gd[key]
140 if val is None:
141 break
142 lst.append(int(val))
143 return datetime.datetime(*lst)
144
145
146def _date_to_string(d):
147 return '%04d-%02d-%02dT%02d:%02d:%02dZ' % (
148 d.year, d.month, d.day,
149 d.hour, d.minute, d.second
150 )
151
152def _escape(text):
153 m = _controlCharPat.search(text)
154 if m is not None:
155 raise ValueError("strings can't contains control characters; "
156 "use bytes instead")
157 text = text.replace("\r\n", "\n") # convert DOS line endings
158 text = text.replace("\r", "\n") # convert Mac line endings
159 text = text.replace("&", "&amp;") # escape '&'
160 text = text.replace("<", "&lt;") # escape '<'
161 text = text.replace(">", "&gt;") # escape '>'
162 return text
163
164class _PlistParser:
Jon Janzence81a922019-09-05 03:11:35 -0500165 def __init__(self, dict_type):
Christian Heimes7e182542008-01-27 15:20:13 +0000166 self.stack = []
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100167 self.current_key = None
Christian Heimes7e182542008-01-27 15:20:13 +0000168 self.root = None
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100169 self._dict_type = dict_type
Christian Heimes7e182542008-01-27 15:20:13 +0000170
171 def parse(self, fileobj):
Ned Deilyb8e59f72011-05-28 02:19:19 -0700172 self.parser = ParserCreate()
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100173 self.parser.StartElementHandler = self.handle_begin_element
174 self.parser.EndElementHandler = self.handle_end_element
175 self.parser.CharacterDataHandler = self.handle_data
Ned Deilyb8e59f72011-05-28 02:19:19 -0700176 self.parser.ParseFile(fileobj)
Christian Heimes7e182542008-01-27 15:20:13 +0000177 return self.root
178
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100179 def handle_begin_element(self, element, attrs):
Christian Heimes7e182542008-01-27 15:20:13 +0000180 self.data = []
181 handler = getattr(self, "begin_" + element, None)
182 if handler is not None:
183 handler(attrs)
184
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100185 def handle_end_element(self, element):
Christian Heimes7e182542008-01-27 15:20:13 +0000186 handler = getattr(self, "end_" + element, None)
187 if handler is not None:
188 handler()
189
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100190 def handle_data(self, data):
Christian Heimes7e182542008-01-27 15:20:13 +0000191 self.data.append(data)
192
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100193 def add_object(self, value):
194 if self.current_key is not None:
Ned Deilyb8e59f72011-05-28 02:19:19 -0700195 if not isinstance(self.stack[-1], type({})):
196 raise ValueError("unexpected element at line %d" %
197 self.parser.CurrentLineNumber)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100198 self.stack[-1][self.current_key] = value
199 self.current_key = None
Christian Heimes7e182542008-01-27 15:20:13 +0000200 elif not self.stack:
201 # this is the root object
202 self.root = value
203 else:
Ned Deilyb8e59f72011-05-28 02:19:19 -0700204 if not isinstance(self.stack[-1], type([])):
205 raise ValueError("unexpected element at line %d" %
206 self.parser.CurrentLineNumber)
Christian Heimes7e182542008-01-27 15:20:13 +0000207 self.stack[-1].append(value)
208
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100209 def get_data(self):
Christian Heimes7e182542008-01-27 15:20:13 +0000210 data = ''.join(self.data)
211 self.data = []
212 return data
213
214 # element handlers
215
216 def begin_dict(self, attrs):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100217 d = self._dict_type()
218 self.add_object(d)
Christian Heimes7e182542008-01-27 15:20:13 +0000219 self.stack.append(d)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100220
Christian Heimes7e182542008-01-27 15:20:13 +0000221 def end_dict(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100222 if self.current_key:
Ned Deilyb8e59f72011-05-28 02:19:19 -0700223 raise ValueError("missing value for key '%s' at line %d" %
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100224 (self.current_key,self.parser.CurrentLineNumber))
Christian Heimes7e182542008-01-27 15:20:13 +0000225 self.stack.pop()
226
227 def end_key(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100228 if self.current_key or not isinstance(self.stack[-1], type({})):
Ned Deilyb8e59f72011-05-28 02:19:19 -0700229 raise ValueError("unexpected key at line %d" %
230 self.parser.CurrentLineNumber)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100231 self.current_key = self.get_data()
Christian Heimes7e182542008-01-27 15:20:13 +0000232
233 def begin_array(self, attrs):
234 a = []
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100235 self.add_object(a)
Christian Heimes7e182542008-01-27 15:20:13 +0000236 self.stack.append(a)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100237
Christian Heimes7e182542008-01-27 15:20:13 +0000238 def end_array(self):
239 self.stack.pop()
240
241 def end_true(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100242 self.add_object(True)
243
Christian Heimes7e182542008-01-27 15:20:13 +0000244 def end_false(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100245 self.add_object(False)
246
Christian Heimes7e182542008-01-27 15:20:13 +0000247 def end_integer(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100248 self.add_object(int(self.get_data()))
249
Christian Heimes7e182542008-01-27 15:20:13 +0000250 def end_real(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100251 self.add_object(float(self.get_data()))
252
Christian Heimes7e182542008-01-27 15:20:13 +0000253 def end_string(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100254 self.add_object(self.get_data())
255
Christian Heimes7e182542008-01-27 15:20:13 +0000256 def end_data(self):
Jon Janzence81a922019-09-05 03:11:35 -0500257 self.add_object(_decode_base64(self.get_data()))
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100258
Christian Heimes7e182542008-01-27 15:20:13 +0000259 def end_date(self):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100260 self.add_object(_date_from_string(self.get_data()))
261
262
263class _DumbXMLWriter:
264 def __init__(self, file, indent_level=0, indent="\t"):
265 self.file = file
266 self.stack = []
267 self._indent_level = indent_level
268 self.indent = indent
269
270 def begin_element(self, element):
271 self.stack.append(element)
272 self.writeln("<%s>" % element)
273 self._indent_level += 1
274
275 def end_element(self, element):
276 assert self._indent_level > 0
277 assert self.stack.pop() == element
278 self._indent_level -= 1
279 self.writeln("</%s>" % element)
280
281 def simple_element(self, element, value=None):
282 if value is not None:
283 value = _escape(value)
284 self.writeln("<%s>%s</%s>" % (element, value, element))
285
286 else:
287 self.writeln("<%s/>" % element)
288
289 def writeln(self, line):
290 if line:
291 # plist has fixed encoding of utf-8
292
293 # XXX: is this test needed?
294 if isinstance(line, str):
295 line = line.encode('utf-8')
296 self.file.write(self._indent_level * self.indent)
297 self.file.write(line)
298 self.file.write(b'\n')
299
300
301class _PlistWriter(_DumbXMLWriter):
302 def __init__(
303 self, file, indent_level=0, indent=b"\t", writeHeader=1,
304 sort_keys=True, skipkeys=False):
305
306 if writeHeader:
307 file.write(PLISTHEADER)
308 _DumbXMLWriter.__init__(self, file, indent_level, indent)
309 self._sort_keys = sort_keys
310 self._skipkeys = skipkeys
311
312 def write(self, value):
313 self.writeln("<plist version=\"1.0\">")
314 self.write_value(value)
315 self.writeln("</plist>")
316
317 def write_value(self, value):
318 if isinstance(value, str):
319 self.simple_element("string", value)
320
321 elif value is True:
322 self.simple_element("true")
323
324 elif value is False:
325 self.simple_element("false")
326
327 elif isinstance(value, int):
Ronald Oussoren6db66532014-01-15 11:32:35 +0100328 if -1 << 63 <= value < 1 << 64:
329 self.simple_element("integer", "%d" % value)
330 else:
331 raise OverflowError(value)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100332
333 elif isinstance(value, float):
334 self.simple_element("real", repr(value))
335
336 elif isinstance(value, dict):
337 self.write_dict(value)
338
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100339 elif isinstance(value, (bytes, bytearray)):
340 self.write_bytes(value)
341
342 elif isinstance(value, datetime.datetime):
343 self.simple_element("date", _date_to_string(value))
344
345 elif isinstance(value, (tuple, list)):
346 self.write_array(value)
347
348 else:
349 raise TypeError("unsupported type: %s" % type(value))
350
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100351 def write_bytes(self, data):
352 self.begin_element("data")
353 self._indent_level -= 1
354 maxlinelength = max(
355 16,
356 76 - len(self.indent.replace(b"\t", b" " * 8) * self._indent_level))
357
358 for line in _encode_base64(data, maxlinelength).split(b"\n"):
359 if line:
360 self.writeln(line)
361 self._indent_level += 1
362 self.end_element("data")
363
364 def write_dict(self, d):
365 if d:
366 self.begin_element("dict")
367 if self._sort_keys:
368 items = sorted(d.items())
369 else:
370 items = d.items()
371
372 for key, value in items:
373 if not isinstance(key, str):
374 if self._skipkeys:
375 continue
376 raise TypeError("keys must be strings")
377 self.simple_element("key", key)
378 self.write_value(value)
379 self.end_element("dict")
380
381 else:
382 self.simple_element("dict")
383
384 def write_array(self, array):
385 if array:
386 self.begin_element("array")
387 for value in array:
388 self.write_value(value)
389 self.end_element("array")
390
391 else:
392 self.simple_element("array")
393
394
395def _is_fmt_xml(header):
396 prefixes = (b'<?xml', b'<plist')
397
398 for pfx in prefixes:
399 if header.startswith(pfx):
400 return True
401
402 # Also check for alternative XML encodings, this is slightly
403 # overkill because the Apple tools (and plistlib) will not
404 # generate files with these encodings.
405 for bom, encoding in (
406 (codecs.BOM_UTF8, "utf-8"),
407 (codecs.BOM_UTF16_BE, "utf-16-be"),
408 (codecs.BOM_UTF16_LE, "utf-16-le"),
409 # expat does not support utf-32
410 #(codecs.BOM_UTF32_BE, "utf-32-be"),
411 #(codecs.BOM_UTF32_LE, "utf-32-le"),
412 ):
413 if not header.startswith(bom):
414 continue
415
416 for start in prefixes:
417 prefix = bom + start.decode('ascii').encode(encoding)
418 if header[:len(prefix)] == prefix:
419 return True
420
421 return False
422
423#
424# Binary Plist
425#
426
427
428class InvalidFileException (ValueError):
429 def __init__(self, message="Invalid file"):
430 ValueError.__init__(self, message)
431
432_BINARY_FORMAT = {1: 'B', 2: 'H', 4: 'L', 8: 'Q'}
433
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200434_undefined = object()
435
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100436class _BinaryPlistParser:
437 """
438 Read or write a binary plist file, following the description of the binary
439 format. Raise InvalidFileException in case of error, otherwise return the
440 root object.
441
442 see also: http://opensource.apple.com/source/CF/CF-744.18/CFBinaryPList.c
443 """
Jon Janzence81a922019-09-05 03:11:35 -0500444 def __init__(self, dict_type):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100445 self._dict_type = dict_type
446
447 def parse(self, fp):
448 try:
449 # The basic file format:
450 # HEADER
451 # object...
452 # refid->offset...
453 # TRAILER
454 self._fp = fp
455 self._fp.seek(-32, os.SEEK_END)
456 trailer = self._fp.read(32)
457 if len(trailer) != 32:
458 raise InvalidFileException()
459 (
460 offset_size, self._ref_size, num_objects, top_object,
461 offset_table_offset
462 ) = struct.unpack('>6xBBQQQ', trailer)
463 self._fp.seek(offset_table_offset)
Serhiy Storchaka06526642014-05-23 16:13:33 +0300464 self._object_offsets = self._read_ints(num_objects, offset_size)
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200465 self._objects = [_undefined] * num_objects
466 return self._read_object(top_object)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100467
Serhiy Storchakadb91e0f2017-10-31 14:05:53 +0200468 except (OSError, IndexError, struct.error, OverflowError,
469 UnicodeDecodeError):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100470 raise InvalidFileException()
471
472 def _get_size(self, tokenL):
473 """ return the size of the next object."""
474 if tokenL == 0xF:
475 m = self._fp.read(1)[0] & 0x3
476 s = 1 << m
477 f = '>' + _BINARY_FORMAT[s]
478 return struct.unpack(f, self._fp.read(s))[0]
479
480 return tokenL
481
Serhiy Storchaka06526642014-05-23 16:13:33 +0300482 def _read_ints(self, n, size):
483 data = self._fp.read(size * n)
484 if size in _BINARY_FORMAT:
485 return struct.unpack('>' + _BINARY_FORMAT[size] * n, data)
486 else:
Serhiy Storchakadb91e0f2017-10-31 14:05:53 +0200487 if not size or len(data) != size * n:
488 raise InvalidFileException()
Serhiy Storchaka06526642014-05-23 16:13:33 +0300489 return tuple(int.from_bytes(data[i: i + size], 'big')
490 for i in range(0, size * n, size))
491
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100492 def _read_refs(self, n):
Serhiy Storchaka06526642014-05-23 16:13:33 +0300493 return self._read_ints(n, self._ref_size)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100494
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200495 def _read_object(self, ref):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100496 """
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200497 read the object by reference.
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100498
499 May recursively read sub-objects (content of an array/dict/set)
500 """
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200501 result = self._objects[ref]
502 if result is not _undefined:
503 return result
504
505 offset = self._object_offsets[ref]
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100506 self._fp.seek(offset)
507 token = self._fp.read(1)[0]
508 tokenH, tokenL = token & 0xF0, token & 0x0F
509
510 if token == 0x00:
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200511 result = None
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100512
513 elif token == 0x08:
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200514 result = False
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100515
516 elif token == 0x09:
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200517 result = True
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100518
519 # The referenced source code also mentions URL (0x0c, 0x0d) and
520 # UUID (0x0e), but neither can be generated using the Cocoa libraries.
521
522 elif token == 0x0f:
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200523 result = b''
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100524
525 elif tokenH == 0x10: # int
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200526 result = int.from_bytes(self._fp.read(1 << tokenL),
527 'big', signed=tokenL >= 3)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100528
529 elif token == 0x22: # real
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200530 result = struct.unpack('>f', self._fp.read(4))[0]
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100531
532 elif token == 0x23: # real
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200533 result = struct.unpack('>d', self._fp.read(8))[0]
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100534
535 elif token == 0x33: # date
536 f = struct.unpack('>d', self._fp.read(8))[0]
537 # timestamp 0 of binary plists corresponds to 1/1/2001
538 # (year of Mac OS X 10.0), instead of 1/1/1970.
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200539 result = (datetime.datetime(2001, 1, 1) +
540 datetime.timedelta(seconds=f))
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100541
542 elif tokenH == 0x40: # data
543 s = self._get_size(tokenL)
Jon Janzence81a922019-09-05 03:11:35 -0500544 result = self._fp.read(s)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100545
546 elif tokenH == 0x50: # ascii string
547 s = self._get_size(tokenL)
548 result = self._fp.read(s).decode('ascii')
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100549
550 elif tokenH == 0x60: # unicode string
551 s = self._get_size(tokenL)
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200552 result = self._fp.read(s * 2).decode('utf-16be')
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100553
Jon Janzenc981ad12019-05-15 22:14:38 +0200554 elif tokenH == 0x80: # UID
555 # used by Key-Archiver plist files
556 result = UID(int.from_bytes(self._fp.read(1 + tokenL), 'big'))
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100557
558 elif tokenH == 0xA0: # array
559 s = self._get_size(tokenL)
560 obj_refs = self._read_refs(s)
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200561 result = []
562 self._objects[ref] = result
563 result.extend(self._read_object(x) for x in obj_refs)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100564
565 # tokenH == 0xB0 is documented as 'ordset', but is not actually
566 # implemented in the Apple reference code.
567
568 # tokenH == 0xC0 is documented as 'set', but sets cannot be used in
569 # plists.
570
571 elif tokenH == 0xD0: # dict
572 s = self._get_size(tokenL)
573 key_refs = self._read_refs(s)
574 obj_refs = self._read_refs(s)
575 result = self._dict_type()
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200576 self._objects[ref] = result
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100577 for k, o in zip(key_refs, obj_refs):
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200578 result[self._read_object(k)] = self._read_object(o)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100579
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200580 else:
581 raise InvalidFileException()
582
583 self._objects[ref] = result
584 return result
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100585
586def _count_to_size(count):
587 if count < 1 << 8:
588 return 1
589
590 elif count < 1 << 16:
591 return 2
592
593 elif count << 1 << 32:
594 return 4
595
596 else:
597 return 8
598
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200599_scalars = (str, int, float, datetime.datetime, bytes)
600
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100601class _BinaryPlistWriter (object):
602 def __init__(self, fp, sort_keys, skipkeys):
603 self._fp = fp
604 self._sort_keys = sort_keys
605 self._skipkeys = skipkeys
606
607 def write(self, value):
608
609 # Flattened object list:
610 self._objlist = []
611
612 # Mappings from object->objectid
613 # First dict has (type(object), object) as the key,
614 # second dict is used when object is not hashable and
615 # has id(object) as the key.
616 self._objtable = {}
617 self._objidtable = {}
618
619 # Create list of all objects in the plist
620 self._flatten(value)
621
622 # Size of object references in serialized containers
623 # depends on the number of objects in the plist.
624 num_objects = len(self._objlist)
625 self._object_offsets = [0]*num_objects
626 self._ref_size = _count_to_size(num_objects)
627
628 self._ref_format = _BINARY_FORMAT[self._ref_size]
629
630 # Write file header
631 self._fp.write(b'bplist00')
632
633 # Write object list
634 for obj in self._objlist:
635 self._write_object(obj)
636
637 # Write refnum->object offset table
638 top_object = self._getrefnum(value)
639 offset_table_offset = self._fp.tell()
640 offset_size = _count_to_size(offset_table_offset)
641 offset_format = '>' + _BINARY_FORMAT[offset_size] * num_objects
642 self._fp.write(struct.pack(offset_format, *self._object_offsets))
643
644 # Write trailer
645 sort_version = 0
646 trailer = (
647 sort_version, offset_size, self._ref_size, num_objects,
648 top_object, offset_table_offset
649 )
650 self._fp.write(struct.pack('>5xBBBQQQ', *trailer))
651
652 def _flatten(self, value):
653 # First check if the object is in the object table, not used for
654 # containers to ensure that two subcontainers with the same contents
655 # will be serialized as distinct values.
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200656 if isinstance(value, _scalars):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100657 if (type(value), value) in self._objtable:
658 return
659
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200660 elif id(value) in self._objidtable:
661 return
662
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100663 # Add to objectreference map
664 refnum = len(self._objlist)
665 self._objlist.append(value)
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200666 if isinstance(value, _scalars):
667 self._objtable[(type(value), value)] = refnum
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200668 else:
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100669 self._objidtable[id(value)] = refnum
670
671 # And finally recurse into containers
672 if isinstance(value, dict):
673 keys = []
674 values = []
675 items = value.items()
676 if self._sort_keys:
677 items = sorted(items)
678
679 for k, v in items:
680 if not isinstance(k, str):
681 if self._skipkeys:
682 continue
683 raise TypeError("keys must be strings")
684 keys.append(k)
685 values.append(v)
686
687 for o in itertools.chain(keys, values):
688 self._flatten(o)
689
690 elif isinstance(value, (list, tuple)):
691 for o in value:
692 self._flatten(o)
693
694 def _getrefnum(self, value):
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200695 if isinstance(value, _scalars):
696 return self._objtable[(type(value), value)]
Serhiy Storchakaa897aee2017-11-30 23:26:11 +0200697 else:
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100698 return self._objidtable[id(value)]
699
700 def _write_size(self, token, size):
701 if size < 15:
702 self._fp.write(struct.pack('>B', token | size))
703
704 elif size < 1 << 8:
705 self._fp.write(struct.pack('>BBB', token | 0xF, 0x10, size))
706
707 elif size < 1 << 16:
708 self._fp.write(struct.pack('>BBH', token | 0xF, 0x11, size))
709
710 elif size < 1 << 32:
711 self._fp.write(struct.pack('>BBL', token | 0xF, 0x12, size))
712
713 else:
714 self._fp.write(struct.pack('>BBQ', token | 0xF, 0x13, size))
715
716 def _write_object(self, value):
717 ref = self._getrefnum(value)
718 self._object_offsets[ref] = self._fp.tell()
719 if value is None:
720 self._fp.write(b'\x00')
721
722 elif value is False:
723 self._fp.write(b'\x08')
724
725 elif value is True:
726 self._fp.write(b'\x09')
727
728 elif isinstance(value, int):
Ronald Oussoren6db66532014-01-15 11:32:35 +0100729 if value < 0:
730 try:
731 self._fp.write(struct.pack('>Bq', 0x13, value))
732 except struct.error:
Ronald Oussoren94e44a92014-02-06 11:19:18 +0100733 raise OverflowError(value) from None
Ronald Oussoren6db66532014-01-15 11:32:35 +0100734 elif value < 1 << 8:
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100735 self._fp.write(struct.pack('>BB', 0x10, value))
736 elif value < 1 << 16:
737 self._fp.write(struct.pack('>BH', 0x11, value))
738 elif value < 1 << 32:
739 self._fp.write(struct.pack('>BL', 0x12, value))
Ronald Oussoren94e44a92014-02-06 11:19:18 +0100740 elif value < 1 << 63:
741 self._fp.write(struct.pack('>BQ', 0x13, value))
742 elif value < 1 << 64:
743 self._fp.write(b'\x14' + value.to_bytes(16, 'big', signed=True))
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100744 else:
Ronald Oussoren94e44a92014-02-06 11:19:18 +0100745 raise OverflowError(value)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100746
747 elif isinstance(value, float):
748 self._fp.write(struct.pack('>Bd', 0x23, value))
749
750 elif isinstance(value, datetime.datetime):
751 f = (value - datetime.datetime(2001, 1, 1)).total_seconds()
752 self._fp.write(struct.pack('>Bd', 0x33, f))
753
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100754 elif isinstance(value, (bytes, bytearray)):
755 self._write_size(0x40, len(value))
756 self._fp.write(value)
757
758 elif isinstance(value, str):
759 try:
760 t = value.encode('ascii')
761 self._write_size(0x50, len(value))
762 except UnicodeEncodeError:
763 t = value.encode('utf-16be')
Serhiy Storchaka7338ebc2016-10-04 20:04:30 +0300764 self._write_size(0x60, len(t) // 2)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100765
766 self._fp.write(t)
767
Jon Janzenc981ad12019-05-15 22:14:38 +0200768 elif isinstance(value, UID):
769 if value.data < 0:
770 raise ValueError("UIDs must be positive")
771 elif value.data < 1 << 8:
772 self._fp.write(struct.pack('>BB', 0x80, value))
773 elif value.data < 1 << 16:
774 self._fp.write(struct.pack('>BH', 0x81, value))
775 elif value.data < 1 << 32:
776 self._fp.write(struct.pack('>BL', 0x83, value))
777 elif value.data < 1 << 64:
778 self._fp.write(struct.pack('>BQ', 0x87, value))
779 else:
780 raise OverflowError(value)
781
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100782 elif isinstance(value, (list, tuple)):
783 refs = [self._getrefnum(o) for o in value]
784 s = len(refs)
785 self._write_size(0xA0, s)
786 self._fp.write(struct.pack('>' + self._ref_format * s, *refs))
787
788 elif isinstance(value, dict):
789 keyRefs, valRefs = [], []
790
791 if self._sort_keys:
792 rootItems = sorted(value.items())
793 else:
794 rootItems = value.items()
795
796 for k, v in rootItems:
797 if not isinstance(k, str):
798 if self._skipkeys:
799 continue
800 raise TypeError("keys must be strings")
801 keyRefs.append(self._getrefnum(k))
802 valRefs.append(self._getrefnum(v))
803
804 s = len(keyRefs)
805 self._write_size(0xD0, s)
806 self._fp.write(struct.pack('>' + self._ref_format * s, *keyRefs))
807 self._fp.write(struct.pack('>' + self._ref_format * s, *valRefs))
808
809 else:
Ronald Oussoren6db66532014-01-15 11:32:35 +0100810 raise TypeError(value)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100811
812
813def _is_fmt_binary(header):
814 return header[:8] == b'bplist00'
815
816
817#
818# Generic bits
819#
820
821_FORMATS={
822 FMT_XML: dict(
823 detect=_is_fmt_xml,
824 parser=_PlistParser,
825 writer=_PlistWriter,
826 ),
827 FMT_BINARY: dict(
828 detect=_is_fmt_binary,
829 parser=_BinaryPlistParser,
830 writer=_BinaryPlistWriter,
831 )
832}
833
834
Jon Janzence81a922019-09-05 03:11:35 -0500835def load(fp, *, fmt=None, dict_type=dict):
Collin Styles0d4f4352019-07-14 02:01:48 -0700836 """Read a .plist file. 'fp' should be a readable and binary file object.
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100837 Return the unpacked root object (which usually is a dictionary).
838 """
839 if fmt is None:
840 header = fp.read(32)
841 fp.seek(0)
842 for info in _FORMATS.values():
843 if info['detect'](header):
Serhiy Storchaka89667592014-07-23 18:49:31 +0300844 P = info['parser']
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100845 break
846
847 else:
848 raise InvalidFileException()
849
850 else:
Serhiy Storchaka89667592014-07-23 18:49:31 +0300851 P = _FORMATS[fmt]['parser']
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100852
Jon Janzence81a922019-09-05 03:11:35 -0500853 p = P(dict_type=dict_type)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100854 return p.parse(fp)
855
856
Jon Janzence81a922019-09-05 03:11:35 -0500857def loads(value, *, fmt=None, dict_type=dict):
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100858 """Read a .plist file from a bytes object.
859 Return the unpacked root object (which usually is a dictionary).
860 """
861 fp = BytesIO(value)
Jon Janzence81a922019-09-05 03:11:35 -0500862 return load(fp, fmt=fmt, dict_type=dict_type)
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100863
864
865def dump(value, fp, *, fmt=FMT_XML, sort_keys=True, skipkeys=False):
Collin Styles0d4f4352019-07-14 02:01:48 -0700866 """Write 'value' to a .plist file. 'fp' should be a writable,
867 binary file object.
Ronald Oussorenc5cf7972013-11-21 15:46:49 +0100868 """
869 if fmt not in _FORMATS:
870 raise ValueError("Unsupported format: %r"%(fmt,))
871
872 writer = _FORMATS[fmt]["writer"](fp, sort_keys=sort_keys, skipkeys=skipkeys)
873 writer.write(value)
874
875
876def dumps(value, *, fmt=FMT_XML, skipkeys=False, sort_keys=True):
877 """Return a bytes object with the contents for a .plist file.
878 """
879 fp = BytesIO()
880 dump(value, fp, fmt=fmt, skipkeys=skipkeys, sort_keys=sort_keys)
881 return fp.getvalue()