blob: 80daf6009a5c82b04350495e37edfae32a617313 [file] [log] [blame]
Damiena5185f42013-10-20 14:41:27 +01001#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
9typedef 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
16unichar str_buf_next_char(str_buf_t *sb) {
17 if (sb->src_cur < sb->src_end) {
18 return *sb->src_cur++;
19 } else {
Damiend99b0522013-12-21 18:17:45 +000020 return MP_LEXER_CHAR_EOF;
Damiena5185f42013-10-20 14:41:27 +010021 }
22}
23
24void 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
Damiend99b0522013-12-21 18:17:45 +000033mp_lexer_t *mp_lexer_new_from_str_len(const char *src_name, const char *str, uint len, bool free_str) {
Damiena5185f42013-10-20 14:41:27 +010034 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;
Damiend99b0522013-12-21 18:17:45 +000039 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);
Damiena5185f42013-10-20 14:41:27 +010040}
41
Damiend99b0522013-12-21 18:17:45 +000042mp_lexer_t *mp_lexer_new_from_file(const char *filename) {
Damiena5185f42013-10-20 14:41:27 +010043 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 Sokolovsky44d0f3c2013-12-29 20:11:05 +020051 int read_size = read(fd, data, size);
Damiena5185f42013-10-20 14:41:27 +010052 close(fd);
Paul Sokolovsky44d0f3c2013-12-29 20:11:05 +020053 if (read_size != size) {
54 printf("error reading file %s\n", filename);
55 m_free(data);
56 return NULL;
57 }
Damiena5185f42013-10-20 14:41:27 +010058
Damiend99b0522013-12-21 18:17:45 +000059 return mp_lexer_new_from_str_len(filename, data, size, true);
Damiena5185f42013-10-20 14:41:27 +010060}