/* See COPYING file for copyright and license details. */
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
#ifndef WINVER
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#else
#include <winsock2.h>
#include <ws2tcpip.h>
#endif

/* plundered from suckless' sic */
int dial(char *host, char *port) {
	static struct addrinfo hints;
	int srv;
	struct addrinfo *res, *r;

	#ifdef WINVER
	WSADATA w;
	if(WSAStartup(MAKEWORD(2,2), &w)!= 0) {
		fprintf(stderr, "error: failed to start winsock\n");
		return -1;
	}
	#endif

	memset(&hints, 0, sizeof hints);
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	if(getaddrinfo(host, port, &hints, &res) != 0) {
		fprintf(stderr, "error: cannot resolve hostname %s\n", host);
		return -1;
	}
	for(r = res; r; r = r->ai_next) {
		if((srv = socket(r->ai_family, r->ai_socktype, r->ai_protocol)) == -1)
			continue;
		if(connect(srv, r->ai_addr, r->ai_addrlen) == 0)
			break;
		close(srv);
	}
	freeaddrinfo(res);
	if(!r) {
		fprintf(stderr, "error: cannot connect to host %s\n", host);
		return -1;
	}
	return srv;
}

int get(char *host, char *path, char *sendcookie, char *savecookie, char **buf) {
	size_t l, res;
	int fd, i, p;
	char h[BUFSIZ] = "";
	char c[COOKIEMAX] = "";
	char t[BUFSIZ];
	char *t2;
	char m[256];

	if((fd = dial(host, "80")) == -1) return 0;

	if(sendcookie && sendcookie[0])
		snprintf(c, COOKIEMAX, "\r\nCookie: %s", sendcookie);
	snprintf(h, BUFSIZ, "GET %s HTTP/1.0\r\nUser-Agent: getxbook-"VERSION \
	                    " (not mozilla)\r\nHost: %s%s\r\n\r\n", path, host, c);
	if(!send(fd, h, strlen(h), 0)) return 0;

	*buf = NULL;
	l = 0;
	h[0] = 0;
	snprintf(m, 256, "Set-Cookie: %%%ds;", COOKIEMAX-1);

	while((res = recv(fd, t, BUFSIZ, 0)) > 0) {
		strncat(h, t, BUFSIZ - strlen(h) - 1);
		if((t2 = strstr(t, "\r\n\r\n")) != NULL && (t2 - t) < (signed)res) {
			/* end of header, save rest to buffer */
			t2+=4;
			l = res - (t2 - t);
			*buf = malloc(sizeof(char *) * l);
			memcpy(*buf, t2, l);
			break;
		}
	}

	if(sscanf(h, "HTTP/%d.%d %d", &i, &i, &p) == 3 && p != 200) {
		if(p == 403)
			fprintf(stderr, "403 forbidden: your IP address may be temporarily blocked\n");
		return 0;
	}
	t2 = h;
	if(savecookie != NULL) {
		while((t2 = strstr(t2, "Set-Cookie: ")) && sscanf(t2, m, c)) {
			strncat(savecookie, c, COOKIEMAX - strlen(savecookie) - 1);
			t2++;
		}
	}

	*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));
	for(; buf != NULL && (res = recv(fd, *buf+l, BUFSIZ, 0)) > 0; l+=res)
		*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));

	return l;
}

int gettofile(char *host, char *url, char *sendcookie, char *savecookie, char *savepath) {
	char *buf = 0;
	FILE *f;
	size_t i, l;

	if(!(l = get(host, url, sendcookie, savecookie, &buf))) {
		fprintf(stderr, "Could not download %s\n", url);
		return 1;
	}
	if((f = fopen(savepath, "wb")) == NULL) {
		fprintf(stderr, "Could not create file %s\n", savepath);
		free(buf); return 1;
	}

	for(i=0; i < l; i+=BUFSIZ)
		if(!fwrite(buf+i, l-i > BUFSIZ ? BUFSIZ : l-i, 1, f)) {
			fprintf(stderr, "Error writing file %s\n", savepath);
			free(buf); fclose(f); return 1;
		}

	free(buf);
	fclose(f);

	return 0;
}

int post(char *host, char *path, char *data, char **buf) {
	size_t l, res;
	int fd, i, p;
	char h[BUFSIZ] = "";
	char t[BUFSIZ];
	char *t2;

	if((fd = dial(host, "80")) == -1) return 0;

	snprintf(h, BUFSIZ, "POST %s HTTP/1.0\r\nUser-Agent: getxbook-"VERSION \
	                    " (not mozilla)\r\nContent-Length: %d\r\n" \
	                    "Content-Type: application/x-www-form-urlencoded\r\n" \
	                    "Host: %s\r\n\r\n%s\r\n",
	                    path, (int)strlen(data), host, data);
	if(!send(fd, h, strlen(h), 0)) return 0;

	*buf = NULL;
	l = 0;
	while((res = recv(fd, t, BUFSIZ, 0)) > 0) {
		if(sscanf(t, "HTTP/%d.%d %d", &i, &i, &p) == 3 && p != 200)
			return 0;
		t2 = t;
		if((t2 = strstr(t, "\r\n\r\n")) != NULL && (t2 - t) < (signed)res) {
			t2+=4;
			l = res - (t2 - t);
			*buf = malloc(sizeof(char *) * l);
			memcpy(*buf, t2, l);
			break;
		}
	}

	*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));
	for(; (res = recv(fd, *buf+l, BUFSIZ, 0)) > 0; l+=res)
		*buf = realloc(*buf, sizeof(char *) * (l+BUFSIZ));

	return l;
}

int renameifjpg(char *path) {
	FILE *f;
	char *newpath, *c;

	if((f = fopen(path, "rb")) == NULL)
		return 1;

	if(fgetc(f) == 255) {
		if((newpath = malloc(strlen(path) + 1)) == NULL)
			return 1;
		strncpy(newpath, path, strlen(path));
		c = strrchr(newpath, '.');
		strncpy(c+1, "jpg\0", 4);

		if(rename(path, newpath))
			return 1;
		free(newpath);
	}

	fclose(f);

	return 0;
}