dan
/
liminix
1
0
Fork 5
Code Issues 2 Pull Requests 1 Projects Releases Wiki Activity

fetch lua glue: handle missing content-length

This commit is contained in:
Daniel Barlow 2024-08-28 19:52:00 +01:00
parent d5d621f310
commit b56f121e04
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
pkgs/fetch-freebsd/lua-glue.c
View File

@ -21,19 +21,34 @@ int lp_error(lua_State *L, int code, char *message) {
} }
int lp_fetched(lua_State *L, FXRETTYPE fp, struct url_stat *stat) { int lp_fetched(lua_State *L, FXRETTYPE fp, struct url_stat *stat) {
void * content = malloc(stat->size); size_t bytes_read = 0;
size_t bytes_read; char * content = 0;
es_read(fp, content, stat->size, &bytes_read); if(stat->size > 0) {
content = (char *) malloc(stat->size);
es_read(fp, content, stat->size, &bytes_read);
if(bytes_read != stat->size) { if(bytes_read != stat->size) {
lp_error(L, -1, "bytes read does not match content-length"); lp_error(L, -1, "bytes read does not match content-length");
}
} else {
/* no content-length header */
int block_size = 128;
int buf_size = block_size;
size_t bytes_read_block;
do {
content = (char *) realloc(content, buf_size);
es_read(fp, content + bytes_read, block_size, &bytes_read_block);
bytes_read += bytes_read_block;
buf_size += block_size;
} while(bytes_read_block > 0);
} }
lua_pushlstring(L, (const char *) content, stat->size);
lua_pushlstring(L, (const char *) content, bytes_read);
lua_newtable(L); lua_newtable(L);
lua_pushliteral(L, "last-modified"); lua_pushliteral(L, "last-modified");
lua_pushinteger(L, stat->mtime); lua_pushinteger(L, stat->mtime);
lua_settable(L, -3); lua_settable(L, -3);
/* free(content) ? */
return 2; return 2;
} }