mirror of https://github.com/rui314/chibicc
136 lines
2.4 KiB
C
136 lines
2.4 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
#include <assert.h>
|
|
#include <ctype.h>
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
typedef struct Type Type;
|
|
typedef struct Node Node;
|
|
|
|
//
|
|
// tokenize.c
|
|
//
|
|
|
|
// Token
|
|
typedef enum {
|
|
TK_IDENT, // Identifiers
|
|
TK_PUNCT, // Punctuators
|
|
TK_KEYWORD, // Keywords
|
|
TK_NUM, // Numeric literals
|
|
TK_EOF, // End-of-file markers
|
|
} TokenKind;
|
|
|
|
// Token type
|
|
typedef struct Token Token;
|
|
struct Token {
|
|
TokenKind kind; // Token kind
|
|
Token *next; // Next token
|
|
int val; // If kind is TK_NUM, its value
|
|
char *loc; // Token location
|
|
int len; // Token length
|
|
};
|
|
|
|
void error(char *fmt, ...);
|
|
void error_at(char *loc, char *fmt, ...);
|
|
void error_tok(Token *tok, char *fmt, ...);
|
|
bool equal(Token *tok, char *op);
|
|
Token *skip(Token *tok, char *op);
|
|
Token *tokenize(char *input);
|
|
|
|
//
|
|
// parse.c
|
|
//
|
|
|
|
// Local variable
|
|
typedef struct Obj Obj;
|
|
struct Obj {
|
|
Obj *next;
|
|
char *name; // Variable name
|
|
int offset; // Offset from RBP
|
|
};
|
|
|
|
// Function
|
|
typedef struct Function Function;
|
|
struct Function {
|
|
Node *body;
|
|
Obj *locals;
|
|
int stack_size;
|
|
};
|
|
|
|
// AST node
|
|
typedef enum {
|
|
ND_ADD, // +
|
|
ND_SUB, // -
|
|
ND_MUL, // *
|
|
ND_DIV, // /
|
|
ND_NEG, // unary -
|
|
ND_EQ, // ==
|
|
ND_NE, // !=
|
|
ND_LT, // <
|
|
ND_LE, // <=
|
|
ND_ASSIGN, // =
|
|
ND_ADDR, // unary &
|
|
ND_DEREF, // unary *
|
|
ND_RETURN, // "return"
|
|
ND_IF, // "if"
|
|
ND_FOR, // "for" or "while"
|
|
ND_BLOCK, // { ... }
|
|
ND_EXPR_STMT, // Expression statement
|
|
ND_VAR, // Variable
|
|
ND_NUM, // Integer
|
|
} NodeKind;
|
|
|
|
// AST node type
|
|
struct Node {
|
|
NodeKind kind; // Node kind
|
|
Node *next; // Next node
|
|
Type *ty; // Type, e.g. int or pointer to int
|
|
Token *tok; // Representative token
|
|
|
|
Node *lhs; // Left-hand side
|
|
Node *rhs; // Right-hand side
|
|
|
|
// "if" or "for" statement
|
|
Node *cond;
|
|
Node *then;
|
|
Node *els;
|
|
Node *init;
|
|
Node *inc;
|
|
|
|
// Block
|
|
Node *body;
|
|
|
|
Obj *var; // Used if kind == ND_VAR
|
|
int val; // Used if kind == ND_NUM
|
|
};
|
|
|
|
Function *parse(Token *tok);
|
|
|
|
//
|
|
// type.c
|
|
//
|
|
|
|
typedef enum {
|
|
TY_INT,
|
|
TY_PTR,
|
|
} TypeKind;
|
|
|
|
struct Type {
|
|
TypeKind kind;
|
|
Type *base;
|
|
};
|
|
|
|
extern Type *ty_int;
|
|
|
|
bool is_integer(Type *ty);
|
|
void add_type(Node *node);
|
|
|
|
//
|
|
// codegen.c
|
|
//
|
|
|
|
void codegen(Function *prog);
|