fix some bugs

[record-dvb] / http.c

diff --git a/http.c b/http.c
index ca756bd5a56cfa57897d047db6d4d934b2e28dec..c7e553cc7511bd7769356fb77c68c00bf52b6791 100644 (file)
--- a/http.c
+++ b/http.c
@@ -27,17 +27,15 @@ int open_http(char *url)
 {
        int fd;
        struct sockaddr_in server;
-       static struct dvb_host *dvbhost = NULL;
-       char buffer[BUFFSIZE];
-       int i;
+       struct dvb_host *dvbhost = NULL;
+       char c, buffer[BUFFSIZE], *pos;
+       int lines = 0;
 
        if(!is_http(url))
                return -1;
 
-       if (!dvbhost) {
-               dvbhost = parse(&(url[7]), "80");
-               dvbhost->socktype = SOCK_STREAM;
-       }
+       dvbhost = parse(&(url[7]), "80");
+       dvbhost->socktype = SOCK_STREAM;
 
        if (resolve(dvbhost, &server) < 0) {
                return -1;
@@ -55,19 +53,61 @@ int open_http(char *url)
 
        snprintf(buffer, BUFFSIZE-1, "GET /%s HTTP/1.0\n\n",dvbhost->location);
        buffer[BUFFSIZE-1] = 0;
-       send(fd, buffer, strlen(buffer), 0);
-       i = 0;
-       while(i < 2) {
-               if (recv(fd, buffer, 1, 0) < 1) {
+       if (send(fd, buffer, strlen(buffer), 0) < 0) {
+               perror("send");
+               return -1;
+       }
+
+       pos = buffer;
+       while (1) {
+               if (recv(fd, &c, 1, 0) < 1) {
                        perror("recv");
                        exit(EXIT_FAILURE);
                }
-               printf("%c",buffer[0]);
-               if (buffer[0] == 0x0a)
-                       i++;
-               else
-                       if (buffer[0] != 0x0d)
-                               i = 0;
+
+               if (pos-buffer >= BUFFSIZE)
+                       pos = buffer;
+
+               *(pos++) = c;
+
+               if (c == 0x0a) {
+                       *(--pos) = 0;
+                       if (pos-buffer > 0 && *(--pos) == 0x0d)
+                               *pos = 0;
+
+                       if (pos-buffer == 0)
+                               break;
+
+#ifdef DEBUG
+                       printf("%d. %s (%d)\n", lines, buffer, pos-buffer);
+#endif
+                       if (lines == 0) {
+                               if (strncasecmp("HTTP/", buffer, 5)) {
+                                       fprintf(stderr, "Wrong answer from server: %s\n", buffer);
+                                       return -1;
+                               }
+
+                               pos = buffer;
+                               while (*pos != 0) {
+                                       if (*(pos++) == ' ') {
+                                               if(strncmp("200", pos, 3)) {
+                                                       fprintf(stderr, "Wrong result-code: %s\n", buffer);
+                                                       return -1;
+                                               } else {
+                                                       break;
+                                               }
+                                       }
+
+                                       if (*(++pos) == 0) {
+                                               fprintf(stderr, "Wrong answer from server: %s\n", buffer);
+                                               return -1;
+                                       }
+                               }
+                       }
+
+                       pos = buffer;
+                       lines++;
+               }
        }
 
        return fd;