diff options
| -rw-r--r-- | src/lib_io.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/lib_io.c b/src/lib_io.c index e9472ba5..5369d450 100644 --- a/src/lib_io.c +++ b/src/lib_io.c | |||
| @@ -17,6 +17,7 @@ | |||
| 17 | #include "lualib.h" | 17 | #include "lualib.h" |
| 18 | 18 | ||
| 19 | #include "lj_obj.h" | 19 | #include "lj_obj.h" |
| 20 | #include "lj_gc.h" | ||
| 20 | #include "lj_err.h" | 21 | #include "lj_err.h" |
| 21 | #include "lj_buf.h" | 22 | #include "lj_buf.h" |
| 22 | #include "lj_str.h" | 23 | #include "lj_str.h" |
| @@ -153,6 +154,7 @@ static int io_file_readline(lua_State *L, FILE *fp, MSize chop) | |||
| 153 | if (n >= m - 64) m += m; | 154 | if (n >= m - 64) m += m; |
| 154 | } | 155 | } |
| 155 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); | 156 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); |
| 157 | lj_gc_check(L); | ||
| 156 | return (int)ok; | 158 | return (int)ok; |
| 157 | } | 159 | } |
| 158 | 160 | ||
| @@ -164,6 +166,7 @@ static void io_file_readall(lua_State *L, FILE *fp) | |||
| 164 | n += (MSize)fread(buf+n, 1, m-n, fp); | 166 | n += (MSize)fread(buf+n, 1, m-n, fp); |
| 165 | if (n != m) { | 167 | if (n != m) { |
| 166 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); | 168 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); |
| 169 | lj_gc_check(L); | ||
| 167 | return; | 170 | return; |
| 168 | } | 171 | } |
| 169 | } | 172 | } |
| @@ -175,6 +178,7 @@ static int io_file_readlen(lua_State *L, FILE *fp, MSize m) | |||
| 175 | char *buf = lj_buf_tmp(L, m); | 178 | char *buf = lj_buf_tmp(L, m); |
| 176 | MSize n = (MSize)fread(buf, 1, m, fp); | 179 | MSize n = (MSize)fread(buf, 1, m, fp); |
| 177 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); | 180 | setstrV(L, L->top++, lj_str_new(L, buf, (size_t)n)); |
| 181 | lj_gc_check(L); | ||
| 178 | return (n > 0 || m == 0); | 182 | return (n > 0 || m == 0); |
| 179 | } else { | 183 | } else { |
| 180 | int c = getc(fp); | 184 | int c = getc(fp); |
