Make http implementation more robust

2 files changed, 1 insertions, 3 deletions

diff --git a/TODO b/TODO
index cb862d5..b4bc96c 100644
--- a/TODO
+++ b/TODO
@@ -6,8 +6,6 @@ use the correct file extension depending on the image type (for google and amazo
 
 bug in get() & post(): if the \r\n\r\n after http headers is cut off between recv buffers
 
-what happens if we receive not a http header? does recv loop forever, in a memory killing manner?
-
 package for osx - https://github.com/kennethreitz/osx-gcc-installer
 
 add djvu convert script
diff --git a/util.c b/util.c
index 207024a..b1f4c69 100644
--- a/util.c
+++ b/util.c
@@ -92,7 +92,7 @@ int get(char *host, char *path, char *sendcookie, char *savecookie, char **buf)
 	}
 
 	*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));
-	for(; (res = recv(fd, *buf+l, BUFSIZ, 0)) > 0; l+=res)
+	for(; buf != NULL && (res = recv(fd, *buf+l, BUFSIZ, 0)) > 0; l+=res)
 		*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));
 
 	return l;