mirror of
https://github.com/ankitects/anki.git
synced 2025-09-20 15:02:21 -04:00
catch attempts to write outside the media folder
big thanks to David Bailey for discovering this
This commit is contained in:
parent
194c02ed16
commit
375c071a26
1 changed files with 7 additions and 3 deletions
|
@ -19,16 +19,20 @@ class AnkiPackageImporter(Anki2Importer):
|
|||
# we need the media dict in advance, and we'll need a map of fname ->
|
||||
# number to use during the import
|
||||
self.nameToNum = {}
|
||||
dir = self.col.media.dir()
|
||||
for k, v in list(json.loads(z.read("media").decode("utf8")).items()):
|
||||
self.nameToNum[v] = k
|
||||
path = os.path.abspath(os.path.join(dir, v))
|
||||
if os.path.commonprefix([path, dir]) != dir:
|
||||
raise Exception("Invalid file")
|
||||
|
||||
self.nameToNum[v] = unicodedata.normalize("NFC", k)
|
||||
# run anki2 importer
|
||||
Anki2Importer.run(self)
|
||||
# import static media
|
||||
for file, c in list(self.nameToNum.items()):
|
||||
if not file.startswith("_") and not file.startswith("latex-"):
|
||||
continue
|
||||
path = os.path.join(self.col.media.dir(),
|
||||
unicodedata.normalize("NFC", file))
|
||||
path = os.path.join(self.col.media.dir(), file)
|
||||
if not os.path.exists(path):
|
||||
open(path, "wb").write(z.read(c))
|
||||
|
||||
|
|
Loading…
Reference in a new issue