-
Notifications
You must be signed in to change notification settings - Fork 0
/
type.c
105 lines (95 loc) · 2.17 KB
/
type.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include "chibicc.h"
Type *ty_int = &(Type){TY_INT, 4};
Type *ty_char = &(Type){TY_CHAR, 1};
Type *ty_short = &(Type){TY_SHORT, 2};
Type *ty_long = &(Type){TY_LONG, 8};
Type *ty_void = &(Type){TY_VOID, 1};
Type *get_type_can_null(Token *tok) {
if (equal(tok, "int"))
return ty_int;
if (equal(tok, "char"))
return ty_char;
if (equal(tok, "short"))
return ty_short;
if (equal(tok, "long"))
return ty_long;
if (equal(tok, "void"))
return ty_void;
return NULL;
}
Type *func_type(Type *return_ty) {
Type *ty = calloc(1, sizeof(Type));
ty -> kind = TY_FUNC;
ty -> return_ty = return_ty;
return ty;
}
Type *pointer_to(Type *base) {
Type *ty = calloc(1, sizeof(Type));
ty -> kind = TY_PTR;
ty -> size = 8;
ty -> base = base;
return ty;
}
Type *array_of(Type *base, int len) {
Type *ty = calloc(1, sizeof(Type));
ty -> kind = TY_ARRAY;
ty -> size = base->size * len;
ty -> base = base;
ty -> array_len = len;
return ty;
}
bool is_integer(Type *ty) {
return ty -> kind == TY_INT || ty -> kind == TY_CHAR
|| ty -> kind == TY_SHORT || ty -> kind == TY_LONG;
}
void add_type(Node *node) {
if (!node || node->ty)
return;
add_type(node->lhs);
add_type(node->rhs);
add_type(node->cond);
add_type(node->then);
add_type(node->_else);
for (Node *n = node->body; n; n = n->next)
add_type(n);
switch (node->kind) {
case ND_ADD:
case ND_SUB:
case ND_MUL:
case ND_DIV:
case ND_NEG:
case ND_ASSIGN:
node->ty = node->lhs->ty;
return;
case ND_EQ:
case ND_NE:
case ND_LT:
case ND_LE:
case ND_NUM:
case ND_FUNCALL:
node -> ty = ty_int;
return;
case ND_VAR:
node -> ty = node -> var -> ty;
return;
case ND_COMMA:
node -> ty = node -> rhs -> ty;
return;
case ND_FIELD:
node -> ty = node -> field -> ty;
return;
case ND_ADDR:
if (node -> lhs -> ty -> kind == TY_ARRAY)
node->ty = pointer_to(node->lhs->ty->base);
else
node->ty = pointer_to(node->lhs->ty);
return;
case ND_DEREF:
if (node->lhs->ty->base == NULL)
error("invalid pointer in type.c:53");
if (node -> lhs -> ty -> base -> kind == TY_VOID)
error("dereferencing a void pointer");
node -> ty = node -> lhs -> ty -> base;
return;
}
}