diff options
Diffstat (limited to 'py')
| -rw-r--r-- | py/builtinimport.c | 4 | ||||
| -rw-r--r-- | py/lexer.c | 2 | ||||
| -rw-r--r-- | py/misc.h | 10 | ||||
| -rw-r--r-- | py/vstr.c | 60 |
4 files changed, 67 insertions, 9 deletions
diff --git a/py/builtinimport.c b/py/builtinimport.c index e64ca82374..7cbb46012b 100644 --- a/py/builtinimport.c +++ b/py/builtinimport.c @@ -211,13 +211,13 @@ mp_obj_t mp_builtin___import__(int n_args, mp_obj_t *args) { vstr_add_char(&path, PATH_SEP_CHAR); vstr_add_str(&path, "__init__.py"); if (mp_import_stat(vstr_str(&path)) != MP_IMPORT_STAT_FILE) { - vstr_cut_tail(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py + vstr_cut_tail_bytes(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py nlr_jump(mp_obj_new_exception_msg_varg(&mp_type_ImportError, "Per PEP-420 a dir without __init__.py (%s) is a namespace package; " "namespace packages are not supported", vstr_str(&path))); } do_load(module_obj, &path); - vstr_cut_tail(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py + vstr_cut_tail_bytes(&path, sizeof("/__init__.py") - 1); // cut off /__init__.py } else { // MP_IMPORT_STAT_FILE do_load(module_obj, &path); // TODO: We cannot just break here, at the very least, we must execute diff --git a/py/lexer.c b/py/lexer.c index be0b1883c5..26fec121c1 100644 --- a/py/lexer.c +++ b/py/lexer.c @@ -516,7 +516,7 @@ STATIC void mp_lexer_next_token_into(mp_lexer_t *lex, mp_token_t *tok, bool firs } // cut off the end quotes from the token text - vstr_cut_tail(&lex->vstr, n_closing); + vstr_cut_tail_bytes(&lex->vstr, n_closing); } else if (is_head_of_identifier(lex)) { tok->kind = MP_TOKEN_NAME; @@ -63,8 +63,8 @@ bool unichar_isxdigit(unichar c); /** variable string *********************************************/ typedef struct _vstr_t { - int alloc; - int len; + uint alloc; + uint len; char *buf; bool had_error : 1; bool fixed_buf : 1; @@ -94,7 +94,11 @@ void vstr_add_str(vstr_t *vstr, const char *str); void vstr_add_strn(vstr_t *vstr, const char *str, int len); //void vstr_add_le16(vstr_t *vstr, unsigned short v); //void vstr_add_le32(vstr_t *vstr, unsigned int v); -void vstr_cut_tail(vstr_t *vstr, int len); +void vstr_ins_byte(vstr_t *vstr, uint byte_pos, byte b); +void vstr_ins_char(vstr_t *vstr, uint char_pos, unichar chr); +void vstr_cut_head_bytes(vstr_t *vstr, uint bytes_to_cut); +void vstr_cut_tail_bytes(vstr_t *vstr, uint bytes_to_cut); +void vstr_cut_out_bytes(vstr_t *vstr, uint byte_pos, uint bytes_to_cut); void vstr_printf(vstr_t *vstr, const char *fmt, ...); /** non-dynamic size-bounded variable buffer/string *************/ @@ -159,12 +159,12 @@ char *vstr_add_len(vstr_t *vstr, int len) { return buf; } -void vstr_add_byte(vstr_t *vstr, byte v) { +void vstr_add_byte(vstr_t *vstr, byte b) { byte *buf = (byte*)vstr_add_len(vstr, 1); if (buf == NULL) { return; } - buf[0] = v; + buf[0] = b; } void vstr_add_char(vstr_t *vstr, unichar c) { @@ -214,7 +214,48 @@ void vstr_add_le32(vstr_t *vstr, unsigned int v) { } */ -void vstr_cut_tail(vstr_t *vstr, int len) { +char *vstr_ins_blank_bytes(vstr_t *vstr, uint byte_pos, uint byte_len) { + if (vstr->had_error) { + return NULL; + } + uint l = vstr->len; + if (byte_pos > l) { + byte_pos = l; + } + if (byte_len > 0) { + // ensure room for the new bytes + if (!vstr_ensure_extra(vstr, byte_len)) { + return NULL; + } + // copy up the string to make room for the new bytes + memmove(vstr->buf + l - 1 + byte_len, vstr->buf + l - 1, l - byte_pos); + // increase the length + vstr->len += byte_len; + vstr->buf[vstr->len] = 0; + } + return vstr->buf + byte_pos; +} + +void vstr_ins_byte(vstr_t *vstr, uint byte_pos, byte b) { + char *s = vstr_ins_blank_bytes(vstr, byte_pos, 1); + if (s != NULL) { + *s = b; + } +} + +void vstr_ins_char(vstr_t *vstr, uint pos, unichar chr) { + // TODO UNICODE + char *s = vstr_ins_blank_bytes(vstr, pos, 1); + if (s != NULL) { + *s = chr; + } +} + +void vstr_cut_head_bytes(vstr_t *vstr, uint bytes_to_cut) { + vstr_cut_out_bytes(vstr, 0, bytes_to_cut); +} + +void vstr_cut_tail_bytes(vstr_t *vstr, uint len) { if (vstr->had_error) { return; } @@ -226,6 +267,19 @@ void vstr_cut_tail(vstr_t *vstr, int len) { vstr->buf[vstr->len] = 0; } +void vstr_cut_out_bytes(vstr_t *vstr, uint byte_pos, uint bytes_to_cut) { + if (vstr->had_error || byte_pos >= vstr->len) { + return; + } else if (byte_pos + bytes_to_cut >= vstr->len) { + vstr->len = byte_pos; + vstr->buf[vstr->len] = 0; + } else { + // move includes +1 for null byte at the end + memmove(vstr->buf + byte_pos, vstr->buf + byte_pos + bytes_to_cut, vstr->len - byte_pos - bytes_to_cut + 1); + vstr->len -= bytes_to_cut; + } +} + void vstr_printf(vstr_t *vstr, const char *fmt, ...) { va_list ap; va_start(ap, fmt); |