1 /* Alexey's C compiler. */
16 unsigned int line
, column
;
20 #define __noreturn __attribute((noreturn))
21 #define __printf(a, b) __attribute__((format(printf, a, b)))
24 #define __printf(a, b)
27 static void warning(struct pos
*pos
, const char *fmt
, ...) __printf(2, 3);
28 static void warning(struct pos
*pos
, const char *fmt
, ...)
32 fprintf(stderr
, "%u:%u: warning: ", pos
->line
, pos
->column
);
34 vfprintf(stderr
, fmt
, args
);
39 static void error_exit(struct pos
*pos
, const char *fmt
, ...) __printf(2, 3) __noreturn
;
40 static void error_exit(struct pos
*pos
, const char *fmt
, ...)
44 fprintf(stderr
, "%u:%u: error: ", pos
->line
, pos
->column
);
46 vfprintf(stderr
, fmt
, args
);
52 static void perror_exit(const char *fmt
, ...) __printf(1, 2) __noreturn
;
53 static void perror_exit(const char *fmt
, ...)
55 int old_errno
= errno
;
58 fputs("acc: ", stderr
);
60 vfprintf(stderr
, fmt
, args
);
68 static void _error_exit(const char *fmt
, ...) __printf(1, 2) __noreturn
;
69 static void _error_exit(const char *fmt
, ...)
73 fputs("acc: error: ", stderr
);
75 vfprintf(stderr
, fmt
, args
);
81 static void *xmalloc(size_t size
)
87 perror_exit("%s: size %zu", __func__
, size
);
91 static void *xmemdup(const void *src
, size_t n
)
100 static ssize_t
_xread(int fd
, void *buf
, size_t count
)
105 rv
= read(fd
, buf
, count
);
106 } while (rv
== -1 && (errno
== EAGAIN
|| errno
== EINTR
));
110 static void xread(int fd
, void *buf
, size_t count
)
115 rv
= _xread(fd
, buf
, count
);
117 perror_exit("read fd %d, buf %p, count %zu", fd
, buf
, count
);
119 _error_exit("fd %d truncated, buf %p, count %zu", fd
, buf
, count
);
121 buf
= (char *)buf
+ rv
;
126 static void convert_from_utf8(uint8_t *_c
, unsigned int _nr_c
, uint32_t **c
, unsigned int *nr_c
)
130 if (_nr_c
>= 0xffffffff / sizeof(uint32_t))
131 _error_exit("integer overflow _nr_c %"PRIu32
, _nr_c
);
133 /* At worse all data is ASCII. */
134 *c
= xmalloc(_nr_c
* sizeof(uint32_t));
139 static const struct {
143 { 0x80, 0x00, 0 }, /* 0xxxxxxx */
144 { 0xe0, 0xc0, 0x80 }, /* 110xxxxx 10xxxxxx */
145 { 0xf0, 0xe0, 0x800 }, /* 1110xxxx 10xxxxxx 10xxxxxx */
146 { 0xf8, 0xf0, 0x10000 }, /* 11110xxx 10xxxxxx 10xxxxxx 10xxxxxx */
148 unsigned int level
, j
;
152 while (level
< sizeof(_mask
) / sizeof(_mask
[0])) {
153 if ((_c
[i
] & _mask
[level
].mask1
) == _mask
[level
].res1
)
157 if (level
== sizeof(_mask
) / sizeof(_mask
[0]))
158 _error_exit("invalid UTF-8 octet sequence at %u: %02"PRIx8
, i
, _c
[i
]);
159 if (i
+ level
>= _nr_c
)
160 _error_exit("truncated UTF-8 octet sequence at %u: %02"PRIx8
, i
, _c
[i
]);
161 for (j
= 0; j
< level
; j
++) {
162 if ((_c
[i
+ j
+ 1] & 0xc0) != 0x80)
163 _error_exit("invalid UTF-8 octet sequence at %u: %02"PRIx8
" ... %02"PRIx8
, i
+ j
+ 1, _c
[i
], _c
[i
+ j
+ 1]);
166 ch
= _c
[i
] & ~_mask
[level
].mask1
;
167 for (j
= 0; j
< level
; j
++)
168 ch
= (ch
<< 6) | (_c
[i
+ j
+ 1] & ~0xc0);
170 if (ch
< _mask
[level
].min
)
171 _error_exit("invalid UTF-8 octet sequence at %u: %02"PRIx8
, i
, _c
[i
]);
180 /* LINE SEPARATOR to catch \n. */
181 #define LS ((uint32_t)0x2028)
183 static void fix_newline(uint32_t *c
, unsigned int *nr_c
)
189 if (c
[i
] == 0x0d && i
+ 1 < *nr_c
&& c
[i
+ 1] == 0x0a) {
190 memmove(&c
[i
], &c
[i
+ 1], *nr_c
- i
- 1);
207 static struct pos
*line_column(const uint32_t *c
, unsigned int nr_c
)
210 unsigned int line
, column
;
213 if (nr_c
>= 0xffffffff / sizeof(struct pos
))
214 _error_exit("integer overflow nr_c %u", nr_c
);
215 pos
= xmalloc(nr_c
* sizeof(struct pos
));
219 for (i
= 0; i
< nr_c
; i
++) {
221 pos
[i
].column
= column
;
232 static void warn_trigraph(const uint32_t *c
, unsigned int nr_c
, struct pos
*pos
)
237 while (i
+ 2 < nr_c
) {
238 if (c
[i
] == '?' && c
[i
+ 1] == '?') {
240 case '=':case ')':case '!':
241 case '(':case '\'':case '>':
242 case '/':case '<':case '-':
243 warning(&pos
[i
], "trigraph sequence ??%c, ignoring", c
[i
+ 2]);
254 static void delete_backslash_newline(uint32_t *c
, unsigned int *nr_c
, struct pos
*pos
)
259 while (i
+ 1 < *nr_c
) {
260 if (c
[i
] == '\\' && c
[i
+ 1] == LS
) {
261 unsigned int nr_to_move
= *nr_c
- i
- 2;
263 memmove(&c
[i
], &c
[i
+ 2], nr_to_move
* sizeof(uint32_t));
264 memmove(&pos
[i
], &pos
[i
+ 2], nr_to_move
* sizeof(struct pos
));
272 struct pp_token
*next
;
274 PP_TOKEN_IDENTIFIER
= UCHAR_MAX
+ 1,
279 #define _2(c1, c2) ((((uint32_t)c1) << 8) | ((uint32_t)c2))
280 #define _3(c1, c2, c3) ((((uint32_t)c1) << 16)| (((uint32_t)c2) << 8) | ((uint32_t)c3))
281 PP_TOKEN_DOTDOTDOT
= _3('.', '.', '.'),
282 PP_TOKEN_DEREFERENCE
= _2('-', '>'),
283 PP_TOKEN_SUB_EQ
= _2('-', '='),
284 PP_TOKEN_DEC
= _2('-', '-'),
285 PP_TOKEN_ADD_EQ
= _2('+', '='),
286 PP_TOKEN_INC
= _2('+', '+'),
287 PP_TOKEN_AND_EQ
= _2('&', '='),
288 PP_TOKEN_AND
= _2('&', '&'),
289 PP_TOKEN_MUL_EQ
= _2('*', '='),
290 PP_TOKEN_NOT_EQ
= _2('!', '='),
291 PP_TOKEN_DIV_EQ
= _2('/', '='),
292 PP_TOKEN_REM_EQ
= _2('%', '='),
293 PP_TOKEN_LSHIFT_EQ
= _3('<', '<', '='),
294 PP_TOKEN_LSHIFT
= _2('<', '<'),
295 PP_TOKEN_LEQ
= _2('<', '='),
296 PP_TOKEN_RSHIFT_EQ
= _3('>', '>', '='),
297 PP_TOKEN_RSHIFT
= _2('>', '>'),
298 PP_TOKEN_GEQ
= _2('>', '='),
299 PP_TOKEN_EQ
= _2('=', '='),
300 PP_TOKEN_XOR_EQ
= _2('^', '='),
301 PP_TOKEN_OR_EQ
= _2('|', '='),
302 PP_TOKEN_OR
= _2('|', '|'),
303 PP_TOKEN_SHARPSHARP
= _2('#', '#'),
307 uint32_t *id
; /* string representation, if type is not enough */
311 static struct pp_token
*pp_token_create(enum pp_token_type type
, struct pos
*pos
)
313 struct pp_token
*ppt
;
315 ppt
= xmalloc(sizeof(struct pp_token
));
324 static void pp_token_add(struct pp_token
*ppt
, const uint32_t *c
, unsigned int start
, unsigned int end
)
326 ppt
->id
= xmemdup(&c
[start
], (end
- start
) * sizeof(uint32_t));
329 static void pp_token_free(struct pp_token
*ppt_head
)
331 struct pp_token
*ppt
;
335 struct pp_token
*next
;
345 static void pp_token_print(struct pp_token
*ppt
)
347 printf("%u:%u:\t", ppt
->pos
.line
, ppt
->pos
.column
);
355 case PP_TOKEN_IDENTIFIER
:
356 printf("pp-identifier");
358 case PP_TOKEN_NUMBER
:
361 case PP_TOKEN_STRING
:
367 case PP_TOKEN_DOTDOTDOT
:
368 case PP_TOKEN_LSHIFT_EQ
:
369 case PP_TOKEN_RSHIFT_EQ
:
370 printf("%c%c%c", (ppt
->type
>> 16) & 0xff, (ppt
->type
>> 8) & 0xff, ppt
->type
& 0xff);
372 case PP_TOKEN_DEREFERENCE
:
373 case PP_TOKEN_SUB_EQ
:
375 case PP_TOKEN_ADD_EQ
:
377 case PP_TOKEN_AND_EQ
:
379 case PP_TOKEN_MUL_EQ
:
380 case PP_TOKEN_NOT_EQ
:
381 case PP_TOKEN_DIV_EQ
:
382 case PP_TOKEN_REM_EQ
:
383 case PP_TOKEN_LSHIFT
:
385 case PP_TOKEN_RSHIFT
:
388 case PP_TOKEN_XOR_EQ
:
391 case PP_TOKEN_SHARPSHARP
:
392 printf("%c%c", (ppt
->type
>> 8) & 0xff, ppt
->type
& 0xff);
395 printf("%c", ppt
->type
);
400 static int pp_nondigit(const uint32_t c
)
402 return ('a' <= c
&& c
<= 'z') || ('A' <= c
&& c
<= 'Z') || c
== '_';
405 static int pp_octdigit(const uint32_t c
)
407 return '0' <= c
&& c
<= '7';
410 static int pp_digit(const uint32_t c
)
412 return '0' <= c
&& c
<= '9';
415 static int pp_hexdigit(const uint32_t c
)
417 return pp_digit(c
) || ('a' <= c
&& c
<= 'f') || ('A' <= c
&& c
<= 'F');
420 static unsigned int _pp_identifier_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
)
426 if (pp_nondigit(c
[i
]) || pp_digit(c
[i
])) {
428 } else if (i
+ 5 < nr_c
&& c
[i
] == '\\' && c
[i
+ 1] == 'u' &&
429 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5])) {
431 } else if (i
+ 9 < nr_c
&& c
[i
] == '\\' && c
[i
+ 1] == 'U' &&
432 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5]) &&
433 pp_hexdigit(c
[i
+ 6]) && pp_hexdigit(c
[i
+ 7]) && pp_hexdigit(c
[i
+ 8]) && pp_hexdigit(c
[i
+ 9])) {
441 static unsigned int pp_number_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
)
447 if ((c
[i
] == 'e' || c
[i
] == 'E' || c
[i
] == 'p' || c
[i
] == 'P') &&
448 i
+ 1 < nr_c
&& (c
[i
+ 1] == '+' || c
[i
+ 1] == '-')) {
450 } else if (pp_digit(c
[i
]) || pp_nondigit(c
[i
]) || c
[i
] == '.') {
452 } else if (c
[i
] == '\\' && i
+ 5 < nr_c
&& c
[i
+ 1] == 'u' &&
453 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5])) {
455 } else if (c
[i
] == '\\' && i
+ 9 < nr_c
&& c
[i
+ 1] == 'U' &&
456 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5]) &&
457 pp_hexdigit(c
[i
+ 6]) && pp_hexdigit(c
[i
+ 7]) && pp_hexdigit(c
[i
+ 8]) && pp_hexdigit(c
[i
+ 9])) {
465 static unsigned int c_comment_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
)
470 while (i
+ 1 < nr_c
) {
471 if (c
[i
] == '*' && c
[i
+ 1] == '/')
478 static unsigned int cpp_comment_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
)
483 while (i
< nr_c
&& c
[i
] != LS
)
488 static unsigned int escape_sequence_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
, struct pos
*_pos
)
490 struct pos
*pos
= &_pos
[start
];
495 error_exit(pos
, "incomplete escape sequence");
497 case '\'':case '"':case '?':case '\\':
498 case 'a':case 'b':case 'f':case 'n':case 'r':case 't':case 'v':
500 case '0':case '1':case '2':case '3':case '4':case '6':case '7':
501 if (i
+ 2 < nr_c
&& pp_octdigit(c
[i
+ 1]) && pp_octdigit(c
[i
+ 2]))
503 if (i
+ 1 < nr_c
&& pp_octdigit(c
[i
+ 1]))
508 while (i
< nr_c
&& pp_hexdigit(c
[i
]))
511 error_exit(pos
, "invalid hexadecimal escape sequence");
515 pp_hexdigit(c
[i
+ 1]) && pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]))
517 error_exit(pos
, "invalid universal character name");
520 pp_hexdigit(c
[i
+ 1]) && pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) &&
521 pp_hexdigit(c
[i
+ 5]) && pp_hexdigit(c
[i
+ 6]) && pp_hexdigit(c
[i
+ 7]) && pp_hexdigit(c
[i
+ 8]))
523 error_exit(pos
, "invalid universal character name");
525 error_exit(pos
, "invalid escape sequence");
529 static unsigned int pp_string_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
, struct pos
*_pos
)
531 struct pos
*pos
= &_pos
[start
];
535 while (i
< nr_c
&& c
[i
] != '"') {
540 i
= escape_sequence_end(c
, nr_c
, i
, _pos
);
551 error_exit(pos
, "incomplete string literal");
554 static unsigned int pp_char_end(const uint32_t *c
, unsigned int nr_c
, unsigned int start
, struct pos
*_pos
)
556 struct pos
*pos
= &_pos
[start
];
560 while (i
< nr_c
&& c
[i
] != '\'') {
565 i
= escape_sequence_end(c
, nr_c
, i
, _pos
);
578 error_exit(pos
, "incomplete character constant");
580 error_exit(pos
, "empty character constant");
583 static struct pp_token
*pp_tokenize(const uint32_t *c
, unsigned int nr_c
, struct pos
*_pos
)
585 struct pp_token
*ppt_head
, *ppt_tail
;
591 struct pos
*pos
= &_pos
[i
];
592 struct pp_token
*ppt
;
599 ppt
= pp_token_create(' ', pos
);
603 ppt
= pp_token_create(LS
, pos
);
615 ppt
= pp_token_create(c
[i
], pos
);
618 case 'a':case 'b':case 'c':case 'd':case 'e':case 'f':case 'g':
619 case 'h':case 'i':case 'j':case 'k':case 'l':case 'm':case 'n':
620 case 'o':case 'p':case 'q':case 'r':case 's':case 't':case 'u':
621 case 'v':case 'w':case 'x':case 'y':case 'z':
622 case 'A':case 'B':case 'C':case 'D':case 'E':case 'F':case 'G':
623 case 'H':case 'I':case 'J':case 'K':case 'L':case 'M':case 'N':
624 case 'O':case 'P':case 'Q':case 'R':case 'S':case 'T':case 'U':
625 case 'V':case 'W':case 'X':case 'Y':case 'Z':
627 ppt
= pp_token_create(PP_TOKEN_IDENTIFIER
, pos
);
628 j
= _pp_identifier_end(c
, nr_c
, i
+ 1);
629 pp_token_add(ppt
, c
, i
, j
);
633 if (i
+ 5 < nr_c
&& c
[i
+ 1] == 'u' &&
634 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5])) {
635 ppt
= pp_token_create(PP_TOKEN_IDENTIFIER
, pos
);
636 j
= _pp_identifier_end(c
, nr_c
, i
+ 2 + 4);
637 pp_token_add(ppt
, c
, i
, j
);
639 } else if (i
+ 9 < nr_c
&& c
[i
+ 1] == 'U' &&
640 pp_hexdigit(c
[i
+ 2]) && pp_hexdigit(c
[i
+ 3]) && pp_hexdigit(c
[i
+ 4]) && pp_hexdigit(c
[i
+ 5]) &&
641 pp_hexdigit(c
[i
+ 6]) && pp_hexdigit(c
[i
+ 7]) && pp_hexdigit(c
[i
+ 8]) && pp_hexdigit(c
[i
+ 9])) {
642 ppt
= pp_token_create(PP_TOKEN_IDENTIFIER
, pos
);
643 j
= _pp_identifier_end(c
, nr_c
, i
+ 2 + 4 + 4);
644 pp_token_add(ppt
, c
, i
, j
);
647 error_exit(pos
, "unknown character %08"PRIx32
, c
[i
]);
649 case '0':case '1':case '2':case '3':case '4':
650 case '5':case '6':case '7':case '8':case '9':
651 ppt
= pp_token_create(PP_TOKEN_NUMBER
, pos
);
652 j
= pp_number_end(c
, nr_c
, i
);
653 pp_token_add(ppt
, c
, i
, j
);
657 if (i
+ 2 < nr_c
&& c
[i
+ 1] == '.' && c
[i
+ 2] == '.') {
658 ppt
= pp_token_create(PP_TOKEN_DOTDOTDOT
, pos
);
660 } else if (i
+ 1 < nr_c
&& pp_digit(c
[i
+ 1])) {
661 ppt
= pp_token_create(PP_TOKEN_NUMBER
, pos
);
662 j
= pp_number_end(c
, nr_c
, i
+ 1);
663 pp_token_add(ppt
, c
, i
, j
);
666 goto pp_token_simple
;
669 ppt
= pp_token_create(PP_TOKEN_STRING
, pos
);
670 j
= pp_string_end(c
, nr_c
, i
, _pos
);
671 pp_token_add(ppt
, c
, i
+ 1, j
- 1);
675 ppt
= pp_token_create(PP_TOKEN_CHAR
, pos
);
676 j
= pp_char_end(c
, nr_c
, i
, _pos
);
677 pp_token_add(ppt
, c
, i
+ 1, j
- 1);
681 if (i
+ 1 < nr_c
&& c
[i
+ 1] == '*') {
682 ppt
= pp_token_create(' ', pos
);
683 i
= c_comment_end(c
, nr_c
, i
);
684 } else if (i
+ 1 < nr_c
&& c
[i
+ 1] == '/') {
685 warning(pos
, "C++ comment");
686 ppt
= pp_token_create(' ', pos
);
687 i
= cpp_comment_end(c
, nr_c
, i
);
688 } else if (i
+ 1 < nr_c
&& c
[i
+ 1] == '=') {
689 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
692 goto pp_token_simple
;
695 if (i
+ 1 < nr_c
&& (c
[i
+ 1] == '>' || c
[i
+ 1] == '=' || c
[i
+ 1] == '-')) {
696 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
699 goto pp_token_simple
;
704 if (i
+ 1 < nr_c
&& (c
[i
+ 1] == '=' || c
[i
+ 1] == c
[i
])) {
705 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
708 goto pp_token_simple
;
715 if (i
+ 1 < nr_c
&& c
[i
+ 1] == '=') {
716 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
719 goto pp_token_simple
;
723 if (i
+ 2 < nr_c
&& c
[i
+ 1] == c
[i
] && c
[i
+ 2] == '=') {
724 ppt
= pp_token_create((c
[i
] << 16) | (c
[i
+ 1] << 8) | c
[i
+ 2], pos
);
726 } else if (i
+ 1 < nr_c
&& (c
[i
+ 1] == c
[i
] || c
[i
+ 1] == '=')) {
727 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
730 goto pp_token_simple
;
733 if (i
+ 1 < nr_c
&& c
[i
+ 1] == '#') {
734 ppt
= pp_token_create((c
[i
] << 8) | c
[i
+ 1], pos
);
737 goto pp_token_simple
;
740 error_exit(pos
, "unknown character %08"PRIx32
, c
[i
]);
746 ppt_tail
->next
= ppt
;
752 int main(int argc
, char *argv
[])
756 unsigned int st_size
;
763 struct pp_token
*ppt_head
;
768 fd
= open(argv
[1], O_RDONLY
);
770 perror_exit("open %s", argv
[1]);
771 if (fstat(fd
, &st
) == -1)
772 perror_exit("fstat %s", argv
[1]);
774 _error_exit("%s: negative st_size %"PRIdMAX
, argv
[1], (intmax_t)st
.st_size
);
775 st_size
= (unsigned int)(uintmax_t)(intmax_t)st
.st_size
;
776 if ((uintmax_t)(intmax_t)st
.st_size
!= (uintmax_t)st_size
)
777 _error_exit("%s: too big st_size %"PRIdMAX
, argv
[1], (intmax_t)st
.st_size
);
779 buf
= xmalloc(st_size
);
780 xread(fd
, buf
, st_size
);
785 /* Skip UTF-8 "BOM" if any. */
786 if (st_size
>= 3 && _c
[0] == 0xef && _c
[1] == 0xbb && _c
[2] == 0xbf) {
790 convert_from_utf8(_c
, _nr_c
, &c
, &nr_c
);
793 fix_newline(c
, &nr_c
);
794 pos
= line_column(c
, nr_c
);
795 warn_trigraph(c
, nr_c
, pos
);
796 delete_backslash_newline(c
, &nr_c
, pos
);
798 ppt_head
= pp_tokenize(c
, nr_c
, pos
);
803 struct pp_token
*ppt
;
805 for (ppt
= ppt_head
; ppt
; ppt
= ppt
->next
)
809 pp_token_free(ppt_head
);