[HaBo]

 
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <string.h>
#include <netinet/in.h>

#define SERVER_OF_BOT "not set"
#define PATH_TO_SIDFILE "not set"

int recvstring(int sockfd, char *buffer, int size){
    int recvd = 1, max;
    max = size;
    while(recvd != 0 && max > 0){
        if((recvd = recv(sockfd, buffer, 1024, 0))==-1)
            return -1;
        max -= recvd;
    }
    return size-max;
}

int sendstring(int sockfd, char *msg){
    int send_n_bytes = strlen(msg), sent;
    while(send_n_bytes != 0){
        if((sent = send(sockfd, msg, strlen(msg), 0))==-1)
            return -1;
        send_n_bytes -= sent;
    }
    return strlen(msg);
}

void usage(char *name){
    printf("usage:%s <HOST> <PATH>\n", name);
    exit(0);
}

void err_quit(char *err){
    printf("ERROR: %s\n", err);
    exit(0);
}


int main(int argc, char **argv){
    int sockfd, status;
    struct addrinfo hints, *res;
    char *server, recvbuffer[40000], *path;
    short flag = 0;
    char query[128];
    FILE *fp;

    if(argc != 3)
        flag++;
    bzero(&hints, sizeof(hints));
    hints.ai_family = AF_INET;
    hints.ai_socktype = SOCK_STREAM;
    
    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
        err_quit("in socket()\n");
                                         
    if(!strcmp(SERVER_OF_BOT, "not set")){
            server = argv[1];
    }else
        server = SERVER_OF_BOT;
    if((status = getaddrinfo(server, "80", &hints, &res)) != 0){
        printf("%s\n", gai_strerror(status));
        exit(0);
    }

    if((status = connect(sockfd, res->ai_addr, res->ai_addrlen)) == -1)
       err_quit("while connecting");

    if(!strcmp(PATH_TO_SIDFILE, "not set")){
            if(flag)
                usage(argv[0]);
            path = argv[2];
    }else
       path = PATH_TO_SIDFILE; 


    strcat(query, "GET /");
    strcat(query, path);
    strcat(query, " HTTP/1.0\n");
    strcat(query, "Host: "); 
    strcat(query, server);
    strcat(query, "\r\n\r\n");
    if((sendstring(sockfd, query)) == -1)
       err_quit("sendstring");

    if((status = recvstring(sockfd, recvbuffer, sizeof(recvbuffer)))==-1)
        err_quit("recvstring");

    printf("recved %d bytes\n", status);
    fp = fopen("SID_FILE", "w+");
    fwrite(recvbuffer, status, 1, fp);
    fclose(fp);
    close(sockfd);
    return 0;
}