Python源码分析5 – 语法分析器PyParser

Introduction

上一篇文章我们分析了Python是如何对语法文件Grammar进行预处理，生成语法数据，并在运行时生成Acclerators加速语法分析的过程。当分析完这些内容之后，下一步便是分析Python中语法分析的机制。回顾一下Python的整个处理流程：

1. PyTokenizer进行词法分析，把源程序分解为Token

2. PyParser根据Token创建CST

3. CST被转换为AST

4. AST被编译为字节码

5. 执行字节码

语法分析处于整个流程的第二步，其目标是处理token生成CST(Concrete Syntax Tree)。因此，在具体分析PyParser的工作方式之前，我们先看一下什么是CST。

CST (Concrete Syntax Tree)

CST (Concrete Syntax Tree) 和AST (Abstract Syntax Tree) 类似，都是语法分析所获得的中间结果。他们的不同之处在于，CST直接对应语法分析的匹配的过程，是直接生成的，含有大量冗余信息。而AST省略了中间的冗余信息，直接对应实际的语义，也就是分析的结果。用例子说明要清楚一些：

假设有这样一个表达式a，

CST是这样的：（->表示从父结点到子结点）

file_input -> stmt -> simple_stmt -> small_stmt -> expr_stmt -> testlist -> test ->or_test ->and_test ->not_test -> comparison -> expr -> xor_expr -> and_expr -> shift_expr -> arith_expr -> term -> factor -> power -> atom -> (NAME, “a”)

而AST则是：

(stmt_ty, expr_kind) -> (expr_ty, name_kind) ->(“a”)

可以看到CST表述了整个分析a的过程，从file_input一直推导到最后的NAME，每一步推导都成了树的结点，而大部分信息都可以说是无用的。AST的结构要简单和直接的多，直接表明a是一个表达式语句（假定a是一个单独的语句），内容是一个标示符，值为”a”。Python的语法分析生成的是CST而非AST，之后Python会调用PyAst_FromNode将CST转换为AST。

CST的结点称为Node，其结构定义在node.h中：

typedef struct _node { short n_type; char *n_str; int n_lineno; int n_col_offset; int n_nchildren; struct _node *n_child; } node;

Field	Description
n_type	结点类型，终结符定义在token.h中，而非终结符定义在graminit.h中
n_str	结点所对应的字符串的内容
n_lineno	对应的行号
n_col_offset	列号
n_nchildren	子结点的个数
n_child	子结点数组，动态分配内存

Python提供了下面的函数/宏来操作CST，同样定义在node.h中：

PyAPI_FUNC(node *) PyNode_New(int type); PyAPI_FUNC(int) PyNode_AddChild(node *n, int type, char *str, int lineno, int col_offset); PyAPI_FUNC(void) PyNode_Free(node *n); /* Node access functions */ #define NCH(n) ((n)->n_nchildren) #define CHILD(n, i) (&(n)->n_child[i]) #define RCHILD(n, i) (CHILD(n, NCH(n) + i)) #define TYPE(n) ((n)->n_type) #define STR(n) ((n)->n_str) /* Assert that the type of a node is what we expect */ #define REQ(n, type) assert(TYPE(n) == (type)) PyAPI_FUNC(void) PyNode_ListTree(node *);

PyNode_New和PyNode_Delete负责创建和释放node结构：

node * PyNode_New(int type) { node *n = (node *) PyObject_MALLOC(1 * sizeof(node)); if (n == NULL) return NULL; n->n_type = type; n->n_str = NULL; n->n_lineno = 0; n->n_nchildren = 0; n->n_child = NULL; return n; } void PyNode_Free(node *n) { if (n != NULL) { freechildren(n); PyObject_FREE(n); } } static void freechildren(node *n) { int i; for (i = NCH(n); --i >= 0; ) freechildren(CHILD(n, i)); if (n->n_child != NULL) PyObject_FREE(n->n_child); if (STR(n) != NULL) PyObject_FREE(STR(n)); }

