walkero
/
physfs
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

physfshttpd: deal with write errors on the socket.

This commit is contained in:
Ryan C. Gordon 2017-08-06 01:34:18 -04:00
parent 61bdee4d78
commit 121ee38fde
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
extras/physfshttpd.c
View File

@ -78,22 +78,30 @@ static char *txt200 =
"Content-Type: text/plain; charset=utf-8\n"
"\n";
static int writeAll(const int fd, const void *buf, const size_t len)
{
return (write(fd, buf, len) == len);
} /* writeAll */
static void feed_file_http(const char *ipstr, int sock, const char *fname)
{
PHYSFS_File *in = PHYSFS_openRead(fname);
char buffer[1024];
printf("%s: requested [%s].\n", ipstr, fname);
if (in == NULL)
{
printf("%s: Can't open [%s]: %s.\n",
ipstr, fname, PHYSFS_getLastError());
write(sock, txt404, strlen(txt404)); /* !!! FIXME: Check retval */
if (!writeAll(sock, txt404, strlen(txt404)))
printf("%s: Write error to socket.\n", ipstr);
return;
} /* if */
else
if (writeAll(sock, txt200, strlen(txt200)))
{
write(sock, txt200, strlen(txt200)); /* !!! FIXME: Check retval */
do
{
char buffer[1024];
PHYSFS_sint64 br = PHYSFS_readBytes(in, buffer, sizeof (buffer));
if (br == -1)
{
@ -101,11 +109,15 @@ static void feed_file_http(const char *ipstr, int sock, const char *fname)
break;
} /* if */
write(sock, buffer, (int) br); /* !!! FIXME: CHECK THIS RETVAL! */
else if (!writeAll(sock, buffer, (size_t) br))
{
printf("%s: Write error to socket.\n", ipstr);
break;
} /* else if */
} while (!PHYSFS_eof(in));
} /* if */
PHYSFS_close(in);
} /* else */
PHYSFS_close(in);
} /* feed_file_http */