Proper fix for escaping xpath attribute values

This commit is contained in:
Kovid Goyal 2016-02-03 09:28:10 +05:30
parent 2638906e69
commit f8c2b77812
3 changed files with 19 additions and 4 deletions

View File

@ -297,3 +297,16 @@ def generate_masthead(title, output_path=None, width=600, height=60):
from calibre.ebooks.covers import generate_masthead
return generate_masthead(title, output_path=output_path, width=width, height=height, font_family=masthead_font_family)
def escape_xpath_attr(value):
if '"' in value:
if "'" in value:
parts = re.split('("+)', value)
ans = []
for x in parts:
if x:
q = "'" if '"' in x else '"'
ans.append(q + x + q)
return 'concat(%s)' % ', '.join(ans)
else:
return "'%s'" % value
return '"%s"' % value

View File

@ -13,6 +13,7 @@ from urlparse import urlparse
from lxml import etree
from calibre.ebooks import escape_xpath_attr
from calibre.ebooks.chardet import xml_to_unicode
from calibre.constants import __appname__, __version__, filesystem_encoding
from calibre.ebooks.metadata.toc import TOC
@ -20,7 +21,7 @@ from calibre.ebooks.metadata import string_to_authors, MetaInformation, check_is
from calibre.ebooks.metadata.book.base import Metadata
from calibre.utils.date import parse_date, isoformat
from calibre.utils.localization import get_lang, canonicalize_lang
from calibre import prints, guess_type, prepare_string_for_xml
from calibre import prints, guess_type
from calibre.utils.cleantext import clean_ascii_chars, clean_xml_chars
from calibre.utils.config import tweaks
@ -1120,7 +1121,7 @@ class OPF(object): # {{{
uuid_elem = self.root.attrib[attr]
break
if uuid_elem:
matches = self.root.xpath('//*[@id="%s"]'%prepare_string_for_xml(uuid_elem, True))
matches = self.root.xpath('//*[@id=%s]'%escape_xpath_attr(uuid_elem))
if matches:
for m in matches:
raw = m.text

View File

@ -16,9 +16,10 @@ from future_builtins import zip
from lxml import etree
from cssutils import replaceUrls, getUrls
from calibre import CurrentDir, prepare_string_for_xml
from calibre import CurrentDir
from calibre.constants import iswindows
from calibre.customize.ui import (plugin_for_input_format, plugin_for_output_format)
from calibre.ebooks import escape_xpath_attr
from calibre.ebooks.chardet import xml_to_unicode
from calibre.ebooks.conversion.plugins.epub_input import (
ADOBE_OBFUSCATION, IDPF_OBFUSCATION, decrypt_font_data)
@ -1103,7 +1104,7 @@ class EpubContainer(Container):
package_id = val
break
if package_id is not None:
for elem in self.opf_xpath('//*[@id="%s"]'%prepare_string_for_xml(package_id, True)):
for elem in self.opf_xpath('//*[@id=%s]'%escape_xpath_attr(package_id)):
if elem.text:
raw_unique_identifier = elem.text
break