NCH/CHILD/RCHILD/TYPE/STR是用来封装对node的成员的访问的。需要提一下的是，CHILD(n, i)是从左边开始算，传入i的是正数，而RCHILD(n, i)则是从右边往左，传入的参数i是负数。

PyNode_AddChild将一个新的子结点加入到子结点数组中。由于结点数量是动态变化的，因此在当前分配的结点数组大小不够的时候，Python会调用realloc重新分配内存。内存分配是一个非常耗时的动作，因此Python在PyNode_AddChild之中用到了和std::vector类似的技巧来尽量减少内存分配的次数，每次增长的时候都会根据某个规则进行RoundUp，而不是需要多少就分配多少。XXXROUNDUP函数负责进行此运算。n<=1时， 返回n。1<n<=128的时候，会RoundUp到4的倍数。n>128, 会调用fancy_roundup来RoundUp到2的幂。

#define XXXROUNDUP(n) ((n) <= 1 ? (n) : / (n) <= 128 ? (((n) + 3) & ~3) : / fancy_roundup(n))

有了XXXROUNDUP之后，实现PyNode_AddChild就非常直接了：

int PyNode_AddChild(register node *n1, int type, char *str, int lineno, int col_offset) { const int nch = n1->n_nchildren; int current_capacity; int required_capacity; node *n; if (nch == INT_MAX || nch < 0) return E_OVERFLOW; current_capacity = XXXROUNDUP(nch); required_capacity = XXXROUNDUP(nch + 1); if (current_capacity < 0 || required_capacity < 0) return E_OVERFLOW; if (current_capacity < required_capacity) { n = n1->n_child; n = (node *) PyObject_REALLOC(n, required_capacity * sizeof(node)); if (n == NULL) return E_NOMEM; n1->n_child = n; } n = &n1->n_child[n1->n_nchildren++]; n->n_type = type; n->n_str = str; n->n_lineno = lineno; n->n_col_offset = col_offset; n->n_nchildren = 0; n->n_child = NULL; return 0; }

值得注意的是该函数并没有记录下当前的最大容量，这个可以通过XXXROUNDUP(nch)计算出来。

PyParser

PyParser所作的事情就是根据token生成CST。整个生成树的过程其实也就是一个遍历语法图的过程。在前一篇文章中我们知道语法图是由多个DFA组成的，而输入的token和当前的所处的状态结点可以决定下一个状态结点。由于PyParser是在多个DFA中遍历，因此当结束了某个DFA的遍历需要回到上一个DFA，这些信息都是由一个专门的栈保存着。PyParser所对应的结构是parser_state，这个结构保存着PyParser的内部状态，如下：

