gbdev/rgbds
1
0
Fork 0
mirror of https://github.com/gbdev/rgbds.git synced 2025-11-20 18:22:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle non-seekable input correctly

Browse Source
This commit is contained in:
Jakub Kądziołka
2019-06-05 20:25:24 +02:00
parent 20f9492899
commit 8d5a53f529
2 changed files with 26 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/asm/lexer.h
View File

@@ -31,7 +31,7 @@ struct yy_buffer_state {
/* Address where the data is initially written after a safety margin */
char *pBufferStart;
char *pBuffer;
uint32_t nBufferSize;
size_t nBufferSize;
uint32_t oAtLineStart;
};

34
src/asm/lexer.c
View File

@@ -156,25 +156,39 @@ YY_BUFFER_STATE yy_create_buffer(FILE *f)
if (pBuffer == NULL)
fatalerror("%s: Out of memory!", __func__);
uint32_t size;
size_t size = 0, capacity;
char *buf = NULL;
fseek(f, 0, SEEK_END);
size = ftell(f);
fseek(f, 0, SEEK_SET);
capacity = ftell(f);
rewind(f);
if (capacity == -1)
capacity = 4096;
while (!feof(f)) {
if (buf == NULL || size >= capacity) {
capacity *= 2;
/* Give extra room for 2 newlines and terminator */
uint32_t capacity = size + 3;
buf = realloc(buf, capacity + SAFETYMARGIN + 3);
pBuffer->pBufferRealStart = malloc(capacity + SAFETYMARGIN);
if (buf == NULL)
fatalerror("%s: Out of memory for buffer!",
__func__);
}
if (pBuffer->pBufferRealStart == NULL)
fatalerror("%s: Out of memory for buffer!", __func__);
char *bufpos = buf + SAFETYMARGIN + size;
size_t read_count = fread(bufpos, 1, capacity - size, f);
pBuffer->pBufferStart = pBuffer->pBufferRealStart + SAFETYMARGIN;
pBuffer->pBuffer = pBuffer->pBufferRealStart + SAFETYMARGIN;
if (read_count == 0)
fatalerror("%s: fread error", __func__);
size = fread(pBuffer->pBuffer, sizeof(uint8_t), size, f);
size += read_count;
}
pBuffer->pBufferRealStart = buf;
pBuffer->pBufferStart = buf + SAFETYMARGIN;
pBuffer->pBuffer = buf + SAFETYMARGIN;
pBuffer->pBuffer[size] = 0;
pBuffer->nBufferSize = size;