From 99614a470f23d3d613576fcef8c1578758333063 Mon Sep 17 00:00:00 2001
From: Mathieu Comandon <strycore@gmail.com>
Date: Mon, 13 Mar 2017 21:08:12 -0700
Subject: [PATCH] Fix registry parsing with keys ending with double backslashes
 (Fixes #541)

---
 lutris/util/wineregistry.py |  6 +++++-
 tests/test_registry.py      | 12 ++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/lutris/util/wineregistry.py b/lutris/util/wineregistry.py
index 27be6997a..00f5ceade 100644
--- a/lutris/util/wineregistry.py
+++ b/lutris/util/wineregistry.py
@@ -215,7 +215,11 @@ class WineRegistryKey(object):
         if line.startswith('#'):
             self.add_meta(line)
         elif line.startswith('"'):
-            key, value = re.split(re.compile(r'(?<=[^\\]\")='), line, maxsplit=1)
+            try:
+                key, value = re.split(re.compile(r"(?<![^\\]\\\")="), line, maxsplit=1)
+            except ValueError as ex:
+                logger.error("Unable to parse line %s", line)
+                raise
             key = key[1:-1]
             self.subkeys[key] = value
         elif line.startswith('@'):
diff --git a/tests/test_registry.py b/tests/test_registry.py
index 050f66d2b..a9b12a0e5 100644
--- a/tests/test_registry.py
+++ b/tests/test_registry.py
@@ -98,3 +98,15 @@ class TestWineRegistryKey(TestCase):
         self.assertEqual(key.name, 'Control Panel/Desktop')
         self.assertEqual(key.raw_name, '[Control Panel\\\\Desktop]')
         self.assertRegexpMatches(key.raw_timestamp, r'\d+\s\d+')
+
+    def test_parse_registry_key(self):
+        key = WineRegistryKey(path='Control Panel/Desktop')
+        key.parse('"C:\\\\users\\\\strider\\\\My Music\\\\iTunes\\\\iTunes Music\\\\Podcasts\\\\"=dword:00000001')
+        self.assertEqual(key.subkeys["C:\\\\users\\\\strider\\\\My Music\\\\iTunes\\\\iTunes Music\\\\Podcasts\\\\"],
+                         'dword:00000001')
+
+        key.parse('"A"=val')
+        self.assertEqual(key.subkeys["A"], 'val')
+
+        key.parse('"String with \\"quotes\\""=val')
+        self.assertEqual(key.subkeys['String with \\"quotes\\"'], 'val')