Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 1 | #include <stdint.h> |
| 2 | #include <stdio.h> |
| 3 | #include <unistd.h> |
| 4 | #include <fcntl.h> |
| 5 | |
| 6 | #include "misc.h" |
| 7 | #include "lexer.h" |
| 8 | |
| 9 | typedef struct _str_buf_t { |
| 10 | bool free; // free src_beg when done |
| 11 | const char *src_beg; // beginning of source |
| 12 | const char *src_cur; // current location in source |
| 13 | const char *src_end; // end (exclusive) of source |
| 14 | } str_buf_t; |
| 15 | |
| 16 | unichar str_buf_next_char(str_buf_t *sb) { |
| 17 | if (sb->src_cur < sb->src_end) { |
| 18 | return *sb->src_cur++; |
| 19 | } else { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 20 | return MP_LEXER_CHAR_EOF; |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 21 | } |
| 22 | } |
| 23 | |
| 24 | void str_buf_free(str_buf_t *sb) { |
| 25 | if (sb) { |
| 26 | if (sb->free) { |
| 27 | m_free((char*)sb->src_beg); |
| 28 | } |
| 29 | m_free(sb); |
| 30 | } |
| 31 | } |
| 32 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 33 | mp_lexer_t *mp_lexer_new_from_str_len(const char *src_name, const char *str, uint len, bool free_str) { |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 34 | str_buf_t *sb = m_new(str_buf_t, 1); |
| 35 | sb->free = free_str; |
| 36 | sb->src_beg = str; |
| 37 | sb->src_cur = str; |
| 38 | sb->src_end = str + len; |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 39 | return mp_lexer_new(src_name, sb, (mp_lexer_stream_next_char_t)str_buf_next_char, (mp_lexer_stream_close_t)str_buf_free); |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 40 | } |
| 41 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 42 | mp_lexer_t *mp_lexer_new_from_file(const char *filename) { |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 43 | int fd = open(filename, O_RDONLY); |
| 44 | if (fd < 0) { |
| 45 | printf("cannot open file %s\n", filename); |
| 46 | return NULL; |
| 47 | } |
| 48 | uint size = lseek(fd, 0, SEEK_END); |
| 49 | lseek(fd, 0, SEEK_SET); |
| 50 | char *data = m_new(char, size); |
Paul Sokolovsky | 44d0f3c | 2013-12-29 20:11:05 +0200 | [diff] [blame] | 51 | int read_size = read(fd, data, size); |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 52 | close(fd); |
Paul Sokolovsky | 44d0f3c | 2013-12-29 20:11:05 +0200 | [diff] [blame] | 53 | if (read_size != size) { |
| 54 | printf("error reading file %s\n", filename); |
| 55 | m_free(data); |
| 56 | return NULL; |
| 57 | } |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 58 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 59 | return mp_lexer_new_from_str_len(filename, data, size, true); |
Damien | a5185f4 | 2013-10-20 14:41:27 +0100 | [diff] [blame] | 60 | } |