summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorMark Dickinson <dickinsm@gmail.com>2010-07-04 18:16:43 (GMT)
committerMark Dickinson <dickinsm@gmail.com>2010-07-04 18:16:43 (GMT)
commit1b9b5727cc85c4aec17c2fa79f9491d986dbb00e (patch)
treeb3477eca8792ea14563fb6167e962820e34b7f59
parent31a0f86e368c980b751bdc0b377e42322cf2566d (diff)
downloadcpython-1b9b5727cc85c4aec17c2fa79f9491d986dbb00e.zip
cpython-1b9b5727cc85c4aec17c2fa79f9491d986dbb00e.tar.gz
cpython-1b9b5727cc85c4aec17c2fa79f9491d986dbb00e.tar.bz2
Merged revisions 82552-82553 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/branches/py3k ........ r82552 | mark.dickinson | 2010-07-04 19:11:51 +0100 (Sun, 04 Jul 2010) | 2 lines Issue #9130: Fix validation of relative imports in parser module. ........ r82553 | mark.dickinson | 2010-07-04 19:15:26 +0100 (Sun, 04 Jul 2010) | 1 line Fix symbol numbers in test_parser test. ........
-rw-r--r--Lib/test/test_parser.py20
-rw-r--r--Misc/NEWS2
-rw-r--r--Modules/parsermodule.c7
3 files changed, 26 insertions, 3 deletions
diff --git a/Lib/test/test_parser.py b/Lib/test/test_parser.py
index 3a7bbe9..17111b0 100644
--- a/Lib/test/test_parser.py
+++ b/Lib/test/test_parser.py
@@ -190,6 +190,12 @@ class RoundtripLegalSyntaxTestCase(unittest.TestCase):
self.check_suite("import sys as system, math")
self.check_suite("import sys, math as my_math")
+ def test_relative_imports(self):
+ self.check_suite("from . import name")
+ self.check_suite("from .. import name")
+ self.check_suite("from .pkg import name")
+ self.check_suite("from ..pkg import name")
+
def test_pep263(self):
self.check_suite("# -*- coding: iso-8859-1 -*-\n"
"pass\n")
@@ -459,6 +465,20 @@ class IllegalSyntaxTestCase(unittest.TestCase):
(0, ''))
self.check_bad_tree(tree, "malformed global ast")
+ def test_missing_import_source(self):
+ # from import fred
+ tree = \
+ (257,
+ (268,
+ (269,
+ (270,
+ (282,
+ (284, (1, 'from'), (1, 'import'),
+ (287, (285, (1, 'fred')))))),
+ (4, ''))),
+ (4, ''), (0, ''))
+ self.check_bad_tree(tree, "from import fred")
+
class CompileTestCase(unittest.TestCase):
diff --git a/Misc/NEWS b/Misc/NEWS
index b2a3160..87af41c 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -75,6 +75,8 @@ C-API
Library
-------
+- Issue #9130: Fix validation of relative imports in parser module.
+
- Issue #9128: Fix validation of class decorators in parser module.
- Issue #5468: urlencode to handle bytes type and other encodings in its query
diff --git a/Modules/parsermodule.c b/Modules/parsermodule.c
index b00037d..c9150d9 100644
--- a/Modules/parsermodule.c
+++ b/Modules/parsermodule.c
@@ -1763,8 +1763,8 @@ count_from_dots(node *tree)
return i-1;
}
-/* 'from' ('.'* dotted_name | '.') 'import' ('*' | '(' import_as_names ')' |
- * import_as_names
+/* import_from: ('from' ('.'* dotted_name | '.'+)
+ * 'import' ('*' | '(' import_as_names ')' | import_as_names))
*/
static int
validate_import_from(node *tree)
@@ -1774,7 +1774,8 @@ validate_import_from(node *tree)
int havename = (TYPE(CHILD(tree, ndots + 1)) == dotted_name);
int offset = ndots + havename;
int res = validate_ntype(tree, import_from)
- && (nch >= 4 + ndots)
+ && (offset >= 1)
+ && (nch >= 3 + offset)
&& validate_name(CHILD(tree, 0), "from")
&& (!havename || validate_dotted_name(CHILD(tree, ndots + 1)))
&& validate_name(CHILD(tree, offset + 1), "import");