diff options
| author | Benjamin Peterson <benjamin@python.org> | 2012-10-31 19:01:42 -0400 |
|---|---|---|
| committer | Benjamin Peterson <benjamin@python.org> | 2012-10-31 19:01:42 -0400 |
| commit | 8144eb153a29e607aa5ee1957e1723557f605ae6 (patch) | |
| tree | 984f78537b2625eadd1336f4e6a383e80d208979 /Python/symtable.c | |
| parent | a544cdd8453491733923297e43fd8e8f31f31a81 (diff) | |
| download | cpython-8144eb153a29e607aa5ee1957e1723557f605ae6.tar.gz | |
check return value of _PyUnicode_AsString
Diffstat (limited to 'Python/symtable.c')
| -rw-r--r-- | Python/symtable.c | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Python/symtable.c b/Python/symtable.c index 1c291bbac0..ff6e8b79b9 100644 --- a/Python/symtable.c +++ b/Python/symtable.c @@ -1202,12 +1202,14 @@ symtable_visit_stmt(struct symtable *st, stmt_ty s) asdl_seq *seq = s->v.Global.names; for (i = 0; i < asdl_seq_LEN(seq); i++) { identifier name = (identifier)asdl_seq_GET(seq, i); - char *c_name = _PyUnicode_AsString(name); long cur = symtable_lookup(st, name); if (cur < 0) return 0; if (cur & (DEF_LOCAL | USE)) { char buf[256]; + char *c_name = _PyUnicode_AsString(name); + if (!c_name) + return 0; if (cur & DEF_LOCAL) PyOS_snprintf(buf, sizeof(buf), GLOBAL_AFTER_ASSIGN, @@ -1229,12 +1231,14 @@ symtable_visit_stmt(struct symtable *st, stmt_ty s) asdl_seq *seq = s->v.Nonlocal.names; for (i = 0; i < asdl_seq_LEN(seq); i++) { identifier name = (identifier)asdl_seq_GET(seq, i); - char *c_name = _PyUnicode_AsString(name); long cur = symtable_lookup(st, name); if (cur < 0) return 0; if (cur & (DEF_LOCAL | USE)) { char buf[256]; + char *c_name = _PyUnicode_AsString(name); + if (!c_name) + return 0; if (cur & DEF_LOCAL) PyOS_snprintf(buf, sizeof(buf), NONLOCAL_AFTER_ASSIGN, |