typedef struct { stack p_stack; /* Stack of parser states */ grammar *p_grammar; /* Grammar to use */ node *p_tree; /* Top of parse tree */ #ifdef PY_PARSER_REQUIRES_FUTURE_KEYWORD unsigned long p_flags; /* see co_flags in Include/code.h */ #endif } parser_state;

Field	Description
p_stack	PyParser状态栈，每一个状态都对应着某个DFA中的某个状态
p_grammar	语法图的指针
p_tree	CST的根结点
p_flags	PyParser的flags，唯一用到的是CO_FUTURE_WITH_STATEMENT

状态栈的定义如下：

typedef struct { stackentry *s_top; /* Top entry */ stackentry s_base[MAXSTACK]; /* Array of stack entries */ /* NB The stack grows down */ } stack;

状态栈中的状态如下：

typedef struct { int s_state; /* State in current DFA */ dfa *s_dfa; /* Current DFA */ struct _node *s_parent; /* Where to add next node */ } stackentry;

Field	Description
s_state	当前DFA中的状态ID
s_dfa	当前DFA指针
s_parent	当前的parent结点。PyParser会把新的结点作为Child加到栈顶状态的s_parent结点中去

PyParser调用下面这些函数来操作栈：

static void s_reset(stack *s); #define s_empty(s) ((s)->s_top == &(s)->s_base[MAXSTACK]) static int s_push(register stack *s, dfa *d, node *parent) #define s_pop(s) (s)->s_top++

这些函数的实现非常简单，不再赘述。

PyParser所支持的“成员”函数如下：

parser_state *PyParser_New(grammar *g, int start); void PyParser_Delete(parser_state *ps); int PyParser_AddToken(parser_state *ps, int type, char *str, int lineno, int col_offset,int *expected_ret); void PyGrammar_AddAccelerators(grammar *g);

PyParser_New & PyParser_Delete显然是用于创建和销毁PyParser的实例的，和PyTokenizer一致。PyGrammar_AddAccelerators在我的前一篇Python源码研究4中提到过，主要用于处理Python的Grammar数据生成Accelerator加快语法分析的速度。在这些函数中，最为核心的是PyParser_AddToken，这个函数的作用是根据PyTokenizer所获得的token和当前所处的状态/DFA，跳转到下一个状态，并添加到CST中。在parsetok函数中，有如下的代码（省略了大部分）：

parser_state *ps; ps = PyParser_New(g, start); for (;;) { char *a, *b; int type; type = PyTokenizer_Get(tok, &a, &b); PyParser_AddToken(ps, (int)type, str, tok->lineno, col_offset, &(err_ret->expected)); }

Parsetok的作用是分析某段代码。可以看到，parsetok会反复调用PyTokenizer_Get获得下一个token，然后将反复将获得的token传给PyParser_AddToken来逐步构造整个CST，当所有token都处理过了之后，整棵树也就建立完毕了。

PyParser API

Python不会直接调用PyParser和PyTokenizer的函数，而是直接调用下面的这些Python API：

PyAPI_FUNC(node *) PyParser_ParseString(const char *, grammar *, int, perrdetail *); PyAPI_FUNC(node *) PyParser_ParseFile (FILE *, const char *, grammar *, int, char *, char *, perrdetail *); PyAPI_FUNC(node *) PyParser_ParseStringFlags(const char *, grammar *, int, perrdetail *, int); PyAPI_FUNC(node *) PyParser_ParseFileFlags(FILE *, const char *, grammar *, int, char *, char *, perrdetail *, int); PyAPI_FUNC(node *) PyParser_ParseStringFlagsFilename(const char *, const char *, grammar *, int, perrdetail *, int); /* Note that he following function is defined in pythonrun.c not parsetok.c. */ PyAPI_FUNC(void) PyParser_SetError(perrdetail *);

PyAPI_FUNC宏是用于定义公用的Python API，表明这些函数可以被外界调用。在Windows上面Python Core被编译成一个DLL，因此PyAPI_FUNC等价于大家常用的__declspec(dllexport)/__declspec(dllimport)。这些函数把PyParser和PyTokenizer对象的接口和细节包装起来，使用者可以直接调用PyParser_ParseXXXX函数来使用PyParser和PyTokenizer的功能而无需知道PyPaser/PyTokenizer的工作方式，这可以看作是一个典型的Façade模式。以PyParser_ParseFile为例，该函数分析传入的FILE返回生成的CST。其他的函数与此类似，只是分析的对象不同和传入参数的不同。

PyParser_AddToken implementation

在全面了解了PyParser的接口和调用方式之后，我们来看一下PyParser_AddToken的具体实现。PyParser_AddToken会调用3个内部函数来做处理：classify, push, shift

Classify根据type和str，确定对应的Label，实现如下：

static int classify(parser_state *ps, int type, char *str) { grammar *g = ps->p_grammar; register int n = g->g_ll.ll_nlabels; if (type == NAME) { register char *s = str; register label *l = g->g_ll.ll_label; register int i; for (i = n; i > 0; i--, l++) { if (l->lb_type != NAME || l->lb_str == NULL || l->lb_str[0] != s[0] || strcmp(l->lb_str, s) != 0) continue; #ifdef PY_PARSER_REQUIRES_FUTURE_KEYWORD if (!(ps->p_flags & CO_FUTURE_WITH_STATEMENT)) { if (s[0] == 'w' && strcmp(s, "with") == 0) break; /* not a keyword yet */ else if (s[0] == 'a' && strcmp(s, "as") == 0) break; /* not a keyword yet */ } #endif D(printf("It's a keyword/n")); return n - i; } } { register label *l = g->g_ll.ll_label; register int i; for (i = n; i > 0; i--, l++) { if (l->lb_type == type && l->lb_str == NULL) { D(printf("It's a token we know/n")); return n - i; } } } D(printf("Illegal token/n")); return -1; }

可以看到classify作了一些针对NAME的特殊处理。有两种NAME，一种是标示符，一种是关键字。语法图中标示符的str是NULL（因为在语法图中无法实现知道标示符的内容，也不用知道），而关键字的str是有值的，不同的关键字对应着不同的语句，因此需要进行比较。classify首先处理关键字的情况，如果不是关键字则当普通情况下处理。两种情况下都需要遍历整个LabelList来确定Label的ID，唯一不同的是关键字的情况下需要比较具体的str内容而其他情况下无须比较。这里的效率看起来还是有改进的余地，在非关键字情况下可以用一个数组来做索引，直接根据type查找到id。在关键字的情况下至少可以把关键字作为一个单独的Token/非终结符来识别，这样就不用和普通标示符的情况弄混了，而目前的实现即使是普通标示符也会进到第一个if语句中进行关键字的判断。还可以使用hash来快速查找等。

Shift改变当前栈顶的状态（注意并不跳转到另外的DFA，这个改变只限于单个DFA中，跳转到另外一个DFA需要调用push压栈），并把当前的type/str作为一个新的子结点加入到栈顶的s_parent结点，通常s_parent结点对应着当前的DFA的结点。

假设我们在DFA0中，当前栈顶为(DFA0, s1)，type=NAME

NAME

DFA0: s0 -> s1 -> s2

从s1跳转到s2不会离开DFA0，不用进栈，只需改变当前(DFA0, s1)到(DFA0, s2)即可。

static int shift(register stack *s, int type, char *str, int newstate, int lineno, int col_offset) { int err; assert(!s_empty(s)); err = PyNode_AddChild(s->s_top->s_parent, type, str, lineno, col_offset); if (err) return err; s->s_top->s_state = newstate; return 0; }

Push同样也会把type/str作为新的子结点n加入到当前s_parent结点并改变当前栈顶的状态为newstate。但是newstate并非是下一个状态，而是当新的DFA遍历完毕之后退栈才会到。然后，把目标DFA压栈。新生成的子结点n作为新的s_parent。

还是举一个例子比较容易说明这个过程：

‘a’

DFA 1: s0 -> s1

DFA1

DFA 0: s1 -> s2

假设当前我们处于(DFA0, s1), type/str告诉我们下一个状态为s2，label是非终结符DFA1，对应着DFA1，因此我们会把当前栈顶(DFA0, s1)修改为(DFA0, s2)，然后跳转到DFA1中进行匹配，同时(DFA1, s0)作为新的栈顶压栈，当(DFA1,s0)退栈之后，说明DFA1匹配完毕，回到(DFA0, s2)。

具体的实现如下：

static int push(register stack *s, int type, dfa *d, int newstate, int lineno, int col_offset) { int err; register node *n; n = s->s_top->s_parent; assert(!s_empty(s)); err = PyNode_AddChild(n, type, (char *)NULL, lineno, col_offset); if (err) return err; s->s_top->s_state = newstate; return s_push(s, d, CHILD(n, NCH(n)-1)); }

在讨论完这些函数之后就可以开始着手研究PyParser_AddToken的实现了：

int PyParser_AddToken(register parser_state *ps, register int type, char *str, int lineno, int col_offset, int *expected_ret) { register int ilabel; int err; D(printf("Token %s/'%s' ... ", _PyParser_TokenNames[type], str)); /* Find out which label this token is */ ilabel = classify(ps, type, str); if (ilabel < 0) return E_SYNTAX;

PyParser_AddToken第一步是根据type和str，调用classify获得对应的label。

PyParser_AddToken获得了对应的label之后，进入一个for loop：

/* Loop until the token is shifted or an error occurred */ for (;;) { /* Fetch the current dfa and state */ register dfa *d = ps->p_stack.s_top->s_dfa; register state *s = &d->d_state[ps->p_stack.s_top->s_state]; D(printf(" DFA '%s', state %d:", d->d_name, ps->p_stack.s_top->s_state));

这个for loop反复拿到当前栈顶，也就是DFA和DFA状态，然后根据当前的状态和Label决定下一步的动作。基本的规则如下：

/* Check accelerator */ if (s->s_lower <= ilabel && ilabel < s->s_upper) { register int x = s->s_accel[ilabel - s->s_lower];

有对应的Accelerator，为x

1. X第8位为1，说明x对应着一个非终结符，记录着目标状态的DFA ID和状态

DFA ID

1

目标状态，7bit

在这种情况下会跳转到另外一个DFA，把目标DFA+状态压栈。

if (x != -1) { if (x & (1<<7)) { /* Push non-terminal */ int nt = (x >> 8) + NT_OFFSET; int arrow = x & ((1<<7)-1); dfa *d1 = PyGrammar_FindDFA( ps->p_grammar, nt); if ((err = push(&ps->p_stack, nt, d1, arrow, lineno, col_offset)) > 0) { D(printf(" MemError: push/n")); return err; } D(printf(" Push .../n")); continue; }

2. 否则，x对应终结符，调用Shift来改变栈顶的状态并把结点添加到CST中。如果栈顶对应着一个Accept状态的话，说明当前DFA已经匹配完毕，反复退栈直到当前状态不为Accept状态为止。

/* Shift the token */ if ((err = shift(&ps->p_stack, type, str, x, lineno, col_offset)) > 0) { D(printf(" MemError: shift./n")); return err; } D(printf(" Shift./n")); /* Pop while we are in an accept-only state */ while (s = &d->d_state [ps->p_stack.s_top->s_state], s->s_accept && s->s_narcs == 1) { D(printf(" DFA '%s', state %d: " "Direct pop./n", d->d_name, ps->p_stack.s_top->s_state)); #ifdef PY_PARSER_REQUIRES_FUTURE_KEYWORD if (d->d_name[0] == 'i' && strcmp(d->d_name, "import_stmt") == 0) future_hack(ps); #endif s_pop(&ps->p_stack); if (s_empty(&ps->p_stack)) { D(printf(" ACCEPT./n")); return E_DONE; } d = ps->p_stack.s_top->s_dfa; } return E_OK;

如果没有Acclerator，有可能该结点已经是Accept状态，同样退栈：

if (s->s_accept) { #ifdef PY_PARSER_REQUIRES_FUTURE_KEYWORD if (d->d_name[0] == 'i' && strcmp(d->d_name, "import_stmt") == 0) future_hack(ps); #endif /* Pop this dfa and try again */ s_pop(&ps->p_stack); D(printf(" Pop .../n")); if (s_empty(&ps->p_stack)) { D(printf(" Error: bottom of stack./n")); return E_SYNTAX; } continue; }

否则，语法错误，假如可能遇到的token只有一种可能的话，设置expected_ret为当前我们所期望看到的token，这样Python才可以显示语法错误信息。

/* Stuck, report syntax error */ D(printf(" Error./n")); if (expected_ret) { if (s->s_lower == s->s_upper - 1) { /* Only one possible expected token */ *expected_ret = ps->p_grammar-> g_ll.ll_label[s->s_lower].lb_type; } else *expected_ret = -1; } return E_SYNTAX;

至此PyParser已经分析完毕。下一篇文章将着重分析从CST到AST转换过程。

作者: ATField
E-Mail: atfield_zhang@hotmail.com
Blog: http://blog.csdn.net/atfield